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 #include "AliStack.h"
  40
  41 class AliAnalysisTaskHFECal : public AliAnalysisTaskSE {
  42  public:
  43   AliAnalysisTaskHFECal();
  44   AliAnalysisTaskHFECal(const char *name);
  45   virtual ~AliAnalysisTaskHFECal();
  46
  47   virtual void   UserCreateOutputObjects();
  48   virtual void   UserExec(Option_t *option);
  49   virtual void   Terminate(Option_t *);
  50
  51   void SetHFECuts(AliHFEcuts * const cuts) { fCuts = cuts; };
  52   void SetOpeningAngleCut (Double_t openingAngle) {fOpeningAngleCut = openingAngle;};
  53   void SetInvariantMassCut (Double_t invmass) {fInvmassCut = invmass;};
  54   void SetQAHist (int qahist) {fqahist = qahist;};
  55   AliHFEpid *GetPID() const { return fPID; }
  56   void SetRejectKinkMother(Bool_t rejectKinkMother = kFALSE) { fRejectKinkMother = rejectKinkMother; };
  57   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, Double_t w, Int_t ibgevent, Bool_t tagpi0, Bool_t tageta);
  58   void FindMother(TParticle* part, int &label, int &pid);
  59   double GetMCweight(double mcPi0pT);
  60   double GetMCweightEta(double mcEtapT);
  61   void FindTriggerClusters();
  62  private:
  63
  64   Bool_t ProcessCutStep(Int_t cutStep, AliVParticle *track);
  65
  66   AliESDEvent           *fESD;                  //!ESD object
  67   AliMCEvent            *fMC;                   //!MC object
  68   AliStack              *stack;                 //!MC object
  69   AliEMCALGeometry      *fGeom;                 // emcal geometry
  70
  71   TList                 *fOutputList;           //! output list
  72   Int_t                 fqahist;
  73
  74   AliESDtrackCuts       *fTrackCuts;            //! ESD track cuts
  75   AliHFEcuts            *fCuts;                 //! Cut Collection
  76   Bool_t                fIdentifiedAsOutInz;    //Out Of Range in z
  77   Bool_t                fPassTheEventCut;       //Pass The Event Cut
  78   Bool_t                fRejectKinkMother;      //Reject Kink Mother
  79   Bool_t                fmcData;
  80   Double_t              fVz;                    //z position of the primary vertex
  81   AliCFManager          *fCFM;                  //! Correction Framework Manager
  82   AliHFEpid             *fPID;                  //! PID
  83   AliHFEpidQAmanager    *fPIDqa;                //! PID QA manager
  84   Double_t              fOpeningAngleCut;       //openingAngle cut value
  85   Double_t              fInvmassCut;            //invariant mass cut value
  86
  87   int ftriggers[48][60];//!
  88   int ftriggersCut[48][60];//!
  89   int ftriggersTime[48][60];//!
  90
  91
  92   TH1F                  *fNoEvents;             //! no of events
  93   THnSparseD            *fEMCAccE;              //! EMC acc
  94   TH1F                  *fTrkpt;                //! track pt
  95   TH2F                  *fTrkEovPBef;           //! track E/p before HFE pid
  96   TH2F                  *fTrkEovPAft;           //! track E/p after HFE pid
  97   TH2F                  *fdEdxBef;              //! track dEdx vs p before HFE pid
  98   TH2F                  *fdEdxAft;              //! track dEdx vs p after HFE pid
  99   TH2F                  *fIncpT;                //! HFE pid electron vs centrality
 100   TH2F                  *fIncpTM20;             //! HFE pid electron vs centrality
 101   THnSparseD            *fInvmassLS;            //! Inv mass of LS (e,e)
 102   THnSparseD            *fInvmassULS;           //! Inv mass of ULS (e,e)
 103   THnSparseD            *fInvmassLSmc;          //! Inv mass of LS (e,e)
 104   THnSparseD            *fInvmassULSmc;         //! Inv mass of ULS (e,e)
 105   TH2D          *fInvmassLSmc0;         //! Inv mass of ULS (e,e)
 106   TH2D          *fInvmassLSmc1;         //! Inv mass of ULS (e,e)
 107   TH2D          *fInvmassLSmc2;         //! Inv mass of ULS (e,e)
 108   TH2D          *fInvmassLSmc3;         //! Inv mass of ULS (e,e)
 109   TH2D          *fInvmassULSmc0;                //! Inv mass of ULS (e,e)
 110   TH2D          *fInvmassULSmc1;                //! Inv mass of ULS (e,e)
 111   TH2D          *fInvmassULSmc2;                //! Inv mass of ULS (e,e)
 112   TH2D          *fInvmassULSmc3;                //! Inv mass of ULS (e,e)
 113   TH1F                  *fOpeningAngleLS;       //! opening angle for LS pairs
 114   TH1F                  *fOpeningAngleULS;      //! opening angle for ULS pairs
 115   TH1F                  *fPhotoElecPt;          //! photonic elec pt
 116   TH2F                  *fPhoElecPt;            //! Pho inclusive ele pt
 117   TH2F                  *fPhoElecPtM20;         //! Pho inclusive ele pt
 118   TH2F                  *fSameElecPt;           //! Same inclusive ele pt
 119   TH2F                  *fSameElecPtM20;                //! Same inclusive ele pt
 120
 121   TH1F                  *fTrackPtBefTrkCuts;    //! Track pt before track cuts
 122   TH1F                  *fTrackPtAftTrkCuts;    //! Track pt after track cuts
 123   TH2F                  *fTPCnsigma;            //! TPC n sigma vs p
 124
 125   TH1F                  *fCent;                 //! centrality
 126   THnSparseD            *fEleInfo;              //! EMC acc
 127   /*
 128   //<---- trigger info
 129   TH1F        *fClsEBftTrigCut; //Cluster E before trigger selection
 130   TH1F        *fClsEAftTrigCut; //Cluster E after trigger selection
 131   TH1F        *fClsEAftTrigCut1;        //Cluster E after trigger selection
 132   TH1F        *fClsEAftTrigCut2;        //Cluster E after trigger selection
 133   TH1F        *fClsEAftTrigCut3;        //Cluster E after trigger selection
 134   TH1F        *fClsEAftTrigCut4;        //Cluster E after trigger selection
 135   TH2F        *fClsETime; //ClsE vs time distribution
 136   TH2F        *fClsETime1; //ClsE vs time distribution
 137   TH1F        *fTrigTimes;// trigger time
 138   TH2F        *fCellCheck;// trigger time
 139   */
 140   //<------ MC
 141   TH2F                  *fInputHFEMC;
 142   TH2F                  *fInputAlle;
 143   TH2F                  *fIncpTMChfe;           //! MC HFE pid electron vs centrality
 144   TH2F                  *fIncpTMChfeAll;                //! MC HFE pid electron vs centrality
 145   TH2F                  *fIncpTMCM20hfe;        //! MC HFE pid electron vs centrality
 146   TH2F                  *fIncpTMCM20hfeAll;     //! MC HFE pid electron vs centrality
 147   TH2F                  *fIncpTMCM20hfeCheck;   //! MC HFE pid electron vs centrality
 148  THnSparseD             *fIncpTMCpho;           //! MC HFE pid electron vs centrality
 149  THnSparseD             *fIncpTMCM20pho;        //! MC HFE pid electron vs centrality
 150  THnSparseD             *fPhoElecPtMC;          //! Pho inclusive ele pt
 151  THnSparseD             *fPhoElecPtMCM20;               //! Pho inclusive ele pt
 152  THnSparseD             *fSameElecPtMC;         //! Same inclusive ele pt
 153  THnSparseD             *fSameElecPtMCM20;              //! Same inclusive ele pt
 154  THnSparseD             *fIncpTMCM20pho_pi0e;   //! MC HFE pid electron vs centrality
 155  THnSparseD             *fPhoElecPtMCM20_pi0e;          //! Pho inclusive ele pt
 156  THnSparseD             *fSameElecPtMCM20_pi0e;         //! Same inclusive ele pt
 157  THnSparseD             *fIncpTMCM20pho_eta;    //! MC HFE pid electron vs centrality
 158  THnSparseD             *fPhoElecPtMCM20_eta;           //! Pho inclusive ele pt
 159  THnSparseD             *fSameElecPtMCM20_eta;          //! Same inclusive ele pt
 160  TH1D                   *CheckNclust;
 161  TH1D                   *CheckNits;
 162  TH2D                   *Hpi0pTcheck;
 163  TH2D                   *HETApTcheck;
 164  TH2D                   *HphopTcheck;
 165  TH1D                   *fpTCheck;
 166  TH2D                   *fMomDtoE;
 167
 168   AliAnalysisTaskHFECal(const AliAnalysisTaskHFECal&); // not implemented
 169   AliAnalysisTaskHFECal& operator=(const AliAnalysisTaskHFECal&); // not implemented
 170
 171   ClassDef(AliAnalysisTaskHFECal, 1); //!example of analysis
 172 };
 173
 174 #endif
 175
 176