def test_crop_func():
entire_screen_pil = Image.open(
os.path.join(get_dir('tests', 'screenshots'),
'53269218_PreFlop_0.png'))
top_left_corner = get_table_template_image("PartyPoker Old",
'topleft_corner')
cropped = crop_screenshot_with_topleft_corner(entire_screen_pil,
top_left_corner)
assert cropped
def test_crop_func():
entire_screen_pil = Image.open(
os.path.join(get_dir('tests', 'screenshots'),
'173280759_PreFlop_0.png'))
top_left_corner = get_table_template_image(
"PartyPoker 6 Players Fast Forward $1-$2 NL Hold'em", 'topleft_corner')
cropped = crop_screenshot_with_topleft_corner(entire_screen_pil,
top_left_corner)
assert cropped
def crop(self):
if not self.original_screenshot:
pop_up("No screenshot taken yet",
"Please take a screenshot first by pressing on the take screenshot button. Then mark a new top "
"left corner or load a previously saved one. After that you can crop the image.")
return
self.load_topleft_corner()
log.debug("Cropping top left corner")
self.original_screenshot, self.tlc = crop_screenshot_with_topleft_corner(self.original_screenshot,
self.top_left_corner_img)
if self.original_screenshot is None:
log.warning("No (or multiple) top left corner found")
pop_up("Top left corner problem: ",
"No or multiple top left corners visible. Please ensure only a single top left corner is visible.")
return
else:
self.signal_update_screenshot_pic.emit(self.original_screenshot)
self.cropped = True
def crop(self):
if not self.original_screenshot:
pop_up(
"No screenshot taken yet",
"Please take a screenshot first by pressing on the take screenshot button. Then mark a new top "
"left corner or load a previously saved one. After that you can crop the image."
)
return
self.load_topleft_corner()
log.debug("Cropping top left corner")
self.original_screenshot, self.tlc = crop_screenshot_with_topleft_corner(
self.original_screenshot, self.top_left_corner_img)
if self.original_screenshot is None:
log.warning("No (or multiple) top left corner found")
pop_up(
"Top left corner not found",
"It was not possible to find the top left corner of the poker window. Please first mark a new "
"topleft corner or load a previously saved one by clicking on the corresponding buttons. This error"
"can also occur if in the table editor a top left corner is visible as well. Taking 2 or 3 consecutive"
"screenshots usually solves the problem.")
return
else:
self.signal_update_screenshot_pic.emit(self.original_screenshot)
self.cropped = True
def crop_from_top_left_corner(self):
"""Crop top left corner based on the current selected table dict and replace self.screnshot with it"""
self.screenshot, self.tlc = crop_screenshot_with_topleft_corner(self.screenshot,
binary_pil_to_cv2(
self.table_dict['topleft_corner']))
return self.screenshot
