Merge pull request #128 from KVSlab/revert-network-bc

Update to Network BCs
KVSlab · Oct 25, 2023 · 29efc7f · 29efc7f
2 parents 7f6c1b8 + aa9d0d7
commit 29efc7f
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 3 deletions.
diff --git a/src/vampy/automatedPreprocessing/automated_preprocessing.py b/src/vampy/automatedPreprocessing/automated_preprocessing.py
@@ -396,7 +396,8 @@ def run_pre_processing(input_model, verbose_print, smoothing_method, smoothing_f
  else:
  mesh = read_polydata(file_name_vtu_mesh)
 
- network, probe_points = setup_model_network(centerlines, file_name_probe_points, region_center, verbose_print)
+ network, probe_points = setup_model_network(centerlines, file_name_probe_points, region_center, verbose_print,
+ has_outlet, is_atrium)
 
  # Load updated parameters following meshing
  parameters = get_parameters(base_path)

diff --git a/src/vampy/automatedPreprocessing/preprocessing_common.py b/src/vampy/automatedPreprocessing/preprocessing_common.py
@@ -653,7 +653,7 @@ def find_boundaries(model_path, mean_inflow_rate, network, mesh, verbose_print,
  write_parameters(info, model_path)
 
 
-def setup_model_network(centerlines, file_name_probe_points, region_center, verbose_print):
+def setup_model_network(centerlines, file_name_probe_points, region_center, verbose_print, has_outlet, is_atrium):
  """
  Sets up network used for network boundary condition model.
 
@@ -662,6 +662,8 @@ def setup_model_network(centerlines, file_name_probe_points, region_center, verb
  file_name_probe_points (str): Save path of probe points
  region_center (list): List of points representing region of refinement
  verbose_print (bool): Prints additional info if True
+ has_outlet (bool): Determines if model has outlet or not
+ is_atrium (bool): Determines if model is atrium or artery
 
  Returns:
  network (Network): Network model
@@ -688,7 +690,11 @@ def setup_model_network(centerlines, file_name_probe_points, region_center, verb
  # Set the flow split and inlet boundary condition
  # Compute the outlet boundary condition percentages.
  flowSplitting = FlowSplitting()
- flowSplitting.ComputeGammas(network, verbose_print)
+ if not is_atrium and has_outlet:
+ flowSplitting.ComputeAlphas(network, verbose_print)
+ flowSplitting.ComputeBetas(network, verbose_print)
+ else:
+ flowSplitting.ComputeGammas(network, verbose_print)
  flowSplitting.CheckTotalFlowRate(network, verbose_print)
 
  return network, probe_points