| 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 March 2006
28
29 #include "G4TrajectoryDrawByOriginVolume.hh"
30 #include "G4TrajectoryDrawerUtils.hh"
31 #include "G4VTrajectory.hh"
32 #include "G4TransportationManager.hh"
33 #include "G4VisTrajContext.hh"
34 #include "G4VTrajectoryPoint.hh"
35
36 G4TrajectoryDrawByOriginVolume::G4TrajectoryDr
37 :G4VTrajectoryModel(name, context)
38 ,fDefault(G4Colour::Grey())
39 {}
40
41 G4TrajectoryDrawByOriginVolume::~G4TrajectoryD
42
43 void
44 G4TrajectoryDrawByOriginVolume::Draw(const G4V
45 {
46 G4VTrajectoryPoint* aTrajectoryPoint = traj.
47 assert (0 != aTrajectoryPoint);
48
49 G4Colour colour(fDefault);
50
51 G4Navigator* navigator =
52 G4TransportationManager::GetTransportationMa
53
54 G4VPhysicalVolume* volume = navigator->Locat
55 (aTrajectoryPoint->GetPosition(), nullptr,fa
56
57 // Logical volumes form basis.
58 G4LogicalVolume* logicalVolume = volume->Get
59 assert (0 != logicalVolume);
60
61 G4String logicalName = logicalVolume->GetNam
62 fMap.GetColour(logicalName, colour);
63
64 // Override with physical volume colouring i
65 G4String physicalName = volume->GetName();
66 fMap.GetColour(physicalName, colour);
67
68 G4VisTrajContext myContext(GetContext());
69
70 myContext.SetLineColour(colour);
71
72 if (GetVerbose()) {
73 G4cout<<"G4TrajectoryDrawByOriginVolume dr
74 G4cout<<", drawing trajectory originating
75 G4cout<<", physical volume "<<physicalName
76 myContext.Print(G4cout);
77 }
78
79 G4TrajectoryDrawerUtils::DrawLineAndPoints(t
80 }
81
82 void
83 G4TrajectoryDrawByOriginVolume::SetDefault(con
84 {
85 G4Colour myColour;
86
87 // Will not modify default colour if colour
88 if (!G4Colour::GetColour(colour, myColour))
89 G4ExceptionDescription ed;
90 ed << "G4Colour with key "<<colour<<" does
91 G4Exception
92 ("G4TrajectoryDrawByOriginParticleID::Se
93 return;
94 }
95
96 SetDefault(myColour);
97 }
98
99 void
100 G4TrajectoryDrawByOriginVolume::SetDefault(con
101 {
102 fDefault = colour;
103 }
104
105 void
106 G4TrajectoryDrawByOriginVolume::Set(const G4St
107 {
108 fMap.Set(particle, colour);
109 }
110
111 void
112 G4TrajectoryDrawByOriginVolume::Set(const G4St
113 {
114 fMap[particle] = colour;
115 }
116
117 void
118 G4TrajectoryDrawByOriginVolume::Print(std::ost
119 {
120 ostr
121 << "G4TrajectoryDrawByOriginVolume model "<<
122 << ", colour scheme: "
123 << ", Default " << fDefault
124 << std::endl;
125
126 fMap.Print(ostr);
127
128 ostr << "Default configuration:" << std::end
129 GetContext().Print(ostr);
130 }
131