mirror of
https://github.com/scratchfoundation/bgfx.git
synced 2024-11-25 00:58:30 -05:00
67 lines
1.9 KiB
Python
67 lines
1.9 KiB
Python
/*
|
|
* Copyright 2014 Stanlo Slasinski. All rights reserved.
|
|
* License: http://www.opensource.org/licenses/BSD-2-Clause
|
|
*/
|
|
|
|
#include "bgfx_compute.sh"
|
|
#include "uniforms.sh"
|
|
|
|
BUFFER_RO(prevPositionBuffer, vec4, 0);
|
|
BUFFER_RO(curPositionBuffer, vec4, 1);
|
|
BUFFER_WR(outPrevPositionBuffer, vec4, 2);
|
|
BUFFER_WR(outCurPositionBuffer, vec4, 3);
|
|
|
|
#define GROUP_SIZE 512
|
|
SHARED vec3 otherEntries[GROUP_SIZE];
|
|
|
|
vec3 calcAcceleration(vec3 _curPosition, vec3 _otherPosition)
|
|
{
|
|
vec3 difference = _otherPosition - _curPosition;
|
|
float dist2 = dot(difference, difference);
|
|
float dist6 = dist2 * dist2 * dist2;
|
|
float invDist3 = 1.0 / (sqrt(dist6) + 0.1);
|
|
return u_gravity * invDist3 * difference;
|
|
}
|
|
|
|
NUM_THREADS(GROUP_SIZE, 1, 1)
|
|
void main()
|
|
{
|
|
vec3 prevPosition = prevPositionBuffer[gl_GlobalInvocationID.x].xyz;
|
|
vec3 curPosition = curPositionBuffer[ gl_GlobalInvocationID.x].xyz;
|
|
|
|
vec3 newAcceleration = vec3_splat(0.0);
|
|
|
|
for (int j = 0; j < int(u_dispatchSize); ++j)
|
|
{
|
|
otherEntries[gl_LocalInvocationIndex] = curPositionBuffer[j * GROUP_SIZE + int(gl_LocalInvocationIndex)].xyz;
|
|
|
|
barrier();
|
|
for (int i = 0; i < GROUP_SIZE; ++i)
|
|
{
|
|
newAcceleration += calcAcceleration(curPosition, otherEntries[i]);
|
|
}
|
|
}
|
|
|
|
newAcceleration += (prevPosition - curPosition) * u_damping;
|
|
float accelerationMagnitude = length(newAcceleration);
|
|
float color = pow(min(accelerationMagnitude / 3.0, 1.0), 0.25);
|
|
if (accelerationMagnitude > 0.0)
|
|
{
|
|
newAcceleration = normalize(newAcceleration) * min(accelerationMagnitude, u_maxAcceleration);
|
|
}
|
|
|
|
vec3 newPosition = 2.0 * curPosition - prevPosition + newAcceleration * u_timeStep;
|
|
|
|
if (any(isnan(newPosition) ) )
|
|
{
|
|
newPosition = vec3_splat(0.0);
|
|
}
|
|
|
|
if (any(isnan(curPosition) ) )
|
|
{
|
|
curPosition = vec3_splat(0.0);
|
|
}
|
|
|
|
outPrevPositionBuffer[gl_GlobalInvocationID.x] = vec4(curPosition, 0.0);
|
|
outCurPositionBuffer[ gl_GlobalInvocationID.x] = vec4(newPosition, color);
|
|
}
|