def test_in_game_caesar_portrait_standard_template_matching_cross_corr(
self):
threshold = unit_constants.ui_standard_template_matching_threshold
wargroove_ctrl.vision.load_frame(
self.__get_eval_folder() +
'in-game/hud-mirror/portrait-caesar-mirrored.png')
match = wargroove_ctrl.find_ui_item_with_template_matching(
'gameplay/wargroove_caesar_portrait', threshold=threshold)
assert_that(match).is_not_none()
assert_that(match.top_left[0]).is_equal_to(153)
assert_that(match.top_left[1]).is_equal_to(13)
assert_that(match.bottom_right[0]).is_equal_to(224)
assert_that(match.bottom_right[1]).is_equal_to(86)
def test_in_game_mercia_portrait_standard_template_matching_cross_corr_img4(
self):
threshold = unit_constants.ui_standard_template_matching_threshold
wargroove_ctrl.vision.load_frame(
self.__get_eval_folder() +
'in-game/hud-mirror/portrait-mercia-low-res.png')
match = wargroove_ctrl.find_ui_item_with_template_matching(
'gameplay/wargroove_mercia_portrait', threshold=threshold)
assert_that(match).is_not_none()
assert_that(match.top_left[0]).is_close_to(6, 1)
assert_that(match.top_left[1]).is_close_to(7, 1)
assert_that(match.bottom_right[0]).is_close_to(42, 1)
assert_that(match.bottom_right[1]).is_close_to(43, 1)
def test_in_game_mercia_portrait_standard_template_matching_square_diff_img4(
self):
threshold = unit_constants.ui_standard_template_matching_square_diff_threshold
wargroove_ctrl.vision.load_frame(
self.__get_eval_folder() +
'in-game/hud-mirror/portrait-mercia-mirrored.png')
match = wargroove_ctrl.find_ui_item_with_template_matching(
'gameplay/wargroove_mercia_portrait',
threshold=threshold,
method=cmp504.computer_vision.TemplateMatchingMethod.
SQUARE_DIFFERENCE_NORMALIZED)
assert_that(match).is_not_none()
assert_that(match.top_left[0]).is_equal_to(153)
assert_that(match.top_left[1]).is_equal_to(13)
assert_that(match.bottom_right[0]).is_equal_to(224)
assert_that(match.bottom_right[1]).is_equal_to(86)
def test_in_game_caesar_portrait_standard_template_matching_corr_coeff(
self):
threshold = unit_constants.ui_standard_template_matching_corr_coeff_threshold
wargroove_ctrl.vision.load_frame(
self.__get_eval_folder() +
'in-game/hud-mirror/portrait-caesar-mirrored.png')
match = wargroove_ctrl.find_ui_item_with_template_matching(
'gameplay/wargroove_caesar_portrait',
threshold=threshold,
method=cmp504.computer_vision.TemplateMatchingMethod.
CORRELATION_COEFFICIENT_NORMALIZED)
assert_that(match).is_not_none()
assert_that(match.top_left[0]).is_equal_to(153)
assert_that(match.top_left[1]).is_equal_to(13)
assert_that(match.bottom_right[0]).is_equal_to(224)
assert_that(match.bottom_right[1]).is_equal_to(86)
def test_in_game_emeric_portrait_standard_template_matching_corr_coeff_img4(
self):
threshold = unit_constants.ui_standard_template_matching_corr_coeff_threshold
wargroove_ctrl.vision.load_frame(
self.__get_eval_folder() +
'in-game/hud-mirror/portrait-emeric-low-res.png')
match = wargroove_ctrl.find_ui_item_with_template_matching(
'gameplay/wargroove_emeric_portrait',
threshold=threshold,
method=cmp504.computer_vision.TemplateMatchingMethod.
CORRELATION_COEFFICIENT_NORMALIZED)
assert_that(match).is_not_none()
assert_that(match.top_left[0]).is_close_to(6, 1)
assert_that(match.top_left[1]).is_close_to(7, 1)
assert_that(match.bottom_right[0]).is_close_to(42, 1)
assert_that(match.bottom_right[1]).is_close_to(43, 1)
def test_event_editor_button_standard_template_matching_cross_corr(self):
pre_process = cmp504.image_processing.ImageProcessingStepChain()
pre_process.append(cmp504.image_processing.BGR2Grayscale())
pre_process.append(cmp504.image_processing.Threshold(200))
threshold = unit_constants.ui_standard_template_matching_threshold
wargroove_ctrl.vision.load_frame(self.__get_eval_folder() +
'change-color.png')
match = wargroove_ctrl.find_ui_item_with_template_matching(
'event_editor_button',
threshold=threshold,
template_pre_processing_chain=pre_process,
frame_pre_processing_chain=pre_process,
method=cmp504.computer_vision.TemplateMatchingMethod.
CROSS_CORRELATION_NORMALIZED)
assert_that(match).is_not_none()
assert_that(match.top_left[0]).is_equal_to(588)
assert_that(match.top_left[1]).is_equal_to(366)
assert_that(match.bottom_right[0]).is_equal_to(690)
assert_that(match.bottom_right[1]).is_equal_to(400)
def test_save_and_exit_button_standard_template_matching_square_diff(self):
pre_process = cmp504.image_processing.ImageProcessingStepChain()
pre_process.append(cmp504.image_processing.BGR2Grayscale())
pre_process.append(cmp504.image_processing.Threshold(200))
threshold = unit_constants.ui_standard_template_matching_threshold
wargroove_ctrl.vision.load_frame(self.__get_eval_folder() +
'change-color.png')
match = wargroove_ctrl.find_ui_item_with_template_matching(
'save_and_exit_button',
threshold=threshold,
template_pre_processing_chain=pre_process,
frame_pre_processing_chain=pre_process,
method=cmp504.computer_vision.TemplateMatchingMethod.
SQUARE_DIFFERENCE_NORMALIZED)
assert_that(match).is_not_none()
assert_that(match.top_left[0]).is_equal_to(590)
assert_that(match.top_left[1]).is_equal_to(450)
assert_that(match.bottom_right[0]).is_equal_to(690)
assert_that(match.bottom_right[1]).is_equal_to(484)
