GenBase.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2022 CERN for the benefit of the ATLAS collaboration
*/
 
#include "GeneratorModules/GenBase.h"
#include "GaudiKernel/ISvcLocator.h"
#include "GaudiKernel/DataSvc.h"
#include <fstream>
 
 
GenBase::GenBase(const std::string& name, ISvcLocator* pSvcLocator)
  : AthAlgorithm(name, pSvcLocator)
{
}
 
 
StatusCode GenBase::initialize() {
  ATH_CHECK( m_mcevents_const.initialize() );
  m_mcEventKey = m_mcevents_const.key();
 
  // Get the particle property service
  if (m_ppSvc.retrieve().isFailure()) {
    ATH_MSG_ERROR("Could not initialize ATLAS Particle Property Service");
    return StatusCode::FAILURE;
  }
  return StatusCode::SUCCESS;
}
 
 
McEventCollection* GenBase::events ATLAS_NOT_CONST_THREAD_SAFE () {
  // Make a new MC event collection if necessary
  McEventCollection* mcevents = nullptr;
  if (!evtStore()->contains<McEventCollection>(m_mcEventKey) && m_mkMcEvent) {
    ATH_MSG_DEBUG("Creating new McEventCollection in the event store");
    mcevents = new McEventCollection();
    if (evtStore()->record(mcevents, m_mcEventKey).isFailure())
      ATH_MSG_ERROR("Failed to record a new McEventCollection");
  }
  else {
    const McEventCollection* mecc = 0;
 
    if (evtStore()->retrieve (mecc, m_mcEventKey).isFailure()) 
      ATH_MSG_ERROR("Failed to retrieve McEventCollection");
 
    mcevents = const_cast<McEventCollection*> (&(*mecc));
  }
 
  return mcevents;
}
 
 
#ifdef HEPMC3
void GenBase::GeVToMeV(HepMC::GenEvent* evt) { for (auto& p: evt->particles()) { p->set_momentum(p->momentum()*1000); p->set_generated_mass(1000* p->generated_mass());}}
void GenBase::MeVToGeV(HepMC::GenEvent* evt) { for (auto& p: evt->particles()) { p->set_momentum(p->momentum()*1.0/1000); p->set_generated_mass(1.0/1000* p->generated_mass());} }
void GenBase::cmTomm(HepMC::GenEvent* evt)   { for (auto& v: evt->vertices())     v->set_position(v->position()*10);}
void GenBase::mmTocm(HepMC::GenEvent* evt)   { for (auto& v: evt->vertices())     v->set_position(v->position()*1.0/10);}
#else
void GenBase::GeVToMeV(HepMC::GenEvent* evt) {
  for (HepMC::GenEvent::particle_iterator p = evt->particles_begin(); p != evt->particles_end(); ++p) {
    const HepMC::FourVector fv((*p)->momentum().px() * 1000,
                               (*p)->momentum().py() * 1000,
                               (*p)->momentum().pz() * 1000,
                               (*p)->momentum().e()  * 1000);
    (*p)->set_momentum(fv);
    (*p)->set_generated_mass(1000 * (*p)->generated_mass());
  }
}
void GenBase::MeVToGeV(HepMC::GenEvent* evt) {
  for (HepMC::GenEvent::particle_iterator p = evt->particles_begin(); p != evt->particles_end(); ++p) {
    const HepMC::FourVector fv((*p)->momentum().px() / 1000,
                               (*p)->momentum().py() / 1000,
                               (*p)->momentum().pz() / 1000,
                               (*p)->momentum().e()  / 1000);
    (*p)->set_momentum(fv);
    (*p)->set_generated_mass((*p)->generated_mass() / 1000);
  }
}
void GenBase::cmTomm(HepMC::GenEvent* evt) {
  for (HepMC::GenEvent::vertex_iterator vtx = evt->vertices_begin(); vtx != evt->vertices_end(); ++vtx) {
    const HepMC::FourVector fv((*vtx)->position().x() * 10,
                               (*vtx)->position().y() * 10,
                               (*vtx)->position().z() * 10,
                               (*vtx)->position().t() * 10);
    (*vtx)->set_position(fv);
  }
}
void GenBase::mmTocm(HepMC::GenEvent* evt) {
  for (HepMC::GenEvent::vertex_iterator vtx = evt->vertices_begin(); vtx != evt->vertices_end(); ++vtx) {
    const HepMC::FourVector fv((*vtx)->position().x() / 10,
                               (*vtx)->position().y() / 10,
                               (*vtx)->position().z() / 10,
                               (*vtx)->position().t() / 10);
    (*vtx)->set_position(fv);
  }
}
#endif