Geant4 Cross Reference

Cross-Referencing   Geant4
Geant4/visualization/modeling/src/G4ArrowModel.cc

Version: [ ReleaseNotes ] [ 1.0 ] [ 1.1 ] [ 2.0 ] [ 3.0 ] [ 3.1 ] [ 3.2 ] [ 4.0 ] [ 4.0.p1 ] [ 4.0.p2 ] [ 4.1 ] [ 4.1.p1 ] [ 5.0 ] [ 5.0.p1 ] [ 5.1 ] [ 5.1.p1 ] [ 5.2 ] [ 5.2.p1 ] [ 5.2.p2 ] [ 6.0 ] [ 6.0.p1 ] [ 6.1 ] [ 6.2 ] [ 6.2.p1 ] [ 6.2.p2 ] [ 7.0 ] [ 7.0.p1 ] [ 7.1 ] [ 7.1.p1 ] [ 8.0 ] [ 8.0.p1 ] [ 8.1 ] [ 8.1.p1 ] [ 8.1.p2 ] [ 8.2 ] [ 8.2.p1 ] [ 8.3 ] [ 8.3.p1 ] [ 8.3.p2 ] [ 9.0 ] [ 9.0.p1 ] [ 9.0.p2 ] [ 9.1 ] [ 9.1.p1 ] [ 9.1.p2 ] [ 9.1.p3 ] [ 9.2 ] [ 9.2.p1 ] [ 9.2.p2 ] [ 9.2.p3 ] [ 9.2.p4 ] [ 9.3 ] [ 9.3.p1 ] [ 9.3.p2 ] [ 9.4 ] [ 9.4.p1 ] [ 9.4.p2 ] [ 9.4.p3 ] [ 9.4.p4 ] [ 9.5 ] [ 9.5.p1 ] [ 9.5.p2 ] [ 9.6 ] [ 9.6.p1 ] [ 9.6.p2 ] [ 9.6.p3 ] [ 9.6.p4 ] [ 10.0 ] [ 10.0.p1 ] [ 10.0.p2 ] [ 10.0.p3 ] [ 10.0.p4 ] [ 10.1 ] [ 10.1.p1 ] [ 10.1.p2 ] [ 10.1.p3 ] [ 10.2 ] [ 10.2.p1 ] [ 10.2.p2 ] [ 10.2.p3 ] [ 10.3 ] [ 10.3.p1 ] [ 10.3.p2 ] [ 10.3.p3 ] [ 10.4 ] [ 10.4.p1 ] [ 10.4.p2 ] [ 10.4.p3 ] [ 10.5 ] [ 10.5.p1 ] [ 10.6 ] [ 10.6.p1 ] [ 10.6.p2 ] [ 10.6.p3 ] [ 10.7 ] [ 10.7.p1 ] [ 10.7.p2 ] [ 10.7.p3 ] [ 10.7.p4 ] [ 11.0 ] [ 11.0.p1 ] [ 11.0.p2 ] [ 11.0.p3, ] [ 11.0.p4 ] [ 11.1 ] [ 11.1.1 ] [ 11.1.2 ] [ 11.1.3 ] [ 11.2 ] [ 11.2.1 ] [ 11.2.2 ] [ 11.3.0 ]

  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 //
 28 // 
 29 // John Allison  15th July 2012
 30 // Model that knows how to draw an arrow.
 31 
 32 #include "G4ArrowModel.hh"
 33 
 34 #include "G4PhysicalConstants.hh"
 35 #include "G4VGraphicsScene.hh"
 36 #include "G4VisAttributes.hh"
 37 #include "G4Tubs.hh"
 38 #include "G4Tet.hh"
 39 #include "G4Polyhedron.hh"
 40 #include "G4Vector3D.hh"
 41 #include "G4Point3D.hh"
 42 #include "G4Transform3D.hh"
 43 #include "G4GeometryTolerance.hh"
 44 
 45 #include <cmath>
 46 
 47 G4ArrowModel::~G4ArrowModel ()
 48 {
 49   delete fpHeadPolyhedron;
 50   delete fpShaftPolyhedron;
 51 }
 52 
 53 G4ArrowModel::G4ArrowModel
 54 (G4double x1, G4double y1, G4double z1,
 55  G4double x2, G4double y2, G4double z2,
 56  G4double width, const G4Colour& colour,
 57  const G4String& description,
 58  G4int lineSegmentsPerCircle,
 59  const G4Transform3D& transform)
 60 : fpShaftPolyhedron(nullptr)
 61 , fpHeadPolyhedron(nullptr)
 62 , fTransform(transform)
 63 {
 64   fType = "G4ArrowModel";
 65   fGlobalTag = fType;
 66   fGlobalDescription = fType + ": " + description;
 67   fExtent = G4VisExtent
 68     (std::min(x1,x2),
 69      std::max(x1,x2),
 70      std::min(y1,y2),
 71      std::max(y1,y2),
 72      std::min(z1,z2),
 73      std::max(z1,z2));
 74 
 75   // Force number of line segments per circle (aka number of rotation steps)
 76   G4int tempN = G4Polyhedron::GetNumberOfRotationSteps();
 77   G4Polyhedron::SetNumberOfRotationSteps(lineSegmentsPerCircle);
 78 
 79   // Make a cylinder slightly shorter than the arrow length so that it
 80   // doesn't stick out of the head.
 81   const G4double tolerance = G4GeometryTolerance::GetInstance()->GetRadialTolerance();
 82   
 83   G4double totalLength = std::hypot(x2-x1, y2-y1, z2-z1);
 84   if (totalLength < tolerance)
 85     {totalLength = tolerance;}
 86   
 87   G4double shaftRadius = width/6.;
 88   if (shaftRadius < tolerance)
 89     {shaftRadius = tolerance;}
 90 
 91   // case 1 - arrow length >> width -> arrow head is width and 1.5x width tall
 92   // case 2 - arrow length <  width -> arrow head is made to be 0.5x length
 93   G4double arrowLength = std::min(1.5*width, 0.5*totalLength);
 94 
 95   G4double shaftLength = totalLength - arrowLength;
 96   if (shaftLength < 2*tolerance)
 97     {shaftLength = 2*tolerance;}
 98 
 99   const G4Tubs shaft("shaft",0.,shaftRadius,0.5*shaftLength,0.,twopi);
100   fpShaftPolyhedron = shaft.CreatePolyhedron();
101   // translate the polyhedron down w.r.t. the centre of the whole arrow
102   if (fpShaftPolyhedron)
103     {fpShaftPolyhedron->Transform(G4Translate3D(0,0,-0.5*arrowLength));}
104 
105   // Locate the head at +halfShaftLength.
106   const G4double zHi  = 0.5*totalLength;
107   const G4double zLow = zHi - arrowLength;
108   const G4double rExt = 0.5*width;
109   const G4double xExt = std::sqrt(3.)*rExt/2.;
110   const G4Tet head("head",
111                    G4ThreeVector(0.,0.,zHi),
112                    G4ThreeVector(0.,rExt,zLow),
113                    G4ThreeVector(xExt,-rExt/2.,zLow),
114                    G4ThreeVector(-xExt,-rExt/2.,zLow));
115   fpHeadPolyhedron = head.CreatePolyhedron();
116 
117   // Transform to position
118   const G4Vector3D arrowDirection = G4Vector3D(x2-x1,y2-y1,z2-z1).unit();
119   const G4double theta = arrowDirection.theta();
120   const G4double phi = arrowDirection.phi();
121   const G4Point3D arrowCentre(0.5*(x1+x2),0.5*(y1+y2),0.5*(z1+z2));
122   const G4Transform3D tr =
123     G4Translate3D(arrowCentre) * G4RotateZ3D(phi) * G4RotateY3D(theta);
124   if (fpShaftPolyhedron) fpShaftPolyhedron->Transform(tr);
125   if (fpHeadPolyhedron) fpHeadPolyhedron->Transform(tr);
126 
127   G4VisAttributes va;
128   va.SetColour(colour);
129   va.SetForceSolid(true);
130   if (fpShaftPolyhedron) fpShaftPolyhedron->SetVisAttributes(va);
131   if (fpHeadPolyhedron) fpHeadPolyhedron->SetVisAttributes(va);
132 
133   // Restore number of line segments per circle
134   G4Polyhedron::SetNumberOfRotationSteps(tempN);
135 }
136 
137 void G4ArrowModel::DescribeYourselfTo (G4VGraphicsScene& sceneHandler)
138 {
139   if (fpShaftPolyhedron && fpHeadPolyhedron) {
140     sceneHandler.BeginPrimitives(fTransform);
141     sceneHandler.AddPrimitive(*fpShaftPolyhedron);
142     sceneHandler.AddPrimitive(*fpHeadPolyhedron);
143     sceneHandler.EndPrimitives();
144   }
145 }
146