def generate_header(src, dst, version_dst): from compat import byte_to_str with open(dst, 'w') as header: header.write('/* THIS FILE IS GENERATED DO NOT EDIT */\n') header.write('#ifndef CS_COMPRESSED_H\n') header.write('#define CS_COMPRESSED_H\n\n') header.write('#ifdef TOOLS_ENABLED\n\n') header.write('#include "core/map.h"\n') header.write('#include "core/ustring.h"\n') inserted_files = '' import os latest_mtime = 0 cs_file_count = 0 for root, _, files in os.walk(src): files = [f for f in files if f.endswith('.cs')] for file in files: cs_file_count += 1 filepath = os.path.join(root, file) filepath_src_rel = os.path.relpath(filepath, src) mtime = os.path.getmtime(filepath) latest_mtime = mtime if mtime > latest_mtime else latest_mtime with open(filepath, 'rb') as f: buf = f.read() decomp_size = len(buf) import zlib buf = zlib.compress(buf) name = str(cs_file_count) header.write('\n') header.write('// ' + filepath_src_rel + '\n') header.write('static const int _cs_' + name + '_compressed_size = ' + str(len(buf)) + ';\n') header.write('static const int _cs_' + name + '_uncompressed_size = ' + str(decomp_size) + ';\n') header.write('static const unsigned char _cs_' + name + '_compressed[] = { ') for i, buf_idx in enumerate(range(len(buf))): if i > 0: header.write(', ') header.write(byte_to_str(buf[buf_idx])) inserted_files += '\tr_files.insert("' + filepath_src_rel.replace('\\', '\\\\') + '", ' \ 'CompressedFile(_cs_' + name + '_compressed_size, ' \ '_cs_' + name + '_uncompressed_size, ' \ '_cs_' + name + '_compressed));\n' header.write(' };\n') header.write('\nstruct CompressedFile\n' '{\n' '\tint compressed_size;\n' '\tint uncompressed_size;\n' '\tconst unsigned char* data;\n' '\n\tCompressedFile(int p_comp_size, int p_uncomp_size, const unsigned char* p_data)\n' '\t{\n' '\t\tcompressed_size = p_comp_size;\n' '\t\tuncompressed_size = p_uncomp_size;\n' '\t\tdata = p_data;\n' '\t}\n' '\n\tCompressedFile() {}\n' '};\n' '\nvoid get_compressed_files(Map& r_files)\n' '{\n' + inserted_files + '}\n' ) header.write('\n#endif // TOOLS_ENABLED\n') header.write('\n#endif // CS_COMPRESSED_H\n') glue_version = int(latest_mtime) # The latest modified time will do for now with open(version_dst, 'w') as version_header: version_header.write('/* THIS FILE IS GENERATED DO NOT EDIT */\n') version_header.write('#ifndef CS_GLUE_VERSION_H\n') version_header.write('#define CS_GLUE_VERSION_H\n\n') version_header.write('#define CS_GLUE_VERSION UINT32_C(' + str(glue_version) + ')\n') version_header.write('\n#endif // CS_GLUE_VERSION_H\n')