PyTouhou: pytouhou/formats/ecl.py annotate

annotate pytouhou/formats/ecl.py @ 316:f0be7ea62330

Fix a bug with ECL instruction 96, and fix overall ECL handling. The issue with instruction 96 was about death callbacks, being executed on the caller of instruction 96 instead of the dying enemies. This was introduced by changeset 5930b33a0370. Additionnaly, ECL processes are now an attribute of the Enemy, and death/timeout conditions are checked right after the ECL frame, even if the ECL script has already ended, just like in the original game.

author	Thibaut Girka <thib@sitedethib.com>
date	Thu, 29 Mar 2012 21:18:35 +0200
parents	4838e9bab0f9
children	3be4c1078095

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'),
286 4838e9bab0f9 Implement dialogs (MSG files). Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 283 diff changeset	160 8: ('', 'call_msg'),
4838e9bab0f9 Implement dialogs (MSG files). Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 283 diff changeset	161 9: ('', 'wait_msg'),
4838e9bab0f9 Implement dialogs (MSG files). Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 283 diff changeset	162 10: ('II', 'resume_ecl'),
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

Mercurial > touhou

annotate pytouhou/formats/ecl.py @ 316:f0be7ea62330