# HG changeset patch
# User Thibaut Girka <thib@sitedethib.com>
# Date 1312211842 -7200
# Node ID 787d2eb13c2d9ef93b1af1f924b2c8fd72933fce
# Parent  ffe0283c27de003e97637c47d37770e1562faf86
Add buggy stageviewer!

diff --git a/pytouhou/utils/matrix.py b/pytouhou/utils/matrix.py
new file mode 100644
--- /dev/null
+++ b/pytouhou/utils/matrix.py
@@ -0,0 +1,66 @@
+#TODO: find/learn to use a proper lib
+
+from math import sin, cos
+
+class Matrix(object):
+    def __init__(self):
+        self.data = [[0] * 4 for i in xrange(4)]
+
+
+    def mult(self, other_matrix):
+        result = Matrix()
+        for i in xrange(4):
+            for j in xrange(4):
+                result.data[i][j] = sum(self.data[i][a] * other_matrix.data[a][j] for a in xrange(4))
+        return result
+
+
+    @classmethod
+    def get_translation_matrix(cls, x, y, z):
+        matrix = cls()
+        matrix.data[0][0] = 1
+        matrix.data[1][1] = 1
+        matrix.data[2][2] = 1
+        matrix.data[3][3] = 1
+        matrix.data[0][3] = x
+        matrix.data[1][3] = y
+        matrix.data[2][3] = z
+        return matrix
+
+
+    @classmethod
+    def get_scaling_matrix(cls, x, y, z):
+        matrix = cls()
+        matrix.data[0][0] = x
+        matrix.data[1][1] = y
+        matrix.data[2][2] = z
+        matrix.data[3][3] = 1
+        return matrix
+
+
+    @classmethod
+    def get_rotation_matrix(cls, angle, axis):
+        """Only handles axis = x, y or z."""
+        matrix = cls()
+        matrix.data[3][3] = 1
+        if axis == 'x':
+            matrix.data[0][0] = 1
+            matrix.data[1][1] = cos(angle)
+            matrix.data[1][2] = -sin(angle)
+            matrix.data[2][1] = sin(angle)
+            matrix.data[2][2] = cos(angle)
+        elif axis == 'y':
+            matrix.data[0][0] = cos(angle)
+            matrix.data[0][2] = sin(angle)
+            matrix.data[2][0] = -sin(angle)
+            matrix.data[2][2] = cos(angle)
+            matrix.data[1][1] = 1
+        elif axis == 'z':
+            matrix.data[0][0] = cos(angle)
+            matrix.data[0][1] = -sin(angle)
+            matrix.data[1][0] = sin(angle)
+            matrix.data[1][1] = cos(angle)
+            matrix.data[2][2] = 1
+        else:
+            raise Exception
+        return matrix
diff --git a/stageviewer.py b/stageviewer.py
new file mode 100644
--- /dev/null
+++ b/stageviewer.py
@@ -0,0 +1,251 @@
+#!/usr/bin/env python
+
+import sys
+import os
+
+import struct
+from math import degrees, radians
+from io import BytesIO
+from itertools import chain
+
+import pygame
+
+import OpenGL
+OpenGL.FORWARD_COMPATIBLE_ONLY = True
+from OpenGL.GL import *
+from OpenGL.GLU import *
+
+from pytouhou.formats.pbg3 import PBG3
+from pytouhou.formats.std import Stage
+from pytouhou.formats.anm0 import Animations
+
+from pytouhou.utils.matrix import Matrix
+
+
+def load_texture(image, alpha_image=None):
+    textureSurface = pygame.image.load(image).convert_alpha()
+
+    if alpha_image:
+        alphaSurface = pygame.image.load(alpha_image)
+        assert textureSurface.get_size() == alphaSurface.get_size()
+        for x in range(alphaSurface.get_width()):
+            for y in range(alphaSurface.get_height()):
+                r, g, b, a = textureSurface.get_at((x, y))
+                color2 = alphaSurface.get_at((x, y))
+                textureSurface.set_at((x, y), (r, g, b, color2[0]))
+
+    textureData = pygame.image.tostring(textureSurface, 'RGBA', 1)
+
+    width = textureSurface.get_width()
+    height = textureSurface.get_height()
+
+    texture = glGenTextures(1)
+    glBindTexture(GL_TEXTURE_2D, texture)
+
+    glTexImage2D(GL_TEXTURE_2D, 0, GL_RGBA, width, height, 0, GL_RGBA,
+        GL_UNSIGNED_BYTE, textureData)
+
+    glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_MAG_FILTER, GL_LINEAR)
+    glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_MIN_FILTER, GL_LINEAR)
+
+    return texture, width, height
+
+
+
+def build_objects_faces(stage, anim):
+    objects_faces = []
+    for i, obj in enumerate(stage.objects):
+        faces = []
+        for script_index, x, y, z, width_override, height_override in obj.quads:
+            #TODO: move mof of it elsewhere
+            vertices = []
+            uvs = []
+            vertmat = Matrix()
+            vertmat.data[0][0] = -.5
+            vertmat.data[1][0] = -.5
+
+            vertmat.data[0][1] = .5
+            vertmat.data[1][1] = -.5
+
+            vertmat.data[0][2] = .5
+            vertmat.data[1][2] = .5
+
+            vertmat.data[0][3] = -.5
+            vertmat.data[1][3] = .5
+
+            for i in range(4):
+                vertmat.data[2][i] = 0.
+                vertmat.data[3][i] = 1.
+
+            properties = {}
+            for time, instr_type, data in anim.scripts[script_index]:
+                if instr_type == 15:
+                    properties[15] = b''
+                    break
+                elif time == 0: #TODO
+                    properties[instr_type] = data
+            #if 15 not in properties: #TODO: Skip properties
+            #    continue
+
+            #TODO: properties 3 and 4
+            if 1 in properties:
+                tx, ty, tw, th = anim.sprites[struct.unpack('<I', properties[1])[0]]
+            width, height = 1., 1.
+            if 2 in properties:
+                width, height = struct.unpack('<ff', properties[2])
+            width = width_override or width * 16.
+            height = height_override or height * 16.
+            transform = Matrix.get_scaling_matrix(width, height, 1.)
+            if 7 in properties:
+                transform = Matrix.get_scaling_matrix(-1., 1., 1.).mult(transform)
+            if 9 in properties:
+                rx, ry, rz = struct.unpack('<fff', properties[9])
+                transform = Matrix.get_rotation_matrix(-rx, 'x').mult(transform)
+                transform = Matrix.get_rotation_matrix(ry, 'y').mult(transform)
+                transform = Matrix.get_rotation_matrix(-rz, 'z').mult(transform) #TODO: minus, really?
+            if 23 in properties: # Reposition
+                transform = Matrix.get_translation_matrix(width / 2., height / 2., 0.).mult(transform)
+
+            transform = Matrix.get_translation_matrix(x, y, z).mult(transform)
+            vertmat = transform.mult(vertmat)
+
+            uvs = [(tx / 256.,         1. - (ty / 256.)),
+                   ((tx + tw) / 256.,  1. - (ty / 256.)),
+                   ((tx + tw) / 256.,  1. - ((ty + th) / 256.)),
+                   (tx / 256.,         1. - ((ty + th) / 256.))]
+
+            for i in xrange(4):
+                w = vertmat.data[3][i]
+                vertices.append((vertmat.data[0][i] / w, vertmat.data[1][i] / w, vertmat.data[2][i] / w))
+            faces.append((vertices, uvs))
+        objects_faces.append(faces)
+    return objects_faces
+
+
+
+def objects_faces_to_vertices_uvs(objects):
+    vertices = tuple(vertex for obj in objects for face in obj for vertex in face[0]) #TODO: check
+    uvs = tuple(uv for obj in objects for face in obj for uv in face[1]) #TODO: check
+    return vertices, uvs
+
+
+
+def main(path, stage_num):
+    # Initialize pygame
+    pygame.init()
+    window = pygame.display.set_mode((384, 448), pygame.OPENGL | pygame.DOUBLEBUF)
+
+    # Initialize OpenGL
+    glHint(GL_PERSPECTIVE_CORRECTION_HINT, GL_NICEST)
+    glEnable(GL_DEPTH_TEST)
+    glEnable(GL_BLEND)
+    glBlendFunc(GL_SRC_ALPHA, GL_ONE_MINUS_SRC_ALPHA)
+    glEnable(GL_TEXTURE_2D)
+    glEnableClientState(GL_VERTEX_ARRAY)
+    glEnableClientState(GL_TEXTURE_COORD_ARRAY)
+
+    # Load data
+    with open(path, 'rb') as file:
+        archive = PBG3.read(file)
+        stage = Stage.read(BytesIO(archive.extract('stage%d.std' % stage_num)))
+        anim = Animations.read(BytesIO(archive.extract('stg%dbg.anm' % stage_num)))
+        textures_components = [None, None]
+        for i, component_name in enumerate((anim.first_name, anim.secondary_name)):
+            if component_name:
+                textures_components[i] = BytesIO(archive.extract(os.path.basename(component_name)))
+        texture = load_texture(*textures_components)
+
+    print(stage.name)
+
+    uvs = []
+    vertices = []
+    objects_faces = build_objects_faces(stage, anim)
+    objects_instances_faces = []
+    for obj, ox, oy, oz in stage.object_instances:
+        obj_id = stage.objects.index(obj)
+
+        obj_instance = []
+        for face_vertices, face_uvs in objects_faces[obj_id]:
+            obj_instance.append((tuple((x + ox, y + oy, z + oz) for x, y, z in face_vertices),
+                                face_uvs))
+        objects_instances_faces.append(obj_instance)
+
+    vertices, uvs = objects_faces_to_vertices_uvs(objects_instances_faces)
+    nb_vertices = len(vertices)
+    vertices_format = 'f' * (3 * nb_vertices)
+    uvs_format = 'f' * (2 * nb_vertices)
+    vertices, uvs = objects_faces_to_vertices_uvs(objects_instances_faces)
+    glVertexPointer(3, GL_FLOAT, 0, struct.pack(vertices_format, *chain(*vertices)))
+    glTexCoordPointer(2, GL_FLOAT, 0, struct.pack(uvs_format, *chain(*uvs)))
+
+    x, y, z = 0, 0, 0
+    frame = 0
+    interpolation = 0, 0, 0
+
+    # Main loop
+    clock = pygame.time.Clock()
+    while True:
+        for event in pygame.event.get():
+            if event.type == pygame.QUIT or (event.type == pygame.KEYDOWN and event.key in (pygame.K_ESCAPE, pygame.K_q)):
+                sys.exit(0)
+            elif event.type == pygame.KEYDOWN:
+                if event.key == pygame.K_RETURN and event.mod & pygame.KMOD_ALT:
+                    pygame.display.toggle_fullscreen()
+
+        glClear(GL_COLOR_BUFFER_BIT | GL_DEPTH_BUFFER_BIT)
+
+        for frame_num, message_type, data in stage.script:
+            if frame_num == frame and message_type == 3:
+                duration, junk1, junk2 = struct.unpack('<III', data)
+                interpolation = frame_num, duration, frame_num + duration
+                old_unknownx, old_dy, old_fov = unknownx, dy, fov
+            if frame_num <= frame and message_type == 0:
+                last_message = frame_num, message_type, data
+            if frame_num <= frame and message_type == 2:
+                next_unknownx, next_dy, next_fov = struct.unpack('<fff', data)
+            if frame_num > frame and message_type == 0:
+                next_message = frame_num, message_type, data
+                break
+
+        if frame < interpolation[2]:
+            truc = float(frame - interpolation[0]) / interpolation[1]
+            unknownx = old_unknownx + (next_unknownx - old_unknownx) * truc
+            dy = old_dy + (next_dy - old_dy) * truc
+            fov = old_fov + (next_fov - old_fov) * truc
+        else:
+            unknownx, dy, fov = next_unknownx, next_dy, next_fov
+
+        x1, y1, z1 = struct.unpack('<fff', last_message[2])
+        x2, y2, z2 = struct.unpack('<fff', next_message[2])
+
+        truc = (float(frame) - last_message[0]) / (next_message[0] - last_message[0])
+
+        #TODO: find proper coordinates (get rid of arbitrary values)
+        x = x1 + (x2 - x1) * truc + 170.
+        y = y1 + (y2 - y1) * truc
+        z = z1 + (z2 - z1) * truc - 720.
+
+        glMatrixMode(GL_PROJECTION)
+        glLoadIdentity()
+        gluPerspective(degrees(fov), float(window.get_width())/window.get_height(), 20, 2000)
+
+        glMatrixMode(GL_MODELVIEW)
+        glLoadIdentity()
+        gluLookAt(x, y, z, x, y - dy, 0., 0., -1., 0.)
+
+        glDrawArrays(GL_QUADS, 0, nb_vertices)
+
+        pygame.display.flip()
+        clock.tick(120)
+        frame += 1
+
+
+
+try:
+    file_path, stage_num = sys.argv[1:]
+    stage_num = int(stage_num)
+except ValueError:
+    print('Usage: %s std_dat_path stage_num' % sys.argv[0])
+else:
+    main(file_path, stage_num)
+