]> git.uio.no Git - u/mrichter/AliRoot.git/blame - PWGJE/EMCALJetTasks/UserTasks/AliAnalysisTaskChargedJetsPA.h
-fix warnings (Ionut)
[u/mrichter/AliRoot.git] / PWGJE / EMCALJetTasks / UserTasks / AliAnalysisTaskChargedJetsPA.h
CommitLineData
8628b70c 1#ifndef ALIANALYSISTASKCHARGEDJETSPA_H
2#define ALIANALYSISTASKCHARGEDJETSPA_H
3
12eb4ac1 4//#define DEBUGMODE
8628b70c 5
6class TH1F;
7class TH2F;
8class TList;
8628b70c 9class TClonesArray;
10class TString;
11class AliEmcalJet;
12class AliRhoParameter;
13class AliVParticle;
14class AliLog;
3fa9cde0 15class AliAnalysisUtils;
b95d0245 16class TRandom3;
8628b70c 17
8628b70c 18class AliAnalysisTaskChargedJetsPA : public AliAnalysisTaskSE {
19 public:
12eb4ac1 20 // ######### CONTRUCTORS/DESTRUCTORS AND STD FUNCTIONS
9f1e4f53 21 AliAnalysisTaskChargedJetsPA() : AliAnalysisTaskSE(), fOutputList(0), fAnalyzeJets(1), fAnalyzeQA(1), fAnalyzeBackground(1), fAnalyzeDeprecatedBackgrounds(1), fAnalyzePythia(0), fHasTracks(0), fHasJets(0), fHasBackgroundJets(0), fIsKinematics(0), fUseVertexCut(1), fUsePileUpCut(1), fSetCentralityToOne(0), fPartialAnalysisNParts(1), fPartialAnalysisIndex(0), fJetArray(0), fTrackArray(0), fBackgroundJetArray(0), fJetArrayName(0), fTrackArrayName(0), fBackgroundJetArrayName(0), fNumPtHardBins(11), fUsePtHardBin(-1), fRhoTaskName(), fNcoll(6.88348), fRandConeRadius(0.4), fSignalJetRadius(0.4), fBackgroundJetRadius(0.4), fTRBackgroundConeRadius(0.6), fNumberRandCones(8), fNumberExcludedJets(-1), fDijetMaxAngleDeviation(10.0), fPhysicalJetRadius(0.6), fSignalJetEtaWindow(0.5), fBackgroundJetEtaWindow(0.5), fTrackEtaWindow(0.9), fMinTrackPt(0.150), fMinJetPt(1.0), fMinJetArea(0.5), fMinBackgroundJetPt(0.0), fMinDijetLeadingPt(10.0), fNumberOfCentralityBins(100), fCentralityType("V0A"), fFirstLeadingJet(0), fSecondLeadingJet(0), fNumberSignalJets(0), fCrossSection(0.0), fTrials(0.0), fRandom(0), fHelperClass(0), fInitialized(0), fTaskInstanceCounter(0), fHistList(0), fHistCount(0), fIsDEBUG(0), fEventCounter(0)
b95d0245 22 {
23 for(Int_t i=0;i<1024;i++)
24 fSignalJets[i] = NULL;
25 }
0983dd5b 26
12eb4ac1 27 AliAnalysisTaskChargedJetsPA(const char *name, const char* trackArrayName, const char* jetArrayName, const char* backgroundJetArrayName);
8628b70c 28 virtual ~AliAnalysisTaskChargedJetsPA();
29 virtual void UserCreateOutputObjects();
30 virtual void UserExec(Option_t *option);
392aa819 31 virtual Bool_t UserNotify();
8628b70c 32 virtual void Terminate(Option_t *);
efb9b161 33
12eb4ac1 34 // ######### SETTERS/GETTERS
35 void SetAnalyzeJets(Bool_t val) {fAnalyzeJets = val;}
b95d0245 36 void SetAnalyzeQA(Bool_t val) {fAnalyzeQA = val;}
12eb4ac1 37 void SetAnalyzeBackground(Bool_t val) {fAnalyzeBackground = val;}
b95d0245 38 void SetAnalyzeDeprecatedBackgrounds(Bool_t val) {fAnalyzeDeprecatedBackgrounds = val;}
12eb4ac1 39 void SetAnalyzePythia(Bool_t val) {fAnalyzePythia = val;}
c2b0976c 40 void SetAnalyzePartialEvents(Int_t nParts, Int_t index) {fPartialAnalysisNParts = nParts; fPartialAnalysisIndex = index;}
392aa819 41 void SetUseVertexCut (Bool_t val) {fUseVertexCut = val;}
42 void SetUsePileUpCut (Bool_t val) {fUsePileUpCut = val;}
9f1e4f53 43 void SetCentralityToOne (Bool_t val) {fSetCentralityToOne = val;}
12eb4ac1 44
45 void SetTrackMinPt(Double_t minPt) {fMinJetPt = minPt;}
46 void SetSignalJetMinPt(Double_t minPt) {fMinJetPt = minPt;}
47 void SetSignalJetMinArea(Double_t minArea) {fMinJetArea = minArea;}
48 void SetBackgroundJetMinPt(Double_t minPt) {fMinBackgroundJetPt = minPt;}
49 void SetDijetLeadingMinPt(Double_t minPt) {fMinDijetLeadingPt = minPt;}
50 void SetNumberOfPtHardBins(Int_t count) {fNumPtHardBins = count;}
b95d0245 51 void SetUsePtHardBin(Int_t number) {fUsePtHardBin = number;}
12eb4ac1 52 void SetNumberOfRandConesPerEvent(Int_t count) {fNumberRandCones = count;}
53 void SetRandConeRadius(Double_t radius) {fRandConeRadius = radius;}
54 void SetSignalJetRadius(Double_t radius) {fSignalJetRadius = radius;}
55 void SetBackgroundJetRadius(Double_t radius) {fBackgroundJetRadius = radius;}
56 void SetTRBackgroundConeRadius(Double_t radius) {fTRBackgroundConeRadius = radius;}
57 void SetCentralityType(const char* type) {fCentralityType = type;}
06b41974 58 void SetExternalRhoTaskName(const char* name) {fRhoTaskName = name;}
12eb4ac1 59 void SetDijetMaxAngleDeviation(Double_t degrees) {fDijetMaxAngleDeviation = degrees/360.0 * TMath::TwoPi();} // degrees are more comfortable
06b41974 60 void SetAcceptanceWindows(Double_t trackEta, Double_t signalJetRadius, Double_t bgrdJetRadius){fTrackEtaWindow = trackEta; fSignalJetRadius = signalJetRadius; fBackgroundJetRadius = bgrdJetRadius; fSignalJetEtaWindow = fTrackEtaWindow-fSignalJetRadius; fBackgroundJetEtaWindow = fTrackEtaWindow-fBackgroundJetRadius;}
12eb4ac1 61 Int_t GetInstanceCounter() {return fTaskInstanceCounter;}
8628b70c 62
63 private:
12eb4ac1 64
65 // ######### MAIN CALCULATION FUNCTIONS
66 void GetSignalJets();
67 Int_t GetLeadingJets(TClonesArray* jetArray, Int_t* jetIDArray, Bool_t isSignalJets);
e8df4140 68 Double_t GetCorrectedJetPt(AliEmcalJet* jet, Double_t background);
c2b0976c 69 Double_t GetDeltaPt(Double_t rho, Double_t leadingJetExclusionProbability = 0);
0cb49d99 70
71 void GetKTBackgroundDensityAll(Int_t numberExcludeLeadingJets, Double_t& rhoPbPb, Double_t& rhoPbPbWithGhosts, Double_t& rhoCMS, Double_t& rhoImprovedCMS, Double_t& rhoMean, Double_t& rhoTrackLike);
72 void GetKTBackgroundDensity(Int_t numberExcludeLeadingJets, Double_t& rhoImprovedCMS);
12eb4ac1 73
b95d0245 74 void GetTRBackgroundDensity(Int_t numberExcludeLeadingJets, Double_t& rhoNoExclusion, Double_t& rhoConeExclusion02, Double_t& rhoConeExclusion04, Double_t& rhoConeExclusion06, Double_t& rhoConeExclusion08, Double_t& rhoExactExclusion);
12eb4ac1 75 void GetTRBackgroundDensity(Int_t numberExcludeLeadingJets, Double_t& rhoMean, Double_t& area, AliEmcalJet* excludeJet1, AliEmcalJet* excludeJet2, Bool_t doSearchPerpendicular);
76 Double_t GetConePt(Double_t eta, Double_t phi, Double_t radius);
77 Double_t GetPtHard();
392aa819 78 Double_t GetPythiaTrials();
12eb4ac1 79 Int_t GetPtHardBin();
06b41974 80 Double_t GetExternalRho();
81
12eb4ac1 82 void GetPerpendicularCone(Double_t vecPhi, Double_t vecTheta, Double_t& conePt);
83
84 // ######### CHECK FUNCTIONS
85 Bool_t IsTrackInAcceptance(AliVParticle* track);
86 Bool_t IsTrackInCone(AliVTrack* track, Double_t eta, Double_t phi, Double_t radius);
b95d0245 87 Bool_t IsTrackInJet(AliEmcalJet* jet, Int_t trackIndex);
88 Bool_t IsJetOverlapping(AliEmcalJet* jet1, AliEmcalJet* jet2);
12eb4ac1 89
0cb49d99 90 Bool_t IsEventInAcceptance(AliVEvent* event);
12eb4ac1 91 Bool_t IsBackgroundJetInAcceptance(AliEmcalJet* jet);
92 Bool_t IsSignalJetInAcceptance(AliEmcalJet* jet);
93 Bool_t IsDijet(AliEmcalJet* jet1, AliEmcalJet* jet2);
8628b70c 94
12eb4ac1 95 // ######### HELPER FUNCTIONS
96 Double_t EtaToTheta(Double_t arg);
97 Double_t ThetaToEta(Double_t arg);
98 Double_t GetDeltaPhi(Double_t phi1, Double_t phi2);
99 Double_t MCGetOverlapCircleRectancle(Double_t cPosX, Double_t cPosY, Double_t cRadius, Double_t rPosXmin, Double_t rPosXmax, Double_t rPosYmin, Double_t rPosYmax);
b95d0245 100 Double_t MCGetOverlapMultipleCirclesRectancle(Int_t numCircles, std::vector<Double_t> cPosX, std::vector<Double_t> cPosY, Double_t cRadius, Double_t rPosXmin, Double_t rPosXmax, Double_t rPosYmin, Double_t rPosYmax);
101
12eb4ac1 102
103 // ######### HISTOGRAM FUNCTIONS
104 void FillHistogram(const char * key, Double_t x);
105 void FillHistogram(const char * key, Double_t x, Double_t y);
106 void FillHistogram(const char * key, Double_t x, Double_t y, Double_t add);
8628b70c 107 const char* GetHistoName(const char* name)
108 {
06b41974 109 if (fIsKinematics)
7e413f22 110 return Form("%s_MC", name);
111 return Form("%s", name);
8628b70c 112 }
8628b70c 113 template <class T> T* AddHistogram1D(const char* name = "CustomHistogram", const char* title = "NO_TITLE", const char* options = "", Int_t xBins = 100, Double_t xMin = 0.0, Double_t xMax = 20.0, const char* xTitle = "x axis", const char* yTitle = "y axis");
8628b70c 114 template <class T> T* AddHistogram2D(const char* name = "CustomHistogram", const char* title = "NO_TITLE", const char* options = "", Int_t xBins = 100, Double_t xMin = 0.0, Double_t xMax = 20.0, Int_t yBins = 100, Double_t yMin = 0.0, Double_t yMax = 20.0, const char* xTitle = "x axis", const char* yTitle = "y axis", const char* zTitle = "z axis");
115
12eb4ac1 116 // ######### STANDARD FUNCTIONS
8628b70c 117 void Calculate(AliVEvent* event);
118 void ExecOnce();
119 void Init ();
120
121 TList* fOutputList; //! Output list
12eb4ac1 122 // ########## USAGE TRIGGERS
8628b70c 123 Bool_t fAnalyzeJets; // trigger if jets should be processed
b95d0245 124 Bool_t fAnalyzeQA; // trigger if QA should be done
8628b70c 125 Bool_t fAnalyzeBackground; // trigger if background should be processed
b95d0245 126 Bool_t fAnalyzeDeprecatedBackgrounds; // trigger if old background estimates should be processed
8628b70c 127 Bool_t fAnalyzePythia; // trigger if pythia properties should be processed
128 Bool_t fHasTracks; // trigger if tracks are actually valid
8628b70c 129 Bool_t fHasJets; // trigger if jets are actually valid
130 Bool_t fHasBackgroundJets; // trigger if background is actually valid
06b41974 131 Bool_t fIsKinematics; // trigger if data is kinematics only (for naming reasons)
392aa819 132 Bool_t fUseVertexCut; // trigger if vertex cut should be done
133 Bool_t fUsePileUpCut; // trigger if pileup cut should be done
9f1e4f53 134 Bool_t fSetCentralityToOne; // trigger if centrality val. should be set to one for every event (failsafe)
c2b0976c 135 Int_t fPartialAnalysisNParts; // take only every Nth event
136 Int_t fPartialAnalysisIndex; // using e.g. only every 5th event, this specifies which one
137
12eb4ac1 138
8628b70c 139 // ########## SOURCE INFORMATION
140 TClonesArray* fJetArray; //! object containing the jets
141 TClonesArray* fTrackArray; //! object containing the tracks
8628b70c 142 TClonesArray* fBackgroundJetArray; //! object containing background jets
143 TString* fJetArrayName; // name of object containing the jets
144 TString* fTrackArrayName; // name of object containing the tracks
8628b70c 145 TString* fBackgroundJetArrayName;// name of object containing event wise bckgrds
146 Int_t fNumPtHardBins; // Number of used pt hard bins
b95d0245 147 Int_t fUsePtHardBin; // That pt hard bin will be analyzed when not -1
06b41974 148 TString fRhoTaskName; // name of rho task for this analysis
8628b70c 149 // ########## JET/DIJET/RC PROPERTIES
06b41974 150 Double_t fNcoll; // Variable for Ncoll
8628b70c 151 Double_t fRandConeRadius; // Radius for the random cones
152 Double_t fSignalJetRadius; // Radius for the signal jets
153 Double_t fBackgroundJetRadius; // Radius for the KT background jets
12eb4ac1 154 Double_t fTRBackgroundConeRadius;// Radius for the jets excluded in track background
8628b70c 155 Int_t fNumberRandCones; // Number of random cones to be put into one event
156 Int_t fNumberExcludedJets; // Number of jets to be excluded from backgrounds
157 Double_t fDijetMaxAngleDeviation;// Max angle deviation from pi between two jets to be accept. as dijet
b95d0245 158 Double_t fPhysicalJetRadius; // Rough size considered for the real jet
8628b70c 159 // ########## CUTS
160 Double_t fSignalJetEtaWindow; // +- window in eta for signal jets
161 Double_t fBackgroundJetEtaWindow;// +- window in eta for background jets
162 Double_t fTrackEtaWindow; // +- window in eta for tracks
8628b70c 163 Double_t fMinTrackPt; // Min track pt to be accepted
8628b70c 164 Double_t fMinJetPt; // Min jet pt to be accepted
165 Double_t fMinJetArea; // Min jet area to be accepted
166 Double_t fMinBackgroundJetPt; // Min jet pt to be accepted as background jet
167 Double_t fMinDijetLeadingPt; // Min jet pt to be accepted as constituent of dijet
b95d0245 168 Int_t fNumberOfCentralityBins;// Number of centrality bins used for histograms
12eb4ac1 169 TString fCentralityType; // Used centrality estimate (V0A, V0C, V0M, ...)
8628b70c 170
171 // ########## EVENT PROPERTIES
172 AliEmcalJet* fFirstLeadingJet; //! leading jet in event
173 AliEmcalJet* fSecondLeadingJet; //! next to leading jet in event
174 Int_t fNumberSignalJets; //! Number of signal jets in event
175 AliEmcalJet* fSignalJets[1024]; //! memory for signal jet pointers
176 Double_t fCrossSection; //! value is filled, if pythia header is accessible
177 Double_t fTrials; //! value is filled, if pythia header is accessible
178
179 // ########## GENERAL VARS
180 TRandom3* fRandom; //! A random number
3fa9cde0 181 AliAnalysisUtils* fHelperClass; //! Vertex selection helper
8628b70c 182 Bool_t fInitialized; //! trigger if tracks/jets are loaded
183 Int_t fTaskInstanceCounter; // for naming reasons
184 TList* fHistList; // Histogram list
185 Int_t fHistCount; // Histogram count
b95d0245 186 Bool_t fIsDEBUG; // Debug trigger
c2b0976c 187 ULong_t fEventCounter; // Internal event counter
12eb4ac1 188 AliAnalysisTaskChargedJetsPA(const AliAnalysisTaskChargedJetsPA&);
189 AliAnalysisTaskChargedJetsPA& operator=(const AliAnalysisTaskChargedJetsPA&);
8628b70c 190
12eb4ac1 191 ClassDef(AliAnalysisTaskChargedJetsPA, 2); // Charged jet analysis for pA
8628b70c 192
193};
194#endif