PyTouhou: pytouhou/formats/exe.py annotate

annotate pytouhou/formats/exe.py @ 235:e59bd7979ddc

Do a little cleanup, and fix PCB SHT usage.

author	Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
date	Sat, 31 Dec 2011 16:31:10 +0100
parents	1c24a6d93c1b
children	741860192b56

rev	line source
229 5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	1 # -- encoding: utf-8 --
5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	2 ##
5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	3 ## Copyright (C) 2011 Thibaut Girka <thib@sitedethib.com>
5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	4 ## Copyright (C) 2011 Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	5 ##
5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	6 ## This program is free software; you can redistribute it and/or modify
5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	7 ## it under the terms of the GNU General Public License as published
5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	8 ## by the Free Software Foundation; version 3 only.
5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	9 ##
5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	10 ## This program is distributed in the hope that it will be useful,
5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	11 ## but WITHOUT ANY WARRANTY; without even the implied warranty of
5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	12 ## MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	13 ## GNU General Public License for more details.
5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	14 ##
5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	15
235 e59bd7979ddc Do a little cleanup, and fix PCB SHT usage. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 230 diff changeset	16 from copy import copy
229 5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	17 from struct import Struct, unpack
235 e59bd7979ddc Do a little cleanup, and fix PCB SHT usage. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 230 diff changeset	18
229 5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	19 from pytouhou.utils.pe import PEFile
5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	20
5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	21 from pytouhou.utils.helpers import get_logger
5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	22
5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	23 logger = get_logger(__name__)
5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	24
5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	25
5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	26 SQ2 = 2. ** 0.5 / 2.
5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	27
5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	28
5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	29 class Shot(object):
5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	30 def __init__(self):
5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	31 self.interval = 0
5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	32 self.delay = 0
5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	33 self.pos = (0., 0.)
5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	34 self.hitbox = (0., 0.)
5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	35 self.angle = 0.
5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	36 self.speed = 0.
5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	37 self.damage = 0
5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	38 self.orb = 0
230 1c24a6d93c1b Improve find_character_defs, clean up a bit, and disable attack types 2 and 3 for now Thibaut Girka <thib@sitedethib.com> parents: 229 diff changeset	39 self.type = 0
229 5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	40 self.sprite = 0
5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	41 self.unknown1 = None
5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	42
5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	43
5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	44 class SHT(object):
5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	45 def __init__(self):
5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	46 #self.unknown1 = None
5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	47 #self.bombs = 0.
5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	48 #self.unknown2 = None
5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	49 self.hitbox = 4.
5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	50 self.graze_hitbox = 42.
5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	51 self.autocollection_speed = 8.
5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	52 self.item_hitbox = 38.
5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	53 # No percentage_of_cherry_loss_on_die
5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	54 self.point_of_collection = 128 #TODO: find the real default.
5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	55 self.horizontal_vertical_speed = 0.
5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	56 self.horizontal_vertical_focused_speed = 0.
5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	57 self.diagonal_speed = 0.
5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	58 self.diagonal_focused_speed = 0.
5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	59 self.shots = {}
5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	60
5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	61
5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	62 @classmethod
230 1c24a6d93c1b Improve find_character_defs, clean up a bit, and disable attack types 2 and 3 for now Thibaut Girka <thib@sitedethib.com> parents: 229 diff changeset	63 def find_character_defs(cls, pe_file):
1c24a6d93c1b Improve find_character_defs, clean up a bit, and disable attack types 2 and 3 for now Thibaut Girka <thib@sitedethib.com> parents: 229 diff changeset	64 """Generator returning the possible VA of character definition blocks.
1c24a6d93c1b Improve find_character_defs, clean up a bit, and disable attack types 2 and 3 for now Thibaut Girka <thib@sitedethib.com> parents: 229 diff changeset	65
1c24a6d93c1b Improve find_character_defs, clean up a bit, and disable attack types 2 and 3 for now Thibaut Girka <thib@sitedethib.com> parents: 229 diff changeset	66 Based on knowledge of the structure, it tries to find valid definition blocks
1c24a6d93c1b Improve find_character_defs, clean up a bit, and disable attack types 2 and 3 for now Thibaut Girka <thib@sitedethib.com> parents: 229 diff changeset	67 without embedding any copyrighted material or hard-coded offsets that would
1c24a6d93c1b Improve find_character_defs, clean up a bit, and disable attack types 2 and 3 for now Thibaut Girka <thib@sitedethib.com> parents: 229 diff changeset	68 only be useful for a specific build of the game.
1c24a6d93c1b Improve find_character_defs, clean up a bit, and disable attack types 2 and 3 for now Thibaut Girka <thib@sitedethib.com> parents: 229 diff changeset	69 """
1c24a6d93c1b Improve find_character_defs, clean up a bit, and disable attack types 2 and 3 for now Thibaut Girka <thib@sitedethib.com> parents: 229 diff changeset	70
229 5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	71 format = Struct('<4f2I')
230 1c24a6d93c1b Improve find_character_defs, clean up a bit, and disable attack types 2 and 3 for now Thibaut Girka <thib@sitedethib.com> parents: 229 diff changeset	72 data_section = [section for section in pe_file.sections
1c24a6d93c1b Improve find_character_defs, clean up a bit, and disable attack types 2 and 3 for now Thibaut Girka <thib@sitedethib.com> parents: 229 diff changeset	73 if section.Name.startswith('.data')][0]
1c24a6d93c1b Improve find_character_defs, clean up a bit, and disable attack types 2 and 3 for now Thibaut Girka <thib@sitedethib.com> parents: 229 diff changeset	74 text_section = [section for section in pe_file.sections
1c24a6d93c1b Improve find_character_defs, clean up a bit, and disable attack types 2 and 3 for now Thibaut Girka <thib@sitedethib.com> parents: 229 diff changeset	75 if section.Name.startswith('.text')][0]
229 5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	76 data_va = pe_file.image_base + data_section.VirtualAddress
230 1c24a6d93c1b Improve find_character_defs, clean up a bit, and disable attack types 2 and 3 for now Thibaut Girka <thib@sitedethib.com> parents: 229 diff changeset	77 data_size = data_section.SizeOfRawData
229 5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	78 text_va = pe_file.image_base + text_section.VirtualAddress
230 1c24a6d93c1b Improve find_character_defs, clean up a bit, and disable attack types 2 and 3 for now Thibaut Girka <thib@sitedethib.com> parents: 229 diff changeset	79 text_size = text_section.SizeOfRawData
229 5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	80
230 1c24a6d93c1b Improve find_character_defs, clean up a bit, and disable attack types 2 and 3 for now Thibaut Girka <thib@sitedethib.com> parents: 229 diff changeset	81 # Search the whole data segment for 4 successive character definitions
1c24a6d93c1b Improve find_character_defs, clean up a bit, and disable attack types 2 and 3 for now Thibaut Girka <thib@sitedethib.com> parents: 229 diff changeset	82 for addr in xrange(data_va, data_va + data_size, 4):
229 5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	83 for character_id in xrange(4):
5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	84 pe_file.seek_to_va(addr + character_id * 24)
230 1c24a6d93c1b Improve find_character_defs, clean up a bit, and disable attack types 2 and 3 for now Thibaut Girka <thib@sitedethib.com> parents: 229 diff changeset	85 (speed1, speed2, speed3, speed4,
1c24a6d93c1b Improve find_character_defs, clean up a bit, and disable attack types 2 and 3 for now Thibaut Girka <thib@sitedethib.com> parents: 229 diff changeset	86 ptr1, ptr2) = format.unpack(pe_file.file.read(format.size))
229 5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	87
230 1c24a6d93c1b Improve find_character_defs, clean up a bit, and disable attack types 2 and 3 for now Thibaut Girka <thib@sitedethib.com> parents: 229 diff changeset	88 # Check whether the character's speed make sense,
1c24a6d93c1b Improve find_character_defs, clean up a bit, and disable attack types 2 and 3 for now Thibaut Girka <thib@sitedethib.com> parents: 229 diff changeset	89 # and whether the function pointers point to valid addresses
1c24a6d93c1b Improve find_character_defs, clean up a bit, and disable attack types 2 and 3 for now Thibaut Girka <thib@sitedethib.com> parents: 229 diff changeset	90 if not (all(0. < x < 10. for x in (speed1, speed2, speed3, speed4))
229 5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	91 and speed2 <= speed1
230 1c24a6d93c1b Improve find_character_defs, clean up a bit, and disable attack types 2 and 3 for now Thibaut Girka <thib@sitedethib.com> parents: 229 diff changeset	92 and 0 <= ptr1 - text_va < text_size - 8
1c24a6d93c1b Improve find_character_defs, clean up a bit, and disable attack types 2 and 3 for now Thibaut Girka <thib@sitedethib.com> parents: 229 diff changeset	93 and 0 <= ptr2 - text_va < text_size - 8):
229 5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	94 break
5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	95
230 1c24a6d93c1b Improve find_character_defs, clean up a bit, and disable attack types 2 and 3 for now Thibaut Girka <thib@sitedethib.com> parents: 229 diff changeset	96 # So far, this character definition seems to be valid.
1c24a6d93c1b Improve find_character_defs, clean up a bit, and disable attack types 2 and 3 for now Thibaut Girka <thib@sitedethib.com> parents: 229 diff changeset	97 # Now, make sure the shoot function wrappers pass valid addresses
229 5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	98 pe_file.seek_to_va(ptr1 + 4)
230 1c24a6d93c1b Improve find_character_defs, clean up a bit, and disable attack types 2 and 3 for now Thibaut Girka <thib@sitedethib.com> parents: 229 diff changeset	99 shtptr1, = unpack('<I', pe_file.file.read(4))
229 5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	100 pe_file.seek_to_va(ptr2 + 4)
230 1c24a6d93c1b Improve find_character_defs, clean up a bit, and disable attack types 2 and 3 for now Thibaut Girka <thib@sitedethib.com> parents: 229 diff changeset	101 shtptr2, = unpack('<I', pe_file.file.read(4))
1c24a6d93c1b Improve find_character_defs, clean up a bit, and disable attack types 2 and 3 for now Thibaut Girka <thib@sitedethib.com> parents: 229 diff changeset	102 if not (0 <= shtptr1 - data_va < data_size - 12
1c24a6d93c1b Improve find_character_defs, clean up a bit, and disable attack types 2 and 3 for now Thibaut Girka <thib@sitedethib.com> parents: 229 diff changeset	103 and 0 <= shtptr2 - data_va < data_size - 12):
1c24a6d93c1b Improve find_character_defs, clean up a bit, and disable attack types 2 and 3 for now Thibaut Girka <thib@sitedethib.com> parents: 229 diff changeset	104 break
229 5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	105
230 1c24a6d93c1b Improve find_character_defs, clean up a bit, and disable attack types 2 and 3 for now Thibaut Girka <thib@sitedethib.com> parents: 229 diff changeset	106 # It is unlikely this character record is not valid, but
1c24a6d93c1b Improve find_character_defs, clean up a bit, and disable attack types 2 and 3 for now Thibaut Girka <thib@sitedethib.com> parents: 229 diff changeset	107 # just to be sure, let's check the first SHT definition.
1c24a6d93c1b Improve find_character_defs, clean up a bit, and disable attack types 2 and 3 for now Thibaut Girka <thib@sitedethib.com> parents: 229 diff changeset	108 pe_file.seek_to_va(shtptr1)
1c24a6d93c1b Improve find_character_defs, clean up a bit, and disable attack types 2 and 3 for now Thibaut Girka <thib@sitedethib.com> parents: 229 diff changeset	109 nb_shots, power, shotsptr = unpack('<III', pe_file.file.read(12))
1c24a6d93c1b Improve find_character_defs, clean up a bit, and disable attack types 2 and 3 for now Thibaut Girka <thib@sitedethib.com> parents: 229 diff changeset	110 if not (0 < nb_shots <= 1000
1c24a6d93c1b Improve find_character_defs, clean up a bit, and disable attack types 2 and 3 for now Thibaut Girka <thib@sitedethib.com> parents: 229 diff changeset	111 and 0 <= power < 1000
1c24a6d93c1b Improve find_character_defs, clean up a bit, and disable attack types 2 and 3 for now Thibaut Girka <thib@sitedethib.com> parents: 229 diff changeset	112 and 0 <= shotsptr - data_va < data_size - 36*nb_shots):
229 5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	113 break
230 1c24a6d93c1b Improve find_character_defs, clean up a bit, and disable attack types 2 and 3 for now Thibaut Girka <thib@sitedethib.com> parents: 229 diff changeset	114
1c24a6d93c1b Improve find_character_defs, clean up a bit, and disable attack types 2 and 3 for now Thibaut Girka <thib@sitedethib.com> parents: 229 diff changeset	115 else:
1c24a6d93c1b Improve find_character_defs, clean up a bit, and disable attack types 2 and 3 for now Thibaut Girka <thib@sitedethib.com> parents: 229 diff changeset	116 # XXX: Obscure python feature! This only gets executed if the
1c24a6d93c1b Improve find_character_defs, clean up a bit, and disable attack types 2 and 3 for now Thibaut Girka <thib@sitedethib.com> parents: 229 diff changeset	117 # XXX: loop ended without a break statement.
1c24a6d93c1b Improve find_character_defs, clean up a bit, and disable attack types 2 and 3 for now Thibaut Girka <thib@sitedethib.com> parents: 229 diff changeset	118 # In our case, it's only executed if all the 4 character
1c24a6d93c1b Improve find_character_defs, clean up a bit, and disable attack types 2 and 3 for now Thibaut Girka <thib@sitedethib.com> parents: 229 diff changeset	119 # definitions are considered valid.
229 5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	120 yield addr
5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	121
5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	122
5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	123 @classmethod
5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	124 def read(cls, file):
5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	125 pe_file = PEFile(file)
5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	126
230 1c24a6d93c1b Improve find_character_defs, clean up a bit, and disable attack types 2 and 3 for now Thibaut Girka <thib@sitedethib.com> parents: 229 diff changeset	127 character_records_va = list(cls.find_character_defs(pe_file))[0]
229 5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	128
5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	129 characters = []
235 e59bd7979ddc Do a little cleanup, and fix PCB SHT usage. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 230 diff changeset	130 shots_offsets = {}
229 5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	131 for character in xrange(4):
5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	132 sht = cls()
5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	133
5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	134 pe_file.seek_to_va(character_records_va + 64character)
5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	135
5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	136 data = unpack('<4f2I', file.read(6*4))
5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	137 (speed, speed_focused, speed_unknown1, speed_unknown2,
5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	138 shots_func_offset, shots_func_offset_focused) = data
5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	139
5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	140 sht.horizontal_vertical_speed = speed
5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	141 sht.horizontal_vertical_focused_speed = speed_focused
5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	142 sht.diagonal_speed = speed * SQ2
5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	143 sht.diagonal_focused_speed = speed_focused * SQ2
5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	144
235 e59bd7979ddc Do a little cleanup, and fix PCB SHT usage. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 230 diff changeset	145 # Characters might have different shot types whether they are
e59bd7979ddc Do a little cleanup, and fix PCB SHT usage. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 230 diff changeset	146 # focused or not, but properties read earlier apply to both modes.
e59bd7979ddc Do a little cleanup, and fix PCB SHT usage. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 230 diff changeset	147 focused_sht = copy(sht)
e59bd7979ddc Do a little cleanup, and fix PCB SHT usage. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 230 diff changeset	148 characters.append((sht, focused_sht))
229 5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	149
235 e59bd7979ddc Do a little cleanup, and fix PCB SHT usage. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 230 diff changeset	150 for sht, func_offset in ((sht, shots_func_offset), (focused_sht, shots_func_offset_focused)):
e59bd7979ddc Do a little cleanup, and fix PCB SHT usage. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 230 diff changeset	151 # Read from “push” operand
e59bd7979ddc Do a little cleanup, and fix PCB SHT usage. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 230 diff changeset	152 pe_file.seek_to_va(func_offset + 4)
e59bd7979ddc Do a little cleanup, and fix PCB SHT usage. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 230 diff changeset	153 offset, = unpack('<I', file.read(4))
e59bd7979ddc Do a little cleanup, and fix PCB SHT usage. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 230 diff changeset	154 if offset not in shots_offsets:
e59bd7979ddc Do a little cleanup, and fix PCB SHT usage. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 230 diff changeset	155 shots_offsets[offset] = []
e59bd7979ddc Do a little cleanup, and fix PCB SHT usage. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 230 diff changeset	156 shots_offsets[offset].append(sht)
229 5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	157
5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	158 character = 0
235 e59bd7979ddc Do a little cleanup, and fix PCB SHT usage. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 230 diff changeset	159 for shots_offset, shts in shots_offsets.iteritems():
229 5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	160 pe_file.seek_to_va(shots_offset)
5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	161
5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	162 level_count = 9
5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	163 levels = []
5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	164 for i in xrange(level_count):
5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	165 shots_count, power, offset = unpack('<III', file.read(3*4))
5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	166 levels.append((shots_count, power, offset))
5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	167
235 e59bd7979ddc Do a little cleanup, and fix PCB SHT usage. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 230 diff changeset	168 shots = {}
229 5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	169
5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	170 for shots_count, power, offset in levels:
235 e59bd7979ddc Do a little cleanup, and fix PCB SHT usage. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 230 diff changeset	171 shots[power] = []
229 5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	172 pe_file.seek_to_va(offset)
5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	173
5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	174 for i in xrange(shots_count):
5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	175 shot = Shot()
5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	176
5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	177 data = unpack('<HH6fHBBhh', file.read(36))
5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	178 (shot.interval, shot.delay, x, y, hitbox_x, hitbox_y,
230 1c24a6d93c1b Improve find_character_defs, clean up a bit, and disable attack types 2 and 3 for now Thibaut Girka <thib@sitedethib.com> parents: 229 diff changeset	179 shot.angle, shot.speed, shot.damage, shot.orb, shot.type,
229 5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	180 shot.sprite, shot.unknown1) = data
5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	181
5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	182 shot.pos = (x, y)
5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	183 shot.hitbox = (hitbox_x, hitbox_y)
5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	184
235 e59bd7979ddc Do a little cleanup, and fix PCB SHT usage. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 230 diff changeset	185 shots[power].append(shot)
e59bd7979ddc Do a little cleanup, and fix PCB SHT usage. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 230 diff changeset	186
e59bd7979ddc Do a little cleanup, and fix PCB SHT usage. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 230 diff changeset	187 for sht in shts:
e59bd7979ddc Do a little cleanup, and fix PCB SHT usage. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 230 diff changeset	188 sht.shots = shots
229 5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	189
5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	190 character += 1
5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	191
5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	192
5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	193 return characters
5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	194

Mercurial > touhou

annotate pytouhou/formats/exe.py @ 235:e59bd7979ddc