Add map merger to operate faslty operation in same map

Author: Renardjojo

Runtime/MapMerger.cs

@@ -0,0 +1,50 @@
+using UnityEngine;
+using UnityEngine.Rendering;
+
+namespace InfluenceMapPackage
+{
+    public class MapMerger
+    {
+        private ComputeShader m_shader;
+        private static readonly int s_shaderPropertyMap1 = Shader.PropertyToID("_Map1");
+        private static readonly int s_shaderPropertyMap2 = Shader.PropertyToID("_Map2");
+        private static readonly int s_shaderPropertyOutput = Shader.PropertyToID("_TextureOut");
+        private int s_kernelIndex = 0;
+
+        public MapMerger(ComputeShader shader)
+        {
+            SetShader(shader);
+        }
+
+        public void SetShader(ComputeShader shader)
+        {
+            m_shader = shader;
+            s_kernelIndex = m_shader.FindKernel("main");
+        }
+
+        public RenderTexture Process(RenderTexture map1, RenderTexture map2)
+        {
+            if (map1.width != map1.height || map2.width != map2.height || map1.width != map2.height)
+            {
+                Debug.LogError("Maps need to be uniforms");
+                return null;
+            }
+
+            int resol = map1.width;
+            RenderTexture rst = RenderTexture.GetTemporary(resol, resol, 0, map1.format);
+            rst.enableRandomWrite = true;
+            rst.filterMode = FilterMode.Point;
+            rst.Create();
+
+            CommandBuffer commandBuffer = new CommandBuffer();
+            commandBuffer.name = "Compute shader";
+            commandBuffer.SetComputeTextureParam(m_shader, s_kernelIndex, s_shaderPropertyMap1, map1);
+            commandBuffer.SetComputeTextureParam(m_shader, s_kernelIndex, s_shaderPropertyMap2, map2);
+            commandBuffer.SetComputeTextureParam(m_shader, s_kernelIndex, s_shaderPropertyOutput, rst);
+            commandBuffer.DispatchCompute(m_shader, s_kernelIndex, resol / 8, resol / 8, 1);
+            Graphics.ExecuteCommandBuffer(commandBuffer);
+
+            return rst;
+        }
+    }
+}

Runtime/MapMerger.cs.meta

@@ -0,0 +1,12 @@
+Texture2D<float> _Map1;
+Texture2D<float> _Map2;
+
+RWTexture2D<float> _TextureOut;
+
+#pragma kernel main
+   
+[numthreads(8,8,1)]                            
+void main (uint2 id : SV_DispatchThreadID)
+{
+    _TextureOut[id.xy] = min(_Map1[id.xy] + _Map2[id.xy], 1.f);
+}

Runtime/Shader/InfluenceMapCS.compute.meta

@@ -0,0 +1,13 @@
+Texture2D<float> _Map1;
+Texture2D<float> _Map2;
+
+RWTexture2D<float> _TextureOut;
+
+#pragma kernel main
+   
+[numthreads(8,8,1)]                            
+void main (uint2 id : SV_DispatchThreadID)
+{
+    // Commune
+    _TextureOut[id.xy] = min((_Map1[id.xy] > 0.f) && (_Map2[id.xy] > 0.f), 1.f);
+}

Runtime/Shader/Operation.meta

@@ -0,0 +1,13 @@
+Texture2D<float> _Map1;
+Texture2D<float> _Map2;
+
+RWTexture2D<float> _TextureOut;
+
+#pragma kernel main
+   
+[numthreads(8,8,1)]                            
+void main (uint2 id : SV_DispatchThreadID)
+{
+    // Combine
+    _TextureOut[id.xy] = min((_Map1[id.xy] > 0.f) || (_Map2[id.xy] > 0.f), 1.f);
+}