1 /**************************************************************************
2 * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
4 * Author: The ALICE Off-line Project. *
5 * Contributors are mentioned in the code where appropriate. *
7 * Permission to use, copy, modify and distribute this software and its *
8 * documentation strictly for non-commercial purposes is hereby granted *
9 * without fee, provided that the above copyright notice appears in all *
10 * copies and that both the copyright notice and this permission notice *
11 * appear in the supporting documentation. The authors make no claims *
12 * about the suitability of this software for any purpose. It is *
13 * provided "as is" without express or implied warranty. *
14 **************************************************************************/
18 //_________________________________________________________________________
19 // A singleton. This class should be used in the analysis stage to get
20 // reconstructed objects: Digits, RecPoints, TrackSegments and RecParticles,
21 // instead of directly reading them from galice.root file. This container
22 // ensures, that one reads Digits, made of these particular digits, RecPoints,
23 // made of these particular RecPoints, TrackSegments and RecParticles.
24 // This becomes non trivial if there are several identical branches, produced with
25 // different set of parameters.
27 // An example of how to use (see also class AliPHOSAnalyser):
28 // for(Int_t irecp = 0; irecp < gime->NRecParticles() ; irecp++)
29 // AliPHOSRecParticle * part = gime->RecParticle(1) ;
31 // please->GetEvent(event) ; // reads new event from galice.root
33 //*-- Author: Yves Schutz (SUBATECH) & Dmitri Peressounko (RRC KI & SUBATECH)
34 //*-- Completely redesigned by Dmitri Peressounko March 2001
36 //*-- YS June 2001 : renamed the original AliPHOSIndexToObject and make
37 //*-- systematic usage of TFolders without changing the interface
38 //////////////////////////////////////////////////////////////////////////////
41 // --- ROOT system ---
47 // --- Standard library ---
49 // --- AliRoot header files ---
51 #include "AliPHOSLoader.h"
53 #include "AliPHOSHit.h"
54 #include "AliPHOSCalibrationDB.h"
55 #include "AliPHOSGetter.h"
57 ClassImp(AliPHOSLoader)
60 const TString AliPHOSLoader::fgkHitsName("HITS");//Name for TClonesArray with hits from one event
61 const TString AliPHOSLoader::fgkSDigitsName("SDIGITS");//Name for TClonesArray
62 const TString AliPHOSLoader::fgkDigitsName("DIGITS");//Name for TClonesArray
63 const TString AliPHOSLoader::fgkEmcRecPointsName("EMCRECPOINTS");//Name for TClonesArray
64 const TString AliPHOSLoader::fgkCpvRecPointsName("CPVRECPOINTS");//Name for TClonesArray
65 const TString AliPHOSLoader::fgkTracksName("TRACKS");//Name for TClonesArray
66 const TString AliPHOSLoader::fgkRecParticlesName("RECPARTICLES");//Name for TClonesArray
68 const TString AliPHOSLoader::fgkEmcRecPointsBranchName("PHOSEmcRP");//Name for branch with EMC Reconstructed Points
69 const TString AliPHOSLoader::fgkCpvRecPointsBranchName("PHOSCpvRP");//Name for branch with CPV Reconstructed Points
70 const TString AliPHOSLoader::fgkTrackSegmentsBranchName("PHOSTS");//Name for branch with TrackSegments
71 const TString AliPHOSLoader::fgkRecParticlesBranchName("PHOSRP");//Name for branch with Reconstructed Particles
72 //____________________________________________________________________________
73 AliPHOSLoader::AliPHOSLoader()
77 //____________________________________________________________________________
78 AliPHOSLoader::AliPHOSLoader(const Char_t *detname,const Char_t *eventfoldername):
79 AliLoader(detname,eventfoldername)
83 //____________________________________________________________________________
85 AliPHOSLoader::~AliPHOSLoader()
87 //remove and delete arrays
89 Clean(fgkSDigitsName);
91 Clean(fgkEmcRecPointsName);
92 Clean(fgkCpvRecPointsName);
94 Clean(fgkRecParticlesName);
96 // set to 0x0 the objgetter in AliGetter ... weird isn it !
97 AliPHOSGetter * gime = AliPHOSGetter::Instance() ; // (AliLoader::GetRunLoader()->GetFileName()).Data()) ;
102 //____________________________________________________________________________
103 void AliPHOSLoader::CleanFolders()
106 AliLoader::CleanFolders();
109 //____________________________________________________________________________
110 Int_t AliPHOSLoader::SetEvent()
112 //Cleans loaded stuff and and sets Files and Directories
113 // do not post any data to folder/tasks
116 Int_t retval = AliLoader::SetEvent();
119 AliError("returned error");
124 if (Hits()) Hits()->Clear();
125 if (SDigits()) SDigits()->Clear();
126 if (Digits()) Digits()->Clear();
127 if (EmcRecPoints()) EmcRecPoints()->Clear();
128 if (CpvRecPoints()) CpvRecPoints()->Clear();
129 if (TrackSegments()) TrackSegments()->Clear();
130 if (RecParticles()) RecParticles()->Clear();
135 //____________________________________________________________________________
136 Int_t AliPHOSLoader::GetEvent()
138 //Overloads GetEvent method called by AliRunLoader::GetEvent(Int_t) method
139 //to add Rec Particles specific for PHOS
141 //First call the original method to get whatever from std. setup is needed
144 retval = AliLoader::GetEvent();
147 AliError("returned error");
151 if (GetHitsDataLoader()->GetBaseDataLoader()->IsLoaded()) ReadHits();
152 if (GetSDigitsDataLoader()->GetBaseDataLoader()->IsLoaded()) ReadSDigits();
153 if (GetDigitsDataLoader()->GetBaseDataLoader()->IsLoaded()) ReadDigits();
154 if (GetRecPointsDataLoader()->GetBaseDataLoader()->IsLoaded()) ReadRecPoints();
155 if (GetTracksDataLoader()->GetBaseDataLoader()->IsLoaded()) ReadTracks();
156 if (GetRecParticlesDataLoader()->GetBaseDataLoader()->IsLoaded()) ReadRecParticles();
159 //Now, check if RecPart were loaded
163 //____________________________________________________________________________
164 const AliPHOS * AliPHOSLoader::PHOS()
166 // returns the PHOS object
167 AliPHOS * phos = dynamic_cast<AliPHOS*>(GetModulesFolder()->FindObject(fDetectorName));
170 cout << "WARNING: AliPHOSLoader::PHOS -> PHOS module not found in Folders" << endl ;
174 //____________________________________________________________________________
175 const AliPHOSGeometry * AliPHOSLoader::PHOSGeometry()
177 // Return PHOS geometry
178 AliPHOSGeometry * rv = 0 ;
180 rv = PHOS()->GetGeometry();
185 //____________________________________________________________________________
186 Int_t AliPHOSLoader::LoadHits(Option_t* opt)
188 //------- Hits ----------------------
189 //Overload (extends) LoadHits implemented in AliLoader
193 //First call the AliLoader's method to send the TreeH to folder
194 res = AliLoader::LoadHits(opt);
198 AliError("returned error");
202 //read the data from tree in folder and send it to folder
208 //____________________________________________________________________________
209 Int_t AliPHOSLoader::LoadSDigits(Option_t* opt)
211 //---------- SDigits -------------------------
213 //First call the AliLoader's method to send the TreeS to folder
214 res = AliLoader::LoadSDigits(opt);
217 AliError("returned error");
220 return ReadSDigits();
223 //____________________________________________________________________________
224 Int_t AliPHOSLoader::LoadDigits(Option_t* opt)
226 //---------- Digits -------------------------
228 //First call the AliLoader's method to send the TreeS to folder
229 res = AliLoader::LoadDigits(opt);
232 AliError("returned error");
237 //____________________________________________________________________________
238 Int_t AliPHOSLoader::LoadRecPoints(Option_t* opt)
240 // -------------- RecPoints -------------------------------------------
242 //First call the AliLoader's method to send the TreeR to folder
243 res = AliLoader::LoadRecPoints(opt);
246 AliError("returned error");
250 TFolder * phosFolder = GetDetectorDataFolder();
251 if ( phosFolder == 0x0 )
253 AliError("Can not get detector data folder");
256 return ReadRecPoints();
258 //____________________________________________________________________________
260 Int_t AliPHOSLoader::LoadTracks(Option_t* opt)
262 //Loads Tracks: Open File, Reads Tree and posts, Read Data and Posts
263 AliDebug(1, Form("opt = %s",opt));
265 res = AliLoader::LoadTracks(opt);
268 AliError("returned error");
274 //____________________________________________________________________________
275 Int_t AliPHOSLoader::LoadRecParticles(Option_t* opt)
277 // -------------- RecPoints -------------------------------------------
279 //First call the AliLoader's method to send the TreeT to folder
280 res = AliLoader::LoadRecParticles(opt);
283 AliError("returned error");
286 return ReadRecParticles();
289 //____________________________________________________________________________
291 Int_t AliPHOSLoader::PostHits()
293 // -------------- Hits -------------------------------------------
294 Int_t reval = AliLoader::PostHits();
297 AliError("returned error");
302 //____________________________________________________________________________
304 Int_t AliPHOSLoader::PostSDigits()
306 // -------------- SDigits -------------------------------------------
307 Int_t reval = AliLoader::PostSDigits();
310 AliError("returned error");
313 return ReadSDigits();
315 //____________________________________________________________________________
317 Int_t AliPHOSLoader::PostDigits()
319 // -------------- Digits -------------------------------------------
320 Int_t reval = AliLoader::PostDigits();
323 AliError("returned error");
328 //____________________________________________________________________________
330 Int_t AliPHOSLoader::PostRecPoints()
332 // -------------- RecPoints -------------------------------------------
333 Int_t reval = AliLoader::PostRecPoints();
336 AliError("returned error");
339 return ReadRecPoints();
342 //____________________________________________________________________________
344 Int_t AliPHOSLoader::PostRecParticles()
346 // -------------- RecParticles -------------------------------------------
347 Int_t reval = AliLoader::PostRecParticles();
350 AliError("returned error");
353 return ReadRecParticles();
355 //____________________________________________________________________________
357 Int_t AliPHOSLoader::PostTracks()
359 // -------------- Tracks -------------------------------------------
360 Int_t reval = AliLoader::PostTracks();
363 AliError("returned error");
368 //____________________________________________________________________________
372 //____________________________________________________________________________
373 Int_t AliPHOSLoader::ReadHits()
375 // If there is no Clones Array in folder creates it and sends to folder
376 // then tries to read
377 // Reads the first entry of PHOS branch in hit tree TreeH()
378 // Reads data from TreeH and stores it in TClonesArray that sits in DetectorDataFolder
380 TObject** hitref = HitsRef();
387 TClonesArray* hits = dynamic_cast<TClonesArray*>(*hitref);
389 TTree* treeh = TreeH();
393 AliError("Cannot read TreeH from folder");
397 TBranch * hitsbranch = treeh->GetBranch(fDetectorName);
400 AliError("Cannot find branch PHOS");
404 AliDebug(1, "Reading Hits");
406 if (hitsbranch->GetEntries() > 1)
408 TClonesArray * tempo = new TClonesArray("AliPHOSHit",1000);
410 hitsbranch->SetAddress(&tempo);
413 for (i = 0 ; i < hitsbranch->GetEntries(); i++)
415 hitsbranch->GetEntry(i) ;
417 for ( j = 0 ; j < tempo->GetEntries() ; j++)
419 AliPHOSHit* hit = (AliPHOSHit*)tempo->At(j);
420 new((*hits)[index]) AliPHOSHit( *hit ) ;
428 hitsbranch->SetAddress(hitref);
429 hitsbranch->GetEntry(0) ;
434 //____________________________________________________________________________
435 Int_t AliPHOSLoader::ReadSDigits()
437 // Read the summable digits tree TreeS():
438 // Check if TClones is in folder
439 // if not create and add to folder
440 // connect to tree if available
443 TObject** sdref = SDigitsRef();
447 sdref = SDigitsRef();
450 TTree * treeS = TreeS();
453 //May happen if file is truncated or new in LoadSDigits
454 //AliError("There is no SDigit Tree");
458 TBranch * branch = treeS->GetBranch(fDetectorName);
460 {//easy, maybe just a new tree
461 //AliError("Cannot find branch PHOS");
465 branch->SetAddress(SDigitsRef());
470 //____________________________________________________________________________
471 Int_t AliPHOSLoader::ReadDigits()
473 // Read the summable digits tree TreeS():
474 // Check if TClones is in folder
475 // if not create and add to folder
476 // connect to tree if available
479 TObject** dref = DigitsRef();
481 {//if there is not array in folder, create it and put it there
486 TTree * treeD = TreeD();
489 //May happen if file is truncated or new in LoadSDigits
490 //AliError("There is no Digit Tree");
494 TBranch * branch = treeD->GetBranch(fDetectorName);
496 {//easy, maybe just a new tree
497 //AliError("Cannot find branch ",fDetectorName.Data());
501 branch->SetAddress(dref);//connect branch to buffer sitting in folder
502 branch->GetEntry(0);//get first event
507 //____________________________________________________________________________
509 void AliPHOSLoader::Track(Int_t itrack)
511 // Read the first entry of PHOS branch in hit tree gAlice->TreeH()
516 AliError("Can not load hits.");
521 TBranch * hitsbranch = dynamic_cast<TBranch*>(TreeH()->GetListOfBranches()->FindObject("PHOS")) ;
524 cout << "WARNING: AliPHOSLoader::ReadTreeH -> Cannot find branch PHOS" << endl ;
527 if(!Hits()) PostHits();
529 hitsbranch->SetAddress(HitsRef());
530 hitsbranch->GetEntry(itrack);
534 //____________________________________________________________________________
535 Int_t AliPHOSLoader::ReadRecPoints()
537 //Creates and posts to folder an array container,
538 //connects branch in tree (if exists), and reads data to array
540 MakeRecPointsArray();
542 TObjArray * cpva = 0x0 ;
543 TObjArray * emca = 0x0 ;
545 TTree * treeR = TreeR();
549 //May happen if file is truncated or new in LoadSDigits
554 TBranch * emcbranch = treeR->GetBranch(fgkEmcRecPointsBranchName);
556 if (emcbranch == 0x0)
558 AliError(Form("Can not get branch with EMC Rec. Points named %s",
559 fgkEmcRecPointsBranchName.Data()));
564 emcbranch->SetAddress(&emca) ;
565 emcbranch->GetEntry(0) ;
567 TBranch * cpvbranch = treeR->GetBranch(fgkCpvRecPointsBranchName);
568 if (cpvbranch == 0x0)
570 AliError(Form("Can not get branch with CPV Rec. Points named %s",
571 fgkCpvRecPointsBranchName.Data()));
576 cpvbranch->SetAddress(&cpva);
577 cpvbranch->GetEntry(0) ;
581 Int_t maxemc = emca->GetEntries() ;
582 for ( ii= 0 ; ii < maxemc ; ii++ )
583 EmcRecPoints()->Add(emca->At(ii)) ;
585 Int_t maxcpv = cpva->GetEntries() ;
586 for ( ii= 0 ; ii < maxcpv ; ii++ )
587 CpvRecPoints()->Add(cpva->At(ii)) ;
592 //____________________________________________________________________________
593 Int_t AliPHOSLoader::ReadTracks()
595 //Creates and posts to folder an array container,
596 //connects branch in tree (if exists), and reads data to arry
598 TObject** trkref = TracksRef();
600 {//Create and post array
601 MakeTrackSegmentsArray();
602 trkref = TracksRef();
605 TTree * treeT = TreeT();
608 //May happen if file is truncated or new in LoadSDigits, or the file is in update mode,
609 //but tracking was not performed yet for a current event
610 //AliError("There is no Tree with Tracks");
614 TBranch * branch = treeT->GetBranch(fgkTrackSegmentsBranchName);
616 {//easy, maybe just a new tree
617 AliError(Form("Cannot find branch named %s",
618 fgkTrackSegmentsBranchName.Data()));
622 branch->SetAddress(trkref);//connect branch to buffer sitting in folder
623 branch->GetEntry(0);//get first event
627 //____________________________________________________________________________
629 Int_t AliPHOSLoader::ReadRecParticles()
631 //Reads Reconstructed Particles from file
632 //Creates and posts to folder an array container,
633 //connects branch in tree (if exists), and reads data to arry
635 TObject** recpartref = RecParticlesRef();
637 if ( recpartref == 0x0 )
638 {//Create and post array
639 MakeRecParticlesArray();
640 recpartref = RecParticlesRef();
643 TTree * treeP = TreeP();
646 //May happen if file is truncated or new in LoadSDigits,
647 //or the file is in update mode,
648 //but tracking was not performed yet for a current event
649 // AliError("There is no Tree with Tracks and Reconstructed Particles");
653 TBranch * branch = treeP->GetBranch(fgkRecParticlesBranchName);
655 {//easy, maybe just a new tree
656 AliError(Form("Cannot find branch %s",
657 fgkRecParticlesBranchName.Data()));
661 branch->SetAddress(recpartref);//connect branch to buffer sitting in folder
662 branch->GetEntry(0);//get first event
668 AliPHOSGeometry* AliPHOSLoader::GetPHOSGeometry()
670 //returns PHOS geometry from gAlice
671 //static Method used by some classes where it is not convienient to pass eventfoldername
674 AliPHOS* phos=dynamic_cast<AliPHOS*>(gAlice->GetDetector("PHOS"));
677 return phos->GetGeometry();
679 /***************************************************************************************/
681 AliPHOSLoader* AliPHOSLoader::GetPHOSLoader(const char* eventfoldername)
683 // Return PHOS loader
684 AliRunLoader* rn = AliRunLoader::GetRunLoader(eventfoldername);
686 printf("Can not find Run Loader in folder %s", eventfoldername);
689 return dynamic_cast<AliPHOSLoader*>(rn->GetLoader("PHOSLoader"));
691 /***************************************************************************************/
693 Bool_t AliPHOSLoader::BranchExists(const TString& recName)
695 // Check if a branch named redName exists
696 if (fBranchTitle.IsNull()) return kFALSE;
697 TString dataname, zername ;
699 if(recName == "SDigits")
702 dataname = GetDetectorName();
703 zername = "AliPHOSSDigitizer" ;
706 if(recName == "Digits"){
708 dataname = GetDetectorName();
709 zername = "AliPHOSDigitizer" ;
712 if(recName == "RecPoints"){
714 dataname = fgkEmcRecPointsBranchName;
715 zername = "AliPHOSClusterizer" ;
718 if(recName == "TrackSegments"){
720 dataname = fgkTrackSegmentsBranchName;
721 zername = "AliPHOSTrackSegmentMaker";
724 if(recName == "RecParticles"){
726 dataname = fgkRecParticlesBranchName;
727 zername = "AliPHOSPID";
736 TObjArray * lob = static_cast<TObjArray*>(tree->GetListOfBranches()) ;
738 TBranch * branch = 0 ;
739 TString titleName(fBranchTitle);
742 while ((branch = (static_cast<TBranch*>(next())))) {
743 TString branchName(branch->GetName() ) ;
744 TString branchTitle(branch->GetTitle() ) ;
745 if ( branchName.BeginsWith(dataname) && branchTitle.BeginsWith(fBranchTitle) ){
746 AliWarning(Form("branch %s with title %s ",
747 dataname.Data(),fBranchTitle.Data()));
750 if ( branchName.BeginsWith(zername) && branchTitle.BeginsWith(titleName) ){
751 AliWarning(Form("branch AliPHOS... with title %s ",
752 branch->GetTitle()));
760 void AliPHOSLoader::SetBranchTitle(const TString& btitle)
763 if (btitle.CompareTo(fBranchTitle) == 0) return;
764 fBranchTitle = btitle;
767 //____________________________________________________________________________
769 void AliPHOSLoader::CleanHits()
772 AliLoader::CleanHits();
774 TClonesArray* hits = Hits();
775 if (hits) hits->Clear();
777 //____________________________________________________________________________
779 void AliPHOSLoader::CleanSDigits()
781 // Clean SDigits array
782 AliLoader::CleanSDigits();
783 TClonesArray* sdigits = SDigits();
784 if (sdigits) sdigits->Clear();
787 //____________________________________________________________________________
789 void AliPHOSLoader::CleanDigits()
791 // Clean Digits array
792 AliLoader::CleanDigits();
793 TClonesArray* digits = Digits();
794 if (digits) digits->Clear();
796 //____________________________________________________________________________
798 void AliPHOSLoader::CleanRecPoints()
800 // Clean RecPoints array
801 AliLoader::CleanRecPoints();
802 TObjArray* recpoints = EmcRecPoints();
803 if (recpoints) recpoints->Clear();
804 recpoints = CpvRecPoints();
805 if (recpoints) recpoints->Clear();
807 //____________________________________________________________________________
809 void AliPHOSLoader::CleanTracks()
811 //Cleans Tracks stuff
812 AliLoader::CleanTracks();//tree
814 //and clear the array
815 TClonesArray* tracks = TrackSegments();
816 if (tracks) tracks->Clear();
819 //____________________________________________________________________________
821 void AliPHOSLoader::CleanRecParticles()
823 // Clean RecParticles array
824 TClonesArray *recpar = RecParticles();
825 if (recpar) recpar->Clear();
829 //____________________________________________________________________________
831 void AliPHOSLoader::ReadCalibrationDB(const char * database,const char * filename)
833 // Read calibration data base from file
834 if(fcdb && (strcmp(database,fcdb->GetTitle())==0))
837 TFile * file = gROOT->GetFile(filename) ;
839 file = TFile::Open(filename);
841 AliError(Form("Cannot open file %s", filename)) ;
846 fcdb = dynamic_cast<AliPHOSCalibrationDB *>(file->Get("AliPHOSCalibrationDB")) ;
848 AliError(Form("No database %s in file %s", database, filename)) ;
850 //____________________________________________________________________________
852 // AliPHOSSDigitizer* AliPHOSLoader::PHOSSDigitizer()
854 // //return PHOS SDigitizer
855 // return dynamic_cast<AliPHOSSDigitizer*>(SDigitizer()) ;
858 //____________________________________________________________________________
859 void AliPHOSLoader::MakeHitsArray()
861 // Add Hits array to the data folder
863 TClonesArray* hits = new TClonesArray("AliPHOSHit",1000);
864 hits->SetName(fgkHitsName);
865 GetDetectorDataFolder()->Add(hits);
868 //____________________________________________________________________________
869 void AliPHOSLoader::MakeSDigitsArray()
871 // Add SDigits array to the data folder
872 if ( SDigits()) return;
873 TClonesArray* sdigits = new TClonesArray("AliPHOSDigit",1);
874 sdigits->SetName(fgkSDigitsName);
875 GetDetectorDataFolder()->Add(sdigits);
878 //____________________________________________________________________________
879 void AliPHOSLoader::MakeDigitsArray()
881 // Add Digits array to the data folder
882 if ( Digits()) return;
883 TClonesArray* digits = new TClonesArray("AliPHOSDigit",1);
884 digits->SetName(fgkDigitsName);
885 GetDetectorDataFolder()->Add(digits);
889 //____________________________________________________________________________
890 void AliPHOSLoader::MakeRecPointsArray()
892 // Add RecPoints array to the data folder
893 if ( EmcRecPoints() == 0x0)
895 AliDebug(9, "Making array for EMC");
896 TObjArray* emc = new TObjArray(100) ;
897 emc->SetName(fgkEmcRecPointsName) ;
898 GetDetectorDataFolder()->Add(emc);
901 if ( CpvRecPoints() == 0x0)
903 AliDebug(9, "Making array for CPV");
904 TObjArray* cpv = new TObjArray(100) ;
905 cpv->SetName(fgkCpvRecPointsName);
906 GetDetectorDataFolder()->Add(cpv);
910 //____________________________________________________________________________
911 void AliPHOSLoader::MakeTrackSegmentsArray()
913 // Add TrackSegments array to the data folder
914 if ( TrackSegments()) return;
915 TClonesArray * ts = new TClonesArray("AliPHOSTrackSegment",100) ;
916 ts->SetName(fgkTracksName);
917 GetDetectorDataFolder()->Add(ts);
921 //____________________________________________________________________________
922 void AliPHOSLoader::MakeRecParticlesArray()
924 // Add RecParticles array to the data folder
925 if ( RecParticles()) return;
926 TClonesArray * rp = new TClonesArray("AliPHOSRecParticle",100) ;
927 rp->SetName(fgkRecParticlesName);
928 GetDetectorDataFolder()->Add(rp);