MakeMCCorrTrain.C

Go to the documentation of this file.

#include "TrainSetup.C"

//====================================================================
class MakeMCCorrTrain : public TrainSetup
{
public:
  MakeMCCorrTrain(const  char* name) 
    : TrainSetup(name)
  {
    fOptions.Set("type", "ESD");
    fOptions.Add("eff", "Effective SPD correction", false);
    fOptions.Add("satellite", "Set-up for satellite-main", false);
    fOptions.Add("reweigh","MODE","Whether to reweigh","");
    fOptions.Add("max-strips", "NUMBER", 
                 "Maximum consequtive strips", 2); 
  }
protected:
  void CreateTasks(AliAnalysisManager* mgr)
  {
    // --- Output file name ------------------------------------------
    AliAnalysisManager::SetCommonFileName("forward_mccorr.root");

    // --- Load libraries/pars ---------------------------------------
    fRailway->LoadLibrary("PWGLFforward2");
    
    // --- Set load path ---------------------------------------------
    gROOT->SetMacroPath(Form("%s:$(ALICE_PHYSICS)/PWGLF/FORWARD/analysis2",
           gROOT->GetMacroPath()));
  
    // --- Options ---------------------------------------------------
    Bool_t   spdEffective = fOptions.Has("eff");
    Bool_t   satellite    = fOptions.AsBool("satellite");
    TString  reweigh      = fOptions.AsString("reweigh","");
    UShort_t maxStrips    = fOptions.AsInt("max-strips");
    UShort_t rflags       = 0;
    if (reweigh.Contains("track")) rflags |= 0x1;
    if (reweigh.Contains("prim"))  rflags |= 0x2;
    if (reweigh.Contains("real"))  rflags |= 0x4;

    // --- Check if this is MC ---------------------------------------
    if (!mgr->GetMCtruthEventHandler()) return;
    
    // --- Task to copy header information ---------------------------
    // gROOT->Macro("AddTaskCopyHeader.C");

    UInt_t mask = AliVEvent::kAny;
    // --- Add the task ----------------------------------------------
    CoupleSECar("AddTaskForwardMCCorr.C",
    Form("%d,0x%x,%d", satellite, rflags, maxStrips), mask); 

    // --- Add the task ----------------------------------------------
    CoupleSECar("AddTaskCentralMCCorr.C",
    Form("%d,%d", satellite, spdEffective), mask);
  }
  //__________________________________________________________________
  void CreatePhysicsSelection(Bool_t mc,
            AliAnalysisManager* mgr)
  {
    TrainSetup::CreatePhysicsSelection(mc, mgr);
#if 0
    // --- Get input event handler -----------------------------------
    AliInputEventHandler* ih =
      dynamic_cast<AliInputEventHandler*>(mgr->GetInputEventHandler());
    if (!ih) 
      Fatal("CreatePhysicsSelection", "Couldn't get input handler (%p)", ih);
    
    // --- Get Physics selection -------------------------------------
    AliPhysicsSelection* ps = 
      dynamic_cast<AliPhysicsSelection*>(ih->GetEventSelection());
    if (!ps) 
      Fatal("CreatePhysicsSelection", "Couldn't get PhysicsSelection (%p)", ps);

    // --- Ignore trigger class when selecting events.  This means ---
    // --- that we get offline+(A,C,E) events too --------------------
    // ps->SetSkipTriggerClassSelection(true);
#endif 
  }
  //__________________________________________________________________
  virtual AliVEventHandler* CreateOutputHandler(UShort_t) { return 0; }
  // void CreateCentralitySelection(Bool_t) {}
  //__________________________________________________________________
  const char* ClassName() const { return "MakeMCCorrTrain"; }
  //__________________________________________________________________
  void SaveSetup(Bool_t asShellScript)
  {
    TrainSetup::SaveSetup(asShellScript);

    SaveExtract();
    SaveSummarize();
    SaveDraw();
  }
  void SaveExtract()
  {
    std::ofstream f("Extract.C");
    if (!f) { 
      Error("SaveSetup", "Failed to open Extract.C");
      return;
    }
    f << "// Generated by " << ClassName() << "\n"
      << "void Extract()\n"
      << "{\n"
      << "  const char* fwd=\"$ALICE_PHYSICS/PWGLF/FORWARD/analysis2\";\n"
      << "  gROOT->LoadMacro(Form(\"%s/corrs/ExtractMCCorr.C\",fwd));\n"
      << "  ExtractMCCorr(\"forward_mccorr.root\");\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 << "// Generated by " << ClassName() << "\n"
      << "// WHAT is a bit mask of\n"
      << "//   0x001     Event inspector\n"
      << "//   0x002     Track density\n"
      << "//   0x004     Vertex bins\n"
      << "//   0x008     Results\n"
      << "//   0x080     Central\n"
      << "//   0x100     Landscape\n"
      << "//   0x200     Pause\n"
      << "//\n"
      << "void Summarize(const char* filename=\"forward_mccorr.root\",\n"
      << "               UShort_t what=0x11F)\n"
      << "{\n"
      << "  const char* fwd=\"$ALICE_PHYSICS/PWGLF/FORWARD/analysis2\";\n"
      << "  gROOT->LoadMacro(Form(\"%s/DrawMCCorrSummary.C\",fwd));\n"
      << "  DrawMCCorrSummary(filename,what);\n"
      << "}\n"
      << "// EOF" << std::endl;
    f.close();
  }
  void SaveDraw()
  {
    std::ofstream f("Draw.C");
    if (!f) { 
      Error("SaveDraw", "Failed to open Draw.C script");
      return;
    }
    f << "// Generated by " << ClassName() << "\n"
      << "// WHAT is a bit mask of\n"
      << "//   0x001     Event inspector\n"
      << "//   0x002     Track density\n"
      << "//   0x004     Vertex bins\n"
      << "//   0x008     Results\n"
      << "//   0x080     Central\n"
      << "//   0x100     Landscape\n"
      << "//   0x200     Pause\n"
      << "//\n"
      << "void Draw(bool details=false)\n"
      << "{\n"
      << "  const char* fwd=\"$ALICE_PHYSICS/PWGLF/FORWARD/analysis2\";\n"
      << "  gROOT->LoadMacro(Form(\"%s/corrs/DrawCorrSecMap.C\",fwd));\n"
      << "  DrawCorrSecMap(999,0,0,999," << fOptions.Has("satellite")
      << ",\"fmd_corrections.root\",details);\n"
      << "}\n"
      << "// EOF" << std::endl;
    f.close();
  }
  void PostShellCode(std::ostream& f)
  {
    f << "  echo \"=== Extracting Corrections ...\"\n"
      << "  aliroot -l -b -q ${prefix}Extract.C\n"
      << "  echo \"=== Summarizing results ...\"\n"
      << "  aliroot -l -b -q ${prefix}Summarize.C\n"
      << "  echo \"=== Draw results ...\"\n"
      << "  aliroot -l -b -q ${prefix}Draw.C\n"
      << "  if test x$dest = x ; then return ; fi\n"
      << "  echo \"=== Uploading to ${dest} ...\"\n"
      << "  aliroot -l -b -q Upload.C\\(\\\"${dest}\\\"\\);"
      << std::endl;
  }
};

//
// EOF
//