]> git.uio.no Git - u/mrichter/AliRoot.git/blobdiff - EMCAL/AliEMCALGetter.h
Changes needed to run simulation and reconstrruction in the same AliRoot session
[u/mrichter/AliRoot.git] / EMCAL / AliEMCALGetter.h
index ee3bceb76035de127d64958243f5f1f1c6f7deaf..5904f8f4e0990bb4dac633d9dea67e56daddac41 100644 (file)
 
 
 // --- ROOT system ---
+#include "TObject.h"  
 #include "TClonesArray.h"
-#include "TFolder.h"  
-#include "TTree.h"
-#include "TFile.h"
-class TString ;
 class TParticle ;
-class TTask ;
+class TTree ; 
+class TGraph ;
+class TF1 ;
 
 // --- Standard library ---
-#include <stdlib.h>
-#include <iostream.h>
 
 // --- AliRoot header files ---
+#include "AliConfig.h" 
+
+// #include "AliRun.h"
+#include "AliEMCALHit.h"   
+#include "AliEMCALRecParticle.h"
+#include "AliEMCALDigitizer.h"
+#include "AliEMCALSDigitizer.h" 
+#include "AliEMCALLoader.h" 
+class AliEMCAL ; 
+class AliEMCALGeometry ; 
+class AliEMCALClusterizer ; 
+class AliEMCALRecPoint ; 
+class AliEMCALTrackSegmentMaker ;
+class AliEMCALTrackSegment ; 
+class AliEMCALPID ; 
+class AliEMCALBeamTestEvent ;
 
