AliAnalysisMuMu classes update

[u/mrichter/AliRoot.git] / PWG / EMCAL / AliEmcalPicoTrackMaker.h

diff --git a/PWG/EMCAL/AliEmcalPicoTrackMaker.h b/PWG/EMCAL/AliEmcalPicoTrackMaker.h
index f65099643bde17225917908b11d29fd75a2531f2..cc737c995feae63a19f5cbd7b4ca4d4a4e6d05da 100644 (file)
--- a/PWG/EMCAL/AliEmcalPicoTrackMaker.h
+++ b/PWG/EMCAL/AliEmcalPicoTrackMaker.h
@@ -1,10 +1,9 @@
 #ifndef ALIEMCALPICOTRACKMAKER_H
 #define ALIEMCALPICOTRACKMAKER_H
 
-// $Id: AliEmcalPicoTrackMaker.h | Fri Dec 6 10:29:04 2013 +0100 | Constantin Loizides  $
-
 class TClonesArray;
-class AliVTrack;
+class AliVParticle;
+class AliNamedArrayI;
 
 #include "AliAnalysisTaskSE.h"
 
@@ -20,14 +19,20 @@ class AliEmcalPicoTrackMaker : public AliAnalysisTaskSE {
   void               SetTrackPtLimits(Double_t min, Double_t max)      { fMaxTrackPt        = max ; fMinTrackPt       = min ; }
   void               SetTracksInName(const char *name)                 { fTracksInName      = name; }
   void               SetTracksOutName(const char *name)                { fTracksOutName     = name; }
+  void               SetMCParticlesName(const char *name)              { fMCParticlesName   = name; }
+  void               SetCopyMCFlag(Bool_t c, const char* name)         { fCopyMCFlag        = c   ; fMCParticlesName  = name; }
+  
 
  protected:
   void               UserCreateOutputObjects();
   void               UserExec(Option_t *option);
 
+  AliVParticle*      GetMCParticle(Int_t label);
+
   Int_t              fAODfilterBits[2];     // AOD track filter bit map
   TString            fTracksOutName;        // name of output track array
   TString            fTracksInName;         // name of input track array
+  TString            fMCParticlesName;      // name of MC particle array, used by IsHIJINGParticle
   Double_t           fMinTrackPt;           // mix pt of tracks
   Double_t           fMaxTrackPt;           // max pt of tracks
   Double_t           fMinTrackEta;          // cut on track eta
@@ -35,13 +40,17 @@ class AliEmcalPicoTrackMaker : public AliAnalysisTaskSE {
   Double_t           fMinTrackPhi;          // cut on track phi
   Double_t           fMaxTrackPhi;          // cut on track phi
   Double_t           fTrackEfficiency;      // track efficiency
+  Bool_t             fCopyMCFlag;           // copy MC flag
   TClonesArray      *fTracksIn;             //!track array in
   TClonesArray      *fTracksOut;            //!track array out
+  TClonesArray      *fMCParticles;          //!MC particle array
+  AliNamedArrayI    *fMCParticlesMap;       //!MC particle map
+  Bool_t             fInit;                 //!true = task initialized
 
  private:
   AliEmcalPicoTrackMaker(const AliEmcalPicoTrackMaker&);            // not implemented
   AliEmcalPicoTrackMaker &operator=(const AliEmcalPicoTrackMaker&); // not implemented
 
-  ClassDef(AliEmcalPicoTrackMaker, 7); // Task to make PicoTracks in AOD/ESD events
+  ClassDef(AliEmcalPicoTrackMaker, 8); // Task to make PicoTracks in AOD/ESD events
 };
 #endif