Removed old macro

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

diff --git a/PWG3/vertexingHF/AliAnalysisTaskSESignificance.h b/PWG3/vertexingHF/AliAnalysisTaskSESignificance.h
index a9015d1ef2fb8b6a12db53170b1c13ed56c2e3c8..5be7b892cb4ac588c2df4bee78911948cda630bc 100644 (file)
--- a/PWG3/vertexingHF/AliAnalysisTaskSESignificance.h
+++ b/PWG3/vertexingHF/AliAnalysisTaskSESignificance.h
@@ -29,13 +29,18 @@ class AliAnalysisTaskSESignificance : public AliAnalysisTaskSE
  
   virtual ~AliAnalysisTaskSESignificance();
 
+  Bool_t CheckConsistency();
   void SetReadMC(Bool_t readMC=kTRUE){fReadMC=readMC;}
   void SetMassLimits(Float_t range,Int_t pdg);
   void SetMassLimits(Float_t lowlimit, Float_t uplimit);
+  void SetNBins(Int_t nbins){fNBins=nbins;}
+  void SetFillWithPartAntiPartBoth(Int_t value){fPartOrAndAntiPart=value;}
   //void SetMultiVector(const AliMultiDimVector *MultiDimVec){fMultiDimVec->CopyStructure(MultiDimVec);}
-  Float_t GetUpperMassLimit(){return fUpmasslimit;}
-  Float_t GetLowerMassLimit(){return fLowmasslimit;}
-  
+  Float_t GetUpperMassLimit()const {return fUpmasslimit;}
+  Float_t GetLowerMassLimit()const {return fLowmasslimit;}
+  Int_t GetNBins()const {return fNBins;}
+  Int_t GetFillWithPartAntiPartBoth()const {return fPartOrAndAntiPart;}
+
   // Implementation of interface methods
   virtual void UserCreateOutputObjects();
   virtual void LocalInit();// {Init();}
@@ -50,12 +55,22 @@ class AliAnalysisTaskSESignificance : public AliAnalysisTaskSE
   Int_t GetSignalHistoIndex(Int_t iPtBin) const { return iPtBin*3+1;}
   Int_t GetBackgroundHistoIndex(Int_t iPtBin) const { return iPtBin*3+2;}
   Int_t GetLSHistoIndex(Int_t iPtBin)const { return iPtBin*5;}
- 
+
+  void CalculateInvMasses(AliAODRecoDecayHF* d,Double_t* &masses,Int_t& nmasses);
+
+  void FillDplus(AliAODRecoDecayHF* d,TClonesArray *arrayMC,Int_t index,Double_t* masses,Int_t isSel);
+  void FillD02p(AliAODRecoDecayHF* d,TClonesArray *arrayMC,Int_t index,Double_t* masses, Int_t isSel);
+  void FillDs(AliAODRecoDecayHF* d,TClonesArray *arrayMC,Int_t index,Double_t* masses,Int_t isSel);
+  void FillDstar(AliAODRecoDecayHF* d,TClonesArray *arrayMC,Int_t index,Double_t* masses,Int_t isSel);
+  void FillD04p(AliAODRecoDecayHF* d,TClonesArray *arrayMC,Int_t index,Double_t* masses,Int_t isSel);
+  void FillLambdac(AliAODRecoDecayHF* d,TClonesArray *arrayMC,Int_t index,Double_t* masses,Int_t isSel);
+
+
   enum {kMaxPtBins=5};
   enum {kMaxCutVar=5};
   enum {kMaxSteps=10};
   enum {kMaxNHist=500000};
-  enum {kDplustoKpipi,kD0toKpi,kDstartoKpipi,kDstoKKpi,kD0toKpipipi,kLambdactoPKpi};
+  enum {kDplustoKpipi,kD0toKpi,kDstartoKpipi,kDstoKKpi,kD0toKpipipi,kLambdactopKpi};
 
   TList   *fOutput; //! list send on output slot 0
   TList *fCutList; //Multidimvector container
@@ -71,8 +86,10 @@ class AliAnalysisTaskSESignificance : public AliAnalysisTaskSE
   Bool_t fReadMC;    //flag for access to MC
   Int_t fDecChannel; //decay channel identifier
   Int_t fSelectionlevel;//selection level: kALL,kTracks,kCandidate
-   
-  ClassDef(AliAnalysisTaskSESignificance,1); // AliAnalysisTaskSE for the MC association of heavy-flavour decay candidates
+  Int_t fNBins;  //number of bins in the mass histograms
+  Int_t fPartOrAndAntiPart;  //fill histograms with particle only (+1), antiparticle only (-1), both (0)
+
+  ClassDef(AliAnalysisTaskSESignificance,2); // AliAnalysisTaskSE for the MC association of heavy-flavour decay candidates
 };
 
 #endif