Merge branch 'master' of https://git.cern.ch/reps/AliRoot

[u/mrichter/AliRoot.git] / PWGLF / FORWARD / analysis2 / corrs / ExtractELoss.C

/**
 * Script to draw the energy loss fits 
 * 
 * @ingroup pwglf_forward_scripts_corr
 */

//____________________________________________________________________
/** 
 * Extract the energy loss correction object from file and rename it 
 * according to the settings 
 * 
 * @param fname  File to extract from 
 * @param mc     Whether this is from MC data or not 
 * @param runNo  Run number 
 * @param sys    Collision system (pp, PbPb)
 * @param sNN    Center of mass energy (in GeV) per nucleon
 * @param fld    L3 magnetic field (-5,0,5) in kGaus
 * @param sat    For satellite interactions 
 * 
 * @ingroup pwglf_forward_scripts_corr
 */
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_ROOT/PWGLF/FORWARD/analysis2";
  gSystem->AddIncludePath(Form("-I%s -I$ALICE_ROOT/include", fwd));
  gROOT->Macro(Form("%s/scripts/LoadLibs.C", fwd));
  gROOT->LoadMacro(Form("%s/corrs/CorrExtractor.C++g", fwd));
  
  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, "ForwardELossSums", "fmd_corrections.root")) 
    fmdEx.Extract(AliFMDCorrELossFit::Class(),
		  "ForwardELossResults/fmdEnergyFitter");

  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
//