Changeset 135:c9b6685ebebf

Timestamp:

Sep 2, 2013, 4:19:57 PM (7 years ago)

Author:

Neil Muller <drnlmuller@…>

Branch:

default

rebase_source:

10cdb285f9c291a4cd0e4e643cc1d04f949b125e

Message:

Add hack'ish 'close polygon' option

File:

• tools/area_editor.py (modified) (4 diffs)

tools/area_editor.py

@@ -90 +90 @@
             self.polygons[index].pop()
 
+    def close_poly(self, index):
+        """Attempts to close the current polygon.
+
+           We allow a small additional step to close the polygon, but
+           it's limited as it's a magic point addition"""
+        if len(self.polygons[index]) < 2:
+            # Too small
+            return False
+        first = self.polygons[index][0]
+        last = self.polygons[index][-1]
+        print first, last
+        print self.fix_angle(index, self.point_to_pygame(first))
+        if self.fix_angle(index, self.point_to_pygame(first)) == first:
+            self.add_point(index, self.point_to_pygame(first))
+            return True
+        candidates = [(first[0] + 10 * i, first[1]) for
+                      i in (-3, -2, -1, 1, 2, 3)]
+        candidates.extend([(first[0], first[1] + 10 * i) for
+                          i in (-3, -2, -1, 1, 2, 3)])
+        candidates.extend([(first[0] + 10 * i, first[1] + 10 * i) for
+                          i in (-3, -2, -1, 1, 2, 3)])
+        candidates.extend([(first[0] + 10 * i, first[1] - 10 * i) for
+                          i in (-3, -2, -1, 1, 2, 3)])
+        min_dist = 99999
+        poss = None
+        for cand in candidates:
+            if self.fix_angle(index, self.point_to_pygame(cand)) == cand:
+                dist = (first[0] - cand[0]) ** 2 + (first[1] - cand[1]) ** 2
+                if dist < min_dist:
+                    poss = cand
+        if poss is not None:
+            self.add_point(index, self.point_to_pygame(poss))
+            self.add_point(index, self.point_to_pygame(first))
+            return True
+        return False
+
     def draw(self, surface, topleft, mouse_pos, mouse_poly, filled):
         self._draw_background(True)
@@ -170 +206 @@
         elif ev.key == pgl.K_f:
             self.set_filled()
+        elif ev.key == pgl.K_c:
+            self.close_poly()
 
     def set_filled(self):
@@ -189 +227 @@
             self.level.add_point(self.cur_poly,
                                  self._level_coordinates(ev.pos))
+
+    def close_poly(self):
+        if self.cur_poly is None:
+            return
+        if self.level.close_poly(self.cur_poly):
+            alert("Successfully closed the polygon")
+            self.change_poly(None)
+        else:
+            alert("Failed to close the polygon")
 
 
@@ -247 +294 @@
         save_but.rect.move_ip(MENU_LEFT, y)
         self.add(save_but)
+        y += MENU_BUTTON_HEIGHT + MENU_PAD
+
+        close_poly_but = Button('Close Polygon',
+                                action=self.level_widget.close_poly)
+        close_poly_but.rect = button_rect.copy()
+        close_poly_but.rect.move_ip(MENU_LEFT, y)
+        self.add(close_poly_but)
         y += MENU_BUTTON_HEIGHT + MENU_PAD