Geant4ParticleMaskAction.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
//
//==========================================================================
 
// Framework include files
#include <DD4hep/Primitives.h>
#include <DD4hep/InstanceCount.h>
#include <DDG4/Geant4Particle.h>
#include <DDG4/Geant4SensDetAction.h>
#include <DDG4/Geant4UserParticleHandler.h>
 
// Geant4 header files
#include <G4Track.hh>
#include <G4VSensitiveDetector.hh>
 
// C/C++ include files
#include <cstdint>
 
namespace dd4hep {
 
  namespace sim {
 
 
    class Geant4ParticleMaskAction : public Geant4UserParticleHandler  {
    protected:
      std::map<std::string, int32_t>  m_nameMasks;
      std::map<std::string, int32_t>  m_typeMasks;
      
    public:
      Geant4ParticleMaskAction(Geant4Context* ctxt, const std::string& nam)
        : Geant4UserParticleHandler(ctxt,nam)
      {
        declareProperty("DetectorNameMasks", m_nameMasks );
        declareProperty("DetectorTypeMasks", m_typeMasks );
        m_needsControl = true;
      }
 
      void mark_track(const G4Track* track, Particle* curr_track )  override final {
        G4LogicalVolume*      vol = track->GetVolume()->GetLogicalVolume();
        G4VSensitiveDetector*  g4 = vol->GetSensitiveDetector();
        Geant4ActionSD*        sd = dynamic_cast<Geant4ActionSD*>(g4);
 
        if( !sd )  {
          return;
        }
        Geant4SensDetActionSequence* sens_det = sd->sequence();
        if( sens_det )  {
          using PropertyMask = dd4hep::detail::ReferenceBitMask<int>;
          PropertyMask mask(curr_track->reason);
          if( !m_nameMasks.empty() )  {
            auto it = m_nameMasks.begin();
            auto nam = sens_det->name();
            for( ; it != m_nameMasks.end(); ++it )  {
              if( it->first == nam || it->first == "*" )  {
                print("+++ Detector: name: %s Masking track %d with mask: %08X",
                      nam.c_str(), curr_track->id, it->second);
                mask.set(it->second);
              }
            }
          }
          if( !m_typeMasks.empty() )  {
            auto it = m_typeMasks.begin();          
            auto nam = sens_det->sensitiveType();
            for( ; it != m_typeMasks.end(); ++it )  {
              if( it->first == nam || it->first == "*" )  {
                print("+++ Detector: type: %s Masking track %d with mask: %08X",
                      nam.c_str(), curr_track->id, it->second);
                mask.set(it->second);
              }
            }
          }
        }
      }
    }; // End class Geant4ParticleMaskAction
  }    // End namespace sim
}      // End namespace dd4hep
 
#include <DDG4/Geant4Particle.h>
#include <DDG4/Factories.h>
using namespace dd4hep::sim;
DECLARE_GEANT4ACTION(Geant4ParticleMaskAction)