1 #ifndef ALICFHEAVYFLAVOURTASKMULTIVARMULTISTEP_H
2 #define ALICFHEAVYFLAVOURTASKMULTIVARMULTISTEP_H
3 /**************************************************************************
4 * Copyright(c) 1998-2009, ALICE Experiment at CERN, All rights reserved. *
6 * Author: The ALICE Off-line Project. *
7 * Contributors are mentioned in the code where appropriate. *
9 * Permission to use, copy, modify and distribute this software and its *
10 * documentation strictly for non-commercial purposes is hereby granted *
11 * without fee, provided that the above copyright notice appears in all *
12 * copies and that both the copyright notice and this permission notice *
13 * appear in the supporting documentation. The authors make no claims *
14 * about the suitability of this software for any purpose. It is *
15 * provided "as is" without express or implied warranty. *
16 **************************************************************************/
18 //-----------------------------------------------------------------------
19 // Class for HF corrections as a function of many variables and step
20 // Author : C. Zampolli, CERN
21 // Base class for HF Unfolding - agrelli@uu.nl
22 //-----------------------------------------------------------------------
25 #include "AliAnalysisTaskSE.h"
32 class AliAODRecoDecay;
33 class AliAODRecoDecayHF2Prong;
34 class AliAODMCParticle;
37 class AliCFHeavyFlavourTaskMultiVarMultiStep : public AliAnalysisTaskSE {
45 kStepReconstructed = 4,
46 kStepRecoAcceptance = 5,
47 kStepRecoITSClusters = 6,
51 AliCFHeavyFlavourTaskMultiVarMultiStep();
52 AliCFHeavyFlavourTaskMultiVarMultiStep(const Char_t* name);
53 AliCFHeavyFlavourTaskMultiVarMultiStep& operator= (const AliCFHeavyFlavourTaskMultiVarMultiStep& c);
54 AliCFHeavyFlavourTaskMultiVarMultiStep(const AliCFHeavyFlavourTaskMultiVarMultiStep& c);
55 virtual ~AliCFHeavyFlavourTaskMultiVarMultiStep();
57 // ANALYSIS FRAMEWORK STUFF to loop on data and fill output objects
58 void UserCreateOutputObjects();
59 void UserExec(Option_t *option);
60 void Terminate(Option_t *);
63 void SetCorrelationMatrix(THnSparse* h) {fCorrelation=h;}
64 void SetAcceptanceUnf(Bool_t AcceptanceUnf) {fAcceptanceUnf = AcceptanceUnf;}
65 Bool_t GetAcceptanceUnf() const {return fAcceptanceUnf;}
68 // CORRECTION FRAMEWORK RELATED FUNCTIONS
69 void SetCFManager(AliCFManager* io) {fCFManager = io;} // global correction manager
70 AliCFManager * GetCFManager() {return fCFManager;} // get corr manager
72 void SetPDG(Int_t code) {fPDG = code; } // defines the PDG code of searched HF
73 Double_t CosThetaStar(AliAODMCParticle* mcPart, AliAODMCParticle* mcPartDaughter0, AliAODMCParticle* mcPartDaughter1) const; // returns cos(ThetaStar) of the D0 decay
74 Double_t CT(AliAODMCParticle* mcPart, AliAODMCParticle* mcPartDaughter0, AliAODMCParticle* mcPartDaughter1) const; // returns cT of the D0 decay
75 void SetFillFromGenerated(Bool_t flag) {fFillFromGenerated = flag;}
76 Bool_t GetFillFromGenerated() const {return fFillFromGenerated;}
77 Bool_t GetGeneratedValuesFromMCParticle(AliAODMCParticle* mcPart, TClonesArray* mcArray, Double_t* vectorMC) const;
78 void SetMinITSClusters(Int_t minITSClusters) {fMinITSClusters = minITSClusters;}
79 Int_t GetMinITSClusters() const {return fMinITSClusters;}
80 Int_t CheckOrigin(AliAODMCParticle* mcPart, TClonesArray* mcArray) const;
82 void SetKeepD0fromB(Bool_t keepD0fromB){fKeepD0fromB=keepD0fromB;}
85 Int_t fPDG; // PDG code of searched V0's
86 AliCFManager *fCFManager; // pointer to the CF manager
87 TH1I *fHistEventsProcessed; //! simple histo for monitoring the number of events processed
88 THnSparse* fCorrelation; // response matrix for unfolding
89 Int_t fCountMC; // MC particle found
90 Int_t fCountAcc; // MC particle found that satisfy acceptance cuts
91 Int_t fCountVertex; // Reco particle found that satisfy vertex constrained
92 Int_t fCountRefit; // Reco particle found that satisfy kTPCrefit and kITSrefit
93 Int_t fCountReco; // Reco particle found that satisfy cuts
94 Int_t fCountRecoAcc; // Reco particle found that satisfy cuts in requested acceptance
95 Int_t fCountRecoITSClusters; // Reco particle found that satisfy cuts in n. of ITS clusters
96 Int_t fCountRecoPPR; // Reco particle found that satisfy cuts in PPR
97 Int_t fEvents; // n. of events
98 Bool_t fFillFromGenerated; // flag to indicate whether data container should be filled
99 // with generated values also for reconstructed particles
100 Int_t fMinITSClusters; // min n. of ITS clusters for RecoDecay
101 Bool_t fAcceptanceUnf; // flag for unfolding before or after cuts.
103 Bool_t fKeepD0fromB; // flag to consider also D0 coming from B
105 ClassDef(AliCFHeavyFlavourTaskMultiVarMultiStep,3); // class for HF corrections as a function of many variables