升级obi

Assets/Obi/Scripts/Common/Backends/Compute/Constraints/Pinhole/ComputePinholeConstraints.cs

@@ -0,0 +1,48 @@
+using System;
+using UnityEngine;
+
+namespace Obi
+{
+    public class ComputePinholeConstraints : ComputeConstraintsImpl<ComputePinholeConstraintsBatch>
+    {
+        public ComputeShader constraintsShader;
+        public int clearKernel;
+        public int initializeKernel;
+        public int projectKernel;
+        public int applyKernel;
+
+        public ComputePinholeConstraints(ComputeSolverImpl solver) : base(solver, Oni.ConstraintType.Pinhole)
+        {
+            constraintsShader = GameObject.Instantiate(Resources.Load<ComputeShader>("Compute/PinholeConstraints"));
+            clearKernel = constraintsShader.FindKernel("Clear");
+            initializeKernel = constraintsShader.FindKernel("Initialize");
+            projectKernel = constraintsShader.FindKernel("Project");
+            applyKernel = constraintsShader.FindKernel("Apply");
+        }
+
+        public override IConstraintsBatchImpl CreateConstraintsBatch()
+        {
+            var dataBatch = new ComputePinholeConstraintsBatch(this);
+            batches.Add(dataBatch);
+            return dataBatch;
+        }
+
+        public override void RemoveBatch(IConstraintsBatchImpl batch)
+        {
+            batches.Remove(batch as ComputePinholeConstraintsBatch);
+            batch.Destroy();
+        }
+
+        public void RequestDataReadback()
+        {
+            foreach (var batch in batches)
+                batch.RequestDataReadback();
+        }
+
+        public void WaitForReadback()
+        {
+            foreach (var batch in batches)
+                batch.WaitForReadback();
+        }
+    }
+}