Upload 18 files

57f1366 verified 8 months ago

3.45 kB

import pyautogui
import chess
import time
import math

class MirrorHandler:
    def __init__(self):
        # Configure pyautogui to be a bit safer/slower if needed
        pyautogui.FAILSAFE = True
        pyautogui.PAUSE = 0.1

    def execute_move(self, move, region, is_flipped=False):
        """
        Execute a chess move on the screen region using mouse drag.
        
        Args:
            move (chess.Move): The move to execute.
            region (dict): {'left', 'top', 'width', 'height'} of the target board.
            is_flipped (bool): If True, board is viewed from Black's perspective (rank 1 at top).
        """
        if not region or not move:
            return

        start_sq = move.from_square
        end_sq = move.to_square

        # Calculate coordinates
        start_x, start_y = self._get_square_center(start_sq, region, is_flipped)
        end_x, end_y = self._get_square_center(end_sq, region, is_flipped)

        # Save current position to restore later? 
        # Actually user said "just use my mouse", implies they expect it to take over.
        # But maybe restoring it is nice? Let's just do the move for now.

        # Perform the drag
        # Move to start
        pyautogui.moveTo(start_x, start_y)
        # Drag to end
        pyautogui.dragTo(end_x, end_y, button='left')
        
        # Check for promotion
        if move.promotion:
            # Assumes auto-queen or standard promotion UI (usually Queen is closest/first)
            # This is tricky for generic sites. 
            # Often, clicking the target square again acts as "Question" or immediate select.
            # For now, let's just click the target square again to select Queen if it pops up.
            time.sleep(0.1)
            pyautogui.click() 

    def _get_square_center(self, square, region, is_flipped):
        """
        Calculate center x, y for a given square index (0-63).
        """
        file_idx = chess.square_file(square)
        rank_idx = chess.square_rank(square)

        if is_flipped:
            # Board is black options (rank 8 at bottom? Wait.)
            # Standard: White at bottom (Rank 0 is bottom).
            # Flipped: Black at bottom (Rank 7 is bottom, Rank 0 is top).
            
            # If Flipped (Black at bottom):
            # File 0 (a) is on the Right? No, usually board just rotates 180.
            # Visual:
            # White bottom: a1 bottom-left.
            # Black bottom: h8 bottom-left.
            
            # Let's assume Standard Rotation (180 degrees):
            # a1 (0,0) becomes top-right.
            # h8 (7,7) becomes bottom-left.
            
            # Let's map visual col/row (0-7 from top-left)
            # Normal:
            # col = file_idx
            # row = 7 - rank_idx
            
            # Flipped:
            # col = 7 - file_idx
            # row = rank_idx
            
            col = 7 - file_idx
            row = rank_idx
        else:
            # Standard (White at bottom)
            # a1 is bottom-left
            col = file_idx
            row = 7 - rank_idx

        sq_w = region['width'] / 8
        sq_h = region['height'] / 8

        center_x = region['left'] + (col * sq_w) + (sq_w / 2)
        center_y = region['top'] + (row * sq_h) + (sq_h / 2)

        return center_x, center_y

	import pyautogui
	import chess
	import time
	import math

	class MirrorHandler:
	def __init__(self):
	# Configure pyautogui to be a bit safer/slower if needed
	pyautogui.FAILSAFE = True
	pyautogui.PAUSE = 0.1

	def execute_move(self, move, region, is_flipped=False):
	"""
	Execute a chess move on the screen region using mouse drag.

	Args:
	move (chess.Move): The move to execute.
	region (dict): {'left', 'top', 'width', 'height'} of the target board.
	is_flipped (bool): If True, board is viewed from Black's perspective (rank 1 at top).
	"""
	if not region or not move:
	return

	start_sq = move.from_square
	end_sq = move.to_square

	# Calculate coordinates
	start_x, start_y = self._get_square_center(start_sq, region, is_flipped)
	end_x, end_y = self._get_square_center(end_sq, region, is_flipped)

	# Save current position to restore later?
	# Actually user said "just use my mouse", implies they expect it to take over.
	# But maybe restoring it is nice? Let's just do the move for now.

	# Perform the drag
	# Move to start
	pyautogui.moveTo(start_x, start_y)
	# Drag to end
	pyautogui.dragTo(end_x, end_y, button='left')

	# Check for promotion
	if move.promotion:
	# Assumes auto-queen or standard promotion UI (usually Queen is closest/first)
	# This is tricky for generic sites.
	# Often, clicking the target square again acts as "Question" or immediate select.
	# For now, let's just click the target square again to select Queen if it pops up.
	time.sleep(0.1)
	pyautogui.click()

	def _get_square_center(self, square, region, is_flipped):
	"""
	Calculate center x, y for a given square index (0-63).
	"""
	file_idx = chess.square_file(square)
	rank_idx = chess.square_rank(square)

	if is_flipped:
	# Board is black options (rank 8 at bottom? Wait.)
	# Standard: White at bottom (Rank 0 is bottom).
	# Flipped: Black at bottom (Rank 7 is bottom, Rank 0 is top).

	# If Flipped (Black at bottom):
	# File 0 (a) is on the Right? No, usually board just rotates 180.
	# Visual:
	# White bottom: a1 bottom-left.
	# Black bottom: h8 bottom-left.

	# Let's assume Standard Rotation (180 degrees):
	# a1 (0,0) becomes top-right.
	# h8 (7,7) becomes bottom-left.

	# Let's map visual col/row (0-7 from top-left)
	# Normal:
	# col = file_idx
	# row = 7 - rank_idx

	# Flipped:
	# col = 7 - file_idx
	# row = rank_idx

	col = 7 - file_idx
	row = rank_idx
	else:
	# Standard (White at bottom)
	# a1 is bottom-left
	col = file_idx
	row = 7 - rank_idx

	sq_w = region['width'] / 8
	sq_h = region['height'] / 8

	center_x = region['left'] + (col * sq_w) + (sq_w / 2)
	center_y = region['top'] + (row * sq_h) + (sq_h / 2)

	return center_x, center_y