PWG3/vertexingHF/AliAODRecoCascadeHF.h

   1 #ifndef ALIAODRECOCASCADEHF_H
   2 #define ALIAODRECOCASCADEHF_H
   3 /* Copyright(c) 1998-2008, ALICE Experiment at CERN, All rights reserved. *
   4  * See cxx source for full Copyright notice                               */
   5
   6 //***********************************************************
   7 // Class AliAODRecoCascadeHF
   8 // base class for AOD reconstructed heavy-flavour cascade decays
   9 // (D*->D0pi, ...)
  10 // The convention is: prong 0 is the bachelor, prong 1 is the "V0"
  11 //
  12 // Author: X-M. Zhang, zhangxm@iopp.ccnu.edu.cn
  13 //***********************************************************
  14
  15 #include <TRef.h>
  16 #include <TRefArray.h>
  17 #include <TClonesArray.h>
  18 #include <TClass.h>
  19 #include "AliAODVertex.h"
  20 #include "AliAODv0.h"
  21 #include "AliAODRecoDecayHF2Prong.h"
  22
  23 class AliAODRecoCascadeHF : public AliAODRecoDecayHF2Prong {
  24
  25  public:
  26
  27   AliAODRecoCascadeHF();
  28   AliAODRecoCascadeHF(AliAODVertex *vtx2, Short_t charge,
  29                       Double_t *px, Double_t *py, Double_t *pz,
  30                       Double_t *d0, Double_t *d0err, Double_t dca);
  31   AliAODRecoCascadeHF(AliAODVertex *vtx2, Short_t charge,
  32                       Double_t *d0, Double_t *d0err, Double_t dca);
  33   virtual ~AliAODRecoCascadeHF();
  34
  35   AliAODRecoCascadeHF(const AliAODRecoCascadeHF& source);
  36   AliAODRecoCascadeHF& operator=(const AliAODRecoCascadeHF& source);
  37
  38   // 2prong (D0 for Dstar)
  39   AliAODRecoDecayHF2Prong* Get2Prong() const {
  40     if ( ! ((AliAODRecoDecay*)GetDaughter(1))->IsA()->InheritsFrom("AliAODRecoDecayHF2Prong") ){
  41       AliWarning("Object is not of type AliAODRecoDecayHF2Prong");
  42       return 0;
  43     }
  44     return (AliAODRecoDecayHF2Prong*)GetDaughter(1);
  45   }
  46
  47   // Bachelor (soft pion for Dstar)
  48   AliAODTrack* GetBachelor() const {return (AliAODTrack*)GetDaughter(0);}
  49
  50   // v0 (Ks or Lambda for Lambda_c)
  51   AliAODv0* Getv0() const {
  52     if ( ! ((AliAODRecoDecay*)GetDaughter(1))->IsA()->InheritsFrom("AliAODv0") ){
  53        AliWarning("Object is not of type v0");
  54        return 0;
  55       }
  56     return (AliAODv0*)GetDaughter(1);
  57     }
  58
  59   // Get v0 positive track
  60   AliAODTrack* Getv0PositiveTrack() const { return  (AliAODTrack*)Getv0()->GetDaughter(0);  }
  61   // Get v0 negative track
  62   AliAODTrack* Getv0NegativeTrack() const { return  (AliAODTrack*)Getv0()->GetDaughter(1);  }
  63
  64   // D*->D0pi, D0->Kpi
  65   Double_t EDstar() const {return E(413);}
  66   Double_t YDstar() const {return Y(413);}
  67   Bool_t   SelectDstar(const Double_t *cutsDstar,const Double_t *cutsD0,Bool_t testD0=kTRUE) const;
  68   Double_t InvMassD0() const {return (Charge()>0 ? Get2Prong()->InvMassD0() : Get2Prong()->InvMassD0bar());}
  69   Double_t InvMassDstarKpipi() const;
  70   Double_t DeltaInvMass() const {return (InvMassDstarKpipi()-InvMassD0());}
  71   Double_t AngleD0dkpPisoft() const;
  72
  73   // Lc invariant mass
  74   Double_t InvMassLctoK0sP() const {
  75     UInt_t pdg[2]={2212,310}; return InvMass(2,pdg);
  76   }
  77   Double_t InvMassLctoLambdaPi() const {
  78     UInt_t pdg[2]={211,3122}; return InvMass(2,pdg);
  79   }
  80   Bool_t SelectLctoV0(const Double_t *cutsLctoV0, Bool_t okLck0sp, Bool_t okLcLpi) const;
  81
  82   Int_t MatchToMC(Int_t pdgabs,Int_t pdgabs2prong,
  83                   Int_t *pdgDg,Int_t *pdgDg2prong,
  84                   TClonesArray *mcArray) const;
  85
  86  protected:
  87
  88   ClassDef(AliAODRecoCascadeHF, 2); // heavy-flavour cascade class
  89 };
  90
  91 #endif