src/BulletMultiThreaded/GpuSoftBodySolvers/DX11/HLSL/UpdateConstants.hlsl

   1 MSTRINGIFY(\r
   2 \r
   3 cbuffer UpdateConstantsCB : register( b0 )\r
   4 {\r
   5         int numLinks;\r
   6         int padding0;\r
   7         int padding1;\r
   8         int padding2;\r
   9 };\r
  10 \r
  11 // Node indices for each link\r
  12 StructuredBuffer<int2> g_linksVertexIndices : register( t0 );\r
  13 StructuredBuffer<float4> g_vertexPositions : register( t1 );\r
  14 StructuredBuffer<float> g_vertexInverseMasses : register( t2 );\r
  15 StructuredBuffer<float> g_linksMaterialLSC : register( t3 );\r
  16 \r
  17 RWStructuredBuffer<float> g_linksMassLSC : register( u0 );\r
  18 RWStructuredBuffer<float> g_linksRestLengthSquared : register( u1 );\r
  19 RWStructuredBuffer<float> g_linksRestLengths : register( u2 );\r
  20 \r
  21 [numthreads(128, 1, 1)]\r
  22 void \r
  23 UpdateConstantsKernel( uint3 Gid : SV_GroupID, uint3 DTid : SV_DispatchThreadID, uint3 GTid : SV_GroupThreadID, uint GI : SV_GroupIndex )\r
  24 {\r
  25         int linkID = DTid.x;\r
  26         if( linkID < numLinks )\r
  27         {       \r
  28                 int2 nodeIndices = g_linksVertexIndices[linkID];\r
  29                 int node0 = nodeIndices.x;\r
  30                 int node1 = nodeIndices.y;\r
  31                 float linearStiffnessCoefficient = g_linksMaterialLSC[ linkID ];\r
  32                 \r
  33                 float3 position0 = g_vertexPositions[node0].xyz;\r
  34                 float3 position1 = g_vertexPositions[node1].xyz;\r
  35                 float inverseMass0 = g_vertexInverseMasses[node0];\r
  36                 float inverseMass1 = g_vertexInverseMasses[node1];\r
  37 \r
  38                 float3 difference = position0 - position1;\r
  39                 float length2 = dot(difference, difference);\r
  40                 float length = sqrt(length2);\r
  41         \r
  42                 g_linksRestLengths[linkID] = length;\r
  43                 g_linksMassLSC[linkID] = (inverseMass0 + inverseMass1)/linearStiffnessCoefficient;\r
  44                 g_linksRestLengthSquared[linkID] = length*length;               \r
  45         }\r
  46 }\r
  47 \r
  48 );