]> git.uio.no Git - u/mrichter/AliRoot.git/blobdiff - ZDC/AliZDC.cxx
Several pointers were set to zero in the default constructors to avoid memory managem...
[u/mrichter/AliRoot.git] / ZDC / AliZDC.cxx
index 80af78be161ca4fbdb5576bed7dc02e57c82fc51..094103e95d12871327ccc346df7dc20b97dbff29 100644 (file)
 
 /*
 $Log$
+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
 
@@ -39,29 +78,35 @@ Introduction of the Copyright and cvs Log
 
 ///////////////////////////////////////////////////////////////////////////////
 //                                                                           //
-//  Zero Degree Calorimeter                                                  //
-//  This class contains the basic functions for the ZDCs                     //
-//  Functions specific to one particular geometry are                       //
-//  contained in the derived classes                                         //
+//                     Zero Degree Calorimeter                              //
+//          This class contains the basic functions for the ZDCs;           //
+//            functions specific to one particular geometry are              //
+//                      contained in the derived classes                     //
 //                                                                           //
 ///////////////////////////////////////////////////////////////////////////////
 
 #include <stdlib.h>
+#include <iostream.h>
 
 // --- ROOT system
 #include <TBRIK.h>
 #include <TNode.h>
-#include "TGeometry.h"
-#include "TTree.h"
+#include <TGeometry.h>
+#include <TFile.h>
+#include <TTree.h>
 
 // --- AliRoot header files
 #include "AliZDC.h"
 #include "AliZDCHit.h"
-#include "AliRun.h"
+#include "AliZDCMergedHit.h"
+#include "AliZDCDigit.h"
+#include "AliZDCMerger.h"
 #include "AliDetector.h"
 #include "AliCallf77.h"
 #include "AliConst.h"
 #include "AliMC.h"
+#include "AliRun.h"
+#include "AliHeader.h"
 
  
 ClassImp(AliZDC)
@@ -73,14 +118,23 @@ AliZDC::AliZDC()
   // Default constructor for the Zero Degree Calorimeter base class
   //
   
-  fIshunt = 1;
+  fIshunt   = 1;
+  fNoShower = 0;
+  fMerger   = 0;
 
-  fNhits = 0;
-  fNStHits = 0;
-  
-  fStHits = new TClonesArray("AliZDCHit",1000);
+  fHits     = 0;
+  fNhits    = 0;
+
+  fDigits   = 0;
+  fNdigits  = 0;
+
+  fMergedHits = 0;
+  fTreeSD = 0;
+  fTreeMD = 0;
+
+  fNRecPoints = 0;
+  fRecPoints = 0;
   
-  fNPrimaryHits = 0;
 }
  
 //_____________________________________________________________________________
@@ -91,82 +145,24 @@ AliZDC::AliZDC(const char *name, const char *title)
   // Standard constructor for the Zero Degree Calorimeter base class
   //
 
-  // Check that DIPO is there (otherwise tracking is wrong!!!)
-  
-  AliModule* DIPO=gAlice->GetModule("DIPO");
-  if(!DIPO) {
-    Error("Constructor","ZDC needs DIPO!!!\n");
-    exit(1);
-  } 
-
-  //
-  // Allocate the array of hits
+  fIshunt   = 1;
+  fNoShower = 0;
+  fMerger   = 0;
 
+  // Allocate the hits array  
   fHits   = new TClonesArray("AliZDCHit",1000);
   gAlice->AddHitList(fHits);
-  
-  fStHits = new TClonesArray("AliZDCHit",1000);
+  // Allocate the merged hits array  
+  fMergedHits = new TClonesArray("AliZDCMergedHit",1000);
 
-  fNStHits = 0;
-
-  fNPrimaryHits = 0;
-  
-  fIshunt =  1;
+  // Allocate the digits array  
+  fDigits = new TClonesArray("AliZDCDigit",1000);
   
-  fDimZN[0] = 3.52;
-  fDimZN[1] = 3.52;
-  fDimZN[2] = 50.;
-  fDimZP[0] = 11.2;
-  fDimZP[1] = 6.;
-  fDimZP[2] = 75.;
-  fPosZN[0] = 0.;
-  fPosZN[1] = 0.;
-  fPosZN[2] = 11650.;
-  fPosZP[0] = -23.;
-  fPosZP[1] = 0.;
-  fPosZP[2] = 11600.;
-  fFibZN[0] = 0.;
-  fFibZN[1] = 0.01825;
-  fFibZN[2] = 50.;
-  fFibZP[0] = 0.;
-  fFibZP[1] = 0.0275;
-  fFibZP[2] = 75.;
-  fGrvZN[0] = 0.03;
-  fGrvZN[1] = 0.03;
-  fGrvZN[2] = 50.;
-  fGrvZP[0] = 0.04;
-  fGrvZP[1] = 0.04;
-  fGrvZP[2] = 75.;
-  fDivZN[0] = 11;
-  fDivZN[1] = 11;
-  fDivZN[2] = 0;
-  fDivZP[0] = 7;
-  fDivZP[1] = 15;
-  fDivZP[2] = 0;
-  fTowZN[0] = 2;
-  fTowZN[1] = 2;
-  fTowZP[0] = 4;
-  fTowZP[1] = 1;
-  
-  // EM Calorimeter
-  fDimZEMPb  = 0.15*(TMath::Sqrt(2.));
-  fDimZEMAir = 0.001;
-  fFibRadZEM = 0.0315;
-  fDivZEM[0] = 92;
-  fDivZEM[1] = 0;
-  fDivZEM[2] = 20;
-  fDimZEM[0] = 2*fDivZEM[2]*(fDimZEMPb+fDimZEMAir+fFibRadZEM*(TMath::Sqrt(2.)));
-  fDimZEM[1] = 3.5;
-  fDimZEM[2] = 3.5;
-  fDimZEM[3] = 45.;
-  fDimZEM[4] = 0.;
-  fDimZEM[5] = 0.;
-  fFibZEM[0] = 0.;
-  fFibZEM[1] = 0.0275;
-  fFibZEM[2] = fDimZEM[2]/TMath::Sin(fDimZEM[3]*kDegrad)-fFibRadZEM;
-  fPosZEM[0] = 0.;
-  fPosZEM[1] = 5.8;
-  fPosZEM[2] = 11600.;
+  fTreeSD = 0;
+  fTreeMD = 0;
+
+  fNRecPoints = 0;
+  fRecPoints = 0;
 
 }
 //____________________________________________________________________________ 
@@ -177,13 +173,19 @@ AliZDC::~AliZDC()
   //
 
   fIshunt   = 0;
-//  delete fHits;
-//  if(fStHits){
-//    fStHits->Delete();
-//    delete fStHits;
-//    fNStHits = 0;
+  
+  if(fMerger) delete fMerger;
+
+//  if(fHits){
+//    fHits->Delete();
+//    delete fHits;
 //  }
-//  delete fDigits;
+
+//  if(fDigits){
+//    fDigits->Delete();
+//    delete fDigits;
+//  }
+
 }
 //_____________________________________________________________________________
 void AliZDC::AddHit(Int_t track, Int_t *vol, Float_t *hits)
@@ -198,86 +200,78 @@ void AliZDC::AddHit(Int_t track, Int_t *vol, Float_t *hits)
   
   static Float_t primKinEn, xImpact, yImpact, sFlag;
 
-  TClonesArray &lsthits = *fStHits;
-
-
-  AliZDCHit *newquad, *curevquad, *curprimquad;
+  AliZDCHit *newquad, *curprimquad;
   newquad = new AliZDCHit(fIshunt, track, vol, hits);
-
   TClonesArray &lhits = *fHits;
-   
-  Int_t i,j,kStHit = 1;
-  for(i=0; i<fNStHits; i++){
-    // If the hits are equal (same track, same volume), sum them.
-     curevquad = (AliZDCHit*) lsthits[i];
-     kStHit = 1;
-     if(*curevquad == *newquad){
-        *curevquad = *curevquad+*newquad;
-        kStHit = 0;
-     } 
-  }
-
-  for(j=0; j<fNhits; j++){
-    // If the hits are equal (same track, same volume), sum them.
-     curprimquad = (AliZDCHit*) lhits[j];
-     if(*curprimquad == *newquad){
-        *curprimquad = *curprimquad+*newquad;
-       delete newquad;
-       return;
-     } 
-  }
   
   if(fNhits==0){
       // First hit -> setting flag for primary or secondary particle
       Int_t primary = gAlice->GetPrimary(track);     
       if(track != primary){
-        newquad->fSFlag = 1;  // Hit created by secondary particle entering the ZDC
+        newquad->fSFlag = 1;  // SECONDARY particle entering the ZDC
       }
       else if(track == primary){
-        newquad->fSFlag = 0;  // Hit created by PRIMARY particle entering the ZDC
+        newquad->fSFlag = 0;  // PRIMARY particle entering the ZDC
       }  
-      fNPrimaryHits = fNPrimaryHits + 1;
-      sFlag = newquad->fSFlag;
+//      fNPrimaryHits += 1;
+      sFlag    = newquad->fSFlag;
       primKinEn = newquad->fPrimKinEn;
-      xImpact = newquad->fXImpact;
-      yImpact = newquad->fYImpact;
+      xImpact  = newquad->fXImpact;
+      yImpact  = newquad->fYImpact;
    }
    else{       
       newquad->fPrimKinEn = primKinEn;
-      newquad->fXImpact = xImpact;
+      newquad->fXImpact        = xImpact;
       newquad->fYImpact = yImpact;
-      newquad->fSFlag = sFlag;
+      newquad->fSFlag  = sFlag;
    }
+  Int_t j;
+  for(j=0; j<fNhits; j++){
+    // If hits are equal (same track, same volume), sum them.
+     curprimquad = (AliZDCHit*) lhits[j];
+     if(*curprimquad == *newquad){
+        *curprimquad = *curprimquad+*newquad;
+       delete newquad;
+       return;
+     } 
+  }
 
     //Otherwise create a new hit
     new(lhits[fNhits]) AliZDCHit(newquad);
     fNhits++;
     
-    if(kStHit){
-      new(lsthits[fNStHits]) AliZDCHit(newquad);
-      fNStHits++;
-    }
-//    printf("\n  Primary Hits --------------------------------------------------------\n");
-//    fHits->Print("");
-//    printf("\n  Event Hits --------------------------------------------------------\n");
-//    fStHits->Print("");
-
     delete newquad;
-  }
-  
+}
+
 //_____________________________________________________________________________
-void AliZDC::ResetDigits()
+void  AliZDC::AddDigit(Int_t *sect, Int_t digit)
 {
-  //
-  // Reset number of digits and the digits array
-  //
-    
-    AliDetector::ResetDigits();
-//    fNStHits = 0;
-//    if(fStHits) fStHits->Clear();
-}
+//
+  AliZDCDigit *newdigit;
+  newdigit = new AliZDCDigit(sect, digit);
 
+//  AliZDCDigit *curdigit;
+//  TClonesArray &ldigits = *fDigits;
+//
+//  Int_t j;
+//  for(j=0; j<fNdigits; j++){
+//     curdigit = (AliZDCDigit*) ldigits[j];
+//     if(*curdigit == *newdigit){
+//     *curdigit = *curdigit+*newdigit;
+//      delete newdigit;
+//      return;
+//     } 
+//  } 
+//
+  
+//  printf("\n AddDigit -> sector[0] = %d, sector[1] = %d, digit = %d",
+//         sect[0], sect[1], digit);
+  new((*fDigits)[fNdigits]) AliZDCDigit(*newdigit);
+  fNdigits++;
+  delete newdigit;
+}
+      
 //_____________________________________________________________________________
 void AliZDC::BuildGeometry()
 {
@@ -289,7 +283,7 @@ void AliZDC::BuildGeometry()
 
   TNode *node, *top;
   TBRIK *brik;
-  const int kColorZDC  = kRed;
+  const int kColorZDC  = kBlue;
   
   //
   top=gAlice->GetGeometry()->GetNode("alice");
@@ -311,11 +305,253 @@ Int_t AliZDC::DistancetoPrimitive(Int_t , Int_t )
   //
   return 9999;
 }
+
+//____________________________________________________________________________
+Float_t AliZDC::ZMin(void) const
+{
+  // Minimum dimension of the ZDC module in z
+  return 11600.;
+}
+
+//____________________________________________________________________________
+Float_t AliZDC::ZMax(void) const
+{
+  // Maximum dimension of the ZDC module in z
+  return  11750.;
+}
+  
+
 //_____________________________________________________________________________
-void AliZDC::StepManager()
+ void AliZDC::MakeBranch(Option_t *opt, const char *file)
 {
   //
-  // Routine called at every step in the Zero Degree Calorimeter
+  // Create Tree branches for the ZDC
   //
+
+  char branchname[10];
+  sprintf(branchname,"%s",GetName());
+  
+  AliDetector::MakeBranch(opt);
+
+  const char *cS = strstr(opt,"S");
+
+  if (gAlice->TreeS() && cS) {
+    if(fMergedHits!=0) fMergedHits->Clear();
+    else fMergedHits = new TClonesArray ("AliZDCMergedHit",1000);
+    MakeBranchInTree(gAlice->TreeS(), 
+                     branchname, &fMergedHits, fBufferSize, file) ;
+    printf("* AliZDC::MakeBranch    * Making Branch %s for SDigits\n\n",branchname);
+  }
+
+    
+  const char *cD = strstr(opt,"D");
+
+  if (gAlice->TreeD() && cD) {
+    if(fDigits!=0) fDigits->Clear();
+    else fDigits = new TClonesArray ("AliZDCDigit",1000);
+    MakeBranchInTree(gAlice->TreeD(), 
+                     branchname, &fDigits, fBufferSize, file) ;
+    printf("* AliZDC::MakeBranch    * Making Branch %s for Digits\n\n",branchname);
+  }
+
+  
+  const char *cR = strstr(opt,"R");
+
+  if (gAlice->TreeR() && cR) {
+    MakeBranchInTree(gAlice->TreeR(), 
+                    branchname, &fRecPoints, fBufferSize, file) ;
+    printf("* AliZDC::MakeBranch    * Making Branch %s for RecPoints\n\n",branchname);   }
+          
+}
+
+//_____________________________________________________________________________
+ void AliZDC::MakeBranchInTreeSD(TTree *treeSD, const char *file)
+{
+  // MakeBranchInTree
+  const Int_t kBufferSize = 4000;
+  char  branchname[20];
+  sprintf(branchname,"%s",GetName());
+  MakeBranchInTree(treeSD, branchname, &fMergedHits, kBufferSize, file) ;
+  printf("* AliZDC::MakeBranch    * Making Branch %s for SDigits\n\n",branchname);
+
+}
+//_____________________________________________________________________________
+ void AliZDC::MakeBranchInTreeD(TTree *treeD, const char *file)
+{
+  // MakeBranchInTree
+  const Int_t kBufferSize = 4000;
+  char  branchname[20];
+  sprintf(branchname,"%s",GetName());
+  MakeBranchInTree(treeD, branchname, &fDigits, kBufferSize, file) ;
+  printf("* AliZDC::MakeBranch    * Making Branch %s for Digits\n\n",branchname);
+
+}
+//_____________________________________________________________________________
+void AliZDC::Hits2SDigits()
+{
+//  printf("\n Entering AliZDC::SDigits2Digits()\n");
+  
+  //----------------------------------------------------------------
+  if(!fMerger){ 
+//    printf("\n       ZDC digitization (without merging)\n");
+
+    AliZDCMergedHit *MHit;
+    Int_t j, sector[2];
+    Float_t MHits[7];
+    fNMergedhits = 0;
+
+    TTree *treeH = gAlice->TreeH();
+    Int_t ntracks = (Int_t) treeH->GetEntries();
+    gAlice->ResetHits();
+  
+    // Tracks loop
+    for(Int_t itrack=0; itrack<ntracks; itrack++){
+       treeH->GetEvent(itrack);
+       for(AliZDCHit* zdcHit=(AliZDCHit*)this->FirstHit(-1); zdcHit;
+                      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();
+       }//Hits loop
+       
+         MHit = new AliZDCMergedHit(sector, MHits);
+         new((*fMergedHits)[fNMergedhits]) AliZDCMergedHit(*MHit);       
+         TClonesArray &sdigits = *fMergedHits;
+         new (sdigits[fNMergedhits]) AliZDCMergedHit(*MHit);
+         fNMergedhits++;
+         delete MHit;
+    }
+//    printf("\n       ### Filling SDigits tree\n");
+    gAlice->TreeS()->Fill();
+    gAlice->TreeS()->Write(0,TObject::kOverwrite);  
+    gAlice->TreeS()->Reset();  
+  }
+  //----------------------------------------------------------------
+  else if(fMerger){
+//    printf("\n         ZDC merging and digitization\n");
+    // ### Initialise merging
+    fMerger -> InitMerging();
+
+    TFile *bgrFile  = fMerger->BgrFile();
+    bgrFile->cd();
+    // SDigits tree
+    Int_t fNEvBgr = fMerger->EvNum();
+    char treeSDBgrName[20];
+    sprintf(treeSDBgrName,"TreeS%d",fNEvBgr);
+    fTreeSD = (TTree*)gDirectory->Get(treeSDBgrName); // TreeH
+    if(!fTreeSD){
+      printf("\n ERROR -> Can't find TreeS%d in background file\n",fNEvBgr);
+    }   
+    // Branch address
+    TBranch *branchSD;
+    char branchSDname[20];
+    sprintf(branchSDname,"%s",GetName());
+    if(fTreeSD && fMergedHits){
+//      printf("\n     fTreeSD!=0 && fMergedHits!=0\n");
+      branchSD = fTreeSD->GetBranch(branchSDname);
+      if(branchSD) branchSD->SetAddress(&fMergedHits);
+    }
+    if(!branchSD) MakeBranchInTreeSD(fTreeSD);
+
+    // ### Get TCA of MergedHits from AliZDCMerger
+    fMergedHits  = fMerger->MergedHits();
+    fNMergedhits = fMerger->GetNMhits();
+//    printf("\n         fNMergedhits (from AliZDCMerger) = %d\n", fNMergedhits);   
+    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);
+    }
+
+//    printf("\n ### Filling SDigits tree\n");
+    bgrFile->cd();
+    fTreeSD->Fill();
+    fTreeSD->Write(0,TObject::kOverwrite);
+  }
+  
+}
+
+//_____________________________________________________________________________
+void AliZDC::SDigits2Digits()
+{
+//  printf("\n Entering AliZDC::SDigits2Digits()\n");
+  if(!fMerger){ // Only digitization
+//    printf("\n        ZDC digitization (without merging) \n");
+    fMerger = new AliZDCMerger();    
+    fMerger->Digitize(fNMergedhits, fMergedHits);
+
+    char hname[30];
+    sprintf(hname,"TreeD%d",gAlice->GetHeader()->GetEvent());
+    gAlice->TreeD()->Fill();
+    gAlice->TreeD()->Write(0,TObject::kOverwrite);
+    gAlice->TreeD()->Reset();  
+  }
+  else if(fMerger){    // Merging and digitization
+//    printf("\n        ZDC merging and digitization\n");
+    fMerger->Digitize(fNMergedhits, fMergedHits);
+
+    TFile *bgrFile = fMerger->BgrFile();
+    bgrFile->cd();
+    // Digits tree
+    Int_t fNEvBgr = fMerger->EvNum();
+    char treeDBgrName[20];
+    sprintf(treeDBgrName,"TreeD%d",fNEvBgr);
+    fTreeMD = (TTree*)gDirectory->Get(treeDBgrName); // TreeH
+    if(!fTreeMD){
+      printf("\n ERROR -> Can't find TreeD%d in background file\n",fNEvBgr);
+    }   
+    // Branch address
+    TBranch *branchD;
+    char branchDname[20];
+    sprintf(branchDname,"%s",GetName());
+    if(fTreeMD && fDigits){
+//      printf("\n     fTreeMD!=0 && fDigits!=0\n");
+      branchD = fTreeMD->GetBranch(branchDname);
+      if(branchD) branchD->SetAddress(&fDigits);
+    }
+    if(!branchD) MakeBranchInTreeD(fTreeMD);
+    
+//    printf("\n ### Filling Digits tree\n");
+    fTreeMD->Fill();
+    fTreeMD->Write(0,TObject::kOverwrite);
+  }
+  
+  
 }
+//_____________________________________________________________________________
+void AliZDC::Hits2Digits()
+{
+    gAlice->Hits2SDigits();
+    gAlice->SDigits2Digits();
+}
+
+//_____________________________________________________________________________
+void AliZDC::Digits2Reco()
+{
+    
+}
+
+//_____________________________________________________________________________
+void   AliZDC::SetMerger(AliZDCMerger* merger)
+{
+// Set pointer to merger 
+    fMerger = merger;
+}
+
+//_____________________________________________________________________________
+AliZDCMerger*  AliZDC::Merger()
+{
+// Return pointer to merger
+    return fMerger;
+}
+