Merge remote-tracking branch 'origin/master' into flatdev

[u/mrichter/AliRoot.git] / PWGHF / hfe / AliHFEInclusiveSpectrum.h

diff --git a/PWGHF/hfe/AliHFEInclusiveSpectrum.h b/PWGHF/hfe/AliHFEInclusiveSpectrum.h
index 88fe13551d64b29c8a7653974934b4c4893e2ab1..5d4d6bdc7e3f625cb0bc1392ba7c4f0724f86bf8 100644 (file)
--- a/PWGHF/hfe/AliHFEInclusiveSpectrum.h
+++ b/PWGHF/hfe/AliHFEInclusiveSpectrum.h
@@ -43,15 +43,18 @@ class AliHFEInclusiveSpectrum : public AliHFECorrectSpectrumBase{
     ~AliHFEInclusiveSpectrum();
     
 
-    virtual Bool_t Init(const AliHFEcontainer *datahfecontainer, const AliHFEcontainer *mchfecontainer, const AliHFEcontainer *v0hfecontainer=0x0, const AliHFEcontainer */*bghfecontainer*/=0x0);
-    virtual Bool_t Correct(Bool_t subtractcontamination=kTRUE);
+    virtual Bool_t Init(const AliHFEcontainer *datahfecontainer, const AliHFEcontainer *mchfecontainer, const AliHFEcontainer */*bghfecontainer*/=0x0, const AliHFEcontainer *v0hfecontainer=0x0,AliCFContainer *photoniccontainerD=0x0);
+    virtual Bool_t Correct(Bool_t subtractcontamination=kTRUE,  Bool_t subtractphotonic=kFALSE);
    
     AliCFDataGrid *SubtractBackground();
+    AliCFDataGrid *SubtractPhotonicBackground();
     AliCFDataGrid *CorrectV0Efficiency(AliCFDataGrid* const bgsubpectrum = 0x0);
     AliCFDataGrid *CorrectParametrizedEfficiency(AliCFDataGrid* const bgsubpectrum = 0x0);
     THnSparse *Unfold(AliCFDataGrid* const bgsubpectrum = 0x0);
     AliCFDataGrid *CorrectForEfficiency(AliCFDataGrid* const bgsubpectrum = 0x0);
 
+    void SetMCNoCentrality() { fNoCentralitySelectionMC = kTRUE; };
+
     void WriteResults(const char *filename);
    
  private:
@@ -60,6 +63,8 @@ class AliHFEInclusiveSpectrum : public AliHFECorrectSpectrumBase{
     virtual void Copy(TObject &o) const;
  
     AliHFEInclusiveSpectrumQA *fQA; // QA
+
+    Bool_t fNoCentralitySelectionMC;  // Explicitly switch off centrality selection
    
     ClassDef(AliHFEInclusiveSpectrum, 1) 
 };