VisualizationAction.cxx

Go to the documentation of this file.

//

#include "larsim/LegacyLArG4/VisualizationAction.h"
#include "larsim/Simulation/LArG4Parameters.h"

#include "art/Framework/Services/Registry/ServiceHandle.h"

#include "Geant4/G4Event.hh"
#include "Geant4/G4TrajectoryContainer.hh"
#include "Geant4/G4UImanager.hh"
#include "Geant4/G4VVisManager.hh"
#include "Geant4/G4ios.hh"

namespace larg4 {

  VisualizationAction::VisualizationAction()
  {
    art::ServiceHandle<sim::LArG4Parameters const> lgp;
    m_energyCutoff = lgp->VisualizationEnergyCut() * CLHEP::GeV;
    m_drawNeutrals = lgp->DrawNeutrals();
  }

  VisualizationAction::~VisualizationAction() {}

  void VisualizationAction::BeginOfRunAction(const G4Run*)
  {
    // If graphics is ON
    if (G4VVisManager::GetConcreteInstance()) {
      G4cout << "Geant4 visualisation is ON" << G4endl;
      G4UImanager* UI = G4UImanager::GetUIpointer();
      UI->ApplyCommand("/tracking/storeTrajectory 1");
      UI->ApplyCommand("/vis/scene/notifyhandlers");
    }
  }

  void VisualizationAction::EndOfRunAction(const G4Run*)
  {
    // If graphics is ON
    if (G4VVisManager::GetConcreteInstance())
      G4UImanager::GetUIpointer()->ApplyCommand("/vis/viewer/update");
  }

  void VisualizationAction::BeginOfEventAction(const G4Event*) {}

  void VisualizationAction::EndOfEventAction(const G4Event* a_event)
  {
    G4VVisManager* pVVisManager = G4VVisManager::GetConcreteInstance();

    if (pVVisManager) {
      G4TrajectoryContainer* trajectoryContainer = a_event->GetTrajectoryContainer();
      G4int n_trajectories = 0;
      if (trajectoryContainer) n_trajectories = trajectoryContainer->entries();

      for (G4int i = 0; i < n_trajectories; i++) {
        G4VTrajectory* trajectory = (*trajectoryContainer)[i];

        // Only draw "interesting" trajectories, according to cuts
        // and limits set by the user.

        G4bool drawThisTrack = false;
        if (trajectory->GetInitialMomentum().mag() > m_energyCutoff) drawThisTrack = true;

        if (!m_drawNeutrals) {
          G4double charge = trajectory->GetCharge();
          // electron-, muon-, and tau-neutrino codes:
          if (charge == 0) drawThisTrack = false;
        }

        if (drawThisTrack) trajectory->DrawTrajectory();
        //trajectory->DrawTrajectory(50);
      }
    }
  }

} // namespace LArG4