X-Git-Url: http://git.uio.no/git/?a=blobdiff_plain;f=PHOS%2FAliPHOSSDigitizer.cxx;h=65a73193f25143a1065f057b5046d4ab8cd1f7c9;hb=b165f59d6b34804592b3ab987e72cd859b9abdd6;hp=227f40f7d3d3fd5d0d0aa5a1575802c11c2a90cf;hpb=6f51f90b52ea5e7fc0f261686f3aac8f779d8d01;p=u%2Fmrichter%2FAliRoot.git diff --git a/PHOS/AliPHOSSDigitizer.cxx b/PHOS/AliPHOSSDigitizer.cxx index 227f40f7d3d..65a73193f25 100644 --- a/PHOS/AliPHOSSDigitizer.cxx +++ b/PHOS/AliPHOSSDigitizer.cxx @@ -16,6 +16,11 @@ /* $Id$ */ +/* History of cvs commits: + * + * $Log$ + */ + //_________________________________________________________________________ // This is a TTask that makes SDigits out of Hits // The name of the TTask is also the title of the branch that will contain @@ -51,190 +56,180 @@ // --- ROOT system --- -#include "TFile.h" -#include "TTask.h" -#include "TTree.h" -#include "TSystem.h" -#include "TROOT.h" -#include "TFolder.h" #include "TBenchmark.h" -#include "TGeometry.h" + //#include "TObjectTable.h" // --- Standard library --- -#include // --- AliRoot header files --- -#include "AliRun.h" -#include "AliHeader.h" +#include "AliLog.h" +#include "AliPHOSGeometry.h" #include "AliPHOSDigit.h" -#include "AliPHOSGeometry.h" #include "AliPHOSGetter.h" #include "AliPHOSHit.h" #include "AliPHOSSDigitizer.h" - + //#include "AliMemoryWatcher.h" ClassImp(AliPHOSSDigitizer) //____________________________________________________________________________ - AliPHOSSDigitizer::AliPHOSSDigitizer():TTask("","") + AliPHOSSDigitizer::AliPHOSSDigitizer():TTask("","") { // ctor - fA = 0; - fB = 10000000.; - fPrimThreshold = 0.01 ; - fSDigitsInRun = 0 ; - fSplitFile = 0 ; + fFirstEvent = fLastEvent = 0 ; + fDefaultInit = kTRUE ; } //____________________________________________________________________________ -AliPHOSSDigitizer::AliPHOSSDigitizer(const char * headerFile, const char * sDigitsTitle):TTask(sDigitsTitle, headerFile) +AliPHOSSDigitizer::AliPHOSSDigitizer(const char * alirunFileName, + const char * eventFolderName): + TTask("PHOS"+AliConfig::Instance()->GetSDigitizerTaskName(), alirunFileName), + fEventFolderName(eventFolderName) { + // ctor - fA = 0; - fB = 10000000.; - fPrimThreshold = 0.01 ; - fSDigitsInRun = 0 ; - fSplitFile = 0 ; + fFirstEvent = fLastEvent = 0 ; // runs one event by defaut + InitParameters() ; Init(); + fDefaultInit = kFALSE ; } //____________________________________________________________________________ -AliPHOSSDigitizer::~AliPHOSSDigitizer() +AliPHOSSDigitizer::AliPHOSSDigitizer(const AliPHOSSDigitizer & sd) + : TTask(sd) { - if (fSplitFile) - if ( fSplitFile->IsOpen() ) - fSplitFile->Close() ; - AliPHOSGetter * gime = AliPHOSGetter::GetInstance() ; - // Close the root file - gime->CloseFile() ; - - // remove the task from the folder list - gime->RemoveTask("S",GetName()) ; - - TString name(GetName()) ; - name.Remove(name.Index(":")) ; + //cpy ctor + + fFirstEvent = sd.fFirstEvent ; + fLastEvent = sd.fLastEvent ; + fA = sd.fA ; + fB = sd.fB ; + fPrimThreshold = sd.fPrimThreshold ; + fSDigitsInRun = sd.fSDigitsInRun ; + SetName(sd.GetName()) ; + SetTitle(sd.GetTitle()) ; + fEventFolderName = sd.fEventFolderName; +} - // remove the Hits from the folder list - gime->RemoveObjects("H",name) ; - // remove the SDigits from the folder list - gime->RemoveObjects("S", name) ; +//____________________________________________________________________________ +AliPHOSSDigitizer::~AliPHOSSDigitizer() { + //dtor + // AliPHOSGetter * gime = + // AliPHOSGetter::Instance(GetTitle(),fEventFolderName.Data()); + AliPHOSGetter * gime = + AliPHOSGetter::Instance(); + gime->PhosLoader()->CleanSDigitizer(); } - //____________________________________________________________________________ void AliPHOSSDigitizer::Init() { - // Initialization: open root-file, allocate arrays for hits and sdigits, - // attach task SDigitizer to the list of PHOS tasks - // - // Initialization can not be done in the default constructor - //============================================================= YS - // The initialisation is now done by AliPHOSGetter + // Uses the getter to access the required files - if( strcmp(GetTitle(), "") == 0 ) - SetTitle("galice.root") ; - - AliPHOSGetter * gime = AliPHOSGetter::GetInstance(GetTitle(), GetName()) ; + fInit = kTRUE ; + + AliPHOSGetter * gime = AliPHOSGetter::Instance(GetTitle(), fEventFolderName.Data()); if ( gime == 0 ) { - cerr << "ERROR: AliPHOSSDigitizer::Init -> Could not obtain the Getter object !" << endl ; + Fatal("Init" ,"Could not obtain the Getter object for file %s and event %s !", GetTitle(), fEventFolderName.Data()) ; return ; } - gime->PostSDigits( GetName(), GetTitle() ) ; + TString opt("SDigits") ; + if(gime->VersionExists(opt) ) { + Error( "Init", "Give a version name different from %s", fEventFolderName.Data() ) ; + fInit = kFALSE ; + } + + gime->PostSDigitizer(this); + gime->PhosLoader()->GetSDigitsDataLoader()->GetBaseTaskLoader()->SetDoNotReload(kTRUE); + +} - TString sdname(GetName() ); - sdname.Append(":") ; - sdname.Append(GetTitle() ) ; - SetName(sdname) ; - gime->PostSDigitizer(this) ; +//____________________________________________________________________________ +void AliPHOSSDigitizer::InitParameters() +{ + // initializes the parameters for digitization + fA = 0; + fB = 10000000.; + fPrimThreshold = 0.01 ; + fSDigitsInRun = 0 ; } //____________________________________________________________________________ void AliPHOSSDigitizer::Exec(Option_t *option) { - // Collects all hits in the same active volume into digit - if( strcmp(GetName(), "") == 0 ) - Init() ; + // Steering method to produce summable digits for events + // in the range from fFirstEvent to fLastEvent. + // This range is optionally set by SetEventRange(). + // if fLastEvent=-1 (by default), then process events until the end. + // + // Summable digit is a sum of all hits in the same active + // volume into digit if (strstr(option, "print") ) { - Print("") ; + Print() ; return ; } if(strstr(option,"tim")) gBenchmark->Start("PHOSSDigitizer"); - - //Check, if this branch already exits - gAlice->GetEvent(0) ; - - TString sdname(GetName()) ; - sdname.Remove(sdname.Index(GetTitle())-1) ; - if(gAlice->TreeS() ) { - TObjArray * lob = static_cast(gAlice->TreeS()->GetListOfBranches()) ; - TIter next(lob) ; - TBranch * branch = 0 ; - Bool_t phosfound = kFALSE, sdigitizerfound = kFALSE ; - - while ( (branch = (static_cast(next()))) && (!phosfound || !sdigitizerfound) ) { - TString thisName( GetName() ) ; - TString branchName( branch->GetTitle() ) ; - branchName.Append(":") ; - if ( (strcmp(branch->GetName(), "PHOS")==0) && thisName.BeginsWith(branchName) ) - phosfound = kTRUE ; - else if ( (strcmp(branch->GetName(), "AliPHOSSDigitizer")==0) && thisName.BeginsWith(branchName) ) - sdigitizerfound = kTRUE ; - } - - if ( phosfound || sdigitizerfound ) { - cerr << "WARNING: AliPHOSSDigitizer::Exec -> SDigits and/or SDigitizer branch with name " << GetName() - << " already exits" << endl ; - return ; - } - } + AliPHOSGetter * gime = AliPHOSGetter::Instance() ; - AliPHOSGetter * gime = AliPHOSGetter::GetInstance() ; + //switch off reloading of this task while getting event + if (!fInit) { // to prevent overwrite existing file + AliError( Form("Give a version name different from %s", fEventFolderName.Data()) ) ; + return ; + } - Int_t nevents = (Int_t) gAlice->TreeE()->GetEntries() ; + if (fLastEvent == -1) + fLastEvent = gime->MaxEvent() - 1 ; + else + fLastEvent = TMath::Min(fFirstEvent, gime->MaxEvent()); // only ine event at the time + Int_t nEvents = fLastEvent - fFirstEvent + 1; Int_t ievent ; - for(ievent = 0; ievent < nevents; ievent++){ - gime->Event(ievent,"H") ; - const TClonesArray * hits = gime->Hits() ; - TClonesArray * sdigits = gime->SDigits(sdname.Data()) ; + + //AliMemoryWatcher memwatcher; + + for (ievent = fFirstEvent; ievent <= fLastEvent; ievent++) { + gime->Event(ievent,"H") ; + TTree * treeS = gime->TreeS(); + TClonesArray * hits = gime->Hits() ; + TClonesArray * sdigits = gime->SDigits() ; sdigits->Clear(); Int_t nSdigits = 0 ; - - //Now make SDigits from hits, for PHOS it is the same, so just copy + Int_t nPrim = static_cast((gime->TreeH())->GetEntries()) ; + // Attention nPrim is the number of primaries tracked by Geant + // and this number could be different to the number of Primaries in TreeK; + Int_t iprim ; - Int_t Nprim = (Int_t) (gAlice->TreeH())->GetEntries(); - // Attention Nprim is the number of primaries tracked by Geant and this number could be different to the number of Primaries in TreeK; - Int_t iprim; - for (iprim=0; iprimTrack(iprim) ; - Int_t i; - for ( i = 0 ; i < hits->GetEntries() ; i++ ) { - AliPHOSHit * hit = dynamic_cast(hits->At(i)) ; - // Assign primary number only if contribution is significant + Int_t i; + for ( i = 0 ; i < hits->GetEntries() ; i++ ) { + AliPHOSHit * hit = dynamic_cast(hits->At(i)) ; + // Assign primary number only if contribution is significant if( hit->GetEnergy() > fPrimThreshold) new((*sdigits)[nSdigits]) AliPHOSDigit(hit->GetPrimary(),hit->GetId(), - Digitize(hit->GetEnergy()), hit->GetTime()) ; + Digitize(hit->GetEnergy()), hit->GetTime()) ; else new((*sdigits)[nSdigits]) AliPHOSDigit( -1 , hit->GetId(), - Digitize(hit->GetEnergy()), hit->GetTime()) ; + Digitize(hit->GetEnergy()), hit->GetTime()) ; nSdigits++ ; - } - + } + } // loop over iprim - + sdigits->Sort() ; - + nSdigits = sdigits->GetEntriesFast() ; + fSDigitsInRun += nSdigits ; sdigits->Expand(nSdigits) ; @@ -244,133 +239,57 @@ void AliPHOSSDigitizer::Exec(Option_t *option) digit->SetIndexInList(i) ; } - if(gAlice->TreeS() == 0) - gAlice->MakeTree("S", fSplitFile); + //Now write SDigits + //First list of sdigits - Int_t bufferSize = 32000 ; - TBranch * sdigitsBranch = gAlice->TreeS()->Branch("PHOS",&sdigits,bufferSize); - sdigitsBranch->SetTitle(sdname); - + + Int_t bufferSize = 32000 ; + TBranch * sdigitsBranch = treeS->Branch("PHOS",&sdigits,bufferSize); + + sdigitsBranch->Fill() ; + + gime->WriteSDigits("OVERWRITE"); + //Next - SDigitizer - Int_t splitlevel = 0 ; - AliPHOSSDigitizer * sd = this ; - TBranch * sdigitizerBranch = gAlice->TreeS()->Branch("AliPHOSSDigitizer","AliPHOSSDigitizer", - &sd,bufferSize,splitlevel); - sdigitizerBranch->SetTitle(sdname); + + gime->WriteSDigitizer("OVERWRITE"); + //gObjectTable->Print() ; - sdigitsBranch->Fill() ; - sdigitizerBranch->Fill() ; - gAlice->TreeS()->AutoSave() ; - if(strstr(option,"deb")) PrintSDigits(option) ; + + //memwatcher.Watch(ievent); + }// event loop - } + Unload(); - if (fSplitFile) - if ( fSplitFile->IsOpen() ) - fSplitFile->Close() ; + // gime->PhosLoader()->GetSDigitsDataLoader()->GetBaseTaskLoader()->SetDoNotReload(kTRUE); - if(strstr(option,"tim")){ gBenchmark->Stop("PHOSSDigitizer"); - cout << "AliPHOSSDigitizer:" << endl ; - cout << " took " << gBenchmark->GetCpuTime("PHOSSDigitizer") << " seconds for SDigitizing " - << gBenchmark->GetCpuTime("PHOSSDigitizer")/nevents << " seconds per event " << endl ; - cout << endl ; - } - - -} - -//__________________________________________________________________ -void AliPHOSSDigitizer::SetSDigitsBranch(const char * title ) -{ - // Setting title to branch SDigits - - TString stitle(title) ; - - // check if branch with title already exists - TBranch * sdigitsBranch = - static_cast(gAlice->TreeS()->GetListOfBranches()->FindObject("PHOS")) ; - TBranch * sdigitizerBranch = - static_cast(gAlice->TreeS()->GetListOfBranches()->FindObject("AliPHOSSDigitizer")) ; - const char * sdigitsTitle = sdigitsBranch ->GetTitle() ; - const char * sdigitizerTitle = sdigitizerBranch ->GetTitle() ; - if ( stitle.CompareTo(sdigitsTitle)==0 || stitle.CompareTo(sdigitizerTitle)==0 ){ - cerr << "ERROR: AliPHOSSdigitizer::SetSDigitsBranch -> Cannot overwrite existing branch with title " << title << endl ; - return ; - } - - cout << "AliPHOSSdigitizer::SetSDigitsBranch -> Changing SDigits file from " << GetName() << " to " << title << endl ; - - SetName(title) ; - - // Post to the WhiteBoard - AliPHOSGetter * gime = AliPHOSGetter::GetInstance() ; - gime->PostSDigits( title, GetTitle()) ; -} - -//__________________________________________________________________ -void AliPHOSSDigitizer::SetSplitFile(const TString splitFileName) -{ - // Diverts the SDigits in a file separate from the hits file - - TDirectory * cwd = gDirectory ; - - if ( !(gAlice->GetTreeSFileName() == splitFileName) ) { - if (gAlice->GetTreeSFile() ) - gAlice->GetTreeSFile()->Close() ; + Info("Exec"," took %f seconds for SDigitizing %f seconds per event", + gBenchmark->GetCpuTime("PHOSSDigitizer"), gBenchmark->GetCpuTime("PHOSSDigitizer")/nEvents) ; } - fSplitFile = gAlice->InitTreeFile("S",splitFileName.Data()); - fSplitFile->cd() ; - if ( !fSplitFile->Get("gAlice") ) - gAlice->Write(); - - TTree *treeE = gAlice->TreeE(); - if (!treeE) { - cerr << "ERROR: AliPHOSSDigitizer::SetSPlitFile -> No TreeE found "<Get("TreeE") ) { - AliHeader *header = new AliHeader(); - treeE->SetBranchAddress("Header", &header); - treeE->SetBranchStatus("*",1); - TTree *treeENew = treeE->CloneTree(); - treeENew->Write(); - } - - // copy AliceGeom - if ( !fSplitFile->Get("AliceGeom") ) { - TGeometry *AliceGeom = static_cast(cwd->Get("AliceGeom")); - if (!AliceGeom) { - cerr << "ERROR: AliPHOSSDigitizer::SetSPlitFile -> AliceGeom was not found in the input file "<Write(); - } - - gAlice->MakeTree("S",fSplitFile); - cwd->cd() ; - cout << "INFO: AliPHOSSDigitizer::SetSPlitMode -> SDigits will be stored in " << splitFileName.Data() << endl ; + //TFile f("out.root","RECREATE"); + //memwatcher.WriteToFile(); + //f.Close(); } //__________________________________________________________________ -void AliPHOSSDigitizer::Print(Option_t* option)const +void AliPHOSSDigitizer::Print(const Option_t *)const { // Prints parameters of SDigitizer - cout << "------------------- "<< GetName() << " -------------" << endl ; - cout << " Writing SDigits to branch with title " << GetName() << endl ; - cout << " with digitization parameters A = " << fA << endl ; - cout << " B = " << fB << endl ; - cout << " Threshold for Primary assignment= " << fPrimThreshold << endl ; - cout << "---------------------------------------------------"<SDigits(sdname.Data()) ; + AliPHOSGetter * gime = AliPHOSGetter::Instance() ; + const TClonesArray * sdigits = gime->SDigits() ; + + Info( "\nPrintSDigits", "event # %d %d sdigits", gAlice->GetEvNumber(), sdigits->GetEntriesFast() ) ; - cout << "AliPHOSSDigitiser: event " << gAlice->GetEvNumber() << endl ; - cout << " Number of entries in SDigits list " << sdigits->GetEntriesFast() << endl ; - cout << endl ; if(strstr(option,"all")||strstr(option,"EMC")){ //loop over digits AliPHOSDigit * digit; - cout << "EMC sdigits " << endl ; - cout << "Digit Id Amplitude Index Nprim Primaries list " << endl; + printf("\nEMC sdigits\n") ; Int_t maxEmc = gime->PHOSGeometry()->GetNModules()*gime->PHOSGeometry()->GetNCristalsInModule() ; Int_t index ; for (index = 0 ; (index < sdigits->GetEntriesFast()) && - (((AliPHOSDigit * ) sdigits->At(index))->GetId() <= maxEmc) ; index++) { - digit = (AliPHOSDigit * ) sdigits->At(index) ; - if(digit->GetNprimary() == 0) continue; - cout << setw(6) << digit->GetId() << " " << setw(10) << digit->GetAmp() << " " - << setw(6) << digit->GetIndexInList() << " " - << setw(5) << digit->GetNprimary() <<" "; - + ((dynamic_cast (sdigits->At(index)))->GetId() <= maxEmc) ; index++) { + digit = dynamic_cast( sdigits->At(index) ) ; + // if(digit->GetNprimary() == 0) + // continue; + printf("%6d %8d %6.5e %4d %2d :\n", + digit->GetId(), digit->GetAmp(), digit->GetTime(), digit->GetIndexInList(), digit->GetNprimary()) ; Int_t iprimary; - for (iprimary=0; iprimaryGetNprimary(); iprimary++) - cout << setw(5) << digit->GetPrimary(iprimary+1) << " "; - cout << endl; + for (iprimary=0; iprimaryGetNprimary(); iprimary++) { + printf("%d ",digit->GetPrimary(iprimary+1) ) ; + } } - cout << endl; } if(strstr(option,"all")||strstr(option,"CPV")){ //loop over CPV digits AliPHOSDigit * digit; - cout << "CPV sdigits " << endl ; - cout << "Digit Id Amplitude Index Nprim Primaries list " << endl; + printf("\nCPV sdigits\n") ; Int_t maxEmc = gime->PHOSGeometry()->GetNModules()*gime->PHOSGeometry()->GetNCristalsInModule() ; Int_t index ; for (index = 0 ; index < sdigits->GetEntriesFast(); index++) { - digit = (AliPHOSDigit * ) sdigits->At(index) ; + digit = dynamic_cast( sdigits->At(index) ) ; if(digit->GetId() > maxEmc){ - cout << setw(6) << digit->GetId() << " " << setw(10) << digit->GetAmp() << " " - << setw(6) << digit->GetIndexInList() << " " - << setw(5) << digit->GetNprimary() <<" "; - + printf("\n%6d %8d %4d %2d :", + digit->GetId(), digit->GetAmp(), digit->GetIndexInList(), digit->GetNprimary()) ; Int_t iprimary; - for (iprimary=0; iprimaryGetNprimary(); iprimary++) - cout << setw(5) << digit->GetPrimary(iprimary+1) << " "; - cout << endl; + for (iprimary=0; iprimaryGetNprimary(); iprimary++) { + printf("%d ",digit->GetPrimary(iprimary+1) ) ; + } } } } - } //____________________________________________________________________________ -void AliPHOSSDigitizer::UseHitsFrom(const char * filename) +void AliPHOSSDigitizer::Unload() const { - SetTitle(filename) ; - Init() ; + // Unloads the objects from the folder + AliPHOSGetter * gime = AliPHOSGetter::Instance() ; + AliPHOSLoader * loader = gime->PhosLoader() ; + loader->UnloadHits() ; + loader->UnloadSDigits() ; }