latest updates to AliAnaElectron for b-jet tagging and non-photonic electron finding

[u/mrichter/AliRoot.git] / PWG4 / PartCorrDep / AliAnaElectron.h

#ifndef ALIANAELECTRON_H\r
#define ALIANAELECTRON_H\r
/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *\r
 * See cxx source for full Copyright notice     */\r
/* $Id:  $ */\r
\r
//_________________________________________________________________________\r
//\r
// Class for the electron identification.\r
// Clusters from EMCAL matched to tracks are selected \r
// and kept in the AOD. Few histograms produced.\r
//\r
\r
//-- Author: J.L. Klay (Cal Poly)\r
\r
// --- ROOT system ---\r
class TH2F ;\r
class TString ;\r
class TNtuple ;\r
class TH3F;\r
\r
// --- ANALYSIS system ---\r
#include "AliAnaPartCorrBaseClass.h"\r
\r
class AliCaloTrackReader;\r
class AliAODTrack;\r
class TList ;\r
\r
class AliAnaElectron : public AliAnaPartCorrBaseClass {\r
\r
public: \r
\r
  AliAnaElectron() ; // default ctor\r
  AliAnaElectron(const AliAnaElectron & g) ; // cpy ctor\r
  AliAnaElectron & operator = (const AliAnaElectron & g) ;//cpy assignment\r
  virtual ~AliAnaElectron() ; //virtual dtor\r
  \r
  TList *  GetCreateOutputObjects();\r
\r
  void Init();\r
\r
  void MakeAnalysisFillAOD()  ;\r
  \r
  void MakeAnalysisFillHistograms() ; \r
  \r
  //B-tagging\r
  Int_t GetDVMBtag(AliAODTrack * tr); //returns # tracks from secvtx\r
\r
  //Temporary local method to get DCA because AliAODTrack is stupid\r
  Bool_t GetDCA(const AliAODTrack* tr,Double_t imp[2], Double_t cov[3]);\r
\r
  Bool_t IsItPhotonic(const AliAODPWG4Particle* part); //check with track list\r
  Bool_t IsItPhotonic2(const AliAODPWG4Particle* part); //check with V0 list\r
\r
  //check if track has been flagged as a non-photonic or DVM electron\r
  //used with the jet tracks to tag bjets\r
  Bool_t CheckTrack(const AliAODTrack* track,const char* type);  \r
\r
  void Print(const Option_t * opt)const;\r
  \r
  TString GetCalorimeter()   const {return fCalorimeter ; }\r
  Double_t GetpOverEmin()   const {return fpOverEmin ; }\r
  Double_t GetpOverEmax()   const {return fpOverEmax ; }\r
  Bool_t GetWriteNtuple()   const {return fWriteNtuple ; }\r
\r
  Double_t GetDrCut() const { return fDrCut; }\r
  Double_t GetPairDcaCut() const { return fPairDcaCut; }\r
  Double_t GetDecayLenCut() const { return fDecayLenCut; }\r
  Double_t GetImpactCut() const { return fImpactCut; }\r
  Double_t GetAssocPtCut() const { return fAssocPtCut; }\r
  Double_t GetMassCut() const { return fMassCut; }\r
  Double_t GetSdcaCut() const { return fSdcaCut; }\r
  Int_t    GetITSCut() const { return fITSCut; }\r
  Int_t    GetNTagTrackCut() const { return fNTagTrkCut; }\r
  Double_t GetIPSigCut() const { return fIPSigCut; }\r
\r
  void SetCalorimeter(TString det)    {fCalorimeter = det ; }\r
  void SetpOverEmin(Double_t min)     {fpOverEmin = min ; }\r
  void SetpOverEmax(Double_t max)     {fpOverEmax = max ; }\r
  void SetResidualCut(Double_t cut)     {fResidualCut = cut ; }\r
  void SetWriteNtuple(Bool_t val)     {fWriteNtuple = val ; }\r
\r
  void SetDrCut(Double_t dr)  { fDrCut = dr; }\r
  void SetPairDcaCut(Double_t pdca) { fPairDcaCut = pdca; }\r
  void SetDecayLenCut(Double_t dlen) { fDecayLenCut = dlen; }\r
  void SetImpactCut(Double_t imp) { fImpactCut = imp; }\r
  void SetAssocPtCut(Double_t pt) { fAssocPtCut = pt; }\r
  void SetMassCut(Double_t mass) { fMassCut = mass; }\r
  void SetSdcaCut(Double_t sdca) { fSdcaCut = sdca; }\r
  void SetITSCut(Int_t its) { fITSCut = its; }\r
  void SetNTagTrackCut(Int_t ntr) { fNTagTrkCut = ntr; }\r
  void SetIPSigCut(Double_t ips) { fIPSigCut = ips; }\r
\r
  void InitParameters();\r
\r
  void Terminate(TList * outputList);\r
  void ReadHistograms(TList * outputList); //Fill histograms with\r
					   //histograms in ouput list,\r
					   //needed in Terminate.            \r
  private:\r
  //For DVM B-tag method\r
  Double_t ComputeSignDca(AliAODTrack *track, AliAODTrack *track2 , float cut1);\r
  //the 2 following functions are internal methods of the b-tagging\r
  //based on transverse impact parameter\r
  Double_t GetIPSignificance(AliAODTrack *tr, Double_t jetPhi);\r
  void GetImpactParamVect(Double_t Pxy[2], Double_t t[2], Double_t Vxy[2], Double_t ip[2]);\r
\r
  private:\r
  TString  fCalorimeter;  //! Which detector? EMCAL or PHOS\r
  Double_t fpOverEmin;    //! Minimum p/E value for Electrons\r
  Double_t fpOverEmax;    //! Maximum p/E value for Electrons\r
  Double_t fResidualCut;  //! Track-cluster matching distance\r
\r
  //DVM B-tagging\r
  Double_t fDrCut;       //max dR\r
  Double_t fPairDcaCut;  //max pair-DCA\r
  Double_t fDecayLenCut; //max 3d-decaylength\r
  Double_t fImpactCut;   //max track impact param\r
  Double_t fAssocPtCut;  //min associated pt\r
  Double_t fMassCut;     //min Minv cut\r
  Double_t fSdcaCut;     //min signDca\r
  Int_t   fITSCut;       //min ITS hits (both)\r
  //IP Sig B-tagging\r
  Int_t    fNTagTrkCut;  //min number of tracks required for IP sig tag\r
  Double_t fIPSigCut;    //min IP significance cut\r
\r
  Bool_t  fWriteNtuple; //flag for filling ntuple or not\r
\r
  TNtuple* fEleNtuple; //! testing ntuple\r
\r
  //matching checks   \r
  TH1F *fh1pOverE;     //! p/E for track-cluster matches\r
  TH1F *fh1dR;         //! distance between projected track and cluster\r
  TH2F *fh2EledEdx;    //! dE/dx vs. momentum for electron candidates\r
  TH2F *fh2MatchdEdx;  //! dE/dx vs. momentum for all matches\r
  TH2F *fh2dEtadPhi;   //! DeltaEta vs. DeltaPhi of all track/cluster\r
		       //! pairs\r
  TH2F *fh2dEtadPhiMatched;   //! DeltaEta vs. DeltaPhi of matched\r
				//! track/cluster pairs\r
  TH2F *fh2dEtadPhiUnmatched;   //! DeltaEta vs. DeltaPhi of unmatched track/cluster pairs\r
\r
  TH2F* fh2TrackPVsClusterE;     //!track momentum vs. cluster energy\r
  TH2F* fh2TrackPtVsClusterE;    //!track pt vs. cluster energy\r
  TH2F* fh2TrackPhiVsClusterPhi; //!track phi vs. cluster phi\r
  TH2F* fh2TrackEtaVsClusterEta; //!track eta vs. cluster eta\r
\r
  //Photonic Electron checks\r
  TH1F* fh1OpeningAngle; //!opening angle between pairs of photon candidates\r
  TH1F* fh1MinvPhoton;   //!invariant mass distribution of electron pairs\r
\r
  //Reconstructed\r
  TH1F * fhPtElectron;  //! Number of identified electron vs transverse momentum \r
  TH2F * fhPhiElectron; //! Azimuthal angle of identified  electron vs transverse momentum \r
  TH2F * fhEtaElectron; //! Pseudorapidity of identified  electron vs tranvserse momentum \r
\r
  TH1F * fhPtNPE;  //! Number of non-photonic electron vs transverse momentum \r
  TH2F * fhPhiNPE; //! Azimuthal angle of non-photonic electron vs transverse momentum \r
  TH2F * fhEtaNPE; //! Pseudorapidity of non-photonic electron vs tranvserse momentum \r
\r
  TH1F * fhPtPE;  //! Number of photonic electron vs transverse momentum \r
  TH2F * fhPhiPE; //! Azimuthal angle of photonic electron vs transverse momentum \r
  TH2F * fhEtaPE; //! Pseudorapidity of photonic electron vs tranvserse momentum \r
\r
  TH1F * fhPtConversion;  //! Number of conversion electron vs transverse momentum \r
  TH2F * fhPhiConversion; //! Azimuthal angle of conversion  electron vs transverse momentum \r
  TH2F * fhEtaConversion; //! Pseudorapidity of conversion electron vs tranvserse momentum \r
\r
  TH1F * fhPtBottom;  //! Number of bottom electron vs transverse momentum \r
  TH2F * fhPhiBottom; //! Azimuthal angle of bottom  electron vs transverse momentum \r
  TH2F * fhEtaBottom; //! Pseudorapidity of bottom electron vs tranvserse momentum \r
\r
  TH1F * fhPtCharm;  //! Number of charm electron vs transverse momentum \r
  TH2F * fhPhiCharm; //! Azimuthal angle of charm  electron vs transverse momentum \r
  TH2F * fhEtaCharm; //! Pseudorapidity of charm electron vs tranvserse momentum \r
\r
  TH1F * fhPtCFromB;  //! Number of charm from bottom electron vs transverse momentum \r
  TH2F * fhPhiCFromB; //! Azimuthal angle of charm from bottom electron vs transverse momentum \r
  TH2F * fhEtaCFromB; //! Pseudorapidity of charm from bottom electron vs tranvserse momentum \r
\r
  TH1F * fhPtDalitz;  //! Number of dalitz electron vs transverse momentum \r
  TH2F * fhPhiDalitz; //! Azimuthal angle of dalitz  electron vs transverse momentum \r
  TH2F * fhEtaDalitz; //! Pseudorapidity of dalitz electron vs tranvserse momentum \r
\r
  TH1F * fhPtWDecay;  //! Number of W-boson electron vs transverse momentum \r
  TH2F * fhPhiWDecay; //! Azimuthal angle of W-boson  electron vs transverse momentum \r
  TH2F * fhEtaWDecay; //! Pseudorapidity of W-boson electron vs tranvserse momentum \r
  		\r
  TH1F * fhPtZDecay;  //! Number of Z-boson electron vs transverse momentum \r
  TH2F * fhPhiZDecay; //! Azimuthal angle of Z-boson  electron vs transverse momentum \r
  TH2F * fhEtaZDecay; //! Pseudorapidity of Z-boson electron vs tranvserse momentum \r
\r
  TH1F * fhPtAll;  //! Number of all electron vs transverse momentum \r
  TH2F * fhPhiAll; //! Azimuthal angle of all  electron vs transverse momentum \r
  TH2F * fhEtaAll; //! Pseudorapidity of all electron vs tranvserse momentum \r
\r
  TH1F * fhPtUnknown;  //! Number of unknown electron vs transverse momentum\r
  TH2F * fhPhiUnknown; //! Azimuthal angle of unknown  electron vs transverse momentum\r
  TH2F * fhEtaUnknown; //! Pseudorapidity of unknown electron vs tranvserse momentum\r
\r
  TH1F * fhPtMisidentified;  //! Number of misidentified electron vs transverse momentum\r
  TH2F * fhPhiMisidentified; //! Azimuthal angle of misidentified electron vs transverse momentum\r
  TH2F * fhEtaMisidentified; //! Pseudorapidity of misidentified electron vs tranvserse momentum \r
\r
  TH1F* fhPtHadron;     //!Pt distribution of reco charged hadrons\r
			//!(pi,k,p) in EMCAL acceptance\r
  TH1F* fhPtEleTrkDet;  //!Pt distribution of reco electrons using\r
			//!pid info from tracking detectors only in\r
			//!EMCAL acceptance\r
  //event QA\r
  TH1F * fhImpactXY;    //! impact parameter of all tracks to primary vertex\r
  TH1F * fhRefMult;     //! refmult (sep 14)\r
  TH1F * fhRefMult2;    //! refmult2 (sep 14)\r
\r
  //DVM B-tagging\r
  TH2F * fhDVMBtagCut1; //! DVM B-tagging result for cut1 (minv>1.0)\r
  TH2F * fhDVMBtagCut2; //! DVM B-tagging result for cut2 (minv>1.5)\r
  TH2F * fhDVMBtagCut3; //! DVM B-tagging result for cut3 (minv>1.8)\r
  TH2F * fhDVMBtagQA1;  //! DVM B-tagging : QA of pairDca vs decaylength\r
  TH2F * fhDVMBtagQA2;  //! DVM B-tagging : QA of signDca vs mass\r
  TH1F * fhDVMBtagQA3;  //! DVM B-tagging : QA (sep 14)\r
  TH1F * fhDVMBtagQA4;  //! DVM B-tagging : QA (sep 14)\r
  TH1F * fhDVMBtagQA5;  //! DVM B-tagging : QA (sep 14)\r
  //IPSig B-tagging\r
  TH1F * fhIPSigBtagQA1; //! IPSig B-tagging : QA of # tag tracks\r
  TH1F * fhIPSigBtagQA2; //! IPSig B-tagging : QA of IP sig\r
\r
  //B-Jet histograms\r
  TH2F* fhJetType;       //! How many of each tag were found vs jet pt\r
  TH2F* fhBJetXsiFF;     //! B-tagged jet FF with xsi = log(pt_Jet/pt_Track)\r
  TH2F* fhBJetPtFF;      //! B-tagged jet FF with pt_Track\r
  TH2F* fhBJetEtaPhi;    //! B-tagged jet eta-phi distribution\r
  TH2F* fhNonBJetXsiFF;  //! Non b-tagged jet FF with xsi = log(pt_Jet/pt_Track)\r
  TH2F* fhNonBJetPtFF;   //! Non b-tagged jet FF with pt_Track\r
  TH2F* fhNonBJetEtaPhi; //! Non b-tagged jet eta-phi distribution\r
\r
  //MC\r
  TNtuple *fMCEleNtuple; //! Ntuple of MC electrons\r
  TH1F* fhPtMCHadron;    //! Pt distribution of MC charged hadrons (pi,k,p) in EMCAL acceptance\r
  TH1F* fhPtMCBottom;    //! Pt distribution of MC bottom electrons in EMCAL\r
  TH1F* fhPtMCCharm;     //! Pt distribution of MC charm electrons in EMCAL \r
  TH1F* fhPtMCCFromB;    //! Pt distribution of MC charm from bottom ele in EMCAL\r
  TH1F* fhPtMCConversion;//! Pt distribution of MC conversion electrons in EMCAL \r
  TH1F* fhPtMCDalitz;    //! Pt distribution of MC Dalitz electrons in EMCAL\r
  TH1F* fhPtMCWDecay;    //! Pt distribution of MC W decay electrons in EMCAL\r
  TH1F* fhPtMCZDecay;    //! Pt distribution of MC Z decay electrons in EMCAL\r
  TH1F* fhPtMCUnknown;   //! Pt distribution of MC unknown electrons in EMCAL\r
\r
  ClassDef(AliAnaElectron,6)\r
\r
} ;\r
 \r
\r
#endif//ALIANAELECTRON_H\r
\r
\r
\r