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 March 2006 28 29 #include "G4TrajectoryDrawByOriginVolume.hh" 30 #include "G4TrajectoryDrawerUtils.hh" 31 #include "G4VTrajectory.hh" 32 #include "G4TransportationManager.hh" 33 #include "G4VisTrajContext.hh" 34 #include "G4VTrajectoryPoint.hh" 35 36 G4TrajectoryDrawByOriginVolume::G4TrajectoryDr 37 :G4VTrajectoryModel(name, context) 38 ,fDefault(G4Colour::Grey()) 39 {} 40 41 G4TrajectoryDrawByOriginVolume::~G4TrajectoryD 42 43 void 44 G4TrajectoryDrawByOriginVolume::Draw(const G4V 45 { 46 G4VTrajectoryPoint* aTrajectoryPoint = traj. 47 assert (0 != aTrajectoryPoint); 48 49 G4Colour colour(fDefault); 50 51 G4Navigator* navigator = 52 G4TransportationManager::GetTransportationMa 53 54 G4VPhysicalVolume* volume = navigator->Locat 55 (aTrajectoryPoint->GetPosition(), nullptr,fa 56 57 // Logical volumes form basis. 58 G4LogicalVolume* logicalVolume = volume->Get 59 assert (0 != logicalVolume); 60 61 G4String logicalName = logicalVolume->GetNam 62 fMap.GetColour(logicalName, colour); 63 64 // Override with physical volume colouring i 65 G4String physicalName = volume->GetName(); 66 fMap.GetColour(physicalName, colour); 67 68 G4VisTrajContext myContext(GetContext()); 69 70 myContext.SetLineColour(colour); 71 72 if (GetVerbose()) { 73 G4cout<<"G4TrajectoryDrawByOriginVolume dr 74 G4cout<<", drawing trajectory originating 75 G4cout<<", physical volume "<<physicalName 76 myContext.Print(G4cout); 77 } 78 79 G4TrajectoryDrawerUtils::DrawLineAndPoints(t 80 } 81 82 void 83 G4TrajectoryDrawByOriginVolume::SetDefault(con 84 { 85 G4Colour myColour; 86 87 // Will not modify default colour if colour 88 if (!G4Colour::GetColour(colour, myColour)) 89 G4ExceptionDescription ed; 90 ed << "G4Colour with key "<<colour<<" does 91 G4Exception 92 ("G4TrajectoryDrawByOriginParticleID::Se 93 return; 94 } 95 96 SetDefault(myColour); 97 } 98 99 void 100 G4TrajectoryDrawByOriginVolume::SetDefault(con 101 { 102 fDefault = colour; 103 } 104 105 void 106 G4TrajectoryDrawByOriginVolume::Set(const G4St 107 { 108 fMap.Set(particle, colour); 109 } 110 111 void 112 G4TrajectoryDrawByOriginVolume::Set(const G4St 113 { 114 fMap[particle] = colour; 115 } 116 117 void 118 G4TrajectoryDrawByOriginVolume::Print(std::ost 119 { 120 ostr 121 << "G4TrajectoryDrawByOriginVolume model "<< 122 << ", colour scheme: " 123 << ", Default " << fDefault 124 << std::endl; 125 126 fMap.Print(ostr); 127 128 ostr << "Default configuration:" << std::end 129 GetContext().Print(ostr); 130 } 131