]> git.uio.no Git - u/mrichter/AliRoot.git/blobdiff - VZERO/AliVZEROReconstructor.cxx
update
[u/mrichter/AliRoot.git] / VZERO / AliVZEROReconstructor.cxx
index d06422a0ea45fb2f2fcbdbb42438397682e80042..2fca0e2e3b6433eb9f38be888b0599f5a567a165 100644 (file)
 ///                                                                          //
 ///////////////////////////////////////////////////////////////////////////////
 
-
+#include "AliRunLoader.h"
+#include "AliRawReader.h"
 #include "AliVZEROReconstructor.h"
+#include "AliVZERORawStream.h"
+#include "AliESDEvent.h"
+#include "AliVZEROTriggerMask.h"
+#include "AliESDfriend.h"
+#include "AliESDVZEROfriend.h"
 
 ClassImp(AliVZEROReconstructor)
 
 //_____________________________________________________________________________
-AliVZEROReconstructor:: AliVZEROReconstructor()
+AliVZEROReconstructor:: AliVZEROReconstructor(): AliReconstructor(),
+   fESDVZERO(0x0),
+   fESD(0x0),
+   fESDVZEROfriend(0x0),
+   fCalibData(GetCalibData())
 {
   // Default constructor  
-  
   // Get calibration data
   
-  fCalibData = GetCalibData(); 
+  // fCalibData = GetCalibData(); 
+
 }
 
 
@@ -52,13 +62,196 @@ AliVZEROReconstructor::~AliVZEROReconstructor()
 {
 // destructor
 
+   delete fESDVZERO;
+   delete fESDVZEROfriend;
+}
+
+//_____________________________________________________________________________
+void AliVZEROReconstructor::Init()
+{
+// initializer
+
+  fESDVZERO  = new AliESDVZERO;
+  fESDVZEROfriend = new AliESDVZEROfriend;
+}
+
+//______________________________________________________________________
+void AliVZEROReconstructor::ConvertDigits(AliRawReader* rawReader, TTree* digitsTree) const
+{
+// converts RAW to digits - pedestal is subtracted 
+
+  if (!digitsTree) {
+    AliError("No digits tree!");
+    return;
+  }
+
+  TClonesArray* digitsArray = new TClonesArray("AliVZEROdigit");
+  digitsTree->Branch("VZERODigit", &digitsArray);
+
+  fESDVZEROfriend->Reset();
+
+  rawReader->Reset();
+  AliVZERORawStream rawStream(rawReader);
+  if (rawStream.Next()) {  
+     Int_t ADC_max[64], adc[64], time[64], width[64], BBFlag[64], BGFlag[64];   
+     for(Int_t i=0; i<64; i++) {
+         // Search for the maximum charge in the train of 21 LHC clocks 
+         // regardless of the integrator which has been operated:
+         ADC_max[i] = 0;
+        Int_t imax = 0;
+         for(Int_t iClock=0; iClock<21; iClock++){
+             if((Int_t)rawStream.GetPedestal(i,iClock) > ADC_max[i])  
+               {ADC_max[i]=(Int_t)rawStream.GetPedestal(i,iClock);
+                imax      = iClock;}
+         }
+        // Convert i (FEE channel numbering) to j (aliroot channel numbering)
+        Int_t j   =  rawStream.GetOfflineChannel(i);
+        adc[j]    =  ADC_max[i];
+        time[j]   =  rawStream.GetTime(i);
+        width[j]  =  rawStream.GetWidth(i);
+        BBFlag[j] =  rawStream.GetBBFlag(i,imax);
+        BGFlag[j] =  rawStream.GetBGFlag(i,imax); 
+
+        // Filling the esd friend object
+        fESDVZEROfriend->SetBBScalers(j,rawStream.GetBBScalers(i));
+        fESDVZEROfriend->SetBGScalers(j,rawStream.GetBGScalers(i));
+        for (Int_t iBunch = 0; iBunch < AliESDVZEROfriend::kNBunches; iBunch++) {
+          fESDVZEROfriend->SetChargeMB(j,iBunch,rawStream.GetChargeMB(i,iBunch));
+          fESDVZEROfriend->SetIntMBFlag(j,iBunch,rawStream.GetIntMBFlag(i,iBunch));
+          fESDVZEROfriend->SetBBMBFlag(j,iBunch,rawStream.GetBBMBFlag(i,iBunch));
+          fESDVZEROfriend->SetBGMBFlag(j,iBunch,rawStream.GetBGMBFlag(i,iBunch));
+        }
+        for (Int_t iEv = 0; iEv < AliESDVZEROfriend::kNEvOfInt; iEv++) {
+          fESDVZEROfriend->SetPedestal(j,iEv,rawStream.GetPedestal(i,iEv));
+          fESDVZEROfriend->SetIntegratorFlag(j,iEv,rawStream.GetIntegratorFlag(i,iEv));
+          fESDVZEROfriend->SetBBFlag(j,iEv,rawStream.GetBBFlag(i,iEv));
+          fESDVZEROfriend->SetBGFlag(j,iEv,rawStream.GetBGFlag(i,iEv));
+        }
+        fESDVZEROfriend->SetTime(j,rawStream.GetTime(i));
+        fESDVZEROfriend->SetWidth(j,rawStream.GetWidth(i));
+     }  
+
+     // Filling the esd friend object
+     fESDVZEROfriend->SetTriggerInputs(rawStream.GetTriggerInputs());
+     fESDVZEROfriend->SetTriggerInputsMask(rawStream.GetTriggerInputsMask());
+
+     for(Int_t iScaler = 0; iScaler < AliESDVZEROfriend::kNScalers; iScaler++)
+       fESDVZEROfriend->SetTriggerScalers(iScaler,rawStream.GetTriggerScalers(iScaler));
+
+     for (Int_t iBunch = 0; iBunch < AliESDVZEROfriend::kNBunches; iBunch++)
+       fESDVZEROfriend->SetBunchNumbersMB(iBunch,rawStream.GetBunchNumbersMB(iBunch));
+     
+
+     // Channels(aliroot numbering) will be ordered in the tree
+     for(Int_t iChannel = 0; iChannel < 64; iChannel++) {
+         new ((*digitsArray)[digitsArray->GetEntriesFast()])
+             AliVZEROdigit(iChannel, adc[iChannel], time[iChannel],
+                          width[iChannel], BBFlag[iChannel], BGFlag[iChannel]);
+     }
+  }
+
+  digitsTree->Fill();
 }
 
