Implimented RGBA buttons

Editor/Scripts/Editor/Nodes/Types/Image/Utilities/ViewNode/ViewImageNode.cs

@@ -1,10 +1,13 @@
 using ImageProcessingGraph.Editor.Nodes.NodeAttributes;
 using UnityEngine;
+using Unity.Collections;
+using Unity.Jobs;
+using Unity.Burst;
 
 namespace ImageProcessingGraph.Editor.Nodes.Types.Image.Utilities.ViewNode
 {
     [NodeInfo("View Texture", "Utility/View Texture", false, null ,typeof(ViewTextureNodeEditor))]
-    public class ViewTextureNode : BaseImageNode
+    public partial class ViewTextureNode : BaseImageNode
     {
         [NodeAttributes.Input("Texture")] public Texture2D texture;
         [NodeAttributes.Input("Image Data")] public ImageData imageData;
@@ -12,6 +15,12 @@ namespace ImageProcessingGraph.Editor.Nodes.Types.Image.Utilities.ViewNode
         public delegate void OnImageUpdated();
         public OnImageUpdated onImageUpdated;
 
+        public Texture2D cachedRGB;
+        public Texture2D cachedR;
+        public Texture2D cachedG;
+        public Texture2D cachedB;
+        public Texture2D cachedA;
+
         public override void Process()
         {
             if (texture == null)
@@ -19,8 +28,7 @@ namespace ImageProcessingGraph.Editor.Nodes.Types.Image.Utilities.ViewNode
                 texture = imageData.ToTexture2D();
             }
 
-            //CHANGE NPOT SETTING
+            GenerateChannelTextures(texture);
-            
 
             onImageUpdated?.Invoke();
         }
@@ -30,5 +38,98 @@ namespace ImageProcessingGraph.Editor.Nodes.Types.Image.Utilities.ViewNode
             texture = null;
             imageData = default;
         }
+
+        private void GenerateChannelTextures(Texture2D source)
+        {
+            var pixels = source.GetPixels();
+            int length = pixels.Length;
+
+            NativeArray<Color> inputPixels = new NativeArray<Color>(pixels, Allocator.TempJob);
+
+            NativeArray<Color> outputR = new NativeArray<Color>(length, Allocator.TempJob);
+            NativeArray<Color> outputG = new NativeArray<Color>(length, Allocator.TempJob);
+            NativeArray<Color> outputB = new NativeArray<Color>(length, Allocator.TempJob);
+            NativeArray<Color> outputA = new NativeArray<Color>(length, Allocator.TempJob);
+
+            var jobR = new ChannelJob(ChannelType.R, inputPixels, outputR);
+            var jobG = new ChannelJob(ChannelType.G, inputPixels, outputG);
+            var jobB = new ChannelJob(ChannelType.B, inputPixels, outputB);
+            var jobA = new ChannelJob(ChannelType.A, inputPixels, outputA);
+
+            JobHandle handleR = jobR.Schedule(length, 64);
+            JobHandle handleG = jobG.Schedule(length, 64);
+            JobHandle handleB = jobB.Schedule(length, 64);
+            JobHandle handleA = jobA.Schedule(length, 64);
+
+            var handles = new NativeArray<JobHandle>(4, Allocator.Temp);
+            handles[0] = handleR;
+            handles[1] = handleG;
+            handles[2] = handleB;
+            handles[3] = handleA;
+
+            JobHandle.CompleteAll(handles);
+            handles.Dispose(); 
+
+
+            cachedRGB = source;
+            cachedR = CreateTexture(source.width, source.height, outputR);
+            cachedG = CreateTexture(source.width, source.height, outputG);
+            cachedB = CreateTexture(source.width, source.height, outputB);
+            cachedA = CreateTexture(source.width, source.height, outputA);
+
+            inputPixels.Dispose();
+            outputR.Dispose();
+            outputG.Dispose();
+            outputB.Dispose();
+            outputA.Dispose();
+        }
+
+        private Texture2D CreateTexture(int width, int height, NativeArray<Color> colors)
+        {
+            Texture2D tex = new Texture2D(width, height, TextureFormat.RGBA32, false);
+            tex.SetPixels(colors.ToArray());
+            tex.Apply();
+            return tex;
+        }
+
+        [BurstCompile]
+        private struct ChannelJob : IJobParallelFor
+        {
+            public ChannelType channelType;
+            [ReadOnly] public NativeArray<Color> input;
+            public NativeArray<Color> output;
+
+            public ChannelJob(ChannelType type, NativeArray<Color> input, NativeArray<Color> output)
+            {
+                this.channelType = type;
+                this.input = input;
+                this.output = output;
+            }
+
+            public void Execute(int index)
+            {
+                Color c = input[index];
+                switch (channelType)
+                {
+                    case ChannelType.R:
+                        output[index] = new Color(c.r, c.r, c.r, 1f);
+                        break;
+                    case ChannelType.G:
+                        output[index] = new Color(c.g, c.g, c.g, 1f);
+                        break;
+                    case ChannelType.B:
+                        output[index] = new Color(c.b, c.b, c.b, 1f);
+                        break;
+                    case ChannelType.A:
+                        output[index] = new Color(c.a, c.a, c.a, 1f);
+                        break;
+                }
+            }
+        }
+
+        private enum ChannelType
+        {
+            R, G, B, A
+        }
     }
 }

