]> git.uio.no Git - u/mrichter/AliRoot.git/blobdiff - ZDC/AliZDC.cxx
In stand-allone mode, pass stack to entries.
[u/mrichter/AliRoot.git] / ZDC / AliZDC.cxx
index 0266e746975dc9cb3640a9ded1a86c9050549789..d66c737674c08cfef6d33873a81876de9f1bd13a 100644 (file)
  * provided "as is" without express or implied warranty.                  *
  **************************************************************************/
 
-/*
-$Log$
-Revision 1.32  2002/10/22 15:35:30  alibrary
-Introducing Riostream.h
-
-Revision 1.31  2002/10/14 14:57:44  hristov
-Merging the VirtualMC branch to the main development branch (HEAD)
-
-Revision 1.28.6.2  2002/07/24 10:10:13  alibrary
-Updating VirtualMC
-
-Revision 1.28.6.1  2002/06/10 15:29:36  hristov
-Merged with v3-08-02
-
-Revision 1.30  2002/06/07 10:19:23  coppedis
-TreeS, TreeD and TreeR for ZDC can be written in a separate file
-
-Revision 1.29  2002/06/04 08:17:04  coppedis
-Reconstruction method improved
-
-Revision 1.28  2002/02/04 09:18:08  coppedis
-Merging and reconstruction code review
-
-Revision 1.26.2.2  2001/11/12 18:41:44  hristov
-All the changes from the head are merged to the release
-
-Revision 1.27  2001/10/21 18:27:45  hristov
-Several pointers were set to zero in the default constructors to avoid memory management problems
-
-Revision 1.26  2001/10/04 14:30:28  coppedis
-Event merging for ZDC
-
-Revision 1.25  2001/10/04 14:24:15  coppedis
-Event merging for ZDC
-
-Revision 1.24  2001/09/26 16:03:41  coppedis
-Merging implemented
-
-Revision 1.23  2001/05/15 13:44:57  coppedis
-Changes in AddHit method
-
-Revision 1.22  2001/05/14 09:53:32  coppedis
-Adding functions ZMin and ZMax
-
-Revision 1.21  2001/04/20 10:05:02  coppedis
-Minor changes
-
-Revision 1.20  2001/03/26 13:39:20  coppedis
-Comment prints
-
-Revision 1.19  2001/03/26 09:10:23  coppedis
-Corrected bug in constructor (fIshunt has to be =1)
-
-Revision 1.18  2001/03/20 08:21:55  coppedis
-ZDC needs PIPE, ABSO, DIPO and SHIL
-
-Revision 1.17  2001/03/16 16:18:03  coppedis
-Correction for superposition of ZDC volumes with MUON arm one
-
-Revision 1.16  2001/03/15 16:01:11  coppedis
-Code review
-
-Revision 1.15  2001/01/26 19:56:27  hristov
-Major upgrade of AliRoot code
-
-Revision 1.14  2000/12/12 13:17:01  coppedis
-Minor corrections suggested by P. Hristov
-
-Revision 1.12  2000/12/01 08:19:01  coppedis
-Adding a message error if ZDC is constructed without DIPO
-
-Revision 1.11  2000/11/30 17:21:03  coppedis
-Introduce hit array fStHits reset only at the end of the event (for digitization)
-
-Revision 1.10  2000/11/22 11:32:58  coppedis
-Major code revision
-
-Revision 1.9  2000/10/02 21:28:20  fca
-Removal of useless dependecies via forward declarations
-
-Revision 1.8  2000/07/10 13:58:01  fca
-New version of ZDC from E.Scomparin & C.Oppedisano
-
-Revision 1.7  2000/01/19 17:17:40  fca
-
-Revision 1.6  1999/09/29 09:24:35  fca
-Introduction of the Copyright and cvs Log
-
-*/
+/* $Id$ */
 
 ///////////////////////////////////////////////////////////////////////////////
 //                                                                           //
@@ -117,24 +29,29 @@ Introduction of the Copyright and cvs Log
 
 // --- ROOT system
 #include <TBRIK.h>
-#include <TNode.h>
-#include <TGeometry.h>
-#include <TFile.h>
-#include <TTree.h>
 #include <TDirectory.h>
 #include <TF1.h>
+#include <TFile.h>
+#include <TGeometry.h>
+#include <TNode.h>
+#include <TTree.h>
+#include <TVirtualMC.h>
 
 // --- AliRoot header files
+#include "AliDetector.h"
 #include "AliZDC.h"
+#include "AliZDCDigit.h"
 #include "AliZDCHit.h"
 #include "AliZDCMergedHit.h"
 #include "AliZDCMerger.h"
-#include "AliZDCDigit.h"
 #include "AliZDCReco.h"
-#include "AliDetector.h"
+
 #include "AliConst.h"
-#include "AliRun.h"
+
 #include "AliHeader.h"
+#include "AliLoader.h"
+#include "AliRun.h"
+#include "AliMC.h"
 
  
 ClassImp(AliZDC)
