From 6ad8ed0396cd42bea0660ff43f38cb6954aa169d Mon Sep 17 00:00:00 2001
From: Fabien Castan <fabcastan@gmail.com>
Date: Fri, 11 Sep 2020 13:04:16 +0200
Subject: [PATCH] [nodes] ImageProcessing: add new option to fix non finite
 pixels

---
 meshroom/multiview.py                         |  1 +
 meshroom/nodes/aliceVision/ImageProcessing.py | 10 +++++++++-
 2 files changed, 10 insertions(+), 1 deletion(-)

diff --git a/meshroom/multiview.py b/meshroom/multiview.py
index f9448f380b..b6c6f02885 100644
--- a/meshroom/multiview.py
+++ b/meshroom/multiview.py
@@ -249,6 +249,7 @@ def panoramaHdrPipeline(graph):
 
     imageProcessing = graph.addNewNode('ImageProcessing',
                                        input=panoramaCompositing.output,
+                                       fixNonFinite=True,
                                        fillHoles=True,
                                        extension='exr')
 
diff --git a/meshroom/nodes/aliceVision/ImageProcessing.py b/meshroom/nodes/aliceVision/ImageProcessing.py
index 6ae880e9bc..32298d8077 100644
--- a/meshroom/nodes/aliceVision/ImageProcessing.py
+++ b/meshroom/nodes/aliceVision/ImageProcessing.py
@@ -86,6 +86,13 @@ class ImageProcessing(desc.CommandLineNode):
             value=False,
             uid=[0],
         ),
+        desc.BoolParam(
+            name='fixNonFinite',
+            label='Fix Non-Finite',
+            description='Fix non-finite pixels based on neighboring pixels average.',
+            value=False,
+            uid=[0],
+        ),
         desc.BoolParam(
             name='exposureCompensation',
             label='Exposure Compensation',
@@ -120,7 +127,8 @@ class ImageProcessing(desc.CommandLineNode):
         desc.BoolParam(
             name='fillHoles',
             label='Fill Holes',
-            description='Fill holes based on the alpha channel.',
+            description='Fill holes based on the alpha channel.\n'
+                        'Note: It will enable fixNonFinite, as it is required for the image pyramid construction used to fill holes.',
             value=False,
             uid=[0],
         ),