From 03f45be0ee41b0eaa6c20b398bfba27924043dbc Mon Sep 17 00:00:00 2001
From: Fabien Castan <fabcastan@gmail.com>
Date: Fri, 11 Sep 2020 21:25:42 +0200
Subject: [PATCH] [nodes] expose image storageDataType on some nodes

---
 meshroom/nodes/aliceVision/LdrToHdrMerge.py       | 13 +++++++++++++
 meshroom/nodes/aliceVision/PanoramaCompositing.py | 13 +++++++++++++
 meshroom/nodes/aliceVision/PanoramaWarping.py     | 13 +++++++++++++
 3 files changed, 39 insertions(+)

diff --git a/meshroom/nodes/aliceVision/LdrToHdrMerge.py b/meshroom/nodes/aliceVision/LdrToHdrMerge.py
index e206d3ba..cd9b4632 100644
--- a/meshroom/nodes/aliceVision/LdrToHdrMerge.py
+++ b/meshroom/nodes/aliceVision/LdrToHdrMerge.py
@@ -140,6 +140,19 @@ class LdrToHdrMerge(desc.CommandLineNode):
             uid=[0],
             enabled= lambda node: node.byPass.enabled and not node.byPass.value and node.highlightCorrectionFactor.value != 0,
         ),
+        desc.ChoiceParam(
+            name='storageDataType',
+            label='Storage Data Type',
+            description='Storage image data type:\n'
+                        ' * Float : Use full floating point (32 bits per channel)\n'
+                        ' * Half : Use half float (16 bits per channel)\n'
+                        ' * HalfFinite : Use half float, but clamp values to avoid non-finite values\n'
+                        ' * Auto : Use half float if all values can fit, else use full float\n',
+            value='Float',
+            values=['Float', 'Half', 'HalfFinite', 'Auto'],
+            exclusive=True,
+            uid=[0],
+        ),
         desc.ChoiceParam(
             name='verboseLevel',
             label='Verbose Level',
diff --git a/meshroom/nodes/aliceVision/PanoramaCompositing.py b/meshroom/nodes/aliceVision/PanoramaCompositing.py
index d4255737..ae5997a1 100644
--- a/meshroom/nodes/aliceVision/PanoramaCompositing.py
+++ b/meshroom/nodes/aliceVision/PanoramaCompositing.py
@@ -64,6 +64,19 @@ Multiple cameras are contributing to the low frequencies and only the best one c
             value=True,
             uid=[0],
         ),
+        desc.ChoiceParam(
+            name='storageDataType',
+            label='Storage Data Type',
+            description='Storage image data type:\n'
+                        ' * Float : Use full floating point (32 bits per channel)\n'
+                        ' * Half : Use half float (16 bits per channel)\n'
+                        ' * HalfFinite : Use half float, but clamp values to avoid non-finite values\n'
+                        ' * Auto : Use half float if all values can fit, else use full float\n',
+            value='Float',
+            values=['Float', 'Half', 'HalfFinite', 'Auto'],
+            exclusive=True,
+            uid=[0],
+        ),
         desc.ChoiceParam(
             name='overlayType',
             label='Overlay Type',
diff --git a/meshroom/nodes/aliceVision/PanoramaWarping.py b/meshroom/nodes/aliceVision/PanoramaWarping.py
index 6ac38a9f..e5dbe114 100644
--- a/meshroom/nodes/aliceVision/PanoramaWarping.py
+++ b/meshroom/nodes/aliceVision/PanoramaWarping.py
@@ -56,6 +56,19 @@ Compute the image warping for each input image in the panorama coordinate system
             enabled=lambda node: (node.estimateResolution.value),
             uid=[0]
         ),
+        desc.ChoiceParam(
+            name='storageDataType',
+            label='Storage Data Type',
+            description='Storage image data type:\n'
+                        ' * Float : Use full floating point (32 bits per channel)\n'
+                        ' * Half : Use half float (16 bits per channel)\n'
+                        ' * HalfFinite : Use half float, but clamp values to avoid non-finite values\n'
+                        ' * Auto : Use half float if all values can fit, else use full float\n',
+            value='Float',
+            values=['Float', 'Half', 'HalfFinite', 'Auto'],
+            exclusive=True,
+            uid=[0],
+        ),
         desc.ChoiceParam(
             name='verboseLevel',
             label='Verbose Level',