EMCAL/AliEMCALJetFinderOutput.h

   1 #ifndef ALIEMCALJETFINDEROUTPUT_H
   2 #define ALIEMCALJETFINDEROUTPUT_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 //  Output object for jetfinder
  11 //
  12 //*-- Author:   Renan Cabrera (LBL)
  13 //              Mark Horner (LBL/UCT)
  14 //
  15
  16 class TClonesArray;
  17 #include "TObject.h"
  18 #include "TParticle.h"
  19 #include "AliEMCALParton.h"
  20 #include "AliEMCALJet.h"
  21 #include "AliEMCALJetFinderTypes.h"
  22
  23 class AliEMCALJetFinderOutput : public TObject
  24 {
  25
  26         public:
  27                 AliEMCALJetFinderOutput();
  28                 ~AliEMCALJetFinderOutput();
  29                 void Reset(AliEMCALJetFinderResetType_t resettype);
  30                 void AddJet(AliEMCALJet *jet);
  31                 void AddParton(AliEMCALParton *parton);
  32                 void AddParticle(TParticle *particle);
  33                 void SetDebug(Int_t debug){fDebug = debug;}
  34                 AliEMCALJet* GetJet(Int_t jetID);
  35                 Int_t GetNJets() const {return fNJets;}
  36                 TClonesArray *GetJets() {return fJetsArray; }
  37                 AliEMCALParton* GetParton(Int_t partonID);
  38                 Int_t GetNPartons() const {return fNPartons;}
  39                 TParticle* GetParticle(Int_t particleID);
  40                 TClonesArray *GetParticles() {return fParticlesArray; }
  41                 Int_t GetNParticles() const {return fNParticles;}
  42
  43         ClassDef(AliEMCALJetFinderOutput,5)
  44         private:
  45                 void InitArrays();
  46                 TClonesArray    *fJetsArray;            // Array of jet objects
  47                 TClonesArray    *fPartonsArray;         // Array of parton objects
  48                 Int_t           fNPartons;              // Number of Partons actually stored
  49                 Int_t           fNJets;                 // Number of jets actually stored
  50                 TClonesArray    *fParticlesArray;       // Array of particles
  51                 Int_t           fNParticles;            // Number of particles actually stored
  52                 Int_t           fNMaxJets;              // Maximum number of jets
  53                 Int_t           fNMaxParticles;         // Maximum number of primary particles
  54                 Int_t           fNMaxPartons;           // Maximum number of primary particles
  55                 Int_t           fDebug;                 // Debug level
  56                 Bool_t          fInitialised;           // stores whether or not the arrays have been initialised
  57
  58 };
  59 #endif