From 73d258316540c375b5730fced145655de7c41444 Mon Sep 17 00:00:00 2001
From: Martin Genet <martin.genet@polytechnique.edu>
Date: Tue, 7 Aug 2018 15:51:54 +0200
Subject: [PATCH] Generate images and image gradients in different files
---
generate_images.py | 18 +++++++++++++++---
1 file changed, 15 insertions(+), 3 deletions(-)
diff --git a/generate_images.py b/generate_images.py
index 5acf74b..fe2a56f 100644
--- a/generate_images.py
+++ b/generate_images.py
@@ -642,6 +642,10 @@ def generateImages(
origin[2] = 0.
vtk_image.SetOrigin(origin)
+ if (generate_image_gradient):
+ vtk_gradient = vtk.vtkImageData()
+ vtk_gradient.DeepCopy(vtk_image)
+
n_points = vtk_image.GetNumberOfPoints()
vtk_image_scalars = myvtk.createFloatArray(
name="ImageScalars",
@@ -649,13 +653,14 @@ def generateImages(
n_tuples=n_points,
verbose=verbose-1)
vtk_image.GetPointData().SetScalars(vtk_image_scalars)
+
if (generate_image_gradient):
- vtk_image_gradient = myvtk.createFloatArray(
+ vtk_gradient_vectors = myvtk.createFloatArray(
name="ImageScalarsGradient",
n_components=3,
n_tuples=n_points,
verbose=verbose-1)
- vtk_image.GetPointData().SetVectors(vtk_image_gradient)
+ vtk_gradient.GetPointData().SetScalars(vtk_gradient_vectors)
if not os.path.exists(images["folder"]):
os.mkdir(images["folder"])
@@ -741,13 +746,20 @@ def generateImages(
else:
assert (0), "n_dim must be \"1\", \"2\" or \"3\". Aborting."
vtk_image_scalars.SetTuple(k_point, I)
- if (generate_image_gradient): vtk_image_gradient.SetTuple(k_point, G)
+ if (generate_image_gradient): vtk_gradient_vectors.SetTuple(k_point, G)
if (I[0] < global_min): global_min = I[0]
if (I[0] > global_max): global_max = I[0]
+ #print vtk_image
myvtk.writeImage(
image=vtk_image,
filename=images["folder"]+"/"+images["basename"]+"_"+str(k_frame).zfill(images["zfill"])+"."+images["ext"],
verbose=verbose-1)
+ if (generate_image_gradient):
+ #print vtk_gradient
+ myvtk.writeImage(
+ image=vtk_gradient,
+ filename=images["folder"]+"/"+images["basename"]+"-grad"+"_"+str(k_frame).zfill(images["zfill"])+"."+images["ext"],
+ verbose=verbose-1)
if (images["data_type"] in ("float")):
pass
--
GitLab