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