o2aod_mc_to_hepmc.cxx

Go to the documentation of this file.

// Copyright 2023-2099 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.
 
#include <Framework/AnalysisHelpers.h>
#include <Framework/AnalysisTask.h>
#include <Generators/AODToHepMC.h>
//--------------------------------------------------------------------
struct AodToHepmc {
  using Converter = o2::eventgen::AODToHepMC;
 
  struct : o2::framework::ConfigurableGroup {
    o2::framework::Configurable<std::string> dump{"hepmc-dump", "",
                                                  "Dump HepMC event to output"};
    o2::framework::Configurable<bool> onlyGen{"hepmc-only-generated", false,
                                              "Only export generated"};
    o2::framework::Configurable<bool> useTree{"hepmc-use-tree", false,
                                              "Export as tree"};
    o2::framework::Configurable<int> precision{"hepmc-precision", 8,
                                               "Export precision in dump"};
    o2::framework::Configurable<bool> recenter{"hepmc-recenter", false,
                                               "Recenter the events at (0,0,0,0)"};
  } configs;
 
  Converter mConverter;
 
  o2::framework::Service<o2::framework::AODToHepMCPostRun> trigger;
 
  using Headers = Converter::Headers;
  using Header = Converter::Header;
  using Tracks = Converter::Tracks;
  using XSections = Converter::XSections;
  using XSection = Converter::XSection;
  using PdfInfos = Converter::PdfInfos;
  using PdfInfo = Converter::PdfInfo;
  using HeavyIons = Converter::HeavyIons;
  using HeavyIon = Converter::HeavyIon;
  void init(o2::framework::InitContext&)
  {
    mConverter.configs = {(std::string)configs.dump, (bool)configs.onlyGen, (bool)configs.useTree, (int)configs.precision, (bool)configs.recenter};
    mConverter.init();
    trigger->ptr = &mConverter;
  }
  void process(Header const& collision,
               Tracks const& tracks,
               XSections const& xsections,
               PdfInfos const& pdfs,
               HeavyIons const& heavyions)
  {
    // Do not run this if --hepmc-no-aux was passed
    if (doPlain) {
      return;
    }
    LOG(debug) << "=== Processing everything ===";
    mConverter.startEvent();
    mConverter.process(collision,
                       xsections,
                       pdfs,
                       heavyions);
    mConverter.process(collision, tracks);
    mConverter.endEvent();
  }
  void processPlain(Header const& collision,
                    Tracks const& tracks)
  {
    // Do not run this if --hepmc-no-aux was not passed
    if (not doPlain) {
      return;
    }
 
    LOG(debug) << "=== Processing only tracks ===";
    mConverter.startEvent();
    mConverter.process(collision, tracks);
    mConverter.endEvent();
  }
  decltype(o2::framework::ProcessConfigurable{&AodToHepmc::processPlain,
                                              "hepmc-no-aux", false,
                                              "Do not process auxiliary "
                                              "information"})
    doPlain = o2::framework::ProcessConfigurable{&AodToHepmc::processPlain,
                                                 "hepmc-no-aux", false,
                                                 "Do not process auxiliary "
                                                 "information"};
};
 
//--------------------------------------------------------------------
// This _must_ be included after our "customize" function above, or
// that function will not be taken into account.
#include <Framework/runDataProcessing.h>
 
//--------------------------------------------------------------------
using WorkflowSpec = o2::framework::WorkflowSpec;
using DataProcessorSpec = o2::framework::DataProcessorSpec;
using ConfigContext = o2::framework::ConfigContext;
 
WorkflowSpec defineDataProcessing(ConfigContext const& cfg)
{
  using o2::framework::adaptAnalysisTask;
 
  // Task: Two entry: header, tracks, and header, tracks, auxiliary
  return WorkflowSpec{adaptAnalysisTask<AodToHepmc>(cfg)};
}
//
// EOF
//