X-Git-Url: http://git.uio.no/git/?a=blobdiff_plain;f=JETAN%2FAliJetReader.h;h=c74f08a9e2af86b65054bdcd0ef193213edb54de;hb=b8187de48ce76b1cacbdbb128e583e7c34d98fbf;hp=300d04adc9999f6de0728b0a6841c398ed1a0604;hpb=7aec0427af26be59d2cc773f41343bf86f41eea8;p=u%2Fmrichter%2FAliRoot.git diff --git a/JETAN/AliJetReader.h b/JETAN/AliJetReader.h index 300d04adc99..c74f08a9e2a 100755 --- a/JETAN/AliJetReader.h +++ b/JETAN/AliJetReader.h @@ -7,22 +7,24 @@ // Jet reader base class // manages the reading of input for jet algorithms // Authors: jgcn@mda.cinvestav.mx -// Magali Estienne +// Magali Estienne #include #include #include -#ifndef ROOT_TTask -#include "TTask.h" -#endif class TTree; +class TChain; class TTask; class TClonesArray; +class TRefArray; class AliJetReaderHeader; +class AliESDEvent; +class AliHeader; class AliJetUnitArray; -class AliJetHadronCorrectionv1; +class AliJetHadronCorrection; class AliJet; +class AliJetFillUnitArray; class AliJetReader : public TObject { @@ -31,29 +33,38 @@ class AliJetReader : public TObject virtual ~AliJetReader(); // Getters - virtual TClonesArray *GetMomentumArray() {return fMomentumArray;} - - virtual AliJetUnitArray *GetUnitArray() const {return fUnitArray;} - virtual AliJetUnitArray *GetUnitArrayNoCuts() const {return fUnitArrayNoCuts;} - - virtual AliJetReaderHeader* GetReaderHeader() { return fReaderHeader;} - virtual Int_t GetSignalFlag(Int_t i) const {return fSignalFlag[i];} - virtual Int_t GetCutFlag(Int_t i) const {return fCutFlag[i];} - virtual Int_t GetArrayInitialised() {return fArrayInitialised;} + virtual TClonesArray* GetMomentumArray() const {return fMomentumArray;} + virtual TRefArray* GetReferences() const {return 0;} + virtual TClonesArray *GetUnitArray() const {return fUnitArray;} + virtual AliJetReaderHeader* GetReaderHeader() const {return fReaderHeader;} + virtual AliHeader *GetAliHeader() const {return fAliHeader;} + virtual Int_t GetSignalFlag(Int_t i) const {return fSignalFlag[i];} + virtual Int_t GetCutFlag(Int_t i) const {return fCutFlag[i];} + virtual Int_t GetArrayInitialised() const {return fArrayInitialised;} + virtual Int_t GetNumCandidate() const {return fNumCandidate;} + virtual Int_t GetNumCandidateCut() const {return fNumCandidateCut;} // Setters - virtual Bool_t FillMomentumArray(Int_t) {return kTRUE;} - virtual void FillUnitArrayFromTPCTracks(Int_t) {} // temporarily not used - virtual void FillUnitArrayFromEMCALHits() {} // temporarily not used - virtual void FillUnitArrayFromEMCALDigits(Int_t) {} // temporarily not used - virtual void FillUnitArrayFromEMCALClusters(Int_t) {} // temporarily not used + virtual Bool_t FillMomentumArray() {return kTRUE;} + virtual Bool_t ReadEventLoader(Int_t) {return kTRUE;} virtual void InitUnitArray() {} + virtual void InitParameters() {} + virtual void CreateTasks(TChain* /*tree*/) {} + virtual Bool_t ExecTasks(Bool_t /*procid*/, TRefArray* /*refArray*/) {return kFALSE;} + // Correction of hadronic energy + virtual void SetHadronCorrector(AliJetHadronCorrection*) {;} + virtual void SetApplyMIPCorrection(Bool_t /*val*/){;} + virtual void SetApplyFractionHadronicCorrection(Bool_t /*val*/){;} + virtual void SetFractionHadronicCorrection(Double_t /*val*/){;} + virtual void SetApplyElectronCorrection(Int_t /*flag*/) {;} virtual void SetReaderHeader(AliJetReaderHeader* header) - {fReaderHeader = header;} - + {fReaderHeader = header;} + virtual void SetESD(AliESDEvent* esd) { fESD = esd;} + // Others virtual void OpenInputFiles() {} - virtual void ConnectTree(TTree* /*tree*/) {} + virtual void SetInputEvent(const TObject* /*esd*/, const TObject* /*aod*/, const TObject* /*mc*/) {;} + virtual void ConnectTree(TTree* /*tree*/, TObject* /*data*/) {} virtual Bool_t GetGenJets(AliJet* /*genJets*/) {return kFALSE;} void ClearArray(); @@ -61,16 +72,31 @@ class AliJetReader : public TObject protected: AliJetReader(const AliJetReader& rJetReader); AliJetReader& operator = (const AliJetReader& rhsr); - TClonesArray *fMomentumArray; // array of particle momenta - TClonesArray *fArrayMC; // array of mc particles - TTask *fFillUnitArray; // task list for filling the UnitArray - AliJetReaderHeader *fReaderHeader; // pointer to 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 - AliJetUnitArray *fUnitArray; // array of digit position and energy - AliJetUnitArray *fUnitArrayNoCuts; // array of digit position and energy - Bool_t fArrayInitialised; // To check that array of units is initialised + + TChain *fChain; // chain for reconstructed tracks + TChain *fTree; // tree for reconstructed tracks + TClonesArray *fMomentumArray; // array of particle momenta + TClonesArray *fArrayMC; //! array of mc particles + TTask *fFillUnitArray; //! task list for filling the UnitArray + AliESDEvent *fESD; // pointer to esd + AliJetReaderHeader *fReaderHeader; // pointer to header + AliHeader *fAliHeader; // AliHeader + 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 + TClonesArray *fUnitArray; // array of digit position and energy + Bool_t fArrayInitialised; // To check that array of units is initialised + AliJetFillUnitArray *fFillUAFromTracks; // For charged particle task + AliJetFillUnitArray *fFillUAFromEMCalDigits; // For neutral particle task + Int_t fNumCandidate; // Number of entries different from zero in unitarray + Int_t fNumCandidateCut; // Number of entries different from zero in unitarray + // which pass pt cut + AliJetHadronCorrection *fHadronCorrector; //! Pointer to hadronic correction + Int_t fHCorrection; // Hadron correction flag + Int_t fECorrection; // Electron correction flag + Bool_t fEFlag; // Electron correction flag + + ClassDef(AliJetReader,1) };