X-Git-Url: https://gerrit.simantics.org/r/gitweb?a=blobdiff_plain;f=org.simantics.plant3d%2Fsrc%2Forg%2Fsimantics%2Fplant3d%2Feditor%2FPlant3DEditor.java;h=273e1062d15ae6e805fbf1507007000df30a6673;hb=refs%2Fchanges%2F93%2F3393%2F1;hp=8387650bb68d5cf7410a53862a2963580d5885bc;hpb=46c28cc32944d0f73c5ab4062bd6b0b232143be7;p=simantics%2F3d.git

diff --git a/org.simantics.plant3d/src/org/simantics/plant3d/editor/Plant3DEditor.java b/org.simantics.plant3d/src/org/simantics/plant3d/editor/Plant3DEditor.java
index 8387650b..273e1062 100644
--- a/org.simantics.plant3d/src/org/simantics/plant3d/editor/Plant3DEditor.java
+++ b/org.simantics.plant3d/src/org/simantics/plant3d/editor/Plant3DEditor.java
@@ -132,7 +132,7 @@ public class Plant3DEditor extends ResourceEditorPart {
 		
 		new ContextMenuListener(panel, contextMenu);
 		
-		cameraAction = new vtkCameraAndSelectorAction(panel);	
+		cameraAction = createCameraAction();
 		panel.setDefaultAction(cameraAction);
 		panel.useDefaultAction();
 		panel.setPickType(4);
@@ -171,12 +171,12 @@ public class Plant3DEditor extends ResourceEditorPart {
 				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);
@@ -208,6 +208,22 @@ public class Plant3DEditor extends ResourceEditorPart {
 		
 	}
 	
+	protected vtkCameraAndSelectorAction createCameraAction() {
+	    return new vtkCameraAndSelectorAction(panel); 
+	}
+	
+	protected NodeSelectionProvider2<Resource,Object> createSelectionProvider() {
+	    return new NodeSelectionProvider2<Resource,Object>(this,mapping,nodeMap);
+	}
+	
+	protected HoverHighlighter createHoverHighlhighter() {
+	    return new HoverHighlighter(panel,nodeMap);
+	}
+	
+	protected SelectionHighlighter<INode> createSelectionHighlighter() {
+	    return new SelectionHighlighter<INode>(panel,nodeMap);
+	}
+	
 	protected void createActions() {
 		translateAction = new TranslateAction(panel,nodeMap);
 		translateInlineAction = new TranslateInlineAction(panel, nodeMap);
@@ -433,6 +449,10 @@ public class Plant3DEditor extends ResourceEditorPart {
 		return result;
 	}
 	
+	public InteractiveVtkComposite getPanel() {
+        return panel;
+    }
+	
 	public P3DRootNode getRootNode() {
 		return rootNode;
 	}
@@ -440,4 +460,8 @@ public class Plant3DEditor extends ResourceEditorPart {
 	public IMapping<Resource, Object> getMapping() {
 		return mapping;
 	}
+	
+	public P3DNodeMap getNodeMap() {
+        return nodeMap;
+    }
 }