Geant4 Cross Reference |
1 // 1 // 2 // ******************************************* 2 // ******************************************************************** 3 // * License and Disclaimer << 3 // * DISCLAIMER * 4 // * 4 // * * 5 // * The Geant4 software is copyright of th << 5 // * The following disclaimer summarizes all the specific disclaimers * 6 // * the Geant4 Collaboration. It is provided << 6 // * of contributors to this software. The specific disclaimers,which * 7 // * conditions of the Geant4 Software License << 7 // * govern, are listed with their locations in: * 8 // * LICENSE and available at http://cern.ch/ << 8 // * http://cern.ch/geant4/license * 9 // * include a list of copyright holders. << 10 // * 9 // * * 11 // * Neither the authors of this software syst 10 // * Neither the authors of this software system, nor their employing * 12 // * institutes,nor the agencies providing fin 11 // * institutes,nor the agencies providing financial support for this * 13 // * work make any representation or warran 12 // * work make any representation or warranty, express or implied, * 14 // * regarding this software system or assum 13 // * regarding this software system or assume any liability for its * 15 // * use. Please see the license in the file << 14 // * use. * 16 // * for the full disclaimer and the limitatio << 17 // * 15 // * * 18 // * This code implementation is the result << 16 // * This code implementation is the intellectual property of the * 19 // * technical work of the GEANT4 collaboratio << 17 // * GEANT4 collaboration. * 20 // * By using, copying, modifying or distri << 18 // * By copying, distributing or modifying the Program (or any work * 21 // * any work based on the software) you ag << 19 // * based on the Program) you indicate your acceptance of this * 22 // * use in resulting scientific publicati << 20 // * statement, and all its terms. * 23 // * acceptance of all terms of the Geant4 Sof << 24 // ******************************************* 21 // ******************************************************************** 25 // 22 // 26 // 23 // >> 24 // $Id: G4RayTrajectory.cc,v 1.14 2004/06/11 14:30:51 gcosmo Exp $ >> 25 // GEANT4 tag $Name: geant4-07-00-patch-01 $ 27 // 26 // 28 // 27 // 29 // 28 // 30 29 31 /////////////////// 30 /////////////////// 32 //G4RayTrajectory.cc 31 //G4RayTrajectory.cc 33 /////////////////// 32 /////////////////// 34 33 35 #include "G4RayTrajectory.hh" 34 #include "G4RayTrajectory.hh" 36 #include "G4RayTrajectoryPoint.hh" 35 #include "G4RayTrajectoryPoint.hh" 37 #include "G4RayTracerSceneHandler.hh" << 38 #include "G4Step.hh" 36 #include "G4Step.hh" >> 37 #include "G4VPhysicalVolume.hh" 39 #include "G4VisManager.hh" 38 #include "G4VisManager.hh" 40 #include "G4VisAttributes.hh" 39 #include "G4VisAttributes.hh" 41 #include "G4Colour.hh" 40 #include "G4Colour.hh" 42 #include "G4TransportationManager.hh" 41 #include "G4TransportationManager.hh" 43 #include "G4ios.hh" 42 #include "G4ios.hh" 44 #include "G4ParallelWorldProcess.hh" << 45 43 46 G4Allocator<G4RayTrajectory>*& rayTrajectoryAl << 44 G4Allocator<G4RayTrajectory> G4RayTrajectoryAllocator; 47 { << 48 G4ThreadLocalStatic G4Allocator<G4RayTraje << 49 return _instance; << 50 } << 51 45 52 G4RayTrajectory :: G4RayTrajectory() 46 G4RayTrajectory :: G4RayTrajectory() 53 { 47 { 54 positionRecord = new std::vector<G4RayTrajec 48 positionRecord = new std::vector<G4RayTrajectoryPoint*>; 55 } 49 } 56 50 57 G4RayTrajectory :: G4RayTrajectory(G4RayTrajec 51 G4RayTrajectory :: G4RayTrajectory(G4RayTrajectory & right) 58 : G4VTrajectory() 52 : G4VTrajectory() 59 { 53 { 60 positionRecord = new std::vector<G4RayTrajec 54 positionRecord = new std::vector<G4RayTrajectoryPoint*>; 61 for(size_t i=0;i<right.positionRecord->size( 55 for(size_t i=0;i<right.positionRecord->size();i++) 62 { 56 { 63 G4RayTrajectoryPoint* rightPoint = (G4RayT 57 G4RayTrajectoryPoint* rightPoint = (G4RayTrajectoryPoint*) 64 ((*(right.positionRecord))[i]); 58 ((*(right.positionRecord))[i]); 65 positionRecord->push_back(new G4RayTraject 59 positionRecord->push_back(new G4RayTrajectoryPoint(*rightPoint)); 66 } 60 } 67 } 61 } 68 62 69 G4RayTrajectory :: ~G4RayTrajectory() 63 G4RayTrajectory :: ~G4RayTrajectory() 70 { 64 { 71 //positionRecord->clearAndDestroy(); 65 //positionRecord->clearAndDestroy(); 72 for(size_t i=0;i<positionRecord->size();i++) 66 for(size_t i=0;i<positionRecord->size();i++) 73 { delete (*positionRecord)[i]; } 67 { delete (*positionRecord)[i]; } 74 positionRecord->clear(); 68 positionRecord->clear(); 75 delete positionRecord; 69 delete positionRecord; 76 } 70 } 77 71 78 void G4RayTrajectory::AppendStep(const G4Step* << 72 void G4RayTrajectory::AppendStep(const G4Step* aStep) 79 { 73 { 80 G4RayTrajectoryPoint* trajectoryPoint = new 74 G4RayTrajectoryPoint* trajectoryPoint = new G4RayTrajectoryPoint(); 81 75 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 76 trajectoryPoint->SetStepLength(aStep->GetStepLength()); 98 77 99 // Surface normal << 78 G4Navigator* theNavigator >> 79 = G4TransportationManager::GetTransportationManager()->GetNavigatorForTracking(); 100 G4bool valid; 80 G4bool valid; 101 G4ThreeVector theLocalNormal = theNavigator- 81 G4ThreeVector theLocalNormal = theNavigator->GetLocalExitNormal(&valid); 102 if(valid) { theLocalNormal = -theLocalNormal 82 if(valid) { theLocalNormal = -theLocalNormal; } 103 G4ThreeVector theGrobalNormal 83 G4ThreeVector theGrobalNormal 104 = theNavigator->GetLocalToGlobalTransform( 84 = theNavigator->GetLocalToGlobalTransform().TransformAxis(theLocalNormal); 105 trajectoryPoint->SetSurfaceNormal(theGrobalN 85 trajectoryPoint->SetSurfaceNormal(theGrobalNormal); 106 86 >> 87 G4VPhysicalVolume* prePhys = aStep->GetPreStepPoint()->GetPhysicalVolume(); >> 88 const G4VisAttributes* preVisAtt = prePhys->GetLogicalVolume()->GetVisAttributes(); 107 G4VisManager* visManager = G4VisManager::Get 89 G4VisManager* visManager = G4VisManager::GetInstance(); 108 G4RayTracerSceneHandler* sceneHandler = << 90 if(visManager) { 109 static_cast<G4RayTracerSceneHandler*>(visMan << 91 G4VViewer* viewer = visManager->GetCurrentViewer(); 110 const auto& sceneVisAttsMap = sceneHandler-> << 92 if (viewer) { 111 << 93 preVisAtt = viewer->GetApplicableVisAttributes(preVisAtt); 112 // Make a path from the preStepPoint touchab << 94 } 113 G4StepPoint* preStepPoint = aStep -> GetPreS << 95 } 114 const G4VTouchable* preTouchable = preStepPo << 96 trajectoryPoint->SetPreStepAtt(preVisAtt); 115 G4int preDepth = preTouchable->GetHistoryDep << 97 116 G4ModelingParameters::PVPointerCopyNoPath lo << 98 const G4VPhysicalVolume* postPhys = aStep->GetPostStepPoint()->GetPhysicalVolume(); 117 for (G4int i = preDepth; i >= 0; --i) { << 99 const G4VisAttributes* postVisAtt = NULL; 118 localPrePVPointerCopyNoPath.push_back << 100 if(postPhys) { 119 (G4ModelingParameters::PVPointerCopyNo << 101 postVisAtt = postPhys->GetLogicalVolume()->GetVisAttributes(); 120 (preTouchable->GetVolume(i),preTouchable- << 102 if(visManager) { 121 } << 103 G4VViewer* viewer = visManager->GetCurrentViewer(); 122 << 104 if (viewer) { 123 // Pick up the vis atts, if any, from the sc << 105 postVisAtt = viewer->GetApplicableVisAttributes(postVisAtt); 124 auto preIterator = sceneVisAttsMap.find(loca << 106 } 125 const G4VisAttributes* preVisAtts; << 107 } 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 } 108 } 152 trajectoryPoint->SetPostStepAtt(postVisAtts) << 109 trajectoryPoint->SetPostStepAtt(postVisAtt); 153 110 154 positionRecord->push_back(trajectoryPoint); 111 positionRecord->push_back(trajectoryPoint); 155 } 112 } 156 113 157 void G4RayTrajectory::ShowTrajectory(std::ostr 114 void G4RayTrajectory::ShowTrajectory(std::ostream&) const 158 { } 115 { } 159 116 160 void G4RayTrajectory::MergeTrajectory(G4VTraje 117 void G4RayTrajectory::MergeTrajectory(G4VTrajectory* secondTrajectory) 161 { 118 { 162 if(!secondTrajectory) return; 119 if(!secondTrajectory) return; 163 120 164 G4RayTrajectory* seco = (G4RayTrajectory*)se 121 G4RayTrajectory* seco = (G4RayTrajectory*)secondTrajectory; 165 G4int ent = seco->GetPointEntries(); 122 G4int ent = seco->GetPointEntries(); 166 for(G4int i=0;i<ent;i++) 123 for(G4int i=0;i<ent;i++) 167 { positionRecord->push_back((G4RayTrajectory 124 { positionRecord->push_back((G4RayTrajectoryPoint*)seco->GetPoint(i)); } 168 seco->positionRecord->clear(); 125 seco->positionRecord->clear(); 169 } 126 } 170 127 171 128