Mercurial > touhou
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 |
rev | line source |
---|---|
4 | 1 #!/usr/bin/env python |
2 | |
3 import sys | |
4 import os | |
5 | |
6 import struct | |
7 from math import degrees, radians | |
8 from io import BytesIO | |
9 from itertools import chain | |
10 | |
11 import pygame | |
12 | |
13 | 13 from pytouhou.formats.pbg3 import PBG3 |
14 from pytouhou.game.background import Background | |
14 | 15 from pytouhou.opengl.texture import load_texture |
13 | 16 |
4 | 17 import OpenGL |
18 OpenGL.FORWARD_COMPATIBLE_ONLY = True | |
19 from OpenGL.GL import * | |
20 from OpenGL.GLU import * | |
21 | |
22 | |
23 def main(path, stage_num): | |
24 # Initialize pygame | |
25 pygame.init() | |
26 window = pygame.display.set_mode((384, 448), pygame.OPENGL | pygame.DOUBLEBUF) | |
27 | |
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 | 33 glEnable(GL_DEPTH_TEST) |
34 glEnable(GL_BLEND) | |
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 | 37 glHint(GL_FOG_HINT, GL_NICEST) |
38 glHint(GL_PERSPECTIVE_CORRECTION_HINT, GL_NICEST) | |
39 glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_MAG_FILTER, GL_LINEAR) | |
40 glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_MIN_FILTER, GL_LINEAR) | |
41 glBlendFunc(GL_SRC_ALPHA, GL_ONE_MINUS_SRC_ALPHA) | |
4 | 42 glEnableClientState(GL_VERTEX_ARRAY) |
43 glEnableClientState(GL_TEXTURE_COORD_ARRAY) | |
44 | |
45 # Load data | |
46 with open(path, 'rb') as file: | |
47 archive = PBG3.read(file) | |
13 | 48 background = Background(archive, stage_num) |
14 | 49 texture = load_texture(archive, background.anim) |
4 | 50 |
13 | 51 print(background.stage.name) |
8 | 52 |
4 | 53 frame = 0 |
54 | |
55 # Main loop | |
56 clock = pygame.time.Clock() | |
57 while True: | |
13 | 58 # Check events |
4 | 59 for event in pygame.event.get(): |
60 if event.type == pygame.QUIT or (event.type == pygame.KEYDOWN and event.key in (pygame.K_ESCAPE, pygame.K_q)): | |
61 sys.exit(0) | |
62 elif event.type == pygame.KEYDOWN: | |
63 if event.key == pygame.K_RETURN and event.mod & pygame.KMOD_ALT: | |
64 pygame.display.toggle_fullscreen() | |
65 | |
13 | 66 # Update game |
67 background.update(frame) | |
68 | |
69 # Draw everything | |
4 | 70 glClear(GL_COLOR_BUFFER_BIT | GL_DEPTH_BUFFER_BIT) |
71 | |
13 | 72 glVertexPointer(3, GL_FLOAT, 0, background._vertices) |
73 glTexCoordPointer(2, GL_FLOAT, 0, background._uvs) | |
4 | 74 |
13 | 75 fog_b, fog_g, fog_r, _, fog_start, fog_end = background.fog_interpolator.values |
76 x, y, z = background.position_interpolator.values | |
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 | 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 | 84 #TODO |
4 | 85 glMatrixMode(GL_MODELVIEW) |
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 | 96 |
13 | 97 glDrawArrays(GL_QUADS, 0, background.nb_vertices) |
4 | 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 | 104 pygame.display.flip() |
105 clock.tick(120) | |
106 frame += 1 | |
107 | |
108 | |
109 | |
110 try: | |
111 file_path, stage_num = sys.argv[1:] | |
112 stage_num = int(stage_num) | |
113 except ValueError: | |
114 print('Usage: %s std_dat_path stage_num' % sys.argv[0]) | |
115 else: | |
116 main(file_path, stage_num) | |
117 |