]>
Commit | Line | Data |
---|---|---|
fdceab34 | 1 | /************************************************************************** |
2 | * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. * | |
3 | * * | |
4 | * Author: The ALICE Off-line Project. * | |
5 | * Contributors are mentioned in the code where appropriate. * | |
6 | * * | |
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 | **************************************************************************/ | |
15 | ||
16 | //----------------------------------------------------------------------- | |
17 | // Author : Marta Verweij - UU | |
18 | //----------------------------------------------------------------------- | |
19 | ||
20 | #ifndef ALIPWG4HIGHPTSPECTRA_H | |
21 | #define ALIPWG4HIGHPTSPECTRA_H | |
22 | ||
23 | #include "AliAnalysisTask.h" | |
24 | #include "AliCFManager.h" | |
25 | ||
26 | class TH1I; | |
b5cc0c6d | 27 | class TH1F; |
fdceab34 | 28 | class TH1D; |
a337a5a9 | 29 | class TH2F; |
b4691ee7 | 30 | class TProfile; |
31 | class TFile; | |
32 | class TList; | |
33 | ||
67ebd013 | 34 | //class AliCFManager; |
fdceab34 | 35 | class AliESDtrackCuts; |
36 | class AliESDEvent; | |
34fc6450 | 37 | class AliESDVertex; |
b4691ee7 | 38 | class AliMCEvent; |
cce400da | 39 | class AliStack; |
b4691ee7 | 40 | class AliGenPythiaEventHeader; |
03372fd1 | 41 | class AliGenHijingEventHeader; |
fdceab34 | 42 | |
43 | class AliPWG4HighPtSpectra : public AliAnalysisTask { | |
b5cc0c6d | 44 | public: |
fdceab34 | 45 | |
46 | enum { | |
ec555a3c | 47 | kStepReconstructed = 0, |
2b553e6f | 48 | kStepSecondaries = 1, |
49 | kStepReconstructedMC = 2, | |
50 | kStepMCAcceptance = 3 | |
fdceab34 | 51 | }; |
52 | ||
53 | AliPWG4HighPtSpectra(); | |
54 | AliPWG4HighPtSpectra(const Char_t* name); | |
67ebd013 | 55 | // AliPWG4HighPtSpectra& operator= (const AliPWG4HighPtSpectra& c); |
56 | // AliPWG4HighPtSpectra(const AliPWG4HighPtSpectra& c); | |
57 | ~AliPWG4HighPtSpectra() {;}; | |
fdceab34 | 58 | |
59 | // ANALYSIS FRAMEWORK STUFF to loop on data and fill output objects | |
e5abcde9 | 60 | virtual void LocalInit(); |
fdceab34 | 61 | virtual void ConnectInputData(Option_t *); |
62 | virtual void CreateOutputObjects(); | |
63 | virtual void Exec(Option_t *option); | |
64 | virtual void Terminate(Option_t *); | |
b4691ee7 | 65 | virtual Bool_t Notify(); //Copied from AliAnalysisTaskJetSpectrum2 |
fdceab34 | 66 | |
2b553e6f | 67 | Bool_t IsPbPb() {return fIsPbPb;} //is PbPb data? |
cce400da | 68 | Bool_t SelectEvent(); //decides if event is used for analysis |
2b553e6f | 69 | Int_t CalculateCentrality(AliESDEvent *esd); |
cce400da | 70 | |
2b553e6f | 71 | //Setters |
72 | void SetIsPbPb(Bool_t cs) {fIsPbPb = cs;} | |
73 | void SetCentralityClass(int cent) {fCentClass=cent;} | |
5d87a047 | 74 | void SetTriggerMask(UInt_t t) {fTriggerMask=t;} |
2b553e6f | 75 | |
fdceab34 | 76 | // CORRECTION FRAMEWORK RELATED FUNCTIONS |
67ebd013 | 77 | void SetCFManagerPos(const AliCFManager* io1) {fCFManagerPos = io1;} // global correction manager |
78 | const AliCFManager * GetCFManagerPos() const {return fCFManagerPos;} // get corr manager | |
79 | void SetCFManagerNeg(const AliCFManager* io2) {fCFManagerNeg = io2;} // global correction manager | |
80 | const AliCFManager * GetCFManagerNeg() const {return fCFManagerNeg;} // get corr manager | |
2b553e6f | 81 | |
82 | //if fTrackType=0 (Global) | |
83 | //if fTrackType=1 (TPConly) | |
84 | //if fTrackType=2 (TPConly constrained) | |
34fc6450 | 85 | void SetTrackType(Int_t trackType) {fTrackType = trackType;} |
fdceab34 | 86 | //AliESDtrackCuts setters |
87 | void SetCuts(AliESDtrackCuts* trackCuts) {fTrackCuts = trackCuts;} | |
327d12da | 88 | void SetCutsReject(AliESDtrackCuts* trackCuts) {fTrackCutsReject = trackCuts;} |
03372fd1 | 89 | void SelectHIJINGOnly(Bool_t b) {fbSelectHIJING = b;} |
90 | ||
91 | Bool_t IsHIJINGParticle(Int_t label); | |
2b553e6f | 92 | |
93 | void SetSigmaConstrainedMax(Double_t sigma) {fSigmaConstrainedMax=sigma;} | |
fdceab34 | 94 | |
95 | // Data types | |
96 | Bool_t IsReadAODData() const {return fReadAODData;} | |
b5cc0c6d | 97 | void SetReadAODData(Bool_t flag=kTRUE) {fReadAODData=flag;} |
b4691ee7 | 98 | |
99 | static AliGenPythiaEventHeader* GetPythiaEventHeader(AliMCEvent *mcEvent); | |
03372fd1 | 100 | static AliGenHijingEventHeader* GetHijingEventHeader(AliMCEvent *mcEvent); |
101 | ||
b4691ee7 | 102 | static Bool_t PythiaInfoFromFile(const char* currFile,Float_t &fXsec,Float_t &fTrials);// get the cross section and the trails either from pyxsec.root or from pysec_hists.root |
03372fd1 | 103 | |
104 | ||
b5cc0c6d | 105 | |
fdceab34 | 106 | protected: |
67ebd013 | 107 | Bool_t fReadAODData ; // flag for AOD/ESD input files |
108 | const AliCFManager *fCFManagerPos ; // pointer to the CF manager for positive charged particles | |
109 | const AliCFManager *fCFManagerNeg ; // pointer to the CF manager for negative charged particles | |
110 | ||
cce400da | 111 | AliESDEvent *fESD; //! ESD object |
112 | AliMCEvent *fMC; //! MC event object | |
113 | AliStack *fStack; //! stack object | |
114 | ||
34fc6450 | 115 | const AliESDVertex *fVtx; //! vertex object |
116 | ||
5d87a047 | 117 | UInt_t fTriggerMask; // Trigger mask to select events |
118 | Bool_t fIsPbPb; // kTRUE if PbPb | |
119 | Int_t fCentClass; // Select only events from predefined centrality class | |
34fc6450 | 120 | |
121 | Int_t fTrackType; // Type of track to be used in analysis | |
1f329128 | 122 | //AliESDtrackCuts options. Must be setted in AddTaskPWG4HighPTSpectra.C. They correspond with different steps in container. |
123 | AliESDtrackCuts *fTrackCuts; // trackCuts applied to global tracks | |
327d12da | 124 | AliESDtrackCuts *fTrackCutsReject; // trackCuts to reject tracks (hybrid case) |
2b553e6f | 125 | |
03372fd1 | 126 | Bool_t fbSelectHIJING; //Select only particles from HIJING event |
127 | ||
2b553e6f | 128 | Double_t fSigmaConstrainedMax; // max sigma on constrained fit |
fdceab34 | 129 | |
130 | private: | |
1f329128 | 131 | AliPWG4HighPtSpectra(const AliPWG4HighPtSpectra&); |
132 | AliPWG4HighPtSpectra& operator=(const AliPWG4HighPtSpectra&); | |
fdceab34 | 133 | |
b4691ee7 | 134 | Float_t fAvgTrials; // Average number of trials |
135 | ||
fdceab34 | 136 | // Histograms |
137 | //Number of events | |
b4691ee7 | 138 | TList *fHistList; //! List of output histograms |
1f329128 | 139 | TH1F *fNEventAll; //! Event counter |
140 | TH1F *fNEventSel; //! Event counter: Selected events for analysis | |
2b553e6f | 141 | TH1F *fNEventReject; //! Book keeping of reason of rejecting events |
142 | ||
143 | TH1F *fh1Centrality; //! Centrality | |
b5cc0c6d | 144 | |
b4691ee7 | 145 | TProfile* fh1Xsec; //! pythia cross section and trials |
146 | TH1F* fh1Trials; //! trials which are added | |
147 | TH1F* fh1PtHard; //! pt hard of the event | |
148 | TH1F* fh1PtHardTrials; //! pt hard of the event | |
149 | ||
a337a5a9 | 150 | TH2F *fPtRelUncertainty1PtPrim; //! Pt vs relUncertainty1Pt for primary particles |
151 | TH2F *fPtRelUncertainty1PtSec; //! Pt vs relUncertainty1Pt for secondary particles | |
152 | ||
2b553e6f | 153 | ClassDef(AliPWG4HighPtSpectra,3); |
fdceab34 | 154 | }; |
155 | ||
156 | #endif |