Geant4TVUserParticleHandler.cpp

Go to the documentation of this file.

//==========================================================================
//  AIDA Detector description implementation
//--------------------------------------------------------------------------
// Copyright (C) Organisation europeenne pour la Recherche nucleaire (CERN)
// All rights reserved.
//
// For the licensing terms see $DD4hepINSTALL/LICENSE.
// For the list of contributors see $DD4hepINSTALL/doc/CREDITS.
//
// Author     : M.Frank
//
//==========================================================================
 
#ifndef DD4HEP_DDG4_GEANT4TVUSERPARTICLEHANDLER_H
#define DD4HEP_DDG4_GEANT4TVUSERPARTICLEHANDLER_H
 
// Framework include files
#include <DD4hep/Primitives.h>
#include <DD4hep/Volumes.h>
#include <DDG4/Geant4UserParticleHandler.h>
 
namespace dd4hep {
 
  namespace sim {
 
 
    class Geant4TVUserParticleHandler : public Geant4UserParticleHandler  {
      Volume m_trackingVolume;
 
    public:
      Geant4TVUserParticleHandler(Geant4Context* context, const std::string& nam);
 
      virtual ~Geant4TVUserParticleHandler() {}
 
 
      virtual void end(const G4Track* track, Particle& particle);
 
      virtual void end(const G4Event* event);
 
    };
  }    // End namespace sim
}      // End namespace dd4hep
 
#endif // DD4HEP_DDG4_GEANT4TVUSERPARTICLEHANDLER_H
 
//====================================================================
//  AIDA Detector description implementation
//--------------------------------------------------------------------
//
//  Author     : M.Frank
//
//====================================================================
// Framework include files
//#include <DDG4/Geant4TVUserParticleHandler.h>
#include <DDG4/Factories.h>
#include <DDG4/Geant4Particle.h>
#include <DDG4/Geant4Kernel.h>
#include "Geant4UserParticleHandlerHelper.h"
 
 
using namespace dd4hep::sim;
DECLARE_GEANT4ACTION(Geant4TVUserParticleHandler)
 
Geant4TVUserParticleHandler::Geant4TVUserParticleHandler(Geant4Context* ctxt, const std::string& nam)
: Geant4UserParticleHandler(ctxt,nam)
{
  m_trackingVolume = ctxt->kernel().detectorDescription().trackingVolume();
}
 
void Geant4TVUserParticleHandler::end(const G4Track* /* track */, Particle& p)  {
 
  std::array<double, 3> start_point = {p.vsx, p.vsy, p.vsz};
  bool starts_in_trk_vol = m_trackingVolume.ptr()->Contains(start_point.data());
 
  std::array<double, 3> end_point = {p.vex, p.vey, p.vez};
  bool ends_in_trk_vol = m_trackingVolume.ptr()->Contains(end_point.data());
 
  setReason(p, starts_in_trk_vol, ends_in_trk_vol);
  setSimulatorStatus(p, starts_in_trk_vol, ends_in_trk_vol);
}
 
void Geant4TVUserParticleHandler::end(const G4Event* /* event */)   {
 
}