2 //Class (base) responsible for management of data:
3 // - opening and closing proper files
4 // - posting data to folders
5 // - writing data from folders to proper files
7 //Author: Alice Offline Group http://alisoft.cern.ch
8 //Responsible: Piotr.Skowronski@cern.ch
10 #include <AliLoader.h>
23 #include <AliRunLoader.h>
24 #include <AliRunDigitizer.h>
25 #include <AliDigitizer.h>
26 #include <AliDetector.h>
28 Int_t AliLoader::fgDebug = 0;
30 const TString AliLoader::fgkDefaultHitsContainerName("TreeH");
31 const TString AliLoader::fgkDefaultDigitsContainerName = "TreeD";
32 const TString AliLoader::fgkDefaultSDigitsContainerName = "TreeS";
33 const TString AliLoader::fgkDefaultRecPointsContainerName = "TreeR";
34 const TString AliLoader::fgkDefaultTracksContainerName = "TreeT";
35 const TString AliLoader::fgkDefaultRecParticlesContainerName = "TreeP";
36 const TString AliLoader::fgkLoaderBaseName("Loader");
39 //___________________________________________________________________
40 /////////////////////////////////////////////////////////////////////
44 // Base class for Loaders. //
45 // Loader provides base I/O fascilities for "standard" data. //
46 // Each detector has a laoder data member //
47 // loader is accessible via folder structure as well //
49 /////////////////////////////////////////////////////////////////////
51 /*****************************************************************************/
53 AliLoader::AliLoader():
58 fDetectorDataFolder(0x0),
66 /******************************************************************/
68 AliLoader::AliLoader(const Char_t* detname,const Char_t* eventfoldername):
69 fDataLoaders(new TObjArray(kNDataTypes)),
73 fDetectorDataFolder(0x0),
79 if (GetDebug()) Info("AliLoader(const Char_t* detname,const Char_t* eventfoldername)",
80 "detname = %s eventfoldername = %s",detname,eventfoldername);
82 //try to find folder eventfoldername in top alice folder
83 //safe because GetTopFolder will abort in case of failure
85 fDetectorName = detname;
86 fName = fDetectorName+"Loader";
89 TObject* fobj = GetTopFolder()->FindObject(eventfoldername);
90 fEventFolder = (fobj)?dynamic_cast<TFolder*>(fobj):0x0;//in case FindObject returns NULL dynamic cast cause seg. fault
91 SetEventFolder(fEventFolder);
94 /*****************************************************************************/
96 AliLoader::AliLoader(const Char_t * detname,TFolder* eventfolder):
97 fDataLoaders(new TObjArray(kNDataTypes)),
98 fDetectorName(detname),
101 fDetectorDataFolder(0x0),
107 fDetectorName = detname;
108 fName = fDetectorName+"Loader";
110 SetEventFolder(eventfolder);
111 //fileoption's don't need to initialized because default TString ctor does it correctly
113 /*****************************************************************************/
115 AliLoader::~AliLoader()
118 if (fDataLoaders) fDataLoaders->SetOwner();
121 /*****************************************************************************/
123 void AliLoader::InitDefaults()
127 dl = new AliDataLoader(fDetectorName + ".Hits.root",fgkDefaultHitsContainerName, "Hits" );
128 fDataLoaders->AddAt(dl,kHits);
131 // S U M M A B L E D I G I T S
132 dl = new AliDataLoader(fDetectorName + ".SDigits.root",fgkDefaultSDigitsContainerName, "Summable Digits");
133 AliTaskLoader* tl = new AliTaskLoader(fDetectorName + AliConfig::Instance()->GetSDigitizerTaskName(),
134 dl,AliRunLoader::GetRunSDigitizer(),kTRUE);
135 dl->SetBaseTaskLoader(tl);
136 fDataLoaders->AddAt(dl,kSDigits);
139 dl = new AliDataLoader(fDetectorName + ".Digits.root",fgkDefaultDigitsContainerName, "Digits");
140 tl = new AliTaskLoader(fDetectorName + AliConfig::Instance()->GetDigitizerTaskName(),
141 dl,AliRunLoader::GetRunDigitizer(),kTRUE);
142 dl->SetBaseTaskLoader(tl);
143 fDataLoaders->AddAt(dl,kDigits);
145 // R E C O N S T R U C T E D P O I N T S aka C L U S T E R S
146 dl = new AliDataLoader(fDetectorName + ".RecPoints.root",fgkDefaultRecPointsContainerName, "Reconstructed Points");
147 tl = new AliTaskLoader(fDetectorName + AliConfig::Instance()->GetReconstructionerTaskName(),
148 dl,AliRunLoader::GetRunReconstructioner(),kTRUE);
149 dl->SetBaseTaskLoader(tl);
150 fDataLoaders->AddAt(dl,kRecPoints);
153 dl = new AliDataLoader(fDetectorName + ".Tracks.root",fgkDefaultTracksContainerName, "Tracks");
154 tl = new AliTaskLoader(fDetectorName + AliConfig::Instance()->GetTrackerTaskName(),
155 dl,AliRunLoader::GetRunTracker(),kTRUE);
156 dl->SetBaseTaskLoader(tl);
157 fDataLoaders->AddAt(dl,kTracks);
159 // R E C O N S T R U C T E D P O I N T S aka C L U S T E R S
160 dl = new AliDataLoader(fDetectorName + ".RecParticles.root",fgkDefaultRecParticlesContainerName, "Reconstructed Particles");
161 tl = new AliTaskLoader(fDetectorName + AliConfig::Instance()->GetPIDTaskName(),
162 dl,AliRunLoader::GetRunPIDTask(),kTRUE);
163 dl->SetBaseTaskLoader(tl);
164 fDataLoaders->AddAt(dl,kRecParticles);
167 /*****************************************************************************/
169 AliDataLoader* AliLoader::GetDataLoader(const char* name)
171 //returns Data Loader with specified name
172 return dynamic_cast<AliDataLoader*>(fDataLoaders->FindObject(name));
174 /*****************************************************************************/
176 Int_t AliLoader::SetEvent()
178 //basically the same that GetEvent but do not post data to folders
179 TIter next(fDataLoaders);
181 while ((dl = (AliDataLoader*)next()))
187 /******************************************************************/
189 Int_t AliLoader::GetEvent()
191 //changes to proper root directory and tries to load data from files to folders
192 // event number is defined in RunLoader
195 // 0 - in case of no error
196 // 1 - event not found
200 TIter next(fDataLoaders);
202 while ((dl = (AliDataLoader*)next()))
204 retval = dl->GetEvent();
207 Error("GetEvent","Error occured while GetEvent for %s",dl->GetName());
215 /******************************************************************/
217 TFolder* AliLoader::GetTopFolder()
219 //returns TOP aliroot folder, just a simple interface to AliConfig (gives shorter notation)
220 return AliConfig::Instance()->GetTopFolder();
223 /******************************************************************/
225 TFolder* AliLoader::GetEventFolder()
227 //get EVENT folder (data that are changing from event to event, even in single run)
230 /******************************************************************/
231 TFolder* AliLoader::GetDataFolder()
233 //returns the folder speciofic to given detector e.g. /Folders/Event/Data/
236 fDataFolder = dynamic_cast<TFolder*>(GetEventFolder()->FindObject(AliConfig::fgkDataFolderName));
240 Fatal("GetDataFolder","Can not find AliRoot data folder. Aborting");
247 /*****************************************************************************/
249 TFolder* AliLoader::GetTasksFolder()
251 //Returns pointer to Folder with Alice Tasks
254 fTasksFolder = dynamic_cast<TFolder*>(GetTopFolder()->FindObject(AliConfig::fgkTasksFolderName));
258 Fatal("GetTasksFolder","Can not find tasks folder. Aborting");
265 /*****************************************************************************/
267 TFolder* AliLoader::GetModulesFolder()
271 fModuleFolder = dynamic_cast<TFolder*>(GetEventFolder()->FindObjectAny(AliConfig::GetModulesFolderName()));
275 Fatal("GetModulesFolder","Can not find modules folder. Aborting");
279 return fModuleFolder;
282 /*****************************************************************************/
284 TFolder* AliLoader::GetQAFolder()
286 //returns folder with Quality assurance
287 if (fQAFolder == 0x0)
289 TObject *obj = GetEventFolder()->FindObjectAny(AliConfig::Instance()->GetQAFolderName());
290 fQAFolder = (obj)?dynamic_cast<TFolder*>(obj):0x0;
292 if (fQAFolder == 0x0)
294 Fatal("GetQAFolder","Can not find Quality Assurance folder. Aborting");
301 /*****************************************************************************/
302 TTask* AliLoader::SDigitizer()
304 //returns SDigitizer task for this detector
305 return GetSDigitsDataLoader()->GetBaseTaskLoader()->Task();
308 /*****************************************************************************/
310 AliDigitizer* AliLoader::Digitizer()
312 //returns Digitizer task for this detector
313 return dynamic_cast<AliDigitizer*>(GetDigitsDataLoader()->GetBaseTaskLoader()->Task());
315 /*****************************************************************************/
317 TTask* AliLoader::Reconstructioner()
319 //returns Recontructioner (Cluster Finder, Cluster Maker,
320 //or whatever you want to call it) task for this detector
321 return GetRecPointsDataLoader()->GetBaseTaskLoader()->Task();
323 /*****************************************************************************/
325 TTask* AliLoader::Tracker()
328 return dynamic_cast<TTask*>(GetTracksDataLoader()->GetBaseTaskLoader()->Task());
331 /*****************************************************************************/
332 TTask* AliLoader::PIDTask()
335 return dynamic_cast<TTask*>(GetRecParticlesDataLoader()->GetBaseTaskLoader()->Task());
338 /*****************************************************************************/
340 TTask* AliLoader::QAtask(const char* name)
342 TTask* qat = AliRunLoader::GetRunQATask();
345 Error("QAtask","Can not get RunQATask. (Name:%s)",GetName());
349 TString dqatname(fDetectorName + AliConfig::Instance()->GetQATaskName());
350 TTask* dqat = dynamic_cast<TTask*>(qat->GetListOfTasks()->FindObject(dqatname));
354 Error("QAtask","Can not find QATask in RunQATask for %s",GetDetectorName().Data());
358 if (strlen(name) == 0) return dqat;
360 TList* list = dqat->GetListOfTasks();
364 while((task = static_cast<TTask *>(it.Next()) ))
366 TString taskname(task->GetName()) ;
367 if(taskname.BeginsWith(name))
370 Error("QAtask","Can not find sub-task with name starting with %s in task %s",name,dqat->GetName());
373 /*****************************************************************************/
375 TDirectory* AliLoader::ChangeDir(TFile* file, Int_t eventno)
377 //changes the root directory in "file" to "dirname" which corresponds to event 'eventno'
378 //in case of success returns the pointer to directory
383 ::Error("AliLoader::ChangeDir","File is null");
386 if (file->IsOpen() == kFALSE)
388 ::Error("AliLoader::ChangeDir","File is not opened");
392 TString dirname("Event");
394 if (AliLoader::fgDebug > 1)
395 ::Info("AliLoader::ChangeDir","Changing Dir to %s in file %s.",dirname.Data(),file->GetName());
399 TDirectory* dir = dynamic_cast<TDirectory*>(file->Get(dirname));
403 if (AliLoader::fgDebug > 1)
404 ::Info("AliLoader::ChangeDir","Can not find directory %s in file %s, creating...",
405 dirname.Data(),file->GetName());
407 if (file->IsWritable() == kFALSE)
409 ::Error("AliLoader::ChangeDir","Can not create directory. File %s in not writable.",
414 TDirectory* newdir = file->mkdir(dirname);
417 ::Error("AliLoader::ChangeDir","Failed to create new directory in file %s.",
421 result = file->cd(dirname);
422 if (result == kFALSE)
429 file->cd();//make a file active
430 file->cd(dirname);//cd event dir
435 /*****************************************************************************/
437 TString AliLoader::GetUnixDir()
439 //This Method will manage jumping through unix directories in case of
440 //run with more events per run than defined in gAlice
446 /*****************************************************************************/
447 /************************************************************/
449 void AliLoader::MakeTree(Option_t *option)
451 //Makes a tree depending on option
454 // S: - Summable Digits
455 // R: - Reconstructed Points (clusters)
456 // T: - Tracks (tracklets)
458 const char *oH = strstr(option,"H");
459 const char *oD = strstr(option,"D");
460 const char *oS = strstr(option,"S");
461 const char *oR = strstr(option,"R");
462 const char *oT = strstr(option,"T");
463 const char *oP = strstr(option,"P");
465 if (oH) MakeHitsContainer();
466 if (oD) MakeDigitsContainer();
467 if (oS) MakeSDigitsContainer();
468 if (oR) MakeRecPointsContainer();
469 if (oT) MakeTracksContainer();
470 if (oP) MakeRecParticlesContainer();
473 /*****************************************************************************/
474 Int_t AliLoader::WriteHits(Option_t* opt)
476 AliDataLoader* dl = GetHitsDataLoader();
477 Int_t ret = dl->WriteData(opt);
480 /*****************************************************************************/
482 Int_t AliLoader::WriteSDigits(Option_t* opt)
484 AliDataLoader* dl = GetSDigitsDataLoader();
485 Int_t ret = dl->WriteData(opt);
489 /*****************************************************************************/
491 Int_t AliLoader::PostSDigitizer(TTask* sdzer)
493 return GetSDigitsDataLoader()->GetBaseTaskLoader()->Post(sdzer);
495 /*****************************************************************************/
497 Int_t AliLoader::PostDigitizer(AliDigitizer* task)
499 return GetDigitsDataLoader()->GetBaseTaskLoader()->Post(task);
501 /*****************************************************************************/
503 Int_t AliLoader::PostReconstructioner(TTask* task)
505 return GetRecPointsDataLoader()->GetBaseTaskLoader()->Post(task);
507 /*****************************************************************************/
509 Int_t AliLoader::PostTracker(TTask* task)
511 return GetTracksDataLoader()->GetBaseTaskLoader()->Post(task);
513 /*****************************************************************************/
515 Int_t AliLoader::PostPIDTask(TTask* task)
517 return GetRecParticlesDataLoader()->GetBaseTaskLoader()->Post(task);
519 /*****************************************************************************/
521 TObject** AliLoader::GetDetectorDataRef(TObject *obj)
527 return GetDetectorDataFolder()->GetListOfFolders()->GetObjectRef(obj) ;
529 /*****************************************************************************/
531 TObject** AliLoader::SDigitizerRef()
533 TTask* rsd = AliRunLoader::GetRunSDigitizer();
538 return rsd->GetListOfTasks()->GetObjectRef(SDigitizer());
540 /*****************************************************************************/
542 TObject** AliLoader::DigitizerRef()
544 TTask* rd = AliRunLoader::GetRunDigitizer();
549 return rd->GetListOfTasks()->GetObjectRef(Digitizer()) ;
551 /*****************************************************************************/
553 TObject** AliLoader::ReconstructionerRef()
555 TTask* rrec = AliRunLoader::GetRunReconstructioner();
560 return rrec->GetListOfTasks()->GetObjectRef(Reconstructioner());
562 /*****************************************************************************/
564 TObject** AliLoader::TrackerRef()
566 TTask* rrec = AliRunLoader::GetRunTracker();
571 return rrec->GetListOfTasks()->GetObjectRef(Tracker());
573 /*****************************************************************************/
575 TObject** AliLoader::PIDTaskRef()
577 TTask* rrec = AliRunLoader::GetRunPIDTask();
582 return rrec->GetListOfTasks()->GetObjectRef(PIDTask());
585 /*****************************************************************************/
586 void AliLoader::CleanFolders()
588 //Cleans all posted objects == removes from folders and deletes them
589 TIter next(fDataLoaders);
591 while ((dl = (AliDataLoader*)next()))
593 if (GetDebug()) Info("CleanFolders","name = %s cleaning",dl->GetName());
597 /*****************************************************************************/
599 /*****************************************************************************/
601 void AliLoader::CleanSDigitizer()
603 //removes and deletes detector task from Run Task
604 TTask* task = AliRunLoader::GetRunSDigitizer();
607 Error("CleanSDigitizer","Can not get RunSDigitizer from folder. Can not clean");
611 if (GetDebug()) Info("CleanSDigitizer","Attempting to delete S Digitizer");
612 delete task->GetListOfTasks()->Remove(SDigitizer()); //TTList::Remove does not delete object
614 /*****************************************************************************/
616 void AliLoader::CleanDigitizer()
618 //removes and deletes detector task from Run Task
619 TTask* task = AliRunLoader::GetRunDigitizer();
622 Error("CleanDigitizer","Can not get RunDigitizer from folder. Can not clean");
627 Info("CleanDigitizer","Attempting to delete Digitizer %X",
628 task->GetListOfTasks()->Remove(Digitizer()));
629 delete task->GetListOfTasks()->Remove(Digitizer()); //TTList::Remove does not delete object
631 /*****************************************************************************/
633 void AliLoader::CleanReconstructioner()
635 //removes and deletes detector Reconstructioner from Run Reconstructioner
636 TTask* task = AliRunLoader::GetRunReconstructioner();
639 Error("CleanReconstructioner","Can not get RunReconstructioner from folder. Can not clean");
644 Info("CleanReconstructioner","Attempting to delete Reconstructioner");
645 delete task->GetListOfTasks()->Remove(Reconstructioner()); //TTList::Remove does not delete object
647 /*****************************************************************************/
649 void AliLoader::CleanTracker()
651 //removes and deletes detector tracker from Run Tracker
652 TTask* task = AliRunLoader::GetRunTracker();
655 Error("CleanTracker","Can not get RunTracker from folder. Can not clean");
660 Info("CleanTracker","Attempting to delete Tracker %X",
661 task->GetListOfTasks()->Remove(Tracker()));
662 delete task->GetListOfTasks()->Remove(Tracker()); //TTList::Remove does not delete object
664 /*****************************************************************************/
666 void AliLoader::CleanPIDTask()
668 //removes and deletes detector Reconstructioner from Run Reconstructioner
669 TTask* task = AliRunLoader::GetRunPIDTask();
672 Error("CleanPIDTask","Can not get Run PID Task from folder. Can not clean");
677 Info("CleanPIDTask","Attempting to delete PID Task");
678 delete task->GetListOfTasks()->Remove(PIDTask()); //TTList::Remove does not delete object
680 /*****************************************************************************/
682 Int_t AliLoader::ReloadAll()
684 TIter next(fDataLoaders);
687 while((dl = (AliDataLoader*)next()))
689 Int_t err = dl->Reload();
692 Error("ReloadAll","Reload returned error for %s",dl->GetName());
698 /*****************************************************************************/
700 void AliLoader::CloseFiles()
702 //close files for data loaders
703 TIter next(fDataLoaders);
705 while((dl = (AliDataLoader*)next()))
710 /*****************************************************************************/
712 Int_t AliLoader::SetEventFolder(TFolder* eventfolder)
714 if (eventfolder == 0x0)
716 Error("SetEventFolder","Stupid joke. Argument is NULL");
720 fEventFolder = eventfolder;
721 TIter next(fDataLoaders);
724 while((dl = (AliDataLoader*)next()))
726 dl->SetEventFolder(fEventFolder);
727 dl->SetFolder(GetDetectorDataFolder()); //Must exists - ensure register is called before
731 }//sets the event folder
732 /*****************************************************************************/
734 Int_t AliLoader::Register(TFolder* eventFolder)
736 //triggers creation of subfolders for a given detector
737 //this method is called when session is read from disk
739 //warning: AliDetector in constructor (not default) calls
740 //creation of folder structure as well (some detectors needs folders
741 //alrady in constructors)
743 if (GetDebug()) Info("Register","Name is %s.",GetName());
744 if (eventFolder == 0x0)
746 Error("Register","Event folder is not set.");
749 Int_t retval = AliConfig::Instance()->AddDetector(eventFolder,fDetectorName,fDetectorName);
752 Error("SetEventFolder","Can not create tasks and/or folders for %s. Event folder name is %s",
753 fDetectorName.Data(),eventFolder->GetName());
756 SetEventFolder(eventFolder);
759 /*****************************************************************************/
760 AliRunLoader* AliLoader::GetRunLoader()
762 //gets the run-loader from event folder
763 AliRunLoader* rg = 0x0;
764 TObject * obj = GetEventFolder()->FindObject(AliRunLoader::fgkRunLoaderName);
765 if (obj) rg = dynamic_cast<AliRunLoader*>(obj);
768 /*****************************************************************************/
769 Bool_t AliLoader::TestFileOption(Option_t* opt)
771 //tests the TFile::Option
772 //if file is truncated at opening moment ("recreate", "new" or "create") returns kFALSE;
773 //else kTRUE (means opened with "read" or "update" mode)
775 if (option.CompareTo("recreate",TString::kIgnoreCase) == 0) return kFALSE;
776 if (option.CompareTo("new",TString::kIgnoreCase) == 0) return kFALSE;
777 if (option.CompareTo("create",TString::kIgnoreCase) == 0) return kFALSE;
780 /*****************************************************************************/
781 void AliLoader::SetDirName(TString& dirname)
783 //adds "dirname/" to each file
784 TIter next(fDataLoaders);
786 while((dl = (AliDataLoader*)next()))
788 dl->SetDirName(dirname);
792 /*****************************************************************************/
794 void AliLoader::SetDigitsFileNameSuffix(const TString& suffix)
796 //adds the suffix before ".root",
797 //e.g. TPC.Digits.root -> TPC.DigitsMerged.root
798 //made on Jiri Chudoba demand
799 GetDigitsDataLoader()->SetFileNameSuffix(suffix);
801 /*****************************************************************************/
803 void AliLoader::SetCompressionLevel(Int_t cl)
805 //sets comression level for data defined by di
806 TIter next(fDataLoaders);
808 while((dl = (AliDataLoader*)next()))
810 dl->SetCompressionLevel(cl);
813 /*****************************************************************************/
815 void AliLoader::Clean()
817 //Cleans all data loaders
818 TIter next(fDataLoaders);
820 while((dl = (AliDataLoader*)next()))
825 /*****************************************************************************/
827 void AliLoader::Clean(const TString& name)
829 TObject* obj = GetDetectorDataFolder()->FindObject(name);
833 Info("Clean(const TString&)","name=%s, cleaning %s.",GetName(),name.Data());
834 GetDetectorDataFolder()->Remove(obj);
839 /*****************************************************************************/
841 Bool_t AliLoader::IsOptionWritable(const TString& opt)
843 if (opt.CompareTo("recreate",TString::kIgnoreCase)) return kTRUE;
844 if (opt.CompareTo("new",TString::kIgnoreCase)) return kTRUE;
845 if (opt.CompareTo("create",TString::kIgnoreCase)) return kTRUE;
846 if (opt.CompareTo("update",TString::kIgnoreCase)) return kTRUE;
849 /*****************************************************************************/
851 void AliLoader::SetTAddrInDet()
853 //calls SetTreeAddress for corresponding detector
854 AliRunLoader* rl = GetRunLoader();
855 if (rl == 0x0) return;
856 AliRun* ar = rl->GetAliRun();
857 if (ar == 0x0) return;
858 AliDetector* det = ar->GetDetector(fDetectorName);
859 if (det == 0x0) return;
860 det->SetTreeAddress();
862 /*****************************************************************************/
864 void AliLoader::Synchronize()
866 //synchrinizes all writtable files
867 TIter next(fDataLoaders);
869 while ((dl = (AliDataLoader*)next()))
875 /*****************************************************************************/
876 /*****************************************************************************/
877 /*****************************************************************************/