PWG3/vertexingHF/AliCFVertexingHF.h

   1 #ifndef ALICFVERTEXINGHF_H
   2 #define ALICFVERTEXINGHF_H
   3
   4
   5 /**************************************************************************
   6  * Copyright(c) 1998-2009, ALICE Experiment at CERN, All rights reserved. *
   7  *                                                                        *
   8  * Author: The ALICE Off-line Project.                                    *
   9  * Contributors are mentioned in the code where appropriate.              *
  10  *                                                                        *
  11  * Permission to use, copy, modify and distribute this software and its   *
  12  * documentation strictly for non-commercial purposes is hereby granted   *
  13  * without fee, provided that the above copyright notice appears in all   *
  14  * copies and that both the copyright notice and this permission notice   *
  15  * appear in the supporting documentation. The authors make no claims     *
  16  * about the suitability of this software for any purpose. It is          *
  17  * provided "as is" without express or implied warranty.                  *
  18  **************************************************************************/
  19
  20 //-----------------------------------------------------------------------
  21 // Class for HF corrections as a function of many variables and step
  22 // Author : C. Zampolli, CERN
  23 // D. Caffarri, Univ & INFN Padova caffarri@pd.infn.it
  24 // Base class for HF Unfolding - agrelli@uu.nl
  25 //-----------------------------------------------------------------------
  26
  27 #include "AliCFContainer.h"
  28 #include "AliAODRecoDecayHF.h"
  29
  30 class TH1I;
  31 class TParticle ;
  32 class TFile ;
  33 class TClonesArray ;
  34 class AliAODMCParticle;
  35 class AliAODMCHeader;
  36 class AliAODEvent;
  37 class THnSparse;
  38 class TClonesArray;
  39 class AliESDtrackCuts;
  40
  41
  42
  43 class AliCFVertexingHF : public TObject {
  44         public:
  45
  46         AliCFVertexingHF() ;
  47         AliCFVertexingHF(TClonesArray *mcArray, UShort_t originDselection);
  48         AliCFVertexingHF(const AliCFVertexingHF& c);
  49         AliCFVertexingHF& operator= (const AliCFVertexingHF& c);
  50
  51         virtual ~AliCFVertexingHF();
  52
  53         virtual Bool_t GetGeneratedValuesFromMCParticle(Double_t* /*vectorMC*/) {return kFALSE;}
  54         virtual Bool_t GetRecoValuesFromCandidate(Double_t* /*vectorReco*/) const {return kFALSE;}
  55         virtual Bool_t CheckMCChannelDecay() const {return kFALSE;}
  56
  57         void   SetFillFromGenerated(Bool_t flag) {fFillFromGenerated = flag;}
  58         Bool_t GetFillFromGenerated() const {return fFillFromGenerated;}
  59
  60         void  SetNVar(Int_t nVar) {fNVar = nVar;}
  61
  62         void  SetRecoPrimVertex (Double_t zPrimVertex) {fzPrimVertex = zPrimVertex;}
  63         void  SetMCPrimaryVertex (Double_t zMCVertex){fzMCVertex = zMCVertex;}
  64         void  SetMCLabel (Int_t mcLabel) {fmcLabel = mcLabel;}
  65         Int_t GetMCLabel () const {return  fmcLabel;}
  66
  67         void   SetMCCandidateParam(Int_t label);
  68
  69
  70         Int_t  MCcquarkCounting(AliAODMCParticle* mcPart) const;
  71         Bool_t CheckMCPartFamily(AliAODMCParticle */*mcPart*/, TClonesArray */*mcArray*/) const;
  72         //      Int_t  CheckOrigin(AliAODMCParticle* mcPart) const;
  73         Int_t  CheckOrigin() const;
  74         Bool_t CheckMCDaughters() const;
  75         Bool_t FillMCContainer(Double_t *containerInputMC);
  76         Bool_t FillRecoContainer(Double_t *containerInput);
  77         Bool_t MCAcceptanceStep() const;
  78         Bool_t MCRefitStep(AliAODEvent *aodEvent, AliESDtrackCuts *trackCuts) const;
  79         Bool_t RecoStep();
  80
  81         Double_t GetEtaProng(Int_t iProng) const;
  82         Double_t GetPtProng(Int_t iProng) const;
  83
  84         Double_t GetPtCand() const {return fRecoCandidate->Pt();}
  85         Double_t GetYCand() const {return fRecoCandidate->Y();}
  86
  87         Bool_t RecoAcceptStep(AliESDtrackCuts *trackCuts) const;
  88
  89         Bool_t FillUnfoldingMatrix(Double_t *fill) const;
  90
  91         void SetNProngs(Int_t nProngs){fProngs = nProngs;}
  92         void SetDselection(UShort_t originDselection);
  93
  94         protected:
  95
  96         TClonesArray      *fmcArray;               //mcArray candidate
  97         AliAODRecoDecayHF *fRecoCandidate;         // Reconstructed HF candidate
  98         AliAODMCParticle  *fmcPartCandidate;
  99         Int_t fNDaughters;
 100         Int_t fNVar;                // get Number of variables for the container from the channel decay
 101         Double_t fzPrimVertex;       //Reco z primary vertex
 102         Double_t fzMCVertex;         //MC z primary vertex
 103
 104         Bool_t fFillFromGenerated;   //  flag to indicate whether data container should be filled
 105         UShort_t fOriginDselection;      // flag to select D0 origins. 0 Only from charm 1 only from beauty 2 both from charm and beauty
 106
 107         Bool_t fKeepDfromB;           //flag for the feed down from b quark decay.
 108         Bool_t fKeepDfromBOnly;       // flag to keep only the charm particles that comes from beauty decays
 109         Int_t fmcLabel;              // results of the MatchToMC()
 110         Int_t fProngs;               // n. of prongs
 111         ClassDef(AliCFVertexingHF, 1);
 112
 113 };
 114
 115 #endif