taisei/scripts/automaterials.py
Andrei Alexeyev 78266ace3f
pbr: refactor material management and shader
Introduced a new PBRMaterial resource type. Materials are now fully
described with .material files instead of being hardcoded. Added some
functions to abstract away and simplify set up of PBR shaders.
2021-09-01 00:21:50 +03:00

67 lines
1.4 KiB
Python
Executable file

#!/usr/bin/env python3
import re
from pathlib import Path
from collections import defaultdict
from taiseilib.common import (
run_main,
)
mat_re = re.compile(r'(.*)_(diffuse|normal|ambient|roughness)\.(?:tex|basis|png|webp)$')
def main(args):
resdir = Path(__file__).parent.parent / 'resources'
materials = defaultdict(set)
gfxdirs = []
for pkgdir in resdir.glob('*.pkgdir'):
gfx = pkgdir / 'gfx'
if not gfx.is_dir():
continue
gfxdirs.append(gfx)
print(gfx)
for p in gfx.glob('**/*'):
rel = p.relative_to(gfx)
m = mat_re.match(str(rel))
if m is None:
continue
matname, mapname = m.groups()
materials[matname].add(mapname)
gfxmain = gfxdirs[0]
for mat, maps in materials.items():
mat_filename = f'{mat}.material'
skip = False
for gfx in gfxdirs:
if (gfx / mat_filename).exists():
print(' - Skip', mat_filename)
skip = True
break
if skip:
continue
outpath = (gfxmain / mat_filename).resolve()
lines = ['']
for mapname in sorted(maps):
lines.append(f'{mapname}_map = {mat}_{mapname}')
lines.append('')
print(' * Write', outpath)
outpath.write_text('\n'.join(lines))
if __name__ == '__main__':
run_main(main)