mirror of
https://github.com/fishyboteso/fishyboteso.git
synced 2024-08-30 18:32:13 +00:00
33 lines
835 B
Python
33 lines
835 B
Python
|
from init import *
|
||
|
|
||
|
|
||
|
class Window:
|
||
|
Screen = None
|
||
|
|
||
|
def __init__(self, name, crop, scale, color):
|
||
|
self.color = color
|
||
|
self.crop = crop
|
||
|
self.scale = scale
|
||
|
self.name = name
|
||
|
|
||
|
@staticmethod
|
||
|
def Loop():
|
||
|
bbox = (0, 0, GetSystemMetrics(0), GetSystemMetrics(1))
|
||
|
Window.Screen = np.array(ImageGrab.grab(bbox=bbox))
|
||
|
|
||
|
@staticmethod
|
||
|
def LoopEnd():
|
||
|
cv2.waitKey(25)
|
||
|
|
||
|
def show(self, show):
|
||
|
if show:
|
||
|
cv2.imshow(self.name, self.getCapture())
|
||
|
else:
|
||
|
cv2.destroyWindow(self.name)
|
||
|
|
||
|
def getCapture(self):
|
||
|
temp_img = cv2.cvtColor(Window.Screen, self.color)
|
||
|
temp_img = temp_img[self.crop[1]:self.crop[3], self.crop[0]:self.crop[2]]
|
||
|
temp_img = imutils.resize(temp_img, width=self.scale)
|
||
|
return temp_img
|