git.mutantstargoat.com Git - bootcensus/blob - src/census/cgmath/cgmath.h

   1 /* gph-cmath - C graphics math library
   2  * Copyright (C) 2018 John Tsiombikas <nuclear@member.fsf.org>
   3  *
   4  * This program is free software. Feel free to use, modify, and/or redistribute
   5  * it under the terms of the MIT/X11 license. See LICENSE for details.
   6  * If you intend to redistribute parts of the code without the LICENSE file
   7  * replace this paragraph with the full contents of the LICENSE file.
   8  *
   9  * Function prefixes signify the data type of their operand(s):
  10  * - cgm_v... functions are operations on cgm_vec3 vectors
  11  * - cgm_w... functions are operations on cgm_vec4 vectors
  12  * - cgm_q... functions are operations on cgm_quat quaternions (w + xi + yj + zk)
  13  * - cgm_m... functions are operations on 4x4 matrices (stored as linear 16 float arrays)
  14  * - cgm_r... functions are operations on cgm_ray rays
  15  *
  16  * NOTE: *ALL* matrix arguments are pointers to 16 floats. Even the functions
  17  * which operate on 3x3 matrices, actually use the upper 3x3 of a 4x4 matrix,
  18  * and still expect an array of 16 floats.
  19  *
  20  * NOTE: matrices are treated by all operations as column-major, to match OpenGL
  21  * conventions, so everything is pretty much transposed.
  22 */
  23 #ifndef CGMATH_H_
  24 #define CGMATH_H_
  25
  26 #include <math.h>
  27 #include <string.h>
  28
  29 typedef struct {
  30         float x, y;
  31 } cgm_vec2;
  32
  33 typedef struct {
  34         float x, y, z;
  35 } cgm_vec3;
  36
  37 typedef struct {
  38         float x, y, z, w;
  39 } cgm_vec4, cgm_quat;
  40
  41 typedef struct {
  42         cgm_vec3 origin, dir;
  43 } cgm_ray;
  44
  45 typedef enum cgm_euler_mode {
  46         CGM_EULER_XYZ,
  47         CGM_EULER_XZY,
  48         CGM_EULER_YXZ,
  49         CGM_EULER_YZX,
  50         CGM_EULER_ZXY,
  51         CGM_EULER_ZYX,
  52         CGM_EULER_ZXZ,
  53         CGM_EULER_ZYZ,
  54         CGM_EULER_YXY,
  55         CGM_EULER_YZY,
  56         CGM_EULER_XYX,
  57         CGM_EULER_XZX
  58 } cgm_euler_mode;
  59
  60 #ifdef __cplusplus
  61 extern "C" {
  62 #endif
  63
  64 /* --- operations on cgm_vec3 --- */
  65 static inline void cgm_vcons(cgm_vec3 *v, float x, float y, float z);
  66
  67 static inline void cgm_vadd(cgm_vec3 *a, const cgm_vec3 *b);
  68 static inline void cgm_vsub(cgm_vec3 *a, const cgm_vec3 *b);
  69 static inline void cgm_vmul(cgm_vec3 *a, const cgm_vec3 *b);
  70 static inline void cgm_vscale(cgm_vec3 *v, float s);
  71 static inline void cgm_vmul_m4v3(cgm_vec3 *v, const float *m);  /* m4x4 * v */
  72 static inline void cgm_vmul_v3m4(cgm_vec3 *v, const float *m);  /* v * m4x4 */
  73 static inline void cgm_vmul_m3v3(cgm_vec3 *v, const float *m);  /* m3x3 * v (m still 16 floats) */
  74 static inline void cgm_vmul_v3m3(cgm_vec3 *v, const float *m);  /* v * m3x3 (m still 16 floats) */
  75
  76 static inline float cgm_vdot(const cgm_vec3 *a, const cgm_vec3 *b);
  77 static inline void cgm_vcross(cgm_vec3 *res, const cgm_vec3 *a, const cgm_vec3 *b);
  78 static inline float cgm_vlength(const cgm_vec3 *v);
  79 static inline float cgm_vlength_sq(const cgm_vec3 *v);
  80 static inline float cgm_vdist(const cgm_vec3 *a, const cgm_vec3 *b);
  81 static inline float cgm_vdist_sq(const cgm_vec3 *a, const cgm_vec3 *b);
  82 static inline void cgm_vnormalize(cgm_vec3 *v);
  83
  84 static inline void cgm_vreflect(cgm_vec3 *v, const cgm_vec3 *n);
  85 static inline void cgm_vrefract(cgm_vec3 *v, const cgm_vec3 *n, float ior);
  86
  87 static inline void cgm_vrotate_quat(cgm_vec3 *v, const cgm_quat *q);
  88 static inline void cgm_vrotate_axis(cgm_vec3 *v, int axis, float angle);
  89 static inline void cgm_vrotate(cgm_vec3 *v, float angle, float x, float y, float z);
  90 static inline void cgm_vrotate_euler(cgm_vec3 *v, float a, float b, float c, enum cgm_euler_mode mode);
  91
  92 static inline void cgm_vlerp(cgm_vec3 *res, const cgm_vec3 *a, const cgm_vec3 *b, float t);
  93
  94 #define cgm_velem(vptr, idx)    ((&(vptr)->x)[idx])
  95
  96 /* --- operations on cgm_vec4 --- */
  97 static inline void cgm_wcons(cgm_vec4 *v, float x, float y, float z, float w);
  98
  99 static inline void cgm_wadd(cgm_vec4 *a, const cgm_vec4 *b);
 100 static inline void cgm_wsub(cgm_vec4 *a, const cgm_vec4 *b);
 101 static inline void cgm_wmul(cgm_vec4 *a, const cgm_vec4 *b);
 102 static inline void cgm_wscale(cgm_vec4 *v, float s);
 103
 104 static inline void cgm_wmul_m4v4(cgm_vec4 *v, const float *m);
 105 static inline void cgm_wmul_v4m4(cgm_vec4 *v, const float *m);
 106 static inline void cgm_wmul_m34v4(cgm_vec4 *v, const float *m); /* doesn't affect w */
 107 static inline void cgm_wmul_v4m43(cgm_vec4 *v, const float *m); /* doesn't affect w */
 108 static inline void cgm_wmul_m3v4(cgm_vec4 *v, const float *m); /* (m still 16 floats) */
 109 static inline void cgm_wmul_v4m3(cgm_vec4 *v, const float *m); /* (m still 16 floats) */
 110
 111 static inline float cgm_wdot(const cgm_vec4 *a, const cgm_vec4 *b);
 112
 113 static inline float cgm_wlength(const cgm_vec4 *v);
 114 static inline float cgm_wlength_sq(const cgm_vec4 *v);
 115 static inline float cgm_wdist(const cgm_vec4 *a, const cgm_vec4 *b);
 116 static inline float cgm_wdist_sq(const cgm_vec4 *a, const cgm_vec4 *b);
 117 static inline void cgm_wnormalize(cgm_vec4 *v);
 118
 119 static inline void cgm_wlerp(cgm_vec4 *res, const cgm_vec4 *a, const cgm_vec4 *b, float t);
 120
 121 #define cgm_welem(vptr, idx)    ((&(vptr)->x)[idx])
 122
 123 /* --- operations on quaternions --- */
 124 static inline void cgm_qcons(cgm_quat *q, float x, float y, float z, float w);
 125
 126 static inline void cgm_qneg(cgm_quat *q);
 127 static inline void cgm_qadd(cgm_quat *a, const cgm_quat *b);
 128 static inline void cgm_qsub(cgm_quat *a, const cgm_quat *b);
 129 static inline void cgm_qmul(cgm_quat *a, const cgm_quat *b);
 130
 131 static inline float cgm_qlength(const cgm_quat *q);
 132 static inline float cgm_qlength_sq(const cgm_quat *q);
 133 static inline void cgm_qnormalize(cgm_quat *q);
 134 static inline void cgm_qconjugate(cgm_quat *q);
 135 static inline void cgm_qinvert(cgm_quat *q);
 136
 137 static inline void cgm_qrotation(cgm_quat *q, float angle, float x, float y, float z);
 138 static inline void cgm_qrotate(cgm_quat *q, float angle, float x, float y, float z);
 139
 140 static inline void cgm_qslerp(cgm_quat *res, const cgm_quat *a, const cgm_quat *b, float t);
 141 static inline void cgm_qlerp(cgm_quat *res, const cgm_quat *a, const cgm_quat *b, float t);
 142
 143 #define cgm_qelem(qptr, idx)    ((&(qptr)->x)[idx])
 144
 145 /* --- operations on matrices --- */
 146 static inline void cgm_mcopy(float *dest, const float *src);
 147 static inline void cgm_mzero(float *m);
 148 static inline void cgm_midentity(float *m);
 149
 150 static inline void cgm_mmul(float *a, const float *b);
 151 static inline void cgm_mpremul(float *a, const float *b);
 152
 153 static inline void cgm_msubmatrix(float *m, int row, int col);
 154 static inline void cgm_mupper3(float *m);
 155 static inline float cgm_msubdet(const float *m, int row, int col);
 156 static inline float cgm_mcofactor(const float *m, int row, int col);
 157 static inline float cgm_mdet(const float *m);
 158 static inline void cgm_mtranspose(float *m);
 159 static inline void cgm_mcofmatrix(float *m);
 160 static inline int cgm_minverse(float *m);       /* returns 0 on success, -1 for singular */
 161
 162 static inline void cgm_mtranslation(float *m, float x, float y, float z);
 163 static inline void cgm_mscaling(float *m, float sx, float sy, float sz);
 164 static inline void cgm_mrotation_x(float *m, float angle);
 165 static inline void cgm_mrotation_y(float *m, float angle);
 166 static inline void cgm_mrotation_z(float *m, float angle);
 167 static inline void cgm_mrotation_axis(float *m, int idx, float angle);
 168 static inline void cgm_mrotation(float *m, float angle, float x, float y, float z);
 169 static inline void cgm_mrotation_euler(float *m, float a, float b, float c, int mode);
 170 static inline void cgm_mrotation_quat(float *m, const cgm_quat *q);
 171
 172 static inline void cgm_mtranslate(float *m, float x, float y, float z);
 173 static inline void cgm_mscale(float *m, float sx, float sy, float sz);
 174 static inline void cgm_mrotate_x(float *m, float angle);
 175 static inline void cgm_mrotate_y(float *m, float angle);
 176 static inline void cgm_mrotate_z(float *m, float angle);
 177 static inline void cgm_mrotate_axis(float *m, int idx, float angle);
 178 static inline void cgm_mrotate(float *m, float angle, float x, float y, float z);
 179 static inline void cgm_mrotate_euler(float *m, float a, float b, float c, int mode);
 180 static inline void cgm_mrotate_quat(float *m, const cgm_quat *q);
 181
 182 static inline void cgm_mpretranslate(float *m, float x, float y, float z);
 183 static inline void cgm_mprescale(float *m, float sx, float sy, float sz);
 184 static inline void cgm_mprerotate_x(float *m, float angle);
 185 static inline void cgm_mprerotate_y(float *m, float angle);
 186 static inline void cgm_mprerotate_z(float *m, float angle);
 187 static inline void cgm_mprerotate_axis(float *m, int idx, float angle);
 188 static inline void cgm_mprerotate(float *m, float angle, float x, float y, float z);
 189 static inline void cgm_mprerotate_euler(float *m, float a, float b, float c, int mode);
 190 static inline void cgm_mprerotate_quat(float *m, const cgm_quat *q);
 191
 192 static inline void cgm_mget_translation(const float *m, cgm_vec3 *res);
 193 static inline void cgm_mget_rotation(const float *m, cgm_quat *res);
 194 static inline void cgm_mget_scaling(const float *m, cgm_vec3 *res);
 195 static inline void cgm_mget_frustum_plane(const float *m, int p, cgm_vec4 *res);
 196
 197 static inline void cgm_mlookat(float *m, const cgm_vec3 *pos, const cgm_vec3 *targ,
 198                 const cgm_vec3 *up);
 199 static inline void cgm_minv_lookat(float *m, const cgm_vec3 *pos, const cgm_vec3 *targ,
 200                 const cgm_vec3 *up);
 201 static inline void cgm_mortho(float *m, float left, float right, float bot, float top,
 202                 float znear, float zfar);
 203 static inline void cgm_mfrustum(float *m, float left, float right, float bot, float top,
 204                 float znear, float zfar);
 205 static inline void cgm_mperspective(float *m, float vfov, float aspect, float znear, float zfar);
 206
 207 static inline void cgm_mmirror(float *m, float a, float b, float c, float d);
 208
 209 /* --- operations on rays --- */
 210 static inline void cgm_rcons(cgm_ray *r, float x, float y, float z, float dx, float dy, float dz);
 211
 212 static inline void cgm_rmul_mr(cgm_ray *ray, const float *m);   /* m4x4 * ray */
 213 static inline void cgm_rmul_rm(cgm_ray *ray, const float *m);   /* ray * m4x4 */
 214
 215 static inline void cgm_rreflect(cgm_ray *ray, const cgm_vec3 *n);
 216 static inline void cgm_rrefract(cgm_ray *ray, const cgm_vec3 *n, float ior);
 217
 218 /* --- miscellaneous utility functions --- */
 219 static inline float cgm_deg_to_rad(float deg);
 220 static inline float cgm_rad_to_deg(float rad);
 221
 222 static inline float cgm_smoothstep(float a, float b, float x);
 223 static inline float cgm_lerp(float a, float b, float t);
 224 static inline float cgm_bezier(float a, float b, float c, float d, float t);
 225 static inline float cgm_bspline(float a, float b, float c, float d, float t);
 226 static inline float cgm_spline(float a, float b, float c, float d, float t);
 227
 228 static inline void cgm_discrand(cgm_vec3 *v, float rad);
 229 static inline void cgm_sphrand(cgm_vec3 *v, float rad);
 230
 231 static inline void cgm_unproject(cgm_vec3 *res, const cgm_vec3 *norm_scrpos,
 232                 const float *inv_viewproj);
 233 static inline void cgm_glu_unproject(float winx, float winy, float winz,
 234                 const float *view, const float *proj, const int *vp,
 235                 float *objx, float *objy, float *objz);
 236
 237 static inline void cgm_pick_ray(cgm_ray *ray, float nx, float ny,
 238                 const float *viewmat, const float *projmat);
 239
 240 static inline void cgm_raypos(cgm_vec3 *p, const cgm_ray *ray, float t);
 241
 242 /* calculate barycentric coordinates of point pt in triangle (a, b, c) */
 243 static inline void cgm_bary(cgm_vec3 *bary, const cgm_vec3 *a,
 244                 const cgm_vec3 *b, const cgm_vec3 *c, const cgm_vec3 *pt);
 245
 246 #include "cgmvec3.inl"
 247 #include "cgmvec4.inl"
 248 #include "cgmquat.inl"
 249 #include "cgmmat.inl"
 250 #include "cgmray.inl"
 251 #include "cgmmisc.inl"
 252
 253 #ifdef __cplusplus
 254 }
 255 #endif
 256
 257 #endif  /* CGMATH_H_ */