Gallinator commited on
Commit
3bffa8c
·
unverified ·
1 Parent(s): 0835c28

preprocessing: programmtically create config.json from input boundary conditions

Browse files
Files changed (1) hide show
  1. 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
- shutil.copy(assets_dir / "data_config.json", session_dir / "assets")
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')