diff --git a/meshroom/core/desc.py b/meshroom/core/desc.py index 05bb6d0a..68a7d4cf 100755 --- a/meshroom/core/desc.py +++ b/meshroom/core/desc.py @@ -1,4 +1,5 @@ from meshroom.common import BaseObject, Property, Variant +from enum import Enum # available by default in python3. For python2: "pip install enum34" class Attribute(BaseObject): @@ -105,10 +106,19 @@ class StringParam(Param): super(StringParam, self).__init__(label=label, description=description, value=value, uid=uid, group=group) +class Level(Enum): + NONE = 0 + NORMAL = 1 + INTENSIVE = 2 + + class Node(object): """ """ internalFolder = '{nodeType}/{uid0}/' + cpu = Level.NORMAL + gpu = Level.NONE + ram = Level.NORMAL def __init__(self): pass diff --git a/meshroom/nodes/aliceVision/DepthMap.py b/meshroom/nodes/aliceVision/DepthMap.py index 881b59b7..717a0856 100644 --- a/meshroom/nodes/aliceVision/DepthMap.py +++ b/meshroom/nodes/aliceVision/DepthMap.py @@ -3,6 +3,7 @@ from meshroom.core import desc class DepthMap(desc.CommandLineNode): internalFolder = '{cache}/{nodeType}/{uid0}/' commandLine = 'CMPMVS {mvsConfigValue} --createDepthmap' + gpu = desc.Level.INTENSIVE mvsConfig = desc.File( label='MVS Configuration file', diff --git a/meshroom/nodes/aliceVision/DepthMapFilter.py b/meshroom/nodes/aliceVision/DepthMapFilter.py index d4d5be62..acaffa37 100644 --- a/meshroom/nodes/aliceVision/DepthMapFilter.py +++ b/meshroom/nodes/aliceVision/DepthMapFilter.py @@ -3,6 +3,7 @@ from meshroom.core import desc class DepthMapFilter(desc.CommandLineNode): internalFolder = '{cache}/{nodeType}/{uid0}/' commandLine = 'CMPMVS {mvsConfigValue} --filterDepthmap' + gpu = desc.Level.NORMAL mvsConfig = desc.File( label='MVS Configuration file', diff --git a/meshroom/nodes/aliceVision/Meshing.py b/meshroom/nodes/aliceVision/Meshing.py index 9996ceab..d683db02 100644 --- a/meshroom/nodes/aliceVision/Meshing.py +++ b/meshroom/nodes/aliceVision/Meshing.py @@ -3,6 +3,8 @@ from meshroom.core import desc class Meshing(desc.CommandLineNode): internalFolder = '{cache}/{nodeType}/{uid0}/' commandLine = 'CMPMVS {mvsConfigValue} --meshing' + cpu = desc.Level.INTENSIVE + ram = desc.Level.INTENSIVE mvsConfig = desc.File( label='MVS Configuration file', diff --git a/meshroom/nodes/aliceVision/Texturing.py b/meshroom/nodes/aliceVision/Texturing.py index 23dd8eb4..474e1425 100644 --- a/meshroom/nodes/aliceVision/Texturing.py +++ b/meshroom/nodes/aliceVision/Texturing.py @@ -3,6 +3,8 @@ from meshroom.core import desc class Texturing(desc.CommandLineNode): internalFolder = '{cache}/{nodeType}/{uid0}/' commandLine = 'CMPMVS {mvsConfigValue} --texturing' + cpu = desc.Level.INTENSIVE + ram = desc.Level.INTENSIVE mvsConfig = desc.File( label='MVS Configuration file',