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 "AliEMCALParton.h"
  20 #include "AliEMCALDigit.h"
  21 #include "AliEMCALJetFinderTypes.h"
  22
  23 class AliEMCALJetFinderInput : public TObject
  24 {
  25         public:
  26                 AliEMCALJetFinderInput();
  27                 ~AliEMCALJetFinderInput();
  28                 void Reset(AliEMCALJetFinderResetType_t resettype);
  29                 void SetDebug(Int_t debug=0){fDebug = debug;}
  30                 void AddEnergyToDigit(Int_t digitID,Int_t denergy);
  31                 void AddTrack(TParticle track);
  32                 void AddParton(AliEMCALParton *parton);
  33                 void AddParticle(TParticle *particle);
  34                 AliEMCALDigit* GetDigit(Int_t digitID);
  35                 Int_t GetNDigits() const {return fNDigits;}
  36                 TParticle* GetTrack(Int_t trackID);
  37                 Int_t GetNTracks() const {return fNTracks;}
  38                 AliEMCALParton* GetParton(Int_t partonID);
  39                 Int_t GetNPartons() const {return fNPartons;}
  40                 TParticle* GetParticle(Int_t particleID);
  41                 Int_t GetNParticles() const {return fNParticles;}
  42
  43         private:
  44                 void InitArrays();
  45                 AliEMCALDigit   fDigitsArray[13824];    // This is the digits array for the EMCAL
  46                 Int_t           fNDigits;       // This is the number of digits
  47                 Int_t           fNMaxDigits;    // This is the max number of digits
  48                 TParticle       fTracksArray[3000];     // This is the track array
  49                 Int_t           fNTracks;       // This stores the number of tracks
  50                 Int_t           fNMaxTracks;    // This stores the maximum number of tracks
  51                 AliEMCALParton  fPartonsArray[4];  //  This is the partons array
  52                 Int_t           fNPartons;      // This stores the number of partons
  53                 Int_t           fNMaxPartons;     // This stores the maximum number of partons
  54                 TParticle       fParticlesArray[2000];// This stores the particles
  55                 Int_t           fNParticles;    // This stores the number of particles
  56                 Int_t           fNMaxParticles; // This stroes the maximum number of particles
  57                 Int_t           fDebug;         // This is the debug value
  58                 Bool_t          fInitialised;   // Stores whether or not the arrays have been initialised
  59
  60         ClassDef(AliEMCALJetFinderInput,3)
  61 };
  62 #endif