def crop_from_top_left_corner2(self, select_table=0):
"""Crop top left corner based on the current selected table dict and replace self.screnshot with it"""
self.screenshots1, self.tlcs1, self.count1 = crop_screenshot_with_topleft_corner(
self.screenshot,
binary_pil_to_cv2(self.table_dict['topleft_corner']),
self.scale_defaults)
self.screenshots2, self.tlcs2, self.count2 = crop_screenshot_with_topleft_corner(
self.screenshot,
binary_pil_to_cv2(self.table_dict['topleft_corner2']),
self.scale_defaults)
if self.screenshots1 is None:
if self.screenshots2 is not None:
self.screenshots = self.screenshots2
self.tlcs = self.tlcs2
else:
return
else:
if self.screenshots2 is None:
self.screenshots = self.screenshots1
self.tlcs = self.tlcs1
else:
self.screenshots = self.screenshots1 + self.screenshots2
self.tlcs = self.tlcs1 + self.tlcs2
self.count = self.count1 + self.count2
if self.count >= 1:
self.screenshot = self.screenshots[select_table]
self.tlc = self.tlcs[self.select_table]
# log.info(f"tlc:{self.tlc}")
return self.screenshot
else:
return
def test_crop_func():
entire_screen_pil = Image.open(os.path.join(get_dir('tests', 'screenshots'), 'test5.png'))
top_left_corner = get_table_template_image('GG_TEST1', 'topleft_corner')
# cv2.imshow('image', top_left_corner)
# cv2.waitKey(0)
cropped = crop_screenshot_with_topleft_corner(entire_screen_pil, top_left_corner, scale_m=1.0)
assert cropped
def test_crop_func():
entire_screen_pil = Image.open(
os.path.join(get_dir('tests', 'screenshots'), 'screenshot1.png'))
top_left_corner = get_table_template_image('default', 'topleft_corner')
cropped = crop_screenshot_with_topleft_corner(entire_screen_pil,
top_left_corner)
assert cropped
def crop2(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_corner2()
log.debug("Cropping 2nd top left corner")
self.original_screenshots, self.tlcs, self.count = crop_screenshot_with_topleft_corner(
self.original_screenshot, self.top_left_corner_img2, scale_m=1.0)
if self.original_screenshot is None:
log.warning("No 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"
)
return
else:
# resp = pop_up1("Select one of tables",
# "Please select one of inactive tables", ok_cancel=False, radio_button=True)
if self.count >= 1:
self.signal_update_screenshot_pic.emit(
self.original_screenshots[self.select_table])
self.original_screenshot = self.original_screenshots[
self.select_table]
else:
self.signal_update_screenshot_pic.emit(
self.original_screenshots)
self.original_screenshot = self.original_screenshots
self.cropped = True
def get_gg_poker_rel(self):
_, gg_poker, count = crop_screenshot_with_topleft_corner(
self.screenshot, binary_pil_to_cv2(self.table_dict['gg_poker']),
self.scale_defaults)
if count >= 1:
self.gg_poker_rel = gg_poker[0]
return True
else:
return False
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,
scale_m=1.0)
assert cropped
def crop_from_top_left_corner(self, select_table=0):
"""Crop top left corner based on the current selected table dict and replace self.screnshot with it"""
self.screenshots, self.tlcs, self.count = crop_screenshot_with_topleft_corner(
self.screenshot,
binary_pil_to_cv2(self.table_dict['topleft_corner']),
self.scale_defaults)
if self.count >= 1:
self.screenshot = self.screenshots[select_table]
self.tlc = self.tlcs[select_table]
return self.screenshot
else:
return
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 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"
)
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
