| Geant4 Cross Reference |
1 // 1
2 // *******************************************
3 // * License and Disclaimer
4 // *
5 // * The Geant4 software is copyright of th
6 // * the Geant4 Collaboration. It is provided
7 // * conditions of the Geant4 Software License
8 // * LICENSE and available at http://cern.ch/
9 // * include a list of copyright holders.
10 // *
11 // * Neither the authors of this software syst
12 // * institutes,nor the agencies providing fin
13 // * work make any representation or warran
14 // * regarding this software system or assum
15 // * use. Please see the license in the file
16 // * for the full disclaimer and the limitatio
17 // *
18 // * This code implementation is the result
19 // * technical work of the GEANT4 collaboratio
20 // * By using, copying, modifying or distri
21 // * any work based on the software) you ag
22 // * use in resulting scientific publicati
23 // * acceptance of all terms of the Geant4 Sof
24 // *******************************************
25 //
26 //
27 // Jane Tinslay August 2006
28 //
29 #include "G4TrajectoryDrawByAttribute.hh"
30 #include "G4AttDef.hh"
31 #include "G4AttFilterUtils.hh"
32 #include "G4AttUtils.hh"
33 #include "G4AttValue.hh"
34 #include "G4TrajectoryDrawerUtils.hh"
35 #include "G4VAttValueFilter.hh"
36 #include "G4VisTrajContext.hh"
37 #include "G4VTrajectory.hh"
38 #include <assert.h>
39
40 G4TrajectoryDrawByAttribute::G4TrajectoryDrawB
41 :G4VTrajectoryModel(name, context)
42 ,fAttName("")
43 ,fFirst(true)
44 ,fWarnedMissingAttribute(false)
45 ,filter(0)
46 {}
47
48 G4TrajectoryDrawByAttribute::~G4TrajectoryDraw
49 {
50 ContextMap::iterator iter = fContextMap.begi
51
52 while (iter != fContextMap.end()) {
53 delete iter->second;
54 iter++;
55 }
56
57 delete filter;
58 }
59
60 void
61 G4TrajectoryDrawByAttribute::Draw(const G4VTra
62 const G4bool& /*visible*/) const
63 {
64 // Return if attribute name has not been set
65 if (fAttName.empty()) {
66
67 if (!fWarnedMissingAttribute) {
68 G4ExceptionDescription ed;
69 ed<<"Null attribute name";
70 G4Exception("G4TrajectoryDrawByAttribute
71 "modeling0116",
72 JustWarning, ed);
73 fWarnedMissingAttribute = true;
74 }
75
76 return;
77 }
78
79 // Basically cache data loaded filter for ef
80 if (fFirst) {
81
82 fFirst = false;
83
84 // Get attribute definition
85 G4AttDef attDef;
86
87 // Expect definition to exist
88 if (!G4AttUtils::ExtractAttDef(object, fAt
89 static G4bool warnedUnableToExtract = fa
90 if (!warnedUnableToExtract) {
91 G4ExceptionDescription ed;
92 ed <<"Unable to extract attribute definition
93 << "Available attributes:\n"
94 << *object.GetAttDefs();
95 G4Exception
96 ("G4TrajectoryDrawByAttribute::Draw",
97 "modeling0117", JustWarning, ed, ". Inval
98 warnedUnableToExtract = true;
99 }
100 return;
101 }
102
103 // Get new G4AttValue filter
104 filter = G4AttFilterUtils::GetNewFilter(at
105 assert (0 != filter);
106
107 // Load both interval and single valued da
108 // override interval data.
109 ContextMap::const_iterator iter = fContext
110
111 while (iter != fContextMap.end()) {
112 if (iter->first.second == G4TrajectoryDr
113 filter->LoadIntervalElement(iter->first.firs
114 }
115 else if (iter->first.second == G4Traject
116 filter->LoadSingleValueElement(iter->first.f
117 }
118 iter++;
119 }
120 }
121
122 // Get attribute value
123 G4AttValue attVal;
124
125 // Expect value to exist
126 if (!G4AttUtils::ExtractAttValue(object, fAt
127 static G4bool warnedUnableToExtract = fals
128 if (!warnedUnableToExtract) {
129 G4ExceptionDescription ed;
130 ed <<"Unable to extract attribute defini
131 << "Available attributes:\n"
132 << *object.GetAttDefs();
133 G4Exception
134 ("G4TrajectoryDrawByAttribute::Draw",
135 "modeling0118", JustWarning, ed, ". Invalid
136 warnedUnableToExtract = true;
137 }
138 return;
139 }
140
141 G4VisTrajContext myContext(GetContext());
142 G4String key;
143
144 // If attribute value passes filter, get cor
145 // key loaded into G4AttValue filter.
146 if (filter->GetValidElement(attVal, key)) {
147
148 // Extract context corresponding to valid
149 // Single value match should have override
150 ContextMap::const_iterator iter = fContext
151
152 G4bool gotContext(false);
153
154 while (!gotContext && (iter != fContextMap
155 if (iter->first.first == key) {
156 myContext = *(iter->second);
157 gotContext = true;
158 }
159 iter++;
160 }
161
162 assert (gotContext);
163 }
164
165 if (GetVerbose()) {
166 G4cout<<"G4TrajectoryDrawByAttribute drawe
167 G4cout<<", drawing style selected accordin
168 G4cout<<" : "<<attVal.GetValue()<<". Sele
169 myContext.Print(G4cout);
170 }
171
172 // Draw the trajectory
173 G4TrajectoryDrawerUtils::DrawLineAndPoints(o
174 }
175
176 void
177 G4TrajectoryDrawByAttribute::Print(std::ostrea
178 {
179 ostr<<"G4TrajectoryDrawByAttribute, dumping
180
181 ostr<<"Default configuration:"<<G4endl;
182 GetContext().Print(ostr);
183
184 ostr<<"\nAttribute name "<<fAttName<<std::en
185 ostr<<"\nKey<->Context map dump:"<<std::endl
186
187 ContextMap::const_iterator iter = fContextMa
188
189 while (iter != fContextMap.end()) {
190 ostr<<"Context for key "<<iter->first.firs
191 iter->second->Print(ostr);
192
193 iter++;
194 }
195 }
196
197 void
198 G4TrajectoryDrawByAttribute::Set(const G4Strin
199 {
200 fAttName = name;
201 }
202
203 void
204 G4TrajectoryDrawByAttribute::AddIntervalContex
205 {
206 // Takes ownership of context
207 std::pair<G4String, Config> myPair(name, G4T
208
209 ContextMap::iterator iter = fContextMap.find
210
211 if (iter != fContextMap.end()) {
212 G4ExceptionDescription ed;
213 ed <<"Interval "<< name <<" already exists
214 G4Exception
215 ("G4TrajectoryDrawByAttribute::AddInterv
216 "modeling0119", FatalErrorInArgument, e
217 }
218
219 fContextMap[myPair] = context;
220 }
221
222 void
223 G4TrajectoryDrawByAttribute::AddValueContext(c
224 {
225 // Takes ownership of context
226 std::pair<G4String, Config> myPair(name, G4T
227
228 ContextMap::iterator iter = fContextMap.find
229
230 if (iter != fContextMap.end()) {
231 G4ExceptionDescription ed;
232 ed <<"Single value "<< name <<" already ex
233 G4Exception
234 ("G4TrajectoryDrawByAttribute::AddSingle
235 "modeling0120", FatalErrorInArgument, e
236 }
237
238 fContextMap[myPair] = context;
239 }
240