]> git.uio.no Git - u/mrichter/AliRoot.git/blobdiff - EMCAL/AliEMCALSDigitizer.cxx
Adding MUON HLT code to the repository.
[u/mrichter/AliRoot.git] / EMCAL / AliEMCALSDigitizer.cxx
index 1639b06bc2256cf909f2a2eb4dc137137d82000d..89c786dcd66fa2fd6bb4e2163c7fbe29631ef017 100644 (file)
 //////////////////////////////////////////////////////////////////////////////
 
 // --- 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 "stdlib.h"
 
 // --- AliRoot header files ---
-#include "AliRun.h"
-#include "AliHeader.h"
+#include "AliLog.h"
 #include "AliEMCALDigit.h"
-#include "AliEMCALGeometry.h"
 #include "AliEMCALGetter.h"
 #include "AliEMCALHit.h"
 #include "AliEMCALSDigitizer.h"
+#include "AliEMCALGeometry.h"
+              //#include "AliMemoryWatcher.h"
 
 ClassImp(AliEMCALSDigitizer)
            
 //____________________________________________________________________________ 
-  AliEMCALSDigitizer::AliEMCALSDigitizer():TTask("AliEMCALSDigitizer","") 
+  AliEMCALSDigitizer::AliEMCALSDigitizer():TTask("","") 
 {
   // ctor
-  InitParameters() ; 
-  fSplitFile           = 0 ; 
-  fToSplit             = kFALSE ;
+  fFirstEvent = fLastEvent  = 0 ;  
   fDefaultInit = kTRUE ; 
 }
 
 //____________________________________________________________________________ 
-AliEMCALSDigitizer::AliEMCALSDigitizer(const char* headerFile, const char *sDigitsTitle, const Bool_t toSplit):
-TTask(sDigitsTitle, headerFile)
+AliEMCALSDigitizer::AliEMCALSDigitizer(const char * alirunFileName, 
+                                      const char * eventFolderName):
+  TTask("EMCAL"+AliConfig::Instance()->GetSDigitizerTaskName(), alirunFileName),
+  fEventFolderName(eventFolderName)
 {
   // ctor
-  fToSplit = toSplit ;
+  fFirstEvent = fLastEvent  = 0 ; // runs one event by defaut  
   Init();
   InitParameters() ; 
   fDefaultInit = kFALSE ; 
 }
 
+
 //____________________________________________________________________________ 
