Mercurial > touhou
annotate pytouhou/game/background.py @ 34:4d93d45ecb62
Fix animation script flow handling
author | Thibaut Girka <thib@sitedethib.com> |
---|---|
date | Sun, 14 Aug 2011 13:34:58 +0200 |
parents | afa91be769ae |
children | a10e3f44a883 |
rev | line source |
---|---|
13 | 1 from io import BytesIO |
2 import os | |
3 import struct | |
4 from itertools import chain | |
5 | |
6 from pytouhou.utils.interpolator import Interpolator | |
15 | 7 from pytouhou.game.sprite import Sprite |
13 | 8 |
9 | |
10 class Background(object): | |
21
bf225780973f
Small refactoring, and Rumia \o/
Thibaut Girka <thib@sitedethib.com>
parents:
17
diff
changeset
|
11 def __init__(self, stage, anm_wrapper): |
15 | 12 self.stage = stage |
21
bf225780973f
Small refactoring, and Rumia \o/
Thibaut Girka <thib@sitedethib.com>
parents:
17
diff
changeset
|
13 self.anm_wrapper = anm_wrapper |
13 | 14 self.objects = [] |
15 self.object_instances = [] | |
16
66ce9bb440ac
Refactor in order to support multiple textures
Thibaut Girka <thib@sitedethib.com>
parents:
15
diff
changeset
|
16 self.objects_by_texture = {} |
15 | 17 |
26
f17122405121
Basic sprite animation support
Thibaut Girka <thib@sitedethib.com>
parents:
21
diff
changeset
|
18 self.position_interpolator = Interpolator((0, 0, 0)) |
f17122405121
Basic sprite animation support
Thibaut Girka <thib@sitedethib.com>
parents:
21
diff
changeset
|
19 self.fog_interpolator = Interpolator((0, 0, 0, 0, 0)) |
f17122405121
Basic sprite animation support
Thibaut Girka <thib@sitedethib.com>
parents:
21
diff
changeset
|
20 self.position2_interpolator = Interpolator((0, 0, 0)) |
f17122405121
Basic sprite animation support
Thibaut Girka <thib@sitedethib.com>
parents:
21
diff
changeset
|
21 |
13 | 22 self.build_objects() |
23 self.build_object_instances() | |
24 | |
25 | |
26 def build_object_instances(self): | |
27 self.object_instances = [] | |
28 for obj, ox, oy, oz in self.stage.object_instances: | |
29 obj_id = self.stage.objects.index(obj) | |
30 | |
31 obj_instance = [] | |
32 for face_vertices, face_uvs in self.objects[obj_id]: | |
33 obj_instance.append((tuple((x + ox, y + oy, z + oz) | |
34 for x, y, z in face_vertices), | |
35 face_uvs)) | |
36 self.object_instances.append(obj_instance) | |
37 # Z-sorting | |
38 def keyfunc(obj): | |
39 return min(z for face in obj for x, y, z in face[0]) | |
40 self.object_instances.sort(key=keyfunc, reverse=True) | |
41 | |
42 | |
43 def object_instances_to_vertices_uvs(self): | |
44 vertices = tuple(vertex for obj in self.object_instances | |
45 for face in obj for vertex in face[0]) | |
46 uvs = tuple(uv for obj in self.object_instances | |
47 for face in obj for uv in face[1]) | |
48 return vertices, uvs | |
49 | |
50 | |
51 def build_objects(self): | |
52 self.objects = [] | |
53 for i, obj in enumerate(self.stage.objects): | |
54 faces = [] | |
15 | 55 for script_index, ox, oy, oz, width_override, height_override in obj.quads: |
21
bf225780973f
Small refactoring, and Rumia \o/
Thibaut Girka <thib@sitedethib.com>
parents:
17
diff
changeset
|
56 #TODO: per-texture rendering |
bf225780973f
Small refactoring, and Rumia \o/
Thibaut Girka <thib@sitedethib.com>
parents:
17
diff
changeset
|
57 anm, sprite = self.anm_wrapper.get_sprite(script_index) |
29
afa91be769ae
Don't lose time updating off-screen enemies' sprites
Thibaut Girka <thib@sitedethib.com>
parents:
26
diff
changeset
|
58 if sprite.update(): |
afa91be769ae
Don't lose time updating off-screen enemies' sprites
Thibaut Girka <thib@sitedethib.com>
parents:
26
diff
changeset
|
59 sprite.update_uvs_vertices(width_override, height_override) |
15 | 60 uvs, vertices = sprite._uvs, tuple((x + ox, y + oy, z + oz) for x, y, z in sprite._vertices) |
13 | 61 faces.append((vertices, uvs)) |
62 self.objects.append(faces) | |
63 | |
64 | |
65 def update(self, frame): | |
16
66ce9bb440ac
Refactor in order to support multiple textures
Thibaut Girka <thib@sitedethib.com>
parents:
15
diff
changeset
|
66 if not self.objects_by_texture: |
13 | 67 vertices, uvs = self.object_instances_to_vertices_uvs() |
16
66ce9bb440ac
Refactor in order to support multiple textures
Thibaut Girka <thib@sitedethib.com>
parents:
15
diff
changeset
|
68 nb_vertices = len(vertices) |
66ce9bb440ac
Refactor in order to support multiple textures
Thibaut Girka <thib@sitedethib.com>
parents:
15
diff
changeset
|
69 vertices_format = 'f' * (3 * nb_vertices) |
66ce9bb440ac
Refactor in order to support multiple textures
Thibaut Girka <thib@sitedethib.com>
parents:
15
diff
changeset
|
70 uvs_format = 'f' * (2 * nb_vertices) |
66ce9bb440ac
Refactor in order to support multiple textures
Thibaut Girka <thib@sitedethib.com>
parents:
15
diff
changeset
|
71 vertices = struct.pack(vertices_format, *chain(*vertices)) |
66ce9bb440ac
Refactor in order to support multiple textures
Thibaut Girka <thib@sitedethib.com>
parents:
15
diff
changeset
|
72 uvs = struct.pack(uvs_format, *chain(*uvs)) |
21
bf225780973f
Small refactoring, and Rumia \o/
Thibaut Girka <thib@sitedethib.com>
parents:
17
diff
changeset
|
73 assert len(self.anm_wrapper.anm_files) == 1 #TODO |
bf225780973f
Small refactoring, and Rumia \o/
Thibaut Girka <thib@sitedethib.com>
parents:
17
diff
changeset
|
74 anm = self.anm_wrapper.anm_files[0] |
bf225780973f
Small refactoring, and Rumia \o/
Thibaut Girka <thib@sitedethib.com>
parents:
17
diff
changeset
|
75 self.objects_by_texture = {(anm.first_name, anm.secondary_name): (nb_vertices, vertices, uvs)} |
13 | 76 |
77 for frame_num, message_type, args in self.stage.script: | |
78 if frame_num == frame: | |
14 | 79 if message_type == 0: |
80 self.position_interpolator.set_interpolation_start(frame_num, args) | |
81 elif message_type == 1: | |
13 | 82 self.fog_interpolator.set_interpolation_end_values(args) |
14 | 83 elif message_type == 2: |
84 self.position2_interpolator.set_interpolation_end_values(args) | |
13 | 85 elif message_type == 3: |
86 duration, = args | |
87 self.position2_interpolator.set_interpolation_end_frame(frame_num + duration) | |
88 elif message_type == 4: | |
89 duration, = args | |
90 self.fog_interpolator.set_interpolation_end_frame(frame_num + duration) | |
91 if frame_num > frame and message_type == 0: | |
92 self.position_interpolator.set_interpolation_end(frame_num, args) | |
93 break | |
94 | |
95 self.position2_interpolator.update(frame) | |
96 self.fog_interpolator.update(frame) | |
97 self.position_interpolator.update(frame) | |
98 |