source: trunk/source/visualization/modeling/src/G4TrajectoryDrawByParticleID.cc @ 1288

Last change on this file since 1288 was 1288, checked in by garnier, 14 years ago

update from CVS

File size: 4.1 KB
Line 
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,v 1.10 2010/05/28 02:00:59 allison Exp $
27// GEANT4 tag $Name:  $
28//
29// Jane Tinslay, John Allison, Joseph Perl November 2005
30
31#include "G4TrajectoryDrawByParticleID.hh"
32#include "G4TrajectoryDrawerUtils.hh"
33#include "G4VisTrajContext.hh"
34#include "G4VTrajectory.hh"
35#include <sstream>
36
37G4TrajectoryDrawByParticleID::G4TrajectoryDrawByParticleID(const G4String& name, G4VisTrajContext* context)
38  :G4VTrajectoryModel(name, context)
39  ,fDefault(G4Colour::Grey())
40{}
41
42G4TrajectoryDrawByParticleID::~G4TrajectoryDrawByParticleID() {}
43
44void
45G4TrajectoryDrawByParticleID::Draw(const G4VTrajectory& object,
46                                   const G4int&,
47                                   const G4bool& visible) const
48{
49  Draw(object, visible);
50}
51
52void
53G4TrajectoryDrawByParticleID::Draw(const G4VTrajectory& traj, const G4bool& visible) const
54{
55  G4Colour colour(fDefault);
56  G4String particle = traj.GetParticleName();
57
58  fMap.GetColour(particle, colour);
59
60  G4VisTrajContext myContext(GetContext());
61 
62  myContext.SetLineColour(colour);
63  myContext.SetVisible(visible);
64 
65  if (GetVerbose()) {
66    G4cout<<"G4TrajectoryDrawByParticleID drawer named "<<Name();
67    G4cout<<", drawing trajectory with particle type, "<<particle<<G4endl;
68    G4cout<<", with configuration:"<<G4endl;
69    myContext.Print(G4cout);
70  }
71
72  G4TrajectoryDrawerUtils::DrawLineAndPoints(traj, myContext);
73}
74
75void
76G4TrajectoryDrawByParticleID::SetDefault(const G4String& colour)
77{
78  G4Colour myColour(G4Colour::White());     
79
80  // Will not modify myColour if colour key does not exist 
81  if (!G4Colour::GetColour(colour, myColour)) {
82    std::ostringstream o;
83    o << "G4Colour with key "<<colour<<" does not exist ";
84    G4Exception
85      ("G4TrajectoryDrawByParticleID::SetDefault(const G4String& colour)",
86       "NonExistentColour", JustWarning, o.str().c_str());
87  }
88
89  SetDefault(myColour);
90}
91
92void
93G4TrajectoryDrawByParticleID::SetDefault(const G4Colour& colour)
94{
95  fDefault = colour;
96}
97
98void
99G4TrajectoryDrawByParticleID::Set(const G4String& particle, const G4String& colour)
100{
101  fMap.Set(particle, colour);
102}
103
104void
105G4TrajectoryDrawByParticleID::Set(const G4String& particle, const G4Colour& colour)
106{
107  fMap[particle] = colour;
108}
109
110void
111G4TrajectoryDrawByParticleID::Print(std::ostream& ostr) const
112{
113  ostr<<"G4TrajectoryDrawByParticleID model "<< Name() <<" colour scheme: "<<std::endl;
114 
115  ostr<<"Default colour: "<<fDefault<<G4endl;
116
117  fMap.Print(ostr);
118
119  ostr<<"Default configuration:"<<G4endl;
120  GetContext().Print(G4cout);
121}
Note: See TracBrowser for help on using the repository browser.