| 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 #include "G4TrajectoryDrawByCharge.hh"
29 #include "G4TrajectoryDrawerUtils.hh"
30 #include "G4VisTrajContext.hh"
31 #include "G4VTrajectory.hh"
32 #include <sstream>
33
34 G4TrajectoryDrawByCharge::G4TrajectoryDrawByCh
35 :G4VTrajectoryModel(name, context)
36 {
37 // Default configuration
38 fMap[Positive] = G4Colour::Blue();
39 fMap[Negative] = G4Colour::Red();
40 fMap[Neutral] = G4Colour::Green();
41 }
42
43 G4TrajectoryDrawByCharge::G4TrajectoryDrawByCh
44 const G4Colour& positive,
45 const G4Colour& negative,
46 const G4Colour& neutral)
47 :G4VTrajectoryModel(name)
48 {
49 fMap[Positive] = positive;
50 fMap[Negative] = negative;
51 fMap[Neutral] = neutral;
52 }
53
54 G4TrajectoryDrawByCharge::~G4TrajectoryDrawByC
55
56 void
57 G4TrajectoryDrawByCharge::Draw(const G4VTrajec
58 {
59 G4Colour colour;
60
61 const G4double charge = traj.GetCharge();
62
63 if(charge>0.) fMap.GetColour(Positive,
64 else if(charge<0.) fMap.GetColour(Negative,
65 else fMap.GetColour(Neutral, c
66
67 G4VisTrajContext myContext(GetContext());
68
69 myContext.SetLineColour(colour);
70
71 if (GetVerbose()) {
72 G4cout<<"G4TrajectoryDrawByCharge drawer n
73 G4cout<<", drawing trajectory with charge,
74 G4cout<<", with configuration:"<<G4endl;
75 myContext.Print(G4cout);
76 }
77
78 G4TrajectoryDrawerUtils::DrawLineAndPoints(t
79 }
80
81 void
82 G4TrajectoryDrawByCharge::Print(std::ostream&
83 {
84 ostr<<"G4TrajectoryDrawByCharge model "<< Na
85 fMap.Print(ostr);
86 ostr<<"Default configuration:"<<G4endl;
87 GetContext().Print(G4cout);
88 }
89
90 void
91 G4TrajectoryDrawByCharge::Set(const Charge& ch
92 {
93 fMap.Set(charge, colour);
94 }
95
96 void
97 G4TrajectoryDrawByCharge::Set(const Charge& ch
98 {
99 fMap[charge] = colour;
100 }
101
102 void
103 G4TrajectoryDrawByCharge::Set(const G4String&
104 {
105 Charge myCharge;
106
107 if (!ConvertToCharge(charge, myCharge)) {
108 G4ExceptionDescription ed;
109 ed << "Invalid charge "<<charge;
110 G4Exception
111 ("G4TrajectoryDrawByCharge::Set(const G4
112 return;
113 }
114
115 return Set(myCharge, colour);
116 }
117
118 void
119 G4TrajectoryDrawByCharge::Set(const G4String&
120 {
121 Charge myCharge;
122
123 if (!ConvertToCharge(charge, myCharge)) {
124 G4ExceptionDescription ed;
125 ed << "Invalid charge "<<charge;
126 G4Exception
127 ("G4TrajectoryDrawByCharge::Set(const G4
128 }
129
130 return Set(myCharge, colour);
131 }
132
133 G4bool
134 G4TrajectoryDrawByCharge::ConvertToCharge(cons
135 {
136 bool result(true);
137
138 G4int charge;
139 std::istringstream is(string.c_str());
140 is >> charge;
141
142 switch (charge) {
143 case 1:
144 myCharge = G4TrajectoryDrawByCharge::Posit
145 break;
146 case 0:
147 myCharge = G4TrajectoryDrawByCharge::Neutr
148 break;
149 case -1:
150 myCharge = G4TrajectoryDrawByCharge::Negat
151 break;
152 default:
153 result = false;
154 }
155
156 return result;
157 }
158