Add participant for visualizing district finder search results

[simantics/district.git] / org.simantics.district.network.ui / src / org / simantics / district / network / ui / DistrictDiagramViewer.java

diff --git a/org.simantics.district.network.ui/src/org/simantics/district/network/ui/DistrictDiagramViewer.java b/org.simantics.district.network.ui/src/org/simantics/district/network/ui/DistrictDiagramViewer.java
index 5b67fa519f418736b5a3cae28a6659ed670fbca9..0b0965aca92e75d65f2c4546c30a3445bb53c54f 100644 (file)
--- a/org.simantics.district.network.ui/src/org/simantics/district/network/ui/DistrictDiagramViewer.java
+++ b/org.simantics.district.network.ui/src/org/simantics/district/network/ui/DistrictDiagramViewer.java
@@ -9,6 +9,9 @@ import java.util.concurrent.TimeUnit;
 import java.util.function.Consumer;
 import java.util.function.Supplier;
 
+import org.eclipse.e4.core.contexts.IEclipseContext;
+import org.eclipse.e4.core.services.events.IEventBroker;
+import org.eclipse.ui.PlatformUI;
 import org.simantics.datatypes.literal.RGB;
 import org.simantics.db.ReadGraph;
 import org.simantics.db.Resource;
@@ -23,6 +26,7 @@ import org.simantics.district.network.ui.internal.Activator;
 import org.simantics.district.network.ui.nodes.DistrictRenderingPreparationNode;
 import org.simantics.district.network.ui.nodes.DistrictSelectionNode;
 import org.simantics.district.network.ui.participants.DNPointerInteractor;
+import org.simantics.district.network.ui.participants.DistrictFinderVisualisationParticipant;
 import org.simantics.district.network.ui.participants.DynamicVisualisationContributionsParticipant;
 import org.simantics.district.network.ui.participants.MapRulerPainter;
 import org.simantics.district.network.visualisations.DynamicVisualisations;
@@ -83,8 +87,19 @@ public class DistrictDiagramViewer extends DiagramViewer {
         DistrictRenderingPreparationNode prepNode = new DistrictRenderingPreparationNode();
         prepNode.setZIndex(Integer.MIN_VALUE / 2);
         spatialRoot.addNode("districtRenderingPrepareNode", prepNode);
+
+        
+        IEclipseContext workbenchContext = getWorkbenchContext();
+        IEventBroker eventBroker = workbenchContext.get(IEventBroker.class);
+        DistrictFinderVisualisationParticipant districtFinderVisualisationParticipant = new DistrictFinderVisualisationParticipant(eventBroker);
+        ctx.add(districtFinderVisualisationParticipant);
+        
     }
-    
+
+    public static IEclipseContext getWorkbenchContext(){
+        return PlatformUI.getWorkbench().getService(IEclipseContext.class);
+    }
+
     protected String getPopupId() {
         return "#DistrictDiagramPopup";
     }