Mercurial > touhou
view scripts/pytouhou @ 561:bafe6361c0af
Use logging for errors in the pytouhou script.
author | Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> |
---|---|
date | Thu, 05 Jun 2014 18:47:31 +0200 |
parents | 653a9f087673 |
children | 5f7f859a72f9 |
line wrap: on
line source
#!/usr/bin/env python2 # -*- encoding: utf-8 -*- ## ## Copyright (C) 2011 Thibaut Girka <thib@sitedethib.com> ## ## This program is free software; you can redistribute it and/or modify ## it under the terms of the GNU General Public License as published ## by the Free Software Foundation; version 3 only. ## ## This program is distributed in the hope that it will be useful, ## but WITHOUT ANY WARRANTY; without even the implied warranty of ## MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the ## GNU General Public License for more details. ## import argparse import os pathsep = os.path.pathsep default_data = (pathsep.join(('CM.DAT', 'th06*_CM.DAT', '*CM.DAT', '*cm.dat')), pathsep.join(('ST.DAT', 'th6*ST.DAT', '*ST.DAT', '*st.dat')), pathsep.join(('IN.DAT', 'th6*IN.DAT', '*IN.DAT', '*in.dat')), pathsep.join(('MD.DAT', 'th6*MD.DAT', '*MD.DAT', '*md.dat')), pathsep.join(('102h.exe', '102*.exe', '東方紅魔郷.exe', '*.exe'))) parser = argparse.ArgumentParser(description='Libre reimplementation of the Touhou 6 engine.') parser.add_argument('data', metavar='DAT', default=default_data, nargs='*', help='Game’s data files') parser.add_argument('-p', '--path', metavar='DIRECTORY', default='.', help='Game directory path.') parser.add_argument('--debug', action='store_true', help='Set unlimited continues, and perhaps other debug features.') parser.add_argument('--verbosity', metavar='VERBOSITY', choices=['DEBUG', 'INFO', 'WARNING', 'ERROR', 'CRITICAL'], help='Select the wanted logging level.') game_group = parser.add_argument_group('Game options') game_group.add_argument('-s', '--stage', metavar='STAGE', type=int, default=None, help='Stage, 1 to 7 (Extra), nothing means story mode.') game_group.add_argument('-r', '--rank', metavar='RANK', type=int, default=0, help='Rank, from 0 (Easy, default) to 3 (Lunatic).') game_group.add_argument('-c', '--character', metavar='CHARACTER', type=int, default=0, help='Select the character to use, from 0 (ReimuA, default) to 3 (MarisaB).') game_group.add_argument('-b', '--boss-rush', action='store_true', help='Fight only bosses.') game_group.add_argument('--game', metavar='GAME', choices=['EoSD'], default='EoSD', help='Select the game engine to use.') game_group.add_argument('--interface', metavar='INTERFACE', choices=['EoSD', 'Sample'], default='EoSD', help='Select the interface to use.') game_group.add_argument('--hints', metavar='HINTS', default=None, help='Hints file, to display text while playing.') replay_group = parser.add_argument_group('Replay options') replay_group.add_argument('--replay', metavar='REPLAY', help='Select a file to replay.') replay_group.add_argument('--save-replay', metavar='REPLAY', help='Save the upcoming game into a replay file.') replay_group.add_argument('--skip-replay', action='store_true', help='Skip the replay and start to play when it’s finished.') netplay_group = parser.add_argument_group('Netplay options') netplay_group.add_argument('--port', metavar='PORT', type=int, default=0, help='Local port to use.') netplay_group.add_argument('--remote', metavar='REMOTE', default=None, help='Remote address.') netplay_group.add_argument('--friendly-fire', action='store_true', help='Allow friendly-fire during netplay.') graphics_group = parser.add_argument_group('Graphics options') graphics_group.add_argument('--backend', metavar='BACKEND', choices=['opengl', 'sdl'], default=['opengl', 'sdl'], nargs='*', help='Which backend to use (opengl or sdl).') graphics_group.add_argument('--fps-limit', metavar='FPS', default=-1, type=int, help='Set fps limit. A value of 0 disables fps limiting, while a negative value limits to 60 fps if and only if vsync doesn’t work.') graphics_group.add_argument('--no-background', action='store_false', help='Disable background display (huge performance boost on slow systems).') graphics_group.add_argument('--no-particles', action='store_false', help='Disable particles handling (huge performance boost on slow systems).') graphics_group.add_argument('--no-sound', action='store_false', help='Disable music and sound effects.') opengl_group = parser.add_argument_group('OpenGL backend options') opengl_group.add_argument('--gl-flavor', choices=['core', 'es', 'compatibility', 'legacy'], default='compatibility', help='OpenGL profile to use.') opengl_group.add_argument('--gl-version', default=2.1, type=float, help='OpenGL version to use.') double_buffer = opengl_group.add_mutually_exclusive_group() double_buffer.add_argument('--double-buffer', dest='double_buffer', action='store_true', default=None, help='Enable double buffering.') double_buffer.add_argument('--single-buffer', dest='double_buffer', action='store_false', default=None, help='Disable double buffering.') args = parser.parse_args() import sys import logging from pytouhou.utils.helpers import get_logger if args.verbosity is not None: logging.basicConfig(level=getattr(logging, args.verbosity)) logger = get_logger('pytouhou') if args.game == 'EoSD': from pytouhou.games.eosd import EoSDCommon as Common, EoSDGame as Game if args.interface == 'EoSD': from pytouhou.interfaces.eosd import EoSDInterface as Interface elif args.interface == 'Sample': from pytouhou.interfaces.sample import SampleInterface as Interface from pytouhou.lib.sdl import SDL, show_simple_message_box from pytouhou.ui.window import Window from pytouhou.resource.loader import Loader from pytouhou.ui.gamerunner import GameRunner from pytouhou.game.player import GameOver from pytouhou.formats.t6rp import T6RP, Level from pytouhou.utils.random import Random from pytouhou.vm.msgrunner import NextStage from pytouhou.formats.hint import Hint from pytouhou.network import Network from importlib import import_module for backend_name in args.backend: if backend_name == 'opengl': options = { 'flavor': args.gl_flavor, 'version': args.gl_version, 'double-buffer': args.double_buffer, } else: options = {} try: backend = import_module('pytouhou.ui.%s.backend' % backend_name) except ImportError as e: logger.error('Failed to import backend %s: %s', backend_name, e) continue try: backend.init(options) except Exception as e: logger.error('Backend %s failed to initialize: %s', backend_name, e) continue GameRenderer = backend.GameRenderer break else: show_simple_message_box(u'No graphical backend could be used, continuing with a windowless game.') backend = None GameRenderer = None class GameBossRush(Game): def run_iter(self, keystates): for i in range(20): skip = not (self.enemies or self.items or self.lasers or self.bullets or self.cancelled_bullets) if skip: keystates = [k & ~1 for k in keystates] Game.run_iter(self, [0 if i else k | 256 for k in keystates]) if not self.enemies and self.frame % 90 == 0: for player in self.players: if player.power < 128: player.power += 1 if not skip: break def cleanup(self): boss_wait = any(ecl_runner.boss_wait for ecl_runner in self.ecl_runners) if not (self.boss or self.msg_wait or boss_wait): self.enemies = [enemy for enemy in self.enemies if enemy.boss_callback or enemy.frame > 1] for laser in self.lasers: if laser.frame <= 1: laser.removed = True self.lasers = [laser for laser in self.lasers if laser.frame > 1] self.bullets = [bullet for bullet in self.bullets if bullet.frame > 1] Game.cleanup(self) def main(window, path, data, stage_num, rank, character, replay, save_filename, skip_replay, boss_rush, debug, enable_background, enable_particles, hints, port, remote, friendly_fire): resource_loader = Loader(path) try: resource_loader.scan_archives(data) except IOError: show_simple_message_box(u'Some data files were not found, did you forget the -p option?') sys.exit(1) if stage_num is None: story = True stage_num = 1 continues = 3 else: story = False continues = 0 if debug: continues = -1 # Infinite lives if replay: with open(replay, 'rb') as file: replay = T6RP.read(file) rank = replay.rank character = replay.character save_keystates = None if save_filename: save_replay = T6RP() save_replay.rank = rank save_replay.character = character difficulty = 16 if port != 0: if remote: remote_addr, remote_port = remote.split(':') addr = remote_addr, int(remote_port) selected_player = 0 else: addr = None selected_player = 1 prng = Random(0) con = Network(port, addr, selected_player) characters = [1, 3] else: con = None selected_player = 0 characters = [character] if hints: with open(hints, 'rb') as file: hints = Hint.read(file) game_class = GameBossRush if boss_rush else Game common = Common(resource_loader, characters, continues, stage_num - 1) interface = Interface(resource_loader, common.players[0]) #XXX common.interface = interface #XXX renderer = GameRenderer(resource_loader, window) if GameRenderer is not None else None runner = GameRunner(window, renderer, common, resource_loader, skip_replay, con) window.set_runner(runner) while True: first_player = common.players[0] if replay: level = replay.levels[stage_num - 1] if not level: raise Exception prng = Random(level.random_seed) #TODO: apply the replay to the other players. #TODO: see if the stored score is used or if it’s the one from the previous stage. if stage_num != 1 and stage_num - 2 in replay.levels: previous_level = replay.levels[stage_num - 1] first_player.score = previous_level.score first_player.effective_score = previous_level.score first_player.points = level.point_items first_player.power = level.power first_player.lives = level.lives first_player.bombs = level.bombs difficulty = level.difficulty elif port == 0: prng = Random() if save_filename: if not replay: save_replay.levels[stage_num - 1] = level = Level() level.random_seed = prng.seed level.score = first_player.score level.point_items = first_player.points level.power = first_player.power level.lives = first_player.lives level.bombs = first_player.bombs level.difficulty = difficulty save_keystates = [] hints_stage = hints.stages[stage_num - 1] if hints else None game = game_class(resource_loader, stage_num, rank, difficulty, common, prng, hints_stage, friendly_fire) if not enable_particles: def new_particle(pos, anim, amp, number=1, reverse=False, duration=24): pass game.new_particle = new_particle background = game.background if enable_background else None runner.load_game(game, background, game.std.bgms, replay, save_keystates) try: # Main loop window.run() break except NextStage: if not story or stage_num == (7 if boss_rush else 6 if rank > 0 else 5): break stage_num += 1 except GameOver: show_simple_message_box(u'Game over!') break finally: if save_filename: last_key = -1 for time, key in enumerate(save_keystates): if key != last_key: level.keys.append((time, key, 0)) last_key = key window.set_runner(None) if save_filename: with open(save_filename, 'wb+') as file: save_replay.write(file) with SDL(sound=args.no_sound): window = Window(backend, fps_limit=args.fps_limit) main(window, args.path, tuple(args.data), args.stage, args.rank, args.character, args.replay, args.save_replay, args.skip_replay, args.boss_rush, args.debug, args.no_background, args.no_particles, args.hints, args.port, args.remote, args.friendly_fire) import gc gc.collect()