EMCAL/AliEMCALJetFinderInput.h

   1 #ifndef ALIEMCALJETFINDERINPUT_H
   2 #define ALIEMCALJETFINDERINPUT_H
   3
   4 /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
   5  *  *  * See cxx source for full Copyright notice     */
   6
   7 /* $Id$ */
   8
   9 //_________________________________________________________________________
  10 //  Initial input object for jetfinder
  11 //
  12 //*-- Author: Mark Horner (LBL/UCT)
  13 //
  14 //
  15
  16
  17 #include "TObject.h"
  18 #include "TParticle.h"
  19 #include "TMCParticle.h"
  20 #include "TClonesArray.h"
  21 #include "AliEMCALParton.h"
  22 #include "AliEMCALDigit.h"
  23 #include "AliEMCALJetFinderTypes.h"
  24
  25 class AliEMCALJetFinderInput : public TObject
  26 {
  27         public:
  28                 AliEMCALJetFinderInput();
  29                 ~AliEMCALJetFinderInput();
  30                 void Reset(AliEMCALJetFinderResetType_t resettype);
  31                 void SetDebug(Int_t debug=0){fDebug = debug;}
  32                 void AddEnergyToDigit(Int_t digitID,Int_t denergy);
  33                 void AddTrack(TParticle track);
  34                 void AddTrack(TMCParticle *track);
  35                 void AddParton(AliEMCALParton *parton);
  36                 void AddParticle(TParticle *particle);
  37                 void AddParticle(TMCParticle *particle);
  38                 AliEMCALDigit* GetDigit(Int_t digitID);
  39                 Int_t GetNDigits() const {return fNDigits;}
  40                 TParticle* GetTrack(Int_t trackID);
  41                 Int_t GetNTracks() const {return fNTracks;}
  42                 AliEMCALParton* GetParton(Int_t partonID);
  43                 Int_t GetNPartons() const {return fNPartons;}
  44                 TParticle* GetParticle(Int_t particleID);
  45                 Int_t GetNParticles() const {return fNParticles;}
  46
  47         private:
  48                 void InitArrays();
  49                 TClonesArray*   fDigitsArray;   //-> This is the digits array for the EMCAL
  50                 Int_t           fNDigits;       // This is the number of digits
  51                 Int_t           fNMaxDigits;    // This is the max number of digits
  52                 TClonesArray*   fTracksArray;   //-> This is the track array
  53                 Int_t           fNTracks;       // This stores the number of tracks
  54                 Int_t           fNMaxTracks;    // This stores the maximum number of tracks
  55                 TClonesArray*   fPartonsArray;  //->  This is the partons array
  56                 Int_t           fNPartons;      // This stores the number of partons
  57                 Int_t           fNMaxPartons;   // This stores the maximum number of partons
  58                 TClonesArray*   fParticlesArray;//-> This stores the particles
  59                 Int_t           fNParticles;    // This stores the number of particles
  60                 Int_t           fNMaxParticles; // This stroes the maximum number of particles
  61                 Int_t           fDebug;         // This is the debug value
  62                 Bool_t          fInitialised;   // Stores whether or not the arrays have been initialised
  63
  64         ClassDef(AliEMCALJetFinderInput,5)
  65 };
  66 #endif