File size: 2,414 Bytes
338c3e4 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 | import json
import shutil
import sys
from pathlib import Path
import numpy as np
PROJECT_ROOT = Path(__file__).resolve().parents[1]
sys.path.insert(0, str(PROJECT_ROOT))
from onescience.utils.YParams import YParams
def resolve_path(path_value):
path = Path(path_value)
return path if path.is_absolute() else PROJECT_ROOT / path
def make_case(case_dir, case_id, rows, cols, frames, rng):
case_dir.mkdir(parents=True, exist_ok=True)
vel_in = 1.0 + 0.05 * case_id
params = {
"vel_in": vel_in,
"density": 1.0 + 0.1 * (case_id % 3),
"viscosity": 0.001 + 0.0001 * (case_id % 5),
"height": 1.0,
"width": 1.0,
}
y = np.linspace(0.0, 1.0, rows, dtype=np.float32)
x = np.linspace(0.0, 1.0, cols, dtype=np.float32)
yy, xx = np.meshgrid(y, x, indexing="ij")
u = np.empty((frames, rows, cols), dtype=np.float32)
v = np.empty_like(u)
phase = 0.2 * case_id
for t in range(frames):
amp = vel_in * (1.0 + 0.03 * t)
u[t] = amp * np.sin(np.pi * yy) * (1.0 - 0.3 * xx) + 0.01 * rng.standard_normal((rows, cols))
v[t] = 0.2 * amp * np.cos(np.pi * xx + phase) * yy + 0.01 * rng.standard_normal((rows, cols))
(case_dir / "case.json").write_text(json.dumps(params, indent=2), encoding="utf-8")
np.save(case_dir / "u.npy", u)
np.save(case_dir / "v.npy", v)
def main():
cfg = YParams(str(PROJECT_ROOT / "conf" / "config.yaml"), "root")
data_dir = resolve_path(cfg.datapipe.source.data_dir)
fake_cfg = cfg.fake_data
data_cfg = cfg.datapipe.data
if data_dir.exists():
shutil.rmtree(data_dir)
rng = np.random.default_rng(fake_cfg.seed)
problem_dir = data_dir / "tube"
case_id = 0
for subset in ("prop", "bc", "geo"):
for subset_idx in range(fake_cfg.num_cases_per_subset):
make_case(
problem_dir / subset / f"case{case_id}",
case_id=case_id + subset_idx,
rows=data_cfg.num_rows,
cols=data_cfg.num_cols,
frames=fake_cfg.num_frames,
rng=rng,
)
case_id += 1
total_cases = fake_cfg.num_cases_per_subset * 3
print(f"Fake CFDBench data written to {data_dir}")
print(f"cases={total_cases}, frames={fake_cfg.num_frames}, grid=({data_cfg.num_rows}, {data_cfg.num_cols})")
if __name__ == "__main__":
main()
|