PyTouhou: pytouhou/resource/loader.py annotate

annotate pytouhou/resource/loader.py @ 429:40d5f3083ebc

Implement PCB’s ANM2 format and vm.

author	Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
date	Sat, 03 Aug 2013 15:48:57 +0200
parents	6d7dbcb31d95
children	c9433188ffdb

rev	line source
285 2100276c289d Document some AnmWrapper related functions. Thibaut Girka <thib@sitedethib.com> parents: 282 diff changeset	1 # -- encoding: utf-8 --
2100276c289d Document some AnmWrapper related functions. Thibaut Girka <thib@sitedethib.com> parents: 282 diff changeset	2 ##
2100276c289d Document some AnmWrapper related functions. Thibaut Girka <thib@sitedethib.com> parents: 282 diff changeset	3 ## Copyright (C) 2012 Thibaut Girka <thib@sitedethib.com>
2100276c289d Document some AnmWrapper related functions. Thibaut Girka <thib@sitedethib.com> parents: 282 diff changeset	4 ##
2100276c289d Document some AnmWrapper related functions. Thibaut Girka <thib@sitedethib.com> parents: 282 diff changeset	5 ## This program is free software; you can redistribute it and/or modify
2100276c289d Document some AnmWrapper related functions. Thibaut Girka <thib@sitedethib.com> parents: 282 diff changeset	6 ## it under the terms of the GNU General Public License as published
2100276c289d Document some AnmWrapper related functions. Thibaut Girka <thib@sitedethib.com> parents: 282 diff changeset	7 ## by the Free Software Foundation; version 3 only.
2100276c289d Document some AnmWrapper related functions. Thibaut Girka <thib@sitedethib.com> parents: 282 diff changeset	8 ##
2100276c289d Document some AnmWrapper related functions. Thibaut Girka <thib@sitedethib.com> parents: 282 diff changeset	9 ## This program is distributed in the hope that it will be useful,
2100276c289d Document some AnmWrapper related functions. Thibaut Girka <thib@sitedethib.com> parents: 282 diff changeset	10 ## but WITHOUT ANY WARRANTY; without even the implied warranty of
2100276c289d Document some AnmWrapper related functions. Thibaut Girka <thib@sitedethib.com> parents: 282 diff changeset	11 ## MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
2100276c289d Document some AnmWrapper related functions. Thibaut Girka <thib@sitedethib.com> parents: 282 diff changeset	12 ## GNU General Public License for more details.
2100276c289d Document some AnmWrapper related functions. Thibaut Girka <thib@sitedethib.com> parents: 282 diff changeset	13 ##
2100276c289d Document some AnmWrapper related functions. Thibaut Girka <thib@sitedethib.com> parents: 282 diff changeset	14
231 c417bb6c98bf Search for 102h.exe in the game directory instead of the current directory. Thibaut Girka <thib@sitedethib.com> parents: 229 diff changeset	15 import os
262 8fa660da5f0c Automatically search data files using different names. Thibaut Girka <thib@sitedethib.com> parents: 231 diff changeset	16 from glob import glob
8fa660da5f0c Automatically search data files using different names. Thibaut Girka <thib@sitedethib.com> parents: 231 diff changeset	17 from itertools import chain
97 ac2e5e1c2c3c Refactor \o/ Thibaut Girka <thib@sitedethib.com> parents: diff changeset	18 from io import BytesIO
ac2e5e1c2c3c Refactor \o/ Thibaut Girka <thib@sitedethib.com> parents: diff changeset	19
429 40d5f3083ebc Implement PCB’s ANM2 format and vm. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 413 diff changeset	20 from pytouhou.formats import WrongFormatError
97 ac2e5e1c2c3c Refactor \o/ Thibaut Girka <thib@sitedethib.com> parents: diff changeset	21 from pytouhou.formats.pbg3 import PBG3
ac2e5e1c2c3c Refactor \o/ Thibaut Girka <thib@sitedethib.com> parents: diff changeset	22 from pytouhou.formats.std import Stage
ac2e5e1c2c3c Refactor \o/ Thibaut Girka <thib@sitedethib.com> parents: diff changeset	23 from pytouhou.formats.ecl import ECL
282 dbb1a86c0235 Rename Animations to ANM0 and prepare AnmWrapper for dialogs and interface. Thibaut Girka <thib@sitedethib.com> parents: 263 diff changeset	24 from pytouhou.formats.anm0 import ANM0
133 2cad2e84a49e Add reading support for the MSG format. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 97 diff changeset	25 from pytouhou.formats.msg import MSG
220 0595315d3880 Fix SHT handling; change a few things to be closer to ZUN’s mind; and first stub of PCB support. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 133 diff changeset	26 from pytouhou.formats.sht import SHT
298 92a6fd2632f1 Improve heuristic to filter out non-game exes (like custom.exe). Thibaut Girka <thib@sitedethib.com> parents: 297 diff changeset	27 from pytouhou.formats.exe import SHT as EoSDSHT, InvalidExeException
321 61adb5453e46 Implement music playback. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 298 diff changeset	28 from pytouhou.formats.music import Track
325 cddfd3cb4797 Add music support for >PCB. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 321 diff changeset	29 from pytouhou.formats.fmt import FMT
97 ac2e5e1c2c3c Refactor \o/ Thibaut Girka <thib@sitedethib.com> parents: diff changeset	30
ac2e5e1c2c3c Refactor \o/ Thibaut Girka <thib@sitedethib.com> parents: diff changeset	31
ac2e5e1c2c3c Refactor \o/ Thibaut Girka <thib@sitedethib.com> parents: diff changeset	32 from pytouhou.resource.anmwrapper import AnmWrapper
ac2e5e1c2c3c Refactor \o/ Thibaut Girka <thib@sitedethib.com> parents: diff changeset	33
298 92a6fd2632f1 Improve heuristic to filter out non-game exes (like custom.exe). Thibaut Girka <thib@sitedethib.com> parents: 297 diff changeset	34 from pytouhou.utils.helpers import get_logger
92a6fd2632f1 Improve heuristic to filter out non-game exes (like custom.exe). Thibaut Girka <thib@sitedethib.com> parents: 297 diff changeset	35
92a6fd2632f1 Improve heuristic to filter out non-game exes (like custom.exe). Thibaut Girka <thib@sitedethib.com> parents: 297 diff changeset	36 logger = get_logger(__name__)
92a6fd2632f1 Improve heuristic to filter out non-game exes (like custom.exe). Thibaut Girka <thib@sitedethib.com> parents: 297 diff changeset	37
92a6fd2632f1 Improve heuristic to filter out non-game exes (like custom.exe). Thibaut Girka <thib@sitedethib.com> parents: 297 diff changeset	38
97 ac2e5e1c2c3c Refactor \o/ Thibaut Girka <thib@sitedethib.com> parents: diff changeset	39
263 ac677dd0ffe0 Add support for reading from a directory instead of a PBG3 (for debugging purposes). Thibaut Girka <thib@sitedethib.com> parents: 262 diff changeset	40 class Directory(object):
ac677dd0ffe0 Add support for reading from a directory instead of a PBG3 (for debugging purposes). Thibaut Girka <thib@sitedethib.com> parents: 262 diff changeset	41 def __init__(self, path):
ac677dd0ffe0 Add support for reading from a directory instead of a PBG3 (for debugging purposes). Thibaut Girka <thib@sitedethib.com> parents: 262 diff changeset	42 self.path = path
ac677dd0ffe0 Add support for reading from a directory instead of a PBG3 (for debugging purposes). Thibaut Girka <thib@sitedethib.com> parents: 262 diff changeset	43
ac677dd0ffe0 Add support for reading from a directory instead of a PBG3 (for debugging purposes). Thibaut Girka <thib@sitedethib.com> parents: 262 diff changeset	44
ac677dd0ffe0 Add support for reading from a directory instead of a PBG3 (for debugging purposes). Thibaut Girka <thib@sitedethib.com> parents: 262 diff changeset	45 def __enter__(self):
ac677dd0ffe0 Add support for reading from a directory instead of a PBG3 (for debugging purposes). Thibaut Girka <thib@sitedethib.com> parents: 262 diff changeset	46 return self
ac677dd0ffe0 Add support for reading from a directory instead of a PBG3 (for debugging purposes). Thibaut Girka <thib@sitedethib.com> parents: 262 diff changeset	47
ac677dd0ffe0 Add support for reading from a directory instead of a PBG3 (for debugging purposes). Thibaut Girka <thib@sitedethib.com> parents: 262 diff changeset	48
ac677dd0ffe0 Add support for reading from a directory instead of a PBG3 (for debugging purposes). Thibaut Girka <thib@sitedethib.com> parents: 262 diff changeset	49 def __exit__(self, type, value, traceback):
ac677dd0ffe0 Add support for reading from a directory instead of a PBG3 (for debugging purposes). Thibaut Girka <thib@sitedethib.com> parents: 262 diff changeset	50 return False
ac677dd0ffe0 Add support for reading from a directory instead of a PBG3 (for debugging purposes). Thibaut Girka <thib@sitedethib.com> parents: 262 diff changeset	51
ac677dd0ffe0 Add support for reading from a directory instead of a PBG3 (for debugging purposes). Thibaut Girka <thib@sitedethib.com> parents: 262 diff changeset	52
ac677dd0ffe0 Add support for reading from a directory instead of a PBG3 (for debugging purposes). Thibaut Girka <thib@sitedethib.com> parents: 262 diff changeset	53 def list_files(self):
ac677dd0ffe0 Add support for reading from a directory instead of a PBG3 (for debugging purposes). Thibaut Girka <thib@sitedethib.com> parents: 262 diff changeset	54 file_list = []
ac677dd0ffe0 Add support for reading from a directory instead of a PBG3 (for debugging purposes). Thibaut Girka <thib@sitedethib.com> parents: 262 diff changeset	55 for path in os.listdir(self.path):
ac677dd0ffe0 Add support for reading from a directory instead of a PBG3 (for debugging purposes). Thibaut Girka <thib@sitedethib.com> parents: 262 diff changeset	56 if os.path.isfile(os.path.join(self.path, path)):
ac677dd0ffe0 Add support for reading from a directory instead of a PBG3 (for debugging purposes). Thibaut Girka <thib@sitedethib.com> parents: 262 diff changeset	57 file_list.append(path)
ac677dd0ffe0 Add support for reading from a directory instead of a PBG3 (for debugging purposes). Thibaut Girka <thib@sitedethib.com> parents: 262 diff changeset	58 return file_list
ac677dd0ffe0 Add support for reading from a directory instead of a PBG3 (for debugging purposes). Thibaut Girka <thib@sitedethib.com> parents: 262 diff changeset	59
ac677dd0ffe0 Add support for reading from a directory instead of a PBG3 (for debugging purposes). Thibaut Girka <thib@sitedethib.com> parents: 262 diff changeset	60
ac677dd0ffe0 Add support for reading from a directory instead of a PBG3 (for debugging purposes). Thibaut Girka <thib@sitedethib.com> parents: 262 diff changeset	61 def extract(self, name):
ac677dd0ffe0 Add support for reading from a directory instead of a PBG3 (for debugging purposes). Thibaut Girka <thib@sitedethib.com> parents: 262 diff changeset	62 with open(os.path.join(self.path, str(name)), 'rb') as file:
ac677dd0ffe0 Add support for reading from a directory instead of a PBG3 (for debugging purposes). Thibaut Girka <thib@sitedethib.com> parents: 262 diff changeset	63 contents = file.read()
ac677dd0ffe0 Add support for reading from a directory instead of a PBG3 (for debugging purposes). Thibaut Girka <thib@sitedethib.com> parents: 262 diff changeset	64 return contents
ac677dd0ffe0 Add support for reading from a directory instead of a PBG3 (for debugging purposes). Thibaut Girka <thib@sitedethib.com> parents: 262 diff changeset	65
ac677dd0ffe0 Add support for reading from a directory instead of a PBG3 (for debugging purposes). Thibaut Girka <thib@sitedethib.com> parents: 262 diff changeset	66
ac677dd0ffe0 Add support for reading from a directory instead of a PBG3 (for debugging purposes). Thibaut Girka <thib@sitedethib.com> parents: 262 diff changeset	67
97 ac2e5e1c2c3c Refactor \o/ Thibaut Girka <thib@sitedethib.com> parents: diff changeset	68 class ArchiveDescription(object):
ac2e5e1c2c3c Refactor \o/ Thibaut Girka <thib@sitedethib.com> parents: diff changeset	69 _formats = {'PBG3': PBG3}
ac2e5e1c2c3c Refactor \o/ Thibaut Girka <thib@sitedethib.com> parents: diff changeset	70
ac2e5e1c2c3c Refactor \o/ Thibaut Girka <thib@sitedethib.com> parents: diff changeset	71 def __init__(self, path, format_class, file_list=None):
ac2e5e1c2c3c Refactor \o/ Thibaut Girka <thib@sitedethib.com> parents: diff changeset	72 self.path = path
ac2e5e1c2c3c Refactor \o/ Thibaut Girka <thib@sitedethib.com> parents: diff changeset	73 self.format_class = format_class
ac2e5e1c2c3c Refactor \o/ Thibaut Girka <thib@sitedethib.com> parents: diff changeset	74 self.file_list = file_list or []
ac2e5e1c2c3c Refactor \o/ Thibaut Girka <thib@sitedethib.com> parents: diff changeset	75
ac2e5e1c2c3c Refactor \o/ Thibaut Girka <thib@sitedethib.com> parents: diff changeset	76
ac2e5e1c2c3c Refactor \o/ Thibaut Girka <thib@sitedethib.com> parents: diff changeset	77 def open(self):
263 ac677dd0ffe0 Add support for reading from a directory instead of a PBG3 (for debugging purposes). Thibaut Girka <thib@sitedethib.com> parents: 262 diff changeset	78 if self.format_class is Directory:
ac677dd0ffe0 Add support for reading from a directory instead of a PBG3 (for debugging purposes). Thibaut Girka <thib@sitedethib.com> parents: 262 diff changeset	79 return self.format_class(self.path)
ac677dd0ffe0 Add support for reading from a directory instead of a PBG3 (for debugging purposes). Thibaut Girka <thib@sitedethib.com> parents: 262 diff changeset	80
97 ac2e5e1c2c3c Refactor \o/ Thibaut Girka <thib@sitedethib.com> parents: diff changeset	81 file = open(self.path, 'rb')
ac2e5e1c2c3c Refactor \o/ Thibaut Girka <thib@sitedethib.com> parents: diff changeset	82 instance = self.format_class.read(file)
ac2e5e1c2c3c Refactor \o/ Thibaut Girka <thib@sitedethib.com> parents: diff changeset	83 return instance
ac2e5e1c2c3c Refactor \o/ Thibaut Girka <thib@sitedethib.com> parents: diff changeset	84
ac2e5e1c2c3c Refactor \o/ Thibaut Girka <thib@sitedethib.com> parents: diff changeset	85
ac2e5e1c2c3c Refactor \o/ Thibaut Girka <thib@sitedethib.com> parents: diff changeset	86 @classmethod
ac2e5e1c2c3c Refactor \o/ Thibaut Girka <thib@sitedethib.com> parents: diff changeset	87 def get_from_path(cls, path):
263 ac677dd0ffe0 Add support for reading from a directory instead of a PBG3 (for debugging purposes). Thibaut Girka <thib@sitedethib.com> parents: 262 diff changeset	88 if os.path.isdir(path):
ac677dd0ffe0 Add support for reading from a directory instead of a PBG3 (for debugging purposes). Thibaut Girka <thib@sitedethib.com> parents: 262 diff changeset	89 instance = Directory(path)
ac677dd0ffe0 Add support for reading from a directory instead of a PBG3 (for debugging purposes). Thibaut Girka <thib@sitedethib.com> parents: 262 diff changeset	90 file_list = instance.list_files()
ac677dd0ffe0 Add support for reading from a directory instead of a PBG3 (for debugging purposes). Thibaut Girka <thib@sitedethib.com> parents: 262 diff changeset	91 return cls(path, Directory, file_list)
97 ac2e5e1c2c3c Refactor \o/ Thibaut Girka <thib@sitedethib.com> parents: diff changeset	92 with open(path, 'rb') as file:
ac2e5e1c2c3c Refactor \o/ Thibaut Girka <thib@sitedethib.com> parents: diff changeset	93 magic = file.read(4)
ac2e5e1c2c3c Refactor \o/ Thibaut Girka <thib@sitedethib.com> parents: diff changeset	94 file.seek(0)
ac2e5e1c2c3c Refactor \o/ Thibaut Girka <thib@sitedethib.com> parents: diff changeset	95 format_class = cls._formats[magic]
ac2e5e1c2c3c Refactor \o/ Thibaut Girka <thib@sitedethib.com> parents: diff changeset	96 instance = format_class.read(file)
ac2e5e1c2c3c Refactor \o/ Thibaut Girka <thib@sitedethib.com> parents: diff changeset	97 file_list = instance.list_files()
ac2e5e1c2c3c Refactor \o/ Thibaut Girka <thib@sitedethib.com> parents: diff changeset	98 return cls(path, format_class, file_list)
ac2e5e1c2c3c Refactor \o/ Thibaut Girka <thib@sitedethib.com> parents: diff changeset	99
ac2e5e1c2c3c Refactor \o/ Thibaut Girka <thib@sitedethib.com> parents: diff changeset	100
ac2e5e1c2c3c Refactor \o/ Thibaut Girka <thib@sitedethib.com> parents: diff changeset	101
ac2e5e1c2c3c Refactor \o/ Thibaut Girka <thib@sitedethib.com> parents: diff changeset	102 class Loader(object):
231 c417bb6c98bf Search for 102h.exe in the game directory instead of the current directory. Thibaut Girka <thib@sitedethib.com> parents: 229 diff changeset	103 def __init__(self, game_dir=None):
298 92a6fd2632f1 Improve heuristic to filter out non-game exes (like custom.exe). Thibaut Girka <thib@sitedethib.com> parents: 297 diff changeset	104 self.exe_files = []
231 c417bb6c98bf Search for 102h.exe in the game directory instead of the current directory. Thibaut Girka <thib@sitedethib.com> parents: 229 diff changeset	105 self.game_dir = game_dir
97 ac2e5e1c2c3c Refactor \o/ Thibaut Girka <thib@sitedethib.com> parents: diff changeset	106 self.known_files = {}
413 6d7dbcb31d95 Let the gc do its work, don’t keep useless references. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 338 diff changeset	107 self.instanced_anms = {} #TODO: remove it someday.
97 ac2e5e1c2c3c Refactor \o/ Thibaut Girka <thib@sitedethib.com> parents: diff changeset	108
ac2e5e1c2c3c Refactor \o/ Thibaut Girka <thib@sitedethib.com> parents: diff changeset	109
262 8fa660da5f0c Automatically search data files using different names. Thibaut Girka <thib@sitedethib.com> parents: 231 diff changeset	110 def scan_archives(self, paths_lists):
8fa660da5f0c Automatically search data files using different names. Thibaut Girka <thib@sitedethib.com> parents: 231 diff changeset	111 for paths in paths_lists:
8fa660da5f0c Automatically search data files using different names. Thibaut Girka <thib@sitedethib.com> parents: 231 diff changeset	112 def _expand_paths():
297 a09ac4650e0d Fix relative path handling and os-specific path separators. Thibaut Girka <thib@sitedethib.com> parents: 285 diff changeset	113 for path in paths.split(os.path.pathsep):
262 8fa660da5f0c Automatically search data files using different names. Thibaut Girka <thib@sitedethib.com> parents: 231 diff changeset	114 if self.game_dir and not os.path.isabs(path):
8fa660da5f0c Automatically search data files using different names. Thibaut Girka <thib@sitedethib.com> parents: 231 diff changeset	115 path = os.path.join(self.game_dir, path)
8fa660da5f0c Automatically search data files using different names. Thibaut Girka <thib@sitedethib.com> parents: 231 diff changeset	116 yield glob(path)
8fa660da5f0c Automatically search data files using different names. Thibaut Girka <thib@sitedethib.com> parents: 231 diff changeset	117 paths = list(chain(*_expand_paths()))
338 65453340ae95 Print an error when all the needed files aren’t present. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 325 diff changeset	118 if not paths:
65453340ae95 Print an error when all the needed files aren’t present. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 325 diff changeset	119 raise IOError
262 8fa660da5f0c Automatically search data files using different names. Thibaut Girka <thib@sitedethib.com> parents: 231 diff changeset	120 path = paths[0]
8fa660da5f0c Automatically search data files using different names. Thibaut Girka <thib@sitedethib.com> parents: 231 diff changeset	121 if os.path.splitext(path)[1] == '.exe':
298 92a6fd2632f1 Improve heuristic to filter out non-game exes (like custom.exe). Thibaut Girka <thib@sitedethib.com> parents: 297 diff changeset	122 self.exe_files.extend(paths)
262 8fa660da5f0c Automatically search data files using different names. Thibaut Girka <thib@sitedethib.com> parents: 231 diff changeset	123 else:
8fa660da5f0c Automatically search data files using different names. Thibaut Girka <thib@sitedethib.com> parents: 231 diff changeset	124 archive_description = ArchiveDescription.get_from_path(path)
8fa660da5f0c Automatically search data files using different names. Thibaut Girka <thib@sitedethib.com> parents: 231 diff changeset	125 for name in archive_description.file_list:
8fa660da5f0c Automatically search data files using different names. Thibaut Girka <thib@sitedethib.com> parents: 231 diff changeset	126 self.known_files[name] = archive_description
97 ac2e5e1c2c3c Refactor \o/ Thibaut Girka <thib@sitedethib.com> parents: diff changeset	127
ac2e5e1c2c3c Refactor \o/ Thibaut Girka <thib@sitedethib.com> parents: diff changeset	128
ac2e5e1c2c3c Refactor \o/ Thibaut Girka <thib@sitedethib.com> parents: diff changeset	129 def get_file_data(self, name):
ac2e5e1c2c3c Refactor \o/ Thibaut Girka <thib@sitedethib.com> parents: diff changeset	130 with self.known_files[name].open() as archive:
ac2e5e1c2c3c Refactor \o/ Thibaut Girka <thib@sitedethib.com> parents: diff changeset	131 content = archive.extract(name)
ac2e5e1c2c3c Refactor \o/ Thibaut Girka <thib@sitedethib.com> parents: diff changeset	132 return content
ac2e5e1c2c3c Refactor \o/ Thibaut Girka <thib@sitedethib.com> parents: diff changeset	133
ac2e5e1c2c3c Refactor \o/ Thibaut Girka <thib@sitedethib.com> parents: diff changeset	134
ac2e5e1c2c3c Refactor \o/ Thibaut Girka <thib@sitedethib.com> parents: diff changeset	135 def get_file(self, name):
ac2e5e1c2c3c Refactor \o/ Thibaut Girka <thib@sitedethib.com> parents: diff changeset	136 with self.known_files[name].open() as archive:
ac2e5e1c2c3c Refactor \o/ Thibaut Girka <thib@sitedethib.com> parents: diff changeset	137 content = archive.extract(name)
ac2e5e1c2c3c Refactor \o/ Thibaut Girka <thib@sitedethib.com> parents: diff changeset	138 return BytesIO(content)
ac2e5e1c2c3c Refactor \o/ Thibaut Girka <thib@sitedethib.com> parents: diff changeset	139
ac2e5e1c2c3c Refactor \o/ Thibaut Girka <thib@sitedethib.com> parents: diff changeset	140
ac2e5e1c2c3c Refactor \o/ Thibaut Girka <thib@sitedethib.com> parents: diff changeset	141 def get_anm(self, name):
ac2e5e1c2c3c Refactor \o/ Thibaut Girka <thib@sitedethib.com> parents: diff changeset	142 if name not in self.instanced_anms:
ac2e5e1c2c3c Refactor \o/ Thibaut Girka <thib@sitedethib.com> parents: diff changeset	143 file = self.get_file(name)
429 40d5f3083ebc Implement PCB’s ANM2 format and vm. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 413 diff changeset	144 self.instanced_anms[name] = ANM0.read(file)
97 ac2e5e1c2c3c Refactor \o/ Thibaut Girka <thib@sitedethib.com> parents: diff changeset	145 return self.instanced_anms[name]
ac2e5e1c2c3c Refactor \o/ Thibaut Girka <thib@sitedethib.com> parents: diff changeset	146
ac2e5e1c2c3c Refactor \o/ Thibaut Girka <thib@sitedethib.com> parents: diff changeset	147
ac2e5e1c2c3c Refactor \o/ Thibaut Girka <thib@sitedethib.com> parents: diff changeset	148 def get_stage(self, name):
413 6d7dbcb31d95 Let the gc do its work, don’t keep useless references. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 338 diff changeset	149 file = self.get_file(name)
6d7dbcb31d95 Let the gc do its work, don’t keep useless references. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 338 diff changeset	150 return Stage.read(file) #TODO: modular
97 ac2e5e1c2c3c Refactor \o/ Thibaut Girka <thib@sitedethib.com> parents: diff changeset	151
ac2e5e1c2c3c Refactor \o/ Thibaut Girka <thib@sitedethib.com> parents: diff changeset	152
ac2e5e1c2c3c Refactor \o/ Thibaut Girka <thib@sitedethib.com> parents: diff changeset	153 def get_ecl(self, name):
413 6d7dbcb31d95 Let the gc do its work, don’t keep useless references. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 338 diff changeset	154 file = self.get_file(name)
6d7dbcb31d95 Let the gc do its work, don’t keep useless references. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 338 diff changeset	155 return ECL.read(file) #TODO: modular
97 ac2e5e1c2c3c Refactor \o/ Thibaut Girka <thib@sitedethib.com> parents: diff changeset	156
ac2e5e1c2c3c Refactor \o/ Thibaut Girka <thib@sitedethib.com> parents: diff changeset	157
133 2cad2e84a49e Add reading support for the MSG format. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 97 diff changeset	158 def get_msg(self, name):
413 6d7dbcb31d95 Let the gc do its work, don’t keep useless references. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 338 diff changeset	159 file = self.get_file(name)
6d7dbcb31d95 Let the gc do its work, don’t keep useless references. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 338 diff changeset	160 return MSG.read(file) #TODO: modular
133 2cad2e84a49e Add reading support for the MSG format. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 97 diff changeset	161
2cad2e84a49e Add reading support for the MSG format. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 97 diff changeset	162
220 0595315d3880 Fix SHT handling; change a few things to be closer to ZUN’s mind; and first stub of PCB support. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 133 diff changeset	163 def get_sht(self, name):
413 6d7dbcb31d95 Let the gc do its work, don’t keep useless references. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 338 diff changeset	164 file = self.get_file(name)
6d7dbcb31d95 Let the gc do its work, don’t keep useless references. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 338 diff changeset	165 return SHT.read(file) #TODO: modular
220 0595315d3880 Fix SHT handling; change a few things to be closer to ZUN’s mind; and first stub of PCB support. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 133 diff changeset	166
0595315d3880 Fix SHT handling; change a few things to be closer to ZUN’s mind; and first stub of PCB support. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 133 diff changeset	167
262 8fa660da5f0c Automatically search data files using different names. Thibaut Girka <thib@sitedethib.com> parents: 231 diff changeset	168 def get_eosd_characters(self):
231 c417bb6c98bf Search for 102h.exe in the game directory instead of the current directory. Thibaut Girka <thib@sitedethib.com> parents: 229 diff changeset	169 #TODO: Move to pytouhou.games.eosd?
298 92a6fd2632f1 Improve heuristic to filter out non-game exes (like custom.exe). Thibaut Girka <thib@sitedethib.com> parents: 297 diff changeset	170 for path in self.exe_files:
92a6fd2632f1 Improve heuristic to filter out non-game exes (like custom.exe). Thibaut Girka <thib@sitedethib.com> parents: 297 diff changeset	171 try:
92a6fd2632f1 Improve heuristic to filter out non-game exes (like custom.exe). Thibaut Girka <thib@sitedethib.com> parents: 297 diff changeset	172 with open(path, 'rb') as file:
92a6fd2632f1 Improve heuristic to filter out non-game exes (like custom.exe). Thibaut Girka <thib@sitedethib.com> parents: 297 diff changeset	173 characters = EoSDSHT.read(file)
92a6fd2632f1 Improve heuristic to filter out non-game exes (like custom.exe). Thibaut Girka <thib@sitedethib.com> parents: 297 diff changeset	174 return characters
92a6fd2632f1 Improve heuristic to filter out non-game exes (like custom.exe). Thibaut Girka <thib@sitedethib.com> parents: 297 diff changeset	175 except InvalidExeException:
92a6fd2632f1 Improve heuristic to filter out non-game exes (like custom.exe). Thibaut Girka <thib@sitedethib.com> parents: 297 diff changeset	176 pass
92a6fd2632f1 Improve heuristic to filter out non-game exes (like custom.exe). Thibaut Girka <thib@sitedethib.com> parents: 297 diff changeset	177 logger.error("Required game exe not found!")
229 5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 220 diff changeset	178
5afc75f71fed Add “SHT” support to EoSD, and do a little cleanup. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 220 diff changeset	179
321 61adb5453e46 Implement music playback. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 298 diff changeset	180 def get_track(self, name):
61adb5453e46 Implement music playback. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 298 diff changeset	181 posname = name.replace('bgm/', '').replace('.mid', '.pos')
413 6d7dbcb31d95 Let the gc do its work, don’t keep useless references. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 338 diff changeset	182 file = self.get_file(posname)
6d7dbcb31d95 Let the gc do its work, don’t keep useless references. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 338 diff changeset	183 return Track.read(file) #TODO: modular
321 61adb5453e46 Implement music playback. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 298 diff changeset	184
61adb5453e46 Implement music playback. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 298 diff changeset	185
325 cddfd3cb4797 Add music support for >PCB. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 321 diff changeset	186 def get_fmt(self, name):
413 6d7dbcb31d95 Let the gc do its work, don’t keep useless references. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 338 diff changeset	187 file = self.get_file(name)
6d7dbcb31d95 Let the gc do its work, don’t keep useless references. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 338 diff changeset	188 return FMT.read(file) #TODO: modular
325 cddfd3cb4797 Add music support for >PCB. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 321 diff changeset	189
cddfd3cb4797 Add music support for >PCB. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 321 diff changeset	190
285 2100276c289d Document some AnmWrapper related functions. Thibaut Girka <thib@sitedethib.com> parents: 282 diff changeset	191 def get_anm_wrapper(self, names, offsets=None):
2100276c289d Document some AnmWrapper related functions. Thibaut Girka <thib@sitedethib.com> parents: 282 diff changeset	192 """Create an AnmWrapper for ANM files “names”.
2100276c289d Document some AnmWrapper related functions. Thibaut Girka <thib@sitedethib.com> parents: 282 diff changeset	193
2100276c289d Document some AnmWrapper related functions. Thibaut Girka <thib@sitedethib.com> parents: 282 diff changeset	194 If one of the files “names” does not exist or is not a valid ANM file,
2100276c289d Document some AnmWrapper related functions. Thibaut Girka <thib@sitedethib.com> parents: 282 diff changeset	195 raises an exception.
2100276c289d Document some AnmWrapper related functions. Thibaut Girka <thib@sitedethib.com> parents: 282 diff changeset	196 """
282 dbb1a86c0235 Rename Animations to ANM0 and prepare AnmWrapper for dialogs and interface. Thibaut Girka <thib@sitedethib.com> parents: 263 diff changeset	197 return AnmWrapper((self.get_anm(name) for name in names), offsets)
97 ac2e5e1c2c3c Refactor \o/ Thibaut Girka <thib@sitedethib.com> parents: diff changeset	198
ac2e5e1c2c3c Refactor \o/ Thibaut Girka <thib@sitedethib.com> parents: diff changeset	199
285 2100276c289d Document some AnmWrapper related functions. Thibaut Girka <thib@sitedethib.com> parents: 282 diff changeset	200 def get_anm_wrapper2(self, names, offsets=None):
2100276c289d Document some AnmWrapper related functions. Thibaut Girka <thib@sitedethib.com> parents: 282 diff changeset	201 """Create an AnmWrapper for ANM files “names”.
2100276c289d Document some AnmWrapper related functions. Thibaut Girka <thib@sitedethib.com> parents: 282 diff changeset	202
2100276c289d Document some AnmWrapper related functions. Thibaut Girka <thib@sitedethib.com> parents: 282 diff changeset	203 Stop at the first non-existent or invalid ANM file if there is one,
2100276c289d Document some AnmWrapper related functions. Thibaut Girka <thib@sitedethib.com> parents: 282 diff changeset	204 and return an AnmWrapper for all the previous correct files.
2100276c289d Document some AnmWrapper related functions. Thibaut Girka <thib@sitedethib.com> parents: 282 diff changeset	205 """
282 dbb1a86c0235 Rename Animations to ANM0 and prepare AnmWrapper for dialogs and interface. Thibaut Girka <thib@sitedethib.com> parents: 263 diff changeset	206 anms = []
dbb1a86c0235 Rename Animations to ANM0 and prepare AnmWrapper for dialogs and interface. Thibaut Girka <thib@sitedethib.com> parents: 263 diff changeset	207
97 ac2e5e1c2c3c Refactor \o/ Thibaut Girka <thib@sitedethib.com> parents: diff changeset	208 try:
ac2e5e1c2c3c Refactor \o/ Thibaut Girka <thib@sitedethib.com> parents: diff changeset	209 for name in names:
282 dbb1a86c0235 Rename Animations to ANM0 and prepare AnmWrapper for dialogs and interface. Thibaut Girka <thib@sitedethib.com> parents: 263 diff changeset	210 anms.append(self.get_anm(name))
97 ac2e5e1c2c3c Refactor \o/ Thibaut Girka <thib@sitedethib.com> parents: diff changeset	211 except KeyError:
ac2e5e1c2c3c Refactor \o/ Thibaut Girka <thib@sitedethib.com> parents: diff changeset	212 pass
ac2e5e1c2c3c Refactor \o/ Thibaut Girka <thib@sitedethib.com> parents: diff changeset	213
282 dbb1a86c0235 Rename Animations to ANM0 and prepare AnmWrapper for dialogs and interface. Thibaut Girka <thib@sitedethib.com> parents: 263 diff changeset	214 return AnmWrapper(anms, offsets)
97 ac2e5e1c2c3c Refactor \o/ Thibaut Girka <thib@sitedethib.com> parents: diff changeset	215

Mercurial > touhou

annotate pytouhou/resource/loader.py @ 429:40d5f3083ebc