Bug fix in filling BB and BG flags in the ESD friend.

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

diff --git a/JETAN/AliJetFinder.h b/JETAN/AliJetFinder.h
index e4bf3f240e2bc9f195126f20c49a58c10c6e868c..b8d70d4bb84ca265733b7b2d606e310d79abe370 100755 (executable)
--- a/JETAN/AliJetFinder.h
+++ b/JETAN/AliJetFinder.h
@@ -17,13 +17,15 @@
 
 #include <TObject.h>
 #include "AliAODJet.h"
-#include "AliJetReader.h"
-#include "AliJetHeader.h"
-#include "AliJetReaderHeader.h"
 
 class TChain;
+class TTree;
 class AliAODJet;
 class AliAODEvent;
+class AliAODJetEventBackground;
+class AliJetHeader;
+class AliJetReader;
+class AliJetReaderHeader;
 
 class AliJetFinder : public TObject 
 {
@@ -32,11 +34,13 @@ class AliJetFinder : public TObject
   virtual ~AliJetFinder();
 
   // Getters
-  virtual AliJetReader *GetReader() {return fReader;}
-  virtual AliJetHeader *GetHeader() {return fHeader;}
+  virtual AliJetReader *GetReader() const {return fReader;}
+  virtual AliJetHeader *GetHeader() const {return fHeader;}
+  virtual AliAODJetEventBackground* GetEventBackground() const {return fAODEvBkg;}
   // Setters
   virtual void          SetJetReader(AliJetReader* r) {fReader=r;}
   virtual void          SetJetHeader(AliJetHeader* h) {fHeader=h;}
+  virtual void          SetEventBackground(AliAODJetEventBackground* bkg) {fAODEvBkg = bkg;}
   // Others
   virtual void          AddJet(AliAODJet jet);
   virtual void          WriteRHeaderToFile();  
@@ -46,7 +50,7 @@ class AliJetFinder : public TObject
   virtual void          Reset() {fNAODjets = 0;}
   virtual void          FindJets() {}
   virtual void          FindJetsC(){}
-  virtual void          WriteJHeaderToFile() {}
+  virtual void          WriteJHeaderToFile() const {}
   // some methods to allow steering from the outside
   virtual Bool_t        ProcessEvent();
   virtual Bool_t        ProcessEvent2();
@@ -58,10 +62,13 @@ class AliJetFinder : public TObject
  protected:
   AliJetFinder(const AliJetFinder& rJetFinder);
   AliJetFinder& operator = (const AliJetFinder& rhsf);
-  AliJetReader* fReader;         //  pointer to reader
-  AliJetHeader* fHeader;         //  pointer to header
-  TClonesArray* fAODjets;        //! reconstructed jets
-  Int_t         fNAODjets;       //! number of reconstructed jets
+  AliJetReader*             fReader;         //  pointer to reader
+  AliJetHeader*             fHeader;         //  pointer to header
+  TClonesArray*             fAODjets;        //! reconstructed jets
+  Int_t                     fNAODjets;       //! number of reconstructed jets
+  AliAODJetEventBackground* fAODEvBkg;       //! bkg object to be store
+  Int_t                     fDebug;          //  debug option, set through the header
+
   ClassDef(AliJetFinder,2)
 };