ATO-98 Extended print. Bug. fix. Initilaization of the flags in AddCorrectionCompact...

[u/mrichter/AliRoot.git] / PWGHF / vertexingHF / AliAnalysisTaskSEDs.h

diff --git a/PWGHF/vertexingHF/AliAnalysisTaskSEDs.h b/PWGHF/vertexingHF/AliAnalysisTaskSEDs.h
index 7a01216600429c4a28007155674ffb9d56dd4f55..9530679a6fb4b1738060404a2eccb4c9b41ed5a4 100644 (file)
--- a/PWGHF/vertexingHF/AliAnalysisTaskSEDs.h
+++ b/PWGHF/vertexingHF/AliAnalysisTaskSEDs.h
@@ -31,7 +31,11 @@ class AliAnalysisTaskSEDs : public AliAnalysisTaskSE
   AliAnalysisTaskSEDs();
   AliAnalysisTaskSEDs(const char *name, AliRDHFCutsDstoKKpi* analysiscuts, Int_t fillNtuple=0);
   virtual ~AliAnalysisTaskSEDs();
-  void SetReadMC(Bool_t readMC=kTRUE){fReadMC=readMC;}
+  void SetReadMC(Bool_t readMC=kTRUE){fReadMC=readMC;} 
+  void SetWriteOnlySignalInNtuple(Bool_t opt=kTRUE){
+    if(fReadMC) fWriteOnlySignal=opt;
+    else AliError("fReadMC has to be kTRUE");
+  }
   void SetFillNtuple(Int_t fill=0){fFillNtuple=fill;}
   void SetMassRange(Double_t rang=0.4){fMassRange=rang;}
   void SetDoCutVarHistos(Bool_t opt=kTRUE) {fDoCutVarHistos=opt;}
@@ -62,8 +66,10 @@ class AliAnalysisTaskSEDs : public AliAnalysisTaskSE
   TH1F*   fHistNEvents;               //! hist. for No. of events  
   TH1F*   fChanHist[4];               //! hist. with KKpi and piKK candidates (sig,bkg,tot)
   TH1F*   fMassHist[4*kMaxPtBins];    //! hist. of mass spectra (sig,bkg,tot)
-  TH1F*   fMassHistPhi[3*kMaxPtBins];    //! hist. of mass spectra via phi (sig,bkg,tot)
-  TH1F*   fMassHistK0st[3*kMaxPtBins];    //! hist. of mass spectra via K0* (sig,bkg,tot)
+  TH1F*   fMassHistPhi[4*kMaxPtBins];    //! hist. of mass spectra via phi (sig,bkg,tot)
+  TH1F*   fMassHistK0st[4*kMaxPtBins];    //! hist. of mass spectra via K0* (sig,bkg,tot)
+  TH1F*   fMassHistKK[kMaxPtBins];    //! hist. of mass spectra of KK
+  TH1F*   fMassHistKpi[kMaxPtBins];    //! hist. of mass spectra of Kpi
   TH1F*   fCosPHist[4*kMaxPtBins];    //! hist. of cos pointing angle (sig,bkg,tot)
   TH1F*   fDLenHist[4*kMaxPtBins];    //! hist. of decay length (sig,bkg,tot)
   TH1F*   fSumd02Hist[4*kMaxPtBins];  //! hist. for sum d02 (Prod Cuts)
@@ -92,6 +98,7 @@ class AliAnalysisTaskSEDs : public AliAnalysisTaskSE
                                     
                                      
   Bool_t  fReadMC;                    //  flag for access to MC
+  Bool_t  fWriteOnlySignal;           //  flag to control ntuple writing in MC
   Bool_t  fDoCutVarHistos;            //  flag to create and fill histos with distributions of cut variables
   Bool_t  fUseSelectionBit;           // flag for useage of HasSelectionBit
   UChar_t fNPtBins;                   // number of Pt bins
@@ -103,7 +110,7 @@ class AliAnalysisTaskSEDs : public AliAnalysisTaskSE
   AliNormalizationCounter *fCounter;//!Counter for normalization
   AliRDHFCutsDstoKKpi *fAnalysisCuts; // Cuts for Analysis
   
-  ClassDef(AliAnalysisTaskSEDs,12);    //  AliAnalysisTaskSE for Ds mass spectra
+  ClassDef(AliAnalysisTaskSEDs,14);    //  AliAnalysisTaskSE for Ds mass spectra
 };
 
 #endif