panels.py

"""panels.py — layer panel, properties panel, underlay panel, selection, and themes."""
import dearpygui.dearpygui as dpg
import state
import geometry
import draw
import history
import i18n
import colorwidget
import underlay

# ── Button themes ─────────────────────────────────────────────────────────────
_sel_btn_theme = None


def _ensure_themes() -> None:
    global _sel_btn_theme
    if _sel_btn_theme is not None:
        return
    with dpg.theme() as t:
        with dpg.theme_component(dpg.mvButton):
            dpg.add_theme_color(dpg.mvThemeCol_Button,        [160, 120, 10, 200])
            dpg.add_theme_color(dpg.mvThemeCol_ButtonHovered, [200, 160, 20, 230])
            dpg.add_theme_color(dpg.mvThemeCol_ButtonActive,  [230, 190, 40, 255])
    _sel_btn_theme = t


# ── Layer panel ───────────────────────────────────────────────────────────────
def refresh_layers() -> None:
    if not dpg.does_item_exist("layer_list"):
        return
    dpg.delete_item("layer_list", children_only=True)
    n = len(state.st["elements"])
    for i, el in enumerate(state.st["elements"]):
        label = el.get("text") or el.get("desc") or i18n.t("layer_empty_label")
        typ   = "T" if el["type"] == "text" else "O"
        col_t = (80, 160, 255, 255) if typ == "T" else (60, 210, 110, 255)
        sel   = (i == state.st["selected"])

        def make_sel(idx):
            def cb(s, u): select(idx)
            return cb

        def make_up(idx):
            def cb(s, u):
                if idx <= 0: return
                history.push_history()
                state.st["elements"][idx], state.st["elements"][idx - 1] = (
                    state.st["elements"][idx - 1], state.st["elements"][idx]
                )
                state.st["selected"] = idx - 1
                refresh_all()
            return cb

        def make_down(idx):
            def cb(s, u):
                if idx >= len(state.st["elements"]) - 1: return
                history.push_history()
                state.st["elements"][idx], state.st["elements"][idx + 1] = (
                    state.st["elements"][idx + 1], state.st["elements"][idx]
                )
                state.st["selected"] = idx + 1
                refresh_all()
            return cb

        def make_del(idx):
            def cb(s, u):
                history.push_history()
                state.st["elements"].pop(idx)
                remaining = len(state.st["elements"])
                state.st["selected"] = min(idx, remaining - 1) if remaining > 0 else -1
                refresh_all()
            return cb

        with dpg.group(horizontal=True, parent="layer_list"):
            dpg.add_text(f"[{typ}]", color=col_t)
            dpg.add_button(label="↑", width=20, enabled=(i > 0),     callback=make_up(i))
            dpg.add_button(label="↓", width=20, enabled=(i < n - 1), callback=make_down(i))
            dpg.add_button(label="✕", width=20,                       callback=make_del(i))
            btn = dpg.add_button(
                label=f"{label[:18]}",
                tag=f"layer_btn_{i}",
                width=-1,
                callback=make_sel(i),
            )
            if sel:
                _ensure_themes()
                dpg.bind_item_theme(btn, _sel_btn_theme)


