Geant4  v4-10.4-release
 모두 클래스 네임스페이스들 파일들 함수 변수 타입정의 열거형 타입 열거형 멤버 Friends 매크로 그룹들 페이지들
G4TrajectoryDrawByParticleID.cc
이 파일의 문서화 페이지로 가기
1 //
2 // ********************************************************************
3 // * License and Disclaimer *
4 // * *
5 // * The Geant4 software is copyright of the Copyright Holders of *
6 // * the Geant4 Collaboration. It is provided under the terms and *
7 // * conditions of the Geant4 Software License, included in the file *
8 // * LICENSE and available at http://cern.ch/geant4/license . These *
9 // * include a list of copyright holders. *
10 // * *
11 // * Neither the authors of this software system, nor their employing *
12 // * institutes,nor the agencies providing financial support for this *
13 // * work make any representation or warranty, express or implied, *
14 // * regarding this software system or assume any liability for its *
15 // * use. Please see the license in the file LICENSE and URL above *
16 // * for the full disclaimer and the limitation of liability. *
17 // * *
18 // * This code implementation is the result of the scientific and *
19 // * technical work of the GEANT4 collaboration. *
20 // * By using, copying, modifying or distributing the software (or *
21 // * any work based on the software) you agree to acknowledge its *
22 // * use in resulting scientific publications, and indicate your *
23 // * acceptance of all terms of the Geant4 Software license. *
24 // ********************************************************************
25 //
26 // $Id: G4TrajectoryDrawByParticleID.cc 66373 2012-12-18 09:41:34Z gcosmo $
27 //
28 // Jane Tinslay, John Allison, Joseph Perl November 2005
29 
32 #include "G4VisTrajContext.hh"
33 #include "G4VTrajectory.hh"
34 
36  :G4VTrajectoryModel(name, context)
37  ,fDefault(G4Colour::Grey())
38 {
39  Set("gamma", "green");
40  Set("e-", "red");
41  Set("e+", "blue");
42  Set("pi+", "magenta");
43  Set("pi-", "magenta");
44  Set("proton", "cyan");
45  Set("neutron", "yellow");
46 }
47 
49 
50 void
51 G4TrajectoryDrawByParticleID::Draw(const G4VTrajectory& traj, const G4bool& visible) const
52 {
53  G4Colour colour(fDefault);
54  G4String particle = traj.GetParticleName();
55 
56  fMap.GetColour(particle, colour);
57 
58  G4VisTrajContext myContext(GetContext());
59 
60  myContext.SetLineColour(colour);
61  myContext.SetVisible(visible);
62 
63  if (GetVerbose()) {
64  G4cout<<"G4TrajectoryDrawByParticleID drawer named "<<Name();
65  G4cout<<", drawing trajectory with particle type, "<<particle<<G4endl;
66  G4cout<<", with configuration:"<<G4endl;
67  myContext.Print(G4cout);
68  }
69 
71 }
72 
73 void
75 {
76  G4Colour myColour(G4Colour::White());
77 
78  // Will not modify myColour if colour key does not exist
79  if (!G4Colour::GetColour(colour, myColour)) {
81  ed << "G4Colour with key "<<colour<<" does not exist ";
83  ("G4TrajectoryDrawByParticleID::SetDefault(const G4String& colour)",
84  "modeling0124", JustWarning, ed);
85  }
86 
87  SetDefault(myColour);
88 }
89 
90 void
92 {
93  fDefault = colour;
94 }
95 
96 void
97 G4TrajectoryDrawByParticleID::Set(const G4String& particle, const G4String& colour)
98 {
99  fMap.Set(particle, colour);
100 }
101 
102 void
103 G4TrajectoryDrawByParticleID::Set(const G4String& particle, const G4Colour& colour)
104 {
105  fMap[particle] = colour;
106 }
107 
108 void
109 G4TrajectoryDrawByParticleID::Print(std::ostream& ostr) const
110 {
111  ostr<<"G4TrajectoryDrawByParticleID model "<< Name() <<" colour scheme: "<<std::endl;
112 
113  ostr<<"Default colour: "<<fDefault<<G4endl;
114 
115  fMap.Print(ostr);
116 
117  ostr<<"Default configuration:"<<G4endl;
119 }
void DrawLineAndPoints(const G4VTrajectory &traj, const G4VisTrajContext &)
const XML_Char * name
Definition: expat.h:151
const G4VisTrajContext & GetContext() const
void Print(std::ostream &ostr) const
std::ostringstream G4ExceptionDescription
Definition: G4Exception.hh:45
virtual void Print(std::ostream &ostr) const
#define G4endl
Definition: G4ios.hh:61
void SetVisible(const G4bool &visible)
bool G4bool
Definition: G4Types.hh:79
G4TrajectoryDrawByParticleID(const G4String &name="Unspecified", G4VisTrajContext *context=0)
virtual G4String GetParticleName() const =0
bool GetColour(const T &, G4Colour &) const
virtual void Draw(const G4VTrajectory &trajectory, const G4bool &visible=true) const
G4String Name() const
void G4Exception(const char *originOfException, const char *exceptionCode, G4ExceptionSeverity severity, const char *description)
Definition: G4Exception.hh:65
const XML_Char * context
Definition: expat.h:434
void Set(const G4String &particle, const G4String &colour)
static G4Colour White()
Definition: G4Colour.hh:155
void Set(const T &, const G4Colour &)
G4GLOB_DLL std::ostream G4cout
G4bool GetVerbose() const
void Print(std::ostream &ostr) const
void SetLineColour(const G4Colour &colour)
static G4bool GetColour(const G4String &key, G4Colour &result)
Definition: G4Colour.cc:163