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

[core] all parameter types could now be used as output attribute

Browse source
This commit is contained in:
Fabien Castan 2024-05-27 21:58:53 +02:00
parent f1d0211758
commit d2ad818b2c
2 changed files with 39 additions and 30 deletions
Download patch file Download diff file Expand all files Collapse all files

28
meshroom/core/attribute.py
View file

@ -63,7 +63,6 @@ class Attribute(BaseObject):
self._node = weakref.ref(node)
self.attributeDesc = attributeDesc
self._isOutput = isOutput
self._value = copy.copy(attributeDesc.value)
self._label = attributeDesc.label
self._enabled = True
self._validValue = True
@ -72,6 +71,10 @@ class Attribute(BaseObject):
# invalidation value for output attributes
self._invalidationValue = ""
self._value = None # Needs to be created before setting the value
# do not emit "value changed" on initialization
self._set_value(copy.copy(attributeDesc.value), emitSignals=False)
@property
def node(self):
return self._node()
@ -183,19 +186,25 @@ class Attribute(BaseObject):
return self.getLinkParam().value
return self._value
def _set_value(self, value):
def _set_value(self, value, emitSignals=True):
if self._value == value:
return
if isinstance(value, Attribute) or Attribute.isLinkExpression(value):
# if we set a link to another attribute
self._value = value
elif isinstance(value, types.FunctionType):
# evaluate the function
self._value = value(self)
else:
# if we set a new value, we use the attribute descriptor validator to check the validity of the value
# and apply some conversion if needed
convertedValue = self.validateValue(value)
self._value = convertedValue
if not emitSignals:
return
if not self.node.isCompatibilityNode:
self.node.onAttributeChanged(self)
# Request graph update when input parameter value is set
@ -232,7 +241,7 @@ class Attribute(BaseObject):
self._set_value(exportedValue)
def resetValue(self):
self._value = self.attributeDesc.value
self.value = self.defaultValue()
def requestGraphUpdate(self):
if self.node.graph:
@ -326,9 +335,9 @@ class Attribute(BaseObject):
def getExportValue(self):
if self.isLink:
return self.getLinkParam().asLinkExpr()
if self.isOutput:
if self.isOutput and self.desc.isExpression:
return self.defaultValue()
return self._value
return self.value
def getEvalValue(self):
'''
@ -497,7 +506,7 @@ class ListAttribute(Attribute):
def resetValue(self):
self._value = ListModel(parent=self)
def _set_value(self, value):
def _set_value(self, value, emitSignals=True):
if self.node.graph:
self.remove(0, len(self))
# Link to another attribute
@ -505,8 +514,13 @@ class ListAttribute(Attribute):
self._value = value
# New value
else:
# During initialization self._value may not be set
if self._value is None:
self._value = ListModel(parent=self)
newValue = self.desc.validateValue(value)
self.extend(newValue)
if not emitSignals:
return
self.requestGraphUpdate()
def upgradeValue(self, exportedValues):
@ -647,7 +661,7 @@ class GroupAttribute(Attribute):
except KeyError:
raise AttributeError(key)
def _set_value(self, exportedValue):
def _set_value(self, exportedValue, emitSignals=True):
value = self.validateValue(exportedValue)
if isinstance(value, dict):
# set individual child attribute values

41
meshroom/core/node.py
View file

@ -747,16 +747,14 @@ class BaseNode(BaseObject):
if attr.isInput:
continue # skip inputs
# Only consider File attributes for command output parameters
if not isinstance(attr.attributeDesc, desc.File):
continue
try:
defaultValue = attr.defaultValue()
except AttributeError as e:
# If we load an old scene, the lambda associated to the 'value' could try to access other params that could not exist yet
logging.warning('Invalid lambda evaluation for "{nodeName}.{attrName}"'.format(nodeName=self.name, attrName=attr.name))
else:
# Apply expressions for File attributes
if attr.attributeDesc.isExpression:
defaultValue = ""
try:
defaultValue = attr.defaultValue()
except AttributeError as e:
# If we load an old scene, the lambda associated to the 'value' could try to access other params that could not exist yet
logging.warning('Invalid lambda evaluation for "{nodeName}.{attrName}"'.format(nodeName=self.name, attrName=attr.name))
try:
attr.value = defaultValue.format(**self._cmdVars)
attr._invalidationValue = defaultValue.format(**cmdVarsNoCache)
@ -1295,8 +1293,7 @@ class Node(BaseNode):
# skip missing attributes
continue
attr = self.attribute(k)
if attr.isInput:
attr.value = v
attr.value = v
def upgradeAttributeValues(self, values):
# initialize attribute values
@ -1305,11 +1302,10 @@ class Node(BaseNode):
# skip missing attributes
continue
attr = self.attribute(k)
if attr.isInput:
try:
attr.upgradeValue(v)
except ValueError:
pass
try:
attr.upgradeValue(v)
except ValueError:
pass
def setInternalAttributeValues(self, values):
# initialize internal attribute values
@ -1327,11 +1323,10 @@ class Node(BaseNode):
# skip missing atributes
continue
attr = self.internalAttribute(k)
if attr.isInput:
try:
attr.upgradeValue(v)
except ValueError:
pass
try:
attr.upgradeValue(v)
except ValueError:
pass
def toDict(self):
inputs = {k: v.getExportValue() for k, v in self._attributes.objects.items() if v.isInput}
@ -1741,7 +1736,7 @@ def nodeFactory(nodeDict, name=None, template=False, uidConflict=False):
break
if compatibilityIssue is None:
node = Node(nodeType, position, **inputs)
node = Node(nodeType, position, **inputs, **outputs)
node.setInternalAttributeValues(internalInputs)
else:
logging.warning("Compatibility issue detected for node '{}': {}".format(name, compatibilityIssue.name))