Minesweeper: add mouse "chording" feature

Author: RetiredWizard

Metro/Metro_RP2350_Minesweeper/code.py

@@ -123,6 +123,7 @@ def update_ui():
 
 buf = array.array("b", [0] * 4)
 waiting_for_release = False
+chord_selected = False
 left_button = right_button = False
 mouse_coords = (display.width // 2, display.height // 2)
 
@@ -231,41 +232,21 @@ def hide_group(group):
 
 menus = (reset_menu, difficulty_menu)
 
-# Mouse state
-last_left_button_state = 0
-last_right_button_state = 0
-left_button_pressed = False
-right_button_pressed = False
-
 # main loop
 while True:
     update_ui()
     # update cursor position, and check for clicks
-    buttons = mouse.update()
+    mouse.update()
+    buttons = mouse.pressed_btns
 
     # Extract button states
-    if buttons is None:
+    if buttons is None or buttons == ():
         left_button = 0
         right_button = 0
     else:
         left_button = 1 if 'left' in buttons else 0
         right_button = 1 if 'right' in buttons else 0
 
-    # Detect button presses
-    if left_button == 1 and last_left_button_state == 0:
-        left_button_pressed = True
-    elif left_button == 0 and last_left_button_state == 1:
-        left_button_pressed = False
-
-    if right_button == 1 and last_right_button_state == 0:
-        right_button_pressed = True
-    elif right_button == 0 and last_right_button_state == 1:
-        right_button_pressed = False
-
-    # Update button states
-    last_left_button_state = left_button
-    last_right_button_state = right_button
-
     mouse_coords = (mouse_tg.x, mouse_tg.y)
 
     if waiting_for_release and not left_button and not right_button:
@@ -282,11 +263,23 @@ def hide_group(group):
     else:
         # process gameboard click if no menu
         ms_board = game_logic.game_board
+
+        if left_button and right_button and not chord_selected:
+            chord_coords = ((mouse_tg.x - ms_board.x) // 16, (mouse_tg.y - ms_board.y) // 16)
+            chord_selected = True
+            game_logic.square_chord_highlight(chord_coords)
+            waiting_for_release = True
+
         if (ms_board.x <= mouse_tg.x <= ms_board.x + game_logic.grid_width * 16 and
             ms_board.y <= mouse_tg.y <= ms_board.y + game_logic.grid_height * 16 and
             not waiting_for_release):
+
             coords = ((mouse_tg.x - ms_board.x) // 16, (mouse_tg.y - ms_board.y) // 16)
-            if right_button:
+            if chord_selected:
+                chord_selected = False
+                game_logic.square_chord_highlight(chord_coords, False)
+                game_logic.square_chorded(chord_coords)
+            elif right_button:
                 game_logic.square_flagged(coords)
             elif left_button:
                 if not game_logic.square_clicked(coords):

Metro/Metro_RP2350_Minesweeper/gamelogic.py

@@ -165,6 +165,62 @@ def square_flagged(self, coords):
                 break
         return True
 
+    def square_chorded(self, coords):
+        if self._status in (STATUS_WON, STATUS_LOST):
+            return False
+
+        x, y = coords
+        if self._get_board(x, y) in (OPEN1, OPEN2, OPEN3, OPEN4, OPEN5, OPEN6, OPEN7, OPEN8):
+            # Count the flags around this square
+            flags = 0
+            for dx in (-1, 0, 1):
+                if x + dx < 0 or x + dx >= self.grid_width:
+                    continue              # off screen
+                for dy in (-1, 0, 1):
+                    if y + dy < 0 or y + dy >= self.grid_height:
+                        continue          # off screen
+                    if dx == 0 and dy == 0:
+                        continue          # don't process where the mine
+                    if self._get_board(x + dx, y + dy) == FLAG:
+                        flags += 1
+            if flags == self._get_board(x, y):
+                # Uncover all non-flagged squares around here
+                for dx in (-1, 0, 1):
+                    if x + dx < 0 or x + dx >= self.grid_width:
+                        continue              # off screen
+                    for dy in (-1, 0, 1):
+                        if y + dy < 0 or y + dy >= self.grid_height:
+                            continue          # off screen
+                        if dx == 0 and dy == 0:
+                            continue          # don't process where the mine
+                        if self._get_board(x + dx, y + dy) != FLAG:
+                            if not self.square_clicked((x + dx, y + dy)):
+                                return False      # lost
+        return True
+
+    def square_chord_highlight(self, coords, highlight=True):
+        if self._status in (STATUS_WON, STATUS_LOST):
+            return False
+
+        x, y = coords
+        if self._get_board(x, y) in (OPEN1, OPEN2, OPEN3, OPEN4, OPEN5, OPEN6, OPEN7, OPEN8):
+            # Highlight all non-flagged squares around here
+            for dx in (-1, 0, 1):
+                if x + dx < 0 or x + dx >= self.grid_width:
+                    continue              # off screen
+                for dy in (-1, 0, 1):
+                    if y + dy < 0 or y + dy >= self.grid_height:
+                        continue          # off screen
+                    if dx == 0 and dy == 0:
+                        continue          # don't process where the mine
+                    if highlight:
+                        if self._get_board(x + dx, y + dy) == BLANK:
+                            self._set_board(x + dx, y + dy,MINE_QUESTION_OPEN)
+                    else:
+                        if self._get_board(x + dx, y + dy) == MINE_QUESTION_OPEN:
+                            self._set_board(x + dx, y + dy, BLANK)
+        return True
+
     def square_clicked(self, coords):
         x, y = coords
 
@@ -215,7 +271,9 @@ def check_for_win(self):
         # first make sure everything has been explored and decided
         for x in range(self.grid_width):
             for y in range(self.grid_height):
-                if self._get_board(x, y) == BLANK or self._get_board(x, y) == MINE_QUESTION:
+                if self._get_board(x, y) == BLANK or \
+                    self._get_board(x, y) == MINE_QUESTION or \
+                    self._get_board(x, y) == MINE_QUESTION_OPEN:
                     return None               # still ignored or question squares
         # then check for mistagged bombs
         for x in range(self.grid_width):