Implemented automatic parallelisation for external generator on HY

Author: jackal1-66

Generators/include/Generators/Generator.h

@@ -99,6 +99,13 @@ class Generator : public FairGenerator
   const std::vector<TParticle>& getParticles() const { return mParticles; }; //!
   static unsigned int getTotalNEvents() { return gTotalNEvents; };
 
+  // Check if simulation is running in Hyperloop mode
+  static bool isHyperloop();
+
+  // Number of parallel sub-generator clones a Hyperloop-aware generator expands into
+  // (currently used by the external-generator-to-hybrid expansion in GeneratorFactory).
+  static constexpr int NHyperloopParallelGenerators = 8;
+
   /** other **/
   void clearParticles() { mParticles.clear(); };
 

Generators/include/Generators/GeneratorPythia8.h

@@ -264,7 +264,7 @@ class GeneratorPythia8 : public Generator
   void seedGenerator();
 
   // Hyperloop flag
-  const bool mIsHyperloop = std::getenv("IS_HYPERLOOP") && std::atoi(std::getenv("IS_HYPERLOOP"));
+  const bool mIsHyperloop = Generator::isHyperloop();
 
   /** Pythia8 **/
   // Show banner only when not running in Hyperloop

Generators/src/Generator.cxx

@@ -28,6 +28,7 @@
 #include "TGrid.h"
 #include "CCDB/BasicCCDBManager.h"
 #include <filesystem>
+#include <cstdlib>
 #ifdef GENERATORS_WITH_TPCLOOPERS
 #include "Generators/TPCLoopers.h"
 #include "Generators/TPCLoopersParam.h"
@@ -40,6 +41,12 @@ namespace eventgen
 
 std::atomic<int> Generator::InstanceCounter{0};
 unsigned int Generator::gTotalNEvents = 0;
+
+bool Generator::isHyperloop()
+{
+  static const bool isHY = std::getenv("IS_HYPERLOOP") && std::atoi(std::getenv("IS_HYPERLOOP"));
+  return isHY;
+}
 /*****************************************************************/
 /*****************************************************************/
 

Generators/src/GeneratorFactory.cxx

@@ -34,6 +34,11 @@
 #if defined(GENERATORS_WITH_PYTHIA8) && defined(GENERATORS_WITH_HEPMC3)
 #include <Generators/GeneratorHybrid.h>
 #include <Generators/GeneratorHybridParam.h>
+#include <rapidjson/document.h>
+#include <rapidjson/writer.h>
+#include <rapidjson/ostreamwrapper.h>
+#include <fstream>
+#include <unistd.h>
 #endif
 #include <Generators/PrimaryGenerator.h>
 #include <Generators/BoxGunParam.h>
