| 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: G4TrajectoryDrawByOriginVolume.cc,v 1.6 2010-05-28 02:00:59 allison Exp $
>> 27 // GEANT4 tag $Name: geant4-09-04-patch-01 $
26 // 28 //
27 // Jane Tinslay March 2006 29 // Jane Tinslay March 2006
28 30
29 #include "G4TrajectoryDrawByOriginVolume.hh" 31 #include "G4TrajectoryDrawByOriginVolume.hh"
30 #include "G4TrajectoryDrawerUtils.hh" 32 #include "G4TrajectoryDrawerUtils.hh"
31 #include "G4VTrajectory.hh" 33 #include "G4VTrajectory.hh"
32 #include "G4TransportationManager.hh" 34 #include "G4TransportationManager.hh"
33 #include "G4VisTrajContext.hh" 35 #include "G4VisTrajContext.hh"
34 #include "G4VTrajectoryPoint.hh" 36 #include "G4VTrajectoryPoint.hh"
>> 37 #include <sstream>
35 38
36 G4TrajectoryDrawByOriginVolume::G4TrajectoryDr 39 G4TrajectoryDrawByOriginVolume::G4TrajectoryDrawByOriginVolume(const G4String& name, G4VisTrajContext* context)
37 :G4VTrajectoryModel(name, context) 40 :G4VTrajectoryModel(name, context)
38 ,fDefault(G4Colour::Grey()) 41 ,fDefault(G4Colour::Grey())
39 {} 42 {}
40 43
41 G4TrajectoryDrawByOriginVolume::~G4TrajectoryD 44 G4TrajectoryDrawByOriginVolume::~G4TrajectoryDrawByOriginVolume() {}
42 45
43 void 46 void
44 G4TrajectoryDrawByOriginVolume::Draw(const G4V << 47 G4TrajectoryDrawByOriginVolume::Draw(const G4VTrajectory& object,
>> 48 const G4int&,
>> 49 const G4bool& visible) const
45 { 50 {
46 G4VTrajectoryPoint* aTrajectoryPoint = traj. << 51 Draw(object, visible);
47 assert (0 != aTrajectoryPoint); << 52 }
48 53
>> 54 void
>> 55 G4TrajectoryDrawByOriginVolume::Draw(const G4VTrajectory& traj, const G4bool& visible) const
>> 56 {
49 G4Colour colour(fDefault); 57 G4Colour colour(fDefault);
50 58
51 G4Navigator* navigator = << 59 G4Navigator* navigator = G4TransportationManager::GetTransportationManager()->GetNavigatorForTracking();
52 G4TransportationManager::GetTransportationMa << 60
53 << 61 G4VTrajectoryPoint* aTrajectoryPoint = traj.GetPoint(0);
54 G4VPhysicalVolume* volume = navigator->Locat << 62 assert (0 != aTrajectoryPoint);
55 (aTrajectoryPoint->GetPosition(), nullptr,fa << 63
>> 64 G4VPhysicalVolume* volume = navigator->LocateGlobalPointAndSetup(aTrajectoryPoint->GetPosition());
56 65
57 // Logical volumes form basis. 66 // Logical volumes form basis.
58 G4LogicalVolume* logicalVolume = volume->Get 67 G4LogicalVolume* logicalVolume = volume->GetLogicalVolume();
59 assert (0 != logicalVolume); 68 assert (0 != logicalVolume);
60 69
61 G4String logicalName = logicalVolume->GetNam 70 G4String logicalName = logicalVolume->GetName();
62 fMap.GetColour(logicalName, colour); 71 fMap.GetColour(logicalName, colour);
63 72
64 // Override with physical volume colouring i 73 // Override with physical volume colouring if it exists
65 G4String physicalName = volume->GetName(); 74 G4String physicalName = volume->GetName();
66 fMap.GetColour(physicalName, colour); 75 fMap.GetColour(physicalName, colour);
67 76
68 G4VisTrajContext myContext(GetContext()); 77 G4VisTrajContext myContext(GetContext());
69 78
70 myContext.SetLineColour(colour); 79 myContext.SetLineColour(colour);
>> 80 myContext.SetVisible(visible);
71 81
72 if (GetVerbose()) { 82 if (GetVerbose()) {
73 G4cout<<"G4TrajectoryDrawByOriginVolume dr 83 G4cout<<"G4TrajectoryDrawByOriginVolume drawer named "<<Name();
74 G4cout<<", drawing trajectory originating 84 G4cout<<", drawing trajectory originating in logical volume, "<<logicalName;
75 G4cout<<", physical volume "<<physicalName << 85 G4cout<<", physical volumed "<<physicalName<<", with configuration:"<<G4endl;
76 myContext.Print(G4cout); 86 myContext.Print(G4cout);
77 } 87 }
78 88
79 G4TrajectoryDrawerUtils::DrawLineAndPoints(t 89 G4TrajectoryDrawerUtils::DrawLineAndPoints(traj, myContext);
80 } 90 }
81 91
82 void 92 void
83 G4TrajectoryDrawByOriginVolume::SetDefault(con 93 G4TrajectoryDrawByOriginVolume::SetDefault(const G4String& colour)
84 { 94 {
85 G4Colour myColour; 95 G4Colour myColour;
86 96
87 // Will not modify default colour if colour 97 // Will not modify default colour if colour key does not exist
88 if (!G4Colour::GetColour(colour, myColour)) 98 if (!G4Colour::GetColour(colour, myColour)) {
89 G4ExceptionDescription ed; << 99 std::ostringstream o;
90 ed << "G4Colour with key "<<colour<<" does << 100 o << "G4Colour with key "<<colour<<" does not exist ";
91 G4Exception 101 G4Exception
92 ("G4TrajectoryDrawByOriginParticleID::Se << 102 ("G4TrajectoryDrawByOriginParticleID::SetDefault(const G4String& colour)",
>> 103 "NonExistentColour", JustWarning, o.str().c_str());
93 return; 104 return;
94 } 105 }
95 106
96 SetDefault(myColour); 107 SetDefault(myColour);
97 } 108 }
98 109
99 void 110 void
100 G4TrajectoryDrawByOriginVolume::SetDefault(con 111 G4TrajectoryDrawByOriginVolume::SetDefault(const G4Colour& colour)
101 { 112 {
102 fDefault = colour; 113 fDefault = colour;
103 } 114 }
104 115
105 void 116 void
106 G4TrajectoryDrawByOriginVolume::Set(const G4St 117 G4TrajectoryDrawByOriginVolume::Set(const G4String& particle, const G4String& colour)
107 { 118 {
108 fMap.Set(particle, colour); 119 fMap.Set(particle, colour);
109 } 120 }
110 121
111 void 122 void
112 G4TrajectoryDrawByOriginVolume::Set(const G4St 123 G4TrajectoryDrawByOriginVolume::Set(const G4String& particle, const G4Colour& colour)
113 { 124 {
114 fMap[particle] = colour; 125 fMap[particle] = colour;
115 } 126 }
116 127
117 void 128 void
118 G4TrajectoryDrawByOriginVolume::Print(std::ost 129 G4TrajectoryDrawByOriginVolume::Print(std::ostream& ostr) const
119 { 130 {
120 ostr << 131 ostr<<"G4TrajectoryDrawByOriginVolume model "<< Name() <<" colour scheme: "<<std::endl;
121 << "G4TrajectoryDrawByOriginVolume model "<< << 132 ostr<<"Default : "<<fDefault<<G4endl;
122 << ", colour scheme: " <<
123 << ", Default " << fDefault <<
124 << std::endl; <<
125 133
126 fMap.Print(ostr); 134 fMap.Print(ostr);
127 135
128 ostr << "Default configuration:" << std::end << 136 ostr<<"Default configuration:"<<G4endl;
129 GetContext().Print(ostr); << 137 GetContext().Print(G4cout);
130 } 138 }
131 139