Skip to content

Commit

Permalink
put common stuff into helper functions
Browse files Browse the repository at this point in the history
  • Loading branch information
Zehvogel committed Jul 29, 2024
1 parent d70e358 commit 6ffa0da
Show file tree
Hide file tree
Showing 4 changed files with 111 additions and 92 deletions.
52 changes: 5 additions & 47 deletions DDG4/plugins/Geant4TCUserParticleHandler.cpp
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
//==========================================================================
// AIDA Detector description implementation
// AIDA Detector description implementation
//--------------------------------------------------------------------------
// Copyright (C) Organisation europeenne pour la Recherche nucleaire (CERN)
// All rights reserved.
Expand Down Expand Up @@ -72,7 +72,7 @@ namespace dd4hep {
#endif // DD4HEP_DDG4_GEANT4TCUSERPARTICLEHANDLER_H

//====================================================================
// AIDA Detector description implementation
// AIDA Detector description implementation
//--------------------------------------------------------------------
//
// Author : M.Frank
Expand All @@ -82,6 +82,7 @@ namespace dd4hep {
//#include <DDG4/Geant4TCUserParticleHandler.h>
#include <DDG4/Geant4Particle.h>
#include <DDG4/Factories.h>
#include "Geant4UserParticleHandlerHelper.h"


using namespace dd4hep::sim;
Expand All @@ -106,58 +107,15 @@ void Geant4TCUserParticleHandler::end(const G4Track* /* track */, Particle& p)
&& z_prod <= m_zTrackerMax
) ;

dd4hep::detail::ReferenceBitMask<int> reason(p.reason);

if( reason.isSet(G4PARTICLE_PRIMARY) ) {
//do nothing
} else if( starts_in_trk_vol && ! reason.isSet(G4PARTICLE_ABOVE_ENERGY_THRESHOLD) ) {
// created in tracking volume but below energy cut
p.reason = 0;
return;
}

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
) ;

// created and ended in calo but not primary particle
//
// we can have particles from the generator only in the calo, if we have a
// long particle with preassigned decay, we need to keep the reason or the
// MChistory will not be updated later on
if( not reason.isSet(G4PARTICLE_PRIMARY) ) {
if( !starts_in_trk_vol ) {
if( !ends_in_trk_vol ){
p.reason = 0;
}
//fg: dont keep backscatter that did not create a tracker hit
else if( ! reason.isSet(G4PARTICLE_CREATED_TRACKER_HIT) ) {
p.reason = 0;
}
}
}

// Set the simulator status bits
dd4hep::detail::ReferenceBitMask<int> simStatus(p.status);

if( ends_in_trk_vol ) {
simStatus.set(G4PARTICLE_SIM_DECAY_TRACKER);
}

// if the particle doesn't end in the tracker volume it must have ended in the calorimeter
if( not ends_in_trk_vol && not simStatus.isSet(G4PARTICLE_SIM_LEFT_DETECTOR) ) {
simStatus.set(G4PARTICLE_SIM_DECAY_CALO);
}

if( not starts_in_trk_vol && ends_in_trk_vol ) {
simStatus.set(G4PARTICLE_SIM_BACKSCATTER);
}

return ;

setReason(p, starts_in_trk_vol, ends_in_trk_vol);
setStatus(p, starts_in_trk_vol, ends_in_trk_vol);
}

/// Post-event action callback
Expand Down
48 changes: 3 additions & 45 deletions DDG4/plugins/Geant4TVUserParticleHandler.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -85,6 +85,7 @@ namespace dd4hep {
#include <DDG4/Factories.h>
#include <DDG4/Geant4Particle.h>
#include <DDG4/Geant4Kernel.h>
#include "Geant4UserParticleHandlerHelper.h"


using namespace dd4hep::sim;
Expand All @@ -103,54 +104,11 @@ 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());

dd4hep::detail::ReferenceBitMask<int> reason(p.reason);

if( reason.isSet(G4PARTICLE_PRIMARY) ) {
//do nothing
} else if( starts_in_trk_vol && ! reason.isSet(G4PARTICLE_ABOVE_ENERGY_THRESHOLD) ) {
// created in tracking volume but below energy cut
p.reason = 0;
return;
}

std::array<double, 3> end_point = {p.vex, p.vey, p.vez};
bool ends_in_trk_vol = m_trackingVolume.ptr()->Contains(end_point.data());

