Cool_Tools/ifstools
1
0
Fork 0
mirror of https://github.com/mon/ifstools.git synced 2024-12-02 20:47:16 +01:00
Code Releases Activity

Proper _super_ support

Browse Source
This commit is contained in:
Will Toohey 2018-02-12 22:53:51 +00:00
parent 471e825c61
commit 2b847d08d7
4 changed files with 24 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
ifstools/__init__.py
View File

@ -1,2 +1,3 @@
from .ifstools import main from .ifstools import main
from .ifs import IFS from .ifs import IFS
from .handlers import GenericFolder, GenericFile

31
ifstools/handlers/GenericFolder.py
View File

@ -1,5 +1,5 @@
from itertools import chain from itertools import chain
from os.path import getmtime, basename, join from os.path import getmtime, basename, dirname, join, realpath
from collections import OrderedDict from collections import OrderedDict
import lxml.etree as etree import lxml.etree as etree
@ -9,14 +9,14 @@ from .Node import Node
class GenericFolder(Node): class GenericFolder(Node):
def __init__(self, ifs_data, obj, parent = None, path = '', name = '', has_super = False): def __init__(self, ifs_data, obj, parent = None, path = '', name = '', supers = None):
# circular dependencies mean we import here # circular dependencies mean we import here
from . import AfpFolder, TexFolder from . import AfpFolder, TexFolder
self.folder_handlers = { self.folder_handlers = {
'afp' : AfpFolder, 'afp' : AfpFolder,
'tex' : TexFolder, 'tex' : TexFolder,
} }
self.has_super = has_super self.supers = supers if supers else []
Node.__init__(self, ifs_data, obj, parent, path, name) Node.__init__(self, ifs_data, obj, parent, path, name)
file_handler = GenericFile file_handler = GenericFile
@ -27,22 +27,29 @@ class GenericFolder(Node):
self.files = OrderedDict() self.files = OrderedDict()
self.folders = {} self.folders = {}
my_path = dirname(realpath(self.ifs_data.file.name))
# muh circular deps
from ..ifs import IFS
for child in element.iterchildren(tag=etree.Element): for child in element.iterchildren(tag=etree.Element):
filename = Node.fix_name(child.tag) filename = Node.fix_name(child.tag)
if filename == '_info_': # metadata if filename == '_info_': # metadata
continue continue
elif filename == '_super_': elif filename == '_super_': # sub-reference
self.has_super = True self.supers.append(IFS(join(my_path, child.text)).data_blob)
# folder: has children or timestamp only # folder: has children or timestamp only, and isn't a reference
elif list(child) or len(child.text.split(' ')) == 1: elif (list(child) or len(child.text.split(' ')) == 1) and child[0].tag != 'i':
# note: files with 'super' references have 'i' tags as backrefs
# We just ignore these
if self.has_super and child[0].tag == 'i':
continue
handler = self.folder_handlers.get(filename, GenericFolder) handler = self.folder_handlers.get(filename, GenericFolder)
self.folders[filename] = handler(self.ifs_data, child, self, self.full_path, filename, self.has_super) self.folders[filename] = handler(self.ifs_data, child, self, self.full_path, filename, self.supers)
else: # file else: # file
self.files[filename] = self.file_handler(self.ifs_data, child, self, self.full_path, filename) self.files[filename] = self.file_handler(self.ifs_data, child, self, self.full_path, filename)
if list(child) and child[0].tag == 'i':
# backref
super_ref = int(child[0].text)
if super_ref > len(self.supers):
raise IOError('IFS references super-IFS {} but we only have {}'.format(super_ref, len(self.supers)))
self.files[filename].ifs_data = self.supers[super_ref - 1]
if not self.full_path: # root if not self.full_path: # root
self.tree_complete() self.tree_complete()

4
ifstools/handlers/MD5Folder.py
View File

@ -6,8 +6,8 @@ from . import GenericFolder
class MD5Folder(GenericFolder): class MD5Folder(GenericFolder):
def __init__(self, ifs_data, parent, obj, path = '', name = '', has_super = False, md5_tag = None, extension = None): def __init__(self, ifs_data, parent, obj, path = '', name = '', supers = None, md5_tag = None, extension = None):
GenericFolder.__init__(self, ifs_data, parent, obj, path, name, has_super) GenericFolder.__init__(self, ifs_data, parent, obj, path, name, supers)
self.md5_tag = md5_tag if md5_tag else self.name self.md5_tag = md5_tag if md5_tag else self.name
self.extension = extension self.extension = extension

4
ifstools/handlers/TexFolder.py
View File

@ -13,8 +13,8 @@ class TextureList(GenericFile):
return k.to_binary() return k.to_binary()
class TexFolder(MD5Folder): class TexFolder(MD5Folder):
def __init__(self, ifs_data, obj, parent = None, path = '', name = '', has_super = False): def __init__(self, ifs_data, obj, parent = None, path = '', name = '', supers = None):
MD5Folder.__init__(self, ifs_data, obj, parent, path, name, has_super, 'image', '.png') MD5Folder.__init__(self, ifs_data, obj, parent, path, name, supers, 'image', '.png')
def tree_complete(self): def tree_complete(self):
MD5Folder.tree_complete(self) MD5Folder.tree_complete(self)