ADD: function add points from edited image

Paulo Henrique Junqueira Amorim
1 parent 6a3795bd
Showing 1 changed file with 15 additions and 0 deletions Show diff stats
invesalius/data/imagedata_utils.py
@@ -109,3 +109,18 @@ def FixGantryTilt(imagedata, tilt):
     imagedata.UpdateInformation()
  
     return imagedata
+
+
+def BuildEditedImage(imagedata, points):
+    """
+    Editing the original image in accordance with the edit
+    points in the editor, it is necessary to generate the
+    vtkPolyData via vtkContourFilter
+    """
+    for point in points:
+        x, y, z = point
+        colour = points[point]
+        imagedata.SetScalarComponentFromDouble(x, y, z, 0, colour)
+        imagedata.Update()
+
+    return imagedata
...	...	@@ -109,3 +109,18 @@ def FixGantryTilt(imagedata, tilt):
109	109	imagedata.UpdateInformation()
110	110
111	111	return imagedata
	112	+
	113	+
	114	+def BuildEditedImage(imagedata, points):
	115	+ """
	116	+ Editing the original image in accordance with the edit
	117	+ points in the editor, it is necessary to generate the
	118	+ vtkPolyData via vtkContourFilter
	119	+ """
	120	+ for point in points:
	121	+ x, y, z = point
	122	+ colour = points[point]
	123	+ imagedata.SetScalarComponentFromDouble(x, y, z, 0, colour)
	124	+ imagedata.Update()
	125	+
	126	+ return imagedata
...	...