# ── Properties panel ──────────────────────────────────────────────────────────
def refresh_props() -> None:
    if not dpg.does_item_exist("props_group"):
        return
    dpg.delete_item("props_group", children_only=True)
    i = state.st["selected"]
    if i < 0 or i >= len(state.st["elements"]):
        dpg.add_text(i18n.t("props_no_selection"), parent="props_group", color=(160, 160, 160, 255))
        return

    el = state.st["elements"][i]
    dpg.add_text(i18n.t("props_layer_title", index=i), parent="props_group", color=(220, 220, 100, 255))
    dpg.add_separator(parent="props_group")

    def set_type(_, val):
        history.push_history()
        el["type"] = val
        if el["type"] == "text" and "text" not in el:
            el["text"] = ""
        refresh_props()
        refresh_layers()
        draw.redraw()

    dpg.add_text(i18n.t("props_type_label"), parent="props_group")
    dpg.add_radio_button(
        ["text", "obj"],
        default_value=el["type"],
        horizontal=True,
        callback=set_type,
        parent="props_group",
    )
    dpg.add_separator(parent="props_group")

    if el["type"] == "text":
        dpg.add_text(i18n.t("props_text_label"), parent="props_group")

        def cb_text(s, v):
            el["text"] = v
            btn_tag = f"layer_btn_{state.st['selected']}"
            if dpg.does_item_exist(btn_tag):
                dpg.set_item_label(btn_tag, (v or i18n.t("layer_empty_label"))[:18])
            draw.redraw()

        dpg.add_input_text(
            default_value=el.get("text", ""),
            width=-1,
            callback=cb_text,
            tag="prop_text",
            parent="props_group",
        )

    dpg.add_text(i18n.t("props_desc_label"), parent="props_group")

    def cb_desc(s, v):
        el["desc"] = v
        if el.get("type") != "text":
            btn_tag = f"layer_btn_{state.st['selected']}"
            if dpg.does_item_exist(btn_tag):
                dpg.set_item_label(btn_tag, (v or i18n.t("layer_empty_label"))[:18])
        draw.redraw()

    dpg.add_input_text(
        default_value=el.get("desc", ""),
        width=-1,
        multiline=True,
        height=100,
        callback=cb_desc,
        tag="prop_desc",
        parent="props_group",
    )

    dpg.add_separator(parent="props_group")
    dpg.add_text(i18n.t("props_bbox_label"), parent="props_group")
    bbox = el["bbox"]

    def make_bbox_cb(idx):
        def cb(s, v):
            el["bbox"][idx] = int(v)
            draw.redraw()
        return cb

    _bbox_labels = ["ymin", "xmin", "ymax", "xmax"]
    for row in range(2):
        with dpg.group(horizontal=True, parent="props_group"):
            for col in range(2):
                idx = row * 2 + col
                with dpg.group():
                    dpg.add_text(_bbox_labels[idx])
                    dpg.add_input_int(
                        default_value=bbox[idx],
                        width=180,
                        min_value=0,
                        max_value=1000,
                        callback=make_bbox_cb(idx),
                        tag=f"prop_bbox_{idx}",
                    )

    dpg.add_separator(parent="props_group")
    dpg.add_text(i18n.t("props_palette_label"), parent="props_group")
    pal_str = ", ".join(el.get("color_palette", []))

    def cb_pal(s, v):
        el["color_palette"] = [c.strip() for c in v.split(",") if c.strip()]

    with dpg.group(horizontal=True, parent="props_group") as _pal_group:
        pal_tag = f"prop_palette_{i}"
        dpg.add_input_text(
            tag=pal_tag,
            default_value=pal_str,
            width=-30,
            callback=cb_pal,
        )
        def _make_pal_change(elem):
            def _cb(new_val):
                elem["color_palette"] = [c.strip() for c in new_val.split(",") if c.strip()]
            return _cb
        colorwidget.build_palette_button(
            parent=_pal_group,
            target_input_tag=pal_tag,
            on_change=_make_pal_change(el),
        )


# ── Underlay panel ────────────────────────────────────────────────────────────
def refresh_underlay_panel() -> None:
    if not dpg.does_item_exist("underlay_panel"):
        return
    dpg.delete_item("underlay_panel", children_only=True)
    ul = state.st["underlay"]

    dpg.add_text(i18n.t("underlay_section"), parent="underlay_panel",
                 color=(200, 200, 100, 255))
    dpg.add_separator(parent="underlay_panel")

    def _open_dlg(s, u):
        import native_dialog
        import os
        native_dialog.open_file(
            title=i18n.t("underlay_section"),
            filetypes=[("Images", "*.png *.jpg *.jpeg"), ("All files", "*")],
            initial_dir=os.path.expanduser("~"),
            on_done=_load_underlay,
            fallback_tag="underlay_dlg",
        )

    def _load_underlay(path: str) -> None:
        underlay.load(path)
        state.g_underlay_dirty = True
        draw.redraw()

    dpg.add_button(label=i18n.t("underlay_add"), width=-1,
                   callback=_open_dlg, parent="underlay_panel")

    if ul["path"] is not None:
        def _remove(s, u):
            underlay.clear()
            state.g_underlay_dirty = True
            draw.redraw()

        dpg.add_button(label=i18n.t("underlay_remove"), width=-1,
                       callback=_remove, parent="underlay_panel")

        def _toggle_visible(s, v):
            ul["visible"] = v
            draw.redraw()

        dpg.add_checkbox(
            label=i18n.t("underlay_visible"),
            default_value=ul["visible"],
            callback=_toggle_visible,
            parent="underlay_panel",
        )

        def _set_opacity(s, v):
            ul["opacity"] = v
            draw.redraw()

        dpg.add_text(i18n.t("underlay_opacity"), parent="underlay_panel")
        dpg.add_slider_float(
            min_value=0.0, max_value=1.0,
            default_value=ul["opacity"],
            width=-1,
            callback=_set_opacity,
            parent="underlay_panel",
        )

        def _set_fit(s, v):
            ul["fit"] = "crop" if v == i18n.t("underlay_fit_crop") else "stretch"
            draw.redraw()

        dpg.add_radio_button(
            [i18n.t("underlay_fit_stretch"), i18n.t("underlay_fit_crop")],
            default_value=i18n.t("underlay_fit_crop") if ul["fit"] == "crop"
                          else i18n.t("underlay_fit_stretch"),
            horizontal=True,
            callback=_set_fit,
            parent="underlay_panel",
        )

    dpg.add_separator(parent="underlay_panel")


# ── Selection ─────────────────────────────────────────────────────────────────
def select(idx: int) -> None:
    state.st["selected"] = idx
    refresh_layers()
    refresh_props()
    draw.redraw()


# ── Convenience ───────────────────────────────────────────────────────────────
def refresh_all() -> None:
    refresh_layers()
    refresh_props()
    refresh_underlay_panel()
    draw.redraw()