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/RE04/src/RE04Trajectory. 26 /// \file runAndEvent/RE04/src/RE04Trajectory.cc 27 /// \brief Implementation of the RE04Trajector 27 /// \brief Implementation of the RE04Trajectory class 28 // 28 // 29 // 29 // 30 #include "RE04Trajectory.hh" 30 #include "RE04Trajectory.hh" 31 << 32 #include "RE04TrajectoryPoint.hh" 31 #include "RE04TrajectoryPoint.hh" 33 << 32 #include "G4ParticleTable.hh" 34 #include "G4AttDef.hh" << 35 #include "G4AttDefStore.hh" 33 #include "G4AttDefStore.hh" >> 34 #include "G4AttDef.hh" 36 #include "G4AttValue.hh" 35 #include "G4AttValue.hh" 37 #include "G4ParticleTable.hh" << 38 #include "G4UIcommand.hh" 36 #include "G4UIcommand.hh" 39 #include "G4UnitsTable.hh" 37 #include "G4UnitsTable.hh" 40 38 41 // #define G4ATTDEBUG << 39 //#define G4ATTDEBUG 42 #ifdef G4ATTDEBUG 40 #ifdef G4ATTDEBUG 43 # include "G4AttCheck.hh" << 41 #include "G4AttCheck.hh" 44 #endif 42 #endif 45 43 46 G4ThreadLocal G4Allocator<RE04Trajectory>* faT << 44 G4ThreadLocal G4Allocator<RE04Trajectory> * faTrajAllocator = 0; 47 45 48 //....oooOO0OOooo........oooOO0OOooo........oo 46 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... 49 RE04Trajectory::RE04Trajectory() 47 RE04Trajectory::RE04Trajectory() 50 : G4VTrajectory(), << 48 : G4VTrajectory(), 51 fPositionRecord(0), << 49 fPositionRecord(0), fTrackID(0), fParentID(0), 52 fTrackID(0), << 50 fPDGEncoding( 0 ), fPDGCharge(0.0), fParticleName(""), 53 fParentID(0), << 51 fInitialKineticEnergy( 0. ), fInitialMomentum( G4ThreeVector() ) 54 fPDGEncoding(0), << 52 {;} 55 fPDGCharge(0.0), << 56 fParticleName(""), << 57 fInitialKineticEnergy(0.), << 58 fInitialMomentum(G4ThreeVector()) << 59 { << 60 ; << 61 } << 62 53 63 //....oooOO0OOooo........oooOO0OOooo........oo 54 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... 64 RE04Trajectory::RE04Trajectory(const G4Track* 55 RE04Trajectory::RE04Trajectory(const G4Track* aTrack) 65 { 56 { 66 G4ParticleDefinition* fpParticleDefinition = << 57 G4ParticleDefinition * fpParticleDefinition = aTrack->GetDefinition(); 67 fParticleName = fpParticleDefinition->GetPar << 58 fParticleName = fpParticleDefinition->GetParticleName(); 68 fPDGCharge = fpParticleDefinition->GetPDGCha << 59 fPDGCharge = fpParticleDefinition->GetPDGCharge(); 69 fPDGEncoding = fpParticleDefinition->GetPDGE << 60 fPDGEncoding = fpParticleDefinition->GetPDGEncoding(); 70 fTrackID = aTrack->GetTrackID(); << 61 fTrackID = aTrack->GetTrackID(); 71 fParentID = aTrack->GetParentID(); << 62 fParentID = aTrack->GetParentID(); 72 fInitialKineticEnergy = aTrack->GetKineticEn << 63 fInitialKineticEnergy = aTrack->GetKineticEnergy(); 73 fInitialMomentum = aTrack->GetMomentum(); << 64 fInitialMomentum = aTrack->GetMomentum(); 74 fPositionRecord = new TrajectoryPointContain << 65 fPositionRecord = new TrajectoryPointContainer(); 75 // Following is for the first trajectory poi << 66 // Following is for the first trajectory point 76 fPositionRecord->push_back(new RE04Trajector << 67 fPositionRecord->push_back(new RE04TrajectoryPoint( >> 68 aTrack->GetPosition(),aTrack->GetMaterial())); 77 } 69 } 78 70 79 //....oooOO0OOooo........oooOO0OOooo........oo 71 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... 80 RE04Trajectory::RE04Trajectory(RE04Trajectory& << 72 RE04Trajectory::RE04Trajectory(RE04Trajectory & right):G4VTrajectory() 81 { 73 { 82 fParticleName = right.fParticleName; 74 fParticleName = right.fParticleName; 83 fPDGCharge = right.fPDGCharge; 75 fPDGCharge = right.fPDGCharge; 84 fPDGEncoding = right.fPDGEncoding; 76 fPDGEncoding = right.fPDGEncoding; 85 fTrackID = right.fTrackID; 77 fTrackID = right.fTrackID; 86 fParentID = right.fParentID; 78 fParentID = right.fParentID; 87 fInitialKineticEnergy = right.fInitialKineti 79 fInitialKineticEnergy = right.fInitialKineticEnergy; 88 fInitialMomentum = right.fInitialMomentum; 80 fInitialMomentum = right.fInitialMomentum; 89 fPositionRecord = new TrajectoryPointContain 81 fPositionRecord = new TrajectoryPointContainer(); 90 82 91 for (size_t i = 0; i < right.fPositionRecord << 83 for(size_t i=0;i<right.fPositionRecord->size();i++) 92 RE04TrajectoryPoint* rightPoint = (RE04Tra << 84 { >> 85 RE04TrajectoryPoint* rightPoint >> 86 = (RE04TrajectoryPoint*)((*(right.fPositionRecord))[i]); 93 fPositionRecord->push_back(new RE04Traject 87 fPositionRecord->push_back(new RE04TrajectoryPoint(*rightPoint)); 94 } 88 } 95 } 89 } 96 90 97 //....oooOO0OOooo........oooOO0OOooo........oo 91 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... 98 RE04Trajectory::~RE04Trajectory() 92 RE04Trajectory::~RE04Trajectory() 99 { 93 { 100 if (fPositionRecord) { 94 if (fPositionRecord) { 101 // fPositionRecord->clearAndDestroy(); 95 // fPositionRecord->clearAndDestroy(); 102 size_t i; 96 size_t i; 103 for (i = 0; i < fPositionRecord->size(); i << 97 for(i=0;i<fPositionRecord->size();i++){ 104 delete (*fPositionRecord)[i]; << 98 delete (*fPositionRecord)[i]; 105 } 99 } 106 fPositionRecord->clear(); 100 fPositionRecord->clear(); 107 delete fPositionRecord; 101 delete fPositionRecord; 108 } 102 } 109 } 103 } 110 104 111 //....oooOO0OOooo........oooOO0OOooo........oo 105 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... 112 void RE04Trajectory::ShowTrajectory(std::ostre 106 void RE04Trajectory::ShowTrajectory(std::ostream& os) const 113 { 107 { 114 // Invoke the default implementation in G4VT 108 // Invoke the default implementation in G4VTrajectory... 115 G4VTrajectory::ShowTrajectory(os); 109 G4VTrajectory::ShowTrajectory(os); 116 // ... or override with your own code here. 110 // ... or override with your own code here. 117 } 111 } 118 112 119 //....oooOO0OOooo........oooOO0OOooo........oo 113 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... 120 void RE04Trajectory::DrawTrajectory() const 114 void RE04Trajectory::DrawTrajectory() const 121 { 115 { 122 // Invoke the default implementation in G4VT 116 // Invoke the default implementation in G4VTrajectory... 123 G4VTrajectory::DrawTrajectory(); 117 G4VTrajectory::DrawTrajectory(); 124 // ... or override with your own code here. 118 // ... or override with your own code here. 125 } 119 } 126 120 127 //....oooOO0OOooo........oooOO0OOooo........oo 121 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... 128 const std::map<G4String, G4AttDef>* RE04Trajec << 122 const std::map<G4String,G4AttDef>* RE04Trajectory::GetAttDefs() const 129 { 123 { 130 G4bool isNew; 124 G4bool isNew; 131 std::map<G4String, G4AttDef>* store = G4AttD << 125 std::map<G4String,G4AttDef>* store >> 126 = G4AttDefStore::GetInstance("RE04Trajectory",isNew); 132 if (isNew) { 127 if (isNew) { >> 128 133 G4String id("ID"); 129 G4String id("ID"); 134 (*store)[id] = G4AttDef(id, "Track ID", "P << 130 (*store)[id] = G4AttDef(id,"Track ID","Physics","","G4int"); 135 131 136 G4String pid("PID"); 132 G4String pid("PID"); 137 (*store)[pid] = G4AttDef(pid, "Parent ID", << 133 (*store)[pid] = G4AttDef(pid,"Parent ID","Physics","","G4int"); 138 134 139 G4String pn("PN"); 135 G4String pn("PN"); 140 (*store)[pn] = G4AttDef(pn, "Particle Name << 136 (*store)[pn] = G4AttDef(pn,"Particle Name","Physics","","G4String"); 141 137 142 G4String ch("Ch"); 138 G4String ch("Ch"); 143 (*store)[ch] = G4AttDef(ch, "Charge", "Phy << 139 (*store)[ch] = G4AttDef(ch,"Charge","Physics","e+","G4double"); 144 140 145 G4String pdg("PDG"); 141 G4String pdg("PDG"); 146 (*store)[pdg] = G4AttDef(pdg, "PDG Encodin << 142 (*store)[pdg] = G4AttDef(pdg,"PDG Encoding","Physics","","G4int"); 147 143 148 G4String ike("IKE"); 144 G4String ike("IKE"); 149 (*store)[ike] = G4AttDef(ike, "Initial kin << 145 (*store)[ike] = >> 146 G4AttDef(ike, "Initial kinetic energy", >> 147 "Physics","G4BestUnit","G4double"); 150 148 151 G4String iMom("IMom"); 149 G4String iMom("IMom"); 152 (*store)[iMom] = G4AttDef(iMom, "Initial m << 150 (*store)[iMom] = G4AttDef(iMom, "Initial momentum", >> 151 "Physics","G4BestUnit","G4ThreeVector"); 153 152 154 G4String iMag("IMag"); 153 G4String iMag("IMag"); 155 (*store)[iMag] = << 154 (*store)[iMag] = 156 G4AttDef(iMag, "Initial momentum magnitu << 155 G4AttDef(iMag, "Initial momentum magnitude", >> 156 "Physics","G4BestUnit","G4double"); 157 157 158 G4String ntp("NTP"); 158 G4String ntp("NTP"); 159 (*store)[ntp] = G4AttDef(ntp, "No. of poin << 159 (*store)[ntp] = G4AttDef(ntp,"No. of points","Physics","","G4int"); >> 160 160 } 161 } 161 return store; 162 return store; 162 } 163 } 163 164 164 //....oooOO0OOooo........oooOO0OOooo........oo 165 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... 165 std::vector<G4AttValue>* RE04Trajectory::Creat 166 std::vector<G4AttValue>* RE04Trajectory::CreateAttValues() const 166 { 167 { 167 std::vector<G4AttValue>* values = new std::v 168 std::vector<G4AttValue>* values = new std::vector<G4AttValue>; 168 169 169 values->push_back(G4AttValue("ID", G4UIcomma << 170 values->push_back >> 171 (G4AttValue("ID",G4UIcommand::ConvertToString(fTrackID),"")); 170 172 171 values->push_back(G4AttValue("PID", G4UIcomm << 173 values->push_back >> 174 (G4AttValue("PID",G4UIcommand::ConvertToString(fParentID),"")); 172 175 173 values->push_back(G4AttValue("PN", fParticle << 176 values->push_back(G4AttValue("PN",fParticleName,"")); 174 177 175 values->push_back(G4AttValue("Ch", G4UIcomma << 178 values->push_back >> 179 (G4AttValue("Ch",G4UIcommand::ConvertToString(fPDGCharge),"")); 176 180 177 values->push_back(G4AttValue("PDG", G4UIcomm << 181 values->push_back >> 182 (G4AttValue("PDG",G4UIcommand::ConvertToString(fPDGEncoding),"")); 178 183 179 values->push_back(G4AttValue("IKE", G4BestUn << 184 values->push_back >> 185 (G4AttValue("IKE",G4BestUnit(fInitialKineticEnergy,"Energy"),"")); 180 186 181 values->push_back(G4AttValue("IMom", G4BestU << 187 values->push_back >> 188 (G4AttValue("IMom",G4BestUnit(fInitialMomentum,"Energy"),"")); 182 189 183 values->push_back(G4AttValue("IMag", G4BestU << 190 values->push_back >> 191 (G4AttValue("IMag",G4BestUnit(fInitialMomentum.mag(),"Energy"),"")); 184 192 185 values->push_back(G4AttValue("NTP", G4UIcomm << 193 values->push_back >> 194 (G4AttValue("NTP",G4UIcommand::ConvertToString(GetPointEntries()),"")); 186 195 187 #ifdef G4ATTDEBUG 196 #ifdef G4ATTDEBUG 188 G4cout << G4AttCheck(values, GetAttDefs()); << 197 G4cout << G4AttCheck(values,GetAttDefs()); 189 #endif 198 #endif 190 199 191 return values; 200 return values; 192 } 201 } 193 202 194 //....oooOO0OOooo........oooOO0OOooo........oo 203 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... 195 void RE04Trajectory::AppendStep(const G4Step* 204 void RE04Trajectory::AppendStep(const G4Step* aStep) 196 { 205 { 197 fPositionRecord->push_back(new RE04Trajector << 206 fPositionRecord->push_back( new RE04TrajectoryPoint( 198 << 207 aStep->GetPostStepPoint()->GetPosition(), >> 208 aStep->GetPreStepPoint()->GetMaterial() )); 199 } 209 } 200 << 210 201 //....oooOO0OOooo........oooOO0OOooo........oo 211 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... 202 G4ParticleDefinition* RE04Trajectory::GetParti 212 G4ParticleDefinition* RE04Trajectory::GetParticleDefinition() 203 { 213 { 204 return (G4ParticleTable::GetParticleTable()- << 214 return (G4ParticleTable::GetParticleTable()->FindParticle(fParticleName)); 205 } 215 } 206 216 207 //....oooOO0OOooo........oooOO0OOooo........oo 217 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... 208 void RE04Trajectory::MergeTrajectory(G4VTrajec 218 void RE04Trajectory::MergeTrajectory(G4VTrajectory* secondTrajectory) 209 { 219 { 210 if (!secondTrajectory) return; << 220 if(!secondTrajectory) return; 211 221 212 RE04Trajectory* seco = (RE04Trajectory*)seco 222 RE04Trajectory* seco = (RE04Trajectory*)secondTrajectory; 213 G4int ent = seco->GetPointEntries(); 223 G4int ent = seco->GetPointEntries(); 214 for (G4int i = 1; i < ent; i++) // initial << 224 for(G4int i=1;i<ent;i++) // initial point of the second trajectory 215 // should n << 225 // should not be merged 216 { << 226 { 217 fPositionRecord->push_back((*(seco->fPosit 227 fPositionRecord->push_back((*(seco->fPositionRecord))[i]); 218 // fPositionRecord->push_back(seco->fPo 228 // fPositionRecord->push_back(seco->fPositionRecord->removeAt(1)); 219 } 229 } 220 delete (*seco->fPositionRecord)[0]; 230 delete (*seco->fPositionRecord)[0]; 221 seco->fPositionRecord->clear(); 231 seco->fPositionRecord->clear(); 222 } 232 } >> 233 >> 234 223 235