| 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 #include "G4VtkPolydataInstanceBakePipeline.hh 26 #include "G4VtkPolydataInstanceBakePipeline.hh"
27 27
28 #include "G4VtkViewer.hh" 28 #include "G4VtkViewer.hh"
29 #include "G4VtkVisContext.hh" 29 #include "G4VtkVisContext.hh"
30 30
31 #include <vtkActor.h> << 31 #include "vtkActor.h"
32 #include <vtkDataArray.h> << 32 #include "vtkDataArray.h"
33 #include <vtkDoubleArray.h> << 33 #include "vtkDoubleArray.h"
34 #include <vtkPointData.h> << 34 #include "vtkPointData.h"
35 #include <vtkPolyData.h> << 35 #include "vtkPolyData.h"
36 #include <vtkPolyDataMapper.h> << 36 #include "vtkPolyDataMapper.h"
37 #include <vtkProperty.h> << 37 #include "vtkProperty.h"
38 38
39 std::size_t G4VtkPolydataInstanceBakePipeline: 39 std::size_t G4VtkPolydataInstanceBakePipeline::MakeHash(const G4Polyhedron& polyhedron,
40 40 const G4VtkVisContext& vc)
41 { 41 {
42 // Get view parameters that the user can for 42 // Get view parameters that the user can force through the vis attributes, thereby over-riding the
43 // current view parameter. 43 // current view parameter.
44 const G4VisAttributes* pVA = 44 const G4VisAttributes* pVA =
45 vc.fViewer->GetApplicableVisAttributes(pol 45 vc.fViewer->GetApplicableVisAttributes(polyhedron.GetVisAttributes());
46 G4Colour colour = pVA->GetColour(); 46 G4Colour colour = pVA->GetColour();
47 47
48 // Hash the vis attributes 48 // Hash the vis attributes
49 std::size_t hash = std::hash<G4double>{}(col 49 std::size_t hash = std::hash<G4double>{}(colour.GetAlpha());
50 std::size_t shash = std::hash<G4double>{}(vc 50 std::size_t shash = std::hash<G4double>{}(vc.fDrawingStyle);
51 51
52 std::hash_combine(hash, shash); 52 std::hash_combine(hash, shash);
53 53
54 return hash; 54 return hash;
55 } 55 }
56 56
57 G4VtkPolydataInstanceBakePipeline::G4VtkPolyda 57 G4VtkPolydataInstanceBakePipeline::G4VtkPolydataInstanceBakePipeline(G4String nameIn,
58 58 const G4VtkVisContext& vcIn)
59 : G4VtkPolydataInstancePipeline(nameIn, vcIn 59 : G4VtkPolydataInstancePipeline(nameIn, vcIn)
60 { 60 {
61 iVert = 0; 61 iVert = 0;
62 iFace = 0; 62 iFace = 0;
63 polydataPointData = vtkSmartPointer<vtkDoubl 63 polydataPointData = vtkSmartPointer<vtkDoubleArray>::New();
64 polydataPointData->SetNumberOfComponents(3); 64 polydataPointData->SetNumberOfComponents(3);
65 polydataPointData->SetName("Colors"); 65 polydataPointData->SetName("Colors");
66 polydata->GetPointData()->SetScalars(polydat 66 polydata->GetPointData()->SetScalars(polydataPointData);
67 67
68 mapper->SetColorModeToDirectScalars(); 68 mapper->SetColorModeToDirectScalars();
69 69
70 actor->GetProperty()->SetOpacity(vc.alpha); 70 actor->GetProperty()->SetOpacity(vc.alpha);
71 } 71 }
72 72
73 void G4VtkPolydataInstanceBakePipeline::SetPol 73 void G4VtkPolydataInstanceBakePipeline::SetPolydata(const G4Polyhedron& polyhedron)
74 { 74 {
75 polyhedronPrototype = &polyhedron; 75 polyhedronPrototype = &polyhedron;
76 } 76 }
77 77
78 void G4VtkPolydataInstanceBakePipeline::Print( 78 void G4VtkPolydataInstanceBakePipeline::Print()
79 { 79 {
80 G4cout << "G4VtkPolydataInstanceBakePipeline 80 G4cout << "G4VtkPolydataInstanceBakePipeline " << GetName() << G4endl;
81 G4VtkPolydataPipeline::Print(); 81 G4VtkPolydataPipeline::Print();
82 } 82 }
83 83
84 void G4VtkPolydataInstanceBakePipeline::addIns 84 void G4VtkPolydataInstanceBakePipeline::addInstance(G4double dx, G4double dy, G4double dz,
85 85 G4double r00, G4double r01, G4double r02,
86 86 G4double r10, G4double r11, G4double r12,
87 87 G4double r20, G4double r21, G4double r22,
88 88 G4double r, G4double g, G4double b, G4double a,
89 89 const G4String& nameIn)
90 { 90 {
91 G4VtkPolydataInstancePipeline::addInstance(d 91 G4VtkPolydataInstancePipeline::addInstance(dx, dy, dz, r00, r01, r02, r10, r11, r12, r20, r21,
92 r 92 r22, r, g, b, a, nameIn);
93 93
94 vtkIdType vStart; 94 vtkIdType vStart;
95 vtkIdType vEnd; 95 vtkIdType vEnd;
96 vtkIdType fStart; 96 vtkIdType fStart;
97 vtkIdType fEnd; 97 vtkIdType fEnd;
98 98
99 vStart = iVert; 99 vStart = iVert;
100 fStart = iFace; 100 fStart = iFace;
101 101
102 G4bool notLastFace; 102 G4bool notLastFace;
103 do { 103 do {
104 G4Point3D vertex[4]; 104 G4Point3D vertex[4];
105 G4int edgeFlag[4]; 105 G4int edgeFlag[4];
106 G4Normal3D normals[4]; 106 G4Normal3D normals[4];
107 G4int nEdges; 107 G4int nEdges;
108 notLastFace = polyhedronPrototype->GetNext 108 notLastFace = polyhedronPrototype->GetNextFacet(nEdges, vertex, edgeFlag, normals);
109 109
110 vtkSmartPointer<vtkIdList> poly = vtkSmart 110 vtkSmartPointer<vtkIdList> poly = vtkSmartPointer<vtkIdList>::New();
111 // loop over vertices 111 // loop over vertices
112 for (int i = 0; i < nEdges; i++) { 112 for (int i = 0; i < nEdges; i++) {
113 // note : G4Transform3D does not have a 113 // note : G4Transform3D does not have a matrix element constructor
114 G4Point3D bakedVertex = 114 G4Point3D bakedVertex =
115 G4Point3D(vertex[i].x() * r00 + vertex 115 G4Point3D(vertex[i].x() * r00 + vertex[i].y() * r01 + vertex[i].z() * r02 + dx,
116 vertex[i].x() * r10 + vertex 116 vertex[i].x() * r10 + vertex[i].y() * r11 + vertex[i].z() * r12 + dy,
117 vertex[i].x() * r20 + vertex 117 vertex[i].x() * r20 + vertex[i].y() * r21 + vertex[i].z() * r22 + dz);
118 polydataPoints->InsertNextPoint(bakedVer 118 polydataPoints->InsertNextPoint(bakedVertex.x(), bakedVertex.y(), bakedVertex.z());
119 poly->InsertNextId(iVert); 119 poly->InsertNextId(iVert);
120 iVert++; 120 iVert++;
121 double cols[] = {r, g, b, 0.5}; 121 double cols[] = {r, g, b, 0.5};
122 polydataPointData->InsertNextTuple(cols) 122 polydataPointData->InsertNextTuple(cols);
123 } 123 }
124 124
125 polydataCells->InsertNextCell(poly); 125 polydataCells->InsertNextCell(poly);
126 iFace++; 126 iFace++;
127 127
128 } while (notLastFace); 128 } while (notLastFace);
129 129
130 vEnd = iVert; 130 vEnd = iVert;
131 fEnd = iFace; 131 fEnd = iFace;
132 132
133 // Add vertex and face to maps to allow for 133 // Add vertex and face to maps to allow for removal later
134 instanceVertexMap[name] = std::pair<vtkIdTyp 134 instanceVertexMap[name] = std::pair<vtkIdType, vtkIdType>(vStart, vEnd);
135 instanceFaceMap[name] = std::pair<vtkIdType, 135 instanceFaceMap[name] = std::pair<vtkIdType, vtkIdType>(fStart, fEnd);
136 } 136 }
137 137
138 void G4VtkPolydataInstanceBakePipeline::remove 138 void G4VtkPolydataInstanceBakePipeline::removeInstance(const G4String& /*name*/) {}
139 139