GeneratorHepMC.h

Go to the documentation of this file.

// 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.
 
 
#ifndef ALICEO2_EVENTGEN_GENERATORHEPMC_H_
#define ALICEO2_EVENTGEN_GENERATORHEPMC_H_
 
#include "Generators/Generator.h"
#include "Generators/GeneratorFileOrCmd.h"
#include "Generators/GeneratorHepMCParam.h"
#include "Generators/GeneratorFileOrCmdParam.h"
#include <iosfwd>
#include <memory>
#include <string>
#include <vector>
 
#ifdef GENERATORS_WITH_HEPMC3_DEPRECATED
namespace HepMC
{
class Reader;
class GenEvent;
class FourVector;
} // namespace HepMC
#else
namespace HepMC3
{
class Reader;
class GenEvent;
class FourVector;
class GenParticle;
} // namespace HepMC3
#endif
 
namespace o2
{
namespace conf
{
class SimConfig;
}
namespace eventgen
{
 
/*****************************************************************/
/*****************************************************************/
 
class GeneratorHepMC : public Generator, public GeneratorFileOrCmd
{
 
 public:
  GeneratorHepMC();
  GeneratorHepMC(const Char_t* name,
                 const Char_t* title = "ALICEo2 HepMC Generator");
  ~GeneratorHepMC() override;
 
  Bool_t Init() override;
 
  void setup(const GeneratorFileOrCmdParam& param0,
             const HepMCGenConfig& param,
             const conf::SimConfig& config);
  // Generator configuration from external local parameters
  void setup(const FileOrCmdGenConfig& param0,
             const HepMCGenConfig& param,
             const conf::SimConfig& config);
  Bool_t generateEvent() override;
  Bool_t importParticles() override;
 
  void stop() override;
 
  void setEventsToSkip(uint64_t val) { mEventsToSkip = val; };
  void setVersion(const int& ver) { mVersion = ver; };
 
 protected:
  GeneratorHepMC(const GeneratorHepMC&);
  GeneratorHepMC& operator=(const GeneratorHepMC&);
 
#ifdef GENERATORS_WITH_HEPMC3_DEPRECATED
  const HepMC::FourVector getBoostedVector(const HepMC::FourVector& vector, Double_t boost);
#else
  const HepMC3::FourVector getBoostedVector(const HepMC3::FourVector& vector, Double_t boost);
#endif
 
  void updateHeader(o2::dataformats::MCEventHeader* eventHeader) override;
  void setupHepMC(const HepMCGenConfig& param);
  bool makeReader();
  void establishEventOrder();
  bool buildIndex(const std::string& filename);
  bool readEntry(int entry);
  Bool_t generateEventOrdered();
 
  typedef bool (*Select)(std::shared_ptr<const HepMC3::GenParticle>);
  void pruneEvent(Select select);
 
  uint64_t mEventsToSkip = 0;
  int mVersion = 0;
  std::shared_ptr<HepMC3::Reader> mReader;
  HepMC3::GenEvent* mEvent = nullptr;
  bool mPrune; 
  std::string mCurrentFileName; 
  std::vector<int> mEventOrder; 
  int mEventCounter = 0; 
  int mEventsServed = 0; 
  int mEventsAvailable = 0; 
  int mLastEntryRead = -1; 
  bool mRandomize = false; 
  bool mRoundRobin = false; 
  bool mReshuffleOnRepeat = true; 
  unsigned int mRngSeed = 0; 
  bool mIndexedHepMC2 = false; 
  std::shared_ptr<std::istream> mIndexedStream; 
  std::vector<std::streamoff> mEventOffsets; 
 
  ClassDefOverride(GeneratorHepMC, 2);
 
}; 
/*****************************************************************/
/*****************************************************************/
 
} // namespace eventgen
} // namespace o2
 
#endif /* ALICEO2_EVENTGEN_GENERATORHEPMC_H_ */