Mercurial > touhou
diff pytouhou/game/enemymanager.py @ 18:ca26a84916cb
Add preliminary ECL viewer/interpreter.
author | Thibaut Girka <thib@sitedethib.com> |
---|---|
date | Tue, 09 Aug 2011 11:40:48 +0200 |
parents | |
children | 6ebf9539c077 |
line wrap: on
line diff
new file mode 100644 --- /dev/null +++ b/pytouhou/game/enemymanager.py @@ -0,0 +1,124 @@ +from itertools import chain +from io import BytesIO +import os +from struct import unpack, pack +from pytouhou.game.sprite import Sprite +from math import cos, sin + + +class Enemy(object): + def __init__(self, pos, life, _type, script, anms): + self.anms = tuple(anms) + self.anm = None + self.script = list(script) + self.x, self.y = pos + self.life = life + self.type = _type + self.frame = 0 + self.sprite = None + + self.angle = 0. + self.speed = 0. + self.rotation_speed = 0. + self.acceleration = 0. + + + def update(self, frame): + if not self.script: + return True + if self.script[0][0] == self.frame: + for instr_type, rank_mask, param_mask, args in self.script.pop(0)[1]: + if instr_type == 1: # delete + return False + elif instr_type == 97: # set_enemy_sprite + script_index, = unpack('<I', args) + if script_index in self.anms[0].scripts: + self.sprite = Sprite(self.anms[0], script_index) + self.anm = self.anms[0] + else: + self.sprite = Sprite(self.anms[1], script_index) + self.anm = self.anms[1] + elif instr_type == 45: # set_angle_speed + self.angle, self.speed = unpack('<ff', args) + elif instr_type == 46: # set_angle + self.rotation_speed, = unpack('<f', args) + elif instr_type == 47: # set_speed + self.speed, = unpack('<f', args) + elif instr_type == 48: # set_acceleration + self.acceleration, = unpack('<f', args) + if self.sprite: + self.sprite.update() + + self.speed += self.acceleration #TODO: units? Execution order? + self.angle += self.rotation_speed #TODO: units? Execution order? + + dx, dy = cos(self.angle) * self.speed, sin(self.angle) * self.speed + if self.type == 2: + self.x -= dx + else: + self.x += dx + self.y += dy + + self.frame += 1 + return True + + + +class EnemyManager(object): + def __init__(self, stage, anims, ecl): + self.stage = stage + self.anims = tuple(anims) + self.main = [] + self.subs = {} + self.objects_by_texture = {} + self.enemies = [] + + # Populate main + for frame, sub, instr_type, args in ecl.main: + if not self.main or self.main[-1][0] < frame: + self.main.append((frame, [(sub, instr_type, args)])) + elif self.main[-1][0] == frame: + self.main[-1][1].append((sub, instr_type, args)) + + + # Populate subs + for i, sub in enumerate(ecl.subs): + for frame, instr_type, rank_mask, param_mask, args in sub: + if i not in self.subs: + self.subs[i] = [] + if not self.subs[i] or self.subs[i][-1][0] < frame: + self.subs[i].append((frame, [(instr_type, rank_mask, param_mask, args)])) + elif self.subs[i][-1][0] == frame: + self.subs[i][-1][1].append((instr_type, rank_mask, param_mask, args)) + + + def update(self, frame): + if self.main and self.main[0][0] == frame: + for sub, instr_type, args in self.main.pop(0)[1]: + if instr_type in (0, 2): # Normal/mirrored enemy + x, y, z, life, unknown1, unknown2, unknown3 = args + self.enemies.append(Enemy((x, y), life, instr_type, self.subs[sub], self.anims)) + + # Update enemies + for enemy in tuple(self.enemies): + if not enemy.update(frame): + self.enemies.remove(enemy) + continue + + # Add enemies to vertices/uvs + self.objects_by_texture = {} + for enemy in self.enemies: + ox, oy = enemy.x, enemy.y + if enemy.sprite: + key = enemy.anm.first_name, enemy.anm.secondary_name + if not key in self.objects_by_texture: + self.objects_by_texture[key] = (0, [], []) + vertices = tuple((x + ox, y + oy, z) for x, y, z in enemy.sprite._vertices) + self.objects_by_texture[key][2].extend(enemy.sprite._uvs) + self.objects_by_texture[key][1].extend(vertices) + for key, (nb_vertices, vertices, uvs) in self.objects_by_texture.items(): + nb_vertices = len(vertices) + vertices = pack('f' * (3 * nb_vertices), *chain(*vertices)) + uvs = pack('f' * (2 * nb_vertices), *chain(*uvs)) + self.objects_by_texture[key] = (nb_vertices, vertices, uvs) +