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 // $Id: G4TrajectoryDrawByAttribute.cc,v 1.6 2010-12-11 16:41:11 allison Exp $ >> 27 // GEANT4 tag $Name: not supported by cvs2svn $ 26 // 28 // 27 // Jane Tinslay August 2006 29 // Jane Tinslay August 2006 28 // 30 // 29 #include "G4TrajectoryDrawByAttribute.hh" 31 #include "G4TrajectoryDrawByAttribute.hh" 30 #include "G4AttDef.hh" 32 #include "G4AttDef.hh" 31 #include "G4AttFilterUtils.hh" 33 #include "G4AttFilterUtils.hh" 32 #include "G4AttUtils.hh" 34 #include "G4AttUtils.hh" 33 #include "G4AttValue.hh" 35 #include "G4AttValue.hh" 34 #include "G4TrajectoryDrawerUtils.hh" 36 #include "G4TrajectoryDrawerUtils.hh" 35 #include "G4VAttValueFilter.hh" 37 #include "G4VAttValueFilter.hh" 36 #include "G4VisTrajContext.hh" 38 #include "G4VisTrajContext.hh" 37 #include "G4VTrajectory.hh" 39 #include "G4VTrajectory.hh" 38 #include <assert.h> 40 #include <assert.h> 39 41 40 G4TrajectoryDrawByAttribute::G4TrajectoryDrawB 42 G4TrajectoryDrawByAttribute::G4TrajectoryDrawByAttribute(const G4String& name, G4VisTrajContext* context) 41 :G4VTrajectoryModel(name, context) 43 :G4VTrajectoryModel(name, context) 42 ,fAttName("") 44 ,fAttName("") 43 ,fFirst(true) 45 ,fFirst(true) 44 ,fWarnedMissingAttribute(false) 46 ,fWarnedMissingAttribute(false) 45 ,filter(0) 47 ,filter(0) 46 {} 48 {} 47 49 48 G4TrajectoryDrawByAttribute::~G4TrajectoryDraw 50 G4TrajectoryDrawByAttribute::~G4TrajectoryDrawByAttribute() 49 { 51 { 50 ContextMap::iterator iter = fContextMap.begi 52 ContextMap::iterator iter = fContextMap.begin(); 51 53 52 while (iter != fContextMap.end()) { 54 while (iter != fContextMap.end()) { 53 delete iter->second; 55 delete iter->second; 54 iter++; 56 iter++; 55 } 57 } 56 58 57 delete filter; 59 delete filter; 58 } 60 } 59 61 60 void 62 void >> 63 G4TrajectoryDrawByAttribute::Draw(const G4VTrajectory& object, >> 64 const G4int&, >> 65 const G4bool& visible) const >> 66 { >> 67 Draw(object, visible); >> 68 } >> 69 >> 70 void 61 G4TrajectoryDrawByAttribute::Draw(const G4VTra 71 G4TrajectoryDrawByAttribute::Draw(const G4VTrajectory& object, 62 const G4bool& /*visible*/) const << 72 const G4bool& visible) const 63 { 73 { 64 // Return if attribute name has not been set 74 // Return if attribute name has not been set. Just print one warning 65 if (fAttName.empty()) { << 75 if (fAttName.isNull()) { 66 76 67 if (!fWarnedMissingAttribute) { 77 if (!fWarnedMissingAttribute) { 68 G4ExceptionDescription ed; 78 G4ExceptionDescription ed; 69 ed<<"Null attribute name"; 79 ed<<"Null attribute name"; 70 G4Exception("G4TrajectoryDrawByAttribute 80 G4Exception("G4TrajectoryDrawByAttribute::Draw", 71 "modeling0116", 81 "modeling0116", 72 JustWarning, ed); 82 JustWarning, ed); 73 fWarnedMissingAttribute = true; 83 fWarnedMissingAttribute = true; 74 } 84 } 75 85 76 return; 86 return; 77 } 87 } 78 88 79 // Basically cache data loaded filter for ef 89 // Basically cache data loaded filter for efficiency 80 if (fFirst) { 90 if (fFirst) { 81 91 82 fFirst = false; 92 fFirst = false; 83 93 84 // Get attribute definition 94 // Get attribute definition 85 G4AttDef attDef; 95 G4AttDef attDef; 86 96 87 // Expect definition to exist 97 // Expect definition to exist 88 if (!G4AttUtils::ExtractAttDef(object, fAt 98 if (!G4AttUtils::ExtractAttDef(object, fAttName, attDef)) { 89 static G4bool warnedUnableToExtract = fa 99 static G4bool warnedUnableToExtract = false; 90 if (!warnedUnableToExtract) { 100 if (!warnedUnableToExtract) { 91 G4ExceptionDescription ed; 101 G4ExceptionDescription ed; 92 ed <<"Unable to extract attribute definition << 102 ed <<"Unable to extract attribute definition named "<<fAttName; 93 << "Available attributes:\n" << 94 << *object.GetAttDefs(); << 95 G4Exception 103 G4Exception 96 ("G4TrajectoryDrawByAttribute::Draw", 104 ("G4TrajectoryDrawByAttribute::Draw", 97 "modeling0117", JustWarning, ed, ". Inval << 105 "modeling0117", JustWarning, ed, "Invalid attribute name"); >> 106 G4cout << "Available attributes:\n" >> 107 << object.GetAttDefs(); 98 warnedUnableToExtract = true; 108 warnedUnableToExtract = true; 99 } 109 } 100 return; 110 return; 101 } 111 } 102 112 103 // Get new G4AttValue filter 113 // Get new G4AttValue filter 104 filter = G4AttFilterUtils::GetNewFilter(at 114 filter = G4AttFilterUtils::GetNewFilter(attDef); 105 assert (0 != filter); 115 assert (0 != filter); 106 116 107 // Load both interval and single valued da 117 // Load both interval and single valued data. Single valued data should 108 // override interval data. 118 // override interval data. 109 ContextMap::const_iterator iter = fContext 119 ContextMap::const_iterator iter = fContextMap.begin(); 110 120 111 while (iter != fContextMap.end()) { 121 while (iter != fContextMap.end()) { 112 if (iter->first.second == G4TrajectoryDr 122 if (iter->first.second == G4TrajectoryDrawByAttribute::Interval) { 113 filter->LoadIntervalElement(iter->first.firs 123 filter->LoadIntervalElement(iter->first.first); 114 } 124 } 115 else if (iter->first.second == G4Traject 125 else if (iter->first.second == G4TrajectoryDrawByAttribute::SingleValue) { 116 filter->LoadSingleValueElement(iter->first.f 126 filter->LoadSingleValueElement(iter->first.first); 117 } 127 } 118 iter++; 128 iter++; 119 } 129 } 120 } 130 } 121 131 122 // Get attribute value 132 // Get attribute value 123 G4AttValue attVal; 133 G4AttValue attVal; 124 134 125 // Expect value to exist 135 // Expect value to exist 126 if (!G4AttUtils::ExtractAttValue(object, fAt 136 if (!G4AttUtils::ExtractAttValue(object, fAttName, attVal)) { 127 static G4bool warnedUnableToExtract = fals 137 static G4bool warnedUnableToExtract = false; 128 if (!warnedUnableToExtract) { 138 if (!warnedUnableToExtract) { 129 G4ExceptionDescription ed; 139 G4ExceptionDescription ed; 130 ed <<"Unable to extract attribute defini << 140 ed <<"Unable to extract attribute value named "<<fAttName; 131 << "Available attributes:\n" << 132 << *object.GetAttDefs(); << 133 G4Exception 141 G4Exception 134 ("G4TrajectoryDrawByAttribute::Draw", 142 ("G4TrajectoryDrawByAttribute::Draw", 135 "modeling0118", JustWarning, ed, ". Invalid << 143 "modeling0118", JustWarning, ed, "Invalid attribute name"); >> 144 G4cout << "Available attributes:\n" >> 145 << object.GetAttDefs(); 136 warnedUnableToExtract = true; 146 warnedUnableToExtract = true; 137 } 147 } 138 return; 148 return; 139 } 149 } 140 150 141 G4VisTrajContext myContext(GetContext()); 151 G4VisTrajContext myContext(GetContext()); 142 G4String key; 152 G4String key; 143 153 144 // If attribute value passes filter, get cor 154 // If attribute value passes filter, get corresponding interval/single value 145 // key loaded into G4AttValue filter. 155 // key loaded into G4AttValue filter. 146 if (filter->GetValidElement(attVal, key)) { 156 if (filter->GetValidElement(attVal, key)) { 147 157 148 // Extract context corresponding to valid 158 // Extract context corresponding to valid key. 149 // Single value match should have override 159 // Single value match should have overriden interval match. 150 ContextMap::const_iterator iter = fContext 160 ContextMap::const_iterator iter = fContextMap.begin(); 151 161 152 G4bool gotContext(false); 162 G4bool gotContext(false); 153 163 154 while (!gotContext && (iter != fContextMap 164 while (!gotContext && (iter != fContextMap.end())) { 155 if (iter->first.first == key) { 165 if (iter->first.first == key) { 156 myContext = *(iter->second); 166 myContext = *(iter->second); 157 gotContext = true; 167 gotContext = true; 158 } 168 } 159 iter++; 169 iter++; 160 } 170 } 161 171 162 assert (gotContext); 172 assert (gotContext); 163 } 173 } >> 174 >> 175 myContext.SetVisible(visible); 164 176 165 if (GetVerbose()) { 177 if (GetVerbose()) { 166 G4cout<<"G4TrajectoryDrawByAttribute drawe 178 G4cout<<"G4TrajectoryDrawByAttribute drawer named "<<Name(); 167 G4cout<<", drawing style selected accordin 179 G4cout<<", drawing style selected according to value of attribute "<<fAttName; 168 G4cout<<" : "<<attVal.GetValue()<<". Sele 180 G4cout<<" : "<<attVal.GetValue()<<". Selected context:"<<G4endl; 169 myContext.Print(G4cout); 181 myContext.Print(G4cout); 170 } 182 } 171 183 172 // Draw the trajectory 184 // Draw the trajectory 173 G4TrajectoryDrawerUtils::DrawLineAndPoints(o 185 G4TrajectoryDrawerUtils::DrawLineAndPoints(object, myContext); 174 } 186 } 175 187 176 void 188 void 177 G4TrajectoryDrawByAttribute::Print(std::ostrea 189 G4TrajectoryDrawByAttribute::Print(std::ostream& ostr) const 178 { 190 { 179 ostr<<"G4TrajectoryDrawByAttribute, dumping 191 ostr<<"G4TrajectoryDrawByAttribute, dumping configuration for model named "<< Name() <<":"<<std::endl;; 180 192 181 ostr<<"Default configuration:"<<G4endl; 193 ostr<<"Default configuration:"<<G4endl; 182 GetContext().Print(ostr); 194 GetContext().Print(ostr); 183 195 184 ostr<<"\nAttribute name "<<fAttName<<std::en 196 ostr<<"\nAttribute name "<<fAttName<<std::endl; 185 ostr<<"\nKey<->Context map dump:"<<std::endl 197 ostr<<"\nKey<->Context map dump:"<<std::endl; 186 198 187 ContextMap::const_iterator iter = fContextMa 199 ContextMap::const_iterator iter = fContextMap.begin(); 188 200 189 while (iter != fContextMap.end()) { 201 while (iter != fContextMap.end()) { 190 ostr<<"Context for key "<<iter->first.firs 202 ostr<<"Context for key "<<iter->first.first<<":"<<std::endl; 191 iter->second->Print(ostr); 203 iter->second->Print(ostr); 192 204 193 iter++; 205 iter++; 194 } 206 } 195 } 207 } 196 208 197 void 209 void 198 G4TrajectoryDrawByAttribute::Set(const G4Strin 210 G4TrajectoryDrawByAttribute::Set(const G4String& name) 199 { 211 { 200 fAttName = name; 212 fAttName = name; 201 } 213 } 202 214 203 void 215 void 204 G4TrajectoryDrawByAttribute::AddIntervalContex 216 G4TrajectoryDrawByAttribute::AddIntervalContext(const G4String& name, G4VisTrajContext* context) 205 { 217 { 206 // Takes ownership of context 218 // Takes ownership of context 207 std::pair<G4String, Config> myPair(name, G4T 219 std::pair<G4String, Config> myPair(name, G4TrajectoryDrawByAttribute::Interval); 208 220 209 ContextMap::iterator iter = fContextMap.find 221 ContextMap::iterator iter = fContextMap.find(myPair); 210 222 211 if (iter != fContextMap.end()) { 223 if (iter != fContextMap.end()) { 212 G4ExceptionDescription ed; 224 G4ExceptionDescription ed; 213 ed <<"Interval "<< name <<" already exists 225 ed <<"Interval "<< name <<" already exists"; 214 G4Exception 226 G4Exception 215 ("G4TrajectoryDrawByAttribute::AddInterv 227 ("G4TrajectoryDrawByAttribute::AddIntervalContext", 216 "modeling0119", FatalErrorInArgument, e << 228 "modeling0119", FatalErrorInArgument, ed, "Invalid interval"); 217 } 229 } 218 230 219 fContextMap[myPair] = context; 231 fContextMap[myPair] = context; 220 } 232 } 221 233 222 void 234 void 223 G4TrajectoryDrawByAttribute::AddValueContext(c 235 G4TrajectoryDrawByAttribute::AddValueContext(const G4String& name, G4VisTrajContext* context) 224 { 236 { 225 // Takes ownership of context 237 // Takes ownership of context 226 std::pair<G4String, Config> myPair(name, G4T 238 std::pair<G4String, Config> myPair(name, G4TrajectoryDrawByAttribute::SingleValue); 227 239 228 ContextMap::iterator iter = fContextMap.find 240 ContextMap::iterator iter = fContextMap.find(myPair); 229 241 230 if (iter != fContextMap.end()) { 242 if (iter != fContextMap.end()) { 231 G4ExceptionDescription ed; 243 G4ExceptionDescription ed; 232 ed <<"Single value "<< name <<" already ex 244 ed <<"Single value "<< name <<" already exists"; 233 G4Exception 245 G4Exception 234 ("G4TrajectoryDrawByAttribute::AddSingle 246 ("G4TrajectoryDrawByAttribute::AddSingleValueContext", 235 "modeling0120", FatalErrorInArgument, e << 247 "modeling0120", FatalErrorInArgument, ed, "Invalid value"); 236 } 248 } 237 249 238 fContextMap[myPair] = context; 250 fContextMap[myPair] = context; 239 } 251 } 240 252