Pulled from first hybrid generator implementation

Author: jackal1-66

Generators/CMakeLists.txt

@@ -24,6 +24,8 @@ o2_add_library(Generators
                        src/GeneratorTGenerator.cxx
                        src/GeneratorExternalParam.cxx
                        src/GeneratorFromFile.cxx
+                       src/GeneratorHybrid.cxx
+                       src/GeneratorHybridParam.cxx
                        src/GeneratorFromO2KineParam.cxx
                        src/GeneratorFileOrCmd.cxx
                        src/GeneratorFileOrCmdParam.cxx
@@ -68,6 +70,8 @@ set(headers
     include/Generators/GeneratorTGenerator.h
     include/Generators/GeneratorExternalParam.h
     include/Generators/GeneratorFromFile.h
+    include/Generators/GeneratorHybrid.h
+    include/Generators/GeneratorHybridParam.h
     include/Generators/GeneratorFromO2KineParam.h
     include/Generators/GeneratorFileOrCmd.h
     include/Generators/GeneratorFileOrCmdParam.h

Generators/include/Generators/GeneratorHybrid.h

@@ -0,0 +1,73 @@
+// Copyright 2019-2020 CERN and copyright holders of ALICE O2.
+// See https://alice-o2.web.cern.ch/copyright for details of the copyright holders.
+// All rights not expressly granted are reserved.
+//
+// This software is distributed under the terms of the GNU General Public
+// License v3 (GPL Version 3), copied verbatim in the file "COPYING".
+//
+// In applying this license CERN does not waive the privileges and immunities
+// granted to it by virtue of its status as an Intergovernmental Organization
+// or submit itself to any jurisdiction.
+
+/// \author M. Giacalone - October 2024
+
+#ifndef ALICEO2_EVENTGEN_GENERATORHYBRID_H_
+#define ALICEO2_EVENTGEN_GENERATORHYBRID_H_
+
+#include "Generators/Generator.h"
+#include "Generators/BoxGenerator.h"
+#include <Generators/GeneratorPythia8.h>
+#include <Generators/GeneratorFromFile.h>
+#include "SimulationDataFormat/MCEventHeader.h"
+#include "SimulationDataFormat/MCGenProperties.h"
+#include "SimulationDataFormat/ParticleStatus.h"
+#include "Generators/GeneratorHybridParam.h"
+#include "Generators/GeneratorFromO2KineParam.h"
+#include <Generators/GeneratorPythia8.h>
+#include <TRandom3.h>
+#include "CommonUtils/ConfigurationMacroHelper.h"
+#include "FairGenerator.h"
+#include <DetectorsBase/Stack.h>
+
+namespace o2
+{
+namespace eventgen
+{
+
+class GeneratorHybrid : public Generator
+{
+
+ public:
+  GeneratorHybrid() = default;
+  GeneratorHybrid(std::vector<std::string> gens);
+  ~GeneratorHybrid() = default;
+  
+  Bool_t Init() override;
+  Bool_t generateEvent() override;
+  Bool_t importParticles() override;
+
+ private:
+  o2::eventgen::Generator* currentgen = nullptr;
+  std::vector<std::unique_ptr<o2::eventgen::Generator>> gens;
+  std::vector<std::unique_ptr<FairGenerator>> mExternal;
+  std::vector<int> mExtFlag;
+  std::vector<std::unique_ptr<FairPrimaryGenerator>> mPrimaryExternal;
+  std::vector<o2::dataformats::MCEventHeader> mExtEventHeader;
+  const std::vector<std::string> generatorNames = {"extkinO2", "boxgen", "external", "hepmc", "pythia8", "pythia8pp", "pythia8hi", "pythia8hf", "pythia8powheg"};
+  std::vector<std::string> mGens;
+  std::vector<std::string> mConfigs;
+  std::vector<std::string> mConfsPythia8;
+  bool mRandomize = false;
+  std::vector<int> mFractions;
+  int mseqCounter = 0;
+  int mCurrentFraction = 0;
+  int mIndex = 0;
+  int mCurrentExt = 0;
+
+  std::unique_ptr<o2::data::Stack> mExtStack;
+};
+
+} // namespace eventgen
+} // namespace o2
+
+#endif

Generators/include/Generators/GeneratorHybridParam.h

@@ -0,0 +1,42 @@
+// Copyright 2019-2020 CERN and copyright holders of ALICE O2.
+// See https://alice-o2.web.cern.ch/copyright for details of the copyright holders.
+// All rights not expressly granted are reserved.
+//
+// This software is distributed under the terms of the GNU General Public
+// License v3 (GPL Version 3), copied verbatim in the file "COPYING".
+//
+// In applying this license CERN does not waive the privileges and immunities
+// granted to it by virtue of its status as an Intergovernmental Organization
+// or submit itself to any jurisdiction.
+
+/// \author M. Giacalone - October 2024
+
+#ifndef ALICEO2_EVENTGEN_GENERATORHYBRIDPARAM_H_
+#define ALICEO2_EVENTGEN_GENERATORHYBRIDPARAM_H_
+
+#include "CommonUtils/ConfigurableParam.h"
+#include "CommonUtils/ConfigurableParamHelper.h"
+
+namespace o2
+{
+namespace eventgen
+{
+
+/**
+ ** a parameter class/struct to keep the settings of
+ ** the Hybrid event generator and
+ ** allow the user to modify them
+ **/
+
+struct GeneratorHybridParam : public o2::conf::ConfigurableParamHelper<GeneratorHybridParam> {
+  std::string Generators = ""; // generators to be used in the cocktail, each divided by a comma
+  std::string Configs = "";    // configurations for the generators, each divided by a colon 
+  bool Randomize = false;      // randomize the order of the generators, if not generator using fractions
+  std::string Fractions = "";  // events fractions for each generator, works only for sequence
+  O2ParamDef(GeneratorHybridParam, "GeneratorHybrid");
+};
+
+} // end namespace eventgen
+} // end namespace o2
+
+#endif // ALICEO2_EVENTGEN_GENERATORHYBRIDPARAM_H_

Generators/src/GeneratorFactory.cxx

@@ -18,6 +18,7 @@
 #include <fairlogger/Logger.h>
 #include <SimConfig/SimConfig.h>
 #include <Generators/GeneratorFromFile.h>
+#include <Generators/GeneratorHybrid.h>
 #include <Generators/GeneratorTParticle.h>
 #include <Generators/GeneratorTParticleParam.h>
 #ifdef GENERATORS_WITH_PYTHIA8
@@ -26,6 +27,7 @@
 #endif
 #include <Generators/GeneratorTGenerator.h>
 #include <Generators/GeneratorExternalParam.h>
+#include <Generators/GeneratorHybridParam.h>
 #include "Generators/GeneratorFromO2KineParam.h"
 #ifdef GENERATORS_WITH_HEPMC3
 #include <Generators/GeneratorHepMC.h>
@@ -240,6 +242,19 @@ void GeneratorFactory::setPrimaryGenerator(o2::conf::SimConfig const& conf, Fair
         primGen->AddGenerator(boxGen);
       }
     }
+  } else if (genconfig.compare("hybrid") == 0) { // hybrid using multiple generators
+    LOG(info) << "Init hybrid generator";
+    auto& hybridparam = GeneratorHybridParam::Instance();
+    std::string genslist = hybridparam.Generators;
+    LOG(info) << "Generators list: " << genslist;
+    std::vector<std::string> generators;
+    std::stringstream ss(genslist);
+    std::string item;
+    while (std::getline(ss, item, ',')) {
+      generators.push_back(item);
+    }
+    auto hybrid = new o2::eventgen::GeneratorHybrid(generators);
+    primGen->AddGenerator(hybrid);
   } else {
     LOG(fatal) << "Invalid generator";
   }