python_output.cc

Go to the documentation of this file.

/*
 *  CepGen: a central exclusive processes event generator
 *  Copyright (C) 2022-2024  Laurent Forthomme
 *
 *  This program is free software: you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License as published by
 *  the Free Software Foundation, either version 3 of the License, or
 *  any later version.
 *
 *  This program is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *  GNU General Public License for more details.
 *
 *  You should have received a copy of the GNU General Public License
 *  along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */
 
// clang-format off
#include "CepGenAddOns/PythonWrapper/Environment.h"
#include "CepGenAddOns/PythonWrapper/Error.h"
#include "CepGenAddOns/PythonWrapper/Utils.h"
// clang-format on
#include "CepGen/Core/RunParameters.h"
#include "CepGen/Generator.h"
#include "CepGen/Modules/ProcessFactory.h"
#include "CepGen/Process/Process.h"
#include "CepGen/Utils/ArgumentsParser.h"
#include "CepGen/Utils/Test.h"
#include "CepGenAddOns/PythonWrapper/ConfigWriter.h"
 
using namespace std;
 
int main(int argc, char* argv[]) {
  string output_file, process;
  cepgen::ArgumentsParser(argc, argv)
      .addOptionalArgument("output,o", "Output python file", &output_file, "py_cfg.py")
      .addOptionalArgument("process,p", "Process name", &process, "")
      .parse();
 
  cepgen::Generator gen;
  if (process.empty())
    process = *cepgen::ProcessFactory::get().modules().begin();
 
  {
    gen.runParameters().setProcess(cepgen::ProcessFactory::get().build(process));
    cepgen::python::ConfigWriter py(cepgen::ParametersList().set("filename", output_file));
    py << gen.runParameters();
  }
 
  try {
    auto env = cepgen::python::Environment(cepgen::ParametersList{});
    const auto path = cepgen::python::pythonPath(output_file);
    env.setProgramName(path);
    auto obj = cepgen::python::ObjectPtr::importModule(path);
    CG_TEST(obj != nullptr, "Module import");
    if (!obj)
      return -1;
    //CG_LOG << cepgen::python::ObjectPtr(PyObject_GenericGetDict(obj.get(), nullptr)).value<cepgen::ParametersList>();
    auto proc = obj.attribute("process");
    CG_TEST(proc != nullptr, "'process' attribute retrieval");
    const auto proc_params = proc.value<cepgen::ParametersList>();
    CG_TEST_EQUAL(proc_params.name(), process, "Process name conservation");
  } catch (const cepgen::python::Error& err) {
    err.dump();
  }
  CG_TEST_SUMMARY;
}