]> git.uio.no Git - u/mrichter/AliRoot.git/blobdiff - EMCAL/AliEMCALSDigitizer.cxx
Adding includes now needed by ROOT
[u/mrichter/AliRoot.git] / EMCAL / AliEMCALSDigitizer.cxx
index 2d7fb96f0a5bcedf2c929515784263faed33419d..4b7d7756b4fa6d8de3e5c54a6d315a3b815cf2cd 100644 (file)
 //////////////////////////////////////////////////////////////////////////////
 
 // --- ROOT system ---
-#include "TBenchmark.h"
-#include "TH1.h"
-#include "TBrowser.h"
-#include "Riostream.h"
-//#include "TObjectTable.h"
+#include <TBenchmark.h>
+#include <TH1.h>
+#include <TBrowser.h>
+#include <Riostream.h>
+#include <TMath.h>
+#include <TROOT.h>
 
 // --- Standard library ---
 #include "stdlib.h"
 ClassImp(AliEMCALSDigitizer)
            
 //____________________________________________________________________________ 
-  AliEMCALSDigitizer::AliEMCALSDigitizer():TTask("","") 
+AliEMCALSDigitizer::AliEMCALSDigitizer()
+  : TTask("",""),
+    fA(0.),fB(0.),fECPrimThreshold(0.),
+    fDefaultInit(kTRUE),
+    fEventFolderName(0),
+    fInit(0),
+    fSDigitsInRun(0),
+    fFirstEvent(0),
+    fLastEvent(0),
+    fSampling(0.),
+    fControlHists(0),
+    fHists(0)
 {
   // ctor
-  fFirstEvent = fLastEvent  = fControlHists = 0;  
-  fDefaultInit = kTRUE ; 
-  fHists = 0;
+  InitParameters();
 }
 
 //____________________________________________________________________________ 
 AliEMCALSDigitizer::AliEMCALSDigitizer(const char * alirunFileName, 
-                                      const char * eventFolderName):
-  TTask("EMCAL"+AliConfig::Instance()->GetSDigitizerTaskName(), alirunFileName),
-  fEventFolderName(eventFolderName)
+                                      const char * eventFolderName)
+  : TTask("EMCAL"+AliConfig::Instance()->GetSDigitizerTaskName(), alirunFileName),
+    fA(0.),fB(0.),fECPrimThreshold(0.),
+    fDefaultInit(kFALSE),
+    fEventFolderName(eventFolderName),
+    fInit(0),
+    fSDigitsInRun(0),
+    fFirstEvent(0),
+    fLastEvent(0),
+    fSampling(0.),
+    fControlHists(1),
+    fHists(0)
 {
   // ctor
-  fFirstEvent = fLastEvent  = fControlHists = 0 ; // runs one event by defaut  
   Init();
   InitParameters() ; 
-  fDefaultInit = kFALSE ; 
-  fHists = 0;
+  if(fControlHists) BookControlHists(1);
 }
 
 
 //____________________________________________________________________________ 
