/************************************************************************** * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. * * * * Author: The ALICE Off-line Project. * * Contributors are mentioned in the code where appropriate. * * * * Permission to use, copy, modify and distribute this software and its * * documentation strictly for non-commercial purposes is hereby granted * * without fee, provided that the above copyright notice appears in all * * copies and that both the copyright notice and this permission notice * * appear in the supporting documentation. The authors make no claims * * about the suitability of this software for any purpose. It is * * provided "as is" without express or implied warranty. * **************************************************************************/ /* $Id: */ /* $Log: 29.05.2001 Yuri Kharlov: Everywhere reading the treese TTree->GetEvent(i) is replaced by reading the branches TBranch->GetEntry(0) */ /* $Log: 08.2002 Dmitri Peressounko: */ //_________________________________________________________________________ // A singleton. This class should be used in the analysis stage to get // reconstructed objects: Digits, RecPoints, TrackSegments and RecParticles, // instead of directly reading them from galice.root file. This container // ensures, that one reads Digits, made of these particular digits, RecPoints, // made of these particular RecPoints, TrackSegments and RecParticles. // This becomes non trivial if there are several identical branches, produced with // different set of parameters. // // An example of how to use (see also class AliEMCALAnalyser): // AliEMCALGetter * gime = AliEMCALGetter::GetInstance("galice.root","test") ; // for(Int_t irecp = 0; irecp < gime->NRecParticles() ; irecp++) // AliEMCALRecParticle * part = gime->RecParticle(1) ; // ................ // please->GetEvent(event) ; // reads new event from galice.root // //*-- Author: Yves Schutz (SUBATECH) & Dmitri Peressounko (RRC KI & SUBATECH) //*-- Completely redesigned by Dmitri Peressounko March 2001 // //*-- YS June 2001 : renamed the original AliEMCALIndexToObject and make //*-- systematic usage of TFolders without changing the interface //*-- YS August 2002 : clone PHOS as closely as possible and intoduction // of new IO (à la PHOS) ////////////////////////////////////////////////////////////////////////////// // --- ROOT system --- #include "TFile.h" #include "TTree.h" #include "TROOT.h" #include "TObjString.h" #include "TFolder.h" #include "TParticle.h" // --- Standard library --- #include // --- AliRoot header files --- #include "AliRun.h" #include "AliConfig.h" #include "AliEMCALGetter.h" #include "AliEMCAL.h" #include "AliEMCALDigitizer.h" #include "AliEMCALSDigitizer.h" #include "AliEMCALClusterizerv1.h" #include "AliEMCALGeometry.h" ClassImp(AliEMCALGetter) AliEMCALGetter * AliEMCALGetter::fgObjGetter = 0 ; TFile * AliEMCALGetter::fFile = 0 ; //____________________________________________________________________________ AliEMCALGetter::AliEMCALGetter(const char* headerFile, const char* branchTitle, const Bool_t toSplit) { // This is the ctor called by GetInstance and the only one that can be used if ( fHeaderFile.Contains("_") ) { Fatal("AliEMCALGetter","Invalid file name (_ not allowed) %s",fHeaderFile.Data()) ; } //Initialize all data fFailed = kFALSE ; fDebug = 0 ; fToSplit = toSplit ; fHeaderFile = headerFile ; fPrimaries = new TObjArray(1) ; fModuleFolder = dynamic_cast(gROOT->FindObjectAny("Folders/Run/Configuration/Modules")); fPrimariesFolder = dynamic_cast(gROOT->FindObjectAny("Folders/RunMC/Event/Data")); fHitsFolder = dynamic_cast(gROOT->FindObjectAny("Folders/RunMC/Event/Data/Hits")); fSDigitsFolder = dynamic_cast(gROOT->FindObjectAny("Folders/RunMC/Event/Data/SDigits")); fDigitsFolder = dynamic_cast(gROOT->FindObjectAny("Folders/Run/Event/Data")); fRecoFolder = dynamic_cast(gROOT->FindObjectAny("Folders/Run/Event/RecData")); //fQAFolder = dynamic_cast(gROOT->FindObjectAny("Folders/Run/Conditions/QA")); fTasksFolder = dynamic_cast(gROOT->FindObjectAny("Folders/Tasks")) ; //Set titles to branches and create PHOS specific folders SetTitle(branchTitle) ; if ( fHeaderFile != "aliroot" ) { // to call the getter without a file //open headers file fFile = static_cast(gROOT->GetFile(fHeaderFile.Data() ) ) ; if(!fFile){ //if file was not opened yet, read gAlice fFile = TFile::Open(fHeaderFile.Data(), "update") ; if (!fFile->IsOpen()) { Error("AliEMCALGetter","Cannot open %s",fHeaderFile.Data()) ; fFailed = kTRUE ; return ; } gAlice = static_cast(fFile->Get("gAlice")) ; } } if (!gAlice) { Error("AliEMCALGetter","Cannot find gAlice in %s",fHeaderFile.Data()) ; fFailed = kTRUE ; return ; } if (!EMCAL()) { if (fDebug) Info("AliEMCALGetter","Posting EMCAL to Folders") ; if (gAlice->GetDetector("EMCAL")) { AliConfig * conf = AliConfig::Instance() ; conf->Add(static_cast(gAlice->GetDetector("EMCAL"))) ; conf->Add(static_cast(gAlice->GetDetector("EMCAL"))) ; } else Error("AliEMCALGetter"," Detector EMCAL not found"); } fDebug=0; } //____________________________________________________________________________ AliEMCALGetter::~AliEMCALGetter() { if (fPrimaries) { fPrimaries->Delete() ; delete fPrimaries ; } TFolder * emcalF = dynamic_cast(fSDigitsFolder->FindObject("EMCAL")) ; TCollection * folderslist = emcalF->GetListOfFolders() ; TIter next(folderslist) ; TFolder * folder = 0 ; while ( (folder = static_cast(next())) ) emcalF->Remove(folder) ; if (fFile) { fFile->Close() ; delete fFile ; fFile = 0 ; } fgObjGetter = 0 ; } //____________________________________________________________________________ void AliEMCALGetter::CloseFile() { delete gAlice ; gAlice = 0 ; } //____________________________________________________________________________ const TFolder * AliEMCALGetter::Folder(const TString what) const { // returns the EMCAL folder required by what // what = hits, sdigits, digits if ( what == "hits" ) return dynamic_cast(fHitsFolder->FindObject("EMCAL")) ; else if ( what == "sdigits" ) return dynamic_cast(fSDigitsFolder->FindObject("EMCAL")) ; else if ( what == "digits" ) return dynamic_cast(fDigitsFolder->FindObject("EMCAL")) ; else { Error("GetFolder","%s illegal option (hits, sdigits, digits) ", what.Data()) ; return 0 ; } } //____________________________________________________________________________ AliEMCALGetter * AliEMCALGetter::GetInstance() { // Returns the pointer of the unique instance already defined if ( fgObjGetter ) { return fgObjGetter ; } else { return 0 ; } } //____________________________________________________________________________ AliEMCALGetter * AliEMCALGetter::GetInstance(const char* headerFile, const char* branchTitle, const Bool_t toSplit) { // Creates and returns the pointer of the unique instance // Must be called only when the environment has changed if(!fgObjGetter){ fgObjGetter = new AliEMCALGetter(headerFile,branchTitle,toSplit) ; if(fgObjGetter->fFailed) return 0; else return fgObjGetter ; } //First checks, if header file already opened if(!fgObjGetter->fFile){ fgObjGetter = new AliEMCALGetter(headerFile,branchTitle,toSplit) ; if(fgObjGetter->fFailed) return 0; else return fgObjGetter ; } if(fgObjGetter->fHeaderFile.CompareTo(headerFile)==0){ //Opened the same header file if((fgObjGetter->fBranchTitle.CompareTo(branchTitle) == 0)&& //Open the same branch title (toSplit==fgObjGetter->fToSplit)){ //Nothing should be cleaned } else{ //Clean all data and AliEMCAL...zers if(fgObjGetter->fToSplit) fgObjGetter->CloseSplitFiles() ; fgObjGetter->CleanWhiteBoard() ; fgObjGetter->fToSplit = toSplit ; fgObjGetter->SetTitle(branchTitle) ; } } else{ //Close already opened files, clean memory and open new header file if(gAlice) delete gAlice ; gAlice= 0; if(fgObjGetter->fFile){ fgObjGetter->fFile->Close() ; fgObjGetter->fFile=0; } if(fgObjGetter->fToSplit) fgObjGetter->CloseSplitFiles() ; fgObjGetter->CleanWhiteBoard() ; fgObjGetter = new AliEMCALGetter(headerFile,branchTitle,toSplit) ; } return fgObjGetter ; } //____________________________________________________________________________ const Bool_t AliEMCALGetter::BranchExists(const TString recName) const { //Looks in the tree Tree"name" if branch with current name olready exists TString filename("") ; TString name, dataname, zername; if(recName == "SDigits"){ filename=fSDigitsFileName ; name = "TreeS0" ; dataname = "EMCAL" ; zername = "AliEMCALSDigitizer" ; } else if(recName == "Digits"){ filename=fDigitsFileName ; name = "TreeD0" ; dataname = "EMCAL" ; zername = "AliEMCALDigitizer" ; } else if(recName =="RecPoints"){ filename=fRecPointsFileName ; name = "TreeR0" ; dataname = "EMCALEmcRP" ; zername = "AliEMCALClusterizer" ; } else if(recName == "TrackSegments"){ filename=fTrackSegmentsFileName ; name = "TreeR0" ; dataname = "EMCALTS" ; zername = "AliEMCALTrackSegmentMaker" ; } else if(recName == "RecParticles"){ filename= fRecParticlesFileName ; name = "TreeR0" ; dataname = "EMCALRP" ; zername = "AliEMCALPID" ; } else return kFALSE ; TFile * file ; TTree * tree ; if(fToSplit){ file = static_cast(gROOT->GetFile(filename.Data() ) ) ; if(!file) file = TFile::Open(fSDigitsFileName.Data(),"update"); } else file = fFile ; tree = (TTree *)file->Get(name.Data()) ; if(!tree ) return kFALSE ; TObjArray * lob = static_cast(tree->GetListOfBranches()) ; TIter next(lob) ; TBranch * branch = 0 ; TString titleName(fBranchTitle); titleName+=":"; while ((branch = (static_cast(next())))) { TString branchName(branch->GetName() ) ; TString branchTitle(branch->GetTitle() ) ; if ( branchName.BeginsWith(dataname) && branchTitle.BeginsWith(fBranchTitle) ){ Warning("BranchExists", "branch %s with title %s already exits in %s", dataname.Data(), fBranchTitle.Data(), name.Data()); return kTRUE ; } if ( branchName.BeginsWith(zername) && branchTitle.BeginsWith(titleName) ){ Warning("BranchExists","Branch AliEMCAL... with title %s already exits in %s",branch->GetTitle(), name.Data()); return kTRUE ; } } //We can't delete three if gAlice points to it... To be redisigned somehow???!!! if(!fToSplit){ if(name.Contains("TreeS")) if(tree!=gAlice->TreeS()) tree->Delete(); if(name.Contains("TreeD")) if(tree!=gAlice->TreeD()) tree->Delete(); if(name.Contains("TreeR")) if(tree!=gAlice->TreeR()) tree->Delete(); } return kFALSE ; } //____________________________________________________________________________ void AliEMCALGetter::ListBranches(Int_t event) const { TBranch * branch = 0 ; if (gAlice->GetEvent(event) == -1) return ; TTree * t = gAlice->TreeH() ; if(t){ Info("ListBranches"," -> ****** Hits : "); TObjArray * lob = t->GetListOfBranches() ; TIter next(lob) ; while ( (branch = static_cast(next())) ) Info("ListBranches"," %s", branch->GetName()); } else Warning("ListBranches"," -> TreeH not found for event %d",event); t = gAlice->TreeS() ; if(t){ Info("ListBranches"," -> ****** SDigits : "); TObjArray * lob = t->GetListOfBranches() ; TIter next(lob) ; while ( (branch = static_cast(next())) ) Info("ListBranches"," %s %s",branch->GetName(),branch->GetTitle()); } else Warning("ListBranches"," -> TreeS not found for event %d",event); t = gAlice->TreeD() ; if(t){ Info("ListBranches"," -> ****** Digits : "); TObjArray * lob = t->GetListOfBranches() ; TIter next(lob) ; while ( (branch = static_cast(next())) ) Info(" %s %s", branch->GetName(), branch->GetTitle()); } else Warning("ListBranches"," -> TreeD not found for event %d", event); t = gAlice->TreeR() ; if(t){ Info("ListBranches"," -> ****** Recon : "); TObjArray * lob = t->GetListOfBranches() ; TIter next(lob) ; while ( (branch = static_cast(next())) ) Info(" %s %s", branch->GetName(), branch->GetTitle()); } else Warning("ListBranches"," -> TreeR not found for event %d",event); } //____________________________________________________________________________ void AliEMCALGetter::NewBranch(TString name, Int_t event) { fBranchTitle = fSDigitsTitle = fDigitsTitle = fRecPointsTitle = fTrackSegmentsTitle = fRecParticlesTitle = name ; Event(event) ; } //____________________________________________________________________________ Bool_t AliEMCALGetter::NewFile(TString name) { fHeaderFile = name ; fFile->Close() ; fFailed = kFALSE; fFile = static_cast(gROOT->GetFile(fHeaderFile.Data() ) ) ; if(!fFile) { //if file was not opened yet, read gAlice fFile = TFile::Open(fHeaderFile.Data(),"update") ; if (!fFile->IsOpen()) { Error("NewFile", " -> Cannot open %s", fHeaderFile.Data()); fFailed = kTRUE ; return fFailed ; } gAlice = static_cast(fFile->Get("gAlice")) ; } if (!gAlice) { Error("NewFile"," -> Cannot find gAlice in %s", fHeaderFile.Data()); fFailed = kTRUE ; return fFailed ; } return fFailed ; } //____________________________________________________________________________ const AliEMCAL * AliEMCALGetter::EMCAL() { // returns the EMCAL object AliEMCAL * emcal = dynamic_cast(fModuleFolder->FindObject("EMCAL")) ; if (!emcal) if (fDebug) Warning("EMCAL"," -> EMCAL module not found in Folders" ); return emcal ; } //____________________________________________________________________________ AliEMCALGeometry * AliEMCALGetter::EMCALGeometry() { AliEMCALGeometry * rv = 0 ; if (EMCAL() ) rv = EMCAL()->GetGeometry() ; return rv ; } //____________________________________________________________________________ const Bool_t AliEMCALGetter::PostPrimaries(void) const { //------- Primaries ---------------------- // the hierarchy is //Folders/RunMC/Event/Data/Primaries TFolder * primariesFolder = dynamic_cast(fPrimariesFolder->FindObject("Primaries")) ; if ( !primariesFolder ) { if (fDebug) { Warning("PostPrimaries", "-> Folder //%s/Primaries/ not found!", fPrimariesFolder->GetName()); Info("PostPrimaries", "-> Adding Folder //%s/Primaries/",fPrimariesFolder->GetName()); } primariesFolder = fPrimariesFolder->AddFolder("Primaries", "Primaries particles from TreeK") ; } TClonesArray *primaries= new TClonesArray("TParticle",1000) ; primaries->SetName("Primaries") ; primariesFolder->Add(primaries) ; return kTRUE; } //____________________________________________________________________________ TObject** AliEMCALGetter::PrimariesRef(void) const { //------- Primaries ---------------------- // the hierarchy is //Folders/RunMC/Event/Data/Primaries if ( !fPrimariesFolder ) { Fatal("PrimariesRef", "-> Folder //%s not found!",fPrimariesFolder); } TFolder * primariesFolder = dynamic_cast(fPrimariesFolder->FindObject("Primaries")) ; if ( !primariesFolder ) { Fatal("PrimariesRef", "-> Folder //%s/Primaries/ not found!",fPrimariesFolder); } TObject * p = primariesFolder->FindObject("Primaries") ; if(!p) { Fatal("PrimariesRef", "-> %s/Primaries not found !",primariesFolder->GetName()); } return primariesFolder->GetListOfFolders()->GetObjectRef(p) ; } //____________________________________________________________________________ const Bool_t AliEMCALGetter::PostHits(void) const { //------- Hits ---------------------- // the hierarchy is //Folders/RunMC/Event/Data/EMCAL/Hits TFolder * emcalFolder = dynamic_cast(fHitsFolder->FindObject("EMCAL")) ; if ( !emcalFolder ) { if (fDebug) { Warning("PostHits", "-> Folder //%s/EMCAL/ not found!", fHitsFolder); Info("PostHits", "-> Adding Folder //%s/EMCAL/",fHitsFolder); } emcalFolder = fHitsFolder->AddFolder("EMCAL", "Hits from EMCAL") ; } TClonesArray *hits= new TClonesArray("AliEMCALHit",1000) ; hits->SetName("Hits") ; emcalFolder->Add(hits) ; return kTRUE; } //____________________________________________________________________________ TObject ** AliEMCALGetter::HitsRef(void) const { //------- Hits ---------------------- // the hierarchy is //Folders/RunMC/Event/Data/EMCAL/Hits if ( !fHitsFolder ) { Error("HitsRef", "-> Folder //%s not found!",fHitsFolder); return 0; } TFolder * emcalFolder = dynamic_cast(fHitsFolder->FindObject("EMCAL")) ; if ( !emcalFolder ) { Error("HitsRef", "-> Folder //%s/EMCAL/ not found!",fHitsFolder); return 0; } TObject * h = emcalFolder->FindObject("Hits") ; if(!h) { Error("HitsRef", "-> %s/Hits not found !",emcalFolder->GetName()); return 0 ; } else return emcalFolder->GetListOfFolders()->GetObjectRef(h) ; } //____________________________________________________________________________ const Bool_t AliEMCALGetter::PostSDigits(const char * name, const char * headerFile) const { //---------- SDigits ------------------------- // the hierarchy is //Folders/RunMC/Event/Data/EMCAL/SDigits/headerFile/sdigitsname // because you can have sdigits from several hit files for mixing TFolder * emcalFolder = dynamic_cast(fSDigitsFolder->FindObject("EMCAL")) ; if ( !emcalFolder ) { if (fDebug) { Warning("PostSDigits", "-> Folder //%s/EMCAL/ not found!", fSDigitsFolder); Info("PostSDigits", "-> Adding Folder //%s/EMCAL/",fHitsFolder); } emcalFolder = fSDigitsFolder->AddFolder("EMCAL", "SDigits from EMCAL") ; } TString subdir(headerFile) ; subdir.ReplaceAll("/", "_") ; TFolder * emcalSubFolder = dynamic_cast(emcalFolder->FindObject(subdir)) ; if ( !emcalSubFolder ) emcalSubFolder = emcalFolder->AddFolder(subdir, ""); TObject * sd = emcalSubFolder->FindObject(name); if ( !sd ) { TClonesArray * sdigits = new TClonesArray("AliEMCALDigit",1) ; sdigits->SetName(name) ; emcalSubFolder->Add(sdigits) ; } return kTRUE; } //____________________________________________________________________________ TObject ** AliEMCALGetter::SDigitsRef(const char * name, const char * file) const { //------- SDigits ---------------------- // the hierarchy is //Folders/RunMC/Event/Data/EMCAL/SDigits/filename/SDigits if ( !fSDigitsFolder ) { Fatal("SDigitsRef", "-> Folder //%s not found!", fSDigitsFolder); } TFolder * emcalFolder = dynamic_cast(fSDigitsFolder->FindObject("EMCAL")) ; if ( !emcalFolder ) { Fatal("SDigitsRef", "-> Folder //%s/EMCAL/ not found!", fSDigitsFolder); } TFolder * emcalSubFolder = 0 ; if(file) emcalSubFolder = dynamic_cast(emcalFolder->FindObject(file)) ; else emcalSubFolder = dynamic_cast(emcalFolder->FindObject(fHeaderFile)) ; if(!emcalSubFolder) { Fatal("SDigitsRef", "-> Folder //Folders/RunMC/Event/Data/EMCAL/%s not found!", file); } TObject * dis = emcalSubFolder->FindObject(name) ; if(!dis) { Fatal("SDigitsRef", "-> object %s not found!", name); } return emcalSubFolder->GetListOfFolders()->GetObjectRef(dis) ; } //____________________________________________________________________________ const Bool_t AliEMCALGetter::PostSDigitizer(AliEMCALSDigitizer * sdigitizer) const { //---------- SDigitizer ------------------------- // the hierarchy is //Folders/Tasks/SDigitizer/EMCAL/sdigitsname TTask * sd = dynamic_cast(fTasksFolder->FindObject("SDigitizer")) ; if ( !sd ) { Error("PostSDigitizer", "-> Task //%s/SDigitizer not found!",fTasksFolder); return kFALSE ; } TTask * emcal = dynamic_cast(sd->GetListOfTasks()->FindObject("EMCAL")) ; if ( !emcal ) { if (fDebug) { Warning("PostSDigitizer", "->//%s/SDigitizer/EMCAL/ not found!",fTasksFolder); Info("PostSDigitizer", "-> Adding //%s/SDigitizer/EMCAL/", fTasksFolder); } emcal = new TTask("EMCAL", "") ; sd->Add(emcal) ; } AliEMCALSDigitizer * emcalsd = dynamic_cast(emcal->GetListOfTasks()->FindObject( sdigitizer->GetName() )); if (emcalsd) { if (fDebug) Info("PostSDigitizer", "-> Task %s already exists",sdigitizer->GetName()); emcal->GetListOfTasks()->Remove(emcalsd) ; } emcal->Add(sdigitizer) ; return kTRUE; } //____________________________________________________________________________ TObject ** AliEMCALGetter::SDigitizerRef(const char * name) const { TTask * sd = dynamic_cast(fTasksFolder->FindObject("SDigitizer")) ; if ( !sd ) { Fatal("SDigitizerRef", "-> Task //%s/SDigitizer not found!", fTasksFolder); } TTask * emcal = dynamic_cast(sd->GetListOfTasks()->FindObject("EMCAL")) ; if ( !emcal ) { Fatal("SDigitizerRef", "-> //%s/SDigitizer/EMCAL not found!", fTasksFolder); } TTask * task = dynamic_cast(emcal->GetListOfTasks()->FindObject(name)) ; return emcal->GetListOfTasks()->GetObjectRef(task) ; } //____________________________________________________________________________ const Bool_t AliEMCALGetter::PostSDigitizer(const char * name, const char * file) const { //---------- SDigitizer ------------------------- // the hierarchy is //Folders/Tasks/SDigitizer/EMCAL/sdigitsname TTask * sd = dynamic_cast(fTasksFolder->FindObject("SDigitizer")) ; if ( !sd ) { Error("PostSDigitizer", "-> Task //%s/SDigitizer not found!", fTasksFolder); return kFALSE ; } TTask * emcal = dynamic_cast(sd->GetListOfTasks()->FindObject("EMCAL")) ; if ( !emcal ) { if (fDebug) { Warning("PostSDigitizer", "-> //%s/SDigitizer/EMCAL/ not found!", fTasksFolder); Info("PostSDigitizer", "-> Adding //%s/SDigitizer/EMCAL", fTasksFolder); } emcal = new TTask("EMCAL", "") ; sd->Add(emcal) ; } TString sdname(name) ; sdname.Append(":") ; sdname.Append(file); sdname.ReplaceAll("/","_") ; AliEMCALSDigitizer * emcalsd = dynamic_cast(emcal->GetListOfTasks()->FindObject( sdname )); if (!emcalsd) { emcalsd = new AliEMCALSDigitizer() ; //Note, we can not call constructor with parameters: it will call Getter and screw up everething emcalsd->SetName(sdname) ; emcalsd->SetTitle(file) ; emcal->Add(emcalsd) ; } return kTRUE; } //____________________________________________________________________________ const Bool_t AliEMCALGetter::PostDigits(const char * name) const { //---------- Digits ------------------------- // the hierarchy is //Folders/Run/Event/Data/EMCAL/SDigits/name TFolder * emcalFolder = dynamic_cast(fDigitsFolder->FindObject("EMCAL")) ; if ( !emcalFolder ) { if (fDebug) { Warning("PostDigits", "-> Folder //%s/EMCAL/ not found!", fDigitsFolder); Info("PostDigits", "-> Adding Folder //%s/EMCAL/", fDigitsFolder); } emcalFolder = fDigitsFolder->AddFolder("EMCAL", "Digits from EMCAL") ; } TObject* dig = emcalFolder->FindObject( name ) ; if ( !dig ) { TClonesArray * digits = new TClonesArray("AliEMCALDigit",1000) ; digits->SetName(name) ; emcalFolder->Add(digits) ; } return kTRUE; } //____________________________________________________________________________ TObject ** AliEMCALGetter::DigitsRef(const char * name) const { //------- Digits ---------------------- // the hierarchy is //Folders/Run/Event/Data/EMCAL/Digits/name if ( !fDigitsFolder ) { Fatal("DigitsRef", "-> Folder //%s not found!", fDigitsFolder); } TFolder * emcalFolder = dynamic_cast(fDigitsFolder->FindObject("EMCAL")) ; if ( !emcalFolder ) { Fatal("DigitsRef", "-> Folder //%s/EMCAL/ not found!", fDigitsFolder); } TObject * d = emcalFolder->FindObject(name) ; if(!d) { Fatal("DigitsRef", "-> object %s not found!", name); } return emcalFolder->GetListOfFolders()->GetObjectRef(d) ; } //____________________________________________________________________________ const Bool_t AliEMCALGetter::PostDigitizer(AliEMCALDigitizer * digitizer) const { //---------- Digitizer ------------------------- TTask * sd = dynamic_cast(fTasksFolder->FindObject("Digitizer")) ; if ( !sd ) { Error("PostDigitizer", "-> Task //%s/Digitizer not found!", fTasksFolder); return kFALSE ; } TTask * emcal = dynamic_cast(sd->GetListOfTasks()->FindObject("EMCAL")) ; if ( !emcal ) { if (fDebug) { Warning("PostDigitizer", "-> //%s/Digitizer/EMCAL not found!", fTasksFolder); Info("PostDigitizer", "-> Adding //%s/Digitizer/EMCAL", fTasksFolder); } emcal = new TTask("EMCAL", "") ; sd->Add(emcal) ; } AliEMCALDigitizer * emcald = dynamic_cast(emcal->GetListOfTasks()->FindObject(digitizer->GetName())) ; if (emcald) { emcald->Delete() ; emcal->GetListOfTasks()->Remove(emcald) ; } emcal->Add(digitizer) ; return kTRUE; } //____________________________________________________________________________ const Bool_t AliEMCALGetter::PostDigitizer(const char * name) const { //---------- Digitizer ------------------------- // the hierarchy is //Folders/Tasks/SDigitizer/EMCAL/sdigitsname TTask * d = dynamic_cast(fTasksFolder->FindObject("Digitizer")) ; if ( !d ) { Error("PostDigitizer", "-> Task //%s/Digitizer not found!", fTasksFolder); return kFALSE ; } TTask * emcal = dynamic_cast(d->GetListOfTasks()->FindObject("EMCAL")) ; if ( !emcal ) { if (fDebug) { Warning("PostDigitizer", "-> //%s/Digitizer/EMCAL not found!", fTasksFolder); Info("PostDigitizer", "-> Adding //%s/Digitizer/EMCAL", fTasksFolder); } emcal = new TTask("EMCAL", "") ; d->Add(emcal) ; } AliEMCALDigitizer * emcald = dynamic_cast(emcal->GetListOfTasks()->FindObject(name)) ; if (!emcald) { emcald = new AliEMCALDigitizer() ; emcald->SetName(fDigitsTitle) ; emcald->SetTitle(fHeaderFile) ; emcal->Add(emcald) ; } return kTRUE; } //____________________________________________________________________________ TObject ** AliEMCALGetter::DigitizerRef(const char * name) const { TTask * sd = dynamic_cast(fTasksFolder->FindObject("Digitizer")) ; if ( !sd ) { Fatal("DigitizerRef", "-> Task //%s/Digitizer not found!", fTasksFolder->GetName()); } TTask * emcal = dynamic_cast(sd->GetListOfTasks()->FindObject("EMCAL")) ; if ( !emcal ) { Fatal("DigitizerRef", "-> //%s/Digitizer/EMCAL", fTasksFolder->GetName()); } TTask * task = dynamic_cast(emcal->GetListOfTasks()->FindObject(name)) ; return emcal->GetListOfTasks()->GetObjectRef(task) ; } //____________________________________________________________________________ const Bool_t AliEMCALGetter::PostRecPoints(const char * name) const { // -------------- RecPoints ------------------------------------------- // the hierarchy is //Folders/Run/Event/RecData/EMCAL/TowerRecPoints/name // the hierarchy is //Folders/Run/Event/RecData/EMCAL/PreShowerRecPoints/name TFolder * emcalFolder = dynamic_cast(fRecoFolder->FindObject("EMCAL")) ; if ( !emcalFolder ) { if (fDebug) { Warning("PostRecPoints", "-> Folder //%s/EMCAL/ not found!", fRecoFolder); Info("PostRecPoints", "-> Adding Folder //%s/EMCAL/", fRecoFolder); } emcalFolder = fRecoFolder->AddFolder("EMCAL", "Reconstructed data from EMCAL") ; } // Tower RecPoints TFolder * emcalRPoTowerFolder = dynamic_cast(emcalFolder->FindObject("TowerRecPoints")) ; if ( !emcalRPoTowerFolder ) { if (fDebug) { Warning("PostRecPoints", "-> Folder //%s/EMCAL/TowerRecPoints/ not found!", fRecoFolder); Info("PostRecPoints", "-> Adding Folder //%s/EMCAL/TowerRecPoints not found!", fRecoFolder); } emcalRPoTowerFolder = emcalFolder->AddFolder("TowerRecPoints", "Tower RecPoints from EMCAL") ; } TObject * erp = emcalFolder->FindObject( name ) ; if ( !erp ) { TObjArray * towerrp = new TObjArray(100) ; towerrp->SetName(name) ; emcalRPoTowerFolder->Add(towerrp) ; } // Pre Shower RecPoints TFolder * emcalRPoPreShoFolder = dynamic_cast(emcalFolder->FindObject("PreShowerRecPoints")) ; if ( !emcalRPoPreShoFolder ) { if (fDebug) { Warning("PostRecPoints", "-> Folder //%s/EMCAL/PreShowerRecPoints/ not found!", fRecoFolder); Info("PostRecPoints", "-> Adding Folder //%s/EMCAL/PreShowerRecPoints/", fRecoFolder); } emcalRPoPreShoFolder = emcalFolder->AddFolder("PreShowerRecPoints", "PreSho RecPoints from EMCAL") ; } TObject * crp = emcalRPoPreShoFolder->FindObject( name ) ; if ( !crp ) { TObjArray * preshorp = new TObjArray(100) ; preshorp->SetName(name) ; emcalRPoPreShoFolder->Add(preshorp) ; } return kTRUE; } //____________________________________________________________________________ TObject ** AliEMCALGetter::TowerRecPointsRef(const char * name) const { // -------------- RecPoints ------------------------------------------- // the hierarchy is //Folders/Run/Event/RecData/EMCAL/TowerRecPoints/name if ( !fRecoFolder ) { Fatal("TowerRecPointsRef", "-> Folder //%s not found!", fRecoFolder); } TFolder * towerFolder = dynamic_cast(fRecoFolder->FindObject("EMCAL/TowerRecPoints")) ; if ( !towerFolder ) { Fatal("TowerRecPointsRef", "-> Folder //%s/EMCAL/TowerRecPoints/ not found!", fRecoFolder); } TObject * trp = towerFolder->FindObject(name ) ; if ( !trp ) { Fatal("TowerRecPointsRef", "-> Object %s not found!", name); } return towerFolder->GetListOfFolders()->GetObjectRef(trp) ; } //____________________________________________________________________________ TObject ** AliEMCALGetter::PreShowerRecPointsRef(const char * name) const { // -------------- RecPoints ------------------------------------------- // the hierarchy is //Folders/Run/Event/RecData/EMCAL/PreShowerRecPoints/name if ( !fRecoFolder ) { Fatal("PreShowerRecPointsRef", "-> Folder //%s not found!", fRecoFolder); } TFolder * preshoFolder = dynamic_cast(fRecoFolder->FindObject("EMCAL/PreShowerRecPoints")) ; if ( !preshoFolder ) { Fatal("PreShowerRecPointsRef", "-> Folder //%s/EMCAL/PreShowerRecPoints/", fRecoFolder); } TObject * prp = preshoFolder->FindObject(name ) ; if ( !prp ) { Fatal("PreShowerRecPointsRef", "-> Object %s not found!", name); } return preshoFolder->GetListOfFolders()->GetObjectRef(prp) ; } //____________________________________________________________________________ const Bool_t AliEMCALGetter::PostClusterizer(AliEMCALClusterizer * clu) const { // ------------------ AliEMCALClusterizer ------------------------ // the hierarchy is //Folders/Tasks/Reconstructioner/EMCAL/sdigitsname TTask * tasks = dynamic_cast(fTasksFolder->FindObject("Reconstructioner")) ; if ( !tasks ) { Error("PostClusterizer", "-> Task //%s/Reconstructioner not found!", fTasksFolder); return kFALSE ; } TTask * emcal = dynamic_cast(tasks->GetListOfTasks()->FindObject("EMCAL")) ; if ( !emcal ) { if (fDebug) { Warning("PostClusterizer", "-> //%s/ReconstructionerEMCAL not found!", fTasksFolder); Info("PostClusterizer", "-> Adding //%s/Reconstructioner/EMCAL", fTasksFolder); } emcal = new TTask("EMCAL", "") ; tasks->Add(emcal) ; } AliEMCALClusterizerv1 * emcalcl = dynamic_cast(emcal->GetListOfTasks()->FindObject(clu->GetName())) ; if (emcalcl) { if (fDebug) Info("PostClusterizer", "-> Task %s already exists", clu->GetName()); emcalcl->Delete() ; emcal->GetListOfTasks()->Remove(emcalcl) ; } emcal->Add(clu) ; return kTRUE; } //____________________________________________________________________________ TObject ** AliEMCALGetter::ClusterizerRef(const char * name) const { // ------------------ AliEMCALClusterizer ------------------------ TTask * tasks = dynamic_cast(fTasksFolder->FindObject("Reconstructioner")) ; if ( !tasks ) { Fatal("ClusterizerRef", "-> Task //%s/Reconstructioner not found!", fTasksFolder->GetName()); } TTask * emcal = dynamic_cast(tasks->GetListOfTasks()->FindObject("EMCAL")) ; if ( !emcal ) { Fatal("ClusterizerRef", "-> //%s/Reconstructioner/EMCAL", fTasksFolder->GetName()); } TList * l = emcal->GetListOfTasks() ; TIter it(l) ; TTask * task ; TTask * clu = 0 ; TString cluname(name) ; cluname+=":clu-" ; while((task = static_cast(it.Next()) )){ TString taskname(task->GetName()) ; if(taskname.BeginsWith(cluname)){ clu = task ; break ; } } if(!clu) { Fatal("ClusterizerRef", "-> task %s not found!", task->GetName()); } return l->GetObjectRef(clu) ; } //____________________________________________________________________________ const Bool_t AliEMCALGetter::PostClusterizer(const char * name) const { // ------------------ AliEMCALClusterizer ------------------------ // the hierarchy is //Folders/Tasks/Reconstructioner/EMCAL/sdigitsname TTask * tasks = dynamic_cast(fTasksFolder->FindObject("Reconstructioner")) ; if ( !tasks ) { Error("PostClusterizer", "-> Task //%s/Reconstructioner not found!", fTasksFolder->GetName()); return kFALSE ; } TTask * emcal = dynamic_cast(tasks->GetListOfTasks()->FindObject("EMCAL")) ; if ( !emcal ) { if (fDebug) { Warning("PostClusterizer", "-> //%s/Reconstructioner/EMCAL not found!", fTasksFolder); Info("PostClusterizer", "-> Adding //%s/Reconstructioner/EMCAL", fTasksFolder); } emcal = new TTask("EMCAL", "") ; tasks->Add(emcal) ; } TList * l = emcal->GetListOfTasks() ; TIter it(l) ; TString clun(name) ; clun+=":clu" ; TTask * task ; while((task = static_cast(it.Next()) )){ TString taskname(task->GetName()) ; if(taskname.BeginsWith(clun)) return kTRUE ; } AliEMCALClusterizerv1 * emcalcl = new AliEMCALClusterizerv1() ; clun+="-v1" ; emcalcl->SetName(clun) ; emcalcl->SetTitle(fHeaderFile) ; emcal->Add(emcalcl) ; return kTRUE; } //____________________________________________________________________________ TTree * AliEMCALGetter::TreeK(TString filename) { // returns TreeK from file filename // usefull in case of split file if ( filename.IsNull() ) filename = fHeaderFile ; TFile * file = 0 ; file = static_cast(gROOT->GetFile(filename.Data() ) ) ; if (!file) { // file not yet open file = TFile::Open(filename.Data(), "read") ; } TString treeName("TreeK") ; treeName += EventNumber() ; TTree * tree = static_cast(file->Get(treeName.Data())) ; if (!tree && fDebug) Warning("TreeK", "-> %s not found in %s", treeName.Data(),filename.Data()); return tree ; } //____________________________________________________________________________ TTree * AliEMCALGetter::TreeH(TString filename) { // returns TreeH from file filename // usefull in case of split file if ( filename.IsNull() ) filename = fHeaderFile ; TFile * file = 0 ; file = static_cast(gROOT->GetFile(filename.Data() ) ) ; if (!file) { // file not open yet file = TFile::Open(filename.Data(), "read") ; } TString treeName("TreeH") ; treeName += EventNumber() ; TTree * tree = static_cast(file->Get(treeName.Data())) ; if (!tree && fDebug) Warning("TreeH", "-> %s not found in %s", treeName.Data(), filename.Data()); return tree ; } //____________________________________________________________________________ TTree * AliEMCALGetter::TreeS(TString filename) { // returns TreeS from file filename // usefull in case of split file if ( filename.IsNull() ) filename = fHeaderFile ; TFile * file = 0 ; file = static_cast(gROOT->GetFile(filename.Data() ) ) ; if (!file) { // file not open yet file = TFile::Open(filename.Data(), "read") ; } TString treeName("TreeS") ; treeName += EventNumber() ; TTree * tree = static_cast(file->Get(treeName.Data())) ; if (!tree && fDebug) Warning("TreeS", "-> %s not found in %s", treeName.Data(), filename.Data()); return tree ; } //____________________________________________________________________________ TTree * AliEMCALGetter::TreeD(TString filename) { // returns TreeD from file filename // usefull in case of split file if ( filename.IsNull() ) filename = fHeaderFile ; TFile * file = 0 ; file = static_cast(gROOT->GetFile(filename.Data() ) ) ; if (!file) { // file not open yet file = TFile::Open(filename.Data(), "read") ; } TString treeName("TreeD") ; treeName += EventNumber() ; TTree * tree = static_cast(file->Get(treeName.Data())) ; if (!tree && fDebug) Warning("TreeD", "-> %s not found in %s", treeName.Data(), filename.Data()); return tree ; } //____________________________________________________________________________ const TParticle * AliEMCALGetter::Primary(Int_t index) const { // Return primary particle numbered by if(index < 0) return 0 ; TParticle * p = 0 ; p = gAlice->Particle(index) ; return p ; } //____________________________________________________________________________ const TParticle * AliEMCALGetter::Secondary(TParticle* p, Int_t index) const { // Return first (index=1) or second (index=2) secondary particle of primary particle p if(index <= 0) return 0 ; if(index > 2) return 0 ; if(p) { Int_t daughterIndex = p->GetDaughter(index-1) ; return gAlice->Particle(daughterIndex) ; } else return 0 ; } //____________________________________________________________________________ Int_t AliEMCALGetter::ReadTreeD(const Int_t event) { // Read the digit tree gAlice->TreeD() TTree * treeD ; if(fToSplit){ TFile * file = static_cast(gROOT->GetFile(fDigitsFileName)); if(!file) file = TFile::Open(fDigitsFileName) ; // Get Digits Tree header from file TString treeName("TreeD") ; treeName += event ; treeD = dynamic_cast(file->Get(treeName.Data())); if(!treeD){ // TreeD not found in header file if (fDebug) Warning("ReadTreeD", "-> Cannot find TreeD in %s", fDigitsFileName.Data()); return 1; } } else treeD = gAlice->TreeD() ; TObjArray * lob = static_cast(treeD->GetListOfBranches()) ; TIter next(lob) ; TBranch * branch = 0 ; TBranch * digitsbranch = 0 ; TBranch * digitizerbranch = 0 ; Bool_t emcalfound = kFALSE, digitizerfound = kFALSE ; while ( (branch = static_cast(next())) && (!emcalfound || !digitizerfound) ) { if ( (strcmp(branch->GetName(), "EMCAL")==0) && (strcmp(branch->GetTitle(), fDigitsTitle)==0) ) { digitsbranch = branch ; emcalfound = kTRUE ; } else if ( (strcmp(branch->GetName(), "AliEMCALDigitizer")==0) && (strcmp(branch->GetTitle(), fDigitsTitle)==0) ) { digitizerbranch = branch ; digitizerfound = kTRUE ; } } if ( !emcalfound || !digitizerfound ) { if (fDebug) Warning("ReadTreeD", "-> Cannot find Digits and/or Digitizer with name %s", fDigitsTitle.Data()); return 2; } //read digits if(!Digits(fDigitsTitle) ) PostDigits(fDigitsTitle); digitsbranch->SetAddress(DigitsRef(fDigitsTitle)) ; digitsbranch->GetEntry(0) ; // read the Digitizer RemoveTask("D", fDigitsTitle) ; // I do not understand why I need that if(!Digitizer(fDigitsTitle)) PostDigitizer(fDigitsTitle) ; digitizerbranch->SetAddress(DigitizerRef(fDigitsTitle)) ; digitizerbranch->GetEntry(0) ; lob->Delete(); if(gAlice->TreeD()!=treeD) treeD->Delete(); return 0 ; } //____________________________________________________________________________ Int_t AliEMCALGetter::ReadTreeH() { // Read the first entry of EMCAL branch in hit tree gAlice->TreeH() TTree * treeH = gAlice->TreeH() ; if(!treeH) {// TreeH not found in header file if (fDebug) Warning("ReadTreeH", "-> Cannot find TreeH in %s", fHeaderFile.Data()); TString searchFileName("EMCAL.HITS") ; if((strcmp(fBranchTitle.Data(),"Default")!=0)&&(strcmp(fBranchTitle.Data(),"")!=0)){ searchFileName+="." ; searchFileName += fBranchTitle ; } searchFileName+=".root" ; if ( (treeH = TreeH(searchFileName)) ) { //found TreeH in the file which contains the hits if (fDebug) Info("ReadTreeH", "-> TreeH found in %s", searchFileName.Data()); } else { Error("ReadTreeH", "-> TreeH not found "); return 1; } } TBranch * hitsbranch = static_cast(gAlice->TreeH()->GetBranch("EMCAL")) ; if ( !hitsbranch ) { if (fDebug) Warning("ReadTreeH", "-> Cannot find branch EMCAL"); return 2; } if(!Hits()) PostHits() ; if (hitsbranch->GetEntries() > 1 ) { (dynamic_cast (*HitsRef()))->Clear() ; TClonesArray * tempo = new TClonesArray("AliEMCALHit",1000) ; TClonesArray * hits = dynamic_cast(*HitsRef()) ; hitsbranch->SetAddress(&tempo) ; Int_t index = 0 ; Int_t i = 0 ; for (i = 0 ; i < hitsbranch->GetEntries() ; i++) { hitsbranch->GetEntry(i) ; Int_t j = 0 ; for ( j = 0 ; j < tempo->GetEntries() ; j++) { const AliEMCALHit * hit = static_cast(tempo->At(j)) ; new((*hits)[index]) AliEMCALHit( *hit ) ; index++ ; } } delete tempo ; } else { (dynamic_cast (*HitsRef()))->Clear() ; hitsbranch->SetAddress(HitsRef()) ; hitsbranch->GetEntry(0) ; } return 0 ; } //____________________________________________________________________________ void AliEMCALGetter::Track(const Int_t itrack) { // Read the first entry of EMCAL branch in hit tree gAlice->TreeH() if(gAlice->TreeH()== 0){ Error("ReadTreeH", "-> Cannot read TreeH "); return ; } TBranch * hitsbranch = dynamic_cast(gAlice->TreeH()->GetListOfBranches()->FindObject("EMCAL")) ; if ( !hitsbranch ) { if (fDebug) Warning("ReadTreeH", "-> Cannot find branch EMCAL"); return ; } if(!Hits()) PostHits() ; (dynamic_cast (*HitsRef()))->Clear() ; hitsbranch->SetAddress(HitsRef()) ; hitsbranch->GetEntry(itrack) ; } //____________________________________________________________________________ void AliEMCALGetter::ReadTreeQA() { if (fDebug) Warning("ReadTreeQA", "-> %s not implemented", ClassName()); } //____________________________________________________________________________ Int_t AliEMCALGetter::ReadTreeR(const Int_t event) { // Read the reconstrunction tree gAlice->TreeR() // A particularity has been introduced here : // if gime->Event(ievent,"R") is called branches with the current title are read, the current title // being for example give in AliEMCALPID(fileName, title) // if gime(Event(ievent, "RA") is called the title of the branches is not checked anymore, "A" stands for any // This is a feature needed by PID to be able to reconstruct several times particles (each time a ther title is given) // from a given set of TrackSegments (with a given name) // This is why any is NOT used to read the branch of RecParticles // any migh have become obsolete : to be checked // See AliEMCALPIDv1 TTree * treeR ; if(fToSplit){ TFile * file = static_cast(gROOT->GetFile(fRecPointsFileName)); if(!file) file = TFile::Open(fRecPointsFileName) ; // Get Digits Tree header from file TString treeName("TreeR") ; treeName += event ; treeR = dynamic_cast(file->Get(treeName.Data())); if(!treeR){ // TreeR not found in header file if (fDebug) Warning("ReadTreeD", "-> Cannot find TreeR in %s", fRecPointsFileName.Data()); return 1; } } else treeR = gAlice->TreeR() ; // RecPoints TObjArray * lob = static_cast(treeR->GetListOfBranches()) ; TIter next(lob) ; TBranch * branch = 0 ; TBranch * towerbranch = 0 ; TBranch * preshowerbranch = 0 ; TBranch * clusterizerbranch = 0 ; Bool_t emcalTowerRPfound = kFALSE, emcalPreShoRPfound = kFALSE, clusterizerfound = kFALSE ; while ( (branch = static_cast(next())) && (!emcalTowerRPfound || !emcalPreShoRPfound || !clusterizerfound) ) { if(strcmp(branch->GetTitle(), fRecPointsTitle)==0 ) { if ( strcmp(branch->GetName(), "EMCALTowerRP")==0) { towerbranch = branch ; emcalTowerRPfound = kTRUE ; } else if ( strcmp(branch->GetName(), "EMCALPreShoRP")==0) { preshowerbranch = branch ; emcalPreShoRPfound = kTRUE ; } else if(strcmp(branch->GetName(), "AliEMCALClusterizer")==0){ clusterizerbranch = branch ; clusterizerfound = kTRUE ; } } } if ( !emcalTowerRPfound || !emcalPreShoRPfound || !clusterizerfound) { if (fDebug) Warning("ReadTreeR", "-> Cannot find RecPoints and/or Clusterizer with name %s", fRecPointsTitle.Data()); } else { if(!TowerRecPoints(fRecPointsTitle) ) PostRecPoints(fRecPointsTitle) ; towerbranch->SetAddress(TowerRecPointsRef(fRecPointsTitle)) ; towerbranch->GetEntry(0) ; preshowerbranch->SetAddress(PreShowerRecPointsRef(fRecPointsTitle)) ; preshowerbranch->GetEntry(0) ; if(!Clusterizer(fRecPointsTitle) ) PostClusterizer(fRecPointsTitle) ; clusterizerbranch->SetAddress(ClusterizerRef(fRecPointsTitle)) ; clusterizerbranch->GetEntry(0) ; } if(gAlice->TreeR()!=treeR) treeR->Delete(); return 0 ; } //____________________________________________________________________________ Int_t AliEMCALGetter::ReadTreeS(const Int_t event) { // Reads the SDigits treeS from all files // Files, which should be opened are listed in emcalF // So, first get list of files TFolder * emcalF = dynamic_cast(fSDigitsFolder->FindObject("EMCAL")) ; if (!emcalF) emcalF = fSDigitsFolder->AddFolder("EMCAL", "SDigits from EMCAL") ; TCollection * folderslist = emcalF->GetListOfFolders() ; // Now iterate over the list of files and read TreeS into Whiteboard TIter next(folderslist) ; TFolder * folder = 0 ; TFile * file; TTree * treeS = 0; while ( (folder = static_cast(next())) ) { TString fileName("") ; if(fToSplit) fileName = folder->GetTitle() ; else fileName = folder->GetName() ; fileName.ReplaceAll("_","/") ; file = static_cast(gROOT->GetFile(fileName)); if(!file) file = TFile::Open(fileName) ; // Get SDigits Tree header from file TString treeName("TreeS") ; treeName += event ; treeS = dynamic_cast(file->Get(treeName.Data())); if(!treeS){ // TreeS not found in header file if (fDebug) Warning("ReadTreeS", "-> Cannot find TreeS in %s", fileName.Data()); return 1; } //set address of the SDigits and SDigitizer TBranch * sdigitsBranch = 0; TBranch * sdigitizerBranch = 0; TBranch * branch = 0 ; TObjArray * lob = static_cast(treeS->GetListOfBranches()) ; TIter next(lob) ; Bool_t emcalfound = kFALSE, sdigitizerfound = kFALSE ; while ( (branch = static_cast(next())) && (!emcalfound || !sdigitizerfound) ) { if ( (strcmp(branch->GetName(), "EMCAL")==0) && (strcmp(branch->GetTitle(), fSDigitsTitle)==0) ) { emcalfound = kTRUE ; sdigitsBranch = branch ; } else if ( (strcmp(branch->GetName(), "AliEMCALSDigitizer")==0) && (strcmp(branch->GetTitle(), fSDigitsTitle)==0) ) { sdigitizerfound = kTRUE ; sdigitizerBranch = branch ; } } if ( !emcalfound || !sdigitizerfound ) { if (fDebug) Warning("ReadSDigits", "-> Digits and/or Digitizer branch with name %s not found", GetName()); return 2; } if ( !folder->FindObject(fSDigitsTitle) ) PostSDigits(fSDigitsTitle,folder->GetName()) ; ((TClonesArray*) (*SDigitsRef(fSDigitsTitle,folder->GetName())))->Clear() ; sdigitsBranch->SetAddress(SDigitsRef(fSDigitsTitle,folder->GetName())) ; sdigitsBranch->GetEntry(0) ; TString sdname(fSDigitsTitle) ; sdname+=":" ; sdname+=folder->GetName() ; if(!SDigitizer(sdname) ) PostSDigitizer(fSDigitsTitle,folder->GetName()) ; sdigitizerBranch->SetAddress(SDigitizerRef(sdname)) ; sdigitizerBranch->GetEntry(0) ; if(gAlice->TreeS()!=treeS) treeS->Delete(); } return 0 ; } //____________________________________________________________________________ void AliEMCALGetter::ReadTreeS(TTree * treeS, Int_t input) { // Read the summable digits fron treeS() TString filename("mergefile") ; filename+= input ; TFolder * emcalFolder = dynamic_cast(fSDigitsFolder->FindObject("EMCAL")) ; if ( !emcalFolder ) { emcalFolder = fSDigitsFolder->AddFolder("EMCAL", "SDigits from EMCAL") ; } TFolder * folder=(TFolder*)emcalFolder->FindObject(filename) ; //set address of the SDigits and SDigitizer TBranch * sdigitsBranch = 0; TBranch * sdigitizerBranch = 0; TBranch * branch = 0 ; TObjArray * lob = (TObjArray*)treeS->GetListOfBranches() ; TIter next(lob) ; Bool_t emcalfound = kFALSE, sdigitizerfound = kFALSE ; while ( (branch = (TBranch*)next()) && (!emcalfound || !sdigitizerfound) ) { if ( strcmp(branch->GetName(), "EMCAL")==0) { emcalfound = kTRUE ; sdigitsBranch = branch ; } else if ( strcmp(branch->GetName(), "AliEMCALSDigitizer")==0) { sdigitizerfound = kTRUE ; sdigitizerBranch = branch ; } } if ( !emcalfound || !sdigitizerfound ) { if (fDebug) Warning("ReadTreeS", "-> Digits and/or Digitizer branch not found"); return ; } if (!folder || !(folder->FindObject(sdigitsBranch->GetTitle()) ) ) PostSDigits(sdigitsBranch->GetTitle(),filename) ; sdigitsBranch->SetAddress(SDigitsRef(sdigitsBranch->GetTitle(),filename)) ; sdigitsBranch->GetEntry(0) ; TString sdname(sdigitsBranch->GetTitle()) ; sdname+=":" ; sdname+=filename ; if(!SDigitizer(sdigitsBranch->GetTitle()) ) PostSDigitizer(sdigitsBranch->GetTitle(),filename) ; sdigitizerBranch->SetAddress(SDigitizerRef(sdname)) ; sdigitizerBranch->GetEntry(0) ; if(gAlice->TreeS()!=treeS) treeS->Delete(); } //____________________________________________________________________________ void AliEMCALGetter::ReadPrimaries() { // Reads specific branches of primaries TClonesArray * ar = 0 ; if(! (ar = Primaries()) ) { PostPrimaries() ; ar = Primaries() ; } ar->Delete() ; if (TreeK(fHeaderFile)) { // treeK found in header file if (fDebug) Info("ReadPrimaries", "-> TreeK found in %s", fHeaderFile.Data()); fNPrimaries = gAlice->GetNtrack() ; } else { // treeK not found in header file Error("ReadPrimaries", "-> TreeK not found "); return ; } Int_t index = 0 ; for (index = 0 ; index < fNPrimaries; index++) { new ((*ar)[index]) TParticle(*(Primary(index))); } } //____________________________________________________________________________ void AliEMCALGetter::Event(const Int_t event, const char* opt) { // Reads the content of all Tree's S, D and R if (event >= gAlice->TreeE()->GetEntries() ) { Error("Event", "-> %d not found in TreeE!", event); return ; } Bool_t any = kFALSE ; if (strstr(opt,"A") ) // do not check the title of the branches any = kTRUE; gAlice->GetEvent(event) ; if( strstr(opt,"R") ) ReadTreeR(event) ; if( strstr(opt,"D") ) ReadTreeD(event) ; if(strstr(opt,"S") ) ReadTreeS(event) ; if(strstr(opt,"H") ) ReadTreeH() ; if( strstr(opt,"Q") ) ReadTreeQA() ; if( strstr(opt,"P") ) ReadPrimaries() ; } //____________________________________________________________________________ TObject * AliEMCALGetter::ReturnO(TString what, TString name, TString file) const { // get the object named "what" from the folder // folders are named like //Folders if ( file.IsNull() ) file = fHeaderFile ; TFolder * folder = 0 ; TObject * emcalO = 0 ; if ( what.CompareTo("Primaries") == 0 ) { folder = dynamic_cast(fPrimariesFolder->FindObject("Primaries")) ; if (folder) emcalO = dynamic_cast(folder->FindObject("Primaries")) ; else return 0 ; } else if ( what.CompareTo("Hits") == 0 ) { folder = dynamic_cast(fHitsFolder->FindObject("EMCAL")) ; if (folder) emcalO = dynamic_cast(folder->FindObject("Hits")) ; } else if ( what.CompareTo("SDigits") == 0 ) { file.ReplaceAll("/","_") ; TString path = "EMCAL/" + file ; folder = dynamic_cast(fSDigitsFolder->FindObject(path.Data())) ; if (folder) { if (name.IsNull()) name = fSDigitsTitle ; emcalO = dynamic_cast(folder->FindObject(name)) ; } } else if ( what.CompareTo("Digits") == 0 ){ folder = dynamic_cast(fDigitsFolder->FindObject("EMCAL")) ; if (folder) { if (name.IsNull()) name = fDigitsTitle ; emcalO = dynamic_cast(folder->FindObject(name)) ; } } else if ( what.CompareTo("TowerRecPoints") == 0 ) { folder = dynamic_cast(fRecoFolder->FindObject("EMCAL/TowerRecPoints")) ; if (folder) { if (name.IsNull()) name = fRecPointsTitle ; emcalO = dynamic_cast(folder->FindObject(name)) ; } } else if ( what.CompareTo("PreShowerRecPoints") == 0 ) { folder = dynamic_cast(fRecoFolder->FindObject("EMCAL/PreShowerRecPoints")) ; if (folder) { if (name.IsNull()) name = fRecPointsTitle ; emcalO = dynamic_cast(folder->FindObject(name)) ; } } if (!emcalO) { if(fDebug) Warning("ReturnO", "-> Object %s not found in %s", what.Data(), folder->GetName()); return 0 ; } return emcalO ; } //____________________________________________________________________________ const TTask * AliEMCALGetter::ReturnT(TString what, TString name) const { // get the TTask named "what" from the folder // folders are named like //Folders/Tasks/what/EMCAL/name TString search(what) ; if ( what.CompareTo("Clusterizer") == 0 ) search = "Reconstructioner" ; else if ( what.CompareTo("TrackSegmentMaker") == 0 ) search = "Reconstructioner" ; else if ( what.CompareTo("PID") == 0 ) search = "Reconstructioner" ; else if ( what.CompareTo("QATasks") == 0 ) search = "QA" ; TTask * tasks = dynamic_cast(fTasksFolder->FindObject(search)) ; if (!tasks) { Error("AliReturnT", "-> Task %s not found!", what.Data()); return 0 ; } TTask * emcalT = dynamic_cast(tasks->GetListOfTasks()->FindObject("EMCAL")) ; if (!emcalT) { Error("ReturnT", "-> Task %s/EMCAL not found!", what.Data()); return 0 ; } TList * list = emcalT->GetListOfTasks() ; if (what.CompareTo("SDigitizer") == 0) { if ( name.IsNull() ) name = fSDigitsTitle ; } else if (what.CompareTo("Digitizer") == 0){ if ( name.IsNull() ) name = fDigitsTitle ; } else if (what.CompareTo("Clusterizer") == 0){ if ( name.IsNull() ) name = fRecPointsTitle ; name.Append(":clu") ; } TIter it(list) ; TTask * task = 0 ; while((task = static_cast(it.Next()) )){ TString taskname(task->GetName()) ; if(taskname.BeginsWith(name)){ return task ;} } if(fDebug) Warning("ReturnT", "-> Task %s/%s not found!", search.Data(), name.Data()); return 0 ; } //____________________________________________________________________________ void AliEMCALGetter::RemoveTask(TString opt, TString name) const { // remove a task from the folder // path is fTasksFolder/SDigitizer/EMCAL/name TTask * task = 0 ; TTask * emcal = 0 ; TList * lofTasks = 0 ; if (opt == "S") { // SDigitizer task = dynamic_cast(fTasksFolder->FindObject("SDigitizer")) ; if (!task) return ; } else if (opt == "D") { // Digitizer task = dynamic_cast(fTasksFolder->FindObject("Digitizer")) ; if (!task) return ; } else if (opt == "C") { // Clusterizer task = dynamic_cast(fTasksFolder->FindObject("Reconstructioner")) ; if (!task) return ; } else { Warning("RemoveTask", "-> Unknown option %s"); return ; } emcal = dynamic_cast(task->GetListOfTasks()->FindObject("EMCAL")) ; if (!emcal) return ; lofTasks = emcal->GetListOfTasks() ; if (!lofTasks) return ; TObject * obj = lofTasks->FindObject(name) ; if (obj) lofTasks->Remove(obj) ; } //____________________________________________________________________________ void AliEMCALGetter::RemoveObjects(TString opt, TString name) const { // remove SDigits from the folder // path is fSDigitsFolder/fHeaderFileName/name TFolder * emcal = 0 ; TFolder * emcalmain = 0 ; if (opt == "H") { // Hits emcal = dynamic_cast(fHitsFolder->FindObject("EMCAL")) ; if (!emcal) return ; name = "Hits" ; } else if ( opt == "S") { // SDigits emcalmain = dynamic_cast(fSDigitsFolder->FindObject("EMCAL")) ; if (!emcalmain) return ; emcal = dynamic_cast(emcalmain->FindObject(fHeaderFile)) ; if (!emcal) return ; } else if (opt == "D") { // Digits emcal = dynamic_cast(fDigitsFolder->FindObject("EMCAL")) ; if (!emcal) return ; } else if (opt == "RT") { // Tower RecPoints emcal = dynamic_cast(fRecoFolder->FindObject("EMCAL/TowerRecPoints")) ; if (!emcal) return ; } else if (opt == "RP") { // Preshower RecPoints emcal = dynamic_cast(fRecoFolder->FindObject("EMCAL/PreShowerRecPoints")) ; if (!emcal) return ; } else if (opt == "T") { // TrackSegments emcal = dynamic_cast(fRecoFolder->FindObject("EMCAL/TrackSegments")) ; if (!emcal) return ; } else if (opt == "P") { // RecParticles emcal = dynamic_cast(fRecoFolder->FindObject("EMCAL/RecParticles")) ; if (!emcal) return ; } else { Warning("RemoveObjects", "-> Unknown option %s", opt.Data()); return ; } TObjArray * ar = dynamic_cast(emcal->FindObject(name)) ; if (ar) { emcal->Remove(ar) ; ar->Delete() ; delete ar ; } if (opt == "S") emcalmain->Remove(emcal) ; } //____________________________________________________________________________ void AliEMCALGetter::RemoveSDigits() const { TFolder * emcal= dynamic_cast(fSDigitsFolder->FindObject("EMCAL")) ; if (!emcal) return ; emcal->SetOwner() ; emcal->Clear() ; } //____________________________________________________________________________ void AliEMCALGetter::CleanWhiteBoard(void){ TFolder * emcalmain = 0 ; TFolder * emcal ; TObjArray * ar ; TList * lofTasks = 0 ; TTask * task = 0 ; TTask * emcalt = 0 ; // Hits emcal = dynamic_cast(fHitsFolder->FindObject("EMCAL")) ; if (emcal){ TObjArray * ar = dynamic_cast(emcal->FindObject("Hits")) ; if (ar) { emcal->Remove(ar) ; ar->Delete() ; delete ar ; } } // SDigits emcalmain = dynamic_cast(fSDigitsFolder->FindObject("EMCAL")) ; if (emcalmain){ emcal = dynamic_cast(emcalmain->FindObject(fHeaderFile)) ; if (emcal) { ar = dynamic_cast(emcal->FindObject(fSDigitsTitle)) ; if (ar) { emcal->Remove(ar) ; ar->Delete() ; delete ar ; } } emcalmain->Remove(emcal) ; } // Digits emcal = dynamic_cast(fDigitsFolder->FindObject("EMCAL")) ; if (emcal){ ar = dynamic_cast(emcal->FindObject(fDigitsTitle)) ; if (ar) { emcal->Remove(ar) ; ar->Delete() ; delete ar ; } } // TowerRecPoints emcal = dynamic_cast(fRecoFolder->FindObject("EMCAL/TowerRecPoints")) ; if (emcal){ ar = dynamic_cast(emcal->FindObject(fRecPointsTitle)) ; if (ar) { emcal->Remove(ar) ; ar->Delete() ; delete ar ; } } // PreShowerRecPoints emcal = dynamic_cast(fRecoFolder->FindObject("EMCAL/PreShowerRecPoints")) ; if (emcal){ ar = dynamic_cast(emcal->FindObject(fRecPointsTitle)) ; if (ar) { emcal->Remove(ar) ; ar->Delete() ; delete ar ; } } // TrackSegments emcal = dynamic_cast(fRecoFolder->FindObject("EMCAL/TrackSegments")) ; if (emcal) { ar = dynamic_cast(emcal->FindObject(fTrackSegmentsTitle)) ; if (ar) { emcal->Remove(ar) ; ar->Delete() ; delete ar ; } } // RecParticles emcal = dynamic_cast(fRecoFolder->FindObject("EMCAL/RecParticles")) ; if (emcal){ ar = dynamic_cast(emcal->FindObject(fRecParticlesTitle)) ; if (ar) { emcal->Remove(ar) ; ar->Delete() ; delete ar ; } } //---- Now Tasks ----------- TObject * obj ; TString sdname(fSDigitsTitle); // Digitizer task = dynamic_cast(fTasksFolder->FindObject("Digitizer")) ; if (task){ emcalt = dynamic_cast(task->GetListOfTasks()->FindObject("EMCAL")) ; if (emcalt){ lofTasks = emcalt->GetListOfTasks() ; if (lofTasks){ obj = lofTasks->FindObject(sdname.Data()) ; if (obj) lofTasks->Remove(obj) ; } } } sdname.Append(":") ; // Clusterizer, TrackSegmentMaker, PID task = dynamic_cast(fTasksFolder->FindObject("Reconstructioner")) ; if (task){ emcalt = dynamic_cast(task->GetListOfTasks()->FindObject("EMCAL")) ; if (emcalt){ lofTasks = emcalt->GetListOfTasks() ; TIter next(lofTasks); while((obj=next())){ TString oname(obj->GetName()) ; if (oname.BeginsWith(sdname)){ lofTasks->Remove(obj) ; } } } } // SDigitizer sdname.Append(fHeaderFile) ; task = dynamic_cast(fTasksFolder->FindObject("SDigitizer")) ; if (task) { emcalt = dynamic_cast(task->GetListOfTasks()->FindObject("EMCAL")) ; if (emcalt){ lofTasks = emcalt->GetListOfTasks() ; if (lofTasks){ obj = lofTasks->FindObject(sdname.Data()) ; if (obj) lofTasks->Remove(obj) ; } } } } //____________________________________________________________________________ void AliEMCALGetter::SetTitle(const char * branchTitle ) { fBranchTitle = branchTitle ; fSDigitsTitle = branchTitle ; fDigitsTitle = branchTitle ; fRecPointsTitle = branchTitle ; fRecParticlesTitle = branchTitle ; fTrackSegmentsTitle = branchTitle ; if(fToSplit){ //First - extract full path if necessary TString sFileName(fHeaderFile) ; Ssiz_t islash = sFileName.Last('/') ; if(islash(fHitsFolder->FindObject("EMCAL")) ; if ( !emcalFolder ) emcalFolder = fHitsFolder->AddFolder("EMCAL", "Hits from EMCAL") ; emcalFolder = dynamic_cast(fSDigitsFolder->FindObject("EMCAL")) ; if ( !emcalFolder ) emcalFolder = fSDigitsFolder->AddFolder("EMCAL", "SDigits from EMCAL") ; //Make folder for SDigits TString subdir(fHeaderFile) ; subdir.ReplaceAll("/","_") ; emcalFolder->AddFolder(subdir, fSDigitsFileName.Data()); emcalFolder = dynamic_cast(fDigitsFolder->FindObject("EMCAL")) ; if ( !emcalFolder ) emcalFolder = fDigitsFolder->AddFolder("EMCAL", "Digits from EMCAL") ; emcalFolder = dynamic_cast(fRecoFolder->FindObject("EMCAL")) ; if ( !emcalFolder ) emcalFolder = fRecoFolder->AddFolder("EMCAL", "Reconstructed data from EMCAL") ; } //____________________________________________________________________________ void AliEMCALGetter::CloseSplitFiles(void){ TFile * file ; file = static_cast(gROOT->GetFile(fSDigitsFileName.Data() ) ) ; if(file) file->Close() ; file = static_cast(gROOT->GetFile(fDigitsFileName.Data() ) ) ; if(file) file->Close() ; file = static_cast(gROOT->GetFile(fRecPointsFileName.Data() ) ) ; if(file) file->Close() ; file = static_cast(gROOT->GetFile(fTrackSegmentsFileName.Data() ) ) ; if(file) file->Close() ; file = static_cast(gROOT->GetFile(fRecParticlesFileName.Data() ) ) ; if(file) file->Close() ; }