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