Mercurial > touhou
annotate pytouhou/formats/std.py @ 97:ac2e5e1c2c3c
Refactor \o/
| author | Thibaut Girka <thib@sitedethib.com> |
|---|---|
| date | Sun, 04 Sep 2011 23:50:00 +0200 |
| parents | ca571697ec83 |
| children | 3ac41b966fed |
| rev | line source |
|---|---|
|
52
ab826bc29aa2
Add some documentation, GPLv3 headers, README and COPYING file.
Thibaut Girka <thib@sitedethib.com>
parents:
20
diff
changeset
|
1 # -*- encoding: utf-8 -*- |
|
ab826bc29aa2
Add some documentation, GPLv3 headers, README and COPYING file.
Thibaut Girka <thib@sitedethib.com>
parents:
20
diff
changeset
|
2 ## |
|
ab826bc29aa2
Add some documentation, GPLv3 headers, README and COPYING file.
Thibaut Girka <thib@sitedethib.com>
parents:
20
diff
changeset
|
3 ## Copyright (C) 2011 Thibaut Girka <thib@sitedethib.com> |
|
ab826bc29aa2
Add some documentation, GPLv3 headers, README and COPYING file.
Thibaut Girka <thib@sitedethib.com>
parents:
20
diff
changeset
|
4 ## |
|
ab826bc29aa2
Add some documentation, GPLv3 headers, README and COPYING file.
Thibaut Girka <thib@sitedethib.com>
parents:
20
diff
changeset
|
5 ## This program is free software; you can redistribute it and/or modify |
|
ab826bc29aa2
Add some documentation, GPLv3 headers, README and COPYING file.
Thibaut Girka <thib@sitedethib.com>
parents:
20
diff
changeset
|
6 ## it under the terms of the GNU General Public License as published |
|
ab826bc29aa2
Add some documentation, GPLv3 headers, README and COPYING file.
Thibaut Girka <thib@sitedethib.com>
parents:
20
diff
changeset
|
7 ## by the Free Software Foundation; version 3 only. |
|
ab826bc29aa2
Add some documentation, GPLv3 headers, README and COPYING file.
Thibaut Girka <thib@sitedethib.com>
parents:
20
diff
changeset
|
8 ## |
|
ab826bc29aa2
Add some documentation, GPLv3 headers, README and COPYING file.
Thibaut Girka <thib@sitedethib.com>
parents:
20
diff
changeset
|
9 ## This program is distributed in the hope that it will be useful, |
|
ab826bc29aa2
Add some documentation, GPLv3 headers, README and COPYING file.
Thibaut Girka <thib@sitedethib.com>
parents:
20
diff
changeset
|
10 ## but WITHOUT ANY WARRANTY; without even the implied warranty of |
|
ab826bc29aa2
Add some documentation, GPLv3 headers, README and COPYING file.
Thibaut Girka <thib@sitedethib.com>
parents:
20
diff
changeset
|
11 ## MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
|
ab826bc29aa2
Add some documentation, GPLv3 headers, README and COPYING file.
Thibaut Girka <thib@sitedethib.com>
parents:
20
diff
changeset
|
12 ## GNU General Public License for more details. |
|
ab826bc29aa2
Add some documentation, GPLv3 headers, README and COPYING file.
Thibaut Girka <thib@sitedethib.com>
parents:
20
diff
changeset
|
13 ## |
|
ab826bc29aa2
Add some documentation, GPLv3 headers, README and COPYING file.
Thibaut Girka <thib@sitedethib.com>
parents:
20
diff
changeset
|
14 |
|
ab826bc29aa2
Add some documentation, GPLv3 headers, README and COPYING file.
Thibaut Girka <thib@sitedethib.com>
parents:
20
diff
changeset
|
15 |
| 0 | 16 from struct import pack, unpack |
| 58 | 17 from pytouhou.utils.helpers import read_string, get_logger |
| 0 | 18 |
| 58 | 19 logger = get_logger(__name__) |
| 0 | 20 |
| 21 | |
|
94
ca571697ec83
Various minor optimisations and refactoring
Thibaut Girka <thib@sitedethib.com>
parents:
58
diff
changeset
|
22 class Model(object): |
| 0 | 23 def __init__(self): |
|
94
ca571697ec83
Various minor optimisations and refactoring
Thibaut Girka <thib@sitedethib.com>
parents:
58
diff
changeset
|
24 self.unknown = 0 |
|
ca571697ec83
Various minor optimisations and refactoring
Thibaut Girka <thib@sitedethib.com>
parents:
58
diff
changeset
|
25 self.bounding_box = (0., 0., 0., |
|
ca571697ec83
Various minor optimisations and refactoring
Thibaut Girka <thib@sitedethib.com>
parents:
58
diff
changeset
|
26 0., 0., 0.) |
| 0 | 27 self.quads = [] |
| 28 | |
| 29 | |
| 30 | |
| 31 class Stage(object): | |
| 97 | 32 def __init__(self): |
| 0 | 33 self.name = '' |
| 34 self.bgms = (('', ''), ('', ''), ('', '')) | |
|
94
ca571697ec83
Various minor optimisations and refactoring
Thibaut Girka <thib@sitedethib.com>
parents:
58
diff
changeset
|
35 self.models = [] |
| 0 | 36 self.object_instances = [] |
| 37 self.script = [] | |
| 38 | |
| 39 | |
| 40 @classmethod | |
| 97 | 41 def read(cls, file): |
| 42 stage = Stage() | |
| 0 | 43 |
|
94
ca571697ec83
Various minor optimisations and refactoring
Thibaut Girka <thib@sitedethib.com>
parents:
58
diff
changeset
|
44 nb_models, nb_faces = unpack('<HH', file.read(4)) |
| 0 | 45 object_instances_offset, script_offset = unpack('<II', file.read(8)) |
| 46 if file.read(4) != b'\x00\x00\x00\x00': | |
| 47 raise Exception #TODO | |
| 48 | |
| 49 stage.name = read_string(file, 128, 'shift-jis') | |
| 50 | |
| 51 bgm_a = read_string(file, 128, 'shift-jis') | |
| 52 bgm_b = read_string(file, 128, 'shift-jis') | |
| 53 bgm_c = read_string(file, 128, 'shift-jis') | |
| 54 bgm_d = read_string(file, 128, 'shift-jis') | |
| 55 | |
| 56 bgm_a_path = read_string(file, 128, 'ascii') | |
| 57 bgm_b_path = read_string(file, 128, 'ascii') | |
| 58 bgm_c_path = read_string(file, 128, 'ascii') | |
| 59 bgm_d_path = read_string(file, 128, 'ascii') | |
| 60 | |
| 61 stage.bgms = [(bgm_a, bgm_a_path), (bgm_b, bgm_b_path), (bgm_c, bgm_c_path), (bgm_d, bgm_d_path)] #TODO: handle ' ' | |
| 62 | |
|
94
ca571697ec83
Various minor optimisations and refactoring
Thibaut Girka <thib@sitedethib.com>
parents:
58
diff
changeset
|
63 # Read model definitions |
|
ca571697ec83
Various minor optimisations and refactoring
Thibaut Girka <thib@sitedethib.com>
parents:
58
diff
changeset
|
64 offsets = unpack('<%s' % ('I' * nb_models), file.read(4 * nb_models)) |
| 0 | 65 for offset in offsets: |
|
94
ca571697ec83
Various minor optimisations and refactoring
Thibaut Girka <thib@sitedethib.com>
parents:
58
diff
changeset
|
66 model = Model() |
|
ca571697ec83
Various minor optimisations and refactoring
Thibaut Girka <thib@sitedethib.com>
parents:
58
diff
changeset
|
67 id_, unknown, x, y, z, width, height, depth = unpack('<HHffffff', file.read(28)) |
|
ca571697ec83
Various minor optimisations and refactoring
Thibaut Girka <thib@sitedethib.com>
parents:
58
diff
changeset
|
68 model.unknown = unknown |
|
ca571697ec83
Various minor optimisations and refactoring
Thibaut Girka <thib@sitedethib.com>
parents:
58
diff
changeset
|
69 model.bounding_box = x, y, z, width, height, depth #TODO: check |
| 0 | 70 while True: |
| 71 unknown, size = unpack('<HH', file.read(4)) | |
| 72 if unknown == 0xffff: | |
| 73 break | |
| 74 if size != 0x1c: | |
| 75 raise Exception #TODO | |
| 76 script_index, _padding, x, y, z, width, height = unpack('<HHfffff', file.read(24)) | |
| 77 #TODO: store script_index, x, y, z, width and height | |
|
94
ca571697ec83
Various minor optimisations and refactoring
Thibaut Girka <thib@sitedethib.com>
parents:
58
diff
changeset
|
78 model.quads.append((script_index, x, y, z, width, height)) |
|
ca571697ec83
Various minor optimisations and refactoring
Thibaut Girka <thib@sitedethib.com>
parents:
58
diff
changeset
|
79 stage.models.append(model) |
| 0 | 80 |
| 81 | |
| 82 # Read object usages | |
| 83 file.seek(object_instances_offset) | |
| 84 while True: | |
| 85 obj_id, unknown, x, y, z = unpack('<HHfff', file.read(16)) | |
| 86 if (obj_id, unknown) == (0xffff, 0xffff): | |
| 87 break | |
| 88 if unknown != 256: | |
| 89 raise Exception #TODO | |
|
94
ca571697ec83
Various minor optimisations and refactoring
Thibaut Girka <thib@sitedethib.com>
parents:
58
diff
changeset
|
90 stage.object_instances.append((obj_id, x, y, z)) |
| 0 | 91 |
| 92 | |
| 93 # Read other funny things (script) | |
| 94 file.seek(script_offset) | |
| 95 while True: | |
| 96 frame, message_type, size = unpack('<IHH', file.read(8)) | |
| 97 if (frame, message_type, size) == (0xffffffff, 0xffff, 0xffff): | |
| 98 break | |
| 99 if size != 0x0c: | |
| 100 raise Exception #TODO | |
| 101 data = file.read(12) | |
| 13 | 102 #TODO: maybe add a name somewhere |
| 103 if message_type == 0: # ViewPos | |
| 104 args = unpack('<fff', data) | |
| 105 elif message_type == 1: # Color | |
| 106 args = unpack('<BBBBff', data) | |
| 107 elif message_type == 2: # ViewPos2 | |
|
20
6ebf9539c077
Handle more enemies types and movements
Thibaut Girka <thib@sitedethib.com>
parents:
15
diff
changeset
|
108 args = unpack('<fff', data) |
| 13 | 109 elif message_type == 3: # StartInterpolatingViewPos2 |
| 110 args = tuple(unpack('<III', data)[:1]) | |
| 111 elif message_type == 4: # StartInterpolatingFog | |
| 112 args = tuple(unpack('<III', data)[:1]) | |
| 113 else: | |
| 114 args = (data,) | |
| 58 | 115 logger.warn('unknown opcode %d (data: %r)', message_type, data) |
| 13 | 116 stage.script.append((frame, message_type, args)) |
| 0 | 117 |
| 118 return stage | |
| 119 |