]> git.uio.no Git - u/mrichter/AliRoot.git/blobdiff - ZDC/AliZDC.cxx
Removing old macros
[u/mrichter/AliRoot.git] / ZDC / AliZDC.cxx
index 97441d7d88c1d8ff2a9e65a6163daad31efd906f..aac2849a6587fc093d6d224791a119717f99c156 100644 (file)
  * provided "as is" without express or implied warranty.                  *
  **************************************************************************/
 
-/*
-$Log$
-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$ */
 
 ///////////////////////////////////////////////////////////////////////////////
 //                                                                           //
-//  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>
-
 // --- ROOT system
 #include <TBRIK.h>
-#include <TNode.h>
 #include <TGeometry.h>
-#include <TFile.h>
+#include <TNode.h>
 #include <TTree.h>
 
 // --- AliRoot header files
+#include "AliDetector.h"
 #include "AliZDC.h"
 #include "AliZDCHit.h"
+#include "AliZDCSDigit.h"
+#include "AliZDCDigit.h"
+#include "AliZDCDigitizer.h"
+#include "AliZDCRawStream.h"
+
+#include "AliRawDataHeader.h"
+#include "AliLoader.h"
 #include "AliRun.h"
-#include "AliDetector.h"
-#include "AliCallf77.h"
-#include "AliConst.h"
 #include "AliMC.h"
 
  
@@ -92,14 +54,15 @@ AliZDC::AliZDC()
   // Default constructor for the Zero Degree Calorimeter base class
   //
   
-  fIshunt = 1;
+  fIshunt     = 1;
+  fNoShower   = 0;
 
-  fNhits = 0;
+  fHits       = 0;
+  fNhits      = 0;
 
-  fNStHits = 0;
+  fDigits     = 0;
+  fNdigits    = 0;
 
-  fNPrimaryHits = 0;
-  fNoShower   = 0;
 }
  
 //_____________________________________________________________________________
@@ -110,30 +73,15 @@ 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* PIPE=gAlice->GetModule("PIPE");
-  AliModule* ABSO=gAlice->GetModule("ABSO");
-  AliModule* DIPO=gAlice->GetModule("DIPO");
-  AliModule* SHIL=gAlice->GetModule("SHIL");
-  if((!PIPE) || (!ABSO) || (!DIPO) || (!SHIL)) {
-    Error("Constructor","ZDC needs PIPE, ABSO, DIPO and SHIL!!!\n");
-    exit(1);
-  } 
-
-  //
-  // Allocate the array of hits
-  
-  fIshunt =  1;
+  fIshunt   = 1;
+  fNoShower = 0;
 
+  // Allocate the hits array  
   fHits   = new TClonesArray("AliZDCHit",1000);
-  gAlice->AddHitList(fHits);
-  
-  fStHits = new TClonesArray("AliZDCHit",1000);
-  fNStHits = 0;
+  gAlice->GetMCApp()->AddHitList(fHits);
 
-  fNPrimaryHits = 0;
-  fNoShower   = 0;
+  fDigits     = 0;
+  fNdigits    = 0;
 
 }
 //____________________________________________________________________________ 
@@ -145,13 +93,6 @@ AliZDC::~AliZDC()
 
   fIshunt   = 0;
 
-//  delete fHits;
-//  if(fStHits){
-//    fStHits->Delete();
-//    delete fStHits;
-//    fNStHits = 0;
-//  }
-//  delete fDigits;
 }
 //_____________________________________________________________________________
 void AliZDC::AddHit(Int_t track, Int_t *vol, Float_t *hits)
@@ -166,84 +107,47 @@ void AliZDC::AddHit(Int_t track, Int_t *vol, Float_t *hits)
   
   static Float_t primKinEn, xImpact, yImpact, sFlag;
 
-  AliZDCHit *newquad, *curevquad, *curprimquad;
+  AliZDCHit *newquad, *curprimquad;
   newquad = new AliZDCHit(fIshunt, track, vol, hits);
