| 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 // John Allison February 2016, based on 27 // John Allison February 2016, based on
28 // G4TrajectoryDrawByVolume.hh Jane Tinslay M 28 // G4TrajectoryDrawByVolume.hh Jane Tinslay March 2006
29 29
30 #include "G4TrajectoryDrawByEncounteredVolume. 30 #include "G4TrajectoryDrawByEncounteredVolume.hh"
31 #include "G4TrajectoryDrawerUtils.hh" 31 #include "G4TrajectoryDrawerUtils.hh"
32 #include "G4VTrajectory.hh" 32 #include "G4VTrajectory.hh"
33 #include "G4VisTrajContext.hh" 33 #include "G4VisTrajContext.hh"
34 #include "G4VTrajectoryPoint.hh" 34 #include "G4VTrajectoryPoint.hh"
35 #include "G4AttDef.hh" <<
36 #include "G4AttValue.hh" 35 #include "G4AttValue.hh"
>> 36 #include "G4RichTrajectory.hh"
37 37
38 G4TrajectoryDrawByEncounteredVolume::G4Traject 38 G4TrajectoryDrawByEncounteredVolume::G4TrajectoryDrawByEncounteredVolume
39 (const G4String& name, G4VisTrajContext* conte 39 (const G4String& name, G4VisTrajContext* context)
40 :G4VTrajectoryModel(name, context) 40 :G4VTrajectoryModel(name, context)
41 ,fDefault(G4Colour::Grey()) 41 ,fDefault(G4Colour::Grey())
42 {} 42 {}
43 43
44 G4TrajectoryDrawByEncounteredVolume::~G4Trajec 44 G4TrajectoryDrawByEncounteredVolume::~G4TrajectoryDrawByEncounteredVolume() {}
45 45
46 void 46 void
47 G4TrajectoryDrawByEncounteredVolume::Draw(cons 47 G4TrajectoryDrawByEncounteredVolume::Draw(const G4VTrajectory& traj, const G4bool& /*visible*/) const
48 { 48 {
49 // Check the required G4Att exists << 49 try
50 const auto& aPointAttDefs = traj.GetPoint(0) << 50 {
51 if (aPointAttDefs->find("PostVPath") != aPoi << 51 const G4RichTrajectory& richTrajectory = dynamic_cast<const G4RichTrajectory&>(traj);
52 <<
53 // Must be a rich trajectory <<
54 const auto& richTrajectory = traj; <<
55 52
56 G4Colour colour(fDefault); 53 G4Colour colour(fDefault);
57 G4String soughtPVName("none"); 54 G4String soughtPVName("none");
58 55
59 for (const auto& item: fMap.GetBasicMap()) 56 for (const auto& item: fMap.GetBasicMap()) {
60 soughtPVName = item.first; 57 soughtPVName = item.first;
61 for (G4int iPoint = 0; iPoint < richTraj 58 for (G4int iPoint = 0; iPoint < richTrajectory.GetPointEntries(); iPoint++) {
62 G4VTrajectoryPoint* point = richTrajec 59 G4VTrajectoryPoint* point = richTrajectory.GetPoint(iPoint);
63 if (!point) continue; 60 if (!point) continue;
64 std::vector<G4AttValue>* attValues = p 61 std::vector<G4AttValue>* attValues = point->CreateAttValues();
65 std::vector<G4AttValue>::const_iterato 62 std::vector<G4AttValue>::const_iterator iAtt;
66 for (iAtt = attValues->begin(); iAtt ! 63 for (iAtt = attValues->begin(); iAtt != attValues->end(); ++iAtt) {
67 if (iAtt->GetName() == "PostVPath" & 64 if (iAtt->GetName() == "PostVPath" &&
68 G4StrUtil::contains(iAtt->GetVal 65 G4StrUtil::contains(iAtt->GetValue(), soughtPVName)) break;
69 } 66 }
70 if (iAtt != attValues->end()) { // Re 67 if (iAtt != attValues->end()) { // Required value found
71 fMap.GetColour(soughtPVName, colour) 68 fMap.GetColour(soughtPVName, colour);
72 break; // First found pvname determ 69 break; // First found pvname determines colour.
73 } 70 }
74 } 71 }
75 } 72 }
76 73
77 G4VisTrajContext myContext(GetContext()); 74 G4VisTrajContext myContext(GetContext());
78 75
79 myContext.SetLineColour(colour); 76 myContext.SetLineColour(colour);
80 77
81 if (GetVerbose()) { 78 if (GetVerbose()) {
82 G4cout 79 G4cout
83 << "G4TrajectoryDrawByEncounteredVolume 80 << "G4TrajectoryDrawByEncounteredVolume drawer named " << Name()
84 << ", drawing trajectory touching physic 81 << ", drawing trajectory touching physical volume " << soughtPVName
85 << ", with configuration:" << G4endl; 82 << ", with configuration:" << G4endl;
86 myContext.Print(G4cout); 83 myContext.Print(G4cout);
87 } 84 }
88 << 85
89 G4TrajectoryDrawerUtils::DrawLineAndPoints 86 G4TrajectoryDrawerUtils::DrawLineAndPoints(richTrajectory, myContext);
>> 87
>> 88 }
90 89
91 } else { << 90 catch (const std::bad_cast&)
92 << 91 {
93 G4ExceptionDescription ed; 92 G4ExceptionDescription ed;
94 ed << "Requires G4RichTrajectory - \"/vis/ 93 ed << "Requires G4RichTrajectory - \"/vis/scene/add/trajectories rich\"";
95 G4Exception 94 G4Exception
96 ("G4TrajectoryDrawByEncounteredVolume::Dra 95 ("G4TrajectoryDrawByEncounteredVolume::Draw(const G4VTrajectory& traj,...",
97 "modeling0125", 96 "modeling0125",
98 JustWarning, ed); 97 JustWarning, ed);
99 << 98 return;
100 } 99 }
101 } 100 }
102 101
103 void 102 void
104 G4TrajectoryDrawByEncounteredVolume::SetDefaul 103 G4TrajectoryDrawByEncounteredVolume::SetDefault(const G4String& colour)
105 { 104 {
106 G4Colour myColour; 105 G4Colour myColour;
107 106
108 // Will not modify default colour if colour 107 // Will not modify default colour if colour key does not exist
109 if (!G4Colour::GetColour(colour, myColour)) 108 if (!G4Colour::GetColour(colour, myColour)) {
110 G4ExceptionDescription ed; 109 G4ExceptionDescription ed;
111 ed << "G4Colour with key "<<colour<<" does 110 ed << "G4Colour with key "<<colour<<" does not exist ";
112 G4Exception 111 G4Exception
113 ("G4TrajectoryDrawByEncounteredParticleI 112 ("G4TrajectoryDrawByEncounteredParticleID::SetDefault(const G4String& colour)",
114 "modeling0123", JustWarning, ed); 113 "modeling0123", JustWarning, ed);
115 return; 114 return;
116 } 115 }
117 116
118 SetDefault(myColour); 117 SetDefault(myColour);
119 } 118 }
120 119
121 void 120 void
122 G4TrajectoryDrawByEncounteredVolume::SetDefaul 121 G4TrajectoryDrawByEncounteredVolume::SetDefault(const G4Colour& colour)
123 { 122 {
124 fDefault = colour; 123 fDefault = colour;
125 } 124 }
126 125
127 void 126 void
128 G4TrajectoryDrawByEncounteredVolume::Set(const 127 G4TrajectoryDrawByEncounteredVolume::Set(const G4String& pvname, const G4String& colour)
129 { 128 {
130 fMap.Set(pvname, colour); 129 fMap.Set(pvname, colour);
131 } 130 }
132 131
133 void 132 void
134 G4TrajectoryDrawByEncounteredVolume::Set(const 133 G4TrajectoryDrawByEncounteredVolume::Set(const G4String& pvname, const G4Colour& colour)
135 { 134 {
136 fMap[pvname] = colour; 135 fMap[pvname] = colour;
137 } 136 }
138 137
139 void 138 void
140 G4TrajectoryDrawByEncounteredVolume::Print(std 139 G4TrajectoryDrawByEncounteredVolume::Print(std::ostream& ostr) const
141 { 140 {
142 ostr 141 ostr
143 << "G4TrajectoryDrawByEncounteredVolume mode 142 << "G4TrajectoryDrawByEncounteredVolume model "<< Name()
144 << ", colour scheme: " 143 << ", colour scheme: "
145 << ", Default " << fDefault 144 << ", Default " << fDefault
146 << std::endl; 145 << std::endl;
147 146
148 fMap.Print(ostr); 147 fMap.Print(ostr);
149 148
150 ostr << "Default configuration:" << std::end 149 ostr << "Default configuration:" << std::endl;
151 GetContext().Print(ostr); 150 GetContext().Print(ostr);
152 } 151 }
153 152