// created and ended in calo but not primary particle
//
// we can have particles from the generator only in the calo, if we have a
// long particle with preassigned decay, we need to keep the reason or the
// MChistory will not be updated later on
if( not reason.isSet(G4PARTICLE_PRIMARY) ) {
if( !starts_in_trk_vol ) {
if( !ends_in_trk_vol ){
p.reason = 0;
}
//fg: dont keep backscatter that did not create a tracker hit
else if( ! reason.isSet(G4PARTICLE_CREATED_TRACKER_HIT) ) {
p.reason = 0;
}
}
}

// Set the simulator status bits
dd4hep::detail::ReferenceBitMask<int> simStatus(p.status);

if( ends_in_trk_vol ) {
simStatus.set(G4PARTICLE_SIM_DECAY_TRACKER);
}

// if the particle doesn't end in the tracker volume it must have ended in the calorimeter
if( not ends_in_trk_vol && not simStatus.isSet(G4PARTICLE_SIM_LEFT_DETECTOR) ) {
simStatus.set(G4PARTICLE_SIM_DECAY_CALO);
}

if( not starts_in_trk_vol && ends_in_trk_vol ) {
simStatus.set(G4PARTICLE_SIM_BACKSCATTER);
}

return ;

setReason(p, starts_in_trk_vol, ends_in_trk_vol);
setStatus(p, starts_in_trk_vol, ends_in_trk_vol);
}

/// Post-event action callback
Expand Down
71 changes: 71 additions & 0 deletions DDG4/plugins/Geant4UserParticleHandlerHelper.cpp
Original file line number Diff line number Diff line change
@@ -0,0 +1,71 @@

//==========================================================================
// 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
//
//==========================================================================

#include "Geant4UserParticleHandlerHelper.h"
#include <DDG4/Geant4Particle.h>
#include <DDG4/Geant4UserParticleHandler.h>

// using namespace dd4hep::sim;

namespace dd4hep::sim {

void setReason(Geant4Particle& p, bool starts_in_trk_vol, bool ends_in_trk_vol) {

dd4hep::detail::ReferenceBitMask<int> reason(p.reason);

if( reason.isSet(G4PARTICLE_PRIMARY) ) {
//do nothing
} else if( starts_in_trk_vol && ! reason.isSet(G4PARTICLE_ABOVE_ENERGY_THRESHOLD) ) {
// created in tracking volume but below energy cut
p.reason = 0;
return;
}

// created and ended in calo but not primary particle
//
// we can have particles from the generator only in the calo, if we have a
// long particle with preassigned decay, we need to keep the reason or the
// MChistory will not be updated later on
if( not reason.isSet(G4PARTICLE_PRIMARY) ) {
if( !starts_in_trk_vol ) {
if( !ends_in_trk_vol ){
p.reason = 0;
}
//fg: dont keep backscatter that did not create a tracker hit
else if( ! reason.isSet(G4PARTICLE_CREATED_TRACKER_HIT) ) {
p.reason = 0;
}
}
}
}

void setStatus(Geant4Particle& p, bool starts_in_trk_vol, bool ends_in_trk_vol) {
// Set the simulator status bits
dd4hep::detail::ReferenceBitMask<int> simStatus(p.status);

if( ends_in_trk_vol ) {
simStatus.set(G4PARTICLE_SIM_DECAY_TRACKER);
}

// if the particle doesn't end in the tracker volume it must have ended in the calorimeter
if( not ends_in_trk_vol && not simStatus.isSet(G4PARTICLE_SIM_LEFT_DETECTOR) ) {
simStatus.set(G4PARTICLE_SIM_DECAY_CALO);
}

if( not starts_in_trk_vol && ends_in_trk_vol ) {
simStatus.set(G4PARTICLE_SIM_BACKSCATTER);
}
}

}
32 changes: 32 additions & 0 deletions DDG4/plugins/Geant4UserParticleHandlerHelper.h
Original file line number Diff line number Diff line change
@@ -0,0 +1,32 @@

//==========================================================================
// 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 DDG4_GEANT4USERPARTICLEHANDLERHELPER_H
#define DDG4_GEANT4USERPARTICLEHANDLERHELPER_H

/// Namespace for the AIDA detector description toolkit
namespace dd4hep {

/// Namespace for the Geant4 based simulation part of the AIDA detector description toolkit
namespace sim {

// Forward declarations
class Geant4Particle;
void setReason(Geant4Particle& p, bool starts_in_trk_volume, bool ends_in_trk_volume);
void setStatus(Geant4Particle& p, bool starts_in_trk_volume, bool ends_in_trk_volume);

}
}

#endif // DDG4_GEANT4USERPARTICLEHANDLERHELPER_H

0 comments on commit 6ffa0da

Please sign in to comment.