@@ -177,7 +94,7 @@ AliZDC::AliZDC(const char *name, const char *title)
 
   // Allocate the hits array  
   fHits   = new TClonesArray("AliZDCHit",1000);
-  gAlice->AddHitList(fHits);
+  gAlice->GetMCApp()->AddHitList(fHits);
   // Allocate the merged hits array  
   fMergedHits = new TClonesArray("AliZDCMergedHit",1000);
 
@@ -219,23 +136,23 @@ void AliZDC::AddHit(Int_t track, Int_t *vol, Float_t *hits)
   
   if(fNhits==0){
       // First hit -> setting flag for primary or secondary particle
-      Int_t primary = gAlice->GetPrimary(track);     
+      Int_t primary = gAlice->GetMCApp()->GetPrimary(track);     
       if(track != primary){
-        newquad->fSFlag = 1;  // SECONDARY particle entering the ZDC
+        newquad->SetSFlag(1);  // SECONDARY particle entering the ZDC
       }
       else if(track == primary){
-        newquad->fSFlag = 0;  // PRIMARY particle entering the ZDC
+        newquad->SetSFlag(0);  // PRIMARY particle entering the ZDC
       }  
-      sFlag    = newquad->fSFlag;
-      primKinEn = newquad->fPrimKinEn;
-      xImpact  = newquad->fXImpact;
-      yImpact  = newquad->fYImpact;
+      sFlag    = newquad->GetSFlag();
+      primKinEn = newquad->GetPrimKinEn();
+      xImpact  = newquad->GetXImpact();
+      yImpact  = newquad->GetYImpact();
    }
    else{       
-      newquad->fPrimKinEn = primKinEn;
-      newquad->fXImpact        = xImpact;
-      newquad->fYImpact = yImpact;
-      newquad->fSFlag  = sFlag;
+      newquad->SetPrimKinEn(primKinEn);
+      newquad->SetXImpact(xImpact);
+      newquad->SetYImpact(yImpact);
+      newquad->SetSFlag(sFlag);
    }
  
   Int_t j;
@@ -328,38 +245,43 @@ Float_t AliZDC::ZMax(void) const
 
   char branchname[10];
   sprintf(branchname,"%s",GetName());
+
+  const char *cH = strstr(opt,"H");
+  
+  if (cH && fLoader->TreeH())
+   fHits   = new TClonesArray("AliZDCHit",1000); 
   
   AliDetector::MakeBranch(opt);
 
   const char *cS = strstr(opt,"S");
 
