up from salvatore

[u/mrichter/AliRoot.git] / PWGJE / EMCALJetTasks / AliJetEmbeddingFromAODTask.h

diff --git a/PWGJE/EMCALJetTasks/AliJetEmbeddingFromAODTask.h b/PWGJE/EMCALJetTasks/AliJetEmbeddingFromAODTask.h
index 3555413139b8983551c9b42b2ba8e978d439f4ce..6b108959e322cfa60589524bf7e88a641dc12aed 100644 (file)
--- a/PWGJE/EMCALJetTasks/AliJetEmbeddingFromAODTask.h
+++ b/PWGJE/EMCALJetTasks/AliJetEmbeddingFromAODTask.h
@@ -28,6 +28,7 @@ class AliJetEmbeddingFromAODTask : public AliJetModelBaseTask {
   void           SetAODTracksName(const char *n)                   { fAODTrackName       = n     ; }
   void           SetAODClusName(const char *n)                     { fAODClusName        = n     ; }
   void           SetAODCellsName(const char *n)                    { fAODCellsName       = n     ; }
+  void           SetAODMCParticlesName(const char *n)              { fAODMCParticlesName = n     ; }
   void           SetCentralityRange(Double_t min, Double_t max)    { fMinCentrality      = min   ; fMaxCentrality    = max ; }
   void           SetTriggerMask(UInt_t mask)                       { fTriggerMask        = mask  ; }
   void           SetAODfilterBits(Int_t b0 = 0, Int_t b1 = 0)      { fAODfilterBits[0]   = b0    ; fAODfilterBits[1] = b1  ; }
@@ -41,36 +42,38 @@ class AliJetEmbeddingFromAODTask : public AliJetModelBaseTask {
   Bool_t         GetNextEntry()        ;// get next entry in current tree
   Bool_t         IsAODEventSelected()  ;// AOD event trigger/centrality selection
 
-  TObjArray     *fFileList         ;//  List of AOD files
-  TString        fAODTreeName      ;//  Name of the tree in the AOD file
-  TString        fAODHeaderName    ;//  Name of the header in the AOD tree
-  TString        fAODVertexName    ;//  Name of the vertex branch in the AOD tree
-  TString        fAODTrackName     ;//  Name of the track collection branch in the AOD tree
-  TString        fAODClusName      ;//  Name of the cluster collection branch in the AOD tree
-  TString        fAODCellsName     ;//  Name of the cell collection branch in the AOD tree
-  Double_t       fMinCentrality    ;//  Minimum centrality
-  Double_t       fMaxCentrality    ;//  Maximum centrality
-  UInt_t         fTriggerMask      ;//  Trigger selection mask
-  Double_t       fZVertexCut       ;//  Z vertex cut
-  Int_t          fAODfilterBits[2] ;//  AOD track filter bit map
-  Bool_t         fIncludeNoITS     ;//  True = includes tracks with failed ITS refit
-  Int_t          fTotalFiles       ;//  Total number of files per pt hard bin
-  Bool_t         fEsdTreeMode      ;//! True = embed from ESD (must be a skimmed ESD!)
-  Int_t          fCurrentFileID    ;//! Current file being processed (trough the event handler)
-  Int_t          fCurrentAODFileID ;//! Current file ID
-  TFile         *fCurrentAODFile   ;//! Current open file
-  Int_t          fPicoTrackVersion ;//! Version of the PicoTrack class (if any) in fCurrentAODFile
-  AliVHeader    *fAODHeader        ;//! AOD header
-  TClonesArray  *fAODVertex        ;//! AOD vertex
-  TClonesArray  *fAODTracks        ;//! AOD track collection
-  TClonesArray  *fAODClusters      ;//! AOD cluster collection
-  AliVCaloCells *fAODCaloCells     ;//! AOD cell collection
-  TH2           *fHistFileIDs      ;//! Current file ID vs. AOD file ID (to be embedded)
+  TObjArray     *fFileList            ;//  List of AOD files
+  TString        fAODTreeName         ;//  Name of the tree in the AOD file
+  TString        fAODHeaderName       ;//  Name of the header in the AOD tree
+  TString        fAODVertexName       ;//  Name of the vertex branch in the AOD tree
+  TString        fAODTrackName        ;//  Name of the track collection branch in the AOD tree
+  TString        fAODClusName         ;//  Name of the cluster collection branch in the AOD tree
+  TString        fAODCellsName        ;//  Name of the cell collection branch in the AOD tree
+  TString        fAODMCParticlesName  ;//  Name of the cell collection branch in the AOD tree
+  Double_t       fMinCentrality       ;//  Minimum centrality
+  Double_t       fMaxCentrality       ;//  Maximum centrality
+  UInt_t         fTriggerMask         ;//  Trigger selection mask
+  Double_t       fZVertexCut          ;//  Z vertex cut
+  Int_t          fAODfilterBits[2]    ;//  AOD track filter bit map
+  Bool_t         fIncludeNoITS        ;//  True = includes tracks with failed ITS refit
+  Int_t          fTotalFiles          ;//  Total number of files per pt hard bin
+  Bool_t         fEsdTreeMode         ;//! True = embed from ESD (must be a skimmed ESD!)
+  Int_t          fCurrentFileID       ;//! Current file being processed (trough the event handler)
+  Int_t          fCurrentAODFileID    ;//! Current file ID
+  TFile         *fCurrentAODFile      ;//! Current open file
+  Int_t          fPicoTrackVersion    ;//! Version of the PicoTrack class (if any) in fCurrentAODFile
+  AliVHeader    *fAODHeader           ;//! AOD header
+  TClonesArray  *fAODVertex           ;//! AOD vertex
+  TClonesArray  *fAODTracks           ;//! AOD track collection
+  TClonesArray  *fAODClusters         ;//! AOD cluster collection
+  AliVCaloCells *fAODCaloCells        ;//! AOD cell collection
+  TClonesArray  *fAODMCParticles      ;//! AOD MC particles collection
+  TH2           *fHistFileIDs         ;//! Current file ID vs. AOD file ID (to be embedded)
 
  private:
   AliJetEmbeddingFromAODTask(const AliJetEmbeddingFromAODTask&);            // not implemented
   AliJetEmbeddingFromAODTask &operator=(const AliJetEmbeddingFromAODTask&); // not implemented
 
-  ClassDef(AliJetEmbeddingFromAODTask, 2) // Jet embedding from AOD task
+  ClassDef(AliJetEmbeddingFromAODTask, 3) // Jet embedding from AOD task
 };
 #endif