G4StationEventAction.cc

Go to the documentation of this file.

#include "G4StationEventAction.h"
#include "G4StationTrackingAction.h"
#include "G4UMDPixelAction.h"

#include <fwk/RunController.h>

#include <G4Event.hh>
#include <G4Timer.hh>
#include <G4SDManager.hh>

#include <string>


namespace G4StationSimulatorOG {

  G4StationEventAction::G4StationEventAction(const bool umdEnabled) :
    fG4StationSimulator(
      dynamic_cast<G4StationSimulator&>(fwk::RunController::GetInstance().GetModule("G4StationSimulatorOG"))
    ),
    fTimer(new G4Timer),
    fIsUMDEnabled(umdEnabled)
  { }


  G4StationEventAction::~G4StationEventAction()
  {
    delete fTimer;
  }


  void
  G4StationEventAction::BeginOfEventAction(const G4Event* const /*event*/)
  {
    fTimer->Start();
    if (fIsUMDEnabled && !fG4StationSimulator.fUMDFastMode && fUMDPixelCollectionID < 0)
      fUMDPixelCollectionID = G4SDManager::GetSDMpointer()->GetCollectionID("pixelCollection");
  }


  void
  G4StationEventAction::EndOfEventAction(const G4Event* const event)
  {
    G4StationTrackingAction::fNumC = 0;
    G4StationTrackingAction::fNumCDelta = 0;
    G4StationTrackingAction::fNumBounces = 0;

    if (fIsUMDEnabled) {
      if (!G4StationSimulator::fgCurrent.GetDetectorStation().ExistsAssociatedCounter())
        return;
      ProcessUMD(event);
    }

    fUMDPixelCollectionID = -1;
    fTimer->Stop();
  }


  void
  G4StationEventAction::ProcessUMD(const G4Event* const event)
  {
    if (fUMDPixelCollectionID < 0)
      return;

    // retrieve the hit collections of this event
    auto hce = event->GetHCofThisEvent();
    if (!hce)
      return;

    auto phc = hce->GetHC(fUMDPixelCollectionID);
    if (!phc)
      return;

    // Pixel collection
    const size_t pixel_hit = phc->GetSize();

    for (size_t i = 0; i < pixel_hit; ++i) {
      auto hit = static_cast<G4UMDPixelHit*>(phc->GetHit(i));
      const int pId = hit->GetPixelId();
      const int mId = hit->GetModuleId();
      const auto& times = hit->GetTimes();
      for (const auto& t : times)
        fG4StationSimulator.AddUMDPhoton(mId, pId, t);
    }

    phc->DrawAllHits();
  }

}