1159 lines
40 KiB
C++
1159 lines
40 KiB
C++
|
|
||
|
/*
|
||
|
Copyright (c) 2013 Advanced Micro Devices, Inc.
|
||
|
|
||
|
This software is provided 'as-is', without any express or implied warranty.
|
||
|
In no event will the authors be held liable for any damages arising from the use of this software.
|
||
|
Permission is granted to anyone to use this software for any purpose,
|
||
|
including commercial applications, and to alter it and redistribute it freely,
|
||
|
subject to the following restrictions:
|
||
|
|
||
|
1. The origin of this software must not be misrepresented; you must not claim that you wrote the original software. If you use this software in a product, an acknowledgment in the product documentation would be appreciated but is not required.
|
||
|
2. Altered source versions must be plainly marked as such, and must not be misrepresented as being the original software.
|
||
|
3. This notice may not be removed or altered from any source distribution.
|
||
|
*/
|
||
|
//Originally written by Erwin Coumans
|
||
|
|
||
|
|
||
|
bool useGpuInitSolverBodies = true;
|
||
|
bool useGpuInfo1 = true;
|
||
|
bool useGpuInfo2= true;
|
||
|
bool useGpuSolveJointConstraintRows=true;
|
||
|
bool useGpuWriteBackVelocities = true;
|
||
|
bool gpuBreakConstraints = true;
|
||
|
|
||
|
#include "b3GpuPgsConstraintSolver.h"
|
||
|
|
||
|
#include "Bullet3Collision/NarrowPhaseCollision/shared/b3RigidBodyData.h"
|
||
|
|
||
|
#include "Bullet3Dynamics/ConstraintSolver/b3TypedConstraint.h"
|
||
|
#include <new>
|
||
|
#include "Bullet3Common/b3AlignedObjectArray.h"
|
||
|
#include <string.h> //for memset
|
||
|
#include "Bullet3Collision/NarrowPhaseCollision/b3Contact4.h"
|
||
|
#include "Bullet3OpenCL/ParallelPrimitives/b3OpenCLArray.h"
|
||
|
#include "Bullet3OpenCL/ParallelPrimitives/b3LauncherCL.h"
|
||
|
|
||
|
#include "Bullet3OpenCL/ParallelPrimitives/b3PrefixScanCL.h"
|
||
|
|
||
|
#include "Bullet3OpenCL/RigidBody/kernels/jointSolver.h" //solveConstraintRowsCL
|
||
|
#include "Bullet3OpenCL/Initialize/b3OpenCLUtils.h"
|
||
|
|
||
|
#define B3_JOINT_SOLVER_PATH "src/Bullet3OpenCL/RigidBody/kernels/jointSolver.cl"
|
||
|
|
||
|
|
||
|
struct b3GpuPgsJacobiSolverInternalData
|
||
|
{
|
||
|
|
||
|
cl_context m_context;
|
||
|
cl_device_id m_device;
|
||
|
cl_command_queue m_queue;
|
||
|
|
||
|
b3PrefixScanCL* m_prefixScan;
|
||
|
|
||
|
cl_kernel m_solveJointConstraintRowsKernels;
|
||
|
cl_kernel m_initSolverBodiesKernel;
|
||
|
cl_kernel m_getInfo1Kernel;
|
||
|
cl_kernel m_initBatchConstraintsKernel;
|
||
|
cl_kernel m_getInfo2Kernel;
|
||
|
cl_kernel m_writeBackVelocitiesKernel;
|
||
|
cl_kernel m_breakViolatedConstraintsKernel;
|
||
|
|
||
|
b3OpenCLArray<unsigned int>* m_gpuConstraintRowOffsets;
|
||
|
|
||
|
b3OpenCLArray<b3GpuSolverBody>* m_gpuSolverBodies;
|
||
|
b3OpenCLArray<b3BatchConstraint>* m_gpuBatchConstraints;
|
||
|
b3OpenCLArray<b3GpuSolverConstraint>* m_gpuConstraintRows;
|
||
|
b3OpenCLArray<unsigned int>* m_gpuConstraintInfo1;
|
||
|
|
||
|
// b3AlignedObjectArray<b3GpuSolverBody> m_cpuSolverBodies;
|
||
|
b3AlignedObjectArray<b3BatchConstraint> m_cpuBatchConstraints;
|
||
|
b3AlignedObjectArray<b3GpuSolverConstraint> m_cpuConstraintRows;
|
||
|
b3AlignedObjectArray<unsigned int> m_cpuConstraintInfo1;
|
||
|
b3AlignedObjectArray<unsigned int> m_cpuConstraintRowOffsets;
|
||
|
|
||
|
b3AlignedObjectArray<b3RigidBodyData> m_cpuBodies;
|
||
|
b3AlignedObjectArray<b3InertiaData> m_cpuInertias;
|
||
|
|
||
|
|
||
|
b3AlignedObjectArray<b3GpuGenericConstraint> m_cpuConstraints;
|
||
|
|
||
|
b3AlignedObjectArray<int> m_batchSizes;
|
||
|
|
||
|
|
||
|
};
|
||
|
|
||
|
|
||
|
/*
|
||
|
static b3Transform getWorldTransform(b3RigidBodyData* rb)
|
||
|
{
|
||
|
b3Transform newTrans;
|
||
|
newTrans.setOrigin(rb->m_pos);
|
||
|
newTrans.setRotation(rb->m_quat);
|
||
|
return newTrans;
|
||
|
}
|
||
|
|
||
|
static const b3Matrix3x3& getInvInertiaTensorWorld(b3InertiaData* inertia)
|
||
|
{
|
||
|
return inertia->m_invInertiaWorld;
|
||
|
}
|
||
|
|
||
|
*/
|
||
|
|
||
|
static const b3Vector3& getLinearVelocity(b3RigidBodyData* rb)
|
||
|
{
|
||
|
return rb->m_linVel;
|
||
|
}
|
||
|
|
||
|
static const b3Vector3& getAngularVelocity(b3RigidBodyData* rb)
|
||
|
{
|
||
|
return rb->m_angVel;
|
||
|
}
|
||
|
|
||
|
b3Vector3 getVelocityInLocalPoint(b3RigidBodyData* rb, const b3Vector3& rel_pos)
|
||
|
{
|
||
|
//we also calculate lin/ang velocity for kinematic objects
|
||
|
return getLinearVelocity(rb) + getAngularVelocity(rb).cross(rel_pos);
|
||
|
|
||
|
}
|
||
|
|
||
|
|
||
|
|
||
|
b3GpuPgsConstraintSolver::b3GpuPgsConstraintSolver (cl_context ctx, cl_device_id device, cl_command_queue queue,bool usePgs)
|
||
|
{
|
||
|
m_usePgs = usePgs;
|
||
|
m_gpuData = new b3GpuPgsJacobiSolverInternalData();
|
||
|
m_gpuData->m_context = ctx;
|
||
|
m_gpuData->m_device = device;
|
||
|
m_gpuData->m_queue = queue;
|
||
|
|
||
|
m_gpuData->m_prefixScan = new b3PrefixScanCL(ctx,device,queue);
|
||
|
|
||
|
m_gpuData->m_gpuConstraintRowOffsets = new b3OpenCLArray<unsigned int>(m_gpuData->m_context,m_gpuData->m_queue);
|
||
|
|
||
|
m_gpuData->m_gpuSolverBodies = new b3OpenCLArray<b3GpuSolverBody>(m_gpuData->m_context,m_gpuData->m_queue);
|
||
|
m_gpuData->m_gpuBatchConstraints = new b3OpenCLArray<b3BatchConstraint>(m_gpuData->m_context,m_gpuData->m_queue);
|
||
|
m_gpuData->m_gpuConstraintRows = new b3OpenCLArray<b3GpuSolverConstraint>(m_gpuData->m_context,m_gpuData->m_queue);
|
||
|
m_gpuData->m_gpuConstraintInfo1 = new b3OpenCLArray<unsigned int>(m_gpuData->m_context,m_gpuData->m_queue);
|
||
|
cl_int errNum=0;
|
||
|
|
||
|
{
|
||
|
cl_program prog = b3OpenCLUtils::compileCLProgramFromString(m_gpuData->m_context,m_gpuData->m_device,solveConstraintRowsCL,&errNum,"",B3_JOINT_SOLVER_PATH);
|
||
|
//cl_program prog = b3OpenCLUtils::compileCLProgramFromString(m_gpuData->m_context,m_gpuData->m_device,0,&errNum,"",B3_JOINT_SOLVER_PATH,true);
|
||
|
b3Assert(errNum==CL_SUCCESS);
|
||
|
m_gpuData->m_solveJointConstraintRowsKernels = b3OpenCLUtils::compileCLKernelFromString(m_gpuData->m_context, m_gpuData->m_device,solveConstraintRowsCL, "solveJointConstraintRows",&errNum,prog);
|
||
|
b3Assert(errNum==CL_SUCCESS);
|
||
|
m_gpuData->m_initSolverBodiesKernel = b3OpenCLUtils::compileCLKernelFromString(m_gpuData->m_context,m_gpuData->m_device,solveConstraintRowsCL,"initSolverBodies",&errNum,prog);
|
||
|
b3Assert(errNum==CL_SUCCESS);
|
||
|
m_gpuData->m_getInfo1Kernel = b3OpenCLUtils::compileCLKernelFromString(m_gpuData->m_context,m_gpuData->m_device,solveConstraintRowsCL,"getInfo1Kernel",&errNum,prog);
|
||
|
b3Assert(errNum==CL_SUCCESS);
|
||
|
m_gpuData->m_initBatchConstraintsKernel = b3OpenCLUtils::compileCLKernelFromString(m_gpuData->m_context,m_gpuData->m_device,solveConstraintRowsCL,"initBatchConstraintsKernel",&errNum,prog);
|
||
|
b3Assert(errNum==CL_SUCCESS);
|
||
|
m_gpuData->m_getInfo2Kernel= b3OpenCLUtils::compileCLKernelFromString(m_gpuData->m_context,m_gpuData->m_device,solveConstraintRowsCL,"getInfo2Kernel",&errNum,prog);
|
||
|
b3Assert(errNum==CL_SUCCESS);
|
||
|
m_gpuData->m_writeBackVelocitiesKernel = b3OpenCLUtils::compileCLKernelFromString(m_gpuData->m_context,m_gpuData->m_device,solveConstraintRowsCL,"writeBackVelocitiesKernel",&errNum,prog);
|
||
|
b3Assert(errNum==CL_SUCCESS);
|
||
|
m_gpuData->m_breakViolatedConstraintsKernel = b3OpenCLUtils::compileCLKernelFromString(m_gpuData->m_context,m_gpuData->m_device,solveConstraintRowsCL,"breakViolatedConstraintsKernel",&errNum,prog);
|
||
|
b3Assert(errNum==CL_SUCCESS);
|
||
|
|
||
|
|
||
|
|
||
|
|
||
|
clReleaseProgram(prog);
|
||
|
}
|
||
|
|
||
|
|
||
|
}
|
||
|
|
||
|
b3GpuPgsConstraintSolver::~b3GpuPgsConstraintSolver ()
|
||
|
{
|
||
|
clReleaseKernel(m_gpuData->m_solveJointConstraintRowsKernels);
|
||
|
clReleaseKernel(m_gpuData->m_initSolverBodiesKernel);
|
||
|
clReleaseKernel(m_gpuData->m_getInfo1Kernel);
|
||
|
clReleaseKernel(m_gpuData->m_initBatchConstraintsKernel);
|
||
|
clReleaseKernel(m_gpuData->m_getInfo2Kernel);
|
||
|
clReleaseKernel(m_gpuData->m_writeBackVelocitiesKernel);
|
||
|
clReleaseKernel(m_gpuData->m_breakViolatedConstraintsKernel);
|
||
|
|
||
|
delete m_gpuData->m_prefixScan;
|
||
|
delete m_gpuData->m_gpuConstraintRowOffsets;
|
||
|
delete m_gpuData->m_gpuSolverBodies;
|
||
|
delete m_gpuData->m_gpuBatchConstraints;
|
||
|
delete m_gpuData->m_gpuConstraintRows;
|
||
|
delete m_gpuData->m_gpuConstraintInfo1;
|
||
|
|
||
|
delete m_gpuData;
|
||
|
}
|
||
|
|
||
|
struct b3BatchConstraint
|
||
|
{
|
||
|
int m_bodyAPtrAndSignBit;
|
||
|
int m_bodyBPtrAndSignBit;
|
||
|
int m_originalConstraintIndex;
|
||
|
int m_batchId;
|
||
|
};
|
||
|
|
||
|
static b3AlignedObjectArray<b3BatchConstraint> batchConstraints;
|
||
|
|
||
|
|
||
|
void b3GpuPgsConstraintSolver::recomputeBatches()
|
||
|
{
|
||
|
m_gpuData->m_batchSizes.clear();
|
||
|
}
|
||
|
|
||
|
|
||
|
|
||
|
|
||
|
b3Scalar b3GpuPgsConstraintSolver::solveGroupCacheFriendlySetup(b3OpenCLArray<b3RigidBodyData>* gpuBodies, b3OpenCLArray<b3InertiaData>* gpuInertias, int numBodies, b3OpenCLArray<b3GpuGenericConstraint>* gpuConstraints,int numConstraints,const b3ContactSolverInfo& infoGlobal)
|
||
|
{
|
||
|
B3_PROFILE("GPU solveGroupCacheFriendlySetup");
|
||
|
batchConstraints.resize(numConstraints);
|
||
|
m_gpuData->m_gpuBatchConstraints->resize(numConstraints);
|
||
|
m_staticIdx = -1;
|
||
|
m_maxOverrideNumSolverIterations = 0;
|
||
|
|
||
|
|
||
|
/* m_gpuData->m_gpuBodies->resize(numBodies);
|
||
|
m_gpuData->m_gpuBodies->copyFromHostPointer(bodies,numBodies);
|
||
|
|
||
|
b3OpenCLArray<b3InertiaData> gpuInertias(m_gpuData->m_context,m_gpuData->m_queue);
|
||
|
gpuInertias.resize(numBodies);
|
||
|
gpuInertias.copyFromHostPointer(inertias,numBodies);
|
||
|
*/
|
||
|
|
||
|
m_gpuData->m_gpuSolverBodies->resize(numBodies);
|
||
|
|
||
|
|
||
|
m_tmpSolverBodyPool.resize(numBodies);
|
||
|
{
|
||
|
|
||
|
if (useGpuInitSolverBodies)
|
||
|
{
|
||
|
B3_PROFILE("m_initSolverBodiesKernel");
|
||
|
|
||
|
b3LauncherCL launcher(m_gpuData->m_queue,m_gpuData->m_initSolverBodiesKernel,"m_initSolverBodiesKernel");
|
||
|
launcher.setBuffer(m_gpuData->m_gpuSolverBodies->getBufferCL());
|
||
|
launcher.setBuffer(gpuBodies->getBufferCL());
|
||
|
launcher.setConst(numBodies);
|
||
|
launcher.launch1D(numBodies);
|
||
|
clFinish(m_gpuData->m_queue);
|
||
|
|
||
|
// m_gpuData->m_gpuSolverBodies->copyToHost(m_tmpSolverBodyPool);
|
||
|
} else
|
||
|
{
|
||
|
gpuBodies->copyToHost(m_gpuData->m_cpuBodies);
|
||
|
for (int i=0;i<numBodies;i++)
|
||
|
{
|
||
|
|
||
|
b3RigidBodyData& body = m_gpuData->m_cpuBodies[i];
|
||
|
b3GpuSolverBody& solverBody = m_tmpSolverBodyPool[i];
|
||
|
initSolverBody(i,&solverBody,&body);
|
||
|
solverBody.m_originalBodyIndex = i;
|
||
|
}
|
||
|
m_gpuData->m_gpuSolverBodies->copyFromHost(m_tmpSolverBodyPool);
|
||
|
}
|
||
|
}
|
||
|
|
||
|
// int totalBodies = 0;
|
||
|
int totalNumRows = 0;
|
||
|
//b3RigidBody* rb0=0,*rb1=0;
|
||
|
//if (1)
|
||
|
{
|
||
|
{
|
||
|
|
||
|
|
||
|
// int i;
|
||
|
|
||
|
m_tmpConstraintSizesPool.resizeNoInitialize(numConstraints);
|
||
|
|
||
|
// b3OpenCLArray<b3GpuGenericConstraint> gpuConstraints(m_gpuData->m_context,m_gpuData->m_queue);
|
||
|
|
||
|
|
||
|
if (useGpuInfo1)
|
||
|
{
|
||
|
B3_PROFILE("info1 and init batchConstraint");
|
||
|
|
||
|
m_gpuData->m_gpuConstraintInfo1->resize(numConstraints);
|
||
|
|
||
|
|
||
|
if (1)
|
||
|
{
|
||
|
B3_PROFILE("getInfo1Kernel");
|
||
|
|
||
|
b3LauncherCL launcher(m_gpuData->m_queue,m_gpuData->m_getInfo1Kernel,"m_getInfo1Kernel");
|
||
|
launcher.setBuffer(m_gpuData->m_gpuConstraintInfo1->getBufferCL());
|
||
|
launcher.setBuffer(gpuConstraints->getBufferCL());
|
||
|
launcher.setConst(numConstraints);
|
||
|
launcher.launch1D(numConstraints);
|
||
|
clFinish(m_gpuData->m_queue);
|
||
|
}
|
||
|
|
||
|
if (m_gpuData->m_batchSizes.size()==0)
|
||
|
{
|
||
|
B3_PROFILE("initBatchConstraintsKernel");
|
||
|
|
||
|
m_gpuData->m_gpuConstraintRowOffsets->resize(numConstraints);
|
||
|
unsigned int total=0;
|
||
|
m_gpuData->m_prefixScan->execute(*m_gpuData->m_gpuConstraintInfo1,*m_gpuData->m_gpuConstraintRowOffsets,numConstraints,&total);
|
||
|
unsigned int lastElem = m_gpuData->m_gpuConstraintInfo1->at(numConstraints-1);
|
||
|
totalNumRows = total+lastElem;
|
||
|
|
||
|
{
|
||
|
B3_PROFILE("init batch constraints");
|
||
|
b3LauncherCL launcher(m_gpuData->m_queue,m_gpuData->m_initBatchConstraintsKernel,"m_initBatchConstraintsKernel");
|
||
|
launcher.setBuffer(m_gpuData->m_gpuConstraintInfo1->getBufferCL());
|
||
|
launcher.setBuffer(m_gpuData->m_gpuConstraintRowOffsets->getBufferCL());
|
||
|
launcher.setBuffer(m_gpuData->m_gpuBatchConstraints->getBufferCL());
|
||
|
launcher.setBuffer(gpuConstraints->getBufferCL());
|
||
|
launcher.setBuffer(gpuBodies->getBufferCL());
|
||
|
launcher.setConst(numConstraints);
|
||
|
launcher.launch1D(numConstraints);
|
||
|
clFinish(m_gpuData->m_queue);
|
||
|
}
|
||
|
//assume the batching happens on CPU, so copy the data
|
||
|
m_gpuData->m_gpuBatchConstraints->copyToHost(batchConstraints);
|
||
|
}
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
totalNumRows = 0;
|
||
|
gpuConstraints->copyToHost(m_gpuData->m_cpuConstraints);
|
||
|
//calculate the total number of contraint rows
|
||
|
for (int i=0;i<numConstraints;i++)
|
||
|
{
|
||
|
unsigned int& info1= m_tmpConstraintSizesPool[i];
|
||
|
// unsigned int info1;
|
||
|
if (m_gpuData->m_cpuConstraints[i].isEnabled())
|
||
|
{
|
||
|
|
||
|
m_gpuData->m_cpuConstraints[i].getInfo1(&info1,&m_gpuData->m_cpuBodies[0]);
|
||
|
} else
|
||
|
{
|
||
|
info1 = 0;
|
||
|
}
|
||
|
|
||
|
totalNumRows += info1;
|
||
|
}
|
||
|
|
||
|
m_gpuData->m_gpuBatchConstraints->copyFromHost(batchConstraints);
|
||
|
m_gpuData->m_gpuConstraintInfo1->copyFromHost(m_tmpConstraintSizesPool);
|
||
|
|
||
|
}
|
||
|
m_tmpSolverNonContactConstraintPool.resizeNoInitialize(totalNumRows);
|
||
|
m_gpuData->m_gpuConstraintRows->resize(totalNumRows);
|
||
|
|
||
|
// b3GpuConstraintArray verify;
|
||
|
|
||
|
if (useGpuInfo2)
|
||
|
{
|
||
|
{
|
||
|
B3_PROFILE("getInfo2Kernel");
|
||
|
b3LauncherCL launcher(m_gpuData->m_queue,m_gpuData->m_getInfo2Kernel,"m_getInfo2Kernel");
|
||
|
launcher.setBuffer(m_gpuData->m_gpuConstraintRows->getBufferCL());
|
||
|
launcher.setBuffer(m_gpuData->m_gpuConstraintInfo1->getBufferCL());
|
||
|
launcher.setBuffer(m_gpuData->m_gpuConstraintRowOffsets->getBufferCL());
|
||
|
launcher.setBuffer(gpuConstraints->getBufferCL());
|
||
|
launcher.setBuffer(m_gpuData->m_gpuBatchConstraints->getBufferCL());
|
||
|
launcher.setBuffer(gpuBodies->getBufferCL());
|
||
|
launcher.setBuffer(gpuInertias->getBufferCL());
|
||
|
launcher.setBuffer(m_gpuData->m_gpuSolverBodies->getBufferCL());
|
||
|
launcher.setConst(infoGlobal.m_timeStep);
|
||
|
launcher.setConst(infoGlobal.m_erp);
|
||
|
launcher.setConst(infoGlobal.m_globalCfm);
|
||
|
launcher.setConst(infoGlobal.m_damping);
|
||
|
launcher.setConst(infoGlobal.m_numIterations);
|
||
|
launcher.setConst(numConstraints);
|
||
|
launcher.launch1D(numConstraints);
|
||
|
clFinish(m_gpuData->m_queue);
|
||
|
|
||
|
if (m_gpuData->m_batchSizes.size()==0)
|
||
|
m_gpuData->m_gpuBatchConstraints->copyToHost(batchConstraints);
|
||
|
//m_gpuData->m_gpuConstraintRows->copyToHost(verify);
|
||
|
//m_gpuData->m_gpuConstraintRows->copyToHost(m_tmpSolverNonContactConstraintPool);
|
||
|
|
||
|
|
||
|
|
||
|
}
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
|
||
|
gpuInertias->copyToHost(m_gpuData->m_cpuInertias);
|
||
|
|
||
|
///setup the b3SolverConstraints
|
||
|
|
||
|
for (int i=0;i<numConstraints;i++)
|
||
|
{
|
||
|
const int& info1 = m_tmpConstraintSizesPool[i];
|
||
|
|
||
|
if (info1)
|
||
|
{
|
||
|
int constraintIndex = batchConstraints[i].m_originalConstraintIndex;
|
||
|
int constraintRowOffset = m_gpuData->m_cpuConstraintRowOffsets[constraintIndex];
|
||
|
|
||
|
b3GpuSolverConstraint* currentConstraintRow = &m_tmpSolverNonContactConstraintPool[constraintRowOffset];
|
||
|
b3GpuGenericConstraint& constraint = m_gpuData->m_cpuConstraints[i];
|
||
|
|
||
|
b3RigidBodyData& rbA = m_gpuData->m_cpuBodies[ constraint.getRigidBodyA()];
|
||
|
//b3RigidBody& rbA = constraint.getRigidBodyA();
|
||
|
// b3RigidBody& rbB = constraint.getRigidBodyB();
|
||
|
b3RigidBodyData& rbB = m_gpuData->m_cpuBodies[ constraint.getRigidBodyB()];
|
||
|
|
||
|
|
||
|
|
||
|
int solverBodyIdA = constraint.getRigidBodyA();//getOrInitSolverBody(constraint.getRigidBodyA(),bodies,inertias);
|
||
|
int solverBodyIdB = constraint.getRigidBodyB();//getOrInitSolverBody(constraint.getRigidBodyB(),bodies,inertias);
|
||
|
|
||
|
b3GpuSolverBody* bodyAPtr = &m_tmpSolverBodyPool[solverBodyIdA];
|
||
|
b3GpuSolverBody* bodyBPtr = &m_tmpSolverBodyPool[solverBodyIdB];
|
||
|
|
||
|
if (rbA.m_invMass)
|
||
|
{
|
||
|
batchConstraints[i].m_bodyAPtrAndSignBit = solverBodyIdA;
|
||
|
} else
|
||
|
{
|
||
|
if (!solverBodyIdA)
|
||
|
m_staticIdx = 0;
|
||
|
batchConstraints[i].m_bodyAPtrAndSignBit = -solverBodyIdA;
|
||
|
}
|
||
|
|
||
|
if (rbB.m_invMass)
|
||
|
{
|
||
|
batchConstraints[i].m_bodyBPtrAndSignBit = solverBodyIdB;
|
||
|
} else
|
||
|
{
|
||
|
if (!solverBodyIdB)
|
||
|
m_staticIdx = 0;
|
||
|
batchConstraints[i].m_bodyBPtrAndSignBit = -solverBodyIdB;
|
||
|
}
|
||
|
|
||
|
|
||
|
int overrideNumSolverIterations = 0;//constraint->getOverrideNumSolverIterations() > 0 ? constraint->getOverrideNumSolverIterations() : infoGlobal.m_numIterations;
|
||
|
if (overrideNumSolverIterations>m_maxOverrideNumSolverIterations)
|
||
|
m_maxOverrideNumSolverIterations = overrideNumSolverIterations;
|
||
|
|
||
|
|
||
|
int j;
|
||
|
for ( j=0;j<info1;j++)
|
||
|
{
|
||
|
memset(¤tConstraintRow[j],0,sizeof(b3GpuSolverConstraint));
|
||
|
currentConstraintRow[j].m_angularComponentA.setValue(0,0,0);
|
||
|
currentConstraintRow[j].m_angularComponentB.setValue(0,0,0);
|
||
|
currentConstraintRow[j].m_appliedImpulse = 0.f;
|
||
|
currentConstraintRow[j].m_appliedPushImpulse = 0.f;
|
||
|
currentConstraintRow[j].m_cfm = 0.f;
|
||
|
currentConstraintRow[j].m_contactNormal.setValue(0,0,0);
|
||
|
currentConstraintRow[j].m_friction = 0.f;
|
||
|
currentConstraintRow[j].m_frictionIndex = 0;
|
||
|
currentConstraintRow[j].m_jacDiagABInv = 0.f;
|
||
|
currentConstraintRow[j].m_lowerLimit = 0.f;
|
||
|
currentConstraintRow[j].m_upperLimit = 0.f;
|
||
|
|
||
|
currentConstraintRow[j].m_originalContactPoint = 0;
|
||
|
currentConstraintRow[j].m_overrideNumSolverIterations = 0;
|
||
|
currentConstraintRow[j].m_relpos1CrossNormal.setValue(0,0,0);
|
||
|
currentConstraintRow[j].m_relpos2CrossNormal.setValue(0,0,0);
|
||
|
currentConstraintRow[j].m_rhs = 0.f;
|
||
|
currentConstraintRow[j].m_rhsPenetration = 0.f;
|
||
|
currentConstraintRow[j].m_solverBodyIdA = 0;
|
||
|
currentConstraintRow[j].m_solverBodyIdB = 0;
|
||
|
|
||
|
currentConstraintRow[j].m_lowerLimit = -B3_INFINITY;
|
||
|
currentConstraintRow[j].m_upperLimit = B3_INFINITY;
|
||
|
currentConstraintRow[j].m_appliedImpulse = 0.f;
|
||
|
currentConstraintRow[j].m_appliedPushImpulse = 0.f;
|
||
|
currentConstraintRow[j].m_solverBodyIdA = solverBodyIdA;
|
||
|
currentConstraintRow[j].m_solverBodyIdB = solverBodyIdB;
|
||
|
currentConstraintRow[j].m_overrideNumSolverIterations = overrideNumSolverIterations;
|
||
|
}
|
||
|
|
||
|
bodyAPtr->internalGetDeltaLinearVelocity().setValue(0.f,0.f,0.f);
|
||
|
bodyAPtr->internalGetDeltaAngularVelocity().setValue(0.f,0.f,0.f);
|
||
|
bodyAPtr->internalGetPushVelocity().setValue(0.f,0.f,0.f);
|
||
|
bodyAPtr->internalGetTurnVelocity().setValue(0.f,0.f,0.f);
|
||
|
bodyBPtr->internalGetDeltaLinearVelocity().setValue(0.f,0.f,0.f);
|
||
|
bodyBPtr->internalGetDeltaAngularVelocity().setValue(0.f,0.f,0.f);
|
||
|
bodyBPtr->internalGetPushVelocity().setValue(0.f,0.f,0.f);
|
||
|
bodyBPtr->internalGetTurnVelocity().setValue(0.f,0.f,0.f);
|
||
|
|
||
|
|
||
|
b3GpuConstraintInfo2 info2;
|
||
|
info2.fps = 1.f/infoGlobal.m_timeStep;
|
||
|
info2.erp = infoGlobal.m_erp;
|
||
|
info2.m_J1linearAxis = currentConstraintRow->m_contactNormal;
|
||
|
info2.m_J1angularAxis = currentConstraintRow->m_relpos1CrossNormal;
|
||
|
info2.m_J2linearAxis = 0;
|
||
|
info2.m_J2angularAxis = currentConstraintRow->m_relpos2CrossNormal;
|
||
|
info2.rowskip = sizeof(b3GpuSolverConstraint)/sizeof(b3Scalar);//check this
|
||
|
///the size of b3GpuSolverConstraint needs be a multiple of b3Scalar
|
||
|
b3Assert(info2.rowskip*sizeof(b3Scalar)== sizeof(b3GpuSolverConstraint));
|
||
|
info2.m_constraintError = ¤tConstraintRow->m_rhs;
|
||
|
currentConstraintRow->m_cfm = infoGlobal.m_globalCfm;
|
||
|
info2.m_damping = infoGlobal.m_damping;
|
||
|
info2.cfm = ¤tConstraintRow->m_cfm;
|
||
|
info2.m_lowerLimit = ¤tConstraintRow->m_lowerLimit;
|
||
|
info2.m_upperLimit = ¤tConstraintRow->m_upperLimit;
|
||
|
info2.m_numIterations = infoGlobal.m_numIterations;
|
||
|
m_gpuData->m_cpuConstraints[i].getInfo2(&info2,&m_gpuData->m_cpuBodies[0]);
|
||
|
|
||
|
///finalize the constraint setup
|
||
|
for ( j=0;j<info1;j++)
|
||
|
{
|
||
|
b3GpuSolverConstraint& solverConstraint = currentConstraintRow[j];
|
||
|
|
||
|
if (solverConstraint.m_upperLimit>=m_gpuData->m_cpuConstraints[i].getBreakingImpulseThreshold())
|
||
|
{
|
||
|
solverConstraint.m_upperLimit = m_gpuData->m_cpuConstraints[i].getBreakingImpulseThreshold();
|
||
|
}
|
||
|
|
||
|
if (solverConstraint.m_lowerLimit<=-m_gpuData->m_cpuConstraints[i].getBreakingImpulseThreshold())
|
||
|
{
|
||
|
solverConstraint.m_lowerLimit = -m_gpuData->m_cpuConstraints[i].getBreakingImpulseThreshold();
|
||
|
}
|
||
|
|
||
|
// solverConstraint.m_originalContactPoint = constraint;
|
||
|
|
||
|
b3Matrix3x3& invInertiaWorldA= m_gpuData->m_cpuInertias[constraint.getRigidBodyA()].m_invInertiaWorld;
|
||
|
{
|
||
|
|
||
|
//b3Vector3 angularFactorA(1,1,1);
|
||
|
const b3Vector3& ftorqueAxis1 = solverConstraint.m_relpos1CrossNormal;
|
||
|
solverConstraint.m_angularComponentA = invInertiaWorldA*ftorqueAxis1;//*angularFactorA;
|
||
|
}
|
||
|
|
||
|
b3Matrix3x3& invInertiaWorldB= m_gpuData->m_cpuInertias[constraint.getRigidBodyB()].m_invInertiaWorld;
|
||
|
{
|
||
|
|
||
|
const b3Vector3& ftorqueAxis2 = solverConstraint.m_relpos2CrossNormal;
|
||
|
solverConstraint.m_angularComponentB = invInertiaWorldB*ftorqueAxis2;//*constraint.getRigidBodyB().getAngularFactor();
|
||
|
}
|
||
|
|
||
|
{
|
||
|
//it is ok to use solverConstraint.m_contactNormal instead of -solverConstraint.m_contactNormal
|
||
|
//because it gets multiplied iMJlB
|
||
|
b3Vector3 iMJlA = solverConstraint.m_contactNormal*rbA.m_invMass;
|
||
|
b3Vector3 iMJaA = invInertiaWorldA*solverConstraint.m_relpos1CrossNormal;
|
||
|
b3Vector3 iMJlB = solverConstraint.m_contactNormal*rbB.m_invMass;//sign of normal?
|
||
|
b3Vector3 iMJaB = invInertiaWorldB*solverConstraint.m_relpos2CrossNormal;
|
||
|
|
||
|
b3Scalar sum = iMJlA.dot(solverConstraint.m_contactNormal);
|
||
|
sum += iMJaA.dot(solverConstraint.m_relpos1CrossNormal);
|
||
|
sum += iMJlB.dot(solverConstraint.m_contactNormal);
|
||
|
sum += iMJaB.dot(solverConstraint.m_relpos2CrossNormal);
|
||
|
b3Scalar fsum = b3Fabs(sum);
|
||
|
b3Assert(fsum > B3_EPSILON);
|
||
|
solverConstraint.m_jacDiagABInv = fsum>B3_EPSILON?b3Scalar(1.)/sum : 0.f;
|
||
|
}
|
||
|
|
||
|
|
||
|
///fix rhs
|
||
|
///todo: add force/torque accelerators
|
||
|
{
|
||
|
b3Scalar rel_vel;
|
||
|
b3Scalar vel1Dotn = solverConstraint.m_contactNormal.dot(rbA.m_linVel) + solverConstraint.m_relpos1CrossNormal.dot(rbA.m_angVel);
|
||
|
b3Scalar vel2Dotn = -solverConstraint.m_contactNormal.dot(rbB.m_linVel) + solverConstraint.m_relpos2CrossNormal.dot(rbB.m_angVel);
|
||
|
|
||
|
rel_vel = vel1Dotn+vel2Dotn;
|
||
|
|
||
|
b3Scalar restitution = 0.f;
|
||
|
b3Scalar positionalError = solverConstraint.m_rhs;//already filled in by getConstraintInfo2
|
||
|
b3Scalar velocityError = restitution - rel_vel * info2.m_damping;
|
||
|
b3Scalar penetrationImpulse = positionalError*solverConstraint.m_jacDiagABInv;
|
||
|
b3Scalar velocityImpulse = velocityError *solverConstraint.m_jacDiagABInv;
|
||
|
solverConstraint.m_rhs = penetrationImpulse+velocityImpulse;
|
||
|
solverConstraint.m_appliedImpulse = 0.f;
|
||
|
|
||
|
}
|
||
|
}
|
||
|
|
||
|
}
|
||
|
}
|
||
|
|
||
|
|
||
|
|
||
|
m_gpuData->m_gpuConstraintRows->copyFromHost(m_tmpSolverNonContactConstraintPool);
|
||
|
m_gpuData->m_gpuConstraintInfo1->copyFromHost(m_tmpConstraintSizesPool);
|
||
|
|
||
|
if (m_gpuData->m_batchSizes.size()==0)
|
||
|
m_gpuData->m_gpuBatchConstraints->copyFromHost(batchConstraints);
|
||
|
else
|
||
|
m_gpuData->m_gpuBatchConstraints->copyToHost(batchConstraints);
|
||
|
|
||
|
m_gpuData->m_gpuSolverBodies->copyFromHost(m_tmpSolverBodyPool);
|
||
|
|
||
|
|
||
|
|
||
|
}//end useGpuInfo2
|
||
|
|
||
|
|
||
|
}
|
||
|
|
||
|
#ifdef B3_SUPPORT_CONTACT_CONSTRAINTS
|
||
|
{
|
||
|
int i;
|
||
|
|
||
|
for (i=0;i<numManifolds;i++)
|
||
|
{
|
||
|
b3Contact4& manifold = manifoldPtr[i];
|
||
|
convertContact(bodies,inertias,&manifold,infoGlobal);
|
||
|
}
|
||
|
}
|
||
|
#endif //B3_SUPPORT_CONTACT_CONSTRAINTS
|
||
|
}
|
||
|
|
||
|
// b3ContactSolverInfo info = infoGlobal;
|
||
|
|
||
|
|
||
|
// int numNonContactPool = m_tmpSolverNonContactConstraintPool.size();
|
||
|
// int numConstraintPool = m_tmpSolverContactConstraintPool.size();
|
||
|
// int numFrictionPool = m_tmpSolverContactFrictionConstraintPool.size();
|
||
|
|
||
|
|
||
|
return 0.f;
|
||
|
|
||
|
}
|
||
|
|
||
|
|
||
|
|
||
|
///a straight copy from GPU/OpenCL kernel, for debugging
|
||
|
__inline void internalApplyImpulse( b3GpuSolverBody* body, const b3Vector3& linearComponent, const b3Vector3& angularComponent,float impulseMagnitude)
|
||
|
{
|
||
|
body->m_deltaLinearVelocity += linearComponent*impulseMagnitude*body->m_linearFactor;
|
||
|
body->m_deltaAngularVelocity += angularComponent*(impulseMagnitude*body->m_angularFactor);
|
||
|
}
|
||
|
|
||
|
|
||
|
void resolveSingleConstraintRowGeneric2( b3GpuSolverBody* body1, b3GpuSolverBody* body2, b3GpuSolverConstraint* c)
|
||
|
{
|
||
|
float deltaImpulse = c->m_rhs-b3Scalar(c->m_appliedImpulse)*c->m_cfm;
|
||
|
float deltaVel1Dotn = b3Dot(c->m_contactNormal,body1->m_deltaLinearVelocity) + b3Dot(c->m_relpos1CrossNormal,body1->m_deltaAngularVelocity);
|
||
|
float deltaVel2Dotn = -b3Dot(c->m_contactNormal,body2->m_deltaLinearVelocity) + b3Dot(c->m_relpos2CrossNormal,body2->m_deltaAngularVelocity);
|
||
|
|
||
|
deltaImpulse -= deltaVel1Dotn*c->m_jacDiagABInv;
|
||
|
deltaImpulse -= deltaVel2Dotn*c->m_jacDiagABInv;
|
||
|
|
||
|
float sum = b3Scalar(c->m_appliedImpulse) + deltaImpulse;
|
||
|
if (sum < c->m_lowerLimit)
|
||
|
{
|
||
|
deltaImpulse = c->m_lowerLimit-b3Scalar(c->m_appliedImpulse);
|
||
|
c->m_appliedImpulse = c->m_lowerLimit;
|
||
|
}
|
||
|
else if (sum > c->m_upperLimit)
|
||
|
{
|
||
|
deltaImpulse = c->m_upperLimit-b3Scalar(c->m_appliedImpulse);
|
||
|
c->m_appliedImpulse = c->m_upperLimit;
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
c->m_appliedImpulse = sum;
|
||
|
}
|
||
|
|
||
|
internalApplyImpulse(body1,c->m_contactNormal*body1->m_invMass,c->m_angularComponentA,deltaImpulse);
|
||
|
internalApplyImpulse(body2,-c->m_contactNormal*body2->m_invMass,c->m_angularComponentB,deltaImpulse);
|
||
|
|
||
|
}
|
||
|
|
||
|
|
||
|
|
||
|
void b3GpuPgsConstraintSolver::initSolverBody(int bodyIndex, b3GpuSolverBody* solverBody, b3RigidBodyData* rb)
|
||
|
{
|
||
|
|
||
|
solverBody->m_deltaLinearVelocity.setValue(0.f,0.f,0.f);
|
||
|
solverBody->m_deltaAngularVelocity.setValue(0.f,0.f,0.f);
|
||
|
solverBody->internalGetPushVelocity().setValue(0.f,0.f,0.f);
|
||
|
solverBody->internalGetTurnVelocity().setValue(0.f,0.f,0.f);
|
||
|
|
||
|
b3Assert(rb);
|
||
|
// solverBody->m_worldTransform = getWorldTransform(rb);
|
||
|
solverBody->internalSetInvMass(b3MakeVector3(rb->m_invMass,rb->m_invMass,rb->m_invMass));
|
||
|
solverBody->m_originalBodyIndex = bodyIndex;
|
||
|
solverBody->m_angularFactor = b3MakeVector3(1,1,1);
|
||
|
solverBody->m_linearFactor = b3MakeVector3(1,1,1);
|
||
|
solverBody->m_linearVelocity = getLinearVelocity(rb);
|
||
|
solverBody->m_angularVelocity = getAngularVelocity(rb);
|
||
|
}
|
||
|
|
||
|
|
||
|
void b3GpuPgsConstraintSolver::averageVelocities()
|
||
|
{
|
||
|
}
|
||
|
|
||
|
|
||
|
b3Scalar b3GpuPgsConstraintSolver::solveGroupCacheFriendlyIterations(b3OpenCLArray<b3GpuGenericConstraint>* gpuConstraints1,int numConstraints,const b3ContactSolverInfo& infoGlobal)
|
||
|
{
|
||
|
//only create the batches once.
|
||
|
//@todo: incrementally update batches when constraints are added/activated and/or removed/deactivated
|
||
|
B3_PROFILE("GpuSolveGroupCacheFriendlyIterations");
|
||
|
|
||
|
bool createBatches = m_gpuData->m_batchSizes.size()==0;
|
||
|
{
|
||
|
|
||
|
if (createBatches)
|
||
|
{
|
||
|
|
||
|
m_gpuData->m_batchSizes.resize(0);
|
||
|
|
||
|
{
|
||
|
m_gpuData->m_gpuBatchConstraints->copyToHost(batchConstraints);
|
||
|
|
||
|
B3_PROFILE("batch joints");
|
||
|
b3Assert(batchConstraints.size()==numConstraints);
|
||
|
int simdWidth =numConstraints+1;
|
||
|
int numBodies = m_tmpSolverBodyPool.size();
|
||
|
sortConstraintByBatch3( &batchConstraints[0], numConstraints, simdWidth , m_staticIdx, numBodies);
|
||
|
|
||
|
m_gpuData->m_gpuBatchConstraints->copyFromHost(batchConstraints);
|
||
|
|
||
|
}
|
||
|
} else
|
||
|
{
|
||
|
/*b3AlignedObjectArray<b3BatchConstraint> cpuCheckBatches;
|
||
|
m_gpuData->m_gpuBatchConstraints->copyToHost(cpuCheckBatches);
|
||
|
b3Assert(cpuCheckBatches.size()==batchConstraints.size());
|
||
|
printf(".\n");
|
||
|
*/
|
||
|
//>copyFromHost(batchConstraints);
|
||
|
}
|
||
|
int maxIterations = infoGlobal.m_numIterations;
|
||
|
|
||
|
bool useBatching = true;
|
||
|
|
||
|
if (useBatching )
|
||
|
{
|
||
|
|
||
|
if (!useGpuSolveJointConstraintRows)
|
||
|
{
|
||
|
B3_PROFILE("copy to host");
|
||
|
m_gpuData->m_gpuSolverBodies->copyToHost(m_tmpSolverBodyPool);
|
||
|
m_gpuData->m_gpuBatchConstraints->copyToHost(batchConstraints);
|
||
|
m_gpuData->m_gpuConstraintRows->copyToHost(m_tmpSolverNonContactConstraintPool);
|
||
|
m_gpuData->m_gpuConstraintInfo1->copyToHost(m_gpuData->m_cpuConstraintInfo1);
|
||
|
m_gpuData->m_gpuConstraintRowOffsets->copyToHost(m_gpuData->m_cpuConstraintRowOffsets);
|
||
|
gpuConstraints1->copyToHost(m_gpuData->m_cpuConstraints);
|
||
|
|
||
|
}
|
||
|
|
||
|
for ( int iteration = 0 ; iteration< maxIterations ; iteration++)
|
||
|
{
|
||
|
|
||
|
int batchOffset = 0;
|
||
|
int constraintOffset=0;
|
||
|
int numBatches = m_gpuData->m_batchSizes.size();
|
||
|
for (int bb=0;bb<numBatches;bb++)
|
||
|
{
|
||
|
int numConstraintsInBatch = m_gpuData->m_batchSizes[bb];
|
||
|
|
||
|
|
||
|
if (useGpuSolveJointConstraintRows)
|
||
|
{
|
||
|
B3_PROFILE("solveJointConstraintRowsKernels");
|
||
|
|
||
|
/*
|
||
|
__kernel void solveJointConstraintRows(__global b3GpuSolverBody* solverBodies,
|
||
|
__global b3BatchConstraint* batchConstraints,
|
||
|
__global b3SolverConstraint* rows,
|
||
|
__global unsigned int* numConstraintRowsInfo1,
|
||
|
__global unsigned int* rowOffsets,
|
||
|
__global b3GpuGenericConstraint* constraints,
|
||
|
int batchOffset,
|
||
|
int numConstraintsInBatch*/
|
||
|
|
||
|
|
||
|
b3LauncherCL launcher(m_gpuData->m_queue,m_gpuData->m_solveJointConstraintRowsKernels,"m_solveJointConstraintRowsKernels");
|
||
|
launcher.setBuffer(m_gpuData->m_gpuSolverBodies->getBufferCL());
|
||
|
launcher.setBuffer(m_gpuData->m_gpuBatchConstraints->getBufferCL());
|
||
|
launcher.setBuffer(m_gpuData->m_gpuConstraintRows->getBufferCL());
|
||
|
launcher.setBuffer(m_gpuData->m_gpuConstraintInfo1->getBufferCL());
|
||
|
launcher.setBuffer(m_gpuData->m_gpuConstraintRowOffsets->getBufferCL());
|
||
|
launcher.setBuffer(gpuConstraints1->getBufferCL());//to detect disabled constraints
|
||
|
launcher.setConst(batchOffset);
|
||
|
launcher.setConst(numConstraintsInBatch);
|
||
|
|
||
|
launcher.launch1D(numConstraintsInBatch);
|
||
|
|
||
|
|
||
|
} else//useGpu
|
||
|
{
|
||
|
|
||
|
|
||
|
|
||
|
for (int b=0;b<numConstraintsInBatch;b++)
|
||
|
{
|
||
|
const b3BatchConstraint& c = batchConstraints[batchOffset+b];
|
||
|
/*printf("-----------\n");
|
||
|
printf("bb=%d\n",bb);
|
||
|
printf("c.batchId = %d\n", c.m_batchId);
|
||
|
*/
|
||
|
b3Assert(c.m_batchId==bb);
|
||
|
b3GpuGenericConstraint* constraint = &m_gpuData->m_cpuConstraints[c.m_originalConstraintIndex];
|
||
|
if (constraint->m_flags&B3_CONSTRAINT_FLAG_ENABLED)
|
||
|
{
|
||
|
int numConstraintRows = m_gpuData->m_cpuConstraintInfo1[c.m_originalConstraintIndex];
|
||
|
int constraintOffset = m_gpuData->m_cpuConstraintRowOffsets[c.m_originalConstraintIndex];
|
||
|
|
||
|
for (int jj=0;jj<numConstraintRows;jj++)
|
||
|
{
|
||
|
//
|
||
|
b3GpuSolverConstraint& constraint = m_tmpSolverNonContactConstraintPool[constraintOffset+jj];
|
||
|
//resolveSingleConstraintRowGenericSIMD(m_tmpSolverBodyPool[constraint.m_solverBodyIdA],m_tmpSolverBodyPool[constraint.m_solverBodyIdB],constraint);
|
||
|
resolveSingleConstraintRowGeneric2(&m_tmpSolverBodyPool[constraint.m_solverBodyIdA],&m_tmpSolverBodyPool[constraint.m_solverBodyIdB],&constraint);
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
}//useGpu
|
||
|
batchOffset+=numConstraintsInBatch;
|
||
|
constraintOffset+=numConstraintsInBatch;
|
||
|
}
|
||
|
}//for (int iteration...
|
||
|
|
||
|
if (!useGpuSolveJointConstraintRows)
|
||
|
{
|
||
|
{
|
||
|
B3_PROFILE("copy from host");
|
||
|
m_gpuData->m_gpuSolverBodies->copyFromHost(m_tmpSolverBodyPool);
|
||
|
m_gpuData->m_gpuBatchConstraints->copyFromHost(batchConstraints);
|
||
|
m_gpuData->m_gpuConstraintRows->copyFromHost(m_tmpSolverNonContactConstraintPool);
|
||
|
}
|
||
|
|
||
|
//B3_PROFILE("copy to host");
|
||
|
//m_gpuData->m_gpuSolverBodies->copyToHost(m_tmpSolverBodyPool);
|
||
|
}
|
||
|
//int sz = sizeof(b3GpuSolverBody);
|
||
|
//printf("cpu sizeof(b3GpuSolverBody)=%d\n",sz);
|
||
|
|
||
|
|
||
|
|
||
|
|
||
|
|
||
|
} else
|
||
|
{
|
||
|
for ( int iteration = 0 ; iteration< maxIterations ; iteration++)
|
||
|
{
|
||
|
int numJoints = m_tmpSolverNonContactConstraintPool.size();
|
||
|
for (int j=0;j<numJoints;j++)
|
||
|
{
|
||
|
b3GpuSolverConstraint& constraint = m_tmpSolverNonContactConstraintPool[j];
|
||
|
resolveSingleConstraintRowGeneric2(&m_tmpSolverBodyPool[constraint.m_solverBodyIdA],&m_tmpSolverBodyPool[constraint.m_solverBodyIdB],&constraint);
|
||
|
}
|
||
|
|
||
|
if (!m_usePgs)
|
||
|
{
|
||
|
averageVelocities();
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
}
|
||
|
clFinish(m_gpuData->m_queue);
|
||
|
return 0.f;
|
||
|
}
|
||
|
|
||
|
|
||
|
|
||
|
|
||
|
static b3AlignedObjectArray<int> bodyUsed;
|
||
|
static b3AlignedObjectArray<int> curUsed;
|
||
|
|
||
|
|
||
|
|
||
|
inline int b3GpuPgsConstraintSolver::sortConstraintByBatch3( b3BatchConstraint* cs, int numConstraints, int simdWidth , int staticIdx, int numBodies)
|
||
|
{
|
||
|
//int sz = sizeof(b3BatchConstraint);
|
||
|
|
||
|
B3_PROFILE("sortConstraintByBatch3");
|
||
|
|
||
|
static int maxSwaps = 0;
|
||
|
int numSwaps = 0;
|
||
|
|
||
|
curUsed.resize(2*simdWidth);
|
||
|
|
||
|
static int maxNumConstraints = 0;
|
||
|
if (maxNumConstraints<numConstraints)
|
||
|
{
|
||
|
maxNumConstraints = numConstraints;
|
||
|
//printf("maxNumConstraints = %d\n",maxNumConstraints );
|
||
|
}
|
||
|
|
||
|
int numUsedArray = numBodies/32+1;
|
||
|
bodyUsed.resize(numUsedArray);
|
||
|
|
||
|
for (int q=0;q<numUsedArray;q++)
|
||
|
bodyUsed[q]=0;
|
||
|
|
||
|
|
||
|
int curBodyUsed = 0;
|
||
|
|
||
|
int numIter = 0;
|
||
|
|
||
|
|
||
|
#if defined(_DEBUG)
|
||
|
for(int i=0; i<numConstraints; i++)
|
||
|
cs[i].m_batchId = -1;
|
||
|
#endif
|
||
|
|
||
|
int numValidConstraints = 0;
|
||
|
// int unprocessedConstraintIndex = 0;
|
||
|
|
||
|
int batchIdx = 0;
|
||
|
|
||
|
|
||
|
{
|
||
|
B3_PROFILE("cpu batch innerloop");
|
||
|
|
||
|
while( numValidConstraints < numConstraints)
|
||
|
{
|
||
|
numIter++;
|
||
|
int nCurrentBatch = 0;
|
||
|
// clear flag
|
||
|
for(int i=0; i<curBodyUsed; i++)
|
||
|
bodyUsed[curUsed[i]/32] = 0;
|
||
|
|
||
|
curBodyUsed = 0;
|
||
|
|
||
|
for(int i=numValidConstraints; i<numConstraints; i++)
|
||
|
{
|
||
|
int idx = i;
|
||
|
b3Assert( idx < numConstraints );
|
||
|
// check if it can go
|
||
|
int bodyAS = cs[idx].m_bodyAPtrAndSignBit;
|
||
|
int bodyBS = cs[idx].m_bodyBPtrAndSignBit;
|
||
|
int bodyA = abs(bodyAS);
|
||
|
int bodyB = abs(bodyBS);
|
||
|
bool aIsStatic = (bodyAS<0) || bodyAS==staticIdx;
|
||
|
bool bIsStatic = (bodyBS<0) || bodyBS==staticIdx;
|
||
|
int aUnavailable = 0;
|
||
|
int bUnavailable = 0;
|
||
|
if (!aIsStatic)
|
||
|
{
|
||
|
aUnavailable = bodyUsed[ bodyA/32 ] & (1<<(bodyA&31));
|
||
|
}
|
||
|
if (!aUnavailable)
|
||
|
if (!bIsStatic)
|
||
|
{
|
||
|
bUnavailable = bodyUsed[ bodyB/32 ] & (1<<(bodyB&31));
|
||
|
}
|
||
|
|
||
|
if( aUnavailable==0 && bUnavailable==0 ) // ok
|
||
|
{
|
||
|
if (!aIsStatic)
|
||
|
{
|
||
|
bodyUsed[ bodyA/32 ] |= (1<<(bodyA&31));
|
||
|
curUsed[curBodyUsed++]=bodyA;
|
||
|
}
|
||
|
if (!bIsStatic)
|
||
|
{
|
||
|
bodyUsed[ bodyB/32 ] |= (1<<(bodyB&31));
|
||
|
curUsed[curBodyUsed++]=bodyB;
|
||
|
}
|
||
|
|
||
|
cs[idx].m_batchId = batchIdx;
|
||
|
|
||
|
if (i!=numValidConstraints)
|
||
|
{
|
||
|
b3Swap(cs[i],cs[numValidConstraints]);
|
||
|
numSwaps++;
|
||
|
}
|
||
|
|
||
|
numValidConstraints++;
|
||
|
{
|
||
|
nCurrentBatch++;
|
||
|
if( nCurrentBatch == simdWidth )
|
||
|
{
|
||
|
nCurrentBatch = 0;
|
||
|
for(int i=0; i<curBodyUsed; i++)
|
||
|
bodyUsed[curUsed[i]/32] = 0;
|
||
|
curBodyUsed = 0;
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
m_gpuData->m_batchSizes.push_back(nCurrentBatch);
|
||
|
batchIdx ++;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
#if defined(_DEBUG)
|
||
|
// debugPrintf( "nBatches: %d\n", batchIdx );
|
||
|
for(int i=0; i<numConstraints; i++)
|
||
|
{
|
||
|
b3Assert( cs[i].m_batchId != -1 );
|
||
|
}
|
||
|
#endif
|
||
|
|
||
|
if (maxSwaps<numSwaps)
|
||
|
{
|
||
|
maxSwaps = numSwaps;
|
||
|
//printf("maxSwaps = %d\n", maxSwaps);
|
||
|
}
|
||
|
|
||
|
return batchIdx;
|
||
|
}
|
||
|
|
||
|
|
||
|
/// b3PgsJacobiSolver Sequentially applies impulses
|
||
|
b3Scalar b3GpuPgsConstraintSolver::solveGroup(b3OpenCLArray<b3RigidBodyData>* gpuBodies, b3OpenCLArray<b3InertiaData>* gpuInertias,
|
||
|
int numBodies, b3OpenCLArray<b3GpuGenericConstraint>* gpuConstraints,int numConstraints, const b3ContactSolverInfo& infoGlobal)
|
||
|
{
|
||
|
|
||
|
B3_PROFILE("solveJoints");
|
||
|
//you need to provide at least some bodies
|
||
|
|
||
|
solveGroupCacheFriendlySetup( gpuBodies, gpuInertias,numBodies,gpuConstraints, numConstraints,infoGlobal);
|
||
|
|
||
|
solveGroupCacheFriendlyIterations(gpuConstraints, numConstraints,infoGlobal);
|
||
|
|
||
|
solveGroupCacheFriendlyFinish(gpuBodies, gpuInertias,numBodies, gpuConstraints, numConstraints, infoGlobal);
|
||
|
|
||
|
return 0.f;
|
||
|
}
|
||
|
|
||
|
void b3GpuPgsConstraintSolver::solveJoints(int numBodies, b3OpenCLArray<b3RigidBodyData>* gpuBodies, b3OpenCLArray<b3InertiaData>* gpuInertias,
|
||
|
int numConstraints, b3OpenCLArray<b3GpuGenericConstraint>* gpuConstraints)
|
||
|
{
|
||
|
b3ContactSolverInfo infoGlobal;
|
||
|
infoGlobal.m_splitImpulse = false;
|
||
|
infoGlobal.m_timeStep = 1.f/60.f;
|
||
|
infoGlobal.m_numIterations = 4;//4;
|
||
|
// infoGlobal.m_solverMode|=B3_SOLVER_USE_2_FRICTION_DIRECTIONS|B3_SOLVER_INTERLEAVE_CONTACT_AND_FRICTION_CONSTRAINTS|B3_SOLVER_DISABLE_VELOCITY_DEPENDENT_FRICTION_DIRECTION;
|
||
|
//infoGlobal.m_solverMode|=B3_SOLVER_USE_2_FRICTION_DIRECTIONS|B3_SOLVER_INTERLEAVE_CONTACT_AND_FRICTION_CONSTRAINTS;
|
||
|
infoGlobal.m_solverMode|=B3_SOLVER_USE_2_FRICTION_DIRECTIONS;
|
||
|
|
||
|
//if (infoGlobal.m_solverMode & B3_SOLVER_INTERLEAVE_CONTACT_AND_FRICTION_CONSTRAINTS)
|
||
|
//if ((infoGlobal.m_solverMode & B3_SOLVER_USE_2_FRICTION_DIRECTIONS) && (infoGlobal.m_solverMode & B3_SOLVER_DISABLE_VELOCITY_DEPENDENT_FRICTION_DIRECTION))
|
||
|
|
||
|
|
||
|
solveGroup(gpuBodies,gpuInertias,numBodies,gpuConstraints,numConstraints,infoGlobal);
|
||
|
|
||
|
}
|
||
|
|
||
|
//b3AlignedObjectArray<b3RigidBodyData> testBodies;
|
||
|
|
||
|
|
||
|
b3Scalar b3GpuPgsConstraintSolver::solveGroupCacheFriendlyFinish(b3OpenCLArray<b3RigidBodyData>* gpuBodies,b3OpenCLArray<b3InertiaData>* gpuInertias,int numBodies,b3OpenCLArray<b3GpuGenericConstraint>* gpuConstraints,int numConstraints,const b3ContactSolverInfo& infoGlobal)
|
||
|
{
|
||
|
B3_PROFILE("solveGroupCacheFriendlyFinish");
|
||
|
// int numPoolConstraints = m_tmpSolverContactConstraintPool.size();
|
||
|
// int i,j;
|
||
|
|
||
|
|
||
|
{
|
||
|
if (gpuBreakConstraints)
|
||
|
{
|
||
|
B3_PROFILE("breakViolatedConstraintsKernel");
|
||
|
b3LauncherCL launcher(m_gpuData->m_queue,m_gpuData->m_breakViolatedConstraintsKernel,"m_breakViolatedConstraintsKernel");
|
||
|
launcher.setBuffer(gpuConstraints->getBufferCL());
|
||
|
launcher.setBuffer(m_gpuData->m_gpuConstraintInfo1->getBufferCL());
|
||
|
launcher.setBuffer(m_gpuData->m_gpuConstraintRowOffsets->getBufferCL());
|
||
|
launcher.setBuffer(m_gpuData->m_gpuConstraintRows->getBufferCL());
|
||
|
launcher.setConst(numConstraints);
|
||
|
launcher.launch1D(numConstraints);
|
||
|
} else
|
||
|
{
|
||
|
gpuConstraints->copyToHost(m_gpuData->m_cpuConstraints);
|
||
|
m_gpuData->m_gpuBatchConstraints->copyToHost(m_gpuData->m_cpuBatchConstraints);
|
||
|
m_gpuData->m_gpuConstraintRows->copyToHost(m_gpuData->m_cpuConstraintRows);
|
||
|
gpuConstraints->copyToHost(m_gpuData->m_cpuConstraints);
|
||
|
m_gpuData->m_gpuConstraintInfo1->copyToHost(m_gpuData->m_cpuConstraintInfo1);
|
||
|
m_gpuData->m_gpuConstraintRowOffsets->copyToHost(m_gpuData->m_cpuConstraintRowOffsets);
|
||
|
|
||
|
for (int cid=0;cid<numConstraints;cid++)
|
||
|
{
|
||
|
int originalConstraintIndex = batchConstraints[cid].m_originalConstraintIndex;
|
||
|
int constraintRowOffset = m_gpuData->m_cpuConstraintRowOffsets[originalConstraintIndex];
|
||
|
int numRows = m_gpuData->m_cpuConstraintInfo1[originalConstraintIndex];
|
||
|
if (numRows)
|
||
|
{
|
||
|
|
||
|
// printf("cid=%d, breakingThreshold =%f\n",cid,breakingThreshold);
|
||
|
for (int i=0;i<numRows;i++)
|
||
|
{
|
||
|
int rowIndex =constraintRowOffset+i;
|
||
|
int orgConstraintIndex = m_gpuData->m_cpuConstraintRows[rowIndex].m_originalConstraintIndex;
|
||
|
float breakingThreshold = m_gpuData->m_cpuConstraints[orgConstraintIndex].m_breakingImpulseThreshold;
|
||
|
// printf("rows[%d].m_appliedImpulse=%f\n",rowIndex,rows[rowIndex].m_appliedImpulse);
|
||
|
if (b3Fabs(m_gpuData->m_cpuConstraintRows[rowIndex].m_appliedImpulse) >= breakingThreshold)
|
||
|
{
|
||
|
|
||
|
m_gpuData->m_cpuConstraints[orgConstraintIndex].m_flags =0;//&= ~B3_CONSTRAINT_FLAG_ENABLED;
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
|
||
|
gpuConstraints->copyFromHost(m_gpuData->m_cpuConstraints);
|
||
|
}
|
||
|
}
|
||
|
|
||
|
{
|
||
|
if (useGpuWriteBackVelocities)
|
||
|
{
|
||
|
B3_PROFILE("GPU write back velocities and transforms");
|
||
|
|
||
|
b3LauncherCL launcher(m_gpuData->m_queue,m_gpuData->m_writeBackVelocitiesKernel,"m_writeBackVelocitiesKernel");
|
||
|
launcher.setBuffer(gpuBodies->getBufferCL());
|
||
|
launcher.setBuffer(m_gpuData->m_gpuSolverBodies->getBufferCL());
|
||
|
launcher.setConst(numBodies);
|
||
|
launcher.launch1D(numBodies);
|
||
|
clFinish(m_gpuData->m_queue);
|
||
|
// m_gpuData->m_gpuSolverBodies->copyToHost(m_tmpSolverBodyPool);
|
||
|
// m_gpuData->m_gpuBodies->copyToHostPointer(bodies,numBodies);
|
||
|
//m_gpuData->m_gpuBodies->copyToHost(testBodies);
|
||
|
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
B3_PROFILE("CPU write back velocities and transforms");
|
||
|
|
||
|
m_gpuData->m_gpuSolverBodies->copyToHost(m_tmpSolverBodyPool);
|
||
|
gpuBodies->copyToHost(m_gpuData->m_cpuBodies);
|
||
|
for ( int i=0;i<m_tmpSolverBodyPool.size();i++)
|
||
|
{
|
||
|
int bodyIndex = m_tmpSolverBodyPool[i].m_originalBodyIndex;
|
||
|
//printf("bodyIndex=%d\n",bodyIndex);
|
||
|
b3Assert(i==bodyIndex);
|
||
|
|
||
|
b3RigidBodyData* body = &m_gpuData->m_cpuBodies[bodyIndex];
|
||
|
if (body->m_invMass)
|
||
|
{
|
||
|
if (infoGlobal.m_splitImpulse)
|
||
|
m_tmpSolverBodyPool[i].writebackVelocityAndTransform(infoGlobal.m_timeStep, infoGlobal.m_splitImpulseTurnErp);
|
||
|
else
|
||
|
m_tmpSolverBodyPool[i].writebackVelocity();
|
||
|
|
||
|
if (m_usePgs)
|
||
|
{
|
||
|
body->m_linVel = m_tmpSolverBodyPool[i].m_linearVelocity;
|
||
|
body->m_angVel = m_tmpSolverBodyPool[i].m_angularVelocity;
|
||
|
} else
|
||
|
{
|
||
|
b3Assert(0);
|
||
|
}
|
||
|
/*
|
||
|
if (infoGlobal.m_splitImpulse)
|
||
|
{
|
||
|
body->m_pos = m_tmpSolverBodyPool[i].m_worldTransform.getOrigin();
|
||
|
b3Quaternion orn;
|
||
|
orn = m_tmpSolverBodyPool[i].m_worldTransform.getRotation();
|
||
|
body->m_quat = orn;
|
||
|
}
|
||
|
*/
|
||
|
}
|
||
|
}//for
|
||
|
|
||
|
gpuBodies->copyFromHost(m_gpuData->m_cpuBodies);
|
||
|
|
||
|
}
|
||
|
}
|
||
|
|
||
|
clFinish(m_gpuData->m_queue);
|
||
|
|
||
|
m_tmpSolverContactConstraintPool.resizeNoInitialize(0);
|
||
|
m_tmpSolverNonContactConstraintPool.resizeNoInitialize(0);
|
||
|
m_tmpSolverContactFrictionConstraintPool.resizeNoInitialize(0);
|
||
|
m_tmpSolverContactRollingFrictionConstraintPool.resizeNoInitialize(0);
|
||
|
|
||
|
m_tmpSolverBodyPool.resizeNoInitialize(0);
|
||
|
return 0.f;
|
||
|
}
|