ExtractELoss.C

Go to the documentation of this file.

//____________________________________________________________________
void
ExtractELoss(const char* fname = "forward_eloss.root",
       Bool_t   mc=false, 
       ULong_t  runNo=0, 
       UShort_t sys=0,
       UShort_t sNN=0,
       Short_t  fld=999,
       Bool_t   sat=false)
{
  const char* fwd = "$ALICE_PHYSICS/PWGLF/FORWARD/analysis2";
  gSystem->AddIncludePath(Form("-I%s "
             "-I$ALICE_ROOT/include "
             "-I$ALICE_PHYSICS/include ", fwd));
  gROOT->Macro(Form("%s/scripts/LoadLibs.C", fwd));
  gROOT->LoadMacro(Form("%s/corrs/CorrExtractor.C+g", fwd));

  TString fileName(fname);
  TString sumDir("ForwardELossSums"); 
  TString resDir("ForwardELossResults");
  if (fileName.Contains("trending")) 
    sumDir = resDir = "forwardQAResults";
 
  CorrExtractor fmdEx(&AliForwardCorrectionManager::Instance());
  fmdEx.SetMC(mc);
  if (runNo >  0)   fmdEx.fRunNo     = runNo;
  if (sys   >  0)   fmdEx.fSys       = sys;
  if (sNN   >  0)   fmdEx.fSNN       = sNN;
  if (fld   <  999) fmdEx.fField     = fld;
  if (sat)          fmdEx.fSatellite = sat;
  if (fmdEx.Init(fname, sumDir, "fmd_corrections.root")) 
    fmdEx.Extract(AliFMDCorrELossFit::Class(),
      Form("%s/fmdEnergyFitter",resDir.Data()));

  std::ofstream f("Upload.C");
  f << "// Generated by ExtractELoss.C\n"
    << "TString MakeDest(const TString& dest, const TString& fname)\n"
    << "{\n"
    << "  TString tmp(dest);\n"
    << "  if (!tmp.IsNull()) {\n"
    << "    if (!tmp.EndsWith(\"/\")) tmp.Append(\"/\");\n"
    << "    tmp.Append(fname);\n"
    << "  }\n"
    << "  return tmp;\n"
    << "}\n\n"
    << "void Upload(const TString& dest=\"\")\n"
    << "{\n"
    << "  gROOT->Macro(\"" << fwd << "/scripts/LoadLibs.C\");\n"
    << "  \n"
    << "  const char* fmdFile = \"fmd_corrections.root\";\n"
    << "  TString fdest = MakeDest(dest, fmdFile);\n"
    << "  \n"
    << "  AliForwardCorrectionManager::Instance().Append(fmdFile, fdest);\n"
    << "}\n"
    << "// EOF\n"
    << std::endl;
  f.close();
}

//____________________________________________________________________
//
// EOF
//