| Geant4 Cross Reference |
1 //
2 // ********************************************************************
3 // * License and Disclaimer *
4 // * *
5 // * The Geant4 software is copyright of the Copyright Holders of *
6 // * the Geant4 Collaboration. It is provided under the terms and *
7 // * conditions of the Geant4 Software License, included in the file *
8 // * LICENSE and available at http://cern.ch/geant4/license . These *
9 // * include a list of copyright holders. *
10 // * *
11 // * Neither the authors of this software system, nor their employing *
12 // * institutes,nor the agencies providing financial support for this *
13 // * work make any representation or warranty, express or implied, *
14 // * regarding this software system or assume any liability for its *
15 // * use. Please see the license in the file LICENSE and URL above *
16 // * for the full disclaimer and the limitation of liability. *
17 // * *
18 // * This code implementation is the result of the scientific and *
19 // * technical work of the GEANT4 collaboration. *
20 // * By using, copying, modifying or distributing the software (or *
21 // * any work based on the software) you agree to acknowledge its *
22 // * use in resulting scientific publications, and indicate your *
23 // * acceptance of all terms of the Geant4 Software license. *
24 // ********************************************************************
25 //
26 //
27 //
28 //
29 //
30
31
32 #include "G4RTSteppingAction.hh"
33
34 #include "G4RayTracerSceneHandler.hh"
35 #include "G4SteppingManager.hh"
36 #include "G4VisManager.hh"
37 #include "G4VisAttributes.hh"
38 #include "G4Colour.hh"
39 #include "G4Track.hh"
40 #include "G4StepStatus.hh"
41 #include "G4TransportationManager.hh"
42
43 G4bool G4RTSteppingAction::ignoreTransparency = false;
44
45 G4RTSteppingAction::G4RTSteppingAction()
46 {}
47
48 void G4RTSteppingAction::UserSteppingAction(const G4Step* aStep)
49 {
50 // Stop the ray particle if it reaches to the coloured volume with its alpha = 1
51 // or coloured volume and the user request to ignore transparency
52
53 G4StepPoint* postStepPoint = aStep -> GetPostStepPoint();
54
55 G4VPhysicalVolume* postVolume_phys=postStepPoint->GetPhysicalVolume();
56 if(!postVolume_phys) return; // Reach to out of the world
57
58 G4VisManager* visManager = G4VisManager::GetInstance();
59 G4RayTracerSceneHandler* sceneHandler =
60 static_cast<G4RayTracerSceneHandler*>(visManager->GetCurrentSceneHandler());
61
62 // Make a path from the touchable
63 const G4VTouchable* postTouchable = postStepPoint->GetTouchable();
64 G4int postDepth = postTouchable->GetHistoryDepth();
65 G4ModelingParameters::PVPointerCopyNoPath localPostPVPointerCopyNoPath;
66 for (G4int i = postDepth; i >= 0; --i) {
67 localPostPVPointerCopyNoPath.push_back
68 (G4ModelingParameters::PVPointerCopyNo
69 (postTouchable->GetVolume(i),postTouchable->GetCopyNumber(i)));
70 }
71
72 // Pick up the vis atts, if any, from the scene handler
73 const auto& sceneVisAttsMap = sceneHandler->GetSceneVisAttsMap();
74 auto postIterator = sceneVisAttsMap.find(localPostPVPointerCopyNoPath);
75 const G4VisAttributes* postVisAtts;
76 if (postIterator != sceneVisAttsMap.end()) {
77 postVisAtts = &postIterator->second;
78 } else {
79 postVisAtts = 0;
80 }
81
82 if((!postVisAtts) || (!(postVisAtts->IsVisible()))) return; // Invisible volume, continue tracking
83
84 if((postVisAtts->IsForceDrawingStyle())
85 &&(postVisAtts->GetForcedDrawingStyle()==G4VisAttributes::wireframe)) return;
86 // Wire frame volume, continue tracking
87
88 G4double postAlpha=(postVisAtts->GetColour()).GetAlpha();
89
90 if(postAlpha==1.0 || ignoreTransparency) // Stop stepping
91 {
92 G4Track* currentTrack = aStep -> GetTrack();
93 currentTrack -> SetTrackStatus(fStopAndKill);
94 }
95 }
96
97 void G4RTSteppingAction::SetIgnoreTransparency(G4bool val)
98 { ignoreTransparency = val; }
99 G4bool G4RTSteppingAction::GetIgnoreTransparency()
100 { return ignoreTransparency; }
101