Geant4PrimaryHandler.h

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 DDG4_GEANT4PRIMARYHANDLER_H
#define DDG4_GEANT4PRIMARYHANDLER_H
 
// Framework include files
#include <DDG4/Geant4GeneratorAction.h>
 
// C/C++ include files
#include <set>
 
namespace dd4hep {
 
  namespace sim {
    class Geant4PrimaryConfig {
    public:
      std::set<int> m_rejectPDGs = {1, 2, 3, 4, 5, 6, 21, 23, 24, 25};
      std::set<int> m_zeroTimePDGs = {11, 13, 15, 17};
 
      std::string toString() const {
        std::stringstream str;
        str << "\nRejectPDGs: ";
        for (int i: m_rejectPDGs) { str << i << ", "; }
        str << "\nzeroTimePDGs: ";
        for (int i: m_zeroTimePDGs) { str << i << ", "; }
        return str.str();
      }
    };
 
 
    class Geant4PrimaryHandler : public Geant4GeneratorAction    {
    public:
      Geant4PrimaryHandler(Geant4Context* context, const std::string& nam);
      virtual ~Geant4PrimaryHandler();
      virtual void operator()(G4Event* event)  override;
 
    public:
      Geant4PrimaryConfig m_primaryConfig{};
 
    };
  }    // End namespace sim
}      // End namespace dd4hep
 
#endif // DDG4_GEANT4PRIMARYHANDLER_H