EventAction.cc

Go to the documentation of this file.

#include "SimG4Core/Application/interface/EventAction.h"
#include "SimG4Core/Application/interface/SimRunInterface.h"
#include "SimG4Core/Notification/interface/G4SimVertex.h"
#include "SimG4Core/Notification/interface/G4SimTrack.h"
#include "SimG4Core/Notification/interface/BeginOfEvent.h"
#include "SimG4Core/Notification/interface/EndOfEvent.h"
#include "SimG4Core/Notification/interface/CMSSteppingVerbose.h"

#include "FWCore/MessageLogger/interface/MessageLogger.h"

#include <fstream>
#include "Randomize.hh"

EventAction::EventAction(const edm::ParameterSet & p,
                         SimRunInterface* rm,
             SimTrackManager* iManager,
             CMSSteppingVerbose* sv) 
    : m_runInterface(rm),
      m_trackManager(iManager),
      m_SteppingVerbose(sv),
      m_stopFile(p.getParameter<std::string>("StopFile")),
      m_printRandom(p.getParameter<bool>("PrintRandomSeed")),
      m_debug(p.getUntrackedParameter<bool>("debug",false))
{
  m_trackManager->setCollapsePrimaryVertices(p.getParameter<bool>("CollapsePrimaryVertices"));
}

EventAction::~EventAction() {}
    
void EventAction::BeginOfEventAction(const G4Event * anEvent)
{
  m_trackManager->reset();

  BeginOfEvent e(anEvent);
  m_beginOfEventSignal(&e);

  if(m_printRandom) 
    {
      edm::LogVerbatim("SimG4CoreApplication") 
    << "BeginOfEvent " << anEvent->GetEventID()
    << " Random number: " << G4UniformRand();  
    }

  if(nullptr != m_SteppingVerbose) { m_SteppingVerbose->BeginOfEvent(anEvent); }
}

void EventAction::EndOfEventAction(const G4Event * anEvent)
{
  if(m_printRandom) 
    {
      edm::LogVerbatim("SimG4CoreApplication") 
    << " EndOfEvent " << anEvent->GetEventID()
    << " Random number: " << G4UniformRand();  
    }
  if (!m_stopFile.empty() && std::ifstream(m_stopFile.c_str()))
    {
      edm::LogWarning("SimG4CoreApplication")
        << "EndOfEventAction: termination signal received at event "
    << anEvent->GetEventID();
      // soft abort run
      m_runInterface->abortRun(true);
    }
  if (anEvent->GetNumberOfPrimaryVertex()==0)
    {
      edm::LogWarning("SimG4CoreApplication")
        << "EndOfEventAction: event " << anEvent->GetEventID() 
    << " must have failed (no G4PrimaryVertices found) and will be skipped ";
      return;
    }

  m_trackManager->storeTracks(m_runInterface->simEvent());

  // dispatch now end of event, and only then delete tracks...
  EndOfEvent e(anEvent);
  m_endOfEventSignal(&e);

  m_trackManager->deleteTracks();
  m_trackManager->cleanTkCaloStateInfoMap();
}

void EventAction::addTkCaloStateInfo(uint32_t t,
                     const std::pair<math::XYZVectorD,math::XYZTLorentzVectorD>& p) 
{
  m_trackManager->addTkCaloStateInfo(t,p);
}

void EventAction::abortEvent()
{
  m_runInterface->abortEvent();
}