PWGGA/CaloTrackCorrelations/AliAnaPi0EbE.h

   1 #ifndef ALIANAPI0EBE_H
   2 #define ALIANAPI0EBE_H
   3 /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
   4  * See cxx source for full Copyright notice     */
   5
   6 //_________________________________________________________________________
   7 //
   8 // Class for the analysis of high pT pi0 event by event
   9 // Pi0/Eta identified by one of the following:
  10 //  -Invariant mass of 2 cluster in calorimeter
  11 //  -Shower shape analysis in calorimeter
  12 //  -Invariant mass of one cluster in calorimeter and one photon reconstructed in TPC (in near future)
  13 //
  14 //-- Author: Gustavo Conesa (INFN-LNF)  &  Raphaelle Ichou (SUBATECH)
  15 //_________________________________________________________________________
  16
  17
  18 // --- ROOT system ---
  19 class TList ;
  20 class TObjString;
  21
  22 // --- ANALYSIS system ---
  23 #include "AliAnaCaloTrackCorrBaseClass.h"
  24
  25 class AliAnaPi0EbE : public AliAnaCaloTrackCorrBaseClass {
  26
  27  public:
  28   AliAnaPi0EbE() ; // default ctor
  29   virtual ~AliAnaPi0EbE() { ; } //virtual dtor
  30
  31   TObjString *   GetAnalysisCuts();
  32
  33   TList      *   GetCreateOutputObjects();
  34
  35   Int_t          GetMCIndex(const Int_t aodTag);
  36
  37   void           Init();
  38
  39   void           InitParameters();
  40
  41   void           MakeAnalysisFillAOD()  ;
  42
  43   void           MakeAnalysisFillHistograms() ;
  44
  45   void           Print(const Option_t * opt) const;
  46
  47   // Main
  48
  49   void           FillSelectedClusterHistograms(AliVCluster* cluster,
  50                                                const Int_t nLocMax,
  51                                                const Int_t tag,
  52                                                const Float_t asy = 0);
  53
  54   void           FillWeightHistograms(AliVCluster *clus);
  55
  56   void           HasPairSameMCMother(AliAODPWG4Particle * photon1,
  57                                      AliAODPWG4Particle * photon2,
  58                                      Int_t & label, Int_t & tag);
  59
  60   void           MakeInvMassInCalorimeter() ;
  61
  62   void           MakeInvMassInCalorimeterAndCTS() ;
  63
  64   void           MakeShowerShapeIdentification() ;
  65
  66   //Setters Getters
  67
  68   //Analysis types
  69   enum anaTypes  {kIMCalo, kSSCalo, kIMCaloTracks};
  70   anaTypes       GetAnalysisType()                     const { return fAnaType                 ; }
  71   void           SetAnalysisType(anaTypes ana)               { fAnaType = ana                  ; }
  72
  73   TString        GetInputAODGammaConvName()            const { return fInputAODGammaConvName   ; }
  74   void           SetInputAODGammaConvName(TString name)      { fInputAODGammaConvName = name   ; }
  75
  76   //Only for pi0 SS identification case
  77   void           SetCalorimeter(TString & det)               { fCalorimeter = det              ; }
  78
  79   void           SetMinDistanceToBadChannel(Float_t m1, Float_t m2, Float_t m3) {
  80                   fMinDist = m1; fMinDist2 = m2; fMinDist3 = m3                               ; }
  81
  82   void           SetTimeCut(Double_t min, Double_t max)      { fTimeCutMin = min;
  83                                                                fTimeCutMax = max               ; }
  84   Double_t       GetTimeCutMin()                       const { return fTimeCutMin              ; }
  85   Double_t       GetTimeCutMax()                       const { return fTimeCutMax              ; }
  86
  87   void           SwitchOnFillWeightHistograms()              { fFillWeightHistograms  = kTRUE  ; }
  88   void           SwitchOffFillWeightHistograms()             { fFillWeightHistograms  = kFALSE ; }
  89
  90   void           SwitchOnTMHistoFill()                       { fFillTMHisto           = kTRUE  ; }
  91   void           SwitchOffTMHistoFill()                      { fFillTMHisto           = kFALSE ; }
  92
  93   void           SwitchOnSelectedClusterHistoFill()          { fFillSelectClHisto     = kTRUE  ; }
  94   void           SwitchOffSelectedClusterHistoFill()         { fFillSelectClHisto     = kFALSE ; }
  95
  96   void           SwitchOnOnlySimpleSSHistoFill()             { fFillOnlySimpleSSHisto = kTRUE  ; }
  97   void           SwitchOffOnlySimpleHistoFill()              { fFillOnlySimpleSSHisto = kFALSE ; }
  98
  99
 100   //For histograms
 101   enum mcTypes   { kmcPhoton = 0, kmcConversion = 1, kmcPi0    = 2,
 102                    kmcEta    = 3, kmcElectron   = 4, kmcHadron = 5 };
 103
 104  private:
 105
 106   anaTypes       fAnaType;                 // Select analysis type
 107
 108   //Only for pi0 SS identification case, kSSCalo
 109   TString        fCalorimeter ;            // Calorimeter where the gamma is searched;
 110   Float_t        fMinDist ;                // Minimal distance to bad channel to accept cluster
 111   Float_t        fMinDist2;                // Cuts on Minimal distance to study acceptance evaluation
 112   Float_t        fMinDist3;                // One more cut on distance used for acceptance-efficiency study
 113   Double_t       fTimeCutMin  ;            // Remove clusters/cells with time smaller than this value, in ns
 114   Double_t       fTimeCutMax  ;            // Remove clusters/cells with time larger than this value, in ns
 115
 116   Bool_t         fFillWeightHistograms ;   // Fill weigth histograms
 117   Bool_t         fFillTMHisto;             // Fill track matching plots
 118   Bool_t         fFillSelectClHisto;       // Fill selected cluster histograms
 119   Bool_t         fFillOnlySimpleSSHisto;   // Fill selected cluster histograms, selected SS histograms
 120
 121   //Only for combination of calorimeter and conversion photons, kIMCaloTracks
 122   TString        fInputAODGammaConvName;   //  Name of AOD branch with conversion photons
 123
 124   //Histograms
 125
 126   TH1F         * fhPt  ;                   //! Number of identified  pi0/eta vs pT
 127   TH1F         * fhE   ;                   //! Number of identified  pi0/eta vs E
 128   TH2F         * fhEEta  ;                 //! E vs eta of identified  pi0/eta
 129   TH2F         * fhEPhi  ;                 //! E vs phi of identified  pi0/eta
 130   TH2F         * fhEtaPhi  ;               //! eta vs phi of identified  pi0/eta
 131
 132   TH1F         * fhPtDecay  ;              //! Number of identified  pi0/eta decay photons vs pT
 133   TH1F         * fhEDecay   ;              //! Number of identified  pi0/eta decay photons vs E
 134
 135   TH2F         * fhEDispersion ;           //! E vs disp of selected cluster
 136   TH2F         * fhELambda0 ;              //! E vs lambda0 of selected cluster
 137   TH2F         * fhELambda1 ;              //! E vs lambda1 of selected cluster
 138   TH2F         * fhELambda0NoTRD ;         //! E vs lambda0 of selected cluster, not behind TRD
 139   TH2F         * fhELambda0FracMaxCellCut ;//! E vs lambda0 of selected cluster, fraction of cluster energy in max cell cut
 140   TH2F         * fhEFracMaxCell ;          //! E vs frac max cell of selected cluster
 141   TH2F         * fhEFracMaxCellNoTRD ;     //! E vs frac max cell of selected cluster, not behind TRD
 142   TH2F         * fhENCells;                //! E vs N cells in selected cluster
 143   TH2F         * fhETime;                  //! E vs Time of selected cluster
 144   TH2F         * fhEPairDiffTime;          //! E vs Pair of clusters time difference vs E
 145
 146   TH2F         * fhDispEtaE ;              //! shower dispersion in eta direction
 147   TH2F         * fhDispPhiE ;              //! shower dispersion in phi direction
 148   TH2F         * fhLambda0DispEta[7] ;     //! shower shape correlation l0 vs disp eta
 149   TH2F         * fhLambda0DispPhi[7] ;     //! shower shape correlation l0 vs disp phi
 150   TH2F         * fhSumEtaE ;               //! shower dispersion in eta direction
 151   TH2F         * fhSumPhiE ;               //! shower dispersion in phi direction
 152   TH2F         * fhSumEtaPhiE ;            //! shower dispersion in eta and phi direction
 153   TH2F         * fhDispEtaPhiDiffE ;       //! shower dispersion eta - phi
 154   TH2F         * fhSphericityE ;           //! shower sphericity in eta vs phi
 155   TH2F         * fhDispEtaDispPhi[7] ;     //! shower dispersion in eta direction vs phi direction for 5 E bins [0-2],[2-4],[4-6],[6-10],[> 10]
 156   TH2F         * fhAsymmetryE ;            //! E asymmetry of 2 splitted clusters vs cluster E
 157   TH2F         * fhAsymmetryLambda0[7] ;   //! E asymmetry of 2 splitted clusters vs lam0 for 5 E bins
 158   TH2F         * fhAsymmetryDispEta[7] ;   //! E asymmetry of 2 splitted clusters vs lam0 for 5 E bins
 159   TH2F         * fhAsymmetryDispPhi[7] ;   //! E asymmetry of 2 splitted clusters vs lam0 for 5 E bins
 160
 161   //MC histograms
 162
 163   TH2F         * fhEMCLambda0[6] ;            //! E vs lambda0 of pi0 pairs but really from MC particle
 164   TH2F         * fhEMCLambda1[6] ;            //! E vs lambda1 of pi0 pairs but really from MC particle
 165   TH2F         * fhEMCDispersion[6] ;         //! E vs dispersion of pi0 pairs but really from MC particle
 166   TH2F         * fhEMCLambda0NoTRD[6] ;         //! E vs lambda0 of pi0 pairs but really from MC particle, not behind TRD
 167   TH2F         * fhEMCLambda0FracMaxCellCut[6] ;//! E vs lambda0 of pi0 pairs but really from MC particle, fraction of cluster energy in max cell cut
 168   TH2F         * fhEMCFracMaxCell[6] ;        //! E vs fraction of max cell
 169
 170   TH2F         * fhMCEDispEta[6] ;            //! shower dispersion in eta direction
 171   TH2F         * fhMCEDispPhi[6] ;            //! shower dispersion in phi direction
 172   TH2F         * fhMCLambda0DispEta[7][6] ;   //! shower shape correlation l0 vs disp eta
 173   TH2F         * fhMCLambda0DispPhi[7][6] ;   //! shower shape correlation l0 vs disp phi
 174   TH2F         * fhMCESumEtaPhi[6] ;          //! shower dispersion in eta vs phi direction
 175   TH2F         * fhMCEDispEtaPhiDiff[6] ;     //! shower dispersion in eta -phi direction
 176   TH2F         * fhMCESphericity[6] ;         //! shower sphericity, eta vs phi
 177   TH2F         * fhMCDispEtaDispPhi[7][6] ;   //! shower dispersion in eta direction vs phi direction for 5 E bins [0-2],[2-4],[4-6],[6-10],[> 10]
 178   TH2F         * fhMCEAsymmetry[6] ;          //! E asymmetry of 2 splitted clusters vs cluster E
 179   TH2F         * fhMCAsymmetryLambda0[7][6] ; //! E asymmetry of 2 splitted clusters vs lam0 for 5 E bins
 180   TH2F         * fhMCAsymmetryDispEta[7][6] ; //! E asymmetry of 2 splitted clusters vs lam0 for 5 E bins
 181   TH2F         * fhMCAsymmetryDispPhi[7][6] ; //! E asymmetry of 2 splitted clusters vs lam0 for 5 E bins
 182
 183   TH1F         * fhMCPt[6];                   //! Number of identified as pi0, coming from X
 184   TH2F         * fhMCPhi[6];                  //! Phi of identified as pi0, coming from X
 185   TH2F         * fhMCEta[6];                  //! eta of identified as pi0, coming from X
 186
 187   TH2F         * fhMCPi0PtGenRecoFraction;    //! SS id, clusters id as pi0 (eta), coming from 2 photon, pi0 primary, pt vs E prim pi0 / E reco
 188   TH2F         * fhMCEtaPtGenRecoFraction;    //! SS id, clusters id as pi0 (eta), coming from 2 photon, eta primary, pt vs E prim eta / E reco
 189   TH1F         * fhMCPi0DecayPt;              //! SS id, clusters id as pi0 (eta), coming from 1 photon, pi0 decay primary, pt
 190   TH2F         * fhMCPi0DecayPtFraction;      //! SS id, clusters id as pi0 (eta), coming from 1 photon, pi0 decay primary, pt vs pt decay / pt mother
 191   TH1F         * fhMCEtaDecayPt;              //! SS id, clusters id as pi0 (eta), coming from 1 photon, eta decay primary, pt
 192   TH2F         * fhMCEtaDecayPtFraction;      //! SS id, clusters id as pi0 (eta), coming from 1 photon, eta decay primary, pt vs pt decay / pt mother
 193   TH1F         * fhMCOtherDecayPt;            //! SS id, clusters id as pi0 (eta), coming from 1 photon, other decay primary, pt
 194
 195   TH2F         * fhMassPairMCPi0;             //! pair mass, origin is same pi0
 196   TH2F         * fhMassPairMCEta;             //! pair mass, origin is same eta
 197   TH2F         * fhAnglePairMCPi0;            //! pair opening angle, origin is same pi0
 198   TH2F         * fhAnglePairMCEta;            //! pair opening angle, origin is same eta
 199
 200   // Weight studies
 201
 202   TH2F         * fhECellClusterRatio;      //! e cell / e cluster vs e cluster for selected photons
 203   TH2F         * fhECellClusterLogRatio;   //! log (e cell / e cluster)  vs e cluster for selected photons
 204   TH2F         * fhEMaxCellClusterRatio;   //! e max cell / e cluster vs e cluster for selected photons
 205   TH2F         * fhEMaxCellClusterLogRatio;//! log (e max cell / e cluster) vs e cluster for selected photons
 206   TH2F         * fhLambda0ForW0[14];       //! L0 for 7 defined w0= 3, 3.5 ... 6 for selected photons
 207   //TH2F         * fhLambda1ForW0[7];        //! L1 for 7 defined w0= 3, 3.5 ... 6 for selected photons
 208
 209   // Track Matching
 210   TH2F         * fhTrackMatchedDEta     ;  //! Eta distance between track and cluster vs cluster E
 211   TH2F         * fhTrackMatchedDPhi     ;  //! Phi distance between track and cluster vs cluster E
 212   TH2F         * fhTrackMatchedDEtaDPhi ;  //! Eta vs Phi distance between track and cluster, E cluster > 0.5 GeV
 213   TH2F         * fhTrackMatchedMCParticle; //! Trace origin of matched particle
 214   TH2F         * fhdEdx  ;                 //! matched track dEdx vs cluster E
 215   TH2F         * fhEOverP;                 //! matched track E cluster over P track vs cluster E
 216   TH2F         * fhEOverPNoTRD;                 //! matched track E cluster over P track vs cluster E, not behind TRD
 217
 218   // Local maxima
 219   TH2F         * fhNLocMax;                //! number of maxima in selected clusters
 220   TH2F         * fhELambda0LocMax[3] ;     //! E vs lambda0 of selected cluster, 1,2,>2 local maxima in cluster
 221   TH2F         * fhELambda1LocMax[3] ;     //! E vs lambda1 of selected cluster, 1,2,>2 local maxima in cluster
 222   TH2F         * fhEDispersionLocMax[3] ;  //! E vs lambda1 of selected cluster, 1,2,>2 local maxima in cluster
 223   TH2F         * fhEDispEtaLocMax[3] ;     //! E vs eta dispersion of selected cluster, 1,2,>2 local maxima in cluster
 224   TH2F         * fhEDispPhiLocMax[3] ;     //! E vs phi dispersion of selected cluster, 1,2,>2 local maxima in cluster
 225   TH2F         * fhESumEtaPhiLocMax[3] ;   //! E vs dispersion in eta and phi direction
 226   TH2F         * fhEDispEtaPhiDiffLocMax[3] ; //! E vs dispersion eta - phi
 227   TH2F         * fhESphericityLocMax[3] ;  //! E vs sphericity in eta vs phi
 228   TH2F         * fhEAsymmetryLocMax[3] ;   //! E asymmetry of 2 splitted clusters vs cluster E for different NLM
 229
 230   TH2F         * fhMassPairLocMax[8];      //! pair mass, origin is same pi0, combine clusters depending on number of maxima
 231
 232   AliAnaPi0EbE(              const AliAnaPi0EbE & pi0ebe) ; // cpy ctor
 233   AliAnaPi0EbE & operator = (const AliAnaPi0EbE & pi0ebe) ; // cpy assignment
 234
 235   ClassDef(AliAnaPi0EbE,20)
 236 } ;
 237
 238
 239 #endif //ALIANAPI0EBE_H
 240
 241
 242