Fixed references from PWG2 -> PWGLF - very efficiently done using ETags.

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

/** 
 * Build (compile) a train script 
 * 
 * @param script Script to compile 
 * @param extra  Extra stuff for AcLic ("", "+", or "+g")
 * @param useTmp Use a temporary file 
 * 
 * @return 
 *
 * @ingroup pwglf_forward_trains
 */
Bool_t
BuildTrain(const char* script, const char* extra="", Bool_t useTmp=false)
{
  // --- Load needed libraries ---------------------------------------
  gROOT->LoadClass("TAlien",               "libRAliEn");
  gROOT->LoadClass("TVirtualMC",           "libVMC");
  gROOT->LoadClass("AliVEvent",            "libSTEERBase");
  gROOT->LoadClass("AliESDEvent",          "libESD");
  gROOT->LoadClass("AliAnalysisManager",   "libANALYSIS");
  gROOT->LoadClass("AliAnalysisTaskSE",    "libANALYSISalice");
  gROOT->LoadClass("AliAODForwardMult",    "libPWGLFforward2");

  // --- Setup script path -------------------------------------------
  const char* aliPath   = gSystem->ExpandPathName("$ALICE_ROOT");
  const char* fwd2Path  = 
    gSystem->ExpandPathName("$ALICE_ROOT/PWGLF/FORWARD/analysis2");
  const char* macroPath = gROOT->GetMacroPath();
  gROOT->SetMacroPath(Form(".:%s:%s/trains:%s/scripts",
			   macroPath,fwd2Path,fwd2Path));
  
  // --- Setup include path ------------------------------------------
  gSystem->AddIncludePath(Form("-I%s", fwd2Path));
  gSystem->AddIncludePath(Form("-I%s/trains", fwd2Path));
  gSystem->AddIncludePath(Form("-I%s", aliPath));
  gSystem->AddIncludePath(Form("-I%s/include", aliPath));

  // --- Check that we can find the script ---------------------------
  TString path = gSystem->Which(gROOT->GetMacroPath(), script);
  if (path.IsNull()) { 
    path = gSystem->Which(gROOT->GetMacroPath(), Form("%s.C", script));
    if (path.IsNull()) { 
      Error("BuildTrain", "Cannot find script %s or %s.C in %s", 
	    script, script, gROOT->GetMacroPath());
      return false;
    }
  }
  
  // --- Compile TrainSetup.C ----------------------------------------
  Int_t ret = gROOT->LoadMacro(Form("%s/trains/TrainSetup.C+%s", 
				    fwd2Path, extra));
  if (ret != 0) { 
    Error("BuildTrain", "Failed to build TrainSetup");
    return false;
  }
  
  // --- Possibly make a temporary copy ------------------------------
  if (useTmp) { 
    TString tmp("Train");
    FILE*   fp = gSystem->TempFileName(tmp, ".");
    fclose(fp);
    gSystem->Unlink(tmp);
    tmp.Append(".C");
    
    gSystem->CopyFile(path, tmp);
    path = tmp;
  }

  // --- Now compile the thing ---------------------------------------
  ret = gROOT->LoadMacro(Form("%s+%s", path.Data(), extra));

  // --- If we did a temporary file, remove stuff --------------------
  if (useTmp) {
    gSystem->Unlink(path);
    path.ReplaceAll(".C",   "_C.d");  gSystem->Unlink(path);
    path.ReplaceAll("_C.d", "_C.so"); gSystem->Unlink(path);
  }

  // --- Warning in case of problems ---------------------------------
  if (ret != 0) 
    Warning("BuildSetup", "Failed to build %s (%s)", path.Data(), script);

  return ret == 0;
}
//
// EOF
//