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