Allow fill histograms for AODs case, remnat of the ESD/AOD different frame

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

#ifndef ALIANABTAG_H\r
#define ALIANABTAG_H\r
/* Copyright(c) 1998-2010, 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 and B-tagging.\r
// Clusters from EMCAL matched to tracks are selected \r
// and kept in the AOD. Few histograms produced.\r
//\r
\r
//-- Author T.R.P-R.Aronsson\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 AliAODMCParticle;\r
class AliCaloTrackReader;\r
class AliAODTrack;\r
class TList ;\r
\r
class AliAnaBtag : public AliAnaPartCorrBaseClass {\r
\r
public: \r
  AliAnaBtag() ; // default ctor\r
  virtual ~AliAnaBtag() ; //virtual dtor\r
private:\r
	AliAnaBtag(const AliAnaBtag & g) ; // cpy ctor\r
	AliAnaBtag & operator = (const AliAnaBtag & g) ;//cpy assignment\r
	\r
public:\r
	\r
  TList *  GetCreateOutputObjects();\r
\r
  //Main functions\r
  void Init();\r
  void MakeAnalysisFillAOD()  ;\r
  void MakeAnalysisFillHistograms() ; \r
  \r
  //B-tagging\r
  Int_t GetDVMBtag(AliAODTrack * tr, Int_t &pairs, Int_t &start, Int_t &stop);//Main tagger\r
\r
  //Temporary local method to get DCA\r
  Bool_t GetDCA(const AliAODTrack* tr,Double_t imp[2], Double_t cov[3]);\r
  Bool_t PhotonicV0(Int_t trackId); //check with V0 list\r
\r
  //used with the jet tracks to tag bjets\r
  Bool_t CheckIfBjet(const AliAODTrack* track);\r
  Bool_t IsMcBJet(Double_t x, Double_t y);\r
  Bool_t IsMcDJet(Double_t x, Double_t y);\r
\r
\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
\r
  void SetWriteNtuple(Int_t w) { fWriteNtuple = w; }\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
\r
  void InitParameters();\r
  void Terminate(TList * outputList);\r
         \r
  private:\r
  //For DVM B-tag method\r
  Double_t ComputeSignDca(AliAODTrack *tr, AliAODTrack *tr2 , Double_t &masscut, Double_t &pdcacut, Double_t &massphoton, Double_t &decay);\r
\r
  AliAODMCParticle* GetMCParticle(Int_t part);\r
\r
\r
  private:\r
  //NTuples!\r
  Int_t fWriteNtuple;    //Will always be no, but might be set to yes in config file.\r
  TNtuple * electrons;   //Electrons\r
  TNtuple * pairs;       //Pairs to the electrons\r
  TNtuple * events;\r
  Int_t fEventNumber;    // For Ntuple to label events (starts at 0)\r
  Int_t fNElec;\r
  Int_t fNElecEv;\r
  Int_t fNPair;\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
  Double_t fJetEtaCut;   //max eta for jets\r
  Double_t fJetPhiMin;   //min phi for jets\r
  Double_t fJetPhiMax;   //max phi for jets\r
\r
  \r
  \r
  //Histograms\r
  TH1F * fhEmcalElectrons;              //All electrons, as id:ed by EMCAL\r
  TH1F * fhTRDElectrons;                //Electrons from TRD\r
  TH1F * fhTPCElectrons;                //Electrons from TPC\r
  TH1F * fhEmcalMCE;                    //All electrons, as id:ed by EMCAL MC\r
  TH1F * fhTRDMCE;                      //Electrons from TRD MC\r
  TH1F * fhTPCMCE;                      //Electrons from TPC MC\r
  TH1F * fhEmcalMCEFake;                //All electrons, as id:ed by EMCAL MC, fake\r
  TH1F * fhTRDMCEFake;                  //Electrons from TRD MC, fake\r
  TH1F * fhTPCMCEFake;                  //Electrons from TPC MC, fake\r
  TH1F * fhEmcalMCP;                    //Pions, as id:ed by EMCAL\r
  TH1F * fhTRDMCP;                      //Pions from TRD\r
  TH1F * fhTPCMCP;                      //Pions from TPC\r
  TH1F * fhEmcalMCK;                    //Kaons from EMCAL\r
  TH1F * fhTRDMCK;                      //Kaons from TRD\r
  TH1F * fhTPCMCK;                      //Kaons from TPC\r
  TH1F * fhSpecies;                     //PDG of id:ed electrons\r
  TH1F * fhDVM1;                        //initial b-tag dvm1\r
  TH1F * fhDVM2;                        //initial b-tag dvm2 \r
  TH1F * fhNVTX;                        //NVtx of all btags \r
  TH1F * fhNVTXMC;                        //NVtx of MC btags \r
  TH2F * fhJets;                        //All jets 2d\r
  TH2F * fhJetsAllEtaPhi;               //Eta phi for all jets\r
  TH2F * fhJetsLeadingBElectronEtaPhi;  //deta dphi btw leading jet and bele\r
  TH2F * fhDVM1EtaPhi;                  //eta phi for b-electrons\r
  TH2F * fhBJetElectronDetaDphi;        //eta phi for jet with b-electrons\r
  TH2F * fhClusterMap;                  //2D eta-phi of EMCAL clusters\r
  TH1F * fhClusterEnergy;               //cluster E of EMCAL\r
  TH1F * fhTestalle;					//Temp histo for EMCAL cluster energy\r
  TH1F * fhResidual;                    //Residuals from trackmatching\r
  TH1F * fhPairPt;                      //Pairs\r
\r
\r
  //Analysis of electrons\r
  TH2F * fhElectrons;\r
  //Analysis for tracks from esd/aod\r
  TH2F * fhTracks;\r
\r
\r
  ClassDef(AliAnaBtag,2)\r
\r
} ;\r
 \r
\r
#endif//ALIANABTAG_H\r
\r
\r
\r