PWGUD/UPC/AliAnalysisTaskUpcPsi2s.h

   1 /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
   2 * See cxx source for full Copyright notice */
   3 /* $Id$ */
   4
   5 #ifndef ALIANALYSISTASKUPCPSI2S_H
   6 #define ALIANALYSISTASKUPCPSI2S_H
   7
   8 class TClonesArray;
   9 class TTree;
  10 class TH1;
  11 class TH2;
  12 class TList;
  13 class AliPIDResponse;
  14 class AliAODEvent;
  15 class AliESDEvent;
  16
  17 #define ntrg 17
  18 #include "AliAnalysisTaskSE.h"
  19
  20 class AliAnalysisTaskUpcPsi2s : public AliAnalysisTaskSE {
  21  public:
  22   AliAnalysisTaskUpcPsi2s();
  23   AliAnalysisTaskUpcPsi2s(const char *name);
  24   virtual ~AliAnalysisTaskUpcPsi2s();
  25
  26   virtual void Init();
  27   virtual void UserCreateOutputObjects();
  28   virtual void UserExec(Option_t *option);
  29   virtual void RunAODtrig();
  30   virtual void RunAODhist();
  31   virtual void RunAODtree();
  32   virtual void RunAODMC(AliAODEvent *aod);
  33   virtual void RunAODsystematics(AliAODEvent *aod);
  34   virtual void RunESDtrig();
  35   virtual void RunESDhist();
  36   virtual void RunESDtree();
  37   virtual void RunESDMC(AliESDEvent *esd);
  38   virtual void Terminate(Option_t *);
  39   void SetRunTree(Bool_t runTree){fRunTree = runTree;}
  40   void SetRunHist(Bool_t runHist){fRunHist = runHist;}
  41   void SetRunSyst(Bool_t runSyst){fRunSystematics = runSyst;}
  42   void SetIsMC(Bool_t MC){isMC = MC;}
  43   void InitSystematics();
  44   Double_t GetMedian(Double_t *daArray);
  45
  46  private:
  47   Int_t fType; // 0 - ESD, 1 - AOD
  48   Bool_t isMC;
  49   Bool_t fRunTree;
  50   Bool_t fRunHist;
  51   Bool_t fRunSystematics;
  52
  53   AliPIDResponse *fPIDResponse;
  54
  55   //event tree
  56   TTree *fJPsiTree;
  57   TTree *fPsi2sTree;
  58   //tree variables
  59   Int_t fRunNum;
  60   UInt_t fPerNum, fOrbNum;
  61   //trigger
  62   Bool_t fTrigger[ntrg];
  63   Bool_t fTriggerInputsMC[4];
  64   UInt_t fL0inputs, fL1inputs;
  65   Bool_t fTOFtrig1, fTOFtrig2;
  66   Double_t fTOFphi[4];
  67
  68   Double_t fPIDTPCMuon[4];
  69   Double_t fPIDTPCElectron[4];
  70   Double_t fPIDTPCPion[4];
  71   Double_t fPIDTPCKaon[4];
  72   Double_t fPIDTPCProton[4];
  73
  74   Double_t fPIDTOFMuon[4];
  75   Double_t fPIDTOFElectron[4];
  76   Double_t fPIDTOFPion[4];
  77   Double_t fPIDTOFKaon[4];
  78   Double_t fPIDTOFProton[4];
  79
  80   Int_t fVtxContrib;
  81   Double_t fVtxPos[3];
  82   Double_t fVtxErr[3];
  83   Double_t fVtxChi2,fVtxNDF;
  84   Double_t fKfVtxPos[3];
  85   Double_t fSpdVtxPos[3];
  86   UShort_t fBCrossNum, fNtracklets, fNLooseTracks;
  87   //vzero, zdc
  88   Double_t fZDCAenergy, fZDCCenergy;
  89   Int_t fV0Adecision, fV0Cdecision;
  90   //input data
  91   TObjString *fDataFilnam;
  92   Short_t fRecoPass;
  93   Long64_t fEvtNum;
  94   //tracks
  95   TClonesArray *fJPsiAODTracks;
  96   TClonesArray *fJPsiESDTracks;
  97   TClonesArray *fPsi2sAODTracks;
  98   TClonesArray *fPsi2sESDTracks;
  99     //mc
 100   TClonesArray *fGenPart;
 101
 102   TList *fListTrig;
 103   TH1D *fHistCcup4TriggersPerRun;
 104   TH1D *fHistCcup7TriggersPerRun;
 105   TH1D *fHistCcup2TriggersPerRun;
 106   TH1D *fHistCint1TriggersPerRun;
 107   TH1D *fHistZedTriggersPerRun;
 108   TH1D *fHistCvlnTriggersPerRun;
 109   TH1D *fHistMBTriggersPerRun;
 110   TH1D *fHistCentralTriggersPerRun;
 111   TH1D *fHistSemiCentralTriggersPerRun;
 112
 113   TList *fListHist;
 114   TH1D *fHistNeventsJPsi;
 115   TH2D *fHistTPCsignalJPsi;
 116   TH2D *fHistDiLeptonPtJPsi;
 117   TH1D *fHistDiElectronMass;
 118   TH1D *fHistDiMuonMass;
 119   TH1D *fHistDiLeptonMass;
 120
 121   TH1D *fHistNeventsPsi2s;
 122   TH2D *fHistPsi2sMassVsPt;
 123   TH1D *fHistPsi2sMassCoherent;
 124
 125   TList *fListSystematics;
 126   TList *fListJPsiLoose;
 127   TList *fListJPsiTight;
 128   TList *fListPsi2sLoose;
 129   TList *fListPsi2sTight;
 130
 131   AliAnalysisTaskUpcPsi2s(const AliAnalysisTaskUpcPsi2s&); //not implemented
 132   AliAnalysisTaskUpcPsi2s& operator =(const AliAnalysisTaskUpcPsi2s&); //not implemented
 133
 134   ClassDef(AliAnalysisTaskUpcPsi2s, 1);
 135 };
 136
 137 #endif
 138
 139
 140
 141
 142
 143
 144
 145
 146