Mercurial > touhou
annotate pytouhou/formats/std.py @ 52:ab826bc29aa2
Add some documentation, GPLv3 headers, README and COPYING file.
| author | Thibaut Girka <thib@sitedethib.com> |
|---|---|
| date | Mon, 22 Aug 2011 22:37:14 +0200 |
| parents | 6ebf9539c077 |
| children | 3da4de9decd0 |
| 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 |
| 17 from pytouhou.utils.helpers import read_string | |
| 18 | |
| 19 | |
| 20 | |
| 21 class Object(object): | |
| 22 def __init__(self): | |
| 23 self.header = (b'\x00') * 28 #TODO | |
| 24 self.quads = [] | |
| 25 | |
| 26 | |
| 27 | |
| 28 class Stage(object): | |
| 15 | 29 def __init__(self, num): |
| 30 self.num = num | |
| 0 | 31 self.name = '' |
| 32 self.bgms = (('', ''), ('', ''), ('', '')) | |
| 33 self.objects = [] | |
| 34 self.object_instances = [] | |
| 35 self.script = [] | |
| 36 | |
| 37 | |
| 38 @classmethod | |
| 15 | 39 def read(cls, file, num): |
| 40 stage = Stage(num) | |
| 0 | 41 |
| 42 nb_objects, nb_faces = unpack('<HH', file.read(4)) | |
| 43 object_instances_offset, script_offset = unpack('<II', file.read(8)) | |
| 44 if file.read(4) != b'\x00\x00\x00\x00': | |
| 45 raise Exception #TODO | |
| 46 | |
| 47 stage.name = read_string(file, 128, 'shift-jis') | |
| 48 | |
| 49 bgm_a = read_string(file, 128, 'shift-jis') | |
| 50 bgm_b = read_string(file, 128, 'shift-jis') | |
| 51 bgm_c = read_string(file, 128, 'shift-jis') | |
| 52 bgm_d = read_string(file, 128, 'shift-jis') | |
| 53 | |
| 54 bgm_a_path = read_string(file, 128, 'ascii') | |
| 55 bgm_b_path = read_string(file, 128, 'ascii') | |
| 56 bgm_c_path = read_string(file, 128, 'ascii') | |
| 57 bgm_d_path = read_string(file, 128, 'ascii') | |
| 58 | |
| 59 stage.bgms = [(bgm_a, bgm_a_path), (bgm_b, bgm_b_path), (bgm_c, bgm_c_path), (bgm_d, bgm_d_path)] #TODO: handle ' ' | |
| 60 | |
| 61 # Read object definitions | |
| 62 offsets = unpack('<%s' % ('I' * nb_objects), file.read(4 * nb_objects)) | |
| 63 for offset in offsets: | |
| 64 obj = Object() | |
| 65 obj.header = file.read(28) #TODO: this has to be reversed! | |
| 66 while True: | |
| 67 unknown, size = unpack('<HH', file.read(4)) | |
| 68 if unknown == 0xffff: | |
| 69 break | |
| 70 if size != 0x1c: | |
| 71 raise Exception #TODO | |
| 72 script_index, _padding, x, y, z, width, height = unpack('<HHfffff', file.read(24)) | |
| 73 #TODO: store script_index, x, y, z, width and height | |
| 74 obj.quads.append((script_index, x, y, z, width, height)) | |
| 75 stage.objects.append(obj) | |
| 76 | |
| 77 | |
| 78 # Read object usages | |
| 79 file.seek(object_instances_offset) | |
| 80 while True: | |
| 81 obj_id, unknown, x, y, z = unpack('<HHfff', file.read(16)) | |
| 82 if (obj_id, unknown) == (0xffff, 0xffff): | |
| 83 break | |
| 84 if unknown != 256: | |
| 85 raise Exception #TODO | |
| 86 stage.object_instances.append((stage.objects[obj_id], x, y, z)) | |
| 87 | |
| 88 | |
| 89 # Read other funny things (script) | |
| 90 file.seek(script_offset) | |
| 91 while True: | |
| 92 frame, message_type, size = unpack('<IHH', file.read(8)) | |
| 93 if (frame, message_type, size) == (0xffffffff, 0xffff, 0xffff): | |
| 94 break | |
| 95 if size != 0x0c: | |
| 96 raise Exception #TODO | |
| 97 data = file.read(12) | |
| 13 | 98 #TODO: maybe add a name somewhere |
| 99 if message_type == 0: # ViewPos | |
| 100 args = unpack('<fff', data) | |
| 101 elif message_type == 1: # Color | |
| 102 args = unpack('<BBBBff', data) | |
| 103 elif message_type == 2: # ViewPos2 | |
|
20
6ebf9539c077
Handle more enemies types and movements
Thibaut Girka <thib@sitedethib.com>
parents:
15
diff
changeset
|
104 args = unpack('<fff', data) |
| 13 | 105 elif message_type == 3: # StartInterpolatingViewPos2 |
| 106 args = tuple(unpack('<III', data)[:1]) | |
| 107 elif message_type == 4: # StartInterpolatingFog | |
| 108 args = tuple(unpack('<III', data)[:1]) | |
| 109 else: | |
| 110 args = (data,) | |
| 111 print('Warning: unknown opcode %d' % message_type) #TODO | |
| 112 stage.script.append((frame, message_type, args)) | |
| 0 | 113 |
| 114 return stage | |
| 115 |