Bug fixes.

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

diff --git a/PWG3/vertexingHF/AliAnalysisTaskSED0Mass.h b/PWG3/vertexingHF/AliAnalysisTaskSED0Mass.h
index 2c713b33b2f6a718f15355155ac8e896528fb638..c203a702cb7ba57d46a4178c98ac5497a2e36983 100644 (file)
--- a/PWG3/vertexingHF/AliAnalysisTaskSED0Mass.h
+++ b/PWG3/vertexingHF/AliAnalysisTaskSED0Mass.h
@@ -20,7 +20,9 @@
 
 #include "AliAnalysisTaskSE.h"
 #include "AliRDHFCutsD0toKpi.h"
+#include "AliNormalizationCounter.h"
 
+class AliAODEvent;
 
 class AliAnalysisTaskSED0Mass : public AliAnalysisTaskSE
 {
@@ -43,29 +45,36 @@ class AliAnalysisTaskSED0Mass : public AliAnalysisTaskSE
 
   void SetReadMC(Bool_t readMC=kFALSE){fReadMC=readMC;}
   void SetCutOnDistr(Bool_t cutondistr=kFALSE){fCutOnDistr=cutondistr;}
+  void SetUsePid4Distr(Bool_t usepid=kTRUE){fUsePid4Distr=usepid;}
+  void SetFillOnlyD0D0bar(Int_t flagfill){fFillOnlyD0D0bar=flagfill;}
+
+  Bool_t GetCutOnDistr() const {return fCutOnDistr;}
+  Bool_t GetUsePid4Distr() const {return fUsePid4Distr;}
+  Int_t  GetFillOnlyD0D0bar() const {return fFillOnlyD0D0bar;}
 
  private:
 
   AliAnalysisTaskSED0Mass(const AliAnalysisTaskSED0Mass &source);
   AliAnalysisTaskSED0Mass& operator=(const AliAnalysisTaskSED0Mass& source); 
-  void     FillMassHists(AliAODRecoDecayHF2Prong *part, TClonesArray *arrMC, AliRDHFCutsD0toKpi *cuts, TList *listout);
-  void     FillVarHists(AliAODRecoDecayHF2Prong *part, TClonesArray *arrMC, AliRDHFCutsD0toKpi *cuts, TList *listout);
-  TList    *fOutputMass; //! list send on output slot 1
-  TList    *fDistr;       //! list send on output slot 2
-  TH1F     *fNentries;    //! histogram with number of events on output slot 3
-  TList    *fChecks;       //! list send on output slot 4
-  TList* fCutList; //!
-  AliRDHFCutsD0toKpi *fCuts;  // Cuts - sent to output slot 5
-  Int_t    fArray;        //   can be D0 or Like Sign candidates
-  Bool_t   fReadMC;       // flag for MC array: kTRUE = read it, kFALSE = do not read it
-  Bool_t    fCutOnDistr;   // Flag to decide if apply cut also on distributions: 0 no cuts, 1 looser cuts, 2 tighter cuts 
-
-  Int_t    fTotPosPairs[5];     //
-  Int_t    fTotNegPairs[5];     // 
-  Double_t fLsNormalization;  //  normalization
-
-
-  ClassDef(AliAnalysisTaskSED0Mass,6); // AliAnalysisTaskSE for the MC association of heavy-flavour decay candidates
+  void     FillMassHists(AliAODEvent *aodev,AliAODRecoDecayHF2Prong *part, TClonesArray *arrMC, AliRDHFCutsD0toKpi *cuts, TList *listout);
+  void     FillVarHists(AliAODEvent *aodev,AliAODRecoDecayHF2Prong *part, TClonesArray *arrMC, AliRDHFCutsD0toKpi *cuts, TList *listout);
+  AliAODVertex* GetPrimaryVtxSkipped(AliAODEvent *aodev,AliAODRecoDecayHF2Prong *d);
+
+  TList    *fOutputMass;          //! list send on output slot 1
+  TList    *fDistr;               //! list send on output slot 2
+  TH1F     *fNentries;            //! histogram with number of events on output slot 3
+  AliRDHFCutsD0toKpi *fCuts;      //  Cuts - sent to output slot 4
+  Int_t     fArray;               //  can be D0 or Like Sign candidates
+  Bool_t    fReadMC;              //  flag for MC array: kTRUE = read it, kFALSE = do not read it
+  Bool_t    fCutOnDistr;          //  flag to decide if apply cut also on distributions: 0 no cuts, 1 looser cuts, 2 tighter cuts 
+  Bool_t    fUsePid4Distr;        //  flag to use the particle identification to fill the signal histograms of distributions. It has effect only with fReadMC=kFALSE
+  AliNormalizationCounter *fCounter;//!AliNormalizationCounter on output slot 6
+  Int_t     fNPtBins;             //  number of pt bins
+  Double_t  fLsNormalization;     //  normalization
+  Int_t     fFillOnlyD0D0bar;     // flag to fill mass histogram with D0/D0bar only (0 = fill with both, 1 = fill with D0 only, 2 = fill with D0bar only)
+
+  ClassDef(AliAnalysisTaskSED0Mass,11); // AliAnalysisTaskSE for the MC association of heavy-flavour decay candidates
 };
 
 #endif
+