rig_picker/constants.py

import gpu
from pathlib import Path

class LazyDict(dict):
    def __getitem__(self, k):
        v = super().__getitem__(k)
        if callable(v):
            v = v()
            super().__setitem__(k, v)
        return v

    def get(self, k, default=None):
        if k in self:
            return self.__getitem__(k)
        return default


PICKERS = {}


MODULE_DIR = Path(__file__).parent
SHADER_DIR = MODULE_DIR / 'shaders'

SHADERS = LazyDict()

vertex_shader = Path(SHADER_DIR, "dash_shader.vert").read_text(encoding='utf-8')
fragment_shader = Path(SHADER_DIR, "dash_shader.frag").read_text(encoding='utf-8')

SHADERS['dashed_line'] = lambda : gpu.types.GPUShader(vertex_shader, fragment_shader)