-#include "AliRun.h"
-#include "AliEMCALv1.h" 
-#include "AliEMCALHit.h" 
-#include "AliEMCALDigit.h" 
-#include "AliEMCALDigitizer.h" 
-#include "AliEMCALSDigitizer.h"
-#include "AliEMCALTowerRecPoint.h"
-class AliEMCALGeometry ;
-class AliEMCALClusterizerv1 ;
-//class AliEMCALTrackSegment ;
-//class AliEMCALTrackSegmentMaker ;
-//class AliEMCALRecParticle ;
-//class AliEMCALPID ;
 
 class AliEMCALGetter : public TObject {
   
- public:
-  
-  AliEMCALGetter(){ 
+ public:  
+  AliEMCALGetter() {    
     // ctor: this is a singleton, the ctor should never be called but cint needs it as public
-    cerr << "ERROR: AliPHOGetter is a singleton default ctor not callable" << endl ;
-    abort() ; 
+    Fatal("ctor", "AliEMCALGetter is a singleton default ctor not callable") ;
   } 
-  AliEMCALGetter(const AliEMCALGetter & obj) {
+  AliEMCALGetter(const AliEMCALGetter & obj):TObject(obj) {
     // cpy ctor requested by Coding Convention 
-    // but not yet needed
-    abort() ; 
+    Fatal("cpy ctor", "not implemented") ;
   } 
   
+  AliEMCALGetter & operator = (const AliEMCALGetter & ) {
+    // assignement operator requested by coding convention, but not needed
+    Fatal("operator =", "not implemented") ;
+    return *this ; 
+  }
   virtual ~AliEMCALGetter() ; 
   
-  void CloseFile() { fFile->Close() ; } 
-  const Bool_t HasFailed(void) const {return fFailed ;} 
-  Bool_t PostHits(void ) const ;  
-  Bool_t PostSDigits(      const char * name,  const char * file = 0) const ;  
-  Bool_t PostDigits(       const char * name ) const ;  
-  Bool_t PostRecPoints(    const char * name ) const ;  
-  //Bool_t PostTrackSegments(const char * name) const ;  
-  //Bool_t PostRecParticles( const char * name) const ;  
-
-  Bool_t PostClusterizer( const char * name) const ;  
-  Bool_t PostClusterizer(AliEMCALClusterizerv1 * clu) const ;  
-  Bool_t PostSDigitizer (AliEMCALSDigitizer * sdigitizer) const ;  
-  Bool_t PostSDigitizer ( const char * name, const char * file ) const ;  
-  Bool_t PostDigitizer (AliEMCALDigitizer * digitizer) const ;  
-  Bool_t PostDigitizer  ( const char * name) const ;  
-  //Bool_t PostTrackSegmentMaker(AliEMCALTrackSegmentMaker * tsm) const ;  
-  //Bool_t PostTrackSegmentMaker(const char * name ) const ;  
-  //Bool_t PostPID  (AliEMCALPID * pid) const ;  
-  //Bool_t PostPID  (const char * name ) const ;  
-  //Bool_t PostQA   (void) const ;
-  
+  //=========== Instantiators ================
+  static AliEMCALGetter * Instance(const char* headerFile,
+                                 const char* version = AliConfig::GetDefaultEventFolderName(),
+                                 Option_t * openingOption = "READ" ) ; 
+  static AliEMCALGetter * Instance() ; 
 
-  void   Event(const Int_t event, const char * opt = "HSDR") ;    
+  static void Print() ; 
+  
+//   //=========== General information about run ==============
+  Bool_t IsLoaded(TString tree) const { return fLoadingStatus.Contains(tree) ; } 
+  void   SetLoaded(TString tree) { fLoadingStatus += tree ; } 
+
+  Int_t  MaxEvent() const ; 
+  Int_t  EventNumber() const ; 
+  Bool_t VersionExists(TString & opt) const ; 
+  UShort_t EventPattern(void) const ; 
+  Float_t  BeamEnergy(void) const ;
+  
+//   //========== EMCALGeometry and EMCAL ============= 
+  AliEMCAL *         EMCAL() const  ;  
+  AliEMCALGeometry * EMCALGeometry() const ; 
+  
+//   //========== Methods to read something from file ==========
+  void   Event(Int_t event, const char * opt = "HSDRP") ;    
   void   Track(Int_t itrack) ;
-
-  //Method to be used when digitizing under AliRunDigitizer, who opens all files etc.
-  void   ReadTreeS(TTree * treeS,Int_t input) ;
   
-  Int_t  EventNumber()       { return (Int_t) gAlice->GetEvNumber() ; }
-  Int_t  MaxEvent()          { return (Int_t) gAlice->TreeE()->GetEntries() ; }
-  static AliEMCALGetter * GetInstance(const char* headerFile,
-                                    const char* branchTitle = "Default", const Option_t * rw="" ) ; 
-  static AliEMCALGetter *   GetInstance() ; 
-
-  const AliEMCALv1 *         EMCAL()  ;  
-  AliEMCALGeometry * EMCALGeometry() ; 
-   // Alarms
-  //TFolder * Alarms() const { return (TFolder*)(ReturnO("Alarms", 0)) ; }
-  //TObjArray *  Alarms(const char * name ) const { return (TObjArray*)(ReturnO("Alarms", name)) ; }
-
-  // QA Tasks
-  //TTask * QATasks(const char * name = 0) const { return (TTask*)(ReturnT("QATasks", name)) ; }
-
-  // Primaries
-  TClonesArray *  Primaries(void) const { return (TClonesArray*)(ReturnO("Primaries")) ; }
+//   //-----------------now getter's data--------------------------------------
+// //  AliEMCALCalibrationDB * CalibrationDB(){return  fcdb; }
+// //  void ReadCalibrationDB(const char * name, const char * filename) ;
+
+  //=========== Primaries ============
+//   TTree *           TreeK(TString filename="") ; 
+  TParticle * Primary(Int_t index) const ;
+  Int_t       NPrimaries() const;
+  TParticle * Secondary(const TParticle * p, Int_t index=1) const ;  
   
+//   //=========== Hits =================
+//   TTree *               TreeH(TString filename="") ; 
+  TClonesArray *  Hits(void) const  ; 
+  AliEMCALHit *    Hit(Int_t index) const { return dynamic_cast<AliEMCALHit*>(Hits()->At(index) );}
+  TTree *         TreeH() const ; 
   
-  // Hits
-  const TClonesArray *  Hits(void) { return static_cast<const TClonesArray*>(ReturnO("Hits")) ; }
-  const AliEMCALHit * Hit(Int_t index)  { return static_cast<const AliEMCALHit*>(Hits()->At(index) );}
+  //=========== SDigits ==============
+  TClonesArray *      SDigits() const ;  
+  AliEMCALDigit *      SDigit(Int_t index) const { return static_cast<AliEMCALDigit *>(SDigits()->At(index)) ;} 
+  TTree *             TreeS() const ; 
+  AliEMCALSDigitizer * SDigitizer() ;  
+
+  TString             GetSDigitsFileName() const { return EmcalLoader()->GetSDigitsFileName() ; }  
+  Int_t               LoadSDigits(Option_t* opt="") const { return EmcalLoader()->LoadSDigits(opt) ; }
+  Int_t               LoadSDigitizer(Option_t* opt="") const { return  EmcalLoader()->LoadSDigitizer(opt) ; }
+  Int_t               WriteSDigits(Option_t* opt="") const { return EmcalLoader()->WriteSDigits(opt) ; }
+  Int_t               WriteSDigitizer(Option_t* opt="") const {
+    return  EmcalLoader()->WriteSDigitizer(opt) ; }
   
-  // SDigits
-  TClonesArray *  SDigits(const char * name = 0, const char * file=0) { 
-    return static_cast<TClonesArray*>(ReturnO("SDigits", name, file)) ; 
-  }
-  const AliEMCALDigit *  SDigit(Int_t index) { return static_cast<const AliEMCALDigit*>(SDigits()->At(index)) ;}
+  //========== Digits ================
+  TClonesArray * Digits() const ;
+  AliEMCALDigit * Digit(Int_t index) const { return static_cast<AliEMCALDigit *>(Digits()->At(index)) ;} 
+  TTree *        TreeD() const ; 
+  AliEMCALDigitizer * Digitizer() ;
+  TString             GetDigitsFileName() const { return EmcalLoader()->GetDigitsFileName() ; }  
+  Int_t               LoadDigits(Option_t* opt="") const { return EmcalLoader()->LoadDigits(opt) ; }
+  Int_t               LoadDigitizer(Option_t* opt="") const {
+    return  EmcalLoader()->LoadDigitizer(opt) ; }
+  Int_t               WriteDigits(Option_t* opt="") const { return EmcalLoader()->WriteDigits(opt) ; }
+  Int_t               WriteDigitizer(Option_t* opt="") const {
+    return  EmcalLoader()->WriteDigitizer(opt) ; }
   
-  AliEMCALSDigitizer *  SDigitizer(const char * name =0) const { 
-    return ((AliEMCALSDigitizer*)(ReturnT("SDigitizer", name))) ; 
-  }
-  // Digits
-   TClonesArray *  Digits(const char * name = 0)const  { 
-    return static_cast<TClonesArray*>(ReturnO("Digits", name)) ; 
-  }
-  const AliEMCALDigit *  Digit(Int_t index) { return static_cast<const AliEMCALDigit *>(Digits()->At(index)) ;}
-  AliEMCALDigitizer *  Digitizer(const char * name =0) const { 
-    return (AliEMCALDigitizer*)(ReturnT("Digitizer", name)) ; 
-  }
+  //========== RecPoints =============
+  TObjArray *             ECARecPoints() const;
+  AliEMCALRecPoint * ECARecPoint(Int_t index) const{ return static_cast<AliEMCALRecPoint *>(ECARecPoints()->At(index)) ;}    
+  TTree *                 TreeR() const ;
+  AliEMCALClusterizer *   Clusterizer()  ;
+  TString                 GetRecPointsFileName() const { return EmcalLoader()->GetRecPointsFileName() ; } 
+  Int_t                   LoadRecPoints(Option_t* opt="") const { return EmcalLoader()->LoadRecPoints(opt) ; }
+  Int_t                   LoadClusterizer(Option_t* opt="") const {
+    return  EmcalLoader()->LoadClusterizer(opt) ; }
+  Int_t                   WriteRecPoints(Option_t* opt="") const { return EmcalLoader()->WriteRecPoints(opt) ; }
+  Int_t                   WriteClusterizer(Option_t* opt="") const {
+    return  EmcalLoader()->WriteClusterizer(opt) ; }
+
+  //========== TrackSegments   TClonesArray * TrackSegments(const char * name = 0) { 
+  TClonesArray *           TrackSegments() const ;
+  AliEMCALTrackSegment *  TrackSegments(Int_t index) const { return static_cast<AliEMCALTrackSegment *>(TrackSegments()->At(index)) ;} 
+  TTree *               TreeT() const ;
+  AliEMCALTrackSegmentMaker * TrackSegmentMaker() ;
+  TString               GetTracksFileName() const { return EmcalLoader()->GetTracksFileName() ; } 
+  Int_t                 LoadTracks(Option_t* opt="") const { return EmcalLoader()->LoadTracks(opt) ; }
+  Int_t                 LoadTrackSegementMaker(Option_t* opt="") const {
+    return  EmcalLoader()->LoadTrackSegmentMaker(opt) ; }
+  Int_t                 WriteTracks(Option_t* opt="") const { return EmcalLoader()->WriteTracks(opt) ; }
+  Int_t                 WriteTrackSegmentMaker(Option_t* opt="") const {
+    return  EmcalLoader()->WriteTracker(opt) ; }
+  //========== RecParticles ===========
+
+  TClonesArray *         RecParticles() const ;
+  AliEMCALRecParticle *  RecParticle(Int_t index) const { return static_cast<AliEMCALRecParticle *>(RecParticles()->At(index)) ;} 
+  TTree *               TreeP() const ;
+  AliEMCALPID * PID() ;
+  TString               GetRecParticlesFileName() const { return EmcalLoader()->GetRecParticlesFileName() ; } 
+  Int_t                 LoadRecParticles(Option_t* opt="") const { return EmcalLoader()->LoadRecParticles(opt) ; }
+  Int_t                 LoadPID(Option_t* opt="") const {
+    return  EmcalLoader()->LoadPID(opt) ; }
+  Int_t                 WriteRecParticles(Option_t* opt="") const { return EmcalLoader()->WriteRecParticles(opt) ; }
+  Int_t                 WritePID(Option_t* opt="") const {
+    return  EmcalLoader()->WritePID(opt) ; }
+
+  //========== Raw ===========
+  Int_t ReadRaw(Int_t event) ; 
+
+  void SetDebug(Int_t level) {fgDebug = level;} // Set debug level 
+    void PostClusterizer(AliEMCALClusterizer * clu) 
+    const{EmcalLoader()->PostClusterizer(clu) ; }
+  void PostPID(AliEMCALPID * pid) 
+    const{EmcalLoader()->PostPID(pid) ; }
+  void PostTrackSegmentMaker(AliEMCALTrackSegmentMaker * tr) 
+    const{EmcalLoader()->PostTrackSegmentMaker(tr) ; }
+  void PostSDigitizer (AliEMCALSDigitizer * sdigitizer) 
+    const {EmcalLoader()->PostSDigitizer(sdigitizer);}    
+  void PostDigitizer (AliEMCALDigitizer * digitizer)    
+    const {EmcalLoader()->PostDigitizer(dynamic_cast<AliDigitizer *>(digitizer));}
+
+  TString Version() const  { return EmcalLoader()->GetTitle() ; } 
+  AliEMCALLoader * EmcalLoader() const { return fgEmcalLoader; }
+  void Reset() ;  
+
+private:
+  
+  AliEMCALGetter(const char* headerFile,
+                const char* version = AliConfig::GetDefaultEventFolderName(),
+                Option_t * openingOption = "READ") ;
 
-  //  RecPoints
-  TObjArray * TowerRecPoints(const char * name = 0) const { 
-    return (TObjArray*)(ReturnO("TowerRecPoints", name)) ; }
-  TObjArray * PreShowerRecPoints(const char * name = 0) const { 
-    return (TObjArray*)(ReturnO("PreShoRecPoints", name)) ; }
-  const AliEMCALTowerRecPoint *  TowerRecPoint(Int_t index) { 
-    return static_cast<const AliEMCALTowerRecPoint *>(TowerRecPoints()->At(index)) ;}
-  const AliEMCALTowerRecPoint *  PreShowerRecPoint(Int_t index) { 
-    return static_cast<const AliEMCALTowerRecPoint *>(PreShowerRecPoints()->At(index)) ;}
-
-    AliEMCALClusterizerv1 * Clusterizer (const char * name =0) const { 
-    return (AliEMCALClusterizerv1*)(ReturnT("Clusterizer", name)) ; }
-
-  // TrackSegments
-  //TClonesArray * TrackSegments(const char * name = 0) const 
-   //                { return (TClonesArray*)(ReturnO("TrackSegments", name)) ; }
-  //AliEMCALTrackSegmentMaker * TrackSegmentMaker (const char * name =0) const 
-    //               { return (AliEMCALTrackSegmentMaker*)(ReturnT("TrackSegmentMaker", name)) ; }
-
-  // RecParticles
-  //TClonesArray * RecParticles(const char * name = 0) const  
-    //               { return (TClonesArray*)(ReturnO("RecParticles", name)) ; }
-    //AliEMCALPID * PID(const char * name =0) const 
-      //             { return (AliEMCALPID*)(ReturnT("PID", name)) ; }
-
-  // Primaries
-  const TParticle *           Primary(Int_t index) const ;
-  const Int_t                 NPrimaries()const { return fNPrimaries; }
-
-  void  SetDebug(Int_t level) {fDebug = level;} // Set debug level
 
-  AliEMCALGetter & operator = (const AliEMCALGetter & ) {
-    // assignement operator requested by coding convention, but not needed
-    abort() ;
-    return *this ; 
-  }
-  
-  TFolder * SDigitsFolder() { return dynamic_cast<TFolder*>(fSDigitsFolder->FindObject("EMCAL")) ; }
-
- private:
-
-  AliEMCALGetter(const char* headerFile, const char* branchTitle ="Default", const Option_t * rw ="") ; 
-  void CreateWhiteBoard() const ; 
-  TObject * ReturnO(TString what, TString name=0, TString file=0) const ; 
-  const TTask * ReturnT(TString what,TString name=0) const ; 
-  void DefineBranchTitles(char* branch, char* branchTitle) ;
-  void ReadTreeD() ;
-  void ReadTreeH() ;
-  void ReadTreeR() ;
-  void ReadTreeS(Int_t event) ;
-  //void ReadTreeQA() ;
-  void ReadPrimaries() ;
-
-  TObject ** HitsRef(void) const ;
-  TObject ** SDigitsRef(const char * name, const char * file = 0 ) const;
-  TObject ** DigitsRef (const char * name)   const ;
-  TObject ** TowerRecPointsRef (const char * name) const ;
-  TObject ** PreShoRecPointsRef (const char * name) const ;
-  //TObject ** TrackSegmentsRef(const char * name)   const ;
-  //TObject ** RecParticlesRef (const char * name)   const ;
-  //TObject ** AlarmsRef (void)   const ;
-
-  TObject ** SDigitizerRef (const char * name) const ; 
-  TObject ** DigitizerRef  (const char * name) const ; 
-  TObject ** ClusterizerRef(const char * name) const ; 
-  //TObject ** TSMakerRef    (const char * name) const ; 
-  //TObject ** PIDRef        (const char * name) const ; 
-
- private:
-
-  static TFile *        fFile ;              //!
-  TString        fHeaderFile ;        //! File in which gAlice lives
-  TString        fBranchTitle ;       //!
-  //TString        fTrackSegmentsTitle ;//! 
-  TString        fRecPointsTitle ;    //!
-  //TString        fRecParticlesTitle ; //!
-  TString        fDigitsTitle ;       //!
-  TString        fSDigitsTitle ;      //!
-
-  Bool_t         fFailed ;            //! true if file is not opened and/or galice not found
-  Int_t          fDebug ;             // Debug level
-
-  Int_t          fNPrimaries ;        //! # of primaries
-  
-  TObjArray *    fPrimaries ;         //! list of lists of primaries-for the case of mixing
-
-  TFolder *      fModuleFolder ;      //!Folder that contains the modules 
-  TFolder *      fHitsFolder ;        //!Folder that contains the Hits 
-  TFolder *      fSDigitsFolder ;     //!Folder that contains the SDigits 
-  TFolder *      fDigitsFolder ;      //!Folder that contains the Digits 
-  TFolder *      fRecoFolder ;        //!Folder that contains the reconstructed objects (RecPoints, TrackSegments, RecParticles) 
-  //TFolder *      fQAFolder ;          //!Folder that contains the QA objects  
-  TFolder *      fTasksFolder ;       //!Folder that contains the Tasks (sdigitizer, digitizer, reconstructioner)
+  Int_t ReadTreeD(void) ;
+  Int_t ReadTreeH(void) ;
+  Int_t ReadTreeR(void) ;
+  Int_t ReadTreeT(void) ;
+  Int_t ReadTreeS(void) ;
+  Int_t ReadTreeP(void) ;
+
+
+  void ReadPrimaries(void) ;
  
+  void FitRaw(Bool_t lowGainFlag, TGraph * gLowGain, TGraph * gHighGain, TF1* signalF, Int_t & amp, Double_t & time) ; 
+
+private:
+
+
+  static Int_t          fgDebug ;             //! Debug level
+
+  TString           fLoadingStatus ;     //! tells which trees are loaded
+
+//  AliEMCALCalibrationDB * fcdb ;       //!
+   
+  static AliEMCALLoader * fgEmcalLoader ; // pointer to EMCAL Loader
   static AliEMCALGetter * fgObjGetter; // pointer to the unique instance of the singleton 
+  
+  enum EDataTypes{kHits,kSDigits,kDigits,kRecPoints,kTracks,kNDataTypes};
+
 
-  ClassDef(AliEMCALGetter,1)  // Algorithm class that provides methods to retrieve objects from a list knowing the index 
+  ClassDef(AliEMCALGetter,6)  // Algorithm class that provides methods to retrieve objects from a list knowing the index 
 
 };