Geant4TCUserParticleHandler.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_GEANT4TCUSERPARTICLEHANDLER_H
#define DD4HEP_DDG4_GEANT4TCUSERPARTICLEHANDLER_H
 
// Framework include files
#include <DD4hep/Primitives.h>
#include <DDG4/Geant4UserParticleHandler.h>
 
namespace dd4hep {
 
  namespace sim {
 
 
    class Geant4TCUserParticleHandler : public Geant4UserParticleHandler  {
      double m_zTrackerMin, m_zTrackerMax, m_rTracker;
    public:
      Geant4TCUserParticleHandler(Geant4Context* context, const std::string& nam);
 
      virtual ~Geant4TCUserParticleHandler() {}
 
 
      virtual void end(const G4Track* track, Particle& particle);
 
      virtual void end(const G4Event* event);
 
    };
  }    // End namespace sim
}      // End namespace dd4hep
 
#endif // DD4HEP_DDG4_GEANT4TCUSERPARTICLEHANDLER_H
 
//====================================================================
//  AIDA Detector description implementation
//--------------------------------------------------------------------
//
//  Author     : M.Frank
//
//====================================================================
// Framework include files
//#include <DDG4/Geant4TCUserParticleHandler.h>
#include <DDG4/Geant4Particle.h>
#include <DDG4/Factories.h>
#include "Geant4UserParticleHandlerHelper.h"
 
 
using namespace dd4hep::sim;
DECLARE_GEANT4ACTION(Geant4TCUserParticleHandler)
 
Geant4TCUserParticleHandler::Geant4TCUserParticleHandler(Geant4Context* ctxt, const std::string& nam)
: Geant4UserParticleHandler(ctxt,nam)
{
  declareProperty("TrackingVolume_Zmin",m_zTrackerMin=-1e100);
  declareProperty("TrackingVolume_Zmax",m_zTrackerMax=1e100);
  declareProperty("TrackingVolume_Rmax",m_rTracker=1e100);
}
 
void Geant4TCUserParticleHandler::end(const G4Track* /* track */, Particle& p)  {
 
  double r_prod = std::sqrt(p.vsx*p.vsx + p.vsy*p.vsy);
  double z_prod = p.vsz;
  bool starts_in_trk_vol = ( r_prod <= m_rTracker
    && z_prod >= (m_zTrackerMin == -1e100? -m_zTrackerMax : m_zTrackerMin)
    && z_prod <= m_zTrackerMax
  )  ;
 
  double r_end  = std::sqrt(p.vex*p.vex + p.vey*p.vey);
  double z_end  = p.vez;
  bool ends_in_trk_vol =  ( r_end <= m_rTracker
     && z_end >= (m_zTrackerMin == -1e100? -m_zTrackerMax : m_zTrackerMin)
     && z_end <= m_zTrackerMax
  ) ;
 
  setReason(p, starts_in_trk_vol, ends_in_trk_vol);
  setSimulatorStatus(p, starts_in_trk_vol, ends_in_trk_vol);
}
 
void Geant4TCUserParticleHandler::end(const G4Event* /* event */)   {
 
}