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