Add event selection depending on high energy clusters existance in EMCal to speed...

[u/mrichter/AliRoot.git] / PWGDQ / dielectron / AliDielectron.h

diff --git a/PWGDQ/dielectron/AliDielectron.h b/PWGDQ/dielectron/AliDielectron.h
index 6ac00143cef454b4194fbf2344c7c9cbc4ee6827..829a9ede7bcc71fd48f0e55d8c584519c2fabbb0 100644 (file)
--- a/PWGDQ/dielectron/AliDielectron.h
+++ b/PWGDQ/dielectron/AliDielectron.h
@@ -23,6 +23,7 @@
 #include <AliKFParticle.h>
 
 #include "AliDielectronHistos.h"
+#include "AliDielectronHF.h"
 
 class AliEventplane;
 class AliVEvent;
@@ -80,7 +81,9 @@ public:
       static_cast<TObjArray*>(fPairCandidates->UncheckedAt(i)):0;}
 
   TObjArray** GetPairArraysPointer() { return &fPairCandidates; }
-  
+  void SetHistogramArray(AliDielectronHF * const histoarray) { fHistoArray=histoarray; }
+  const TObjArray * GetHistogramArray() const { return fHistoArray?fHistoArray->GetHistArray():0x0; }
+
   void SetHistogramManager(AliDielectronHistos * const histos) { fHistos=histos; }
   AliDielectronHistos* GetHistoManager() const { return fHistos; }
   const THashList * GetHistogramList() const { return fHistos?fHistos->GetHistogramList():0x0; }
@@ -144,7 +147,8 @@ private:
 
   Bool_t fNoPairing;    // if to skip pairing, can be used for track QA only
   Bool_t fUseKF;    // if to skip pairing, can be used for track QA only
-    
+
+  AliDielectronHF *fHistoArray;   // Histogram framework
   AliDielectronHistos *fHistos;   // Histogram manager
                                   //  Streaming and merging should be handled
                                   //  by the analysis framework
@@ -195,6 +199,7 @@ private:
   void ProcessMC(AliVEvent *ev1);
   
   void  FillHistograms(const AliVEvent *ev, Bool_t pairInfoOnly=kFALSE);
+  void  FillMCHistograms(const AliVEvent *ev);
   void  FillHistogramsMC(const AliMCEvent *ev,  AliVEvent *ev1);
   void  FillHistogramsPair(AliDielectronPair *pair,Bool_t fromPreFilter=kFALSE);
   void  FillHistogramsTracks(TObjArray **tracks);