-
-  TClonesArray &lsthits = *fStHits;
   TClonesArray &lhits = *fHits;
-   
-  Int_t i,j,kStHit = 1;
-  for(i=0; i<fNStHits; i++){
-    // If 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 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);     
+      Int_t primary = gAlice->GetMCApp()->GetPrimary(track);     
       if(track != primary){
-        newquad->fSFlag = 1;  // Hit created by secondary particle entering the ZDC
+        newquad->SetSFlag(1);  // SECONDARY particle entering the ZDC
       }
       else if(track == primary){
-        newquad->fSFlag = 0;  // Hit created by PRIMARY particle entering the ZDC
+        newquad->SetSFlag(0);  // PRIMARY particle entering the ZDC
       }  
-      fNPrimaryHits += 1;
-      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;
+  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++;
-    }
-//    if(fDebug == 1){ 
-//      printf("\n  Primary Hits --------------------------------------------------------\n");
-//      fHits->Print("");
-//      printf("\n  Event Hits --------------------------------------------------------\n");
-//      fStHits->Print("");
-//    }
-
     delete newquad;
-  }
-  
-//_____________________________________________________________________________
-void AliZDC::ResetDigits()
-{
-  //
-  // Reset number of digits and the digits array
-  //
-    
-    AliDetector::ResetDigits();
-//    fNStHits = 0;
-//    if(fStHits) fStHits->Clear();
 }
 
 //_____________________________________________________________________________
@@ -257,7 +161,7 @@ void AliZDC::BuildGeometry()
 
   TNode *node, *top;
   TBRIK *brik;
-  const int kColorZDC  = kRed;
+  const int kColorZDC  = kBlue;
   
   //
   top=gAlice->GetGeometry()->GetNode("alice");
