#!/usr/bin/env python3

import nanogui
from nanogui import glfw
from nanogui import BoxLayout, Color, Label, Orientation, Screen, Widget, Window, Vector2i
from nanogui.nanovg import RGBAf, RGB


class CustomWidget(Widget):
    def __init__(self, parent, size=(100, 100), color=Color(192, 0, 0, 255)):
        super().__init__(parent)
        self._size = size
        self._color = RGBAf(color.r, color.g, color.b, color.w)

    def preferred_size(self, ctx):
        return Vector2i(self._size)

    def draw(self, ctx):
        pos = self.position()
        size = self.size()

        ctx.BeginPath()
        ctx.Rect(pos[0], pos[1], size[0], size[1])
        ctx.FillColor(RGB(128, 128, 0))
        ctx.Fill()
        ctx.BeginPath()
        ctx.StrokeWidth(5.0)
        ctx.StrokeColor(self._color)
        ctx.MoveTo(pos[0], pos[1])
        ctx.LineTo(pos[0] + size[0], pos[1] + size[1] / 3.0)
        ctx.Stroke()
        ctx.ClosePath()


class CustomWidgetApp(Screen):
    def __init__(self, size=(400, 300)):
        super().__init__(size, "NanoGUI CustomWidget")
        self.win = Window(self, "Demo Window")

        self.win.set_layout(BoxLayout(Orientation.Vertical, margin=20, spacing=20))
        self.resize_event(size)

        Label(self.win, "NanoGUI CustomWidget Demo", "sans-bold")

        cw = CustomWidget(self.win, (200, 100))

        self.draw_all()
        self.set_visible(True)
        self.perform_layout()

    def resize_event(self, size):
        self.win.set_fixed_size(size)
        self.win.set_size(size)
        self.win.center()
        self.win.perform_layout(self.nvg_context())
        super().resize_event(size)
        return True

    def keyboard_event(self, key, scancode, action, modifiers):
        if super().keyboard_event(key, scancode, action, modifiers):
            return True

        if key == glfw.KEY_ESCAPE and action == glfw.PRESS:
            self.set_visible(False)
            return True

        return False


if __name__ == "__main__":
    import gc

    nanogui.init()
    app = CustomWidgetApp()
    nanogui.mainloop(refresh=1 / 60.0 * 1000)
    del app
    gc.collect()
    nanogui.shutdown()