Mentions légales du service

Skip to content
Snippets Groups Projects
Commit 4c4f2dc9 authored by PATTE Cecile's avatar PATTE Cecile
Browse files

add field_name option

parent 8cbe4001
No related branches found
No related tags found
No related merge requests found
Hide whitespace changes
Inline Side-by-side
compute_warped_images.py
+ 8
− 6
View file @ 4c4f2dc9
...@@ -29,6 +29,7 @@ def compute_warped_images( ...@@ -29,6 +29,7 @@ def compute_warped_images(
ref_frame=0, ref_frame=0,
ref_image_model=None, ref_image_model=None,
working_ext="vtu", working_ext="vtu",
field_name="displacement",
verbose=0): verbose=0):
ref_image_zfill = len(glob.glob(ref_image_folder+"/"+ref_image_basename+"_*.vti")[0].rsplit("_")[-1].split(".")[0]) ref_image_zfill = len(glob.glob(ref_image_folder+"/"+ref_image_basename+"_*.vti")[0].rsplit("_")[-1].split(".")[0])
...@@ -56,7 +57,7 @@ def compute_warped_images( ...@@ -56,7 +57,7 @@ def compute_warped_images(
working_zfill = len(glob.glob(working_folder+"/"+working_basename+"_*."+working_ext)[0].rsplit("_")[-1].split(".")[0]) working_zfill = len(glob.glob(working_folder+"/"+working_basename+"_*."+working_ext)[0].rsplit("_")[-1].split(".")[0])
n_frames = len(glob.glob(working_folder+"/"+working_basename+"_"+"[0-9]"*working_zfill+"."+working_ext)) n_frames = len(glob.glob(working_folder+"/"+working_basename+"_"+"[0-9]"*working_zfill+"."+working_ext))
#n_frames = 1 # n_frames = 1
X = numpy.empty(3) X = numpy.empty(3)
U = numpy.empty(3) U = numpy.empty(3)
...@@ -68,7 +69,8 @@ def compute_warped_images( ...@@ -68,7 +69,8 @@ def compute_warped_images(
mesh = myvtk.readUGrid( mesh = myvtk.readUGrid(
filename=working_folder+"/"+working_basename+"_"+str(k_frame).zfill(working_zfill)+"."+working_ext) filename=working_folder+"/"+working_basename+"_"+str(k_frame).zfill(working_zfill)+"."+working_ext)
#print mesh # print mesh
mesh.GetPointData().SetActiveVectors(field_name)
warp = vtk.vtkWarpVector() warp = vtk.vtkWarpVector()
if (vtk.vtkVersion.GetVTKMajorVersion() >= 6): if (vtk.vtkVersion.GetVTKMajorVersion() >= 6):
...@@ -77,9 +79,9 @@ def compute_warped_images( ...@@ -77,9 +79,9 @@ def compute_warped_images(
warp.SetInput(mesh) warp.SetInput(mesh)
warp.Update() warp.Update()
warped_mesh = warp.GetOutput() warped_mesh = warp.GetOutput()
#myvtk.writeUGrid( myvtk.writeUGrid(
#ugrid=warped_mesh, ugrid=warped_mesh,
#filename=working_folder+"/"+working_basename+"-warped_"+str(k_frame).zfill(working_zfill)+"."+working_ext) filename=working_folder+"/"+working_basename+"-warped_"+str(k_frame).zfill(working_zfill)+"."+working_ext)
probe = vtk.vtkProbeFilter() probe = vtk.vtkProbeFilter()
if (vtk.vtkVersion.GetVTKMajorVersion() >= 6): if (vtk.vtkVersion.GetVTKMajorVersion() >= 6):
...@@ -91,7 +93,7 @@ def compute_warped_images( ...@@ -91,7 +93,7 @@ def compute_warped_images(
probe.Update() probe.Update()
probed_image = probe.GetOutput() probed_image = probe.GetOutput()
scalars_mask = probed_image.GetPointData().GetArray("vtkValidPointMask") scalars_mask = probed_image.GetPointData().GetArray("vtkValidPointMask")
scalars_U = probed_image.GetPointData().GetArray("displacement") scalars_U = probed_image.GetPointData().GetArray(field_name)
#myvtk.writeImage( #myvtk.writeImage(
#image=probed_image, #image=probed_image,
#filename=working_folder+"/"+working_basename+"_"+str(k_frame).zfill(working_zfill)+".vti") #filename=working_folder+"/"+working_basename+"_"+str(k_frame).zfill(working_zfill)+".vti")
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment