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