SimG4Core/Application/interface/TrackingAction.h - master - githubmirror/cmssw

SimG4Core/Application/interface/TrackingAction.h

54 строки2 KB

Vladimir

Added possibility to handle tracks without track information data

30 сен 2025, 18:03

30 сен 2025, 18:03adf360a

#ifndef SimG4Core_TrackingAction_H
#define SimG4Core_TrackingAction_H

#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "SimG4Core/Notification/interface/SimActivityRegistry.h"

#include "G4UserTrackingAction.hh"
#include "G4Region.hh"

#include <vector>

class SimTrackManager;
class CMSG4TrackInterface;
class TrackWithHistory;
class BeginOfTrack;
class EndOfTrack;
class CMSSteppingVerbose;
class TrackInformation;

class TrackingAction : public G4UserTrackingAction {
public:
  explicit TrackingAction(SimTrackManager*, CMSSteppingVerbose*, const edm::ParameterSet& ps);
  ~TrackingAction() override = default;

  void PreUserTrackingAction(const G4Track* aTrack) override;
  void PostUserTrackingAction(const G4Track* aTrack) override;

  inline TrackWithHistory* currentTrackWithHistory() { return currentHistory_; }
  inline const G4Track* geant4Track() const { return g4Track_; }
  inline G4TrackingManager* getTrackManager() { return fpTrackingManager; }

  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;
  TrackWithHistory* currentHistory_ = nullptr;
  int endPrintTrackID_;
  bool checkTrack_;
  bool doFineCalo_;
  bool saveCaloBoundaryInformation_;
  double ekinMin_;
  std::vector<double> ekinMinRegion_;
  std::vector<G4Region*> ptrRegion_;
};

#endif