general.odin 18 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669
  1. package linalg
  2. import "core:math"
  3. import "base:builtin"
  4. import "base:intrinsics"
  5. @require import "base:runtime"
  6. // Generic
  7. TAU :: 6.28318530717958647692528676655900576
  8. PI :: 3.14159265358979323846264338327950288
  9. E :: 2.71828182845904523536
  10. τ :: TAU
  11. π :: PI
  12. e :: E
  13. SQRT_TWO :: 1.41421356237309504880168872420969808
  14. SQRT_THREE :: 1.73205080756887729352744634150587236
  15. SQRT_FIVE :: 2.23606797749978969640917366873127623
  16. LN2 :: 0.693147180559945309417232121458176568
  17. LN10 :: 2.30258509299404568401799145468436421
  18. MAX_F64_PRECISION :: 16 // Maximum number of meaningful digits after the decimal point for 'f64'
  19. MAX_F32_PRECISION :: 8 // Maximum number of meaningful digits after the decimal point for 'f32'
  20. RAD_PER_DEG :: TAU/360.0
  21. DEG_PER_RAD :: 360.0/TAU
  22. @private IS_NUMERIC :: intrinsics.type_is_numeric
  23. @private IS_QUATERNION :: intrinsics.type_is_quaternion
  24. @private IS_ARRAY :: intrinsics.type_is_array
  25. @private IS_FLOAT :: intrinsics.type_is_float
  26. @private BASE_TYPE :: intrinsics.type_base_type
  27. @private ELEM_TYPE :: intrinsics.type_elem_type
  28. @(require_results)
  29. scalar_dot :: proc "contextless" (a, b: $T) -> T where IS_FLOAT(T), !IS_ARRAY(T) {
  30. return a * b
  31. }
  32. @(require_results)
  33. vector_dot :: proc "contextless" (a, b: $T/[$N]$E) -> (c: E) where IS_NUMERIC(E) #no_bounds_check {
  34. for i in 0..<N {
  35. c += a[i] * b[i]
  36. }
  37. return
  38. }
  39. @(require_results)
  40. quaternion64_dot :: proc "contextless" (a, b: $T/quaternion64) -> (c: f16) {
  41. return a.w*b.w + a.x*b.x + a.y*b.y + a.z*b.z
  42. }
  43. @(require_results)
  44. quaternion128_dot :: proc "contextless" (a, b: $T/quaternion128) -> (c: f32) {
  45. return a.w*b.w + a.x*b.x + a.y*b.y + a.z*b.z
  46. }
  47. @(require_results)
  48. quaternion256_dot :: proc "contextless" (a, b: $T/quaternion256) -> (c: f64) {
  49. return a.w*b.w + a.x*b.x + a.y*b.y + a.z*b.z
  50. }
  51. dot :: proc{scalar_dot, vector_dot, quaternion64_dot, quaternion128_dot, quaternion256_dot}
  52. inner_product :: dot
  53. outer_product :: intrinsics.outer_product
  54. @(require_results)
  55. quaternion_inverse :: proc "contextless" (q: $Q) -> Q where IS_QUATERNION(Q) {
  56. return conj(q) * quaternion(w=1.0/dot(q, q), x=0, y=0, z=0)
  57. }
  58. @(require_results)
  59. scalar_cross :: proc "contextless" (a, b: $T) -> T where IS_FLOAT(T), !IS_ARRAY(T) {
  60. return a * b
  61. }
  62. @(require_results)
  63. vector_cross2 :: proc "contextless" (a, b: $T/[2]$E) -> E where IS_NUMERIC(E) {
  64. return a[0]*b[1] - b[0]*a[1]
  65. }
  66. @(require_results)
  67. vector_cross3 :: proc "contextless" (a, b: $T/[3]$E) -> (c: T) where IS_NUMERIC(E) {
  68. c[0] = a[1]*b[2] - b[1]*a[2]
  69. c[1] = a[2]*b[0] - b[2]*a[0]
  70. c[2] = a[0]*b[1] - b[0]*a[1]
  71. return
  72. }
  73. @(require_results)
  74. quaternion_cross :: proc "contextless" (q1, q2: $Q) -> (q3: Q) where IS_QUATERNION(Q) {
  75. q3.x = q1.w * q2.x + q1.x * q2.w + q1.y * q2.z - q1.z * q2.y
  76. q3.y = q1.w * q2.y + q1.y * q2.w + q1.z * q2.x - q1.x * q2.z
  77. q3.z = q1.w * q2.z + q1.z * q2.w + q1.x * q2.y - q1.y * q2.x
  78. q3.w = q1.w * q2.w - q1.x * q2.x - q1.y * q2.y - q1.z * q2.z
  79. return
  80. }
  81. vector_cross :: proc{scalar_cross, vector_cross2, vector_cross3}
  82. cross :: proc{scalar_cross, vector_cross2, vector_cross3, quaternion_cross}
  83. @(require_results)
  84. vector_normalize :: proc "contextless" (v: $T/[$N]$E) -> T where IS_FLOAT(E) {
  85. return v / length(v)
  86. }
  87. @(require_results)
  88. quaternion_normalize :: proc "contextless" (q: $Q) -> Q where IS_QUATERNION(Q) {
  89. return q/abs(q)
  90. }
  91. normalize :: proc{vector_normalize, quaternion_normalize}
  92. @(require_results)
  93. vector_normalize0 :: proc "contextless" (v: $T/[$N]$E) -> T where IS_FLOAT(E) {
  94. m := length(v)
  95. return 0 if m == 0 else v/m
  96. }
  97. @(require_results)
  98. quaternion_normalize0 :: proc "contextless" (q: $Q) -> Q where IS_QUATERNION(Q) {
  99. m := abs(q)
  100. return 0 if m == 0 else q/m
  101. }
  102. normalize0 :: proc{vector_normalize0, quaternion_normalize0}
  103. @(require_results)
  104. vector_length :: proc "contextless" (v: $T/[$N]$E) -> E where IS_FLOAT(E) {
  105. return math.sqrt(dot(v, v))
  106. }
  107. @(require_results)
  108. vector_length2 :: proc "contextless" (v: $T/[$N]$E) -> E where IS_NUMERIC(E) {
  109. return dot(v, v)
  110. }
  111. @(require_results)
  112. quaternion_length :: proc "contextless" (q: $Q) -> Q where IS_QUATERNION(Q) {
  113. return abs(q)
  114. }
  115. @(require_results)
  116. quaternion_length2 :: proc "contextless" (q: $Q) -> Q where IS_QUATERNION(Q) {
  117. return dot(q, q)
  118. }
  119. @(require_results)
  120. scalar_triple_product :: proc "contextless" (a, b, c: $T/[$N]$E) -> E where IS_NUMERIC(E) {
  121. // a . (b x c)
  122. // b . (c x a)
  123. // c . (a x b)
  124. return dot(a, cross(b, c))
  125. }
  126. @(require_results)
  127. vector_triple_product :: proc "contextless" (a, b, c: $T/[$N]$E) -> T where IS_NUMERIC(E) {
  128. // a x (b x c)
  129. // (a . c)b - (a . b)c
  130. return cross(a, cross(b, c))
  131. }
  132. length :: proc{vector_length, quaternion_length}
  133. length2 :: proc{vector_length2, quaternion_length2}
  134. @(require_results)
  135. clamp_length :: proc "contextless" (v: $T/[$N]$E, a: E) -> T where IS_FLOAT(E) {
  136. if a <= 0 {
  137. return 0
  138. }
  139. m2 := length2(v)
  140. return v if (m2 <= a*a) else (v / sqrt(m2) * a) // returns original when m2 is 0
  141. }
  142. @(require_results)
  143. projection :: proc "contextless" (x, normal: $T/[$N]$E) -> T where IS_NUMERIC(E) {
  144. return dot(x, normal) / dot(normal, normal) * normal
  145. }
  146. @(require_results)
  147. identity_array_based_matrix :: proc "contextless" ($T: typeid/[$N][N]$E) -> (m: T) #no_bounds_check {
  148. for i in 0..<N {
  149. m[i][i] = E(1)
  150. }
  151. return m
  152. }
  153. @(require_results)
  154. identity_matrix :: proc "contextless" ($T: typeid/matrix[$N, N]$E) -> T #no_bounds_check {
  155. return 1
  156. }
  157. identity :: proc{
  158. identity_array_based_matrix,
  159. identity_matrix,
  160. }
  161. transpose :: intrinsics.transpose
  162. @(require_results)
  163. matrix_mul :: proc "contextless" (a, b: $M/matrix[$N, N]$E) -> (c: M)
  164. where !IS_ARRAY(E), IS_NUMERIC(E) #no_bounds_check {
  165. return a * b
  166. }
  167. @(require_results)
  168. matrix_comp_mul :: proc "contextless" (a, b: $M/matrix[$I, $J]$E) -> (c: M)
  169. where !IS_ARRAY(E), IS_NUMERIC(E) #no_bounds_check {
  170. return hadamard_product(a, b)
  171. }
  172. @(require_results)
  173. matrix_mul_differ :: proc "contextless" (a: $A/matrix[$I, $J]$E, b: $B/matrix[J, $K]E) -> (c: matrix[I, K]E)
  174. where !IS_ARRAY(E), IS_NUMERIC(E), I != K #no_bounds_check {
  175. return a * b
  176. }
  177. @(require_results)
  178. matrix_mul_vector :: proc "contextless" (a: $A/matrix[$I, $J]$E, b: $B/[J]E) -> (c: B)
  179. where !IS_ARRAY(E), IS_NUMERIC(E) #no_bounds_check {
  180. return a * b
  181. }
  182. @(require_results)
  183. quaternion_mul_quaternion :: proc "contextless" (q1, q2: $Q) -> Q where IS_QUATERNION(Q) {
  184. return q1 * q2
  185. }
  186. @(require_results)
  187. quaternion64_mul_vector3 :: proc "contextless" (q: $Q/quaternion64, v: $V/[3]$F/f16) -> V {
  188. q := transmute(runtime.Raw_Quaternion64_Vector_Scalar)q
  189. v := v
  190. t := cross(2*q.vector, v)
  191. return V(v + q.scalar*t + cross(q.vector, t))
  192. }
  193. @(require_results)
  194. quaternion128_mul_vector3 :: proc "contextless" (q: $Q/quaternion128, v: $V/[3]$F/f32) -> V {
  195. q := transmute(runtime.Raw_Quaternion128_Vector_Scalar)q
  196. v := v
  197. t := cross(2*q.vector, v)
  198. return V(v + q.scalar*t + cross(q.vector, t))
  199. }
  200. @(require_results)
  201. quaternion256_mul_vector3 :: proc "contextless" (q: $Q/quaternion256, v: $V/[3]$F/f64) -> V {
  202. q := transmute(runtime.Raw_Quaternion256_Vector_Scalar)q
  203. v := v
  204. t := cross(2*q.vector, v)
  205. return V(v + q.scalar*t + cross(q.vector, t))
  206. }
  207. quaternion_mul_vector3 :: proc{quaternion64_mul_vector3, quaternion128_mul_vector3, quaternion256_mul_vector3}
  208. mul :: proc{
  209. matrix_mul,
  210. matrix_mul_differ,
  211. matrix_mul_vector,
  212. quaternion64_mul_vector3,
  213. quaternion128_mul_vector3,
  214. quaternion256_mul_vector3,
  215. quaternion_mul_quaternion,
  216. }
  217. @(require_results)
  218. vector_to_ptr :: proc "contextless" (v: ^$V/[$N]$E) -> ^E where IS_NUMERIC(E), N > 0 #no_bounds_check {
  219. return &v[0]
  220. }
  221. @(require_results)
  222. matrix_to_ptr :: proc "contextless" (m: ^$A/matrix[$I, $J]$E) -> ^E where IS_NUMERIC(E), I > 0, J > 0 #no_bounds_check {
  223. return &m[0, 0]
  224. }
  225. to_ptr :: proc{vector_to_ptr, matrix_to_ptr}
  226. vector_angle_between :: proc "contextless" (a, b: $V/[$N]$E) -> E {
  227. a0 := normalize0(a)
  228. b0 := normalize0(b)
  229. d := clamp(dot(a0, b0), -1, +1)
  230. return math.acos(d)
  231. }
  232. quaternion64_angle_between :: proc "contextless" (a, b: $Q/quaternion64) -> f16 {
  233. c := normalize0(conj(a) * b)
  234. return math.acos(c.w)
  235. }
  236. quaternion128_angle_between :: proc "contextless" (a, b: $Q/quaternion128) -> f32 {
  237. c := normalize0(conj(a) * b)
  238. return math.acos(c.w)
  239. }
  240. quaternion256_angle_between :: proc "contextless" (a, b: $Q/quaternion256) -> f64 {
  241. c := normalize0(conj(a) * b)
  242. return math.acos(c.w)
  243. }
  244. angle_between :: proc{
  245. vector_angle_between,
  246. quaternion64_angle_between,
  247. quaternion128_angle_between,
  248. quaternion256_angle_between,
  249. }
  250. // Splines
  251. @(require_results)
  252. vector_slerp :: proc "contextless" (x, y: $T/[$N]$E, a: E) -> T #no_bounds_check {
  253. cos_alpha := dot(x, y)
  254. alpha := math.acos(cos_alpha)
  255. sin_alpha := math.sin(alpha)
  256. t1 := math.sin((1 - a) * alpha) / sin_alpha
  257. t2 := math.sin(a * alpha) / sin_alpha
  258. return x * t1 + y * t2
  259. }
  260. @(require_results)
  261. catmull_rom :: proc "contextless" (v1, v2, v3, v4: $T/[$N]$E, s: E) -> T #no_bounds_check {
  262. s2 := s*s
  263. s3 := s2*s
  264. f1 := -s3 + 2 * s2 - s
  265. f2 := 3 * s3 - 5 * s2 + 2
  266. f3 := -3 * s3 + 4 * s2 + s
  267. f4 := s3 - s2
  268. return (f1 * v1 + f2 * v2 + f3 * v3 + f4 * v4) * 0.5
  269. }
  270. @(require_results)
  271. hermite :: proc "contextless" (v1, t1, v2, t2: $T/[$N]$E, s: E) -> T #no_bounds_check {
  272. s2 := s*s
  273. s3 := s2*s
  274. f1 := 2 * s3 - 3 * s2 + 1
  275. f2 := -2 * s3 + 3 * s2
  276. f3 := s3 - 2 * s2 + s
  277. f4 := s3 - s2
  278. return f1 * v1 + f2 * v2 + f3 * t1 + f4 * t2
  279. }
  280. @(require_results)
  281. cubic :: proc "contextless" (v1, v2, v3, v4: $T/[$N]$E, s: E) -> T #no_bounds_check {
  282. return ((v1 * s + v2) * s + v3) * s + v4
  283. }
  284. @(require_results)
  285. array_cast :: proc "contextless" (v: $A/[$N]$T, $Elem_Type: typeid) -> (w: [N]Elem_Type) #no_bounds_check {
  286. for i in 0..<N {
  287. w[i] = Elem_Type(v[i])
  288. }
  289. return
  290. }
  291. @(require_results)
  292. matrix_cast :: proc "contextless" (v: $A/matrix[$M, $N]$T, $Elem_Type: typeid) -> (w: matrix[M, N]Elem_Type) #no_bounds_check {
  293. for j in 0..<N {
  294. for i in 0..<M {
  295. w[i, j] = Elem_Type(v[i, j])
  296. }
  297. }
  298. return
  299. }
  300. @(require_results) to_f32 :: #force_inline proc(v: $A/[$N]$T) -> [N]f32 { return array_cast(v, f32) }
  301. @(require_results) to_f64 :: #force_inline proc(v: $A/[$N]$T) -> [N]f64 { return array_cast(v, f64) }
  302. @(require_results) to_i8 :: #force_inline proc(v: $A/[$N]$T) -> [N]i8 { return array_cast(v, i8) }
  303. @(require_results) to_i16 :: #force_inline proc(v: $A/[$N]$T) -> [N]i16 { return array_cast(v, i16) }
  304. @(require_results) to_i32 :: #force_inline proc(v: $A/[$N]$T) -> [N]i32 { return array_cast(v, i32) }
  305. @(require_results) to_i64 :: #force_inline proc(v: $A/[$N]$T) -> [N]i64 { return array_cast(v, i64) }
  306. @(require_results) to_int :: #force_inline proc(v: $A/[$N]$T) -> [N]int { return array_cast(v, int) }
  307. @(require_results) to_u8 :: #force_inline proc(v: $A/[$N]$T) -> [N]u8 { return array_cast(v, u8) }
  308. @(require_results) to_u16 :: #force_inline proc(v: $A/[$N]$T) -> [N]u16 { return array_cast(v, u16) }
  309. @(require_results) to_u32 :: #force_inline proc(v: $A/[$N]$T) -> [N]u32 { return array_cast(v, u32) }
  310. @(require_results) to_u64 :: #force_inline proc(v: $A/[$N]$T) -> [N]u64 { return array_cast(v, u64) }
  311. @(require_results) to_uint :: #force_inline proc(v: $A/[$N]$T) -> [N]uint { return array_cast(v, uint) }
  312. @(require_results) to_complex32 :: #force_inline proc(v: $A/[$N]$T) -> [N]complex32 { return array_cast(v, complex32) }
  313. @(require_results) to_complex64 :: #force_inline proc(v: $A/[$N]$T) -> [N]complex64 { return array_cast(v, complex64) }
  314. @(require_results) to_complex128 :: #force_inline proc(v: $A/[$N]$T) -> [N]complex128 { return array_cast(v, complex128) }
  315. @(require_results) to_quaternion64 :: #force_inline proc(v: $A/[$N]$T) -> [N]quaternion64 { return array_cast(v, quaternion64) }
  316. @(require_results) to_quaternion128 :: #force_inline proc(v: $A/[$N]$T) -> [N]quaternion128 { return array_cast(v, quaternion128) }
  317. @(require_results) to_quaternion256 :: #force_inline proc(v: $A/[$N]$T) -> [N]quaternion256 { return array_cast(v, quaternion256) }
  318. hadamard_product :: intrinsics.hadamard_product
  319. matrix_flatten :: intrinsics.matrix_flatten
  320. determinant :: proc{
  321. matrix1x1_determinant,
  322. matrix2x2_determinant,
  323. matrix3x3_determinant,
  324. matrix4x4_determinant,
  325. }
  326. adjugate :: proc{
  327. matrix1x1_adjugate,
  328. matrix2x2_adjugate,
  329. matrix3x3_adjugate,
  330. matrix4x4_adjugate,
  331. }
  332. inverse_transpose :: proc{
  333. matrix1x1_inverse_transpose,
  334. matrix2x2_inverse_transpose,
  335. matrix3x3_inverse_transpose,
  336. matrix4x4_inverse_transpose,
  337. }
  338. inverse :: proc{
  339. matrix1x1_inverse,
  340. matrix2x2_inverse,
  341. matrix3x3_inverse,
  342. matrix4x4_inverse,
  343. }
  344. @(require_results)
  345. hermitian_adjoint :: proc "contextless" (m: $M/matrix[$N, N]$T) -> M where intrinsics.type_is_complex(T), N >= 1 #no_bounds_check {
  346. return conj(transpose(m))
  347. }
  348. @(require_results)
  349. trace :: proc "contextless" (m: $M/matrix[$N, N]$T) -> (trace: T) #no_bounds_check {
  350. for i in 0..<N {
  351. trace += m[i, i]
  352. }
  353. return
  354. }
  355. @(require_results)
  356. matrix_minor :: proc "contextless" (m: $M/matrix[$N, N]$T, #any_int row, column: int) -> (minor: T) where N > 1 #no_bounds_check {
  357. K :: int(N-1)
  358. cut_down: matrix[K, K]T
  359. for col_idx in 0..<K {
  360. j := col_idx + int(col_idx >= column)
  361. for row_idx in 0..<K {
  362. i := row_idx + int(row_idx >= row)
  363. cut_down[row_idx, col_idx] = m[i, j]
  364. }
  365. }
  366. return determinant(cut_down)
  367. }
  368. @(require_results)
  369. matrix1x1_determinant :: proc "contextless" (m: $M/matrix[1, 1]$T) -> (det: T) #no_bounds_check {
  370. return m[0, 0]
  371. }
  372. @(require_results)
  373. matrix2x2_determinant :: proc "contextless" (m: $M/matrix[2, 2]$T) -> (det: T) #no_bounds_check {
  374. return m[0, 0]*m[1, 1] - m[0, 1]*m[1, 0]
  375. }
  376. @(require_results)
  377. matrix3x3_determinant :: proc "contextless" (m: $M/matrix[3, 3]$T) -> (det: T) #no_bounds_check {
  378. a := +m[0, 0] * (m[1, 1] * m[2, 2] - m[1, 2] * m[2, 1])
  379. b := -m[0, 1] * (m[1, 0] * m[2, 2] - m[1, 2] * m[2, 0])
  380. c := +m[0, 2] * (m[1, 0] * m[2, 1] - m[1, 1] * m[2, 0])
  381. return a + b + c
  382. }
  383. @(require_results)
  384. matrix4x4_determinant :: proc "contextless" (m: $M/matrix[4, 4]$T) -> (det: T) #no_bounds_check {
  385. a := adjugate(m)
  386. for i in 0..<4 {
  387. det += m[0, i] * a[0, i]
  388. }
  389. return
  390. }
  391. @(require_results)
  392. matrix1x1_adjugate :: proc "contextless" (x: $M/matrix[1, 1]$T) -> (y: M) #no_bounds_check {
  393. y = x
  394. return
  395. }
  396. @(require_results)
  397. matrix2x2_adjugate :: proc "contextless" (x: $M/matrix[2, 2]$T) -> (y: M) #no_bounds_check {
  398. y[0, 0] = +x[1, 1]
  399. y[0, 1] = -x[1, 0]
  400. y[1, 0] = -x[0, 1]
  401. y[1, 1] = +x[0, 0]
  402. return
  403. }
  404. @(require_results)
  405. matrix3x3_adjugate :: proc "contextless" (m: $M/matrix[3, 3]$T) -> (y: M) #no_bounds_check {
  406. y[0, 0] = +(m[1, 1] * m[2, 2] - m[2, 1] * m[1, 2])
  407. y[0, 1] = -(m[1, 0] * m[2, 2] - m[2, 0] * m[1, 2])
  408. y[0, 2] = +(m[1, 0] * m[2, 1] - m[2, 0] * m[1, 1])
  409. y[1, 0] = -(m[0, 1] * m[2, 2] - m[2, 1] * m[0, 2])
  410. y[1, 1] = +(m[0, 0] * m[2, 2] - m[2, 0] * m[0, 2])
  411. y[1, 2] = -(m[0, 0] * m[2, 1] - m[2, 0] * m[0, 1])
  412. y[2, 0] = +(m[0, 1] * m[1, 2] - m[1, 1] * m[0, 2])
  413. y[2, 1] = -(m[0, 0] * m[1, 2] - m[1, 0] * m[0, 2])
  414. y[2, 2] = +(m[0, 0] * m[1, 1] - m[1, 0] * m[0, 1])
  415. return
  416. }
  417. @(require_results)
  418. matrix4x4_adjugate :: proc "contextless" (x: $M/matrix[4, 4]$T) -> (y: M) #no_bounds_check {
  419. for i in 0..<4 {
  420. for j in 0..<4 {
  421. sign: T = 1 if (i + j) % 2 == 0 else -1
  422. y[i, j] = sign * matrix_minor(x, i, j)
  423. }
  424. }
  425. return
  426. }
  427. @(require_results)
  428. matrix1x1_inverse_transpose :: proc "contextless" (x: $M/matrix[1, 1]$T) -> (y: M) #no_bounds_check {
  429. y[0, 0] = 1/x[0, 0]
  430. return
  431. }
  432. @(require_results)
  433. matrix2x2_inverse_transpose :: proc "contextless" (x: $M/matrix[2, 2]$T) -> (y: M) #no_bounds_check {
  434. d := x[0, 0]*x[1, 1] - x[0, 1]*x[1, 0]
  435. when intrinsics.type_is_integer(T) {
  436. y[0, 0] = +x[1, 1] / d
  437. y[1, 0] = -x[0, 1] / d
  438. y[0, 1] = -x[1, 0] / d
  439. y[1, 1] = +x[0, 0] / d
  440. } else {
  441. id := 1 / d
  442. y[0, 0] = +x[1, 1] * id
  443. y[1, 0] = -x[0, 1] * id
  444. y[0, 1] = -x[1, 0] * id
  445. y[1, 1] = +x[0, 0] * id
  446. }
  447. return
  448. }
  449. @(require_results)
  450. matrix3x3_inverse_transpose :: proc "contextless" (x: $M/matrix[3, 3]$T) -> (y: M) #no_bounds_check {
  451. a := adjugate(x)
  452. d := determinant(x)
  453. when intrinsics.type_is_integer(T) {
  454. for i in 0..<3 {
  455. for j in 0..<3 {
  456. y[i, j] = a[i, j] / d
  457. }
  458. }
  459. } else {
  460. id := 1/d
  461. for i in 0..<3 {
  462. for j in 0..<3 {
  463. y[i, j] = a[i, j] * id
  464. }
  465. }
  466. }
  467. return
  468. }
  469. @(require_results)
  470. matrix4x4_inverse_transpose :: proc "contextless" (x: $M/matrix[4, 4]$T) -> (y: M) #no_bounds_check {
  471. a := adjugate(x)
  472. d: T
  473. for i in 0..<4 {
  474. d += x[0, i] * a[0, i]
  475. }
  476. when intrinsics.type_is_integer(T) {
  477. for i in 0..<4 {
  478. for j in 0..<4 {
  479. y[i, j] = a[i, j] / d
  480. }
  481. }
  482. } else {
  483. id := 1/d
  484. for i in 0..<4 {
  485. for j in 0..<4 {
  486. y[i, j] = a[i, j] * id
  487. }
  488. }
  489. }
  490. return
  491. }
  492. @(require_results)
  493. matrix1x1_inverse :: proc "contextless" (x: $M/matrix[1, 1]$T) -> (y: M) #no_bounds_check {
  494. y[0, 0] = 1/x[0, 0]
  495. return
  496. }
  497. @(require_results)
  498. matrix2x2_inverse :: proc "contextless" (x: $M/matrix[2, 2]$T) -> (y: M) #no_bounds_check {
  499. d := x[0, 0]*x[1, 1] - x[0, 1]*x[1, 0]
  500. when intrinsics.type_is_integer(T) {
  501. y[0, 0] = +x[1, 1] / d
  502. y[0, 1] = -x[0, 1] / d
  503. y[1, 0] = -x[1, 0] / d
  504. y[1, 1] = +x[0, 0] / d
  505. } else {
  506. id := 1 / d
  507. y[0, 0] = +x[1, 1] * id
  508. y[0, 1] = -x[0, 1] * id
  509. y[1, 0] = -x[1, 0] * id
  510. y[1, 1] = +x[0, 0] * id
  511. }
  512. return
  513. }
  514. @(require_results)
  515. matrix3x3_inverse :: proc "contextless" (x: $M/matrix[3, 3]$T) -> (y: M) #no_bounds_check {
  516. a := adjugate(x)
  517. d := determinant(x)
  518. when intrinsics.type_is_integer(T) {
  519. for i in 0..<3 {
  520. for j in 0..<3 {
  521. y[i, j] = a[j, i] / d
  522. }
  523. }
  524. } else {
  525. id := 1/d
  526. for i in 0..<3 {
  527. for j in 0..<3 {
  528. y[i, j] = a[j, i] * id
  529. }
  530. }
  531. }
  532. return
  533. }
  534. @(require_results)
  535. matrix4x4_inverse :: proc "contextless" (x: $M/matrix[4, 4]$T) -> (y: M) #no_bounds_check {
  536. a := adjugate(x)
  537. d: T
  538. for i in 0..<4 {
  539. d += x[0, i] * a[0, i]
  540. }
  541. when intrinsics.type_is_integer(T) {
  542. for i in 0..<4 {
  543. for j in 0..<4 {
  544. y[i, j] = a[j, i] / d
  545. }
  546. }
  547. } else {
  548. id := 1/d
  549. for i in 0..<4 {
  550. for j in 0..<4 {
  551. y[i, j] = a[j, i] * id
  552. }
  553. }
  554. }
  555. return
  556. }