Mercurial > touhou
annotate pytouhou/resource/loader.py @ 615:d1f0bb0b7a17
Don’t inherit explicitely from object, we are not on Python 2.7 anymore. :)
| author | Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> |
|---|---|
| date | Sat, 28 Mar 2015 21:40:51 +0100 |
| parents | e15672733c93 |
| children | d18c0bf11138 |
| 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 | 18 |
|
429
40d5f3083ebc
Implement PCB’s ANM2 format and vm.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents:
413
diff
changeset
|
19 from pytouhou.formats import WrongFormatError |
| 97 | 20 from pytouhou.formats.pbg3 import PBG3 |
| 21 from pytouhou.formats.std import Stage | |
| 22 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
|
23 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
|
24 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
|
25 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
|
26 from pytouhou.formats.exe import SHT as EoSDSHT, InvalidExeException |
|
321
61adb5453e46
Implement music playback.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents:
298
diff
changeset
|
27 from pytouhou.formats.music import Track |
|
325
cddfd3cb4797
Add music support for >PCB.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents:
321
diff
changeset
|
28 from pytouhou.formats.fmt import FMT |
| 97 | 29 |
|
298
92a6fd2632f1
Improve heuristic to filter out non-game exes (like custom.exe).
Thibaut Girka <thib@sitedethib.com>
parents:
297
diff
changeset
|
30 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
|
31 |
|
92a6fd2632f1
Improve heuristic to filter out non-game exes (like custom.exe).
Thibaut Girka <thib@sitedethib.com>
parents:
297
diff
changeset
|
32 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
|
33 |
|
92a6fd2632f1
Improve heuristic to filter out non-game exes (like custom.exe).
Thibaut Girka <thib@sitedethib.com>
parents:
297
diff
changeset
|
34 |
| 97 | 35 |
|
615
d1f0bb0b7a17
Don’t inherit explicitely from object, we are not on Python 2.7 anymore. :)
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents:
590
diff
changeset
|
36 class Directory: |
|
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
|
37 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
|
38 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
|
39 |
|
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 |
|
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 __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
|
42 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
|
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 __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
|
46 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
|
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 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
|
50 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
|
51 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
|
52 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
|
53 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
|
54 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
|
55 |
|
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 |
|
536
6b76c9ba3975
Make archives return files by default, instead of bytes.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents:
504
diff
changeset
|
57 def get_file(self, name): |
|
6b76c9ba3975
Make archives return files by default, instead of bytes.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents:
504
diff
changeset
|
58 return open(os.path.join(self.path, str(name)), 'rb') |
|
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
|
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 |
|
615
d1f0bb0b7a17
Don’t inherit explicitely from object, we are not on Python 2.7 anymore. :)
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents:
590
diff
changeset
|
62 class ArchiveDescription: |
|
590
e15672733c93
Switch to Python 3.x instead of 2.7.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents:
536
diff
changeset
|
63 _formats = {b'PBG3': PBG3} |
| 97 | 64 |
| 65 def __init__(self, path, format_class, file_list=None): | |
| 66 self.path = path | |
| 67 self.format_class = format_class | |
| 68 self.file_list = file_list or [] | |
| 69 | |
| 70 | |
| 71 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
|
72 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
|
73 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
|
74 |
| 97 | 75 file = open(self.path, 'rb') |
| 76 instance = self.format_class.read(file) | |
| 77 return instance | |
| 78 | |
| 79 | |
| 80 @classmethod | |
| 81 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
|
82 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
|
83 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
|
84 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
|
85 return cls(path, Directory, file_list) |
| 97 | 86 with open(path, 'rb') as file: |
| 87 magic = file.read(4) | |
| 88 file.seek(0) | |
| 89 format_class = cls._formats[magic] | |
| 90 instance = format_class.read(file) | |
| 91 file_list = instance.list_files() | |
| 92 return cls(path, format_class, file_list) | |
| 93 | |
| 94 | |
| 95 | |
|
615
d1f0bb0b7a17
Don’t inherit explicitely from object, we are not on Python 2.7 anymore. :)
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents:
590
diff
changeset
|
96 class Loader: |
|
231
c417bb6c98bf
Search for 102h.exe in the game directory instead of the current directory.
Thibaut Girka <thib@sitedethib.com>
parents:
229
diff
changeset
|
97 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
|
98 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
|
99 self.game_dir = game_dir |
| 97 | 100 self.known_files = {} |
|
504
69c73023f7a0
Make ANM garbage collectable.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents:
430
diff
changeset
|
101 self.instanced_anms = {} # Cache for the textures. |
|
69c73023f7a0
Make ANM garbage collectable.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents:
430
diff
changeset
|
102 self.loaded_anms = [] # For the double loading warnings. |
| 97 | 103 |
| 104 | |
|
262
8fa660da5f0c
Automatically search data files using different names.
Thibaut Girka <thib@sitedethib.com>
parents:
231
diff
changeset
|
105 def scan_archives(self, paths_lists): |
|
8fa660da5f0c
Automatically search data files using different names.
Thibaut Girka <thib@sitedethib.com>
parents:
231
diff
changeset
|
106 for paths in paths_lists: |
|
8fa660da5f0c
Automatically search data files using different names.
Thibaut Girka <thib@sitedethib.com>
parents:
231
diff
changeset
|
107 def _expand_paths(): |
|
297
a09ac4650e0d
Fix relative path handling and os-specific path separators.
Thibaut Girka <thib@sitedethib.com>
parents:
285
diff
changeset
|
108 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
|
109 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
|
110 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
|
111 yield glob(path) |
|
8fa660da5f0c
Automatically search data files using different names.
Thibaut Girka <thib@sitedethib.com>
parents:
231
diff
changeset
|
112 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
|
113 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
|
114 raise IOError |
|
262
8fa660da5f0c
Automatically search data files using different names.
Thibaut Girka <thib@sitedethib.com>
parents:
231
diff
changeset
|
115 path = paths[0] |
|
8fa660da5f0c
Automatically search data files using different names.
Thibaut Girka <thib@sitedethib.com>
parents:
231
diff
changeset
|
116 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
|
117 self.exe_files.extend(paths) |
|
262
8fa660da5f0c
Automatically search data files using different names.
Thibaut Girka <thib@sitedethib.com>
parents:
231
diff
changeset
|
118 else: |
|
8fa660da5f0c
Automatically search data files using different names.
Thibaut Girka <thib@sitedethib.com>
parents:
231
diff
changeset
|
119 archive_description = ArchiveDescription.get_from_path(path) |
|
8fa660da5f0c
Automatically search data files using different names.
Thibaut Girka <thib@sitedethib.com>
parents:
231
diff
changeset
|
120 for name in archive_description.file_list: |
|
8fa660da5f0c
Automatically search data files using different names.
Thibaut Girka <thib@sitedethib.com>
parents:
231
diff
changeset
|
121 self.known_files[name] = archive_description |
| 97 | 122 |
| 123 | |
| 124 def get_file(self, name): | |
| 125 with self.known_files[name].open() as archive: | |
|
536
6b76c9ba3975
Make archives return files by default, instead of bytes.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents:
504
diff
changeset
|
126 return archive.get_file(name) |
| 97 | 127 |
| 128 | |
| 129 def get_anm(self, name): | |
|
504
69c73023f7a0
Make ANM garbage collectable.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents:
430
diff
changeset
|
130 if name in self.loaded_anms: |
|
69c73023f7a0
Make ANM garbage collectable.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents:
430
diff
changeset
|
131 logger.warn('ANM0 %s already loaded', name) |
|
69c73023f7a0
Make ANM garbage collectable.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents:
430
diff
changeset
|
132 file = self.get_file(name) |
|
69c73023f7a0
Make ANM garbage collectable.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents:
430
diff
changeset
|
133 anm = ANM0.read(file) |
|
69c73023f7a0
Make ANM garbage collectable.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents:
430
diff
changeset
|
134 self.instanced_anms[name] = anm |
|
69c73023f7a0
Make ANM garbage collectable.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents:
430
diff
changeset
|
135 self.loaded_anms.append(name) |
|
69c73023f7a0
Make ANM garbage collectable.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents:
430
diff
changeset
|
136 return anm |
| 97 | 137 |
| 138 | |
| 139 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
|
140 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
|
141 return Stage.read(file) #TODO: modular |
| 97 | 142 |
| 143 | |
| 144 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
|
145 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
|
146 return ECL.read(file) #TODO: modular |
| 97 | 147 |
| 148 | |
|
133
2cad2e84a49e
Add reading support for the MSG format.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents:
97
diff
changeset
|
149 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
|
150 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
|
151 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
|
152 |
|
2cad2e84a49e
Add reading support for the MSG format.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents:
97
diff
changeset
|
153 |
|
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
|
154 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
|
155 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
|
156 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
|
157 |
|
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
|
158 |
|
262
8fa660da5f0c
Automatically search data files using different names.
Thibaut Girka <thib@sitedethib.com>
parents:
231
diff
changeset
|
159 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
|
160 #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
|
161 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
|
162 try: |
|
92a6fd2632f1
Improve heuristic to filter out non-game exes (like custom.exe).
Thibaut Girka <thib@sitedethib.com>
parents:
297
diff
changeset
|
163 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
|
164 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
|
165 return characters |
|
92a6fd2632f1
Improve heuristic to filter out non-game exes (like custom.exe).
Thibaut Girka <thib@sitedethib.com>
parents:
297
diff
changeset
|
166 except InvalidExeException: |
|
92a6fd2632f1
Improve heuristic to filter out non-game exes (like custom.exe).
Thibaut Girka <thib@sitedethib.com>
parents:
297
diff
changeset
|
167 pass |
|
92a6fd2632f1
Improve heuristic to filter out non-game exes (like custom.exe).
Thibaut Girka <thib@sitedethib.com>
parents:
297
diff
changeset
|
168 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
|
169 |
|
5afc75f71fed
Add “SHT” support to EoSD, and do a little cleanup.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents:
220
diff
changeset
|
170 |
|
321
61adb5453e46
Implement music playback.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents:
298
diff
changeset
|
171 def get_track(self, name): |
|
61adb5453e46
Implement music playback.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents:
298
diff
changeset
|
172 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
|
173 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
|
174 return Track.read(file) #TODO: modular |
|
321
61adb5453e46
Implement music playback.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents:
298
diff
changeset
|
175 |
|
61adb5453e46
Implement music playback.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents:
298
diff
changeset
|
176 |
|
325
cddfd3cb4797
Add music support for >PCB.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents:
321
diff
changeset
|
177 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
|
178 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
|
179 return FMT.read(file) #TODO: modular |
|
325
cddfd3cb4797
Add music support for >PCB.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents:
321
diff
changeset
|
180 |
|
cddfd3cb4797
Add music support for >PCB.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents:
321
diff
changeset
|
181 |
|
430
c9433188ffdb
Remove AnmWrapper, since ANMs are lists of entries now.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents:
429
diff
changeset
|
182 def get_single_anm(self, name): |
|
c9433188ffdb
Remove AnmWrapper, since ANMs are lists of entries now.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents:
429
diff
changeset
|
183 """Hack for EoSD, since it doesn’t support multi-entries ANMs.""" |
|
c9433188ffdb
Remove AnmWrapper, since ANMs are lists of entries now.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents:
429
diff
changeset
|
184 anm = self.get_anm(name) |
|
c9433188ffdb
Remove AnmWrapper, since ANMs are lists of entries now.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents:
429
diff
changeset
|
185 assert len(anm) == 1 |
|
c9433188ffdb
Remove AnmWrapper, since ANMs are lists of entries now.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents:
429
diff
changeset
|
186 return anm[0] |
| 97 | 187 |
| 188 | |
|
430
c9433188ffdb
Remove AnmWrapper, since ANMs are lists of entries now.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents:
429
diff
changeset
|
189 def get_multi_anm(self, names): |
|
c9433188ffdb
Remove AnmWrapper, since ANMs are lists of entries now.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents:
429
diff
changeset
|
190 """Hack for EoSD, since it doesn’t support multi-entries ANMs.""" |
|
c9433188ffdb
Remove AnmWrapper, since ANMs are lists of entries now.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents:
429
diff
changeset
|
191 return sum((self.get_anm(name) for name in names), []) |