GenBase.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2022 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef GENERATORMODULES_GENBASE_H
#define GENERATORMODULES_GENBASE_H
 
#include "AthenaBaseComps/AthAlgorithm.h"
#include "GaudiKernel/ServiceHandle.h"
#include "GaudiKernel/IPartPropSvc.h"
#include "GaudiKernel/IIncidentSvc.h"
#include "AthenaKernel/errorcheck.h"
#include "CxxUtils/checker_macros.h"
#include "StoreGate/ReadHandleKey.h"
#include "GeneratorObjects/McEventCollection.h"
 
#include "AtlasHepMC/GenEvent.h"
#include "AtlasHepMC/GenVertex.h"
#include "TruthUtils/MagicNumbers.h"
 
#include "HepPDT/ParticleData.hh"
#include "HepPDT/ParticleDataTable.hh"
 
#include <string>
#include <vector>
#include <map>
 
 
class GenBase : public AthAlgorithm {
public:
 
 
 
  GenBase(const std::string& name, ISvcLocator* pSvcLocator);
 
  virtual ~GenBase() { }
 
 
 
 
  virtual StatusCode initialize() override;
  virtual StatusCode execute() override { return StatusCode::SUCCESS; }
 
 
 
 
  HepMC::GenEvent* event ATLAS_NOT_CONST_THREAD_SAFE () {
    if (events()->empty())
      ATH_MSG_ERROR("McEventCollection is empty during first event access");
    return *(events()->begin());
  }
 
  const HepMC::GenEvent* event_const() const {
    if (events_const()->empty())
      ATH_MSG_ERROR("Const McEventCollection is empty during first event access");
    return *(events_const()->begin());
  }
 
  McEventCollection* events ATLAS_NOT_CONST_THREAD_SAFE ();
 
  const McEventCollection* events_const() const {
    return events_const( getContext() );
  }
  const McEventCollection* events_const( const EventContext& ctx ) const {
    SG::ReadHandle<McEventCollection> ret = SG::makeHandle(m_mcevents_const, ctx);
    if (!ret.isValid())
      ATH_MSG_ERROR("No McEventCollection found in StoreGate with key " << m_mcevents_const.key());
    return ret.cptr();
  }
 
 
 
 
 
  const ServiceHandle<IPartPropSvc> partPropSvc() const {
    return m_ppSvc;
  }
 
  const HepPDT::ParticleDataTable& particleTable() const {
    return *(m_ppSvc->PDT());
  }
 
  const HepPDT::ParticleDataTable& pdt() const { return particleTable(); }
 
  const HepPDT::ParticleData* particleData(int pid) const {
    return pdt().particle(HepPDT::ParticleID(std::abs(pid)));
  }
 
 
protected:
 
 
  std::string m_mcEventKey{};
  BooleanProperty m_mkMcEvent{this, "MakeMcEvent", false, "Create a new MC event collection if it doesn't exist"};
 
 
 
  void GeVToMeV(HepMC::GenEvent* evt);
  void MeVToGeV(HepMC::GenEvent* evt);
  void cmTomm(HepMC::GenEvent* evt);
  void mmTocm(HepMC::GenEvent* evt);
 
 
private:
 
  ServiceHandle<IPartPropSvc> m_ppSvc{this, "PartPropSvc", "PartPropSvc"};
 
  SG::ReadHandleKey<McEventCollection> m_mcevents_const{ this, "McEventKey", "GEN_EVENT", "StoreGate key of the MC event collection" };
 
};
 
 
#endif