PWGPP/AliTaskCDBconnect.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 #include "AliTaskCDBconnect.h"
  17
  18 #include <TChain.h>
  19 #include <TFile.h>
  20 #include <TGeoGlobalMagField.h>
  21 #include "TGeoManager.h"
  22
  23 #include "AliAnalysisManager.h"
  24 #include "AliGeomManager.h"
  25 #include "AliCDBManager.h"
  26 #include "AliGRPManager.h"
  27 #include "AliVEvent.h"
  28 #include "AliInputEventHandler.h"
  29 #include "AliLog.h"
  30
  31 ClassImp(AliTaskCDBconnect)
  32
  33 //______________________________________________________________________________
  34 AliTaskCDBconnect::AliTaskCDBconnect():
  35            AliAnalysisTask(),
  36            fRun(0),
  37            fGRPManager(NULL)
  38 {
  39 // Dummy constructor
  40 }
  41
  42 //______________________________________________________________________________
  43 AliTaskCDBconnect::AliTaskCDBconnect(const char* name, const char *storage, Int_t run)
  44           :AliAnalysisTask(name, "ESD analysis tender car"),
  45            fRun(run),
  46            fGRPManager(NULL)
  47 {
  48 // Default constructor
  49   AliCDBManager *cdb = AliCDBManager::Instance();
  50   cdb->SetDefaultStorage(storage);
  51   DefineInput (0, TChain::Class());
  52   if (run>0) InitGRP();
  53 }
  54
  55 //______________________________________________________________________________
  56 AliTaskCDBconnect::~AliTaskCDBconnect()
  57 {
  58   // Destructor
  59   delete fGRPManager;
  60 }
  61
  62 //______________________________________________________________________________
  63 void AliTaskCDBconnect::InitGRP()
  64 {
  65   // Initialize geometry and mag. field
  66   AliCDBManager *cdb = AliCDBManager::Instance();
  67   if (!cdb->IsDefaultStorageSet()) cdb->SetDefaultStorage("raw://");
  68   cdb->SetRun(fRun);
  69   if (!fGRPManager) fGRPManager = new AliGRPManager();
  70   AliInfo("AliCDBconnect: #### Loading GRP to init B-field...");
  71   if(!fGRPManager->ReadGRPEntry()) AliFatal("Cannot get GRP entry");
  72   if(!fGRPManager->SetMagField())  AliFatal("Problem with magnetic field setup");
  73   //
  74   // geometry
  75   if (!gGeoManager) {
  76     AliInfo("AliCDBconnect: #### Loading geometry...");
  77     AliGeomManager::LoadGeometry("geometry.root");
  78     if(!AliGeomManager::ApplyAlignObjsFromCDB("GRP ITS TPC TRD")) AliWarning("Problem with align objects");
  79   }
  80 }
  81
  82 //______________________________________________________________________________
  83 void AliTaskCDBconnect::CreateOutputObjects()
  84 {
  85   // Init CDB locally if run number is defined.
  86   //
  87   //  try to init before the analysis set
  88   AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
  89   if (!mgr) AliFatal("No analysis manager");
  90   if ( fRun>0 && !fGRPManager) {
  91     // in the proof or plugin mode the initialization done in the constructor is not available
  92     InitGRP();
  93   }
  94   else {
  95     AliInfo("Run number is not available at this stage, InitGRP will be called in the execution loop");
  96   }
  97   //
  98 }
  99
 100 //______________________________________________________________________________
 101 void AliTaskCDBconnect::Exec(Option_t* /*option*/)
 102 {
 103 //
 104 // Execute all supplied analysis of one event. Notify run change via RunChanged().
 105   AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
 106   if (!mgr) AliFatal("No analysis manager");
 107   AliInputEventHandler* inp = (AliInputEventHandler*)mgr->GetInputEventHandler();
 108   if (!inp) AliFatal("No input event handler connected");
 109   //
 110   AliVEvent* ev = inp->GetEvent();
 111   if (!ev) AliFatal("No event returned");
 112   int run = ev->GetRunNumber();
 113   // Intercept when the run number changed
 114   if (fRun != run) {
 115     fRun = run;
 116     InitGRP();
 117   }
 118 }
 119
 120 //______________________________________________________________________________
 121 void AliTaskCDBconnect::SetSpecificStorage(const char* calibType, const char* dbString)
 122 {
 123     AliCDBManager *cdb = AliCDBManager::Instance();
 124     cdb->SetSpecificStorage(calibType,dbString);
 125  }