-AliEMCALSDigitizer::~AliEMCALSDigitizer()
-{
+AliEMCALSDigitizer::AliEMCALSDigitizer(const AliEMCALSDigitizer & sd) : TTask(sd) {
+  //cpy ctor 
+
+  fFirstEvent    = sd.fFirstEvent ; 
+  fLastEvent     = sd.fLastEvent ;
+  fA             = sd.fA ;
+  fB             = sd.fB ;
+  fECPrimThreshold  = sd.fECPrimThreshold ;
+  fSDigitsInRun  = sd.fSDigitsInRun ;
+  SetName(sd.GetName()) ; 
+  SetTitle(sd.GetTitle()) ; 
+  fEventFolderName = sd.fEventFolderName;
+}
+
+
+//____________________________________________________________________________ 
+AliEMCALSDigitizer::~AliEMCALSDigitizer() {
   // dtor
-  fSplitFile = 0 ; 
+  AliEMCALGetter * gime = 
+    //   AliEMCALGetter::Instance(GetTitle(), fEventFolderName.Data());  
+    AliEMCALGetter::Instance();  
+  gime->EmcalLoader()->CleanSDigitizer();
 }
 
 //____________________________________________________________________________ 
@@ -108,108 +122,84 @@ void AliEMCALSDigitizer::Init(){
   //============================================================= YS
   //  The initialisation is now done by the getter
 
-  if( strcmp(GetTitle(), "") == 0 )
-    SetTitle("galice.root") ;
+  fInit = kTRUE ; 
    
-  AliEMCALGetter * gime = AliEMCALGetter::GetInstance(GetTitle(), GetName(), fToSplit) ; 
+  AliEMCALGetter * gime = AliEMCALGetter::Instance(GetTitle(), fEventFolderName.Data());  
   if ( gime == 0 ) {
-    Error("Init", "Could not obtain the Getter object !" ) ;  
+    Fatal("Init", "Could not obtain the Getter objectfor file %s and event %s !", GetTitle(), fEventFolderName.Data()) ;  
     return ;
   } 
   
-  gime->PostSDigits( GetName(), GetTitle() ) ; 
-  
-  fSplitFile = 0 ;
-  if(fToSplit){
-    // construct the name of the file as /path/EMCAL.SDigits.root
-    // First - extract full path if necessary
-    TString sDigitsFileName(GetTitle()) ;
-    Ssiz_t islash = sDigitsFileName.Last('/') ;
-    if(islash<sDigitsFileName.Length())
-      sDigitsFileName.Remove(islash+1,sDigitsFileName.Length()) ;
-    else
-      sDigitsFileName="" ;
-
-    // Next - append the file name 
-    sDigitsFileName+="EMCAL.SDigits." ;
-    if((strcmp(GetName(),"Default")!=0)&&(strcmp(GetName(),"")!=0)){
-      sDigitsFileName+=GetName() ;
-      sDigitsFileName+="." ;
-    }
-    sDigitsFileName+="root" ;
-
-    // Finally - check if the file already opened or open the file
-    fSplitFile = static_cast<TFile*>(gROOT->GetFile(sDigitsFileName.Data()));   
-    if(!fSplitFile)
-      fSplitFile =  TFile::Open(sDigitsFileName.Data(),"update") ;
+  TString opt("SDigits") ; 
+  if(gime->VersionExists(opt) ) { 
+    Error( "Init", "Give a version name different from %s", fEventFolderName.Data() ) ;
+    fInit = kFALSE ; 
   }
-
-  TString sdname(GetName() );
-  sdname.Append(":") ;
-  sdname.Append(GetTitle() ) ;
-  SetName(sdname) ;
-  gime->PostSDigitizer(this) ;
-
+  
+  gime->PostSDigitizer(this);
+  gime->EmcalLoader()->GetSDigitsDataLoader()->GetBaseTaskLoader()->SetDoNotReload(kTRUE);
+  
 }
 
 //____________________________________________________________________________ 
 void AliEMCALSDigitizer::InitParameters()
 {
+  // Initializes parameters
   fA                      = 0;
   fB                      = 10000000.;
 
-  AliEMCALGetter * gime = AliEMCALGetter::GetInstance() ;
+  AliEMCALGetter * gime = AliEMCALGetter::Instance() ;
   const AliEMCALGeometry * geom = gime->EMCALGeometry() ; 
   if (geom->GetSampling() == 0.) {
-    Error("InitParameters", "Sampling factor not set !") ; 
-    abort() ;
+    Fatal("InitParameters", "Sampling factor not set !") ; 
   }
-  else
-    Info("InitParameters", "Sampling factor set to %f\n", geom->GetSampling()) ; 
+//   else
+//     Info("InitParameters", "Sampling factor set to %f", geom->GetSampling()) ; 
   
   // this threshold corresponds approximately to 100 MeV
-  fECPrimThreshold     = 100E-3 / ( geom->GetSampling() * ( geom->GetNPRLayers() + geom->GetNECLayers()) ) * geom->GetNECLayers() ;
-  fPREPrimThreshold    = 100E-3 / ( geom->GetSampling() * ( geom->GetNPRLayers() + geom->GetNECLayers()) ) * geom->GetNPRLayers() ; 
-  fHCPrimThreshold     = fECPrimThreshold/5. ; // 5 is totally arbitrary
-
+  fECPrimThreshold     = 100E-3;
 }
 
 //____________________________________________________________________________
 void AliEMCALSDigitizer::Exec(Option_t *option) 
 { 
-  // Collects all hits in the section (PRE/ECAL/HCAL) of the same tower into digit
-
-  if( strcmp(GetName(), "") == 0 )
-    Init() ;
-  
+  // Collects all hit of the same tower into digits
   if (strstr(option, "print") ) {
-    Print("") ; 
+    Print() ; 
     return ; 
   }
   
   if(strstr(option,"tim"))
     gBenchmark->Start("EMCALSDigitizer");
 
-  //Check, if this branch already exits
-  AliEMCALGetter * gime = AliEMCALGetter::GetInstance() ;
-  if(gime->BranchExists("SDigits") ) 
-    return;   
+  //AliEMCALGetter * gime = AliEMCALGetter::Instance(GetTitle()) ;
+  AliEMCALGetter * gime = AliEMCALGetter::Instance() ;
 
-  TString sdname(GetName()) ;
-  sdname.Remove(sdname.Index(GetTitle())-1) ;
+  //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 ;
+    }
+
+  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 nevents = gime->MaxEvent() ; 
   Int_t ievent ;   
-  for(ievent = 0; ievent < nevents; ievent++){     
+
+  //AliMemoryWatcher memwatcher;
+
+  for (ievent = fFirstEvent; ievent <= fLastEvent; ievent++) {
     gime->Event(ievent,"H") ;  
-    const TClonesArray * hits = gime->Hits() ; 
-    TClonesArray * sdigits = gime->SDigits(sdname.Data()) ;
+    TTree * treeS = gime->TreeS(); 
+    TClonesArray * hits = gime->Hits() ; 
+    TClonesArray * sdigits = gime->SDigits() ;
     sdigits->Clear();
     Int_t nSdigits = 0 ;
-    
     //Now make SDigits from hits, for EMCAL it is the same, so just copy    
-    Int_t nPrim =  static_cast<Int_t>((gAlice->TreeH())->GetEntries()) ; 
+    Int_t nPrim =  static_cast<Int_t>((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 ;
@@ -224,24 +214,8 @@ void AliEMCALSDigitizer::Exec(Option_t *option)
        Bool_t newsdigit = kTRUE; 
 
        // Assign primary number only if deposited energy is significant
-
-       const AliEMCALGeometry * geom = gime->EMCALGeometry() ; 
-
-       if (gDebug) 
-         Info("Exec", "id = %d energy = %f thresholdPRE = %f thresholdEC = %f thresholdHC = %f \n", 
-              hit->GetId(), hit->GetEnergy(), fPREPrimThreshold, fECPrimThreshold, fHCPrimThreshold) ;    
-       
-       if( geom->IsInPRE(hit->GetId()) )  
-         if( hit->GetEnergy() > fPREPrimThreshold )
-           curSDigit =  new AliEMCALDigit( hit->GetPrimary(),
-                                           hit->GetIparent(), hit->GetId(), 
-                                           Digitize(hit->GetEnergy()), hit->GetTime() ) ;
-         else
-           curSDigit =  new AliEMCALDigit( -1               , 
-                                           -1               ,
-                                           hit->GetId(), 
-                                           Digitize(hit->GetEnergy()), hit->GetTime() ) ;
-       else if( geom->IsInECAL(hit->GetId()) )
+       AliEMCALGeometry * geom = gime->EMCALGeometry() ; 
+       if( geom->IsInECA(hit->GetId()) ){
          if( hit->GetEnergy() >  fECPrimThreshold )
            curSDigit =  new AliEMCALDigit( hit->GetPrimary(),
                                            hit->GetIparent(), hit->GetId(), 
@@ -251,41 +225,36 @@ void AliEMCALSDigitizer::Exec(Option_t *option)
                                            -1               ,
                                            hit->GetId(), 
                                            Digitize(hit->GetEnergy()), hit->GetTime() ) ;
-       else if( geom->IsInHCAL(hit->GetId()) )
-         if( hit->GetEnergy() >  fHCPrimThreshold )
-           
-           curSDigit =  new AliEMCALDigit( hit->GetPrimary(),
-                                           hit->GetIparent(), hit->GetId(), 
-                                           Digitize(hit->GetEnergy()), hit->GetTime() ) ;
-         else
-           curSDigit =  new AliEMCALDigit( -1               , 
-                                           -1               ,
-                                           hit->GetId(), 
-                                           Digitize(hit->GetEnergy()), hit->GetTime() ) ;
-       
+       }
+       else{
+         newsdigit = kFALSE;
+       }
+
        Int_t check = 0 ;
+
        for(check= 0; check < nSdigits ; check++) {
          sdigit = dynamic_cast<AliEMCALDigit *>(sdigits->At(check)) ;
-         if( sdigit->GetId() == curSDigit->GetId()) { // Are we in the same ECAL/HCAL/preshower tower ?              
+          if(curSDigit != 0){
+           if( sdigit->GetId() == curSDigit->GetId()) { // Are we in the same ECAL tower ?              
            *sdigit = *sdigit + *curSDigit;
            newsdigit = kFALSE;
+           }
          }
        }
-       if (newsdigit) { 
+       if (newsdigit) {
          new((*sdigits)[nSdigits])  AliEMCALDigit(*curSDigit);
          nSdigits++ ;  
        }
        delete curSDigit ; 
-      }  // loop over all hits (hit = deposited energy/layer/entering particle)
+      }  // loop over all hits (hit = deposited energy/entering particle)
     } // loop over iprim
-    
     sdigits->Sort() ;
     
     nSdigits = sdigits->GetEntriesFast() ;
     fSDigitsInRun += nSdigits ;  
     sdigits->Expand(nSdigits) ;
 
-    Int_t NPrimarymax = -1 ; 
+    Int_t nPrimarymax = -1 ; 
     Int_t i ;
     for (i = 0 ; i < sdigits->GetEntriesFast() ; i++) { 
       AliEMCALDigit * sdigit = dynamic_cast<AliEMCALDigit *>(sdigits->At(i)) ;
@@ -293,99 +262,58 @@ void AliEMCALSDigitizer::Exec(Option_t *option)
     }
     
     for (i = 0 ; i < sdigits->GetEntriesFast() ; i++) {   
-      if (((dynamic_cast<AliEMCALDigit *>(sdigits->At(i)))->GetNprimary()) > NPrimarymax)
-       NPrimarymax = ((dynamic_cast<AliEMCALDigit *>(sdigits->At(i)))->GetNprimary()) ;
+      if (((dynamic_cast<AliEMCALDigit *>(sdigits->At(i)))->GetNprimary()) > nPrimarymax)
+       nPrimarymax = ((dynamic_cast<AliEMCALDigit *>(sdigits->At(i)))->GetNprimary()) ;
     }
     
-    // Now write SDigits
-    
-    if(gAlice->TreeS() == 0 || (fSplitFile))  //<--- To be checked: we should not create TreeS if it is already here
-      gAlice->MakeTree("S",fSplitFile);
-  
-    if(fSplitFile)
-      fSplitFile->cd() ;
-    
+    // Now write SDigits    
     //First list of sdigits
+
     Int_t bufferSize = 32000 ;    
-    TBranch * sdigitsBranch = gAlice->TreeS()->Branch("EMCAL",&sdigits,bufferSize);
-    sdigitsBranch->SetTitle(sdname);
+    TBranch * sdigitsBranch = treeS->Branch("EMCAL",&sdigits,bufferSize);
+    sdigitsBranch->Fill() ;
+
+    gime->WriteSDigits("OVERWRITE");
     
     //NEXT - SDigitizer
-    Int_t splitlevel = 0 ;
-    AliEMCALSDigitizer * sd = this ;
-    TBranch * sdigitizerBranch = gAlice->TreeS()->Branch("AliEMCALSDigitizer","AliEMCALSDigitizer",
-                                                        &sd,bufferSize,splitlevel); 
-    sdigitizerBranch->SetTitle(sdname);
-    
-    sdigitsBranch->Fill() ; 
-    sdigitizerBranch->Fill() ; 
-  
-    gAlice->TreeS()->AutoSave() ;
+
+    gime->WriteSDigitizer("OVERWRITE");
     
     if(strstr(option,"deb"))
       PrintSDigits(option) ;  
-  }
-   
+    
+    //gObjectTable->Print() ; 
+    //memwatcher.Watch(ievent); 
+  }// event loop
+
+  Unload();
+  
+  gime->EmcalLoader()->GetSDigitsDataLoader()->GetBaseTaskLoader()->SetDoNotReload(kTRUE);
+  
   if(strstr(option,"tim")){
     gBenchmark->Stop("EMCALSDigitizer"); 
-    Info("Exec", "took %f seconds for SDigitizing %f seconds per event", 
-        gBenchmark->GetCpuTime("EMCALSDigitizer"), gBenchmark->GetCpuTime("EMCALSDigitizer") ) ; 
+    printf("Exec: took %f seconds for SDigitizing %f seconds per event", 
+        gBenchmark->GetCpuTime("EMCALSDigitizer"), gBenchmark->GetCpuTime("EMCALSDigitizer")/nEvents ) ; 
   }
-}
 
-//__________________________________________________________________
-void AliEMCALSDigitizer::SetSDigitsBranch(const char * title ){
-  // Setting title to branch SDigits 
-
-  TString stitle(title) ;
-
-  // check if branch with title already exists
-  TBranch * sdigitsBranch    = 
-    static_cast<TBranch*>(gAlice->TreeS()->GetListOfBranches()->FindObject("EMCAL")) ; 
-  TBranch * sdigitizerBranch =  
-    static_cast<TBranch*>(gAlice->TreeS()->GetListOfBranches()->FindObject("AliEMCALSDigitizer")) ;
-  const char * sdigitsTitle    = sdigitsBranch ->GetTitle() ;  
-  const char * sdigitizerTitle = sdigitizerBranch ->GetTitle() ;
-  if ( stitle.CompareTo(sdigitsTitle)==0 || stitle.CompareTo(sdigitizerTitle)==0 ){
-    Error("SetSDigitsBranch", "Cannot overwrite existing branch with title %s", title) ;
-    return ;
-  }
-  
-  Info("SetSDigitsBranch", "Changing SDigits file from %s to %s", GetName(), title) ;
-
-  SetName(title) ; 
-    
-  // Post to the WhiteBoard
-  AliEMCALGetter * gime = AliEMCALGetter::GetInstance() ; 
-  gime->PostSDigits( title, GetTitle()) ; 
+  //TFile f("out.root","RECREATE");
+  //memwatcher.WriteToFile(); 
+  //f.Close();
 }
 
 
 //__________________________________________________________________
-void AliEMCALSDigitizer::Print(Option_t* option)const
+void AliEMCALSDigitizer::Print()const
 { 
   // Prints parameters of SDigitizer
+  printf("Print: \n------------------- %s -------------", GetName() ) ; 
+  printf("   Writing SDigits to branch with title  %s\n", fEventFolderName.Data()) ;
+  printf("   with digitization parameters  A = %f\n", fA) ; 
+  printf("                                 B = %f\n", fB) ;
+  printf("   Threshold for EC Primary assignment= %f\n", fECPrimThreshold)  ; 
+  printf("---------------------------------------------------\n") ;
 
-  TString message("\n") ; 
-  message += "------------------- "; 
-  message += GetName() ; 
-  message += " -------------\n" ;
-  message += "   Writing SDigitis to branch with title  " ; 
-  message += GetName() ;
-  message += "\n   with digitization parameters  A               = " ; 
-  message += fA ;
-  message += "\n                                 B               = " ; 
-  message += fB ; 
-  message += "\n   Threshold for Primary assignment in PreShower = " ; 
-  message += fPREPrimThreshold ; 
-  message += "\n   Threshold for Primary assignment in EC section= " ; 
-  message += fECPrimThreshold ; 
-  message += "\n   Threshold for Primary assignment in HC section= " ; 
-  message += fHCPrimThreshold ; 
-  message += "\n---------------------------------------------------" ;
-  
-  Info("Print", message.Data() ) ; 
 }
 
 //__________________________________________________________________
@@ -393,72 +321,54 @@ Bool_t AliEMCALSDigitizer::operator==( AliEMCALSDigitizer const &sd )const
 {
   // Equal operator.
   // SDititizers are equal if their pedestal, slope and threshold are equal
-
   if( (fA==sd.fA)&&(fB==sd.fB)&&
-      (fECPrimThreshold==sd.fECPrimThreshold) &&
-      (fHCPrimThreshold==sd.fHCPrimThreshold) &&
-      (fPREPrimThreshold==sd.fPREPrimThreshold))
+      (fECPrimThreshold==sd.fECPrimThreshold))
     return kTRUE ;
   else
     return kFALSE ;
 }
 
 //__________________________________________________________________
-void AliEMCALSDigitizer::PrintSDigits(Option_t * option){
+void AliEMCALSDigitizer::PrintSDigits(Option_t * option)
+{
   //Prints list of digits produced at the current pass of AliEMCALDigitizer
   
-  AliEMCALGetter * gime = AliEMCALGetter::GetInstance() ; 
-  TString sdname(GetName()) ;
-  sdname.Remove(sdname.Index(GetTitle())-1) ;
-  const TClonesArray * sdigits = gime->SDigits(sdname.Data()) ; 
   
-  TString message("\n") ;  
-  message += "event " ; 
-  message += gAlice->GetEvNumber() ;
-  message += "\n      Number of entries in SDigits list " ;
-  message += sdigits->GetEntriesFast() ; 
+  AliEMCALGetter * gime = AliEMCALGetter::Instance() ; 
+  const TClonesArray * sdigits = gime->SDigits() ; 
+  
+  printf("\n") ;  
+  printf("event %i", gAlice->GetEvNumber()) ;
+  printf("\n      Number of entries in SDigits list %i", sdigits->GetEntriesFast()); 
   if(strstr(option,"all")||strstr(option,"EMC")){
     
     //loop over digits
     AliEMCALDigit * digit;
-    message += "\n   Id  Amplitude    Time          Index Nprim: Primaries list \n" ;    
+    printf("\n   Id  Amplitude    Time          Index Nprim: Primaries list \n") ;    
     Int_t index ;
     char * tempo = new char[8192]; 
     for (index = 0 ; index < sdigits->GetEntries() ; index++) {
       digit = dynamic_cast<AliEMCALDigit *>( sdigits->At(index) ) ;
       sprintf(tempo, "\n%6d  %8d    %6.5e %4d      %2d :",
              digit->GetId(), digit->GetAmp(), digit->GetTime(), digit->GetIndexInList(), digit->GetNprimary()) ;  
-      message += tempo 
+      printf(tempo)
       
       Int_t iprimary;
       for (iprimary=0; iprimary<digit->GetNprimary(); iprimary++) {
        sprintf(tempo, "%d ",digit->GetPrimary(iprimary+1) ) ; 
-       message += tempo 
+       printf(tempo)
       }         
     }
     delete tempo ;
   }
-  Info("PrintSDigits", message.Data() ) ; 
 }
 
-//_______________________________________________________________________________________
-// void AliEMCALSDigitizer::TestTowerID(void)
-// {
-//   Int_t j;
-
-//   Bool_t preshower = kFALSE;
-//   for (j = 0 ; j < 10 ; j++){  // loop over hit id
-//     Int_t i;
-//    for (i = 0 ; i <= 2 ; i++){  // loop over 
-//      Int_t k = i*96*144+j*144+1;
-//       Info("TestTowerID", " Hit Index = %d  %d   TOWERID = %d", k, j*10, Layer2TowerID(k, preshower) ) ;
-//     }
-//   }
-// }
-
 //____________________________________________________________________________ 
-void AliEMCALSDigitizer::UseHitsFrom(const char * filename)
+void AliEMCALSDigitizer::Unload() const
 {
-  SetTitle(filename) ; 
-  Init() ; 
+  // Unload Hits and SDigits from the folder
+  AliEMCALGetter * gime = AliEMCALGetter::Instance() ; 
+  AliEMCALLoader * loader = gime->EmcalLoader() ; 
+  loader->UnloadHits() ; 
+  loader->UnloadSDigits() ; 
 }