Merge pull request #47549 from civanch/extended_sim_exception

[FULLSIM] Extended SIM exception

Author: cmsbuild

SimG4Core/Application/interface/CMSG4TrackInterface.h

@@ -0,0 +1,44 @@
+#ifndef SimG4Core_CMSG4TrackInterface_h
+#define SimG4Core_CMSG4TrackInterface_h 1
+
+//
+// Package:     Application
+// Class  :     SimTrackInterface
+//
+// 10.03.2025   V.Ivantchenko
+//
+// An interface between Geant4 and CMSSW
+
+#include "G4ThreadLocalSingleton.hh"
+
+class G4Track;
+
+class CMSG4TrackInterface {
+  friend class G4ThreadLocalSingleton<CMSG4TrackInterface>;
+
+public:
+  static CMSG4TrackInterface* instance();
+
+  ~CMSG4TrackInterface();
+
+  const G4Track* getCurrentTrack() { return track_; }
+
+  void setCurrentTrack(const G4Track* p) { track_ = p; }
+
+  int getThreadID() { return threadID_; }
+
+  void setThreadID(int n) { threadID_ = n; }
+
+  CMSG4TrackInterface(CMSG4TrackInterface&) = delete;
+  CMSG4TrackInterface& operator=(const CMSG4TrackInterface& right) = delete;
+
+private:
+  CMSG4TrackInterface();
+
+  static G4ThreadLocal CMSG4TrackInterface* interface_;
+
+  const G4Track* track_{nullptr};
+  int threadID_{0};
+};
+
+#endif

SimG4Core/Application/interface/ExceptionHandler.h

