]> git.uio.no Git - u/mrichter/AliRoot.git/blobdiff - PWGDQ/dielectron/AliDielectron.h
Add event selection depending on high energy clusters existance in EMCal to speed...
[u/mrichter/AliRoot.git] / PWGDQ / dielectron / AliDielectron.h
index 7e23a20510b363ff2fac65c0b4fd14c2c8267121..829a9ede7bcc71fd48f0e55d8c584519c2fabbb0 100644 (file)
@@ -23,6 +23,7 @@
 #include <AliKFParticle.h>
 
 #include "AliDielectronHistos.h"
+#include "AliDielectronHF.h"
 
 class AliEventplane;
 class AliVEvent;
@@ -74,12 +75,15 @@ public:
   Int_t GetLeg2Pdg()   const { return fPdgLeg2;   }
 
   void SetNoPairing(Bool_t noPairing=kTRUE) { fNoPairing=noPairing; }
+  void SetUseKF(Bool_t useKF=kTRUE) { fUseKF=useKF; }
   const TObjArray* GetTrackArray(Int_t i) const {return (i>=0&&i<4)?&fTracks[i]:0;}
   const TObjArray* GetPairArray(Int_t i)  const {return (i>=0&&i<11)?
       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; }
@@ -142,7 +146,9 @@ private:
   TObjArray* fSignalsMC;      // array of AliDielectronSignalMC
 
   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
@@ -193,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);