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