From 8fec8bda8abae22db33adc39cc6f7268be2eb708 Mon Sep 17 00:00:00 2001 From: kerim371 Date: Fri, 18 Mar 2022 22:20:18 +0300 Subject: [PATCH] ENH: improved control over IO volume types It is now possible to add custom volume types to node selectors: `slicer.modules.SimpleFiltersInstance.scalarVolumeNodeClasses=['...']` --- SimpleFilters/SimpleFilters.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/SimpleFilters/SimpleFilters.py b/SimpleFilters/SimpleFilters.py index bcdfddc..0af9ced 100644 --- a/SimpleFilters/SimpleFilters.py +++ b/SimpleFilters/SimpleFilters.py @@ -32,6 +32,7 @@ class SimpleFilters: import inspect __file__ = inspect.getframeinfo(inspect.currentframe())[0] + scalarVolumeNodeClasses = ["vtkMRMLScalarVolumeNode", "vtkMRMLLabelMapVolumeNode"] ICON_DIR = os.path.dirname(os.path.realpath(__file__)) + '/Resources/Icons/' JSON_DIR = os.path.dirname(os.path.realpath(__file__)) + '/Resources/json/' @@ -842,7 +843,7 @@ def create(self, json): self.outputSelector = slicer.qMRMLNodeComboBox() self.widgets.append(self.outputSelector) - self.outputSelector.nodeTypes = ["vtkMRMLScalarVolumeNode", "vtkMRMLLabelMapVolumeNode"] + self.outputSelector.nodeTypes = SimpleFilters.scalarVolumeNodeClasses self.outputSelector.selectNodeUponCreation = True self.outputSelector.addEnabled = True self.outputSelector.removeEnabled = False @@ -884,7 +885,7 @@ def create(self, json): def createInputWidget(self,n, noneEnabled=False): inputSelector = slicer.qMRMLNodeComboBox() self.widgets.append(inputSelector) - inputSelector.nodeTypes = ["vtkMRMLScalarVolumeNode", "vtkMRMLLabelMapVolumeNode"] + inputSelector.nodeTypes = SimpleFilters.scalarVolumeNodeClasses inputSelector.selectNodeUponCreation = True inputSelector.addEnabled = False inputSelector.removeEnabled = False