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 /// \file runAndEvent/RE01/src/RE01Trajectory.cc 27 /// \brief Implementation of the RE01Trajectory class 28 // 29 // 30 // 31 32 #include "RE01Trajectory.hh" 33 34 #include "RE01TrackInformation.hh" 35 36 #include "G4AttDef.hh" 37 #include "G4AttDefStore.hh" 38 #include "G4AttValue.hh" 39 #include "G4Circle.hh" 40 #include "G4Colour.hh" 41 #include "G4DynamicParticle.hh" 42 #include "G4ParticleTable.hh" 43 #include "G4ParticleTypes.hh" 44 #include "G4Polyline.hh" 45 #include "G4PrimaryParticle.hh" 46 #include "G4UIcommand.hh" 47 #include "G4UnitsTable.hh" 48 #include "G4VVisManager.hh" 49 #include "G4VisAttributes.hh" 50 51 G4ThreadLocal G4Allocator<RE01Trajectory>* myTrajectoryAllocator = 0; 52 53 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... 54 RE01Trajectory::RE01Trajectory(const G4Track* aTrack) 55 : G4VTrajectory(), fPositionRecord(0), fParticleDefinition(0) 56 { 57 fParticleDefinition = aTrack->GetDefinition(); 58 fParticleName = fParticleDefinition->GetParticleName(); 59 fPDGCharge = fParticleDefinition->GetPDGCharge(); 60 fPDGEncoding = fParticleDefinition->GetPDGEncoding(); 61 if (fParticleName == "unknown") { 62 G4PrimaryParticle* pp = aTrack->GetDynamicParticle()->GetPrimaryParticle(); 63 if (pp) { 64 if (pp->GetCharge() < DBL_MAX) fPDGCharge = pp->GetCharge(); 65 fPDGEncoding = pp->GetPDGcode(); 66 if (pp->GetG4code() != 0) { 67 fParticleName += " : "; 68 fParticleName += pp->GetG4code()->GetParticleName(); 69 } 70 } 71 } 72 fTrackID = aTrack->GetTrackID(); 73 RE01TrackInformation* trackInfo = (RE01TrackInformation*)(aTrack->GetUserInformation()); 74 fTrackStatus = trackInfo->GetTrackingStatus(); 75 if (fTrackStatus == 1) { 76 fParentID = aTrack->GetParentID(); 77 } 78 else if (fTrackStatus == 2) { 79 fParentID = trackInfo->GetSourceTrackID(); 80 } 81 else { 82 fParentID = -1; 83 } 84 fPositionRecord = new RE01TrajectoryPointContainer(); 85 fPositionRecord->push_back(new G4TrajectoryPoint(aTrack->GetPosition())); 86 fMomentum = aTrack->GetMomentum(); 87 fVertexPosition = aTrack->GetPosition(); 88 fGlobalTime = aTrack->GetGlobalTime(); 89 } 90 91 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... 92 RE01Trajectory::~RE01Trajectory() 93 { 94 size_t i; 95 for (i = 0; i < fPositionRecord->size(); i++) { 96 delete (*fPositionRecord)[i]; 97 } 98 fPositionRecord->clear(); 99 100 delete fPositionRecord; 101 } 102 103 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... 104 void RE01Trajectory::ShowTrajectory(std::ostream& os) const 105 { 106 os << G4endl << "TrackID =" << fTrackID << " : ParentID=" << fParentID 107 << " : TrackStatus=" << fTrackStatus << G4endl; 108 os << "Particle name : " << fParticleName << " PDG code : " << fPDGEncoding 109 << " Charge : " << fPDGCharge << G4endl; 110 os << "Original momentum : " << G4BestUnit(fMomentum, "Energy") << G4endl; 111 os << "Vertex : " << G4BestUnit(fVertexPosition, "Length") 112 << " Global time : " << G4BestUnit(fGlobalTime, "Time") << G4endl; 113 os << " Current trajectory has " << fPositionRecord->size() << " points." << G4endl; 114 115 for (size_t i = 0; i < fPositionRecord->size(); i++) { 116 G4TrajectoryPoint* aTrajectoryPoint = (G4TrajectoryPoint*)((*fPositionRecord)[i]); 117 os << "Point[" << i << "]" 118 << " Position= " << aTrajectoryPoint->GetPosition() << G4endl; 119 } 120 } 121 122 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... 123 void RE01Trajectory::DrawTrajectory() const 124 { 125 G4VVisManager* pVVisManager = G4VVisManager::GetConcreteInstance(); 126 G4ThreeVector pos; 127 128 G4Polyline pPolyline; 129 for (size_t i = 0; i < fPositionRecord->size(); i++) { 130 G4TrajectoryPoint* aTrajectoryPoint = (G4TrajectoryPoint*)((*fPositionRecord)[i]); 131 pos = aTrajectoryPoint->GetPosition(); 132 pPolyline.push_back(pos); 133 } 134 135 G4Colour colour(0.2, 0.2, 0.2); 136 if (fParticleDefinition == G4Gamma::GammaDefinition()) 137 colour = G4Colour(0., 0., 1.); 138 else if (fParticleDefinition == G4Electron::ElectronDefinition() 139 || fParticleDefinition == G4Positron::PositronDefinition()) 140 colour = G4Colour(1., 1., 0.); 141 else if (fParticleDefinition == G4MuonMinus::MuonMinusDefinition() 142 || fParticleDefinition == G4MuonPlus::MuonPlusDefinition()) 143 colour = G4Colour(0., 1., 0.); 144 else if (fParticleDefinition->GetParticleType() == "meson") { 145 if (fPDGCharge != 0.) 146 colour = G4Colour(1., 0., 0.); 147 else 148 colour = G4Colour(0.5, 0., 0.); 149 } 150 else if (fParticleDefinition->GetParticleType() == "baryon") { 151 if (fPDGCharge != 0.) 152 colour = G4Colour(0., 1., 1.); 153 else 154 colour = G4Colour(0., 0.5, 0.5); 155 } 156 157 G4VisAttributes attribs(colour); 158 pPolyline.SetVisAttributes(attribs); 159 if (pVVisManager) pVVisManager->Draw(pPolyline); 160 } 161 162 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... 163 const std::map<G4String, G4AttDef>* RE01Trajectory::GetAttDefs() const 164 { 165 G4bool isNew; 166 std::map<G4String, G4AttDef>* store = G4AttDefStore::GetInstance("RE01Trajectory", isNew); 167 if (isNew) { 168 G4String id("ID"); 169 (*store)[id] = G4AttDef(id, "Track ID", "Bookkeeping", "", "G4int"); 170 171 G4String pid("PID"); 172 (*store)[pid] = G4AttDef(pid, "Parent ID", "Bookkeeping", "", "G4int"); 173 174 G4String status("Status"); 175 (*store)[status] = G4AttDef(status, "Track Status", "Bookkeeping", "", "G4int"); 176 177 G4String pn("PN"); 178 (*store)[pn] = G4AttDef(pn, "Particle Name", "Bookkeeping", "", "G4String"); 179 180 G4String ch("Ch"); 181 (*store)[ch] = G4AttDef(ch, "Charge", "Physics", "e+", "G4double"); 182 183 G4String pdg("PDG"); 184 (*store)[pdg] = G4AttDef(pdg, "PDG Encoding", "Bookkeeping", "", "G4int"); 185 186 G4String imom("IMom"); 187 (*store)[imom] = G4AttDef(imom, "Momentum of track at start of trajectory", "Physics", 188 "G4BestUnit", "G4ThreeVector"); 189 190 G4String imag("IMag"); 191 (*store)[imag] = G4AttDef(imag, "Magnitude of momentum of track at start of trajectory", 192 "Physics", "G4BestUnit", "G4double"); 193 194 G4String vtxPos("VtxPos"); 195 (*store)[vtxPos] = 196 G4AttDef(vtxPos, "Vertex position", "Physics", "G4BestUnit", "G4ThreeVector"); 197 198 G4String ntp("NTP"); 199 (*store)[ntp] = G4AttDef(ntp, "No. of points", "Bookkeeping", "", "G4int"); 200 } 201 return store; 202 } 203 204 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... 205 std::vector<G4AttValue>* RE01Trajectory::CreateAttValues() const 206 { 207 std::vector<G4AttValue>* values = new std::vector<G4AttValue>; 208 209 values->push_back(G4AttValue("ID", G4UIcommand::ConvertToString(fTrackID), "")); 210 211 values->push_back(G4AttValue("PID", G4UIcommand::ConvertToString(fParentID), "")); 212 213 values->push_back(G4AttValue("Status", G4UIcommand::ConvertToString(fTrackStatus), "")); 214 215 values->push_back(G4AttValue("PN", fParticleName, "")); 216 217 values->push_back(G4AttValue("Ch", G4UIcommand::ConvertToString(fPDGCharge), "")); 218 219 values->push_back(G4AttValue("PDG", G4UIcommand::ConvertToString(fPDGEncoding), "")); 220 221 values->push_back(G4AttValue("IMom", G4BestUnit(fMomentum, "Energy"), "")); 222 223 values->push_back(G4AttValue("IMag", G4BestUnit(fMomentum.mag(), "Energy"), "")); 224 225 values->push_back(G4AttValue("VtxPos", G4BestUnit(fVertexPosition, "Length"), "")); 226 227 values->push_back(G4AttValue("NTP", G4UIcommand::ConvertToString(GetPointEntries()), "")); 228 229 return values; 230 } 231 232 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... 233 void RE01Trajectory::AppendStep(const G4Step* aStep) 234 { 235 fPositionRecord->push_back(new G4TrajectoryPoint(aStep->GetPostStepPoint()->GetPosition())); 236 } 237 238 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... 239 void RE01Trajectory::MergeTrajectory(G4VTrajectory* secondTrajectory) 240 { 241 if (!secondTrajectory) return; 242 243 RE01Trajectory* seco = (RE01Trajectory*)secondTrajectory; 244 G4int ent = seco->GetPointEntries(); 245 // 246 // initial point of the second trajectory should not be merged 247 for (int i = 1; i < ent; i++) { 248 fPositionRecord->push_back((*(seco->fPositionRecord))[i]); 249 } 250 delete (*seco->fPositionRecord)[0]; 251 seco->fPositionRecord->clear(); 252 } 253