Cleanup the code. Fix memory leak. Now inherit from AliAnalysisTaskSE (Antoine, Phili...

[u/mrichter/AliRoot.git] / PWG3 / hfe / AliAnalysisTaskHFEpidQA.h

diff --git a/PWG3/hfe/AliAnalysisTaskHFEpidQA.h b/PWG3/hfe/AliAnalysisTaskHFEpidQA.h
index 4b2e6c63c50812c88cf7f640ed3ee634de6662ca..42aa892eb1ea4db41e7602c8fd41cafc34c7ad59 100644 (file)
--- a/PWG3/hfe/AliAnalysisTaskHFEpidQA.h
+++ b/PWG3/hfe/AliAnalysisTaskHFEpidQA.h
@@ -1,3 +1,6 @@
+#ifndef ALIANALYSISTASKHFEPIDQA_H
+#define ALIANALYSISTASKHFEPIDQA_H
+
 /**************************************************************************
 * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
 *                                                                        *
@@ -12,20 +15,24 @@
 * about the suitability of this software for any purpose. It is          *
 * provided "as is" without express or implied warranty.                  *
 **************************************************************************/
+
+/* $Id$ */ 
+
 //
 // Task for PID QA
 // Using AliHFEpidQA and AliHFEMCpidQA
 // More information can be found in the source file
 //
-#ifndef ALIANALYSISTASKHFEPIDQA_H
-#define ALIANALYSISTASKHFEPIDQA_H
-
 #ifndef ALIANALYSISTASKSE_H
 #include "AliAnalysisTaskSE.h"
 #endif
 
 class TH1;
 class TList;
+class TFile;
+
+class AliLog;
+class AliMCEvent;
 
 class AliHFEpidQA;
 
@@ -45,6 +52,8 @@ class AliAnalysisTaskHFEpidQA : public AliAnalysisTaskSE{
     void SetV0pidQA(Bool_t v0pidQA = kTRUE) { SetBit(kV0pidQA, v0pidQA); };
     void SetRecalculateTRDpid(Bool_t recal = kTRUE) { SetBit(kRecalculateTRDpid, recal); };
 
+    void SetNNref(TFile *f) { fNNref = f; };
+
   private:
     enum{
       kV0pidQA = BIT(22),
@@ -55,6 +64,7 @@ class AliAnalysisTaskHFEpidQA : public AliAnalysisTaskSE{
     AliHFEpidQA *fPIDqa;    //! The heart of the analysis  
     TList *fOutput;         //! Container for output histos
     TH1 *fEvents;           //! Number of Events
+    TFile  *fNNref;         //  reference file for NN
 
     ClassDef(AliAnalysisTaskHFEpidQA, 1)
 };