c7364436eed7eeb5b2993b933a4f9cdce3484130
[u/mrichter/AliRoot.git] / PWGGA / GammaConv / AliConversionMesonCuts.h
1 #ifndef ALICONVERSIONMESONCUTS_H
2 #define ALICONVERSIONMESONCUTS_H
3
4 // Class handling all kinds of selection cuts for Gamma Conversion analysis
5 // Authors: Svein Lindal, Daniel Lohner                                                                                         *
6
7 #include "AliAODpidUtil.h"
8 #include "AliConversionPhotonBase.h"
9 #include "AliAODConversionMother.h"
10 #include "AliAODTrack.h"
11 #include "AliESDtrack.h"
12 #include "AliVTrack.h"
13 #include "AliAODTrack.h"
14 #include "AliStack.h"
15 #include "AliAnalysisCuts.h"
16 #include "TH1F.h"
17
18 class AliESDEvent;
19 class AliAODEvent;
20 class AliConversionPhotonBase;
21 class AliKFVertex;
22 class TH1F;
23 class TH2F;
24 class AliPIDResponse;
25 class AliAnalysisCuts;
26 class iostream;
27 class TList;
28 class AliAnalysisManager;
29
30
31 using namespace std;
32
33 class AliConversionMesonCuts : public AliAnalysisCuts {
34         
35  public: 
36
37
38   enum cutIds {
39         kMesonKind,
40         kBackgroundScheme,            
41         kNumberOfBGEvents,           
42         kDegreesForRotationMethod,    
43         kRapidityMesonCut,    
44         kRCut,
45         kalphaMesonCut,               
46         kchi2MesonCut,
47         kElecShare,
48         kToCloseV0s,
49         kuseMCPSmearing,
50         kNCuts
51   };
52
53   Bool_t SetCutIds(TString cutString); 
54   Int_t fCuts[kNCuts];
55   Bool_t SetCut(cutIds cutID, Int_t cut);
56   Bool_t UpdateCutString();
57
58   static const char * fgkCutNames[kNCuts];
59
60   Bool_t InitializeCutsFromCutString(const TString analysisCutSelection);
61   void FillElectonLabelArray(AliAODConversionPhoton* photon, Int_t nV0);
62    
63   AliConversionMesonCuts(const char *name="MesonCuts", const char * title="Meson Cuts");
64   AliConversionMesonCuts(const AliConversionMesonCuts&);
65   AliConversionMesonCuts& operator=(const AliConversionMesonCuts&); // not implemented
66
67   virtual ~AliConversionMesonCuts();                            //virtual destructor
68
69   virtual Bool_t IsSelected(TObject* /*obj*/){return kTRUE;}
70   virtual Bool_t IsSelected(TList* /*list*/) {return kTRUE;}
71
72   TString GetCutNumber();
73
74   // Cut Selection
75   Bool_t MesonIsSelected(AliAODConversionMother *pi0,Bool_t IsSignal=kTRUE);
76   Bool_t MesonIsSelectedMC(TParticle *fMCMother,AliStack *fMCStack);
77   Bool_t MesonIsSelectedMCDalitz(TParticle *fMCMother,AliStack *fMCStack);
78   void PrintCuts();
79
80   void InitCutHistograms(TString name="");
81   void SetFillCutHistograms(TString name=""){if(!fHistograms){InitCutHistograms(name);};}
82   TList *GetCutHistograms(){return fHistograms;}
83   void SmearParticle(AliAODConversionPhoton * photon);
84   
85   ///Cut functions
86   Bool_t RejectSharedElectronV0s(AliAODConversionPhoton* photon, Int_t nV0, Int_t nV0s);
87   Bool_t RejectToCloseV0s(AliAODConversionPhoton* photon, TList *photons, Int_t nV0);
88   // Set Individual Cuts
89   Bool_t SetRCut(Int_t RCut);
90   Bool_t SetMesonKind(Int_t mesonKind);
91   Bool_t SetChi2MesonCut(Int_t chi2MesonCut);
92   Bool_t SetAlphaMesonCut(Int_t alphaMesonCut);
93   Bool_t SetRapidityMesonCut(Int_t RapidityMesonCut);
94   Bool_t SetBackgroundScheme(Int_t BackgroundScheme);
95   Bool_t SetNDegreesForRotationMethod(Int_t DegreesForRotationMethod);
96   Bool_t SetNumberOfBGEvents(Int_t NumberOfBGEvents);
97   Bool_t SetMCPSmearing(Int_t useMCPSmearing);
98   Bool_t SetSharedElectronCut(Int_t sharedElec);
99   Bool_t SetToCloseV0sCut(Int_t toClose);
100
101
102   // Request Flags
103   Bool_t UseRotationMethod(){return fUseRotationMethodInBG;}
104   Bool_t UseTrackMultiplicity(){return fUseTrackMultiplicityForBG;}
105   Int_t GetNumberOfBGEvents(){return fNumberOfBGEvents;}
106   Int_t NDegreesRotation(){return fnDegreeRotationPMForBG;}
107   Bool_t DoBGCalculation(){return fDoBG;}
108   Bool_t DoBGProbability(){return fdoBGProbability;}
109   Bool_t UseElecSharingCut(){return fDoSharedElecCut;}
110   Bool_t UseToCloseV0sCut(){return fDoToCloseV0sCut;}
111   Bool_t UseMCPSmearing(){return fUseMCPSmearing;}
112   Int_t BackgroundHandlerType(){return fBackgroundHandler;}
113   
114   protected:
115   TList *fHistograms;
116   //cuts
117   Int_t fMesonKind;
118   Double_t fMaxR; //r cut  
119   Double_t fChi2CutMeson; //chicut meson
120   Double_t fAlphaMinCutMeson; // min value for meson alpha cut
121   Double_t fAlphaCutMeson; // max value for meson alpha cut
122   Double_t fRapidityCutMeson; // max value for meson rapidity
123   Bool_t fUseRotationMethodInBG; // flag to apply rotation method for meson bg estimation
124   Bool_t fDoBG; // flag to intialize BG
125   Bool_t fdoBGProbability; // flag to use probability method for meson bg estimation
126   Bool_t fUseTrackMultiplicityForBG; // flag to use track multiplicity for meson bg estimation (else V0 mult)
127   Int_t fnDegreeRotationPMForBG; //
128   Int_t fNumberOfBGEvents; //
129   Float_t fOpeningAngle; // min opening angle for meson
130   Bool_t fDoToCloseV0sCut; //
131   Double_t fminV0Dist; //
132   Bool_t fDoSharedElecCut; //
133   Bool_t fUseMCPSmearing; // flag
134   Double_t fPBremSmearing;//
135   Double_t fPSigSmearing; //
136   Double_t fPSigSmearingCte; //
137   TF1 *fBrem; //
138   TRandom3 fRandom; //
139   Int_t fElectronArraySize; // size of electron label array
140   Int_t *fElectronLabelArray; //[fElectronArraySize]
141   Int_t fBackgroundHandler; //
142   
143   // Histograms
144   TObjString *fCutString; // cut number used for analysis
145   TH1F *hMesonCuts; // bookkeeping for meson cuts
146   TH1F *hMesonBGCuts; // bookkeeping for meson bg cuts
147
148   ClassDef(AliConversionMesonCuts,3)
149 };
150
151
152 #endif