package org.simantics.plant3d.editor;
+import java.util.ArrayList;
import java.util.HashSet;
import java.util.List;
import java.util.Set;
import org.simantics.db.Session;
import org.simantics.db.common.request.ReadRequest;
import org.simantics.db.exception.DatabaseException;
-import org.simantics.g3d.scenegraph.IG3DNode;
import org.simantics.g3d.scenegraph.NodeMap;
+import org.simantics.g3d.scenegraph.RenderListener;
import org.simantics.g3d.scenegraph.base.INode;
import org.simantics.g3d.vtk.action.RemoveAction;
import org.simantics.g3d.vtk.common.HoverHighlighter;
import org.simantics.g3d.vtk.common.NodeSelectionProvider2;
import org.simantics.g3d.vtk.common.SelectionHighlighter;
+import org.simantics.g3d.vtk.common.VTKSelectionItem;
import org.simantics.g3d.vtk.common.VtkView;
import org.simantics.g3d.vtk.shape.vtkShape;
import org.simantics.g3d.vtk.swt.ContextMenuListener;
+import org.simantics.g3d.vtk.swt.FocusAction;
import org.simantics.g3d.vtk.swt.InteractiveVtkComposite;
import org.simantics.g3d.vtk.swt.RotateAction;
import org.simantics.g3d.vtk.swt.TranslateAction;
import org.simantics.plant3d.actions.AddComponentAction;
import org.simantics.plant3d.actions.AddEquipmentAction;
import org.simantics.plant3d.actions.AddNozzleAction;
+import org.simantics.plant3d.actions.RemoveAndSplitAction;
import org.simantics.plant3d.actions.RoutePipeAction;
+import org.simantics.plant3d.actions.TranslateFreeVariableLengthAction;
import org.simantics.plant3d.actions.TranslateInlineAction;
+import org.simantics.plant3d.ontology.Plant3D;
import org.simantics.plant3d.scenegraph.EndComponent;
import org.simantics.plant3d.scenegraph.Equipment;
import org.simantics.plant3d.scenegraph.IP3DNode;
import org.simantics.plant3d.scenegraph.InlineComponent;
import org.simantics.plant3d.scenegraph.Nozzle;
import org.simantics.plant3d.scenegraph.P3DRootNode;
-import org.simantics.plant3d.scenegraph.PipeRun;
import org.simantics.plant3d.scenegraph.PipelineComponent;
import org.simantics.plant3d.scenegraph.SchemaBuilder;
import org.simantics.plant3d.scenegraph.TurnComponent;
import org.simantics.plant3d.scenegraph.controlpoint.ControlPointFactory;
-import org.simantics.plant3d.scenegraph.controlpoint.PipeControlPoint;
import org.simantics.plant3d.scenegraph.controlpoint.PipingRules;
import org.simantics.plant3d.utils.ComponentUtils;
import org.simantics.plant3d.utils.Item;
import vtk.vtkActor;
import vtk.vtkCameraPass;
import vtk.vtkDefaultPass;
+import vtk.vtkGaussianBlurPass;
import vtk.vtkLightsPass;
+import vtk.vtkProp;
import vtk.vtkRenderPassCollection;
import vtk.vtkRenderer;
+import vtk.vtkSSAAPass;
import vtk.vtkSequencePass;
+import vtk.vtkSimpleMotionBlurPass;
public class Plant3DEditor extends ResourceEditorPart {
private InteractiveVtkComposite panel;
private P3DRootNode rootNode;
- private IMapping<Resource,Object> mapping;
+ private IMapping<Resource,INode> mapping;
- private NodeSelectionProvider2<Resource,Object> selectionProvider;
+ protected NodeSelectionProvider2<Resource, INode> selectionProvider;
- private vtkCameraAndSelectorAction cameraAction;
- private TranslateAction translateAction;
- private TranslateInlineAction translateInlineAction;
- private RotateAction rotateAction;
- private RemoveAction removeAction;
- private RoutePipeAction routePipeAction;
- private AddComponentAction addComponentAction;
+
+ protected vtkCameraAndSelectorAction cameraAction;
+ protected FocusAction focusAction;
+ protected TranslateAction translateAction;
+ protected TranslateInlineAction translateInlineAction;
+ protected TranslateFreeVariableLengthAction translateFreeVariableLengthAction;
+ protected RotateAction rotateAction;
+ protected RemoveAction removeAction;
+ protected RemoveAndSplitAction removeSplitAction;
+ protected RoutePipeAction routePipeAction;
+ protected AddComponentAction addComponentAction;
private P3DNodeMap nodeMap;
hookContextMenu();
createScene();
-
- //component.syncPopulate();
new ContextMenuListener(panel, contextMenu);
- cameraAction = new vtkCameraAndSelectorAction(panel);
+ cameraAction = createCameraAction();
panel.setDefaultAction(cameraAction);
panel.useDefaultAction();
panel.setPickType(4);
try {
- ControlPointFactory.preloadCache();
+ ControlPointFactory.preloadCache(getLibraryUri());
ComponentUtils.preloadCache();
} catch (Exception e) {
ExceptionUtils.logAndShowError("Cannot open Plant3D editor",e);
try {
getSession().syncRequest(new ReadRequest() {
- @SuppressWarnings({ "rawtypes", "unchecked" })
@Override
public void run(ReadGraph graph) throws DatabaseException {
+ System.out.println("START PLANT3D LOAD");
PipingRules.setEnabled(false);
- IMappingSchema<Resource, Object> schema = getSchema(graph);
+ IMappingSchema<Resource,INode> schema = getSchema(graph);
mapping = Mappings.createWithListening(schema);
rootNode = (P3DRootNode)mapping.map(graph, input);
// update control points.
// TODO : this should be optimized.
+
try {
P3DUtil.finalizeDBLoad(rootNode);
-
- } catch (Exception e) {
- throw new DatabaseException(e);
- }
- nodeMap = createNodeMap(getSession(), mapping, panel,rootNode);
+ nodeMap = createNodeMap(getSession(), mapping, panel,rootNode);
+
+ } catch (Exception e) {
+ throw new DatabaseException(e);
+ }
+
+ System.out.println("END PLANT3D LOAD");
}
});
throw new RuntimeException("Scenegraph loading failed.");
populate();
- selectionProvider = new NodeSelectionProvider2<Resource,Object>(this,mapping,nodeMap);
+ selectionProvider = createSelectionProvider();
cameraAction.addSelectionChangedListener(selectionProvider);
- cameraAction.addHoverChangedListener(new HoverHighlighter(panel,nodeMap));
- selectionProvider.addSelectionChangedListener(new SelectionHighlighter(panel,nodeMap));
+ cameraAction.addHoverChangedListener(createHoverHighlhighter());
+ selectionProvider.addSelectionChangedListener(createSelectionHighlighter());
getSite().setSelectionProvider(selectionProvider);
getSite().getPage().addPostSelectionListener(selectionProvider);
return;
}
+ createActions();
+
+ }
+
+ protected vtkCameraAndSelectorAction createCameraAction() {
+ return new vtkCameraAndSelectorAction(panel);
+ }
+
+ protected NodeSelectionProvider2<Resource,INode> createSelectionProvider() {
+ return new NodeSelectionProvider2<Resource,INode>(this,mapping,nodeMap);
+ }
+
+ protected HoverHighlighter<Resource> createHoverHighlhighter() {
+ return new HoverHighlighter<>(panel,nodeMap);
+ }
+
+ protected SelectionHighlighter<Resource> createSelectionHighlighter() {
+ return new SelectionHighlighter<Resource>(panel,nodeMap);
+ }
+
+ protected String getLibraryUri() {
+ return Plant3D.URIs.Builtin;
+ }
+
+ protected void createActions() {
+ focusAction = new FocusAction(panel, cameraAction);
translateAction = new TranslateAction(panel,nodeMap);
translateInlineAction = new TranslateInlineAction(panel, nodeMap);
+ translateFreeVariableLengthAction = new TranslateFreeVariableLengthAction(panel, getRootNode());
rotateAction = new RotateAction(panel,nodeMap);
- removeAction = new RemoveAction(nodeMap) {
- public void setNode(IG3DNode node) {
- super.setNode(node);
-
-
- }
- };
+ removeAction = new RemoveAction(nodeMap);
+ removeSplitAction = new RemoveAndSplitAction(nodeMap);
routePipeAction = new RoutePipeAction(panel,rootNode);
- addComponentAction = new AddComponentAction(panel, rootNode);
-
+ addComponentAction = new AddComponentAction(panel, rootNode, getLibraryUri());
}
public void populate() {
@Override
public void run() {
nodeMap.populate();
+ panel.addListener(new RenderListener() {
+
+ @Override
+ public void preRender() {
+
+ }
+
+ @Override
+ public void postRender() {
+ try {
+ P3DUtil.finalizeDBLoad2(rootNode);
+ if (nodeMap.isRangeModified());
+ nodeMap.commit("Load sync");
+ } catch (Exception e) {
+ //throw new DatabaseException(e);
+ }
+ panel.removeListener(this);
+
+ }
+ });
+
}
});
}
- protected IMappingSchema<Resource, Object> getSchema(ReadGraph graph) throws DatabaseException {
- IMappingSchema<Resource,Object> schema = SchemaBuilder.getSchema(graph);
+ protected IMappingSchema<Resource, INode> getSchema(ReadGraph graph) throws DatabaseException {
+ IMappingSchema<Resource,INode> schema = SchemaBuilder.getSchema(graph);
return schema;
}
- protected P3DNodeMap createNodeMap(Session session, IMapping<Resource, Object> mapping, VtkView panel, P3DRootNode rootNode) {
- return new P3DNodeMap(session, mapping, panel,rootNode);
+ protected P3DNodeMap createNodeMap(Session session, IMapping<Resource, INode> mapping, VtkView panel, P3DRootNode rootNode) {
+ return new P3DNodeMap(session, mapping, panel,rootNode);
}
@Override
vtkRenderer ren1 = panel.getRenderer();
boolean multiPass = false;
+ boolean blur = false;
+ boolean ssaa = false;
+ //boolean sobel = true;
+ boolean mblur = false;
if (multiPass) {
vtkLightsPass lightsPass = new vtkLightsPass();
vtkCameraPass cameraPass = new vtkCameraPass();
cameraPass.SetDelegatePass(seq);
- ren1.SetPass(cameraPass);
-
+ if (blur) {
+ vtkGaussianBlurPass blurPass = new vtkGaussianBlurPass();
+ blurPass.SetDelegatePass(cameraPass);
+ ren1.SetPass(blurPass);
+ } else if (ssaa) {
+ vtkSSAAPass ssaaPass = new vtkSSAAPass();
+ ssaaPass.SetDelegatePass(cameraPass);
+ ren1.SetPass(ssaaPass);
+ } else if (mblur) {
+ vtkSimpleMotionBlurPass mBlurPass = new vtkSimpleMotionBlurPass();
+ mBlurPass.SetDelegatePass(cameraPass);
+ ren1.SetPass(mBlurPass);
+// } else if (sobel) {
+// vtkSobelGradientMagnitudePass sobelPass = new vtkSobelGradientMagnitudePass();
+// sobelPass.SetDelegatePass(sobelPass);
+// ren1.SetPass(sobelPass);
+ } else {
+ ren1.SetPass(cameraPass);
+ }
+
}
// ren1.GetRenderWindow().LineSmoothingOn();
// ren1.GetRenderWindow().PointSmoothingOn();
ren1.SetBackground(0.9,0.9,0.9);
ren1.SetGradientBackground(true);
- // vtkActor grid = vtkShape.createGridActor(8,1.0,1|2|4);
- vtkActor grid = vtkShape.createGridActor(8,1.0, 2 );
- grid.SetPickable(0);
- ren1.AddActor(grid);
- panel.addDeletable(grid);
-
- AxesDisplay axesDisplay = new AxesDisplay(panel);
- axesDisplay.show();
+ // vtkActor grid = vtkShape.createGridActor(8,1.0,1|2|4);
+ vtkActor grid = vtkShape.createGridActor(8,1.0, 2 );
+ grid.SetPickable(0);
+ ren1.AddActor(grid);
+ panel.addDeletable(grid);
+
+ AxesDisplay axesDisplay = new AxesDisplay(panel);
+ axesDisplay.show();
}
protected Menu contextMenu;
protected void hookContextMenu() {
- MenuManager menuMgr = new MenuManager("#PopupMenu");
- menuMgr.setRemoveAllWhenShown(true);
- menuMgr.addMenuListener(new IMenuListener() {
- public void menuAboutToShow(IMenuManager manager) {
- createContextMenu(manager);
- }
- });
+ MenuManager menuMgr = new MenuManager("#PopupMenu");
+ menuMgr.setRemoveAllWhenShown(true);
+ menuMgr.addMenuListener(new IMenuListener() {
+ public void menuAboutToShow(IMenuManager manager) {
+ createContextMenu(manager);
+ }
+ });
- contextMenu = menuMgr.createContextMenu(parent);
- }
+ contextMenu = menuMgr.createContextMenu(parent);
+ }
protected void createContextMenu(IMenuManager m) {
- List<IG3DNode> selected = selectionProvider.getSelectedNodes();
+ List<INode> selected = selectionProvider.getSelectedNodes();
+ List<VTKSelectionItem<Resource>> selectedItems = selectionProvider.getSelectedItems();
+ if (selectedItems.size() > 0) {
+ List<vtkProp> props = new ArrayList<>();
+ for (VTKSelectionItem<Resource> i : selectedItems) {
+ vtkProp p = (vtkProp)i.getAdapter(vtkProp.class);
+ if (p != null)
+ props.add(p);
+ }
+ if (props.size() > 0) {
+ focusAction.setProps(props);
+ m.add(focusAction);
+ }
+ }
try {
if (selected.size() == 0) {
- for (Item eq : P3DUtil.getEquipments()) {
+ for (Item eq : P3DUtil.getEquipments(getLibraryUri())) {
m.add(new AddEquipmentAction(rootNode, eq));
}
} else if (selected.size() == 1) {
if (node instanceof Equipment) {
m.add(translateAction);
m.add(rotateAction);
- for (Item eq : P3DUtil.getNozzles()) {
+ for (Item eq : P3DUtil.getNozzles(getLibraryUri())) {
AddNozzleAction add = new AddNozzleAction(rootNode, eq);
add.setEquipment((Equipment)node);
m.add(add);
}
+ m.add(removeAction);
} else if (node instanceof Nozzle) {
- Nozzle nozzle = (Nozzle)node;
- if (!nozzle.isFixed()) {
- m.add(translateAction);
- m.add(rotateAction);
- }
+ Nozzle nozzle = (Nozzle)node;
+ if (!nozzle.isFixed()) {
+ m.add(translateAction);
+ m.add(rotateAction);
+ }
m.add(routePipeAction);
routePipeAction.setComponent(nozzle);
routePipeAction.setEnabled(nozzle.getNext() == null && nozzle.getPrevious() == null);
m.add(addComponentAction);
addComponentAction.setComponent(nozzle);
+ m.add(removeAction);
} else if (node instanceof TurnComponent) {
m.add(translateAction);
TurnComponent component = (TurnComponent)node;
routePipeAction.setEnabled(component.getNext() == null || component.getPrevious() == null);
m.add(addComponentAction);
addComponentAction.setComponent(component);
+ m.add(removeAction);
+ m.add(removeSplitAction);
+ removeSplitAction.setNode(node);
} else if (node instanceof EndComponent) {
m.add(translateAction);
m.add(addComponentAction);
addComponentAction.setComponent((PipelineComponent)node);
+ m.add(removeAction);
} else if (node instanceof InlineComponent) {
//m.add(translateInlineAction);
InlineComponent component = (InlineComponent)node;
- m.add(translateInlineAction);
+ if (component.isVariableLength())
+ m.add(translateFreeVariableLengthAction);
+ else
+ m.add(translateInlineAction);
m.add(routePipeAction);
routePipeAction.setComponent(component);
m.add(addComponentAction);
addComponentAction.setComponent(component);
+ m.add(removeAction);
+ m.add(removeSplitAction);
+ removeSplitAction.setNode(node);
+ } else {
+ m.add(removeAction);
}
- m.add(removeAction);
+
translateAction.setNode(node);
translateInlineAction.setNode(node);
+ translateFreeVariableLengthAction.setNode(node);
rotateAction.setNode(node);
removeAction.setNode(node);
}
} catch (DatabaseException e) {
- ExceptionUtils.logAndShowError(e);
- }
+ ExceptionUtils.logAndShowError(e);
+ }
}
private IContentOutlinePage createOutline() {
return outlinePage;
}
- @SuppressWarnings("rawtypes")
@Override
- public Object getAdapter(Class adapter) {
- if (IPropertyPage.class.equals(adapter))
- return new StandardPropertyPage(getSite(),getPropertyContexts());
- if (IContentOutlinePage.class.equals(adapter)) {
- return createOutline();
+ public <T> T getAdapter(Class<T> adapter) {
+ if (adapter.isAssignableFrom(IPropertyPage.class))
+ return adapter.cast(new StandardPropertyPage(getSite(),getPropertyContexts()));
+ if (adapter.isAssignableFrom(ISelectionProvider.class))
+ return adapter.cast(selectionProvider);
+ if (adapter.isAssignableFrom(IContentOutlinePage.class)) {
+ return adapter.cast(createOutline());
}
- if (NodeMap.class.equals(adapter)) {
- return nodeMap;
+ if (adapter.isAssignableFrom(NodeMap.class)) {
+ return adapter.cast(nodeMap);
}
- if (INode.class.equals(adapter)) {
- return rootNode;
+ if (adapter.isAssignableFrom(INode.class)) {
+ return adapter.cast(rootNode);
}
- if (IMapping.class.equals(adapter)) {
- return mapping;
+ if (adapter.isAssignableFrom(IMapping.class)) {
+ return adapter.cast(mapping);
}
-// if (InteractiveVtkPanel.class.equals(adapter)) {
-// return panel;
+// if (adapter.isAssignableFrom(InteractiveVtkPanel.class)) {
+// return adapter.cast(panel);
// }
- if (VtkView.class.equals(adapter))
- return panel;
- if (ISelectionProvider.class.equals(adapter))
- return selectionProvider;
+ if (adapter.isAssignableFrom(VtkView.class))
+ return adapter.cast(panel);
return super.getAdapter(adapter);
}
return result;
}
+ public InteractiveVtkComposite getPanel() {
+ return panel;
+ }
+
public P3DRootNode getRootNode() {
return rootNode;
}
- public IMapping<Resource, Object> getMapping() {
+ public IMapping<Resource, INode> getMapping() {
return mapping;
}
+
+ public P3DNodeMap getNodeMap() {
+ return nodeMap;
+ }
}
Code Review / simantics / 3d.git / blobdiff