from OpenGL.GL import shaders
from libs import *
from entity import Entity
from mapgen import MapGen
from mesh import SquareMesh
from player import Player
from region import WorldRegion

VS = f"""
#version 120
{"\n".join([f"attribute {a};" for a in VERTEX_ATTRS])}
{"\n".join([f"varying {a}F;" for a in VERTEX_ATTRS])}
{"\n".join([f"uniform {v[1]} {k};" for (k, v) in UNIFORMS.items() if v[0] == "vert"])}
void main() {{
    gl_Position = MVP * vec4(position, 1.0);
{"\n".join([f"    {a.split()[1]}F = {a.split()[1]};" for a in VERTEX_ATTRS])}
}}
"""

FS = f"""
#version 120
{"\n".join([f"varying {a}F;" for a in VERTEX_ATTRS])}
{"\n".join([f"uniform {v[1]} {k};" for (k, v) in UNIFORMS.items() if v[0] == "frag"])}
void main() {{
    float ambient = reflsF.x;
    vec3 lightDir = normalize(lightPos.xyz - positionF);
    float lambert = dot(normalF, lightDir);
    float diffuse = reflsF.y * lightPos.w * lambert / pow(distance(lightPos.xyz, positionF), 2);
    float specular = reflsF.z * lightPos.w * pow(max(dot(normalize(camPos - positionF), reflect(-lightDir, normalF)), 0.0), reflsF.w);
    gl_FragColor = vec4((ambient + diffuse + specular) * colF.xyz, colF.w);
}}
"""

FPS = 60

def debug_mode(scene, cam):
    pass

def click_mesh(self, button, coords):
    match button:
        case 1:
            self.set_voxel(coords[1])
        case 2:
            print(f"coords {coords}")
        case 3:
            offset = axis2offset(coords[0])
            self.set_voxel(coords[1] + offset, 2)
    if button % 2 == 1:
        for i in range(len(VERTEX_ATTRS)):
            update_attr(self, i)

def build_scene(mg):
    box = WorldRegion()
    print(f"map generated with {mg.world_natural(box)} voxels")
    box.remesh()
    box.onclick = click_mesh
    return [box]

def main():
    pygame.init()
    pygame.display.set_mode((640, 480), pygame.DOUBLEBUF | pygame.OPENGL | pygame.RESIZABLE)
    pygame.display.set_caption("OpenGL")
    sp = shaders.compileProgram(
        shaders.compileShader(VS, GL_VERTEX_SHADER),
        shaders.compileShader(FS, GL_FRAGMENT_SHADER)
    )
    for (k, v) in UNIFORMS.items():
        v[2] = glGetUniformLocation(sp, k)
    glUseProgram(sp)
    glEnable(GL_DEPTH_TEST)
    glClearColor(0.1, 0.1, 0.1, 1.0)
    glUniform4fv(UNIFORMS["lightPos"][2], 1, vec([5, 10, 10, 10]))
    mg = MapGen()
    scene = build_scene(mg)
    clock = pygame.time.Clock()
    cam = Player(speed=10/FPS)
    cam.pos = [8, 16, 8]
    cam.set_focus(True)
    while True:
        for ev in pygame.event.get():
            if ev.type == pygame.QUIT or ev.type == pygame.KEYUP and ev.unicode == "q":
                pygame.quit()
                return
            elif ev.type == pygame.KEYUP and ev.unicode == "d":
                debug_mode(scene, cam)
            cam.handle_event(ev, scene)
        cam.update_pos()
        glClear(GL_COLOR_BUFFER_BIT | GL_DEPTH_BUFFER_BIT)
        for ent in scene:
            cam.draw(ent)
        perf = clock.get_fps() / FPS
        cam.draw_hud(scene[0], perf)
        pygame.display.flip()
        clock.tick(FPS)

main()