PWGGA/CaloTrackCorrelations/AliAnaElectron.h

   1 #ifndef ALIANAELECTRON_H
   2 #define ALIANAELECTRON_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 electron identification,
   9 // Clusters from calorimeters are identified as electrons
  10 // and kept in the AOD. Few histograms produced.
  11 // Copy of AliAnaPhoton just add electron id.
  12 //
  13
  14 //-- Author: Gustavo Conesa (LPSC-IN2P3-CNRS)
  15
  16 // --- ROOT system ---
  17 class TH2F ;
  18 class TH1F;
  19 class TH3D;
  20 class TString ;
  21 class TObjString;
  22
  23 // --- ANALYSIS system ---
  24 #include "AliAnaCaloTrackCorrBaseClass.h"
  25 class AliStack;
  26 class TParticle;
  27
  28 class TList ;
  29
  30 class AliAnaElectron : public AliAnaCaloTrackCorrBaseClass {
  31
  32  public:
  33
  34   AliAnaElectron() ;                                       // default ctor
  35
  36   virtual ~AliAnaElectron() { ; }                          // virtual dtor
  37
  38   //---------------------------------------
  39   // General analysis frame methods
  40   //---------------------------------------
  41
  42   TObjString * GetAnalysisCuts();
  43
  44   TList      * GetCreateOutputObjects();
  45
  46   void         Init();
  47
  48   void         InitParameters();
  49
  50   void         MakeAnalysisFillAOD()  ;
  51
  52   void         MakeAnalysisFillHistograms() ;
  53
  54   void         Print(const Option_t * opt)const;
  55
  56
  57   // Analysis methods
  58
  59   Bool_t       ClusterSelected(AliVCluster* cl, TLorentzVector mom) ;
  60
  61   void         FillShowerShapeHistograms( AliVCluster* cluster, const Int_t mcTag , const Int_t pidTag) ;
  62
  63   void         SwitchOnFillShowerShapeHistograms()    { fFillSSHistograms = kTRUE  ; }
  64   void         SwitchOffFillShowerShapeHistograms()   { fFillSSHistograms = kFALSE ; }
  65
  66   void         WeightHistograms(AliVCluster *clus);
  67
  68   void         SwitchOnFillWeightHistograms()         { fFillWeightHistograms = kTRUE  ; }
  69   void         SwitchOffFillWeightHistograms()        { fFillWeightHistograms = kFALSE ; }
  70
  71   //---------------------------------------
  72   // Analysis parameters setters getters
  73   //---------------------------------------
  74
  75   TString      GetCalorimeter()                 const { return fCalorimeter        ; }
  76   void         SetCalorimeter(TString  & det)         { fCalorimeter = det         ; }
  77
  78   // ** Cluster selection methods **
  79
  80   void         SetdEdxCut(Double_t min, Double_t max) { fdEdxMin    = min ;
  81                                                         fdEdxMax    = max          ; }
  82
  83   void         SetEOverP(Double_t min, Double_t max)  { fEOverPMin  = min ;
  84                                                         fEOverPMax  = max          ; }
  85
  86
  87   void         SetMinDistanceToBadChannel(Float_t m1, Float_t m2, Float_t m3) {
  88                 fMinDist = m1; fMinDist2 = m2; fMinDist3 = m3; }
  89
  90   void         SetTimeCut(Double_t min, Double_t max) { fTimeCutMin = min;
  91                                                         fTimeCutMax = max          ; }
  92   Double_t     GetTimeCutMin()                  const { return fTimeCutMin         ; }
  93   Double_t     GetTimeCutMax()                  const { return fTimeCutMax         ; }
  94
  95   void         SetNCellCut(Int_t n)                   { fNCellsCut = n             ; }
  96   Double_t     GetNCellCut()                    const { return fNCellsCut          ; }
  97
  98   void         FillNOriginHistograms(Int_t n)         { fNOriginHistograms = n ;
  99                                                         if(n > 10) fNOriginHistograms = 10; }
 100
 101
 102   void         FillAODWithElectrons()                 { fAODParticle = AliCaloPID::kElectron      ; }
 103   void         FillAODWithHadrons()                   { fAODParticle = AliCaloPID::kChargedHadron ; }
 104   void         FillAODWithAny()                       { fAODParticle = 0 ; }
 105
 106   void         SwitchOnOnlySimpleSSHistoFill()        { fFillOnlySimpleSSHisto = kTRUE  ; }
 107   void         SwitchOffOnlySimpleHistoFill()         { fFillOnlySimpleSSHisto = kFALSE ; }
 108
 109   // For histograms in arrays, index in the array, corresponding to a particle
 110   enum mcTypes    { kmcPhoton = 0,        kmcPi0Decay = 1,       kmcOtherDecay = 2,
 111                     kmcPi0 = 3,           kmcEta = 4,            kmcElectron = 5,
 112                     kmcConversion = 6,    kmcOther = 7,          kmcAntiNeutron = 8,
 113                     kmcAntiProton = 9                                                 };
 114
 115   enum mcssTypes  { kmcssPhoton = 0,      kmcssOther = 1,       kmcssPi0 = 2,
 116                     kmcssEta = 3,         kmcssConversion = 4,  kmcssElectron = 5       };
 117
 118   private:
 119
 120   TString  fCalorimeter ;                      // Calorimeter where the gamma is searched;
 121   Float_t  fMinDist ;                          // Minimal distance to bad channel to accept cluster
 122   Float_t  fMinDist2;                          // Cuts on Minimal distance to study acceptance evaluation
 123   Float_t  fMinDist3;                          // One more cut on distance used for acceptance-efficiency study
 124   Double_t fTimeCutMin  ;                      // Remove clusters/cells with time smaller than this value, in ns
 125   Double_t fTimeCutMax  ;                      // Remove clusters/cells with time larger than this value, in ns
 126   Int_t    fNCellsCut ;                        // Accept for the analysis clusters with more than fNCellsCut cells
 127   Bool_t   fFillSSHistograms ;                 // Fill shower shape histograms
 128   Bool_t   fFillOnlySimpleSSHisto;   // Fill selected cluster histograms, selected SS histograms
 129   Bool_t   fFillWeightHistograms ;             // Fill weigth histograms
 130   Int_t    fNOriginHistograms;                 // Fill only NOriginHistograms of the 14 defined types
 131
 132   Float_t  fdEdxMin;                           // Max dEdx for electrons
 133   Float_t  fdEdxMax;                           // Min dEdx for electrons
 134   Float_t  fEOverPMin;                         // Max E/p for electrons, after dEdx cut
 135   Float_t  fEOverPMax;                         // Min E/p for electrons, after dEdx cut
 136
 137   Int_t    fAODParticle;                       // Select the type of particle to put in AODs for other analysis
 138
 139   //Histograms
 140   TH2F * fhdEdxvsE;                            //! matched track dEdx vs cluster E
 141   TH2F * fhdEdxvsP;                            //! matched track dEdx vs track P
 142   TH2F * fhEOverPvsE;                          //! matched track E cluster over P track vs cluster E, after dEdx cut
 143   TH2F * fhEOverPvsP;                          //! matched track E cluster over P track vs track P, after dEdx cut
 144
 145   TH2F * fhNCellsE[2];                         //! number of cells in cluster vs E
 146   TH2F * fhMaxCellDiffClusterE[2];             //! Fraction of energy carried by cell with maximum energy
 147   TH2F * fhTimeE[2];                           //! E vs Time of selected cluster
 148
 149   TH1F * fhE[2]    ;                           //! Number of identified electron vs energy
 150   TH1F * fhPt[2]   ;                           //! Number of identified electron vs transerse momentum
 151   TH2F * fhPhi[2]  ;                           //! Azimuthal angle of identified  electron vs transerse momentum
 152   TH2F * fhEta[2]  ;                           //! Pseudorapidity of identified  electron vs transerse momentum
 153   TH2F * fhEtaPhi[2]  ;                        //! Pseudorapidity vs Phi of identified  electron for transerse momentum > 0.5
 154   TH2F * fhEtaPhi05[2]  ;                      //! Pseudorapidity vs Phi of identified  electron for transerse momentum < 0.5
 155
 156   //Shower shape
 157
 158   TH2F * fhDispE[2];                           //! cluster dispersion vs E
 159   TH2F * fhLam0E[2];                           //! cluster lambda0 vs  E
 160   TH2F * fhLam1E[2];                           //! cluster lambda1 vs  E
 161
 162   TH2F * fhDispETRD[2];                        //! cluster dispersion vs E, SM covered by TRD
 163   TH2F * fhLam0ETRD[2];                        //! cluster lambda0 vs  E, SM covered by TRD
 164   TH2F * fhLam1ETRD[2];                        //! cluster lambda1 vs  E, SM covered by TRD
 165
 166   TH2F * fhNCellsLam0LowE[2];                  //! cluster N cells vs lambda0, E<2
 167   TH2F * fhNCellsLam0HighE[2];                 //! cluster N Cells vs lambda0, E>2
 168
 169   TH2F * fhEtaLam0LowE[2];                     //! cluster eta vs lambda0, E<2
 170   TH2F * fhPhiLam0LowE[2];                     //! cluster phi vs lambda0, E<2
 171   TH2F * fhEtaLam0HighE[2];                    //! cluster eta vs lambda0, E>2
 172   TH2F * fhPhiLam0HighE[2];                    //! cluster phi vs lambda0, E>2
 173
 174   TH2F * fhDispEtaE[2] ;                       //! shower dispersion in eta direction
 175   TH2F * fhDispPhiE[2] ;                       //! shower dispersion in phi direction
 176   TH2F * fhSumEtaE[2] ;                        //! shower dispersion in eta direction
 177   TH2F * fhSumPhiE[2] ;                        //! shower dispersion in phi direction
 178   TH2F * fhSumEtaPhiE[2] ;                     //! shower dispersion in eta and phi direction
 179   TH2F * fhDispEtaPhiDiffE[2] ;                //! shower dispersion eta - phi
 180   TH2F * fhSphericityE[2] ;                    //! shower sphericity in eta vs phi
 181   TH2F * fhDispEtaDispPhiEBin[2][5] ;          //! shower dispersion in eta direction vs phi direction for 5 E bins [0-2],[2-4],[4-6],[6-10],[> 10]
 182
 183   // Weight studies
 184
 185   TH2F * fhECellClusterRatio;                  //! e cell / e cluster vs e cluster for selected electrons
 186   TH2F * fhECellClusterLogRatio;               //! log (e cell / e cluster)  vs e cluster for selected electrons
 187   TH2F * fhEMaxCellClusterRatio;               //! e max cell / e cluster vs e cluster for selected electrons
 188   TH2F * fhEMaxCellClusterLogRatio;            //! log (e max cell / e cluster) vs e cluster for selected electrons
 189   TH2F * fhLambda0ForW0[14];                   //! L0 for 7 defined w0= 3, 3.5 ... 6 for selected electrons
 190   //TH2F * fhLambda1ForW0[14];                    //! L1 for 7 defined w0= 3, 3.5 ... 6 for selected electrons
 191
 192   //Fill MC dependent histograms, Origin of this cluster is ...
 193
 194   TH2F * fhMCDeltaE[2][10]  ;                  //! MC-Reco E distribution coming from MC particle
 195   TH2F * fhMC2E[2][10]  ;                      //! E distribution, Reco vs MC coming from MC particle
 196
 197   TH1F * fhMCE[2][10];                          //! Number of identified electron vs cluster energy coming from MC particle
 198   TH1F * fhMCPt[2][10];                         //! Number of identified electron vs cluster energy coming from MC particle
 199   TH2F * fhMCPhi[2][10];                        //! Phi of identified electron coming from MC particle
 200   TH2F * fhMCEta[2][10];                        //! eta of identified electron coming from MC particle
 201
 202   // Shower Shape MC
 203
 204   TH2F * fhMCELambda0[2][6] ;                   //! E vs Lambda0 from MC particle
 205
 206   TH2F * fhMCEDispEta[2][6] ;                   //! shower dispersion in eta direction from MC particle
 207   TH2F * fhMCEDispPhi[2][6] ;                   //! shower dispersion in phi direction from MC particle
 208   TH2F * fhMCESumEtaPhi[2][6] ;                 //! shower dispersion in eta vs phi direction from MC particle
 209   TH2F * fhMCEDispEtaPhiDiff[2][6] ;            //! shower dispersion in eta -phi direction from MC particle
 210   TH2F * fhMCESphericity[2][6] ;                //! shower sphericity, eta vs phi from MC particle
 211
 212   TH2F * fhMCElectronELambda0NoOverlap ;        //! E vs Lambda0 from MC electrons, no overlap
 213   TH2F * fhMCElectronELambda0TwoOverlap ;       //! E vs Lambda0 from MC electrons, 2 particles overlap
 214   TH2F * fhMCElectronELambda0NOverlap ;         //! E vs Lambda0 from MC electrons, N particles overlap
 215
 216   //Embedding
 217   TH2F * fhEmbeddedSignalFractionEnergy ;       //! Fraction of electron energy of embedded signal vs cluster energy
 218
 219   TH2F * fhEmbedElectronELambda0FullSignal ;    //!  Lambda0 vs E for embedded electrons with more than 90% of the cluster energy
 220   TH2F * fhEmbedElectronELambda0MostlySignal ;  //!  Lambda0 vs E for embedded electrons with 90%<fraction<50%
 221   TH2F * fhEmbedElectronELambda0MostlyBkg ;     //!  Lambda0 vs E for embedded electrons with 50%<fraction<10%
 222   TH2F * fhEmbedElectronELambda0FullBkg ;       //!  Lambda0 vs E for embedded electrons with less than 10% of the cluster energy
 223
 224   AliAnaElectron(              const AliAnaElectron & el) ; // cpy ctor
 225   AliAnaElectron & operator = (const AliAnaElectron & el) ; // cpy assignment
 226
 227   ClassDef(AliAnaElectron,4)
 228
 229 } ;
 230
 231
 232 #endif//ALIANAELECTRON_H
 233
 234
 235