GeezerVison

class ImguiFigure
def __init__(self, *args, **kwargs):
    super()
    self.imgui_renderer ....
    # more imgui init stuff
    self._guis = list()

def render(self):
    super()
    # imgui render stuff
    for gui in self.guis:
        gui.update()
    imgui.end_frame()
    ...
    canvas.request_draw()

def add_ui(self, edge: str, callable_or_instance_TBD):  # edge is the top, bottom, left, or right edge of the canvas
    # parse edge and set Figure bbox, i.e. [x, y, width, height] that the pygfx renderer uses
    self._guis["edge"] = callable_or_instance_TBD

def set_ui_size(self, edge: str, size: int):
    # set imgui reserved size
    self._edges["edge"].size = size

@property
def width(self) -> int
    return self.renderer.logical_size[0] - self._edges["left"] -self._edges["right"]

# likewise properties for height, x, and y
# use these [x, y, width, height] properties elsewhere in the `Figure`
# such as the subplots and other `PlotAreas` instead of `renderer.logical_size`

from imgui_bundle import imgui, icons_fontawesome_6 as fa, imgui_ctx

from .._plot_area import PlotArea


ID_COUNTER = 0


class SubplotToolbar:
    def __init__(self, subplot: PlotArea, icons: imgui.ImFont):
        self._subplot = subplot
        self.icons = icons

        # required to prevent conflict with multiple Figures
        global ID_COUNTER
        ID_COUNTER += 1

        self.id = ID_COUNTER

        self.bbox = subplot.get_rect()
        self._subplot.get_rect = self.get_rect

    def get_rect(self):
        x, y, width, height = self.bbox

        return x + 10, y + 50, width - 10, height - 60

    def update(self):
        x, y, width, height = self._subplot.get_rect()

        # pos = (x, y + height)

        size = self.bbox[2:]
        pos = self.bbox[:2]

        imgui.set_next_window_size(size, imgui.Cond_.appearing)
        imgui.set_next_window_pos(pos, imgui.Cond_.appearing)
        flags = imgui.WindowFlags_.no_collapse | imgui.WindowFlags_.no_background #| imgui.WindowFlags_.no_title_bar

        imgui.begin(f"Toolbar-{self._subplot.position}", p_open=None, flags=flags)

        width, height = imgui.get_window_size()
        x, y = imgui.get_window_pos()


        imgui.push_font(self.icons)

        imgui.push_id(self.id)  # push ID to prevent conflict between multiple figs with same UI
        with imgui_ctx.begin_horizontal(f"toolbar-{self._subplot.position}"):
            # autoscale button
            if imgui.button(fa.ICON_FA_MAXIMIZE):
                self._subplot.auto_scale()
            imgui.pop_font()
            if imgui.is_item_hovered(0):
                imgui.set_tooltip("autoscale scene")

            # center scene
            imgui.push_font(self.icons)
            if imgui.button(fa.ICON_FA_ALIGN_CENTER):
                self._subplot.center_scene()

            # checkbox controller
            _, self._subplot.controller.enabled = imgui.checkbox(fa.ICON_FA_COMPUTER_MOUSE, self._subplot.controller.enabled)

            # checkbox maintain_apsect
            _, self._subplot.camera.maintain_aspect = imgui.checkbox(fa.ICON_FA_EXPAND, self._subplot.camera.maintain_aspect)

            imgui.pop_font()

            _, flip_y = imgui.checkbox("flip-y", self._subplot.camera.local.scale_y < 0)
            if flip_y and self._subplot.camera.local.scale_y > 0:
                self._subplot.camera.local.scale_y *= -1
            elif not flip_y and self._subplot.camera.local.scale_y < 0:
                self._subplot.camera.local.scale_y *= -1

        imgui.pop_id()

        self.bbox = (x, y, width, height)
        self._subplot.set_viewport_rect()

        imgui.end()