Changeset 33415 in osm for applications/editors/josm/plugins/pt_assistant/src

Timestamp:

2017-06-28T14:42:16+02:00 (8 years ago)

Author:

giackserva

Message:

[pt_assistant] #josm14933 - supports undoredo

File:

• applications/editors/josm/plugins/pt_assistant/src/org/openstreetmap/josm/plugins/pt_assistant/actions/SplitRoundaboutAction.java (modified) (10 diffs)

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

@@ -24 +24 @@
 import org.openstreetmap.josm.actions.SplitWayAction.SplitWayResult;
 import org.openstreetmap.josm.actions.downloadtasks.DownloadOsmTask;
+import org.openstreetmap.josm.command.ChangeCommand;
+import org.openstreetmap.josm.command.Command;
+import org.openstreetmap.josm.command.SequenceCommand;
 import org.openstreetmap.josm.data.Bounds;
 import org.openstreetmap.josm.data.osm.BBox;
@@ -47 +50 @@
 public class SplitRoundaboutAction extends JosmAction {
 
-    private static final String actionName = "Split Roundabout";
+    private static final String ACTION_NAME = "Split Roundabout";
     private static final long serialVersionUID = 8912249304286025356L;
 
@@ -54 +57 @@
      */
     public SplitRoundaboutAction() {
-        super(actionName, "icons/splitroundabout", actionName, null, true);
+        super(ACTION_NAME, "icons/splitroundabout", ACTION_NAME, null, true);
     }
 
@@ -99 +102 @@
         Map<Relation, List<Integer>> savedPositions = getSavedPositions(roundabout);
 
+        //remove the roundabout from each relation
+        Main.main.undoRedo.add(getRemoveRoundaboutFromRelationsCommand(roundabout));
+
         //split the roundabout on the designed nodes
         List<Node> splitNodes = getSplitNodes(roundabout);
         SplitWayResult result = SplitWayAction.split(getLayerManager().getEditLayer(),
                         roundabout, splitNodes, Collections.emptyList());
-        result.getCommand().executeCommand();
+        Main.main.undoRedo.add(result.getCommand());
         Collection<Way> splitWays = result.getNewWays();
         splitWays.add(result.getOriginalWay());
 
         //update the relations.
-        updateRelations(savedPositions, splitNodes, splitWays);
-    }
-
-    public void updateRelations(Map<Relation, List<Integer>> savedPositions,
+        Main.main.undoRedo.add(getUpdateRelationsCommand(savedPositions, splitNodes, splitWays));
+    }
+
+    public Command getUpdateRelationsCommand(Map<Relation,
+            List<Integer>> savedPositions,
             List<Node> splitNodes, Collection<Way> splitWays) {
+
+        Map<Relation, Relation> changingRelations =
+                updateRelations(savedPositions, splitNodes, splitWays);
+
+        List<Command> commands = new ArrayList<>();
+        changingRelations.forEach((oldR, newR) ->
+            commands.add(new ChangeCommand(oldR, newR)));
+
+        return new SequenceCommand("Updating Relations for SplitRoundabout", commands);
+    }
+
+    private Map<Relation, Relation> updateRelations(Map<Relation,
+            List<Integer>> savedPositions,
+            List<Node> splitNodes, Collection<Way> splitWays) {
+        Map<Relation, Relation> changingRelation = new HashMap<>();
         Map<Relation, Integer> memberOffset = new HashMap<>();
         savedPositions.forEach((r, positions) ->
             positions.forEach(i -> {
 
+                if(!changingRelation.containsKey(r))
+                    changingRelation.put(r, new Relation(r));
+
+                Relation c = changingRelation.get(r);
+
                 if(!memberOffset.containsKey(r))
                     memberOffset.put(r, 0);
                 int offset = memberOffset.get(r);
 
-                Pair<Way, Way> entryExitWays= getEntryExitWays(r, i + offset);
+                Pair<Way, Way> entryExitWays= getEntryExitWays(c, i + offset);
                 Way entryWay = entryExitWays.a;
                 Way exitWay = entryExitWays.b;
@@ -141 +168 @@
 
                 while(!curr.lastNode().equals(exitNode)) {
-                    r.addMember(i + offset++, new RelationMember(null, curr));
+                    c.addMember(i + offset++, new RelationMember(null, curr));
                     parents = curr.lastNode().getParentWays();
                     parents.remove(curr);
@@ -147 +174 @@
                     curr = parents.get(0);
                 }
-                r.addMember(i + offset++, new RelationMember(null, curr));
+                c.addMember(i + offset++, new RelationMember(null, curr));
                 memberOffset.put(r, offset);
             }));
+        return changingRelation;
     }
 
@@ -184 +212 @@
             parents.remove(roundabout);
             for(Way parent: parents) {
-                for(OsmPrimitive prim : parent.getReferrers()) {
-                    if(prim.getType() == OsmPrimitiveType.RELATION &&
-                            RouteUtils.isPTRoute((Relation) prim))
+                for(Relation r : OsmPrimitive.getFilteredList(
+                        parent.getReferrers(), Relation.class)) {
+                    if(RouteUtils.isPTRoute(r))
                         return false;
                 }
@@ -196 +224 @@
     }
 
+    public Command getRemoveRoundaboutFromRelationsCommand(Way roundabout) {
+        List <Relation> referrers = OsmPrimitive.getFilteredList(
+                roundabout.getReferrers(), Relation.class);
+        referrers.removeIf(r -> !RouteUtils.isPTRoute(r));
+
+        List<Command> commands = new ArrayList<>();
+        referrers.forEach(r -> {
+            Relation c = new Relation(r);
+            c.removeMembersFor(roundabout);
+            commands.add(new ChangeCommand(r, c));
+        });
+
+        return new SequenceCommand("Remove roundabout from relations", commands);
+    }
+
     //save the position of the roundabout inside each public transport route
     //it is contained in
@@ -201 +244 @@
 
         Map<Relation, List<Integer>> savedPositions = new HashMap<>();
-        List <OsmPrimitive> referrers = roundabout.getReferrers();
-        referrers.removeIf(r -> r.getType() != OsmPrimitiveType.RELATION
-                || !RouteUtils.isPTRoute((Relation) r));
-
-        for(OsmPrimitive currPrim : referrers) {
-            Relation curr = (Relation) currPrim;
+        List <Relation> referrers = OsmPrimitive.getFilteredList(
+                roundabout.getReferrers(), Relation.class);
+        referrers.removeIf(r -> !RouteUtils.isPTRoute(r));
+
+        for(Relation curr : referrers) {
             for(int j = 0; j < curr.getMembersCount(); j++) {
                 if(curr.getMember(j).getUniqueId() == roundabout.getUniqueId()) {
@@ -215 +257 @@
                 }
             }
-
-            if(savedPositions.containsKey(curr))
-                curr.removeMembersFor(roundabout);
         }