forked from mttu-developers/konabot
37 lines
1.2 KiB
Python
37 lines
1.2 KiB
Python
"""
|
|
headless moderngl
|
|
"""
|
|
import pathlib
|
|
import moderngl
|
|
import numpy as np
|
|
from PIL import Image
|
|
from konabot.plugins.marchtoy.command import Scene
|
|
|
|
async def render(command: str, res: tuple[int, int]):
|
|
fs = Scene(command).compile()
|
|
PATH = pathlib.Path(__file__).parent / "shaders"
|
|
with (PATH / "vert.glsl").open(encoding='utf-8') as f:
|
|
vs = f.read()
|
|
ctx = moderngl.create_context(standalone=True, backend='egl')
|
|
ctx.gc_mode = "auto"
|
|
try:
|
|
program = ctx.program(
|
|
vertex_shader=vs,
|
|
fragment_shader=fs
|
|
)
|
|
except Exception as e:
|
|
raise Exception(f"cannot compile glsl: {e}") from e
|
|
uniform = program['u_resolution']
|
|
uniform.write(np.array(res, dtype=np.float32))
|
|
vertices = np.array([-1.0, -1.0, -1.0, 1.0, 1.0, -1.0, 1.0, 1.0], dtype=np.float32)
|
|
indices = np.array([0, 1, 2, 1, 2, 3], dtype=np.int32)
|
|
ibo = ctx.buffer(indices)
|
|
vbo = ctx.buffer(vertices)
|
|
vao = ctx.vertex_array(program, vbo, 'in_position', index_buffer = ibo)
|
|
fbo = ctx.simple_framebuffer(res)
|
|
fbo.use()
|
|
fbo.clear(0.0, 0.0, 0.0, 0.0)
|
|
vao.render(moderngl.TRIANGLES)
|
|
return Image.frombytes('RGBA', fbo.size, fbo.read(components=4), 'raw', 'RGBA', 0, -1)
|
|
|