PWG4/PartCorrDep/AliAnaBtag.h

   1 #ifndef ALIANABTAG_H\r
   2 #define ALIANABTAG_H\r
   3 /* Copyright(c) 1998-2010, ALICE Experiment at CERN, All rights reserved. *\r
   4  * See cxx source for full Copyright notice     */\r
   5 /* $Id:  $ */\r
   6 \r
   7 //_________________________________________________________________________\r
   8 //\r
   9 // Class for the electron identification and B-tagging.\r
  10 // Clusters from EMCAL matched to tracks are selected \r
  11 // and kept in the AOD. Few histograms produced.\r
  12 //\r
  13 \r
  14 //-- Author T.R.P-R.Aronsson, J.Klay\r
  15 \r
  16 // --- ROOT system ---\r
  17 class TH2F ;\r
  18 class TString ;\r
  19 class TNtuple ;\r
  20 class TH3F;\r
  21 \r
  22 // --- ANALYSIS system ---\r
  23 #include "AliAnaPartCorrBaseClass.h"\r
  24 \r
  25 class AliAODMCParticle;\r
  26 class AliCaloTrackReader;\r
  27 class AliAODTrack;\r
  28 class TList ;\r
  29 \r
  30 class AliAnaBtag : public AliAnaPartCorrBaseClass {\r
  31 \r
  32 public: \r
  33   AliAnaBtag() ; // default ctor\r
  34   virtual ~AliAnaBtag() ; //virtual dtor\r
  35 private:\r
  36         AliAnaBtag(const AliAnaBtag & g) ; // cpy ctor\r
  37         AliAnaBtag & operator = (const AliAnaBtag & g) ;//cpy assignment\r
  38         \r
  39 public:\r
  40         \r
  41   TList *  GetCreateOutputObjects();\r
  42 \r
  43   //Main functions\r
  44   void Init();\r
  45   void MakeAnalysisFillAOD()  ;\r
  46   void MakeAnalysisFillHistograms() ; \r
  47   \r
  48   //B-tagging\r
  49   Int_t GetDVMBtag(AliAODTrack * tr);//Main tagger\r
  50 \r
  51   //Temporary local method to get DCA\r
  52   Bool_t GetDCA(const AliAODTrack* tr,Double_t imp[2], Double_t cov[3]);\r
  53   Bool_t PhotonicV0(Int_t trackId); //check with V0 list\r
  54 \r
  55   //used with the jet tracks to tag bjets\r
  56   Bool_t CheckIfBjet(const AliAODTrack* track);\r
  57   Bool_t IsMcBJet(Double_t x, Double_t y);\r
  58   Bool_t IsMcDJet(Double_t x, Double_t y);\r
  59 \r
  60   void Print(const Option_t * opt)const;\r
  61   \r
  62   TString GetCalorimeter()   const {return fCalorimeter ; }\r
  63   Double_t GetpOverEmin()   const {return fpOverEmin ; }\r
  64   Double_t GetpOverEmax()   const {return fpOverEmax ; }\r
  65 \r
  66   Double_t GetDrCut() const { return fDrCut; }\r
  67   Double_t GetPairDcaCut() const { return fPairDcaCut; }\r
  68   Double_t GetDecayLenCut() const { return fDecayLenCut; }\r
  69   Double_t GetImpactCut() const { return fImpactCut; }\r
  70   Double_t GetAssocPtCut() const { return fAssocPtCut; }\r
  71   Double_t GetMassCut() const { return fMassCut; }\r
  72   Double_t GetSdcaCut() const { return fSdcaCut; }\r
  73   Int_t    GetITSCut() const { return fITSCut; }\r
  74   Int_t    GetNTagTrackCut() const { return fNTagTrkCut; }\r
  75   Double_t GetIPSigCut() const { return fIPSigCut; }\r
  76   Double_t GetMinClusEne() const { return fMinClusEne; }\r
  77 \r
  78   void SetCalorimeter(TString det)    {fCalorimeter = det ; }\r
  79   void SetpOverEmin(Double_t min)     {fpOverEmin = min ; }\r
  80   void SetpOverEmax(Double_t max)     {fpOverEmax = max ; }\r
  81   void SetResidualCut(Double_t cut)     {fResidualCut = cut ; }\r
  82 \r
  83   void SetDrCut(Double_t dr)  { fDrCut = dr; }\r
  84   void SetPairDcaCut(Double_t pdca) { fPairDcaCut = pdca; }\r
  85   void SetDecayLenCut(Double_t dlen) { fDecayLenCut = dlen; }\r
  86   void SetImpactCut(Double_t imp) { fImpactCut = imp; }\r
  87   void SetAssocPtCut(Double_t pt) { fAssocPtCut = pt; }\r
  88   void SetMassCut(Double_t mass) { fMassCut = mass; }\r
  89   void SetSdcaCut(Double_t sdca) { fSdcaCut = sdca; }\r
  90   void SetITSCut(Int_t its) { fITSCut = its; }\r
  91   void SetNTagTrackCut(Int_t ntr) { fNTagTrkCut = ntr; }\r
  92   void SetIPSigCut(Double_t ips) { fIPSigCut = ips; }\r
  93   void SetMinClusEne(Double_t ene) { fMinClusEne = ene; }\r
  94 \r
  95   void InitParameters();\r
  96   void Terminate(TList * outputList);\r
  97          \r
  98   private:\r
  99   //For DVM B-tag method\r
 100   Double_t ComputeSignDca(AliAODTrack *track, AliAODTrack *track2, float cut1, double pdcacut);\r
 101 \r
 102   //Int_t GetMCSource(Int_t mctag);\r
 103   AliAODMCParticle* GetMCParticle(Int_t part);\r
 104 \r
 105 \r
 106   private:\r
 107   TString  fCalorimeter;  //! Which detector? EMCAL or PHOS\r
 108   Double_t fpOverEmin;    //! Minimum p/E value for Electrons\r
 109   Double_t fpOverEmax;    //! Maximum p/E value for Electrons\r
 110   Double_t fResidualCut;  //! Track-cluster matching distance\r
 111   Double_t fMinClusEne;   //! Min clus energy for matching\r
 112 \r
 113   //DVM B-tagging\r
 114   Double_t fDrCut;       //max dR\r
 115   Double_t fPairDcaCut;  //max pair-DCA\r
 116   Double_t fDecayLenCut; //max 3d-decaylength\r
 117   Double_t fImpactCut;   //max track impact param\r
 118   Double_t fAssocPtCut;  //min associated pt\r
 119   Double_t fMassCut;     //min Minv cut\r
 120   Double_t fSdcaCut;     //min signDca\r
 121   Int_t   fITSCut;       //min ITS hits (both)\r
 122   //IP Sig B-tagging\r
 123   Int_t    fNTagTrkCut;  //min number of tracks required for IP sig tag\r
 124   Double_t fIPSigCut;    //min IP significance cut\r
 125 \r
 126   Double_t fJetEtaCut;   //max eta for jets\r
 127   Double_t fJetPhiMin;   //min phi for jets\r
 128   Double_t fJetPhiMax;   //max phi for jets\r
 129 \r
 130   \r
 131   \r
 132   //Histograms\r
 133   TH1F * fhEmcalElectrons;              //All electrons, as id:ed by EMCAL\r
 134   TH1F * fhTRDElectrons;                //Electrons from TRD\r
 135   TH1F * fhTPCElectrons;                //Electrons from TPC\r
 136   TH1F * fhDVM1;                        //initial b-tag dvm1\r
 137   TH1F * fhDVM2;                        //initial b-tag dvm2 \r
 138   TH2F * fhJets;                        //All jets 2d\r
 139   TH2F * fhJetsAllEtaPhi;               //Eta phi for all jets\r
 140   TH2F * fhJetsLeadingBElectronEtaPhi;  //deta dphi btw leading jet and bele\r
 141   TH2F * fhDVM1EtaPhi;                  //eta phi for b-electrons\r
 142   TH2F * fhBJetElectronDetaDphi;        //eta phi for jet with b-electrons\r
 143   TH1F * fhClusterEnergy;               //cluster E of EMCAL\r
 144   TH1F * fhTestalle;                                    //Temp histo for EMCAL cluster energy\r
 145   TH1F * fhResidual;                    //Residuals from trackmatching\r
 146 \r
 147   //Analysis of electrons\r
 148   TH2F * fhElectrons;\r
 149   //Analysis for tracks from esd/aod\r
 150   TH2F * fhTracks;\r
 151 \r
 152 \r
 153   ClassDef(AliAnaBtag,2)\r
 154 \r
 155 } ;\r
 156  \r
 157 \r
 158 #endif//ALIANABTAG_H\r
 159 \r
 160 \r
 161 \r