3pp-mirror/Meshroom
1
0
Fork 0
mirror of https://github.com/alicevision/Meshroom.git synced 2025-07-17 16:47:19 +02:00
Code Issues Projects Releases Packages Wiki Activity

[core] node: add some typing

Browse source
This commit is contained in:
Fabien Castan 2025-03-28 12:15:39 +01:00
parent 2f08448310
commit 0c32060264
1 changed files with 12 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

21
meshroom/core/node.py
View file

@ -68,15 +68,18 @@ class StatusData(BaseObject):
def __init__(self, nodeName='', nodeType='', packageName='', packageVersion='', mrNodeType: MrNodeType = MrNodeType.NONE, parent: BaseObject = None):
super(StatusData, self).__init__(parent)
self.reset()
self.nodeName: str = nodeName
self.nodeType: str = nodeType
self.packageName: str = packageName
self.packageVersion: str = packageVersion
self.mrNodeType = mrNodeType
self.sessionUid: Optional[str] = meshroom.core.sessionUid
self.submitterSessionUid: Optional[str] = None
self.resetDynamicValues()
def merge(self, other):
self.startDateTime = min(self.startDateTime, other.startDateTime)
self.endDateTime = max(self.endDateTime, other.endDateTime)
@ -1420,7 +1423,7 @@ class BaseNode(BaseObject):
self._hasDuplicates = bool(len(newList))
self.hasDuplicatesChanged.emit()
def statusInThisSession(self):
def statusInThisSession(self) -> bool:
if not self._chunks:
return False
for chunk in self._chunks:
@ -1428,7 +1431,7 @@ class BaseNode(BaseObject):
return False
return True
def submitterStatusInThisSession(self):
def submitterStatusInThisSession(self) -> bool:
if not self._chunks:
return False
for chunk in self._chunks:
@ -1436,8 +1439,8 @@ class BaseNode(BaseObject):
return False
return True
def initFromThisSession(self):
if not self._chunks:
def initFromThisSession(self) -> bool:
if len(self._chunks) == 0:
return False
for chunk in self._chunks:
mrNodeType = chunk.node.getMrNodeType()
@ -1447,7 +1450,7 @@ class BaseNode(BaseObject):
return True
@Slot(result=bool)
def canBeStopped(self):
def canBeStopped(self) -> bool:
# Only locked nodes running in local with the same
# sessionUid as the Meshroom instance can be stopped
return (self.getGlobalStatus() == Status.RUNNING and
@ -1455,14 +1458,14 @@ class BaseNode(BaseObject):
self.initFromThisSession())
@Slot(result=bool)
def canBeCanceled(self):
def canBeCanceled(self) -> bool:
# Only locked nodes submitted in local with the same
# sessionUid as the Meshroom instance can be canceled
return (self.getGlobalStatus() == Status.SUBMITTED and
self.globalExecMode == ExecMode.LOCAL.name and
self.initFromThisSession())
def hasImageOutputAttribute(self):
def hasImageOutputAttribute(self) -> bool:
"""
Return True if at least one attribute has the 'image' semantic (and can thus be loaded in the 2D Viewer),
False otherwise.
@ -1472,7 +1475,7 @@ class BaseNode(BaseObject):
return True
return False
def hasSequenceOutputAttribute(self):
def hasSequenceOutputAttribute(self) -> bool:
"""
Return True if at least one attribute has the 'sequence' semantic (and can thus be loaded in the 2D Viewer),
False otherwise.