-  if (gAlice->TreeS() && cS) {
+  if (fLoader->TreeS() && cS) {
     if(fMergedHits!=0) fMergedHits->Clear();
     else fMergedHits = new TClonesArray ("AliZDCMergedHit",1000);
-    MakeBranchInTree(gAlice->TreeS(), 
+    MakeBranchInTree(fLoader->TreeS(), 
                      branchname, &fMergedHits, fBufferSize, file) ;
-    printf("* AliZDC::MakeBranch    * Making Branch %s for SDigits\n\n",branchname);
+    if (GetDebug()) printf("* AliZDC::MakeBranch    * Making Branch %s for SDigits\n\n",branchname);
   }
 
     
   const char *cD = strstr(opt,"D");
 
-  if (gAlice->TreeD() && cD) {
+  if (fLoader->TreeD() && cD) {
     if(fDigits!=0) fDigits->Clear();
     else fDigits = new TClonesArray ("AliZDCDigit",1000);
-    MakeBranchInTree(gAlice->TreeD(), 
+    MakeBranchInTree(fLoader->TreeD(), 
                      branchname, &fDigits, fBufferSize, file) ;
-    printf("* AliZDC::MakeBranch    * Making Branch %s for Digits\n\n",branchname);
+    if (GetDebug()) printf("* AliZDC::MakeBranch    * Making Branch %s for Digits\n\n",branchname);
   }
 
   
   const char *cR = strstr(opt,"R");
 
-  if (gAlice->TreeR() && cR) {
+  if (fLoader->TreeR() && cR) {
     if(fRecPoints==0) fRecPoints = new TClonesArray("AliZDCReco",1000);
-    MakeBranchInTree(gAlice->TreeR(), 
+    MakeBranchInTree(fLoader->TreeR(), 
                     branchname, &fRecPoints, fBufferSize, file) ;
-    printf("* AliZDC::MakeBranch    * Making Branch %s for RecPoints\n\n",branchname);   }
+    if (GetDebug()) printf("* AliZDC::MakeBranch    * Making Branch %s for RecPoints\n\n",branchname);   }
           
 }
 
@@ -370,8 +292,9 @@ Float_t AliZDC::ZMax(void) const
   const Int_t kBufferSize = 4000;
   char  branchname[20];
   sprintf(branchname,"%s",GetName());
+  if (fMergedHits==0x0) fMergedHits = new TClonesArray("AliZDCMergedHit",1000);
   MakeBranchInTree(treeS, branchname, &fMergedHits, kBufferSize, file) ;
-  printf("* AliZDC::MakeBranch    * Making Branch %s for SDigits\n\n",branchname);
+  if (GetDebug()) printf("* AliZDC::MakeBranch    * Making Branch %s for SDigits\n\n",branchname);
 
 }
 //_____________________________________________________________________________
@@ -381,8 +304,9 @@ Float_t AliZDC::ZMax(void) const
   const Int_t kBufferSize = 4000;
   char  branchname[20];
   sprintf(branchname,"%s",GetName());
+  if (fDigits == 0x0) fDigits = new TClonesArray("AliZDCDigit",1000);
   MakeBranchInTree(treeD, branchname, &fDigits, kBufferSize, file) ;
-  printf("* AliZDC::MakeBranch    * Making Branch %s for Digits\n\n",branchname);
+  if (GetDebug()) printf("* AliZDC::MakeBranch    * Making Branch %s for Digits\n\n",branchname);
 
 }
 //_____________________________________________________________________________
@@ -393,24 +317,33 @@ Float_t AliZDC::ZMax(void) const
   char  branchname[20];
   sprintf(branchname,"%s",GetName());
   MakeBranchInTree(treeR, branchname, &fRecPoints, kBufferSize, file) ;
-  printf("* AliZDC::MakeBranch    * Making Branch %s for RecPoints\n\n",branchname);
+  if (GetDebug()) printf("* AliZDC::MakeBranch    * Making Branch %s for RecPoints\n\n",branchname);
 
 }
 //_____________________________________________________________________________
 void AliZDC::Hits2SDigits()
 {
-  printf("\n   Entering AliZDC::SDigits2Digits() ");
+  if (GetDebug()) printf("\n   Entering AliZDC::SDigits2Digits() ");
   
+  fLoader->LoadHits("read");
+  fLoader->LoadSDigits("recreate");
+  AliRunLoader* runLoader = fLoader->GetRunLoader(); 
+
+  for (Int_t iEvent = 0; iEvent < runLoader->GetNumberOfEvents(); iEvent++) {
+    runLoader->GetEvent(iEvent);
+    if (!fLoader->TreeS()) fLoader->MakeTree("S");
+    MakeBranch("S");
+
   //----------------------------------------------------------------
   if(!fMerger){ 
-    printf("   ZDC digitization (without merging)\n");
+    if (GetDebug()) printf("   ZDC digitization (without merging)\n");
 
-    AliZDCMergedHit *MHit;
+    AliZDCMergedHit *mHit;
     Int_t j, sector[2];
-    Float_t MHits[7];
+    Float_t mHits[7];
     fNMergedhits = 0;
 
-    TTree *treeH = gAlice->TreeH();
+    TTree *treeH = TreeH();
     Int_t ntracks = (Int_t) treeH->GetEntries();
     gAlice->ResetHits();
   
@@ -421,45 +354,50 @@ void AliZDC::Hits2SDigits()
                       zdcHit = (AliZDCHit*)this->NextHit()){ 
                      
           for(j=0; j<2; j++) sector[j] = zdcHit->GetVolume(j);
-          MHits[0] = zdcHit->GetPrimKinEn();
-          MHits[1] = zdcHit->GetXImpact();
-          MHits[2] = zdcHit->GetYImpact();
-          MHits[3] = zdcHit->GetSFlag();
-          MHits[4] = zdcHit->GetLightPMQ();
-          MHits[5] = zdcHit->GetLightPMC();
-          MHits[6] = zdcHit->GetEnergy();
+          mHits[0] = zdcHit->GetPrimKinEn();
+          mHits[1] = zdcHit->GetXImpact();
+          mHits[2] = zdcHit->GetYImpact();
+          mHits[3] = zdcHit->GetSFlag();
+          mHits[4] = zdcHit->GetLightPMQ();
+          mHits[5] = zdcHit->GetLightPMC();
+          mHits[6] = zdcHit->GetEnergy();
        }//Hits loop
        
-         MHit = new AliZDCMergedHit(sector, MHits);
-         new((*fMergedHits)[fNMergedhits]) AliZDCMergedHit(*MHit);       
+         mHit = new AliZDCMergedHit(sector, mHits);
+         new((*fMergedHits)[fNMergedhits]) AliZDCMergedHit(*mHit);       
          TClonesArray &sdigits = *fMergedHits;
-         new (sdigits[fNMergedhits]) AliZDCMergedHit(*MHit);
+         new (sdigits[fNMergedhits]) AliZDCMergedHit(*mHit);
          fNMergedhits++;
-         delete MHit;
+         delete mHit;
     }
-    gAlice->TreeS()->Fill();
-    gAlice->TreeS()->AutoSave(); 
-    gAlice->TreeS()->Reset();  
+      fLoader->TreeS()->Fill();
+      fLoader->TreeS()->AutoSave(); 
+      fLoader->TreeS()->Reset();  
   }
   //----------------------------------------------------------------
   else if(fMerger){
-    printf("   ZDC merging and digitization\n");
+    if (GetDebug()) printf("   ZDC merging and digitization\n");
     // ### Initialise merging
     fMerger -> InitMerging();
 
     // SDigits tree
-    TTree *treeS = 0;
-    char treeSName[20];
-    sprintf(treeSName,"TreeS%d",fMerger->EvNum());
-    if(gAlice->GetTreeSFile()){
-      gAlice->GetTreeSFile()->cd();
-      treeS = (TTree*)gAlice->GetTreeSFile()->Get(treeSName);
-    }
-    else {
-      treeS = gAlice->TreeS();
-    }
+
+
+
+    TTree *treeS = fLoader->TreeS();
+    if (treeS == 0x0)
+     {
+      Int_t retval = fLoader->LoadSDigits();
+      if (retval)
+       {
+         Error("Hits2SDigits","Error while loading S. Digits");
+         return;
+       }
+      treeS = fLoader->TreeS();
+     }
+    
     if(!treeS){
-      printf("\n ERROR -> Can't find TreeS%d in background file\n",fMerger->EvNum());
+      if (GetDebug()) printf("\n ERROR -> Can't find TreeS%d in background file\n",fMerger->EvNum());
     }   
 
     // ### Get TCA of MergedHits from AliZDCMerger
@@ -474,50 +412,61 @@ void AliZDC::Hits2SDigits()
       if(branchSD) branchSD->SetAddress(&fMergedHits);
       else if(!branchSD) MakeBranchInTreeS(treeS);
     }
-    AliZDCMergedHit *MHit;
+    AliZDCMergedHit *mHit;
     TClonesArray &sdigits = *fMergedHits;
     Int_t imhit;
     //Merged Hits loop
     for(imhit=0; imhit<fNMergedhits; imhit++){
-       MHit = (AliZDCMergedHit*) fMergedHits->UncheckedAt(imhit);
-       new (sdigits[imhit]) AliZDCMergedHit(*MHit);
+       mHit = (AliZDCMergedHit*) fMergedHits->UncheckedAt(imhit);
+       new (sdigits[imhit]) AliZDCMergedHit(*mHit);
     }
     treeS->Fill();
     treeS->AutoSave();
   }
   
