Changeset 20662 in osm for applications/editors/josm/plugins/michigan_left

Timestamp:

2010-03-25T03:47:36+01:00 (14 years ago)

Author:

nakor

Message:

Cleanup; Disable plugin with no layer

File:

• applications/editors/josm/plugins/michigan_left/src/MichiganLeft/MichiganLeft.java (modified) (2 diffs)

applications/editors/josm/plugins/michigan_left/src/MichiganLeft/MichiganLeft.java

@@ -1 +1 @@
 package MichiganLeft;
 
+import java.awt.event.ActionEvent;
+import java.awt.event.KeyEvent;
+import java.util.Collection;
+import java.util.Enumeration;
+import java.util.Hashtable;
+import java.util.LinkedList;
+import java.util.ArrayList;
+
 import javax.swing.JMenuItem;
+import javax.swing.JOptionPane;
+
+import static org.openstreetmap.josm.tools.I18n.tr;
 
 import org.openstreetmap.josm.Main;
+import org.openstreetmap.josm.actions.JosmAction;
+import org.openstreetmap.josm.command.AddCommand;
+import org.openstreetmap.josm.command.Command;
+import org.openstreetmap.josm.command.SequenceCommand;
+import org.openstreetmap.josm.data.osm.Node;
+import org.openstreetmap.josm.data.osm.OsmPrimitive;
+import org.openstreetmap.josm.data.osm.Relation;
+import org.openstreetmap.josm.data.osm.RelationMember;
+import org.openstreetmap.josm.data.osm.Way;
 import org.openstreetmap.josm.gui.MainMenu;
 import org.openstreetmap.josm.plugins.Plugin;
 import org.openstreetmap.josm.plugins.PluginInformation;
+import org.openstreetmap.josm.tools.Shortcut;
+
 
 public class MichiganLeft extends Plugin {
@@ -16 +38 @@
 
     }
