X-Git-Url: http://git.uio.no/git/?a=blobdiff_plain;f=JETAN%2FAliJetReader.h;h=c74f08a9e2af86b65054bdcd0ef193213edb54de;hb=e3f643f049dacb8959076a4b9bebd1baf65d741d;hp=3b63c3282cd59d00a7675ce4102322621fddacae;hpb=ac088e8a82f8f179416604f4764b4c562cbc5898;p=u%2Fmrichter%2FAliRoot.git diff --git a/JETAN/AliJetReader.h b/JETAN/AliJetReader.h index 3b63c3282cd..c74f08a9e2a 100755 --- a/JETAN/AliJetReader.h +++ b/JETAN/AliJetReader.h @@ -7,23 +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 { @@ -32,29 +33,37 @@ class AliJetReader : public TObject virtual ~AliJetReader(); // Getters - virtual TClonesArray* GetMomentumArray() const {return fMomentumArray;} - virtual TRefArray* GetReferences() const {return 0;} - virtual AliJetUnitArray* GetUnitArray() const {return fUnitArray;} - virtual AliJetUnitArray* GetUnitArrayNoCuts() const {return fUnitArrayNoCuts;} - - virtual AliJetReaderHeader* GetReaderHeader() const {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() const {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 SetInputEvent(TObject* /*esd*/, TObject* /*aod*/, TObject* /*mc*/) {;} + 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;} @@ -63,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) };