EventAction.cc

Go to the documentation of this file.

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

#include <fstream>

using std::cout;
using std::endl;

EventAction::EventAction(const edm::ParameterSet & p,
                         RunManager* rm,
             SimTrackManager* iManager) 
    : m_runManager(rm),
      m_trackManager(iManager),
      m_stopFile(p.getParameter<std::string>("StopFile")),
      m_debug(p.getUntrackedParameter<bool>("debug",false))
{
  m_trackManager->setCollapsePrimaryVertices(p.getParameter<bool>("CollapsePrimaryVertices"));
}

EventAction::~EventAction() {}
    
void EventAction::BeginOfEventAction(const G4Event * anEvent)
{
    if (std::ifstream(m_stopFile.c_str()))
    {
        cout << "BeginOfEventAction: termination signal received at event "
             << anEvent->GetEventID() << endl;
        //RunManager::instance()->abortRun(true);
    m_runManager->abortRun(true);
    }

    m_trackManager->reset();
    BeginOfEvent e(anEvent);
    m_beginOfEventSignal(&e);

}

void EventAction::EndOfEventAction(const G4Event * anEvent)
{
    if (std::ifstream(m_stopFile.c_str()))
    {
        cout << "EndOfEventAction: termination signal received at event "
             << anEvent->GetEventID() << endl;
    // soft abort run
    m_runManager->abortRun(true);
    }
    if (anEvent->GetNumberOfPrimaryVertex()==0)
    {
        cout << " EndOfEventAction: event " << anEvent->GetEventID()
             << " must have failed (no G4PrimaryVertices found) and will be skipped " 
         << endl;
        return;
    }

    // m_trackManager->storeTracks(RunManager::instance()->simEvent());
    m_trackManager->storeTracks(m_runManager->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::addTrack(TrackWithHistory* iTrack, bool inHistory, bool withAncestor)
{
  m_trackManager->addTrack(iTrack, inHistory, withAncestor);
}

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

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