games/semicongine: src/zamikongine/math/matrix.nim annotate

annotate src/zamikongine/math/matrix.nim @ 38:c3c963e7c1a6

did: tons of stuff, input, refactoring, fix some errors, some template improvment, sorry for super-commit

author	Sam <sam@basx.dev>
date	Wed, 18 Jan 2023 09:52:03 +0700
parents	71bbe11d8de8
children

rev	line source
16 617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	1 import std/math
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	2 import std/macros
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	3 import std/random
13 a571db114152 add: vector/matrix modules, tests, nim config script Sam <sam@basx.dev> parents: diff changeset	4 import std/strutils
a571db114152 add: vector/matrix modules, tests, nim config script Sam <sam@basx.dev> parents: diff changeset	5 import std/typetraits
a571db114152 add: vector/matrix modules, tests, nim config script Sam <sam@basx.dev> parents: diff changeset	6
a571db114152 add: vector/matrix modules, tests, nim config script Sam <sam@basx.dev> parents: diff changeset	7 import ./vector
a571db114152 add: vector/matrix modules, tests, nim config script Sam <sam@basx.dev> parents: diff changeset	8
38 c3c963e7c1a6 did: tons of stuff, input, refactoring, fix some errors, some template improvment, sorry for super-commit Sam <sam@basx.dev> parents: 24 diff changeset	9 export math
c3c963e7c1a6 did: tons of stuff, input, refactoring, fix some errors, some template improvment, sorry for super-commit Sam <sam@basx.dev> parents: 24 diff changeset	10
13 a571db114152 add: vector/matrix modules, tests, nim config script Sam <sam@basx.dev> parents: diff changeset	11 type
a571db114152 add: vector/matrix modules, tests, nim config script Sam <sam@basx.dev> parents: diff changeset	12 # layout is row-first
a571db114152 add: vector/matrix modules, tests, nim config script Sam <sam@basx.dev> parents: diff changeset	13 # having an object instead of directly aliasing the array seems a bit ugly at
a571db114152 add: vector/matrix modules, tests, nim config script Sam <sam@basx.dev> parents: diff changeset	14 # first, but is necessary to be able to work correctly with distinguished
a571db114152 add: vector/matrix modules, tests, nim config script Sam <sam@basx.dev> parents: diff changeset	15 # types (i.e. Mat23 and Mat32 would be an alias for the same type array[6, T]
a571db114152 add: vector/matrix modules, tests, nim config script Sam <sam@basx.dev> parents: diff changeset	16 # which prevents the type system from identifying the correct type at times)
a571db114152 add: vector/matrix modules, tests, nim config script Sam <sam@basx.dev> parents: diff changeset	17 #
a571db114152 add: vector/matrix modules, tests, nim config script Sam <sam@basx.dev> parents: diff changeset	18 # Though, great news is that objects have zero overhead!
a571db114152 add: vector/matrix modules, tests, nim config script Sam <sam@basx.dev> parents: diff changeset	19 Mat22*[T: SomeNumber] = object
38 c3c963e7c1a6 did: tons of stuff, input, refactoring, fix some errors, some template improvment, sorry for super-commit Sam <sam@basx.dev> parents: 24 diff changeset	20 data*: array[4, T]
13 a571db114152 add: vector/matrix modules, tests, nim config script Sam <sam@basx.dev> parents: diff changeset	21 Mat23*[T: SomeNumber] = object
38 c3c963e7c1a6 did: tons of stuff, input, refactoring, fix some errors, some template improvment, sorry for super-commit Sam <sam@basx.dev> parents: 24 diff changeset	22 data*: array[6, T]
13 a571db114152 add: vector/matrix modules, tests, nim config script Sam <sam@basx.dev> parents: diff changeset	23 Mat32*[T: SomeNumber] = object
38 c3c963e7c1a6 did: tons of stuff, input, refactoring, fix some errors, some template improvment, sorry for super-commit Sam <sam@basx.dev> parents: 24 diff changeset	24 data*: array[6, T]
13 a571db114152 add: vector/matrix modules, tests, nim config script Sam <sam@basx.dev> parents: diff changeset	25 Mat33*[T: SomeNumber] = object
38 c3c963e7c1a6 did: tons of stuff, input, refactoring, fix some errors, some template improvment, sorry for super-commit Sam <sam@basx.dev> parents: 24 diff changeset	26 data*: array[9, T]
13 a571db114152 add: vector/matrix modules, tests, nim config script Sam <sam@basx.dev> parents: diff changeset	27 Mat34*[T: SomeNumber] = object
38 c3c963e7c1a6 did: tons of stuff, input, refactoring, fix some errors, some template improvment, sorry for super-commit Sam <sam@basx.dev> parents: 24 diff changeset	28 data*: array[12, T]
13 a571db114152 add: vector/matrix modules, tests, nim config script Sam <sam@basx.dev> parents: diff changeset	29 Mat43*[T: SomeNumber] = object
38 c3c963e7c1a6 did: tons of stuff, input, refactoring, fix some errors, some template improvment, sorry for super-commit Sam <sam@basx.dev> parents: 24 diff changeset	30 data*: array[12, T]
13 a571db114152 add: vector/matrix modules, tests, nim config script Sam <sam@basx.dev> parents: diff changeset	31 Mat44*[T: SomeNumber] = object
38 c3c963e7c1a6 did: tons of stuff, input, refactoring, fix some errors, some template improvment, sorry for super-commit Sam <sam@basx.dev> parents: 24 diff changeset	32 data*: array[16, T]
13 a571db114152 add: vector/matrix modules, tests, nim config script Sam <sam@basx.dev> parents: diff changeset	33 MatMM* = Mat22\|Mat33\|Mat44
a571db114152 add: vector/matrix modules, tests, nim config script Sam <sam@basx.dev> parents: diff changeset	34 MatMN* = Mat23\|Mat32\|Mat34\|Mat43
a571db114152 add: vector/matrix modules, tests, nim config script Sam <sam@basx.dev> parents: diff changeset	35 Mat* = MatMM\|MatMN
a571db114152 add: vector/matrix modules, tests, nim config script Sam <sam@basx.dev> parents: diff changeset	36
16 617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	37 func unit22[T: SomeNumber](): auto {.compiletime.} = Mat22[T](data:[
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	38 T(1), T(0),
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	39 T(0), T(1),
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	40 ])
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	41 func unit33[T: SomeNumber](): auto {.compiletime.} = Mat33[T](data:[
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	42 T(1), T(0), T(0),
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	43 T(0), T(1), T(0),
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	44 T(0), T(0), T(1),
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	45 ])
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	46 func unit44[T: SomeNumber](): auto {.compiletime.} = Mat44[T](data: [
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	47 T(1), T(0), T(0), T(0),
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	48 T(0), T(1), T(0), T(0),
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	49 T(0), T(0), T(1), T(0),
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	50 T(0), T(0), T(0), T(1),
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	51 ])
13 a571db114152 add: vector/matrix modules, tests, nim config script Sam <sam@basx.dev> parents: diff changeset	52
16 617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	53 # generates constants: Unit
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	54 # Also for Y, Z, R, G, B
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	55 # not sure if this is necessary or even a good idea...
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	56 macro generateAllConsts() =
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	57 result = newStmtList()
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	58 for theType in ["int", "int8", "int16", "int32", "int64", "float", "float32", "float64"]:
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	59 var typename = theType[0 .. 0]
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	60 if theType[^2].isDigit:
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	61 typename = typename & theType[^2]
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	62 if theType[^1].isDigit:
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	63 typename = typename & theType[^1]
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	64 result.add(newConstStmt(
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	65 postfix(ident("Unit22" & typename), "*"),
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	66 newCall(nnkBracketExpr.newTree(ident("unit22"), ident(theType)))
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	67 ))
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	68 result.add(newConstStmt(
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	69 postfix(ident("Unit33" & typename), "*"),
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	70 newCall(nnkBracketExpr.newTree(ident("unit33"), ident(theType)))
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	71 ))
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	72 result.add(newConstStmt(
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	73 postfix(ident("Unit44" & typename), "*"),
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	74 newCall(nnkBracketExpr.newTree(ident("unit44"), ident(theType)))
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	75 ))
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	76
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	77 generateAllConsts()
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	78
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	79 const Unit22* = unit22[float]()
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	80 const Unit33* = unit33[float]()
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	81 const Unit44* = unit44[float]()
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	82
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	83 template rowCount*(m: typedesc): int =
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	84 when m is Mat22: 2
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	85 elif m is Mat23: 2
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	86 elif m is Mat32: 3
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	87 elif m is Mat33: 3
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	88 elif m is Mat34: 3
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	89 elif m is Mat43: 4
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	90 elif m is Mat44: 4
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	91 template columnCount*(m: typedesc): int =
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	92 when m is Mat22: 2
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	93 elif m is Mat23: 3
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	94 elif m is Mat32: 2
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	95 elif m is Mat33: 3
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	96 elif m is Mat34: 4
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	97 elif m is Mat43: 3
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	98 elif m is Mat44: 4
13 a571db114152 add: vector/matrix modules, tests, nim config script Sam <sam@basx.dev> parents: diff changeset	99
a571db114152 add: vector/matrix modules, tests, nim config script Sam <sam@basx.dev> parents: diff changeset	100
a571db114152 add: vector/matrix modules, tests, nim config script Sam <sam@basx.dev> parents: diff changeset	101 func toString[T](value: T): string =
a571db114152 add: vector/matrix modules, tests, nim config script Sam <sam@basx.dev> parents: diff changeset	102 var
a571db114152 add: vector/matrix modules, tests, nim config script Sam <sam@basx.dev> parents: diff changeset	103 strvalues: seq[string]
a571db114152 add: vector/matrix modules, tests, nim config script Sam <sam@basx.dev> parents: diff changeset	104 maxwidth = 0
a571db114152 add: vector/matrix modules, tests, nim config script Sam <sam@basx.dev> parents: diff changeset	105
a571db114152 add: vector/matrix modules, tests, nim config script Sam <sam@basx.dev> parents: diff changeset	106 for n in value.data:
a571db114152 add: vector/matrix modules, tests, nim config script Sam <sam@basx.dev> parents: diff changeset	107 let strval = $n
a571db114152 add: vector/matrix modules, tests, nim config script Sam <sam@basx.dev> parents: diff changeset	108 strvalues.add(strval)
a571db114152 add: vector/matrix modules, tests, nim config script Sam <sam@basx.dev> parents: diff changeset	109 if strval.len > maxwidth:
a571db114152 add: vector/matrix modules, tests, nim config script Sam <sam@basx.dev> parents: diff changeset	110 maxwidth = strval.len
a571db114152 add: vector/matrix modules, tests, nim config script Sam <sam@basx.dev> parents: diff changeset	111
a571db114152 add: vector/matrix modules, tests, nim config script Sam <sam@basx.dev> parents: diff changeset	112 for i in 0 ..< strvalues.len:
a571db114152 add: vector/matrix modules, tests, nim config script Sam <sam@basx.dev> parents: diff changeset	113 let filler = " ".repeat(maxwidth - strvalues[i].len)
16 617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	114 if i mod T.columnCount == T.columnCount - 1:
13 a571db114152 add: vector/matrix modules, tests, nim config script Sam <sam@basx.dev> parents: diff changeset	115 result &= filler & strvalues[i] & "\n"
a571db114152 add: vector/matrix modules, tests, nim config script Sam <sam@basx.dev> parents: diff changeset	116 else:
16 617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	117 if i mod T.columnCount == 0:
13 a571db114152 add: vector/matrix modules, tests, nim config script Sam <sam@basx.dev> parents: diff changeset	118 result &= " "
a571db114152 add: vector/matrix modules, tests, nim config script Sam <sam@basx.dev> parents: diff changeset	119 result &= filler & strvalues[i] & " "
a571db114152 add: vector/matrix modules, tests, nim config script Sam <sam@basx.dev> parents: diff changeset	120 result = $T & "\n" & result
a571db114152 add: vector/matrix modules, tests, nim config script Sam <sam@basx.dev> parents: diff changeset	121
a571db114152 add: vector/matrix modules, tests, nim config script Sam <sam@basx.dev> parents: diff changeset	122 func `$`*(v: Mat22[SomeNumber]): string = toString[Mat22[SomeNumber]](v)
a571db114152 add: vector/matrix modules, tests, nim config script Sam <sam@basx.dev> parents: diff changeset	123 func `$`*(v: Mat23[SomeNumber]): string = toString[Mat23[SomeNumber]](v)
a571db114152 add: vector/matrix modules, tests, nim config script Sam <sam@basx.dev> parents: diff changeset	124 func `$`*(v: Mat32[SomeNumber]): string = toString[Mat32[SomeNumber]](v)
a571db114152 add: vector/matrix modules, tests, nim config script Sam <sam@basx.dev> parents: diff changeset	125 func `$`*(v: Mat33[SomeNumber]): string = toString[Mat33[SomeNumber]](v)
a571db114152 add: vector/matrix modules, tests, nim config script Sam <sam@basx.dev> parents: diff changeset	126 func `$`*(v: Mat34[SomeNumber]): string = toString[Mat34[SomeNumber]](v)
a571db114152 add: vector/matrix modules, tests, nim config script Sam <sam@basx.dev> parents: diff changeset	127 func `$`*(v: Mat43[SomeNumber]): string = toString[Mat43[SomeNumber]](v)
a571db114152 add: vector/matrix modules, tests, nim config script Sam <sam@basx.dev> parents: diff changeset	128 func `$`*(v: Mat44[SomeNumber]): string = toString[Mat44[SomeNumber]](v)
a571db114152 add: vector/matrix modules, tests, nim config script Sam <sam@basx.dev> parents: diff changeset	129
16 617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	130 func `[]`[T: Mat](m: T, row, col: int): auto = m.data[col + row T.columnCount]
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	131 proc `[]=`[T: Mat, U](m: var T, row, col: int, value: U) = m.data[col + row T.columnCount] = value
13 a571db114152 add: vector/matrix modules, tests, nim config script Sam <sam@basx.dev> parents: diff changeset	132
a571db114152 add: vector/matrix modules, tests, nim config script Sam <sam@basx.dev> parents: diff changeset	133 func row*[T: Mat22](m: T, i: 0..1): auto = Vec2([m[i, 0], m[i, 1]])
a571db114152 add: vector/matrix modules, tests, nim config script Sam <sam@basx.dev> parents: diff changeset	134 func row*[T: Mat32](m: T, i: 0..2): auto = Vec2([m[i, 0], m[i, 1]])
a571db114152 add: vector/matrix modules, tests, nim config script Sam <sam@basx.dev> parents: diff changeset	135 func row*[T: Mat23](m: T, i: 0..1): auto = Vec3([m[i, 0], m[i, 1], m[i, 2]])
a571db114152 add: vector/matrix modules, tests, nim config script Sam <sam@basx.dev> parents: diff changeset	136 func row*[T: Mat33](m: T, i: 0..2): auto = Vec3([m[i, 0], m[i, 1], m[i, 2]])
a571db114152 add: vector/matrix modules, tests, nim config script Sam <sam@basx.dev> parents: diff changeset	137 func row*[T: Mat43](m: T, i: 0..3): auto = Vec3([m[i, 0], m[i, 1], m[i, 2]])
a571db114152 add: vector/matrix modules, tests, nim config script Sam <sam@basx.dev> parents: diff changeset	138 func row*[T: Mat34](m: T, i: 0..2): auto = Vec4([m[i, 0], m[i, 1], m[i, 2], m[i, 3]])
a571db114152 add: vector/matrix modules, tests, nim config script Sam <sam@basx.dev> parents: diff changeset	139 func row*[T: Mat44](m: T, i: 0..3): auto = Vec4([m[i, 0], m[i, 1], m[i, 2], m[i, 3]])
a571db114152 add: vector/matrix modules, tests, nim config script Sam <sam@basx.dev> parents: diff changeset	140
a571db114152 add: vector/matrix modules, tests, nim config script Sam <sam@basx.dev> parents: diff changeset	141 func col*[T: Mat22](m: T, i: 0..1): auto = Vec2([m[0, i], m[1, i]])
a571db114152 add: vector/matrix modules, tests, nim config script Sam <sam@basx.dev> parents: diff changeset	142 func col*[T: Mat23](m: T, i: 0..2): auto = Vec2([m[0, i], m[1, i]])
a571db114152 add: vector/matrix modules, tests, nim config script Sam <sam@basx.dev> parents: diff changeset	143 func col*[T: Mat32](m: T, i: 0..1): auto = Vec3([m[0, i], m[1, i], m[2, i]])
a571db114152 add: vector/matrix modules, tests, nim config script Sam <sam@basx.dev> parents: diff changeset	144 func col*[T: Mat33](m: T, i: 0..2): auto = Vec3([m[0, i], m[1, i], m[2, i]])
a571db114152 add: vector/matrix modules, tests, nim config script Sam <sam@basx.dev> parents: diff changeset	145 func col*[T: Mat34](m: T, i: 0..3): auto = Vec3([m[0, i], m[1, i], m[2, i]])
a571db114152 add: vector/matrix modules, tests, nim config script Sam <sam@basx.dev> parents: diff changeset	146 func col*[T: Mat43](m: T, i: 0..2): auto = Vec4([m[0, i], m[1, i], m[2, i], m[3, i]])
a571db114152 add: vector/matrix modules, tests, nim config script Sam <sam@basx.dev> parents: diff changeset	147 func col*[T: Mat44](m: T, i: 0..3): auto = Vec4([m[0, i], m[1, i], m[2, i], m[3, i]])
a571db114152 add: vector/matrix modules, tests, nim config script Sam <sam@basx.dev> parents: diff changeset	148
16 617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	149 proc createMatMatMultiplicationOperator(leftType: typedesc, rightType: typedesc, outType: typedesc): NimNode =
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	150 var data = nnkBracket.newTree()
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	151 for i in 0 ..< rowCount(leftType):
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	152 for j in 0 ..< rightType.columnCount:
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	153 data.add(newCall(
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	154 ident("sum"),
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	155 infix(
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	156 newCall(newDotExpr(ident("a"), ident("row")), newLit(i)),
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	157 "*",
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	158 newCall(newDotExpr(ident("b"), ident("col")), newLit(j))
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	159 )
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	160 ))
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	161
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	162 return newProc(
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	163 postfix(nnkAccQuoted.newTree(ident("")), ""),
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	164 params=[
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	165 ident("auto"),
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	166 newIdentDefs(ident("a"), ident(leftType.name)),
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	167 newIdentDefs(ident("b"), ident(rightType.name))
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	168 ],
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	169 body=nnkObjConstr.newTree(ident(outType.name), nnkExprColonExpr.newTree(ident("data"), data)),
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	170 procType=nnkFuncDef,
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	171 )
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	172
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	173 proc createVecMatMultiplicationOperator(matType: typedesc, vecType: typedesc): NimNode =
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	174 var data = nnkBracket.newTree()
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	175 for i in 0 ..< matType.rowCount:
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	176 data.add(newCall(
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	177 ident("sum"),
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	178 infix(
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	179 ident("v"),
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	180 "*",
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	181 newCall(newDotExpr(ident("m"), ident("row")), newLit(i))
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	182 )
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	183 ))
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	184
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	185 let resultVec = newCall(
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	186 nnkBracketExpr.newTree(ident(vecType.name), ident("T")),
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	187 data,
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	188 )
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	189 let name = postfix(nnkAccQuoted.newTree(ident("")), "")
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	190 let genericParams = nnkGenericParams.newTree(nnkIdentDefs.newTree(ident("T"), ident("SomeNumber"), newEmptyNode()))
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	191 let formalParams = nnkFormalParams.newTree(
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	192 ident("auto"),
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	193 newIdentDefs(ident("m"), nnkBracketExpr.newTree(ident(matType.name), ident("T"))),
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	194 newIdentDefs(ident("v"), nnkBracketExpr.newTree(ident(vecType.name), ident("T"))),
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	195 )
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	196
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	197 return nnkFuncDef.newTree(
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	198 name,
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	199 newEmptyNode(),
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	200 genericParams,
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	201 formalParams,
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	202 newEmptyNode(),
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	203 newEmptyNode(),
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	204 resultVec
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	205 )
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	206
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	207
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	208 proc createMatScalarOperator(matType: typedesc, op: string): NimNode =
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	209 result = newStmtList()
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	210
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	211 var data = nnkBracket.newTree()
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	212 for i in 0 ..< matType.rowCount * matType.columnCount:
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	213 data.add(infix(nnkBracketExpr.newTree(newDotExpr(ident("a"), ident("data")), newLit(i)), op, ident("b")))
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	214 result.add(newProc(
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	215 postfix(nnkAccQuoted.newTree(ident(op)), "*"),
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	216 params=[
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	217 ident("auto"),
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	218 newIdentDefs(ident("a"), ident(matType.name)),
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	219 newIdentDefs(ident("b"), ident("SomeNumber")),
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	220 ],
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	221 body=nnkObjConstr.newTree(ident(matType.name), nnkExprColonExpr.newTree(ident("data"), data)),
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	222 procType=nnkFuncDef,
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	223 ))
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	224 result.add(newProc(
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	225 postfix(nnkAccQuoted.newTree(ident(op)), "*"),
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	226 params=[
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	227 ident("auto"),
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	228 newIdentDefs(ident("b"), ident("SomeNumber")),
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	229 newIdentDefs(ident("a"), ident(matType.name)),
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	230 ],
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	231 body=nnkObjConstr.newTree(ident(matType.name), nnkExprColonExpr.newTree(ident("data"), data)),
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	232 procType=nnkFuncDef,
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	233 ))
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	234 if op == "-":
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	235 var data2 = nnkBracket.newTree()
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	236 for i in 0 ..< matType.rowCount * matType.columnCount:
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	237 data2.add(prefix(nnkBracketExpr.newTree(newDotExpr(ident("a"), ident("data")), newLit(i)), op))
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	238 result.add(newProc(
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	239 postfix(nnkAccQuoted.newTree(ident(op)), "*"),
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	240 params=[
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	241 ident("auto"),
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	242 newIdentDefs(ident("a"), ident(matType.name)),
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	243 ],
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	244 body=nnkObjConstr.newTree(ident(matType.name), nnkExprColonExpr.newTree(ident("data"), data2)),
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	245 procType=nnkFuncDef,
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	246 ))
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	247
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	248 macro createAllMultiplicationOperators() =
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	249 result = newStmtList()
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	250
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	251 for op in ["+", "-", "*", "/"]:
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	252 result.add(createMatScalarOperator(Mat22, op))
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	253 result.add(createMatScalarOperator(Mat23, op))
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	254 result.add(createMatScalarOperator(Mat32, op))
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	255 result.add(createMatScalarOperator(Mat33, op))
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	256 result.add(createMatScalarOperator(Mat34, op))
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	257 result.add(createMatScalarOperator(Mat43, op))
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	258 result.add(createMatScalarOperator(Mat44, op))
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	259
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	260 result.add(createMatMatMultiplicationOperator(Mat22, Mat22, Mat22))
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	261 result.add(createMatMatMultiplicationOperator(Mat22, Mat23, Mat23))
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	262 result.add(createMatMatMultiplicationOperator(Mat23, Mat32, Mat22))
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	263 result.add(createMatMatMultiplicationOperator(Mat23, Mat33, Mat23))
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	264 result.add(createMatMatMultiplicationOperator(Mat32, Mat22, Mat32))
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	265 result.add(createMatMatMultiplicationOperator(Mat32, Mat23, Mat33))
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	266 result.add(createMatMatMultiplicationOperator(Mat33, Mat32, Mat32))
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	267 result.add(createMatMatMultiplicationOperator(Mat33, Mat33, Mat33))
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	268 result.add(createMatMatMultiplicationOperator(Mat33, Mat34, Mat34))
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	269 result.add(createMatMatMultiplicationOperator(Mat43, Mat33, Mat43))
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	270 result.add(createMatMatMultiplicationOperator(Mat43, Mat34, Mat44))
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	271 result.add(createMatMatMultiplicationOperator(Mat44, Mat43, Mat43))
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	272 result.add(createMatMatMultiplicationOperator(Mat44, Mat44, Mat44))
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	273
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	274 result.add(createVecMatMultiplicationOperator(Mat22, Vec2))
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	275 result.add(createVecMatMultiplicationOperator(Mat33, Vec3))
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	276 result.add(createVecMatMultiplicationOperator(Mat44, Vec4))
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	277
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	278 createAllMultiplicationOperators()
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	279
617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	280
13 a571db114152 add: vector/matrix modules, tests, nim config script Sam <sam@basx.dev> parents: diff changeset	281 func transposed*[T](m: Mat22[T]): Mat22[T] = Mat22[T](data: [
a571db114152 add: vector/matrix modules, tests, nim config script Sam <sam@basx.dev> parents: diff changeset	282 m[0, 0], m[1, 0],
a571db114152 add: vector/matrix modules, tests, nim config script Sam <sam@basx.dev> parents: diff changeset	283 m[0, 1], m[1, 1],
a571db114152 add: vector/matrix modules, tests, nim config script Sam <sam@basx.dev> parents: diff changeset	284 ])
a571db114152 add: vector/matrix modules, tests, nim config script Sam <sam@basx.dev> parents: diff changeset	285 func transposed*[T](m: Mat23[T]): Mat32[T] = Mat32[T](data: [
a571db114152 add: vector/matrix modules, tests, nim config script Sam <sam@basx.dev> parents: diff changeset	286 m[0, 0], m[1, 0],
a571db114152 add: vector/matrix modules, tests, nim config script Sam <sam@basx.dev> parents: diff changeset	287 m[0, 1], m[1, 1],
a571db114152 add: vector/matrix modules, tests, nim config script Sam <sam@basx.dev> parents: diff changeset	288 m[0, 2], m[1, 2],
a571db114152 add: vector/matrix modules, tests, nim config script Sam <sam@basx.dev> parents: diff changeset	289 ])
a571db114152 add: vector/matrix modules, tests, nim config script Sam <sam@basx.dev> parents: diff changeset	290 func transposed*[T](m: Mat32[T]): Mat23[T] = Mat23[T](data: [
a571db114152 add: vector/matrix modules, tests, nim config script Sam <sam@basx.dev> parents: diff changeset	291 m[0, 0], m[1, 0], m[2, 0],
a571db114152 add: vector/matrix modules, tests, nim config script Sam <sam@basx.dev> parents: diff changeset	292 m[0, 1], m[1, 1], m[2, 1],
a571db114152 add: vector/matrix modules, tests, nim config script Sam <sam@basx.dev> parents: diff changeset	293 ])
a571db114152 add: vector/matrix modules, tests, nim config script Sam <sam@basx.dev> parents: diff changeset	294 func transposed*[T](m: Mat33[T]): Mat33[T] = Mat33[T](data: [
a571db114152 add: vector/matrix modules, tests, nim config script Sam <sam@basx.dev> parents: diff changeset	295 m[0, 0], m[1, 0], m[2, 0],
a571db114152 add: vector/matrix modules, tests, nim config script Sam <sam@basx.dev> parents: diff changeset	296 m[0, 1], m[1, 1], m[2, 1],
a571db114152 add: vector/matrix modules, tests, nim config script Sam <sam@basx.dev> parents: diff changeset	297 m[0, 2], m[1, 2], m[2, 2],
a571db114152 add: vector/matrix modules, tests, nim config script Sam <sam@basx.dev> parents: diff changeset	298 ])
a571db114152 add: vector/matrix modules, tests, nim config script Sam <sam@basx.dev> parents: diff changeset	299 func transposed*[T](m: Mat43[T]): Mat34[T] = Mat34[T](data: [
a571db114152 add: vector/matrix modules, tests, nim config script Sam <sam@basx.dev> parents: diff changeset	300 m[0, 0], m[1, 0], m[2, 0], m[3, 0],
a571db114152 add: vector/matrix modules, tests, nim config script Sam <sam@basx.dev> parents: diff changeset	301 m[0, 1], m[1, 1], m[2, 1], m[3, 1],
a571db114152 add: vector/matrix modules, tests, nim config script Sam <sam@basx.dev> parents: diff changeset	302 m[0, 2], m[1, 2], m[2, 2], m[3, 2],
a571db114152 add: vector/matrix modules, tests, nim config script Sam <sam@basx.dev> parents: diff changeset	303 ])
a571db114152 add: vector/matrix modules, tests, nim config script Sam <sam@basx.dev> parents: diff changeset	304 func transposed*[T](m: Mat34[T]): Mat43[T] = Mat43[T](data: [
a571db114152 add: vector/matrix modules, tests, nim config script Sam <sam@basx.dev> parents: diff changeset	305 m[0, 0], m[1, 0], m[2, 0],
a571db114152 add: vector/matrix modules, tests, nim config script Sam <sam@basx.dev> parents: diff changeset	306 m[0, 1], m[1, 1], m[2, 1],
a571db114152 add: vector/matrix modules, tests, nim config script Sam <sam@basx.dev> parents: diff changeset	307 m[0, 2], m[1, 2], m[2, 2],
a571db114152 add: vector/matrix modules, tests, nim config script Sam <sam@basx.dev> parents: diff changeset	308 m[0, 3], m[1, 3], m[2, 3],
a571db114152 add: vector/matrix modules, tests, nim config script Sam <sam@basx.dev> parents: diff changeset	309 ])
a571db114152 add: vector/matrix modules, tests, nim config script Sam <sam@basx.dev> parents: diff changeset	310 func transposed*[T](m: Mat44[T]): Mat44[T] = Mat44[T](data: [
a571db114152 add: vector/matrix modules, tests, nim config script Sam <sam@basx.dev> parents: diff changeset	311 m[0, 0], m[1, 0], m[2, 0], m[3, 0],
a571db114152 add: vector/matrix modules, tests, nim config script Sam <sam@basx.dev> parents: diff changeset	312 m[0, 1], m[1, 1], m[2, 1], m[3, 1],
a571db114152 add: vector/matrix modules, tests, nim config script Sam <sam@basx.dev> parents: diff changeset	313 m[0, 2], m[1, 2], m[2, 2], m[3, 2],
a571db114152 add: vector/matrix modules, tests, nim config script Sam <sam@basx.dev> parents: diff changeset	314 m[0, 3], m[1, 3], m[2, 3], m[3, 3],
a571db114152 add: vector/matrix modules, tests, nim config script Sam <sam@basx.dev> parents: diff changeset	315 ])
a571db114152 add: vector/matrix modules, tests, nim config script Sam <sam@basx.dev> parents: diff changeset	316
22 b45a5d338cd0 did: improve alotof_triangles example, remove glslang lib, use only binary, easier cross-compilation Sam <sam@basx.dev> parents: 19 diff changeset	317 func translate2d*[T](x, y: T): Mat33[T] = Mat33[T](data: [
b45a5d338cd0 did: improve alotof_triangles example, remove glslang lib, use only binary, easier cross-compilation Sam <sam@basx.dev> parents: 19 diff changeset	318 T(1), T(0), x,
b45a5d338cd0 did: improve alotof_triangles example, remove glslang lib, use only binary, easier cross-compilation Sam <sam@basx.dev> parents: 19 diff changeset	319 T(0), T(1), y,
b45a5d338cd0 did: improve alotof_triangles example, remove glslang lib, use only binary, easier cross-compilation Sam <sam@basx.dev> parents: 19 diff changeset	320 T(0), T(0), T(1),
b45a5d338cd0 did: improve alotof_triangles example, remove glslang lib, use only binary, easier cross-compilation Sam <sam@basx.dev> parents: 19 diff changeset	321 ])
b45a5d338cd0 did: improve alotof_triangles example, remove glslang lib, use only binary, easier cross-compilation Sam <sam@basx.dev> parents: 19 diff changeset	322 func scale2d*[T](sx, sy: T): Mat33[T] = Mat33[T](data: [
b45a5d338cd0 did: improve alotof_triangles example, remove glslang lib, use only binary, easier cross-compilation Sam <sam@basx.dev> parents: 19 diff changeset	323 sx, T(0), T(0),
b45a5d338cd0 did: improve alotof_triangles example, remove glslang lib, use only binary, easier cross-compilation Sam <sam@basx.dev> parents: 19 diff changeset	324 T(0), sy, T(0),
b45a5d338cd0 did: improve alotof_triangles example, remove glslang lib, use only binary, easier cross-compilation Sam <sam@basx.dev> parents: 19 diff changeset	325 T(0), T(0), T(1),
b45a5d338cd0 did: improve alotof_triangles example, remove glslang lib, use only binary, easier cross-compilation Sam <sam@basx.dev> parents: 19 diff changeset	326 ])
b45a5d338cd0 did: improve alotof_triangles example, remove glslang lib, use only binary, easier cross-compilation Sam <sam@basx.dev> parents: 19 diff changeset	327 func rotate2d*[T](angle: T): Mat33[T] = Mat33[T](data: [
b45a5d338cd0 did: improve alotof_triangles example, remove glslang lib, use only binary, easier cross-compilation Sam <sam@basx.dev> parents: 19 diff changeset	328 cos(angle), -sin(angle), T(0),
b45a5d338cd0 did: improve alotof_triangles example, remove glslang lib, use only binary, easier cross-compilation Sam <sam@basx.dev> parents: 19 diff changeset	329 sin(angle), cos(angle), T(0),
b45a5d338cd0 did: improve alotof_triangles example, remove glslang lib, use only binary, easier cross-compilation Sam <sam@basx.dev> parents: 19 diff changeset	330 T(0), T(0), T(1),
b45a5d338cd0 did: improve alotof_triangles example, remove glslang lib, use only binary, easier cross-compilation Sam <sam@basx.dev> parents: 19 diff changeset	331 ])
b45a5d338cd0 did: improve alotof_triangles example, remove glslang lib, use only binary, easier cross-compilation Sam <sam@basx.dev> parents: 19 diff changeset	332 func translate3d*[T](x, y, z: T): Mat44[T] = Mat44[T](data: [
b45a5d338cd0 did: improve alotof_triangles example, remove glslang lib, use only binary, easier cross-compilation Sam <sam@basx.dev> parents: 19 diff changeset	333 T(1), T(0), T(0), x,
b45a5d338cd0 did: improve alotof_triangles example, remove glslang lib, use only binary, easier cross-compilation Sam <sam@basx.dev> parents: 19 diff changeset	334 T(0), T(1), T(0), y,
b45a5d338cd0 did: improve alotof_triangles example, remove glslang lib, use only binary, easier cross-compilation Sam <sam@basx.dev> parents: 19 diff changeset	335 T(0), T(0), T(1), z,
b45a5d338cd0 did: improve alotof_triangles example, remove glslang lib, use only binary, easier cross-compilation Sam <sam@basx.dev> parents: 19 diff changeset	336 T(0), T(0), T(0), T(1),
b45a5d338cd0 did: improve alotof_triangles example, remove glslang lib, use only binary, easier cross-compilation Sam <sam@basx.dev> parents: 19 diff changeset	337 ])
b45a5d338cd0 did: improve alotof_triangles example, remove glslang lib, use only binary, easier cross-compilation Sam <sam@basx.dev> parents: 19 diff changeset	338 func scale3d*[T](sx, sy, sz: T): Mat44[T] = Mat44[T](data: [
b45a5d338cd0 did: improve alotof_triangles example, remove glslang lib, use only binary, easier cross-compilation Sam <sam@basx.dev> parents: 19 diff changeset	339 sx, T(0), T(0), T(0),
b45a5d338cd0 did: improve alotof_triangles example, remove glslang lib, use only binary, easier cross-compilation Sam <sam@basx.dev> parents: 19 diff changeset	340 T(0), sy, T(0), T(0),
b45a5d338cd0 did: improve alotof_triangles example, remove glslang lib, use only binary, easier cross-compilation Sam <sam@basx.dev> parents: 19 diff changeset	341 T(0), T(0), sz, T(0),
b45a5d338cd0 did: improve alotof_triangles example, remove glslang lib, use only binary, easier cross-compilation Sam <sam@basx.dev> parents: 19 diff changeset	342 T(0), T(0), T(0), T(1),
b45a5d338cd0 did: improve alotof_triangles example, remove glslang lib, use only binary, easier cross-compilation Sam <sam@basx.dev> parents: 19 diff changeset	343 ])
b45a5d338cd0 did: improve alotof_triangles example, remove glslang lib, use only binary, easier cross-compilation Sam <sam@basx.dev> parents: 19 diff changeset	344 func rotate3d*[T](angle: T, a: Vec3[T]): Mat44[T] =
b45a5d338cd0 did: improve alotof_triangles example, remove glslang lib, use only binary, easier cross-compilation Sam <sam@basx.dev> parents: 19 diff changeset	345 let
b45a5d338cd0 did: improve alotof_triangles example, remove glslang lib, use only binary, easier cross-compilation Sam <sam@basx.dev> parents: 19 diff changeset	346 cosa = cos(angle)
b45a5d338cd0 did: improve alotof_triangles example, remove glslang lib, use only binary, easier cross-compilation Sam <sam@basx.dev> parents: 19 diff changeset	347 sina = sin(angle)
38 c3c963e7c1a6 did: tons of stuff, input, refactoring, fix some errors, some template improvment, sorry for super-commit Sam <sam@basx.dev> parents: 24 diff changeset	348 x = a[0]
c3c963e7c1a6 did: tons of stuff, input, refactoring, fix some errors, some template improvment, sorry for super-commit Sam <sam@basx.dev> parents: 24 diff changeset	349 y = a[1]
c3c963e7c1a6 did: tons of stuff, input, refactoring, fix some errors, some template improvment, sorry for super-commit Sam <sam@basx.dev> parents: 24 diff changeset	350 z = a[2]
22 b45a5d338cd0 did: improve alotof_triangles example, remove glslang lib, use only binary, easier cross-compilation Sam <sam@basx.dev> parents: 19 diff changeset	351 Mat44[T](data: [
b45a5d338cd0 did: improve alotof_triangles example, remove glslang lib, use only binary, easier cross-compilation Sam <sam@basx.dev> parents: 19 diff changeset	352 x * x * (1 - cosa) + cosa, y * x * (1 - cosa) - z * sina, z * x * (1 - cosa) + y * sina, T(0),
b45a5d338cd0 did: improve alotof_triangles example, remove glslang lib, use only binary, easier cross-compilation Sam <sam@basx.dev> parents: 19 diff changeset	353 x * y * (1 - cosa) + z * sina, y * y * (1 - cosa) + cosa, z * y * (1 - cosa) - x * sina, T(0),
b45a5d338cd0 did: improve alotof_triangles example, remove glslang lib, use only binary, easier cross-compilation Sam <sam@basx.dev> parents: 19 diff changeset	354 x * z * (1 - cosa) - y * sina, y * z * (1 - cosa) + x * sina, z * z * (1 - cosa) + cosa, T(0),
38 c3c963e7c1a6 did: tons of stuff, input, refactoring, fix some errors, some template improvment, sorry for super-commit Sam <sam@basx.dev> parents: 24 diff changeset	355 T(0), T(0), T(0), T(1),
22 b45a5d338cd0 did: improve alotof_triangles example, remove glslang lib, use only binary, easier cross-compilation Sam <sam@basx.dev> parents: 19 diff changeset	356 ])
b45a5d338cd0 did: improve alotof_triangles example, remove glslang lib, use only binary, easier cross-compilation Sam <sam@basx.dev> parents: 19 diff changeset	357
b45a5d338cd0 did: improve alotof_triangles example, remove glslang lib, use only binary, easier cross-compilation Sam <sam@basx.dev> parents: 19 diff changeset	358
16 617c6dcddbe2 add: matrix multiplications, tests Sam <sam@basx.dev> parents: 13 diff changeset	359 # call e.g. Mat32[int]().randomized() to get a random matrix
13 a571db114152 add: vector/matrix modules, tests, nim config script Sam <sam@basx.dev> parents: diff changeset	360 template makeRandomInit(mattype: typedesc) =
a571db114152 add: vector/matrix modules, tests, nim config script Sam <sam@basx.dev> parents: diff changeset	361 proc randomized*[T: SomeInteger](m: mattype[T]): mattype[T] =
a571db114152 add: vector/matrix modules, tests, nim config script Sam <sam@basx.dev> parents: diff changeset	362 for i in 0 ..< result.data.len:
a571db114152 add: vector/matrix modules, tests, nim config script Sam <sam@basx.dev> parents: diff changeset	363 result.data[i] = rand(low(typeof(m.data[0])) .. high(typeof(m.data[0])))
a571db114152 add: vector/matrix modules, tests, nim config script Sam <sam@basx.dev> parents: diff changeset	364 proc randomized*[T: SomeFloat](m: mattype[T]): mattype[T] =
a571db114152 add: vector/matrix modules, tests, nim config script Sam <sam@basx.dev> parents: diff changeset	365 for i in 0 ..< result.data.len:
a571db114152 add: vector/matrix modules, tests, nim config script Sam <sam@basx.dev> parents: diff changeset	366 result.data[i] = rand(1.0)
a571db114152 add: vector/matrix modules, tests, nim config script Sam <sam@basx.dev> parents: diff changeset	367
a571db114152 add: vector/matrix modules, tests, nim config script Sam <sam@basx.dev> parents: diff changeset	368 makeRandomInit(Mat22)
a571db114152 add: vector/matrix modules, tests, nim config script Sam <sam@basx.dev> parents: diff changeset	369 makeRandomInit(Mat23)
a571db114152 add: vector/matrix modules, tests, nim config script Sam <sam@basx.dev> parents: diff changeset	370 makeRandomInit(Mat32)
a571db114152 add: vector/matrix modules, tests, nim config script Sam <sam@basx.dev> parents: diff changeset	371 makeRandomInit(Mat33)
a571db114152 add: vector/matrix modules, tests, nim config script Sam <sam@basx.dev> parents: diff changeset	372 makeRandomInit(Mat34)
a571db114152 add: vector/matrix modules, tests, nim config script Sam <sam@basx.dev> parents: diff changeset	373 makeRandomInit(Mat43)
a571db114152 add: vector/matrix modules, tests, nim config script Sam <sam@basx.dev> parents: diff changeset	374 makeRandomInit(Mat44)
38 c3c963e7c1a6 did: tons of stuff, input, refactoring, fix some errors, some template improvment, sorry for super-commit Sam <sam@basx.dev> parents: 24 diff changeset	375
c3c963e7c1a6 did: tons of stuff, input, refactoring, fix some errors, some template improvment, sorry for super-commit Sam <sam@basx.dev> parents: 24 diff changeset	376 func perspective*[T: SomeFloat](fovy, aspect, zNear, zFar: T): Mat44[T] =
c3c963e7c1a6 did: tons of stuff, input, refactoring, fix some errors, some template improvment, sorry for super-commit Sam <sam@basx.dev> parents: 24 diff changeset	377 let tanHalfFovy = tan(fovy / T(2))
c3c963e7c1a6 did: tons of stuff, input, refactoring, fix some errors, some template improvment, sorry for super-commit Sam <sam@basx.dev> parents: 24 diff changeset	378 return Mat44[T](data:[
c3c963e7c1a6 did: tons of stuff, input, refactoring, fix some errors, some template improvment, sorry for super-commit Sam <sam@basx.dev> parents: 24 diff changeset	379 T(1) / (aspect * tanHalfFovy), T(0), T(0), T(0),
c3c963e7c1a6 did: tons of stuff, input, refactoring, fix some errors, some template improvment, sorry for super-commit Sam <sam@basx.dev> parents: 24 diff changeset	380 T(0), T(1) / tanHalfFovy, T(0), T(0),
c3c963e7c1a6 did: tons of stuff, input, refactoring, fix some errors, some template improvment, sorry for super-commit Sam <sam@basx.dev> parents: 24 diff changeset	381 T(0), T(0), T(zFar / (zFar - zNear)), T(-(zFar * zNear) / (zFar - zNear)),
c3c963e7c1a6 did: tons of stuff, input, refactoring, fix some errors, some template improvment, sorry for super-commit Sam <sam@basx.dev> parents: 24 diff changeset	382 T(0), T(0), T(1), T(1),
c3c963e7c1a6 did: tons of stuff, input, refactoring, fix some errors, some template improvment, sorry for super-commit Sam <sam@basx.dev> parents: 24 diff changeset	383 ])
c3c963e7c1a6 did: tons of stuff, input, refactoring, fix some errors, some template improvment, sorry for super-commit Sam <sam@basx.dev> parents: 24 diff changeset	384
c3c963e7c1a6 did: tons of stuff, input, refactoring, fix some errors, some template improvment, sorry for super-commit Sam <sam@basx.dev> parents: 24 diff changeset	385 func ortho*[T: SomeFloat](left, right, bottom, top, zNear, zFar: T): Mat44[T] =
c3c963e7c1a6 did: tons of stuff, input, refactoring, fix some errors, some template improvment, sorry for super-commit Sam <sam@basx.dev> parents: 24 diff changeset	386 Mat44[T](data:[
c3c963e7c1a6 did: tons of stuff, input, refactoring, fix some errors, some template improvment, sorry for super-commit Sam <sam@basx.dev> parents: 24 diff changeset	387 T(2) / (right - left), T(0), T(0), -(right + left) / (right - left),
c3c963e7c1a6 did: tons of stuff, input, refactoring, fix some errors, some template improvment, sorry for super-commit Sam <sam@basx.dev> parents: 24 diff changeset	388 T(0), T(2) / (top - bottom), T(0), -(top + bottom) / (top - bottom),
c3c963e7c1a6 did: tons of stuff, input, refactoring, fix some errors, some template improvment, sorry for super-commit Sam <sam@basx.dev> parents: 24 diff changeset	389 T(0), T(0), T(1) / (zFar - zNear), -zNear / (zFar - zNear),
c3c963e7c1a6 did: tons of stuff, input, refactoring, fix some errors, some template improvment, sorry for super-commit Sam <sam@basx.dev> parents: 24 diff changeset	390 T(0), T(0), T(1), T(1),
c3c963e7c1a6 did: tons of stuff, input, refactoring, fix some errors, some template improvment, sorry for super-commit Sam <sam@basx.dev> parents: 24 diff changeset	391 ])

Mercurial > games > semicongine

annotate src/zamikongine/math/matrix.nim @ 38:c3c963e7c1a6