guess the run number from the input file path

[u/mrichter/AliRoot.git] / PWGPP / TPC / AliTaskConfigOCDB.cxx

/**************************************************************************
 * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
 *                                                                        *
 * Author: The ALICE Off-line Project.                                    *
 * Contributors are mentioned in the code where appropriate.              *
 *                                                                        *
 * Permission to use, copy, modify and distribute this software and its   *
 * documentation strictly for non-commercial purposes is hereby granted   *
 * without fee, provided that the above copyright notice appears in all   *
 * copies and that both the copyright notice and this permission notice   *
 * appear in the supporting documentation. The authors make no claims     *
 * about the suitability of this software for any purpose. It is          *
 * provided "as is" without express or implied warranty.                  *
 **************************************************************************/

#include "AliTaskConfigOCDB.h"

#include <TChain.h>
#include <TFile.h>
#include <TGeoGlobalMagField.h>
#include "TGeoManager.h"
#include <TRegexp.h>
 
#include "AliAnalysisManager.h"
#include "AliGeomManager.h"
#include "AliCDBManager.h"
#include "AliGRPManager.h"
#include "AliESDEvent.h"
#include "AliESDInputHandler.h"
#include "AliLog.h"

ClassImp(AliTaskConfigOCDB)

//______________________________________________________________________________
AliTaskConfigOCDB::AliTaskConfigOCDB():
           AliAnalysisTask(),
           fRun(0),
           fRunChanged(kFALSE),
           fESDhandler(NULL),
           fESD(NULL),
           fGRPManager(NULL)
{
// Dummy constructor
}

//______________________________________________________________________________
AliTaskConfigOCDB::AliTaskConfigOCDB(const char* name, const char *storage, Int_t run)
          :AliAnalysisTask(name, "configure OCDB field geom"),
           fRun(run),
           fRunChanged(kFALSE),
           fESDhandler(NULL),
           fESD(NULL),
           fGRPManager(NULL)
{
// Default constructor
  AliCDBManager *cdb = AliCDBManager::Instance();
  cdb->SetDefaultStorage(storage);
  cdb->SetRun(run);
  DefineInput (0, TChain::Class());
}

//______________________________________________________________________________
AliTaskConfigOCDB::~AliTaskConfigOCDB()
{
// Destructor
  delete fGRPManager;
}  

//______________________________________________________________________________
void AliTaskConfigOCDB::LocalInit()
{
// Init CDB locally if run number is defined.
}
  
//______________________________________________________________________________
void AliTaskConfigOCDB::ConnectInputData(Option_t* /*option*/)
{
// Connect the input data, create CDB manager.
}

//______________________________________________________________________________
void AliTaskConfigOCDB::InitGRP()
{
// Initialize geometry and mag. field
  if (!fGRPManager) {
  // magnetic field
    if (!TGeoGlobalMagField::Instance()->GetField()) {
      printf("AliCDBconnect: #### Loading field map...\n");
      fGRPManager = new AliGRPManager();
      if(!fGRPManager->ReadGRPEntry()) { 
        AliError("Cannot get GRP entry"); 
      }
      if( !fGRPManager->SetMagField() ) { 
        AliError("Problem with magnetic field setup"); 
      }
    }

    // geometry
    if (!gGeoManager) {
      printf("AliCDBconnect: #### Loading geometry...\n");
      AliGeomManager::LoadGeometry();
      if( !AliGeomManager::ApplyAlignObjsFromCDB("GRP ITS TPC TRD") ) {
        AliError("Problem with align objects"); 
      }
    }  
  }  
}

//______________________________________________________________________________
void AliTaskConfigOCDB::CreateOutputObjects()
{
// Init CDB locally if run number is defined.
  AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
  if (!mgr) AliFatal("No analysis manager");
  fESDhandler = dynamic_cast<AliESDInputHandler *>(mgr->GetInputEventHandler());
    
  if (!fESDhandler) {
     AliFatal("No ESD input event handler connected");
     return;
  }   
  // Try to get event number before the first event is read (this has precedence
  // over existing fRun)
  TTree* inputTree = mgr->GetTree();
  if (!inputTree) { AliError("no input tree"); return; }
  TFile* inputFile = inputTree->GetCurrentFile();
  if (!inputFile) { AliError("no input file"); return; }
  TString inputFileName(inputFile->GetName());
  Int_t run = guessRunNumber(inputFileName);
  mgr->SetRunFromPath(run);

  if (!run && !fRun) {
     AliError("AliTaskConfigOCDB: Run not set - no CDB connection");
     return;
  }
  // Create CDB manager
  AliCDBManager *cdb = AliCDBManager::Instance();
  // If CDB is already locked, return
  if (cdb->GetLock()) return;
  // SetDefault storage. Specific storages must be set by TaskCDBconnectSupply::Init()
  //  cdb->SetDefaultStorage("local://$ALICE_ROOT/OCDB");
//  if (!cdb->GetRaw()) {
//     cdb->SetDefaultStorage("raw://");
//  }   
  if (run && (run != fRun)) {
     fRunChanged = kTRUE;
     fRun = run;
  } else {
     fRunChanged = kFALSE;
  }
  // Set run
  if (fRunChanged || !fGRPManager) {
     printf("AliCDBconnect: #### Setting run to: %d\n", fRun);
     cdb->SetRun(fRun);
     // Initialize GRP manager only once
     if (fRun) InitGRP();
  }   
}

//______________________________________________________________________________
Bool_t AliTaskConfigOCDB::Notify()
{
// Init CDB locally if run number is defined.
  CreateOutputObjects();
  return kTRUE;
}

//______________________________________________________________________________
void AliTaskConfigOCDB::Exec(Option_t* /*option*/)
{
//
// Execute all supplied analysis of one event. Notify run change via RunChanged().
  fESD = fESDhandler->GetEvent();
  // Intercept when the run number changed
  if (fRun != fESD->GetRunNumber()) {
    fRunChanged = kTRUE;
    fRun = fESD->GetRunNumber();
    CreateOutputObjects();
  }
}

//______________________________________________________________________________
void AliTaskConfigOCDB::Terminate(Option_t *)
{
  // Initialize CDB also in Terminate
  //   CreateOutputObjects();
}

Int_t AliTaskConfigOCDB::guessRunNumber(TString path)
{
  //guess the runnumber from datapath
  //works also on the LEGO train where the data path looks like this:
  //workdir/testdata/__alice__data__2010__LHC10b__000114924__ESDs__pass2_root_archive_AliESDs_2/10000114924018.100/root_archive.zip
  TObjArray* a = path.Tokenize("/_");
  TRegexp r("^000[0-9][0-9][0-9][0-9][0-9][0-9]$");
  TString sub;
  for (Int_t i=0; i<a->GetEntries();i++)
  {
    TObjString* subobj = (TObjString*)a->At(i);
    TString subtmp = subobj->GetString();
    if (subtmp.Contains(r)) 
    {
      sub=subtmp;
      break;
    }
  }
  Int_t runNumber=sub.Atoi();
  AliInfo(Form("guessed run: %i\n",runNumber));
  a->Delete();
  delete a;
  return runNumber;
}