123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878 |
- /*
- 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
- #define B3_CONSTRAINT_FLAG_ENABLED 1
- #define B3_GPU_POINT2POINT_CONSTRAINT_TYPE 3
- #define B3_GPU_FIXED_CONSTRAINT_TYPE 4
- #define MOTIONCLAMP 100000 //unused, for debugging/safety in case constraint solver fails
- #define B3_INFINITY 1e30f
- #define mymake_float4 (float4)
- __inline float dot3F4(float4 a, float4 b)
- {
- float4 a1 = mymake_float4(a.xyz,0.f);
- float4 b1 = mymake_float4(b.xyz,0.f);
- return dot(a1, b1);
- }
- typedef float4 Quaternion;
- typedef struct
- {
- float4 m_row[3];
- }Matrix3x3;
- __inline
- float4 mtMul1(Matrix3x3 a, float4 b);
- __inline
- float4 mtMul3(float4 a, Matrix3x3 b);
- __inline
- float4 mtMul1(Matrix3x3 a, float4 b)
- {
- float4 ans;
- ans.x = dot3F4( a.m_row[0], b );
- ans.y = dot3F4( a.m_row[1], b );
- ans.z = dot3F4( a.m_row[2], b );
- ans.w = 0.f;
- return ans;
- }
- __inline
- float4 mtMul3(float4 a, Matrix3x3 b)
- {
- float4 colx = mymake_float4(b.m_row[0].x, b.m_row[1].x, b.m_row[2].x, 0);
- float4 coly = mymake_float4(b.m_row[0].y, b.m_row[1].y, b.m_row[2].y, 0);
- float4 colz = mymake_float4(b.m_row[0].z, b.m_row[1].z, b.m_row[2].z, 0);
- float4 ans;
- ans.x = dot3F4( a, colx );
- ans.y = dot3F4( a, coly );
- ans.z = dot3F4( a, colz );
- return ans;
- }
- typedef struct
- {
- Matrix3x3 m_invInertiaWorld;
- Matrix3x3 m_initInvInertia;
- } BodyInertia;
- typedef struct
- {
- Matrix3x3 m_basis;//orientation
- float4 m_origin;//transform
- }b3Transform;
- typedef struct
- {
- // b3Transform m_worldTransformUnused;
- float4 m_deltaLinearVelocity;
- float4 m_deltaAngularVelocity;
- float4 m_angularFactor;
- float4 m_linearFactor;
- float4 m_invMass;
- float4 m_pushVelocity;
- float4 m_turnVelocity;
- float4 m_linearVelocity;
- float4 m_angularVelocity;
- union
- {
- void* m_originalBody;
- int m_originalBodyIndex;
- };
- int padding[3];
- } b3GpuSolverBody;
- typedef struct
- {
- float4 m_pos;
- Quaternion m_quat;
- float4 m_linVel;
- float4 m_angVel;
- unsigned int m_shapeIdx;
- float m_invMass;
- float m_restituitionCoeff;
- float m_frictionCoeff;
- } b3RigidBodyCL;
- typedef struct
- {
- float4 m_relpos1CrossNormal;
- float4 m_contactNormal;
- float4 m_relpos2CrossNormal;
- //float4 m_contactNormal2;//usually m_contactNormal2 == -m_contactNormal
- float4 m_angularComponentA;
- float4 m_angularComponentB;
-
- float m_appliedPushImpulse;
- float m_appliedImpulse;
- int m_padding1;
- int m_padding2;
- float m_friction;
- float m_jacDiagABInv;
- float m_rhs;
- float m_cfm;
-
- float m_lowerLimit;
- float m_upperLimit;
- float m_rhsPenetration;
- int m_originalConstraint;
- int m_overrideNumSolverIterations;
- int m_frictionIndex;
- int m_solverBodyIdA;
- int m_solverBodyIdB;
- } b3SolverConstraint;
- typedef struct
- {
- int m_bodyAPtrAndSignBit;
- int m_bodyBPtrAndSignBit;
- int m_originalConstraintIndex;
- int m_batchId;
- } b3BatchConstraint;
- typedef struct
- {
- int m_constraintType;
- int m_rbA;
- int m_rbB;
- float m_breakingImpulseThreshold;
- float4 m_pivotInA;
- float4 m_pivotInB;
- Quaternion m_relTargetAB;
- int m_flags;
- int m_padding[3];
- } b3GpuGenericConstraint;
- /*b3Transform getWorldTransform(b3RigidBodyCL* rb)
- {
- b3Transform newTrans;
- newTrans.setOrigin(rb->m_pos);
- newTrans.setRotation(rb->m_quat);
- return newTrans;
- }*/
- __inline
- float4 cross3(float4 a, float4 b)
- {
- return cross(a,b);
- }
- __inline
- float4 fastNormalize4(float4 v)
- {
- v = mymake_float4(v.xyz,0.f);
- return fast_normalize(v);
- }
- __inline
- Quaternion qtMul(Quaternion a, Quaternion b);
- __inline
- Quaternion qtNormalize(Quaternion in);
- __inline
- float4 qtRotate(Quaternion q, float4 vec);
- __inline
- Quaternion qtInvert(Quaternion q);
- __inline
- Quaternion qtMul(Quaternion a, Quaternion b)
- {
- Quaternion ans;
- ans = cross3( a, b );
- ans += a.w*b+b.w*a;
- // ans.w = a.w*b.w - (a.x*b.x+a.y*b.y+a.z*b.z);
- ans.w = a.w*b.w - dot3F4(a, b);
- return ans;
- }
- __inline
- Quaternion qtNormalize(Quaternion in)
- {
- return fastNormalize4(in);
- // in /= length( in );
- // return in;
- }
- __inline
- float4 qtRotate(Quaternion q, float4 vec)
- {
- Quaternion qInv = qtInvert( q );
- float4 vcpy = vec;
- vcpy.w = 0.f;
- float4 out = qtMul(qtMul(q,vcpy),qInv);
- return out;
- }
- __inline
- Quaternion qtInvert(Quaternion q)
- {
- return (Quaternion)(-q.xyz, q.w);
- }
- __inline void internalApplyImpulse(__global b3GpuSolverBody* body, float4 linearComponent, float4 angularComponent,float impulseMagnitude)
- {
- body->m_deltaLinearVelocity += linearComponent*impulseMagnitude*body->m_linearFactor;
- body->m_deltaAngularVelocity += angularComponent*(impulseMagnitude*body->m_angularFactor);
- }
- void resolveSingleConstraintRowGeneric(__global b3GpuSolverBody* body1, __global b3GpuSolverBody* body2, __global b3SolverConstraint* c)
- {
- float deltaImpulse = c->m_rhs-c->m_appliedImpulse*c->m_cfm;
- float deltaVel1Dotn = dot3F4(c->m_contactNormal,body1->m_deltaLinearVelocity) + dot3F4(c->m_relpos1CrossNormal,body1->m_deltaAngularVelocity);
- float deltaVel2Dotn = -dot3F4(c->m_contactNormal,body2->m_deltaLinearVelocity) + dot3F4(c->m_relpos2CrossNormal,body2->m_deltaAngularVelocity);
- deltaImpulse -= deltaVel1Dotn*c->m_jacDiagABInv;
- deltaImpulse -= deltaVel2Dotn*c->m_jacDiagABInv;
- float sum = c->m_appliedImpulse + deltaImpulse;
- if (sum < c->m_lowerLimit)
- {
- deltaImpulse = c->m_lowerLimit-c->m_appliedImpulse;
- c->m_appliedImpulse = c->m_lowerLimit;
- }
- else if (sum > c->m_upperLimit)
- {
- deltaImpulse = c->m_upperLimit-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);
- }
- __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
- )
- {
- int b = get_global_id(0);
- if (b>=numConstraintsInBatch)
- return;
- __global b3BatchConstraint* c = &batchConstraints[b+batchOffset];
- int originalConstraintIndex = c->m_originalConstraintIndex;
- if (constraints[originalConstraintIndex].m_flags&B3_CONSTRAINT_FLAG_ENABLED)
- {
- int numConstraintRows = numConstraintRowsInfo1[originalConstraintIndex];
- int rowOffset = rowOffsets[originalConstraintIndex];
- for (int jj=0;jj<numConstraintRows;jj++)
- {
- __global b3SolverConstraint* constraint = &rows[rowOffset+jj];
- resolveSingleConstraintRowGeneric(&solverBodies[constraint->m_solverBodyIdA],&solverBodies[constraint->m_solverBodyIdB],constraint);
- }
- }
- };
- __kernel void initSolverBodies(__global b3GpuSolverBody* solverBodies,__global b3RigidBodyCL* bodiesCL, int numBodies)
- {
- int i = get_global_id(0);
- if (i>=numBodies)
- return;
- __global b3GpuSolverBody* solverBody = &solverBodies[i];
- __global b3RigidBodyCL* bodyCL = &bodiesCL[i];
- solverBody->m_deltaLinearVelocity = (float4)(0.f,0.f,0.f,0.f);
- solverBody->m_deltaAngularVelocity = (float4)(0.f,0.f,0.f,0.f);
- solverBody->m_pushVelocity = (float4)(0.f,0.f,0.f,0.f);
- solverBody->m_pushVelocity = (float4)(0.f,0.f,0.f,0.f);
- solverBody->m_invMass = (float4)(bodyCL->m_invMass,bodyCL->m_invMass,bodyCL->m_invMass,0.f);
- solverBody->m_originalBodyIndex = i;
- solverBody->m_angularFactor = (float4)(1,1,1,0);
- solverBody->m_linearFactor = (float4) (1,1,1,0);
- solverBody->m_linearVelocity = bodyCL->m_linVel;
- solverBody->m_angularVelocity = bodyCL->m_angVel;
- }
- __kernel void breakViolatedConstraintsKernel(__global b3GpuGenericConstraint* constraints, __global unsigned int* numConstraintRows, __global unsigned int* rowOffsets, __global b3SolverConstraint* rows, int numConstraints)
- {
- int cid = get_global_id(0);
- if (cid>=numConstraints)
- return;
- int numRows = numConstraintRows[cid];
- if (numRows)
- {
- for (int i=0;i<numRows;i++)
- {
- int rowIndex = rowOffsets[cid]+i;
- float breakingThreshold = constraints[cid].m_breakingImpulseThreshold;
- if (fabs(rows[rowIndex].m_appliedImpulse) >= breakingThreshold)
- {
- constraints[cid].m_flags =0;//&= ~B3_CONSTRAINT_FLAG_ENABLED;
- }
- }
- }
- }
- __kernel void getInfo1Kernel(__global unsigned int* infos, __global b3GpuGenericConstraint* constraints, int numConstraints)
- {
- int i = get_global_id(0);
- if (i>=numConstraints)
- return;
- __global b3GpuGenericConstraint* constraint = &constraints[i];
- switch (constraint->m_constraintType)
- {
- case B3_GPU_POINT2POINT_CONSTRAINT_TYPE:
- {
- infos[i] = 3;
- break;
- }
- case B3_GPU_FIXED_CONSTRAINT_TYPE:
- {
- infos[i] = 6;
- break;
- }
- default:
- {
- }
- }
- }
- __kernel void initBatchConstraintsKernel(__global unsigned int* numConstraintRows, __global unsigned int* rowOffsets,
- __global b3BatchConstraint* batchConstraints,
- __global b3GpuGenericConstraint* constraints,
- __global b3RigidBodyCL* bodies,
- int numConstraints)
- {
- int i = get_global_id(0);
- if (i>=numConstraints)
- return;
- int rbA = constraints[i].m_rbA;
- int rbB = constraints[i].m_rbB;
- batchConstraints[i].m_bodyAPtrAndSignBit = bodies[rbA].m_invMass != 0.f ? rbA : -rbA;
- batchConstraints[i].m_bodyBPtrAndSignBit = bodies[rbB].m_invMass != 0.f ? rbB : -rbB;
- batchConstraints[i].m_batchId = -1;
- batchConstraints[i].m_originalConstraintIndex = i;
- }
- typedef struct
- {
- // integrator parameters: frames per second (1/stepsize), default error
- // reduction parameter (0..1).
- float fps,erp;
- // for the first and second body, pointers to two (linear and angular)
- // n*3 jacobian sub matrices, stored by rows. these matrices will have
- // been initialized to 0 on entry. if the second body is zero then the
- // J2xx pointers may be 0.
- union
- {
- __global float4* m_J1linearAxisFloat4;
- __global float* m_J1linearAxis;
- };
- union
- {
- __global float4* m_J1angularAxisFloat4;
- __global float* m_J1angularAxis;
- };
- union
- {
- __global float4* m_J2linearAxisFloat4;
- __global float* m_J2linearAxis;
- };
- union
- {
- __global float4* m_J2angularAxisFloat4;
- __global float* m_J2angularAxis;
- };
- // elements to jump from one row to the next in J's
- int rowskip;
- // right hand sides of the equation J*v = c + cfm * lambda. cfm is the
- // "constraint force mixing" vector. c is set to zero on entry, cfm is
- // set to a constant value (typically very small or zero) value on entry.
- __global float* m_constraintError;
- __global float* cfm;
- // lo and hi limits for variables (set to -/+ infinity on entry).
- __global float* m_lowerLimit;
- __global float* m_upperLimit;
- // findex vector for variables. see the LCP solver interface for a
- // description of what this does. this is set to -1 on entry.
- // note that the returned indexes are relative to the first index of
- // the constraint.
- __global int *findex;
- // number of solver iterations
- int m_numIterations;
- //damping of the velocity
- float m_damping;
- } b3GpuConstraintInfo2;
- void getSkewSymmetricMatrix(float4 vecIn, __global float4* v0,__global float4* v1,__global float4* v2)
- {
- *v0 = (float4)(0. ,-vecIn.z ,vecIn.y,0.f);
- *v1 = (float4)(vecIn.z ,0. ,-vecIn.x,0.f);
- *v2 = (float4)(-vecIn.y ,vecIn.x ,0.f,0.f);
- }
- void getInfo2Point2Point(__global b3GpuGenericConstraint* constraint,b3GpuConstraintInfo2* info,__global b3RigidBodyCL* bodies)
- {
- float4 posA = bodies[constraint->m_rbA].m_pos;
- Quaternion rotA = bodies[constraint->m_rbA].m_quat;
- float4 posB = bodies[constraint->m_rbB].m_pos;
- Quaternion rotB = bodies[constraint->m_rbB].m_quat;
- // anchor points in global coordinates with respect to body PORs.
-
- // set jacobian
- info->m_J1linearAxis[0] = 1;
- info->m_J1linearAxis[info->rowskip+1] = 1;
- info->m_J1linearAxis[2*info->rowskip+2] = 1;
- float4 a1 = qtRotate(rotA,constraint->m_pivotInA);
- {
- __global float4* angular0 = (__global float4*)(info->m_J1angularAxis);
- __global float4* angular1 = (__global float4*)(info->m_J1angularAxis+info->rowskip);
- __global float4* angular2 = (__global float4*)(info->m_J1angularAxis+2*info->rowskip);
- float4 a1neg = -a1;
- getSkewSymmetricMatrix(a1neg,angular0,angular1,angular2);
- }
- if (info->m_J2linearAxis)
- {
- info->m_J2linearAxis[0] = -1;
- info->m_J2linearAxis[info->rowskip+1] = -1;
- info->m_J2linearAxis[2*info->rowskip+2] = -1;
- }
-
- float4 a2 = qtRotate(rotB,constraint->m_pivotInB);
-
- {
- // float4 a2n = -a2;
- __global float4* angular0 = (__global float4*)(info->m_J2angularAxis);
- __global float4* angular1 = (__global float4*)(info->m_J2angularAxis+info->rowskip);
- __global float4* angular2 = (__global float4*)(info->m_J2angularAxis+2*info->rowskip);
- getSkewSymmetricMatrix(a2,angular0,angular1,angular2);
- }
-
- // set right hand side
- // float currERP = (m_flags & B3_P2P_FLAGS_ERP) ? m_erp : info->erp;
- float currERP = info->erp;
- float k = info->fps * currERP;
- int j;
- float4 result = a2 + posB - a1 - posA;
- float* resultPtr = &result;
- for (j=0; j<3; j++)
- {
- info->m_constraintError[j*info->rowskip] = k * (resultPtr[j]);
- }
- }
- Quaternion nearest( Quaternion first, Quaternion qd)
- {
- Quaternion diff,sum;
- diff = first- qd;
- sum = first + qd;
-
- if( dot(diff,diff) < dot(sum,sum) )
- return qd;
- return (-qd);
- }
- float b3Acos(float x)
- {
- if (x<-1)
- x=-1;
- if (x>1)
- x=1;
- return acos(x);
- }
- float getAngle(Quaternion orn)
- {
- if (orn.w>=1.f)
- orn.w=1.f;
- float s = 2.f * b3Acos(orn.w);
- return s;
- }
- void calculateDiffAxisAngleQuaternion( Quaternion orn0,Quaternion orn1a,float4* axis,float* angle)
- {
- Quaternion orn1 = nearest(orn0,orn1a);
-
- Quaternion dorn = qtMul(orn1,qtInvert(orn0));
- *angle = getAngle(dorn);
- *axis = (float4)(dorn.x,dorn.y,dorn.z,0.f);
-
- //check for axis length
- float len = dot3F4(*axis,*axis);
- if (len < FLT_EPSILON*FLT_EPSILON)
- *axis = (float4)(1,0,0,0);
- else
- *axis /= sqrt(len);
- }
- void getInfo2FixedOrientation(__global b3GpuGenericConstraint* constraint,b3GpuConstraintInfo2* info,__global b3RigidBodyCL* bodies, int start_row)
- {
- Quaternion worldOrnA = bodies[constraint->m_rbA].m_quat;
- Quaternion worldOrnB = bodies[constraint->m_rbB].m_quat;
- int s = info->rowskip;
- int start_index = start_row * s;
- // 3 rows to make body rotations equal
- info->m_J1angularAxis[start_index] = 1;
- info->m_J1angularAxis[start_index + s + 1] = 1;
- info->m_J1angularAxis[start_index + s*2+2] = 1;
- if ( info->m_J2angularAxis)
- {
- info->m_J2angularAxis[start_index] = -1;
- info->m_J2angularAxis[start_index + s+1] = -1;
- info->m_J2angularAxis[start_index + s*2+2] = -1;
- }
-
- float currERP = info->erp;
- float k = info->fps * currERP;
- float4 diff;
- float angle;
- float4 qrelCur = qtMul(worldOrnA,qtInvert(worldOrnB));
-
- calculateDiffAxisAngleQuaternion(constraint->m_relTargetAB,qrelCur,&diff,&angle);
- diff*=-angle;
-
- float* resultPtr = &diff;
-
- for (int j=0; j<3; j++)
- {
- info->m_constraintError[(3+j)*info->rowskip] = k * resultPtr[j];
- }
-
- }
- __kernel void writeBackVelocitiesKernel(__global b3RigidBodyCL* bodies,__global b3GpuSolverBody* solverBodies,int numBodies)
- {
- int i = get_global_id(0);
- if (i>=numBodies)
- return;
- if (bodies[i].m_invMass)
- {
- // if (length(solverBodies[i].m_deltaLinearVelocity)<MOTIONCLAMP)
- {
- bodies[i].m_linVel += solverBodies[i].m_deltaLinearVelocity;
- }
- // if (length(solverBodies[i].m_deltaAngularVelocity)<MOTIONCLAMP)
- {
- bodies[i].m_angVel += solverBodies[i].m_deltaAngularVelocity;
- }
- }
- }
- __kernel void getInfo2Kernel(__global b3SolverConstraint* solverConstraintRows,
- __global unsigned int* infos,
- __global unsigned int* constraintRowOffsets,
- __global b3GpuGenericConstraint* constraints,
- __global b3BatchConstraint* batchConstraints,
- __global b3RigidBodyCL* bodies,
- __global BodyInertia* inertias,
- __global b3GpuSolverBody* solverBodies,
- float timeStep,
- float globalErp,
- float globalCfm,
- float globalDamping,
- int globalNumIterations,
- int numConstraints)
- {
- int i = get_global_id(0);
- if (i>=numConstraints)
- return;
-
- //for now, always initialize the batch info
- int info1 = infos[i];
-
- __global b3SolverConstraint* currentConstraintRow = &solverConstraintRows[constraintRowOffsets[i]];
- __global b3GpuGenericConstraint* constraint = &constraints[i];
- __global b3RigidBodyCL* rbA = &bodies[ constraint->m_rbA];
- __global b3RigidBodyCL* rbB = &bodies[ constraint->m_rbB];
- int solverBodyIdA = constraint->m_rbA;
- int solverBodyIdB = constraint->m_rbB;
- __global b3GpuSolverBody* bodyAPtr = &solverBodies[solverBodyIdA];
- __global b3GpuSolverBody* bodyBPtr = &solverBodies[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;
- }
- if (info1)
- {
- 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(b3SolverConstraint));
- currentConstraintRow[j].m_angularComponentA = (float4)(0,0,0,0);
- currentConstraintRow[j].m_angularComponentB = (float4)(0,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 = (float4)(0,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_originalConstraint = i;
- currentConstraintRow[j].m_overrideNumSolverIterations = 0;
- currentConstraintRow[j].m_relpos1CrossNormal = (float4)(0,0,0,0);
- currentConstraintRow[j].m_relpos2CrossNormal = (float4)(0,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->m_deltaLinearVelocity = (float4)(0,0,0,0);
- bodyAPtr->m_deltaAngularVelocity = (float4)(0,0,0,0);
- bodyAPtr->m_pushVelocity = (float4)(0,0,0,0);
- bodyAPtr->m_turnVelocity = (float4)(0,0,0,0);
- bodyBPtr->m_deltaLinearVelocity = (float4)(0,0,0,0);
- bodyBPtr->m_deltaAngularVelocity = (float4)(0,0,0,0);
- bodyBPtr->m_pushVelocity = (float4)(0,0,0,0);
- bodyBPtr->m_turnVelocity = (float4)(0,0,0,0);
- int rowskip = sizeof(b3SolverConstraint)/sizeof(float);//check this
-
- b3GpuConstraintInfo2 info2;
- info2.fps = 1.f/timeStep;
- info2.erp = globalErp;
- info2.m_J1linearAxisFloat4 = ¤tConstraintRow->m_contactNormal;
- info2.m_J1angularAxisFloat4 = ¤tConstraintRow->m_relpos1CrossNormal;
- info2.m_J2linearAxisFloat4 = 0;
- info2.m_J2angularAxisFloat4 = ¤tConstraintRow->m_relpos2CrossNormal;
- info2.rowskip = sizeof(b3SolverConstraint)/sizeof(float);//check this
- ///the size of b3SolverConstraint needs be a multiple of float
- // b3Assert(info2.rowskip*sizeof(float)== sizeof(b3SolverConstraint));
- info2.m_constraintError = ¤tConstraintRow->m_rhs;
- currentConstraintRow->m_cfm = globalCfm;
- info2.m_damping = globalDamping;
- info2.cfm = ¤tConstraintRow->m_cfm;
- info2.m_lowerLimit = ¤tConstraintRow->m_lowerLimit;
- info2.m_upperLimit = ¤tConstraintRow->m_upperLimit;
- info2.m_numIterations = globalNumIterations;
- switch (constraint->m_constraintType)
- {
- case B3_GPU_POINT2POINT_CONSTRAINT_TYPE:
- {
- getInfo2Point2Point(constraint,&info2,bodies);
- break;
- }
- case B3_GPU_FIXED_CONSTRAINT_TYPE:
- {
- getInfo2Point2Point(constraint,&info2,bodies);
- getInfo2FixedOrientation(constraint,&info2,bodies,3);
- break;
- }
- default:
- {
- }
- }
- ///finalize the constraint setup
- for ( j=0;j<info1;j++)
- {
- __global b3SolverConstraint* solverConstraint = ¤tConstraintRow[j];
- if (solverConstraint->m_upperLimit>=constraint->m_breakingImpulseThreshold)
- {
- solverConstraint->m_upperLimit = constraint->m_breakingImpulseThreshold;
- }
- if (solverConstraint->m_lowerLimit<=-constraint->m_breakingImpulseThreshold)
- {
- solverConstraint->m_lowerLimit = -constraint->m_breakingImpulseThreshold;
- }
- // solverConstraint->m_originalContactPoint = constraint;
-
- Matrix3x3 invInertiaWorldA= inertias[constraint->m_rbA].m_invInertiaWorld;
- {
- //float4 angularFactorA(1,1,1);
- float4 ftorqueAxis1 = solverConstraint->m_relpos1CrossNormal;
- solverConstraint->m_angularComponentA = mtMul1(invInertiaWorldA,ftorqueAxis1);//*angularFactorA;
- }
-
- Matrix3x3 invInertiaWorldB= inertias[constraint->m_rbB].m_invInertiaWorld;
- {
- float4 ftorqueAxis2 = solverConstraint->m_relpos2CrossNormal;
- solverConstraint->m_angularComponentB = mtMul1(invInertiaWorldB,ftorqueAxis2);//*constraint->m_rbB.getAngularFactor();
- }
- {
- //it is ok to use solverConstraint->m_contactNormal instead of -solverConstraint->m_contactNormal
- //because it gets multiplied iMJlB
- float4 iMJlA = solverConstraint->m_contactNormal*rbA->m_invMass;
- float4 iMJaA = mtMul3(solverConstraint->m_relpos1CrossNormal,invInertiaWorldA);
- float4 iMJlB = solverConstraint->m_contactNormal*rbB->m_invMass;//sign of normal?
- float4 iMJaB = mtMul3(solverConstraint->m_relpos2CrossNormal,invInertiaWorldB);
- float sum = dot3F4(iMJlA,solverConstraint->m_contactNormal);
- sum += dot3F4(iMJaA,solverConstraint->m_relpos1CrossNormal);
- sum += dot3F4(iMJlB,solverConstraint->m_contactNormal);
- sum += dot3F4(iMJaB,solverConstraint->m_relpos2CrossNormal);
- float fsum = fabs(sum);
- if (fsum>FLT_EPSILON)
- {
- solverConstraint->m_jacDiagABInv = 1.f/sum;
- } else
- {
- solverConstraint->m_jacDiagABInv = 0.f;
- }
- }
- ///fix rhs
- ///todo: add force/torque accelerators
- {
- float rel_vel;
- float vel1Dotn = dot3F4(solverConstraint->m_contactNormal,rbA->m_linVel) + dot3F4(solverConstraint->m_relpos1CrossNormal,rbA->m_angVel);
- float vel2Dotn = -dot3F4(solverConstraint->m_contactNormal,rbB->m_linVel) + dot3F4(solverConstraint->m_relpos2CrossNormal,rbB->m_angVel);
- rel_vel = vel1Dotn+vel2Dotn;
- float restitution = 0.f;
- float positionalError = solverConstraint->m_rhs;//already filled in by getConstraintInfo2
- float velocityError = restitution - rel_vel * info2.m_damping;
- float penetrationImpulse = positionalError*solverConstraint->m_jacDiagABInv;
- float velocityImpulse = velocityError *solverConstraint->m_jacDiagABInv;
- solverConstraint->m_rhs = penetrationImpulse+velocityImpulse;
- solverConstraint->m_appliedImpulse = 0.f;
- }
- }
- }
- }
|