Cool_Tools/vgmstream
1
0
Fork 0
mirror of https://github.com/vgmstream/vgmstream.git synced 2025-02-17 19:19:16 +01:00
Code Issues Packages Projects Releases Wiki Activity

txtp_maker: extra subsong/renaming options

Browse Source
This commit is contained in:
bnnm 2021-09-19 23:40:31 +02:00
parent 2cddb50da9
commit 6405113c82
1 changed files with 34 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

40
cli/tools/txtp_maker.py
View File

@ -32,6 +32,8 @@ class Cli(object):
" - make .txtp for in .bnk including only subsong names that start with 'bgm'\n\n"
" %(prog)s *.fsb -n \"{fn}<__{ss}>< [{in}]>\" -z 4 -o\n"
" - make .txtp for all fsb, adding subsongs and stream name if they exist\n\n"
" %(prog)s bgm.awb -or -os \"[alt%%%%i]\"\n"
" - make .txtp and rename repeated names with a custom suffix per repeat 'i'\n\n"
)
p = argparse.ArgumentParser(description=description, epilog=epilog, formatter_class=argparse.RawTextHelpFormatter)
@ -48,6 +50,8 @@ class Cli(object):
p.add_argument('-z', dest='zero_fill', help="Zero-fill subsong number (default: auto per subsongs)", type=int)
p.add_argument('-ie', dest='no_internal_ext', help="Remove internal name's extension if any", action='store_true')
p.add_argument('-m', dest='mini_txtp', help="Create mini-txtp", action='store_true')
p.add_argument('-s', dest='subsong_start', help="Start subsong", type=int)
p.add_argument('-S', dest='subsong_end', help="End subsong", type=int)
p.add_argument('-o', dest='overwrite', help="Overwrite existing .txtp\n(beware when using with internal names alone)", action='store_true')
p.add_argument('-oi', dest='overwrite_ignore', help="Ignore repeated rather than overwritting.txtp\n", action='store_true')
p.add_argument('-or', dest='overwrite_rename', help="Rename rather than overwriting", action='store_true')
@ -264,8 +268,22 @@ class TxtpMaker(object):
exists = os.path.exists(outname)
if exists and (cfg.overwrite_rename or cfg.overwrite_suffix):
must_rename = True
if cfg.overwrite_suffix:
outname = outname.replace(".txtp", "%s.txtp" % (cfg.overwrite_suffix))
suffix = cfg.overwrite_suffix
if suffix:
outname = outname.replace(".txtp", "%s.txtp" % (suffix))
if '%' in suffix:
if outname in self.rename_map:
rename_count = self.rename_map[outname]
else:
rename_count = 0
self.rename_map[outname] = rename_count + 1
if rename_count:
outname = outname % (rename_count)
else:
#outname = outname.replace("%i", "")
outname = re.sub(r"\%[0-9]*i", "", outname)
must_rename = os.path.exists(outname)
if must_rename:
@ -459,8 +477,10 @@ class App(object):
files = []
for root, dirnames, filenames in os.walk(dir):
for filename in fnmatch.filter(filenames, pattern):
files.append(os.path.join(root, filename))
# manually test name as is too, as filter wouldn't handle stuff like "bgm[us].wav"
for filename in filenames:
if filename == pattern or fnmatch.fnmatch(filename, pattern):
files.append(os.path.join(root, filename))
if not self.cfg.recursive:
break
@ -496,7 +516,13 @@ class App(object):
created = 0
dupes = 0
errors = 0
target_subsong = 1
subsong_start = self.cfg.subsong_start
subsong_end = self.cfg.subsong_end
if not subsong_start:
subsong_start = 1
target_subsong = subsong_start
while True:
try:
# main call to vgmstream
@ -512,7 +538,7 @@ class App(object):
errors += 1
break
if target_subsong == 1:
if target_subsong == subsong_start:
log.debug("processing %s...", filename_in_clean)
if not maker.is_ignorable():
@ -526,6 +552,8 @@ class App(object):
if not maker.has_more_subsongs(target_subsong):
break
if target_subsong >= subsong_end:
break
target_subsong += 1
if target_subsong % 200 == 0: