Source code for wolfhece._bridge_manager

"""
Companion manager for bridge menus and actions.
Extracted from PyDraw.WolfMapViewer.
"""
from __future__ import annotations
import logging
import wx
from typing import TYPE_CHECKING

from .PyTranslate import _
from ._action_kind import ActionKind

if TYPE_CHECKING:
    from .PyDraw import WolfMapViewer




class BridgeManager:
    """Manages the &Bridges menu and its associated actions."""

    def __init__(self, viewer: 'WolfMapViewer') -> None:


        self._viewer = viewer



        self._menu: wx.Menu | None = None



        self._item_add: wx.MenuItem | None = None



        self._item_find: wx.MenuItem | None = None



        self._item_edit: wx.MenuItem | None = None


    # ------------------------------------------------------------------
    # Menu build
    # ------------------------------------------------------------------



    def menu_build(self) -> None:
        v = self._viewer
        if self._menu is not None:
            return

        self._menu = wx.Menu()
        v.menubar.Append(self._menu, _('&Bridges'))

        self._item_add  = self._menu.Append(wx.ID_ANY, _('New bridge'),  _('Add a new bridge to the active collection...'))
        self._item_find = self._menu.Append(wx.ID_ANY, _('Pick bridge'), _("Pick the mouse's nearest bridge"))
        self._item_edit = self._menu.Append(wx.ID_ANY, _('Edit bridge'), _('Edit the active bridge'))

        v.Bind(wx.EVT_MENU, self.on_add,  self._item_add)
        v.Bind(wx.EVT_MENU, self.on_edit, self._item_edit)
        v.Bind(wx.EVT_MENU, self.on_find, self._item_find)


    # ------------------------------------------------------------------
    # Action handlers
    # ------------------------------------------------------------------



    def on_add(self, e: wx.Event) -> None:
        from .PyDraw import draw_type

        v = self._viewer
        if v.active_bridges is None:
            logging.warning(_('No bridge collection !'))
            return

        dlg = wx.TextEntryDialog(None, _('Enter the new bridge id'), _('New bridge id'), 'bridge')
        if dlg.ShowModal() == wx.ID_OK:
            newid = dlg.GetValue()
            while newid in v.get_list_keys(drawing_type=draw_type.VECTORS):
                newid = newid + '_'
            newbridge = v.active_bridges.addnew(newid)
            v.add_object('vector', newobj=newbridge, id=newid)




    def on_edit(self, e: wx.Event) -> None:
        from .PyDraw import draw_type

        v = self._viewer
        if v.active_bridge is None:
            logging.warning(_('No active bridge to edit !'))
            return

        keys = v.get_list_keys(drawing_type=draw_type.VECTORS)
        newid = v.active_bridge.idx
        while newid in keys:
            newid = newid + '_'
        v.add_object('vector', newobj=v.active_bridge, id=newid)




    def on_find(self, e: wx.Event) -> None:
        self._viewer.start_action(ActionKind.PICK_BRIDGE, _('Right click to pick the nearest bridge'))




    def pick(self, x: float, y: float) -> None:
        v = self._viewer
        if v.active_bridges is None:
            logging.warning(_('No bridges to pick !'))
            return
        v.active_bridge = v.active_bridges.find_nearest(x, y)