MakeFMDELossTrain.C

Go to the documentation of this file.

#include "TrainSetup.C"

//====================================================================
class MakeFMDELossTrain : public TrainSetup
{
public:
  MakeFMDELossTrain(const char* name  = "FMD Energy Loss")
    : TrainSetup(name)
  {
    fOptions.Add("only-mb", "Only collect statistics from MB events");
    fOptions.Add("config",  "FILE", "Configuration", "elossFitConfig.C");
    fOptions.Add("corr",    "DIR",  "Corrections dir", "");
    fOptions.Add("dead",    "FILE", "Additional dead-map script", "");
    fOptions.Add("low-cut", "VALUE","Low cut on eloss", -1);
    fOptions.Set("type", "ESD");
  }
protected:
  //------------------------------------------------------------------
  virtual AliAnalysisManager* CreateAnalysisManager(const char* name)
  {
    AliAnalysisManager* mgr = TrainSetup::CreateAnalysisManager(name);
    // mgr->SetAutoBranchLoading(false);
    return mgr;
  }
  //__________________________________________________________________
  void CreateTasks(AliAnalysisManager* mgr)
  {
    // --- Output file name ------------------------------------------
    AliAnalysisManager::SetCommonFileName("forward_eloss.root");

    // --- Load libraries/pars ---------------------------------------
    fRailway->LoadLibrary("PWGLFforward2");
    
    // --- Set load path ---------------------------------------------
    gROOT->SetMacroPath(Form("%s:$(ALICE_PHYSICS)/PWGLF/FORWARD/analysis2",
           gROOT->GetMacroPath()));

    // --- Check if this is MC ---------------------------------------
    Bool_t   mc     = HasMCHandler();
    Bool_t   onlyMB = fOptions.AsBool("only-mb");
    Double_t lowCut = fOptions.AsDouble("low-cut", -1);
    TString  config = fOptions.Get("config"); 
    TString  corrs  = "";
    TString  dead   = "";
    if (fOptions.Has("corr")) corrs = fOptions.Get("corr"); 
    if (fOptions.Has("dead")) dead = fOptions.Get("dead"); 

    // --- Add the task ----------------------------------------------
    AliAnalysisTask* task = 
      CoupleSECar("AddTaskFMDELoss.C", Form("%d,%d,%f,\"%s\",\"%s\",\"%s\"", 
              mc, onlyMB,  lowCut,
              config.Data(), 
              corrs.Data(),
              dead.Data()), AliVEvent::kAny);
    if (!task)
      Fatal("CoupleCars", "Failed to add energy loss task");
    
    fRailway->LoadAux(gSystem->Which(gROOT->GetMacroPath(), config), true);
    if (!corrs.IsNull())
      fRailway->LoadAux(Form("%s/fmd_corrections.root",corrs.Data()), true);
    if (!dead.IsNull())
      fRailway->LoadAux(Form("%s",dead.Data()), true);
    
  }
  virtual void CreateCentralitySelection(Bool_t mc)
  {
    if (!fOptions.Has("cent")) return;
    TrainSetup::CreateCentralitySelection(mc);    
  }
  AliVEventHandler* CreateOutputHandler(UShort_t) { return 0; }

  //__________________________________________________________________
  const char* ClassName() const { return "MakeFMDELossTrain"; }
  //__________________________________________________________________
  void SaveSetup(Bool_t asShellScript)
  {
    TrainSetup::SaveSetup(asShellScript);
    SaveExtract();
    SaveSummarize();
    SaveReFit();
  }
  void SaveExtract()
  {
    std::ofstream f("Extract.C");
    if (!f) { 
      Error("SaveSetup", "Failed to open Extract.C");
      return;
    }
    f << "// Generated by " << ClassName() << "\n"
      << "void Extract(Bool_t mc=" << HasMCHandler() << ",\n"
      << "             const char* file=\"forward_eloss.root\")\n"
      << "{\n"
      << "  const char* fwd=\"$ALICE_PHYSICS/PWGLF/FORWARD/analysis2\";\n"
      << "  gROOT->LoadMacro(Form(\"%s/corrs/ExtractELoss.C\",fwd));\n"
      << "  ExtractELoss(file,mc);\n" 
      << "}\n"
      << "// EOF" << std::endl;
    f.close();
  }
  void SaveSummarize()
  {
    std::ofstream f("Summarize.C");
    if (!f) { 
      Error("SaveSummarize", "Failed to open Summarize.C script");
      return;
    }
    f << std::boolalpha 
      << "// Generated by " << ClassName() << "\n"
      << "//\n"
      << "void Summarize(Bool_t mc=" << HasMCHandler() << ",\n"
      << "               const char* filename=\"forward_eloss.root\")\n"
      << "{\n"
      << "  const char* fwd=\"$ALICE_PHYSICS/PWGLF/FORWARD/analysis2\";\n"
      << "  gROOT->LoadMacro(Form(\"%s/corrs/DrawCorrELoss.C\",fwd));\n"
      << "  DrawCorrELoss(mc,filename);\n"
      << "}\n"
      << "// EOF" << std::endl;
    f.close();
  }
  void SaveReFit()
  {
    std::ofstream f("ReFit.C");
    if (!f) { 
      Error("SaveRerun", "Failed to open ReFit.C script");
      return;
    }
    f << std::boolalpha 
      << "// Generated by " << ClassName() << "\n"
      << "// If force=true, then force set parameters\n"
      << "// If shift=true, enable extra shift in Delta from sigma/xi\n"
      << "//\n"
      << "void ReFit(Bool_t      force=false,\n"
      << "           Bool_t      shift=true,\n"
      << "           const char* filename=\"forward_eloss.root\")\n"
      << "{\n"
      << "  const char* fwd=\"$ALICE_PHYSICS/PWGLF/FORWARD/analysis2\";\n"
      << "  gROOT->LoadMacro(Form(\"%s/corrs/RerunELossFits.C\",fwd));\n"
      << "  RerunELossFits(force,filename);\n"
      << "}\n"
      << "// EOF" << std::endl;
    f.close();
  }
  void PostShellCode(std::ostream& f)
  {
    f << "  mc=" << std::boolalpha << HasMCHandler() << "\n"
      << "  echo \"=== Extracting Corrections ...\"\n"
      << "  aliroot -l -b -q ${prefix}Extract.C\\($mc\\)\n"
      << "  echo \"=== Summarizing results ...\"\n"
      << "  aliroot -l -b -q ${prefix}Summarize.C\\($mc\\)\n"
      << "  if test x$dest = x ; then return ; fi\n"
      << "  echo \"=== Uploading to ${dest} ...\"\n"
      << "  aliroot -l -b -q Upload.C\\(\\\"${dest}\\\"\\);\n"
      << std::endl;
  }
};

//
// EOF
//