Geant4 Cross Reference |
1 // 1 // 2 // ******************************************* 2 // ******************************************************************** 3 // * License and Disclaimer 3 // * License and Disclaimer * 4 // * 4 // * * 5 // * The Geant4 software is copyright of th 5 // * The Geant4 software is copyright of the Copyright Holders of * 6 // * the Geant4 Collaboration. It is provided 6 // * the Geant4 Collaboration. It is provided under the terms and * 7 // * conditions of the Geant4 Software License 7 // * conditions of the Geant4 Software License, included in the file * 8 // * LICENSE and available at http://cern.ch/ 8 // * LICENSE and available at http://cern.ch/geant4/license . These * 9 // * include a list of copyright holders. 9 // * include a list of copyright holders. * 10 // * 10 // * * 11 // * Neither the authors of this software syst 11 // * Neither the authors of this software system, nor their employing * 12 // * institutes,nor the agencies providing fin 12 // * institutes,nor the agencies providing financial support for this * 13 // * work make any representation or warran 13 // * work make any representation or warranty, express or implied, * 14 // * regarding this software system or assum 14 // * regarding this software system or assume any liability for its * 15 // * use. Please see the license in the file 15 // * use. Please see the license in the file LICENSE and URL above * 16 // * for the full disclaimer and the limitatio 16 // * for the full disclaimer and the limitation of liability. * 17 // * 17 // * * 18 // * This code implementation is the result 18 // * This code implementation is the result of the scientific and * 19 // * technical work of the GEANT4 collaboratio 19 // * technical work of the GEANT4 collaboration. * 20 // * By using, copying, modifying or distri 20 // * By using, copying, modifying or distributing the software (or * 21 // * any work based on the software) you ag 21 // * any work based on the software) you agree to acknowledge its * 22 // * use in resulting scientific publicati 22 // * use in resulting scientific publications, and indicate your * 23 // * acceptance of all terms of the Geant4 Sof 23 // * acceptance of all terms of the Geant4 Software license. * 24 // ******************************************* 24 // ******************************************************************** 25 // 25 // 26 // 26 // 27 // 27 // 28 // 28 // 29 // John Allison 15th July 2012 29 // John Allison 15th July 2012 30 // Model that knows how to draw an arrow. 30 // Model that knows how to draw an arrow. 31 31 32 #include "G4ArrowModel.hh" 32 #include "G4ArrowModel.hh" 33 33 34 #include "G4PhysicalConstants.hh" 34 #include "G4PhysicalConstants.hh" 35 #include "G4VGraphicsScene.hh" 35 #include "G4VGraphicsScene.hh" 36 #include "G4VisAttributes.hh" 36 #include "G4VisAttributes.hh" 37 #include "G4Tubs.hh" 37 #include "G4Tubs.hh" 38 #include "G4Tet.hh" 38 #include "G4Tet.hh" 39 #include "G4Polyhedron.hh" 39 #include "G4Polyhedron.hh" 40 #include "G4Vector3D.hh" 40 #include "G4Vector3D.hh" 41 #include "G4Point3D.hh" 41 #include "G4Point3D.hh" 42 #include "G4Transform3D.hh" 42 #include "G4Transform3D.hh" 43 #include "G4GeometryTolerance.hh" 43 #include "G4GeometryTolerance.hh" 44 44 45 #include <cmath> << 46 << 47 G4ArrowModel::~G4ArrowModel () 45 G4ArrowModel::~G4ArrowModel () 48 { 46 { 49 delete fpHeadPolyhedron; 47 delete fpHeadPolyhedron; 50 delete fpShaftPolyhedron; 48 delete fpShaftPolyhedron; 51 } 49 } 52 50 53 G4ArrowModel::G4ArrowModel 51 G4ArrowModel::G4ArrowModel 54 (G4double x1, G4double y1, G4double z1, 52 (G4double x1, G4double y1, G4double z1, 55 G4double x2, G4double y2, G4double z2, 53 G4double x2, G4double y2, G4double z2, 56 G4double width, const G4Colour& colour, 54 G4double width, const G4Colour& colour, 57 const G4String& description, 55 const G4String& description, 58 G4int lineSegmentsPerCircle, << 56 G4int lineSegmentsPerCircle) 59 const G4Transform3D& transform) << 60 : fpShaftPolyhedron(nullptr) 57 : fpShaftPolyhedron(nullptr) 61 , fpHeadPolyhedron(nullptr) 58 , fpHeadPolyhedron(nullptr) 62 , fTransform(transform) << 63 { 59 { 64 fType = "G4ArrowModel"; 60 fType = "G4ArrowModel"; 65 fGlobalTag = fType; 61 fGlobalTag = fType; 66 fGlobalDescription = fType + ": " + descript 62 fGlobalDescription = fType + ": " + description; 67 fExtent = G4VisExtent 63 fExtent = G4VisExtent 68 (std::min(x1,x2), 64 (std::min(x1,x2), 69 std::max(x1,x2), 65 std::max(x1,x2), 70 std::min(y1,y2), 66 std::min(y1,y2), 71 std::max(y1,y2), 67 std::max(y1,y2), 72 std::min(z1,z2), 68 std::min(z1,z2), 73 std::max(z1,z2)); 69 std::max(z1,z2)); 74 70 75 // Force number of line segments per circle 71 // Force number of line segments per circle (aka number of rotation steps) 76 G4int tempN = G4Polyhedron::GetNumberOfRotat 72 G4int tempN = G4Polyhedron::GetNumberOfRotationSteps(); 77 G4Polyhedron::SetNumberOfRotationSteps(lineS 73 G4Polyhedron::SetNumberOfRotationSteps(lineSegmentsPerCircle); 78 74 79 // Make a cylinder slightly shorter than the 75 // Make a cylinder slightly shorter than the arrow length so that it 80 // doesn't stick out of the head. 76 // doesn't stick out of the head. 81 const G4double tolerance = G4GeometryToleran 77 const G4double tolerance = G4GeometryTolerance::GetInstance()->GetRadialTolerance(); 82 << 78 G4double shaftLength = std::sqrt 83 G4double totalLength = std::hypot(x2-x1, y2- << 79 (std::pow(x2-x1,2)+std::pow(y2-y1,2)+std::pow(z2-z1,2)); 84 if (totalLength < tolerance) << 80 if (shaftLength < tolerance) shaftLength = tolerance; 85 {totalLength = tolerance;} << 81 G4double shaftRadius = width/2.; 86 << 82 if (shaftRadius > shaftLength/100.) shaftRadius = shaftLength/100.; 87 G4double shaftRadius = width/6.; << 83 if (shaftRadius < tolerance) shaftRadius = tolerance; 88 if (shaftRadius < tolerance) << 84 const G4double halfShaftLength = shaftLength/2.; 89 {shaftRadius = tolerance;} << 85 const G4double halfReduction = 4.*shaftRadius; 90 << 86 G4double halfLength = halfShaftLength - halfReduction; 91 // case 1 - arrow length >> width -> arrow h << 87 if (halfLength < tolerance) halfLength = tolerance; 92 // case 2 - arrow length < width -> arrow h << 88 const G4Tubs shaft("shaft",0.,shaftRadius,halfLength,0.,twopi); 93 G4double arrowLength = std::min(1.5*width, 0 << 94 << 95 G4double shaftLength = totalLength - arrowLe << 96 if (shaftLength < 2*tolerance) << 97 {shaftLength = 2*tolerance;} << 98 << 99 const G4Tubs shaft("shaft",0.,shaftRadius,0. << 100 fpShaftPolyhedron = shaft.CreatePolyhedron() 89 fpShaftPolyhedron = shaft.CreatePolyhedron(); 101 // translate the polyhedron down w.r.t. the << 90 // Move it a little so that the tail is at z = -halfShaftLength. 102 if (fpShaftPolyhedron) 91 if (fpShaftPolyhedron) 103 {fpShaftPolyhedron->Transform(G4Translate3 << 92 fpShaftPolyhedron->Transform(G4Translate3D(0,0,-halfReduction)); 104 93 105 // Locate the head at +halfShaftLength. 94 // Locate the head at +halfShaftLength. 106 const G4double zHi = 0.5*totalLength; << 95 const G4double zHi = halfShaftLength; 107 const G4double zLow = zHi - arrowLength; << 96 const G4double zLow = halfShaftLength - 12.*shaftRadius; 108 const G4double rExt = 0.5*width; << 97 const G4double rExt = 8. * shaftRadius; 109 const G4double xExt = std::sqrt(3.)*rExt/2.; 98 const G4double xExt = std::sqrt(3.)*rExt/2.; 110 const G4Tet head("head", 99 const G4Tet head("head", 111 G4ThreeVector(0.,0.,zHi), 100 G4ThreeVector(0.,0.,zHi), 112 G4ThreeVector(0.,rExt,zLow) 101 G4ThreeVector(0.,rExt,zLow), 113 G4ThreeVector(xExt,-rExt/2. 102 G4ThreeVector(xExt,-rExt/2.,zLow), 114 G4ThreeVector(-xExt,-rExt/2 103 G4ThreeVector(-xExt,-rExt/2.,zLow)); 115 fpHeadPolyhedron = head.CreatePolyhedron(); 104 fpHeadPolyhedron = head.CreatePolyhedron(); 116 105 117 // Transform to position 106 // Transform to position 118 const G4Vector3D arrowDirection = G4Vector3D 107 const G4Vector3D arrowDirection = G4Vector3D(x2-x1,y2-y1,z2-z1).unit(); 119 const G4double theta = arrowDirection.theta( 108 const G4double theta = arrowDirection.theta(); 120 const G4double phi = arrowDirection.phi(); 109 const G4double phi = arrowDirection.phi(); 121 const G4Point3D arrowCentre(0.5*(x1+x2),0.5* 110 const G4Point3D arrowCentre(0.5*(x1+x2),0.5*(y1+y2),0.5*(z1+z2)); 122 const G4Transform3D tr = 111 const G4Transform3D tr = 123 G4Translate3D(arrowCentre) * G4RotateZ3D(p 112 G4Translate3D(arrowCentre) * G4RotateZ3D(phi) * G4RotateY3D(theta); 124 if (fpShaftPolyhedron) fpShaftPolyhedron->Tr 113 if (fpShaftPolyhedron) fpShaftPolyhedron->Transform(tr); 125 if (fpHeadPolyhedron) fpHeadPolyhedron->Tran 114 if (fpHeadPolyhedron) fpHeadPolyhedron->Transform(tr); 126 115 127 G4VisAttributes va; 116 G4VisAttributes va; 128 va.SetColour(colour); 117 va.SetColour(colour); 129 va.SetForceSolid(true); 118 va.SetForceSolid(true); 130 if (fpShaftPolyhedron) fpShaftPolyhedron->Se 119 if (fpShaftPolyhedron) fpShaftPolyhedron->SetVisAttributes(va); 131 if (fpHeadPolyhedron) fpHeadPolyhedron->SetV 120 if (fpHeadPolyhedron) fpHeadPolyhedron->SetVisAttributes(va); 132 121 133 // Restore number of line segments per circl 122 // Restore number of line segments per circle 134 G4Polyhedron::SetNumberOfRotationSteps(tempN 123 G4Polyhedron::SetNumberOfRotationSteps(tempN); 135 } 124 } 136 125 137 void G4ArrowModel::DescribeYourselfTo (G4VGrap 126 void G4ArrowModel::DescribeYourselfTo (G4VGraphicsScene& sceneHandler) 138 { 127 { 139 if (fpShaftPolyhedron && fpHeadPolyhedron) { << 128 sceneHandler.BeginPrimitives(); 140 sceneHandler.BeginPrimitives(fTransform); << 129 if (fpShaftPolyhedron) sceneHandler.AddPrimitive(*fpShaftPolyhedron); 141 sceneHandler.AddPrimitive(*fpShaftPolyhedr << 130 if (fpHeadPolyhedron) sceneHandler.AddPrimitive(*fpHeadPolyhedron); 142 sceneHandler.AddPrimitive(*fpHeadPolyhedro << 131 sceneHandler.EndPrimitives(); 143 sceneHandler.EndPrimitives(); << 144 } << 145 } 132 } 146 133