Derive from AliAnalysisTaskSE.
[u/mrichter/AliRoot.git] / JETAN / AliJetESDReader.h
index 8074fdf..fb4e9bb 100755 (executable)
@@ -18,7 +18,6 @@
 #include "AliJetUnitArray.h"
 #include "AliJetGrid.h"
 class AliJetESDReaderHeader;
-class AliEMCALGeometry;
 class AliJetDummyGeo;
 class AliJetHadronCorrection;
 class AliJetUnitArray;
@@ -30,11 +29,19 @@ class AliJetESDReader : public AliJetReader
  public: 
   AliJetESDReader();
   virtual ~AliJetESDReader();
+
+  // Getters
+  Float_t GetTrackMass() const {return fMass;}  // returns mass of the track
+  Int_t   GetTrackSign() const {return fSign;}  // returns sign of the track
+
   // Setters
-  Bool_t FillMomentumArray(Int_t event); 
+  Bool_t FillMomentumArray(); 
   void   OpenInputFiles();
   void   InitUnitArray();
-  void   ConnectTree(TTree* tree, TObject* data);
+  void   CreateTasks();
+  //  void   ExecTasks(Int_t event);
+  Bool_t   ExecTasks(Int_t event);
+  void   SetInputEvent(TObject* esd, TObject* aod, TObject* mc);
   virtual void SetTPCGrid(AliJetGrid *grid)   {fTpcGrid = grid;}
   virtual void SetEMCalGrid(AliJetGrid *grid) {fEmcalGrid = grid;}
   // Correction of hadronic energy
@@ -50,19 +57,25 @@ class AliJetESDReader : public AliJetReader
   AliJetHadronCorrectionv1   *fHadCorr;          //! Pointer to Hadron Correction Object 
   AliJetGrid                 *fTpcGrid;          //! Pointer to grid object
   AliJetGrid                 *fEmcalGrid;        //! Pointer to grid object
+  AliJetGrid                 *fGrid0;            // Pointer to grid object
+  AliJetGrid                 *fGrid1;            // Pointer to grid object
+  AliJetGrid                 *fGrid2;            // Pointer to grid object
+  AliJetGrid                 *fGrid3;            // Pointer to grid object
+  AliJetGrid                 *fGrid4;            // Pointer to grid object
   Float_t                     fPtCut;            // Pt cut for tracks to minimise background contribution
   Int_t                       fHCorrection;      // Hadron correction flag
   Int_t                       fNumUnits;         // Number of units in the unit object array
                                                  // (same as num towers in EMCAL)
   Int_t                       fDebug;            // Debug option
+  Float_t                     fMass;    // Particle mass
+  Int_t                       fSign;    // Particle sign
   Int_t                       fNIn;              // Number of Array filled in UnitArray
   Int_t                       fOpt;              // Detector to be used for jet reconstruction
+  Bool_t                      fDZ;               // Use or not dead zones
   Int_t                       fNeta;             // Number of bins in eta of tpc grid
   Int_t                       fNphi;             // Number of bins in phi of tpc grid
   Bool_t                      fArrayInitialised; // To check that array of units is initialised
   
-
-
   ClassDef(AliJetESDReader,1)
 };