| 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, John Allison, Joseph Perl Nov
28
29 #include "G4TrajectoryDrawByParticleID.hh"
30 #include "G4TrajectoryDrawerUtils.hh"
31 #include "G4VisTrajContext.hh"
32 #include "G4VTrajectory.hh"
33
34 G4TrajectoryDrawByParticleID::G4TrajectoryDraw
35 :G4VTrajectoryModel(name, context)
36 ,fDefault(G4Colour::Grey())
37 {
38 Set("gamma", "green");
39 Set("e-", "red");
40 Set("e+", "blue");
41 Set("pi+", "magenta");
42 Set("pi-", "magenta");
43 Set("proton", "cyan");
44 Set("neutron", "yellow");
45 }
46
47 G4TrajectoryDrawByParticleID::~G4TrajectoryDra
48
49 void
50 G4TrajectoryDrawByParticleID::Draw(const G4VTr
51 {
52 G4Colour colour(fDefault);
53 G4String particle = traj.GetParticleName();
54
55 fMap.GetColour(particle, colour);
56
57 G4VisTrajContext myContext(GetContext());
58
59 myContext.SetLineColour(colour);
60
61 if (GetVerbose()) {
62 G4cout<<"G4TrajectoryDrawByParticleID draw
63 G4cout<<", drawing trajectory with particl
64 G4cout<<", with configuration:"<<G4endl;
65 myContext.Print(G4cout);
66 }
67
68 G4TrajectoryDrawerUtils::DrawLineAndPoints(t
69 }
70
71 void
72 G4TrajectoryDrawByParticleID::SetDefault(const
73 {
74 G4Colour myColour(G4Colour::White());
75
76 // Will not modify myColour if colour key do
77 if (!G4Colour::GetColour(colour, myColour))
78 G4ExceptionDescription ed;
79 ed << "G4Colour with key "<<colour<<" does
80 G4Exception
81 ("G4TrajectoryDrawByParticleID::SetDefau
82 "modeling0124", JustWarning, ed);
83 }
84
85 SetDefault(myColour);
86 }
87
88 void
89 G4TrajectoryDrawByParticleID::SetDefault(const
90 {
91 fDefault = colour;
92 }
93
94 void
95 G4TrajectoryDrawByParticleID::Set(const G4Stri
96 {
97 fMap.Set(particle, colour);
98 }
99
100 void
101 G4TrajectoryDrawByParticleID::Set(const G4Stri
102 {
103 fMap[particle] = colour;
104 }
105
106 void
107 G4TrajectoryDrawByParticleID::Print(std::ostre
108 {
109 ostr<<"G4TrajectoryDrawByParticleID model "<
110
111 ostr<<"Default colour: "<<fDefault<<G4endl;
112
113 fMap.Print(ostr);
114
115 ostr<<"Default configuration:"<<G4endl;
116 GetContext().Print(G4cout);
117 }
118