1 /*******************************************************************************
\r
2 * Copyright (c) 2007 VTT Technical Research Centre of Finland and others.
\r
3 * All rights reserved. This program and the accompanying materials
\r
4 * are made available under the terms of the Eclipse Public License v1.0
\r
5 * which accompanies this distribution, and is available at
\r
6 * http://www.eclipse.org/legal/epl-v10.html
\r
9 * VTT Technical Research Centre of Finland - initial API and implementation
\r
10 *******************************************************************************/
\r
11 package org.simantics.processeditor.views;
\r
13 import java.util.ArrayList;
\r
14 import java.util.Collection;
\r
15 import java.util.HashMap;
\r
16 import java.util.List;
\r
17 import java.util.Map;
\r
19 import org.eclipse.jface.action.Action;
\r
20 import org.eclipse.jface.viewers.ISelection;
\r
21 import org.eclipse.jface.viewers.ISelectionChangedListener;
\r
22 import org.eclipse.jface.viewers.SelectionChangedEvent;
\r
23 import org.eclipse.swt.widgets.Composite;
\r
24 import org.eclipse.ui.IWorkbenchPart;
\r
25 import org.eclipse.ui.views.contentoutline.IContentOutlinePage;
\r
26 import org.simantics.db.Graph;
\r
27 import org.simantics.db.Resource;
\r
28 import org.simantics.db.Session;
\r
29 import org.simantics.db.management.ISessionContext;
\r
30 import org.simantics.layer0.utils.EntityFactory;
\r
31 import org.simantics.layer0.utils.IEntity;
\r
32 import org.simantics.layer0.utils.Property;
\r
33 import org.simantics.processeditor.Activator;
\r
34 import org.simantics.processeditor.ProcessResource;
\r
35 import org.simantics.processeditor.actions.InsertComponentAction;
\r
36 import org.simantics.processeditor.actions.InsertEquipmentAction;
\r
37 import org.simantics.processeditor.actions.InsertNozzleAction;
\r
38 import org.simantics.processeditor.actions.RoutePipeAction;
\r
39 import org.simantics.processeditor.common.ControlPointTools;
\r
40 import org.simantics.processeditor.common.PipingRules;
\r
41 import org.simantics.processeditor.dialogs.FloorConfigureDialog;
\r
42 import org.simantics.processeditor.scenegraph.NonVisibleNode;
\r
43 import org.simantics.processeditor.scenegraph.PipeComponentNode;
\r
44 import org.simantics.processeditor.scenegraph.PipeRunNode;
\r
45 import org.simantics.processeditor.scenegraph.PipelineComponentNode;
\r
46 import org.simantics.processeditor.stubs.PipeControlPoint;
\r
47 import org.simantics.processeditor.stubs.PipeRun;
\r
48 import org.simantics.processeditor.stubs.Plant;
\r
49 import org.simantics.processeditor.stubs.Plant3DResource;
\r
50 import org.simantics.processeditor.tools.PlantEditContribution;
\r
51 import org.simantics.processeditor.tools.PlantVisualizationContribution;
\r
52 import org.simantics.proconf.g3d.base.JmeRenderingComponent;
\r
53 import org.simantics.proconf.g3d.base.ScenegraphAdapter;
\r
54 import org.simantics.proconf.g3d.base.ScenegraphAdapterImpl;
\r
55 import org.simantics.proconf.g3d.base.SelectionAdapter;
\r
56 import org.simantics.proconf.g3d.base.ThreeDimensionalEditorBase;
\r
57 import org.simantics.proconf.g3d.common.StructuredResourceSelection;
\r
58 import org.simantics.proconf.g3d.scenegraph.IGeometryNode;
\r
59 import org.simantics.proconf.g3d.scenegraph.IGraphicsNode;
\r
60 import org.simantics.proconf.g3d.scenegraph.ISelectableNode;
\r
61 import org.simantics.proconf.g3d.scenegraph.ParameterizedModelNode;
\r
62 import org.simantics.proconf.g3d.shapes.FloorShape;
\r
63 import org.simantics.proconf.g3d.stubs.G3DNode;
\r
64 import org.simantics.utils.ui.ErrorLogger;
\r
65 import org.simantics.utils.ui.jface.MenuTools;
\r
67 import com.jme.math.Vector3f;
\r
68 import com.jme.scene.Geometry;
\r
71 public class ProcessEditor extends ThreeDimensionalEditorBase {
\r
73 private Resource plantResource = null;
\r
75 private Action configureFloorAction = null;
\r
77 private Geometry floorShape = null;
\r
79 public ProcessEditor(ISessionContext session) {
\r
81 addEditorContribution(new PlantEditContribution(this));
\r
82 addEditorContribution(new PlantVisualizationContribution(this));
\r
85 public ProcessEditor(ISessionContext session,JmeRenderingComponent component) {
\r
86 super(session,component);
\r
87 addEditorContribution(new PlantEditContribution(this));
\r
88 addEditorContribution(new PlantVisualizationContribution(this));
\r
92 protected ScenegraphAdapter createScenegraphAdapter() {
\r
93 return new ProcessEditorAdapter(session,getRenderingComponent());
\r
97 public void createControl(Graph graph,Composite parent) {
\r
98 super.createControl(graph,parent);
\r
100 floorShape = FloorShape.getShape(getRenderingComponent().getDisplaySystem().getRenderer(), 100.f,0.2f);
\r
101 getRenderingComponent().getNoCastRoot().attachChild(floorShape);
\r
102 floorShape.setLocalTranslation(new Vector3f(0.f,-0.01f,0.f));
\r
106 protected void makeActions(Graph graph) {
\r
107 super.makeActions(graph);
\r
109 //actions.add(new ShowTrendsAction(this));
\r
111 configureFloorAction = new Action() {
\r
112 public void run() {
\r
113 FloorConfigureDialog dialog = new FloorConfigureDialog(ProcessEditor.this.parent.getShell());
\r
114 if (dialog.open() == FloorConfigureDialog.CANCEL)
\r
116 if (dialog.isFloorEnabled()) {
\r
117 if (floorShape.getParent() == null)
\r
118 getRenderingComponent().getNoCastRoot().attachChild(floorShape);
\r
120 floorShape.removeFromParent();
\r
122 floorShape.setLocalTranslation(new Vector3f(0.f,(float)dialog.getFloorHeight(),0.f));
\r
126 configureFloorAction.setText("Configure floor");
\r
127 configureFloorAction.setImageDescriptor(Activator.imageDescriptorFromPlugin("fi.vtt.proconf.ode", "icons/silk/shape_align_bottom.png"));
\r
129 // ContextActionFactory extended[] = ContextActionRegistry.getActions("fi.vtt.proconf.shapeeditor.processeditorview");
\r
130 // for (ContextActionFactory c : extended) {
\r
131 // actions.add(c.createAction(this));
\r
135 protected void fillLocalPullDown() {
\r
136 super.fillLocalPullDown();
\r
137 MenuTools.getOrCreate(getMenuID(),"Advanced", menuManager).add(configureFloorAction);
\r
140 protected class ProcessEditorAdapter extends ScenegraphAdapterImpl {
\r
142 public ProcessEditorAdapter(Session session, JmeRenderingComponent component) {
\r
143 super(session, component);
\r
146 private class NormalScenegraphQuery extends ScenegraphQuery {
\r
148 public NormalScenegraphQuery(Resource node) {
\r
154 public void shapeAdded(Graph graph, IGraphicsNode node) {
\r
155 // FIXME : this won't work like in previous ProConf
\r
159 private Map<Resource,PipeRunControlPointQuery> pipeRunQueries = new HashMap<Resource, PipeRunControlPointQuery>();
\r
161 protected ScenegraphQuery newSubnodeListener(G3DNode node) {
\r
162 if (node.isInstanceOf(ProcessResource.plant3Dresource.PipeRun)) {
\r
163 PipeRunControlPointQuery query = new PipeRunControlPointQuery(node.getResource());
\r
164 pipeRunQueries.put(node.getResource(), query);
\r
165 node.getGraph().performQuery(query);
\r
167 return new NormalScenegraphQuery(node.getResource());
\r
172 protected NodePropertyQuery newRootPropertyListener(G3DNode root) {
\r
173 // currently Plant does not have any properties.
\r
177 private class TransformationQuery extends NodeTransformationQuery {
\r
179 public TransformationQuery(Resource res) {
\r
184 public void shapeUpdated(Graph graph, IGraphicsNode shape) {
\r
185 //if (shape instanceof IGeometryNode) {
\r
186 // updateGeometry((IGeometryNode)shape);
\r
188 shape.updateTransform(graph);
\r
194 protected NodeTransformationQuery newTransformationListener(G3DNode root) {
\r
195 return new TransformationQuery(root.getResource());
\r
198 private class NormalNodePropertyQuery extends org.simantics.proconf.g3d.base.ScenegraphAdapterImpl.NodePropertyQuery {
\r
200 public NormalNodePropertyQuery(Resource resource) {
\r
205 public void shapeUpdated(Graph graph,IGraphicsNode shape) {
\r
206 if (shape instanceof IGeometryNode) {
\r
207 updateGeometry((IGeometryNode)shape);
\r
209 shape.updateTransform(graph);
\r
214 protected NodePropertyQuery newPropertyListener(G3DNode node) {
\r
215 return new NormalNodePropertyQuery(node.getResource());
\r
219 protected IGraphicsNode instantiateNode(IGraphicsNode parent,
\r
221 Plant3DResource p3r = ProcessResource.plant3Dresource;
\r
222 IGraphicsNode newNode = null;
\r
224 if (node.isInstanceOf(p3r.Equipment)) {
\r
225 newNode = new ParameterizedModelNode(
\r
226 ProcessEditor.this, parent, node.getGraph(),
\r
227 node.getResource(), p3r.HasGraphics);
\r
228 } else if (node.isInstanceOf(p3r.PipeRun)) {
\r
229 newNode = new PipeRunNode(parent, node.getGraph(), node.getResource());
\r
230 } else if (node.isInstanceOf(p3r.Nozzle)) {
\r
231 newNode = new ParameterizedModelNode(
\r
232 ProcessEditor.this, parent, node.getGraph(),
\r
233 node.getResource(), p3r.HasGraphics);
\r
234 // CodedComponent must be handled first since it uses
\r
235 // hard-coded geometries
\r
236 // TODO : is this really necessary, or could we unify
\r
237 // PipeComponentNode, InlineComponentNode,...
\r
238 } else if (node.isInstanceOf(p3r.CodedComponent)) {
\r
239 newNode = new PipeComponentNode(ProcessEditor.this,
\r
240 parent, node.getGraph(), node.getResource());
\r
241 } else if (node.isInstanceOf(p3r.NonVisibleComponent)) {
\r
242 newNode = new NonVisibleNode(parent, node.getGraph(), node.getResource());
\r
243 } else if (node.isInstanceOf(p3r.PipelineComponent)) {
\r
244 newNode = new PipelineComponentNode(ProcessEditor.this,
\r
245 parent, node.getGraph(), node.getResource());
\r
248 // } else if (node instanceof Shape) // Markers (ar/mobile)
\r
250 // newNode = new ShapeNode(TestProcessEditor.this,parent,node);
\r
251 if (newNode != null) {
\r
252 if (newNode instanceof ISelectableNode)
\r
253 ((ISelectableNode) newNode).setVisible(true);
\r
254 if (newNode instanceof IGeometryNode) {
\r
255 updateGeometry((IGeometryNode) newNode);
\r
259 } catch (Exception e) {
\r
260 ErrorLogger.defaultLogError("Cannot handle node " + node.getResource(), e);
\r
263 ErrorLogger.defaultLogError("Cannot handle node " + node.getResource(), null);
\r
269 * This is used to create elbows and straight pipes to pipeline TODO :
\r
270 * this should be done with rule-engine!
\r
273 * @author Marko Luukkainen
\r
276 protected class PipeRunControlPointQuery extends NodeQuery {
\r
277 private List<Resource> removed = new ArrayList<Resource>();
\r
278 private List<Resource> added = new ArrayList<Resource>();
\r
280 public PipeRunControlPointQuery(Resource r) {
\r
282 if (DEBUG) System.out.println("Created PipeRunControlPointQuery for " + r);
\r
287 protected Object compute2(Graph graph) {
\r
288 PipeRun run = new PipeRun(graph, nodeResource);
\r
289 Collection<IEntity> cps = run
\r
290 .getRelatedObjects(ProcessResource.plant3Dresource.HasControlPoints);
\r
291 List<Resource> res = new ArrayList<Resource>();
\r
292 for (IEntity t : cps)
\r
293 res.add(t.getResource());
\r
298 public boolean updated(Graph graph, Object oldResult,
\r
299 Object newResult) {
\r
304 List<Resource> oldCps = (List<Resource>) oldResult;
\r
305 List<Resource> newCps = (List<Resource>) newResult;
\r
306 if (oldCps == null)
\r
307 oldCps = new ArrayList<Resource>();
\r
309 for (Resource r : oldCps) {
\r
310 if (!newCps.contains(r))
\r
314 for (Resource r : newCps) {
\r
315 if (!oldCps.contains(r))
\r
318 for (Resource r : removed)
\r
319 removeControlPoint(graph, r);
\r
320 for (Resource r : added) {
\r
321 addControlPoint(graph, r);
\r
322 // ControlPointTools.addControlPoint(new
\r
323 // PipeRun(graph,pipeRun), new PipeControlPoint(graph, r));
\r
326 return (added.size() > 0 || removed.size() > 0);
\r
330 public void dispose() {
\r
332 for (ControlPointPropertyQuery q : controlPointPropertyQueries.values())
\r
334 controlPointPropertyQueries.clear();
\r
337 private Map<Resource,ControlPointPropertyQuery> controlPointPropertyQueries = new HashMap<Resource, ControlPointPropertyQuery>();
\r
339 private void addControlPoint(Graph graph, Resource resource) {
\r
340 ControlPointPropertyQuery query = new ControlPointPropertyQuery(resource);
\r
341 graph.performQuery(query);
\r
342 controlPointPropertyQueries.put(resource,query);
\r
345 private void removeControlPoint(Graph graph, Resource resource) {
\r
346 ControlPointPropertyQuery query = controlPointPropertyQueries.remove(resource);
\r
348 ControlPointTools.removeControlPoint(new PipeControlPoint(
\r
354 protected class ControlPointPropertyQuery extends NodeQuery {
\r
355 boolean initialized = false;
\r
357 public ControlPointPropertyQuery(Resource r) {
\r
359 if (DEBUG) System.out.println("Created ControlPointPropertyQuery for " + r);
\r
363 public List<Object> compute2(Graph g) {
\r
364 IEntity t = EntityFactory.create(g,nodeResource);
\r
366 Collection<Property> properties = t.getRelatedProperties(ProcessResource.builtins.HasProperty);
\r
367 List<Object> propertyValues = new ArrayList<Object>();
\r
368 p(properties,propertyValues);
\r
370 return propertyValues;
\r
373 private void p(Collection<Property> properties, List<Object> propertyValues) {
\r
374 for (Property p : properties) {
\r
375 Collection<Property> subProperties = p.getRelatedProperties(p.getGraph().getBuiltins().HasProperty);
\r
376 if (subProperties.size() != 0) {
\r
377 p(subProperties,propertyValues);
\r
380 propertyValues.add(p.getValue());
\r
386 public boolean updated(Graph graph, Object oldResult, Object newResult) {
\r
387 PipingRules.pipeControlPointPositionUpdate(graph, this.nodeResource);
\r
389 //PipingRules.pipeControlPointPositionUpdate(graph, this.nodeResource);
\r
391 initialized = true;
\r
398 protected void removeNode(Resource parent, Resource r) {
\r
399 super.removeNode(parent, r);
\r
400 PipeRunControlPointQuery q = pipeRunQueries.get(r);
\r
406 public void dispose() {
\r
412 protected void pageSelectionChanged(IWorkbenchPart part, ISelection selection) {
\r
413 if (!(selection instanceof StructuredResourceSelection)) {
\r
417 StructuredResourceSelection s = (StructuredResourceSelection) selection;
\r
418 selectionAdapter.setCurrentSelection(s);
\r
419 viewChanged = true;
\r
421 //if (s.getRootSelection() == null) {
\r
422 if (!(part instanceof ProcessEditor)) {
\r
423 //System.out.println("ShapeEditorView.pageSelectionChanged() no root selection");
\r
424 ((ProcessEditorSelectionAdapter)selectionAdapter).setEditorSelection(true);
\r
427 //if (!s.getRootSelection().getResource().getId().equals(plant.getResource().getId())) {
\r
428 ProcessEditor sender = (ProcessEditor)part;
\r
429 if (!sender.getPlantResource().equals(plantResource)) {
\r
430 // System.out.println("ShapeEditorView.pageSelectionChanged() not right group "
\r
431 // + s.getRootSelection().getResource().getId() + " != " + model.getResource().getId());
\r
432 selectionAdapter.setCurrentSelection(new StructuredResourceSelection());
\r
433 ((ProcessEditorSelectionAdapter)selectionAdapter).setEditorSelection(false);
\r
436 selectionAdapter.setEditorSelection();
\r
441 protected void reloadFrom(IEntity thing) {
\r
442 if (plantResource != null) {
\r
443 throw new UnsupportedOperationException("Reloading instantiated viewer not supported");
\r
445 if (thing.isInstanceOf(ProcessResource.plant3Dresource.Plant)) {
\r
446 plantResource = thing.getResource();
\r
447 G3DNode plant = new G3DNode(thing);
\r
448 adapter.setRootNode(plant);
\r
449 //adapter.addOutbound(plant);
\r
450 ControlPointTools.reloadCache(thing.getGraph(),plant.getResource());
\r
452 throw new IllegalArgumentException("Resource is not a plant");
\r
456 public Resource getPlantResource() {
\r
457 return plantResource;
\r
460 public Plant getPlant(Graph g) {
\r
461 return new Plant(g, plantResource);
\r
465 protected SelectionAdapter createSelectionAdapter() {
\r
466 return new ProcessEditorSelectionAdapter(adapter);
\r
469 protected class ProcessEditorSelectionAdapter extends SelectionAdapter {
\r
472 public ProcessEditorSelectionAdapter(ScenegraphAdapter adapter) {
\r
477 public void setEditorSelection() {
\r
478 List<IGraphicsNode> sel = getSelectedObjects();
\r
479 for (IGraphicsNode o : adapter.getNodes())
\r
480 if (o instanceof ISelectableNode) {
\r
481 if (sel.contains(o)) {
\r
482 ((ISelectableNode)o).setSelected(true);
\r
484 ((ISelectableNode)o).setSelected(false);
\r
489 public void setEditorSelection(boolean addShapes) {
\r
491 List<IGraphicsNode> sel = getSelectedObjects();
\r
492 for (IGraphicsNode o : adapter.getNodes())
\r
493 if (o instanceof ISelectableNode) {
\r
494 if (sel.contains(o)) {
\r
495 ((ISelectableNode)o).setSelected(true);
\r
497 ((ISelectableNode)o).setSelected(false);
\r
503 protected void setEditorHighlightSelection() {
\r
504 List<IGraphicsNode> sel = getInteractiveSelectedObjects();
\r
505 for (IGraphicsNode o : adapter.getNodes())
\r
506 if (o instanceof ISelectableNode) {
\r
507 if (sel.contains(o)) {
\r
508 ((ISelectableNode)o).setHighlighted(true);
\r
510 ((ISelectableNode)o).setHighlighted(false);
\r
517 protected void hookDragAndDrop() {
\r
518 super.hookDragAndDrop();
\r
519 dropTarget.addDropListener(new InsertEquipmentAction(this));
\r
520 dropTarget.addDropListener(new InsertNozzleAction(this));
\r
521 dropTarget.addDropListener(new InsertComponentAction(this));
\r
522 dropTarget.addDropListener(new RoutePipeAction(this));
\r
526 public Object getAdapter(Class adapter) {
\r
527 if (adapter == IContentOutlinePage.class) {
\r
528 if (getPlantResource() == null)
\r
530 final PlantStructureOutlinePage page = new PlantStructureOutlinePage(sessionContext,getPlantResource());
\r
532 getSelectionAdapter().addSelectionChangedListener(new ISelectionChangedListener() {
\r
534 public void selectionChanged(SelectionChangedEvent event) {
\r
535 page.setSelection(event.getSelection());
\r
539 parent.getDisplay().asyncExec(new Runnable() {
\r
541 public void run() {
\r
542 page.addSelectionChangedListener(new ISelectionChangedListener() {
\r
544 public void selectionChanged(SelectionChangedEvent event) {
\r
545 selectionAdapter.setSelection(SelectionAdapter.transformSelection(event.getSelection()));
\r