refactor config
This commit is contained in:
@@ -1,7 +1,7 @@
|
||||
import random
|
||||
from time import time
|
||||
from window_capture import WindowCapture
|
||||
from stun_vision import StunVision
|
||||
from vision import Vision
|
||||
import cv2 as cv
|
||||
import pytesseract
|
||||
from hsvfilter import HsvFilter
|
||||
@@ -19,7 +19,7 @@ def run():
|
||||
# initialize the StunWindowCapture class
|
||||
try:
|
||||
capture_window = WindowCapture(
|
||||
None, "stun", config)
|
||||
None, "dig", config)
|
||||
video_mode = False
|
||||
except:
|
||||
# StunWindowCapture.list_window_names()
|
||||
@@ -28,7 +28,7 @@ def run():
|
||||
video_mode = True
|
||||
|
||||
# initialize the StunVision class
|
||||
vision_stun = StunVision()
|
||||
vision_stun = Vision()
|
||||
# initialize the StunOverlay class
|
||||
hsv_filter = HsvFilter(0, 0, 124, 15, 255, 168, 0, 255, 0, 0)
|
||||
|
||||
@@ -67,24 +67,24 @@ def run():
|
||||
#continue
|
||||
|
||||
needles = []
|
||||
needles.append(cv.imread("wtf.jpg", cv.IMREAD_UNCHANGED))
|
||||
needles.append(cv.imread("Brown0.jpg", cv.IMREAD_UNCHANGED))
|
||||
needles.append(cv.imread("1.jpg", cv.IMREAD_UNCHANGED))
|
||||
needles.append(cv.imread("2.jpg", cv.IMREAD_UNCHANGED))
|
||||
needles.append(cv.imread("3.jpg", cv.IMREAD_UNCHANGED))
|
||||
needles.append(cv.imread("4.jpg", cv.IMREAD_UNCHANGED))
|
||||
needles.append(cv.imread("H1.jpg", cv.IMREAD_UNCHANGED))
|
||||
needles.append(cv.imread("H2.jpg", cv.IMREAD_UNCHANGED))
|
||||
needles.append(cv.imread("H3.jpg", cv.IMREAD_UNCHANGED))
|
||||
needles.append(cv.imread("H4.jpg", cv.IMREAD_UNCHANGED))
|
||||
needles.append(cv.imread("D1.jpg", cv.IMREAD_UNCHANGED))
|
||||
needles.append(cv.imread("D2.jpg", cv.IMREAD_UNCHANGED))
|
||||
needles.append(cv.imread("D3.jpg", cv.IMREAD_UNCHANGED))
|
||||
needles.append(cv.imread("D3.jpg", cv.IMREAD_UNCHANGED))
|
||||
needles.append(cv.imread("D4.jpg", cv.IMREAD_UNCHANGED))
|
||||
needles.append(cv.imread("D5.jpg", cv.IMREAD_UNCHANGED))
|
||||
needles.append(cv.imread("D6.jpg", cv.IMREAD_UNCHANGED))
|
||||
needles.append(cv.imread("D7.jpg", cv.IMREAD_UNCHANGED))
|
||||
#needles.append(cv.imread("wtf.jpg", cv.IMREAD_UNCHANGED))
|
||||
needles.append(cv.imread("dig/Brown0.jpg", cv.IMREAD_UNCHANGED))
|
||||
needles.append(cv.imread("dig/1.jpg", cv.IMREAD_UNCHANGED))
|
||||
needles.append(cv.imread("dig/2.jpg", cv.IMREAD_UNCHANGED))
|
||||
needles.append(cv.imread("dig/3.jpg", cv.IMREAD_UNCHANGED))
|
||||
needles.append(cv.imread("dig/4.jpg", cv.IMREAD_UNCHANGED))
|
||||
#needles.append(cv.imread("H1.jpg", cv.IMREAD_UNCHANGED))
|
||||
#needles.append(cv.imread("H2.jpg", cv.IMREAD_UNCHANGED))
|
||||
#needles.append(cv.imread("H3.jpg", cv.IMREAD_UNCHANGED))
|
||||
#needles.append(cv.imread("H4.jpg", cv.IMREAD_UNCHANGED))
|
||||
# needles.append(cv.imread("D1.jpg", cv.IMREAD_UNCHANGED))
|
||||
# needles.append(cv.imread("D2.jpg", cv.IMREAD_UNCHANGED))
|
||||
# needles.append(cv.imread("D3.jpg", cv.IMREAD_UNCHANGED))
|
||||
# needles.append(cv.imread("D3.jpg", cv.IMREAD_UNCHANGED))
|
||||
# needles.append(cv.imread("D4.jpg", cv.IMREAD_UNCHANGED))
|
||||
# needles.append(cv.imread("D5.jpg", cv.IMREAD_UNCHANGED))
|
||||
# needles.append(cv.imread("D6.jpg", cv.IMREAD_UNCHANGED))
|
||||
# needles.append(cv.imread("D7.jpg", cv.IMREAD_UNCHANGED))
|
||||
|
||||
for needle in needles:
|
||||
# do object detection
|
||||
@@ -111,8 +111,8 @@ def run():
|
||||
|
||||
left = int(round(rectangles[0][0] / size, 0)) # 4
|
||||
down = int(round(rectangles[0][1] / size, 0)) # 23
|
||||
offset_left = config.returnStunWindowPos()[2]
|
||||
offset_down = config.returnStunWindowPos()[3]
|
||||
offset_left = config.returnMagicWindowPos()[2]
|
||||
offset_down = config.returnMagicWindowPos()[3]
|
||||
# 167 1055 start
|
||||
# 3x47 left 26x right to 30
|
||||
# 1x down 22x up to 24
|
||||
@@ -124,15 +124,15 @@ def run():
|
||||
for i in range(start_left, start_left + (size * 30), size):
|
||||
pydirectinput.moveTo(i + offset_left, f + offset_down)
|
||||
pydirectinput.mouseDown()
|
||||
w = random.randint(1, 100)
|
||||
cv.waitKey(150 + w)
|
||||
w = random.randint(1, 50)
|
||||
cv.waitKey(850 + w)
|
||||
pydirectinput.mouseUp()
|
||||
if keyboard.is_pressed('p') == True or pause == True:
|
||||
pause = True
|
||||
break
|
||||
|
||||
screenshot = capture_window.get_screenshot()
|
||||
rectangles = vision_stun.find(screenshot, cv.imread("ok_button.jpg", cv.IMREAD_UNCHANGED), 0.8, 1)
|
||||
rectangles = vision_stun.find(screenshot, cv.imread("dig/ok_button.jpg", cv.IMREAD_UNCHANGED), 0.8, 1)
|
||||
# draw the detection results onto the original image
|
||||
output_image = vision_stun.draw_rectangles(screenshot, rectangles)
|
||||
if len(rectangles) == 1:
|
||||
|
||||
Reference in New Issue
Block a user