tinyphysicsengine.h 29 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948
  1. #ifndef TINYPHYSICSENGINE_H
  2. #define TINYPHYSICSENGINE_H
  3. /**
  4. author: Miloslav Ciz
  5. license: CC0 1.0 (public domain)
  6. found at https://creativecommons.org/publicdomain/zero/1.0/
  7. + additional waiver of all IP
  8. version: 0.1d
  9. This is a suckless library for simple 3D (and 2D) physics simulation. The
  10. physics is based on the Newtonian model but is further simplified,
  11. particularly in the area of rotation: there is no moment of inertia for
  12. objects, i.e. every object rotates as if it was a ball, and the object can be
  13. rotating around at most one axis at a time, i.e. it is not possible to
  14. simulate e.g. the Dzhanibekov effect.
  15. CONVENTIONS:
  16. - Compatibility and simple usage with small3dlib is intended, so most
  17. convention and data types copy those of small3dlib (which takes a lot of
  18. conventions of OpenGL).
  19. - No floating point is used, we instead use integers (effectively a fixed
  20. point). TPE_FRACTIONS_PER_UNIT is an equivalent to 1.0 in floating point and
  21. all numbers are normalized by this constant.
  22. - Units: for any measure only an abstract mathematical unit is used. This unit
  23. always has TPE_FRACTIONS_PER_UNIT parts. You can assign any correcpondence
  24. with real life units to these units. E.g. 1 spatial unit (which you can see
  25. as e.g. 1 meter) is equal to TPE_FRACTIONS_PER_UNIT. Same with temporatl
  26. (e.g. 1 second) and mass (e.g. 1 kilogram) units, and also any derived
  27. units, e.g. a unit of velocity (e.g. 1 m/s) is also equal to 1
  28. TPE_FRACTIONS_PER_UNIT. A full angle is also split into
  29. TPE_FRACTIONS_PER_UNIT parts (instead of 2 * PI or degrees).
  30. - Quaternions are represented as vec4 where x ~ i, y ~ j, z ~ k, w ~ real.
  31. */
  32. #include <stdint.h>
  33. typedef int32_t TPE_Unit;
  34. /** How many fractions a unit is split into. This is NOT SUPPOSED TO BE
  35. REDEFINED, so rather don't do it (otherwise things may overflow etc.). */
  36. #define TPE_FRACTIONS_PER_UNIT 512
  37. #define TPE_INFINITY 2147483647
  38. #define TPE_PI 1608 ///< pi in TPE_Units
  39. #define TPE_SHAPE_POINT 0 ///< single point in space
  40. #define TPE_SHAPE_SPHERE 1 ///< sphere, params.: radius
  41. #define TPE_SHAPE_CUBOID 2 ///< cuboid, params.: width, height, depth
  42. #define TPE_SHAPE_PLANE 3 ///< plane, params.: width, depth
  43. #define TPE_SHAPE_CYLINDER 4 ///< cylinder, params.: radius, height
  44. #define TPE_SHAPE_TRIMESH 5 /**< triangle mesh, params.:
  45. vertex count,
  46. triangle count
  47. vertices (int32_t pointer),
  48. indices (uint16_t pointer) */
  49. #define TPE_MAX_SHAPE_PARAMS 3
  50. #define TPE_MAX_SHAPE_PARAMPOINTERS 2
  51. #define TPE_BODY_FLAG_DISABLED 0x00 ///< won't take part in simul. at all
  52. #define TPE_BODY_FLAG_NONCOLLIDING 0x01 ///< simulated but won't collide
  53. TPE_Unit TPE_wrap(TPE_Unit value, TPE_Unit mod);
  54. TPE_Unit TPE_clamp(TPE_Unit v, TPE_Unit v1, TPE_Unit v2);
  55. static inline TPE_Unit TPE_nonZero(TPE_Unit x);
  56. /** Returns an integer square root of given value. */
  57. TPE_Unit TPE_sqrt(TPE_Unit value);
  58. /** Returns a sine of given arguments, both in TPE_Units (see the library
  59. conventions). */
  60. TPE_Unit TPE_sin(TPE_Unit x);
  61. TPE_Unit TPE_cos(TPE_Unit x);
  62. TPE_Unit TPE_asin(TPE_Unit x);
  63. TPE_Unit TPE_acos(TPE_Unit x);
  64. typedef struct
  65. {
  66. TPE_Unit x;
  67. TPE_Unit y;
  68. TPE_Unit z;
  69. TPE_Unit w;
  70. } TPE_Vec4;
  71. /** Initializes vec4 to a zero vector. */
  72. void TPE_initVec4(TPE_Vec4 *v);
  73. void TPE_vec4Set(TPE_Vec4 *v, TPE_Unit x, TPE_Unit y, TPE_Unit z, TPE_Unit w);
  74. void TPE_vec3Add(TPE_Vec4 a, TPE_Vec4 b, TPE_Vec4 *result);
  75. void TPE_vec4Add(TPE_Vec4 a, TPE_Vec4 b, TPE_Vec4 *result);
  76. void TPE_vec3Substract(TPE_Vec4 a, TPE_Vec4 b, TPE_Vec4 *result);
  77. void TPE_vec4Substract(TPE_Vec4 a, TPE_Vec4 b, TPE_Vec4 *result);
  78. void TPE_vec3Multiply(TPE_Vec4 v, TPE_Unit f, TPE_Vec4 *result);
  79. void TPE_vec4Multiply(TPE_Vec4 v, TPE_Unit f, TPE_Vec4 *result);
  80. TPE_Unit TPE_vec3Len(TPE_Vec4 v);
  81. TPE_Unit TPE_vec4Len(TPE_Vec4 v);
  82. TPE_Unit TPE_vec3DotProduct(TPE_Vec4 v1, TPE_Vec4 v2);
  83. void TPE_vec3CrossProduct(TPE_Vec4 a, TPE_Vec4 b, TPE_Vec4 *result);
  84. void TPE_vec3Normalize(TPE_Vec4 *v);
  85. void TPE_vec4Normalize(TPE_Vec4 *v);
  86. void TPE_vec3Project(TPE_Vec4 v, TPE_Vec4 base, TPE_Vec4 *result);
  87. /** Converts a linear velocity of an orbiting point to the angular velocity
  88. (angle units per time units). This depends on the distance of the point from
  89. the center of rotation. */
  90. TPE_Unit TPE_linearVelocityToAngular(TPE_Unit velocity, TPE_Unit distance);
  91. /** Holds a rotation state around a single axis, in a way that prevents rounding
  92. errors from distorting the rotation over time. In theory rotation of a body
  93. could be represented as
  94. [current orientation, axis of rotation,angular velocity]
  95. However applying the rotation and normalizing the orientation quaternion each
  96. simulation step leads to error cumulation and the rotation gets aligned with
  97. one principal axis after some time. Because of this we rather represent the
  98. rotation state as
  99. [original orientation, axis of rotation, angular velocity, current angle]
  100. From this we can at each simulation step compute the current orientation by
  101. applying rotation by current angle to the original rotation without error
  102. cumulation. */
  103. typedef struct
  104. {
  105. TPE_Vec4 originalOrientation; /**< quaternion holding the original
  106. orientation of the body at the time when it
  107. has taken on this rotational state */
  108. TPE_Vec4 axisVelocity; /**< axis of rotation (x,y,z) and a
  109. non-negative angular velocity around this
  110. axis (w), determined ny the right hand
  111. rule */
  112. TPE_Unit currentAngle; /**< angle the body has already rotated along
  113. the rotation axis (from the original
  114. orientation) */
  115. } TPE_RotationState;
  116. #define TPE_BODY_SHAPE_SPHERE 0
  117. #define TPE_BODY_SHAPE_CUBOID 1
  118. #define TPE_BODY_SHAPE_PLANE 2
  119. #define TPE_BODY_SHAPE_CYLINDER 3
  120. #define TPE_BODY_SHAPE_MESH 4
  121. #define TPE_BODY_SHAPE_MULTIBODY 5
  122. typedef struct
  123. {
  124. TPE_Unit radius;
  125. } TPE_ShapeSphereParams;
  126. typedef struct
  127. {
  128. TPE_Unit width;
  129. TPE_Unit height;
  130. TPE_Unit depth;
  131. } TPE_ShapeCuboidParams;
  132. typedef struct
  133. {
  134. TPE_Unit *vertices;
  135. uint16_t vertexCount;
  136. uint16_t *triangles;
  137. uint16_t triangleCount;
  138. } TPE_ShapeMeshParams;
  139. typedef struct
  140. {
  141. uint8_t shape;
  142. union shapeParams
  143. {
  144. TPE_ShapeSphereParams sphere;
  145. TPE_ShapeCuboidParams cuboid;
  146. TPE_ShapeMeshParams mesh;
  147. };
  148. TPE_Unit shapeParams[TPE_MAX_SHAPE_PARAMS]; ///< parameters of the body type
  149. void *shapeParamPointers[TPE_MAX_SHAPE_PARAMPOINTERS]; ///< pointer parameters
  150. uint8_t flags;
  151. TPE_Unit mass; /**< body mass, setting this to TPE_INFINITY will
  152. make the object static (not moving at all)
  153. which may help performance */
  154. TPE_Vec4 position; ///< position of the body's center of mass
  155. TPE_Vec4 velocity; ///< linear velocity vector
  156. TPE_RotationState rotation; /**< holds the state related to rotation, i.e.
  157. the rotation axis, angular momentum and data
  158. from which current orientation can be
  159. inferred */
  160. } TPE_Body;
  161. /** Initializes a physical body, this should be called on all TPE_Bodys that
  162. are created.*/
  163. void TPE_bodyInit(TPE_Body *body);
  164. /** Computes a 4x4 transform matrix of given body. The matrix has the same
  165. format as S3L_Mat4 from small3dlib. */
  166. void TPE_bodyGetTransformMatrix(const TPE_Body *body, TPE_Unit matrix[4][4]);
  167. void TPE_bodyGetOrientation(const TPE_Body *body, TPE_Vec4 *quaternion);
  168. void TPE_bodyStep(TPE_Body *body);
  169. void TPE_bodySetRotation(TPE_Body *body, TPE_Vec4 axis, TPE_Unit velocity);
  170. void TPE_bodyAddRotation(TPE_Body *body, TPE_Vec4 axis, TPE_Unit velocity);
  171. /** Applies a velocity change to a body at a specific point (relative to the
  172. body center), which will change its linear and/or angular velocity. This is
  173. similar to an impulse but doesn't take mass into account, only velocity. */
  174. void TPE_bodyApplyVelocity(TPE_Body *body, TPE_Vec4 point, TPE_Vec4 velocity);
  175. #define TPE_PRINTF_VEC4(v) printf("[%d %d %d %d]\n",(v).x,(v).y,(v).z,(v).w);
  176. typedef struct
  177. {
  178. uint16_t bodyCount;
  179. TPE_Body *bodies;
  180. } TPE_PhysicsWorld;
  181. /** Multiplies two quaternions which can be seen as chaining two rotations
  182. represented by them. This is not commutative (a*b != b*a)! Rotations a is
  183. performed firth, then rotation b is performed. */
  184. void TPE_quaternionMultiply(TPE_Vec4 a, TPE_Vec4 b, TPE_Vec4 *result);
  185. /** Initializes quaternion to the a rotation identity (i.e. NOT zero
  186. quaternion). */
  187. void TPE_quaternionInit(TPE_Vec4 *quaternion);
  188. /** Converts a rotation given as an axis and angle around this axis (by right
  189. hand rule) to a rotation quaternion. */
  190. void TPE_rotationToQuaternion(TPE_Vec4 axis, TPE_Unit angle,
  191. TPE_Vec4 *quaternion);
  192. void TPE_quaternionToRotation(TPE_Vec4 quaternion, TPE_Vec4 *axis,
  193. TPE_Unit *angle);
  194. /** Converts a rotation quaternion to a 4x4 rotation matrix. The matrix is
  195. indexed as [column][row] and is in the same format as S3L_Mat4 from
  196. small3dlib. */
  197. void TPE_quaternionToRotationMatrix(TPE_Vec4 quaternion, TPE_Unit matrix[4][4]);
  198. void TPE_getVelocitiesAfterCollision(
  199. TPE_Unit *v1,
  200. TPE_Unit *v2,
  201. TPE_Unit m1,
  202. TPE_Unit m2,
  203. TPE_Unit elasticity
  204. );
  205. //------------------------------------------------------------------------------
  206. void TPE_initVec4(TPE_Vec4 *v)
  207. {
  208. v->x = 0;
  209. v->y = 0;
  210. v->z = 0;
  211. v->w = 0;
  212. }
  213. void TPE_vec4Set(TPE_Vec4 *v, TPE_Unit x, TPE_Unit y, TPE_Unit z, TPE_Unit w)
  214. {
  215. v->x = x;
  216. v->y = y;
  217. v->z = z;
  218. v->w = w;
  219. }
  220. TPE_Unit TPE_wrap(TPE_Unit value, TPE_Unit mod)
  221. {
  222. return value >= 0 ? (value % mod) : (mod + (value % mod) - 1);
  223. }
  224. TPE_Unit TPE_clamp(TPE_Unit v, TPE_Unit v1, TPE_Unit v2)
  225. {
  226. return v >= v1 ? (v <= v2 ? v : v2) : v1;
  227. }
  228. TPE_Unit TPE_nonZero(TPE_Unit x)
  229. {
  230. return x + (x == 0);
  231. }
  232. #define TPE_SIN_TABLE_LENGTH 128
  233. static const TPE_Unit TPE_sinTable[TPE_SIN_TABLE_LENGTH] =
  234. {
  235. /* 511 was chosen here as a highest number that doesn't overflow during
  236. compilation for TPE_FRACTIONS_PER_UNIT == 1024 */
  237. (0*TPE_FRACTIONS_PER_UNIT)/511, (6*TPE_FRACTIONS_PER_UNIT)/511,
  238. (12*TPE_FRACTIONS_PER_UNIT)/511, (18*TPE_FRACTIONS_PER_UNIT)/511,
  239. (25*TPE_FRACTIONS_PER_UNIT)/511, (31*TPE_FRACTIONS_PER_UNIT)/511,
  240. (37*TPE_FRACTIONS_PER_UNIT)/511, (43*TPE_FRACTIONS_PER_UNIT)/511,
  241. (50*TPE_FRACTIONS_PER_UNIT)/511, (56*TPE_FRACTIONS_PER_UNIT)/511,
  242. (62*TPE_FRACTIONS_PER_UNIT)/511, (68*TPE_FRACTIONS_PER_UNIT)/511,
  243. (74*TPE_FRACTIONS_PER_UNIT)/511, (81*TPE_FRACTIONS_PER_UNIT)/511,
  244. (87*TPE_FRACTIONS_PER_UNIT)/511, (93*TPE_FRACTIONS_PER_UNIT)/511,
  245. (99*TPE_FRACTIONS_PER_UNIT)/511, (105*TPE_FRACTIONS_PER_UNIT)/511,
  246. (111*TPE_FRACTIONS_PER_UNIT)/511, (118*TPE_FRACTIONS_PER_UNIT)/511,
  247. (124*TPE_FRACTIONS_PER_UNIT)/511, (130*TPE_FRACTIONS_PER_UNIT)/511,
  248. (136*TPE_FRACTIONS_PER_UNIT)/511, (142*TPE_FRACTIONS_PER_UNIT)/511,
  249. (148*TPE_FRACTIONS_PER_UNIT)/511, (154*TPE_FRACTIONS_PER_UNIT)/511,
  250. (160*TPE_FRACTIONS_PER_UNIT)/511, (166*TPE_FRACTIONS_PER_UNIT)/511,
  251. (172*TPE_FRACTIONS_PER_UNIT)/511, (178*TPE_FRACTIONS_PER_UNIT)/511,
  252. (183*TPE_FRACTIONS_PER_UNIT)/511, (189*TPE_FRACTIONS_PER_UNIT)/511,
  253. (195*TPE_FRACTIONS_PER_UNIT)/511, (201*TPE_FRACTIONS_PER_UNIT)/511,
  254. (207*TPE_FRACTIONS_PER_UNIT)/511, (212*TPE_FRACTIONS_PER_UNIT)/511,
  255. (218*TPE_FRACTIONS_PER_UNIT)/511, (224*TPE_FRACTIONS_PER_UNIT)/511,
  256. (229*TPE_FRACTIONS_PER_UNIT)/511, (235*TPE_FRACTIONS_PER_UNIT)/511,
  257. (240*TPE_FRACTIONS_PER_UNIT)/511, (246*TPE_FRACTIONS_PER_UNIT)/511,
  258. (251*TPE_FRACTIONS_PER_UNIT)/511, (257*TPE_FRACTIONS_PER_UNIT)/511,
  259. (262*TPE_FRACTIONS_PER_UNIT)/511, (268*TPE_FRACTIONS_PER_UNIT)/511,
  260. (273*TPE_FRACTIONS_PER_UNIT)/511, (278*TPE_FRACTIONS_PER_UNIT)/511,
  261. (283*TPE_FRACTIONS_PER_UNIT)/511, (289*TPE_FRACTIONS_PER_UNIT)/511,
  262. (294*TPE_FRACTIONS_PER_UNIT)/511, (299*TPE_FRACTIONS_PER_UNIT)/511,
  263. (304*TPE_FRACTIONS_PER_UNIT)/511, (309*TPE_FRACTIONS_PER_UNIT)/511,
  264. (314*TPE_FRACTIONS_PER_UNIT)/511, (319*TPE_FRACTIONS_PER_UNIT)/511,
  265. (324*TPE_FRACTIONS_PER_UNIT)/511, (328*TPE_FRACTIONS_PER_UNIT)/511,
  266. (333*TPE_FRACTIONS_PER_UNIT)/511, (338*TPE_FRACTIONS_PER_UNIT)/511,
  267. (343*TPE_FRACTIONS_PER_UNIT)/511, (347*TPE_FRACTIONS_PER_UNIT)/511,
  268. (352*TPE_FRACTIONS_PER_UNIT)/511, (356*TPE_FRACTIONS_PER_UNIT)/511,
  269. (361*TPE_FRACTIONS_PER_UNIT)/511, (365*TPE_FRACTIONS_PER_UNIT)/511,
  270. (370*TPE_FRACTIONS_PER_UNIT)/511, (374*TPE_FRACTIONS_PER_UNIT)/511,
  271. (378*TPE_FRACTIONS_PER_UNIT)/511, (382*TPE_FRACTIONS_PER_UNIT)/511,
  272. (386*TPE_FRACTIONS_PER_UNIT)/511, (391*TPE_FRACTIONS_PER_UNIT)/511,
  273. (395*TPE_FRACTIONS_PER_UNIT)/511, (398*TPE_FRACTIONS_PER_UNIT)/511,
  274. (402*TPE_FRACTIONS_PER_UNIT)/511, (406*TPE_FRACTIONS_PER_UNIT)/511,
  275. (410*TPE_FRACTIONS_PER_UNIT)/511, (414*TPE_FRACTIONS_PER_UNIT)/511,
  276. (417*TPE_FRACTIONS_PER_UNIT)/511, (421*TPE_FRACTIONS_PER_UNIT)/511,
  277. (424*TPE_FRACTIONS_PER_UNIT)/511, (428*TPE_FRACTIONS_PER_UNIT)/511,
  278. (431*TPE_FRACTIONS_PER_UNIT)/511, (435*TPE_FRACTIONS_PER_UNIT)/511,
  279. (438*TPE_FRACTIONS_PER_UNIT)/511, (441*TPE_FRACTIONS_PER_UNIT)/511,
  280. (444*TPE_FRACTIONS_PER_UNIT)/511, (447*TPE_FRACTIONS_PER_UNIT)/511,
  281. (450*TPE_FRACTIONS_PER_UNIT)/511, (453*TPE_FRACTIONS_PER_UNIT)/511,
  282. (456*TPE_FRACTIONS_PER_UNIT)/511, (459*TPE_FRACTIONS_PER_UNIT)/511,
  283. (461*TPE_FRACTIONS_PER_UNIT)/511, (464*TPE_FRACTIONS_PER_UNIT)/511,
  284. (467*TPE_FRACTIONS_PER_UNIT)/511, (469*TPE_FRACTIONS_PER_UNIT)/511,
  285. (472*TPE_FRACTIONS_PER_UNIT)/511, (474*TPE_FRACTIONS_PER_UNIT)/511,
  286. (476*TPE_FRACTIONS_PER_UNIT)/511, (478*TPE_FRACTIONS_PER_UNIT)/511,
  287. (481*TPE_FRACTIONS_PER_UNIT)/511, (483*TPE_FRACTIONS_PER_UNIT)/511,
  288. (485*TPE_FRACTIONS_PER_UNIT)/511, (487*TPE_FRACTIONS_PER_UNIT)/511,
  289. (488*TPE_FRACTIONS_PER_UNIT)/511, (490*TPE_FRACTIONS_PER_UNIT)/511,
  290. (492*TPE_FRACTIONS_PER_UNIT)/511, (494*TPE_FRACTIONS_PER_UNIT)/511,
  291. (495*TPE_FRACTIONS_PER_UNIT)/511, (497*TPE_FRACTIONS_PER_UNIT)/511,
  292. (498*TPE_FRACTIONS_PER_UNIT)/511, (499*TPE_FRACTIONS_PER_UNIT)/511,
  293. (501*TPE_FRACTIONS_PER_UNIT)/511, (502*TPE_FRACTIONS_PER_UNIT)/511,
  294. (503*TPE_FRACTIONS_PER_UNIT)/511, (504*TPE_FRACTIONS_PER_UNIT)/511,
  295. (505*TPE_FRACTIONS_PER_UNIT)/511, (506*TPE_FRACTIONS_PER_UNIT)/511,
  296. (507*TPE_FRACTIONS_PER_UNIT)/511, (507*TPE_FRACTIONS_PER_UNIT)/511,
  297. (508*TPE_FRACTIONS_PER_UNIT)/511, (509*TPE_FRACTIONS_PER_UNIT)/511,
  298. (509*TPE_FRACTIONS_PER_UNIT)/511, (510*TPE_FRACTIONS_PER_UNIT)/511,
  299. (510*TPE_FRACTIONS_PER_UNIT)/511, (510*TPE_FRACTIONS_PER_UNIT)/511,
  300. (510*TPE_FRACTIONS_PER_UNIT)/511, (510*TPE_FRACTIONS_PER_UNIT)/511
  301. };
  302. #define TPE_SIN_TABLE_UNIT_STEP\
  303. (TPE_FRACTIONS_PER_UNIT / (TPE_SIN_TABLE_LENGTH * 4))
  304. TPE_Unit TPE_sqrt(TPE_Unit value)
  305. {
  306. int8_t sign = 1;
  307. if (value < 0)
  308. {
  309. sign = -1;
  310. value *= -1;
  311. }
  312. uint32_t result = 0;
  313. uint32_t a = value;
  314. uint32_t b = 1u << 30;
  315. while (b > a)
  316. b >>= 2;
  317. while (b != 0)
  318. {
  319. if (a >= result + b)
  320. {
  321. a -= result + b;
  322. result = result + 2 * b;
  323. }
  324. b >>= 2;
  325. result >>= 1;
  326. }
  327. return result * sign;
  328. }
  329. TPE_Unit TPE_sin(TPE_Unit x)
  330. {
  331. x = TPE_wrap(x / TPE_SIN_TABLE_UNIT_STEP,TPE_SIN_TABLE_LENGTH * 4);
  332. int8_t positive = 1;
  333. if (x < TPE_SIN_TABLE_LENGTH)
  334. {
  335. }
  336. else if (x < TPE_SIN_TABLE_LENGTH * 2)
  337. {
  338. x = TPE_SIN_TABLE_LENGTH * 2 - x - 1;
  339. }
  340. else if (x < TPE_SIN_TABLE_LENGTH * 3)
  341. {
  342. x = x - TPE_SIN_TABLE_LENGTH * 2;
  343. positive = 0;
  344. }
  345. else
  346. {
  347. x = TPE_SIN_TABLE_LENGTH - (x - TPE_SIN_TABLE_LENGTH * 3) - 1;
  348. positive = 0;
  349. }
  350. return positive ? TPE_sinTable[x] : -1 * TPE_sinTable[x];
  351. }
  352. TPE_Unit TPE_cos(TPE_Unit x)
  353. {
  354. return TPE_sin(x + TPE_FRACTIONS_PER_UNIT / 4);
  355. }
  356. void TPE_bodyInit(TPE_Body *body)
  357. {
  358. // TODO
  359. TPE_initVec4(&(body->position));
  360. TPE_initVec4(&(body->velocity));
  361. // init orientation to identity unit quaternion (1,0,0,0):
  362. TPE_quaternionInit(&(body->rotation.originalOrientation));
  363. TPE_vec4Set(&(body->rotation.axisVelocity),TPE_FRACTIONS_PER_UNIT,0,0,0);
  364. body->rotation.currentAngle = 0;
  365. }
  366. void TPE_bodyGetOrientation(const TPE_Body *body, TPE_Vec4 *quaternion)
  367. {
  368. TPE_Vec4 axisRotation;
  369. TPE_rotationToQuaternion(
  370. body->rotation.axisVelocity,
  371. body->rotation.currentAngle,
  372. &axisRotation);
  373. TPE_quaternionMultiply(
  374. body->rotation.originalOrientation,
  375. axisRotation,
  376. quaternion);
  377. TPE_vec4Normalize(quaternion);
  378. }
  379. void TPE_vec3CrossProduct(TPE_Vec4 a, TPE_Vec4 b, TPE_Vec4 *result)
  380. {
  381. TPE_Vec4 r;
  382. r.x = a.y * b.z - a.z * b.y;
  383. r.y = a.z * b.x - a.x * b.z;
  384. r.z = a.x * b.y - a.y * b.x;
  385. *result = r;
  386. }
  387. void TPE_bodyApplyVelocity(TPE_Body *body, TPE_Vec4 point, TPE_Vec4 velocity)
  388. {
  389. TPE_Vec4 linearVelocity, angularVelocity, rotationAxis;
  390. TPE_vec3Add(body->velocity,velocity,&(body->velocity));
  391. TPE_Unit pointDistance = TPE_vec3Len(point);
  392. if (pointDistance != 0)
  393. {
  394. /* normalize the point, we don't use the function as we don't want to
  395. recompute the vector length */
  396. point.x = (point.x * TPE_FRACTIONS_PER_UNIT) / pointDistance;
  397. point.y = (point.y * TPE_FRACTIONS_PER_UNIT) / pointDistance;
  398. point.z = (point.z * TPE_FRACTIONS_PER_UNIT) / pointDistance;
  399. /* Now we take only a part of the applied velocity, the part projected
  400. to a plane perpendicular to the point vector, and this part will
  401. contribute to the body rotation. */
  402. TPE_Vec4 tmp;
  403. TPE_vec3Project(velocity,point,&tmp);
  404. TPE_vec3Substract(velocity,tmp,&angularVelocity);
  405. TPE_vec3CrossProduct(point,angularVelocity,&rotationAxis);
  406. TPE_bodyAddRotation(body,rotationAxis,
  407. TPE_linearVelocityToAngular(
  408. TPE_vec3Len(angularVelocity),-1 * pointDistance));
  409. }
  410. }
  411. TPE_Unit TPE_linearVelocityToAngular(TPE_Unit velocity, TPE_Unit distance)
  412. {
  413. TPE_Unit circumfence = (2 * TPE_PI * distance) / TPE_FRACTIONS_PER_UNIT;
  414. return (velocity * TPE_FRACTIONS_PER_UNIT) / circumfence;
  415. }
  416. void TPE_bodyStep(TPE_Body *body)
  417. {
  418. TPE_vec3Add(body->position,body->velocity,&(body->position));
  419. body->rotation.currentAngle += body->rotation.axisVelocity.w;
  420. }
  421. void TPE_bodySetRotation(TPE_Body *body, TPE_Vec4 axis, TPE_Unit velocity)
  422. {
  423. TPE_bodyGetOrientation(body,&(body->rotation.originalOrientation));
  424. if (velocity < 0)
  425. {
  426. axis.x *= -1;
  427. axis.y *= -1;
  428. axis.z *= -1;
  429. velocity *= -1;
  430. }
  431. TPE_vec3Normalize(&axis);
  432. body->rotation.axisVelocity = axis;
  433. body->rotation.axisVelocity.w = velocity;
  434. body->rotation.currentAngle = 0;
  435. }
  436. void TPE_bodyAddRotation(TPE_Body *body, TPE_Vec4 axis, TPE_Unit velocity)
  437. {
  438. /* Rotation is added like this: we convert both the original and added
  439. rotation to vectors whose direction is along the rotations axis and
  440. magnitude is the rotation speed, then we add these vectors and convert
  441. the final vector back to normalized rotation axis + scalar rotation
  442. speed.*/
  443. body->rotation.axisVelocity.x =
  444. (body->rotation.axisVelocity.x * body->rotation.axisVelocity.w)
  445. / TPE_FRACTIONS_PER_UNIT;
  446. body->rotation.axisVelocity.y =
  447. (body->rotation.axisVelocity.y * body->rotation.axisVelocity.w)
  448. / TPE_FRACTIONS_PER_UNIT;
  449. body->rotation.axisVelocity.z =
  450. (body->rotation.axisVelocity.z * body->rotation.axisVelocity.w)
  451. / TPE_FRACTIONS_PER_UNIT;
  452. TPE_vec3Normalize(&axis);
  453. axis.x = (axis.x * velocity) / TPE_FRACTIONS_PER_UNIT;
  454. axis.y = (axis.y * velocity) / TPE_FRACTIONS_PER_UNIT;
  455. axis.z = (axis.z * velocity) / TPE_FRACTIONS_PER_UNIT;
  456. TPE_vec3Add(body->rotation.axisVelocity,axis,&axis);
  457. axis.w = TPE_vec3Len(axis);
  458. TPE_bodySetRotation(body,axis,axis.w);
  459. }
  460. void TPE_quaternionMultiply(TPE_Vec4 a, TPE_Vec4 b, TPE_Vec4 *result)
  461. {
  462. TPE_Vec4 r; // in case result is identical to a or b
  463. r.x =
  464. (a.w * b.x +
  465. a.x * b.w +
  466. a.y * b.z -
  467. a.z * b.y) / TPE_FRACTIONS_PER_UNIT;
  468. r.y =
  469. (a.w * b.y -
  470. a.x * b.z +
  471. a.y * b.w +
  472. a.z * b.x) / TPE_FRACTIONS_PER_UNIT;
  473. r.z =
  474. (a.w * b.z +
  475. a.x * b.y -
  476. a.y * b.x +
  477. a.z * b.w) / TPE_FRACTIONS_PER_UNIT;
  478. r.w =
  479. (a.w * b.w -
  480. a.x * b.x -
  481. a.y * b.y -
  482. a.z * b.z) / TPE_FRACTIONS_PER_UNIT;
  483. result->x = r.x;
  484. result->y = r.y;
  485. result->z = r.z;
  486. result->w = r.w;
  487. }
  488. void TPE_rotationToQuaternion(TPE_Vec4 axis, TPE_Unit angle, TPE_Vec4 *quaternion)
  489. {
  490. TPE_vec3Normalize(&axis);
  491. angle /= 2;
  492. TPE_Unit s = TPE_sin(angle);
  493. quaternion->x = (s * axis.x) / TPE_FRACTIONS_PER_UNIT;
  494. quaternion->y = (s * axis.y) / TPE_FRACTIONS_PER_UNIT;
  495. quaternion->z = (s * axis.z) / TPE_FRACTIONS_PER_UNIT;
  496. quaternion->w = TPE_cos(angle);
  497. }
  498. TPE_Unit TPE_asin(TPE_Unit x)
  499. {
  500. x = TPE_clamp(x,-TPE_FRACTIONS_PER_UNIT,TPE_FRACTIONS_PER_UNIT);
  501. int8_t sign = 1;
  502. if (x < 0)
  503. {
  504. sign = -1;
  505. x *= -1;
  506. }
  507. int16_t low = 0;
  508. int16_t high = TPE_SIN_TABLE_LENGTH -1;
  509. int16_t middle;
  510. while (low <= high) // binary search
  511. {
  512. middle = (low + high) / 2;
  513. TPE_Unit v = TPE_sinTable[middle];
  514. if (v > x)
  515. high = middle - 1;
  516. else if (v < x)
  517. low = middle + 1;
  518. else
  519. break;
  520. }
  521. middle *= TPE_SIN_TABLE_UNIT_STEP;
  522. return sign * middle;
  523. }
  524. TPE_Unit TPE_acos(TPE_Unit x)
  525. {
  526. return TPE_asin(-1 * x) + TPE_FRACTIONS_PER_UNIT / 4;
  527. }
  528. void TPE_quaternionToRotation(TPE_Vec4 quaternion, TPE_Vec4 *axis, TPE_Unit *angle)
  529. {
  530. *angle = 2 * TPE_acos(quaternion.x);
  531. TPE_Unit tmp =
  532. TPE_nonZero(TPE_sqrt(
  533. (TPE_FRACTIONS_PER_UNIT -
  534. (quaternion.x * quaternion.x) / TPE_FRACTIONS_PER_UNIT
  535. ) * TPE_FRACTIONS_PER_UNIT));
  536. axis->x = (quaternion.x * TPE_FRACTIONS_PER_UNIT) / tmp;
  537. axis->y = (quaternion.y * TPE_FRACTIONS_PER_UNIT) / tmp;
  538. axis->z = (quaternion.z * TPE_FRACTIONS_PER_UNIT) / tmp;
  539. }
  540. void TPE_quaternionToRotationMatrix(TPE_Vec4 quaternion, TPE_Unit matrix[4][4])
  541. {
  542. TPE_Unit
  543. _2x2 = (2 * quaternion.x * quaternion.x) / TPE_FRACTIONS_PER_UNIT,
  544. _2y2 = (2 * quaternion.y * quaternion.y) / TPE_FRACTIONS_PER_UNIT,
  545. _2z2 = (2 * quaternion.z * quaternion.z) / TPE_FRACTIONS_PER_UNIT,
  546. _2xy = (2 * quaternion.x * quaternion.y) / TPE_FRACTIONS_PER_UNIT,
  547. _2xw = (2 * quaternion.x * quaternion.w) / TPE_FRACTIONS_PER_UNIT,
  548. _2zw = (2 * quaternion.z * quaternion.w) / TPE_FRACTIONS_PER_UNIT,
  549. _2xz = (2 * quaternion.x * quaternion.z) / TPE_FRACTIONS_PER_UNIT,
  550. _2yw = (2 * quaternion.y * quaternion.w) / TPE_FRACTIONS_PER_UNIT,
  551. _2yz = (2 * quaternion.y * quaternion.z) / TPE_FRACTIONS_PER_UNIT;
  552. #define ONE TPE_FRACTIONS_PER_UNIT
  553. matrix[0][0] = ONE - _2y2 - _2z2;
  554. matrix[1][0] = _2xy - _2zw;
  555. matrix[2][0] = _2xz + _2yw;
  556. matrix[3][0] = 0;
  557. matrix[0][1] = _2xy + _2zw;
  558. matrix[1][1] = ONE - _2x2 - _2z2;
  559. matrix[2][1] = _2yz - _2xw;
  560. matrix[3][1] = 0;
  561. matrix[0][2] = _2xz - _2yw;
  562. matrix[1][2] = _2yz + _2xw;
  563. matrix[2][2] = ONE - _2x2 - _2y2;
  564. matrix[3][2] = 0;
  565. matrix[0][3] = 0;
  566. matrix[1][3] = 0;
  567. matrix[2][3] = 0;
  568. matrix[3][3] = ONE;
  569. #undef ONE
  570. }
  571. void TPE_vec3Add(const TPE_Vec4 a, const TPE_Vec4 b, TPE_Vec4 *result)
  572. {
  573. result->x = a.x + b.x;
  574. result->y = a.y + b.y;
  575. result->z = a.z + b.z;
  576. }
  577. void TPE_vec4Add(const TPE_Vec4 a, const TPE_Vec4 b, TPE_Vec4 *result)
  578. {
  579. result->x = a.x + b.x;
  580. result->y = a.y + b.y;
  581. result->z = a.z + b.z;
  582. result->w = a.w + b.w;
  583. }
  584. void TPE_vec3Substract(const TPE_Vec4 a, const TPE_Vec4 b, TPE_Vec4 *result)
  585. {
  586. result->x = a.x - b.x;
  587. result->y = a.y - b.y;
  588. result->z = a.z - b.z;
  589. }
  590. void TPE_vec4Substract(const TPE_Vec4 a, const TPE_Vec4 b, TPE_Vec4 *result)
  591. {
  592. result->x = a.x - b.x;
  593. result->y = a.y - b.y;
  594. result->z = a.z - b.z;
  595. result->w = a.w - b.w;
  596. }
  597. void TPE_vec3Multiply(const TPE_Vec4 v, TPE_Unit f, TPE_Vec4 *result)
  598. {
  599. result->x = (v.x * f) / TPE_FRACTIONS_PER_UNIT;
  600. result->y = (v.y * f) / TPE_FRACTIONS_PER_UNIT;
  601. result->z = (v.z * f) / TPE_FRACTIONS_PER_UNIT;
  602. }
  603. void TPE_vec4Multiply(const TPE_Vec4 v, TPE_Unit f, TPE_Vec4 *result)
  604. {
  605. result->x = (v.x * f) / TPE_FRACTIONS_PER_UNIT;
  606. result->y = (v.y * f) / TPE_FRACTIONS_PER_UNIT;
  607. result->z = (v.z * f) / TPE_FRACTIONS_PER_UNIT;
  608. result->w = (v.w * f) / TPE_FRACTIONS_PER_UNIT;
  609. }
  610. TPE_Unit TPE_vec3Len(TPE_Vec4 v)
  611. {
  612. return TPE_sqrt(v.x * v.x + v.y * v.y + v.z * v.z);
  613. }
  614. TPE_Unit TPE_vec4Len(TPE_Vec4 v)
  615. {
  616. return TPE_sqrt(v.x * v.x + v.y * v.y + v.z * v.z + v.w * v.w);
  617. }
  618. TPE_Unit TPE_vec3DotProduct(const TPE_Vec4 v1, const TPE_Vec4 v2)
  619. {
  620. return
  621. (v1.x * v2.x + v1.y * v2.y + v1.z * v2.z) / TPE_FRACTIONS_PER_UNIT;
  622. }
  623. void TPE_vec3Normalize(TPE_Vec4 *v)
  624. {
  625. TPE_Unit l = TPE_vec3Len(*v);
  626. if (l == 0)
  627. {
  628. v->x = TPE_FRACTIONS_PER_UNIT;
  629. return;
  630. }
  631. v->x = (v->x * TPE_FRACTIONS_PER_UNIT) / l;
  632. v->y = (v->y * TPE_FRACTIONS_PER_UNIT) / l;
  633. v->z = (v->z * TPE_FRACTIONS_PER_UNIT) / l;
  634. }
  635. void TPE_vec4Normalize(TPE_Vec4 *v)
  636. {
  637. TPE_Unit l = TPE_vec4Len(*v);
  638. if (l == 0)
  639. {
  640. v->x = TPE_FRACTIONS_PER_UNIT;
  641. return;
  642. }
  643. v->x = (v->x * TPE_FRACTIONS_PER_UNIT) / l;
  644. v->y = (v->y * TPE_FRACTIONS_PER_UNIT) / l;
  645. v->z = (v->z * TPE_FRACTIONS_PER_UNIT) / l;
  646. v->w = (v->w * TPE_FRACTIONS_PER_UNIT) / l;
  647. }
  648. void TPE_vec3Project(TPE_Vec4 v, TPE_Vec4 base, TPE_Vec4 *result)
  649. {
  650. TPE_Unit p = TPE_vec3DotProduct(v,base);
  651. result->x = (p * base.x) / TPE_FRACTIONS_PER_UNIT;
  652. result->y = (p * base.y) / TPE_FRACTIONS_PER_UNIT;
  653. result->z = (p * base.z) / TPE_FRACTIONS_PER_UNIT;
  654. }
  655. void TPE_getVelocitiesAfterCollision(
  656. TPE_Unit *v1,
  657. TPE_Unit *v2,
  658. TPE_Unit m1,
  659. TPE_Unit m2,
  660. TPE_Unit elasticity
  661. )
  662. {
  663. /* in the following a lot of TPE_FRACTIONS_PER_UNIT cancel out, feel free to
  664. check if confused */
  665. #define ANTI_OVERFLOW 30000
  666. #define ANTI_OVERFLOW_SCALE 128
  667. uint8_t overflowDanger = m1 > ANTI_OVERFLOW || *v1 > ANTI_OVERFLOW ||
  668. m2 > ANTI_OVERFLOW || *v2 > ANTI_OVERFLOW;
  669. if (overflowDanger)
  670. {
  671. m1 = (m1 != 0) ? TPE_nonZero(m1 / ANTI_OVERFLOW_SCALE) : 0;
  672. m2 = (m2 != 0) ? TPE_nonZero(m2 / ANTI_OVERFLOW_SCALE) : 0;
  673. *v1 = (*v1 != 0) ? TPE_nonZero(*v1 / ANTI_OVERFLOW_SCALE) : 0;
  674. *v2 = (*v2 != 0) ? TPE_nonZero(*v2 / ANTI_OVERFLOW_SCALE) : 0;
  675. }
  676. TPE_Unit m1Pm2 = m1 + m2;
  677. TPE_Unit v2Mv1 = *v2 - *v1;
  678. TPE_Unit m1v1Pm2v2 = ((m1 * *v1) + (m2 * *v2));
  679. *v1 = (((elasticity * m2 / TPE_FRACTIONS_PER_UNIT) * v2Mv1)
  680. + m1v1Pm2v2) / m1Pm2;
  681. *v2 = (((elasticity * m1 / TPE_FRACTIONS_PER_UNIT) * -1 * v2Mv1)
  682. + m1v1Pm2v2) / m1Pm2;
  683. if (overflowDanger)
  684. {
  685. *v1 *= ANTI_OVERFLOW_SCALE;
  686. *v2 *= ANTI_OVERFLOW_SCALE;
  687. }
  688. #undef ANTI_OVERFLOW
  689. #undef ANTI_OVERFLOW_SCALE
  690. }
  691. void TPE_resolvePointCollision(
  692. TPE_Vec4 collisionPoint,
  693. TPE_Vec4 collisionNormal,
  694. TPE_Unit elasticity,
  695. TPE_Vec4 linVelocity1,
  696. TPE_Vec4 rotVelocity1,
  697. TPE_Unit m1,
  698. TPE_Vec4 linVelocity2,
  699. TPE_Vec4 rotVelocity2,
  700. TPE_Unit m2)
  701. {
  702. TPE_Vec4 v1, v2, v1New, v2New;
  703. TPE_initVec4(&v1);
  704. TPE_initVec4(&v2);
  705. TPE_initVec4(&v1New);
  706. TPE_initVec4(&v2New);
  707. // add lin. and rot. velocities to get the overall vel. of both points:
  708. TPE_vec4Add(linVelocity1,rotVelocity1,&v1);
  709. TPE_vec4Add(linVelocity2,rotVelocity2,&v2);
  710. /* project both of these velocities to the collision normal as we'll apply
  711. the collision equation only in the direction of this normal: */
  712. TPE_vec3Project(v1,collisionNormal,&v1New);
  713. TPE_vec3Project(v2,collisionNormal,&v2New);
  714. // get the velocities of the components
  715. TPE_Unit
  716. v1NewMag = TPE_vec3Len(v1New),
  717. v2NewMag = TPE_vec3Len(v2New);
  718. /* now also substract this component from the original velocity (so that it
  719. will now be in the collision plane), we'll later add back the updated
  720. velocity to it */
  721. TPE_vec4Substract(v1,v1New,&v1);
  722. TPE_vec4Substract(v2,v2New,&v2);
  723. // apply the 1D collision equation to velocities along the normal:
  724. TPE_getVelocitiesAfterCollision(
  725. &v1NewMag,
  726. &v2NewMag,
  727. m1,
  728. m2,
  729. elasticity);
  730. // add back the updated velocities to get the new overall velocities:
  731. v1New.x += (collisionNormal.x * v1NewMag) / TPE_FRACTIONS_PER_UNIT;
  732. v1New.y += (collisionNormal.y * v1NewMag) / TPE_FRACTIONS_PER_UNIT;
  733. v1New.z += (collisionNormal.z * v1NewMag) / TPE_FRACTIONS_PER_UNIT;
  734. v2New.x += (collisionNormal.x * v2NewMag) / TPE_FRACTIONS_PER_UNIT;
  735. v2New.y += (collisionNormal.y * v2NewMag) / TPE_FRACTIONS_PER_UNIT;
  736. v2New.z += (collisionNormal.z * v2NewMag) / TPE_FRACTIONS_PER_UNIT;
  737. // TODO
  738. }
  739. void TPE_bodyGetTransformMatrix(const TPE_Body *body, TPE_Unit matrix[4][4])
  740. {
  741. TPE_Vec4 orientation;
  742. TPE_bodyGetOrientation(body,&orientation);
  743. TPE_quaternionToRotationMatrix(orientation,matrix);
  744. matrix[0][3] = body->position.x;
  745. matrix[1][3] = body->position.y;
  746. matrix[2][3] = body->position.z;
  747. }
  748. void TPE_quaternionInit(TPE_Vec4 *quaternion)
  749. {
  750. quaternion->x = 0;
  751. quaternion->y = 0;
  752. quaternion->z = 0;
  753. quaternion->w = TPE_FRACTIONS_PER_UNIT;
  754. }
  755. #endif // guard