Mercurial > touhou

annotate pytouhou/formats/ecl.py @ 166:dcf32488a2c9

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Better enemy death, with animation and (hopefully) correct flags handling.
author Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
date Tue, 18 Oct 2011 13:17:23 -0700
parents c8c60291c56f
children 2f3665a77f11
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
52
ab826bc29aa2 Add some documentation, GPLv3 headers, README and COPYING file.
Thibaut Girka <thib@sitedethib.com>
parents: 51
diff changeset
1 # -*- encoding: utf-8 -*-
ab826bc29aa2 Add some documentation, GPLv3 headers, README and COPYING file.
Thibaut Girka <thib@sitedethib.com>
parents: 51
diff changeset
2 ##
ab826bc29aa2 Add some documentation, GPLv3 headers, README and COPYING file.
Thibaut Girka <thib@sitedethib.com>
parents: 51
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: 51
diff changeset
4 ##
ab826bc29aa2 Add some documentation, GPLv3 headers, README and COPYING file.
Thibaut Girka <thib@sitedethib.com>
parents: 51
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: 51
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: 51
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: 51
diff changeset
8 ##
ab826bc29aa2 Add some documentation, GPLv3 headers, README and COPYING file.
Thibaut Girka <thib@sitedethib.com>
parents: 51
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: 51
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: 51
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: 51
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: 51
diff changeset
13 ##
ab826bc29aa2 Add some documentation, GPLv3 headers, README and COPYING file.
Thibaut Girka <thib@sitedethib.com>
parents: 51
diff changeset
14
113
732c64662f87 Minor changes
Thibaut Girka <thib@sitedethib.com>
parents: 112
diff changeset
15 import struct
95
e2d8f2a56ea4 Handle ECL opcodes with string.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents: 87
diff changeset
16 from struct import pack, unpack, calcsize
18
ca26a84916cb Add preliminary ECL viewer/interpreter.
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
17
58
3da4de9decd0 Use logging module
Thibaut Girka <thib@sitedethib.com>
parents: 52
diff changeset
18 from pytouhou.utils.helpers import get_logger
18
ca26a84916cb Add preliminary ECL viewer/interpreter.
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
19
58
3da4de9decd0 Use logging module
Thibaut Girka <thib@sitedethib.com>
parents: 52
diff changeset
20 logger = get_logger(__name__)
18
ca26a84916cb Add preliminary ECL viewer/interpreter.
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
21
ca26a84916cb Add preliminary ECL viewer/interpreter.
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
22 class ECL(object):
42
1b0ca2fb89f9 Refactor ECL parsing/etc.
Thibaut Girka <thib@sitedethib.com>
parents: 20
diff changeset
23 _instructions = {0: ('', 'noop?'),
1b0ca2fb89f9 Refactor ECL parsing/etc.
Thibaut Girka <thib@sitedethib.com>
parents: 20
diff changeset
24 1: ('I', 'delete?'),
1b0ca2fb89f9 Refactor ECL parsing/etc.
Thibaut Girka <thib@sitedethib.com>
parents: 20
diff changeset
25 2: ('Ii', 'relative_jump'),
1b0ca2fb89f9 Refactor ECL parsing/etc.
Thibaut Girka <thib@sitedethib.com>
parents: 20
diff changeset
26 3: ('Iii', 'relative_jump_ex'),
46
25ca15f714ad Add/fix a few opcodes
Thibaut Girka <thib@sitedethib.com>
parents: 43
diff changeset
27 4: ('ii', 'set_int'),
25ca15f714ad Add/fix a few opcodes
Thibaut Girka <thib@sitedethib.com>
parents: 43
diff changeset
28 5: ('if', 'set_float'),
25ca15f714ad Add/fix a few opcodes
Thibaut Girka <thib@sitedethib.com>
parents: 43
diff changeset
29 6: ('ii', 'set_random_int'),
25ca15f714ad Add/fix a few opcodes
Thibaut Girka <thib@sitedethib.com>
parents: 43
diff changeset
30 8: ('if', 'set_random_float'),
59
4fe37a620b22 Handle set_random_angle properly! At last!
Thibaut Girka <thib@sitedethib.com>
parents: 58
diff changeset
31 9: ('iff', 'set_random_float2'),
75
b3bd421bb895 Handle a few more ECL instructions
Thibaut Girka <thib@sitedethib.com>
parents: 70
diff changeset
32 10: ('i', 'store_x'),
87
fda176f07d6d Fix add_int
Thibaut Girka <thib@sitedethib.com>
parents: 79
diff changeset
33 13: ('iii', 'add_int'),
63
8527fe640844 Implement simple arithmetic instructions.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents: 59
diff changeset
34 14: ('iii', 'substract_int'),
8527fe640844 Implement simple arithmetic instructions.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents: 59
diff changeset
35 15: ('iii', 'multiply_int'),
8527fe640844 Implement simple arithmetic instructions.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents: 59
diff changeset
36 16: ('iii', 'divide_int'),
8527fe640844 Implement simple arithmetic instructions.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents: 59
diff changeset
37 17: ('iii', 'modulo'),
96
54929d495654 Handle ECL instruction 18
Thibaut Girka <thib@sitedethib.com>
parents: 95
diff changeset
38 18: ('i', 'increment'),
63
8527fe640844 Implement simple arithmetic instructions.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents: 59
diff changeset
39 20: ('iff', 'add_float'),
8527fe640844 Implement simple arithmetic instructions.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents: 59
diff changeset
40 21: ('iff', 'substract_float'),
8527fe640844 Implement simple arithmetic instructions.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents: 59
diff changeset
41 23: ('iff', 'divide_float'),
77
6fa6d74a049a Handle a new ECL instruction, and add some names.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents: 76
diff changeset
42 25: ('iffff', 'get_direction'),
42
1b0ca2fb89f9 Refactor ECL parsing/etc.
Thibaut Girka <thib@sitedethib.com>
parents: 20
diff changeset
43 26: ('i', None),
46
25ca15f714ad Add/fix a few opcodes
Thibaut Girka <thib@sitedethib.com>
parents: 43
diff changeset
44 27: ('ii', 'compare_ints'),
63
8527fe640844 Implement simple arithmetic instructions.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents: 59
diff changeset
45 28: ('ff', 'compare_floats'),
64
d469012368b3 Implement conditional jumps.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents: 63
diff changeset
46 29: ('ii', 'relative_jump_if_lower_than'),
d469012368b3 Implement conditional jumps.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents: 63
diff changeset
47 30: ('ii', 'relative_jump_if_lower_or_equal'),
46
25ca15f714ad Add/fix a few opcodes
Thibaut Girka <thib@sitedethib.com>
parents: 43
diff changeset
48 31: ('ii', 'relative_jump_if_equal'),
64
d469012368b3 Implement conditional jumps.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents: 63
diff changeset
49 32: ('ii', 'relative_jump_if_greater_than'),
d469012368b3 Implement conditional jumps.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents: 63
diff changeset
50 33: ('ii', 'relative_jump_if_greater_or_equal'),
d469012368b3 Implement conditional jumps.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents: 63
diff changeset
51 34: ('ii', 'relative_jump_if_not_equal'),
46
25ca15f714ad Add/fix a few opcodes
Thibaut Girka <thib@sitedethib.com>
parents: 43
diff changeset
52 35: ('iif', 'call'),
42
1b0ca2fb89f9 Refactor ECL parsing/etc.
Thibaut Girka <thib@sitedethib.com>
parents: 20
diff changeset
53 36: ('', 'return?'),
46
25ca15f714ad Add/fix a few opcodes
Thibaut Girka <thib@sitedethib.com>
parents: 43
diff changeset
54 39: ('iifii', 'call_if_equal'),
42
1b0ca2fb89f9 Refactor ECL parsing/etc.
Thibaut Girka <thib@sitedethib.com>
parents: 20
diff changeset
55 43: ('fff', 'set_position'),
1b0ca2fb89f9 Refactor ECL parsing/etc.
Thibaut Girka <thib@sitedethib.com>
parents: 20
diff changeset
56 45: ('ff', 'set_angle_and_speed'),
1b0ca2fb89f9 Refactor ECL parsing/etc.
Thibaut Girka <thib@sitedethib.com>
parents: 20
diff changeset
57 46: ('f', 'set_rotation_speed'),
1b0ca2fb89f9 Refactor ECL parsing/etc.
Thibaut Girka <thib@sitedethib.com>
parents: 20
diff changeset
58 47: ('f', 'set_speed'),
1b0ca2fb89f9 Refactor ECL parsing/etc.
Thibaut Girka <thib@sitedethib.com>
parents: 20
diff changeset
59 48: ('f', 'set_acceleration'),
70
7c1f20407b3e Add set_random_angle support
Thibaut Girka <thib@sitedethib.com>
parents: 67
diff changeset
60 49: ('ff', 'set_random_angle'),
7c1f20407b3e Add set_random_angle support
Thibaut Girka <thib@sitedethib.com>
parents: 67
diff changeset
61 50: ('ff', 'set_random_angle_ex'),
42
1b0ca2fb89f9 Refactor ECL parsing/etc.
Thibaut Girka <thib@sitedethib.com>
parents: 20
diff changeset
62 51: ('ff', 'set_speed_towards_player'),
77
6fa6d74a049a Handle a new ECL instruction, and add some names.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents: 76
diff changeset
63 52: ('iff', 'move_in_decel'),
76
f305c0e406d6 Handle all move_to_* ECL instructions.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents: 75
diff changeset
64 56: ('ifff', 'move_to_linear'),
f305c0e406d6 Handle all move_to_* ECL instructions.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents: 75
diff changeset
65 57: ('ifff', 'move_to_decel'),
f305c0e406d6 Handle all move_to_* ECL instructions.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents: 75
diff changeset
66 59: ('iffi', 'move_to_accel'),
46
25ca15f714ad Add/fix a few opcodes
Thibaut Girka <thib@sitedethib.com>
parents: 43
diff changeset
67 61: ('i', 'stop_in'),
75
b3bd421bb895 Handle a few more ECL instructions
Thibaut Girka <thib@sitedethib.com>
parents: 70
diff changeset
68 63: ('i', 'stop_in_accel'),
50
811cefefb5c8 Fix a few bugs and add support for a few instructions
Thibaut Girka <thib@sitedethib.com>
parents: 46
diff changeset
69 65: ('ffff', 'set_screen_box'),
811cefefb5c8 Fix a few bugs and add support for a few instructions
Thibaut Girka <thib@sitedethib.com>
parents: 46
diff changeset
70 66: ('', 'clear_screen_box'),
42
1b0ca2fb89f9 Refactor ECL parsing/etc.
Thibaut Girka <thib@sitedethib.com>
parents: 20
diff changeset
71 67: ('hhiiffffi', 'set_bullet_attributes'),
1b0ca2fb89f9 Refactor ECL parsing/etc.
Thibaut Girka <thib@sitedethib.com>
parents: 20
diff changeset
72 68: ('hhiiffffi', 'set_bullet_attributes2'),
1b0ca2fb89f9 Refactor ECL parsing/etc.
Thibaut Girka <thib@sitedethib.com>
parents: 20
diff changeset
73 69: ('hhiiffffi', 'set_bullet_attributes3'),
1b0ca2fb89f9 Refactor ECL parsing/etc.
Thibaut Girka <thib@sitedethib.com>
parents: 20
diff changeset
74 70: ('hhiiffffi', 'set_bullet_attributes4'),
46
25ca15f714ad Add/fix a few opcodes
Thibaut Girka <thib@sitedethib.com>
parents: 43
diff changeset
75 71: ('hhiiffffi', 'set_bullet_attributes5'),
25ca15f714ad Add/fix a few opcodes
Thibaut Girka <thib@sitedethib.com>
parents: 43
diff changeset
76 74: ('hhiiffffi', 'set_bullet_attributes6'),
25ca15f714ad Add/fix a few opcodes
Thibaut Girka <thib@sitedethib.com>
parents: 43
diff changeset
77 75: ('hhiiffffi', 'set_bullet_attributes7'),
78
bcf965ede96c Fix/rename/comment a few things
Thibaut Girka <thib@sitedethib.com>
parents: 77
diff changeset
78 76: ('i', 'set_bullet_interval'),
bcf965ede96c Fix/rename/comment a few things
Thibaut Girka <thib@sitedethib.com>
parents: 77
diff changeset
79 77: ('i', 'set_bullet_interval_ex'),
46
25ca15f714ad Add/fix a few opcodes
Thibaut Girka <thib@sitedethib.com>
parents: 43
diff changeset
80 78: ('', 'delay_attack'),
25ca15f714ad Add/fix a few opcodes
Thibaut Girka <thib@sitedethib.com>
parents: 43
diff changeset
81 79: ('', 'no_delay_attack'),
42
1b0ca2fb89f9 Refactor ECL parsing/etc.
Thibaut Girka <thib@sitedethib.com>
parents: 20
diff changeset
82 81: ('fff', 'set_bullet_launch_offset'),
78
bcf965ede96c Fix/rename/comment a few things
Thibaut Girka <thib@sitedethib.com>
parents: 77
diff changeset
83 82: ('iiiiffff', 'set_extended_bullet_attributes'),
42
1b0ca2fb89f9 Refactor ECL parsing/etc.
Thibaut Girka <thib@sitedethib.com>
parents: 20
diff changeset
84 83: ('', None),
1b0ca2fb89f9 Refactor ECL parsing/etc.
Thibaut Girka <thib@sitedethib.com>
parents: 20
diff changeset
85 84: ('i', None),
46
25ca15f714ad Add/fix a few opcodes
Thibaut Girka <thib@sitedethib.com>
parents: 43
diff changeset
86 85: ('hhffffffiiiiii', 'laser'),
25ca15f714ad Add/fix a few opcodes
Thibaut Girka <thib@sitedethib.com>
parents: 43
diff changeset
87 86: ('hhffffffiiiiii', 'laser2'),
25ca15f714ad Add/fix a few opcodes
Thibaut Girka <thib@sitedethib.com>
parents: 43
diff changeset
88 87: ('i', 'set_upcoming_id'),
25ca15f714ad Add/fix a few opcodes
Thibaut Girka <thib@sitedethib.com>
parents: 43
diff changeset
89 88: ('if','alter_laser_angle'),
42
1b0ca2fb89f9 Refactor ECL parsing/etc.
Thibaut Girka <thib@sitedethib.com>
parents: 20
diff changeset
90 90: ('iiii', None),
1b0ca2fb89f9 Refactor ECL parsing/etc.
Thibaut Girka <thib@sitedethib.com>
parents: 20
diff changeset
91 92: ('i', None),
95
e2d8f2a56ea4 Handle ECL opcodes with string.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents: 87
diff changeset
92 93: ('hhs', 'set_spellcard'),
67
e2cb9d434dc2 Add support for a few opcodes.
Thibaut Girka <thib@sitedethib.com>
parents: 66
diff changeset
93 94: ('', 'end_spellcard'),
42
1b0ca2fb89f9 Refactor ECL parsing/etc.
Thibaut Girka <thib@sitedethib.com>
parents: 20
diff changeset
94 95: ('ifffhhi', None),
1b0ca2fb89f9 Refactor ECL parsing/etc.
Thibaut Girka <thib@sitedethib.com>
parents: 20
diff changeset
95 96: ('', None),
1b0ca2fb89f9 Refactor ECL parsing/etc.
Thibaut Girka <thib@sitedethib.com>
parents: 20
diff changeset
96 97: ('i', 'set_anim'),
46
25ca15f714ad Add/fix a few opcodes
Thibaut Girka <thib@sitedethib.com>
parents: 43
diff changeset
97 98: ('hhhhhxx', 'set_multiple_anims'),
42
1b0ca2fb89f9 Refactor ECL parsing/etc.
Thibaut Girka <thib@sitedethib.com>
parents: 20
diff changeset
98 99: ('ii', None),
1b0ca2fb89f9 Refactor ECL parsing/etc.
Thibaut Girka <thib@sitedethib.com>
parents: 20
diff changeset
99 100: ('i', 'set_death_anim'),
1b0ca2fb89f9 Refactor ECL parsing/etc.
Thibaut Girka <thib@sitedethib.com>
parents: 20
diff changeset
100 101: ('i', 'set_boss_mode?'),
77
6fa6d74a049a Handle a new ECL instruction, and add some names.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents: 76
diff changeset
101 102: ('iffff', 'create_squares'),
42
1b0ca2fb89f9 Refactor ECL parsing/etc.
Thibaut Girka <thib@sitedethib.com>
parents: 20
diff changeset
102 103: ('fff', 'set_enemy_hitbox'),
79
ffe2c2b9912c Handle a few more ECL instructions. Prepare for bullet handling \o/
Thibaut Girka <thib@sitedethib.com>
parents: 78
diff changeset
103 104: ('i', 'set_collidable'),
67
e2cb9d434dc2 Add support for a few opcodes.
Thibaut Girka <thib@sitedethib.com>
parents: 66
diff changeset
104 105: ('i', 'set_damageable'),
42
1b0ca2fb89f9 Refactor ECL parsing/etc.
Thibaut Girka <thib@sitedethib.com>
parents: 20
diff changeset
105 106: ('i', 'play_sound'),
75
b3bd421bb895 Handle a few more ECL instructions
Thibaut Girka <thib@sitedethib.com>
parents: 70
diff changeset
106 107: ('i', 'set_death_flags'),
42
1b0ca2fb89f9 Refactor ECL parsing/etc.
Thibaut Girka <thib@sitedethib.com>
parents: 20
diff changeset
107 108: ('i', 'set_death_callback?'),
46
25ca15f714ad Add/fix a few opcodes
Thibaut Girka <thib@sitedethib.com>
parents: 43
diff changeset
108 109: ('ii', 'memory_write_int'),
25ca15f714ad Add/fix a few opcodes
Thibaut Girka <thib@sitedethib.com>
parents: 43
diff changeset
109 111: ('i', 'set_life'),
67
e2cb9d434dc2 Add support for a few opcodes.
Thibaut Girka <thib@sitedethib.com>
parents: 66
diff changeset
110 112: ('i', 'set_ellapsed_time'),
42
1b0ca2fb89f9 Refactor ECL parsing/etc.
Thibaut Girka <thib@sitedethib.com>
parents: 20
diff changeset
111 113: ('i', 'set_low_life_trigger'),
1b0ca2fb89f9 Refactor ECL parsing/etc.
Thibaut Girka <thib@sitedethib.com>
parents: 20
diff changeset
112 114: ('i', 'set_low_life_callback'),
1b0ca2fb89f9 Refactor ECL parsing/etc.
Thibaut Girka <thib@sitedethib.com>
parents: 20
diff changeset
113 115: ('i', 'set_timeout'),
67
e2cb9d434dc2 Add support for a few opcodes.
Thibaut Girka <thib@sitedethib.com>
parents: 66
diff changeset
114 116: ('i', 'set_timeout_callback'),
e2cb9d434dc2 Add support for a few opcodes.
Thibaut Girka <thib@sitedethib.com>
parents: 66
diff changeset
115 117: ('i', 'set_touchable'),
42
1b0ca2fb89f9 Refactor ECL parsing/etc.
Thibaut Girka <thib@sitedethib.com>
parents: 20
diff changeset
116 118: ('iihh', None),
46
25ca15f714ad Add/fix a few opcodes
Thibaut Girka <thib@sitedethib.com>
parents: 43
diff changeset
117 119: ('i', 'drop_bonus'),
75
b3bd421bb895 Handle a few more ECL instructions
Thibaut Girka <thib@sitedethib.com>
parents: 70
diff changeset
118 120: ('i', 'set_automatic_orientation'),
107
5d9052b9a4e8 (almost) implement Cirno's freezing spellcard
Thibaut Girka <thib@sitedethib.com>
parents: 96
diff changeset
119 121: ('ii', 'call_special_function'),
42
1b0ca2fb89f9 Refactor ECL parsing/etc.
Thibaut Girka <thib@sitedethib.com>
parents: 20
diff changeset
120 122: ('i', None),
75
b3bd421bb895 Handle a few more ECL instructions
Thibaut Girka <thib@sitedethib.com>
parents: 70
diff changeset
121 123: ('i', 'skip_frames'),
77
6fa6d74a049a Handle a new ECL instruction, and add some names.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents: 76
diff changeset
122 124: ('i', 'drop_specific_bonus'),
42
1b0ca2fb89f9 Refactor ECL parsing/etc.
Thibaut Girka <thib@sitedethib.com>
parents: 20
diff changeset
123 125: ('', None),
1b0ca2fb89f9 Refactor ECL parsing/etc.
Thibaut Girka <thib@sitedethib.com>
parents: 20
diff changeset
124 126: ('i', 'set_remaining_lives'),
1b0ca2fb89f9 Refactor ECL parsing/etc.
Thibaut Girka <thib@sitedethib.com>
parents: 20
diff changeset
125 127: ('i', None),
1b0ca2fb89f9 Refactor ECL parsing/etc.
Thibaut Girka <thib@sitedethib.com>
parents: 20
diff changeset
126 128: ('i', None),
1b0ca2fb89f9 Refactor ECL parsing/etc.
Thibaut Girka <thib@sitedethib.com>
parents: 20
diff changeset
127 129: ('ii', None),
1b0ca2fb89f9 Refactor ECL parsing/etc.
Thibaut Girka <thib@sitedethib.com>
parents: 20
diff changeset
128 130: ('i', None),
1b0ca2fb89f9 Refactor ECL parsing/etc.
Thibaut Girka <thib@sitedethib.com>
parents: 20
diff changeset
129 131: ('ffiiii', None),
1b0ca2fb89f9 Refactor ECL parsing/etc.
Thibaut Girka <thib@sitedethib.com>
parents: 20
diff changeset
130 132: ('i', None),
1b0ca2fb89f9 Refactor ECL parsing/etc.
Thibaut Girka <thib@sitedethib.com>
parents: 20
diff changeset
131 133: ('', None),
1b0ca2fb89f9 Refactor ECL parsing/etc.
Thibaut Girka <thib@sitedethib.com>
parents: 20
diff changeset
132 134: ('', None),
1b0ca2fb89f9 Refactor ECL parsing/etc.
Thibaut Girka <thib@sitedethib.com>
parents: 20
diff changeset
133 135: ('i', None)} #TODO
1b0ca2fb89f9 Refactor ECL parsing/etc.
Thibaut Girka <thib@sitedethib.com>
parents: 20
diff changeset
134
165
c8c60291c56f Implement item dropping by enemies.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents: 158
diff changeset
135 _main_instructions = {0: ('fffhhHH', 'spawn_enemy'),
c8c60291c56f Implement item dropping by enemies.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents: 158
diff changeset
136 2: ('fffhhHH', 'spawn_enemy_mirrored'),
c8c60291c56f Implement item dropping by enemies.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents: 158
diff changeset
137 4: ('fffhhHH', 'spawn_enemy_random'),
c8c60291c56f Implement item dropping by enemies.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents: 158
diff changeset
138 6: ('fffhhHH', 'spawn_enemy_mirrored_random'),
112
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
139 8: ('', None),
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
140 9: ('', None),
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
141 10: ('II', None),
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
142 12: ('', None)}
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
143
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
144
18
ca26a84916cb Add preliminary ECL viewer/interpreter.
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
145 def __init__(self):
ca26a84916cb Add preliminary ECL viewer/interpreter.
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
146 self.main = []
ca26a84916cb Add preliminary ECL viewer/interpreter.
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
147 self.subs = [[]]
ca26a84916cb Add preliminary ECL viewer/interpreter.
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
148
ca26a84916cb Add preliminary ECL viewer/interpreter.
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
149
ca26a84916cb Add preliminary ECL viewer/interpreter.
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
150 @classmethod
ca26a84916cb Add preliminary ECL viewer/interpreter.
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
151 def read(cls, file):
ca26a84916cb Add preliminary ECL viewer/interpreter.
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
152 sub_count, main_offset = unpack('<II', file.read(8))
ca26a84916cb Add preliminary ECL viewer/interpreter.
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
153 if file.read(8) != b'\x00\x00\x00\x00\x00\x00\x00\x00':
ca26a84916cb Add preliminary ECL viewer/interpreter.
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
154 raise Exception #TODO
112
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
155 sub_offsets = unpack('<%dI' % sub_count, file.read(4 * sub_count))
18
ca26a84916cb Add preliminary ECL viewer/interpreter.
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
156
ca26a84916cb Add preliminary ECL viewer/interpreter.
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
157 ecl = cls()
ca26a84916cb Add preliminary ECL viewer/interpreter.
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
158 ecl.subs = []
ca26a84916cb Add preliminary ECL viewer/interpreter.
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
159 ecl.main = []
ca26a84916cb Add preliminary ECL viewer/interpreter.
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
160
ca26a84916cb Add preliminary ECL viewer/interpreter.
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
161 # Read subs
43
7195aaf95f6e Fix set_counter, and relative_jump(_ex)
Thibaut Girka <thib@sitedethib.com>
parents: 42
diff changeset
162 for sub_offset in sub_offsets:
7195aaf95f6e Fix set_counter, and relative_jump(_ex)
Thibaut Girka <thib@sitedethib.com>
parents: 42
diff changeset
163 file.seek(sub_offset)
18
ca26a84916cb Add preliminary ECL viewer/interpreter.
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
164 ecl.subs.append([])
43
7195aaf95f6e Fix set_counter, and relative_jump(_ex)
Thibaut Girka <thib@sitedethib.com>
parents: 42
diff changeset
165
7195aaf95f6e Fix set_counter, and relative_jump(_ex)
Thibaut Girka <thib@sitedethib.com>
parents: 42
diff changeset
166 instruction_offsets = []
7195aaf95f6e Fix set_counter, and relative_jump(_ex)
Thibaut Girka <thib@sitedethib.com>
parents: 42
diff changeset
167
18
ca26a84916cb Add preliminary ECL viewer/interpreter.
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
168 while True:
43
7195aaf95f6e Fix set_counter, and relative_jump(_ex)
Thibaut Girka <thib@sitedethib.com>
parents: 42
diff changeset
169 instruction_offsets.append(file.tell() - sub_offset)
7195aaf95f6e Fix set_counter, and relative_jump(_ex)
Thibaut Girka <thib@sitedethib.com>
parents: 42
diff changeset
170
18
ca26a84916cb Add preliminary ECL viewer/interpreter.
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
171 time, opcode = unpack('<IH', file.read(6))
ca26a84916cb Add preliminary ECL viewer/interpreter.
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
172 if time == 0xffffffff or opcode == 0xffff:
ca26a84916cb Add preliminary ECL viewer/interpreter.
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
173 break
ca26a84916cb Add preliminary ECL viewer/interpreter.
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
174 size, rank_mask, param_mask = unpack('<HHH', file.read(6))
ca26a84916cb Add preliminary ECL viewer/interpreter.
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
175 data = file.read(size - 12)
42
1b0ca2fb89f9 Refactor ECL parsing/etc.
Thibaut Girka <thib@sitedethib.com>
parents: 20
diff changeset
176 if opcode in cls._instructions:
95
e2d8f2a56ea4 Handle ECL opcodes with string.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents: 87
diff changeset
177 fmt = '<%s' % cls._instructions[opcode][0]
e2d8f2a56ea4 Handle ECL opcodes with string.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents: 87
diff changeset
178 if fmt.endswith('s'):
e2d8f2a56ea4 Handle ECL opcodes with string.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents: 87
diff changeset
179 fmt = fmt[:-1]
e2d8f2a56ea4 Handle ECL opcodes with string.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents: 87
diff changeset
180 fmt = '%s%ds' % (fmt, size - 12 - calcsize(fmt))
e2d8f2a56ea4 Handle ECL opcodes with string.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents: 87
diff changeset
181 args = unpack(fmt, data)
e2d8f2a56ea4 Handle ECL opcodes with string.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents: 87
diff changeset
182 if fmt.endswith('s'):
e2d8f2a56ea4 Handle ECL opcodes with string.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents: 87
diff changeset
183 args = args[:-1] + (args[-1].decode('shift_jis'),)
42
1b0ca2fb89f9 Refactor ECL parsing/etc.
Thibaut Girka <thib@sitedethib.com>
parents: 20
diff changeset
184 else:
1b0ca2fb89f9 Refactor ECL parsing/etc.
Thibaut Girka <thib@sitedethib.com>
parents: 20
diff changeset
185 args = (data, )
58
3da4de9decd0 Use logging module
Thibaut Girka <thib@sitedethib.com>
parents: 52
diff changeset
186 logger.warn('unknown opcode %d', opcode)
43
7195aaf95f6e Fix set_counter, and relative_jump(_ex)
Thibaut Girka <thib@sitedethib.com>
parents: 42
diff changeset
187
42
1b0ca2fb89f9 Refactor ECL parsing/etc.
Thibaut Girka <thib@sitedethib.com>
parents: 20
diff changeset
188 ecl.subs[-1].append((time, opcode, rank_mask, param_mask, args))
18
ca26a84916cb Add preliminary ECL viewer/interpreter.
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
189
43
7195aaf95f6e Fix set_counter, and relative_jump(_ex)
Thibaut Girka <thib@sitedethib.com>
parents: 42
diff changeset
190
7195aaf95f6e Fix set_counter, and relative_jump(_ex)
Thibaut Girka <thib@sitedethib.com>
parents: 42
diff changeset
191 # Translate offsets to instruction pointers
7195aaf95f6e Fix set_counter, and relative_jump(_ex)
Thibaut Girka <thib@sitedethib.com>
parents: 42
diff changeset
192 for instr_offset, (i, instr) in zip(instruction_offsets, enumerate(ecl.subs[-1])):
7195aaf95f6e Fix set_counter, and relative_jump(_ex)
Thibaut Girka <thib@sitedethib.com>
parents: 42
diff changeset
193 time, opcode, rank_mask, param_mask, args = instr
66
a701a89192a9 Add offset translation for all relative jumps.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents: 64
diff changeset
194 if opcode in (2, 29, 30, 31, 32, 33, 34): # relative_jump
43
7195aaf95f6e Fix set_counter, and relative_jump(_ex)
Thibaut Girka <thib@sitedethib.com>
parents: 42
diff changeset
195 frame, relative_offset = args
7195aaf95f6e Fix set_counter, and relative_jump(_ex)
Thibaut Girka <thib@sitedethib.com>
parents: 42
diff changeset
196 args = frame, instruction_offsets.index(instr_offset + relative_offset)
7195aaf95f6e Fix set_counter, and relative_jump(_ex)
Thibaut Girka <thib@sitedethib.com>
parents: 42
diff changeset
197 elif opcode == 3: # relative_jump_ex
7195aaf95f6e Fix set_counter, and relative_jump(_ex)
Thibaut Girka <thib@sitedethib.com>
parents: 42
diff changeset
198 frame, relative_offset, counter_id = args
7195aaf95f6e Fix set_counter, and relative_jump(_ex)
Thibaut Girka <thib@sitedethib.com>
parents: 42
diff changeset
199 args = frame, instruction_offsets.index(instr_offset + relative_offset), counter_id
7195aaf95f6e Fix set_counter, and relative_jump(_ex)
Thibaut Girka <thib@sitedethib.com>
parents: 42
diff changeset
200 ecl.subs[-1][i] = time, opcode, rank_mask, param_mask, args
7195aaf95f6e Fix set_counter, and relative_jump(_ex)
Thibaut Girka <thib@sitedethib.com>
parents: 42
diff changeset
201
7195aaf95f6e Fix set_counter, and relative_jump(_ex)
Thibaut Girka <thib@sitedethib.com>
parents: 42
diff changeset
202
18
ca26a84916cb Add preliminary ECL viewer/interpreter.
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
203 # Read main
ca26a84916cb Add preliminary ECL viewer/interpreter.
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
204 file.seek(main_offset)
ca26a84916cb Add preliminary ECL viewer/interpreter.
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
205 while True:
ca26a84916cb Add preliminary ECL viewer/interpreter.
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
206 time, = unpack('<H', file.read(2))
ca26a84916cb Add preliminary ECL viewer/interpreter.
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
207 if time == 0xffff:
ca26a84916cb Add preliminary ECL viewer/interpreter.
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
208 break
112
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
209
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
210 sub, opcode, size = unpack('<HHH', file.read(6))
18
ca26a84916cb Add preliminary ECL viewer/interpreter.
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
211 data = file.read(size - 8)
112
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
212
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
213 if opcode in cls._main_instructions:
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
214 args = unpack('<%s' % cls._main_instructions[opcode][0], data)
18
ca26a84916cb Add preliminary ECL viewer/interpreter.
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
215 else:
ca26a84916cb Add preliminary ECL viewer/interpreter.
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
216 args = (data,)
112
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
217 logger.warn('unknown main opcode %d', opcode)
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
218
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
219 ecl.main.append((time, sub, opcode, args))
18
ca26a84916cb Add preliminary ECL viewer/interpreter.
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
220
ca26a84916cb Add preliminary ECL viewer/interpreter.
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
221 return ecl
ca26a84916cb Add preliminary ECL viewer/interpreter.
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
222
112
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
223
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
224 def write(self, file):
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
225 sub_count = len(self.subs)
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
226 sub_offsets = []
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
227 main_offset = 0
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
228
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
229 # Skip header, it will be written later
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
230 file.seek(8+8+4*sub_count)
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
231
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
232 # Write subs
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
233 for sub in self.subs:
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
234 sub_offsets.append(file.tell())
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
235
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
236 instruction_offsets = []
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
237 instruction_datas = []
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
238 for time, opcode, rank_mask, param_mask, args in sub:
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
239 format = self._instructions[opcode][0]
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
240 if format.endswith('s'):
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
241 args = list(args)
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
242 args[-1] = args[-1].encode('shift_jis')
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
243 format = '%s%ds' % (format[:-1], len(args[-1]))
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
244 format = '<IHHHH%s' % format
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
245 size = calcsize(format)
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
246 instruction_offsets.append((instruction_offsets[-1] + len(instruction_datas[-1])) if instruction_offsets else 0)
113
732c64662f87 Minor changes
Thibaut Girka <thib@sitedethib.com>
parents: 112
diff changeset
247 try:
732c64662f87 Minor changes
Thibaut Girka <thib@sitedethib.com>
parents: 112
diff changeset
248 instruction_datas.append(pack(format, time, opcode, size, rank_mask, param_mask, *args))
732c64662f87 Minor changes
Thibaut Girka <thib@sitedethib.com>
parents: 112
diff changeset
249 except struct.error:
732c64662f87 Minor changes
Thibaut Girka <thib@sitedethib.com>
parents: 112
diff changeset
250 logger.error('Failed to assemble opcode %d' % opcode)
732c64662f87 Minor changes
Thibaut Girka <thib@sitedethib.com>
parents: 112
diff changeset
251 raise
112
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
252
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
253 #TODO: clean up this mess
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
254 for instruction, data, offset in zip(sub, instruction_datas, instruction_offsets):
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
255 time, opcode, rank_mask, param_mask, args = instruction
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
256 if opcode in (2, 29, 30, 31, 32, 33, 34): # relative_jump
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
257 frame, index = args
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
258 args = frame, instruction_offsets[index] - offset
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
259 format = '<IHHHH%s' % self._instructions[opcode][0]
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
260 size = calcsize(format)
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
261 data = pack(format, time, opcode, size, rank_mask, param_mask, *args)
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
262 elif opcode == 3: # relative_jump_ex
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
263 frame, index, counter_id = args
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
264 args = frame, instruction_offsets[index] - offset, counter_id
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
265 format = '<IHHHH%s' % self._instructions[opcode][0]
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
266 size = calcsize(format)
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
267 data = pack(format, time, opcode, size, rank_mask, param_mask, *args)
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
268 file.write(data)
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
269 file.write(b'\xff' * 6 + b'\x0c\x00\x00\xff\xff\x00')
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
270
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
271 # Write main
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
272 main_offset = file.tell()
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
273 for time, sub, opcode, args in self.main:
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
274 format = '<HHHH%s' % self._main_instructions[opcode][0]
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
275 size = calcsize(format)
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
276
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
277 file.write(pack(format, time, sub, opcode, size, *args))
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
278 file.write(b'\xff\xff\x04\x00')
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
279
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
280 # Patch header
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
281 file.seek(0)
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
282 file.write(pack('<IIII%dI' % sub_count, sub_count, main_offset, 0, 0, *sub_offsets))
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
283