boxes_common.py

import json
import math


PARSED_JSON_TYPE = "JRW_PARSED_JSON"


def default_outputs():
    return (0, 0, 0, 0, 0, 0)


def invalid_value(error_on_invalid, message, default_factory):
    if error_on_invalid:
        raise ValueError(message)
    return default_factory()


def parse_json_string(json_string):
    try:
        parsed_json = json.loads(json_string) if json_string else None
    except json.JSONDecodeError as exc:
        return None, f"json_string is not valid JSON: {exc.msg}"

    return parsed_json, None


def integer_box_values(x1, y1, x2, y2):
    left = math.floor(min(x1, x2))
    right = math.ceil(max(x1, x2))
    top = math.floor(min(y1, y2))
    bottom = math.ceil(max(y1, y2))

    return (int(left), int(right), int(top), int(bottom), int(right - left), int(bottom - top))


def extract_box_values(parsed_json, index):
    if not isinstance(parsed_json, (list, tuple)):
        return None, "parsed_json must be a list of boxes."

    if index < 0 or index >= len(parsed_json):
        return None, "index must point to an existing box in parsed_json."

    box = parsed_json[index]
    if not isinstance(box, (list, tuple)) or len(box) != 4:
        return None, "Selected box must contain exactly 4 values: [x1, y1, x2, y2]."

    try:
        x1, y1, x2, y2 = (float(value) for value in box)
    except (TypeError, ValueError):
        return None, "Selected box contains a value that cannot be converted to float."

    return integer_box_values(x1, y1, x2, y2), None