def doSuccessExecAccumultor(self, _edPlugin=None):
with self.semAccumulator:
self.DEBUG("EDPluginControlAlignStackv1_0.doSuccessExecAccumultor")
self.retrieveSuccessMessages(
_edPlugin,
"EDPluginControlAlignStackv1_0.doSuccessExecAccumultor")
for query in _edPlugin.dataOutput.getQuery():
self.addExtraTime(60)
_edPlugin.addExtraTime(60)
accType = query.getItem()[0].getValue().split()[0]
listInt = [
int(i.getValue().split()[1]) for i in query.getItem()
]
if accType == "raw":
listFrame = [self.getFrameRef(i) for i in listInt]
xsdata = XSDataInputMeasureOffset(image=listFrame)
doSIFT = False
if self.xsdMeasureOffset is not None:
xsdata.setCropBorders(
self.xsdMeasureOffset.getCropBorders())
xsdata.setSmoothBorders(
self.xsdMeasureOffset.getSmoothBorders())
xsdata.setBackgroundSubtraction(
self.xsdMeasureOffset.getRemoveBackground())
if self.xsdMeasureOffset.useSift is not None:
doSIFT = self.xsdMeasureOffset.useSift.value
if max(listInt
) > EDPluginControlAlignStackv1_0.__iRefFrame:
listInt.sort()
else:
listInt.sort(reverse=True)
xsdata.setIndex([XSDataInteger(i) for i in listInt])
if doSIFT:
edPluginExecMeasure = self.loadPlugin(
self.__strControlledPluginMeasureSIFT)
else:
edPluginExecMeasure = self.loadPlugin(
self.__strControlledPluginMeasureFFT)
edPluginExecMeasure.setDataInput(xsdata)
edPluginExecMeasure.connectSUCCESS(
self.doSuccessExecMeasureOffset)
edPluginExecMeasure.connectFAILURE(
self.doFailureExecMeasureOffset)
self.queue.put(edPluginExecMeasure)
elif accType == "shift":
shift_1 = 0.0
shift_2 = 0.0
for frame in listInt:
shift_1 += EDPluginControlAlignStackv1_0.__dictRelShift[
frame][0]
shift_2 += EDPluginControlAlignStackv1_0.__dictRelShift[
frame][1]
if listInt[0] > EDPluginControlAlignStackv1_0.__iRefFrame:
iFrameShift = max(listInt)
else:
iFrameShift = min(listInt)
EDPluginControlAlignStackv1_0.__dictAbsShift[
iFrameShift] = (shift_1, shift_2)
self.screen(
"Frame number %i has absolute offset of %.3f,%.3f" %
(iFrameShift, shift_1, shift_2))
edPluginExecShift = self.loadPlugin(
self.__strControlledPluginShift)
edPluginExecShift.dataInput = XSDataInputShiftImage(
index=XSDataInteger(iFrameShift),
offset=[XSDataDouble(shift_1),
XSDataDouble(shift_2)],
inputImage=self.getFrameRef(iFrameShift),
outputImage=XSDataImageExt(
shared=XSDataString("Shifted-%06i" % iFrameShift)))
edPluginExecShift.connectSUCCESS(
self.doSuccessExecShiftImage)
edPluginExecShift.connectFAILURE(
self.doFailureExecShiftImage)
self.queue.put(edPluginExecShift)
self.DEBUG("Items: %s" % EDPluginAccumulatorv1_0.getItems())
self.DEBUG("Queries: %s" % EDPluginAccumulatorv1_0.getQueries())
def process(self, _edObject=None):
EDPluginControl.process(self)
self.DEBUG("EDPluginControlAlignStackv1_0.process")
for iFrame in self.iFrames:
edPluginExecAccumulator = self.loadPlugin(
self.__strControlledPluginAccumulator)
queryRaw = XSDataQuery()
queryShift = XSDataQuery()
queryRaw.setRemoveItems(XSDataBoolean(False))
queryShift.setRemoveItems(XSDataBoolean(False))
xsdataAcc = XSDataInputAccumulator()
if (EDPluginControlAlignStackv1_0.__iRefFrame
== iFrame) or (self.bDoAlign == False):
EDPluginControlAlignStackv1_0.__dictAbsShift[iFrame] = (0.0,
0.0)
EDPluginControlAlignStackv1_0.__dictRelShift[iFrame] = (0.0,
0.0)
self.hdf5_offset(index=iFrame, offset=[0.0, 0.0])
edPluginExecShift = self.loadPlugin(
self.__strControlledPluginShift)
xsdata = XSDataInputShiftImage(
index=XSDataInteger(iFrame),
offset=[
XSDataDouble(i) for i in
EDPluginControlAlignStackv1_0.__dictAbsShift[iFrame]
],
inputImage=self.getFrameRef(iFrame),
outputImage=XSDataImageExt(
shared=XSDataString("Shifted-%06i" % iFrame)))
edPluginExecShift.setDataInput(xsdata)
edPluginExecShift.connectSUCCESS(self.doSuccessExecShiftImage)
edPluginExecShift.connectFAILURE(self.doFailureExecShiftImage)
self.queue.put(edPluginExecShift)
if (self.bDoAlign == False):
self.executeControlledPlugins()
return
elif EDPluginControlAlignStackv1_0.__iRefFrame < iFrame:
if self.bAlwaysMOvsRef:
queryRaw.setItem([
XSDataString(
"raw %04i" %
(EDPluginControlAlignStackv1_0.__iRefFrame)),
XSDataString("raw %04i" % iFrame)
])
xsdataAcc.setQuery([queryRaw])
else:
queryRaw.setItem([
XSDataString("raw %04i" % (iFrame - 1)),
XSDataString("raw %04i" % iFrame)
])
queryShift.setItem([
XSDataString("shift %04i" % i) for i in range(
EDPluginControlAlignStackv1_0.__iRefFrame +
1, iFrame + 1)
])
xsdataAcc.setQuery([queryRaw, queryShift])
elif EDPluginControlAlignStackv1_0.__iRefFrame > iFrame:
if self.bAlwaysMOvsRef:
queryRaw.setItem([
XSDataString("raw %04i" % iFrame),
XSDataString(
"raw %04i" %
(EDPluginControlAlignStackv1_0.__iRefFrame))
])
xsdataAcc.setQuery([queryRaw])
else:
queryRaw.setItem([
XSDataString("raw %04i" % (iFrame + 1)),
XSDataString("raw %04i" % iFrame)
])
queryShift.setItem([
XSDataString("shift %04i" % i) for i in range(
EDPluginControlAlignStackv1_0.__iRefFrame -
1, iFrame - 1, -1)
])
xsdataAcc.setQuery([queryRaw, queryShift])
if (EDPluginControlAlignStackv1_0.__iRefFrame == iFrame):
self.saveReferenceFrame(iFrame)
xsdataAcc.setItem([XSDataString("raw %04i" % iFrame)])
edPluginExecAccumulator.setDataInput(xsdataAcc)
edPluginExecAccumulator.connectSUCCESS(
self.doSuccessExecAccumultor)
edPluginExecAccumulator.connectFAILURE(
self.doFailureExecAccumulator)
self.queue.put(edPluginExecAccumulator)
self.executeControlledPlugins()
def doSuccessExecMeasureOffset(self, _edPlugin=None):
with self.semMeasure:
self.DEBUG(
"EDPluginControlAlignStackv1_0.doSuccessExecMeasureOffset")
self.retrieveSuccessMessages(
_edPlugin,
"EDPluginControlAlignStackv1_0.doSuccessExecMeasureOffset")
listIndex = [i.getValue() for i in _edPlugin.dataInput.index]
listIndex.sort()
dataOutput = _edPlugin.dataOutput
if self.bAlwaysMOvsRef:
if min(listIndex) < EDPluginControlAlignStackv1_0.__iRefFrame:
iToShift, iRef = tuple(listIndex)
EDPluginControlAlignStackv1_0.__dictAbsShift[
iToShift] = tuple(
[-i.getValue() for i in dataOutput.getOffset()])
else:
iRef, iToShift = tuple(listIndex)
EDPluginControlAlignStackv1_0.__dictAbsShift[
iToShift] = tuple(
[i.getValue() for i in dataOutput.getOffset()])
self.screen(
"Frame number %i has absolute offset of %.3f,%.3f" %
(iToShift,
EDPluginControlAlignStackv1_0.__dictAbsShift[iToShift][0],
EDPluginControlAlignStackv1_0.__dictAbsShift[iToShift][1])
)
edPluginExecShift = self.loadPlugin(
self.__strControlledPluginShift)
xsdata = XSDataInputShiftImage(
index=XSDataInteger(iToShift),
offset=[
XSDataDouble(i) for i in
EDPluginControlAlignStackv1_0.__dictAbsShift[iToShift]
],
inputImage=self.getFrameRef(iToShift),
outputImage=XSDataImageExt(
shared=XSDataString("Shifted-%06i" % iToShift)))
edPluginExecShift.setDataInput(xsdata)
edPluginExecShift.connectSUCCESS(self.doSuccessExecShiftImage)
edPluginExecShift.connectFAILURE(self.doFailureExecShiftImage)
self.queue.put(edPluginExecShift)
else:
if min(listIndex) < EDPluginControlAlignStackv1_0.__iRefFrame:
iToShift, iRef = tuple(listIndex)
EDPluginControlAlignStackv1_0.__dictRelShift[
iToShift] = tuple(
[-i.value for i in dataOutput.offset])
else:
iRef, iToShift = tuple(listIndex)
EDPluginControlAlignStackv1_0.__dictRelShift[
iToShift] = tuple([i.value for i in dataOutput.offset])
self.screen(
"Frame number %i has relative offset of %.3f,%.3f" %
(iToShift,
EDPluginControlAlignStackv1_0.__dictRelShift[iToShift][0],
EDPluginControlAlignStackv1_0.__dictRelShift[iToShift][1])
)
xsdata = XSDataInputAccumulator(
item=[XSDataString("shift %04i" % iToShift)])
edPluginExecAccumulator = self.loadPlugin(
self.__strControlledPluginAccumulator)
edPluginExecAccumulator.setDataInput(xsdata)
edPluginExecAccumulator.connectSUCCESS(
self.doSuccessExecAccumultor)
edPluginExecAccumulator.connectFAILURE(
self.doFailureExecAccumulator)
self.queue.put(edPluginExecAccumulator)
def testCheckParameters(self):
xsDataInput = XSDataInputShiftImage()
edPluginExecShiftImage = self.createPlugin()
edPluginExecShiftImage.setDataInput(xsDataInput)
edPluginExecShiftImage.checkParameters()