SimG4Core/SensitiveDetector/interface/SensitiveDetector.h - master - githubmirror/cmssw

SimG4Core/SensitiveDetector/interface/SensitiveDetector.h

63 строки2 KB

Vladimir

code format

30 сен 2025, 19:28

30 сен 2025, 19:280a72539

#ifndef SimG4Core_SensitiveDetector_H
#define SimG4Core_SensitiveDetector_H

#include "DataFormats/GeometryVector/interface/LocalPoint.h"
#include "SimG4Core/Geometry/interface/SensitiveDetectorCatalog.h"
#include "SimG4Core/Notification/interface/TrackInformation.h"

#include "G4VSensitiveDetector.hh"

#include <string>
#include <cstdint>

class G4Track;
class G4Step;
class G4HCofThisEvent;
class G4TouchableHistory;
class G4VPhysicalVolume;

class SensitiveDetector : public G4VSensitiveDetector {
public:
  explicit SensitiveDetector(const std::string& iname,
                             const SensitiveDetectorCatalog&,
                             bool calo,
                             const std::string& newcollname = "");

  ~SensitiveDetector() override = default;

  G4bool ProcessHits(G4Step* step, G4TouchableHistory* tHistory) override = 0;

  virtual uint32_t setDetUnitId(const G4Step* step) = 0;
  virtual void clearHits() = 0;

  inline const std::vector<std::string>& getNames() const { return m_namesOfSD; }

  inline bool isCaloSD() const { return m_isCalo; }

protected:
  // generic geometry methods, all coordinates in mm
  enum coordinates { WorldCoordinates, LocalCoordinates };
  Local3DPoint InitialStepPosition(const G4Step* step, coordinates) const;
  Local3DPoint FinalStepPosition(const G4Step* step, coordinates) const;

  // local coordinates to the volume, in which the step is done
  Local3DPoint LocalPreStepPosition(const G4Step* step) const;
  Local3DPoint LocalPostStepPosition(const G4Step* step) const;

  inline Local3DPoint ConvertToLocal3DPoint(const G4ThreeVector& point) const {
    return Local3DPoint(point.x(), point.y(), point.z());
  }

  void setNames(const std::vector<std::string>&);
  void NaNTrap(const G4Step* step) const;

  TrackInformation* cmsTrackInformation(const G4Track* aTrack);

private:
  void AssignSD(const std::string& vname);

  std::vector<std::string> m_namesOfSD;
  bool m_isCalo;
};

#endif