bilateral image filtering

Author: mlavik1

Assets/Scripts/Gradients/CentralDifferenceGradientComputator.cs

@@ -1,11 +1,14 @@
+using itk.simple;
+using openDicom.Image;
 using System;
+using System.Runtime.InteropServices;
 using UnityEngine;
 
 namespace UnityVolumeRendering
 {
     public class CentralDifferenceGradientComputator : GradientComputator
     {
-        public CentralDifferenceGradientComputator(VolumeDataset dataset) : base(dataset)
+        public CentralDifferenceGradientComputator(VolumeDataset dataset, bool smootheDataValues) : base(dataset, smootheDataValues)
         {
         }
 

Assets/Scripts/Gradients/GradientComputator.cs

@@ -1,4 +1,6 @@
+using itk.simple;
 using System;
+using System.Runtime.InteropServices;
 using UnityEngine;
 
 namespace UnityVolumeRendering
@@ -8,14 +10,62 @@ public abstract class GradientComputator
         protected float[] data;
         protected int dimX, dimY, dimZ;
 
-        public GradientComputator(VolumeDataset dataset)
+        public GradientComputator(VolumeDataset dataset, bool smootheDataValues)
         {
             this.data = dataset.data;
             this.dimX = dataset.dimX;
             this.dimY = dataset.dimY;
             this.dimZ = dataset.dimZ;
+            
+            if (smootheDataValues)
+            {
+#if UVR_USE_SIMPLEITK
+                Image image = new Image((uint)dimX, (uint)dimY, (uint)dimZ, PixelIDValueEnum.sitkFloat32);
+
+                for (uint z = 0; z < dimZ; z++)
+                {
+                    for (uint y = 0; y < dimY; y++)
+                    {
+                        for (uint x = 0; x < dimX; x++)
+                        {
+                            float value = data[x + y * dimX + z * (dimX * dimY)];
+                            image.SetPixelAsFloat(new VectorUInt32() { x, y, z }, value);
+                        }
+                    }
+                }
+
+                BilateralImageFilter filter = new BilateralImageFilter();
+                image = filter.Execute(image);
+
+                this.data = new float[data.Length];
+                IntPtr imgBuffer = image.GetBufferAsFloat();
+                Marshal.Copy(imgBuffer, data, 0, data.Length);
+#else
+            Debug.LogWarning("SimpleITK is required to generate smooth gradients.");
+#endif
+            }
         }
 
         public abstract Vector3 ComputeGradient(int x, int y, int z, float minValue, float maxRange);
     }
+
+    public class GradientComputatorFactory
+    {
+        public static GradientComputator CreateGradientComputator(VolumeDataset dataset, GradientType gradientType)
+        {
+            switch (gradientType)
+            {
+                case GradientType.CentralDifference:
+                    return new CentralDifferenceGradientComputator(dataset, false);
+                case GradientType.SmoothedCentralDifference:
+                    return new CentralDifferenceGradientComputator(dataset, true);
+                case GradientType.Sobel:
+                    return new SobelGradientComputator(dataset, false);
+                case GradientType.SmoothedSobel:
+                    return new SobelGradientComputator(dataset, true);
+                default:
+                    throw new NotImplementedException();
+            }
+        }
+    }
 }

Assets/Scripts/Gradients/GradientType.cs

@@ -6,7 +6,9 @@ namespace UnityVolumeRendering
     public enum GradientType
     {
         CentralDifference,
-        Sobel
+        SmoothedCentralDifference,
+        Sobel,
+        SmoothedSobel
     }
 
     public class GradientTypeUtils
@@ -15,7 +17,6 @@ public static GradientType GetDefaultGradientType()
         {
             GradientType gradientType = GradientType.CentralDifference;
             Enum.TryParse(PlayerPrefs.GetString("DefaultGradientType"), out gradientType);
-            Debug.Log(gradientType);
             return gradientType;
         }
     }

Assets/Scripts/Gradients/SobelGradientComputator.cs

@@ -1,11 +1,12 @@
-using System;
+using System;
+using System.Drawing;
 using UnityEngine;
 
 namespace UnityVolumeRendering
 {
     public class SobelGradientComputator : GradientComputator
     {
-        public SobelGradientComputator(VolumeDataset dataset) : base(dataset)
+        public SobelGradientComputator(VolumeDataset dataset, bool smootheDataValues) : base(dataset, smootheDataValues)
         {
         }
 

Assets/Scripts/Importing/ImageSequenceImporter/SimpleITK/SimpleITKDICOMImporter.cs

@@ -153,6 +153,10 @@ private void ImportSeriesInternal(VectorString dicomNames, ImageSequenceSeries s
             // Cast to 32-bit float
             image = SimpleITK.Cast(image, PixelIDValueEnum.sitkFloat32);
 
+            //SmoothingRecursiveGaussianImageFilter filter = new SmoothingRecursiveGaussianImageFilter();
+            //filter.SetSigma(2);
+            //image = filter.Execute(image);
+
             size = image.GetSize();
 
             int numPixels = 1;

Assets/Scripts/VolumeData/VolumeDataset.cs

@@ -377,7 +377,7 @@ await Task.Run(() => {
                 Texture3D textureTmp = new Texture3D(dimX, dimY, dimZ, texformat, false);
                 textureTmp.wrapMode = TextureWrapMode.Clamp;
 
-                GradientComputator gradientComputator = gradientType == GradientType.Sobel ? new SobelGradientComputator(this) : new CentralDifferenceGradientComputator(this);
+                GradientComputator gradientComputator = GradientComputatorFactory.CreateGradientComputator(this, gradientType);
 
                 for (int x = 0; x < dimX; x++)
                 {
@@ -405,7 +405,7 @@ await Task.Run(() => {
 
             progressHandler.StartStage(0.6f, "Creating gradient texture");
             await Task.Run(() => {
-                GradientComputator gradientComputator = gradientType == GradientType.Sobel ? new SobelGradientComputator(this) : new CentralDifferenceGradientComputator(this);
+                GradientComputator gradientComputator = GradientComputatorFactory.CreateGradientComputator(this, gradientType);
 
                 for (int z = 0; z < dimZ; z++)
                 {