]> gerrit.simantics Code Review - simantics/district.git/blobdiff - org.simantics.district.network.ui/src/org/simantics/district/network/ui/nodes/NetworkDrawingNode.java
Allow starting/ending of manual network creation to vertices only
[simantics/district.git] / org.simantics.district.network.ui / src / org / simantics / district / network / ui / nodes / NetworkDrawingNode.java
index b2121bf70128b358aada26990e75deae7a93b100..524bf7221e42133f76b0644130760af8fd43842a 100644 (file)
@@ -17,9 +17,12 @@ import org.simantics.db.Resource;
 import org.simantics.db.WriteGraph;
 import org.simantics.db.common.request.WriteRequest;
 import org.simantics.db.exception.DatabaseException;
+import org.simantics.db.request.Write;
 import org.simantics.diagram.elements.DiagramNodeUtil;
 import org.simantics.diagram.ui.DiagramModelHints;
+import org.simantics.district.network.DistrictNetworkUtil;
 import org.simantics.district.network.ModelledCRS;
+import org.simantics.district.network.ontology.DistrictNetworkResource;
 import org.simantics.district.network.ui.DNEdgeBuilder;
 import org.simantics.district.network.ui.NetworkDrawingParticipant;
 import org.simantics.g2d.canvas.Hints;