@@ -19,7 +19,7 @@
 class ExceptionHandler : public G4VExceptionHandler {
 public:
   explicit ExceptionHandler(double th, bool tr);
-  ~ExceptionHandler() override;
+  ~ExceptionHandler() override = default;
 
   int operator==(const ExceptionHandler &right) const { return (this == &right); }
   int operator!=(const ExceptionHandler &right) const { return (this != &right); }

SimG4Core/Application/interface/TrackingAction.h

@@ -10,6 +10,7 @@
 #include <vector>
 
 class SimTrackManager;
+class CMSG4TrackInterface;
 class TrackWithHistory;
 class BeginOfTrack;
 class EndOfTrack;
@@ -31,8 +32,12 @@ class TrackingAction : public G4UserTrackingAction {
   SimActivityRegistry::BeginOfTrackSignal m_beginOfTrackSignal;
   SimActivityRegistry::EndOfTrackSignal m_endOfTrackSignal;
 
+  TrackingAction(TrackingAction&) = delete;
+  TrackingAction& operator=(const TrackingAction& right) = delete;
+
 private:
   SimTrackManager* trackManager_;
+  CMSG4TrackInterface* interface_;
   CMSSteppingVerbose* steppingVerbose_;
   const G4Track* g4Track_ = nullptr;
   TrackInformation* trkInfo_ = nullptr;

SimG4Core/Application/src/CMSG4TrackInterface.cc

@@ -0,0 +1,15 @@
+#include "SimG4Core/Application/interface/CMSG4TrackInterface.h"
+
+G4ThreadLocal CMSG4TrackInterface* CMSG4TrackInterface::interface_ = nullptr;
+
+CMSG4TrackInterface* CMSG4TrackInterface::instance() {
+  if (nullptr == interface_) {
+    static G4ThreadLocalSingleton<CMSG4TrackInterface> inst;
+    interface_ = inst.Instance();
+  }
+  return interface_;
+}
+
+CMSG4TrackInterface::CMSG4TrackInterface() {};
+
+CMSG4TrackInterface::~CMSG4TrackInterface() {};

SimG4Core/Application/src/ExceptionHandler.cc

@@ -1,4 +1,5 @@
 #include "SimG4Core/Application/interface/ExceptionHandler.h"
+#include "SimG4Core/Application/interface/CMSG4TrackInterface.h"
 
 #include "FWCore/MessageLogger/interface/MessageLogger.h"
 #include "FWCore/Utilities/interface/Exception.h"
@@ -11,8 +12,6 @@
 
 ExceptionHandler::ExceptionHandler(double th, bool tr) : m_eth(th), m_trace(tr) {}
 
-ExceptionHandler::~ExceptionHandler() {}
-
 bool ExceptionHandler::Notify(const char* exceptionOrigin,
                               const char* exceptionCode,
                               G4ExceptionSeverity severity,
@@ -22,12 +21,14 @@ bool ExceptionHandler::Notify(const char* exceptionOrigin,
   static const G4String ws_banner = "\n-------- WWWW ------- G4Exception-START -------- WWWW -------\n";
   static const G4String we_banner = "\n-------- WWWW -------- G4Exception-END --------- WWWW -------\n";
 
-  G4Track* track = G4EventManager::GetEventManager()->GetTrackingManager()->GetTrack();
+  auto ig4 = CMSG4TrackInterface::instance();
+  const G4Track* track = ig4->getCurrentTrack();
+  int id = ig4->getThreadID();
   double ekin = m_eth;
 
   std::stringstream message;
   message << "*** G4Exception : " << exceptionCode << "\n"
-          << "      issued by : " << exceptionOrigin << "\n"
+          << "      issued by : " << exceptionOrigin << " in threadID=" << id << "\n"
           << description;
 
   // part of exception happens outside tracking loop
@@ -61,10 +62,10 @@ bool ExceptionHandler::Notify(const char* exceptionOrigin,
   G4String code;
   mescode >> code;
 
-  // track is killed
+  // track should be killed, const_cast is allowed in this place
   if (ekin < m_eth && (code == "GeomNav0003" || code == "GeomField0003")) {
     localSeverity = JustWarning;
-    track->SetTrackStatus(fStopAndKill);
+    const_cast<G4Track*>(track)->SetTrackStatus(fStopAndKill);
     ++m_number;
   }
 

SimG4Core/Application/src/RunManagerMTWorker.cc

@@ -12,6 +12,7 @@
 #include "SimG4Core/Application/interface/CustomUIsessionToFile.h"
 #include "SimG4Core/Application/interface/ExceptionHandler.h"
 #include "SimG4Core/Application/interface/CMSGDMLWriteStructure.h"
+#include "SimG4Core/Application/interface/CMSG4TrackInterface.h"
 
 #include "SimG4Core/Geometry/interface/CustomUIsession.h"
 
@@ -178,6 +179,7 @@ RunManagerMTWorker::RunManagerMTWorker(const edm::ParameterSet& p, edm::Consumes
   edm::LogVerbatim("SimG4CoreApplication") << "RunManagerMTWorker for the thread " << id;
 
   // Initialize per-thread output
+  CMSG4TrackInterface::instance()->setThreadID(id);
   G4Threading::G4SetThreadId(id);
   G4UImanager::GetUIpointer()->SetUpForAThread(id);
   auto iPset = p.getUntrackedParameter<edm::ParameterSet>("CustomUIsession");
@@ -194,8 +196,6 @@ RunManagerMTWorker::RunManagerMTWorker(const edm::ParameterSet& p, edm::Consumes
         << "', valid are MessageLogger, MessageLoggerThreadPrefix, FilePerThread";
   }
   G4UImanager::GetUIpointer()->SetCoutDestination(m_UIsession);
-  //G4PhysListUtil::InitialiseParameters();
-  //G4LossTableManager::Instance();
 
   // sensitive detectors
   std::vector<std::string> onlySDs = p.getParameter<std::vector<std::string>>("OnlySDs");

SimG4Core/Application/src/TrackingAction.cc

@@ -1,4 +1,5 @@
 #include "SimG4Core/Application/interface/TrackingAction.h"
+#include "SimG4Core/Application/interface/CMSG4TrackInterface.h"
 
 #include "SimG4Core/Notification/interface/CurrentG4Track.h"
 #include "SimG4Core/Notification/interface/BeginOfTrack.h"
@@ -25,6 +26,7 @@ TrackingAction::TrackingAction(SimTrackManager* stm, CMSSteppingVerbose* sv, con
       saveCaloBoundaryInformation_(p.getParameter<bool>("SaveCaloBoundaryInformation")),
       ekinMin_(p.getParameter<double>("PersistencyEmin") * CLHEP::GeV),
       ekinMinRegion_(p.getParameter<std::vector<double>>("RegionEmin")) {
+  interface_ = CMSG4TrackInterface::instance();
   double eth = p.getParameter<double>("EminFineTrack") * CLHEP::MeV;
   if (doFineCalo_ && eth < ekinMin_) {
     ekinMin_ = eth;
@@ -39,6 +41,7 @@ TrackingAction::TrackingAction(SimTrackManager* stm, CMSSteppingVerbose* sv, con
 void TrackingAction::PreUserTrackingAction(const G4Track* aTrack) {
   g4Track_ = aTrack;
   currentTrack_ = new TrackWithHistory(aTrack, aTrack->GetParentID());
+  interface_->setCurrentTrack(aTrack);
 
   BeginOfTrack bt(aTrack);
   m_beginOfTrackSignal(&bt);