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