Restoring EMCAL digitization and reconstruction
[u/mrichter/AliRoot.git] / JETAN / AliJetReader.h
1 #ifndef ALIJETREADER_H
2 #define ALIJETREADER_H
3  
4 /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
5  * See cxx source for full Copyright notice                               */
6  
7 //---------------------------------------------------------------------
8 // Jet reader base class
9 // manages the reading of input for jet algorithms
10 // Author: jgcn@mda.cinvestav.mx
11 //---------------------------------------------------------------------
12   
13 #include <TObject.h>
14 #include <TChain.h>
15 #include <TArrayI.h>
16
17 class TClonesArray;
18 class AliJetReaderHeader;
19 class AliESD;
20 class AliHeader;
21
22
23 class AliJetReader : public TObject 
24 {
25  public: 
26   AliJetReader();
27   virtual ~AliJetReader();
28
29   // Getters
30   virtual TClonesArray *GetMomentumArray() {return fMomentumArray;}
31   virtual Int_t GetChainEntries() {return fChain->GetEntries();} 
32   virtual AliJetReaderHeader* GetReaderHeader() { return fReaderHeader;}
33   virtual AliHeader* GetAliHeader() { return fAliHeader;}
34   virtual Int_t GetSignalFlag(Int_t i) const {return fSignalFlag[i];}
35
36   // Setters
37   virtual void FillMomentumArray(Int_t) {}
38   virtual void SetReaderHeader(AliJetReaderHeader* header) 
39     {fReaderHeader = header;}
40           
41   // others
42   virtual void OpenInputFiles() {}
43   void ClearArray();
44  
45  protected:
46   TChain                  *fChain;         // chain for reconstructed tracks
47   TChain                  *fChainMC;       // chain for mc information
48   TClonesArray            *fMomentumArray; // array of particle momenta
49   TClonesArray            *fArrayMC;       // array of mc particles
50   AliESD                  *fESD;           // pointer to esd
51   AliJetReaderHeader      *fReaderHeader;  // pointer to header
52   AliHeader               *fAliHeader;     // pointer to event header
53   TArrayI fSignalFlag;   // to flag if a particle comes from pythia or 
54                         // from the underlying event
55
56   ClassDef(AliJetReader,1)
57 };
58  
59 #endif