Changeset 33440 in osm for applications/editors/josm/plugins/pt_assistant/src/org

Timestamp:

2017-07-12T11:29:19+02:00 (7 years ago)

Author:

giackserva

Message:

[pt_assistant] #josm14977 - sort also same way stops

Location:

applications/editors/josm/plugins/pt_assistant/src/org/openstreetmap/josm/plugins/pt_assistant

Files:

• actions/SortPTStopsAction.java (modified) (4 diffs)
• data/PTStop.java (modified) (1 diff)

applications/editors/josm/plugins/pt_assistant/src/org/openstreetmap/josm/plugins/pt_assistant/actions/SortPTStopsAction.java

@@ -20 +20 @@
 import org.openstreetmap.josm.actions.JosmAction;
 import org.openstreetmap.josm.actions.relation.DownloadSelectedIncompleteMembersAction;
+import org.openstreetmap.josm.command.ChangeCommand;
+import org.openstreetmap.josm.command.Command;
+import org.openstreetmap.josm.data.coor.EastNorth;
+import org.openstreetmap.josm.data.osm.Node;
 import org.openstreetmap.josm.data.osm.OsmPrimitive;
 import org.openstreetmap.josm.data.osm.OsmPrimitiveType;
@@ -32 +36 @@
 import org.openstreetmap.josm.tools.Utils;
 
+/**
+ * Sorts the stop positions in a PT route according to the assigned ways
+ *
+ * @author giacomo
+ *
+ */
 public class SortPTStopsAction extends JosmAction {
 
@@ -77 +87 @@
 
     private void continueAfterDownload(Relation rel) {
-        List<RelationMember> members = rel.getMembers();
-
+        Main.main.undoRedo.add(getSortPTStopCommand(rel));
+    }
+
+    public Command getSortPTStopCommand(Relation rel) {
+        Relation newRel = new Relation(rel);
+        List<RelationMember> members = newRel.getMembers();
         for (int i = 0; i < members.size(); i++) {
-            rel.removeMember(0);
-        }
-
+            newRel.removeMember(0);
+        }
         members = new RelationSorter().sortMembers(members);
 
@@ -130 +143 @@
         });
 
-        wayMembers.forEach(wm -> {
-            if (wm.getType() != OsmPrimitiveType.WAY)
-                return;
-            List<PTStop> stps = wayStop.get(wm.getWay());
-            if (stps == null)
-                return;
-            stps.forEach(stop -> {
-                if (stop != null) {
-                    if (stop.getStopPositionRM() != null)
-                        rel.addMember(stop.getStopPositionRM());
-                    if (stop.getPlatformRM() != null)
-                        rel.addMember(stop.getPlatformRM());
+        Way prev = null;
+        for (RelationMember wm : wayMembers) {
+            if (wm.getType() == OsmPrimitiveType.WAY) {
+                Way curr = wm.getWay();
+                List<PTStop> stps = wayStop.get(curr);
+                if (stps != null) {
+                    if (stps.size() > 1)
+                        stps = sortSameWayStops(stps, curr, prev);
+                    stps.forEach(stop -> {
+                        if (stop != null) {
+                            if (stop.getStopPositionRM() != null)
+                                newRel.addMember(stop.getStopPositionRM());
+                            if (stop.getPlatformRM() != null)
+                                newRel.addMember(stop.getPlatformRM());
+                        }
+                    });
                 }
-            });
-        });
-
-        wayMembers.forEach(rel::addMember);
+                prev = curr;
+            }
+        }
+
+        wayMembers.forEach(newRel::addMember);
+
+        return new ChangeCommand(rel, newRel);
+    }
+
+    private List<PTStop> sortSameWayStops(List<PTStop> stps, Way way, Way prev) {
+        Map<Node, List<PTStop>> closeNodes = new HashMap<>();
+        List<PTStop> noLocationStops = new ArrayList<>();
+        List<Node> nodes = way.getNodes();
+        for (PTStop stop : stps) {
+            Node closest = findClosestNode(stop, nodes);
+            if (closest == null) {
+                noLocationStops.add(stop);
+                continue;
+            }
+            if (!closeNodes.containsKey(closest)) {
+                closeNodes.put(closest, new ArrayList<>());
+            }
+            closeNodes.get(closest).add(stop);
+        }
+
+        boolean reverse = prev.firstNode().equals(way.lastNode())
+                || prev.lastNode().equals(way.lastNode());
+
+        if (reverse)
+            Collections.reverse(nodes);
+
+        List<PTStop> ret = new ArrayList<>();
+        for (int i = 0; i < nodes.size(); i++) {
+            Node n = nodes.get(i);
+            Node prevNode = i > 0 ? nodes.get(i - 1) : n;
+            List<PTStop> stops = closeNodes.get(n);
+            if (stops != null) {
+                if (stops.size() > 1) {
+                    stops.sort((s1, s2) -> {
+                        Double d1 = stopEastNorth(s1).distance(prevNode.getEastNorth());
+                        Double d2 = stopEastNorth(s2).distance(prevNode.getEastNorth());
+                        return d1.compareTo(d2);
+                    });
+                }
+                stops.forEach(ret::add);
+            }
+        }
+
+        ret.addAll(noLocationStops);
+        return ret;
+    }
+
+    private Node findClosestNode(PTStop stop, List<Node> nodes) {
+        EastNorth stopEN = stopEastNorth(stop);
+        if (stopEN == null)
+            return null;
+        double minDist = Double.MAX_VALUE;
+        Node closest = null;
+        for (Node node : nodes) {
+            double dist = node.getEastNorth().distance(stopEN);
+            if (dist < minDist) {
+                minDist = dist;
+                closest = node;
+            }
+        }
+        return closest;
+    }
+
+    private EastNorth stopEastNorth(PTStop stop) {
+        if (stop.getStopPosition() != null)
+            return stop.getStopPosition().getEastNorth();
+        OsmPrimitive prim = stop.getPlatform();
+        if (prim.getType() == OsmPrimitiveType.WAY)
+            return ((Way) prim).firstNode().getEastNorth();
+        else if (prim.getType() == OsmPrimitiveType.NODE)
+            return ((Node) prim).getEastNorth();
+        else
+            return null;
     }
 

applications/editors/josm/plugins/pt_assistant/src/org/openstreetmap/josm/plugins/pt_assistant/data/PTStop.java

@@ -150 +150 @@
 
         Collection<Node> allNodes = platform.getDataSet().getNodes();
+        String platName = platform.get("name");
         for (Node currentNode : allNodes) {
-            if (platformBBox.bounds(currentNode.getBBox()) && currentNode.hasTag("public_transport", "stop_position")) {
-                potentialStopPositions.add(currentNode);
+            String nodeName = currentNode.get("name");
+            if (platformBBox.bounds(currentNode.getBBox())
+                    && currentNode.hasTag("public_transport", "stop_position")
+                    && ((platName == null || nodeName == null)
+                        || platName.equals(nodeName))) {
+                    potentialStopPositions.add(currentNode);
             }
         }