44 lines
1.4 KiB
Python
44 lines
1.4 KiB
Python
import os
|
|
import sys
|
|
import re
|
|
from pathlib import Path
|
|
|
|
def safe_var_name(path):
|
|
return re.sub("__+", "_", ''.join(c if c.isalnum() else '_' for c in str(path)))
|
|
|
|
def safe_namespace_name(path):
|
|
return re.sub(":::+", "::", ''.join(c if (c.isalnum() or c == '_') else '::' for c in str(path)))
|
|
|
|
def embed(input_path, output_path, namespace, var_name):
|
|
with open(input_path, "rt") as f:
|
|
contents = f.read()
|
|
out = (
|
|
f"// Generated from {input_path}\n"
|
|
f"#pragma once\n\n"
|
|
f"namespace {safe_namespace_name(namespace)}" "\n{\n"
|
|
f"\tinline constexpr char {safe_var_name(var_name)}_str[] = R\"(\"{contents}\")\";\n""}"
|
|
)
|
|
os.makedirs(os.path.dirname(output_path), exist_ok=True)
|
|
with open(output_path.with_suffix(output_path.suffix + ".h"), "w") as f:
|
|
f.write(out)
|
|
|
|
def main(input_dir, output_dir):
|
|
input_path = Path(input_dir)
|
|
|
|
for root, _, files in os.walk(input_dir):
|
|
for file in files:
|
|
print(f"Processing file: %{file}%")
|
|
file_path = Path(root) / file
|
|
|
|
parent_path = file_path.relative_to(input_dir)
|
|
|
|
output_path = Path(output_dir) / parent_path
|
|
|
|
embed(file_path, output_path, parent_path, file_path.name)
|
|
|
|
if __name__ == "__main__":
|
|
if len(sys.argv) != 3:
|
|
print("Expected two arguments: input_dir output_dir")
|
|
exit(1)
|
|
main(sys.argv[1], sys.argv[2])
|