Geant4  10.01.p01
G4ArrowModel.cc
Go to the documentation of this file.
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 // $Id: G4ArrowModel.cc 76285 2013-11-08 13:01:44Z gcosmo $
28 //
29 //
30 // John Allison 15th July 2012
31 // Model that knows how to draw an arrow.
32 
33 #include "G4ArrowModel.hh"
34 
35 #include "G4PhysicalConstants.hh"
36 #include "G4VGraphicsScene.hh"
37 #include "G4VisAttributes.hh"
38 #include "G4Tubs.hh"
39 #include "G4GenericPolycone.hh"
40 #include "G4Polyhedron.hh"
41 #include "G4Vector3D.hh"
42 #include "G4Point3D.hh"
43 #include "G4Transform3D.hh"
44 
46 {
47  delete fpHeadPolyhedron;
48  delete fpShaftPolyhedron;
49 }
50 
53  G4double x2, G4double y2, G4double z2,
54  G4double width, const G4Colour& colour,
55  const G4String& description)
56 {
57  fType = "G4ArrowModel";
58  fGlobalTag = fType;
59  fGlobalDescription = fType + ": " + description;
60  fExtent = G4VisExtent
61  (std::min(x1,x2),
62  std::max(x1,x2),
63  std::min(y1,y2),
64  std::max(y1,y2),
65  std::min(z1,z2),
66  std::max(z1,z2));
67 
68  // Make a cylinder slightly shorter than the arrow length so that it
69  // doesn't stick out of the head.
70  const G4double shaftLength = std::sqrt
71  (std::pow(x2-x1,2)+std::pow(y2-y1,2)+std::pow(z2-z1,2));
72  G4double shaftRadius = width/2.;
73  // Limit the radius
74  if (shaftRadius > shaftLength/100.) shaftRadius = shaftLength/100.;
75  const G4double halfShaftLength = shaftLength/2.;
76  const G4double halfReduction = 4.*shaftRadius;
77  const G4double halfLength = halfShaftLength-halfReduction;
78  G4Tubs shaft("shaft",0.,shaftRadius,halfLength,0.,twopi);
79  fpShaftPolyhedron = shaft.CreatePolyhedron();
80  // Move it a little so that the tail is at z = -halfShaftLength.
81  fpShaftPolyhedron->Transform(G4Translate3D(0,0,-halfReduction));
82 
83  // Locate the head at +halfShaftLength.
84  const G4int numRZ = 3;
85  G4double r[] = {0,4,0};
86  G4double z[] = {0,-6,-4};
87  for (G4int i = 0; i < numRZ; i++) {
88  r[i] *= 2.*shaftRadius;
89  z[i] = halfShaftLength + z[i] * 2.*shaftRadius;
90  }
91  G4GenericPolycone head("head",0,twopi,numRZ,r,z);
92  fpHeadPolyhedron = head.CreatePolyhedron();
93 
94  // Transform to position
95  const G4Vector3D arrowDirection = G4Vector3D(x2-x1,y2-y1,z2-z1).unit();
96  const G4double theta = arrowDirection.theta();
97  const G4double phi = arrowDirection.phi();
98  const G4Point3D arrowCentre(0.5*(x1+x2),0.5*(y1+y2),0.5*(z1+z2));
99  const G4Transform3D tr =
100  G4Translate3D(arrowCentre) * G4RotateZ3D(phi) * G4RotateY3D(theta);
101  fpShaftPolyhedron->Transform(tr);
102  fpHeadPolyhedron->Transform(tr);
103 
104  G4VisAttributes va;
105  va.SetColour(colour);
106  va.SetForceSolid(true);
107  fpShaftPolyhedron->SetVisAttributes(va);
108  fpHeadPolyhedron->SetVisAttributes(va);
109 }
110 
112 {
113  sceneHandler.BeginPrimitives();
114  sceneHandler.AddPrimitive(*fpShaftPolyhedron);
115  sceneHandler.AddPrimitive(*fpHeadPolyhedron);
116  sceneHandler.EndPrimitives();
117 }
void SetColour(const G4Colour &)
HepGeom::RotateY3D G4RotateY3D
G4double z
Definition: TRTMaterials.hh:39
Definition: G4Tubs.hh:85
virtual void BeginPrimitives(const G4Transform3D &objectTransformation=G4Transform3D())=0
#define width
HepGeom::Point3D< G4double > G4Point3D
Definition: G4Point3D.hh:35
HepGeom::Vector3D< G4double > G4Vector3D
Definition: G4Vector3D.hh:35
void SetForceSolid(G4bool)
int G4int
Definition: G4Types.hh:78
HepGeom::RotateZ3D G4RotateZ3D
virtual void AddPrimitive(const G4Polyline &)=0
G4Polyhedron * CreatePolyhedron() const
G4Polyhedron * CreatePolyhedron() const
Definition: G4Tubs.cc:1919
HepGeom::Transform3D G4Transform3D
G4Polyhedron * fpShaftPolyhedron
Definition: G4ArrowModel.hh:67
virtual ~G4ArrowModel()
Definition: G4ArrowModel.cc:45
T max(const T t1, const T t2)
brief Return the largest of the two arguments
G4ArrowModel(G4double x1, G4double y1, G4double z1, G4double x2, G4double y2, G4double z2, G4double width, const G4Colour &colour, const G4String &description="")
Definition: G4ArrowModel.cc:52
T min(const T t1, const T t2)
brief Return the smallest of the two arguments
HepGeom::Translate3D G4Translate3D
virtual void EndPrimitives()=0
virtual void DescribeYourselfTo(G4VGraphicsScene &)
double G4double
Definition: G4Types.hh:76
G4Polyhedron * fpHeadPolyhedron
Definition: G4ArrowModel.hh:68