Merge branch 'feature-movesplit'

[u/mrichter/AliRoot.git] / PWGJE / FlavourJetTasks / AliAnalysisTaskSEDmesonsFilterCJ.h

diff --git a/PWGJE/FlavourJetTasks/AliAnalysisTaskSEDmesonsFilterCJ.h b/PWGJE/FlavourJetTasks/AliAnalysisTaskSEDmesonsFilterCJ.h
index eaae3ccf56900535a4c9dea873ba139817ecdbf1..19db5db6d7215d9c1f87ed1c828118ea246ea761 100644 (file)
--- a/PWGJE/FlavourJetTasks/AliAnalysisTaskSEDmesonsFilterCJ.h
+++ b/PWGJE/FlavourJetTasks/AliAnalysisTaskSEDmesonsFilterCJ.h
@@ -60,11 +60,17 @@ class AliAnalysisTaskSEDmesonsFilterCJ : public AliAnalysisTaskSE
   void   SetMC(Bool_t theMCon) { fUseMCInfo = theMCon; }
   Bool_t GetMC() const { return fUseMCInfo; }
   
+  // set usage of generated or reconstucted quantities (relevant for MC)
+  void SetUseReco(Bool_t useReco=kTRUE) { fUseReco= useReco;}
+  Bool_t GetUseReco() const {return fUseReco;}
+ 
   void SetMassLimits(Double_t range, Int_t pdg);
   void SetMassLimits(Double_t lowlimit, Double_t uplimit);
 
   // Array of D0 width for the Dstar
   Bool_t SetD0WidthForDStar(Int_t nptbins, Float_t *width);
+  
+  Float_t DeltaR(AliVParticle *p1, AliVParticle *p2) const;
 
  private :
   
@@ -72,6 +78,7 @@ class AliAnalysisTaskSEDmesonsFilterCJ : public AliAnalysisTaskSE
   AliAnalysisTaskSEDmesonsFilterCJ& operator=(const AliAnalysisTaskSEDmesonsFilterCJ& source); 
 
   Bool_t fUseMCInfo;               //  Use MC info
+  Bool_t fUseReco;                 // use reconstructed tracks when running on MC
 
   UInt_t  fCandidateType;          //  Dstar or D0
   TString fCandidateName;          //  Dstar or D0
@@ -83,16 +90,20 @@ class AliAnalysisTaskSEDmesonsFilterCJ : public AliAnalysisTaskSE
 
   TString fBranchName;             //  AOD branch name
   TList  *fOutput;                 //! user output
-//TList *fOutputCandidates;        //! output of array of candidates (kExchange)
 
   AliRDHFCuts *fCuts;              // Cuts 
   Double_t fMinMass;               //  mass lower limit histogram
   Double_t fMaxMass;               //  mass upper limit histogram
 
   TClonesArray *fCandidateArray;   //! contains candidates selected by AliRDHFCuts
-//TClonesArray *fIsSelectedArray;  //! contains result of IsSelected for candidates which pass the cuts (needed for D0)
-
-  ClassDef(AliAnalysisTaskSEDmesonsFilterCJ,1); // class for charm-jet correlations
+  TClonesArray *fSideBandArray;    //! contains candidates selected by AliRDHFCuts::IsSelected(kTracks), to be used for side bands (DStar case only!!)
+  //Histograms
+  TH2F* fhImpPar;                  //!
+  TH2F* fhImpParB;                 //!
+  TH1F* fhInvMassS;                //!
+  TH1F* fhInvMassB;                //!
+  
+  ClassDef(AliAnalysisTaskSEDmesonsFilterCJ,3); // class for charm-jet correlations
 };
 
 #endif