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