-AliEMCALSDigitizer::AliEMCALSDigitizer(const AliEMCALSDigitizer & sd) : TTask(sd) {
+AliEMCALSDigitizer::AliEMCALSDigitizer(const AliEMCALSDigitizer & sd) 
+  : TTask(sd.GetName(),sd.GetTitle()),
+    fA(sd.fA),
+    fB(sd.fB),
+    fECPrimThreshold(sd.fECPrimThreshold),
+    fDefaultInit(sd.fDefaultInit),
+    fEventFolderName(sd.fEventFolderName),
+    fInit(sd.fInit),
+    fSDigitsInRun(sd.fSDigitsInRun),
+    fFirstEvent(sd.fFirstEvent),
+    fLastEvent(sd.fLastEvent),
+    fSampling(sd.fSampling),
+    fControlHists(sd.fControlHists),
+    fHists(sd.fHists)
+{
   //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
   AliLoader *emcalLoader=0;
   if ((emcalLoader = AliRunLoader::GetRunLoader()->GetDetectorLoader("EMCAL")))
       emcalLoader->CleanSDigitizer();
@@ -127,7 +149,7 @@ void AliEMCALSDigitizer::Init(){
   //============================================================= YS
   //  The initialisation is now done by the getter
 
-  fInit = kTRUE 
+  fInit = kTRUE; 
    
   AliEMCALLoader *emcalLoader = dynamic_cast<AliEMCALLoader*>(AliRunLoader::GetRunLoader()->GetDetectorLoader("EMCAL"));
 
@@ -144,6 +166,8 @@ void AliEMCALSDigitizer::Init(){
 //____________________________________________________________________________ 
 void AliEMCALSDigitizer::InitParameters()
 {
+  //initialize parameters for sdigitization
+
   const AliEMCALGeometry * geom = AliEMCALGeometry::GetInstance();
   if (geom->GetSampling() == 0.) {
     Fatal("InitParameters", "Sampling factor not set !") ; 
@@ -155,6 +179,20 @@ void AliEMCALSDigitizer::InitParameters()
 
  // threshold for deposit energy of hit
   fECPrimThreshold  = 0.; // 24-nov-04 - was 1.e-6;
+
+  AliDebug(2,Form("Print: \n------------------- %s -------------\n",GetName()));
+  AliDebug(2,Form("   fInit                                 %i\n", int(fInit)));
+  AliDebug(2,Form("   fFirstEvent                           %i\n", fFirstEvent));
+  AliDebug(2,Form("   fLastEvent                            %i\n", fLastEvent));
+  AliDebug(2,Form("   Writing SDigits to branch with title  %s\n", fEventFolderName.Data()));
+  AliDebug(2,Form("   with digitization parameters       A = %f\n", fA));
+  AliDebug(2,Form("                                      B = %f\n", fB));
+  AliDebug(2,Form("   Threshold for EC Primary assignment  = %f\n", fECPrimThreshold));
+  AliDebug(2,Form("   Sampling                             = %f\n", fSampling));
+  AliDebug(2,Form("---------------------------------------------------\n"));
+
+  //  Print("");
+
 }
 
 //____________________________________________________________________________
@@ -163,7 +201,19 @@ void AliEMCALSDigitizer::Exec(Option_t *option)
   // Collects all hit of the same tower into digits
   TString o(option); o.ToUpper();
   if (strstr(option, "print") ) {
-    Print() ; 
+
+    AliDebug(2,Form("Print: \n------------------- %s -------------\n",GetName()));
+    AliDebug(2,Form("   fInit                                 %i\n", int(fInit)));
+    AliDebug(2,Form("   fFirstEvent                           %i\n", fFirstEvent));
+    AliDebug(2,Form("   fLastEvent                            %i\n", fLastEvent));
+    AliDebug(2,Form("   Writing SDigits to branch with title  %s\n", fEventFolderName.Data()));
+    AliDebug(2,Form("   with digitization parameters       A = %f\n", fA));
+    AliDebug(2,Form("                                      B = %f\n", fB));
+    AliDebug(2,Form("   Threshold for EC Primary assignment  = %f\n", fECPrimThreshold));
+    AliDebug(2,Form("   Sampling                             = %f\n", fSampling));
+    AliDebug(2,Form("---------------------------------------------------\n"));
+
+    //    Print();
     return ; 
   }
   
@@ -201,73 +251,62 @@ void AliEMCALSDigitizer::Exec(Option_t *option)
     TClonesArray * hits = emcalLoader->Hits() ; 
     TClonesArray * sdigits = emcalLoader->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>((emcalLoader->TreeH())->GetEntries()) ; 
-    // This is not true: there is only one list of hits (MvL jan 2006)
-    // 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 ; iprim < nPrim ; iprim++ ) { 
-      //=========== Get the EMCAL branch from Hits Tree for the Primary iprim
-      rl->Stack()->Particle(iprim) ;
-      Int_t i;
-      AliEMCALGeometry *geom = AliEMCALGeometry::GetInstance(); 
-      for ( i = 0 ; i < hits->GetEntries() ; i++ ) {
-       AliEMCALHit * hit = dynamic_cast<AliEMCALHit*>(hits->At(i)) ;
-       AliEMCALDigit * curSDigit = 0 ;
-       AliEMCALDigit * sdigit = 0 ;
-       Bool_t newsdigit = kTRUE; 
-
-        // hit->GetId() - Absolute Id number EMCAL segment
-       if(geom->CheckAbsCellId(hit->GetId())) { // was IsInECA(hit->GetId())
-          energy = hit->GetEnergy() * fSampling; // 23-nov-04
-         if(energy >  fECPrimThreshold )
-       // Assign primary number only if deposited energy is significant
-           curSDigit =  new AliEMCALDigit( hit->GetPrimary(),
-                                           hit->GetIparent(), hit->GetId(), 
-                                           Digitize(energy), hit->GetTime() ) ;
+    Int_t i;
+    AliEMCALGeometry *geom = AliEMCALGeometry::GetInstance(); 
+    for ( i = 0 ; i < hits->GetEntries() ; i++ ) {
+      AliEMCALHit * hit = dynamic_cast<AliEMCALHit*>(hits->At(i)) ;
+      AliEMCALDigit * curSDigit = 0 ;
+      AliEMCALDigit * sdigit = 0 ;
+      Bool_t newsdigit = kTRUE; 
+      
+      // hit->GetId() - Absolute Id number EMCAL segment
+      if(geom->CheckAbsCellId(hit->GetId())) { // was IsInECA(hit->GetId())
+       energy = hit->GetEnergy() * fSampling; // 23-nov-04
+       if(energy >  fECPrimThreshold )
+         // Assign primary number only if deposited energy is significant
+         curSDigit =  new AliEMCALDigit( hit->GetPrimary(),
+                                         hit->GetIparent(), hit->GetId(), 
+                                         Digitize(energy), hit->GetTime() ) ;
          else
            curSDigit =  new AliEMCALDigit( -1               , 
                                            -1               ,
                                            hit->GetId(), 
                                            Digitize(energy), hit->GetTime() ) ;
-       } else {
-          Warning("Exec"," abs id %i is bad \n", hit->GetId());
-          newsdigit = kFALSE;
-         curSDigit = 0;
-        }
-
-       if(curSDigit != 0){
-         for(Int_t check= 0; check < nSdigits ; check++) {
-           sdigit = dynamic_cast<AliEMCALDigit *>(sdigits->At(check)) ;
-           
-           if( sdigit->GetId() == curSDigit->GetId()) { // Are we in the same ECAL tower ?              
-             *sdigit = *sdigit + *curSDigit;
-             newsdigit = kFALSE;
-           }
+      } else {
+       Warning("Exec"," abs id %i is bad \n", hit->GetId());
+       newsdigit = kFALSE;
+       curSDigit = 0;
+      }
+      
+      if(curSDigit != 0){
+       for(Int_t check= 0; check < nSdigits ; check++) {
+         sdigit = dynamic_cast<AliEMCALDigit *>(sdigits->At(check)) ;
+         
+         if( sdigit->GetId() == curSDigit->GetId()) { // Are we in the same ECAL tower ?              
+           *sdigit = *sdigit + *curSDigit;
+           newsdigit = kFALSE;
          }
        }
-       if (newsdigit) {
-         new((*sdigits)[nSdigits])  AliEMCALDigit(*curSDigit);
-         nSdigits++ ;  
-       }
-       delete curSDigit ; 
-      }  // loop over all hits (hit = deposited energy/entering particle)
-    } // loop over iprim
+      }
+      if (newsdigit) {
+       new((*sdigits)[nSdigits])  AliEMCALDigit(*curSDigit);
+       nSdigits++ ;  
+      }
+      delete curSDigit ; 
+    }  // loop over all hits (hit = deposited energy/entering particle)
     sdigits->Sort() ;
     
     nSdigits = sdigits->GetEntriesFast() ;
     fSDigitsInRun += nSdigits ;  
-
-    Int_t nPrimarymax = -1 ; 
-    Int_t i ;
+    
     Double_t e=0.,esum=0.;
     AliEMCALHistoUtilities::FillH1(fHists, 0, double(sdigits->GetEntriesFast()));
     for (i = 0 ; i < sdigits->GetEntriesFast() ; i++) { 
       AliEMCALDigit * sdigit = dynamic_cast<AliEMCALDigit *>(sdigits->At(i)) ;
       sdigit->SetIndexInList(i) ;
-
+      
       AliEMCALHistoUtilities::FillH1(fHists, 2, double(sdigit->GetAmp()));
       e = double(Calibrate(sdigit->GetAmp()));
       esum += e;
@@ -276,20 +315,15 @@ void AliEMCALSDigitizer::Exec(Option_t *option)
     }
     if(esum>0.) AliEMCALHistoUtilities::FillH1(fHists, 1, esum);
     
-    for (i = 0 ; i < sdigits->GetEntriesFast() ; i++) { // for what 
-      if (((dynamic_cast<AliEMCALDigit *>(sdigits->At(i)))->GetNprimary()) > nPrimarymax)
-       nPrimarymax = ((dynamic_cast<AliEMCALDigit *>(sdigits->At(i)))->GetNprimary()) ;
-    }
-
     // Now write SDigits    
-
+    
     Int_t bufferSize = 32000 ;    
     TBranch * sdigitsBranch = treeS->GetBranch("EMCAL");
     if (sdigitsBranch)
       sdigitsBranch->SetAddress(&sdigits);
     else
       treeS->Branch("EMCAL",&sdigits,bufferSize);
+    
     treeS->Fill();
     
     emcalLoader->WriteSDigits("OVERWRITE");
@@ -300,7 +334,7 @@ void AliEMCALSDigitizer::Exec(Option_t *option)
     if(strstr(option,"deb"))
       PrintSDigits(option) ;  
   }
-
+  
   Unload();
   
   emcalLoader->GetSDigitsDataLoader()->GetBaseTaskLoader()->SetDoNotReload(kTRUE);
@@ -308,10 +342,26 @@ void AliEMCALSDigitizer::Exec(Option_t *option)
   if(strstr(option,"tim")){
     gBenchmark->Stop("EMCALSDigitizer"); 
     printf("\n Exec: took %f seconds for SDigitizing %f seconds per event\n", 
-        gBenchmark->GetCpuTime("EMCALSDigitizer"), gBenchmark->GetCpuTime("EMCALSDigitizer")/nEvents ) ; 
+          gBenchmark->GetCpuTime("EMCALSDigitizer"), gBenchmark->GetCpuTime("EMCALSDigitizer")/nEvents ) ; 
   }
 }
 
+//__________________________________________________________________
+
+Int_t AliEMCALSDigitizer::Digitize(Float_t energy)const {
+  // Digitize the energy
+    Double_t aSignal = fA + energy*fB;
+    if (TMath::Abs(aSignal)>2147483647.0) { 
+      //PH 2147483647 is the max. integer
+      //PH This apparently is a problem which needs investigation
+      AliWarning(Form("Too big or too small energy %f",aSignal));
+      aSignal = TMath::Sign((Double_t)2147483647,aSignal);
+    }
+    return (Int_t ) aSignal;
+  }
+
+//__________________________________________________________________
 
 void AliEMCALSDigitizer::Print1(Option_t * option)
 {
@@ -323,7 +373,7 @@ void AliEMCALSDigitizer::Print1(Option_t * option)
 void AliEMCALSDigitizer::Print(Option_t *option) const
 { 
   // Prints parameters of SDigitizer
-  printf("Print: \n------------------- %s -------------\n", GetName() ) ; 
+  printf("Print: \n------------------- %s ------------- option %s\n", GetName() , option) ; 
   printf("   fInit                                 %i\n", int(fInit));
   printf("   fFirstEvent                           %i\n", fFirstEvent);
   printf("   fLastEvent                            %i\n", fLastEvent);
@@ -401,10 +451,13 @@ void AliEMCALSDigitizer::Browse(TBrowser* b)
 }
 
 TList *AliEMCALSDigitizer::BookControlHists(int var)
-{ // 22-nov-04
+{ 
+  //book histograms for monitoring sdigitization
+  // 22-nov-04
   gROOT->cd();
   const AliEMCALGeometry *geom = AliEMCALGeometry::GetInstance() ;
   if(var>=1){
+    AliDebug(1, " BookControlHists() in action ");
     new TH1F("HSDigiN",  "#EMCAL  sdigits ", 1001, -0.5, 1000.5);
     new TH1F("HSDigiSumEnergy","Sum.EMCAL energy", 1000, 0.0, 100.);
     new TH1F("HSDigiAmp",  "EMCAL sdigits amplitude", 1000, 0., 2.e+9);
@@ -412,8 +465,10 @@ TList *AliEMCALSDigitizer::BookControlHists(int var)
     new TH1F("HSDigiAbsId","EMCAL absID for sdigits",
     geom->GetNCells(), 0.5, Double_t(geom->GetNCells())+0.5);
   }
+
   fHists = AliEMCALHistoUtilities::MoveHistsToList("EmcalSDigiControlHists", kFALSE);
-  fHists = 0;
+  //  fHists = 0; ??
+
   return fHists;
 }