1 package org.simantics.plant3d.scenegraph;
3 import java.util.ArrayList;
4 import java.util.Collection;
5 import java.util.Collections;
6 import java.util.HashMap;
7 import java.util.HashSet;
12 import org.jcae.opencascade.jni.TopoDS_Shape;
13 import org.simantics.g3d.math.MathTools;
14 import org.simantics.g3d.scenegraph.GeometryProvider;
15 import org.simantics.g3d.scenegraph.MeshProvider;
16 import org.simantics.g3d.scenegraph.NodeHighlighter.HighlightEventType;
17 import org.simantics.g3d.scenegraph.ParametricGeometryProvider;
18 import org.simantics.g3d.shape.Mesh;
19 import org.simantics.g3d.vtk.common.VtkView;
20 import org.simantics.g3d.vtk.shape.MeshActor;
21 import org.simantics.g3d.vtk.shape.vtkMeshObject;
22 import org.simantics.opencascade.OccTriangulator;
23 import org.simantics.opencascade.SolidModelProvider;
24 import org.simantics.opencascade.vtk.vtkSolidObject;
28 import vtk.vtkProperty;
30 public class GeometryComponent {
32 private TopoDS_Shape solidModel;
33 private vtkSolidObject solidObject;
35 private vtkMeshObject meshObject;
37 private Map<String,Object> currentParameters;
38 private Map<String,Object> calculatedParameters;
40 private boolean parametersUpdated = true;
42 private GeometryProvider geometryProvider;
44 public GeometryComponent() {
45 currentParameters = new HashMap<>();
46 calculatedParameters = new HashMap<>();
49 public void visualize(VtkView panel) {
50 if (geometryProvider != null) {
53 if (geometryProvider instanceof SolidModelProvider) {
54 SolidModelProvider solidModelProvider = (SolidModelProvider)geometryProvider;
55 if (solidModel == null || parametersUpdated) {
56 createGeometry(solidModelProvider);
58 if ((solidObject == null || parametersUpdated) && solidModel != null) {
59 solidObject = new vtkSolidObject(panel, solidModel);
61 if (solidObject != null) {
62 solidObject.visualizeSolid(true,true, false,false);
63 updateVisuals(true, true);
65 } else if (geometryProvider instanceof MeshProvider) {
66 MeshProvider provider = (MeshProvider)geometryProvider;
67 if (mesh == null || parametersUpdated) {
68 mesh = provider.getMesh();
70 if ((meshObject== null || parametersUpdated) && mesh != null) {
71 meshObject = new vtkMeshObject(panel, mesh);
74 if (meshObject != null) {
75 meshObject.visualizeMesh();
76 updateVisuals(true, true);
81 parametersUpdated = false;
85 public void updateParameters() {
86 //if (parametersUpdated && geometryProvider instanceof ParametricGeometryProvider) {
87 if (geometryProvider instanceof ParametricGeometryProvider) {
88 ((ParametricGeometryProvider)geometryProvider).setProperties(currentParameters);
89 ((ParametricGeometryProvider)geometryProvider).updateCalculatedProperties(calculatedParameters);
93 private void createGeometry(SolidModelProvider solidModelProvider) {
94 Collection<TopoDS_Shape> shapes;
96 shapes = solidModelProvider.getModel();
97 if (shapes.size() == 1) {
98 solidModel = shapes.iterator().next();
100 solidModel = OccTriangulator.makeCompound(shapes.toArray(new TopoDS_Shape[shapes.size()]));
101 for (TopoDS_Shape shape : shapes) {
106 } catch (Exception e) {
107 // TODO Auto-generated catch block
112 public Map<String, Object> getParameterMap() {
113 return Collections.unmodifiableMap(currentParameters);
116 public Map<String, Object> getCalculatedParameters() {
117 return calculatedParameters;
120 public Map<String, Object> getTotalParameters() {
121 Map<String,Object> params = new HashMap<>();
122 params.putAll(currentParameters);
123 params.putAll(calculatedParameters);
127 public void setParametersUpdated(boolean parametersUpdated) {
128 this.parametersUpdated = parametersUpdated;
131 public Set<String> setParameterMap(Map<String, Object> parameters) {
132 Set<String> ids = new HashSet<String>();
133 for (String id : parameters.keySet()) {
134 Object currentValue = currentParameters.get(id);
135 Object newValue = parameters.get(id);
136 if (currentValue == newValue)
138 if (currentValue instanceof Double) {
139 if (Math.abs((Double)currentValue-(Double)newValue) < MathTools.NEAR_ZERO)
142 currentParameters.put(id, newValue);
143 parametersUpdated = true;
149 public GeometryProvider getGeometry() {
150 return geometryProvider;
153 public void setGeometry(final GeometryProvider provider) {
154 if (provider != null && provider.equals(geometryProvider))
157 if (geometryProvider != null) {
160 geometryProvider = provider;
163 private void deleteData() {
164 geometryProvider = null;
165 if (solidObject != null) {
166 solidObject.clearActors();
169 if (solidModel != null) {
173 if (meshObject != null) {
174 meshObject.clearActors();
182 public Collection<vtkProp3D> getActors() {
183 List<vtkProp3D> list = new ArrayList<vtkProp3D>();
184 if (solidObject != null)
185 list.addAll(solidObject.getActors());
186 if (meshObject != null && meshObject.getActor() != null)
187 list.add(meshObject.getActor());
191 public void stopVisualize() {
192 if (solidObject != null) {
193 solidObject.clearActorsVTK();
196 if (solidModel != null) {
200 if (meshObject != null) {
201 meshObject.clearActorsVTK();
209 private boolean selected = false;
210 private boolean hover = false;
212 public void highlight(HighlightEventType type) {
213 if (type == HighlightEventType.Selection || type == HighlightEventType.ClearSelection) {
214 selected = type == HighlightEventType.Selection;
215 // hingeA.visualizeSolid(selected,true,false);
216 // hingeB.visualizeSolid(selected,true,false);
217 updateVisuals(true, false);
220 hover = type == HighlightEventType.Hover;
221 updateVisuals(false, true);
227 protected double[] getSelectedColor() {
228 return new double[]{1,0,0};
231 protected double[] getColor() {
232 return new double[]{1,1,0};
235 public void updateVisuals(boolean s, boolean h) {
236 if (solidObject != null) {
240 color = getSelectedColor();
246 for (vtkProp3D prop : solidObject.getSolid()) {
247 vtkProperty property = ((vtkActor)prop).GetProperty();
248 property.SetColor(color);
253 double color[] = new double[]{0,0,0};
255 color = new double[]{1,0,1};
256 for (vtkProp3D prop : solidObject.getEdges()) {
257 vtkProperty property = ((vtkActor)prop).GetProperty();
258 property.SetColor(color);
263 } else if (meshObject != null) {
267 color = getSelectedColor();
273 vtkProp3D prop = meshObject.getActor();
275 vtkProperty property = ((vtkActor)prop).GetProperty();
276 property.SetColor(color);
282 double color[] = new double[]{0,0,0};
283 MeshActor actor = meshObject.getActor();
285 vtkProperty property = actor.GetProperty();
287 color = new double[]{1,0,1};
288 property.SetEdgeVisibility(1);
289 property.SetEdgeColor(color);
291 property.SetEdgeVisibility(0);
292 property.SetEdgeColor(color);
299 // axes.addToRenderer();
300 // axes.setAxesVisibility(selected);