| 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: G4TrajectoryDrawByParticleID.cc,v 1.11 2010-06-01 21:17:41 allison Exp $
>> 27 // GEANT4 tag $Name: not supported by cvs2svn $
26 // 28 //
27 // Jane Tinslay, John Allison, Joseph Perl Nov 29 // Jane Tinslay, John Allison, Joseph Perl November 2005
28 30
29 #include "G4TrajectoryDrawByParticleID.hh" 31 #include "G4TrajectoryDrawByParticleID.hh"
30 #include "G4TrajectoryDrawerUtils.hh" 32 #include "G4TrajectoryDrawerUtils.hh"
31 #include "G4VisTrajContext.hh" 33 #include "G4VisTrajContext.hh"
32 #include "G4VTrajectory.hh" 34 #include "G4VTrajectory.hh"
>> 35 #include <sstream>
33 36
34 G4TrajectoryDrawByParticleID::G4TrajectoryDraw 37 G4TrajectoryDrawByParticleID::G4TrajectoryDrawByParticleID(const G4String& name, G4VisTrajContext* context)
35 :G4VTrajectoryModel(name, context) 38 :G4VTrajectoryModel(name, context)
36 ,fDefault(G4Colour::Grey()) 39 ,fDefault(G4Colour::Grey())
37 { 40 {
38 Set("gamma", "green"); 41 Set("gamma", "green");
39 Set("e-", "red"); 42 Set("e-", "red");
40 Set("e+", "blue"); 43 Set("e+", "blue");
41 Set("pi+", "magenta"); 44 Set("pi+", "magenta");
42 Set("pi-", "magenta"); 45 Set("pi-", "magenta");
43 Set("proton", "cyan"); 46 Set("proton", "cyan");
44 Set("neutron", "yellow"); 47 Set("neutron", "yellow");
45 } 48 }
46 49
47 G4TrajectoryDrawByParticleID::~G4TrajectoryDra 50 G4TrajectoryDrawByParticleID::~G4TrajectoryDrawByParticleID() {}
48 51
49 void 52 void
50 G4TrajectoryDrawByParticleID::Draw(const G4VTr << 53 G4TrajectoryDrawByParticleID::Draw(const G4VTrajectory& object,
>> 54 const G4int&,
>> 55 const G4bool& visible) const
>> 56 {
>> 57 Draw(object, visible);
>> 58 }
>> 59
>> 60 void
>> 61 G4TrajectoryDrawByParticleID::Draw(const G4VTrajectory& traj, const G4bool& visible) const
51 { 62 {
52 G4Colour colour(fDefault); 63 G4Colour colour(fDefault);
53 G4String particle = traj.GetParticleName(); 64 G4String particle = traj.GetParticleName();
54 65
55 fMap.GetColour(particle, colour); 66 fMap.GetColour(particle, colour);
56 67
57 G4VisTrajContext myContext(GetContext()); 68 G4VisTrajContext myContext(GetContext());
58 69
59 myContext.SetLineColour(colour); 70 myContext.SetLineColour(colour);
>> 71 myContext.SetVisible(visible);
60 72
61 if (GetVerbose()) { 73 if (GetVerbose()) {
62 G4cout<<"G4TrajectoryDrawByParticleID draw 74 G4cout<<"G4TrajectoryDrawByParticleID drawer named "<<Name();
63 G4cout<<", drawing trajectory with particl 75 G4cout<<", drawing trajectory with particle type, "<<particle<<G4endl;
64 G4cout<<", with configuration:"<<G4endl; 76 G4cout<<", with configuration:"<<G4endl;
65 myContext.Print(G4cout); 77 myContext.Print(G4cout);
66 } 78 }
67 79
68 G4TrajectoryDrawerUtils::DrawLineAndPoints(t 80 G4TrajectoryDrawerUtils::DrawLineAndPoints(traj, myContext);
69 } 81 }
70 82
71 void 83 void
72 G4TrajectoryDrawByParticleID::SetDefault(const 84 G4TrajectoryDrawByParticleID::SetDefault(const G4String& colour)
73 { 85 {
74 G4Colour myColour(G4Colour::White()); 86 G4Colour myColour(G4Colour::White());
75 87
76 // Will not modify myColour if colour key do 88 // Will not modify myColour if colour key does not exist
77 if (!G4Colour::GetColour(colour, myColour)) 89 if (!G4Colour::GetColour(colour, myColour)) {
78 G4ExceptionDescription ed; << 90 std::ostringstream o;
79 ed << "G4Colour with key "<<colour<<" does << 91 o << "G4Colour with key "<<colour<<" does not exist ";
80 G4Exception 92 G4Exception
81 ("G4TrajectoryDrawByParticleID::SetDefau 93 ("G4TrajectoryDrawByParticleID::SetDefault(const G4String& colour)",
82 "modeling0124", JustWarning, ed); << 94 "NonExistentColour", JustWarning, o.str().c_str());
83 } 95 }
84 96
85 SetDefault(myColour); 97 SetDefault(myColour);
86 } 98 }
87 99
88 void 100 void
89 G4TrajectoryDrawByParticleID::SetDefault(const 101 G4TrajectoryDrawByParticleID::SetDefault(const G4Colour& colour)
90 { 102 {
91 fDefault = colour; 103 fDefault = colour;
92 } 104 }
93 105
94 void 106 void
95 G4TrajectoryDrawByParticleID::Set(const G4Stri 107 G4TrajectoryDrawByParticleID::Set(const G4String& particle, const G4String& colour)
96 { 108 {
97 fMap.Set(particle, colour); 109 fMap.Set(particle, colour);
98 } 110 }
99 111
100 void 112 void
101 G4TrajectoryDrawByParticleID::Set(const G4Stri 113 G4TrajectoryDrawByParticleID::Set(const G4String& particle, const G4Colour& colour)
102 { 114 {
103 fMap[particle] = colour; 115 fMap[particle] = colour;
104 } 116 }
105 117
106 void 118 void
107 G4TrajectoryDrawByParticleID::Print(std::ostre 119 G4TrajectoryDrawByParticleID::Print(std::ostream& ostr) const
108 { 120 {
109 ostr<<"G4TrajectoryDrawByParticleID model "< 121 ostr<<"G4TrajectoryDrawByParticleID model "<< Name() <<" colour scheme: "<<std::endl;
110 122
111 ostr<<"Default colour: "<<fDefault<<G4endl; 123 ostr<<"Default colour: "<<fDefault<<G4endl;
112 124
113 fMap.Print(ostr); 125 fMap.Print(ostr);
114 126
115 ostr<<"Default configuration:"<<G4endl; 127 ostr<<"Default configuration:"<<G4endl;
116 GetContext().Print(G4cout); 128 GetContext().Print(G4cout);
117 } 129 }
118 130