EMCAL/jetfinder/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 #include <RVersion.h>
  17 #include "TObject.h"
  18 #include "TParticle.h"
  19 #if ROOT_VERSION_CODE > 331524
  20 #include "TMCParticle.h"
  21 #else
  22 #include "TMCParticle6.h"
  23 #endif
  24 #include "TClonesArray.h"
  25 #include "AliEMCALParton.h"
  26 #include "AliEMCALDigit.h"
  27 #include "AliEMCALJetFinderTypes.h"
  28
  29 class AliEMCALJetFinderInput : public TObject
  30 {
  31         public:
  32                 AliEMCALJetFinderInput();
  33                 ~AliEMCALJetFinderInput();
  34                 void Reset(AliEMCALJetFinderResetType_t resettype);
  35                 void SetDebug(Int_t debug=0){fDebug = debug;}
  36                 void AddEnergyToDigit(Int_t digitID,Int_t denergy);
  37                 void AddTrack(TParticle track);
  38                 void AddTrack(TMCParticle *track);
  39                 void AddParton(AliEMCALParton *parton);
  40                 void AddParticle(TParticle *particle);
  41                 void AddParticle(TMCParticle *particle);
  42                 AliEMCALDigit* GetDigit(Int_t digitID);
  43                 Int_t GetNDigits() const {return fNDigits;}
  44                 TParticle* GetTrack(Int_t trackID);
  45                 Int_t GetNTracks() const {return fNTracks;}
  46                 AliEMCALParton* GetParton(Int_t partonID);
  47                 Int_t GetNPartons() const {return fNPartons;}
  48                 TParticle* GetParticle(Int_t particleID);
  49                 Int_t GetNParticles() const {return fNParticles;}
  50
  51                 AliEMCALJetFinderInput (const AliEMCALJetFinderInput&);
  52                 AliEMCALJetFinderInput & operator = (const AliEMCALJetFinderInput &) {
  53                   Fatal("operator =", "not implemented") ;
  54                   return *this ;
  55                 }
  56
  57         private:
  58                 void InitArrays();
  59                 TClonesArray*   fDigitsArray;   //-> This is the digits array for the EMCAL
  60                 Int_t           fNDigits;       // This is the number of digits
  61                 Int_t           fNMaxDigits;    // This is the max number of digits
  62                 TClonesArray*   fTracksArray;   //-> This is the track array
  63                 Int_t           fNTracks;       // This stores the number of tracks
  64                 Int_t           fNMaxTracks;    // This stores the maximum number of tracks
  65                 TClonesArray*   fPartonsArray;  //->  This is the partons array
  66                 Int_t           fNPartons;      // This stores the number of partons
  67                 Int_t           fNMaxPartons;   // This stores the maximum number of partons
  68                 TClonesArray*   fParticlesArray;//-> This stores the particles
  69                 Int_t           fNParticles;    // This stores the number of particles
  70                 Int_t           fNMaxParticles; // This stroes the maximum number of particles
  71                 Int_t           fDebug;         // This is the debug value
  72                 Bool_t          fInitialised;   // Stores whether or not the arrays have been initialised
  73
  74         ClassDef(AliEMCALJetFinderInput,5)
  75 };
  76 #endif