| 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 // Filter trajectories according to encountered volume name. Only registered
28 // volumes will pass the filter.
29 //
30 // John Allison, Feb 2016
31 //
32 #include "G4TrajectoryEncounteredVolumeFilter.hh"
33 #include "G4TransportationManager.hh"
34 #include "G4VTrajectoryPoint.hh"
35 #include "G4AttDef.hh"
36 #include "G4AttValue.hh"
37
38 G4TrajectoryEncounteredVolumeFilter::G4TrajectoryEncounteredVolumeFilter(const G4String& name)
39 :G4SmartFilter<G4VTrajectory>(name)
40 {}
41
42 G4TrajectoryEncounteredVolumeFilter::~G4TrajectoryEncounteredVolumeFilter() {}
43
44 bool
45 G4TrajectoryEncounteredVolumeFilter::Evaluate(const G4VTrajectory& traj) const
46 {
47 // Check the required G4Att exists
48 const auto& aPointAttDefs = traj.GetPoint(0)->GetAttDefs();
49 if (aPointAttDefs->find("PostVPath") != aPointAttDefs->end()) {
50
51 // Must be a rich trajectory
52 const auto& richTrajectory = traj;
53
54 for (const auto& pvname: fVolumes) {
55 for (G4int iPoint = 0; iPoint < richTrajectory.GetPointEntries(); iPoint++) {
56 G4VTrajectoryPoint* point = richTrajectory.GetPoint(iPoint);
57 if (!point) continue;
58 std::vector<G4AttValue>* attValues = point->CreateAttValues();
59 std::vector<G4AttValue>::const_iterator iAtt;
60 for (iAtt = attValues->begin(); iAtt != attValues->end(); ++iAtt) {
61 if (iAtt->GetName() == "PostVPath" &&
62 G4StrUtil::contains(iAtt->GetValue(),pvname)) break;
63 }
64 if (iAtt != attValues->end()) { // Required value found
65 return true; // First found pvname determines selection.
66 }
67 }
68 }
69 return false;
70
71 } else {
72
73 G4ExceptionDescription ed;
74 ed << "Requires G4RichTrajectory - \"/vis/scene/add/trajectories rich\"";
75 G4Exception
76 ("G4TrajectoryEncounteredVolumeFilter::Evaluate(const G4VTrajectory& traj)",
77 "modeling0126",
78 JustWarning, ed);
79 return false;
80
81 }
82 }
83
84 void
85 G4TrajectoryEncounteredVolumeFilter::Add(const G4String& volume)
86 {
87 fVolumes.push_back(volume);
88 }
89
90 void
91 G4TrajectoryEncounteredVolumeFilter::Print(std::ostream& ostr) const
92 {
93 ostr<<"Volume names registered: "<<G4endl;
94 std::vector<G4String>::const_iterator iter = fVolumes.begin();
95
96 while (iter != fVolumes.end()) {
97 ostr<<*iter<<G4endl;
98 iter++;
99 }
100 }
101
102 void
103 G4TrajectoryEncounteredVolumeFilter::Clear()
104 {
105 // Clear volume vector
106 fVolumes.clear();
107 }
108