From 60a987485916be2b4d2ea2452ff923d68fa3e7b4 Mon Sep 17 00:00:00 2001
From: Martin Genet <martin.genet@polytechnique.edu>
Date: Fri, 16 Mar 2018 21:52:14 +0100
Subject: [PATCH] adding option for reference mesh extention
---
compute_strains.py | 8 +++++---
1 file changed, 5 insertions(+), 3 deletions(-)
diff --git a/compute_strains.py b/compute_strains.py
index 73e7e2c..6346951 100644
--- a/compute_strains.py
+++ b/compute_strains.py
@@ -29,6 +29,7 @@ def compute_strains(
strain_array_name="Strain",
mesh_w_local_basis_folder=None,
mesh_w_local_basis_basename=None,
+ mesh_w_local_basis_ext="vtk",
CYL_or_PPS="PPS",
write_strains=1,
temporal_offset=None,
@@ -40,9 +41,8 @@ def compute_strains(
verbose=1):
if (mesh_w_local_basis_folder is not None) and (mesh_w_local_basis_basename is not None):
- mesh_w_local_basis_filename = mesh_w_local_basis_folder+"/"+mesh_w_local_basis_basename+"-WithLocalBasis.vtk"
mesh_w_local_basis = myvtk.readUGrid(
- filename=mesh_w_local_basis_filename,
+ filename=mesh_w_local_basis_folder+"/"+mesh_w_local_basis_basename+"."+mesh_w_local_basis_ext,
verbose=verbose)
mesh_w_local_basis_n_cells = mesh_w_local_basis.GetNumberOfCells()
if (verbose): print "mesh_w_local_basis_n_cells = " + str(mesh_w_local_basis_n_cells)
@@ -95,13 +95,15 @@ def compute_strains(
farray_F0 = ref_mesh.GetCellData().GetArray(defo_grad_array_name)
for k_frame in xrange(n_frames):
+ print "k_frame = "+str(k_frame)
+
mesh_filename = working_folder+"/"+working_basename+"_"+str(k_frame).zfill(working_zfill)+"."+working_ext
mesh = myvtk.readUGrid(
filename=mesh_filename,
verbose=verbose)
n_cells = mesh.GetNumberOfCells()
if (mesh_w_local_basis is not None):
- assert (n_cells == mesh_w_local_basis_n_cells)
+ assert (n_cells == mesh_w_local_basis_n_cells), "mesh_w_local_basis_n_cells ("+str(mesh_w_local_basis_n_cells)+") ≠n_cells ("+str(n_cells)+"). Aborting."
mesh.GetCellData().AddArray(mesh_w_local_basis.GetCellData().GetArray("part_id"))
mesh.GetCellData().AddArray(mesh_w_local_basis.GetCellData().GetArray("sector_id"))
myvtk.addDeformationGradients(
--
GitLab