Spaces:
Sleeping
Sleeping
Gallinator commited on
preprocessing: programmtically create config.json from input boundary conditions
Browse files- app/preprocessing.py +8 -1
app/preprocessing.py
CHANGED
|
@@ -51,7 +51,7 @@ def create_session_folders(assets_dir: str, session_dir: str, predict_input: Pre
|
|
| 51 |
session_dir.mkdir(parents=True, exist_ok=True)
|
| 52 |
|
| 53 |
shutil.copytree(assets_dir / "openfoam-case-template", session_dir / "assets" / "openfoam-case-template")
|
| 54 |
-
|
| 55 |
split_dir = session_dir / "assets" / "meshes" / "split"
|
| 56 |
split_dir.mkdir(exist_ok=True, parents=True)
|
| 57 |
shutil.copy(assets_dir / "transforms.json", split_dir)
|
|
@@ -59,6 +59,13 @@ def create_session_folders(assets_dir: str, session_dir: str, predict_input: Pre
|
|
| 59 |
data_dir = session_dir / "data"
|
| 60 |
data_dir.mkdir(parents=True, exist_ok=True)
|
| 61 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 62 |
|
| 63 |
def path_to_obj(x: list[float], y: list[float], dest_path: str):
|
| 64 |
bpy.ops.object.select_all(action='SELECT')
|
|
|
|
| 51 |
session_dir.mkdir(parents=True, exist_ok=True)
|
| 52 |
|
| 53 |
shutil.copytree(assets_dir / "openfoam-case-template", session_dir / "assets" / "openfoam-case-template")
|
| 54 |
+
|
| 55 |
split_dir = session_dir / "assets" / "meshes" / "split"
|
| 56 |
split_dir.mkdir(exist_ok=True, parents=True)
|
| 57 |
shutil.copy(assets_dir / "transforms.json", split_dir)
|
|
|
|
| 59 |
data_dir = session_dir / "data"
|
| 60 |
data_dir.mkdir(parents=True, exist_ok=True)
|
| 61 |
|
| 62 |
+
model_dir = "pipn" if "pipn" in predict_input.model else "pi-gano"
|
| 63 |
+
shutil.copy(assets_dir / model_dir / "data_config.json", session_dir / "assets")
|
| 64 |
+
|
| 65 |
+
copy_config_with_boundary_conditions(assets_dir / "config.json",
|
| 66 |
+
session_dir / "assets" / "meshes" / "split",
|
| 67 |
+
predict_input)
|
| 68 |
+
|
| 69 |
|
| 70 |
def path_to_obj(x: list[float], y: list[float], dest_path: str):
|
| 71 |
bpy.ops.object.select_all(action='SELECT')
|