Geant4 Cross Reference |
1 // 1 // 2 // ******************************************* 2 // ******************************************************************** 3 // * License and Disclaimer 3 // * License and Disclaimer * 4 // * 4 // * * 5 // * The Geant4 software is copyright of th 5 // * The Geant4 software is copyright of the Copyright Holders of * 6 // * the Geant4 Collaboration. It is provided 6 // * the Geant4 Collaboration. It is provided under the terms and * 7 // * conditions of the Geant4 Software License 7 // * conditions of the Geant4 Software License, included in the file * 8 // * LICENSE and available at http://cern.ch/ 8 // * LICENSE and available at http://cern.ch/geant4/license . These * 9 // * include a list of copyright holders. 9 // * include a list of copyright holders. * 10 // * 10 // * * 11 // * Neither the authors of this software syst 11 // * Neither the authors of this software system, nor their employing * 12 // * institutes,nor the agencies providing fin 12 // * institutes,nor the agencies providing financial support for this * 13 // * work make any representation or warran 13 // * work make any representation or warranty, express or implied, * 14 // * regarding this software system or assum 14 // * regarding this software system or assume any liability for its * 15 // * use. Please see the license in the file 15 // * use. Please see the license in the file LICENSE and URL above * 16 // * for the full disclaimer and the limitatio 16 // * for the full disclaimer and the limitation of liability. * 17 // * 17 // * * 18 // * This code implementation is the result 18 // * This code implementation is the result of the scientific and * 19 // * technical work of the GEANT4 collaboratio 19 // * technical work of the GEANT4 collaboration. * 20 // * By using, copying, modifying or distri 20 // * By using, copying, modifying or distributing the software (or * 21 // * any work based on the software) you ag 21 // * any work based on the software) you agree to acknowledge its * 22 // * use in resulting scientific publicati 22 // * use in resulting scientific publications, and indicate your * 23 // * acceptance of all terms of the Geant4 Sof 23 // * acceptance of all terms of the Geant4 Software license. * 24 // ******************************************* 24 // ******************************************************************** 25 // 25 // 26 // G4SmoothTrajectory 26 // G4SmoothTrajectory 27 // 27 // 28 // Class description: 28 // Class description: 29 // 29 // 30 // This class represents the trajectory of a p 30 // This class represents the trajectory of a particle tracked. 31 // It includes information of 31 // It includes information of 32 // 1) List of trajectory points which compos 32 // 1) List of trajectory points which compose the trajectory, 33 // 2) static information of particle which g 33 // 2) static information of particle which generates the 34 // trajectory, 34 // trajectory, 35 // 3) trackID and parent particle ID of the 35 // 3) trackID and parent particle ID of the trajectory, 36 // 4) Auxiliary points will be associated to 36 // 4) Auxiliary points will be associated to G4SmoothTrajectoryPoint 37 // to assist drawing smoothly curved traj 37 // to assist drawing smoothly curved trajectory. 38 38 39 // Contact: 39 // Contact: 40 // Questions and comments to this code shoul 40 // Questions and comments to this code should be sent to 41 // Katsuya Amako (e-mail: Katsuya.Amako@k 41 // Katsuya Amako (e-mail: Katsuya.Amako@kek.jp) 42 // Makoto Asai (e-mail: asai@slac.stanf 42 // Makoto Asai (e-mail: asai@slac.stanford.edu) 43 // Takashi Sasaki (e-mail: Takashi.Sasaki@ 43 // Takashi Sasaki (e-mail: Takashi.Sasaki@kek.jp) 44 // ------------------------------------------- 44 // -------------------------------------------------------------------- 45 #ifndef G4SmoothTrajectory_hh 45 #ifndef G4SmoothTrajectory_hh 46 #define G4SmoothTrajectory_hh 1 46 #define G4SmoothTrajectory_hh 1 47 47 48 #include "G4Allocator.hh" 48 #include "G4Allocator.hh" 49 #include "G4ParticleDefinition.hh" // Include 49 #include "G4ParticleDefinition.hh" // Include from 'particle+matter' 50 #include "G4SmoothTrajectoryPoint.hh" // Incl 50 #include "G4SmoothTrajectoryPoint.hh" // Include from 'tracking' 51 #include "G4Step.hh" 51 #include "G4Step.hh" 52 #include "G4Track.hh" 52 #include "G4Track.hh" 53 #include "G4VTrajectory.hh" 53 #include "G4VTrajectory.hh" 54 #include "G4ios.hh" // Include from 'system' 54 #include "G4ios.hh" // Include from 'system' 55 #include "globals.hh" // Include from 'global 55 #include "globals.hh" // Include from 'global' 56 56 57 #include "trkgdefs.hh" 57 #include "trkgdefs.hh" 58 #include <stdlib.h> // Include from 'system' 58 #include <stdlib.h> // Include from 'system' 59 59 60 #include <vector> 60 #include <vector> 61 61 62 class G4Polyline; 62 class G4Polyline; 63 class G4ClonedSmoothTrajectory; << 64 63 65 class G4SmoothTrajectory : public G4VTrajector 64 class G4SmoothTrajectory : public G4VTrajectory 66 { 65 { 67 using G4TrajectoryPointContainer = std::vect 66 using G4TrajectoryPointContainer = std::vector<G4VTrajectoryPoint*>; 68 67 69 friend class G4ClonedSmoothTrajectory; << 70 << 71 public: 68 public: 72 // Constructors/Destructor 69 // Constructors/Destructor 73 // 70 // 74 G4SmoothTrajectory() = default; 71 G4SmoothTrajectory() = default; 75 G4SmoothTrajectory(const G4Track* aTrack); 72 G4SmoothTrajectory(const G4Track* aTrack); 76 ~G4SmoothTrajectory() override; 73 ~G4SmoothTrajectory() override; 77 G4SmoothTrajectory(G4SmoothTrajectory&); 74 G4SmoothTrajectory(G4SmoothTrajectory&); 78 G4SmoothTrajectory& operator=(const G4Smooth 75 G4SmoothTrajectory& operator=(const G4SmoothTrajectory&) = delete; 79 76 80 // Operators 77 // Operators 81 // 78 // 82 inline G4bool operator==(const G4SmoothTraje 79 inline G4bool operator==(const G4SmoothTrajectory& r) const; 83 inline void* operator new(size_t); 80 inline void* operator new(size_t); 84 inline void operator delete(void*); 81 inline void operator delete(void*); 85 << 86 // cloning with the master thread allocator << 87 G4VTrajectory* CloneForMaster() const overri << 88 82 89 // Get/Set functions 83 // Get/Set functions 90 // 84 // 91 inline G4int GetTrackID() const override { r 85 inline G4int GetTrackID() const override { return fTrackID; } 92 inline G4int GetParentID() const override { 86 inline G4int GetParentID() const override { return fParentID; } 93 inline G4String GetParticleName() const over 87 inline G4String GetParticleName() const override { return ParticleName; } 94 inline G4double GetCharge() const override { 88 inline G4double GetCharge() const override { return PDGCharge; } 95 inline G4int GetPDGEncoding() const override 89 inline G4int GetPDGEncoding() const override { return PDGEncoding; } 96 inline G4double GetInitialKineticEnergy() co 90 inline G4double GetInitialKineticEnergy() const { return initialKineticEnergy; } 97 inline G4ThreeVector GetInitialMomentum() co 91 inline G4ThreeVector GetInitialMomentum() const override { return initialMomentum; } 98 92 99 // Other member functions 93 // Other member functions 100 // 94 // 101 void ShowTrajectory(std::ostream& os = G4cou 95 void ShowTrajectory(std::ostream& os = G4cout) const override; 102 void DrawTrajectory() const override; 96 void DrawTrajectory() const override; 103 void AppendStep(const G4Step* aStep) overrid 97 void AppendStep(const G4Step* aStep) override; 104 G4int GetPointEntries() const override { ret 98 G4int GetPointEntries() const override { return (G4int)positionRecord->size(); } 105 G4VTrajectoryPoint* GetPoint(G4int i) const 99 G4VTrajectoryPoint* GetPoint(G4int i) const override { return (*positionRecord)[i]; } 106 void MergeTrajectory(G4VTrajectory* secondTr 100 void MergeTrajectory(G4VTrajectory* secondTrajectory) override; 107 101 108 G4ParticleDefinition* GetParticleDefinition( 102 G4ParticleDefinition* GetParticleDefinition(); 109 103 110 // Get method for HEPRep style attributes 104 // Get method for HEPRep style attributes 111 // 105 // 112 const std::map<G4String, G4AttDef>* GetAttDe 106 const std::map<G4String, G4AttDef>* GetAttDefs() const override; 113 std::vector<G4AttValue>* CreateAttValues() c 107 std::vector<G4AttValue>* CreateAttValues() const override; 114 108 115 private: 109 private: 116 G4TrajectoryPointContainer* positionRecord = 110 G4TrajectoryPointContainer* positionRecord = nullptr; 117 G4int fTrackID = 0; 111 G4int fTrackID = 0; 118 G4int fParentID = 0; 112 G4int fParentID = 0; 119 G4int PDGEncoding = 0; 113 G4int PDGEncoding = 0; 120 G4double PDGCharge = 0.0; 114 G4double PDGCharge = 0.0; 121 G4String ParticleName = ""; 115 G4String ParticleName = ""; 122 G4double initialKineticEnergy = 0.0; 116 G4double initialKineticEnergy = 0.0; 123 G4ThreeVector initialMomentum; 117 G4ThreeVector initialMomentum; 124 }; 118 }; 125 119 126 extern G4TRACKING_DLL G4Allocator<G4SmoothTraj 120 extern G4TRACKING_DLL G4Allocator<G4SmoothTrajectory>*& aSmoothTrajectoryAllocator(); 127 121 128 inline void* G4SmoothTrajectory::operator new( 122 inline void* G4SmoothTrajectory::operator new(size_t) 129 { 123 { 130 if (aSmoothTrajectoryAllocator() == nullptr) 124 if (aSmoothTrajectoryAllocator() == nullptr) { 131 aSmoothTrajectoryAllocator() = new G4Alloc 125 aSmoothTrajectoryAllocator() = new G4Allocator<G4SmoothTrajectory>; 132 } 126 } 133 return (void*)aSmoothTrajectoryAllocator()-> 127 return (void*)aSmoothTrajectoryAllocator()->MallocSingle(); 134 } 128 } 135 129 136 inline void G4SmoothTrajectory::operator delet 130 inline void G4SmoothTrajectory::operator delete(void* aTrajectory) 137 { 131 { 138 aSmoothTrajectoryAllocator()->FreeSingle((G4 132 aSmoothTrajectoryAllocator()->FreeSingle((G4SmoothTrajectory*)aTrajectory); 139 } 133 } 140 134 141 inline G4bool G4SmoothTrajectory::operator==(c 135 inline G4bool G4SmoothTrajectory::operator==(const G4SmoothTrajectory& r) const 142 { 136 { 143 return (this == &r); 137 return (this == &r); 144 } 138 } 145 139 146 #endif 140 #endif 147 141