3pp-mirror/Meshroom
1
0
Fork 0
mirror of https://github.com/alicevision/Meshroom.git synced 2025-05-22 05:26:29 +02:00
Code Issues Projects Releases Packages Wiki Activity

[nodes] KeyframeSelection: Set a dynamic size for the node

Browse source 
The size is set with the minimum between the input frames (when available)
and the set maximum number of output keyframes. This allows the nodes
following the KeyframeSelection one to be chunked if applicable.
This commit is contained in:
Candice Bentéjac 2023-06-07 17:38:53 +02:00
parent f80e688e64
commit b6fc017db1
1 changed files with 48 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

49
meshroom/nodes/aliceVision/KeyframeSelection.py
View file

@ -1,11 +1,58 @@
__version__ = "4.0"
__version__ = "4.1"
import os
from meshroom.core import desc
class KeyframeSelectionNodeSize(desc.DynamicNodeSize):
def computeSize(self, node):
inputPathsSize = super(KeyframeSelectionNodeSize, self).computeSize(node)
s = 0
finalSize = 0
defaultParam = self._param
# Compute the size for each entry in the list of input paths
for input in node.attribute("inputPaths").value:
self._param = input.getFullName()
s = s + super(KeyframeSelectionNodeSize, self).computeSize(node)
# Retrieve the maximum number of keyframes for the smart selection (which is high by default)
maxFramesSmart = node.attribute("selectionMethod.smartSelection.maxNbOutFrames").value
# If the smart selection is enabled and the number of input frames is available (s is not equal to the number of input paths),
# set the size as the minimum between the number of input frames and maximum number of output keyframes. If the number of
# input frames is not available, set the size to the maximum number of output keyframes.
smartSelectionOn = node.attribute("selectionMethod.useSmartSelection").value
if smartSelectionOn:
if s != inputPathsSize:
finalSize = min(s, maxFramesSmart)
else:
finalSize = maxFramesSmart
# If the smart selection is not enabled, the maximum number of output keyframes for the regular mode can be used
# if and only if it has been set, in the same way as for the smart selection. If the maximum number of frames has
# not been set, then the size is either the minimum between the maximum number of output keyframes for the smart
# selection and the number of input frames if it is available, or the maximum number of output keyframes for the
# smart selection if the number of input frames is not available.
else:
maxFrames = node.attribute("selectionMethod.regularSelection.maxNbOutFrames").value
if maxFrames > 0 and s != inputPathsSize:
finalSize = min(s, maxFrames)
elif maxFrames > 0 and s == inputPathsSize:
finalSize = maxFrames
elif maxFrames <= 0 and s != inputPathsSize:
finalSize = min(s, maxFramesSmart)
else:
finalSize = maxFramesSmart
# Reset the param used to compute size to the default one: if the size is computed again,
# this will prevent having an inputPathsSize that is erroneous
self._param = defaultParam
return finalSize
class KeyframeSelection(desc.AVCommandLineNode):
commandLine = 'aliceVision_keyframeSelection {allParams}'
size = KeyframeSelectionNodeSize('inputPaths')
category = 'Utils'
documentation = '''