72 lines
2.3 KiB
Python
72 lines
2.3 KiB
Python
|
import argparse
|
||
|
|
import os
|
||
|
|
import re
|
||
|
|
|
||
|
|
def reindent_code(shader_code: str) -> str:
|
||
|
|
indented_code = []
|
||
|
|
indentation_level = 0
|
||
|
|
tab_size = 4 # You can change this to set the desired tab size
|
||
|
|
|
||
|
|
for line in shader_code.split('\n'):
|
||
|
|
line = line.strip()
|
||
|
|
if line.endswith('}'):
|
||
|
|
indentation_level -= 1
|
||
|
|
|
||
|
|
indented_code.append(' ' * (tab_size * indentation_level) + line)
|
||
|
|
|
||
|
|
if line.endswith('{'):
|
||
|
|
indentation_level += 1
|
||
|
|
|
||
|
|
return '\n'.join(indented_code)
|
||
|
|
|
||
|
|
def c_to_glsl(input_filename: str):
|
||
|
|
# Ensure the output directory exists
|
||
|
|
if not os.path.exists("engine/shaders"):
|
||
|
|
os.makedirs("engine/shaders")
|
||
|
|
|
||
|
|
# Read the C code from the file
|
||
|
|
with open(input_filename, 'r') as f:
|
||
|
|
lines = f.readlines()
|
||
|
|
|
||
|
|
capturing = False
|
||
|
|
shader_code = ""
|
||
|
|
output_filename = None
|
||
|
|
|
||
|
|
for line in lines:
|
||
|
|
if not capturing:
|
||
|
|
match = re.search(r'static const char \*const ([a-zA-Z0-9_]+) =', line)
|
||
|
|
if match:
|
||
|
|
output_filename = f"engine/shaders/{match.group(1)}.glsl"
|
||
|
|
capturing = True
|
||
|
|
continue # skip the current line since it's just the declaration
|
||
|
|
|
||
|
|
if capturing:
|
||
|
|
if line.strip().endswith('";'):
|
||
|
|
shader_code += line.rstrip('";\n')
|
||
|
|
# Process the captured shader code
|
||
|
|
shader_code = re.sub(r'/\*.*?\*/', '', shader_code, flags=re.DOTALL) # Remove C-style comments
|
||
|
|
shader_code = re.sub(r'//.*', '', shader_code) # Remove C++ style comments
|
||
|
|
shader_code = shader_code.replace(r'"\n"', '\n').replace(r'\n', '').replace(r'"', '')
|
||
|
|
|
||
|
|
# Save to a .glsl file
|
||
|
|
with open(output_filename, 'w') as f:
|
||
|
|
shader_code = shader_code.strip()
|
||
|
|
f.write(reindent_code(shader_code.strip()))
|
||
|
|
|
||
|
|
# Reset for the next shader code
|
||
|
|
capturing = False
|
||
|
|
shader_code = ""
|
||
|
|
output_filename = None
|
||
|
|
else:
|
||
|
|
shader_code += line
|
||
|
|
|
||
|
|
def main():
|
||
|
|
parser = argparse.ArgumentParser()
|
||
|
|
parser.add_argument("--input", required=True)
|
||
|
|
|
||
|
|
args = parser.parse_args()
|
||
|
|
c_to_glsl(args.input)
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
main()
|