1 #ifndef ALICONVEVENTCUTS_H
2 #define ALICONVEVENTCUTS_H
4 // Class handling all kinds of selection cuts for Gamma Conversion analysis
5 // Authors: Svein Lindal, Daniel Lohner *
7 #include "AliAODTrack.h"
8 #include "AliESDtrack.h"
10 #include "AliMCEvent.h"
11 #include "AliVTrack.h"
13 #include "AliAnalysisCuts.h"
16 #include "AliAnalysisUtils.h"
17 #include "AliAnalysisManager.h"
25 class AliAnalysisCuts;
28 class AliAnalysisManager;
29 class AliAODMCParticle;
33 class AliConvEventCuts : public AliAnalysisCuts {
40 kSelectSpecialTriggerAlias,
41 kSelectSubTriggerClass,
47 Bool_t SetCutIds(TString cutString);
49 Bool_t SetCut(cutIds cutID, Int_t cut);
50 Bool_t UpdateCutString();
51 static const char * fgkCutNames[kNCuts];
53 Bool_t InitializeCutsFromCutString(const TString analysisCutSelection);
54 void SelectCollisionCandidates(UInt_t offlineTriggerMask = AliVEvent::kAny) {
55 fOfflineTriggerMask = offlineTriggerMask;
56 fTriggerSelectedManually = kTRUE;
58 void SelectSpecialTrigger(UInt_t offlineTriggerMask = AliVEvent::kAny, TString TriggerClassName = "AliVEvent::kAny" ) {
59 fOfflineTriggerMask = offlineTriggerMask;
60 fSpecialTriggerName = TriggerClassName;
61 cout << fSpecialTriggerName.Data() << endl;
65 void SetAcceptedHeader(TList *HeaderList){fHeaderList = HeaderList;}
66 TString *GetFoundHeader(){return fGeneratorNames;}
68 Int_t GetEventQuality(){return fEventQuality;}
69 Bool_t GetIsFromPileup(){return fRemovePileUp;}
71 AliConvEventCuts(const char *name="EventCuts", const char * title="Event Cuts");
72 AliConvEventCuts(const AliConvEventCuts&);
73 AliConvEventCuts& operator=(const AliConvEventCuts&);
75 virtual ~AliConvEventCuts(); //virtual destructor
77 // static AliConvEventCuts * GetStandardCuts2010PbPb();
78 // static AliConvEventCuts * GetStandardCuts2010pp();
80 virtual Bool_t IsSelected(TObject* /*obj*/){return kTRUE;}
81 virtual Bool_t IsSelected(TList* /*list*/) {return kTRUE;}
83 TString GetCutNumber();
85 void GetCentralityRange(Double_t range[2]){range[0]=10*fCentralityMin;range[1]=10*fCentralityMax;};
88 Bool_t EventIsSelected(AliVEvent *fInputEvent, AliVEvent *fMCEvent);
89 Int_t IsEventAcceptedByCut(AliConvEventCuts *ReaderCuts, AliVEvent *InputEvent, AliMCEvent *MCEvent, Int_t isHeavyIon);
92 void PrintCutsWithValues();
94 void InitCutHistograms(TString name="",Bool_t preCut = kTRUE);
95 void SetFillCutHistograms(TString name="",Bool_t preCut = kTRUE){if(!fHistograms){InitCutHistograms(name,preCut);};}
96 TList *GetCutHistograms(){return fHistograms;}
99 Int_t IsParticleFromBGEvent(Int_t index, AliStack *MCStack, AliVEvent *InputEvent = 0x0);
100 void GetNotRejectedParticles(Int_t rejection, TList *HeaderList, AliVEvent *MCEvent);
101 void SetUseReweightingWithHistogramFromFile( Bool_t pi0reweight=kTRUE, Bool_t etareweight=kFALSE, Bool_t k0sreweight=kFALSE, TString path="$ALICE_ROOT/PWGGA/GammaConv/MCSpectraInput.root",
102 TString histoNamePi0 = "", TString histoNameEta = "", TString histoNameK0s = "",
103 TString fitNamePi0 = "", TString fitNameEta = "", TString fitNameK0s ="" ) {
104 AliInfo(Form("enabled reweighting for: pi0 : %i, eta: %i, K0s: %i",pi0reweight, etareweight, k0sreweight));
105 fDoReweightHistoMCPi0 = pi0reweight;
106 fDoReweightHistoMCEta = etareweight;
107 fDoReweightHistoMCK0s = k0sreweight;
108 fPathTrFReweighting=path;
109 fNameHistoReweightingPi0 =histoNamePi0;
110 fNameHistoReweightingEta =histoNameEta;
111 fNameHistoReweightingK0s =histoNameK0s;
112 fNameFitDataPi0 =fitNamePi0;
113 fNameFitDataEta =fitNameEta;
114 fNameFitDataK0s =fitNameK0s;
116 void LoadReweightingHistosMCFromFile ();
117 void SetAddedSignalPDGCode(Int_t addedSignalPDGcode) {fAddedSignalPDGCode = addedSignalPDGcode;}
120 Bool_t IsCentralitySelected(AliVEvent *fInputEvent, AliVEvent *fMCEvent = NULL);
121 Double_t GetCentrality(AliVEvent *event);
122 Int_t GetNumberOfContributorsVtx(AliVEvent *event);
123 Bool_t VertexZCut(AliVEvent *fInputEvent);
124 Bool_t IsTriggerSelected(AliVEvent *fInputEvent);
125 Bool_t HasV0AND(){return fHasV0AND;}
126 Bool_t IsSDDFired(){return fIsSDDFired;}
127 Int_t IsSpecialTrigger(){return fSpecialTrigger;}
128 TString GetSpecialTriggerName(){return fSpecialTriggerName;}
131 Int_t IsHeavyIon(){return fIsHeavyIon;}
132 Float_t GetWeightForMeson(TString period, Int_t index, AliStack *MCStack, AliVEvent *InputEvent = 0x0);
133 void SetPreSelectionCutFlag(Bool_t preSelFlag){fPreSelCut = preSelFlag;}
136 Int_t GetMultiplicityMethod(){return fMultiplicityMethod;}
137 Int_t GetSignalRejection(){return fRejectExtraSignals;}
138 Int_t GetNAcceptedHeaders(){return fnHeaders; }
139 TString * GetAcceptedHeaderNames(){return fGeneratorNames;}
140 Int_t * GetAcceptedHeaderStart(){return fNotRejectedStart;}
141 Int_t * GetAcceptedHeaderEnd(){return fNotRejectedEnd;}
142 TList* GetAcceptedHeader(){return fHeaderList;}
146 void SetEtaShift(Double_t etaShift) {
147 fEtaShift = etaShift;
149 void SetEtaShift(TString pPbOrPbp) {
150 Double_t etaShift = 0.0;
151 if(!pPbOrPbp.CompareTo("pPb")) etaShift = -0.465;
152 else if(!pPbOrPbp.CompareTo("Pbp")) etaShift = 0.465;
154 fEtaShift = etaShift;
156 Double_t GetEtaShift() {return fEtaShift;}
157 Bool_t GetDoEtaShift(){return fDoEtaShift;}
158 void DoEtaShift(Bool_t doEtaShift){fDoEtaShift = doEtaShift;}
159 void GetCorrectEtaShiftFromPeriod(TString periodName);
161 Bool_t SetIsHeavyIon(Int_t isHeavyIon);
162 Bool_t SetCentralityMax(Int_t centralityBin);
163 Bool_t SetCentralityMin(Int_t centralityBin);
164 Bool_t SetRemovePileUp(Int_t removePileUp);
165 Bool_t SetMultiplicityMethod(Int_t multiplicityMethod);
166 Bool_t SetSelectSpecialTrigger(Int_t selectSpecialTrigger);
167 Bool_t SetSelectSubTriggerClass (Int_t selectSpecialSubTriggerClass);
168 Bool_t SetRejectExtraSignalsCut(Int_t extraSignal);
169 void SetV0ReaderName(TString name) {fV0ReaderName = name;}
175 Int_t fEventQuality; // EventQuality
177 Int_t fIsHeavyIon; // flag for heavy ion
178 Int_t fDetectorCentrality; // centrality detecotor V0M or CL1
179 Int_t fModCentralityClass; // allows to select smaller centrality classes
180 Double_t fMaxVertexZ; // max z offset of vertex
181 Int_t fCentralityMin; // centrality selection lower bin value
182 Int_t fCentralityMax; // centrality selection upper bin value
183 Int_t fMultiplicityMethod; // selected multiplicity method
184 Int_t fSpecialTrigger; // flag
185 Int_t fSpecialSubTrigger; // flag
186 Bool_t fRemovePileUp; // flag
187 Int_t fRejectExtraSignals; //
188 UInt_t fOfflineTriggerMask; // Task processes collision candidates only
189 Bool_t fHasV0AND; // V0AND Offline Trigger
190 Bool_t fIsSDDFired; // SDD FIRED to select with SDD events
192 Int_t fnHeaders; // Number of Headers
193 Int_t *fNotRejectedStart; //[fnHeaders]
194 Int_t *fNotRejectedEnd; //[fnHeaders]
195 TString *fGeneratorNames; //[fnHeaders]
196 TObjString *fCutString; // cut number used for analysis
197 AliAnalysisUtils *fUtils;
199 Bool_t fDoEtaShift; // Flag for Etashift
200 Bool_t fDoReweightHistoMCPi0; // Flag for reweighting Pi0 input with histogram
201 Bool_t fDoReweightHistoMCEta; // Flag for reweighting Eta input with histogram
202 Bool_t fDoReweightHistoMCK0s; // Flag for reweighting K0s input with histogram
203 TString fPathTrFReweighting; // Path for file used in reweighting
204 TString fNameHistoReweightingPi0; // Histogram name for reweighting Pi0
205 TString fNameHistoReweightingEta; // Histogram name for reweighting Eta
206 TString fNameHistoReweightingK0s; // Histogram name for reweighting K0s
207 TString fNameFitDataPi0; // Fit name for fit to spectrum of pi0s in Data
208 TString fNameFitDataEta; // Fit name for fit to spectrum of etas in Data
209 TString fNameFitDataK0s; // Fit name for fit to spectrum of k0s in Data
211 TH1F *fHistoEventCuts; // bookkeeping for event selection cuts
212 TH1F *hCentrality; // centrality distribution for selected events
213 TH2F *hCentralityVsNumberOfPrimaryTracks; // centrality distribution for selected events
214 TH1F *hVertexZ; // vertex z distribution for selected events
215 TH1F *hTriggerClass; // fired offline trigger class
216 TH1F *hTriggerClassSelected; // selected fired offline trigger class
217 TH1D *hReweightMCHistPi0; // histogram input for reweighting Pi0
218 TH1D *hReweightMCHistEta; // histogram input for reweighting Eta
219 TH1D *hReweightMCHistK0s; // histogram input for reweighting K0s
220 TF1 *fFitDataPi0; // fit to pi0 spectrum in Data
221 TF1 *fFitDataEta; // fit to eta spectrum in Data
222 TF1 *fFitDataK0s; // fit to K0s spectrum in Data
223 Int_t fAddedSignalPDGCode;
224 Bool_t fPreSelCut; // Flag for preselection cut used in V0Reader
225 Bool_t fTriggerSelectedManually; // Flag for manual trigger selection
226 TString fSpecialTriggerName; // Name of the Special Triggers
227 TString fSpecialSubTriggerName; // Name of the Special Triggers
228 Int_t fNSpecialSubTriggerOptions;
229 TString fV0ReaderName; // Name of V0Reader
234 ClassDef(AliConvEventCuts,2)