PWGHF/vertexingHF/AliCFVertexingHF3Prong.h

   1 #ifndef ALICFVERTEXINGHF3PRONG_H\r
   2 #define ALICFVERTEXINGHF3PRONG_H\r
   3 \r
   4 /* Copyright(c) 2007-2009, ALICE Experiment at CERN, All rights reserved. *\r
   5  * See cxx source for full Copyright notice                               */\r
   6 \r
   7 /* $Id$ */\r
   8 \r
   9 ///////////////////////////////////////////////////////////////////\r
  10 //                                                               //\r
  11 // Class to compute variables for correction framework           //  \r
  12 // for 3-body decays of D mesons (D+, Ds, Lc)                    //\r
  13 // in bins of cut variables                                      //\r
  14 // Origin:       Francesco Prino (prino@to.infn.it)              //\r
  15 //               Renu Bala       (bala@to.infn.it)               //\r
  16 //                                                               //\r
  17 ///////////////////////////////////////////////////////////////////\r
  18 \r
  19 #include "AliCFVertexingHF.h"\r
  20 #include "AliAODRecoDecayHF.h"\r
  21 #include "AliAODRecoDecayHF3Prong.h"\r
  22 \r
  23 class AliAODMCParticle;\r
  24 class TClonesArray;\r
  25 class AliCFVertexingHF;\r
  26 class AliESDtrack;\r
  27 class TDatabasePDG;\r
  28 \r
  29 class AliCFVertexingHF3Prong : public AliCFVertexingHF{\r
  30         public:\r
  31                 \r
  32   AliCFVertexingHF3Prong(Int_t decay);\r
  33   AliCFVertexingHF3Prong(TClonesArray *mcArray, UShort_t originDselection, Int_t decay);  \r
  34   virtual ~AliCFVertexingHF3Prong(){};\r
  35   \r
  36   \r
  37   \r
  38   Bool_t GetGeneratedValuesFromMCParticle(Double_t* /*vectorMC*/);\r
  39   Bool_t GetRecoValuesFromCandidate(Double_t* /*vectorReco*/ ) const;\r
  40   Bool_t CheckMCChannelDecay()const;\r
  41   \r
  42   Bool_t SetRecoCandidateParam(AliAODRecoDecayHF *recoCand);\r
  43   virtual void SetDecay3Prong(Int_t decay){fDecay=decay;}\r
  44  \r
  45   void SetGeneratedDsOption(Int_t opt) {fGenDsOption=opt;}\r
  46   Int_t GetGeneratedDsOption() const {return fGenDsOption;}\r
  47 \r
  48   enum EDsSel {kCountAllDsKKpi=0, kCountPhipi=1, kCountK0stK=2, kCountResonant=3, kCountNonResonant=4};\r
  49 \r
  50  protected:\r
  51   \r
  52   \r
  53   \r
  54  private:       \r
  55   AliCFVertexingHF3Prong(const AliCFVertexingHF3Prong& c);\r
  56   AliCFVertexingHF3Prong& operator= (const AliCFVertexingHF3Prong& other);\r
  57   \r
  58   Int_t fDecay;   // decay mode id\r
  59   Int_t fGenDsOption;  // option for selection Ds (see enum)\r
  60 \r
  61   ClassDef(AliCFVertexingHF3Prong, 2);\r
  62   \r
  63 };\r
  64 \r
  65 #endif\r