ECCE @ EIC Software
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Groups Pages
G4TrajectoryDrawByOriginVolume.cc
Go to the documentation of this file. Or view the newest version in sPHENIX GitHub for file G4TrajectoryDrawByOriginVolume.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 //
27 // Jane Tinslay March 2006
28 
31 #include "G4VTrajectory.hh"
33 #include "G4VisTrajContext.hh"
34 #include "G4VTrajectoryPoint.hh"
35 
37  :G4VTrajectoryModel(name, context)
38  ,fDefault(G4Colour::Grey())
39 {}
40 
42 
43 void
44 G4TrajectoryDrawByOriginVolume::Draw(const G4VTrajectory& traj, const G4bool& /*visible*/) const
45 {
46  G4VTrajectoryPoint* aTrajectoryPoint = traj.GetPoint(0);
47  assert (0 != aTrajectoryPoint);
48 
49  G4Colour colour(fDefault);
50 
53 
55  (aTrajectoryPoint->GetPosition(), nullptr,false,true);
56 
57  // Logical volumes form basis.
58  G4LogicalVolume* logicalVolume = volume->GetLogicalVolume();
59  assert (0 != logicalVolume);
60 
61  G4String logicalName = logicalVolume->GetName();
62  fMap.GetColour(logicalName, colour);
63 
64  // Override with physical volume colouring if it exists
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 drawer named "<<Name();
74  G4cout<<", drawing trajectory originating in logical volume, "<<logicalName;
75  G4cout<<", physical volume "<<physicalName<<", with configuration:"<<G4endl;
76  myContext.Print(G4cout);
77  }
78 
80 }
81 
82 void
84 {
85  G4Colour myColour;
86 
87  // Will not modify default colour if colour key does not exist
88  if (!G4Colour::GetColour(colour, myColour)) {
90  ed << "G4Colour with key "<<colour<<" does not exist ";
92  ("G4TrajectoryDrawByOriginParticleID::SetDefault(const G4String& colour)", "modeling0123", JustWarning, ed);
93  return;
94  }
95 
96  SetDefault(myColour);
97 }
98 
99 void
101 {
102  fDefault = colour;
103 }
104 
105 void
107 {
108  fMap.Set(particle, colour);
109 }
110 
111 void
113 {
114  fMap[particle] = colour;
115 }
116 
117 void
118 G4TrajectoryDrawByOriginVolume::Print(std::ostream& ostr) const
119 {
120  ostr
121  << "G4TrajectoryDrawByOriginVolume model "<< Name()
122  << ", colour scheme: "
123  << ", Default " << fDefault
124  << std::endl;
125 
126  fMap.Print(ostr);
127 
128  ostr << "Default configuration:" << std::endl;
129  GetContext().Print(ostr);
130 }