#include "TFile.h"
#include "TTree.h"
#include "TROOT.h"
#include "AliObjectLoader.h"
#include "AliLog.h"
#include "AliPHOSLoader.h"
#include "AliPHOS.h"
#include "AliPHOSHit.h"
using std::cout;
using std::endl;
ClassImp(AliPHOSLoader)
const TString AliPHOSLoader::fgkHitsName("HITS");
const TString AliPHOSLoader::fgkSDigitsName("SDIGITS");
const TString AliPHOSLoader::fgkDigitsName("DIGITS");
const TString AliPHOSLoader::fgkEmcRecPointsName("EMCRECPOINTS");
const TString AliPHOSLoader::fgkCpvRecPointsName("CPVRECPOINTS");
const TString AliPHOSLoader::fgkTracksName("TRACKS");
const TString AliPHOSLoader::fgkRecParticlesName("RECPARTICLES");
const TString AliPHOSLoader::fgkEmcRecPointsBranchName("PHOSEmcRP");
const TString AliPHOSLoader::fgkCpvRecPointsBranchName("PHOSCpvRP");
const TString AliPHOSLoader::fgkTrackSegmentsBranchName("PHOSTS");
const TString AliPHOSLoader::fgkRecParticlesBranchName("PHOSRP");
AliPHOSLoader::AliPHOSLoader() : fBranchTitle(), fDebug(0), fTmpHits(0x0)
{
fTmpHits = new TClonesArray("AliPHOSHit",1000);
}
AliPHOSLoader::AliPHOSLoader(const Char_t *detname,const Char_t *eventfoldername) :
AliLoader(detname, eventfoldername),
fBranchTitle(), fDebug(0), fTmpHits(0x0)
{
}
AliPHOSLoader::AliPHOSLoader(const Char_t *detname,TFolder *topfolder):
AliLoader(detname,topfolder),
fBranchTitle(), fDebug(0), fTmpHits(0x0)
{
fTmpHits = new TClonesArray("AliPHOSHit",1000);
}
AliPHOSLoader::~AliPHOSLoader()
{
Clean(fgkHitsName);
Clean(fgkSDigitsName);
Clean(fgkDigitsName);
Clean(fgkEmcRecPointsName);
Clean(fgkCpvRecPointsName);
Clean(fgkTracksName);
Clean(fgkRecParticlesName);
CleanFolders() ;
}
void AliPHOSLoader::CleanFolders()
{
CleanRecParticles();
AliLoader::CleanFolders();
}
Int_t AliPHOSLoader::SetEvent()
{
Int_t retval = AliLoader::SetEvent();
if (retval)
{
AliError("returned error");
return retval;
}
if (Hits()) Hits()->Clear();
if (SDigits()) SDigits()->Clear();
if (Digits()) Digits()->Clear();
if (EmcRecPoints()) EmcRecPoints()->Clear();
if (CpvRecPoints()) CpvRecPoints()->Clear();
if (TrackSegments()) TrackSegments()->Clear();
if (RecParticles()) RecParticles()->Clear();
return 0;
}
Int_t AliPHOSLoader::GetEvent()
{
Int_t retval;
retval = AliLoader::GetEvent();
if (retval)
{
AliError("returned error");
return retval;
}
if (GetHitsDataLoader()->GetBaseDataLoader()->IsLoaded()) ReadHits();
if (GetSDigitsDataLoader()->GetBaseDataLoader()->IsLoaded()) ReadSDigits();
if (GetDigitsDataLoader()->GetBaseDataLoader()->IsLoaded()) ReadDigits();
if (GetRecPointsDataLoader()->GetBaseDataLoader()->IsLoaded()) ReadRecPoints();
if (GetTracksDataLoader()->GetBaseDataLoader()->IsLoaded()) ReadTracks();
if (GetRecParticlesDataLoader()->GetBaseDataLoader()->IsLoaded()) ReadRecParticles();
return 0;
}
Int_t AliPHOSLoader::LoadHits(Option_t* opt)
{
Int_t res;
res = AliLoader::LoadHits(opt);
if (res)
{
AliError("returned error");
return res;
}
res = ReadHits();
return 0;
}
Int_t AliPHOSLoader::LoadSDigits(Option_t* opt)
{
Int_t res;
res = AliLoader::LoadSDigits(opt);
if (res)
{
AliError("returned error");
return res;
}
return ReadSDigits();
}
Int_t AliPHOSLoader::LoadDigits(Option_t* opt)
{
Int_t res;
res = AliLoader::LoadDigits(opt);
if (res)
{
AliError("returned error");
return res;
}
return ReadDigits();
}
Int_t AliPHOSLoader::LoadRecPoints(Option_t* opt)
{
Int_t res;
res = AliLoader::LoadRecPoints(opt);
if (res)
{
AliError("returned error");
return res;
}
TFolder * phosFolder = GetDetectorDataFolder();
if ( phosFolder == 0x0 )
{
AliError("Can not get detector data folder");
return 1;
}
return ReadRecPoints();
}
Int_t AliPHOSLoader::LoadTracks(Option_t* opt)
{
AliDebug(1, Form("opt = %s",opt));
Int_t res;
res = AliLoader::LoadTracks(opt);
if (res)
{
AliError("returned error");
return res;
}
return ReadTracks();
}
Int_t AliPHOSLoader::LoadRecParticles(Option_t* opt)
{
Int_t res;
res = AliLoader::LoadRecParticles(opt);
if (res)
{
AliError("returned error");
return res;
}
return ReadRecParticles();
}
Int_t AliPHOSLoader::PostHits()const
{
Int_t reval = AliLoader::PostHits();
if (reval)
{
AliError("returned error");
return reval;
}
return const_cast<AliPHOSLoader *>(this)->ReadHits();
}
Int_t AliPHOSLoader::PostSDigits()const
{
Int_t reval = AliLoader::PostSDigits();
if (reval)
{
AliError("returned error");
return reval;
}
return const_cast<AliPHOSLoader *>(this)->ReadSDigits();
}
Int_t AliPHOSLoader::PostDigits()const
{
Int_t reval = AliLoader::PostDigits();
if (reval)
{
AliError("returned error");
return reval;
}
return const_cast<AliPHOSLoader *>(this)->ReadDigits();
}
Int_t AliPHOSLoader::PostRecPoints()const
{
Int_t reval = AliLoader::PostRecPoints();
if (reval)
{
AliError("returned error");
return reval;
}
return const_cast<AliPHOSLoader *>(this)->ReadRecPoints();
}
Int_t AliPHOSLoader::PostRecParticles()const
{
Int_t reval = AliLoader::PostRecParticles();
if (reval)
{
AliError("returned error");
return reval;
}
return const_cast<AliPHOSLoader *>(this)->ReadRecParticles();
}
Int_t AliPHOSLoader::PostTracks()const
{
Int_t reval = AliLoader::PostTracks();
if (reval)
{
AliError("returned error");
return reval;
}
return const_cast<AliPHOSLoader *>(this)->ReadTracks();
}
Int_t AliPHOSLoader::ReadHits()
{
TObject** hitref = HitsRef();
if(hitref == 0x0) {
MakeHitsArray();
hitref = HitsRef();
}
TClonesArray* hits = static_cast<TClonesArray*>(*hitref);
TTree* treeh = TreeH();
if(treeh == 0) {
AliError("Cannot read TreeH from folder");
return 1;
}
TBranch * hitsbranch = treeh->GetBranch(fDetectorName);
if (hitsbranch == 0) {
AliError("Cannot find branch PHOS");
return 1;
}
AliDebug(1, "Reading Hits");
if (hitsbranch->GetEntries() > 1) {
hitsbranch->SetAddress(&fTmpHits);
Int_t index = 0 ;
for (Int_t i = 0 ; i < hitsbranch->GetEntries(); i++) {
hitsbranch->GetEntry(i) ;
for (Int_t j = 0 ; j < fTmpHits->GetEntriesFast() ; j++) {
AliPHOSHit* hit = (AliPHOSHit*)fTmpHits->At(j);
new((*hits)[index]) AliPHOSHit( *hit ) ;
index++ ;
}
}
fTmpHits->Clear();
}
else {
hitsbranch->SetAddress(hitref);
hitsbranch->GetEntry(0) ;
}
return 0;
}
Int_t AliPHOSLoader::ReadSDigits()
{
TObject** sdref = SDigitsRef();
if(sdref == 0x0)
{
MakeSDigitsArray();
sdref = SDigitsRef();
}
TTree * treeS = TreeS();
if(treeS==0)
{
return 0;
}
TBranch * branch = treeS->GetBranch(fDetectorName);
if (branch == 0)
{
return 0;
}
branch->SetAddress(sdref);
branch->GetEntry(0);
return 0;
}
Int_t AliPHOSLoader::ReadDigits()
{
TObject** dref = DigitsRef();
if(dref == 0x0)
{
MakeDigitsArray();
dref = DigitsRef();
}
TTree * treeD = TreeD();
if(treeD==0)
{
return 0;
}
TBranch * branch = treeD->GetBranch(fDetectorName);
if (branch == 0)
{
return 0;
}
branch->SetAddress(dref);
branch->GetEntry(0);
return 0;
}
void AliPHOSLoader::Track(Int_t itrack)
{
if(TreeH()== 0)
{
if (LoadHits())
{
AliError("Can not load hits.");
return;
}
}
TBranch * hitsbranch = dynamic_cast<TBranch*>(TreeH()->GetListOfBranches()->FindObject("PHOS")) ;
if ( !hitsbranch ) {
if (fDebug)
cout << "WARNING: AliPHOSLoader::ReadTreeH -> Cannot find branch PHOS" << endl ;
return ;
}
if(!Hits()) PostHits();
hitsbranch->SetAddress(HitsRef());
hitsbranch->GetEntry(itrack);
}
Int_t AliPHOSLoader::ReadRecPoints()
{
MakeRecPointsArray();
TObjArray * cpva = 0x0 ;
TObjArray * emca = 0x0 ;
TTree * treeR = TreeR();
if(treeR==0)
{
return 0;
}
Int_t retval = 0;
TBranch * emcbranch = treeR->GetBranch(fgkEmcRecPointsBranchName);
if (emcbranch == 0x0)
{
AliError(Form("Can not get branch with EMC Rec. Points named %s",
fgkEmcRecPointsBranchName.Data()));
retval = 1;
}
else
{
emcbranch->SetAddress(&emca) ;
emcbranch->GetEntry(0) ;
}
TBranch * cpvbranch = treeR->GetBranch(fgkCpvRecPointsBranchName);
if (cpvbranch == 0x0)
{
AliError(Form("Can not get branch with CPV Rec. Points named %s",
fgkCpvRecPointsBranchName.Data()));
retval = 2;
}
else
{
cpvbranch->SetAddress(&cpva);
cpvbranch->GetEntry(0) ;
}
Int_t ii ;
if (emca != 0) {
Int_t maxemc = emca->GetEntries() ;
for ( ii= 0 ; ii < maxemc ; ii++ )
EmcRecPoints()->Add(emca->At(ii)) ;
}
if (cpva != 0) {
Int_t maxcpv = cpva->GetEntries() ;
for ( ii= 0 ; ii < maxcpv ; ii++ )
CpvRecPoints()->Add(cpva->At(ii)) ;
}
return retval;
}
Int_t AliPHOSLoader::ReadTracks()
{
TObject** trkref = TracksRef();
if ( trkref == 0x0 )
{
MakeTrackSegmentsArray();
trkref = TracksRef();
}
TTree * treeT = TreeT();
if(treeT==0)
{
return 0;
}
TBranch * branch = treeT->GetBranch(fgkTrackSegmentsBranchName);
if (branch == 0)
{
AliError(Form("Cannot find branch named %s",
fgkTrackSegmentsBranchName.Data()));
return 0;
}
branch->SetAddress(trkref);
branch->GetEntry(0);
return 0;
}
Int_t AliPHOSLoader::ReadRecParticles()
{
TObject** recpartref = RecParticlesRef();
if ( recpartref == 0x0 )
{
MakeRecParticlesArray();
recpartref = RecParticlesRef();
}
TTree * treeP = TreeP();
if(treeP==0)
{
return 0;
}
TBranch * branch = treeP->GetBranch(fgkRecParticlesBranchName);
if (branch == 0)
{
AliError(Form("Cannot find branch %s",
fgkRecParticlesBranchName.Data()));
return 0;
}
branch->SetAddress(recpartref);
branch->GetEntry(0);
return 0;
}
AliPHOSLoader* AliPHOSLoader::GetPHOSLoader(const char* eventfoldername)
{
AliRunLoader* rn = AliRunLoader::GetRunLoader(eventfoldername);
if (rn == 0x0) {
printf("Can not find Run Loader in folder %s", eventfoldername);
return 0x0 ;
}
return dynamic_cast<AliPHOSLoader*>(rn->GetLoader("PHOSLoader"));
}
Bool_t AliPHOSLoader::BranchExists(const TString& recName)
{
if (fBranchTitle.IsNull()) return kFALSE;
TString dataname, zername ;
TTree* tree;
if(recName == "SDigits")
{
tree = TreeS();
dataname = GetDetectorName();
zername = "AliPHOSSDigitizer" ;
}
else
if(recName == "Digits"){
tree = TreeD();
dataname = GetDetectorName();
zername = "AliPHOSDigitizer" ;
}
else
if(recName == "RecPoints"){
tree = TreeR();
dataname = fgkEmcRecPointsBranchName;
zername = "AliPHOSClusterizer" ;
}
else
if(recName == "TrackSegments"){
tree = TreeT();
dataname = fgkTrackSegmentsBranchName;
zername = "AliPHOSTrackSegmentMaker";
}
else
if(recName == "RecParticles"){
tree = TreeP();
dataname = fgkRecParticlesBranchName;
zername = "AliPHOSPID";
}
else
return kFALSE ;
if(!tree )
return kFALSE ;
TObjArray * lob = static_cast<TObjArray*>(tree->GetListOfBranches()) ;
TIter next(lob) ;
TBranch * branch = 0 ;
TString titleName(fBranchTitle);
titleName+=":";
while ((branch = (static_cast<TBranch*>(next())))) {
TString branchName(branch->GetName() ) ;
TString branchTitle(branch->GetTitle() ) ;
if ( branchName.BeginsWith(dataname) && branchTitle.BeginsWith(fBranchTitle) ){
AliWarning(Form("branch %s with title %s ",
dataname.Data(),fBranchTitle.Data()));
return kTRUE ;
}
if ( branchName.BeginsWith(zername) && branchTitle.BeginsWith(titleName) ){
AliWarning(Form("branch AliPHOS... with title %s ",
branch->GetTitle()));
return kTRUE ;
}
}
return kFALSE ;
}
void AliPHOSLoader::SetBranchTitle(const TString& btitle)
{
if (btitle.CompareTo(fBranchTitle) == 0) return;
fBranchTitle = btitle;
ReloadAll();
}
void AliPHOSLoader::CleanHits()const
{
AliLoader::CleanHits();
TClonesArray* hits = const_cast<AliPHOSLoader *>(this)->Hits();
if (hits) hits->Clear();
}
void AliPHOSLoader::CleanSDigits()const
{
AliLoader::CleanSDigits();
TClonesArray* sdigits = const_cast<AliPHOSLoader *>(this)->SDigits();
if (sdigits) sdigits->Clear();
}
void AliPHOSLoader::CleanDigits()const
{
AliLoader::CleanDigits();
TClonesArray* digits = const_cast<AliPHOSLoader *>(this)->Digits();
if (digits) digits->Clear();
}
void AliPHOSLoader::CleanRecPoints()const
{
AliLoader::CleanRecPoints();
TObjArray* recpoints = const_cast<AliPHOSLoader *>(this)->EmcRecPoints();
if (recpoints) recpoints->Clear();
recpoints = const_cast<AliPHOSLoader *>(this)->CpvRecPoints();
if (recpoints) recpoints->Clear();
}
void AliPHOSLoader::CleanTracks()const
{
AliLoader::CleanTracks();
TClonesArray* tracks = const_cast<AliPHOSLoader *>(this)->TrackSegments();
if (tracks) tracks->Clear();
}
void AliPHOSLoader::CleanRecParticles()
{
TClonesArray *recpar = RecParticles();
if (recpar) recpar->Clear();
}
void AliPHOSLoader::MakeHitsArray()
{
if (Hits()) return;
TClonesArray* hits = new TClonesArray("AliPHOSHit",1000);
hits->SetName(fgkHitsName);
GetDetectorDataFolder()->Add(hits);
}
void AliPHOSLoader::MakeSDigitsArray()
{
if ( SDigits()) return;
TClonesArray* sdigits = new TClonesArray("AliPHOSDigit",1);
sdigits->SetName(fgkSDigitsName);
GetDetectorDataFolder()->Add(sdigits);
}
void AliPHOSLoader::MakeDigitsArray()
{
if ( Digits()) return;
TClonesArray* digits = new TClonesArray("AliPHOSDigit",1);
digits->SetName(fgkDigitsName);
GetDetectorDataFolder()->Add(digits);
}
void AliPHOSLoader::MakeRecPointsArray()
{
if ( EmcRecPoints() == 0x0)
{
AliDebug(9, "Making array for EMC");
TObjArray* emc = new TObjArray(100) ;
emc->SetName(fgkEmcRecPointsName) ;
GetDetectorDataFolder()->Add(emc);
}
if ( CpvRecPoints() == 0x0)
{
AliDebug(9, "Making array for CPV");
TObjArray* cpv = new TObjArray(100) ;
cpv->SetName(fgkCpvRecPointsName);
GetDetectorDataFolder()->Add(cpv);
}
}
void AliPHOSLoader::MakeTrackSegmentsArray()
{
if ( TrackSegments()) return;
TClonesArray * ts = new TClonesArray("AliPHOSTrackSegment",100) ;
ts->SetName(fgkTracksName);
GetDetectorDataFolder()->Add(ts);
}
void AliPHOSLoader::MakeRecParticlesArray()
{
if ( RecParticles()) return;
TClonesArray * rp = new TClonesArray("AliPHOSRecParticle",100) ;
rp->SetName(fgkRecParticlesName);
GetDetectorDataFolder()->Add(rp);
}