升级obi

Assets/Obi/Scripts/Common/Backends/Compute/Constraints/Stitch/ComputeStitchConstraints.cs

@@ -0,0 +1,32 @@
+using System;
+using UnityEngine;
+
+namespace Obi
+{
+    public class ComputeStitchConstraints : ComputeConstraintsImpl<ComputeStitchConstraintsBatch>
+    {
+        public ComputeShader constraintsShader;
+        public int projectKernel;
+        public int applyKernel;
+
+        public ComputeStitchConstraints(ComputeSolverImpl solver) : base(solver, Oni.ConstraintType.Stitch)
+        {
+            constraintsShader = GameObject.Instantiate(Resources.Load<ComputeShader>("Compute/StitchConstraints"));
+            projectKernel = constraintsShader.FindKernel("Project");
+            applyKernel = constraintsShader.FindKernel("Apply");
+        }
+
+        public override IConstraintsBatchImpl CreateConstraintsBatch()
+        {
+            var dataBatch = new ComputeStitchConstraintsBatch(this);
+            batches.Add(dataBatch);
+            return dataBatch;
+        }
+
+        public override void RemoveBatch(IConstraintsBatchImpl batch)
+        {
+            batches.Remove(batch as ComputeStitchConstraintsBatch);
+            batch.Destroy();
+        }
+    }
+}

Assets/Obi/Scripts/Common/Backends/Compute/Constraints/Stitch/ComputeStitchConstraints.cs.meta

@@ -0,0 +1,11 @@
+fileFormatVersion: 2
+guid: 35350420ce5c24323abc6770d4de9655
+MonoImporter:
+  externalObjects: {}
+  serializedVersion: 2
+  defaultReferences: []
+  executionOrder: 0
+  icon: {instanceID: 0}
+  userData: 
+  assetBundleName: 
+  assetBundleVariant: 

Assets/Obi/Scripts/Common/Backends/Compute/Constraints/Stitch/ComputeStitchConstraintsBatch.cs

@@ -0,0 +1,71 @@
+using UnityEngine;
+
+namespace Obi
+{
+    public class ComputeStitchConstraintsBatch : ComputeConstraintsBatchImpl, IStitchConstraintsBatchImpl
+    {
+        GraphicsBuffer stiffnesses;
+
+        public ComputeStitchConstraintsBatch(ComputeStitchConstraints constraints)
+        {
+            m_Constraints = constraints;
+            m_ConstraintType = Oni.ConstraintType.Stitch;
+        }
+
+        public void SetStitchConstraints(ObiNativeIntList particleIndices, ObiNativeFloatList stiffnesses, ObiNativeFloatList lambdas, int count)
+        {
+            this.particleIndices = particleIndices.AsComputeBuffer<int>();
+            this.stiffnesses = stiffnesses.AsComputeBuffer<float>();
+            this.lambdas = lambdas.AsComputeBuffer<float>();
+
+            m_ConstraintCount = count;
+        }
+
+        public override void Evaluate(float stepTime, float substepTime, int steps, float timeLeft)
+        {
+            if (m_ConstraintCount > 0)
+            {
+                var shader = ((ComputeStitchConstraints)m_Constraints).constraintsShader;
+                int projectKernel = ((ComputeStitchConstraints)m_Constraints).projectKernel;
+
+                shader.SetBuffer(projectKernel, "particleIndices", particleIndices);
+                shader.SetBuffer(projectKernel, "stiffnesses", stiffnesses);
+                shader.SetBuffer(projectKernel, "lambdas", lambdas);
+
+                shader.SetBuffer(projectKernel, "positions", solverImplementation.positionsBuffer);
+                shader.SetBuffer(projectKernel, "invMasses", solverImplementation.invMassesBuffer);
+                shader.SetBuffer(projectKernel, "deltasAsInt", solverImplementation.positionDeltasIntBuffer);
+                shader.SetBuffer(projectKernel, "positionConstraintCounts", solverImplementation.positionConstraintCountBuffer);
+
+                shader.SetInt("activeConstraintCount", m_ConstraintCount);
+                shader.SetFloat("deltaTime", substepTime);
+
+                int threadGroups = ComputeMath.ThreadGroupCount(m_ConstraintCount, 128);
+                shader.Dispatch(projectKernel, threadGroups, 1, 1);
+            }
+        }
+
+        public override void Apply(float substepTime)
+        {
+            if (m_ConstraintCount > 0)
+            {
+                var parameters = solverAbstraction.GetConstraintParameters(m_ConstraintType);
+
+                var shader = ((ComputeStitchConstraints)m_Constraints).constraintsShader;
+                int applyKernel = ((ComputeStitchConstraints)m_Constraints).applyKernel;
+
+                shader.SetBuffer(applyKernel, "particleIndices", particleIndices);
+                shader.SetBuffer(applyKernel, "positions", solverImplementation.positionsBuffer);
+                shader.SetBuffer(applyKernel, "deltasAsInt", solverImplementation.positionDeltasIntBuffer);
+                shader.SetBuffer(applyKernel, "positionConstraintCounts", solverImplementation.positionConstraintCountBuffer);
+
+                shader.SetInt("activeConstraintCount", m_ConstraintCount);
+                shader.SetFloat("sorFactor", parameters.SORFactor);
+
+                int threadGroups = ComputeMath.ThreadGroupCount(m_ConstraintCount, 128);
+                shader.Dispatch(applyKernel, threadGroups, 1, 1);
+            }
+        }
+
+    }
+}

Assets/Obi/Scripts/Common/Backends/Compute/Constraints/Stitch/ComputeStitchConstraintsBatch.cs.meta

@@ -0,0 +1,11 @@
+fileFormatVersion: 2
+guid: d82f622203a3d402dbbd5f47dd78fed3
+MonoImporter:
+  externalObjects: {}
+  serializedVersion: 2
+  defaultReferences: []
+  executionOrder: 0
+  icon: {instanceID: 0}
+  userData: 
+  assetBundleName: 
+  assetBundleVariant: