SensitiveDetector.h

Go to the documentation of this file.

#ifndef SimG4Core_SensitiveDetector_H
#define SimG4Core_SensitiveDetector_H

#include "FWCore/ParameterSet/interface/ParameterSet.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;

namespace edm {
  class EventSetup;
}

class SensitiveDetector : public G4VSensitiveDetector {
public:
  explicit SensitiveDetector(const std::string& iname,
                             const edm::EventSetup& es,
                             const SensitiveDetectorCatalog&,
                             edm::ParameterSet const& p,
                             bool calo);

  ~SensitiveDetector() override;

  void Initialize(G4HCofThisEvent* eventHC) override;
  G4bool ProcessHits(G4Step* step, G4TouchableHistory* tHistory) override = 0;
  void EndOfEvent(G4HCofThisEvent* eventHC) override;

  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