def createOutputStep(self):
from convert import readSetOfClasses3D
numberOfClasses = self.numberOfRef
imgSet = self.inputParticles.get()
volumes = self._createSetOfVolumes()
volumes.setSamplingRate(imgSet.getSamplingRate())
fileparList = []
volumeList = []
for ref in range(1, numberOfClasses + 1):
vol = Volume()
filepar = self._getBaseName('output_par_class', iter=self._getLastIter(), ref=ref)
volFn = self._getFileName('iter_vol_class', iter=self._getLastIter(), ref=ref)
vol.setFileName(volFn)
volumes.append(vol)
fileparList.append(filepar)
volumeList.append(volFn)
classes = self._createSetOfClasses3D(imgSet)
readSetOfClasses3D(classes, fileparList, volumeList)
# Define the outputs and relations
self._defineOutputs(outputClasses=classes)
self._defineOutputs(outputVolumes=volumes)
#TODO: save alignment
self._defineSourceRelation(imgSet, classes)
self._defineSourceRelation(self.input3DReference.get(), classes)
self._defineSourceRelation(imgSet, volumes)
self._defineSourceRelation(self.input3DReference.get(), volumes)
def createOutputStep(self):
numberOfClasses = self.numberOfRef
imgSet = self._getInputParticles()
volumes = self._createSetOfVolumes()
volumes.setSamplingRate(imgSet.getSamplingRate())
for ref in range(1, numberOfClasses + 1):
vol = Volume()
volFn = self._getFileName("iter_vol_class", iter=self._getLastIter(), ref=ref)
vol.setFileName(volFn)
volumes.append(vol)
clsSet = self._createSetOfClasses3D(imgSet)
self._fill3DClasses(clsSet, numberOfClasses)
# Define the outputs and relations
self._defineOutputs(outputClasses=clsSet)
self._defineOutputs(outputVolumes=volumes)
# TODO: save alignment
self._defineSourceRelation(self._getInputParticlesPointer(), clsSet)
self._defineSourceRelation(self._getInputParticlesPointer(), volumes)
if not self.doContinue:
self._defineSourceRelation(self.input3DReference, clsSet)
self._defineSourceRelation(self.input3DReference, volumes)
