PWG1/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();
 103       if( !AliGeomManager::ApplyAlignObjsFromCDB("GRP ITS TPC") ) {
 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
 123   Int_t run = mgr->GetRunFromPath();
 124   if (!run && fESD) run = fESD->GetRunNumber();
 125   if (!run) {
 126      AliWarning("AliTaskCDBconnect: Could not set run from path");
 127      if (!fRun) return;
 128   }
 129   // Create CDB manager
 130   AliCDBManager *cdb = AliCDBManager::Instance();
 131   // SetDefault storage. Specific storages must be set by TaskCDBconnectSupply::Init()
 132 //  cdb->SetDefaultStorage("local://$ALICE_ROOT/OCDB");
 133   cdb->SetDefaultStorage("raw://");
 134   if (run && (run != fRun)) {
 135      fRunChanged = kTRUE;
 136      fRun = run;
 137   } else {
 138      fRunChanged = kFALSE;
 139   }
 140   // Set run
 141   printf("AliCDBconnect: #### Setting run to: %d\n", fRun);
 142   cdb->SetRun(fRun);
 143   // Initialize GRP manager only once
 144   if (fRun) InitGRP();
 145 }
 146
 147 //______________________________________________________________________________
 148 Bool_t AliTaskCDBconnect::Notify()
 149 {
 150 // Init CDB locally if run number is defined.
 151   CreateOutputObjects();
 152   return kTRUE;
 153 }
 154
 155 //______________________________________________________________________________
 156 void AliTaskCDBconnect::Exec(Option_t* /*option*/)
 157 {
 158 //
 159 // Execute all supplied analysis of one event. Notify run change via RunChanged().
 160   fESD = fESDhandler->GetEvent();
 161   // Intercept when the run number changed
 162   if (fRun != fESD->GetRunNumber()) {
 163     fRunChanged = kTRUE;
 164     fRun = fESD->GetRunNumber();
 165     Warning("Exec", "Run number changed in ESDEvent to %d", fRun);
 166     AliCDBManager::Instance()->SetRun(fRun);
 167   }
 168 //  PostData(0, fESD);
 169 }