main.py

from pyglet.gl import *
from pyglet.graphics.shader import Shader, ShaderProgram
from pyglet.math import Vec3
from pyglet.window import key
import pyglet


from sombra_engine.camera import FPSCamera
from sombra_engine.debug.gizmo import Gizmo
from sombra_engine.models import Model, Wireframe
from sombra_engine.models.obj import OBJLoader
from sombra_engine.scene import Scene


window = pyglet.window.Window(caption="Sombra Engine")
camera = FPSCamera(
    window, position=Vec3(0.0, 0.0, -15.0), pitch=90, yaw=90
)
batch = pyglet.graphics.Batch()
model: Model | None = None
wf: Wireframe | None = None
gizmo = Gizmo(batch=batch)


# def on_key_press(symbol, mod):
#     if symbol == key.P:
#         pyglet.image.get_buffer_manager().get_color_buffer().save(
#             'docs/screenshot.png'
#         )


@window.event
def on_draw():
    window.clear()
    glEnable(GL_CULL_FACE)
    glEnable(GL_DEPTH_TEST)
    glDepthFunc(GL_LESS)
    batch.draw()


def update(dt: float):
    mesh = model.meshes[0]
    rx = mesh.transform.rotation.x
    mesh.rotate_x(rx + dt / 3)


def main():
    global model, wf
    with open('sombra_engine/shaders/default.vert') as f:
        vert_shader = Shader(f.read(), 'vertex')
    # with open('sombra_engine/shaders/blinn_barycentric.frag') as f:
    with open('sombra_engine/shaders/blinn.frag') as f:
    # with open('sombra_engine/shaders/lambert.frag') as f:
    # with open('sombra_engine/shaders/solid.frag') as f:
    # with open('sombra_engine/shaders/normals.frag') as f:
    # with open('sombra_engine/shaders/normals_from_bump.frag') as f:
    # with open('sombra_engine/shaders/specular.frag') as f:
        frag_shader = Shader(f.read(), 'fragment')
    program = ShaderProgram(vert_shader, frag_shader)

    scene = Scene()
    scene.create_light(Vec3(100.0, 150.0, -7.0), Vec3(1.0, 1.0, 1.0))
    program['light.position'] = scene.lights[0].position
    program['light.color'] = scene.lights[0].color

    program['eye'] = camera.position

    # model_group = pyglet.graphics.Group()
    # model_group.visible = False
    model = OBJLoader.load(
        # "tests/data/ancient_house.obj", "House",
        "tests/data/yoda/yoda.obj", "Yoda",
        # group=model_group,
        program=program, batch=batch
    )
    print(f"object {model.name} loaded")
    wf = Wireframe(model.meshes[0], vert_shader, batch=batch)

    # window.push_handlers(on_key_press)

    # pyglet.clock.schedule_interval(update, 1 / 60)
    pyglet.app.run()


if __name__ == '__main__':
    main()