Continued network mode implementation

lib/pygogui.py

@@ -10,7 +10,7 @@ from sgc.locals import *
 
 
 class GUI:
-    def __init__(self, goban):
+    def __init__(self, goban, settings):
         # Basic screen init
         pygame.init()
         # screen = pygame.display.set_mode((1000, 800))
@@ -28,16 +28,34 @@ class GUI:
         # Build the dict of image objects
         self.img_res = _build_img_res()
 
+        self.goban = goban
+        self.settings = settings
+
+        # Network-related settings
+        self.net_thread = None
+        self.socket = None
+        self.network_mode = False
+        self.our_color = None
+
+
         self.board_size = 800
         self.board_inc = self.board_size / 19
 
         self.screen.fill((250, 250, 250))
+
+        self.join_btn = sgc.widgets.Button(label="Join Game", pos=(850,400))
+        self.join_btn.activate = self.join_game
+        self.join_btn.add()
+
+        self.wait_btn = sgc.widgets.Button(label="Listen", pos=(850,475))
+        self.wait_btn.activate = self.wait_for_game
+        self.wait_btn.add()
     
-        self.pass_btn = sgc.widgets.Button(label="Pass", pos=(850,500))
+        self.pass_btn = sgc.widgets.Button(label="Pass", pos=(850,550))
         self.pass_btn.activate = goban.pass_move
         self.pass_btn.add()
 
-        self.resign_btn = sgc.widgets.Button(label="Resign", pos=(850,600))
+        self.resign_btn = sgc.widgets.Button(label="Resign", pos=(850,625))
         self.resign_btn.activate = goban.resign
         self.resign_btn.add()
 
@@ -45,33 +63,46 @@ class GUI:
         self.quit_btn.activate = sys.exit
         self.quit_btn.add()
 
+#        self.waiting
+
         # pygame.time.set_timer(USEREVENT, 1000)
 
 
-
-    def do_event(self, goban, network_mode, our_color):
+    def do_event(self):
         event = pygame.event.wait()
         sgc.widgets.event(event)
 
         if event.type == QUIT:
             return
         
-        # This set of events should only be called if we can currently play
-        if network_mode == False or goban.to_move == our_color:
+
+        if self.network_mode:
+            # This set of events should only be called if we can currently play
+            if self.goban.to_move == self.our_color:
+                # Hover a transparent stone over our
+                # cursor position, assuming play is legal
+                if event.type == MOUSEMOTION:
+                    with self.net_thread.goban_lock:
+                        self.do_hover(event)
+
+                if event.type == MOUSEBUTTONDOWN:
+                    x, y = event.pos
+                    row = y / self.board_inc
+                    col = x / self.board_inc
+
+                    if x <= self.board_size:
+                        if event.button == 1:
+                            with self.net_thread.send_lock:
+                                col_letter = chr(col + 96)
+                                self.socket.send(col_letter + str(row))
+                            self.goban.play_move((row, col))
+
+        # Local play mode
+        else:
             # Hover a transparent stone over our
             # cursor position, assuming play is legal
             if event.type == MOUSEMOTION:
-                x, y = event.pos
-                row = y / self.board_inc
-                col = x / self.board_inc
-
-                if _magnitude(event.rel) < 3:
-                    if x <= self.board_size:
-                        goban.set_hover((row,col))
-                    else:
-                        goban.clear_hover()
-                elif goban.hover != goban._real_pos((row,col)):
-                    goban.clear_hover()
+                self.do_hover(event)
 
             # Place a stone on left-click
             if event.type == MOUSEBUTTONDOWN:
@@ -81,7 +112,7 @@ class GUI:
 
                 if x <= self.board_size:
                     if event.button == 1:
-                        goban.play_move((row, col))
+                        self.goban.play_move((row, col))
 
         # if event.type == USEREVENT:
         #     goban.elapsed_time += 1
@@ -92,18 +123,42 @@ class GUI:
         #         dialogs.remove(widget)
 
 
-
-    def update(self, goban):
-        board = goban.draw_board(self.board_size, self.img_res)
+    def update(self):
+        board = self.goban.draw_board(self.board_size, self.img_res)
         self.screen.blit(board, (0,0))
     
-        text = goban.draw_info()
+        text = self.goban.draw_info()
         self.screen.blit(text, (815, 25))
 
         sgc.widgets.update(0)
         pygame.display.flip()
 
 
+    # fixme: this uses localhost as a stub
+    def join_game(self):
+        pass
+
+
+    # fixme: this uses localhost as a stub
+    def wait_for_game(self):
+        pass
+
+
+    def do_hover(self, event):
+        x, y = event.pos
+        row = y / self.board_inc
+        col = x / self.board_inc
+
+        if _magnitude(event.rel) < 3:
+            if x <= self.board_size:
+                self.goban.set_hover((row,col))
+            else:
+                self.goban.clear_hover()
+        elif self.goban.hover != self.goban._real_pos((row,col)):
+            self.goban.clear_hover()
+
+
+
 
 def _magnitude(vector):
     x,y = vector