def __init__(self): super(QObject, self).__init__() self.data_pack = "datas" self.coords_folder = "coords" self.screens_folder = "screens" self.currentScreensFolder = "1080x2220" self.manage_default_currentScreensPath() self.buttons_folder = "buttons.json" self.movements_folder = "movements.json" self.static_coords_folder = "static_coords.json" self.ui_color = (0, 255, 255) self.ui_bgcolor = (43, 43, 43) self.ui_lines_color_rgb = (0, 255, 0) self.ui_lines_color_rgb_selected = (255, 0, 255) self.linePermittedSizes = [i for i in range(1, 20, 1)] self.currentLineWidth = self.linePermittedSizes[int(len(self.linePermittedSizes) / 2.0)] self.currentFiles = {} self.currentDict = {} self.currentMovements = {} self.currentFrameChecks = {} self.screensFolders = readAllSizesFolders() self.device_connector = UsbConnector() self.device_connector.connectionChangedFunctions.append(self.onDeviceConnectionChanged)
def initDataFolders(self): self.dataFolders = readAllSizesFolders() deviceFolder = buildDataFolder(self.width, self.heigth) first_folder = list(self.dataFolders.keys())[0] if deviceFolder not in self.dataFolders: print("Error: not having %s coordinates. Trying with %s" % (deviceFolder, first_folder)) deviceFolder = first_folder self.changeCurrentDataFolder(deviceFolder)
import numpy as np from PIL import Image from GameScreenConnector import GameScreenConnector from Utils import readAllSizesFolders # from pure_adb_connector import * def getImageFrame(path: str): with Image.open(path, "r") as im: pixval = np.array(im.getdata()) return pixval screens_data = readAllSizesFolders() keys = [k for k in screens_data.keys()] for i in range(len(keys)): print("%d: %s" % (i, keys[i])) choosen = input("Select your number") folder = keys[int(choosen)] screens_path = os.path.join("datas", folder, "screens") print("Using %s" % screens_path) selected_debug = input( "Do you wish to debug? (set yes only if you did it already once and found some rows with NO_DETECTION): (y/n):" ) debug = False if selected_debug != None: if selected_debug == "y" or selected_debug == "yes":