Skip to content
Snippets Groups Projects
Commit 3602753c authored by Florian Zill's avatar Florian Zill
Browse files

[propertylib] small fix for mesh_dependent funcs

parent b441a1af
No related branches found
No related tags found
1 merge request!134MeshSeries aggregation function
...@@ -48,11 +48,16 @@ def depth(mesh: pv.UnstructuredGrid, use_coords: bool = False) -> np.ndarray: ...@@ -48,11 +48,16 @@ def depth(mesh: pv.UnstructuredGrid, use_coords: bool = False) -> np.ndarray:
] ]
edge_centers = edges.cell_centers().points edge_centers = edges.cell_centers().points
adj_cells = [mesh.find_closest_cell(point) for point in edge_centers] adj_cells = [mesh.find_closest_cell(point) for point in edge_centers]
adj_cell_centers = mesh.extract_cells(adj_cells).cell_centers().points adj_centers = np.vstack(
are_above = ( [
edge_centers[..., vertical_dim] mesh.extract_cells(adj_cell).cell_centers().points
> adj_cell_centers[..., vertical_dim] for adj_cell in adj_cells
]
) )
are_above = [
edge_center[vertical_dim] > adj_center[vertical_dim]
for edge_center, adj_center in zip(edge_centers, adj_centers)
]
are_non_vertical = np.asarray(edge_horizontal_extent) > 1e-12 are_non_vertical = np.asarray(edge_horizontal_extent) > 1e-12
top_cells = are_above & are_non_vertical top_cells = are_above & are_non_vertical
top = edges.extract_cells(top_cells) top = edges.extract_cells(top_cells)
...@@ -110,8 +115,10 @@ def fluid_pressure_criterion( ...@@ -110,8 +115,10 @@ def fluid_pressure_criterion(
""" """
Qty = u_reg.Quantity Qty = u_reg.Quantity
sigma = -Qty(mesh[mesh_property.data_name], mesh_property.data_unit) sigma = mesh[mesh_property.data_name]
return p_fluid(mesh) - eigenvalues(sigma)[..., 0] return p_fluid(mesh) - Qty(
eigenvalues(-sigma)[..., 0], mesh_property.data_unit
)
def dilatancy_critescu( def dilatancy_critescu(
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment