[u/mrichter/AliRoot.git] / JETAN / AliJetReader.h

#ifndef ALIJETREADER_H
#define ALIJETREADER_H
 
/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
 * See cxx source for full Copyright notice                               */
 
// Jet reader base class
// manages the reading of input for jet algorithms
// Author: jgcn@mda.cinvestav.mx
  
#include <TObject.h>
#include <TChain.h>
#include <TArrayI.h>

class TClonesArray;
class AliJetReaderHeader;
class AliESD;
class AliHeader;

class AliJetReader : public TObject 
{
 public: 
  AliJetReader();
  virtual ~AliJetReader();

  // Getters
  virtual TClonesArray *GetMomentumArray() {return fMomentumArray;}
  virtual Int_t GetChainEntries() {return fChain->GetEntries();} 
  virtual AliJetReaderHeader* GetReaderHeader() { return fReaderHeader;}
  virtual AliHeader* GetAliHeader() {return fAliHeader;}
  virtual Int_t GetSignalFlag(Int_t i) const {return fSignalFlag[i];}
  virtual Int_t GetCutFlag(Int_t i) const {return fCutFlag[i];}

  // Setters
  virtual void FillMomentumArray(Int_t) {}
  virtual void SetReaderHeader(AliJetReaderHeader* header) 
    {fReaderHeader = header;}
	  
  // others
  virtual void OpenInputFiles() {}
  void ClearArray();
 
 protected:
  AliJetReader(const AliJetReader& rJetReader);
  AliJetReader& operator = (const AliJetReader& rhsr);

  TChain                  *fChain;         // chain for reconstructed tracks
  TChain                  *fChainMC;       // chain for mc information
  TClonesArray            *fMomentumArray; // array of particle momenta
  TClonesArray            *fArrayMC;       // array of mc particles
  AliESD                  *fESD;           // pointer to esd
  AliJetReaderHeader      *fReaderHeader;  // pointer to header
  AliHeader               *fAliHeader;     // pointer to event header
  TArrayI fSignalFlag;   // to flag if a particle comes from pythia or 
                         // from the underlying event
  TArrayI fCutFlag;      // to flag if a particle passed the pt cut or not

  ClassDef(AliJetReader,1)
};
 
#endif
Commit	Line	Data
99e5fe42	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
99e5fe42	7	// Jet reader base class
	8	// manages the reading of input for jet algorithms
	9	// Author: jgcn@mda.cinvestav.mx
99e5fe42	10
	11	#include <TObject.h>
	12	#include <TChain.h>
8011d399	13	#include <TArrayI.h>
99e5fe42	14
	15	class TClonesArray;
	16	class AliJetReaderHeader;
	17	class AliESD;
	18	class AliHeader;
	19
99e5fe42	20	class AliJetReader : public TObject
	21	{
	22	public:
	23	AliJetReader();
	24	virtual ~AliJetReader();
	25
	26	// Getters
	27	virtual TClonesArray *GetMomentumArray() {return fMomentumArray;}
	28	virtual Int_t GetChainEntries() {return fChain->GetEntries();}
	29	virtual AliJetReaderHeader* GetReaderHeader() { return fReaderHeader;}
83a444b1	30	virtual AliHeader* GetAliHeader() {return fAliHeader;}
8011d399	31	virtual Int_t GetSignalFlag(Int_t i) const {return fSignalFlag[i];}
83a444b1	32	virtual Int_t GetCutFlag(Int_t i) const {return fCutFlag[i];}
99e5fe42	33
	34	// Setters
	35	virtual void FillMomentumArray(Int_t) {}
8011d399	36	virtual void SetReaderHeader(AliJetReaderHeader* header)
8011d399	37	{fReaderHeader = header;}
99e5fe42	38
	39	// others
	40	virtual void OpenInputFiles() {}
	41	void ClearArray();
	42
	43	protected:
1b7d5d7e	44	AliJetReader(const AliJetReader& rJetReader);
	45	AliJetReader& operator = (const AliJetReader& rhsr);
	46
99e5fe42	47	TChain *fChain; // chain for reconstructed tracks
	48	TChain *fChainMC; // chain for mc information
	49	TClonesArray *fMomentumArray; // array of particle momenta
	50	TClonesArray *fArrayMC; // array of mc particles
	51	AliESD *fESD; // pointer to esd
	52	AliJetReaderHeader *fReaderHeader; // pointer to header
	53	AliHeader *fAliHeader; // pointer to event header
8011d399	54	TArrayI fSignalFlag; // to flag if a particle comes from pythia or
83a444b1	55	// from the underlying event
83a444b1	56	TArrayI fCutFlag; // to flag if a particle passed the pt cut or not
8011d399	57
99e5fe42	58	ClassDef(AliJetReader,1)
	59	};
	60
	61	#endif