+//______________________________________________________________________
+void AliVZEROReconstructor::FillESD(TTree* digitsTree, TTree* /*clustersTree*/,
+                                   AliESDEvent* esd) const
+{
+// fills multiplicities to the ESD
+
+  if (!digitsTree) {
+    AliError("No digits tree!");
+    return;
+  }
+
+  TClonesArray* digitsArray = NULL;
+  TBranch* digitBranch = digitsTree->GetBranch("VZERODigit");
+  digitBranch->SetAddress(&digitsArray);
+
+  Short_t Multiplicity[64];
+  Float_t   mult[64];  
+  Short_t    adc[64]; 
+  Short_t   time[64]; 
+  Short_t  width[64];
+  Bool_t  BBFlag[64];
+  Bool_t  BGFlag[64];
+   
+  for (Int_t i=0; i<64; i++){
+       adc[i]    = 0;
+       mult[i]   = 0.0;
+       time[i]   = 0;
+       width[i]  = 0;
+       BBFlag[i] = kFALSE;
+       BGFlag[i] = kFALSE;
+  }
+     
+  // loop over VZERO entries to get multiplicity
+  Int_t nEntries = (Int_t)digitsTree->GetEntries();
+  for (Int_t e=0; e<nEntries; e++) {
+    digitsTree->GetEvent(e);
+
+    Int_t nDigits = digitsArray->GetEntriesFast();
+    
+    for (Int_t d=0; d<nDigits; d++) {    
+        AliVZEROdigit* digit = (AliVZEROdigit*)digitsArray->At(d);      
+        Int_t  pmNumber      = digit->PMNumber(); 
+        // Pedestal retrieval and suppression: 
+        Int_t  pedestal      = int(fCalibData->GetPedestal(d));
+        adc[pmNumber]   = (Short_t) digit->ADC() - pedestal; 
+        time[pmNumber]  = (Short_t) digit->Time();
+               width[pmNumber] = (Short_t) digit->Width();
+               BBFlag[pmNumber]= digit->BBFlag();
+               BGFlag[pmNumber]= digit->BGFlag();
+        // printf("PM = %d,  MIP per ADC channel = %f \n",pmNumber, fCalibData->GetMIPperADC(pmNumber));
+        // cut of ADC at 1MIP/2 
+        if (adc[pmNumber] > (int(1.0/fCalibData->GetMIPperADC(pmNumber)) /2) ) 
+           mult[pmNumber] += float(adc[pmNumber])*fCalibData->GetMIPperADC(pmNumber);
+    } // end of loop over digits
+  } // end of loop over events in digits tree
+  
+  for (Int_t j=0; j<64; j++) Multiplicity[j] = short(mult[j]+0.5); 
+        
+  fESDVZERO->SetMultiplicity(Multiplicity);
+  fESDVZERO->SetADC(adc);
+  fESDVZERO->SetTime(time);
+  fESDVZERO->SetWidth(width);
+  fESDVZERO->SetBBFlag(BBFlag);
+  fESDVZERO->SetBGFlag(BGFlag);
+
+  // now get the trigger mask
+
+  AliVZEROTriggerMask *TriggerMask = new AliVZEROTriggerMask();
+  TriggerMask->SetAdcThreshold(20.0/2.0);
+  TriggerMask->SetTimeWindowWidthBBA(50);
+  TriggerMask->SetTimeWindowWidthBGA(20);
+  TriggerMask->SetTimeWindowWidthBBC(50);
+  TriggerMask->SetTimeWindowWidthBGC(20);
+  TriggerMask->FillMasks(digitsTree,digitsArray);
+
+  fESDVZERO->SetBBtriggerV0A(TriggerMask->GetBBtriggerV0A());
+  fESDVZERO->SetBGtriggerV0A(TriggerMask->GetBGtriggerV0A());
+  fESDVZERO->SetBBtriggerV0C(TriggerMask->GetBBtriggerV0C());
+  fESDVZERO->SetBGtriggerV0C(TriggerMask->GetBGtriggerV0C());
   
+  if (esd) { 
+     AliDebug(1, Form("Writing VZERO data to ESD tree"));
+     esd->SetVZEROData(fESDVZERO);
+  }
+
+  if (esd) {
+    AliESDfriend *fr = (AliESDfriend*)esd->FindListObject("AliESDfriend");
+    if (fr) {
+      AliDebug(1, Form("Writing VZERO friend data to ESD tree"));
+      fr->SetVZEROfriend(fESDVZEROfriend);
+    }
+  }
+}
+
 //_____________________________________________________________________________
 AliCDBStorage* AliVZEROReconstructor::SetStorage(const char *uri) 
 {
-  
+// Sets the storage  
+
   Bool_t deleteManager = kFALSE;
   
   AliCDBManager *manager = AliCDBManager::Instance();
@@ -73,8 +266,8 @@ AliCDBStorage* AliVZEROReconstructor::SetStorage(const char *uri)
   AliCDBStorage *storage = manager->GetDefaultStorage();
 
   if(deleteManager){
-    AliCDBManager::Instance()->UnsetDefaultStorage();
-    defstorage = 0;   // the storage is killed by AliCDBManager::Instance()->Destroy()
+     AliCDBManager::Instance()->UnsetDefaultStorage();
+     defstorage = 0;   // the storage is killed by AliCDBManager::Instance()->Destroy()
   }
 
   return storage; 
@@ -83,8 +276,7 @@ AliCDBStorage* AliVZEROReconstructor::SetStorage(const char *uri)
 //_____________________________________________________________________________
 AliVZEROCalibData* AliVZEROReconstructor::GetCalibData() const
 {
-
-  // Getting calibration object for VZERO set
+  // Gets calibration object for VZERO set
 
   AliCDBManager *man = AliCDBManager::Instance();
 
@@ -92,22 +284,21 @@ AliVZEROCalibData* AliVZEROReconstructor::GetCalibData() const
 
   entry = man->Get("VZERO/Calib/Data");
 
-  if(!entry){
-    AliWarning("Load of calibration data from default storage failed!");
-    AliWarning("Calibration data will be loaded from local storage ($ALICE_ROOT)");
-    Int_t runNumber = man->GetRun();
-    entry = man->GetStorage("local://$ALICE_ROOT")
-      ->Get("VZERO/Calib/Data",runNumber);
-       
-  }
+//   if(!entry){
+//     AliWarning("Load of calibration data from default storage failed!");
+//     AliWarning("Calibration data will be loaded from local storage ($ALICE_ROOT)");
+//     Int_t runNumber = man->GetRun();
+//     entry = man->GetStorage("local://$ALICE_ROOT/OCDB")
+//       ->Get("VZERO/Calib/Data",runNumber);
+//     
+//   }
 
   // Retrieval of data in directory VZERO/Calib/Data:
 
-
   AliVZEROCalibData *calibdata = 0;
 
   if (entry) calibdata = (AliVZEROCalibData*) entry->GetObject();
-  if (!calibdata)  AliError("No calibration data from calibration database !");
+  if (!calibdata)  AliFatal("No calibration data from calibration database !");
 
   return calibdata;
 }