Skip to content

Commit 031956f

Browse files
committed
BUG: Dynamically generate extractor and shrinker
To support adding the image or label image after initialization.
1 parent 2030d43 commit 031956f

File tree

1 file changed

+4
-6
lines changed

1 file changed

+4
-6
lines changed

itkwidgets/widget_viewer.py

Lines changed: 4 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -403,12 +403,6 @@ def __init__(self, **kwargs): # noqa: C901
403403
for dim in range(dimension):
404404
if size[dim] > self.size_limit_3d[dim]:
405405
self._downsampling = True
406-
if self._downsampling and self.image:
407-
self.extractor = itk.ExtractImageFilter.New(self.image)
408-
self.shrinker = itk.BinShrinkImageFilter.New(self.extractor)
409-
if self._downsampling and self.label_image:
410-
self.label_image_extractor = itk.ExtractImageFilter.New(self.label_image)
411-
self.label_image_shrinker = itk.ShrinkImageFilter.New(self.label_image_extractor)
412406
self._update_rendered_image()
413407
if self._downsampling:
414408
self.observe(self._on_roi_changed, ['roi'])
@@ -488,8 +482,12 @@ def f():
488482
self.size_limit_3d, dimension, size)
489483
self._scale_factors = np.array(scale_factors, dtype=np.uint8)
490484
if self.image:
485+
self.extractor = itk.ExtractImageFilter.New(self.image)
486+
self.shrinker = itk.ShrinkImageFilter.New(self.extractor)
491487
self.shrinker.SetShrinkFactors(scale_factors[:dimension])
492488
if self.label_image:
489+
self.label_image_extractor = itk.ExtractImageFilter.New(self.label_image)
490+
self.label_image_shrinker = itk.ShrinkImageFilter.New(self.label_image_extractor)
493491
self.label_image_shrinker.SetShrinkFactors(scale_factors[:dimension])
494492

495493
region = itk.ImageRegion[dimension]()

0 commit comments

Comments
 (0)