STEER/AliRun.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 //  Control class for Alice C++                                              //
  21 //  Only one single instance of this class exists.                           //
  22 //  The object is created in main program aliroot                            //
  23 //  and is pointed by the global gAlice.                                     //
  24 //                                                                           //
  25 //   -Supports the list of all Alice Detectors (fModules).                 //
  26 //   -Supports the list of particles (fParticles).                           //
  27 //   -Supports the Trees.                                                    //
  28 //   -Supports the geometry.                                                 //
  29 //   -Supports the event display.                                            //
  30 //Begin_Html
  31 /*
  32 <img src="picts/AliRunClass.gif">
  33 */
  34 //End_Html
  35 //Begin_Html
  36 /*
  37 <img src="picts/alirun.gif">
  38 */
  39 //End_Html
  40 //                                                                           //
  41 ///////////////////////////////////////////////////////////////////////////////
  42
  43 #include <TCint.h>
  44 #include <TROOT.h>
  45 #include <TRandom3.h>
  46 #include <TSystem.h>
  47 #include <TVirtualMC.h>
  48 #include <TGeoManager.h>
  49 //
  50 #include "AliLog.h"
  51 #include "AliDetector.h"
  52 #include "AliHeader.h"
  53 #include "AliMC.h"
  54 #include "AliPDG.h"
  55 #include "AliRun.h"
  56 #include "AliStack.h"
  57 #include "AliCDBManager.h"
  58 #include "AliAlignObj.h"
  59 #include "AliSimulation.h"
  60 #include "AliLego.h"
  61
  62 AliRun *gAlice;
  63
  64 ClassImp(AliRun)
  65
  66 //_______________________________________________________________________
  67 AliRun::AliRun():
  68   fRun(-1),
  69   fEvent(0),
  70   fEventNrInRun(-1),
  71   fModules(0),
  72   fMCApp(0),
  73   fNdets(0),
  74   fConfigFunction(""),
  75   fRandom(0),
  76   fBaseFileName(""),
  77   fIsRootGeometry(kFALSE),
  78   fGeometryFromCDB(kFALSE),
  79   fGeometryFileName(""),
  80   fTriggerDescriptor(""),
  81   fRunLoader(0x0)
  82 {
  83   //
  84   // Default constructor for AliRun
  85   //
  86   AliConfig::Instance();//skowron 29 Feb 2002
  87                         //ensures that the folder structure is build
  88
  89 }
  90
  91 //_____________________________________________________________________________
  92 AliRun::AliRun(const char *name, const char *title):
  93   TNamed(name,title),
  94   fRun(-1),
  95   fEvent(0),
  96   fEventNrInRun(-1),
  97   fModules(new TObjArray(77)), // Support list for the Detectors
  98   fMCApp(new AliMC(GetName(),GetTitle())),
  99   fNdets(0),
 100   fConfigFunction("Config();"),
 101   fRandom(new TRandom3()),
 102   fBaseFileName(""),
 103   fIsRootGeometry(kFALSE),
 104   fGeometryFromCDB(kFALSE),
 105   fGeometryFileName(""),
 106   fTriggerDescriptor(""),
 107   fRunLoader(0x0)
 108 {
 109   //
 110   //  Constructor for the main processor.
 111   //  Creates the geometry
 112   //  Creates the list of Detectors.
 113   //  Creates the list of particles.
 114   //
 115
 116   gAlice     = this;
 117
 118   // Set random number generator
 119   gRandom = fRandom;
 120
 121   if (gSystem->Getenv("CONFIG_SEED")) {
 122      gRandom->SetSeed(static_cast<UInt_t>(atoi(gSystem->Getenv("CONFIG_SEED"))));
 123   }
 124
 125   // Add to list of browsable
 126   gROOT->GetListOfBrowsables()->Add(this,name);
 127
 128 }
 129
 130
 131 //_______________________________________________________________________
 132 AliRun::~AliRun()
 133 {
 134   //
 135   // Default AliRun destructor
 136   //
 137   gROOT->GetListOfBrowsables()->Remove(this);
 138
 139   if (fRunLoader)
 140    {
 141     TFolder* evfold = fRunLoader->GetEventFolder();
 142     TFolder* modfold = dynamic_cast<TFolder*>(evfold->FindObjectAny(AliConfig::GetModulesFolderName()));
 143     TIter next(fModules);
 144     AliModule *mod;
 145     while((mod = (AliModule*)next()))
 146      {
 147        modfold->Remove(mod);
 148      }
 149    }
 150
 151   delete fMCApp;
 152   delete gMC; gMC=0;
 153   if (fModules) {
 154     fModules->Delete();
 155     delete fModules;
 156   }
 157
 158 }
 159
 160
 161 //_______________________________________________________________________
 162 void AliRun::SetRootGeometry(Bool_t flag)
 163 {
 164 // Instruct application that the geometry is to be retreived from a root file.
 165    fIsRootGeometry = flag;
 166    if (flag && gMC) gMC->SetRootGeometry();
 167 }
 168
 169 //_______________________________________________________________________
 170 void AliRun::SetGeometryFromCDB()
 171 {
 172   // Set the loading of geometry from cdb instead of creating it
 173   // A default CDB storage needs to be set before this method is called
 174   if(AliCDBManager::Instance()->IsDefaultStorageSet() &&
 175         AliCDBManager::Instance()->GetRun() >= 0){
 176     SetRootGeometry();
 177     fGeometryFromCDB = kTRUE;
 178   }else{
 179     AliError("!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!");
 180     AliError("Loading of geometry from CDB ignored. First set a default CDB storage!");
 181     AliError("!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!");
 182   }
 183 }
 184
 185 //_____________________________________________________________________________
 186
 187 void AliRun::InitLoaders()
 188 {
 189   //creates list of getters
 190   AliDebug(1, "");
 191   TIter next(fModules);
 192   AliModule *mod;
 193   while((mod = (AliModule*)next()))
 194    {
 195      mod->SetRunLoader(fRunLoader);
 196      AliDetector *det = dynamic_cast<AliDetector*>(mod);
 197      if (det)
 198       {
 199         AliDebug(2, Form("Adding %s", det->GetName()));
 200         fRunLoader->AddLoader(det);
 201       }
 202    }
 203   AliDebug(1, "Done");
 204 }
 205
 206 //_______________________________________________________________________
 207 void AliRun::Announce() const
 208 {
 209   //
 210   // Announce the current version of AliRoot
 211   //
 212   printf("%70s",
 213          "****************************************************************\n");
 214   printf("%6s","*");printf("%64s","*\n");
 215
 216   printf("%6s","*");
 217   printf("    You are running AliRoot version NewIO\n");
 218
 219   printf("%6s","*");
 220   printf("    The SVN version for the current program is $Id$\n");
 221
 222   printf("%6s","*");printf("%64s","*\n");
 223   printf("%70s",
 224          "****************************************************************\n");
 225 }
 226
 227 //_______________________________________________________________________
 228 AliModule *AliRun::GetModule(const char *name) const
 229 {
 230   //
 231   // Return pointer to detector from name
 232   //
 233   return dynamic_cast<AliModule*>(fModules->FindObject(name));
 234 }
 235
 236 //_______________________________________________________________________
 237 AliDetector *AliRun::GetDetector(const char *name) const
 238 {
 239   //
 240   // Return pointer to detector from name
 241   //
 242   return dynamic_cast<AliDetector*>(fModules->FindObject(name));
 243 }
 244
 245 //_______________________________________________________________________
 246 Int_t AliRun::GetModuleID(const char *name) const
 247 {
 248   //
 249   // Return galice internal detector identifier from name
 250   //
 251   Int_t i=-1;
 252   TObject *mod=fModules->FindObject(name);
 253   if(mod) i=fModules->IndexOf(mod);
 254   return i;
 255 }
 256
 257 //_______________________________________________________________________
 258 Int_t AliRun::GetEvent(Int_t event)
 259 {
 260 //
 261 // Reloads data containers in folders # event
 262 // Set branch addresses
 263 //
 264   if (fRunLoader == 0x0)
 265    {
 266      AliError("RunLoader is not set. Can not load data.");
 267      return -1;
 268    }
 269 /*****************************************/
 270 /****   P R E    R E L O A D I N G    ****/
 271 /*****************************************/
 272 // Reset existing structures
 273   fMCApp->ResetHits();
 274   fMCApp->ResetTrackReferences();
 275   fMCApp->ResetDigits();
 276   fMCApp->ResetSDigits();
 277
 278 /*****************************************/
 279 /****       R  E  L  O  A  D          ****/
 280 /*****************************************/
 281
 282   AliRunLoader::GetRunLoader()->GetEvent(event);
 283
 284 /*****************************************/
 285 /****  P O S T    R E L O A D I N G   ****/
 286 /*****************************************/
 287
 288   // Set Trees branch addresses
 289   TIter next(fModules);
 290   AliDetector *detector;
 291   while((detector = dynamic_cast<AliDetector*>(next())))
 292    {
 293      detector->SetTreeAddress();
 294    }
 295
 296   return AliRunLoader::GetRunLoader()->GetHeader()->GetNtrack();
 297 }
 298
 299 //_______________________________________________________________________
 300 void AliRun::SetBaseFile(const char *filename)
 301 {
 302   fBaseFileName = filename;
 303 }
 304
 305
 306 //_______________________________________________________________________
 307 void AliRun::Hits2Digits(const char *selected)
 308 {
 309
 310    // Convert Hits to sumable digits
 311    //
 312    for (Int_t nevent=0; nevent<AliRunLoader::GetRunLoader()->TreeE()->GetEntries(); nevent++) {
 313      GetEvent(nevent);
 314      Hits2SDigits(selected);
 315      SDigits2Digits(selected);
 316    }
 317 }
 318
 319
 320 //_______________________________________________________________________
 321
 322 void AliRun::Tree2Tree(Option_t *option, const char *selected)
 323 {
 324   //
 325   // Function to transform the content of
 326   //
 327   // - TreeH to TreeS (option "S")
 328   // - TreeS to TreeD (option "D")
 329   // - TreeD to TreeR (option "R")
 330   //
 331   // If multiple options are specified ("SDR"), transformation will be done in sequence for
 332   // selected detector and for all detectors if none is selected (detector string
 333   // can contain blank separated list of detector names).
 334
 335
 336    const char *oS = strstr(option,"S");
 337    const char *oD = strstr(option,"D");
 338    const char *oR = strstr(option,"R");
 339
 340    TObjArray *detectors = Detectors();
 341
 342    TIter next(detectors);
 343
 344    AliDetector *detector = 0;
 345
 346    while((detector = dynamic_cast<AliDetector*>(next()))) {
 347      if (selected)
 348        if (strcmp(detector->GetName(),selected)) continue;
 349      if (detector->IsActive())
 350       {
 351
 352        AliLoader* loader = detector->GetLoader();
 353        if (loader == 0x0) continue;
 354
 355        if (oS)
 356         {
 357           AliDebug(1, Form("Processing Hits2SDigits for %s ...", detector->GetName()));
 358           loader->LoadHits("read");
 359           if (loader->TreeS() == 0x0) loader->MakeTree("S");
 360           detector->MakeBranch(option);
 361           detector->SetTreeAddress();
 362           detector->Hits2SDigits();
 363           loader->UnloadHits();
 364           loader->UnloadSDigits();
 365         }
 366        if (oD)
 367         {
 368           AliDebug(1, Form("Processing SDigits2Digits for %s ...", detector->GetName()));
 369           loader->LoadSDigits("read");
 370           if (loader->TreeD() == 0x0) loader->MakeTree("D");
 371           detector->MakeBranch(option);
 372           detector->SetTreeAddress();
 373           detector->SDigits2Digits();
 374           loader->UnloadSDigits();
 375           loader->UnloadDigits();
 376         }
 377        if (oR)
 378         {
 379           AliDebug(1, Form("Processing Digits2Reco for %s ...", detector->GetName()));
 380           loader->LoadDigits("read");
 381           if (loader->TreeR() == 0x0) loader->MakeTree("R");
 382           detector->MakeBranch(option);
 383           detector->SetTreeAddress();
 384           detector->Digits2Reco();
 385           loader->UnloadDigits();
 386           loader->UnloadRecPoints();
 387
 388         }
 389      }
 390    }
 391 }
 392
 393
 394 //_______________________________________________________________________
 395 void AliRun::Streamer(TBuffer &R__b)
 396 {
 397   // Stream an object of class AliRun.
 398
 399   if (R__b.IsReading()) {
 400     if (!gAlice) gAlice = this;
 401     AliRun::Class()->ReadBuffer(R__b, this);
 402     gROOT->GetListOfBrowsables()->Add(this,"Run");
 403
 404     gRandom = fRandom;
 405   } else {
 406     AliRun::Class()->WriteBuffer(R__b, this);
 407   }
 408 }
 409 //_______________________________________________________________________
 410
 411 void AliRun::SetGenEventHeader(AliGenEventHeader* header)
 412 {
 413   AliRunLoader::GetRunLoader()->GetHeader()->SetGenEventHeader(header);
 414 }
 415
 416
 417 //_______________________________________________________________________
 418
 419 Int_t AliRun::GetEvNumber() const
 420 {
 421 //Returns number of current event
 422   if (fRunLoader == 0x0)
 423    {
 424      AliError("RunLoader is not set. Can not load data.");
 425      return -1;
 426    }
 427
 428   return fRunLoader->GetEventNumber();
 429 }
 430 //_______________________________________________________________________
 431
 432 void AliRun::SetRunLoader(AliRunLoader* rloader)
 433 {
 434   //
 435   // Set the loader of the run
 436   //
 437   fRunLoader = rloader;
 438   if (fRunLoader == 0x0) return;
 439
 440   TString evfoldname;
 441   TFolder* evfold = fRunLoader->GetEventFolder();
 442   if (evfold) evfoldname = evfold->GetName();
 443   else AliWarning("Did not get Event Folder from Run Loader");
 444
 445   if ( fRunLoader->GetAliRun() )
 446    {//if alrun already exists in folder
 447     if (fRunLoader->GetAliRun() != this )
 448      {//and is different than this - crash
 449        AliFatal("AliRun is already in Folder and it is not this object");
 450        return;//pro forma
 451      }//else do nothing
 452    }
 453   else
 454    {
 455      evfold->Add(this);//Post this AliRun to Folder
 456    }
 457
 458   TIter next(fModules);
 459   AliModule *module;
 460   while((module = (AliModule*)next()))
 461    {
 462      if (evfold) AliConfig::Instance()->Add(module,evfoldname);
 463      module->SetRunLoader(fRunLoader);
 464      AliDetector* detector = dynamic_cast<AliDetector*>(module);
 465      if (detector)
 466       {
 467         AliLoader* loader = fRunLoader->GetLoader(detector);
 468         if (loader == 0x0)
 469          {
 470            AliError(Form("Can not get loader for detector %s", detector->GetName()));
 471          }
 472         else
 473          {
 474            AliDebug(1, Form("Setting loader for detector %s", detector->GetName()));
 475            detector->SetLoader(loader);
 476          }
 477       }
 478    }
 479 }
 480
 481 void AliRun::AddModule(AliModule* mod)
 482 {
 483   //
 484   // Add a module to the module list
 485   //
 486   if (mod == 0x0) return;
 487   if (strlen(mod->GetName()) == 0) return;
 488   if (GetModuleID(mod->GetName()) >= 0) return;
 489
 490   AliDebug(1, mod->GetName());
 491   if (fRunLoader == 0x0) AliConfig::Instance()->Add(mod);
 492   else AliConfig::Instance()->Add(mod,fRunLoader->GetEventFolder()->GetName());
 493
 494   Modules()->Add(mod);
 495
 496   fNdets++;
 497 }
 498