Update (Andrea)

[u/mrichter/AliRoot.git] / PWG3 / vertexingHF / AliCFHeavyFlavourTaskMultiVarMultiStep.h

diff --git a/PWG3/vertexingHF/AliCFHeavyFlavourTaskMultiVarMultiStep.h b/PWG3/vertexingHF/AliCFHeavyFlavourTaskMultiVarMultiStep.h
index c6e84ed46619505d463bc6cd1419e23170ef85b4..eb711e14a2378939b464cbae562717b874514e04 100644 (file)
--- a/PWG3/vertexingHF/AliCFHeavyFlavourTaskMultiVarMultiStep.h
+++ b/PWG3/vertexingHF/AliCFHeavyFlavourTaskMultiVarMultiStep.h
@@ -15,6 +15,8 @@
  * provided "as is" without express or implied warranty.                  *
  **************************************************************************/
 
+/* $Id$ */ 
+
 //-----------------------------------------------------------------------
 // Class for HF corrections as a function of many variables and step 
 // Author : C. Zampolli, CERN
@@ -33,23 +35,26 @@ class AliAODRecoDecay;
 class AliAODRecoDecayHF2Prong;
 class AliAODMCParticle;
 class THnSparse;
+class AliRDHFCutsD0toKpi;
 
 class AliCFHeavyFlavourTaskMultiVarMultiStep : public AliAnalysisTaskSE {
   public:
 
   enum {
-    kStepGenerated       = 0,
-    kStepAcceptance      = 1,
-    kStepVertex          = 2,
-    kStepRefit           = 3,
-    kStepReconstructed   = 4,
-    kStepRecoAcceptance  = 5,
-    kStepRecoITSClusters = 6,
-    kStepRecoPPR         = 7
+    kStepGeneratedLimAcc = 0,
+    kStepGenerated       = 1,
+    kStepAcceptance      = 2,
+    kStepVertex          = 3,
+    kStepRefit           = 4,
+    kStepReconstructed   = 5,
+    kStepRecoAcceptance  = 6,
+    kStepRecoITSClusters = 7,
+    kStepRecoPPR         = 8,
+    kStepRecoPID         = 9
   };
 
   AliCFHeavyFlavourTaskMultiVarMultiStep();
-  AliCFHeavyFlavourTaskMultiVarMultiStep(const Char_t* name);
+  AliCFHeavyFlavourTaskMultiVarMultiStep(const Char_t* name, AliRDHFCutsD0toKpi* cuts);
   AliCFHeavyFlavourTaskMultiVarMultiStep& operator= (const AliCFHeavyFlavourTaskMultiVarMultiStep& c);
   AliCFHeavyFlavourTaskMultiVarMultiStep(const AliCFHeavyFlavourTaskMultiVarMultiStep& c);
   virtual ~AliCFHeavyFlavourTaskMultiVarMultiStep();
@@ -57,6 +62,8 @@ class AliCFHeavyFlavourTaskMultiVarMultiStep : public AliAnalysisTaskSE {
   // ANALYSIS FRAMEWORK STUFF to loop on data and fill output objects
   void     UserCreateOutputObjects();
   void     UserExec(Option_t *option);
+  void     Init();
+  void     LocalInit() {Init();}
   void     Terminate(Option_t *);
 
  // UNFOLDING
@@ -70,14 +77,25 @@ class AliCFHeavyFlavourTaskMultiVarMultiStep : public AliAnalysisTaskSE {
   AliCFManager * GetCFManager()                 {return fCFManager;} // get corr manager
 
   void     SetPDG(Int_t code) {fPDG = code; }     // defines the PDG code of searched HF
-  Double_t CosThetaStar(AliAODMCParticle* mcPart, AliAODMCParticle* mcPartDaughter0, AliAODMCParticle* mcPartDaughter1) const;  // returns cos(ThetaStar) of the D0 decay
-  Double_t CT(AliAODMCParticle* mcPart, AliAODMCParticle* mcPartDaughter0, AliAODMCParticle* mcPartDaughter1) const;            // returns cT of the D0 decay
+  Double_t CosThetaStar(const AliAODMCParticle* mcPart, const AliAODMCParticle* mcPartDaughter0, const AliAODMCParticle* mcPartDaughter1) const;  // returns cos(ThetaStar) of the D0 decay
+  Double_t CT(const AliAODMCParticle* mcPart, const AliAODMCParticle* mcPartDaughter0, const AliAODMCParticle* mcPartDaughter1) const;            // returns cT of the D0 decay
   void     SetFillFromGenerated(Bool_t flag) {fFillFromGenerated = flag;}
   Bool_t   GetFillFromGenerated() const {return fFillFromGenerated;}
-  Bool_t   GetGeneratedValuesFromMCParticle(AliAODMCParticle* mcPart, TClonesArray* mcArray, Double_t* vectorMC) const;
+  Bool_t   GetGeneratedValuesFromMCParticle(AliAODMCParticle* mcPart, const TClonesArray* mcArray, Double_t* vectorMC) const;
   void     SetMinITSClusters(Int_t minITSClusters) {fMinITSClusters = minITSClusters;}
   Int_t    GetMinITSClusters() const {return fMinITSClusters;}
-  Int_t    CheckOrigin(AliAODMCParticle* mcPart, TClonesArray* mcArray) const;
+  Int_t    CheckOrigin(const AliAODMCParticle* mcPart, const TClonesArray* mcArray) const;
+
+  void SetKeepD0fromB(Bool_t keepD0fromB) {fKeepD0fromB=keepD0fromB;}
+  Bool_t GetKeepD0fromB() const {return fKeepD0fromB;}
+  void SetKeepD0fromBOnly(Bool_t keepD0fromBOnly){ fKeepD0fromBOnly=keepD0fromBOnly; }
+  Bool_t GetKeepD0fromBOnly() const { return fKeepD0fromBOnly;}
+  void SetUseWeight(Bool_t useWeight){fUseWeight=useWeight;}
+  Bool_t GetUseWeight() const {return fUseWeight;}
+  Double_t GetWeight(Float_t pt);
+  Double_t DodNdptFit(Float_t pt, const Double_t* par) const;
+  void SetSign(Char_t isSign) {fSign = isSign;}
+  Char_t GetSign() const {return fSign;}
 
  protected:
   Int_t           fPDG;         //  PDG code of searched V0's
@@ -92,13 +110,21 @@ class AliCFHeavyFlavourTaskMultiVarMultiStep : public AliAnalysisTaskSE {
   Int_t fCountRecoAcc;          //  Reco particle found that satisfy cuts in requested acceptance
   Int_t fCountRecoITSClusters;  //  Reco particle found that satisfy cuts in n. of ITS clusters
   Int_t fCountRecoPPR;          //  Reco particle found that satisfy cuts in PPR
+  Int_t fCountRecoPID;          //  Reco particle found that satisfy cuts in PPR, and PID condition
   Int_t fEvents;                //  n. of events
   Bool_t fFillFromGenerated;    //  flag to indicate whether data container should be filled 
                                 //  with generated values also for reconstructed particles
   Int_t fMinITSClusters;        //  min n. of ITS clusters for RecoDecay
   Bool_t fAcceptanceUnf;        //  flag for unfolding before or after cuts.
+
+  Bool_t fKeepD0fromB;          // flag to consider also D0 coming from B
+  Bool_t fKeepD0fromBOnly;      // flag to consider _only_ D0 coming from B
+  AliRDHFCutsD0toKpi* fCuts;    // cuts
+  Bool_t fUseWeight;            // flag to decide whether to use weights != 1 when filling the container or not
+  Double_t fWeight;             // weight used to fill the container
+  Char_t fSign;                 // flag to decide wheter to keep D0 only (0), D0bar only (1), or both D0 and D0bar (2)
   
-  ClassDef(AliCFHeavyFlavourTaskMultiVarMultiStep,3); // class for HF corrections as a function of many variables
+  ClassDef(AliCFHeavyFlavourTaskMultiVarMultiStep,7); // class for HF corrections as a function of many variables
 };
 
 #endif