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 <stdlib.h> // Include from 'system' 49 #include "G4ParticleDefinition.hh" // Include << 49 #include <vector> 50 #include "G4SmoothTrajectoryPoint.hh" // Incl << 51 #include "G4Step.hh" << 52 #include "G4Track.hh" << 53 #include "G4VTrajectory.hh" << 54 #include "G4ios.hh" // Include from 'system' << 55 #include "globals.hh" // Include from 'global << 56 50 57 #include "trkgdefs.hh" 51 #include "trkgdefs.hh" 58 #include <stdlib.h> // Include from 'system' << 52 #include "G4VTrajectory.hh" 59 << 53 #include "G4Allocator.hh" 60 #include <vector> << 54 #include "G4ios.hh" // Include from 'system' >> 55 #include "globals.hh" // Include from 'global' >> 56 #include "G4ParticleDefinition.hh" // Include from 'particle+matter' >> 57 #include "G4SmoothTrajectoryPoint.hh" // Include from 'tracking' >> 58 #include "G4Track.hh" >> 59 #include "G4Step.hh" 61 60 62 class G4Polyline; 61 class G4Polyline; 63 class G4ClonedSmoothTrajectory; << 64 62 65 class G4SmoothTrajectory : public G4VTrajector 63 class G4SmoothTrajectory : public G4VTrajectory 66 { 64 { 67 using G4TrajectoryPointContainer = std::vect 65 using G4TrajectoryPointContainer = std::vector<G4VTrajectoryPoint*>; 68 66 69 friend class G4ClonedSmoothTrajectory; << 67 public: 70 68 71 public: << 69 // Constructors/Destructor 72 // Constructors/Destructor << 70 // 73 // << 71 G4SmoothTrajectory(); 74 G4SmoothTrajectory() = default; << 72 G4SmoothTrajectory(const G4Track* aTrack); 75 G4SmoothTrajectory(const G4Track* aTrack); << 73 G4SmoothTrajectory(G4SmoothTrajectory &); 76 ~G4SmoothTrajectory() override; << 74 virtual ~G4SmoothTrajectory(); 77 G4SmoothTrajectory(G4SmoothTrajectory&); << 75 78 G4SmoothTrajectory& operator=(const G4Smooth << 76 // Operators 79 << 77 // 80 // Operators << 78 G4SmoothTrajectory& operator= (const G4SmoothTrajectory&) = delete; 81 // << 79 inline G4int operator == (const G4SmoothTrajectory& r) const; 82 inline G4bool operator==(const G4SmoothTraje << 80 inline void* operator new(size_t); 83 inline void* operator new(size_t); << 81 inline void operator delete(void*); 84 inline void operator delete(void*); << 82 85 << 83 // Get/Set functions 86 // cloning with the master thread allocator << 84 // 87 G4VTrajectory* CloneForMaster() const overri << 85 inline G4int GetTrackID() const 88 << 86 { return fTrackID; } 89 // Get/Set functions << 87 inline G4int GetParentID() const 90 // << 88 { return fParentID; } 91 inline G4int GetTrackID() const override { r << 89 inline G4String GetParticleName() const 92 inline G4int GetParentID() const override { << 90 { return ParticleName; } 93 inline G4String GetParticleName() const over << 91 inline G4double GetCharge() const 94 inline G4double GetCharge() const override { << 92 { return PDGCharge; } 95 inline G4int GetPDGEncoding() const override << 93 inline G4int GetPDGEncoding() const 96 inline G4double GetInitialKineticEnergy() co << 94 { return PDGEncoding; } 97 inline G4ThreeVector GetInitialMomentum() co << 95 inline G4double GetInitialKineticEnergy() const 98 << 96 { return initialKineticEnergy; } 99 // Other member functions << 97 inline G4ThreeVector GetInitialMomentum() const 100 // << 98 { return initialMomentum; } 101 void ShowTrajectory(std::ostream& os = G4cou << 99 102 void DrawTrajectory() const override; << 100 // Other member functions 103 void AppendStep(const G4Step* aStep) overrid << 101 // 104 G4int GetPointEntries() const override { ret << 102 virtual void ShowTrajectory(std::ostream& os=G4cout) const; 105 G4VTrajectoryPoint* GetPoint(G4int i) const << 103 virtual void DrawTrajectory() const; 106 void MergeTrajectory(G4VTrajectory* secondTr << 104 virtual void AppendStep(const G4Step* aStep); 107 << 105 virtual G4int GetPointEntries() const 108 G4ParticleDefinition* GetParticleDefinition( << 106 { return (G4int)positionRecord->size(); } 109 << 107 virtual G4VTrajectoryPoint* GetPoint(G4int i) const 110 // Get method for HEPRep style attributes << 108 { return (*positionRecord)[i]; } 111 // << 109 virtual void MergeTrajectory(G4VTrajectory* secondTrajectory); 112 const std::map<G4String, G4AttDef>* GetAttDe << 110 113 std::vector<G4AttValue>* CreateAttValues() c << 111 G4ParticleDefinition* GetParticleDefinition(); 114 << 112 115 private: << 113 // Get method for HEPRep style attributes 116 G4TrajectoryPointContainer* positionRecord = << 114 // 117 G4int fTrackID = 0; << 115 virtual const std::map<G4String,G4AttDef>* GetAttDefs() const; 118 G4int fParentID = 0; << 116 virtual std::vector<G4AttValue>* CreateAttValues() const; 119 G4int PDGEncoding = 0; << 117 120 G4double PDGCharge = 0.0; << 118 private: 121 G4String ParticleName = ""; << 119 122 G4double initialKineticEnergy = 0.0; << 120 G4TrajectoryPointContainer* positionRecord = nullptr; 123 G4ThreeVector initialMomentum; << 121 G4int fTrackID = 0; >> 122 G4int fParentID = 0; >> 123 G4int PDGEncoding = 0; >> 124 G4double PDGCharge = 0.0; >> 125 G4String ParticleName = ""; >> 126 G4double initialKineticEnergy = 0.0; >> 127 G4ThreeVector initialMomentum; 124 }; 128 }; 125 129 126 extern G4TRACKING_DLL G4Allocator<G4SmoothTraj << 130 extern G4TRACKING_DLL >> 131 G4Allocator<G4SmoothTrajectory>*& aSmoothTrajectoryAllocator(); 127 132 128 inline void* G4SmoothTrajectory::operator new( 133 inline void* G4SmoothTrajectory::operator new(size_t) 129 { 134 { 130 if (aSmoothTrajectoryAllocator() == nullptr) << 135 if (aSmoothTrajectoryAllocator() == nullptr) >> 136 { 131 aSmoothTrajectoryAllocator() = new G4Alloc 137 aSmoothTrajectoryAllocator() = new G4Allocator<G4SmoothTrajectory>; 132 } 138 } 133 return (void*)aSmoothTrajectoryAllocator()-> 139 return (void*)aSmoothTrajectoryAllocator()->MallocSingle(); 134 } 140 } 135 141 136 inline void G4SmoothTrajectory::operator delet 142 inline void G4SmoothTrajectory::operator delete(void* aTrajectory) 137 { 143 { 138 aSmoothTrajectoryAllocator()->FreeSingle((G4 144 aSmoothTrajectoryAllocator()->FreeSingle((G4SmoothTrajectory*)aTrajectory); 139 } 145 } 140 146 141 inline G4bool G4SmoothTrajectory::operator==(c << 147 inline G4int G4SmoothTrajectory::operator== (const G4SmoothTrajectory& r) const 142 { 148 { 143 return (this == &r); << 149 return (this==&r); 144 } 150 } 145 151 146 #endif 152 #endif 147 153