mrabbitt / bgee_sikuli_auto_roller Goto Github PK

# Find total roll region
def get_roll_value():
    wait(total_roll_pattern)
    total_roll = find(total_roll_pattern)

    tens_digit_region = Region(
        int(total_roll.getTopRight().getX() - expand_pixels),
        int(total_roll.getTopRight().getY() - expand_pixels) - 1,
        digit_width + (2 * expand_pixels), 
        digit_height+ (2 * expand_pixels)
    )
    ones_digit_region = Region(
        int(total_roll.getTopRight().getX() - expand_pixels) + digit_offset_width,
        int(total_roll.getTopRight().getY() - expand_pixels) - 1,
        digit_width + (2 * expand_pixels), 
        digit_height + (2 * expand_pixels)
    )

    tens_digits = get_digits_in_region(tens_digit_region)
    ones_digits = get_digits_in_region(ones_digit_region)
    if len(tens_digits) == len(ones_digits) == 1:
        # check for triple digit total
        if tens_digits[0] == 0 or tens_digits[0] == 1:
            tens_digits[0] = 10
        roll_value = tens_digits[0] * 10 + ones_digits[0]
        return roll_value
    else:
        logging.error('Mutiple matches. tens: %s, ones: %s', tens_digits, ones_digits)
        if len(tens_digits) != 1:
            tens_digit_region.highlight(3)
        if len(ones_digits) != 1:
            ones_digit_region.highlight(3)
        return 0