#!/usr/bin/env python

import copy
import json

from pathlib import Path

SIZE = [384, 384, 384]

with (Path(__file__).parent.parent / "templates" / "strong-gpu.json").open(encoding="utf8") as f:
    template = json.load(f)

configs = [
    [  1,  1,  1],
    [  1,  1,  2],
    [  1,  2,  2],
    [  2,  2,  2],
    [  2,  2,  4],
    [  2,  4,  4],
    [  4,  4,  4],
    [  4,  4,  8],
]

out_path = Path(__file__).parent.parent / "generated" / "config"

for c in configs:
    nc = copy.deepcopy(template)
    nc["Geometry"]["blockcount"] = c
    nc["Geometry"]["blocksize"] = [bs // bc for bc, bs in zip(c, SIZE)]

    nc_out_path = out_path / f"strong-gpu-{c[0]:02}-{c[1]:02}-{c[2]:02}.json"
    print(f"Dumping {(c[0] * c[1] * c[2])} to {nc_out_path}")
    with nc_out_path.open("w", encoding="utf8") as f:
        json.dump(nc, f)