VZERO/AliVZEROReconstructor.cxx

   1 /**************************************************************************
   2  * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
   3  *                                                                        *
   4  * Author: The ALICE Off-line Project.                                    *
   5  * Contributors are mentioned in the code where appropriate.              *
   6  *                                                                        *
   7  * Permission to use, copy, modify and distribute this software and its   *
   8  * documentation strictly for non-commercial purposes is hereby granted   *
   9  * without fee, provided that the above copyright notice appears in all   *
  10  * copies and that both the copyright notice and this permission notice   *
  11  * appear in the supporting documentation. The authors make no claims     *
  12  * about the suitability of this software for any purpose. It is          *
  13  * provided "as is" without express or implied warranty.                  *
  14  **************************************************************************/
  15
  16 /* $Id$ */
  17
  18 ///////////////////////////////////////////////////////////////////////////////
  19 ///                                                                          //
  20 /// class for VZERO reconstruction                                           //
  21 ///                                                                          //
  22 ///////////////////////////////////////////////////////////////////////////////
  23
  24 #include "AliRunLoader.h"
  25 #include "AliRawReader.h"
  26 #include "AliGRPObject.h"
  27 #include "AliCDBManager.h"
  28 #include "AliCDBStorage.h"
  29 #include "AliCDBEntry.h"
  30 #include "AliVZEROReconstructor.h"
  31 #include "AliVZERORawStream.h"
  32 #include "AliESDEvent.h"
  33 #include "AliVZEROTriggerMask.h"
  34 #include "AliESDfriend.h"
  35 #include "AliESDVZEROfriend.h"
  36 #include "AliVZEROdigit.h"
  37 #include "AliVZEROCalibData.h"
  38 #include "AliVZEROTriggerData.h"
  39
  40 ClassImp(AliVZEROReconstructor)
  41
  42 //_____________________________________________________________________________
  43 AliVZEROReconstructor:: AliVZEROReconstructor(): AliReconstructor(),
  44                         fESDVZERO(0x0),
  45                         fESD(0x0),
  46                         fESDVZEROfriend(0x0),
  47                         fCalibData(GetCalibData()),
  48                         fTriggerData(GetTriggerData()),
  49                         fCollisionMode(0),
  50                         fBeamEnergy(0.)
  51 {
  52   // Default constructor
  53   // Get calibration data
  54
  55   // fCalibData = GetCalibData();
  56
  57 }
  58
  59
  60 //_____________________________________________________________________________
  61 AliVZEROReconstructor& AliVZEROReconstructor::operator =
  62   (const AliVZEROReconstructor& /*reconstructor*/)
  63 {
  64 // assignment operator
  65
  66   Fatal("operator =", "assignment operator not implemented");
  67   return *this;
  68 }
  69
  70 //_____________________________________________________________________________
  71 AliVZEROReconstructor::~AliVZEROReconstructor()
  72 {
  73 // destructor
  74
  75    delete fESDVZERO;
  76    delete fESDVZEROfriend;
  77 }
  78
  79 //_____________________________________________________________________________
  80 void AliVZEROReconstructor::Init()
  81 {
  82 // initializer
  83
  84   fESDVZERO  = new AliESDVZERO;
  85   fESDVZEROfriend = new AliESDVZEROfriend;
  86
  87   GetCollisionMode();  // fCollisionMode =1 for Pb-Pb simulated data
  88 }
  89
  90 //______________________________________________________________________
  91 void AliVZEROReconstructor::ConvertDigits(AliRawReader* rawReader, TTree* digitsTree) const
  92 {
  93 // converts RAW to digits - pedestal is subtracted
  94
  95   if (!digitsTree) {
  96     AliError("No digits tree!");
  97     return;
  98   }
  99
 100   TClonesArray* digitsArray = new TClonesArray("AliVZEROdigit");
 101   digitsTree->Branch("VZERODigit", &digitsArray);
 102
 103   fESDVZEROfriend->Reset();
 104
 105   rawReader->Reset();
 106   AliVZERORawStream rawStream(rawReader);
 107   if (rawStream.Next()) {
 108      Float_t ADC_max[64], adc[64];
 109      Bool_t BBFlag[64], BGFlag[64], integrator[64];
 110      Float_t time[64], width[64];
 111      for(Int_t i=0; i<64; i++) {
 112          // Search for the maximum charge in the train of 21 LHC clocks
 113          // regardless of the integrator which has been operated:
 114          ADC_max[i] = 0.0;
 115          Int_t imax = 0;
 116          for(Int_t iClock=0; iClock<21; iClock++){
 117              if(rawStream.GetPedestal(i,iClock) > ADC_max[i])
 118                 {ADC_max[i]= rawStream.GetPedestal(i,iClock);
 119                  imax      = iClock;}
 120          }
 121          // Convert i (FEE channel numbering) to j (aliroot channel numbering)
 122          Int_t j       =  rawStream.GetOfflineChannel(i);
 123          Int_t board   = j / 8;
 124          adc[j]        =  ADC_max[i];
 125          time[j]       =  rawStream.GetTime(i)/ (25./256.) * fTriggerData->GetTimeResolution(board);
 126          width[j]      =  rawStream.GetWidth(i) / 0.4 * fTriggerData->GetWidthResolution(board);
 127          BBFlag[j]     =  rawStream.GetBBFlag(i,imax);
 128          BGFlag[j]     =  rawStream.GetBGFlag(i,imax);
 129          integrator[j] =  rawStream.GetIntegratorFlag(i,imax);
 130
 131          // Filling the esd friend object
 132          fESDVZEROfriend->SetBBScalers(j,rawStream.GetBBScalers(i));
 133          fESDVZEROfriend->SetBGScalers(j,rawStream.GetBGScalers(i));
 134          for (Int_t iBunch = 0; iBunch < AliESDVZEROfriend::kNBunches; iBunch++) {
 135              fESDVZEROfriend->SetChargeMB(j,iBunch,rawStream.GetChargeMB(i,iBunch));
 136              fESDVZEROfriend->SetIntMBFlag(j,iBunch,rawStream.GetIntMBFlag(i,iBunch));
 137              fESDVZEROfriend->SetBBMBFlag(j,iBunch,rawStream.GetBBMBFlag(i,iBunch));
 138              fESDVZEROfriend->SetBGMBFlag(j,iBunch,rawStream.GetBGMBFlag(i,iBunch));
 139          }
 140          for (Int_t iEv = 0; iEv < AliESDVZEROfriend::kNEvOfInt; iEv++) {
 141              fESDVZEROfriend->SetPedestal(j,iEv,rawStream.GetPedestal(i,iEv));
 142              fESDVZEROfriend->SetIntegratorFlag(j,iEv,rawStream.GetIntegratorFlag(i,iEv));
 143              fESDVZEROfriend->SetBBFlag(j,iEv,rawStream.GetBBFlag(i,iEv));
 144              fESDVZEROfriend->SetBGFlag(j,iEv,rawStream.GetBGFlag(i,iEv));
 145          }
 146          fESDVZEROfriend->SetTime(j,time[j]);
 147          fESDVZEROfriend->SetWidth(j,width[j]);
 148      }
 149
 150      // Filling the esd friend object
 151      fESDVZEROfriend->SetTriggerInputs(rawStream.GetTriggerInputs());
 152      fESDVZEROfriend->SetTriggerInputsMask(rawStream.GetTriggerInputsMask());
 153
 154      for(Int_t iScaler = 0; iScaler < AliESDVZEROfriend::kNScalers; iScaler++)
 155          fESDVZEROfriend->SetTriggerScalers(iScaler,rawStream.GetTriggerScalers(iScaler));
 156
 157      for(Int_t iBunch = 0; iBunch < AliESDVZEROfriend::kNBunches; iBunch++)
 158          fESDVZEROfriend->SetBunchNumbersMB(iBunch,rawStream.GetBunchNumbersMB(iBunch));
 159
 160
 161      // Channels(aliroot numbering) will be ordered in the tree
 162      for(Int_t iChannel = 0; iChannel < 64; iChannel++) {
 163          new ((*digitsArray)[digitsArray->GetEntriesFast()])
 164              AliVZEROdigit(iChannel, adc[iChannel], time[iChannel],
 165                            width[iChannel], BBFlag[iChannel], BGFlag[iChannel],integrator[iChannel]);
 166      }
 167      digitsTree->Fill();
 168   }
 169 }
 170
 171 //______________________________________________________________________
 172 void AliVZEROReconstructor::FillESD(TTree* digitsTree, TTree* /*clustersTree*/,
 173                                     AliESDEvent* esd) const
 174 {
 175 // fills multiplicities to the ESD
 176
 177   if (!digitsTree) {
 178       AliError("No digits tree!");
 179       return;
 180   }
 181
 182   TClonesArray* digitsArray = NULL;
 183   TBranch* digitBranch = digitsTree->GetBranch("VZERODigit");
 184   digitBranch->SetAddress(&digitsArray);
 185
 186   Float_t   mult[64];
 187   Float_t    adc[64];
 188   Float_t   time[64];
 189   Float_t  width[64];
 190   Bool_t  BBFlag[64];
 191   Bool_t  BGFlag[64];
 192
 193   for (Int_t i=0; i<64; i++){
 194        adc[i]    = 0.0;
 195        mult[i]   = 0.0;
 196        time[i]   = 0.0;
 197        width[i]  = 0.0;
 198        BBFlag[i] = kFALSE;
 199        BGFlag[i] = kFALSE;
 200   }
 201
 202   // loop over VZERO entries to get multiplicity
 203   Int_t nEntries = (Int_t)digitsTree->GetEntries();
 204   for (Int_t e=0; e<nEntries; e++) {
 205     digitsTree->GetEvent(e);
 206
 207     Int_t nDigits = digitsArray->GetEntriesFast();
 208
 209     for (Int_t d=0; d<nDigits; d++) {
 210         AliVZEROdigit* digit = (AliVZEROdigit*)digitsArray->At(d);
 211         Int_t  pmNumber      = digit->PMNumber();
 212         // Pedestal retrieval and suppression:
 213         Float_t  pedestal      = fCalibData->GetPedestal(d);
 214         adc[pmNumber]   =  digit->ADC() - pedestal;
 215         time[pmNumber]  =  digit->Time();
 216         width[pmNumber] =  digit->Width();
 217         BBFlag[pmNumber]=  digit->BBFlag();
 218         BGFlag[pmNumber]=  digit->BGFlag();
 219         // printf("PM = %d,  MIP per ADC channel = %f \n",pmNumber, fCalibData->GetMIPperADC(pmNumber));
 220                 //AliInfo(Form("PM = %d,  ADC = %f TDC %f",pmNumber, digit->ADC(),digit->Time()));
 221         // cut of ADC at 1MIP/2
 222         if(fCollisionMode >0) {
 223            Float_t MIP = 2.0;
 224            if (adc[pmNumber] > (MIP/2.) ) mult[pmNumber] += adc[pmNumber]*(1.0/MIP) ;
 225            }
 226            else{
 227            if (adc[pmNumber] > (1.0/fCalibData->GetMIPperADC(pmNumber) /2.) )
 228                mult[pmNumber] += adc[pmNumber]*fCalibData->GetMIPperADC(pmNumber);
 229         }
 230     } // end of loop over digits
 231   } // end of loop over events in digits tree
 232
 233   fESDVZERO->SetMultiplicity(mult);
 234   fESDVZERO->SetADC(adc);
 235   fESDVZERO->SetTime(time);
 236   fESDVZERO->SetWidth(width);
 237   fESDVZERO->SetBBFlag(BBFlag);
 238   fESDVZERO->SetBGFlag(BGFlag);
 239
 240   // now get the trigger mask
 241
 242   AliVZEROTriggerMask *TriggerMask = new AliVZEROTriggerMask();
 243   TriggerMask->SetAdcThreshold(10.0/2.0);
 244   TriggerMask->SetTimeWindowWidthBBA(50);
 245   TriggerMask->SetTimeWindowWidthBGA(20);
 246   TriggerMask->SetTimeWindowWidthBBC(50);
 247   TriggerMask->SetTimeWindowWidthBGC(20);
 248   TriggerMask->FillMasks(digitsTree,digitsArray);
 249
 250   fESDVZERO->SetBBtriggerV0A(TriggerMask->GetBBtriggerV0A());
 251   fESDVZERO->SetBGtriggerV0A(TriggerMask->GetBGtriggerV0A());
 252   fESDVZERO->SetBBtriggerV0C(TriggerMask->GetBBtriggerV0C());
 253   fESDVZERO->SetBGtriggerV0C(TriggerMask->GetBGtriggerV0C());
 254
 255   if (esd) {
 256      AliDebug(1, Form("Writing VZERO data to ESD tree"));
 257      esd->SetVZEROData(fESDVZERO);
 258   }
 259
 260   if (esd) {
 261      AliESDfriend *fr = (AliESDfriend*)esd->FindListObject("AliESDfriend");
 262      if (fr) {
 263         AliDebug(1, Form("Writing VZERO friend data to ESD tree"));
 264         fr->SetVZEROfriend(fESDVZEROfriend);
 265     }
 266   }
 267 }
 268
 269 //_____________________________________________________________________________
 270 AliCDBStorage* AliVZEROReconstructor::SetStorage(const char *uri)
 271 {
 272 // Sets the storage
 273
 274   Bool_t deleteManager = kFALSE;
 275
 276   AliCDBManager *manager = AliCDBManager::Instance();
 277   AliCDBStorage *defstorage = manager->GetDefaultStorage();
 278
 279   if(!defstorage || !(defstorage->Contains("VZERO"))){
 280      AliWarning("No default storage set or default storage doesn't contain VZERO!");
 281      manager->SetDefaultStorage(uri);
 282      deleteManager = kTRUE;
 283   }
 284
 285   AliCDBStorage *storage = manager->GetDefaultStorage();
 286
 287   if(deleteManager){
 288      AliCDBManager::Instance()->UnsetDefaultStorage();
 289      defstorage = 0;   // the storage is killed by AliCDBManager::Instance()->Destroy()
 290   }
 291
 292   return storage;
 293 }
 294
 295 //____________________________________________________________________________
 296 void AliVZEROReconstructor::GetCollisionMode()
 297 {
 298 // Retrieves the collision mode from GRP data
 299
 300 // Initialization of the GRP entry
 301 // To be replace by a call to AliRunInfo object
 302
 303   AliCDBEntry*  entry = AliCDBManager::Instance()->Get("GRP/GRP/Data");
 304   AliGRPObject* grpData = 0x0;
 305
 306   if(entry){
 307     TMap* m = dynamic_cast<TMap*>(entry->GetObject());  // old GRP entry
 308     if(m){
 309        grpData = new AliGRPObject();
 310        grpData->ReadValuesFromMap(m);
 311     }
 312     else{
 313        grpData = dynamic_cast<AliGRPObject*>(entry->GetObject());  // new GRP entry
 314     }
 315   }
 316
 317   if(!grpData) { AliError("No GRP entry found in OCDB!");
 318                   return; }
 319
 320 // Retrieval of simulated collision mode
 321
 322   TString beamType = grpData->GetBeamType();
 323   if(beamType==AliGRPObject::GetInvalidString()){
 324      AliError("GRP/GRP/Data entry:  missing value for the beam type !");
 325      AliError("\t VZERO cannot retrieve beam type\n");
 326      return;
 327   }
 328
 329    if( (beamType.CompareTo("P-P") ==0)  || (beamType.CompareTo("p-p") ==0) ){
 330        fCollisionMode=0;
 331   }
 332    else if( (beamType.CompareTo("Pb-Pb") ==0)  || (beamType.CompareTo("A-A") ==0) ){
 333        fCollisionMode=1;
 334    }
 335
 336   fBeamEnergy = grpData->GetBeamEnergy();
 337   if(fBeamEnergy==AliGRPObject::GetInvalidFloat()) {
 338      AliError("GRP/GRP/Data entry:  missing value for the beam energy ! Using 0");
 339      fBeamEnergy = 0.;
 340   }
 341
 342 //    printf("\n ++++++ Beam type and collision mode retrieved as %s %d @ %1.3f GeV ++++++\n\n",beamType.Data(), fCollisionMode, fBeamEnergy);
 343
 344 }
 345
 346 //_____________________________________________________________________________
 347 AliVZEROCalibData* AliVZEROReconstructor::GetCalibData() const
 348 {
 349   // Gets calibration object for VZERO set
 350
 351   AliCDBManager *man = AliCDBManager::Instance();
 352
 353   AliCDBEntry *entry=0;
 354
 355   entry = man->Get("VZERO/Calib/Data");
 356
 357 //   if(!entry){
 358 //     AliWarning("Load of calibration data from default storage failed!");
 359 //     AliWarning("Calibration data will be loaded from local storage ($ALICE_ROOT)");
 360 //     Int_t runNumber = man->GetRun();
 361 //     entry = man->GetStorage("local://$ALICE_ROOT/OCDB")
 362 //       ->Get("VZERO/Calib/Data",runNumber);
 363 //
 364 //   }
 365
 366   // Retrieval of data in directory VZERO/Calib/Data:
 367
 368   AliVZEROCalibData *calibdata = 0;
 369
 370   if (entry) calibdata = (AliVZEROCalibData*) entry->GetObject();
 371   if (!calibdata)  AliFatal("No calibration data from calibration database !");
 372
 373   return calibdata;
 374 }
 375
 376 //_____________________________________________________________________________
 377 AliVZEROTriggerData* AliVZEROReconstructor::GetTriggerData() const
 378 {
 379   // Gets calibration object for VZERO set
 380
 381   AliCDBManager *man = AliCDBManager::Instance();
 382
 383   AliCDBEntry *entry=0;
 384
 385   entry = man->Get("VZERO/Trigger/Data");
 386
 387 //   if(!entry){
 388 //     AliWarning("Load of calibration data from default storage failed!");
 389 //     AliWarning("Calibration data will be loaded from local storage ($ALICE_ROOT)");
 390 //     Int_t runNumber = man->GetRun();
 391 //     entry = man->GetStorage("local://$ALICE_ROOT/OCDB")
 392 //       ->Get("VZERO/Calib/Data",runNumber);
 393 //
 394 //   }
 395
 396   // Retrieval of data in directory VZERO/Calib/Data:
 397
 398   AliVZEROTriggerData *triggerdata = 0;
 399
 400   if (entry) triggerdata = (AliVZEROTriggerData*) entry->GetObject();
 401   if (!triggerdata)  AliFatal("No trigger data from calibration database !");
 402
 403   return triggerdata;
 404 }