including switch to set on/off iso-track core removal, cleaning and bug fix

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

diff --git a/PWGDQ/dielectron/AliAnalysisTaskDielectronFilter.h b/PWGDQ/dielectron/AliAnalysisTaskDielectronFilter.h
index 77c3779623804bd2262328146c94f2208942a16d..2f64a896232274994a4de36fe31d4c7bdb329b40 100644 (file)
--- a/PWGDQ/dielectron/AliAnalysisTaskDielectronFilter.h
+++ b/PWGDQ/dielectron/AliAnalysisTaskDielectronFilter.h
@@ -30,6 +30,7 @@ Add a sattelite AOD with the array of candidates.
 class AliDielectron;
 class TH1D;
 class AliTriggerAnalysis;
+class AliAODCaloCluster;
 
 class AliAnalysisTaskDielectronFilter : public AliAnalysisTaskSE {
   
@@ -60,7 +61,9 @@ public:
 
   void SetStoreLikeSignCandidates(Bool_t storeLS) { fStoreLikeSign = storeLS; }
   void SetStoreRotatedPairs(Bool_t storeTR) { fStoreRotatedPairs = storeTR; }
-  void SetStoreTrackLegs(Bool_t storeTrackRef) { fStoreTrackLegs = storeTrackRef; }
+  void SetStoreEventsWithSingleTracks(Bool_t storeSingleTrk) { fStoreEventsWithSingleTracks = storeSingleTrk; }
+  void SetCreateNanoAODs(Bool_t storeTrackRef) { fCreateNanoAOD = storeTrackRef; }
+  void SetStoreHeader(Bool_t storeHeader) { fStoreHeader = storeHeader; }
 
   void SetEventFilter(AliAnalysisCuts * const filter) {fEventFilter=filter;}
 
@@ -83,7 +86,9 @@ private:
   
   Bool_t fStoreLikeSign;        // flag to store like-sign candidates
   Bool_t fStoreRotatedPairs;    // flag to store rotation
-  Bool_t fStoreTrackLegs;       // flag to store track legs
+  Bool_t fStoreEventsWithSingleTracks;    // flag to store events with a least one reconstructed track 
+  Bool_t fCreateNanoAOD;        // flag to create nanoAODs 
+  Bool_t fStoreHeader;          // flag to store header for all events 
 
   AliAnalysisCuts *fEventFilter;     // event filter