aboutsummaryrefslogtreecommitdiff
path: root/scripts/zstd.py
diff options
context:
space:
mode:
authorAqua-sama <aqua@iserlohn-fortress.net>2020-03-30 14:08:41 +0300
committerAqua-sama <aqua@iserlohn-fortress.net>2020-03-30 14:17:00 +0300
commitf9ea03470e3a0bbe67ccb0a531dcf0e39aec1e90 (patch)
treea2079e6810e083ed437d2c0f1b9b1fe0bba27776 /scripts/zstd.py
parentDrop dependency on serge-sans-paille/frozen (diff)
downloadrcc-f9ea03470e3a0bbe67ccb0a531dcf0e39aec1e90.tar.xz
Move scripts/rcc to top level
- add rcc generator for use when importing as subproject
Diffstat (limited to 'scripts/zstd.py')
-rw-r--r--scripts/zstd.py43
1 files changed, 0 insertions, 43 deletions
diff --git a/scripts/zstd.py b/scripts/zstd.py
deleted file mode 100644
index 2cfa149..0000000
--- a/scripts/zstd.py
+++ /dev/null
@@ -1,43 +0,0 @@
-import subprocess
-from rcc_format import *
-
-def zstd(filelist, args):
- if args.train is not None:
- train(filelist, args.train, args.binary, args.dsize)
- return
-
- write_header(args.output, args.namespace)
-
- for f in filelist:
- with open(f.path, 'rb') as contents:
- write_item(args.output, f.variable, compress(contents, args.binary, args.level, dictionary=args.dict))
-
- write_entries(args.output, filelist)
- if args.dict is not None:
- write_item(args.output, 'dict', args.dict.read())
- print("constexpr auto dictionary = std::span(dict, dict_len);", file=args.output)
- else:
- print("constexpr std::span<const uint8_t> dictionary {};", file=args.output)
-
- print("constexpr auto compression = embed::Zstd;", file=args.output)
-
- write_footer(args.output, args.namespace)
-
-def train(filelist, output, zstd_bin, maxdict):
- cmd = [ zstd_bin, '--train', '--maxdict=' + str(maxdict), '-o', output.name ]
-
- for f in filelist:
- cmd.append(f.path)
-
- subprocess.run(cmd)
-
-def compress(file, zstd_bin, level, dictionary=None):
- cmd = [ zstd_bin, '--compress', '--stdout', '-' + str(level) ]
-
- if dictionary is not None:
- cmd.append('-D')
- cmd.append(dictionary.name)
-
- cmd.append(file.name)
- return subprocess.run(cmd, capture_output=True).stdout
-