VisualizationAction.cxx

Go to the documentation of this file.

//

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

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

#include <cmath>
#include <cstring>

namespace larg4 {

  VisualizationAction::VisualizationAction()
  {
    art::ServiceHandle<sim::LArG4Parameters> 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