WIP-ADD functions for coloring

Author: eleniv3d

src/gh/components/DF_vizualization/code.py

@@ -13,7 +13,7 @@
 import diffCheck
 from diffCheck import diffcheck_bindings
 from diffCheck import df_cvt_bindings
-from diffCheck import df_error_estimation
+from diffCheck import df_error_estimation, df_vizualization
 
 import diffCheck.df_util
 
@@ -30,18 +30,22 @@ def RunScript(self,
             ghenv.Component.AddRuntimeMessage(RML.Warning, "Please provide both objects of type point clouds to compare")
             return None
 
-        #temp
-        o_source = []
-        o_target = []
-
+        # check if target is a pcl
         o_source = [df_cvt_bindings.cvt_dfcloud_2_rhcloud(src) for src in i_results.source]
+
+        # by default we color the target
+        min_value = min(min(sublist) for sublist in i_results.distances)
+        max_value = max(max(sublist) for sublist in i_results.distances)
+        
+        o_source = [df_vizualization.add_color(src, dist, min_value, max_value) for src, dist in zip(o_source, i_results.distances)]
+
         o_target = [df_cvt_bindings.cvt_dfcloud_2_rhcloud(trg) for trg in i_results.target]
 
         o_legend = []
-    
-        #color the source pointcloud based on viz_settings
 
-        #make a legend
+        # color the source pointcloud based on viz_settings
+
+         #make a legend
 
         return o_source, o_target, o_legend
 

src/gh/diffCheck/diffCheck/df_vizualization.py

@@ -7,6 +7,7 @@
 import open3d as o3d
 from diffCheck import diffcheck_bindings
 import Rhino.Geometry as rg
+import System.Drawing
 
 class DFVizSettings:
     """
@@ -21,3 +22,53 @@ def __init__(self, source_valueType, target_valueType, upper_threshold, lower_th
         self.upper_threshold = upper_threshold
         self.lower_threshold = lower_threshold
         self.palette = palette
+
+
+def interpolate_color(color1, color2, t):
+    """Interpolate between two colors."""
+
+    r = int(color1.R + (color2.R - color1.R) * t)
+    g = int(color1.G + (color2.G - color1.G) * t)
+    b = int(color1.B + (color2.B - color1.B) * t)
+    return System.Drawing.Color.FromArgb(r, g, b)
+
+
+def value_to_color(value, min_value, max_value):
+    """Map a value to a color based on a spectral colormap."""
+
+    # Define the spectral colormap (simplified)
+    colormap = [
+        System.Drawing.Color.FromArgb(0, 0, 255),  # Blue
+        System.Drawing.Color.FromArgb(0, 255, 255),  # Cyan
+        System.Drawing.Color.FromArgb(0, 255, 0),  # Green
+        System.Drawing.Color.FromArgb(255, 255, 0),  # Yellow
+        System.Drawing.Color.FromArgb(255, 0, 0),  # Red
+        System.Drawing.Color.FromArgb(255, 0, 255)  # Magenta
+    ]
+
+    # Normalize the value within the range
+    if min_value == max_value:
+        t = 0.5
+    else:
+        t = (value - min_value) / (max_value - min_value)
+
+    # Determine the segment in the colormap
+    n = len(colormap) - 1
+    idx = int(t * n)
+    if idx >= n:
+        idx = n - 1
+    t = (t * n) - idx
+
+    # Interpolate between the two colors
+    color1 = colormap[idx]
+    color2 = colormap[idx + 1]
+
+    return interpolate_color(color1, color2, t)
+
+
+def add_color(pcd, values, min_value, max_values):
+
+    for i, p in enumerate(pcd):
+        mapped_color = value_to_color(values[i], min_value, max_values)
+        p.Color = mapped_color
+    return pcd