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 // 26 // >> 27 // $Id: G4RayTrajectory.cc,v 1.15 2006-06-29 21:24:21 gunter Exp $ >> 28 // GEANT4 tag $Name: not supported by cvs2svn $ 27 // 29 // 28 // 30 // 29 // 31 // 30 32 31 /////////////////// 33 /////////////////// 32 //G4RayTrajectory.cc 34 //G4RayTrajectory.cc 33 /////////////////// 35 /////////////////// 34 36 35 #include "G4RayTrajectory.hh" 37 #include "G4RayTrajectory.hh" 36 #include "G4RayTrajectoryPoint.hh" 38 #include "G4RayTrajectoryPoint.hh" 37 #include "G4RayTracerSceneHandler.hh" << 38 #include "G4Step.hh" 39 #include "G4Step.hh" >> 40 #include "G4VPhysicalVolume.hh" 39 #include "G4VisManager.hh" 41 #include "G4VisManager.hh" 40 #include "G4VisAttributes.hh" 42 #include "G4VisAttributes.hh" 41 #include "G4Colour.hh" 43 #include "G4Colour.hh" 42 #include "G4TransportationManager.hh" 44 #include "G4TransportationManager.hh" 43 #include "G4ios.hh" 45 #include "G4ios.hh" 44 #include "G4ParallelWorldProcess.hh" << 45 46 46 G4Allocator<G4RayTrajectory>*& rayTrajectoryAl << 47 G4Allocator<G4RayTrajectory> G4RayTrajectoryAllocator; 47 { << 48 G4ThreadLocalStatic G4Allocator<G4RayTraje << 49 return _instance; << 50 } << 51 48 52 G4RayTrajectory :: G4RayTrajectory() 49 G4RayTrajectory :: G4RayTrajectory() 53 { 50 { 54 positionRecord = new std::vector<G4RayTrajec 51 positionRecord = new std::vector<G4RayTrajectoryPoint*>; 55 } 52 } 56 53 57 G4RayTrajectory :: G4RayTrajectory(G4RayTrajec 54 G4RayTrajectory :: G4RayTrajectory(G4RayTrajectory & right) 58 : G4VTrajectory() 55 : G4VTrajectory() 59 { 56 { 60 positionRecord = new std::vector<G4RayTrajec 57 positionRecord = new std::vector<G4RayTrajectoryPoint*>; 61 for(size_t i=0;i<right.positionRecord->size( 58 for(size_t i=0;i<right.positionRecord->size();i++) 62 { 59 { 63 G4RayTrajectoryPoint* rightPoint = (G4RayT 60 G4RayTrajectoryPoint* rightPoint = (G4RayTrajectoryPoint*) 64 ((*(right.positionRecord))[i]); 61 ((*(right.positionRecord))[i]); 65 positionRecord->push_back(new G4RayTraject 62 positionRecord->push_back(new G4RayTrajectoryPoint(*rightPoint)); 66 } 63 } 67 } 64 } 68 65 69 G4RayTrajectory :: ~G4RayTrajectory() 66 G4RayTrajectory :: ~G4RayTrajectory() 70 { 67 { 71 //positionRecord->clearAndDestroy(); 68 //positionRecord->clearAndDestroy(); 72 for(size_t i=0;i<positionRecord->size();i++) 69 for(size_t i=0;i<positionRecord->size();i++) 73 { delete (*positionRecord)[i]; } 70 { delete (*positionRecord)[i]; } 74 positionRecord->clear(); 71 positionRecord->clear(); 75 delete positionRecord; 72 delete positionRecord; 76 } 73 } 77 74 78 void G4RayTrajectory::AppendStep(const G4Step* << 75 void G4RayTrajectory::AppendStep(const G4Step* aStep) 79 { 76 { 80 G4RayTrajectoryPoint* trajectoryPoint = new 77 G4RayTrajectoryPoint* trajectoryPoint = new G4RayTrajectoryPoint(); 81 78 82 const G4Step* aStep = theStep; << 83 G4Navigator* theNavigator << 84 = G4TransportationManager::GetTransportati << 85 << 86 // Take care of parallel world(s) << 87 if(G4ParallelWorldProcess::GetHyperStep()) << 88 { << 89 aStep = G4ParallelWorldProcess::GetHyperSt << 90 G4int navID = G4ParallelWorldProcess::GetH << 91 std::vector<G4Navigator*>::iterator iNav = << 92 G4TransportationManager::GetTransportati << 93 GetAc << 94 theNavigator = iNav[navID]; << 95 } << 96 << 97 trajectoryPoint->SetStepLength(aStep->GetSte 79 trajectoryPoint->SetStepLength(aStep->GetStepLength()); 98 80 99 // Surface normal << 81 G4Navigator* theNavigator >> 82 = G4TransportationManager::GetTransportationManager()->GetNavigatorForTracking(); 100 G4bool valid; 83 G4bool valid; 101 G4ThreeVector theLocalNormal = theNavigator- 84 G4ThreeVector theLocalNormal = theNavigator->GetLocalExitNormal(&valid); 102 if(valid) { theLocalNormal = -theLocalNormal 85 if(valid) { theLocalNormal = -theLocalNormal; } 103 G4ThreeVector theGrobalNormal 86 G4ThreeVector theGrobalNormal 104 = theNavigator->GetLocalToGlobalTransform( 87 = theNavigator->GetLocalToGlobalTransform().TransformAxis(theLocalNormal); 105 trajectoryPoint->SetSurfaceNormal(theGrobalN 88 trajectoryPoint->SetSurfaceNormal(theGrobalNormal); 106 89 >> 90 G4VPhysicalVolume* prePhys = aStep->GetPreStepPoint()->GetPhysicalVolume(); >> 91 const G4VisAttributes* preVisAtt = prePhys->GetLogicalVolume()->GetVisAttributes(); 107 G4VisManager* visManager = G4VisManager::Get 92 G4VisManager* visManager = G4VisManager::GetInstance(); 108 G4RayTracerSceneHandler* sceneHandler = << 93 if(visManager) { 109 static_cast<G4RayTracerSceneHandler*>(visMan << 94 G4VViewer* viewer = visManager->GetCurrentViewer(); 110 const auto& sceneVisAttsMap = sceneHandler-> << 95 if (viewer) { 111 << 96 preVisAtt = viewer->GetApplicableVisAttributes(preVisAtt); 112 // Make a path from the preStepPoint touchab << 97 } 113 G4StepPoint* preStepPoint = aStep -> GetPreS << 98 } 114 const G4VTouchable* preTouchable = preStepPo << 99 trajectoryPoint->SetPreStepAtt(preVisAtt); 115 G4int preDepth = preTouchable->GetHistoryDep << 100 116 G4ModelingParameters::PVPointerCopyNoPath lo << 101 const G4VPhysicalVolume* postPhys = aStep->GetPostStepPoint()->GetPhysicalVolume(); 117 for (G4int i = preDepth; i >= 0; --i) { << 102 const G4VisAttributes* postVisAtt = NULL; 118 localPrePVPointerCopyNoPath.push_back << 103 if(postPhys) { 119 (G4ModelingParameters::PVPointerCopyNo << 104 postVisAtt = postPhys->GetLogicalVolume()->GetVisAttributes(); 120 (preTouchable->GetVolume(i),preTouchable- << 105 if(visManager) { 121 } << 106 G4VViewer* viewer = visManager->GetCurrentViewer(); 122 << 107 if (viewer) { 123 // Pick up the vis atts, if any, from the sc << 108 postVisAtt = viewer->GetApplicableVisAttributes(postVisAtt); 124 auto preIterator = sceneVisAttsMap.find(loca << 109 } 125 const G4VisAttributes* preVisAtts; << 110 } 126 if (preIterator != sceneVisAttsMap.end()) { << 127 preVisAtts = &preIterator->second; << 128 } else { << 129 preVisAtts = 0; << 130 } << 131 trajectoryPoint->SetPreStepAtt(preVisAtts); << 132 << 133 // Make a path from the postStepPoint toucha << 134 G4StepPoint* postStepPoint = aStep -> GetPos << 135 const G4VTouchable* postTouchable = postStep << 136 G4int postDepth = postTouchable->GetHistoryD << 137 G4ModelingParameters::PVPointerCopyNoPath lo << 138 for (G4int i = postDepth; i >= 0; --i) { << 139 localPostPVPointerCopyNoPath.push_back << 140 (G4ModelingParameters::PVPointerCopyNo << 141 (postTouchable->GetVolume(i),postTouchabl << 142 } << 143 << 144 // Pick up the vis atts, if any, from the sc << 145 auto postIterator = sceneVisAttsMap.find(loc << 146 const G4VisAttributes* postVisAtts; << 147 if (postIterator != sceneVisAttsMap.end()) { << 148 postVisAtts = &postIterator->second; << 149 } else { << 150 postVisAtts = 0; << 151 } 111 } 152 trajectoryPoint->SetPostStepAtt(postVisAtts) << 112 trajectoryPoint->SetPostStepAtt(postVisAtt); 153 113 154 positionRecord->push_back(trajectoryPoint); 114 positionRecord->push_back(trajectoryPoint); 155 } 115 } 156 116 157 void G4RayTrajectory::ShowTrajectory(std::ostr 117 void G4RayTrajectory::ShowTrajectory(std::ostream&) const 158 { } 118 { } 159 119 160 void G4RayTrajectory::MergeTrajectory(G4VTraje 120 void G4RayTrajectory::MergeTrajectory(G4VTrajectory* secondTrajectory) 161 { 121 { 162 if(!secondTrajectory) return; 122 if(!secondTrajectory) return; 163 123 164 G4RayTrajectory* seco = (G4RayTrajectory*)se 124 G4RayTrajectory* seco = (G4RayTrajectory*)secondTrajectory; 165 G4int ent = seco->GetPointEntries(); 125 G4int ent = seco->GetPointEntries(); 166 for(G4int i=0;i<ent;i++) 126 for(G4int i=0;i<ent;i++) 167 { positionRecord->push_back((G4RayTrajectory 127 { positionRecord->push_back((G4RayTrajectoryPoint*)seco->GetPoint(i)); } 168 seco->positionRecord->clear(); 128 seco->positionRecord->clear(); 169 } 129 } 170 130 171 131