re-implement equipment with masks

This commit is contained in:
2022-05-15 00:50:42 +02:00
parent e7b15cc48a
commit 883a6f77aa
292 changed files with 454 additions and 173 deletions

View File

@@ -13,11 +13,14 @@ EMITTER_MUSH = "mushroom"
EMITTER_AMU = "amulett"
EMITTER_SWORD = "sword"
EMITTER_STAFF = "staff"
EMITTER_WAND = "wand"
EMITTER_RING = "ring"
HSV_DEFAULT = HsvFilter(0, 0, 0, 179, 255, 255, 0, 0, 0, 0)
def run():
EMITTER_TO_USE = EMITTER_STAFF
EMITTER_TO_USE = EMITTER_RING
SPAWN_COUNT = 1
# initialize the user-class
config = UserConfigs()
@@ -32,38 +35,50 @@ def run():
# capture_window = cv.VideoCapture("snip_slam.mp4")
video_mode = True
# initialize the StunVision class
vision_stun = Vision()
loop_time = time()
needles = []
hsv = []
hsvs = []
tresholds = []
masks = []
if EMITTER_TO_USE == EMITTER_MUSH:
include_books(needles, hsv)
include_mushs(needles, hsv)
include_pots(needles, hsv)
include_bags(needles, hsv)
include_books(needles, hsvs, tresholds, masks)
include_mushs(needles, hsvs, tresholds, masks)
include_pots(needles, hsvs, tresholds, masks)
include_bags(needles, hsvs, tresholds, masks)
elif EMITTER_TO_USE == EMITTER_AMU:
include_books(needles, hsv)
include_amus(needles, hsv)
include_bags(needles, hsv)
include_books(needles, hsvs, tresholds, masks)
include_amus(needles, hsvs, tresholds, masks)
include_bags(needles, hsvs, tresholds, masks)
elif EMITTER_TO_USE == EMITTER_MAIN:
include_books(needles, hsv)
include_chests(needles, hsv)
include_keys(needles, hsv)
include_pots(needles, hsv)
include_bags(needles, hsv)
include_coins(needles, hsv)
#include_runes(needles, hsv)
include_books(needles, hsvs, tresholds, masks)
include_chests(needles, hsvs, tresholds, masks)
include_keys(needles, hsvs, tresholds, masks)
include_pots(needles, hsvs, tresholds, masks)
include_bags(needles, hsvs, tresholds, masks)
include_coins(needles, hsvs, tresholds, masks)
include_runes(needles, hsvs, tresholds, masks)
elif EMITTER_TO_USE == EMITTER_SWORD:
include_books(needles, hsv)
include_swords(needles, hsv)
include_bags(needles, hsv)
include_books(needles, hsvs, tresholds, masks)
include_swords(needles, hsvs, tresholds, masks)
include_bags(needles, hsvs, tresholds, masks)
elif EMITTER_TO_USE == EMITTER_STAFF:
include_books(needles, hsv)
include_staffs(needles, hsv)
include_bags(needles, hsv)
include_books(needles, hsvs, tresholds, masks)
include_staffs(needles, hsvs, tresholds, masks)
include_bags(needles, hsvs, tresholds, masks)
elif EMITTER_TO_USE == EMITTER_RING:
#include_books(needles, hsvs, tresholds, masks)
include_rings(needles, hsvs, tresholds, masks)
#include_bags(needles, hsvs, tresholds, masks)
elif EMITTER_TO_USE == EMITTER_WAND:
include_books(needles, hsvs, tresholds, masks)
include_wands(needles, hsvs, tresholds, masks)
include_bags(needles, hsvs, tresholds, masks)
pause = True
while True:
@@ -97,7 +112,7 @@ def run():
if len(spawn_1) == 1:
spawn_button_active = True
points = vision_stun.get_click_points(spawn_1)
for i in range(0, 15, 1):
for i in range(0, SPAWN_COUNT, 1):
click_point(config, points[0][0], points[0][1])
if keyboard.is_pressed('p') == True or pause == True:
pause = True
@@ -112,17 +127,27 @@ def run():
break
continue
cv.waitKey(500)
for rer in range(0, len(needles), 1):
while True:
# do object detection
screenshot = capture_window.get_screenshot()
processed_screenshot = vision_stun.apply_hsv_filter(screenshot, hsv[rer])
processed_needle = vision_stun.apply_hsv_filter(needles[rer], hsv[rer])
rectangles = vision_stun.find(processed_screenshot, processed_needle, 0.8, 5)
processed_screenshot = vision_stun.apply_hsv_filter(screenshot, hsvs[rer])
processed_needle = vision_stun.apply_hsv_filter(needles[rer], hsvs[rer])
trho = 0.8
mask = None
norm = False
try:
trho = tresholds[rer]
mask = masks[rer]
norm = True
except:
pass
rectangles = vision_stun.find(processed_screenshot, processed_needle, trho, 5, norm, mask)
# draw the detection results onto the original image
# output_image = vision_stun.draw_rectangles(processed_screenshot, rectangles)
# cv.imshow("output_image", output_image)
# cv.waitKey(150)
output_image = vision_stun.draw_rectangles(processed_screenshot, rectangles)
cv.imshow("output_image", output_image)
cv.waitKey(150)
if len(rectangles) is not 5:
break
@@ -133,21 +158,21 @@ def run():
points = vision_stun.get_click_points(rectangles)
check_and_move_tile(capture_window, vision_stun, config,
[70, 70, rectangles[0][0] + 70, rectangles[0][1]], needles[rer],
hsv[rer],
hsvs[rer],
points[1], [points[0][0] + 70, points[0][1]])
if keyboard.is_pressed('p') == True or pause == True:
pause = True
break
check_and_move_tile(capture_window, vision_stun, config,
[70, 70, rectangles[0][0], rectangles[0][1] + 70], needles[rer],
hsv[rer],
hsvs[rer],
points[2], [points[0][0], points[0][1] + 70])
if keyboard.is_pressed('p') == True or pause == True:
pause = True
break
check_and_move_tile(capture_window, vision_stun, config,
[70, 70, rectangles[0][0] + 70, rectangles[0][1] + 70],
needles[rer], hsv[rer],
needles[rer], hsvs[rer],
points[3], [points[0][0] + 70, points[0][1] + 70])
move_tile(config, points[4], [points[0][0], points[0][1]])
if keyboard.is_pressed('p') == True or pause == True:
@@ -159,186 +184,431 @@ def run():
cv.waitKey(150)
def include_chests(needles, hsv):
needles.append(cv.imread("equip/chest_1_32.jpg", cv.IMREAD_UNCHANGED))
hsv.append(HsvFilter(13, 40, 85, 135, 255, 255, 0, 0, 55, 53))
#needles.append(cv.imread("equip/chest_1_32-removebg-preview.png", cv.IMREAD_UNCHANGED))
#hsv.append(HsvFilter(0, 0, 0, 179, 255, 255, 0, 0, 0, 0))
needles.append(cv.imread("equip/chest_2_32.jpg", cv.IMREAD_UNCHANGED))
hsv.append(HsvFilter(14, 0, 0, 179, 255, 255, 0, 0, 0, 0))
needles.append(cv.imread("equip/chest_3_32.jpg", cv.IMREAD_UNCHANGED))
hsv.append(HsvFilter(0, 0, 0, 179, 255, 255, 0, 0, 0, 0))
def include_chests(needles, hsv, tresh, mask):
needles.append(cv.imread("equip/chests/chest_1_32.jpg", cv.IMREAD_UNCHANGED))
hsv.append(HSV_DEFAULT)
tresh.append(0.95)
mask.append(cv.imread("equip/chests/chest_1_32-mask.png", cv.IMREAD_COLOR))
needles.append(cv.imread("equip/chests/chest_2_32.jpg", cv.IMREAD_UNCHANGED))
hsv.append(HSV_DEFAULT)
tresh.append(0.95)
mask.append(cv.imread("equip/chests/chest_2_32-mask.png", cv.IMREAD_COLOR))
needles.append(cv.imread("equip/chests/chest_3_32.jpg", cv.IMREAD_UNCHANGED))
hsv.append(HSV_DEFAULT)
tresh.append(0.91)
mask.append(cv.imread("equip/chests/chest_3_32-mask.png", cv.IMREAD_COLOR))
def include_books(needles, hsv, tresh, mask):
needles.append(cv.imread("equip/books/book_1_32.jpg", cv.IMREAD_UNCHANGED))
hsv.append(HSV_DEFAULT)
tresh.append(0.91)
mask.append(cv.imread("equip/books/book_1_32-mask.png", cv.IMREAD_COLOR))
needles.append(cv.imread("equip/books/book_2_32.jpg", cv.IMREAD_UNCHANGED))
hsv.append(HSV_DEFAULT)
tresh.append(0.9)
mask.append(cv.imread("equip/books/book_2_32-mask.png", cv.IMREAD_COLOR))
needles.append(cv.imread("equip/books/book_3_32.jpg", cv.IMREAD_UNCHANGED))
hsv.append(HSV_DEFAULT)
tresh.append(0.87)
mask.append(cv.imread("equip/books/book_3_32-mask.png", cv.IMREAD_COLOR))
needles.append(cv.imread("equip/books/book_4_32.jpg", cv.IMREAD_UNCHANGED))
hsv.append(HSV_DEFAULT)
tresh.append(0.93)
mask.append(cv.imread("equip/books/book_4_32-mask.png", cv.IMREAD_COLOR))
needles.append(cv.imread("equip/books/book_5_32.jpg", cv.IMREAD_UNCHANGED))
hsv.append(HSV_DEFAULT)
tresh.append(0.91)
mask.append(cv.imread("equip/books/book_5_32-mask.png", cv.IMREAD_COLOR))
needles.append(cv.imread("equip/books/book_6_32.jpg", cv.IMREAD_UNCHANGED))
hsv.append(HSV_DEFAULT)
tresh.append(0.9)
mask.append(cv.imread("equip/books/book_6_32-mask.png", cv.IMREAD_COLOR))
needles.append(cv.imread("equip/books/book_7_32.jpg", cv.IMREAD_UNCHANGED))
hsv.append(HSV_DEFAULT)
tresh.append(0.91)
mask.append(cv.imread("equip/books/book_7_32-mask.png", cv.IMREAD_COLOR))
needles.append(cv.imread("equip/books/book_8_32.jpg", cv.IMREAD_UNCHANGED))
hsv.append(HSV_DEFAULT)
tresh.append(0.91)
mask.append(cv.imread("equip/books/book_8_32-mask.png", cv.IMREAD_COLOR))
def include_books(needles, hsv):
needles.append(cv.imread("equip/book_1_32.jpg", cv.IMREAD_UNCHANGED))
hsv.append(HsvFilter(0, 128, 0, 179, 255, 255, 0, 0, 0, 0))
needles.append(cv.imread("equip/book_2_32.jpg", cv.IMREAD_UNCHANGED))
hsv.append(HsvFilter(49, 0, 0, 179, 255, 255, 0, 0, 0, 0))
needles.append(cv.imread("equip/book_3_32.jpg", cv.IMREAD_UNCHANGED))
hsv.append(HsvFilter(0, 0, 0, 179, 255, 196, 0, 0, 0, 0))
needles.append(cv.imread("equip/book_4_32.jpg", cv.IMREAD_UNCHANGED))
hsv.append(HsvFilter(0, 156, 0, 179, 255, 255, 0, 0, 0, 0))
needles.append(cv.imread("equip/book_5_32.jpg", cv.IMREAD_UNCHANGED))
hsv.append(HsvFilter(0, 95, 137, 179, 255, 255, 0, 0, 0, 0))
def include_keys(needles, hsv, tresh, mask):
needles.append(cv.imread("equip/keys/key_1_32.jpg", cv.IMREAD_UNCHANGED))
hsv.append(HSV_DEFAULT)
tresh.append(0.91)
mask.append(cv.imread("equip/keys/key_1_32-mask.png", cv.IMREAD_COLOR))
needles.append(cv.imread("equip/keys/key_2_32.jpg", cv.IMREAD_UNCHANGED))
hsv.append(HSV_DEFAULT)
tresh.append(0.91)
mask.append(cv.imread("equip/keys/key_2_32-mask.png", cv.IMREAD_COLOR))
needles.append(cv.imread("equip/keys/key_3_32.jpg", cv.IMREAD_UNCHANGED))
hsv.append(HSV_DEFAULT)
tresh.append(0.91)
mask.append(cv.imread("equip/keys/key_3_32-mask.png", cv.IMREAD_COLOR))
def include_keys(needles, hsv):
needles.append(cv.imread("equip/key_1_32.jpg", cv.IMREAD_UNCHANGED))
hsv.append(HsvFilter(0, 0, 0, 179, 255, 255, 0, 0, 0, 0))
needles.append(cv.imread("equip/key_2_32.jpg", cv.IMREAD_UNCHANGED))
hsv.append(HsvFilter(0, 0, 0, 179, 255, 255, 0, 0, 0, 0))
needles.append(cv.imread("equip/key_3_32.jpg", cv.IMREAD_UNCHANGED))
hsv.append(HsvFilter(0, 0, 0, 179, 255, 255, 0, 0, 0, 0))
def include_pots(needles, hsv, tresh, mask):
needles.append(cv.imread("equip/pots/pot_1_32.jpg", cv.IMREAD_UNCHANGED))
hsv.append(HSV_DEFAULT)
tresh.append(0.91)
mask.append(cv.imread("equip/pots/pot_1_32-mask.png", cv.IMREAD_COLOR))
needles.append(cv.imread("equip/pots/pot_2_32.jpg", cv.IMREAD_UNCHANGED))
hsv.append(HSV_DEFAULT)
tresh.append(0.91)
mask.append(cv.imread("equip/pots/pot_2_32-mask.png", cv.IMREAD_COLOR))
needles.append(cv.imread("equip/pots/pot_3_32.jpg", cv.IMREAD_UNCHANGED))
hsv.append(HSV_DEFAULT)
tresh.append(0.94)
mask.append(cv.imread("equip/pots/pot_3_32-mask.png", cv.IMREAD_COLOR))
needles.append(cv.imread("equip/pots/pot_4_32.jpg", cv.IMREAD_UNCHANGED))
hsv.append(HSV_DEFAULT)
tresh.append(0.91)
mask.append(cv.imread("equip/pots/pot_4_32-mask.png", cv.IMREAD_COLOR))
needles.append(cv.imread("equip/pots/pot_5_32.jpg", cv.IMREAD_UNCHANGED))
hsv.append(HSV_DEFAULT)
tresh.append(0.91)
mask.append(cv.imread("equip/pots/pot_5_32-mask.png", cv.IMREAD_COLOR))
needles.append(cv.imread("equip/pots/pot_6_32.jpg", cv.IMREAD_UNCHANGED))
hsv.append(HSV_DEFAULT)
tresh.append(0.91)
mask.append(cv.imread("equip/pots/pot_6_32-mask.png", cv.IMREAD_COLOR))
def include_pots(needles, hsv):
needles.append(cv.imread("equip/pot_1_32.jpg", cv.IMREAD_UNCHANGED))
hsv.append(HsvFilter(0, 0, 0, 179, 255, 255, 0, 0, 0, 0))
needles.append(cv.imread("equip/pot_2_32.jpg", cv.IMREAD_UNCHANGED))
hsv.append(HsvFilter(0, 0, 0, 179, 255, 255, 0, 0, 0, 0))
needles.append(cv.imread("equip/pot_3_32.jpg", cv.IMREAD_UNCHANGED))
hsv.append(HsvFilter(0, 0, 0, 179, 255, 255, 0, 0, 0, 0))
needles.append(cv.imread("equip/pot_4_32.jpg", cv.IMREAD_UNCHANGED))
hsv.append(HsvFilter(0, 0, 0, 179, 255, 255, 0, 0, 0, 0))
needles.append(cv.imread("equip/pot_5_32.jpg", cv.IMREAD_UNCHANGED))
hsv.append(HsvFilter(0, 0, 0, 179, 255, 255, 0, 0, 0, 0))
needles.append(cv.imread("equip/pot_6_32.jpg", cv.IMREAD_UNCHANGED))
hsv.append(HsvFilter(0, 0, 0, 179, 255, 255, 0, 0, 0, 0))
def include_bags(needles, hsv, tresh, mask):
needles.append(cv.imread("equip/bags/bag_1_32.jpg", cv.IMREAD_UNCHANGED))
hsv.append(HSV_DEFAULT)
tresh.append(0.91)
mask.append(cv.imread("equip/bags/bag_1_32-mask.png", cv.IMREAD_COLOR))
needles.append(cv.imread("equip/bags/bag_2_32.jpg", cv.IMREAD_UNCHANGED))
hsv.append(HSV_DEFAULT)
tresh.append(0.88)
mask.append(cv.imread("equip/bags/bag_2_32-mask.png", cv.IMREAD_COLOR))
needles.append(cv.imread("equip/bags/bag_3_32.jpg", cv.IMREAD_UNCHANGED))
hsv.append(HSV_DEFAULT)
tresh.append(0.91)
mask.append(cv.imread("equip/bags/bag_3_32-mask.png", cv.IMREAD_COLOR))
needles.append(cv.imread("equip/bags/bag_4_32.jpg", cv.IMREAD_UNCHANGED))
hsv.append(HSV_DEFAULT)
tresh.append(0.91)
mask.append(cv.imread("equip/bags/bag_4_32-mask.png", cv.IMREAD_COLOR))
needles.append(cv.imread("equip/bags/bag_5_32.jpg", cv.IMREAD_UNCHANGED))
hsv.append(HSV_DEFAULT)
tresh.append(0.91)
mask.append(cv.imread("equip/bags/bag_5_32-mask.png", cv.IMREAD_COLOR))
needles.append(cv.imread("equip/bags/bag_6_32.jpg", cv.IMREAD_UNCHANGED))
hsv.append(HSV_DEFAULT)
tresh.append(0.91)
mask.append(cv.imread("equip/bags/bag_6_32-mask.png", cv.IMREAD_COLOR))
needles.append(cv.imread("equip/bags/bag_7_32.jpg", cv.IMREAD_UNCHANGED))
hsv.append(HSV_DEFAULT)
tresh.append(0.91)
mask.append(cv.imread("equip/bags/bag_7_32-mask.png", cv.IMREAD_COLOR))
def include_coins(needles, hsv, tresh, mask):
needles.append(cv.imread("equip/coins/coin_1_32.jpg", cv.IMREAD_UNCHANGED))
hsv.append(HSV_DEFAULT)
tresh.append(0.91)
mask.append(cv.imread("equip/coins/coin_1_32-mask.png", cv.IMREAD_COLOR))
needles.append(cv.imread("equip/coins/coin_2_32.jpg", cv.IMREAD_UNCHANGED))
hsv.append(HSV_DEFAULT)
tresh.append(0.91)
mask.append(cv.imread("equip/coins/coin_2_32-mask.png", cv.IMREAD_COLOR))
needles.append(cv.imread("equip/coins/coin_3_32.jpg", cv.IMREAD_UNCHANGED))
hsv.append(HSV_DEFAULT)
tresh.append(0.91)
mask.append(cv.imread("equip/coins/coin_3_32-mask.png", cv.IMREAD_COLOR))
needles.append(cv.imread("equip/coins/coin_4_32.jpg", cv.IMREAD_UNCHANGED))
hsv.append(HSV_DEFAULT)
tresh.append(0.91)
mask.append(cv.imread("equip/coins/coin_4_32-mask.png", cv.IMREAD_COLOR))
needles.append(cv.imread("equip/coins/coin_5_32.jpg", cv.IMREAD_UNCHANGED))
hsv.append(HSV_DEFAULT)
tresh.append(0.91)
mask.append(cv.imread("equip/coins/coin_5_32-mask.png", cv.IMREAD_COLOR))
def include_runes(needles, hsv, tresh, mask):
needles.append(cv.imread("equip/runes/rune_1_32.jpg", cv.IMREAD_UNCHANGED))
hsv.append(HSV_DEFAULT)
tresh.append(0.91)
mask.append(cv.imread("equip/runes/rune_1_32-mask.png", cv.IMREAD_COLOR))
needles.append(cv.imread("equip/runes/rune_2_32.jpg", cv.IMREAD_UNCHANGED))
hsv.append(HSV_DEFAULT)
tresh.append(0.91)
mask.append(cv.imread("equip/runes/rune_2_32-mask.png", cv.IMREAD_COLOR))
def include_bags(needles, hsv):
needles.append(cv.imread("equip/bag_1_32.jpg", cv.IMREAD_UNCHANGED))
hsv.append(HsvFilter(0, 0, 0, 179, 255, 255, 0, 0, 0, 0))
needles.append(cv.imread("equip/bag_2_32.jpg", cv.IMREAD_UNCHANGED))
hsv.append(HsvFilter(0, 0, 0, 179, 255, 255, 0, 0, 0, 0))
needles.append(cv.imread("equip/bag_3_32.jpg", cv.IMREAD_UNCHANGED))
hsv.append(HsvFilter(0, 0, 0, 179, 255, 255, 0, 0, 0, 0))
needles.append(cv.imread("equip/bag_4_32.jpg", cv.IMREAD_UNCHANGED))
hsv.append(HsvFilter(0, 0, 0, 179, 255, 255, 0, 0, 0, 0))
needles.append(cv.imread("equip/bag_5_32.jpg", cv.IMREAD_UNCHANGED))
hsv.append(HsvFilter(0, 0, 0, 179, 255, 255, 0, 0, 0, 0))
needles.append(cv.imread("equip/bag_6_32.jpg", cv.IMREAD_UNCHANGED))
hsv.append(HsvFilter(0, 0, 0, 179, 255, 255, 0, 0, 0, 0))
def include_mushs(needles, hsv, tresh, mask):
needles.append(cv.imread("equip/mushrooms/mush_1_32.jpg", cv.IMREAD_UNCHANGED))
hsv.append(HSV_DEFAULT)
tresh.append(0.93)
mask.append(cv.imread("equip/mushrooms/mush_1_32-mask.png", cv.IMREAD_COLOR))
needles.append(cv.imread("equip/mushrooms/mush_2_32.jpg", cv.IMREAD_UNCHANGED))
hsv.append(HSV_DEFAULT)
tresh.append(0.93)
mask.append(cv.imread("equip/mushrooms/mush_2_32-mask.png", cv.IMREAD_COLOR))
needles.append(cv.imread("equip/mushrooms/mush_3_32.jpg", cv.IMREAD_UNCHANGED))
hsv.append(HSV_DEFAULT)
tresh.append(0.93)
mask.append(cv.imread("equip/mushrooms/mush_3_32-mask.png", cv.IMREAD_COLOR))
needles.append(cv.imread("equip/mushrooms/mush_4_32.jpg", cv.IMREAD_UNCHANGED))
hsv.append(HSV_DEFAULT)
tresh.append(0.93)
mask.append(cv.imread("equip/mushrooms/mush_4_32-mask.png", cv.IMREAD_COLOR))
needles.append(cv.imread("equip/mushrooms/mush_5_32.jpg", cv.IMREAD_UNCHANGED))
hsv.append(HSV_DEFAULT)
tresh.append(0.93)
mask.append(cv.imread("equip/mushrooms/mush_5_32-mask.png", cv.IMREAD_COLOR))
needles.append(cv.imread("equip/mushrooms/mush_6_32.jpg", cv.IMREAD_UNCHANGED))
hsv.append(HSV_DEFAULT)
tresh.append(0.93)
mask.append(cv.imread("equip/mushrooms/mush_6_32-mask.png", cv.IMREAD_COLOR))
needles.append(cv.imread("equip/mushrooms/mush_7_32.jpg", cv.IMREAD_UNCHANGED))
hsv.append(HSV_DEFAULT)
tresh.append(0.93)
mask.append(cv.imread("equip/mushrooms/mush_7_32-mask.png", cv.IMREAD_COLOR))
def include_amus(needles, hsv, tresh, mask):
needles.append(cv.imread("equip/amus/amu_1_32.jpg", cv.IMREAD_UNCHANGED))
hsv.append(HSV_DEFAULT)
tresh.append(0.93)
mask.append(cv.imread("equip/amus/amu_1_32-mask.png", cv.IMREAD_COLOR))
needles.append(cv.imread("equip/amus/amu_2_32.jpg", cv.IMREAD_UNCHANGED))
hsv.append(HSV_DEFAULT)
tresh.append(0.93)
mask.append(cv.imread("equip/amus/amu_2_32-mask.png", cv.IMREAD_COLOR))
needles.append(cv.imread("equip/amus/amu_3_32.jpg", cv.IMREAD_UNCHANGED))
hsv.append(HSV_DEFAULT)
tresh.append(0.93)
mask.append(cv.imread("equip/amus/amu_3_32-mask.png", cv.IMREAD_COLOR))
needles.append(cv.imread("equip/amus/amu_4_32.jpg", cv.IMREAD_UNCHANGED))
hsv.append(HSV_DEFAULT)
tresh.append(0.93)
mask.append(cv.imread("equip/amus/amu_4_32-mask.png", cv.IMREAD_COLOR))
needles.append(cv.imread("equip/amus/amu_5_32.jpg", cv.IMREAD_UNCHANGED))
hsv.append(HSV_DEFAULT)
tresh.append(0.93)
mask.append(cv.imread("equip/amus/amu_5_32-mask.png", cv.IMREAD_COLOR))
needles.append(cv.imread("equip/amus/amu_6_32.jpg", cv.IMREAD_UNCHANGED))
hsv.append(HSV_DEFAULT)
tresh.append(0.93)
mask.append(cv.imread("equip/amus/amu_6_32-mask.png", cv.IMREAD_COLOR))
needles.append(cv.imread("equip/amus/amu_7_32.jpg", cv.IMREAD_UNCHANGED))
hsv.append(HSV_DEFAULT)
tresh.append(0.93)
mask.append(cv.imread("equip/amus/amu_7_32-mask.png", cv.IMREAD_COLOR))
def include_coins(needles, hsv):
needles.append(cv.imread("equip/coin_1_32.jpg", cv.IMREAD_UNCHANGED))
hsv.append(HsvFilter(0, 0, 0, 179, 255, 255, 0, 0, 0, 0))
needles.append(cv.imread("equip/coin_2_32.jpg", cv.IMREAD_UNCHANGED))
hsv.append(HsvFilter(0, 0, 0, 179, 255, 255, 0, 0, 0, 0))
needles.append(cv.imread("equip/coin_3_32.jpg", cv.IMREAD_UNCHANGED))
hsv.append(HsvFilter(0, 0, 0, 179, 255, 255, 0, 0, 0, 0))
needles.append(cv.imread("equip/coin_4_32.jpg", cv.IMREAD_UNCHANGED))
hsv.append(HsvFilter(0, 0, 0, 179, 255, 255, 0, 0, 0, 0))
needles.append(cv.imread("equip/coin_5_32.jpg", cv.IMREAD_UNCHANGED))
hsv.append(HsvFilter(0, 0, 0, 179, 255, 255, 0, 0, 0, 0))
def include_swords(needles, hsv, tresh, mask):
needles.append(cv.imread("equip/swords/sword_1_32.jpg", cv.IMREAD_UNCHANGED))
hsv.append(HSV_DEFAULT)
tresh.append(0.93)
mask.append(cv.imread("equip/swords/sword_1_32-mask.png", cv.IMREAD_COLOR))
def include_runes(needles, hsv):
needles.append(cv.imread("equip/rune_1_32.jpg", cv.IMREAD_UNCHANGED))
hsv.append(HsvFilter(0, 0, 0, 179, 255, 255, 0, 0, 0, 0))
needles.append(cv.imread("equip/rune_2_32.jpg", cv.IMREAD_UNCHANGED))
hsv.append(HsvFilter(0, 0, 0, 179, 255, 255, 0, 0, 0, 0))
needles.append(cv.imread("equip/swords/sword_2_32.jpg", cv.IMREAD_UNCHANGED))
hsv.append(HSV_DEFAULT)
tresh.append(0.93)
mask.append(cv.imread("equip/swords/sword_2_32-mask.png", cv.IMREAD_COLOR))
needles.append(cv.imread("equip/swords/sword_3_32.jpg", cv.IMREAD_UNCHANGED))
hsv.append(HSV_DEFAULT)
tresh.append(0.93)
mask.append(cv.imread("equip/swords/sword_3_32-mask.png", cv.IMREAD_COLOR))
def include_mushs(needles, hsv):
needles.append(cv.imread("equip/mush_1_32.jpg", cv.IMREAD_UNCHANGED))
hsv.append(HsvFilter(0, 0, 0, 179, 255, 255, 0, 0, 0, 0))
needles.append(cv.imread("equip/mush_2_32.jpg", cv.IMREAD_UNCHANGED))
hsv.append(HsvFilter(0, 0, 0, 179, 255, 255, 0, 0, 0, 0))
needles.append(cv.imread("equip/mush_3_32.jpg", cv.IMREAD_UNCHANGED))
hsv.append(HsvFilter(0, 0, 0, 179, 255, 255, 0, 0, 0, 0))
needles.append(cv.imread("equip/mush_4_32.jpg", cv.IMREAD_UNCHANGED))
hsv.append(HsvFilter(0, 0, 0, 179, 255, 255, 0, 0, 0, 0))
needles.append(cv.imread("equip/mush_5_32.jpg", cv.IMREAD_UNCHANGED))
hsv.append(HsvFilter(0, 0, 0, 179, 255, 255, 0, 0, 0, 0))
needles.append(cv.imread("equip/mush_6_32.jpg", cv.IMREAD_UNCHANGED))
hsv.append(HsvFilter(0, 0, 0, 179, 255, 255, 0, 0, 0, 0))
needles.append(cv.imread("equip/mush_7_32.jpg", cv.IMREAD_UNCHANGED))
hsv.append(HsvFilter(0, 0, 0, 179, 255, 255, 0, 0, 0, 0))
needles.append(cv.imread("equip/mush_8_32.jpg", cv.IMREAD_UNCHANGED))
hsv.append(HsvFilter(0, 0, 0, 179, 255, 255, 0, 0, 0, 0))
needles.append(cv.imread("equip/swords/sword_4_32.jpg", cv.IMREAD_UNCHANGED))
hsv.append(HSV_DEFAULT)
tresh.append(0.93)
mask.append(cv.imread("equip/swords/sword_4_32-mask.png", cv.IMREAD_COLOR))
needles.append(cv.imread("equip/swords/sword_5_32.jpg", cv.IMREAD_UNCHANGED))
hsv.append(HSV_DEFAULT)
tresh.append(0.93)
mask.append(cv.imread("equip/swords/sword_5_32-mask.png", cv.IMREAD_COLOR))
def include_amus(needles, hsv):
needles.append(cv.imread("equip/amu_1_32.jpg", cv.IMREAD_UNCHANGED))
hsv.append(HsvFilter(0, 0, 0, 179, 255, 255, 0, 0, 0, 0))
needles.append(cv.imread("equip/amu_2_32.jpg", cv.IMREAD_UNCHANGED))
hsv.append(HsvFilter(0, 0, 0, 179, 255, 255, 0, 0, 0, 0))
needles.append(cv.imread("equip/amu_3_32.jpg", cv.IMREAD_UNCHANGED))
hsv.append(HsvFilter(0, 0, 0, 179, 255, 255, 0, 0, 0, 0))
needles.append(cv.imread("equip/amu_4_32.jpg", cv.IMREAD_UNCHANGED))
hsv.append(HsvFilter(0, 0, 0, 179, 255, 255, 0, 0, 0, 0))
needles.append(cv.imread("equip/amu_5_32.jpg", cv.IMREAD_UNCHANGED))
hsv.append(HsvFilter(0, 0, 0, 179, 255, 255, 0, 0, 0, 0))
needles.append(cv.imread("equip/amu_6_32.jpg", cv.IMREAD_UNCHANGED))
hsv.append(HsvFilter(0, 0, 0, 179, 255, 255, 0, 0, 0, 0))
needles.append(cv.imread("equip/amu_7_32.jpg", cv.IMREAD_UNCHANGED))
hsv.append(HsvFilter(0, 0, 0, 179, 255, 255, 0, 0, 0, 0))
def include_staffs(needles, hsv, tresh, mask):
needles.append(cv.imread("equip/staffs/staff_1_32.jpg", cv.IMREAD_UNCHANGED))
hsv.append(HSV_DEFAULT)
tresh.append(0.93)
mask.append(cv.imread("equip/staffs/staff_1_32-mask.png", cv.IMREAD_COLOR))
needles.append(cv.imread("equip/staffs/staff_2_32.jpg", cv.IMREAD_UNCHANGED))
hsv.append(HSV_DEFAULT)
tresh.append(0.93)
mask.append(cv.imread("equip/staffs/staff_2_32-mask.png", cv.IMREAD_COLOR))
def include_swords(needles, hsv):
needles.append(cv.imread("equip/sword_1_32.jpg", cv.IMREAD_UNCHANGED))
hsv.append(HsvFilter(0, 0, 0, 179, 255, 255, 0, 0, 0, 0))
needles.append(cv.imread("equip/sword_2_32.jpg", cv.IMREAD_UNCHANGED))
hsv.append(HsvFilter(0, 0, 0, 179, 255, 255, 0, 0, 0, 0))
needles.append(cv.imread("equip/sword_3_32.jpg", cv.IMREAD_UNCHANGED))
hsv.append(HsvFilter(0, 0, 0, 179, 255, 255, 0, 0, 0, 0))
needles.append(cv.imread("equip/sword_4_32.jpg", cv.IMREAD_UNCHANGED))
hsv.append(HsvFilter(0, 0, 0, 179, 255, 255, 0, 0, 0, 0))
needles.append(cv.imread("equip/sword_5_32.jpg", cv.IMREAD_UNCHANGED))
hsv.append(HsvFilter(0, 0, 0, 179, 255, 255, 0, 0, 0, 0))
needles.append(cv.imread("equip/staffs/staff_3_32.jpg", cv.IMREAD_UNCHANGED))
hsv.append(HSV_DEFAULT)
tresh.append(0.93)
mask.append(cv.imread("equip/staffs/staff_3_32-mask.png", cv.IMREAD_COLOR))
def include_staffs(needles, hsv):
needles.append(cv.imread("equip/staff_1_32.jpg", cv.IMREAD_UNCHANGED))
hsv.append(HsvFilter(0, 0, 0, 179, 255, 255, 0, 0, 0, 0))
needles.append(cv.imread("equip/staff_2_32.jpg", cv.IMREAD_UNCHANGED))
hsv.append(HsvFilter(0, 0, 0, 179, 255, 255, 0, 0, 0, 0))
needles.append(cv.imread("equip/staff_3_32.jpg", cv.IMREAD_UNCHANGED))
hsv.append(HsvFilter(0, 0, 0, 179, 255, 255, 0, 0, 0, 0))
needles.append(cv.imread("equip/staff_4_32.jpg", cv.IMREAD_UNCHANGED))
hsv.append(HsvFilter(0, 0, 0, 179, 255, 255, 0, 0, 0, 0))
#needles.append(cv.imread("equip/sword_5_32.jpg", cv.IMREAD_UNCHANGED))
#hsv.append(HsvFilter(0, 0, 0, 179, 255, 255, 0, 0, 0, 0))
needles.append(cv.imread("equip/staffs/staff_4_32.jpg", cv.IMREAD_UNCHANGED))
hsv.append(HSV_DEFAULT)
tresh.append(0.93)
mask.append(cv.imread("equip/staffs/staff_4_32-mask.png", cv.IMREAD_COLOR))
def include_rings(needles, hsv, tresh, mask):
needles.append(cv.imread("equip/rings/ring_1_32.jpg", cv.IMREAD_UNCHANGED))
hsv.append(HSV_DEFAULT)
tresh.append(0.92)
mask.append(cv.imread("equip/rings/ring_1_32-mask.png", cv.IMREAD_COLOR))
needles.append(cv.imread("equip/rings/ring_2_32.jpg", cv.IMREAD_UNCHANGED))
hsv.append(HSV_DEFAULT)
tresh.append(0.94)
mask.append(cv.imread("equip/rings/ring_2_32-mask.png", cv.IMREAD_COLOR))
needles.append(cv.imread("equip/rings/ring_3_32.jpg", cv.IMREAD_UNCHANGED))
hsv.append(HSV_DEFAULT)
tresh.append(0.93)
mask.append(cv.imread("equip/rings/ring_3_32-mask.png", cv.IMREAD_COLOR))
needles.append(cv.imread("equip/rings/ring_4_32.jpg", cv.IMREAD_UNCHANGED))
hsv.append(HSV_DEFAULT)
tresh.append(0.91)
mask.append(cv.imread("equip/rings/ring_4_32-mask.png", cv.IMREAD_COLOR))
needles.append(cv.imread("equip/rings/ring_5_32.jpg", cv.IMREAD_UNCHANGED))
hsv.append(HSV_DEFAULT)
tresh.append(0.93)
mask.append(cv.imread("equip/rings/ring_5_32-mask.png", cv.IMREAD_COLOR))
def include_wands(needles, hsv, tresh, mask):
needles.append(cv.imread("equip/wands/wand_1_32.jpg", cv.IMREAD_UNCHANGED))
hsv.append(HSV_DEFAULT)
tresh.append(0.93)
mask.append(cv.imread("equip/wands/wand_1_32-mask.png", cv.IMREAD_COLOR))
needles.append(cv.imread("equip/wands/wand_2_32.jpg", cv.IMREAD_UNCHANGED))
hsv.append(HSV_DEFAULT)
tresh.append(0.93)
mask.append(cv.imread("equip/wands/wand_2_32-mask.png", cv.IMREAD_COLOR))
needles.append(cv.imread("equip/wands/wand_3_32.jpg", cv.IMREAD_UNCHANGED))
hsv.append(HSV_DEFAULT)
tresh.append(0.93)
mask.append(cv.imread("equip/wands/wand_3_32-mask.png", cv.IMREAD_COLOR))
needles.append(cv.imread("equip/wands/wand_4_32.jpg", cv.IMREAD_UNCHANGED))
hsv.append(HSV_DEFAULT)
tresh.append(0.93)
mask.append(cv.imread("equip/wands/wand_4_32-mask.png", cv.IMREAD_COLOR))
def move_tile(conf, point_source, point_dest):
offset_left = conf.returnEquipmentWindowPos()[2]
offset_down = conf.returnEquipmentWindowPos()[3]
pydirectinput.moveTo(point_source[0] + offset_left, point_source[1] + offset_down)
pydirectinput.mouseDown()
w = random.randint(1, 100)
cv.waitKey(250 + w)
w = random.randint(1, 50)
cv.waitKey(100 + w)
pydirectinput.moveTo(point_dest[0] + offset_left, point_dest[1] + offset_down)
pydirectinput.mouseUp()
cv.waitKey(250 + w)
cv.waitKey(500)
def find_emitter(emitter_to_use, vis, screen, layer):
if emitter_to_use == EMITTER_MUSH:
if layer == 0:
return vis.find(screen, cv.imread("equip/mush_e_32.jpg", cv.IMREAD_UNCHANGED), 0.7, 1)
needle = cv.imread("equip/emitters/mush_e_32.jpg", cv.IMREAD_UNCHANGED)
mask = cv.imread("equip/emitters/mush_e_32-mask.png", cv.IMREAD_COLOR)
return vis.find(screen, needle, 0.95, 1, True, mask)
else:
return vis.find(screen, cv.imread("equip/mush_e2_32.jpg", cv.IMREAD_UNCHANGED), 0.4, 1)
needle = cv.imread("equip/emitters/mush_e2_32.jpg", cv.IMREAD_UNCHANGED)
mask = cv.imread("equip/emitters/mush_e2_32-mask.png", cv.IMREAD_COLOR)
return vis.find(screen, needle, 0.96, 1, True, mask)
elif emitter_to_use == EMITTER_AMU:
if layer == 0:
return vis.find(screen, cv.imread("equip/amu_e_32.jpg", cv.IMREAD_UNCHANGED), 0.7, 1)
needle = cv.imread("equip/emitters/amu_e_32.jpg", cv.IMREAD_UNCHANGED)
mask = cv.imread("equip/emitters/amu_e_32-mask.png", cv.IMREAD_COLOR)
return vis.find(screen, needle, 0.95, 1, True, mask)
else:
return vis.find(screen, cv.imread("equip/amu_e2_32.jpg", cv.IMREAD_UNCHANGED), 0.4, 1)
needle = cv.imread("equip/emitters/amu_e2_32.jpg", cv.IMREAD_UNCHANGED)
mask = cv.imread("equip/emitters/amu_e2_32-mask.png", cv.IMREAD_COLOR)
return vis.find(screen, needle, 0.95, 1, True, mask)
elif emitter_to_use == EMITTER_MAIN:
if layer == 0:
return vis.find(screen, cv.imread("equip/main_e1_32.jpg", cv.IMREAD_UNCHANGED), 0.8, 1)
needle = cv.imread("equip/emitters/main_e1_32.jpg", cv.IMREAD_UNCHANGED)
mask = cv.imread("equip/emitters/main_e1_32-mask.png", cv.IMREAD_COLOR)
return vis.find(screen, needle, 0.95, 1, True, mask)
else:
return vis.find(screen, cv.imread("equip/main_e2_32.jpg", cv.IMREAD_UNCHANGED), 0.5, 1)
needle = cv.imread("equip/emitters/main_e2_32.jpg", cv.IMREAD_UNCHANGED)
mask = cv.imread("equip/emitters/main_e2_32-mask.png", cv.IMREAD_COLOR)
return vis.find(screen, needle, 0.95, 1, True, mask)
elif emitter_to_use == EMITTER_SWORD:
if layer == 0:
return vis.find(screen, cv.imread("equip/sword_e1_32.jpg", cv.IMREAD_UNCHANGED), 0.8, 1)
needle = cv.imread("equip/emitters/sword_e1_32.jpg", cv.IMREAD_UNCHANGED)
mask = cv.imread("equip/emitters/sword_e1_32-mask.png", cv.IMREAD_COLOR)
return vis.find(screen, needle, 0.95, 1, True, mask)
else:
return vis.find(screen, cv.imread("equip/sword_e2_32.jpg", cv.IMREAD_UNCHANGED), 0.7, 1)
needle = cv.imread("equip/emitters/sword_e2_32.jpg", cv.IMREAD_UNCHANGED)
mask = cv.imread("equip/emitters/sword_e2_32-mask.png", cv.IMREAD_COLOR)
return vis.find(screen, needle, 0.98, 1, True, mask)
elif emitter_to_use == EMITTER_STAFF:
if layer == 0:
return vis.find(screen, cv.imread("equip/staff_e1_32.jpg", cv.IMREAD_UNCHANGED), 0.8, 1)
needle = cv.imread("equip/emitters/staff_e1_32.jpg", cv.IMREAD_UNCHANGED)
mask = cv.imread("equip/emitters/staff_e1_32-mask.png", cv.IMREAD_COLOR)
return vis.find(screen, needle, 0.95, 1, True, mask)
else:
return vis.find(screen, cv.imread("equip/staff_e2_32.jpg", cv.IMREAD_UNCHANGED), 0.7, 1)
needle = cv.imread("equip/emitters/staff_e2_32.jpg", cv.IMREAD_UNCHANGED)
mask = cv.imread("equip/emitters/staff_e2_32-mask.png", cv.IMREAD_COLOR)
return vis.find(screen, needle, 0.99, 1, True, mask)
elif emitter_to_use == EMITTER_RING:
if layer == 0:
needle = cv.imread("equip/emitters/ring_e1_32.jpg", cv.IMREAD_UNCHANGED)
mask = cv.imread("equip/emitters/ring_e1_32-mask.png", cv.IMREAD_COLOR)
return vis.find(screen, needle, 0.95, 1, True, mask)
else:
needle = cv.imread("equip/emitters/ring_e2_32.jpg", cv.IMREAD_UNCHANGED)
mask = cv.imread("equip/emitters/ring_e2_32-mask.png", cv.IMREAD_COLOR)
return vis.find(screen, needle, 0.95, 1, True, mask)
elif emitter_to_use == EMITTER_WAND:
if layer == 0:
needle = cv.imread("equip/emitters/wand_e1_32.jpg", cv.IMREAD_UNCHANGED)
mask = cv.imread("equip/emitters/wand_e1_32-mask.png", cv.IMREAD_COLOR)
return vis.find(screen, needle, 0.95, 1, True, mask)
else:
needle = cv.imread("equip/emitters/wand_e2_32.jpg", cv.IMREAD_UNCHANGED)
mask = cv.imread("equip/emitters/wand_e2_32-mask.png", cv.IMREAD_COLOR)
return vis.find(screen, needle, 0.96, 1, True, mask)
def check_and_move_tile(capture_win, visio, conf, rect, needl, hsv, point_source, point_dest):
screenshot_pos = capture_win.get_screenshot_by_area(rect)