Mercurial > touhou

annotate pytouhou/formats/ecl.py @ 283:b6c068c8f7f2

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Fix ECL time flow. Spellcard do not stop time. Instruction 0xc does.
author Thibaut Girka <thib@sitedethib.com>
date Sun, 12 Feb 2012 02:09:16 +0100
parents f037bca24f2d
children 4838e9bab0f9
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
204
88361534c77e Add some documentation (argh, so much left to document!)
Thibaut Girka <thib@sitedethib.com>
parents: 171
diff changeset
15 """ECL files handling.
88361534c77e Add some documentation (argh, so much left to document!)
Thibaut Girka <thib@sitedethib.com>
parents: 171
diff changeset
16
88361534c77e Add some documentation (argh, so much left to document!)
Thibaut Girka <thib@sitedethib.com>
parents: 171
diff changeset
17 This module provides classes for handling the ECL file format.
88361534c77e Add some documentation (argh, so much left to document!)
Thibaut Girka <thib@sitedethib.com>
parents: 171
diff changeset
18 The ECL format is a format used in Touhou 6: EoSD to script most of the gameplay
88361534c77e Add some documentation (argh, so much left to document!)
Thibaut Girka <thib@sitedethib.com>
parents: 171
diff changeset
19 aspects of the game, such as enemy movements, attacks, and so on.
88361534c77e Add some documentation (argh, so much left to document!)
Thibaut Girka <thib@sitedethib.com>
parents: 171
diff changeset
20 """
88361534c77e Add some documentation (argh, so much left to document!)
Thibaut Girka <thib@sitedethib.com>
parents: 171
diff changeset
21
113
732c64662f87 Minor changes
Thibaut Girka <thib@sitedethib.com>
parents: 112
diff changeset
22 import struct
95
e2d8f2a56ea4 Handle ECL opcodes with string.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents: 87
diff changeset
23 from struct import pack, unpack, calcsize
18
ca26a84916cb Add preliminary ECL viewer/interpreter.
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
24
58
3da4de9decd0 Use logging module
Thibaut Girka <thib@sitedethib.com>
parents: 52
diff changeset
25 from pytouhou.utils.helpers import get_logger
18
ca26a84916cb Add preliminary ECL viewer/interpreter.
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
26
58
3da4de9decd0 Use logging module
Thibaut Girka <thib@sitedethib.com>
parents: 52
diff changeset
27 logger = get_logger(__name__)
18
ca26a84916cb Add preliminary ECL viewer/interpreter.
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
28
ca26a84916cb Add preliminary ECL viewer/interpreter.
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
29 class ECL(object):
204
88361534c77e Add some documentation (argh, so much left to document!)
Thibaut Girka <thib@sitedethib.com>
parents: 171
diff changeset
30 """Handle Touhou 6 ECL files.
88361534c77e Add some documentation (argh, so much left to document!)
Thibaut Girka <thib@sitedethib.com>
parents: 171
diff changeset
31
88361534c77e Add some documentation (argh, so much left to document!)
Thibaut Girka <thib@sitedethib.com>
parents: 171
diff changeset
32 ECL files are binary script files used to describe the behavior of enemies.
88361534c77e Add some documentation (argh, so much left to document!)
Thibaut Girka <thib@sitedethib.com>
parents: 171
diff changeset
33 They are basically composed of a header and two additional sections.
88361534c77e Add some documentation (argh, so much left to document!)
Thibaut Girka <thib@sitedethib.com>
parents: 171
diff changeset
34 The first section is a list of subroutines, each composed of a list of timed
88361534c77e Add some documentation (argh, so much left to document!)
Thibaut Girka <thib@sitedethib.com>
parents: 171
diff changeset
35 instructions.
88361534c77e Add some documentation (argh, so much left to document!)
Thibaut Girka <thib@sitedethib.com>
parents: 171
diff changeset
36 The second section is a list of a different set of instructions describing
88361534c77e Add some documentation (argh, so much left to document!)
Thibaut Girka <thib@sitedethib.com>
parents: 171
diff changeset
37 enemy waves, triggering dialogs and level completion.
88361534c77e Add some documentation (argh, so much left to document!)
Thibaut Girka <thib@sitedethib.com>
parents: 171
diff changeset
38
88361534c77e Add some documentation (argh, so much left to document!)
Thibaut Girka <thib@sitedethib.com>
parents: 171
diff changeset
39 Instance variables:
88361534c77e Add some documentation (argh, so much left to document!)
Thibaut Girka <thib@sitedethib.com>
parents: 171
diff changeset
40 main -- list of instructions describing waves and triggering dialogs
88361534c77e Add some documentation (argh, so much left to document!)
Thibaut Girka <thib@sitedethib.com>
parents: 171
diff changeset
41 subs -- list of subroutines
88361534c77e Add some documentation (argh, so much left to document!)
Thibaut Girka <thib@sitedethib.com>
parents: 171
diff changeset
42 """
88361534c77e Add some documentation (argh, so much left to document!)
Thibaut Girka <thib@sitedethib.com>
parents: 171
diff changeset
43
42
1b0ca2fb89f9 Refactor ECL parsing/etc.
Thibaut Girka <thib@sitedethib.com>
parents: 20
diff changeset
44 _instructions = {0: ('', 'noop?'),
1b0ca2fb89f9 Refactor ECL parsing/etc.
Thibaut Girka <thib@sitedethib.com>
parents: 20
diff changeset
45 1: ('I', 'delete?'),
1b0ca2fb89f9 Refactor ECL parsing/etc.
Thibaut Girka <thib@sitedethib.com>
parents: 20
diff changeset
46 2: ('Ii', 'relative_jump'),
1b0ca2fb89f9 Refactor ECL parsing/etc.
Thibaut Girka <thib@sitedethib.com>
parents: 20
diff changeset
47 3: ('Iii', 'relative_jump_ex'),
46
25ca15f714ad Add/fix a few opcodes
Thibaut Girka <thib@sitedethib.com>
parents: 43
diff changeset
48 4: ('ii', 'set_int'),
25ca15f714ad Add/fix a few opcodes
Thibaut Girka <thib@sitedethib.com>
parents: 43
diff changeset
49 5: ('if', 'set_float'),
25ca15f714ad Add/fix a few opcodes
Thibaut Girka <thib@sitedethib.com>
parents: 43
diff changeset
50 6: ('ii', 'set_random_int'),
25ca15f714ad Add/fix a few opcodes
Thibaut Girka <thib@sitedethib.com>
parents: 43
diff changeset
51 8: ('if', 'set_random_float'),
59
4fe37a620b22 Handle set_random_angle properly! At last!
Thibaut Girka <thib@sitedethib.com>
parents: 58
diff changeset
52 9: ('iff', 'set_random_float2'),
75
b3bd421bb895 Handle a few more ECL instructions
Thibaut Girka <thib@sitedethib.com>
parents: 70
diff changeset
53 10: ('i', 'store_x'),
87
fda176f07d6d Fix add_int
Thibaut Girka <thib@sitedethib.com>
parents: 79
diff changeset
54 13: ('iii', 'add_int'),
63
8527fe640844 Implement simple arithmetic instructions.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents: 59
diff changeset
55 14: ('iii', 'substract_int'),
8527fe640844 Implement simple arithmetic instructions.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents: 59
diff changeset
56 15: ('iii', 'multiply_int'),
8527fe640844 Implement simple arithmetic instructions.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents: 59
diff changeset
57 16: ('iii', 'divide_int'),
8527fe640844 Implement simple arithmetic instructions.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents: 59
diff changeset
58 17: ('iii', 'modulo'),
96
54929d495654 Handle ECL instruction 18
Thibaut Girka <thib@sitedethib.com>
parents: 95
diff changeset
59 18: ('i', 'increment'),
63
8527fe640844 Implement simple arithmetic instructions.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents: 59
diff changeset
60 20: ('iff', 'add_float'),
8527fe640844 Implement simple arithmetic instructions.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents: 59
diff changeset
61 21: ('iff', 'substract_float'),
8527fe640844 Implement simple arithmetic instructions.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents: 59
diff changeset
62 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
63 25: ('iffff', 'get_direction'),
204
88361534c77e Add some documentation (argh, so much left to document!)
Thibaut Girka <thib@sitedethib.com>
parents: 171
diff changeset
64 26: ('i', 'float_to_unit_circle'), #TODO: find a better name
46
25ca15f714ad Add/fix a few opcodes
Thibaut Girka <thib@sitedethib.com>
parents: 43
diff changeset
65 27: ('ii', 'compare_ints'),
63
8527fe640844 Implement simple arithmetic instructions.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents: 59
diff changeset
66 28: ('ff', 'compare_floats'),
64
d469012368b3 Implement conditional jumps.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents: 63
diff changeset
67 29: ('ii', 'relative_jump_if_lower_than'),
d469012368b3 Implement conditional jumps.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents: 63
diff changeset
68 30: ('ii', 'relative_jump_if_lower_or_equal'),
46
25ca15f714ad Add/fix a few opcodes
Thibaut Girka <thib@sitedethib.com>
parents: 43
diff changeset
69 31: ('ii', 'relative_jump_if_equal'),
64
d469012368b3 Implement conditional jumps.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents: 63
diff changeset
70 32: ('ii', 'relative_jump_if_greater_than'),
d469012368b3 Implement conditional jumps.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents: 63
diff changeset
71 33: ('ii', 'relative_jump_if_greater_or_equal'),
d469012368b3 Implement conditional jumps.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents: 63
diff changeset
72 34: ('ii', 'relative_jump_if_not_equal'),
46
25ca15f714ad Add/fix a few opcodes
Thibaut Girka <thib@sitedethib.com>
parents: 43
diff changeset
73 35: ('iif', 'call'),
204
88361534c77e Add some documentation (argh, so much left to document!)
Thibaut Girka <thib@sitedethib.com>
parents: 171
diff changeset
74 36: ('', 'return'),
46
25ca15f714ad Add/fix a few opcodes
Thibaut Girka <thib@sitedethib.com>
parents: 43
diff changeset
75 39: ('iifii', 'call_if_equal'),
42
1b0ca2fb89f9 Refactor ECL parsing/etc.
Thibaut Girka <thib@sitedethib.com>
parents: 20
diff changeset
76 43: ('fff', 'set_position'),
1b0ca2fb89f9 Refactor ECL parsing/etc.
Thibaut Girka <thib@sitedethib.com>
parents: 20
diff changeset
77 45: ('ff', 'set_angle_and_speed'),
1b0ca2fb89f9 Refactor ECL parsing/etc.
Thibaut Girka <thib@sitedethib.com>
parents: 20
diff changeset
78 46: ('f', 'set_rotation_speed'),
1b0ca2fb89f9 Refactor ECL parsing/etc.
Thibaut Girka <thib@sitedethib.com>
parents: 20
diff changeset
79 47: ('f', 'set_speed'),
1b0ca2fb89f9 Refactor ECL parsing/etc.
Thibaut Girka <thib@sitedethib.com>
parents: 20
diff changeset
80 48: ('f', 'set_acceleration'),
70
7c1f20407b3e Add set_random_angle support
Thibaut Girka <thib@sitedethib.com>
parents: 67
diff changeset
81 49: ('ff', 'set_random_angle'),
7c1f20407b3e Add set_random_angle support
Thibaut Girka <thib@sitedethib.com>
parents: 67
diff changeset
82 50: ('ff', 'set_random_angle_ex'),
42
1b0ca2fb89f9 Refactor ECL parsing/etc.
Thibaut Girka <thib@sitedethib.com>
parents: 20
diff changeset
83 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
84 52: ('iff', 'move_in_decel'),
76
f305c0e406d6 Handle all move_to_* ECL instructions.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents: 75
diff changeset
85 56: ('ifff', 'move_to_linear'),
f305c0e406d6 Handle all move_to_* ECL instructions.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents: 75
diff changeset
86 57: ('ifff', 'move_to_decel'),
f305c0e406d6 Handle all move_to_* ECL instructions.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents: 75
diff changeset
87 59: ('iffi', 'move_to_accel'),
46
25ca15f714ad Add/fix a few opcodes
Thibaut Girka <thib@sitedethib.com>
parents: 43
diff changeset
88 61: ('i', 'stop_in'),
75
b3bd421bb895 Handle a few more ECL instructions
Thibaut Girka <thib@sitedethib.com>
parents: 70
diff changeset
89 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
90 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
91 66: ('', 'clear_screen_box'),
42
1b0ca2fb89f9 Refactor ECL parsing/etc.
Thibaut Girka <thib@sitedethib.com>
parents: 20
diff changeset
92 67: ('hhiiffffi', 'set_bullet_attributes'),
1b0ca2fb89f9 Refactor ECL parsing/etc.
Thibaut Girka <thib@sitedethib.com>
parents: 20
diff changeset
93 68: ('hhiiffffi', 'set_bullet_attributes2'),
1b0ca2fb89f9 Refactor ECL parsing/etc.
Thibaut Girka <thib@sitedethib.com>
parents: 20
diff changeset
94 69: ('hhiiffffi', 'set_bullet_attributes3'),
1b0ca2fb89f9 Refactor ECL parsing/etc.
Thibaut Girka <thib@sitedethib.com>
parents: 20
diff changeset
95 70: ('hhiiffffi', 'set_bullet_attributes4'),
46
25ca15f714ad Add/fix a few opcodes
Thibaut Girka <thib@sitedethib.com>
parents: 43
diff changeset
96 71: ('hhiiffffi', 'set_bullet_attributes5'),
25ca15f714ad Add/fix a few opcodes
Thibaut Girka <thib@sitedethib.com>
parents: 43
diff changeset
97 74: ('hhiiffffi', 'set_bullet_attributes6'),
25ca15f714ad Add/fix a few opcodes
Thibaut Girka <thib@sitedethib.com>
parents: 43
diff changeset
98 75: ('hhiiffffi', 'set_bullet_attributes7'),
78
bcf965ede96c Fix/rename/comment a few things
Thibaut Girka <thib@sitedethib.com>
parents: 77
diff changeset
99 76: ('i', 'set_bullet_interval'),
bcf965ede96c Fix/rename/comment a few things
Thibaut Girka <thib@sitedethib.com>
parents: 77
diff changeset
100 77: ('i', 'set_bullet_interval_ex'),
46
25ca15f714ad Add/fix a few opcodes
Thibaut Girka <thib@sitedethib.com>
parents: 43
diff changeset
101 78: ('', 'delay_attack'),
25ca15f714ad Add/fix a few opcodes
Thibaut Girka <thib@sitedethib.com>
parents: 43
diff changeset
102 79: ('', 'no_delay_attack'),
42
1b0ca2fb89f9 Refactor ECL parsing/etc.
Thibaut Girka <thib@sitedethib.com>
parents: 20
diff changeset
103 81: ('fff', 'set_bullet_launch_offset'),
78
bcf965ede96c Fix/rename/comment a few things
Thibaut Girka <thib@sitedethib.com>
parents: 77
diff changeset
104 82: ('iiiiffff', 'set_extended_bullet_attributes'),
204
88361534c77e Add some documentation (argh, so much left to document!)
Thibaut Girka <thib@sitedethib.com>
parents: 171
diff changeset
105 83: ('', 'change_bullets_in_star_bonus'),
42
1b0ca2fb89f9 Refactor ECL parsing/etc.
Thibaut Girka <thib@sitedethib.com>
parents: 20
diff changeset
106 84: ('i', None),
46
25ca15f714ad Add/fix a few opcodes
Thibaut Girka <thib@sitedethib.com>
parents: 43
diff changeset
107 85: ('hhffffffiiiiii', 'laser'),
25ca15f714ad Add/fix a few opcodes
Thibaut Girka <thib@sitedethib.com>
parents: 43
diff changeset
108 86: ('hhffffffiiiiii', 'laser2'),
25ca15f714ad Add/fix a few opcodes
Thibaut Girka <thib@sitedethib.com>
parents: 43
diff changeset
109 87: ('i', 'set_upcoming_id'),
25ca15f714ad Add/fix a few opcodes
Thibaut Girka <thib@sitedethib.com>
parents: 43
diff changeset
110 88: ('if','alter_laser_angle'),
274
f037bca24f2d Partially implement lasers.
Thibaut Girka <thib@sitedethib.com>
parents: 204
diff changeset
111 90: ('ifff', 'reposition_laser'),
204
88361534c77e Add some documentation (argh, so much left to document!)
Thibaut Girka <thib@sitedethib.com>
parents: 171
diff changeset
112 92: ('i', 'cancel_laser'),
95
e2d8f2a56ea4 Handle ECL opcodes with string.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents: 87
diff changeset
113 93: ('hhs', 'set_spellcard'),
67
e2cb9d434dc2 Add support for a few opcodes.
Thibaut Girka <thib@sitedethib.com>
parents: 66
diff changeset
114 94: ('', 'end_spellcard'),
204
88361534c77e Add some documentation (argh, so much left to document!)
Thibaut Girka <thib@sitedethib.com>
parents: 171
diff changeset
115 95: ('ifffhhi', 'spawn_enemy'),
88361534c77e Add some documentation (argh, so much left to document!)
Thibaut Girka <thib@sitedethib.com>
parents: 171
diff changeset
116 96: ('', 'kill_all_enemies'),
42
1b0ca2fb89f9 Refactor ECL parsing/etc.
Thibaut Girka <thib@sitedethib.com>
parents: 20
diff changeset
117 97: ('i', 'set_anim'),
46
25ca15f714ad Add/fix a few opcodes
Thibaut Girka <thib@sitedethib.com>
parents: 43
diff changeset
118 98: ('hhhhhxx', 'set_multiple_anims'),
42
1b0ca2fb89f9 Refactor ECL parsing/etc.
Thibaut Girka <thib@sitedethib.com>
parents: 20
diff changeset
119 99: ('ii', None),
1b0ca2fb89f9 Refactor ECL parsing/etc.
Thibaut Girka <thib@sitedethib.com>
parents: 20
diff changeset
120 100: ('i', 'set_death_anim'),
1b0ca2fb89f9 Refactor ECL parsing/etc.
Thibaut Girka <thib@sitedethib.com>
parents: 20
diff changeset
121 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
122 102: ('iffff', 'create_squares'),
42
1b0ca2fb89f9 Refactor ECL parsing/etc.
Thibaut Girka <thib@sitedethib.com>
parents: 20
diff changeset
123 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
124 104: ('i', 'set_collidable'),
67
e2cb9d434dc2 Add support for a few opcodes.
Thibaut Girka <thib@sitedethib.com>
parents: 66
diff changeset
125 105: ('i', 'set_damageable'),
42
1b0ca2fb89f9 Refactor ECL parsing/etc.
Thibaut Girka <thib@sitedethib.com>
parents: 20
diff changeset
126 106: ('i', 'play_sound'),
75
b3bd421bb895 Handle a few more ECL instructions
Thibaut Girka <thib@sitedethib.com>
parents: 70
diff changeset
127 107: ('i', 'set_death_flags'),
42
1b0ca2fb89f9 Refactor ECL parsing/etc.
Thibaut Girka <thib@sitedethib.com>
parents: 20
diff changeset
128 108: ('i', 'set_death_callback?'),
46
25ca15f714ad Add/fix a few opcodes
Thibaut Girka <thib@sitedethib.com>
parents: 43
diff changeset
129 109: ('ii', 'memory_write_int'),
25ca15f714ad Add/fix a few opcodes
Thibaut Girka <thib@sitedethib.com>
parents: 43
diff changeset
130 111: ('i', 'set_life'),
67
e2cb9d434dc2 Add support for a few opcodes.
Thibaut Girka <thib@sitedethib.com>
parents: 66
diff changeset
131 112: ('i', 'set_ellapsed_time'),
42
1b0ca2fb89f9 Refactor ECL parsing/etc.
Thibaut Girka <thib@sitedethib.com>
parents: 20
diff changeset
132 113: ('i', 'set_low_life_trigger'),
1b0ca2fb89f9 Refactor ECL parsing/etc.
Thibaut Girka <thib@sitedethib.com>
parents: 20
diff changeset
133 114: ('i', 'set_low_life_callback'),
1b0ca2fb89f9 Refactor ECL parsing/etc.
Thibaut Girka <thib@sitedethib.com>
parents: 20
diff changeset
134 115: ('i', 'set_timeout'),
67
e2cb9d434dc2 Add support for a few opcodes.
Thibaut Girka <thib@sitedethib.com>
parents: 66
diff changeset
135 116: ('i', 'set_timeout_callback'),
e2cb9d434dc2 Add support for a few opcodes.
Thibaut Girka <thib@sitedethib.com>
parents: 66
diff changeset
136 117: ('i', 'set_touchable'),
42
1b0ca2fb89f9 Refactor ECL parsing/etc.
Thibaut Girka <thib@sitedethib.com>
parents: 20
diff changeset
137 118: ('iihh', None),
46
25ca15f714ad Add/fix a few opcodes
Thibaut Girka <thib@sitedethib.com>
parents: 43
diff changeset
138 119: ('i', 'drop_bonus'),
75
b3bd421bb895 Handle a few more ECL instructions
Thibaut Girka <thib@sitedethib.com>
parents: 70
diff changeset
139 120: ('i', 'set_automatic_orientation'),
107
5d9052b9a4e8 (almost) implement Cirno's freezing spellcard
Thibaut Girka <thib@sitedethib.com>
parents: 96
diff changeset
140 121: ('ii', 'call_special_function'),
42
1b0ca2fb89f9 Refactor ECL parsing/etc.
Thibaut Girka <thib@sitedethib.com>
parents: 20
diff changeset
141 122: ('i', None),
75
b3bd421bb895 Handle a few more ECL instructions
Thibaut Girka <thib@sitedethib.com>
parents: 70
diff changeset
142 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
143 124: ('i', 'drop_specific_bonus'),
42
1b0ca2fb89f9 Refactor ECL parsing/etc.
Thibaut Girka <thib@sitedethib.com>
parents: 20
diff changeset
144 125: ('', None),
1b0ca2fb89f9 Refactor ECL parsing/etc.
Thibaut Girka <thib@sitedethib.com>
parents: 20
diff changeset
145 126: ('i', 'set_remaining_lives'),
1b0ca2fb89f9 Refactor ECL parsing/etc.
Thibaut Girka <thib@sitedethib.com>
parents: 20
diff changeset
146 127: ('i', None),
204
88361534c77e Add some documentation (argh, so much left to document!)
Thibaut Girka <thib@sitedethib.com>
parents: 171
diff changeset
147 128: ('i', 'set_smooth_disappear'),
42
1b0ca2fb89f9 Refactor ECL parsing/etc.
Thibaut Girka <thib@sitedethib.com>
parents: 20
diff changeset
148 129: ('ii', None),
1b0ca2fb89f9 Refactor ECL parsing/etc.
Thibaut Girka <thib@sitedethib.com>
parents: 20
diff changeset
149 130: ('i', None),
204
88361534c77e Add some documentation (argh, so much left to document!)
Thibaut Girka <thib@sitedethib.com>
parents: 171
diff changeset
150 131: ('ffiiii', 'set_difficulty_coeffs'),
88361534c77e Add some documentation (argh, so much left to document!)
Thibaut Girka <thib@sitedethib.com>
parents: 171
diff changeset
151 132: ('i', 'set_invisible'),
42
1b0ca2fb89f9 Refactor ECL parsing/etc.
Thibaut Girka <thib@sitedethib.com>
parents: 20
diff changeset
152 133: ('', None),
1b0ca2fb89f9 Refactor ECL parsing/etc.
Thibaut Girka <thib@sitedethib.com>
parents: 20
diff changeset
153 134: ('', None),
204
88361534c77e Add some documentation (argh, so much left to document!)
Thibaut Girka <thib@sitedethib.com>
parents: 171
diff changeset
154 135: ('i', 'enable_spellcard_bonus')} #TODO
42
1b0ca2fb89f9 Refactor ECL parsing/etc.
Thibaut Girka <thib@sitedethib.com>
parents: 20
diff changeset
155
171
2f3665a77f11 Add support for the last unknown value of the enemy spawning.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents: 165
diff changeset
156 _main_instructions = {0: ('fffhhI', 'spawn_enemy'),
2f3665a77f11 Add support for the last unknown value of the enemy spawning.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents: 165
diff changeset
157 2: ('fffhhI', 'spawn_enemy_mirrored'),
2f3665a77f11 Add support for the last unknown value of the enemy spawning.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents: 165
diff changeset
158 4: ('fffhhI', 'spawn_enemy_random'),
2f3665a77f11 Add support for the last unknown value of the enemy spawning.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents: 165
diff changeset
159 6: ('fffhhI', 'spawn_enemy_mirrored_random'),
112
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
160 8: ('', None),
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
161 9: ('', None),
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
162 10: ('II', None),
283
b6c068c8f7f2 Fix ECL time flow. Spellcard do not stop time. Instruction 0xc does.
Thibaut Girka <thib@sitedethib.com>
parents: 274
diff changeset
163 12: ('', 'stop_time')}
112
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
164
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
165
18
ca26a84916cb Add preliminary ECL viewer/interpreter.
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
166 def __init__(self):
ca26a84916cb Add preliminary ECL viewer/interpreter.
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
167 self.main = []
ca26a84916cb Add preliminary ECL viewer/interpreter.
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
168 self.subs = [[]]
ca26a84916cb Add preliminary ECL viewer/interpreter.
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
169
ca26a84916cb Add preliminary ECL viewer/interpreter.
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
170
ca26a84916cb Add preliminary ECL viewer/interpreter.
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
171 @classmethod
ca26a84916cb Add preliminary ECL viewer/interpreter.
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
172 def read(cls, file):
204
88361534c77e Add some documentation (argh, so much left to document!)
Thibaut Girka <thib@sitedethib.com>
parents: 171
diff changeset
173 """Read an ECL file.
88361534c77e Add some documentation (argh, so much left to document!)
Thibaut Girka <thib@sitedethib.com>
parents: 171
diff changeset
174
88361534c77e Add some documentation (argh, so much left to document!)
Thibaut Girka <thib@sitedethib.com>
parents: 171
diff changeset
175 Raise an exception if the file is invalid.
88361534c77e Add some documentation (argh, so much left to document!)
Thibaut Girka <thib@sitedethib.com>
parents: 171
diff changeset
176 Return a ECL instance otherwise.
88361534c77e Add some documentation (argh, so much left to document!)
Thibaut Girka <thib@sitedethib.com>
parents: 171
diff changeset
177 """
88361534c77e Add some documentation (argh, so much left to document!)
Thibaut Girka <thib@sitedethib.com>
parents: 171
diff changeset
178
18
ca26a84916cb Add preliminary ECL viewer/interpreter.
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
179 sub_count, main_offset = unpack('<II', file.read(8))
ca26a84916cb Add preliminary ECL viewer/interpreter.
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
180 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
181 raise Exception #TODO
112
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
182 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
183
ca26a84916cb Add preliminary ECL viewer/interpreter.
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
184 ecl = cls()
ca26a84916cb Add preliminary ECL viewer/interpreter.
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
185 ecl.subs = []
ca26a84916cb Add preliminary ECL viewer/interpreter.
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
186 ecl.main = []
ca26a84916cb Add preliminary ECL viewer/interpreter.
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
187
ca26a84916cb Add preliminary ECL viewer/interpreter.
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
188 # Read subs
43
7195aaf95f6e Fix set_counter, and relative_jump(_ex)
Thibaut Girka <thib@sitedethib.com>
parents: 42
diff changeset
189 for sub_offset in sub_offsets:
7195aaf95f6e Fix set_counter, and relative_jump(_ex)
Thibaut Girka <thib@sitedethib.com>
parents: 42
diff changeset
190 file.seek(sub_offset)
18
ca26a84916cb Add preliminary ECL viewer/interpreter.
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
191 ecl.subs.append([])
43
7195aaf95f6e Fix set_counter, and relative_jump(_ex)
Thibaut Girka <thib@sitedethib.com>
parents: 42
diff changeset
192
7195aaf95f6e Fix set_counter, and relative_jump(_ex)
Thibaut Girka <thib@sitedethib.com>
parents: 42
diff changeset
193 instruction_offsets = []
7195aaf95f6e Fix set_counter, and relative_jump(_ex)
Thibaut Girka <thib@sitedethib.com>
parents: 42
diff changeset
194
18
ca26a84916cb Add preliminary ECL viewer/interpreter.
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
195 while True:
43
7195aaf95f6e Fix set_counter, and relative_jump(_ex)
Thibaut Girka <thib@sitedethib.com>
parents: 42
diff changeset
196 instruction_offsets.append(file.tell() - sub_offset)
7195aaf95f6e Fix set_counter, and relative_jump(_ex)
Thibaut Girka <thib@sitedethib.com>
parents: 42
diff changeset
197
18
ca26a84916cb Add preliminary ECL viewer/interpreter.
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
198 time, opcode = unpack('<IH', file.read(6))
ca26a84916cb Add preliminary ECL viewer/interpreter.
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
199 if time == 0xffffffff or opcode == 0xffff:
ca26a84916cb Add preliminary ECL viewer/interpreter.
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
200 break
204
88361534c77e Add some documentation (argh, so much left to document!)
Thibaut Girka <thib@sitedethib.com>
parents: 171
diff changeset
201
18
ca26a84916cb Add preliminary ECL viewer/interpreter.
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
202 size, rank_mask, param_mask = unpack('<HHH', file.read(6))
ca26a84916cb Add preliminary ECL viewer/interpreter.
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
203 data = file.read(size - 12)
42
1b0ca2fb89f9 Refactor ECL parsing/etc.
Thibaut Girka <thib@sitedethib.com>
parents: 20
diff changeset
204 if opcode in cls._instructions:
95
e2d8f2a56ea4 Handle ECL opcodes with string.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents: 87
diff changeset
205 fmt = '<%s' % cls._instructions[opcode][0]
e2d8f2a56ea4 Handle ECL opcodes with string.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents: 87
diff changeset
206 if fmt.endswith('s'):
e2d8f2a56ea4 Handle ECL opcodes with string.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents: 87
diff changeset
207 fmt = fmt[:-1]
e2d8f2a56ea4 Handle ECL opcodes with string.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents: 87
diff changeset
208 fmt = '%s%ds' % (fmt, size - 12 - calcsize(fmt))
e2d8f2a56ea4 Handle ECL opcodes with string.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents: 87
diff changeset
209 args = unpack(fmt, data)
e2d8f2a56ea4 Handle ECL opcodes with string.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents: 87
diff changeset
210 if fmt.endswith('s'):
e2d8f2a56ea4 Handle ECL opcodes with string.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents: 87
diff changeset
211 args = args[:-1] + (args[-1].decode('shift_jis'),)
42
1b0ca2fb89f9 Refactor ECL parsing/etc.
Thibaut Girka <thib@sitedethib.com>
parents: 20
diff changeset
212 else:
1b0ca2fb89f9 Refactor ECL parsing/etc.
Thibaut Girka <thib@sitedethib.com>
parents: 20
diff changeset
213 args = (data, )
58
3da4de9decd0 Use logging module
Thibaut Girka <thib@sitedethib.com>
parents: 52
diff changeset
214 logger.warn('unknown opcode %d', opcode)
43
7195aaf95f6e Fix set_counter, and relative_jump(_ex)
Thibaut Girka <thib@sitedethib.com>
parents: 42
diff changeset
215
42
1b0ca2fb89f9 Refactor ECL parsing/etc.
Thibaut Girka <thib@sitedethib.com>
parents: 20
diff changeset
216 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
217
43
7195aaf95f6e Fix set_counter, and relative_jump(_ex)
Thibaut Girka <thib@sitedethib.com>
parents: 42
diff changeset
218
204
88361534c77e Add some documentation (argh, so much left to document!)
Thibaut Girka <thib@sitedethib.com>
parents: 171
diff changeset
219 # Translate offsets to instruction pointers.
88361534c77e Add some documentation (argh, so much left to document!)
Thibaut Girka <thib@sitedethib.com>
parents: 171
diff changeset
220 # Indeed, jump instructions are relative and byte-based.
88361534c77e Add some documentation (argh, so much left to document!)
Thibaut Girka <thib@sitedethib.com>
parents: 171
diff changeset
221 # Since our representation doesn't conserve offsets, we have to
88361534c77e Add some documentation (argh, so much left to document!)
Thibaut Girka <thib@sitedethib.com>
parents: 171
diff changeset
222 # keep trace of where the jump is supposed to end up.
43
7195aaf95f6e Fix set_counter, and relative_jump(_ex)
Thibaut Girka <thib@sitedethib.com>
parents: 42
diff changeset
223 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
224 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
225 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
226 frame, relative_offset = args
7195aaf95f6e Fix set_counter, and relative_jump(_ex)
Thibaut Girka <thib@sitedethib.com>
parents: 42
diff changeset
227 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
228 elif opcode == 3: # relative_jump_ex
7195aaf95f6e Fix set_counter, and relative_jump(_ex)
Thibaut Girka <thib@sitedethib.com>
parents: 42
diff changeset
229 frame, relative_offset, counter_id = args
7195aaf95f6e Fix set_counter, and relative_jump(_ex)
Thibaut Girka <thib@sitedethib.com>
parents: 42
diff changeset
230 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
231 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
232
7195aaf95f6e Fix set_counter, and relative_jump(_ex)
Thibaut Girka <thib@sitedethib.com>
parents: 42
diff changeset
233
18
ca26a84916cb Add preliminary ECL viewer/interpreter.
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
234 # Read main
ca26a84916cb Add preliminary ECL viewer/interpreter.
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
235 file.seek(main_offset)
ca26a84916cb Add preliminary ECL viewer/interpreter.
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
236 while True:
ca26a84916cb Add preliminary ECL viewer/interpreter.
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
237 time, = unpack('<H', file.read(2))
ca26a84916cb Add preliminary ECL viewer/interpreter.
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
238 if time == 0xffff:
ca26a84916cb Add preliminary ECL viewer/interpreter.
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
239 break
112
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
240
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
241 sub, opcode, size = unpack('<HHH', file.read(6))
18
ca26a84916cb Add preliminary ECL viewer/interpreter.
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
242 data = file.read(size - 8)
112
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
243
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
244 if opcode in cls._main_instructions:
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
245 args = unpack('<%s' % cls._main_instructions[opcode][0], data)
18
ca26a84916cb Add preliminary ECL viewer/interpreter.
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
246 else:
ca26a84916cb Add preliminary ECL viewer/interpreter.
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
247 args = (data,)
112
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
248 logger.warn('unknown main opcode %d', opcode)
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
249
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
250 ecl.main.append((time, sub, opcode, args))
18
ca26a84916cb Add preliminary ECL viewer/interpreter.
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
251
ca26a84916cb Add preliminary ECL viewer/interpreter.
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
252 return ecl
ca26a84916cb Add preliminary ECL viewer/interpreter.
Thibaut Girka <thib@sitedethib.com>
parents:
diff changeset
253
112
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
254
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
255 def write(self, file):
204
88361534c77e Add some documentation (argh, so much left to document!)
Thibaut Girka <thib@sitedethib.com>
parents: 171
diff changeset
256 """Write to an ECL file."""
88361534c77e Add some documentation (argh, so much left to document!)
Thibaut Girka <thib@sitedethib.com>
parents: 171
diff changeset
257
112
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
258 sub_count = len(self.subs)
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
259 sub_offsets = []
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
260 main_offset = 0
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
261
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
262 # Skip header, it will be written later
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
263 file.seek(8+8+4*sub_count)
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
264
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
265 # Write subs
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
266 for sub in self.subs:
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
267 sub_offsets.append(file.tell())
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
268
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
269 instruction_offsets = []
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
270 instruction_datas = []
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
271 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
272 format = self._instructions[opcode][0]
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
273 if format.endswith('s'):
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
274 args = list(args)
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
275 args[-1] = args[-1].encode('shift_jis')
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
276 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
277 format = '<IHHHH%s' % format
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
278 size = calcsize(format)
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
279 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
280 try:
732c64662f87 Minor changes
Thibaut Girka <thib@sitedethib.com>
parents: 112
diff changeset
281 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
282 except struct.error:
732c64662f87 Minor changes
Thibaut Girka <thib@sitedethib.com>
parents: 112
diff changeset
283 logger.error('Failed to assemble opcode %d' % opcode)
732c64662f87 Minor changes
Thibaut Girka <thib@sitedethib.com>
parents: 112
diff changeset
284 raise
112
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
285
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
286 #TODO: clean up this mess
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
287 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
288 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
289 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
290 frame, index = args
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
291 args = frame, instruction_offsets[index] - offset
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
292 format = '<IHHHH%s' % self._instructions[opcode][0]
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
293 size = calcsize(format)
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
294 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
295 elif opcode == 3: # relative_jump_ex
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
296 frame, index, counter_id = args
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
297 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
298 format = '<IHHHH%s' % self._instructions[opcode][0]
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
299 size = calcsize(format)
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
300 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
301 file.write(data)
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
302 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
303
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
304 # Write main
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
305 main_offset = file.tell()
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
306 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
307 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
308 size = calcsize(format)
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
309
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
310 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
311 file.write(b'\xff\xff\x04\x00')
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
312
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
313 # Patch header
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
314 file.seek(0)
e544f9a7966d Add writing support to pytouhou.formats.ecl!
Thibaut Girka <thib@sitedethib.com>
parents: 107
diff changeset
315 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
316