PWG3/vertexingHF/AliAODRecoDecayHF.h

   1 #ifndef ALIAODRECODECAYHF_H
   2 #define ALIAODRECODECAYHF_H
   3 /* Copyright(c) 1998-2006, ALICE Experiment at CERN, All rights reserved. *
   4  * See cxx source for full Copyright notice                               */
   5
   6 //***********************************************************
   7 // Class AliAODRecoDecayHF
   8 // base class for AOD reconstructed heavy-flavour decays
   9 // Author: A.Dainese, andrea.dainese@lnl.infn.it
  10 //***********************************************************
  11
  12 #include <TRef.h>
  13 #include <TList.h>
  14 #include "AliAODTrack.h"
  15 #include "AliAODRecoDecay.h"
  16
  17 class AliAODEvent;
  18 class AliRDHFCuts;
  19 class AliKFParticle;
  20
  21 class AliAODRecoDecayHF : public AliAODRecoDecay {
  22
  23  public:
  24
  25   AliAODRecoDecayHF();
  26   AliAODRecoDecayHF(AliAODVertex *vtx2,Int_t nprongs,Short_t charge,
  27                     Double_t *px,Double_t *py,Double_t *pz,
  28                     Double_t *d0,Double_t *d0err);
  29   AliAODRecoDecayHF(AliAODVertex *vtx2,Int_t nprongs,Short_t charge,
  30                     Double_t *d0,Double_t *d0err);
  31   AliAODRecoDecayHF(Double_t vtx1[3],Double_t vtx2[3],
  32                     Int_t nprongs,Short_t charge,
  33                     Double_t *px,Double_t *py,Double_t *pz,Double_t *d0);
  34   virtual ~AliAODRecoDecayHF();
  35
  36   AliAODRecoDecayHF(const AliAODRecoDecayHF& source);
  37   AliAODRecoDecayHF& operator=(const AliAODRecoDecayHF& source);
  38
  39
  40   // primary vertex
  41   void SetPrimaryVtxRef(TObject *vtx) { fEventPrimaryVtx = vtx; }
  42   AliAODVertex* GetPrimaryVtxRef() const { return (AliAODVertex*)(fEventPrimaryVtx.GetObject()); }
  43   void SetOwnPrimaryVtx(const AliAODVertex *vtx) { UnsetOwnPrimaryVtx(); fOwnPrimaryVtx = new AliAODVertex(*vtx);}
  44   void CheckOwnPrimaryVtx() const
  45     {if(!fOwnPrimaryVtx) printf("fOwnPrimaryVtx not set"); return;}
  46   AliAODVertex* GetOwnPrimaryVtx() const {return fOwnPrimaryVtx;}
  47   void GetOwnPrimaryVtx(Double_t vtx[3]) const
  48     {CheckOwnPrimaryVtx();fOwnPrimaryVtx->GetPosition(vtx);}
  49   void UnsetOwnPrimaryVtx() {if(fOwnPrimaryVtx) {delete fOwnPrimaryVtx; fOwnPrimaryVtx=0;} return;}
  50   AliAODVertex* GetPrimaryVtx() const { return (GetOwnPrimaryVtx() ? GetOwnPrimaryVtx() : GetPrimaryVtxRef()); }
  51   AliAODVertex* RemoveDaughtersFromPrimaryVtx(AliAODEvent *aod);
  52
  53
  54   // kinematics & topology
  55   Double_t DecayLength2() const
  56     { return AliAODRecoDecay::DecayLength2(GetPrimaryVtx());}
  57   Double_t DecayLength() const
  58     { return AliAODRecoDecay::DecayLength(GetPrimaryVtx());}
  59   Double_t DecayLengthError() const
  60     { return AliAODRecoDecay::DecayLengthError(GetPrimaryVtx());}
  61   Double_t NormalizedDecayLength() const
  62     { return AliAODRecoDecay::NormalizedDecayLength(GetPrimaryVtx());}
  63   Double_t NormalizedDecayLength2() const
  64     { return AliAODRecoDecay::NormalizedDecayLength2(GetPrimaryVtx());}
  65   Double_t DecayLengthXY() const
  66     { return AliAODRecoDecay::DecayLengthXY(GetPrimaryVtx());}
  67   Double_t DecayLengthXYError() const
  68     { return AliAODRecoDecay::DecayLengthXYError(GetPrimaryVtx());}
  69   Double_t NormalizedDecayLengthXY() const
  70     { return AliAODRecoDecay::NormalizedDecayLengthXY(GetPrimaryVtx());}
  71   Double_t Ct(UInt_t pdg) const
  72     { return AliAODRecoDecay::Ct(pdg,GetPrimaryVtx());}
  73   Double_t CosPointingAngle() const
  74     { return AliAODRecoDecay::CosPointingAngle(GetPrimaryVtx());}
  75   Double_t CosPointingAngleXY() const
  76     { return AliAODRecoDecay::CosPointingAngleXY(GetPrimaryVtx());}
  77   Double_t ImpParXY() const
  78     { return AliAODRecoDecay::ImpParXY(GetPrimaryVtx());}
  79   Double_t QtProngFlightLine(Int_t ip) const
  80     { return AliAODRecoDecay::QtProngFlightLine(ip,GetPrimaryVtx());}
  81   Double_t QlProngFlightLine(Int_t ip) const
  82     { return AliAODRecoDecay::QlProngFlightLine(ip,GetPrimaryVtx());}
  83
  84   // prongs
  85   Double_t Getd0errProng(Int_t ip) const {return fd0err[ip];}
  86   Double_t Normalizedd0Prong(Int_t ip) const
  87     {return Getd0Prong(ip)/Getd0errProng(ip);}
  88
  89   void SetProngIDs(Int_t nIDs,UShort_t *id);
  90   UShort_t GetProngID(Int_t ip) const
  91     {if(fProngID) {return fProngID[ip];} else {return 9999;}}
  92
  93   // ITS clustermap for daughters
  94   Bool_t DaughterHasPointOnITSLayer(Int_t dg,Int_t l) const;
  95
  96   // check if it is like-sign
  97   Bool_t IsLikeSign() const;
  98
  99   // list of cuts
 100   void SetListOfCutsRef(TObject *obj) {fListOfCuts=obj;}
 101   TList *GetListOfCuts() const {return (TList*)(fListOfCuts.GetObject());}
 102   AliRDHFCuts *GetCuts(const char* name) const;
 103
 104   // vertexing KF:
 105   AliKFParticle *ApplyVertexingKF(Int_t *iprongs,Int_t nprongs,Int_t *pdgs,
 106                                  Bool_t topoCostraint,Double_t bzkG,
 107                                  Double_t *mass) const;
 108
 109   // misalign
 110   void Misalign(TString misal="null");
 111
 112  protected:
 113
 114   AliAODVertex *fOwnPrimaryVtx; // primary vertex for this candidate
 115   TRef          fEventPrimaryVtx; // ref to primary vertex of the event
 116   TRef          fListOfCuts;  // ref to the list of analysis cuts
 117   Double_t     *fd0err;  //[fNProngs] error on prongs rphi impact param [cm]
 118   UShort_t     *fProngID;  //[fNProngs] track ID of daughters
 119
 120   ClassDef(AliAODRecoDecayHF,4)  // base class for AOD reconstructed heavy-flavour decays
 121 };
 122
 123 inline void AliAODRecoDecayHF::SetProngIDs(Int_t nIDs,UShort_t *id)
 124 {
 125   if(nIDs!=GetNProngs()) {
 126     printf("Wrong number of IDs, must be nProngs\n");
 127     return;
 128   }
 129   if(fProngID) delete [] fProngID;
 130   fProngID = new UShort_t[nIDs];
 131   for(Int_t i=0;i<nIDs;i++)
 132     fProngID[i] = id[i];
 133   return;
 134 }
 135
 136 inline Bool_t AliAODRecoDecayHF::IsLikeSign() const
 137 {
 138   // check if it is like-sign
 139
 140   Int_t ndg=GetNDaughters();
 141   if(!ndg) {
 142     printf("Daughters not available\n");
 143     return kFALSE;
 144   }
 145   Int_t chargeDg0 = ((AliAODTrack*)GetDaughter(0))->Charge();
 146
 147   for(Int_t i=1; i<ndg; i++) {
 148     if(chargeDg0!=((AliAODTrack*)GetDaughter(i))->Charge()) return kFALSE;
 149   }
 150
 151   return kTRUE;
 152 }
 153
 154 inline AliRDHFCuts *AliAODRecoDecayHF::GetCuts(const char* name) const
 155 {
 156   // returns the analysis cuts
 157
 158   TList *list = GetListOfCuts();
 159   if(!list) return 0;
 160
 161
 162   return (AliRDHFCuts*)list->FindObject(name);
 163 }
 164
 165 inline Bool_t AliAODRecoDecayHF::DaughterHasPointOnITSLayer(Int_t dg,Int_t l) const
 166 {
 167   // ITS clustermap for daughters
 168
 169   if(l<0 || l>5) {
 170     printf("ERROR: layer has to be in the range 0-5\n");
 171     return kFALSE;
 172   }
 173   AliAODTrack *t = (AliAODTrack*)GetDaughter(dg);
 174   if(!t) return kFALSE;
 175
 176   return TESTBIT(t->GetITSClusterMap(),l);
 177 }
 178
 179 #endif
 180