+  }
+
+  fLoader->UnloadHits();
+  fLoader->UnloadSDigits();
 }
 
 //_____________________________________________________________________________
 void AliZDC::SDigits2Digits()
 {
   if(!fMerger){ // Only digitization
-    printf("   ZDC digitization (no merging) \n");
+    if (GetDebug()) printf("   ZDC digitization (no merging) \n");
     fMerger = new AliZDCMerger();    
     fMerger->Digitize(fNMergedhits, fMergedHits);
 
     char hname[30];
-    sprintf(hname,"TreeD%d",gAlice->GetHeader()->GetEvent());
-    gAlice->TreeD()->Fill();
-    gAlice->TreeD()->AutoSave();
-    gAlice->TreeD()->Reset();  
+    AliRunLoader * rl = fLoader->GetRunLoader();
+    sprintf(hname,"TreeD%d",rl->GetHeader()->GetEvent());
+    fLoader->TreeD()->Fill();
+    fLoader->TreeD()->AutoSave();
+    fLoader->TreeD()->Reset();  
   }
   else if(fMerger){    // Merging and digitization
-    printf("   ZDC merging and digitization\n");
+    if (GetDebug()) printf("   ZDC merging and digitization\n");
     fMerger->Digitize(fNMergedhits, fMergedHits);
 
     // Digits tree
-    TTree *treeD = 0;
-    char treeDName[20];
-    sprintf(treeDName,"TreeD%d",fMerger->EvNum());  
-    if(gAlice->GetTreeDFile()){
-      treeD = (TTree*)gAlice->GetTreeDFile()->Get(treeDName);
-    }
-    else {
-      treeD = gAlice->TreeD();
-    }
+
+    TTree *treeD = fLoader->TreeD();
+    if (treeD == 0x0)
+     {
+      Int_t retval = fLoader->LoadDigits();
+      if (retval)
+       {
+         Error("SDigits2Digits","Error while loading Digits");
+         return;
+       }
+      treeD = fLoader->TreeD();
+     }
+
+
+
     if(!treeD){
-      printf("\n ERROR -> Can't find TreeD%d in background file\n",fMerger->EvNum());
+      if (GetDebug()) printf("\n ERROR -> Can't find TreeD%d in background file\n",fMerger->EvNum());
     }   
     // Branch address
     char branchDname[20];
@@ -543,66 +492,70 @@ void AliZDC::Hits2Digits()
 //_____________________________________________________________________________
 void AliZDC::Digits2Reco()
 {
-    printf("   Entering AliZDC::Digits2Reco\n");
-    AliDetector *ZDC  = gAlice->GetDetector("ZDC");
-    TClonesArray *ZDCdigits = ZDC->Digits();
+    if (GetDebug()) printf("   Entering AliZDC::Digits2Reco\n");
+    AliDetector *zdcd  = gAlice->GetDetector("ZDC");
+    TClonesArray *zdcdigits = zdcd->Digits();
     
-    TTree *TD = 0;
-    char treeDame[20];
-    sprintf(treeDame,"TreeD%d",fMerger->EvNum());
-    if(gAlice->GetTreeDFile()){
-      TD = (TTree*)gAlice->GetTreeDFile()->Get(treeDame);
-    }
-    else {
-      TD = gAlice->TreeD();
-    }
-    if(TD){
+    TTree *td = fLoader->TreeD();
+    if (td == 0x0)
+     {
+      Int_t retval = fLoader->LoadDigits();
+      if (retval)
+       {
+         Error("Digits2Reco","Error while loading Digits");
+         return;
+       }
+      td = fLoader->TreeD();
+     }
+
+
+    if(td){
       char brname[20];
-      sprintf(brname,"%s",ZDC->GetName());
-      TBranch *br = TD->GetBranch(brname);
-      if(br) br->SetAddress(&ZDCdigits);
+      sprintf(brname,"%s",zdcd->GetName());
+      TBranch *br = td->GetBranch(brname);
+      if(br) br->SetAddress(&zdcdigits);
     }
-    else if(!TD) printf("      ERROR -> TreeD NOT found in gAlice object\n");
+    else if(!td) printf("      ERROR -> TreeD NOT found in gAlice object\n");
     
-    Int_t nt = (Int_t) (TD->GetEntries());
+    Int_t nt = (Int_t) (td->GetEntries());
     gAlice->ResetDigits();    
     
     AliZDCDigit *dig;
-    Int_t j, idig, ndigits, ZNraw=0, ZPraw=0, ZEMraw=0;
+    Int_t j, idig, ndigits, znraw=0, zpraw=0, zemraw=0;
     // ---      Summing raw ADCs for each detector to obtain total light
     for(j=0; j<nt; j++){
-      TD->GetEvent(j);
-      ndigits = ZDCdigits->GetEntries();
-      ZNraw=0;
-      ZPraw=0; 
-      ZEMraw=0;
+      td->GetEvent(j);
+      ndigits = zdcdigits->GetEntries();
+      znraw=0;
+      zpraw=0; 
+      zemraw=0;
       //  ---  Loop over event digits
       for(idig=0; idig<ndigits; idig++){
-         dig = (AliZDCDigit*) ZDCdigits->UncheckedAt(idig);
-         if(dig->GetSector(0) == 1)     ZNraw  += dig->GetADCValue();
-         else if(dig->GetSector(0) == 2) ZPraw  += dig->GetADCValue();
-         else if(dig->GetSector(0) == 3) ZEMraw += dig->GetADCValue();
+         dig = (AliZDCDigit*) zdcdigits->UncheckedAt(idig);
+         if(dig->GetSector(0) == 1)     znraw  += dig->GetADCValue();
+         else if(dig->GetSector(0) == 2) zpraw  += dig->GetADCValue();
+         else if(dig->GetSector(0) == 3) zemraw += dig->GetADCValue();
       } // Digits loop
     } //  TreeD entries loop
-    printf("\n ---     ZNraw = %d, ZPraw = %d, ZEMraw = %d\n",ZNraw, ZPraw, ZEMraw);
+    if (GetDebug()) printf("\n ---     znraw = %d, zpraw = %d, zemraw = %d\n",znraw, zpraw, zemraw);
     
   //  ---      Pedestal subtraction
-  Int_t ZNcorr, ZPcorr, ZEMcorr, MeanPed=50;
-  ZNcorr  = ZNraw  - 5*MeanPed;
-  ZPcorr  = ZPraw  - 5*MeanPed;
-  ZEMcorr = ZEMraw - 2*MeanPed;
-  if(ZNcorr<0)  ZNcorr=0;
-  if(ZPcorr<0)  ZPcorr=0;
-  if(ZEMcorr<0) ZEMcorr=0;
printf("\n    ZNcorr = %d, ZPcorr = %d, ZEMcorr = %d\n",ZNcorr,ZPcorr,ZEMcorr);
+  Int_t zncorr, zpcorr, zemcorr, meanPed=50;
+  zncorr  = znraw  - 5*meanPed;
+  zpcorr  = zpraw  - 5*meanPed;
+  zemcorr = zemraw - 2*meanPed;
+  if(zncorr<0)  zncorr=0;
+  if(zpcorr<0)  zpcorr=0;
+  if(zemcorr<0) zemcorr=0;
if (GetDebug()) printf("\n    zncorr = %d, zpcorr = %d, zemcorr = %d\n",zncorr,zpcorr,zemcorr);
   
   //  ---      ADCchannel -> photoelectrons
   // NB-> PM gain = 10^(5), ADC resolution = 6.4*10^(-7)
-  Float_t ZNphe, ZPphe, ZEMphe, ConvFactor = 0.064;
-  ZNphe  = ZNcorr/ConvFactor;
-  ZPphe  = ZPcorr/ConvFactor;
-  ZEMphe = ZEMcorr/ConvFactor;
-  printf("\n    ZNphe = %f, ZPphe = %f, ZEMphe = %f\n",ZNphe, ZPphe, ZEMphe);
+  Float_t znphe, zpphe, zemphe, convFactor = 0.064;
+  znphe  = zncorr/convFactor;
+  zpphe  = zpcorr/convFactor;
+  zemphe = zemcorr/convFactor;
+  if (GetDebug()) printf("\n    znphe = %f, zpphe = %f, zemphe = %f\n",znphe, zpphe, zemphe);
   
   //  ---      Energy calibration
   // Conversion factors for hadronic ZDCs goes from phe yield to TRUE incident 
@@ -611,26 +564,26 @@ void AliZDC::Digits2Reco()
   // NB -> ZN and ZP conversion factors are constant since incident spectators
   // have all the same energy, ZEM energy is obtained through a fit over the whole
   // range of incident particle energies (obtained with full HIJING simulations) 
-  Float_t ZNenergy, ZPenergy, ZEMenergy, ZDCenergy;
-  Float_t ZNphexTeV=329., ZPphexTeV=369.;
-  ZNenergy  = ZNphe/ZNphexTeV;
-  ZPenergy  = ZPphe/ZPphexTeV;
-  ZDCenergy = ZNenergy+ZPenergy;
-  ZEMenergy = -4.81+0.3238*ZEMphe;
-  if(ZEMenergy<0) ZEMenergy=0;
-  printf("    ZNenergy = %f TeV, ZPenergy = %f TeV, ZDCenergy = %f GeV, "
-         "\n           ZEMenergy = %f TeV\n", ZNenergy, ZPenergy, 
-        ZDCenergy, ZEMenergy);
+  Float_t znenergy, zpenergy, zemenergy, zdcenergy;
+  Float_t znphexTeV=329., zpphexTeV=369.;
+  znenergy  = znphe/znphexTeV;
+  zpenergy  = zpphe/zpphexTeV;
+  zdcenergy = znenergy+zpenergy;
+  zemenergy = -4.81+0.3238*zemphe;
+  if(zemenergy<0) zemenergy=0;
+  if (GetDebug()) printf("    znenergy = %f TeV, zpenergy = %f TeV, zdcenergy = %f GeV, "
+         "\n           zemenergy = %f TeV\n", znenergy, zpenergy, 
+        zdcenergy, zemenergy);
   
-  if(ZDCenergy==0)
-    printf("\n\n       ###     ATTENZIONE!!! -> ev# %d: ZNenergy = %f TeV, ZPenergy = %f TeV, ZDCenergy = %f GeV, "
-         " ZEMenergy = %f TeV\n\n", fMerger->EvNum(), ZNenergy, ZPenergy, ZDCenergy, ZEMenergy); 
+  if(zdcenergy==0)
+    if (GetDebug()) printf("\n\n       ###     ATTENZIONE!!! -> ev# %d: znenergy = %f TeV, zpenergy = %f TeV, zdcenergy = %f GeV, "
+         " zemenergy = %f TeV\n\n", fMerger->EvNum(), znenergy, zpenergy, zdcenergy, zemenergy); 
   
   //  ---      Number of incident spectator nucleons
-  Int_t NDetSpecN, NDetSpecP;
-  NDetSpecN = (Int_t) (ZNenergy/2.760);
-  NDetSpecP = (Int_t) (ZPenergy/2.760);
-  printf("\n    NDetSpecN = %d, NDetSpecP = %d\n",NDetSpecN, NDetSpecP);
+  Int_t nDetSpecN, nDetSpecP;
+  nDetSpecN = (Int_t) (znenergy/2.760);
+  nDetSpecP = (Int_t) (zpenergy/2.760);
+  if (GetDebug()) printf("\n    nDetSpecN = %d, nDetSpecP = %d\n",nDetSpecN, nDetSpecP);
   
   //  ---      Number of generated spectator nucleons and impact parameter
   // --------------------------------------------------------------------------------------------------
@@ -692,76 +645,76 @@ void AliZDC::Digits2Reco()
   TF1 *fZEMsp = new TF1("fZEMsp","208.7-0.09006*x+0.000009526*x*x",0.,4000.);
   TF1 *fZEMb  = new TF1("fZEMb","16.06-0.01633*x+1.44e-5*x*x-6.778e-9*x*x*x+1.438e-12*x*x*x*x-1.112e-16*x*x*x*x*x",0.,4000.);
   
-  Int_t NGenSpecN=0, NGenSpecP=0, NGenSpec=0;
-  Double_t ImpPar=0;
+  Int_t nGenSpecN=0, nGenSpecP=0, nGenSpec=0;
+  Double_t impPar=0;
   // Cut value for Ezem (GeV)
   // [1] ### Results in Chiara's PhD thesis -> 0<b<15 fm (Dec 2001)
-  //Float_t EZEMCut = 360.; 
+  //Float_t eZEMCut = 360.; 
   // [2] ### Results from a new production  -> 0<b<18 fm (Apr 2002)
-  Float_t EZEMCut = 420.;
-  Float_t DeltaEZEMSup = 690.; 
-  Float_t DeltaEZEMInf = 270.; 
-  if(ZEMenergy > (EZEMCut+DeltaEZEMSup)){
-    NGenSpecN = (Int_t) (fZNCen->Eval(ZNenergy));
-    NGenSpecP = (Int_t) (fZPCen->Eval(ZPenergy));
-    NGenSpec  = (Int_t) (fZDCCen->Eval(ZDCenergy));
-    ImpPar    = fbCen->Eval(ZDCenergy);
-    //printf("    fZNCen = %f, fZPCen = %f, fZDCCen = %f\n",fZNCen->Eval(ZNenergy),
-    //            fZPCen->Eval(ZPenergy),fZDCCen->Eval(ZDCenergy));
+  Float_t eZEMCut = 420.;
+  Float_t deltaEZEMSup = 690.; 
+  Float_t deltaEZEMInf = 270.; 
+  if(zemenergy > (eZEMCut+deltaEZEMSup)){
+    nGenSpecN = (Int_t) (fZNCen->Eval(znenergy));
+    nGenSpecP = (Int_t) (fZPCen->Eval(zpenergy));
+    nGenSpec  = (Int_t) (fZDCCen->Eval(zdcenergy));
+    impPar    = fbCen->Eval(zdcenergy);
+    //printf("    fZNCen = %f, fZPCen = %f, fZDCCen = %f\n",fZNCen->Eval(znenergy),
+    //            fZPCen->Eval(zpenergy),fZDCCen->Eval(zdcenergy));
   }
-  else if(ZEMenergy < (EZEMCut-DeltaEZEMInf)){
-    NGenSpecN = (Int_t) (fZNPer->Eval(ZNenergy)); 
-    NGenSpecP = (Int_t) (fZPPer->Eval(ZPenergy));
-    NGenSpec  = (Int_t) (fZDCPer->Eval(ZDCenergy));
-    ImpPar    = fbPer->Eval(ZDCenergy);
-    //printf("    fZNPer = %f, fZPPer = %f, fZDCPer = %f\n",fZNPer->Eval(ZNenergy),
-    //            fZPPer->Eval(ZPenergy),fZDCPer->Eval(ZDCenergy));
+  else if(zemenergy < (eZEMCut-deltaEZEMInf)){
+    nGenSpecN = (Int_t) (fZNPer->Eval(znenergy)); 
+    nGenSpecP = (Int_t) (fZPPer->Eval(zpenergy));
+    nGenSpec  = (Int_t) (fZDCPer->Eval(zdcenergy));
+    impPar    = fbPer->Eval(zdcenergy);
+    //printf("    fZNPer = %f, fZPPer = %f, fZDCPer = %f\n",fZNPer->Eval(znenergy),
+    //            fZPPer->Eval(zpenergy),fZDCPer->Eval(zdcenergy));
   }
-  else if(ZEMenergy >= (EZEMCut-DeltaEZEMInf) && ZEMenergy <= (EZEMCut+DeltaEZEMSup)){
-    NGenSpecN = (Int_t) (fZEMn->Eval(ZEMenergy));
-    NGenSpecP = (Int_t) (fZEMp->Eval(ZEMenergy));
-    NGenSpec  = (Int_t)(fZEMsp->Eval(ZEMenergy));
-    ImpPar    =  fZEMb->Eval(ZEMenergy);
-    //printf("    Nspec ZEM = %f, Nspec ZDC = %f\n",fZEMsp->Eval(ZNenergy),fZDCPer->Eval(ZDCenergy));
+  else if(zemenergy >= (eZEMCut-deltaEZEMInf) && zemenergy <= (eZEMCut+deltaEZEMSup)){
+    nGenSpecN = (Int_t) (fZEMn->Eval(zemenergy));
+    nGenSpecP = (Int_t) (fZEMp->Eval(zemenergy));
+    nGenSpec  = (Int_t)(fZEMsp->Eval(zemenergy));
+    impPar    =  fZEMb->Eval(zemenergy);
+    //printf("    Nspec ZEM = %f, Nspec ZDC = %f\n",fZEMsp->Eval(znenergy),fZDCPer->Eval(zdcenergy));
   }
   // [1] ### Results in Chiara's PhD thesis -> 0<b<15 fm (Dec 2001)
-  /*if(ZNenergy>158.5)  NGenSpecN = (Int_t) (fZEMn->Eval(ZEMenergy));
-  if(ZPenergy>58.91)  NGenSpecP = (Int_t) (fZEMp->Eval(ZEMenergy));
-  if(ZDCenergy>220.4) NGenSpec  = (Int_t)(fZEMsp->Eval(ZEMenergy));
-  if(ZDCenergy>225.)  ImpPar    =          fZEMb->Eval(ZEMenergy);*/
+  /*if(znenergy>158.5)  nGenSpecN = (Int_t) (fZEMn->Eval(zemenergy));
+  if(zpenergy>58.91)  nGenSpecP = (Int_t) (fZEMp->Eval(zemenergy));
+  if(zdcenergy>220.4) nGenSpec  = (Int_t)(fZEMsp->Eval(zemenergy));
+  if(zdcenergy>225.)  impPar    =          fZEMb->Eval(zemenergy);*/
   // [2] ### Results from a new production  -> 0<b<18 fm (Apr 2002)
-  if(ZNenergy>162.)  NGenSpecN = (Int_t) (fZEMn->Eval(ZEMenergy));
-  if(ZPenergy>59.75)  NGenSpecP = (Int_t) (fZEMp->Eval(ZEMenergy));
-  if(ZDCenergy>221.5) NGenSpec  = (Int_t)(fZEMsp->Eval(ZEMenergy));
-  if(ZDCenergy>220.)  ImpPar    =  fZEMb->Eval(ZEMenergy);
+  if(znenergy>162.)  nGenSpecN = (Int_t) (fZEMn->Eval(zemenergy));
+  if(zpenergy>59.75)  nGenSpecP = (Int_t) (fZEMp->Eval(zemenergy));
+  if(zdcenergy>221.5) nGenSpec  = (Int_t)(fZEMsp->Eval(zemenergy));
+  if(zdcenergy>220.)  impPar    =  fZEMb->Eval(zemenergy);
   
-  if(NGenSpecN>125)    NGenSpecN=125;
-  else if(NGenSpecN<0) NGenSpecN=0;
-  if(NGenSpecP>82)     NGenSpecP=82;
-  else if(NGenSpecP<0) NGenSpecP=0;
-  if(NGenSpec>207)     NGenSpec=207;
-  else if(NGenSpec<0)  NGenSpec=0;
-  //printf("    NRecSpecN = %d, NRecSpecP = %d, NRecSpec = %d\n",NGenSpecN,NGenSpecP,NGenSpec);
+  if(nGenSpecN>125)    nGenSpecN=125;
+  else if(nGenSpecN<0) nGenSpecN=0;
+  if(nGenSpecP>82)     nGenSpecP=82;
+  else if(nGenSpecP<0) nGenSpecP=0;
+  if(nGenSpec>207)     nGenSpec=207;
+  else if(nGenSpec<0)  nGenSpec=0;
+  //printf("    NRecSpecN = %d, NRecSpecP = %d, NRecSpec = %d\n",nGenSpecN,nGenSpecP,nGenSpec);
   
   //  ---      Number of participants
-  Int_t NPart, NPartTot;
-  NPart = 207-NGenSpecN-NGenSpecP;
-  NPartTot = 207-NGenSpec;
-  //printf("   ###     NPart(ZP+ZN) = %d, NPart(ZDC) = %d, b = %f fm\n",NPart,NPartTot,ImpPar);
-  printf("     ###     NPart = %d, b = %f fm\n",NPartTot,ImpPar);
+  Int_t nPart, nPartTot;
+  nPart = 207-nGenSpecN-nGenSpecP;
+  nPartTot = 207-nGenSpec;
+  //printf("   ###     nPart(ZP+ZN) = %d, nPart(ZDC) = %d, b = %f fm\n",nPart,nPartTot,impPar);
+  if (GetDebug()) printf("     ###     nPart = %d, b = %f fm\n",nPartTot,impPar);
   
   //  ---     Writing RecPoints TCA
   // Allocate the RecPoints TCA 
   fRecPoints = new TClonesArray("AliZDCReco",1000);
-  AliZDCReco *reco = new AliZDCReco(ZNenergy,ZPenergy,ZDCenergy,ZEMenergy,
-             NDetSpecN,NDetSpecP,NGenSpecN,NGenSpecP,NGenSpec,NPartTot,ImpPar);
+  AliZDCReco *reco = new AliZDCReco(znenergy,zpenergy,zdcenergy,zemenergy,
+             nDetSpecN,nDetSpecP,nGenSpecN,nGenSpecP,nGenSpec,nPartTot,impPar);
   new((*fRecPoints)[fNRecPoints]) AliZDCReco(*reco);
   //fNRecPoints++;
   //fRecPoints->Dump();
   delete reco;
   
   // TreeR
-  TTree *treeR = gAlice->TreeR();
+  TTree *treeR = fLoader->TreeR();
   if(!treeR) printf("\n ERROR -> Can't find TreeR%d in background file\n",fMerger->EvNum());
   // Branch address
   char branchRname[20];
@@ -776,4 +729,21 @@ void AliZDC::Digits2Reco()
   treeR->Reset();
 }
 
+//______________________________________________________________________
+void AliZDC::SetTreeAddress(){
+  // Set branch address for the Trees.
+  // Inputs:
+  //      none.
+  // Outputs:
+  //      none.
+  // Return:
+  //      none.
+  if (fLoader->TreeH() && (fHits == 0x0))
+    fHits   = new TClonesArray("AliZDCHit",1000);
+      
+  if (fLoader->TreeD() && (fDigits == 0x0))
+    fDigits = new TClonesArray("AliZDCDigit",1000);
+      
+  AliDetector::SetTreeAddress();
+}