diff --git a/craft_table/craft_overlay.py b/craft_table/craft_overlay.py new file mode 100644 index 0000000..b4a1a3c --- /dev/null +++ b/craft_table/craft_overlay.py @@ -0,0 +1,93 @@ +# Run tkinter code in another thread +import threading +import tkinter as tk +from tkinter import ttk + + +class CraftOverlay(threading.Thread): + + def __init__(self): + threading.Thread.__init__(self) + + self.run_mode = 'init' + + self.root = tk.Tk + + self.crafts_use = tk.StringVar + self.mode_use = tk.StringVar + + self.CraftsLabel = tk.Label + self.CraftsEntry = tk.Entry + self.ModeLabel = tk.Label + self.Mode_Box = ttk.Combobox + self.ButtonFrame = tk.Frame + + self.StartButton = tk.Button + self.StopButton = tk.Button + self.TkPosition = '133x92+60+600' + + self.StatusLabel = tk.Label + + self.start() + + def run(self): + self.root = tk.Tk() + + self.crafts_use = tk.StringVar(self.root, value='100') + self.mode_use = tk.StringVar(self.root, value='m25x1') + + self.CraftsLabel = tk.Label(self.root, text="Crafts:", font=("Helvetica", 10, "bold"), + background="grey") + self.CraftsLabel.grid(row=0, column=0, sticky='w') + self.CraftsEntry = tk.Entry(self.root, textvariable=self.crafts_use, font=("Helvetica", 10, "bold"), width='12') + + self.CraftsEntry.grid(row=0, column=1, columnspan=2) + + self.ModeLabel = tk.Label(self.root, text="Mode:", font=("Helvetica", 10, "bold"), + background="grey") + self.ModeLabel.grid(row=2, column=0, sticky='w') + self.Mode_Box = ttk.Combobox(self.root, textvariable=self.mode_use, font=("Helvetica", 8, "bold"), + width='11') + self.Mode_Box['values'] = ('m25x1', 'm10x2', 'm5x3', 'simple', 'm1x5_T', 'm1x5_X', 'house') + + self.Mode_Box.grid(row=2, column=1, columnspan=2, sticky='w') + + self.ButtonFrame = tk.Frame(self.root) + self.StartButton = tk.Button(self.ButtonFrame, text="Start", command=self.start_button_callback, width='8') + self.StartButton.grid(row=0, column=0) + self.StopButton = tk.Button(self.ButtonFrame, text="Stop", command=self.stop_button_callback, width='8', + state=tk.DISABLED) + self.StopButton.grid(row=0, column=1) + + self.ButtonFrame.grid(row=3, column=0, columnspan=2) + + self.StatusLabel = tk.Label(self.root, text="", font=("Helvetica", 10, "bold"), + background="grey") + self.StatusLabel.grid(row=4, column=0, columnspan=2) + + # self.ClearButton.pack(side="top") + self.root.geometry(self.TkPosition) + self.root.overrideredirect(1) # fenster ohne aussen rum :-) + # self.root.attributes('-alpha', 0.7) # fenster transparent + self.root.attributes('-topmost', 1) # fenster immer im vordergrund + # self.root.wm_attributes("-disabled", True) + self.root.configure(background='black') + self.root.mainloop() + + def start_button_callback(self): + self.StartButton.configure(state=tk.DISABLED) + self.StopButton.configure(state=tk.NORMAL) + self.StatusLabel.configure(text='') + self.run_mode = 'started' + + def stop_button_callback(self): + self.StartButton.configure(state=tk.NORMAL) + self.StopButton.configure(state=tk.DISABLED) + self.run_mode = 'stopped' + + def get_run_mode(self): + return self.run_mode + + def update_status_label(self, mode, energy_to_go): + text = mode + ": " + str(energy_to_go) + self.StatusLabel.configure(text=text) diff --git a/craft_table/main_craft.py b/craft_table/main_craft.py new file mode 100644 index 0000000..73c885f --- /dev/null +++ b/craft_table/main_craft.py @@ -0,0 +1,335 @@ +from time import time +import sys +import cv2 as cv +import keyboard +from utils import get_click_point, dig_point, check_for_ok_button, check_for_craft_button +import pydirectinput +from craft_overlay import CraftOverlay +from config_file import UserConfigs +from window_capture import WindowCapture +from vision import Vision + +MODE25X1 = "m25x1" +MODE10X2 = "m10x2" +MODE5X3 = "m5x3" +MODE_SIMPLE_STACK = "simple" +MODE1X5_T = "m1x5_T" +MODE1X5_X = "m1x5_X" +MODE_HOUSE = "house" + +def run(): + overlay = CraftOverlay() + + while True: + if overlay.run_mode == 'stopped': + overlay.update_status_label("stopped", "0") + overlay.run_mode = 'init' + continue + elif overlay.run_mode == 'started': + cv.waitKey(500) + elif overlay.run_mode == 'init': + continue + elif overlay.run_mode == 'finished': + overlay.update_status_label("finished", "0") + overlay.run_mode = 'init' + continue + else: + #boelk + pass + + mode = str(overlay.mode_use.get()) + run_target = int(overlay.crafts_use.get()) + + run_counter = 0 + + while True: + if overlay.run_mode == 'stopped': + break + + if run_counter >= run_target: + run_counter = 0 + overlay.run_mode = 'finished' + break + + ab1 = [420, 1180, 142, 142] + ab2 = [562, 1180, 142, 142] + ab3 = [704, 1180, 142, 142] + ab4 = [846, 1180, 142, 142] + ab5 = [986, 1180, 142, 142] + ab6 = [1128, 1180, 142, 142] + ab10 = [1698, 1180, 142, 142] + + if mode == MODE25X1: + click_square_center(ab1) + + # grid 770 1630 160 1020 172 o + dim = 172 + for i in range(0, 5, 1): + for e in range(0, 5, 1): + click_square_center([770 + (i * dim), 160 + (e * dim), dim, dim]) + if overlay.run_mode == 'stopped': + break + + click_craft_button() + + for i in range(0, 5, 1): + for e in range(0, 5, 1): + collect_grid_reverse_click([770 + (i * dim), 160 + (e * dim), dim, dim]) + run_counter = run_counter + 1 + overlay.update_status_label(mode, run_target - run_counter) + if overlay.run_mode == 'stopped': + break + click_square_center(ab10) + elif mode == MODE10X2: + for r in range(0, 2, 1): + # grid 770 1630 160 1020 172 + dim = 172 + if r == 0: + click_square_center(ab1) + start = 0 + target = 3 + else: + click_square_center(ab2) + start = 1 + target = 4 + + for i in range(start, target, 2): + for e in range(0, 5, 1): + click_square_center([770 + (i * dim), 160 + (e * dim), dim, dim]) + if overlay.run_mode == 'stopped': + break + + click_craft_button() + + for i in range(0, 4, 2): + for e in range(0, 5, 1): + collect_grid_reverse_click([770 + (i * dim), 160 + (e * dim), dim, dim]) + run_counter = run_counter + 1 + overlay.update_status_label(mode, run_target - run_counter) + if overlay.run_mode == 'stopped': + break + click_square_center(ab10) + elif mode == MODE5X3: + if overlay.run_mode == 'stopped': + break + for r in range(0, 3, 1): + # grid 770 1630 160 1020 172oo + dim = 172 + if r == 0: + click_square_center(ab1) + start = 0 + target = 1 + elif r == 1: + click_square_center(ab2) + start = 1 + target = 2 + else: + click_square_center(ab3) + start = 2 + target = 3 + + for i in range(start, target, 1): + for e in range(0, 5, 1): + click_square_center([770 + (i * dim), 160 + (e * dim), dim, dim]) + if overlay.run_mode == 'stopped': + break + + click_craft_button() + + for i in range(0, 1, 1): + if overlay.run_mode == 'stopped': + break + for e in range(0, 5, 1): + collect_grid_reverse_click([770 + (i * dim), 160 + (e * dim), dim, dim]) + run_counter = run_counter + 1 + overlay.update_status_label(mode, run_target - run_counter) + if overlay.run_mode == 'stopped': + break + click_square_center(ab10) + elif mode == MODE1X5_T: + dim = 172 + + tps = [] + tps.append(ab1) + tps.append([770 + (0 * dim), 160 + (2 * dim), dim, dim]) + tps.append([770 + (1 * dim), 160 + (2 * dim), dim, dim]) + tps.append(ab2) + tps.append([770 + (2 * dim), 160 + (1 * dim), dim, dim]) + tps.append([770 + (2 * dim), 160 + (2 * dim), dim, dim]) + tps.append([770 + (2 * dim), 160 + (3 * dim), dim, dim]) + + result_tp = [770 + (0 * dim), 160 + (3 * dim), dim, dim] + + for tp in tps: + click_square_center(tp) + if overlay.run_mode == 'stopped': + break + + click_craft_button() + if overlay.run_mode == 'stopped': + break + click_square_center(result_tp) + if overlay.run_mode == 'stopped': + break + click_square_center(ab10) + + run_counter = run_counter + 1 + overlay.update_status_label(mode, run_target - run_counter) + + if overlay.run_mode == 'stopped': + break + elif mode == MODE1X5_X: + dim = 172 + + tps = [] + tps.append(ab1) + tps.append([770 + (0 * dim), 160 + (3 * dim), dim, dim]) + tps.append([770 + (1 * dim), 160 + (3 * dim), dim, dim]) + tps.append([770 + (2 * dim), 160 + (3 * dim), dim, dim]) + tps.append([770 + (1 * dim), 160 + (4 * dim), dim, dim]) + tps.append([770 + (1 * dim), 160 + (2 * dim), dim, dim]) + + result_tp = [770 + (0 * dim), 160 + (4 * dim), dim, dim] + + for tp in tps: + click_square_center(tp) + if overlay.run_mode == 'stopped': + break + + click_craft_button() + if overlay.run_mode == 'stopped': + break + click_square_center(result_tp) + if overlay.run_mode == 'stopped': + break + click_square_center(ab10) + + run_counter = run_counter + 1 + overlay.update_status_label(mode, run_target - run_counter) + + if overlay.run_mode == 'stopped': + break + elif mode == MODE_HOUSE: + dim = 172 + + cements = [] + cements.append([770 + (0 * dim), 160 + (4 * dim), dim, dim]) + cements.append([770 + (1 * dim), 160 + (4 * dim), dim, dim]) + cements.append([770 + (2 * dim), 160 + (4 * dim), dim, dim]) + cements.append([770 + (3 * dim), 160 + (4 * dim), dim, dim]) + cements.append([770 + (4 * dim), 160 + (4 * dim), dim, dim]) + + bricks = [] + bricks.append([770 + (0 * dim), 160 + (3 * dim), dim, dim]) + bricks.append([770 + (1 * dim), 160 + (3 * dim), dim, dim]) + bricks.append([770 + (3 * dim), 160 + (3 * dim), dim, dim]) + bricks.append([770 + (4 * dim), 160 + (3 * dim), dim, dim]) + bricks.append([770 + (0 * dim), 160 + (2 * dim), dim, dim]) + bricks.append([770 + (2 * dim), 160 + (2 * dim), dim, dim]) + bricks.append([770 + (4 * dim), 160 + (2 * dim), dim, dim]) + + boards = [] + boards.append([770 + (2 * dim), 160 + (3 * dim), dim, dim]) + boards.append([770 + (0 * dim), 160 + (1 * dim), dim, dim]) + boards.append([770 + (1 * dim), 160 + (1 * dim), dim, dim]) + boards.append([770 + (2 * dim), 160 + (1 * dim), dim, dim]) + boards.append([770 + (3 * dim), 160 + (1 * dim), dim, dim]) + boards.append([770 + (4 * dim), 160 + (1 * dim), dim, dim]) + boards.append([770 + (2 * dim), 160 + (0 * dim), dim, dim]) + + glasses = [] + glasses.append([770 + (1 * dim), 160 + (2 * dim), dim, dim]) + glasses.append([770 + (3 * dim), 160 + (2 * dim), dim, dim]) + + tp_hammer = [770 + (1 * dim), 160 + (0 * dim), dim, dim] + tp_nails = [770 + (3 * dim), 160 + (0 * dim), dim, dim] + tp_result = [770 + (0 * dim), 160 + (4 * dim), dim, dim] + + click_square_center(ab1) + + for cement in cements: + click_square_center(cement) + if overlay.run_mode == 'stopped': + break + + click_square_center(ab2) + + for brick in bricks: + click_square_center(brick) + if overlay.run_mode == 'stopped': + break + + click_square_center(ab3) + + for board in boards: + click_square_center(board) + if overlay.run_mode == 'stopped': + break + + click_square_center(ab4) + + for glass in glasses: + click_square_center(glass) + if overlay.run_mode == 'stopped': + break + + click_square_center(ab5) + click_square_center(tp_hammer) + if overlay.run_mode == 'stopped': + break + + click_square_center(ab6) + click_square_center(tp_nails) + if overlay.run_mode == 'stopped': + break + + click_craft_button() + if overlay.run_mode == 'stopped': + break + + click_square_center(tp_result) + click_square_center(ab10) + if overlay.run_mode == 'stopped': + break + + run_counter = run_counter + 1 + overlay.update_status_label(mode, run_target - run_counter) + + if overlay.run_mode == 'stopped': + break + elif mode == MODE_SIMPLE_STACK: + # initialize the StunWindowCapture class + capture_config = UserConfigs() + #capture_window = WindowCapture(None, "screen_conf", capture_config) + + capture_window2 = WindowCapture(None, "bla", capture_config) + + # initialize the StunVision class + vision_stun = Vision() + # odig_overlay = DiggingOverlay(config) + check_for_craft_button(capture_window2, vision_stun, capture_config) + run_counter = run_counter + 1 + overlay.update_status_label(mode, run_target - run_counter) + #check_for_ok_button(capture_window, vision_stun, capture_config) + +def collect_grid_reverse_click(square): + pydirectinput.moveTo(get_click_point(square)[0], get_click_point(square)[1]) + cv.waitKey(50) + pydirectinput.mouseUp() + cv.waitKey(50) + pydirectinput.mouseDown() + cv.waitKey(100) + +def click_craft_button(): + dig_point(1910, 330, 100) + cv.waitKey(1500) + + +def click_square_center(square): + cp = get_click_point(square) + dig_point(cp[0], cp[1], 100) + cv.waitKey(100) + +if __name__ == "__main__": + run() + sys.exit() diff --git a/craft_table/ok_button.jpg b/craft_table/ok_button.jpg new file mode 100644 index 0000000..d78f586 Binary files /dev/null and b/craft_table/ok_button.jpg differ diff --git a/craft_table/play.jpg b/craft_table/play.jpg new file mode 100644 index 0000000..4464a07 Binary files /dev/null and b/craft_table/play.jpg differ diff --git a/craft_table/screen_panel.jpg b/craft_table/screen_panel.jpg new file mode 100644 index 0000000..a680413 Binary files /dev/null and b/craft_table/screen_panel.jpg differ