Editor/Scripts/Editor/Nodes/Types/Image/Utilities/ViewNode/ViewImageNodeEditor.cs

@@ -18,25 +18,13 @@ namespace ImageProcessingGraph.Editor.Nodes.Types.Image.Utilities.ViewNode
             IPTPort tex2DPort = InputPorts[0] as IPTPort;
             IPTPort imageDataPort = InputPorts[1] as IPTPort;
 
-            tex2DPort.OnPortConnected += () =>
+            tex2DPort.OnPortConnected += () => { imageDataPort.style.display = DisplayStyle.None; };
-            {
-                imageDataPort.style.display = DisplayStyle.None;
-            };
 
-            tex2DPort.OnPortDisconnected += () =>
+            tex2DPort.OnPortDisconnected += () => { imageDataPort.style.display = DisplayStyle.Flex; };
-            {
-                imageDataPort.style.display = DisplayStyle.Flex;
-            };
 
-            imageDataPort.OnPortConnected += () =>
+            imageDataPort.OnPortConnected += () => { tex2DPort.style.display = DisplayStyle.None; };
-            {
-                tex2DPort.style.display = DisplayStyle.None;
-            };
 
-            imageDataPort.OnPortDisconnected += () =>
+            imageDataPort.OnPortDisconnected += () => { tex2DPort.style.display = DisplayStyle.Flex; };
-            {
-                tex2DPort.style.display = DisplayStyle.Flex;
-            };
 
             this.Q("output").style.display = DisplayStyle.None;
 
@@ -60,10 +48,7 @@ namespace ImageProcessingGraph.Editor.Nodes.Types.Image.Utilities.ViewNode
             viewableImage.style.backgroundColor = Color.black;
             foldout.Add(viewableImage);
 
-            viewImageNode.onImageUpdated += () =>
+            viewImageNode.onImageUpdated += () => { viewableImage.image = viewImageNode.texture; };
-            {
-                viewableImage.image = viewImageNode.texture;
-            };
 
             buttonRow = new VisualElement
             {
@@ -93,7 +78,27 @@ namespace ImageProcessingGraph.Editor.Nodes.Types.Image.Utilities.ViewNode
 
         private void OnChannelClicked(ViewTextureNode viewNode, string channel)
         {
-            Debug.Log($"[ViewTextureNodeEditor] Channel Selected: {channel}");
+            switch (channel)
+            {
+                case "RGB":
+                    viewableImage.image = viewNode.cachedRGB;
+                    break;
+                case "R":
+                    viewableImage.image = viewNode.cachedR;
+                    break;
+                case "G":
+                    viewableImage.image = viewNode.cachedG;
+                    break;
+                case "B":
+                    viewableImage.image = viewNode.cachedB;
+                    break;
+                case "A":
+                    viewableImage.image = viewNode.cachedA;
+                    break;
+                default:
+                    Debug.LogError("[ViewTextureNodeEditor] Unknown channel clicked! Did you hack reality again?");
+                    break;
+            }
         }
     }
 }