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 // G4Trajectory 26 // G4Trajectory 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 being tracked. 31 // It includes information of: 31 // It includes information of: 32 // 1) List of trajectory points which comp 32 // 1) List of trajectory points which compose the trajectory; 33 // 2) Static information of the particle w << 33 // 2) Static information of the particle which generated the 34 // trajectory; 34 // trajectory; 35 // 3) Track ID and parent particle ID of t 35 // 3) Track ID and parent particle ID of the trajectory. 36 36 37 // Contact: 37 // Contact: 38 // Questions and comments to this code shoul 38 // Questions and comments to this code should be sent to 39 // Katsuya Amako (e-mail: Katsuya.Amako@k 39 // Katsuya Amako (e-mail: Katsuya.Amako@kek.jp) 40 // Makoto Asai (e-mail: asai@slac.stanf 40 // Makoto Asai (e-mail: asai@slac.stanford.edu) 41 // Takashi Sasaki (e-mail: Takashi.Sasaki@ 41 // Takashi Sasaki (e-mail: Takashi.Sasaki@kek.jp) 42 // ------------------------------------------- 42 // -------------------------------------------------------------------- 43 #ifndef G4Trajectory_hh 43 #ifndef G4Trajectory_hh 44 #define G4Trajectory_hh 1 44 #define G4Trajectory_hh 1 45 45 >> 46 #include <stdlib.h> // Include from 'system' >> 47 #include <vector> >> 48 >> 49 #include "trkgdefs.hh" >> 50 #include "G4VTrajectory.hh" 46 #include "G4Allocator.hh" 51 #include "G4Allocator.hh" >> 52 #include "G4ios.hh" // Include from 'system' >> 53 #include "globals.hh" // Include from 'global' 47 #include "G4ParticleDefinition.hh" // Include 54 #include "G4ParticleDefinition.hh" // Include from 'particle+matter' 48 #include "G4Step.hh" << 55 #include "G4TrajectoryPoint.hh" // Include from 'tracking' 49 #include "G4Track.hh" 56 #include "G4Track.hh" 50 #include "G4TrajectoryPoint.hh" // Include fr << 57 #include "G4Step.hh" 51 #include "G4VTrajectory.hh" << 52 #include "G4ios.hh" // Include from 'system' << 53 #include "globals.hh" // Include from 'global << 54 #include "G4Threading.hh" << 55 << 56 #include "trkgdefs.hh" << 57 #include <stdlib.h> // Include from 'system' << 58 << 59 #include <vector> << 60 58 61 class G4Polyline; 59 class G4Polyline; 62 class G4ClonedTrajectory; << 63 60 64 class G4Trajectory : public G4VTrajectory 61 class G4Trajectory : public G4VTrajectory 65 { 62 { >> 63 66 using G4TrajectoryPointContainer = std::vect 64 using G4TrajectoryPointContainer = std::vector<G4VTrajectoryPoint*>; 67 65 68 friend class G4ClonedTrajectory; << 66 public: 69 67 70 public: << 68 // Constructors/Destructor 71 // Constructors/Destructor << 72 69 73 G4Trajectory() = default; << 70 G4Trajectory(); 74 G4Trajectory(const G4Track* aTrack); << 71 G4Trajectory(const G4Track* aTrack); 75 G4Trajectory(G4Trajectory&); << 72 G4Trajectory(G4Trajectory &); 76 ~G4Trajectory() override; << 73 virtual ~G4Trajectory(); 77 << 74 78 // Operators << 75 // Operators 79 << 76 80 inline void* operator new(size_t); << 77 inline void* operator new(size_t); 81 inline void operator delete(void*); << 78 inline void operator delete(void*); 82 inline G4bool operator==(const G4Trajectory& << 79 inline G4int operator == (const G4Trajectory& r) const; 83 << 80 84 // cloning with the master thread allocator << 81 // Get/Set functions 85 G4VTrajectory* CloneForMaster() const overri << 82 86 << 83 inline G4int GetTrackID() const 87 // Get/Set functions << 84 { return fTrackID; } 88 << 85 inline G4int GetParentID() const 89 inline G4int GetTrackID() const override { r << 86 { return fParentID; } 90 inline G4int GetParentID() const override { << 87 inline G4String GetParticleName() const 91 inline G4String GetParticleName() const over << 88 { return ParticleName; } 92 inline G4double GetCharge() const override { << 89 inline G4double GetCharge() const 93 inline G4int GetPDGEncoding() const override << 90 { return PDGCharge; } 94 inline G4double GetInitialKineticEnergy() co << 91 inline G4int GetPDGEncoding() const 95 inline G4ThreeVector GetInitialMomentum() co << 92 { return PDGEncoding; } 96 << 93 inline G4double GetInitialKineticEnergy() const 97 // Other member functions << 94 { return initialKineticEnergy; } 98 << 95 inline G4ThreeVector GetInitialMomentum() const 99 void ShowTrajectory(std::ostream& os = G4cou << 96 { return initialMomentum; } 100 void DrawTrajectory() const override; << 97 101 void AppendStep(const G4Step* aStep) overrid << 98 // Other member functions 102 G4int GetPointEntries() const override { ret << 99 103 G4VTrajectoryPoint* GetPoint(G4int i) const << 100 virtual void ShowTrajectory(std::ostream& os=G4cout) const; 104 void MergeTrajectory(G4VTrajectory* secondTr << 101 virtual void DrawTrajectory() const; 105 << 102 virtual void AppendStep(const G4Step* aStep); 106 G4ParticleDefinition* GetParticleDefinition( << 103 virtual G4int GetPointEntries() const 107 << 104 { return G4int(positionRecord->size()); } 108 const std::map<G4String, G4AttDef>* GetAttDe << 105 virtual G4VTrajectoryPoint* GetPoint(G4int i) const 109 std::vector<G4AttValue>* CreateAttValues() c << 106 { return (*positionRecord)[i]; } 110 << 107 virtual void MergeTrajectory(G4VTrajectory* secondTrajectory); 111 private: << 108 112 G4TrajectoryPointContainer* positionRecord = << 109 G4ParticleDefinition* GetParticleDefinition(); 113 G4int fTrackID = 0; << 110 114 G4int fParentID = 0; << 111 virtual const std::map<G4String,G4AttDef>* GetAttDefs() const; 115 G4int PDGEncoding = 0; << 112 virtual std::vector<G4AttValue>* CreateAttValues() const; 116 G4double PDGCharge = 0.0; << 113 117 G4String ParticleName = "dummy"; << 114 private: 118 G4double initialKineticEnergy = 0.0; << 115 119 G4ThreeVector initialMomentum; << 116 G4TrajectoryPointContainer* positionRecord = nullptr; >> 117 G4int fTrackID = 0; >> 118 G4int fParentID = 0; >> 119 G4int PDGEncoding = 0; >> 120 G4double PDGCharge = 0.0; >> 121 G4String ParticleName = ""; >> 122 G4double initialKineticEnergy = 0.0; >> 123 G4ThreeVector initialMomentum; 120 }; 124 }; 121 125 122 extern G4TRACKING_DLL G4Allocator<G4Trajectory 126 extern G4TRACKING_DLL G4Allocator<G4Trajectory>*& aTrajectoryAllocator(); 123 127 124 inline void* G4Trajectory::operator new(size_t 128 inline void* G4Trajectory::operator new(size_t) 125 { 129 { 126 if (aTrajectoryAllocator() == nullptr) { << 130 if (aTrajectoryAllocator() == nullptr) >> 131 { 127 aTrajectoryAllocator() = new G4Allocator<G 132 aTrajectoryAllocator() = new G4Allocator<G4Trajectory>; 128 } 133 } 129 return (void*)aTrajectoryAllocator()->Malloc 134 return (void*)aTrajectoryAllocator()->MallocSingle(); 130 } 135 } 131 136 132 inline void G4Trajectory::operator delete(void 137 inline void G4Trajectory::operator delete(void* aTrajectory) 133 { 138 { 134 aTrajectoryAllocator()->FreeSingle((G4Trajec 139 aTrajectoryAllocator()->FreeSingle((G4Trajectory*)aTrajectory); 135 } 140 } 136 141 137 inline G4bool G4Trajectory::operator==(const G << 142 inline G4int G4Trajectory::operator == (const G4Trajectory& r) const >> 143 { >> 144 return (this==&r); >> 145 } 138 146 139 #endif 147 #endif 140 148