Geant4  10.03.p03
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Groups Pages
G4TrajectoryDrawByEncounteredVolume.cc
Go to the documentation of this file.
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 // $Id: G4TrajectoryDrawByEncounteredVolume.cc 95107 2016-01-26 12:41:11Z allison $
27 //
28 // John Allison February 2016, based on
29 // G4TrajectoryDrawByVolume.hh Jane Tinslay March 2006
30 
33 #include "G4VTrajectory.hh"
34 #include "G4VisTrajContext.hh"
35 #include "G4VTrajectoryPoint.hh"
36 #include "G4AttValue.hh"
37 #include "G4RichTrajectory.hh"
38 
41  :G4VTrajectoryModel(name, context)
42  ,fDefault(G4Colour::Grey())
43 {}
44 
46 
47 void
49 {
50  try
51  {
52  const G4RichTrajectory& richTrajectory = dynamic_cast<const G4RichTrajectory&>(traj);
53 
54  G4Colour colour(fDefault);
55  G4String soughtPVName("none");
56 
57  for (const auto& item: fMap.GetBasicMap()) {
58  soughtPVName = item.first;
59  for (G4int iPoint = 0; iPoint < richTrajectory.GetPointEntries(); iPoint++) {
60  G4VTrajectoryPoint* point = richTrajectory.GetPoint(iPoint);
61  if (!point) continue;
62  std::vector<G4AttValue>* attValues = point->CreateAttValues();
63  std::vector<G4AttValue>::const_iterator iAtt;
64  for (iAtt = attValues->begin(); iAtt != attValues->end(); ++iAtt) {
65  if (iAtt->GetName() == "PostVPath" &&
66  iAtt->GetValue().contains(soughtPVName)) break;
67  }
68  if (iAtt != attValues->end()) { // Required value found
69  fMap.GetColour(soughtPVName, colour);
70  break; // First found pvname determines colour.
71  }
72  }
73  }
74 
75  G4VisTrajContext myContext(GetContext());
76 
77  myContext.SetLineColour(colour);
78  myContext.SetVisible(visible);
79 
80  if (GetVerbose()) {
81  G4cout
82  << "G4TrajectoryDrawByEncounteredVolume drawer named " << Name()
83  << ", drawing trajectory touching physical volume " << soughtPVName
84  << ", with configuration:" << G4endl;
85  myContext.Print(G4cout);
86  }
87 
88  G4TrajectoryDrawerUtils::DrawLineAndPoints(richTrajectory, myContext);
89 
90  }
91 
92  catch (std::bad_cast)
93  {
95  ed << "Requires G4RichTrajectory - \"/vis/scene/add/trajectories rich\"";
97  ("G4TrajectoryDrawByEncounteredVolume::Draw(const G4VTrajectory& traj,...",
98  "modeling0125",
99  JustWarning, ed);
100  return;
101  }
102 }
103 
104 void
106 {
107  G4Colour myColour;
108 
109  // Will not modify default colour if colour key does not exist
110  if (!G4Colour::GetColour(colour, myColour)) {
112  ed << "G4Colour with key "<<colour<<" does not exist ";
114  ("G4TrajectoryDrawByEncounteredParticleID::SetDefault(const G4String& colour)",
115  "modeling0123", JustWarning, ed);
116  return;
117  }
118 
119  SetDefault(myColour);
120 }
121 
122 void
124 {
125  fDefault = colour;
126 }
127 
128 void
130 {
131  fMap.Set(pvname, colour);
132 }
133 
134 void
136 {
137  fMap[pvname] = colour;
138 }
139 
140 void
142 {
143  ostr
144  << "G4TrajectoryDrawByEncounteredVolume model "<< Name()
145  << ", colour scheme: "
146  << ", Default " << fDefault
147  << std::endl;
148 
149  fMap.Print(ostr);
150 
151  ostr << "Default configuration:" << std::endl;
152  GetContext().Print(ostr);
153 }
const XML_Char * name
Definition: expat.h:151
const std::map< T, G4Colour > & GetBasicMap() const
std::ostringstream G4ExceptionDescription
Definition: globals.hh:76
void SetLineColour(const G4Colour &colour)
virtual void Print(std::ostream &ostr) const
G4int first(char) const
void Set(const G4String &pvname, const G4String &colour)
static G4bool GetColour(const G4String &key, G4Colour &result)
Definition: G4Colour.cc:126
void Set(const T &, const G4Colour &)
int GetPointEntries() const
int G4int
Definition: G4Types.hh:78
G4String Name() const
virtual std::vector< G4AttValue > * CreateAttValues() const
const G4VisTrajContext & GetContext() const
G4GLOB_DLL std::ostream G4cout
bool GetColour(const T &, G4Colour &) const
void DrawLineAndPoints(const G4VTrajectory &traj, const G4VisTrajContext &)
void Print(std::ostream &ostr) const
static G4Colour Grey()
Definition: G4Colour.hh:145
bool G4bool
Definition: G4Types.hh:79
void SetVisible(const G4bool &visible)
virtual void Draw(const G4VTrajectory &trajectory, const G4bool &visible=true) const
void G4Exception(const char *originOfException, const char *exceptionCode, G4ExceptionSeverity severity, const char *comments)
Definition: G4Exception.cc:41
G4TrajectoryDrawByEncounteredVolume(const G4String &name="Unspecified", G4VisTrajContext *context=0)
const XML_Char * context
Definition: expat.h:434
#define G4endl
Definition: G4ios.hh:61
G4bool GetVerbose() const
void Print(std::ostream &ostr) const
G4VTrajectoryPoint * GetPoint(G4int i) const