ExtractMCCorr.C

Go to the documentation of this file.

//____________________________________________________________________
void
ExtractMCCorr(const char* fname,
        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_PHYSICS/include", fwd));
  gROOT->Macro(Form("%s/scripts/LoadLibs.C", fwd));
  gROOT->LoadMacro(Form("%s/corrs/CorrExtractor.C++g", fwd));
  
  CorrExtractor fmdEx(&AliForwardCorrectionManager::Instance());
  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, "ForwardCorrSums", "fmd_corrections.root")) 
    fmdEx.Extract(AliFMDCorrSecondaryMap::Class(),
      "ForwardCorrResults");

  CorrExtractor spdEx(&AliCentralCorrectionManager::Instance());
  if (runNo > 0)   spdEx.fRunNo     = runNo;
  if (sys   > 0)   spdEx.fSys       = sys;
  if (sNN   > 0)   spdEx.fSNN       = sNN;
  if (fld   < 999) spdEx.fField     = fld;
  if (sat)         spdEx.fSatellite = sat;
  if (spdEx.Init(fname, "CentralCorrSums", "spd_corrections.root")) {
    spdEx.Extract(AliCentralCorrSecondaryMap::Class(), "CentralCorrResults");
    spdEx.Extract(AliCentralCorrAcceptance::Class(), "CentralCorrResults");
  }

  std::ofstream f("Upload.C");
  f << "// Generated by ExtractMCCorr.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"
    << "  const char* spdFile = \"spd_corrections.root\";\n"
    << "  TString fdest = MakeDest(dest, fmdFile);\n"
    << "  TString sdest = MakeDest(dest, spdFile);\n"
    << "  \n"
    << "  AliForwardCorrectionManager::Instance().Append(fmdFile, fdest);\n"
    << "  AliCentralCorrectionManager::Instance().Append(spdFile, sdest);\n"
    << "}\n"
    << "// EOF\n"
    << std::endl;
  f.close();
}

//____________________________________________________________________
//
// EOF
//