@@ -49,6 +54,48 @@ namespace o2
 namespace eventgen
 {
 
+#if defined(GENERATORS_WITH_PYTHIA8) && defined(GENERATORS_WITH_HEPMC3)
+// Builds a parallel GeneratorHybrid JSON configuration out of
+// 8 (Generator::NHyperloopParallelGenerators) identical "external" sub-generator entries,
+// using the configured GeneratorExternalParam. This makes the simulation parallelisation automatic
+// on Hyperloop.
+// To-do: Define the behaviour with ini configuration which are already definining a hybrid gen
+std::string buildHyperloopExternalHybridConfig(GeneratorExternalParam const& extparams)
+{
+  rapidjson::Document doc;
+  doc.SetObject();
+  auto& alloc = doc.GetAllocator();
+  doc.AddMember("mode", "parallel", alloc);
+
+  rapidjson::Value generators(rapidjson::kArrayType);
+  rapidjson::Value fractions(rapidjson::kArrayType);
+  for (int i = 0; i < Generator::NHyperloopParallelGenerators; ++i) {
+    rapidjson::Value config(rapidjson::kObjectType);
+    config.AddMember("fileName", rapidjson::Value(extparams.fileName.c_str(), alloc), alloc);
+    config.AddMember("funcName", rapidjson::Value(extparams.funcName.c_str(), alloc), alloc);
+    config.AddMember("iniFile", "", alloc);
+
+    rapidjson::Value generator(rapidjson::kObjectType);
+    generator.AddMember("name", "external", alloc);
+    generator.AddMember("config", config, alloc);
+    generators.PushBack(generator, alloc);
+    fractions.PushBack(1, alloc);
+  }
+  doc.AddMember("generators", generators, alloc);
+  doc.AddMember("fractions", fractions, alloc);
+
+  std::string path = "hyperloop_exttohybrid_" + std::to_string(getpid()) + ".json";
+  std::ofstream ofs(path);
+  if (!ofs.is_open()) {
+    LOG(fatal) << "Failed to open " << path << " for writing the Hyperloop hybrid generator configuration";
+  }
+  rapidjson::OStreamWrapper osw(ofs);
+  rapidjson::Writer<rapidjson::OStreamWrapper> writer(osw);
+  doc.Accept(writer);
+  return path;
+}
+#endif
+
 // reusable helper class
 // main purpose is to init a FairPrimGen given some (Sim)Config
 void GeneratorFactory::setPrimaryGenerator(o2::conf::SimConfig const& conf, FairPrimaryGenerator* primGen)
@@ -89,9 +136,20 @@ void GeneratorFactory::setPrimaryGenerator(o2::conf::SimConfig const& conf, Fair
   o2::O2DatabasePDG::addALICEParticles(TDatabasePDG::Instance());
   auto genconfig = conf.getGenerator();
 #if defined(GENERATORS_WITH_PYTHIA8) && defined(GENERATORS_WITH_HEPMC3)
-  if (GeneratorHybridParam::Instance().switchExtToHybrid && (genconfig.compare("external") == 0 || genconfig.compare("extgen") == 0)) {
-    LOG(info) << "Switching external generator to hybrid mode";
-    genconfig = "hybrid";
+  std::string hyperloopExtHybridConfigFile; // set when IS_HYPERLOOP is defined
+  if (genconfig.compare("external") == 0 || genconfig.compare("extgen") == 0) {
+    if (GeneratorHybridParam::Instance().switchExtToHybrid) {
+      LOG(info) << "Switching external generator to hybrid mode";
+      genconfig = "hybrid";
+    } else if (Generator::isHyperloop()) {
+      // Running under Hyperloop: transparently expand the single external generator
+      // configuration into a parallel hybrid of Generator::NHyperloopParallelGenerators
+      // clones, to increase on-the-fly MC-generation throughput.
+      LOG(info) << "IS_HYPERLOOP detected: expanding external generator into "
+                << Generator::NHyperloopParallelGenerators << " parallel hybrid sub-generators";
+      hyperloopExtHybridConfigFile = buildHyperloopExternalHybridConfig(GeneratorExternalParam::Instance());
+      genconfig = "hybrid";
+    }
   }
 #endif
   LOG(info) << "** Generator to use: '" << genconfig << "'";
@@ -269,8 +327,7 @@ void GeneratorFactory::setPrimaryGenerator(o2::conf::SimConfig const& conf, Fair
 #if defined(GENERATORS_WITH_PYTHIA8) && defined(GENERATORS_WITH_HEPMC3)
   } else if (genconfig.compare("hybrid") == 0) { // hybrid using multiple generators
     LOG(info) << "Init hybrid generator";
-    auto& hybridparam = GeneratorHybridParam::Instance();
-    std::string config = hybridparam.configFile;
+    std::string config = !hyperloopExtHybridConfigFile.empty() ? hyperloopExtHybridConfigFile : GeneratorHybridParam::Instance().configFile;
     // check if config string points to an existing and not empty file
     if (config.empty()) {
       LOG(fatal) << "No configuration file provided for hybrid generator";