X-Git-Url: http://git.tomasm.cz/imago.git/blobdiff_plain/33b793fe326a2706173b38d2fc4b454982fbf900..81e4eb0f1e3e06ab476c9707fb950c0e1225d33f:/src/imago.py?ds=sidebyside

diff --git a/src/imago.py b/src/imago.py
index 0dd970a..698ef2d 100755
--- a/src/imago.py
+++ b/src/imago.py
@@ -1,9 +1,6 @@
 #!/usr/bin/env python
 
-"""Go image recognition.
-
-This is the main UI module of Imago.
-"""
+"""Go image recognition."""
 
 import sys
 import os
@@ -17,7 +14,6 @@ except ImportError, msg:
     sys.exit(1)
 
 import linef
-import manual
 import intrsc
 import gridf3 as gridf
 import output
@@ -97,6 +93,7 @@ def main():
             pass
         
     if args.manual_mode:
+        import manual
         try:
             lines = manual.find_lines(image)
         except manual.UserQuitError:
@@ -133,10 +130,12 @@ def main():
                 draw.line(l, fill=(64, 255, 64), width=1)
             do_something(im_g, "grid", name="grid")
 
-    board = intrsc.board(image, lines, show_all, do_something, logger)
+    intersections = intrsc.b_intersects(image, lines, show_all, do_something, logger)
+    board = intrsc.board(image, intersections, show_all, do_something, logger)
 
     logger("finished")
 
+    # TODO! refactor this mess:
     if len(args.files) == 1:
 
         if args.sgf_output:
@@ -156,7 +155,10 @@ def main():
                 print >> sys.stderr, "Opening", f
             if image.mode == 'P':
                 image = image.convert('RGB')
-            board = intrsc.board(image, lines, show_all, do_something, logger)
+            if image.size[0] > args.w:
+                image = image.resize((args.w, int((float(args.w)/image.size[0]) *
+                              image.size[1])), Image.ANTIALIAS)
+            board = intrsc.board(image, intersections, show_all, do_something, logger)
             if args.sgf_output:
                 game.addMove(board)
             else:
@@ -173,14 +175,15 @@ class Imsave():
         self.saving_num = 0
 
     def save(self, image, title='', name=None):
+        im_format = ('.png', 'PNG')
         if name:
-            filename = self.saving_dir + name + '.jpg'
+            filename = self.saving_dir + name + im_format[0]
         else:
-            filename = self.saving_dir + "{0:0>2}".format(self.saving_num) + '.jpg'
+            filename = self.saving_dir + "{0:0>3}".format(self.saving_num) + im_format[0]
             self.saving_num += 1
         if not os.path.isdir(self.saving_dir):
             os.makedirs(self.saving_dir)
-        image.save(filename, 'JPEG')
+        image.save(filename, im_format[1])
 
 if __name__ == '__main__':
     try: