Merge pull request #48337 from civanch/update_regions

[SIM] Update handling of G4Regions, improved printouts, added Phase2 stacking action

Author: cmsbuild

SimG4Core/Application/interface/Phase2StackingAction.h

@@ -0,0 +1,113 @@
+#ifndef SimG4Core_Phase2StackingAction_H
+#define SimG4Core_Phase2StackingAction_H
+
+#include "FWCore/ParameterSet/interface/ParameterSet.h"
+
+#include "G4UserStackingAction.hh"
+#include "G4Region.hh"
+#include "G4Track.hh"
+#include "G4LogicalVolume.hh"
+
+#include <string>
+#include <vector>
+
+class TrackingAction;
+class CMSSteppingVerbose;
+class G4VProcess;
+class CMSG4TrackInterface;
+
+class Phase2StackingAction : public G4UserStackingAction {
+public:
+  explicit Phase2StackingAction(const edm::ParameterSet& ps, const CMSSteppingVerbose*);
+
+  ~Phase2StackingAction() override = default;
+
+  G4ClassificationOfNewTrack ClassifyNewTrack(const G4Track* aTrack) final;
+
+  void NewStage() override;
+  void PrepareNewEvent() override;
+
+private:
+  void initPointer();
+
+  int isItPrimaryDecayProductOrConversion(const int subtype, const G4Track&) const;
+
+  int isItFromPrimary(const G4Track&, int) const;
+
+  bool rrApplicable(const G4Track*, const G4Track&) const;
+
+  bool isItOutOfTimeWindow(const G4Region*, const double&) const;
+
+  bool isThisRegion(const G4Region*, std::vector<const G4Region*>&) const;
+
+  void printRegions(const std::vector<const G4Region*>& reg, const std::string& word) const;
+
+private:
+  bool savePDandCinTracker, savePDandCinCalo;
+  bool savePDandCinMuon, saveFirstSecondary;
+  bool savePDandCinAll;
+  bool killInCalo{false};
+  bool killInCaloEfH{false};
+  bool killHeavy, trackNeutrino, killDeltaRay;
+  bool killExtra;
+  bool killGamma;
+  double limitEnergyForVacuum;
+  double kmaxIon, kmaxNeutron, kmaxProton;
+  double kmaxGamma;
+  double maxTrackTime;
+  double maxTrackTimeForward;
+  double maxZCentralCMS;
+  unsigned int numberTimes;
+  std::vector<double> maxTrackTimes;
+  std::vector<std::string> maxTimeNames;
+  std::vector<std::string> deadRegionNames;
+
+  std::vector<const G4Region*> maxTimeRegions;
+  std::vector<const G4Region*> trackerRegions;
+  std::vector<const G4Region*> muonRegions;
+  std::vector<const G4Region*> caloRegions;
+  std::vector<const G4Region*> lowdensRegions;
+  std::vector<const G4Region*> deadRegions;
+
+  G4VSolid* worldSolid;
+  CMSG4TrackInterface* m_trackInterface;
+  const CMSSteppingVerbose* steppingVerbose;
+  const G4VProcess* m_Compton{nullptr};
+
+  // Russian roulette regions
+  const G4Region* regionEcal{nullptr};
+  const G4Region* regionHcal{nullptr};
+  const G4Region* regionMuonIron{nullptr};
+  const G4Region* regionPreShower{nullptr};
+  const G4Region* regionCastor{nullptr};
+  const G4Region* regionZDC{nullptr};
+  const G4Region* regionHGcal{nullptr};
+  const G4Region* regionWorld{nullptr};
+
+  // Russian roulette energy limits
+  double gRusRoEnerLim;
+  double nRusRoEnerLim;
+
+  // Russian roulette factors
+  double gRusRoEcal;
+  double nRusRoEcal;
+  double gRusRoHcal;
+  double nRusRoHcal;
+  double gRusRoMuonIron;
+  double nRusRoMuonIron;
+  double gRusRoPreShower;
+  double nRusRoPreShower;
+  double gRusRoCastor;
+  double nRusRoCastor;
+  double gRusRoZDC;
+  double nRusRoZDC;
+  double gRusRoHGcal;
+  double nRusRoHGcal;
+  double gRusRoWorld;
+  double nRusRoWorld;
+  // flags
+  bool gRRactive{false};
+  bool nRRactive{false};
+};
+
+#endif

SimG4Core/Application/interface/RunManagerMT.h

@@ -17,6 +17,7 @@ class CustomUIsession;
 class SimWatcher;
 class SimProducer;
 class G4SimEvent;
+class G4LogicalVolume;
 
 class RunAction;
 
@@ -87,6 +88,15 @@ class RunManagerMT {
 
   void runForPhase2();
 
+  void addRegions();
+
+  void addG4Region(const std::vector<G4LogicalVolume*>&,
+                   const std::string& regName,
+                   double cutg,
+                   double cute,
+                   double cutp,
+                   double cuti);
+
   G4MTRunManagerKernel* m_kernel;
 
   CustomUIsession* m_UIsession;