Commit 9a65a3410352e6637ff61fc6c5172b7ea6b63056
1 parent
553830f4
Exists in
master
and in
5 other branches
FIX: Workaround to solve problem when using multiprocessing under Windows.
Showing
1 changed file
with
24 additions
and
2 deletions
Show diff stats
invesalius/data/surface_process.py
| ... | ... | @@ -7,10 +7,31 @@ import vtk |
| 7 | 7 | |
| 8 | 8 | import i18n |
| 9 | 9 | import converters |
| 10 | -import imagedata_utils as iu | |
| 10 | +# import imagedata_utils as iu | |
| 11 | 11 | |
| 12 | 12 | from scipy import ndimage |
| 13 | 13 | |
| 14 | +# TODO: Code duplicated from file {imagedata_utils.py}. | |
| 15 | +def ResampleImage3D(imagedata, value): | |
| 16 | + """ | |
| 17 | + Resample vtkImageData matrix. | |
| 18 | + """ | |
| 19 | + spacing = imagedata.GetSpacing() | |
| 20 | + extent = imagedata.GetExtent() | |
| 21 | + size = imagedata.GetDimensions() | |
| 22 | + | |
| 23 | + width = float(size[0]) | |
| 24 | + height = float(size[1]/value) | |
| 25 | + | |
| 26 | + resolution = (height/(extent[1]-extent[0])+1)*spacing[1] | |
| 27 | + | |
| 28 | + resample = vtk.vtkImageResample() | |
| 29 | + resample.SetInput(imagedata) | |
| 30 | + resample.SetAxisMagnificationFactor(0, resolution) | |
| 31 | + resample.SetAxisMagnificationFactor(1, resolution) | |
| 32 | + | |
| 33 | + return resample.GetOutput() | |
| 34 | + | |
| 14 | 35 | class SurfaceProcess(multiprocessing.Process): |
| 15 | 36 | |
| 16 | 37 | def __init__(self, pipe, filename, shape, dtype, mask_filename, |
| ... | ... | @@ -100,7 +121,8 @@ class SurfaceProcess(multiprocessing.Process): |
| 100 | 121 | del a_image |
| 101 | 122 | |
| 102 | 123 | if self.imagedata_resolution: |
| 103 | - image = iu.ResampleImage3D(image, self.imagedata_resolution) | |
| 124 | + # image = iu.ResampleImage3D(image, self.imagedata_resolution) | |
| 125 | + image = ResampleImage3D(image, self.imagedata_resolution) | |
| 104 | 126 | |
| 105 | 127 | flip = vtk.vtkImageFlip() |
| 106 | 128 | flip.SetInput(image) | ... | ... |