]> git.uio.no Git - u/mrichter/AliRoot.git/blobdiff - PHOS/AliPHOSRaw2Digits.cxx
Combine TEPEMGEN and EPEMGEN in one library to avoid circular dependences. Compilatio...
[u/mrichter/AliRoot.git] / PHOS / AliPHOSRaw2Digits.cxx
index 541223d59fbfd1e5406e50282a30fbcb134ebfb9..18d0995581685803f444088963fbe99137d7b5b7 100644 (file)
 
 /* $Id$ */
 
+/* History of cvs commits:
+ *
+ * $Log$
+ * Revision 1.14  2007/01/23 10:27:37  alibrary
+ * Adding include files where needed for latest ROOT
+ *
+ * Revision 1.13  2006/09/07 18:31:08  kharlov
+ * Effective c++ corrections (T.Pocheptsov)
+ *
+ * Revision 1.12  2005/05/28 14:19:04  schutz
+ * Compilation warnings fixed by T.P.
+ *
+ */
+
 //_________________________________________________________________________
-// Short description  
+// Class designed to convert raw data to aliroot format. We assume, that
+// prototype is situated in the center of 3 PHOS module and convert prototype
+// outpur to AliPHOSDigits. In addition, we fill branch of TreeE with 
+// AliPHOSBeamTestEvent, contaning description of event(triggers etc).
+// Note, that one byte per channel in raw data is transvormed to class 
+// AliPHOSDigit, so finale zise increase ~ 100 times. So output can be split 
+// into peases of reasonable size: each file can not contain more than 
+// fMaxPerFile: if there are still events in raw file, then new directory 
+// is created and header+digits files are written to it.
+// 
+// Use Case:
+//   AliPHOSRaw2Digits * r = new AliPHOSRaw2Digits("path/input.file") ;
+//                   //note, that it can be gzipped file!
+//   //Set position of the target in the given run.
+//   //Z axis along beam direction, from target to prototype (0-surface of prototype)
+//   //X axis along columns of prototype (0-center of prototype)
+//   //Y axis along raws of prototype    (0-center of prototype)
+//   Double_t pos[3]={0,0,-452.} ;
+//   r->SetTargetPosition(pos) ;
+//   //Read/create connection Table:
+//   TFile f("ConTableDB.root") ;
+//   AliPHOSConTableDB * cdb = f.Get("AliPHOSConTableDB") ;
+//   f.Close() ;
+//   r->SetConTableDB(cdb) ;
+//   r->ExecuteTask() ;
+//
+// As a result files galice.root and PHOS.Digits.root should be produced in 
+// current dir, and, possibly, dirs 1,2,3... each with galice.root and PHOS.Digits.root,
+// where the rest of data are written. 
 //
 /*-- Author: Maxim Volkov (RRC KI)
              Dmitri Peressounko (RRC KI & SUBATECH)
 //////////////////////////////////////////////////////////////////////////////
 
 // --- ROOT system ---
-#include "TClonesArray.h"
-#include "TFile.h"
-#include "TTree.h"
+#include <Bytes.h>
+#include <TClonesArray.h>
+#include <TFile.h>
+#include <TSystem.h>
+#include <TTree.h>
 
 // --- Standard library ---
-//#include <sys/types.h>
-#include <sys/stat.h>
-#include <fcntl.h>
-#include <unistd.h>
-#include <ctype.h>
+
 #include <stdio.h>
-#include <stdlib.h>
-#include <unistd.h>
-#include <string.h>
-#include <netinet/in.h>
 
 // --- AliRoot header files ---
 #include "AliPHOSDigit.h"
 #include "AliPHOSConTableDB.h"
 #include "AliPHOSBeamTestEvent.h"
 #include "AliPHOSRaw2Digits.h"
-#include "AliPHOSv1.h"
-#include "../EVGEN/AliGenBox.h"
 #include "AliRun.h"
 
 ClassImp(AliPHOSRaw2Digits)
   
-  
 //____________________________________________________________________________ 
-  AliPHOSRaw2Digits::AliPHOSRaw2Digits():TTask() 
+AliPHOSRaw2Digits::AliPHOSRaw2Digits() : 
+  fDigits(0),
+  fPHOSHeader(0),
+  fctdb(0),
+  fHeaderFile(0),
+  fDigitsFile(0),
+  fBeamEnergy(0.f),
+  fMaxPerFile(20000),
+  fEvent(0),
+  fStatus(0),
+  fInName(""),
+  fDebug(kFALSE),
+  fIsInitialized(kFALSE),
+  fMK1(0x0123CDEF),
+  fMK2(0x80708070),
+  fMK3(0x4321ABCD),
+  fMK4(0x80618061),
+  fCKW(0x4640E400)
 {
-  fInName="";  
-  fMK1 = 0x0123CDEF ;
-  fMK2 = 0x80708070 ;
-  fMK3 = 0x4321ABCD ;
-  fMK4 = 0x80618061 ;
-  fCKW = 0x4640E400 ;
-  fDebug = kFALSE;             //  Debug flag
-  fIsInitialized = kFALSE ;
+  //As one can easily see, this is constructor.
   fTarget[0] = 0 ;
   fTarget[1] = 0 ;
   fTarget[2] = 0 ;
-  fDigits = 0 ;
-  fPHOSHeader =0 ;
-  fEvent = 0 ;
-  fctdb = 0;
 }
+
 //____________________________________________________________________________ 
-  AliPHOSRaw2Digits::AliPHOSRaw2Digits(const char * filename):TTask("Default","") 
+AliPHOSRaw2Digits::AliPHOSRaw2Digits(const char * filename) : 
+  TTask("Default",""),
+  fDigits(0),
+  fPHOSHeader(0),
+  fctdb(0),
+  fHeaderFile(0),
+  fDigitsFile(0),
+  fBeamEnergy(0.f),
+  fMaxPerFile(20000),
+  fEvent(0),
+  fStatus(0),
+  fInName(filename),
+  fDebug(kFALSE),
+  fIsInitialized(kFALSE),
+  fMK1(0x0123CDEF),
+  fMK2(0x80708070),
+  fMK3(0x4321ABCD),
+  fMK4(0x80618061),
+  fCKW(0x4640E400)
 {
-  fInName=filename;
+  //this constructor should be normally used. Parameters: input file 
   TString outname(fInName) ;
   outname.ToLower() ;
   outname.ReplaceAll(".fz",".root") ;
   outname.ReplaceAll(".gz","") ;
-  SetTitle(outname) ;
-
-  fMK1 = 0x0123CDEF ;
-  fMK2 = 0x80708070 ;
-  fMK3 = 0x4321ABCD ;
-  fMK4 = 0x80618061 ;
-  fCKW = 0x4640E400 ;
-  fDebug = kFALSE;             //  Debug flag
-  fIsInitialized = kFALSE ;
+  SetTitle(outname);
+
   fTarget[0] = 0 ;
   fTarget[1] = 0 ;
   fTarget[2] = 0 ;
-  fDigits = 0 ;
-  fPHOSHeader =0 ;
-  fEvent = 0 ;
-  fctdb = 0;
+}
+
+//____________________________________________________________________________ 
+AliPHOSRaw2Digits::AliPHOSRaw2Digits(AliPHOSRaw2Digits & r2d) :
+  TTask(r2d.GetName(), r2d.GetTitle()),
+  fDigits(r2d.fDigits),
+  fPHOSHeader(r2d.fPHOSHeader),
+  fctdb(new AliPHOSConTableDB(*r2d.fctdb)),
+  fHeaderFile(new TFile(r2d.fHeaderFile->GetName(), "new" )),
+  fDigitsFile(new TFile(r2d.fDigitsFile->GetName(), "new" )),
+  fBeamEnergy(r2d.fBeamEnergy),
+  fMaxPerFile(r2d.fMaxPerFile),
+  fEvent(r2d.fEvent),
+  fStatus(r2d.fStatus),
+  fInName(r2d.fInName),
+  fDebug(kFALSE),
+  fIsInitialized(kFALSE),
+  fMK1(r2d.fMK1),
+  fMK2(r2d.fMK2),
+  fMK3(r2d.fMK3),
+  fMK4(r2d.fMK4),
+  fCKW(r2d.fCKW)
+{
+  // cpy ctor. wrong. because dtor can delete fDigits twice (or n times you copy AliPHOSRaw2Digits)
+  //because fHeaderFile and fDigitsFile will recreate existing files etc.
+  fTarget[0] = r2d.fTarget[0] ;
+  fTarget[1] = r2d.fTarget[1] ;
+  fTarget[2] = r2d.fTarget[2] ;
 }
 
 //____________________________________________________________________________ 
 AliPHOSRaw2Digits::~AliPHOSRaw2Digits()
 {
+//destructor
   if(fPHOSHeader)
     fPHOSHeader->Delete() ;
   if(fDigits){
@@ -110,7 +186,7 @@ AliPHOSRaw2Digits::~AliPHOSRaw2Digits()
   
 }
 //____________________________________________________________________________ 
-void AliPHOSRaw2Digits::Exec(Option_t * option){
+void AliPHOSRaw2Digits::Exec(const Option_t *){
   //This is steering method performing all the conversion
 
   if(!fIsInitialized) //need initialization
@@ -119,72 +195,135 @@ void AliPHOSRaw2Digits::Exec(Option_t * option){
 
   ProcessRawFile() ;
 
-  FinishRun() ;
 } 
 //____________________________________________________________________________ 
 Bool_t AliPHOSRaw2Digits::Init(void){
-  //Create PHOS geometry, sets magnetic field to zero, 
-  //create Generator - to store target position, 
-  //opens out file, creates TreeE and make initialization of contaniers
-
+  //Makes initialization of contaniers
 
   if(fIsInitialized)
     return kTRUE;
 
-  //Create PHOS
-  new AliPHOSv1("PHOS","GPS2") ;
-
-  //Set Magnetic field
-  gAlice->SetField(0,2);  
-
-  //Set positin of the virtex
-  AliGenBox *gener = new AliGenBox(1);
-  Float_t ox = fTarget[1]; 
-  Float_t oy = fTarget[2]-460.; 
-  Float_t oz = fTarget[0];
-  gener->SetOrigin(ox, oy, oz);
-
-  //  Create the output file
-  TString outname("") ;
-  if(strstr(GetTitle(),"root")){
-    outname=GetTitle();
-  }
-  else{
-    outname = fInName ;
-    outname.ToLower() ;
-    outname.ReplaceAll(".fz",".root") ;
-  }
-  TFile *rootfile = new TFile(outname,"recreate");
-  rootfile->SetCompressionLevel(2);
-
-  // Create the Root Trees
-  gAlice->MakeTree("E") ;
-
   //Make container for digits
   fDigits = new TClonesArray("AliPHOSDigit",1000) ;
-
-  //Fill now TreeE
   fPHOSHeader = new  AliPHOSBeamTestEvent() ;
-  Int_t splitlevel = 0 ;
-  Int_t bufferSize = 32000 ;    
-  TBranch * headerBranch = gAlice->TreeE()->Branch("AliPHOSBeamTestEvent", 
-                                                  "AliPHOSBeamTestEvent", 
-                                                  &fPHOSHeader,bufferSize,splitlevel);
-  headerBranch->SetName("AliPHOSBeamTestEvent") ;
-
   fIsInitialized = kTRUE ;
+  return StartRootFiles() ;
+
+}
+//____________________________________________________________________________ 
+Bool_t AliPHOSRaw2Digits::StartRootFiles(void ) const {
+//   //Create PHOS geometry, sets magnetic field to zero, 
+//   //create Generator - to store target position, 
+//   //opens out file, creates TreeE 
+
+//   //create gAlice if nececcary
+//   if(!gAlice)
+//     new AliRun("gAlice","The ALICE Off-line Simulation Framework") ;
+
+//   //Create PHOS
+//   if(!gAlice->GetModule("PHOS"))
+//     new AliPHOSv1("PHOS","GPS2") ;
+
+//   //Set Magnetic field
+//   gAlice->SetField(0,2);  
+
+//   //Set positin of the virtex
+//   AliGenerator * gener = gAlice->Generator() ; 
+//   if(!gener)    
+//     gener = new AliGenBox(1);
+//   Float_t ox = fTarget[1]; 
+//   Float_t oy = fTarget[2]+460.; 
+//   Float_t oz = fTarget[0];
+//   gener->SetOrigin(ox, oy, oz);
+
+//   //make directory 
+//   Int_t nRootFile = (fEvent+1)/fMaxPerFile ;        
+//   if(nRootFile){
+//     char dname[20];
+//     sprintf(dname,"%d",nRootFile) ;
+//     if(gSystem->AccessPathName(dname)) //strange return: 0 if exists
+//       if(gSystem->MakeDirectory(dname)!=0)
+//     Fatal("StartRootFiles","Can not make directory %s \n",dname) ;
+    
+//     if(!gSystem->ChangeDirectory(dname))
+//       Fatal("StartRootFiles","Can not cd to %s\n",dname) ;
+//   }
+
+//   //  Create the output file
+//   TString outname("") ;
+//   if(strstr(GetTitle(),"root")){
+//     outname=GetTitle();
+//   }
+//   else{
+//     outname = fInName ;
+//     outname.ToLower() ;
+//     outname.ReplaceAll(".fz",".root") ;
+//   }
+
+//   fHeaderFile = new TFile(outname,"recreate");
+//   fHeaderFile->SetCompressionLevel(2);
+  
+//   // Create the Root Trees
+  
+//   gime->MakeTree("E") ;
+  
+//   //Fill now TreeE
+//   Int_t splitlevel = 0 ;
+//   Int_t bufferSize = 32000 ;    
+//   TBranch * headerBranch = gAlice->TreeE()->Branch("AliPHOSBeamTestEvent", 
+//                                                "AliPHOSBeamTestEvent", 
+//                                                &fPHOSHeader,bufferSize,splitlevel);
+//   headerBranch->SetName("AliPHOSBeamTestEvent") ;
+
+// //   if(fToSplit){
+// //     fDigitsFile = new TFile("PHOS.Digits.root","recreate") ;
+// //     fDigitsFile->SetCompressionLevel(2) ;
+// //   }
+   return kTRUE ;
+}
+//____________________________________________________________________________ 
+Bool_t AliPHOSRaw2Digits::CloseRootFiles(void ){
+  //cleans everething to start next root file
+  if(fHeaderFile){
+    printf("writing gAlice \n") ;
+    fHeaderFile->cd() ;
+    gAlice->Write(0,TObject::kOverwrite);
+    gAlice->TreeE()->Write(0,TObject::kOverwrite);
+  }
+
+  delete gAlice ;
+  
+  if(fHeaderFile){
+    fHeaderFile->Close() ;
+    delete fHeaderFile ;
+    fHeaderFile = 0;
+  }   
+       
+  if(fDigitsFile){
+    fDigitsFile->Close() ;
+    delete fDigitsFile ;
+    fDigitsFile = 0 ;
+  }
+  
+  Int_t nRootFile = (fEvent-1)/fMaxPerFile ;   
+  if(nRootFile){
+    if(!gSystem->ChangeDirectory("../")){
+     Fatal("CloseRootFile","Can not return to initial dir \n") ;      
+     return kFALSE ;
+    }
+  }
   return kTRUE ;
 }
 //____________________________________________________________________________ 
 Bool_t AliPHOSRaw2Digits::ProcessRawFile(){
-
+  
   //Method opens zebra file and reads successively bytes from it,
   //filling corresponding fields in header and digits.
-
-
+  
+  
   fStatus= -3 ;
   //First of all, open file and check if it is a zebra file
-
+  
   char command[256];
   sprintf(command,"zcat %s",fInName.Data());
   FILE *dataFile = popen(command, "r");
@@ -194,7 +333,6 @@ Bool_t AliPHOSRaw2Digits::ProcessRawFile(){
     fStatus = -1 ;
     return kFALSE ;
   }
-  printf("Open pipe: %s\n",command);
 
   // Check if byte ordering is little-endian 
   UInt_t w = 0x12345678;
@@ -276,7 +414,14 @@ Bool_t AliPHOSRaw2Digits::ProcessRawFile(){
 
     //    StartNewEvent() ;
     fDigits->Delete() ;
-    gAlice->SetEvent(fEvent) ;
+    if((fEvent%fMaxPerFile == 0) && fEvent ){
+      CloseRootFiles() ;
+      StartRootFiles() ;
+    }
+    gAlice->SetEvent(fEvent%fMaxPerFile) ;
+
+    //Set Beam Energy
+    fPHOSHeader->SetBeamEnergy(fBeamEnergy) ;
          
     Int_t i ;
     for(i=0;i<16;i++)
@@ -356,7 +501,7 @@ Bool_t AliPHOSRaw2Digits::ProcessRawFile(){
     UChar_t *byteptr=(UChar_t*)recptr;
     
     // Trigger bit register  
-    pattern=ntohs(*(UShort_t*)byteptr);
+    pattern=net2host(*(UShort_t*)byteptr);
     fPHOSHeader->SetPattern(pattern) ;
     byteptr+=sizeof(UShort_t);
     
@@ -364,7 +509,7 @@ Bool_t AliPHOSRaw2Digits::ProcessRawFile(){
     //or Kurchatov 64+2 channel ADC 
     //(the rest of the channels padded with 0xffff) 
     for(i=0;i<80;i++){
-      Int_t peak = static_cast<Int_t>(ntohs(*(UShort_t*)byteptr));
+      Int_t peak = static_cast<Int_t>(net2host(*(UShort_t*)byteptr));
       //make digit
       Int_t absID = fctdb->Raw2AbsId(i) ;
       if(absID > 0)
@@ -378,28 +523,28 @@ Bool_t AliPHOSRaw2Digits::ProcessRawFile(){
     
     // Scanning ADCs, 4 modulesX8=32 channels
     for(i=0;i<32;i++){
-      scanning[i]=ntohs(*(UShort_t*)byteptr);
+      scanning[i]=net2host(*(UShort_t*)byteptr);
       byteptr+=sizeof(UShort_t);
     }
     fPHOSHeader->SetScanning(scanning) ;
     
     // Charge ADCs, 1 moduleX12=12 channels
     for(i=0;i<12;i++){
-      charge[i]=ntohs(*(UShort_t*)byteptr);
+      charge[i]=net2host(*(UShort_t*)byteptr);
       byteptr+=sizeof(UShort_t);
     }
     fPHOSHeader->SetCharge(charge) ;
     
     // Scalers, 1 moduleX12=12 (4 byte) channels
     for(i=0;i<12;i++){
-      scaler[i]=ntohl(*(UInt_t*)byteptr);
+      scaler[i]=net2host(*(UInt_t*)byteptr);
       byteptr+=sizeof(UInt_t);
     }
     fPHOSHeader->SetScaler(scaler) ;
     
     // LeCroy TDC 2228A, 4 moduleX8=32 channels
     for(i=0;i<8;i++){
-      tdc2228[i]=ntohs(*(UShort_t*)byteptr);
+      tdc2228[i]=net2host(*(UShort_t*)byteptr);
       byteptr+=sizeof(UShort_t);
     }
     fPHOSHeader->SetTDC(tdc2228) ;
@@ -409,7 +554,7 @@ Bool_t AliPHOSRaw2Digits::ProcessRawFile(){
       Info("ProcessRawFile", "event= %d written", fEvent) ;
  
     // Read next record 
-    UInt_t nb = nwphr *sizeof(UInt_t);
+    nb = nwphr *sizeof(UInt_t);
     n = fread( recBuf, nb,1,dataFile);
     if (n < 0 ){
       perror(fInName);
@@ -423,12 +568,15 @@ Bool_t AliPHOSRaw2Digits::ProcessRawFile(){
     }
     fEvent++ ;
   }
+  CloseRootFiles() ;
   
   fStatus = 1 ;  
   return kTRUE ;  
 }
+
 //____________________________________________________________________________ 
-void AliPHOSRaw2Digits::Swab4(void *from, void *to, size_t nwords){
+void AliPHOSRaw2Digits::Swab4(void *from, void *to, size_t nwords)const
+{
   // The function swaps 4 bytes: byte#3<-->byte#0, byte#2<-->byte#1 
   register char *pf=static_cast<char*>(from) ;
   register char *pt=static_cast<char*>(to) ;
@@ -446,8 +594,9 @@ void AliPHOSRaw2Digits::Swab4(void *from, void *to, size_t nwords){
 }
 
 //____________________________________________________________________________ 
-void AliPHOSRaw2Digits::Swab2(void *from, void *to, size_t nwords)
-{ //The function swaps 2x2 bytes: byte#0<-->byte#1, byte#2<-->byte#3 
+void AliPHOSRaw2Digits::Swab2(void *from, void *to, size_t nwords)const
+{ 
+  //The function swaps 2x2 bytes: byte#0<-->byte#1, byte#2<-->byte#3 
   register char *pf=static_cast<char*>(from) ;
   register char *pt=static_cast<char*>(to);
   register char c;   
@@ -463,13 +612,6 @@ void AliPHOSRaw2Digits::Swab2(void *from, void *to, size_t nwords)
   }
 }
 
-//____________________________________________________________________________ 
-void AliPHOSRaw2Digits::FinishRun(){
-  //Write geometry and header tree
-  gAlice->Write(0,TObject::kOverwrite);
-  gAlice->TreeE()->Write(0,TObject::kOverwrite);
-  
-}
 //____________________________________________________________________________ 
 void AliPHOSRaw2Digits::WriteDigits(void){
   //In this method we create TreeD, write digits and Raw2Digits to it
@@ -482,7 +624,7 @@ void AliPHOSRaw2Digits::WriteDigits(void){
     static_cast<AliPHOSDigit*>(fDigits->At(i))->SetIndexInList(i) ;
 
   char hname[30];
-  sprintf(hname,"TreeD%d",fEvent);
+  sprintf(hname,"TreeD%d",fEvent%fMaxPerFile);
   TTree * treeD = new TTree(hname,"Digits");
   //treeD->Write(0,TObject::kOverwrite);
   
@@ -497,7 +639,9 @@ void AliPHOSRaw2Digits::WriteDigits(void){
   TBranch * digitizerBranch = treeD->Branch("AliPHOSRaw2Digits", 
                                            "AliPHOSRaw2Digits", &d,bufferSize,splitlevel); 
   digitizerBranch->SetTitle("Default");
-  
+
+  if(fDigitsFile)
+    fDigitsFile->cd() ;
   digitsBranch->Fill() ;
   digitizerBranch->Fill() ; 
   treeD->Write(0,TObject::kOverwrite);
@@ -505,39 +649,34 @@ void AliPHOSRaw2Digits::WriteDigits(void){
   delete treeD ;
 
   //Write header
+  fHeaderFile->cd() ;
   gAlice->TreeE()->Fill();
 }
 //____________________________________________________________________________ 
-void AliPHOSRaw2Digits::Print(Option_t * option)const{
-  
-  TString message ;
-  message  = "----------AliPHOSRaw2Digits---------- \n" ;
-  message += "Input stream \n" ;
-  message += "Current input  File: %s\n" ; 
-  message += "Current output File: %s\n" ; 
-  message += "Events processes in the last file %d\n" ; 
-  message += "Input file status\n" ;
+void AliPHOSRaw2Digits::Print(const Option_t *)const{
+  //prints current configuration and status.
+
+  printf("----------AliPHOSRaw2Digits---------- \n") ;
+  printf("Current input  File: %s\n",fInName.Data()) ; 
+  printf("Current output File: %s\n", GetTitle()); 
+  printf("Events processes in the last file %d\n",fEvent) ; 
+  printf("Input file status\n") ;
   switch (fStatus){
-  case 0: message += "`Have not processed yet'\n" ;
+  case 0: printf("`Have not processed yet'\n") ;
     break ;
-  case 1: message += "`Processed normally'\n" ;
+  case 1: printf("`Processed normally'\n") ;
     break ;
-  case -1: message += "`File not found'\n" ;
+  case -1: printf("`File not found'\n") ;
     break ;
-  case -2: message += "`Error in reading'\n" ;
+  case -2: printf("`Error in reading'\n") ;
     break ;
-  case -3: message += "'Interupted'\n" ;
+  case -3: printf("'Interupted'\n") ;
   default: ;
   }
-  message += "Connection table: " ;
+  printf("Connection table: " );
   if(fctdb)
-    message += "%s %s \n" ; 
+    printf("%s %s \n",fctdb->GetName(), fctdb->GetTitle() ) ; 
   else
-    message += " no DB \n" ;
+    printf(" no DB \n" );
 
-  Info("Print", message.Data(),  
-       fInName.Data(), 
-       GetTitle(), 
-       fEvent, 
-       fctdb->GetName(), fctdb->GetTitle() ) ; 
 }