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 << 27 // 26 // 28 // Class description: << 29 // 27 // 30 // This class represents the trajectory of a p << 28 //--------------------------------------------------------------- 31 // It includes information of << 29 // 32 // 1) List of trajectory points which compos << 30 // G4SmoothTrajectory.hh 33 // 2) static information of particle which g << 31 // 34 // trajectory, << 32 // class description: 35 // 3) trackID and parent particle ID of the << 33 // This class represents the trajectory of a particle tracked. 36 // 4) Auxiliary points will be associated to << 34 // It includes information of 37 // to assist drawing smoothly curved traj << 35 // 1) List of trajectory points which compose the trajectory, 38 << 36 // 2) static information of particle which generated the 39 // Contact: << 37 // trajectory, 40 // Questions and comments to this code shoul << 38 // 3) trackID and parent particle ID of the trajectory, 41 // Katsuya Amako (e-mail: Katsuya.Amako@k << 39 // 4) Auxiliary points will be associated to G4SmoothTrajectoryPoint 42 // Makoto Asai (e-mail: asai@slac.stanf << 40 // to assist drawing smoothly curved trajectory. 43 // Takashi Sasaki (e-mail: Takashi.Sasaki@ << 41 // 44 // ------------------------------------------- << 42 // --------------------------------------------------------------- 45 #ifndef G4SmoothTrajectory_hh << 46 #define G4SmoothTrajectory_hh 1 << 47 43 48 #include "G4Allocator.hh" << 44 class G4SmoothTrajectory; 49 #include "G4ParticleDefinition.hh" // Include << 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 45 57 #include "trkgdefs.hh" << 46 #ifndef G4SmoothTrajectory_h 58 #include <stdlib.h> // Include from 'system' << 47 #define G4SmoothTrajectory_h 1 59 48 >> 49 #include <stdlib.h> // Include from 'system' 60 #include <vector> 50 #include <vector> 61 51 62 class G4Polyline; << 52 #include "trkgdefs.hh" 63 class G4ClonedSmoothTrajectory; << 53 #include "G4VTrajectory.hh" >> 54 #include "G4Allocator.hh" >> 55 #include "G4ios.hh" // Include from 'system' >> 56 #include "globals.hh" // Include from 'global' >> 57 #include "G4ParticleDefinition.hh" // Include from 'particle+matter' >> 58 #include "G4SmoothTrajectoryPoint.hh" // Include from 'tracking' >> 59 #include "G4Track.hh" >> 60 #include "G4Step.hh" >> 61 >> 62 class G4Polyline; // Forward declaration >> 63 >> 64 typedef std::vector<G4VTrajectoryPoint*> TrajectoryPointContainer; 64 65 65 class G4SmoothTrajectory : public G4VTrajector 66 class G4SmoothTrajectory : public G4VTrajectory 66 { 67 { 67 using G4TrajectoryPointContainer = std::vect << 68 68 69 friend class G4ClonedSmoothTrajectory; << 69 //-------- >> 70 public: // with description >> 71 //-------- >> 72 >> 73 // Constructor/Destrcutor >> 74 >> 75 G4SmoothTrajectory(); >> 76 >> 77 G4SmoothTrajectory(const G4Track* aTrack); >> 78 G4SmoothTrajectory(G4SmoothTrajectory &); >> 79 virtual ~G4SmoothTrajectory(); >> 80 >> 81 private: >> 82 G4SmoothTrajectory& operator= (const G4SmoothTrajectory&); >> 83 >> 84 // Operators >> 85 public: >> 86 inline void* operator new(size_t); >> 87 inline void operator delete(void*); >> 88 inline int operator == (const G4SmoothTrajectory& right) const >> 89 {return (this==&right);} >> 90 >> 91 // Get/Set functions >> 92 inline G4int GetTrackID() const >> 93 { return fTrackID; } >> 94 inline G4int GetParentID() const >> 95 { return fParentID; } >> 96 inline G4String GetParticleName() const >> 97 { return ParticleName; } >> 98 inline G4double GetCharge() const >> 99 { return PDGCharge; } >> 100 inline G4int GetPDGEncoding() const >> 101 { return PDGEncoding; } >> 102 inline G4double GetInitialKineticEnergy() const >> 103 { return initialKineticEnergy; } >> 104 inline G4ThreeVector GetInitialMomentum() const >> 105 { return initialMomentum; } >> 106 >> 107 // Other member functions >> 108 virtual void ShowTrajectory(std::ostream& os=G4cout) const; >> 109 virtual void DrawTrajectory() const; >> 110 virtual void AppendStep(const G4Step* aStep); >> 111 virtual int GetPointEntries() const { return positionRecord->size(); } >> 112 virtual G4VTrajectoryPoint* GetPoint(G4int i) const >> 113 { return (*positionRecord)[i]; } >> 114 virtual void MergeTrajectory(G4VTrajectory* secondTrajectory); >> 115 >> 116 G4ParticleDefinition* GetParticleDefinition(); >> 117 >> 118 // Get method for HEPRep style attributes >> 119 virtual const std::map<G4String,G4AttDef>* GetAttDefs() const; >> 120 virtual std::vector<G4AttValue>* CreateAttValues() const; >> 121 >> 122 //--------- >> 123 private: >> 124 //--------- >> 125 >> 126 TrajectoryPointContainer* positionRecord; >> 127 G4int fTrackID; >> 128 G4int fParentID; >> 129 G4int PDGEncoding; >> 130 G4double PDGCharge; >> 131 G4String ParticleName; >> 132 G4double initialKineticEnergy; >> 133 G4ThreeVector initialMomentum; 70 134 71 public: << 72 // Constructors/Destructor << 73 // << 74 G4SmoothTrajectory() = default; << 75 G4SmoothTrajectory(const G4Track* aTrack); << 76 ~G4SmoothTrajectory() override; << 77 G4SmoothTrajectory(G4SmoothTrajectory&); << 78 G4SmoothTrajectory& operator=(const G4Smooth << 79 << 80 // Operators << 81 // << 82 inline G4bool operator==(const G4SmoothTraje << 83 inline void* operator new(size_t); << 84 inline void operator delete(void*); << 85 << 86 // cloning with the master thread allocator << 87 G4VTrajectory* CloneForMaster() const overri << 88 << 89 // Get/Set functions << 90 // << 91 inline G4int GetTrackID() const override { r << 92 inline G4int GetParentID() const override { << 93 inline G4String GetParticleName() const over << 94 inline G4double GetCharge() const override { << 95 inline G4int GetPDGEncoding() const override << 96 inline G4double GetInitialKineticEnergy() co << 97 inline G4ThreeVector GetInitialMomentum() co << 98 << 99 // Other member functions << 100 // << 101 void ShowTrajectory(std::ostream& os = G4cou << 102 void DrawTrajectory() const override; << 103 void AppendStep(const G4Step* aStep) overrid << 104 G4int GetPointEntries() const override { ret << 105 G4VTrajectoryPoint* GetPoint(G4int i) const << 106 void MergeTrajectory(G4VTrajectory* secondTr << 107 << 108 G4ParticleDefinition* GetParticleDefinition( << 109 << 110 // Get method for HEPRep style attributes << 111 // << 112 const std::map<G4String, G4AttDef>* GetAttDe << 113 std::vector<G4AttValue>* CreateAttValues() c << 114 << 115 private: << 116 G4TrajectoryPointContainer* positionRecord = << 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; << 124 }; 135 }; 125 136 126 extern G4TRACKING_DLL G4Allocator<G4SmoothTraj 137 extern G4TRACKING_DLL G4Allocator<G4SmoothTrajectory>*& aSmoothTrajectoryAllocator(); 127 138 128 inline void* G4SmoothTrajectory::operator new( 139 inline void* G4SmoothTrajectory::operator new(size_t) 129 { 140 { 130 if (aSmoothTrajectoryAllocator() == nullptr) << 141 if (!aSmoothTrajectoryAllocator()) 131 aSmoothTrajectoryAllocator() = new G4Alloc << 142 { aSmoothTrajectoryAllocator() = new G4Allocator<G4SmoothTrajectory>; } 132 } << 133 return (void*)aSmoothTrajectoryAllocator()-> 143 return (void*)aSmoothTrajectoryAllocator()->MallocSingle(); 134 } 144 } 135 145 136 inline void G4SmoothTrajectory::operator delet 146 inline void G4SmoothTrajectory::operator delete(void* aTrajectory) 137 { 147 { 138 aSmoothTrajectoryAllocator()->FreeSingle((G4 148 aSmoothTrajectoryAllocator()->FreeSingle((G4SmoothTrajectory*)aTrajectory); 139 } << 140 << 141 inline G4bool G4SmoothTrajectory::operator==(c << 142 { << 143 return (this == &r); << 144 } 149 } 145 150 146 #endif 151 #endif 147 152