From d83afeed3bcd189e347bb9560472abd0e98d71d4 Mon Sep 17 00:00:00 2001 From: Martin Genet <martin.genet@polytechnique.edu> Date: Tue, 4 Jun 2019 18:11:59 +0200 Subject: [PATCH] Cleaning --- compute_displacements.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/compute_displacements.py b/compute_displacements.py index 8b539df..cca255e 100644 --- a/compute_displacements.py +++ b/compute_displacements.py @@ -56,7 +56,11 @@ def compute_displacements( filename=working_folder+"/"+working_basename+"_"+str(k_frame).zfill(working_zfill)+"."+working_ext, verbose=verbose) cur_disp_farray = cur_mesh.GetPointData().GetVectors() - [warped_disp_farray.SetTuple(k_point, numpy.array(cur_disp_farray.GetTuple(k_point)) - numpy.array(ref_disp_farray.GetTuple(k_point))) for k_point in xrange(n_points)] + [warped_disp_farray.SetTuple( + k_point, + numpy.substract( + cur_disp_farray.GetTuple(k_point), + ref_disp_farray.GetTuple(k_point))) for k_point in xrange(n_points)] myvtk.writeUGrid( ugrid=warped_mesh, filename=working_folder+"/"+working_basename+("-"+suffix)*(suffix!="")+"_"+str(k_frame).zfill(working_zfill)+"."+working_ext, -- GitLab