1 /**************************************************************************
2 * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
4 * Author: The ALICE Off-line Project. *
5 * Contributors are mentioned in the code where appropriate. *
7 * Permission to use, copy, modify and distribute this software and its *
8 * documentation strictly for non-commercial purposes is hereby granted *
9 * without fee, provided that the above copyright notice appears in all *
10 * copies and that both the copyright notice and this permission notice *
11 * appear in the supporting documentation. The authors make no claims *
12 * about the suitability of this software for any purpose. It is *
13 * provided "as is" without express or implied warranty. *
14 **************************************************************************/
18 //_________________________________________________________________________
19 // A test analysis task to check the pt of tracks distribution in simulated data
22 //////////////////////////////////////////////////////////////////////////////
32 #include "AliAnalysisTaskPt.h"
36 //________________________________________________________________________
37 AliAnalysisTaskPt::AliAnalysisTaskPt(const char *name) :
38 AliAnalysisTask(name,""),
43 // Input slot #0 works with an Ntuple
44 DefineInput(0, TChain::Class());
45 // Output slot #0 writes into a TH1 container
46 DefineOutput(0, TObjArray::Class());
49 //________________________________________________________________________
50 void AliAnalysisTaskPt::ConnectInputData(Option_t *)
52 // Initialisation of branch container and histograms
54 AliInfo(Form("*** Initialization of %s", GetName())) ;
57 fChain = dynamic_cast<TChain *>(GetInputData(0)) ;
59 AliError(Form("Input 0 for %s not found\n", GetName()));
63 // One should first check if the branch address was taken by some other task
64 char ** address = (char **)GetBranchAddress(0, "ESD");
66 fESD = (AliESD*)(*address);
69 SetBranchAddress(0, "ESD", &fESD);
73 //________________________________________________________________________
74 void AliAnalysisTaskPt::CreateOutputObjects()
80 fhPt = new TH1F("fhPt","This is the Pt distribution",15,0.1,3.1);
81 fhPt->SetStats(kTRUE);
82 fhPt->GetXaxis()->SetTitle("P_{T} [GeV]");
83 fhPt->GetYaxis()->SetTitle("#frac{dN}{dP_{T}}");
84 fhPt->GetXaxis()->SetTitleColor(1);
85 fhPt->SetMarkerStyle(kFullCircle);
87 // create output container
89 fOutputContainer = new TObjArray(1) ;
90 fOutputContainer->SetName(GetName()) ;
92 fOutputContainer->AddAt(fhPt, 0) ;
95 //________________________________________________________________________
96 void AliAnalysisTaskPt::Exec(Option_t *)
98 // Processing of one event
100 Long64_t entry = fChain->GetReadEntry() ;
103 AliError("fESD is not connected to the input!") ;
107 if ( !((entry-1)%100) )
108 AliInfo(Form("%s ----> Processing event # %lld", (dynamic_cast<TChain *>(fChain))->GetFile()->GetName(), entry)) ;
110 //************************ Pt tracks *************************************
112 for(Int_t iTracks = 0; iTracks < fESD->GetNumberOfTracks(); iTracks++) {
113 AliESDtrack * ESDTrack = fESD->GetTrack(iTracks);
114 Double_t momentum[3];
115 ESDTrack->GetPxPyPz(momentum);
116 Double_t Pt = sqrt(pow(momentum[0],2) + pow(momentum[1],2));
120 PostData(0, fOutputContainer);
123 //________________________________________________________________________
124 void AliAnalysisTaskPt::Terminate(Option_t *)
126 // Processing when the event loop is ended
128 Bool_t problem=kFALSE;
130 AliInfo(Form(" *** %s Report:", GetName())) ;
132 TCanvas *c1 = new TCanvas("c1","Pt",10,10,310,310);
133 c1->SetFillColor(10);
134 c1->SetHighLightColor(10);
136 c1->cd(1)->SetLeftMargin(0.15);
137 c1->cd(1)->SetBottomMargin(0.15);
138 if (fhPt->GetMaximum() > 0 )
139 c1->cd(1)->SetLogy();
140 fOutputContainer = (TObjArray*)GetOutputData(0);
141 fhPt = (TH1F*)fOutputContainer->At(0);
142 if (fhPt) fhPt->DrawCopy("E");
144 c1->Print("TracksPt.eps");
147 sprintf(line, ".!tar -zcf %s.tar.gz *.eps", GetName()) ;
148 gROOT->ProcessLine(line);
149 sprintf(line, ".!rm -fR *.eps");
150 gROOT->ProcessLine(line);
152 AliInfo(Form("!!! All the eps files are in %s.tar.gz !!!", GetName())) ;
156 report="Problems found, please check!!!";
160 AliInfo(Form("*** %s Summary Report: %s \n",GetName(), report.Data())) ;