import numpy as np
from backend.data_types import DataField


def test_flip_field():
    from backend.nodes.flip import FlipField
    from backend.node_registry import get_node_info

    node = FlipField()
    data = np.arange(1, 10, dtype=np.float64).reshape(3, 3)
    markup_overlay = {
        "kind": "markup",
        "shapes": [
            {"kind": "line", "x1": 0.1, "y1": 0.2, "x2": 0.9, "y2": 0.8, "width": 2, "color": "#ffffff"},
            {"kind": "rectangle", "x1": 0.15, "y1": 0.1, "x2": 0.45, "y2": 0.6, "width": 3, "color": "#ff0000"},
        ],
    }
    annotation_overlay = {"kind": "annotation", "show_scale_bar": True, "show_color_map": False, "text_size": 14.0}
    field = DataField(data=data, xreal=3.0, yreal=4.0, xoff=10.0, yoff=20.0, si_unit_xy="nm", si_unit_z="nm", overlays=[markup_overlay, annotation_overlay])

    assert get_node_info("FlipField")["category"] == "Geometry"

    flipped_x, = node.process(field, axis="x")
    assert np.array_equal(flipped_x.data, np.flipud(data))
    assert flipped_x.xreal == field.xreal
    assert flipped_x.yreal == field.yreal
    assert flipped_x.xoff == field.xoff
    assert flipped_x.yoff == field.yoff
    assert np.isclose(flipped_x.overlays[0]["shapes"][0]["x1"], 0.1)
    assert np.isclose(flipped_x.overlays[0]["shapes"][0]["y1"], 0.8)
    assert np.isclose(flipped_x.overlays[0]["shapes"][0]["x2"], 0.9)
    assert np.isclose(flipped_x.overlays[0]["shapes"][0]["y2"], 0.2)
    assert np.isclose(flipped_x.overlays[0]["shapes"][1]["x1"], 0.15)
    assert np.isclose(flipped_x.overlays[0]["shapes"][1]["y1"], 0.4)
    assert np.isclose(flipped_x.overlays[0]["shapes"][1]["x2"], 0.45)
    assert np.isclose(flipped_x.overlays[0]["shapes"][1]["y2"], 0.9)
    assert flipped_x.overlays[1] == annotation_overlay

    flipped_y, = node.process(field, axis="y")
    assert np.array_equal(flipped_y.data, np.fliplr(data))
    assert np.isclose(flipped_y.overlays[0]["shapes"][0]["x1"], 0.9)
    assert np.isclose(flipped_y.overlays[0]["shapes"][0]["y1"], 0.2)
    assert np.isclose(flipped_y.overlays[0]["shapes"][0]["x2"], 0.1)
    assert np.isclose(flipped_y.overlays[0]["shapes"][0]["y2"], 0.8)

    assert field.overlays[0]["shapes"][0]["x1"] == markup_overlay["shapes"][0]["x1"]

    try:
        node.process(field, axis="diagonal")
        raise AssertionError("Expected invalid flip axis to raise ValueError")
    except ValueError:
        pass