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 "AliESDEvent.h"
  28 #include "AliESDInputHandler.h"
  29 #include "AliLog.h"
  30
  31 ClassImp(AliTaskCDBconnect)
  32
  33 //______________________________________________________________________________
  34 AliTaskCDBconnect::AliTaskCDBconnect():
  35            AliAnalysisTask(),
  36            fRun(0),
  37            fRunChanged(kFALSE),
  38            fESDhandler(NULL),
  39            fESD(NULL),
  40            fGRPManager(NULL)
  41 {
  42 // Dummy constructor
  43 }
  44
  45 //______________________________________________________________________________
  46 AliTaskCDBconnect::AliTaskCDBconnect(const char* name):
  47            AliAnalysisTask(name, "ESD analysis tender car"),
  48            fRun(0),
  49            fRunChanged(kFALSE),
  50            fESDhandler(NULL),
  51            fESD(NULL),
  52            fGRPManager(NULL)
  53 {
  54 // Default constructor
  55   AliCDBManager *cdb = AliCDBManager::Instance();
  56   cdb->SetDefaultStorage("local://$ALICE_ROOT/OCDB");
  57   cdb->SetRun(0);
  58   DefineInput (0, TChain::Class());
  59 //  DefineOutput(0,  AliESDEvent::Class());
  60 }
  61
  62 //______________________________________________________________________________
  63 AliTaskCDBconnect::~AliTaskCDBconnect()
  64 {
  65 // Destructor
  66   if (fGRPManager) delete fGRPManager;
  67 //  if (gGeoManager) delete gGeoManager;
  68 }
  69
  70 //______________________________________________________________________________
  71 void AliTaskCDBconnect::LocalInit()
  72 {
  73 // Init CDB locally if run number is defined.
  74 }
  75
  76 //______________________________________________________________________________
  77 void AliTaskCDBconnect::ConnectInputData(Option_t* /*option*/)
  78 {
  79 // Connect the input data, create CDB manager.
  80 }
  81
  82 //______________________________________________________________________________
  83 void AliTaskCDBconnect::InitGRP()
  84 {
  85 // Initialize geometry and mag. field
  86   if (!fGRPManager) {
  87   // magnetic field
  88     if (!TGeoGlobalMagField::Instance()->GetField()) {
  89       printf("AliCDBconnect: #### Loading field map...\n");
  90       fGRPManager = new AliGRPManager();
  91       if(!fGRPManager->ReadGRPEntry()) {
  92         AliError("Cannot get GRP entry");
  93       }
  94       if( !fGRPManager->SetMagField() ) {
  95         AliError("Problem with magnetic field setup");
  96       }
  97     }
  98
  99     // geometry
 100     if (!gGeoManager) {
 101       printf("AliCDBconnect: #### Loading geometry...\n");
 102       AliGeomManager::LoadGeometry("geometry.root");
 103       if( !AliGeomManager::ApplyAlignObjsFromCDB("GRP ITS TPC TRD") ) {
 104         AliError("Problem with align objects");
 105       }
 106     }
 107   }
 108 }
 109
 110 //______________________________________________________________________________
 111 void AliTaskCDBconnect::CreateOutputObjects()
 112 {
 113 // Init CDB locally if run number is defined.
 114   AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
 115   if (!mgr) AliFatal("No analysis manager");
 116   fESDhandler = dynamic_cast<AliESDInputHandler *>(mgr->GetInputEventHandler());
 117
 118   if (!fESDhandler) {
 119      AliFatal("No ESD input event handler connected");
 120      return;
 121   }
 122   // Try to get event number before the first event is read (this has precedence
 123   // over existing fRun)
 124   Int_t run = mgr->GetRunFromPath();
 125   if (!run && !fRun) {
 126      AliError("AliTaskCDBconnect: Run not set - no CDB connection");
 127      return;
 128   }
 129   // Create CDB manager
 130   AliCDBManager *cdb = AliCDBManager::Instance();
 131   // If CDB is already locked, return
 132   if (cdb->GetLock()) return;
 133   // SetDefault storage. Specific storages must be set by TaskCDBconnectSupply::Init()
 134   //  cdb->SetDefaultStorage("local://$ALICE_ROOT/OCDB");
 135   if (!cdb->GetRaw()) {
 136      cdb->SetDefaultStorage("raw://");
 137   }
 138   if (run && (run != fRun)) {
 139      fRunChanged = kTRUE;
 140      fRun = run;
 141   } else {
 142      fRunChanged = kFALSE;
 143   }
 144   // Set run
 145   if (fRunChanged || !fGRPManager) {
 146      printf("AliCDBconnect: #### Setting run to: %d\n", fRun);
 147      cdb->SetRun(fRun);
 148      // Initialize GRP manager only once
 149      if (fRun) InitGRP();
 150   }
 151 }
 152
 153 //______________________________________________________________________________
 154 Bool_t AliTaskCDBconnect::Notify()
 155 {
 156 // Init CDB locally if run number is defined.
 157   CreateOutputObjects();
 158   return kTRUE;
 159 }
 160
 161 //______________________________________________________________________________
 162 void AliTaskCDBconnect::Exec(Option_t* /*option*/)
 163 {
 164 //
 165 // Execute all supplied analysis of one event. Notify run change via RunChanged().
 166   fESD = fESDhandler->GetEvent();
 167   // Intercept when the run number changed
 168   if (fRun != fESD->GetRunNumber()) {
 169     fRunChanged = kTRUE;
 170     fRun = fESD->GetRunNumber();
 171     CreateOutputObjects();
 172   }
 173 }
 174
 175 //______________________________________________________________________________
 176 void AliTaskCDBconnect::Terminate(Option_t *)
 177 {
 178 // Initialize CDB also in Terminate
 179 //   CreateOutputObjects();
 180 }