da5a4f1f4ff82d2329fa804c87f4ed432f7000bb
[u/mrichter/AliRoot.git] / PWGHF / hfe / AliAnalysisTaskHFECal.h
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 #ifndef ALIANALYSISTASKHFECAL_H
17 #define ALIANALYSISTASKHFECAL_H
18
19 class THnSparse;
20 class TH2F;
21 class TLorentzVector;
22
23 class AliEMCALTrack;
24 class AliMagF;
25 class AliESDEvent;
26 class AliAODEvent;
27 class AliEMCALGeometry;
28 class AliEMCALRecoUtils;
29 class AliAnalysisFilter;
30 class AliESDtrackCuts;
31 class AliESDtrack;
32 class AliHFEcontainer;
33 class AliHFEcuts;
34 class AliHFEpid;
35 class AliHFEpidQAmanager;
36 class AliCFManager;
37
38 #include "AliAnalysisTaskSE.h"
39
40 class AliAnalysisTaskHFECal : public AliAnalysisTaskSE {
41  public:
42   AliAnalysisTaskHFECal();
43   AliAnalysisTaskHFECal(const char *name);
44   virtual ~AliAnalysisTaskHFECal();
45   
46   virtual void   UserCreateOutputObjects();
47   virtual void   UserExec(Option_t *option);
48   virtual void   Terminate(Option_t *);
49   
50   void SetHFECuts(AliHFEcuts * const cuts) { fCuts = cuts; };
51   void SetOpeningAngleCut (Double_t openingAngle) {fOpeningAngleCut = openingAngle;};
52   void SetInvariantMassCut (Double_t invmass) {fInvmassCut = invmass;};
53   void SetQAHist (int qahist) {fqahist = qahist;};
54   AliHFEpid *GetPID() const { return fPID; }
55   void SetRejectKinkMother(Bool_t rejectKinkMother = kFALSE) { fRejectKinkMother = rejectKinkMother; };
56   void SelectPhotonicElectron(Int_t itrack, Double_t cent, AliESDtrack *track, Bool_t &fFlagPhotonicElec, Bool_t &fFlagConvinatElec, Double_t nSig, Double_t shower, Double_t ep, Double_t mce);
57   void FindTriggerClusters();
58  private:
59   
60   Bool_t ProcessCutStep(Int_t cutStep, AliVParticle *track);
61   
62   AliESDEvent           *fESD;                  //!ESD object
63   AliMCEvent            *fMC;                   //!MC object
64   AliEMCALGeometry      *fGeom;                 // emcal geometry 
65     
66   TList                 *fOutputList;           //! output list
67   Int_t                 fqahist;  
68
69   AliESDtrackCuts       *fTrackCuts;            //! ESD track cuts
70   AliHFEcuts            *fCuts;                 //! Cut Collection
71   Bool_t                fIdentifiedAsOutInz;    //Out Of Range in z
72   Bool_t                fPassTheEventCut;       //Pass The Event Cut
73   Bool_t                fRejectKinkMother;      //Reject Kink Mother
74   Bool_t                fmcData;
75   Double_t              fVz;                    //z position of the primary vertex
76   AliCFManager          *fCFM;                  //! Correction Framework Manager
77   AliHFEpid             *fPID;                  //! PID
78   AliHFEpidQAmanager    *fPIDqa;                //! PID QA manager
79   Double_t              fOpeningAngleCut;       //openingAngle cut value
80   Double_t              fInvmassCut;            //invariant mass cut value
81  
82   int ftriggers[48][60];//!
83   int ftriggersCut[48][60];//!
84   int ftriggersTime[48][60];//!
85  
86
87   TH1F                  *fNoEvents;             //! no of events
88   THnSparseD            *fEMCAccE;              //! EMC acc
89   TH1F                  *fTrkpt;                //! track pt
90   TH2F                  *fTrkEovPBef;           //! track E/p before HFE pid
91   TH2F                  *fTrkEovPAft;           //! track E/p after HFE pid
92   TH2F                  *fdEdxBef;              //! track dEdx vs p before HFE pid
93   TH2F                  *fdEdxAft;              //! track dEdx vs p after HFE pid
94   TH2F                  *fIncpT;                //! HFE pid electron vs centrality
95   TH2F                  *fIncpTM20;             //! HFE pid electron vs centrality
96   THnSparseD            *fInvmassLS;            //! Inv mass of LS (e,e)
97   THnSparseD            *fInvmassULS;           //! Inv mass of ULS (e,e)
98   TH1F                  *fOpeningAngleLS;       //! opening angle for LS pairs
99   TH1F                  *fOpeningAngleULS;      //! opening angle for ULS pairs
100   TH1F                  *fPhotoElecPt;          //! photonic elec pt 
101   TH2F                  *fPhoElecPt;            //! Pho inclusive ele pt
102   TH2F                  *fPhoElecPtM20;         //! Pho inclusive ele pt
103   TH2F                  *fSameElecPt;           //! Same inclusive ele pt
104   TH2F                  *fSameElecPtM20;                //! Same inclusive ele pt
105   
106   TH1F                  *fTrackPtBefTrkCuts;    //! Track pt before track cuts  
107   TH1F                  *fTrackPtAftTrkCuts;    //! Track pt after track cuts
108   TH2F                  *fTPCnsigma;            //! TPC n sigma vs p    
109   
110   TH1F                  *fCent;                 //! centrality
111   THnSparseD            *fEleInfo;              //! EMC acc
112   /*
113   //<---- trigger info
114   TH1F        *fClsEBftTrigCut; //Cluster E before trigger selection
115   TH1F        *fClsEAftTrigCut; //Cluster E after trigger selection
116   TH1F        *fClsEAftTrigCut1;        //Cluster E after trigger selection
117   TH1F        *fClsEAftTrigCut2;        //Cluster E after trigger selection
118   TH1F        *fClsEAftTrigCut3;        //Cluster E after trigger selection
119   TH1F        *fClsEAftTrigCut4;        //Cluster E after trigger selection
120   TH2F        *fClsETime; //ClsE vs time distribution
121   TH2F        *fClsETime1; //ClsE vs time distribution
122   TH1F        *fTrigTimes;// trigger time
123   TH2F        *fCellCheck;// trigger time
124   */
125   //<------ MC
126   TH2F                  *fInputHFEMC;
127   TH2F                  *fInputAlle;
128   TH2F                  *fIncpTMChfe;           //! MC HFE pid electron vs centrality
129   TH2F                  *fIncpTMChfeAll;                //! MC HFE pid electron vs centrality
130   TH2F                  *fIncpTMCM20hfe;        //! MC HFE pid electron vs centrality
131   TH2F                  *fIncpTMCM20hfeAll;     //! MC HFE pid electron vs centrality
132   TH2F                  *fIncpTMCpho;           //! MC HFE pid electron vs centrality
133   TH2F                  *fIncpTMCM20pho;        //! MC HFE pid electron vs centrality
134   TH2F                  *fPhoElecPtMC;          //! Pho inclusive ele pt
135   TH2F                  *fPhoElecPtMCM20;               //! Pho inclusive ele pt
136   TH2F                  *fSameElecPtMC;         //! Same inclusive ele pt
137   TH2F                  *fSameElecPtMCM20;              //! Same inclusive ele pt
138   
139   AliAnalysisTaskHFECal(const AliAnalysisTaskHFECal&); // not implemented
140   AliAnalysisTaskHFECal& operator=(const AliAnalysisTaskHFECal&); // not implemented
141   
142   ClassDef(AliAnalysisTaskHFECal, 1); //!example of analysis
143 };
144
145 #endif
146
147