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