+    
+    
+    private class MichiganLeftAction extends JosmAction {
+        private LinkedList<Command> cmds = new LinkedList<Command>();
+
+        public MichiganLeftAction() {
+        super(tr("Michigan Left"), "michigan_left", tr("Adds no left turn for sets of 4 or 5 ways."), Shortcut.registerShortcut("tools:michigan_left", tr("Tool: {0}", tr("Michigan Left")),
+        KeyEvent.VK_M, Shortcut.GROUP_EDIT, KeyEvent.SHIFT_DOWN_MASK|KeyEvent.ALT_DOWN_MASK), true);
+      }
+
+      public void actionPerformed(ActionEvent e) {
+        Collection<OsmPrimitive> selection = Main.main.getCurrentDataSet().getSelected();
+
+        int ways = 0;
+        for (OsmPrimitive prim : selection) {
+          if (prim instanceof Way)  ways++;
+        }
+
+        if ((ways != 4) && (ways !=5)) {
+          JOptionPane.showMessageDialog(Main.parent, tr("Please select 4 or 5 ways to assign no left turns."));
+          return;
+        }
+
+        if (ways == 4)
+        {
+          // Find extremities of ways
+          Hashtable<Node, Integer> ExtremNodes=new Hashtable<Node, Integer>();
+          for (OsmPrimitive prim : selection)
+          {
+            if (prim instanceof Way)
+            {
+              Way way = (Way) prim;
+              incrementHashtable(ExtremNodes, way.firstNode());
+              incrementHashtable(ExtremNodes, way.lastNode());
+            }
+          }
+          //System.out.println(tr("{0} extrem nodes.", ExtremNodes.size()));
+          if (ExtremNodes.size() != 4)
+          {
+            JOptionPane.showMessageDialog(Main.parent, tr("Please select 4 ways that form a closed relation."));
+            return;
+          }
+
+          // order the ways
+          ArrayList<Way> orderedWays=new ArrayList<Way>();
+          Way currentWay=(Way) selection.iterator().next();
+          orderedWays.add((Way) currentWay);
+          selection.remove(currentWay);
+          while (selection.size()>0)
+          {
+              boolean found=false;
+              Node nextNode=currentWay.lastNode();
+              for (OsmPrimitive prim : selection)
+              {
+                  Way tmpWay=(Way) prim;
+                  if (tmpWay.firstNode() == nextNode)
+                  {
+                    orderedWays.add(tmpWay);
+                    selection.remove(prim);
+                    currentWay=tmpWay;
+                    found=true;
+                    break;
+                  }
+              }
+              if (!found)
+              {
+                JOptionPane.showMessageDialog(Main.parent, tr("Unable to order the ways. Please verify their directions"));
+                return;
+              }
+          }
+
+          // Build relations
+          for (int index=0 ; index<4 ; index++)
+          {
+            Way firstWay=orderedWays.get(index);
+            Way lastWay=orderedWays.get( (index+1) % 4);
+            Node lastNode = firstWay.lastNode();
+
+            buildRelation(firstWay, lastWay, lastNode);
+           }
+          Command c = new SequenceCommand(tr("Create Michigan left turn restriction"), cmds);
+          Main.main.undoRedo.add(c);
+          cmds.clear();
+        }
+
+        if (ways == 5)
+        {
+          // Find extremities of ways
+          Hashtable<Node, Integer> ExtremNodes=new Hashtable<Node, Integer>();
+          for (OsmPrimitive prim : selection) {
+            if (prim instanceof Way)
+            {
+              Way way = (Way) prim;
+              incrementHashtable(ExtremNodes, way.firstNode());
+              incrementHashtable(ExtremNodes, way.lastNode());
+            }
+          }
+          //System.out.println(tr("{0} extrem nodes.", ExtremNodes.size()));
+
+          ArrayList<Node> viaNodes=new ArrayList<Node>();
+          // find via nodes (they have 3 occurences in the list)
+          for (Enumeration enumKey = ExtremNodes.keys() ; enumKey.hasMoreElements(); )
+          {
+            Node extrem=(Node)enumKey.nextElement();
+            Integer nb=(Integer) ExtremNodes.get(extrem);
+            //System.out.println(tr("Via node {0}, {1}", extrem.getId(), nb.intValue()));
+            if (nb.intValue() == 3)
+            {
+              viaNodes.add(extrem);
+            }
+          }
+          //System.out.println(tr("{0} via nodes.", viaNodes.size()));
+
+          if (viaNodes.size() != 2) {
+            JOptionPane.showMessageDialog(Main.parent, tr("Unable to find via nodes. Please check your selection"));
+            return;
+          }
+
+          Node viaFirst = viaNodes.get(0);
+          Node viaLast = viaNodes.get(1);      // Find middle segment
+
+          Way middle = null;
+          for (OsmPrimitive prim : selection) {
+            if (prim instanceof Way)
+            {
+              Way way = (Way) prim;
+              Node first = way.firstNode();
+              Node last = way.lastNode();
+
+
+              if ((first.equals(viaFirst) && last.equals(viaLast)) || (first.equals(viaLast) && last.equals(viaFirst)))
+                middle=way;
+            }
+          }
+          //System.out.println(tr("Middle way: {0}", middle.getId()));
+
+          // Build relations
+          for (OsmPrimitive prim : selection) {
+            if (prim instanceof Way)
+            {
+              Way way = (Way) prim;
+              if (way != middle)
+              {
+                Node first = way.firstNode();
+                Node last = way.lastNode();
+
+                if (first==viaFirst)
+                  buildRelation(middle, way, viaNodes.get(0));
+                else if (first==viaLast)
+                  buildRelation(middle, way, viaNodes.get(1));
+                else if (last==viaFirst)
+                  buildRelation(way, middle, viaNodes.get(0));
+                else if (last==viaLast)
+                  buildRelation(way, middle, viaNodes.get(1));
+              }
+            }
+          }
+          Command c = new SequenceCommand(tr("Create Michigan left turn restriction"), cmds);
+          Main.main.undoRedo.add(c);
+          cmds.clear();
+        }
+      }
+
+      public void incrementHashtable(Hashtable<Node, Integer> hash, Node node)
+      {
+        //System.out.println(tr("Processing {0}", node.getId()));
+        if (hash.containsKey(node))
+        {
+          Integer nb=(Integer) hash.get(node);
+          hash.put(node, new Integer (nb.intValue()+1));
+          //System.out.println(tr("Old value", nb.intValue()));
+        }
+        else
+          hash.put(node, new Integer (1));
+      }
+
+      public void buildRelation(Way fromWay, Way toWay, Node viaNode)
+      {
+        //System.out.println(tr("Relation: from {0} to {1} via {2}", fromWay.getId(), toWay.getId(), viaNode.getId()));
+
+        Relation relation = new Relation();
+
+        RelationMember from = new RelationMember("from", fromWay);
+        relation.addMember(from);
+
+        RelationMember to = new RelationMember("to", toWay);
+        relation.addMember(to);
+
+        RelationMember via = new RelationMember("via", viaNode);
+        relation.addMember(via);
+
+        relation.put("type", "restriction");
+        relation.put("restriction", "no_left_turn");
+
+        cmds.add(new AddCommand(relation));
+      }
+      @Override
+      protected void updateEnabledState() {
+          setEnabled(getEditLayer() != null);
+      }
+
+      @Override
+      protected void updateEnabledState(
+              Collection<? extends OsmPrimitive> selection) {
+          // do nothing
+      }
+
+}    
 }