/************************************************************************** * 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$ */ //_________________________________________________________________________ //*-- Author : Dmitri Peressounko (SUBATECH & Kurchatov Institute) ////////////////////////////////////////////////////////////////////////////// // This TTask performs digitization of Summable digits (in the PHOS case it is just // the sum of contributions from all primary particles into a given cell). // In addition it performs mixing of summable digits from different events. // The name of the TTask is also the title of the branch that will contain // the created SDigits // The title of the TTAsk is the name of the file that contains the hits from // which the SDigits are created // // For each event two branches are created in TreeD: // "PHOS" - list of digits // "AliPHOSDigitizer" - AliPHOSDigitizer with all parameters used in digitization // // Note, that one can set a title for new digits branch, and repeat digitization with // another set of parameters. // // Use case: // root[0] AliPHOSDigitizer * d = new AliPHOSDigitizer() ; // root[1] d->ExecuteTask() // Warning in : object already instantiated // //Digitizes SDigitis in all events found in file galice.root // // root[2] AliPHOSDigitizer * d1 = new AliPHOSDigitizer("galice1.root") ; // // Will read sdigits from galice1.root // root[3] d1->MixWith("galice2.root") // Warning in : object already instantiated // // Reads another set of sdigits from galice2.root // root[3] d1->MixWith("galice3.root") // // Reads another set of sdigits from galice3.root // root[4] d->ExecuteTask("deb timing") // // Reads SDigits from files galice1.root, galice2.root .... // // mixes them and stores produced Digits in file galice1.root // // deb - prints number of produced digits // // deb all - prints list of produced digits // // timing - prints time used for digitization // // --- ROOT system --- #include "TFile.h" #include "TTree.h" #include "TSystem.h" #include "TROOT.h" #include "TFolder.h" #include "TObjString.h" #include "TBenchmark.h" // --- Standard library --- #include // --- AliRoot header files --- #include "AliRun.h" #include "AliRunDigitizer.h" #include "AliPHOSDigit.h" #include "AliPHOS.h" #include "AliPHOSGetter.h" #include "AliPHOSDigitizer.h" #include "AliPHOSSDigitizer.h" #include "AliPHOSGeometry.h" ClassImp(AliPHOSDigitizer) //____________________________________________________________________________ AliPHOSDigitizer::AliPHOSDigitizer() { // ctor fPinNoise = 0.01 ; fEMCDigitThreshold = 0.01 ; fCPVNoise = 0.01; fCPVDigitThreshold = 0.09 ; fTimeResolution = 1.0e-9 ; fDigitsInRun = 0 ; fPedestal = 0.; // Calibration parameters fSlope = 10000000. ; fARD = 0 ; // We work in the standalong mode } //____________________________________________________________________________ AliPHOSDigitizer::AliPHOSDigitizer(const char *headerFile,const char * name) { // ctor SetName(name) ; SetTitle(headerFile) ; fPinNoise = 0.01 ; fEMCDigitThreshold = 0.01 ; fCPVNoise = 0.01; fCPVDigitThreshold = 0.09 ; fDigitsInRun = 0 ; fPedestal = 0.; // Calibration parameters fSlope = 10000000. ; fARD = 0 ; // We work in the standalong mode Init() ; } //____________________________________________________________________________ AliPHOSDigitizer::AliPHOSDigitizer(AliRunDigitizer * ard) { // ctor fARD = ard ; SetName("Default"); SetTitle("aliroot") ; fPinNoise = 0.01 ; fEMCDigitThreshold = 0.01 ; fCPVNoise = 0.01; fCPVDigitThreshold = 0.09 ; fDigitsInRun = 0 ; fPedestal = 0.; // Calibration parameters fSlope = 10000000. ; Init() ; } //____________________________________________________________________________ AliPHOSDigitizer::~AliPHOSDigitizer() { // dtor } //____________________________________________________________________________ void AliPHOSDigitizer::Digitize(const Int_t event) { // Makes the digitization of the collected summable digits. // It first creates the array of all PHOS modules // filled with noise (different for EMC, CPV and PPSD) and // then adds contributions from SDigits. // This design avoids scanning over the list of digits to add // contribution to new SDigits only. if( strcmp(GetName(), "") == 0 ) Init() ; AliPHOSGetter * gime = AliPHOSGetter::GetInstance() ; TClonesArray * digits = gime->Digits(GetName()) ; digits->Clear() ; const AliPHOSGeometry *geom = gime->PHOSGeometry() ; //Making digits with noise, first EMC Int_t nEMC = geom->GetNModules()*geom->GetNPhi()*geom->GetNZ(); Int_t nCPV ; Int_t absID ; TString name = geom->GetName() ; nCPV = nEMC + geom->GetNumberOfCPVPadsZ()*geom->GetNumberOfCPVPadsPhi()* geom->GetNModules() ; digits->Expand(nCPV) ; // sdigitize random gaussian noise and add it to all cells (EMCA+CPV+PPSD) // get first the sdigitizer from the tasks list (must have same name as the digitizer) const AliPHOSSDigitizer * sDigitizer = gime->SDigitizer(GetName()); if ( !sDigitizer) { cerr << "ERROR: AliPHOSDigitizer::Digitize -> SDigitizer with name " << GetName() << " not found " << endl ; abort() ; } // loop through the sdigits posted to the White Board and add them to the noise TCollection * folderslist = ((TFolder*)gROOT->FindObjectAny("YSAlice/WhiteBoard/SDigits/PHOS"))->GetListOfFolders() ; TIter next(folderslist) ; TFolder * folder = 0 ; TClonesArray * sdigits = 0 ; Int_t input = 0 ; TObjArray * sdigArray = new TObjArray(2) ; while ( (folder = (TFolder*)next()) ) if ( (sdigits = (TClonesArray*)folder->FindObject(GetName()) ) ) { cout << "INFO: AliPHOSDigitizer::Digitize -> Adding SDigits " << GetName() << " from " << folder->GetName() << endl ; sdigArray->AddAt(sdigits, input) ; input++ ; } //Find the first crystall with sygnal Int_t nextSig = 200000 ; Int_t i; for(i=0; iAt(i) ; Int_t curNext = ((AliPHOSDigit *)sdigits->At(0))->GetId() ; if(curNext < nextSig) nextSig = curNext ; } TArrayF * energies = new TArrayF(input); TArrayF * times = new TArrayF(input) ; TArrayI index(input) ; index.Reset() ; //Set all indexes to zero AliPHOSDigit * digit ; AliPHOSDigit * curSDigit ; //Put Noise contribution for(absID = 1; absID <= nEMC; absID++){ Float_t noise = gRandom->Gaus(0., fPinNoise) ; new((*digits)[absID-1]) AliPHOSDigit( -1,absID,sDigitizer->Digitize(noise), TimeOfNoise() ) ; //look if we have to add signal? if(absID==nextSig){ //Add SDigits from all inputs digit = (AliPHOSDigit *) digits->At(absID-1) ; Int_t contrib = 0 ; energies[contrib] = digit->GetAmp() ; times[contrib] = digit->GetTime() ; //loop over inputs for(i=0; iAt(i))->At(index[i]) ; //May be several digits will contribute from the same input while(curSDigit && curSDigit->GetId() == absID){ //Shift primary to separate primaries belonging different inputs Int_t primaryoffset ; if(fARD) primaryoffset = fARD->GetMask(i) ; else primaryoffset = 10000000*i ; curSDigit->ShiftPrimary(primaryoffset) ; *digit = *digit + *curSDigit ; //add energies energies[contrib] = curSDigit->GetAmp() ; times[contrib] = curSDigit->GetTime() ; contrib++ ; index[i]++ ; curSDigit = (AliPHOSDigit*)((TClonesArray *)sdigArray->At(i))->At(index[i]) ; } } //calculate and set time Float_t time = FrontEdgeTime(energies, times) ; digit->SetTime(time) ; energies->Reset() ; times->Reset() ; //Find next signal module for(i=0; iAt(i)) ; Int_t curNext = ((AliPHOSDigit *) sdigits->At(index[i]))->GetId() ; if(curNext < nextSig) nextSig = curNext ; } } } //Now CPV digits (different noise and no timing) for(absID = nEMC+1; absID <= nCPV; absID++){ Float_t noise = gRandom->Gaus(0., fCPVNoise) ; new((*digits)[absID-1]) AliPHOSDigit( -1,absID,sDigitizer->Digitize(noise), TimeOfNoise() ) ; //look if we have to add signal? if(absID==nextSig){ digit = (AliPHOSDigit *) digits->At(absID-1) ; //Add SDigits from all inputs for(i=0; iAt(i))->At(index[i]) ; //May be several digits will contribute from the same input while(curSDigit && curSDigit->GetId() == absID){ //Shift primary to separate primaries belonging different inputs Int_t primaryoffset ; if(fARD) primaryoffset = fARD->GetMask(i) ; else primaryoffset = 10000000*i ; curSDigit->ShiftPrimary(primaryoffset) ; //add energies *digit = *digit + *curSDigit ; index[i]++ ; curSDigit = (AliPHOSDigit*)((TClonesArray *)sdigArray->At(i))->At(index[i]) ; } } //Find next signal module for(i=0; iAt(i) ; Int_t curNext = ((AliPHOSDigit *) sdigits->At(index[i]))->GetId() ; if(curNext < nextSig) nextSig = curNext ; } } } delete energies ; delete times ; delete sdigArray ; //We should not delete its contents //remove digits below thresholds for(absID = 0; absID < nEMC ; absID++) if(sDigitizer->Calibrate(((AliPHOSDigit*)digits->At(absID))->GetAmp()) < fEMCDigitThreshold) digits->RemoveAt(absID) ; for(absID = nEMC; absID < nCPV ; absID++) if(sDigitizer->Calibrate(((AliPHOSDigit*)digits->At(absID))->GetAmp()) < fCPVDigitThreshold) digits->RemoveAt(absID) ; digits->Compress() ; Int_t ndigits = digits->GetEntriesFast() ; digits->Expand(ndigits) ; //Set indexes in list of digits for (i = 0 ; i < ndigits ; i++) { AliPHOSDigit * digit = (AliPHOSDigit *) digits->At(i) ; digit->SetIndexInList(i) ; } } //____________________________________________________________________________ void AliPHOSDigitizer::Exec(Option_t *option) { // Managing method if( strcmp(GetName(), "") == 0 ) Init() ; if (strstr(option,"print")) { Print(""); return ; } if(strstr(option,"tim")) gBenchmark->Start("PHOSDigitizer"); AliPHOSGetter * gime = AliPHOSGetter::GetInstance() ; Int_t nevents ; TTree * treeD ; if(fARD){ treeD = fARD->GetTreeD() ; nevents = 1 ; // Will process only one event } else { gAlice->GetEvent(0) ; nevents = (Int_t) gAlice->TreeE()->GetEntries() ; treeD=gAlice->TreeD() ; } if(treeD == 0 ){ cerr << " AliPHOSDigitizer :: Can not find TreeD " << endl ; return ; } //Check, if this branch already exits TObjArray * lob = (TObjArray*)treeD->GetListOfBranches() ; TIter next(lob) ; TBranch * branch = 0 ; Bool_t phosfound = kFALSE, digitizerfound = kFALSE ; while ( (branch = (TBranch*)next()) && (!phosfound || !digitizerfound) ) { if ( (strcmp(branch->GetName(), "PHOS")==0) && (strcmp(branch->GetTitle(), GetName())==0) ) phosfound = kTRUE ; else if ( (strcmp(branch->GetName(), "AliPHOSDigitizer")==0) && (strcmp(branch->GetTitle(), GetName())==0) ) digitizerfound = kTRUE ; } if ( phosfound ) { cerr << "WARNING: AliPHOSDigitizer -> Digits branch with name " << GetName() << " already exits" << endl ; return ; } if ( digitizerfound ) { cerr << "WARNING: AliPHOSDigitizer -> Digitizer branch with name " << GetName() << " already exits" << endl ; return ; } Int_t ievent ; for(ievent = 0; ievent < nevents; ievent++){ if(fARD){ Int_t input ; for(input = 0 ; input < fARD->GetNinputs(); input ++){ TTree * treeS = fARD->GetInputTreeS(input) ; if(!treeS){ cerr << "AliPHOSDigitizer -> No Input " << endl ; return ; } gime->ReadTreeS(treeS,input) ; } } else gime->Event(ievent,"S") ; Digitize(ievent) ; //Add prepared SDigits to digits and add the noise WriteDigits(ievent) ; if(strstr(option,"deb")) PrintDigits(option); } if(strstr(option,"tim")){ gBenchmark->Stop("PHOSDigitizer"); cout << "AliPHOSDigitizer:" << endl ; cout << " took " << gBenchmark->GetCpuTime("PHOSDigitizer") << " seconds for Digitizing " << gBenchmark->GetCpuTime("PHOSDigitizer")/nevents << " seconds per event " << endl ; cout << endl ; } } //____________________________________________________________________________ Float_t AliPHOSDigitizer::FrontEdgeTime(TArrayF * energies, TArrayF * times) { // Float_t curtime = times->At(0) ; Float_t time = curtime ; Int_t i = 1 ; while(curtime){ if(time > curtime) time = curtime ; curtime = times->At(i++) ; } return time ; } //____________________________________________________________________________ Bool_t AliPHOSDigitizer::Init() { // Makes all memory allocations // Adds Digitizer task to the folder of PHOS tasks //============================================================= YS // The initialisation is now done by AliPHOSGetter if( strcmp(GetTitle(), "") == 0 ) SetTitle("galice.root") ; // the SDigits name is stored by AliPHOSGetter as the name of the TClones Array // //YSAlice/WhiteBoard/SDigits/PHOS/headerFile/branchname and has branchTitle as title. AliPHOSGetter * gime = AliPHOSGetter::GetInstance(GetTitle(), GetName()) ; if ( gime == 0 ) { cerr << "ERROR: AliPHOSDigitizer::Init -> Could not obtain the Getter object !" << endl ; return kFALSE; } // create a folder on the white board //YSAlice/WhiteBoard/Digits/PHOS/headerFile/digitsTitle gime->PostDigits(GetName() ) ; //add Task to //YSAlice/tasks/Digitizer/PHOS gime->PostDigitizer(this) ; //Mark that we will use current header file if(!fARD){ gime->PostSDigits(GetName(),GetTitle()) ; gime->PostSDigitizer(GetName(),GetTitle()) ; } return kTRUE ; } //__________________________________________________________________ void AliPHOSDigitizer::MixWith(const char* headerFile) { // Allows to produce digits by superimposing background and signal event. // It is assumed, that headers file with SIGNAL events is opened in // the constructor. // Sets the BACKGROUND event, with which the SIGNAL event is to be mixed // Thus we avoid writing (changing) huge and expensive // backgound files: all output will be writen into SIGNAL, i.e. // opened in constructor file. // // One can open as many files to mix with as one needs. // However only Sdigits with the same name (i.e. constructed with the same SDigitizer) // can be mixed. if( strcmp(GetName(), "") == 0 ) Init() ; if(fARD){ cout << "Can not use this method under AliRunDigitizer " << endl ; return ; } // check if the specified SDigits do not already exist on the White Board: // //YSAlice/WhiteBoard/SDigits/PHOS/headerFile/sDigitsTitle TString path = "YSAlice/WhiteBoard/SDigits/PHOS/" ; path += headerFile ; path += "/" ; path += GetName() ; if ( gROOT->FindObjectAny(path.Data()) ) { cerr << "WARNING: AliPHOSDigitizer::MixWith -> Entry already exists, do not add" << endl ; return; } AliPHOSGetter * gime = AliPHOSGetter::GetInstance() ; gime->PostSDigits(GetName(),headerFile) ; // check if the requested file is already open or exist and if SDigits Branch exist TFile * file = (TFile*)gROOT->FindObject(headerFile); if ( !file ) { file = new TFile(headerFile, "READ") ; if (!file) { cerr << "ERROR: AliPHOSDigitizer::MixWith -> File " << headerFile << " does not exist!" << endl ; return ; } } } //__________________________________________________________________ void AliPHOSDigitizer::Print(Option_t* option)const { // Print Digitizer's parameters if( strcmp(GetName(), "") != 0 ){ cout << "------------------- "<< GetName() << " -------------" << endl ; cout << "Digitizing sDigits from file(s): " <FindObjectAny("YSAlice/WhiteBoard/SDigits/PHOS"))->GetListOfFolders() ; TIter next(folderslist) ; TFolder * folder = 0 ; while ( (folder = (TFolder*)next()) ) { if ( folder->FindObject(GetName()) ) cout << "Adding SDigits " << GetName() << " from " << folder->GetName() << endl ; } cout << endl ; cout << "Writing digits to " << GetTitle() << endl ; cout << endl ; cout << "With following parameters: " << endl ; cout << " Electronics noise in EMC (fPinNoise) = " << fPinNoise << endl ; cout << " Threshold in EMC (fEMCDigitThreshold) = " << fEMCDigitThreshold << endl ; ; cout << " Noise in CPV (fCPVNoise) = " << fCPVNoise << endl ; cout << " Threshold in CPV (fCPVDigitThreshold) = " << fCPVDigitThreshold << endl ; cout << "---------------------------------------------------" << endl ; } else cout << "AliPHOSDigitizer not initialized " << endl ; } //__________________________________________________________________ void AliPHOSDigitizer::PrintDigits(Option_t * option){ // Print a table of digits AliPHOSGetter * gime = AliPHOSGetter::GetInstance() ; TClonesArray * digits = gime->Digits() ; cout << "AliPHOSDigitiser: event " << gAlice->GetEvNumber() << endl ; cout << " Number of entries in Digits list " << digits->GetEntriesFast() << endl ; cout << endl ; if(strstr(option,"all")||strstr(option,"EMC")){ //loop over digits AliPHOSDigit * digit; cout << "Digit Id Amplitude Index " << " Nprim " << " Primaries list " << endl; Int_t maxEmc = gime->PHOSGeometry()->GetNModules()*gime->PHOSGeometry()->GetNCristalsInModule() ; Int_t index ; for (index = 0 ; (index < digits->GetEntries()) && (((AliPHOSDigit * ) digits->At(index))->GetId() <= maxEmc) ; index++) { digit = (AliPHOSDigit * ) digits->At(index) ; if(digit->GetNprimary() == 0) continue; cout << setw(6) << digit->GetId() << " " << setw(10) << digit->GetAmp() << " " << setw(6) << digit->GetIndexInList() << " " << setw(5) << digit->GetNprimary() <<" "; Int_t iprimary; for (iprimary=0; iprimaryGetNprimary(); iprimary++) cout << setw(5) << digit->GetPrimary(iprimary+1) << " "; cout << endl; } cout << endl; } if(strstr(option,"all")||strstr(option,"CPV")){ //loop over CPV digits AliPHOSDigit * digit; cout << "Digit Id " << " Amplitude " << " Index " << " Nprim " << " Primaries list " << endl; Int_t maxEmc = gime->PHOSGeometry()->GetNModules()*gime->PHOSGeometry()->GetNCristalsInModule() ; Int_t index ; for (index = 0 ; index < digits->GetEntries(); index++) { digit = (AliPHOSDigit * ) digits->At(index) ; if(digit->GetId() > maxEmc){ cout << setw(6) << digit->GetId() << " " << setw(10) << digit->GetAmp() << " " << setw(6) << digit->GetIndexInList() << " " << setw(5) << digit->GetNprimary() <<" "; Int_t iprimary; for (iprimary=0; iprimaryGetNprimary(); iprimary++) cout << setw(5) << digit->GetPrimary(iprimary+1) << " "; cout << endl; } } } } //__________________________________________________________________ void AliPHOSDigitizer::SetSDigitsBranch(const char* title) { // we set title (comment) of the SDigits branch in the first! header file if( strcmp(GetName(), "") == 0 ) Init() ; AliPHOSGetter::GetInstance()->SDigits()->SetName(title) ; } //__________________________________________________________________ Float_t AliPHOSDigitizer::TimeOfNoise(void) { // Calculates the time signal generated by noise //to be rewritten, now returns just big number return 1. ; } //____________________________________________________________________________ void AliPHOSDigitizer::Reset() { // sets current event number to the first simulated event if( strcmp(GetName(), "") == 0 ) Init() ; // Int_t inputs ; // for(inputs = 0; inputs < fNinputs ;inputs++) // fIevent->AddAt(-1, inputs ) ; } //____________________________________________________________________________ void AliPHOSDigitizer::WriteDigits(Int_t event) { // Makes TreeD in the output file. // Check if branch already exists: // if yes, exit without writing: ROOT TTree does not support overwriting/updating of // already existing branches. // else creates branch with Digits, named "PHOS", title "...", // and branch "AliPHOSDigitizer", with the same title to keep all the parameters // and names of files, from which digits are made. AliPHOSGetter * gime = AliPHOSGetter::GetInstance() ; TClonesArray * digits = gime->Digits(GetName()) ; TTree * treeD ; if(fARD) treeD = fARD->GetTreeD() ; else treeD = gAlice->TreeD(); // create new branches // -- generate file name if necessary char * file =0; if(gSystem->Getenv("CONFIG_SPLIT_FILE")){ //generating file name file = new char[strlen(gAlice->GetBaseFile())+20] ; sprintf(file,"%s/PHOS.Digits.root",gAlice->GetBaseFile()) ; } TDirectory *cwd = gDirectory; // -- create Digits branch Int_t bufferSize = 32000 ; TBranch * digitsBranch = treeD->Branch("PHOS",&digits,bufferSize); digitsBranch->SetTitle(GetName()); if (file) { digitsBranch->SetFile(file); TIter next( digitsBranch->GetListOfBranches()); TBranch * sbr ; while ((sbr=(TBranch*)next())) { sbr->SetFile(file); } cwd->cd(); } // -- Create Digitizer branch Int_t splitlevel = 0 ; AliPHOSDigitizer * d = gime->Digitizer(GetName()) ; TBranch * digitizerBranch = treeD->Branch("AliPHOSDigitizer", "AliPHOSDigitizer", &d,bufferSize,splitlevel); digitizerBranch->SetTitle(GetName()); if (file) { digitizerBranch->SetFile(file); TIter next( digitizerBranch->GetListOfBranches()); TBranch * sbr; while ((sbr=(TBranch*)next())) { sbr->SetFile(file); } cwd->cd(); } digitsBranch->Fill() ; digitizerBranch->Fill() ; treeD->Write(0,kOverwrite) ; }