mirror of
https://github.com/fishyboteso/fishyboteso.git
synced 2024-08-30 18:32:13 +00:00
103 lines
2.5 KiB
Python
103 lines
2.5 KiB
Python
import logging
|
|
from enum import Enum
|
|
from threading import Thread
|
|
|
|
import cv2
|
|
import math
|
|
|
|
import pywintypes
|
|
import win32gui
|
|
from win32api import GetSystemMetrics
|
|
|
|
import numpy as np
|
|
from PIL import ImageGrab
|
|
|
|
from fishy.helper.config import config
|
|
|
|
|
|
class Status(Enum):
|
|
CRASHED = -1
|
|
STOPPED = 0
|
|
RUNNING = 1
|
|
|
|
|
|
class WindowServer:
|
|
"""
|
|
Records the game window, and allows to create instance to process it
|
|
"""
|
|
Screen = None
|
|
windowOffset = None
|
|
titleOffset = None
|
|
hwnd = None
|
|
status = Status.STOPPED
|
|
|
|
|
|
def init():
|
|
"""
|
|
Executed once before the main loop,
|
|
Finds the game window, and calculates the offset to remove the title bar
|
|
"""
|
|
try:
|
|
WindowServer.hwnd = win32gui.FindWindow(None, "Elder Scrolls Online")
|
|
rect = win32gui.GetWindowRect(WindowServer.hwnd)
|
|
client_rect = win32gui.GetClientRect(WindowServer.hwnd)
|
|
WindowServer.windowOffset = math.floor(((rect[2] - rect[0]) - client_rect[2]) / 2)
|
|
WindowServer.titleOffset = ((rect[3] - rect[1]) - client_rect[3]) - WindowServer.windowOffset
|
|
if config.get("borderless"):
|
|
WindowServer.titleOffset = 0
|
|
WindowServer.status = Status.RUNNING
|
|
except pywintypes.error:
|
|
logging.error("Game window not found")
|
|
WindowServer.status = Status.CRASHED
|
|
|
|
|
|
def loop():
|
|
"""
|
|
Executed in the start of the main loop
|
|
finds the game window location and captures it
|
|
"""
|
|
bbox = (0, 0, GetSystemMetrics(0), GetSystemMetrics(1))
|
|
|
|
temp_screen = np.array(ImageGrab.grab(bbox=bbox))
|
|
|
|
temp_screen = cv2.cvtColor(temp_screen, cv2.COLOR_BGR2RGB)
|
|
|
|
rect = win32gui.GetWindowRect(WindowServer.hwnd)
|
|
crop = (
|
|
rect[0] + WindowServer.windowOffset, rect[1] + WindowServer.titleOffset, rect[2] - WindowServer.windowOffset,
|
|
rect[3] - WindowServer.windowOffset)
|
|
|
|
WindowServer.Screen = temp_screen[crop[1]:crop[3], crop[0]:crop[2]]
|
|
|
|
if WindowServer.Screen.size == 0:
|
|
logging.error("Don't minimize or drag game window outside the screen")
|
|
WindowServer.status = Status.CRASHED
|
|
|
|
|
|
def loop_end():
|
|
cv2.waitKey(25)
|
|
|
|
|
|
def run():
|
|
# todo use config
|
|
while WindowServer.status == Status.RUNNING:
|
|
loop()
|
|
loop_end()
|
|
|
|
|
|
def start():
|
|
if WindowServer.status == Status.RUNNING:
|
|
return
|
|
|
|
init()
|
|
if WindowServer.status == Status.RUNNING:
|
|
Thread(target=run).start()
|
|
|
|
|
|
def screen_ready():
|
|
return WindowServer.Screen is not None or WindowServer.status == Status.CRASHED
|
|
|
|
|
|
def stop():
|
|
WindowServer.status = Status.STOPPED
|