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.shape.MeshActor;
20 import org.simantics.g3d.vtk.shape.vtkMeshObject;
21 import org.simantics.opencascade.OccTriangulator;
22 import org.simantics.opencascade.SolidModelProvider;
23 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<String, Object>();
46 calculatedParameters = new HashMap<String, Object>();
49 public void visualize(vtkPanel 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 void setParametersUpdated(boolean parametersUpdated) {
121 this.parametersUpdated = parametersUpdated;
124 public Set<String> setParameterMap(Map<String, Object> parameters) {
125 Set<String> ids = new HashSet<String>();
126 for (String id : parameters.keySet()) {
127 Object currentValue = currentParameters.get(id);
128 Object newValue = parameters.get(id);
129 if (currentValue == newValue)
131 if (currentValue instanceof Double) {
132 if (Math.abs((Double)currentValue-(Double)newValue) < MathTools.NEAR_ZERO)
135 currentParameters.put(id, newValue);
136 parametersUpdated = true;
142 public GeometryProvider getGeometry() {
143 return geometryProvider;
146 public void setGeometry(final GeometryProvider provider) {
147 if (provider != null && provider.equals(geometryProvider))
150 if (geometryProvider != null) {
153 geometryProvider = provider;
156 private void deleteData() {
157 geometryProvider = null;
158 if (solidObject != null) {
159 solidObject.clearActors();
162 if (solidModel != null) {
166 if (meshObject != null) {
167 meshObject.clearActors();
175 public Collection<vtkProp3D> getActors() {
176 List<vtkProp3D> list = new ArrayList<vtkProp3D>();
177 if (solidObject != null)
178 list.addAll(solidObject.getActors());
179 if (meshObject != null && meshObject.getActor() != null)
180 list.add(meshObject.getActor());
184 public void stopVisualize() {
185 if (solidObject != null) {
186 solidObject.clearActorsAWT();
189 if (solidModel != null) {
193 if (meshObject != null) {
194 meshObject.clearActorsAWT();
202 private boolean selected = false;
203 private boolean hover = false;
205 public void highlight(HighlightEventType type) {
206 if (type == HighlightEventType.Selection || type == HighlightEventType.ClearSelection) {
207 selected = type == HighlightEventType.Selection;
208 // hingeA.visualizeSolid(selected,true,false);
209 // hingeB.visualizeSolid(selected,true,false);
210 updateVisuals(true, false);
213 hover = type == HighlightEventType.Hover;
214 updateVisuals(false, true);
220 protected double[] getSelectedColor() {
221 return new double[]{1,0,0};
224 protected double[] getColor() {
225 return new double[]{1,1,0};
228 public void updateVisuals(boolean s, boolean h) {
229 if (solidObject != null) {
233 color = getSelectedColor();
239 for (vtkProp3D prop : solidObject.getSolid()) {
240 vtkProperty property = ((vtkActor)prop).GetProperty();
241 property.SetColor(color);
246 double color[] = new double[]{0,0,0};
248 color = new double[]{1,0,1};
249 for (vtkProp3D prop : solidObject.getEdges()) {
250 vtkProperty property = ((vtkActor)prop).GetProperty();
251 property.SetColor(color);
256 } else if (meshObject != null) {
260 color = getSelectedColor();
266 vtkProp3D prop = meshObject.getActor();
268 vtkProperty property = ((vtkActor)prop).GetProperty();
269 property.SetColor(color);
275 double color[] = new double[]{0,0,0};
276 MeshActor actor = meshObject.getActor();
278 vtkProperty property = actor.GetProperty();
280 color = new double[]{1,0,1};
281 property.SetEdgeVisibility(1);
282 property.SetEdgeColor(color);
284 property.SetEdgeVisibility(0);
285 property.SetEdgeColor(color);
292 // axes.addToRenderer();
293 // axes.setAxesVisibility(selected);