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)