Merge pull request #839 from alicevision/dev/imageProcessing

[nodes] new ImageProcessing node

meshroom/nodes/aliceVision/ImageProcessing.py

@@ -0,0 +1,110 @@
+__version__ = "1.1"
+
+from meshroom.core import desc
+
+
+class ImageProcessing(desc.CommandLineNode):
+    commandLine = 'aliceVision_utils_imageProcessing {allParams}'
+    size = desc.DynamicNodeSize('input')
+    # parallelization = desc.Parallelization(blockSize=40)
+    # commandLineRange = '--rangeStart {rangeStart} --rangeSize {rangeBlockSize}'
+
+    inputs = [
+        desc.File(
+            name='input',
+            label='Input',
+            description='SfMData file.',
+            value='',
+            uid=[0],
+        ),
+        desc.ChoiceParam(
+            name='extension',
+            label='File Extension',
+            description='File Extension.',
+            value='',
+            values=['', 'exr', 'jpg', 'tiff', 'png'],
+            exclusive=True,
+            uid=[0],
+        ),
+        desc.BoolParam(
+            name='reconstructedViewsOnly',
+            label='Only Reconstructed Views',
+            description='Process Only Reconstructed Views',
+            value=False,
+            uid=[0],
+        ),
+        desc.BoolParam(
+            name='exposureCompensation',
+            label='Exposure Compensation',
+            description='Exposure Compensation',
+            value=False,
+            uid=[0],
+        ),
+        desc.FloatParam(
+            name='downscale',
+            label='Downscale',
+            description='Downscale.',
+            value=1.0,
+            range=(0.0, 1.0, 0.01),
+            uid=[0],
+        ),
+        desc.FloatParam(
+            name='contrast',
+            label='Contrast',
+            description='Contrast.',
+            value=1.0,
+            range=(0.0, 100.0, 0.1),
+            uid=[0],
+        ),
+        desc.IntParam(
+            name='medianFilter',
+            label='Median Filter',
+            description='Median Filter.',
+            value=0,
+            range=(0, 10, 1),
+            uid=[0],
+        ),
+        desc.IntParam(
+            name='sharpenWidth',
+            label='Sharpen Width',
+            description='Sharpen Width.',
+            value=1,
+            range=(1, 9, 2),
+            uid=[0],
+        ),
+        desc.FloatParam(
+            name='sharpenContrast',
+            label='Sharpen Contrast',
+            description='Sharpen Contrast.',
+            value=1.0,
+            range=(0.0, 100.0, 0.1),
+            uid=[0],
+        ),
+        desc.FloatParam(
+            name='sharpenThreshold',
+            label='Sharpen Threshold',
+            description='Sharpen Threshold.',
+            value=0.0,
+            range=(0.0, 1.0, 0.01),
+            uid=[0],
+        ),
+        desc.ChoiceParam(
+            name='verboseLevel',
+            label='Verbose Level',
+            description='verbosity level (fatal, error, warning, info, debug, trace).',
+            value='info',
+            values=['fatal', 'error', 'warning', 'info', 'debug', 'trace'],
+            exclusive=True,
+            uid=[],
+        )
+    ]
+
+    outputs = [
+        desc.File(
+            name='outSfMData',
+            label='Output sfmData',
+            description='Output sfmData.',
+            value=desc.Node.internalFolder + 'sfmData.abc',
+            uid=[],
+        ),
+    ]