annotate stageviewer.py @ 14:07a7f28c8aaa

Minor refactoring
author Thibaut Girka <thib@sitedethib.com>
date Fri, 05 Aug 2011 14:54:32 +0200
parents 58bc264aba38
children 07fba4e1da65
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
4
787d2eb13c2d Add buggy stageviewer!
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
1 #!/usr/bin/env python
787d2eb13c2d Add buggy stageviewer!
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
2
787d2eb13c2d Add buggy stageviewer!
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
3 import sys
787d2eb13c2d Add buggy stageviewer!
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
4 import os
787d2eb13c2d Add buggy stageviewer!
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
5
787d2eb13c2d Add buggy stageviewer!
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
6 import struct
787d2eb13c2d Add buggy stageviewer!
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
7 from math import degrees, radians
787d2eb13c2d Add buggy stageviewer!
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
8 from io import BytesIO
787d2eb13c2d Add buggy stageviewer!
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
9 from itertools import chain
787d2eb13c2d Add buggy stageviewer!
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
10
787d2eb13c2d Add buggy stageviewer!
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
11 import pygame
787d2eb13c2d Add buggy stageviewer!
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
12
13
58bc264aba38 Refactor
Thibaut Girka <thib@sitedethib.com>
parents: 11
diff changeset
13 from pytouhou.formats.pbg3 import PBG3
58bc264aba38 Refactor
Thibaut Girka <thib@sitedethib.com>
parents: 11
diff changeset
14 from pytouhou.game.background import Background
14
07a7f28c8aaa Minor refactoring
Thibaut Girka <thib@sitedethib.com>
parents: 13
diff changeset
15 from pytouhou.opengl.texture import load_texture
13
58bc264aba38 Refactor
Thibaut Girka <thib@sitedethib.com>
parents: 11
diff changeset
16
4
787d2eb13c2d Add buggy stageviewer!
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
17 import OpenGL
787d2eb13c2d Add buggy stageviewer!
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
18 OpenGL.FORWARD_COMPATIBLE_ONLY = True
787d2eb13c2d Add buggy stageviewer!
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
19 from OpenGL.GL import *
787d2eb13c2d Add buggy stageviewer!
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
20 from OpenGL.GLU import *
787d2eb13c2d Add buggy stageviewer!
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
21
787d2eb13c2d Add buggy stageviewer!
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
22
787d2eb13c2d Add buggy stageviewer!
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
23 def main(path, stage_num):
787d2eb13c2d Add buggy stageviewer!
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
24 # Initialize pygame
787d2eb13c2d Add buggy stageviewer!
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
25 pygame.init()
787d2eb13c2d Add buggy stageviewer!
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
26 window = pygame.display.set_mode((384, 448), pygame.OPENGL | pygame.DOUBLEBUF)
787d2eb13c2d Add buggy stageviewer!
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
27
787d2eb13c2d Add buggy stageviewer!
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
28 # Initialize OpenGL
5
aa201d8cfc19 Fix camera handling, thanks, elghinn!
Thibaut Girka <thib@sitedethib.com>
parents: 4
diff changeset
29 glMatrixMode(GL_PROJECTION)
aa201d8cfc19 Fix camera handling, thanks, elghinn!
Thibaut Girka <thib@sitedethib.com>
parents: 4
diff changeset
30 glLoadIdentity()
10
059ea0ea8d38 Use the original game's near and far planes.
Thibaut Girka <thib@sitedethib.com>
parents: 9
diff changeset
31 gluPerspective(30, float(window.get_width())/window.get_height(), 101010101./2010101., 101010101./10101.)
5
aa201d8cfc19 Fix camera handling, thanks, elghinn!
Thibaut Girka <thib@sitedethib.com>
parents: 4
diff changeset
32
4
787d2eb13c2d Add buggy stageviewer!
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
33 glEnable(GL_DEPTH_TEST)
787d2eb13c2d Add buggy stageviewer!
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
34 glEnable(GL_BLEND)
787d2eb13c2d Add buggy stageviewer!
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
35 glEnable(GL_TEXTURE_2D)
7
02a5f5314a19 Add preliminary fog support
Thibaut Girka <thib@sitedethib.com>
parents: 6
diff changeset
36 glEnable(GL_FOG)
14
07a7f28c8aaa Minor refactoring
Thibaut Girka <thib@sitedethib.com>
parents: 13
diff changeset
37 glHint(GL_FOG_HINT, GL_NICEST)
07a7f28c8aaa Minor refactoring
Thibaut Girka <thib@sitedethib.com>
parents: 13
diff changeset
38 glHint(GL_PERSPECTIVE_CORRECTION_HINT, GL_NICEST)
07a7f28c8aaa Minor refactoring
Thibaut Girka <thib@sitedethib.com>
parents: 13
diff changeset
39 glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_MAG_FILTER, GL_LINEAR)
07a7f28c8aaa Minor refactoring
Thibaut Girka <thib@sitedethib.com>
parents: 13
diff changeset
40 glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_MIN_FILTER, GL_LINEAR)
07a7f28c8aaa Minor refactoring
Thibaut Girka <thib@sitedethib.com>
parents: 13
diff changeset
41 glBlendFunc(GL_SRC_ALPHA, GL_ONE_MINUS_SRC_ALPHA)
4
787d2eb13c2d Add buggy stageviewer!
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
42 glEnableClientState(GL_VERTEX_ARRAY)
787d2eb13c2d Add buggy stageviewer!
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
43 glEnableClientState(GL_TEXTURE_COORD_ARRAY)
787d2eb13c2d Add buggy stageviewer!
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
44
787d2eb13c2d Add buggy stageviewer!
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
45 # Load data
787d2eb13c2d Add buggy stageviewer!
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
46 with open(path, 'rb') as file:
787d2eb13c2d Add buggy stageviewer!
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
47 archive = PBG3.read(file)
13
58bc264aba38 Refactor
Thibaut Girka <thib@sitedethib.com>
parents: 11
diff changeset
48 background = Background(archive, stage_num)
14
07a7f28c8aaa Minor refactoring
Thibaut Girka <thib@sitedethib.com>
parents: 13
diff changeset
49 texture = load_texture(archive, background.anim)
4
787d2eb13c2d Add buggy stageviewer!
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
50
13
58bc264aba38 Refactor
Thibaut Girka <thib@sitedethib.com>
parents: 11
diff changeset
51 print(background.stage.name)
8
f3ff96192476 Fix z-sorting issues
Thibaut Girka <thib@sitedethib.com>
parents: 7
diff changeset
52
4
787d2eb13c2d Add buggy stageviewer!
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
53 frame = 0
787d2eb13c2d Add buggy stageviewer!
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
54
787d2eb13c2d Add buggy stageviewer!
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
55 # Main loop
787d2eb13c2d Add buggy stageviewer!
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
56 clock = pygame.time.Clock()
787d2eb13c2d Add buggy stageviewer!
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
57 while True:
13
58bc264aba38 Refactor
Thibaut Girka <thib@sitedethib.com>
parents: 11
diff changeset
58 # Check events
4
787d2eb13c2d Add buggy stageviewer!
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
59 for event in pygame.event.get():
787d2eb13c2d Add buggy stageviewer!
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
60 if event.type == pygame.QUIT or (event.type == pygame.KEYDOWN and event.key in (pygame.K_ESCAPE, pygame.K_q)):
787d2eb13c2d Add buggy stageviewer!
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
61 sys.exit(0)
787d2eb13c2d Add buggy stageviewer!
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
62 elif event.type == pygame.KEYDOWN:
787d2eb13c2d Add buggy stageviewer!
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
63 if event.key == pygame.K_RETURN and event.mod & pygame.KMOD_ALT:
787d2eb13c2d Add buggy stageviewer!
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
64 pygame.display.toggle_fullscreen()
787d2eb13c2d Add buggy stageviewer!
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
65
13
58bc264aba38 Refactor
Thibaut Girka <thib@sitedethib.com>
parents: 11
diff changeset
66 # Update game
58bc264aba38 Refactor
Thibaut Girka <thib@sitedethib.com>
parents: 11
diff changeset
67 background.update(frame)
58bc264aba38 Refactor
Thibaut Girka <thib@sitedethib.com>
parents: 11
diff changeset
68
58bc264aba38 Refactor
Thibaut Girka <thib@sitedethib.com>
parents: 11
diff changeset
69 # Draw everything
4
787d2eb13c2d Add buggy stageviewer!
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
70 glClear(GL_COLOR_BUFFER_BIT | GL_DEPTH_BUFFER_BIT)
787d2eb13c2d Add buggy stageviewer!
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
71
13
58bc264aba38 Refactor
Thibaut Girka <thib@sitedethib.com>
parents: 11
diff changeset
72 glVertexPointer(3, GL_FLOAT, 0, background._vertices)
58bc264aba38 Refactor
Thibaut Girka <thib@sitedethib.com>
parents: 11
diff changeset
73 glTexCoordPointer(2, GL_FLOAT, 0, background._uvs)
4
787d2eb13c2d Add buggy stageviewer!
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
74
13
58bc264aba38 Refactor
Thibaut Girka <thib@sitedethib.com>
parents: 11
diff changeset
75 fog_b, fog_g, fog_r, _, fog_start, fog_end = background.fog_interpolator.values
58bc264aba38 Refactor
Thibaut Girka <thib@sitedethib.com>
parents: 11
diff changeset
76 x, y, z = background.position_interpolator.values
58bc264aba38 Refactor
Thibaut Girka <thib@sitedethib.com>
parents: 11
diff changeset
77 unknownx, dy, dz = background.position2_interpolator.values
7
02a5f5314a19 Add preliminary fog support
Thibaut Girka <thib@sitedethib.com>
parents: 6
diff changeset
78
02a5f5314a19 Add preliminary fog support
Thibaut Girka <thib@sitedethib.com>
parents: 6
diff changeset
79 glFogi(GL_FOG_MODE, GL_LINEAR)
02a5f5314a19 Add preliminary fog support
Thibaut Girka <thib@sitedethib.com>
parents: 6
diff changeset
80 glFogf(GL_FOG_START, fog_start)
13
58bc264aba38 Refactor
Thibaut Girka <thib@sitedethib.com>
parents: 11
diff changeset
81 glFogf(GL_FOG_END, fog_end)
7
02a5f5314a19 Add preliminary fog support
Thibaut Girka <thib@sitedethib.com>
parents: 6
diff changeset
82 glFogfv(GL_FOG_COLOR, (fog_r / 255., fog_g / 255., fog_b / 255., 1.))
02a5f5314a19 Add preliminary fog support
Thibaut Girka <thib@sitedethib.com>
parents: 6
diff changeset
83
13
58bc264aba38 Refactor
Thibaut Girka <thib@sitedethib.com>
parents: 11
diff changeset
84 #TODO
4
787d2eb13c2d Add buggy stageviewer!
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
85 glMatrixMode(GL_MODELVIEW)
787d2eb13c2d Add buggy stageviewer!
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
86 glLoadIdentity()
11
548662d70860 Add some explanations to the magic constants
Thibaut Girka <thib@sitedethib.com>
parents: 10
diff changeset
87 # Some explanations on the magic constants:
548662d70860 Add some explanations to the magic constants
Thibaut Girka <thib@sitedethib.com>
parents: 10
diff changeset
88 # 192. = 384. / 2. = width / 2.
548662d70860 Add some explanations to the magic constants
Thibaut Girka <thib@sitedethib.com>
parents: 10
diff changeset
89 # 224. = 448. / 2. = height / 2.
548662d70860 Add some explanations to the magic constants
Thibaut Girka <thib@sitedethib.com>
parents: 10
diff changeset
90 # 835.979370 = 224./math.tan(math.radians(15)) = (height/2.)/math.tan(math.radians(fov/2))
548662d70860 Add some explanations to the magic constants
Thibaut Girka <thib@sitedethib.com>
parents: 10
diff changeset
91 # This is so that objects on the (O, x, y) plane use pixel coordinates
7
02a5f5314a19 Add preliminary fog support
Thibaut Girka <thib@sitedethib.com>
parents: 6
diff changeset
92 gluLookAt(192., 224., - 835.979370 * dz,
02a5f5314a19 Add preliminary fog support
Thibaut Girka <thib@sitedethib.com>
parents: 6
diff changeset
93 192., 224. - dy, 750 - 835.979370 * dz, 0., -1., 0.) #TODO: 750 might not be accurate
5
aa201d8cfc19 Fix camera handling, thanks, elghinn!
Thibaut Girka <thib@sitedethib.com>
parents: 4
diff changeset
94 #print(glGetFloat(GL_MODELVIEW_MATRIX))
aa201d8cfc19 Fix camera handling, thanks, elghinn!
Thibaut Girka <thib@sitedethib.com>
parents: 4
diff changeset
95 glTranslatef(-x, -y, -z)
4
787d2eb13c2d Add buggy stageviewer!
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
96
13
58bc264aba38 Refactor
Thibaut Girka <thib@sitedethib.com>
parents: 11
diff changeset
97 glDrawArrays(GL_QUADS, 0, background.nb_vertices)
4
787d2eb13c2d Add buggy stageviewer!
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
98
11
548662d70860 Add some explanations to the magic constants
Thibaut Girka <thib@sitedethib.com>
parents: 10
diff changeset
99 #TODO: show the game itself
548662d70860 Add some explanations to the magic constants
Thibaut Girka <thib@sitedethib.com>
parents: 10
diff changeset
100 # It is displayed on (0, 0, 0), (0, 448, 0), (388, 448, 0), (388, 0, 0)
548662d70860 Add some explanations to the magic constants
Thibaut Girka <thib@sitedethib.com>
parents: 10
diff changeset
101 # using a camera at (192, 224, -835.979370) looking right behind itself
548662d70860 Add some explanations to the magic constants
Thibaut Girka <thib@sitedethib.com>
parents: 10
diff changeset
102 # Depth test should be disabled when rendering the game
548662d70860 Add some explanations to the magic constants
Thibaut Girka <thib@sitedethib.com>
parents: 10
diff changeset
103
4
787d2eb13c2d Add buggy stageviewer!
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
104 pygame.display.flip()
787d2eb13c2d Add buggy stageviewer!
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
105 clock.tick(120)
787d2eb13c2d Add buggy stageviewer!
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
106 frame += 1
787d2eb13c2d Add buggy stageviewer!
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
107
787d2eb13c2d Add buggy stageviewer!
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
108
787d2eb13c2d Add buggy stageviewer!
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
109
787d2eb13c2d Add buggy stageviewer!
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
110 try:
787d2eb13c2d Add buggy stageviewer!
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
111 file_path, stage_num = sys.argv[1:]
787d2eb13c2d Add buggy stageviewer!
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
112 stage_num = int(stage_num)
787d2eb13c2d Add buggy stageviewer!
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
113 except ValueError:
787d2eb13c2d Add buggy stageviewer!
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
114 print('Usage: %s std_dat_path stage_num' % sys.argv[0])
787d2eb13c2d Add buggy stageviewer!
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
115 else:
787d2eb13c2d Add buggy stageviewer!
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
116 main(file_path, stage_num)
787d2eb13c2d Add buggy stageviewer!
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
117