@@ -279,11 +183,269 @@ 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());
+
+  const char *cH = strstr(opt,"H");
+  
+  if (cH && fLoader->TreeH())
+   fHits   = new TClonesArray("AliZDCHit",1000); 
+  
+  AliDetector::MakeBranch(opt);
+}
+
+//_____________________________________________________________________________
+void AliZDC::Hits2SDigits()
+{
+  // Create summable digits from hits
+  
+  if (GetDebug()) printf("\n   Entering AliZDC::Hits2Digits() ");
+  
+  fLoader->LoadHits("read");
+  fLoader->LoadSDigits("recreate");
+  AliRunLoader* runLoader = fLoader->GetRunLoader();
+  AliZDCSDigit sdigit;
+  AliZDCSDigit* psdigit = &sdigit;
+
+  // Event loop
+  for (Int_t iEvent = 0; iEvent < runLoader->GetNumberOfEvents(); iEvent++) {
+    Float_t pmCZN = 0, pmCZP = 0, pmQZN[4], pmQZP[4], pmZEM1 = 0, pmZEM2 = 0;
+    for (Int_t i = 0; i < 4; i++) pmQZN[i] = pmQZP[i] = 0;
+
+    runLoader->GetEvent(iEvent);
+    TTree* treeH = fLoader->TreeH();
+    Int_t ntracks = (Int_t) treeH->GetEntries();
+    ResetHits();
+
+    // Tracks loop
+    Int_t sector[2];
+    for (Int_t itrack = 0; itrack < ntracks; itrack++) {
+      treeH->GetEntry(itrack);
+      for (AliZDCHit* zdcHit = (AliZDCHit*)FirstHit(-1); zdcHit;
+                      zdcHit = (AliZDCHit*)NextHit()) { 
+                     
+       sector[0] = zdcHit->GetVolume(0);
+       sector[1] = zdcHit->GetVolume(1);
+       if ((sector[1] < 1) || (sector[1] > 4)) {
+         Error("Hits2SDigits", "sector[0] = %d, sector[1] = %d", 
+               sector[0], sector[1]);
+         continue;
+       }
+       Float_t lightQ = zdcHit->GetLightPMQ();
+       Float_t lightC = zdcHit->GetLightPMC();
+     
+       if (sector[0] == 1) {          //ZN 
+         pmCZN += lightC;
+         pmQZN[sector[1]-1] += lightQ;
+       } else if (sector[0] == 2) {   //ZP 
+         pmCZP += lightC;
+         pmQZP[sector[1]-1] += lightQ;
+       } else if (sector[0] == 3) {   //ZEM 
+         if (sector[1] == 1) pmZEM1 += lightC;
+         else                pmZEM2 += lightQ;
+       }
+      }//Hits loop
+    }
+
+    // create the output tree
+    fLoader->MakeTree("S");
+    TTree* treeS = fLoader->TreeS();
+    const Int_t kBufferSize = 4000;
+    treeS->Branch(GetName(), "AliZDCSDigit", &psdigit, kBufferSize);
+
+    // Create sdigits for ZN
+    sector[0] = 1; // Detector = ZN
+    sector[1] = 0; // Common PM ADC
+    new(psdigit) AliZDCSDigit(sector, pmCZN);
+    if (pmCZN > 0) treeS->Fill();
+    for (Int_t j = 0; j < 4; j++) {
+      sector[1] = j+1; // Towers PM ADCs
+      new(psdigit) AliZDCSDigit(sector, pmQZN[j]);
+      if (pmQZN[j] > 0) treeS->Fill();
+    }
+  
+    // Create sdigits for ZP
+    sector[0] = 2; // Detector = ZP
+    sector[1] = 0; // Common PM ADC
+    new(psdigit) AliZDCSDigit(sector, pmCZP);
+    if (pmCZP > 0) treeS->Fill();
+    for (Int_t j = 0; j < 4; j++) {
+      sector[1] = j+1; // Towers PM ADCs
+      new(psdigit) AliZDCSDigit(sector, pmQZP[j]);
+      if (pmQZP[j] > 0) treeS->Fill();
+    }
+
+    // Create sdigits for ZEM
+    sector[0] = 3; 
+    sector[1] = 1; // Detector = ZEM1
+    new(psdigit) AliZDCSDigit(sector, pmZEM1);
+    if (pmZEM1 > 0) treeS->Fill();
+    sector[1] = 2; // Detector = ZEM2
+    new(psdigit) AliZDCSDigit(sector, pmZEM2);
+    if (pmZEM2 > 0) treeS->Fill();
+
+    // write the output tree
+    fLoader->WriteSDigits("OVERWRITE");
+  }
+
+  fLoader->UnloadHits();
+  fLoader->UnloadSDigits();
+}
+
+//_____________________________________________________________________________
+AliDigitizer* AliZDC::CreateDigitizer(AliRunDigitizer* manager) const
+{
+  // Create the digitizer for ZDC
+
+  return new AliZDCDigitizer(manager);
+}
+
+//_____________________________________________________________________________
+void AliZDC::Digits2Raw()
+{
+  // Convert ZDC digits to raw data
+
+  // preliminary format: 12 interger values (ZNC, ZNQ1-4, ZPC, ZPQ1-4, ZEM1,2)
+  // For the CAEN module V965 we have an header, the Data Words and an End Of Block
+  UInt_t ADCHeader; 
+  UInt_t ADCData[24];
+  UInt_t ADCEndBlock;
+
+  // load the digits
+  fLoader->LoadDigits("read");
+  AliZDCDigit digit;
+  AliZDCDigit* pdigit = &digit;
+  TTree* treeD = fLoader->TreeD();
+  if (!treeD) return;
+  treeD->SetBranchAddress("ZDC", &pdigit);
+
+  // Fill data array
+  // ADC header
+  UInt_t ADCHeaderGEO = 0;
+  UInt_t ADCHeaderCRATE = 0;
+  UInt_t ADCHeaderCNT = (UInt_t) treeD->GetEntries();
+    
+  ADCHeader = ADCHeaderGEO << 27 | 0x1 << 25 | ADCHeaderCRATE << 16 |
+              ADCHeaderCNT << 8 ;
+
+  //printf("ADCHeader = %d\n",ADCHeader);
+      
+  // ADC data word
+  UInt_t ADCDataGEO = ADCHeaderGEO;
+  UInt_t ADCDataValue[24];
+  UInt_t ADCDataOvFlw[24];
+  for(Int_t i = 0; i < 24; i++){
+    ADCDataValue[i] = 0;
+    ADCDataOvFlw[i] = 0;
+  }
+  UInt_t ADCDataChannel = 0;
+  
+  // loop over digits
+  for (Int_t iDigit = 0; iDigit < treeD->GetEntries(); iDigit++) {
+    treeD->GetEntry(iDigit);
+    if (!pdigit) continue;
+
+    //ADC data
+    Int_t index = 0;
+    if(digit.GetSector(0)!=3){
+      index = (digit.GetSector(0)-1) + digit.GetSector(1)*4;
+      ADCDataChannel = (digit.GetSector(0)-1)*8 + digit.GetSector(1);
+    }
+    else {
+      index = 19 + digit.GetSector(1);
+      ADCDataChannel = 5 + digit.GetSector(1)*8;
+    }
+     
+    if ((index < 0) || (index >= 22)) {
+      Error("Digits2Raw", "sector[0] = %d, sector[1] = %d", 
+           digit.GetSector(0), digit.GetSector(1));
+      continue;
+    }
+    
+    ADCDataValue[index] = digit.GetADCValue(0);
+    if (ADCDataValue[index] > 2047) ADCDataOvFlw[index] = 1; 
+    ADCDataValue[index+2] = digit.GetADCValue(1);
+    if (ADCDataValue[index+2] > 2047) ADCDataOvFlw[index+2] = 1; 
+    
+    ADCData[index] =   ADCDataGEO << 27 | ADCDataChannel << 17 | 
+                       ADCDataOvFlw[index] << 12 | (ADCDataValue[index] & 0xfff); 
+    ADCData[index+2] = ADCDataGEO << 27 | ADCDataChannel << 17 | 0x1 << 16 |
+                       ADCDataOvFlw[index+2] << 12 | (ADCDataValue[index+2] & 0xfff);                    
+  }
+  //for (Int_t i=0;i<24;i++)printf("ADCData[%d] = %d\n",i,ADCData[i]);
+  
+  // End of Block
+  UInt_t ADCEndBlockGEO = ADCHeaderGEO;
+  UInt_t ADCEndBlockEvCount = gAlice->GetEventNrInRun();
+  
+  ADCEndBlock = ADCEndBlockGEO << 27 | 0x1 << 26 | ADCEndBlockEvCount;
+  
+  //printf("ADCEndBlock = %d\n",ADCEndBlock);
+
+
+  // open the output file
+  char fileName[30];
+  sprintf(fileName, "ZDC_%d.ddl", AliZDCRawStream::kDDLOffset);
+#ifndef __DECCXX
+  ofstream file(fileName, ios::binary);
+#else
+  ofstream file(fileName);
+#endif
+
+  // write the DDL data header
+  AliRawDataHeader header;
+  header.fSize = sizeof(header) + sizeof(ADCHeader) + 
+                 sizeof(ADCData) + sizeof(ADCEndBlock);
+  //printf("sizeof header = %d, ADCHeader = %d, ADCData = %d, ADCEndBlock = %d\n",
+  //        sizeof(header),sizeof(ADCHeader),sizeof(ADCData),sizeof(ADCEndBlock));
+  header.SetAttribute(0);  // valid data
+  file.write((char*)(&header), sizeof(header));
+
+  // write the raw data and close the file
+  file.write((char*) &ADCHeader, sizeof (ADCHeader));
+  file.write((char*)(ADCData), sizeof(ADCData));
+  file.write((char*) &ADCEndBlock, sizeof(ADCEndBlock));
+  file.close();
+
+  // unload the digits
+  fLoader->UnloadDigits();
 }
+
+//______________________________________________________________________
+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);
+      
+  AliDetector::SetTreeAddress();
+}