@@ -38,11 +41,17 @@ import org.simantics.scenegraph.utils.NodeUtil;
 
 public class NetworkDrawingNode extends G2DNode {
 
+    static class DrawingNode {
+
+        private List<Point2D> routeNodes = new ArrayList<>();
+    }
+
     private static final long serialVersionUID = -3475301184009620573L;
     
     private Point2D currentMousePos = null;
-    
-    private List<Point2D> nodes = new ArrayList<>();
+
+    private List<DrawingNode> nodes = new ArrayList<>();
+    private DrawingNode currentRouteNode = null;
 
     private Resource diagramResource;
 
@@ -82,35 +91,41 @@ public class NetworkDrawingNode extends G2DNode {
     public void render(Graphics2D g2d) {
         if (nodes.isEmpty())
             return;
-        
-        Path2D path = new Path2D.Double();
-        Iterator<Point2D> nodeIter = nodes.iterator();
-        if (nodeIter.hasNext()) {
-            Point2D node = nodeIter.next();
-            path.moveTo(node.getX(), node.getY());
-        }
-        while (nodeIter.hasNext()) {
-            Point2D node = nodeIter.next();
-            path.lineTo(node.getX(), node.getY());
-        }
-        if (currentMousePos != null)
-            path.lineTo(currentMousePos.getX(), currentMousePos.getY());
-        
+
         Color old = g2d.getColor();
         Stroke oldStroke = g2d.getStroke();
-        
-        if (DASHED_STROKE != null) {
-            if (scaleStroke && DASHED_STROKE instanceof BasicStroke) {
-                BasicStroke bs = GeometryUtils.scaleStroke(DASHED_STROKE, (float) (1.0 / GeometryUtils.getScale(g2d.getTransform())));
-                g2d.setStroke(bs);
-            } else {
-                g2d.setStroke(DASHED_STROKE);
+
+        Iterator<DrawingNode> dnodeIterator = nodes.iterator();
+        while (dnodeIterator.hasNext()) {
+            Path2D path = new Path2D.Double();
+            DrawingNode dnode = dnodeIterator.next();
+            Iterator<Point2D> nodeIter = dnode.routeNodes.iterator();
+            if (nodeIter.hasNext()) {
+                Point2D node = nodeIter.next();
+                path.moveTo(node.getX(), node.getY());
             }
-        }
-        
-        g2d.setColor(BLUE_ALPHA);
+            while (nodeIter.hasNext()) {
+                Point2D node = nodeIter.next();
+                path.lineTo(node.getX(), node.getY());
+            }
+            if (!dnodeIterator.hasNext()) {
+                if (currentMousePos != null)
+                    path.lineTo(currentMousePos.getX(), currentMousePos.getY());
+            }
+            
+            if (DASHED_STROKE != null) {
+                if (scaleStroke && DASHED_STROKE instanceof BasicStroke) {
+                    BasicStroke bs = GeometryUtils.scaleStroke(DASHED_STROKE, (float) (1.0 / GeometryUtils.getScale(g2d.getTransform())));
+                    g2d.setStroke(bs);
+                } else {
+                    g2d.setStroke(DASHED_STROKE);
+                }
+            }
+            
+            g2d.setColor(BLUE_ALPHA);
 
-        g2d.draw(path);
+            g2d.draw(path);
+        }
         
         g2d.setStroke(oldStroke);
         g2d.setColor(old);
@@ -131,35 +146,39 @@ public class NetworkDrawingNode extends G2DNode {
         // nodes to path2d
         IToolMode mode = getToolMode();
         if (mode == Hints.CONNECTTOOL || e.hasAnyModifier(MouseEvent.ALT_MASK | MouseEvent.ALT_GRAPH_MASK)) {
-            Point2D start = null;
-            Point2D end = null;
-            Iterator<Point2D> nodeIter = nodes.iterator();
-            while (nodeIter.hasNext()) {
-                if (end == null) {
-                    start = nodeIter.next();
-                    if (!nodeIter.hasNext()) {
-                        break;
+            // ok, new routenode starts from here
+            Point2D localPos = NodeUtil.worldToLocal(this, e.controlPosition, new Point2D.Double());
+            Point2D.Double pos = new Point2D.Double(localPos.getX(), localPos.getY());
+            if (currentRouteNode != null) {
+                //currentRouteNode.routeNodes.add(pos);
+                currentRouteNode = new DrawingNode();
+                currentRouteNode.routeNodes.add(pos);
+                nodes.add(currentRouteNode);
+            } else {
+                // ok, this must be creation of dh_point
+                double scale = getTransform().getScaleY();
+                double x = ModelledCRS.xToLongitude(pos.getX() / scale);
+                double y = ModelledCRS.yToLatitude(-pos.getY() / scale);
+                Simantics.getSession().asyncRequest(new Write() {
+                    
+                    @Override
+                    public void perform(WriteGraph graph) throws DatabaseException {
+                        graph.markUndoPoint();
+                        Resource defaultMapping = graph.getSingleObject(diagramResource, DistrictNetworkResource.getInstance(graph).VertexDefaultMapping);
+                        DistrictNetworkUtil.createVertex(graph, diagramResource, new double[] { x, y }, 0, defaultMapping); // TODO: elevation can be fetched from e.g. elevation API
                     }
-                } else {
-                    start = end;
-                }
-                
-                end = nodeIter.next();
-                
-                createEdge(start, end);
+                });
             }
-            
-            nodes.clear();
-            committed = true;
-            
             repaint();
-            
             return true;
         }
         return super.mouseDoubleClicked(e);
     }
 
-    private void createEdge(Point2D start, Point2D end) {
+    private void createEdge(DrawingNode node) {
+        
+        Point2D start = node.routeNodes.get(0);
+        Point2D end = node.routeNodes.get(node.routeNodes.size() - 1);
         
         double currentPadding = DistrictNetworkVertexNode.width;
         AffineTransform test = getTransform();
@@ -188,12 +207,21 @@ public class NetworkDrawingNode extends G2DNode {
         double[] startCoords = new double[] { startLon, startLat };
         double[] endCoords = new double[] { endLon, endLat };
         
+        double[] detailedGeometryCoords = new double[node.routeNodes.size() * 2];
+        int i = 0;
+        for (Point2D p : node.routeNodes) {
+            double lat = ModelledCRS.yToLatitude(-p.getY() / scaleY);
+            double lon = ModelledCRS.xToLongitude(p.getX() / scaleX);
+            detailedGeometryCoords[i++] = lon;
+            detailedGeometryCoords[i++] = lat;
+        }
+        
         DNEdgeBuilder builder = new DNEdgeBuilder(diagramResource, diagram);
         Simantics.getSession().asyncRequest(new WriteRequest() {
             
             @Override
             public void perform(WriteGraph graph) throws DatabaseException {
-                builder.create(graph, startCoords, 0, endCoords, 0, padding);
+                builder.create(graph, startCoords, 0, endCoords, 0, detailedGeometryCoords, padding);
             }
         });
         
@@ -212,14 +240,35 @@ public class NetworkDrawingNode extends G2DNode {
                 nodes.remove(nodes.size() - 1);
             } else if (e.button == MouseEvent.LEFT_BUTTON) {
                 Point2D localPos = NodeUtil.worldToLocal(this, e.controlPosition, new Point2D.Double());
-                nodes.add(new Point2D.Double(localPos.getX(), localPos.getY()));
+                if (currentRouteNode == null && canStartEdge(localPos)) {
+                    // ok, we can start from here
+                    currentRouteNode = new DrawingNode();
+                    currentRouteNode.routeNodes.add(new Point2D.Double(localPos.getX(), localPos.getY()));
+                    nodes.add(currentRouteNode);
+                } else if (currentRouteNode != null && canStartEdge(localPos)) {
+                    // let's commit our new routenode
+                    currentRouteNode.routeNodes.add(new Point2D.Double(localPos.getX(), localPos.getY()));
+                    Iterator<DrawingNode> nodeIter = nodes.iterator();
+                    while (nodeIter.hasNext()) {
+                        createEdge(nodeIter.next());
+                    }
+                    currentRouteNode = null;
+                    nodes.clear();
+                    committed = true;
+                } else if (currentRouteNode != null) {
+                    currentRouteNode.routeNodes.add(new Point2D.Double(localPos.getX(), localPos.getY()));
+                }
             }
             repaint();
             return true;
         }
         return super.mouseClicked(e);
     }
-    
+
+    private boolean canStartEdge(Point2D currentPos) {
+        return participant.isHoveringOverNode(currentPos);
+    }
+
     private IToolMode getToolMode() {
         return participant.getHint(Hints.KEY_TOOL);
     }
@@ -249,6 +298,7 @@ public class NetworkDrawingNode extends G2DNode {
     @Override
     protected boolean keyPressed(KeyPressedEvent e) {
         if (e.keyCode == java.awt.event.KeyEvent.VK_ESCAPE) {
+            currentRouteNode = null;
             nodes.clear();
             repaint();
             return true;