Switching off reclustering in tender supply for ET analysis

[u/mrichter/AliRoot.git] / PWGLF / FORWARD / trains / GridTerminate.C

/**
 * @file   Terminate.C
 * @author Christian Holm Christensen <cholm@master.hehi.nbi.dk>
 * @date   Wed Jan 23 21:22:28 2013
 * 
 * @brief  Worker script to run terminate part for Grid  
 * 
 * 
 */
/** 
 * Load a library 
 * 
 * @param libName Library name 
 * 
 * @return true on success 
 */
Bool_t LoadLib(const char* libName)
{
  if (gSystem->Load(libName) < 0) {
    Error("GridTerminate", "Failed to load library %s",libName);
    return false;
  }
  Info("GridTerminate","Loaded library %s",libName);
  return true;
}
/** 
 * Load a PAR 
 * 
 * @param parName PAR file name 
 * 
 * @return true on success
 */
Bool_t LoadPar(const char* parName)
{
  if (!AliAnalysisAlien::SetupPar(parName)) {
    Error("GridTerminate","Failed to load PAR %s",parName);
    return false;
  }
  Info("GridTerminate","Loaded package %s",parName);
  return true;
}
/** 
 * Submit the terminate job 
 * 
 * @param name Name of the job
 * @param libs Libraries to load (space separated string)
 * @param pars PARs to load (space separated string)
 * @param srcs Sources to load (space separated string)
 * 
 * @return true on success 
 */
Bool_t GridTerminate(const TString& name, 
		     const TString& libs, 
		     const TString& pars, 
		     const TString& srcs)
{
  // Load basic ROOT libraries
  gSystem->AddDynamicPath("/usr/lib");
  if (gSystem->Load("libTree.so")       < 0) return false;
  if (gSystem->Load("libGeom.so")       < 0) return false;
  if (gSystem->Load("libVMC.so")        < 0) return false;
  if (gSystem->Load("libPhysics.so")    < 0) return false;
  if (gSystem->Load("libMinuit.so")     < 0) return false;

  // Load basic AliROOT libraries
  if (gSystem->Load("libSTEERBase")     < 0) return false;
  if (gSystem->Load("libESD")           < 0) return false;
  if (gSystem->Load("libAOD")           < 0) return false;
  if (gSystem->Load("libANALYSIS")      < 0) return false;
  if (gSystem->Load("libOADB")          < 0) return false;
  if (gSystem->Load("libANALYSISalice") < 0) return false;

  // Load libraries
  TObjArray*  libsArray = libs.Tokenize(" ");
  TObjString* lib       = 0;
  TIter       nextLib(libsArray);
  while ((lib = static_cast<TObjString*>(nextLib()))) {
    const TString& libName = lib->String();
    if (libName.Contains("libSTEERBase") ||
	libName.Contains("libESD")       ||
	libName.Contains("libAOD")       ||
	libName.Contains("libANALYSIS")  ||
	libName.Contains("libOADB")      ||
	libName.Contains("libANALYSISalice")) continue;
    if (!libName.Contains(".so")) continue;
    if (!LoadLib(libName.Data())) return false;
  }
  libsArray.Delete();
  
  // Load packages
  TObjArray*  parArray = pars.Tokenize(" ");
  TObjString* par      = 0;
  TIter       nextPar(parArray);
  while ((par = static_cast<TObjString*>(nextPar()))) { 
    TString parName(par->String());
    if (parName.EndsWith(".par")) parName.ReplaceAll(".par", "");
    if (parName.Contains("STEERBase") ||
	parName.Contains("ESD")       ||
	parName.Contains("AOD")       ||
	parName.Contains("ANALYSIS")  ||
	parName.Contains("OADB")      ||
	parName.Contains("ANALYSISalice")) continue;
    if (!LoadPar(parName.Data())) return false;
  }

  // Load sources
  TObjArray*  srcArray = srcs.Tokenize(" ");
  TObjString* src      = 0;
  TIter       nextSrc(srcArray);
  while ((src = static_cast<TObjString*>(nextSrc()))) { 
    const TString& srcName = src->String();
    gROOT->ProcessLine(Form(".L %s+g", srcName.Data()));
  }

  // Load the analysis manager from file
  TString base(name);
  base.Append(".root");
  if (gSystem->AccessPathName(base.Data())) {
    // Couldn't read from current directory, try sub-dir
    TString sub(gSystem->ConcatFileName(name, base));
    if (gSystem->AccessPathName(sub)) {
      Error("GridTerminate","Couldn't find manager file %s",base.Data());
      return false;
    }
    base = sub;
  }
  gEnv->SetValue("XSec.GSI.DelegProxy", "2");
  AliAnalysisManager* mgr= AliAnalysisAlien::LoadAnalysisManager(base);
  if (!mgr) {
    Error("GridTerminate", "Failed to load manager from %s",base.Data());
    return false;
  }
  if (!name.EqualTo(mgr->GetName())) {
    Error("GridTerminate","Read manager %s is not %s",mgr->GetName(),name.Data());
    return false;
  }
  Info("GridTerminate","Loaded analysis manager");

  // Load plugin
  TFile* plug = TFile::Open(Form("%s_plugin.root",name.Data()),"READ");
  if (!plug) {
    Error("GridTerminate","Failed to open %s_plugin.root",name.Data());
    return false;
  }
  AliAnalysisAlien* handler = 
    static_cast<AliAnalysisAlien*>(plug->Get("plugin"));
  if (!handler) {
    Error("GridTerminate","Failed to load plugin");
    return false;
  }
  Info("GridTerminate","Setting grid handler");
  handler->SetRunMode("terminate");
  mgr->SetGridHandler(handler);

  // Run the terminate job
  Info("GridTerminate","Starting terminate job");
  if (mgr->StartAnalysis("grid") < 0) return false;
  // Info("GridTerminate","Job IDs: %s",handler->GetGridJobIDs().Data());
  return true;
}
// 
// EOF
//