class AutoSegmentationStep(WorkflowStepMountPoint):
'''
Skeleton step which is intended to be used as a starting point
for new steps.
'''
def __init__(self, location):
super(AutoSegmentationStep, self).__init__('Automatic Segmenter', location)
self._state = StepState()
self._icon = QtGui.QImage(':/autosegmentation/images/autoseg.png')
self.addPort(('http://physiomeproject.org/workflow/1.0/rdf-schema#port', 'http://physiomeproject.org/workflow/1.0/rdf-schema#uses', 'http://physiomeproject.org/workflow/1.0/rdf-schema#images'))
self.addPort(('http://physiomeproject.org/workflow/1.0/rdf-schema#port', 'http://physiomeproject.org/workflow/1.0/rdf-schema#provides', 'http://physiomeproject.org/workflow/1.0/rdf-schema#pointcloud'))
# The widget will be the interface widget for the user to see the
# autosegmentation step and interact with it.
self._widget = None
self._category = 'Segmentation'
# This step only requires an identifier which we can generate from
# a char set. This makes the configuration of the step trivial.
self._identifier = generateIdentifier()
self._configured = True
self._dataIn = None
def configure(self):
return self._configured
def getIdentifier(self):
return self._identifier
def setIdentifier(self, identifier):
self._identifier = identifier
def serialize(self, location):
'''There is nothing to be done here for this step.
'''
pass
def deserialize(self, location):
'''There is nothing to be done here for this step.
'''
pass
def execute(self):
if not self._widget:
self._widget = AutoSegmentationWidget(self._dataIn)
self._widget._ui.doneButton.clicked.connect(self._doneExecution)
self._setCurrentWidget(self._widget)
def setPortData(self, portId, dataIn):
self._dataIn = dataIn
def getPortData(self, index):
point_cloud = self._widget.getPointCloud()
return point_cloud
def execute(self):
if not self._widget:
self._widget = AutoSegmentationWidget(self._dataIn)
self._widget._ui.doneButton.clicked.connect(self._doneExecution)
self._setCurrentWidget(self._widget)
