File indexing completed on 2024-04-06 12:31:42
0001 #ifndef TrackPropagation_Geant4ePropagator_h
0002 #define TrackPropagation_Geant4ePropagator_h
0003
0004 #include <memory>
0005
0006
0007
0008 #include "TrackingTools/GeomPropagators/interface/Propagator.h"
0009
0010
0011 #include "G4ErrorPropagatorData.hh"
0012 #include "G4ErrorPropagatorManager.hh"
0013 #include "G4ErrorSurfaceTarget.hh"
0014
0015
0016
0017
0018
0019
0020 class Geant4ePropagator : public Propagator {
0021 public:
0022
0023
0024
0025
0026
0027
0028 Geant4ePropagator(const MagneticField *field = nullptr,
0029 std::string particleName = "mu",
0030 PropagationDirection dir = alongMomentum,
0031 double plimit = 1.0);
0032
0033 ~Geant4ePropagator() override;
0034
0035
0036
0037
0038
0039
0040
0041
0042 std::pair<TrajectoryStateOnSurface, double> propagateWithPath(const FreeTrajectoryState &,
0043 const Plane &) const override;
0044
0045 std::pair<TrajectoryStateOnSurface, double> propagateWithPath(const FreeTrajectoryState &,
0046 const Cylinder &) const override;
0047
0048 std::pair<TrajectoryStateOnSurface, double> propagateWithPath(const TrajectoryStateOnSurface &,
0049 const Plane &) const override;
0050
0051 std::pair<TrajectoryStateOnSurface, double> propagateWithPath(const TrajectoryStateOnSurface &,
0052 const Cylinder &) const override;
0053
0054 Geant4ePropagator *clone() const override { return new Geant4ePropagator(*this); }
0055
0056 const MagneticField *magneticField() const override { return theField; }
0057
0058 private:
0059 typedef std::pair<TrajectoryStateOnSurface, double> TsosPP;
0060 typedef std::pair<bool, std::shared_ptr<G4ErrorTarget>> ErrorTargetPair;
0061
0062
0063 const MagneticField *theField;
0064
0065
0066 std::string theParticleName;
0067
0068
0069 G4ErrorPropagatorManager *theG4eManager;
0070 G4ErrorPropagatorData *theG4eData;
0071 double plimit_;
0072
0073
0074
0075 template <class SurfaceType>
0076 ErrorTargetPair transformToG4SurfaceTarget(const SurfaceType &pDest, bool moveTargetToEndOfSurface) const;
0077
0078
0079
0080
0081
0082
0083
0084
0085 std::string generateParticleName(int charge) const;
0086
0087
0088
0089
0090
0091 template <class SurfaceType>
0092 std::pair<TrajectoryStateOnSurface, double> propagateGeneric(const FreeTrajectoryState &ftsStart,
0093 const SurfaceType &pDest) const;
0094
0095
0096
0097
0098 template <class SurfaceType>
0099 bool configurePropagation(G4ErrorMode &mode,
0100 SurfaceType const &pDest,
0101 GlobalPoint const &cmsInitPos,
0102 GlobalVector const &cmsInitMom) const;
0103
0104
0105
0106
0107
0108 template <class SurfaceType>
0109 bool configureAnyPropagation(G4ErrorMode &mode,
0110 SurfaceType const &pDest,
0111 GlobalPoint const &cmsInitPos,
0112 GlobalVector const &cmsInitMom) const;
0113
0114
0115
0116
0117
0118
0119 void ensureGeant4eIsInitilized(bool forceInit) const;
0120
0121
0122 template <class SurfaceType>
0123 std::string getSurfaceType(SurfaceType const &surface) const;
0124
0125 void debugReportPlaneSetup(GlobalPoint const &posPlane,
0126 HepGeom::Point3D<double> const &surfPos,
0127 GlobalVector const &normalPlane,
0128 HepGeom::Normal3D<double> const &surfNorm,
0129 const Plane &pDest) const;
0130
0131 template <class SurfaceType>
0132 void debugReportTrackState(std::string const ¤tContext,
0133 GlobalPoint const &cmsInitPos,
0134 CLHEP::Hep3Vector const &g4InitPos,
0135 GlobalVector const &cmsInitMom,
0136 CLHEP::Hep3Vector const &g4InitMom,
0137 const SurfaceType &pDest) const;
0138 };
0139
0140 #endif