updated
[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   AliHFEpid *GetPID() const { return fPID; }
54   void SetRejectKinkMother(Bool_t rejectKinkMother = kFALSE) { fRejectKinkMother = rejectKinkMother; };
55   void SelectPhotonicElectron(Int_t itrack, Double_t cent, AliESDtrack *track, Bool_t &fFlagPhotonicElec, Bool_t &fFlagConvinatElec);
56   void FindTriggerClusters();
57  private:
58   
59   Bool_t ProcessCutStep(Int_t cutStep, AliVParticle *track);
60   
61   AliESDEvent           *fESD;                  //!ESD object
62   AliEMCALGeometry      *fGeom;                 // emcal geometry 
63     
64   TList                 *fOutputList;           //! output list
65   
66   AliESDtrackCuts       *fTrackCuts;            //! ESD track cuts
67   AliHFEcuts            *fCuts;                 //! Cut Collection
68   Bool_t                fIdentifiedAsOutInz;    //Out Of Range in z
69   Bool_t                fPassTheEventCut;       //Pass The Event Cut
70   Bool_t                fRejectKinkMother;      //Reject Kink Mother
71   Double_t              fVz;                    //z position of the primary vertex
72   AliCFManager          *fCFM;                  //! Correction Framework Manager
73   AliHFEpid             *fPID;                  //! PID
74   AliHFEpidQAmanager    *fPIDqa;                //! PID QA manager
75   Double_t              fOpeningAngleCut;       //openingAngle cut value
76   Double_t              fInvmassCut;            //invariant mass cut value
77  
78   int ftriggers[48][60];//!
79   int ftriggersCut[48][60];//!
80   int ftriggersTime[48][60];//!
81  
82   TH1F                  *fNoEvents;             //! no of events
83   THnSparseD            *fEMCAccE;              //! EMC acc
84   TH1F                  *fTrkpt;                //! track pt
85   TH2F                  *fTrkEovPBef;           //! track E/p before HFE pid
86   TH2F                  *fTrkEovPAft;           //! track E/p after HFE pid
87   TH2F                  *fdEdxBef;              //! track dEdx vs p before HFE pid
88   TH2F                  *fdEdxAft;              //! track dEdx vs p after HFE pid
89   TH2F                  *fIncpT;                //! HFE pid electron vs centrality
90   TH2F                  *fIncpTM20;             //! HFE pid electron vs centrality
91   THnSparseD            *fInvmassLS;            //! Inv mass of LS (e,e)
92   THnSparseD            *fInvmassULS;           //! Inv mass of ULS (e,e)
93   TH1F                  *fOpeningAngleLS;       //! opening angle for LS pairs
94   TH1F                  *fOpeningAngleULS;      //! opening angle for ULS pairs
95   TH1F                  *fPhotoElecPt;          //! photonic elec pt 
96   TH2F                  *fPhoElecPt;            //! Pho inclusive ele pt
97   TH2F                  *fPhoElecPtM20;         //! Pho inclusive ele pt
98   TH2F                  *fSameElecPt;           //! Same inclusive ele pt
99   TH2F                  *fSameElecPtM20;                //! Same inclusive ele pt
100   
101   TH1F                  *fTrackPtBefTrkCuts;    //! Track pt before track cuts  
102   TH1F                  *fTrackPtAftTrkCuts;    //! Track pt after track cuts
103   TH2F                  *fTPCnsigma;            //! TPC n sigma vs p    
104   
105   TH1F                  *fCent;                 //! centrality
106   THnSparseD            *fEleInfo;              //! EMC acc
107   
108   //<---- trigger info
109   TH1F        *fClsEBftTrigCut; //Cluster E before trigger selection
110   TH1F        *fClsEAftTrigCut; //Cluster E after trigger selection
111   TH1F        *fClsEAftTrigCut1;        //Cluster E after trigger selection
112   TH1F        *fClsEAftTrigCut2;        //Cluster E after trigger selection
113   TH1F        *fClsEAftTrigCut3;        //Cluster E after trigger selection
114   TH1F        *fClsEAftTrigCut4;        //Cluster E after trigger selection
115   TH2F        *fClsETime; //ClsE vs time distribution
116   TH2F        *fClsETime1; //ClsE vs time distribution
117   TH1F        *fTrigTimes;// trigger time
118   TH2F        *fCellCheck;// trigger time
119
120
121   AliAnalysisTaskHFECal(const AliAnalysisTaskHFECal&); // not implemented
122   AliAnalysisTaskHFECal& operator=(const AliAnalysisTaskHFECal&); // not implemented
123   
124   ClassDef(AliAnalysisTaskHFECal, 1); //!example of analysis
125 };
126
127 #endif
128
129