Mercurial > games > semicongine
diff semiconginev2/core/matrix.nim @ 1218:56781cc0fc7c compiletime-tests
did: renamge main package
author | sam <sam@basx.dev> |
---|---|
date | Wed, 17 Jul 2024 21:01:37 +0700 |
parents | semicongine/core/matrix.nim@a3eb305bcac2 |
children | c4f98eb4bb05 |
line wrap: on
line diff
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/semiconginev2/core/matrix.nim Wed Jul 17 21:01:37 2024 +0700 @@ -0,0 +1,467 @@ +export math + +type + # layout is row-first + # having an object instead of directly aliasing the array seems a bit ugly at + # first, but is necessary to be able to work correctly with distinguished + # types (i.e. TMat23 and TMat32 would be an alias for the same type array[6, T] + # which prevents the type system from identifying the correct type at times) + # + # Though, great news is that objects have zero overhead! + TMat2*[T: SomeNumber] = object + data*: array[4, T] + TMat23*[T: SomeNumber] = object + data*: array[6, T] + TMat32*[T: SomeNumber] = object + data*: array[6, T] + TMat3*[T: SomeNumber] = object + data*: array[9, T] + TMat34*[T: SomeNumber] = object + data*: array[12, T] + TMat43*[T: SomeNumber] = object + data*: array[12, T] + TMat4*[T: SomeNumber] = object + data*: array[16, T] + TMat* = TMat2|TMat3|TMat4|TMat23|TMat32|TMat34|TMat43 + Mat2* = TMat2[float32] + Mat23* = TMat23[float32] + Mat32* = TMat32[float32] + Mat3* = TMat3[float32] + Mat34* = TMat34[float32] + Mat43* = TMat43[float32] + Mat4* = TMat4[float32] + +func MakeUnit2*[T: SomeNumber](): auto {.compiletime.} = TMat2[T](data: [ + T(1), T(0), + T(0), T(1), +]) +func MakeUnit3*[T: SomeNumber](): auto {.compiletime.} = TMat3[T](data: [ + T(1), T(0), T(0), + T(0), T(1), T(0), + T(0), T(0), T(1), +]) +func MakeUnit4*[T: SomeNumber](): auto {.compiletime.} = TMat4[T](data: [ + T(1), T(0), T(0), T(0), + T(0), T(1), T(0), T(0), + T(0), T(0), T(1), T(0), + T(0), T(0), T(0), T(1), +]) + +# generates constants: Unit +# Also for Y, Z, R, G, B +# not sure if this is necessary or even a good idea... +macro generateAllMatrixConsts() = + result = newStmtList() + for theType in ["int", "int8", "int16", "int32", "int64", "float", "float32", "float64"]: + var typename = theType[0 .. 0] + if theType[^2].isDigit: + typename = typename & theType[^2] + if theType[^1].isDigit: + typename = typename & theType[^1] + result.add(newConstStmt( + postfix(ident("Unit2" & typename), "*"), + newCall(nnkBracketExpr.newTree(ident("MakeUnit2"), ident(theType))) + )) + result.add(newConstStmt( + postfix(ident("Unit3" & typename), "*"), + newCall(nnkBracketExpr.newTree(ident("MakeUnit3"), ident(theType))) + )) + result.add(newConstStmt( + postfix(ident("Unit4" & typename), "*"), + newCall(nnkBracketExpr.newTree(ident("MakeUnit4"), ident(theType))) + )) + +generateAllMatrixConsts() + +const Unit2* = MakeUnit2[float32]() +const Unit3* = MakeUnit3[float32]() +const Unit4* = MakeUnit4[float32]() + +template RowCount*(m: typedesc): int = + when m is TMat2: 2 + elif m is TMat23: 2 + elif m is TMat32: 3 + elif m is TMat3: 3 + elif m is TMat34: 3 + elif m is TMat43: 4 + elif m is TMat4: 4 +template ColumnCount*(m: typedesc): int = + when m is TMat2: 2 + elif m is TMat23: 3 + elif m is TMat32: 2 + elif m is TMat3: 3 + elif m is TMat34: 4 + elif m is TMat43: 3 + elif m is TMat4: 4 +template matlen(m: typedesc): int = + when m is TMat2: 4 + elif m is TMat23: 6 + elif m is TMat32: 6 + elif m is TMat3: 9 + elif m is TMat34: 12 + elif m is TMat43: 12 + elif m is TMat4: 16 + + +func toString[T: TMat](value: T): string = + var + strvalues: seq[string] + maxwidth = 0 + + for n in value.data: + let strval = &"{float(n):.4f}" + strvalues.add(strval) + if strval.len > maxwidth: + maxwidth = strval.len + + for i in 0 ..< strvalues.len: + let filler = " ".repeat(maxwidth - strvalues[i].len) + if i mod T.ColumnCount == T.ColumnCount - 1: + result &= filler & strvalues[i] & "\n" + else: + if i mod T.ColumnCount == 0: + result &= " " + result &= filler & strvalues[i] & " " + +func `$`*(v: TMat2[SomeNumber]): string = toString[TMat2[SomeNumber]](v) +func `$`*(v: TMat23[SomeNumber]): string = toString[TMat23[SomeNumber]](v) +func `$`*(v: TMat32[SomeNumber]): string = toString[TMat32[SomeNumber]](v) +func `$`*(v: TMat3[SomeNumber]): string = toString[TMat3[SomeNumber]](v) +func `$`*(v: TMat34[SomeNumber]): string = toString[TMat34[SomeNumber]](v) +func `$`*(v: TMat43[SomeNumber]): string = toString[TMat43[SomeNumber]](v) +func `$`*(v: TMat4[SomeNumber]): string = toString[TMat4[SomeNumber]](v) + +func `[]`*[T: TMat](m: T, row, col: int): auto = m.data[col + row * T.ColumnCount] +func `[]=`*[T: TMat, U](m: var T, row, col: int, value: U) = m.data[col + row * T.ColumnCount] = value +func `[]`*[T: TMat](m: T, i: int): auto = m.data[i] +func `[]=`*[T: TMat, U](m: var T, i: int, value: U) = m.data[i] = value + +func Row*[T: TMat2](m: T, i: 0..1): auto = TVec2([m[i, 0], m[i, 1]]) +func Row*[T: TMat32](m: T, i: 0..2): auto = TVec2([m[i, 0], m[i, 1]]) +func Row*[T: TMat23](m: T, i: 0..1): auto = TVec3([m[i, 0], m[i, 1], m[i, 2]]) +func Row*[T: TMat3](m: T, i: 0..2): auto = TVec3([m[i, 0], m[i, 1], m[i, 2]]) +func Row*[T: TMat43](m: T, i: 0..3): auto = TVec3([m[i, 0], m[i, 1], m[i, 2]]) +func Row*[T: TMat34](m: T, i: 0..2): auto = TVec4([m[i, 0], m[i, 1], m[i, 2], m[i, 3]]) +func Row*[T: TMat4](m: T, i: 0..3): auto = TVec4([m[i, 0], m[i, 1], m[i, 2], m[i, 3]]) + +func Col*[T: TMat2](m: T, i: 0..1): auto = TVec2([m[0, i], m[1, i]]) +func Col*[T: TMat23](m: T, i: 0..2): auto = TVec2([m[0, i], m[1, i]]) +func Col*[T: TMat32](m: T, i: 0..1): auto = TVec3([m[0, i], m[1, i], m[2, i]]) +func Col*[T: TMat3](m: T, i: 0..2): auto = TVec3([m[0, i], m[1, i], m[2, i]]) +func Col*[T: TMat34](m: T, i: 0..3): auto = TVec3([m[0, i], m[1, i], m[2, i]]) +func Col*[T: TMat43](m: T, i: 0..2): auto = TVec4([m[0, i], m[1, i], m[2, i], m[3, i]]) +func Col*[T: TMat4](m: T, i: 0..3): auto = TVec4([m[0, i], m[1, i], m[2, i], m[3, i]]) + +proc createMatMatMultiplicationOperator(leftType: typedesc, rightType: typedesc, outType: typedesc): NimNode = + var data = nnkBracket.newTree() + for i in 0 ..< RowCount(leftType): + for j in 0 ..< rightType.ColumnCount: + data.add(newCall( + ident("sum"), + infix( + newCall(newDotExpr(ident("a"), ident("Row")), newLit(i)), + "*", + newCall(newDotExpr(ident("b"), ident("Col")), newLit(j)) + ) + )) + + return newProc( + postfix(nnkAccQuoted.newTree(ident("*")), "*"), + params = [ + ident("auto"), + newIdentDefs(ident("a"), ident(leftType.name)), + newIdentDefs(ident("b"), ident(rightType.name)) + ], + body = nnkObjConstr.newTree(ident(outType.name), nnkExprColonExpr.newTree(ident("data"), data)), + procType = nnkFuncDef, + ) + +proc createMatMatAdditionOperator(theType: typedesc): NimNode = + var data = nnkBracket.newTree() + for i in 0 ..< matlen(theType): + data.add( + infix( + nnkBracketExpr.newTree(ident("a"), newLit(i)), + "+", + nnkBracketExpr.newTree(ident("b"), newLit(i)), + )) + + return newProc( + postfix(nnkAccQuoted.newTree(ident("+")), "*"), + params = [ + ident("auto"), + newIdentDefs(ident("a"), ident(theType.name)), + newIdentDefs(ident("b"), ident(theType.name)) + ], + body = nnkObjConstr.newTree(ident(theType.name), nnkExprColonExpr.newTree(ident("data"), data)), + procType = nnkFuncDef, + ) + +proc createVecMatMultiplicationOperator(matType: typedesc, vecType: typedesc): NimNode = + var data = nnkBracket.newTree() + for i in 0 ..< matType.RowCount: + data.add(newCall( + ident("sum"), + infix( + ident("v"), + "*", + newCall(newDotExpr(ident("m"), ident("Row")), newLit(i)) + ) + )) + + let resultVec = newCall( + nnkBracketExpr.newTree(ident(vecType.name), ident("T")), + data, + ) + let name = postfix(nnkAccQuoted.newTree(ident("*")), "*") + let genericParams = nnkGenericParams.newTree(nnkIdentDefs.newTree(ident("T"), ident("SomeNumber"), newEmptyNode())) + let formalParams = nnkFormalParams.newTree( + ident("auto"), + newIdentDefs(ident("m"), nnkBracketExpr.newTree(ident(matType.name), ident("T"))), + newIdentDefs(ident("v"), nnkBracketExpr.newTree(ident(vecType.name), ident("T"))), + ) + + return nnkFuncDef.newTree( + name, + newEmptyNode(), + genericParams, + formalParams, + newEmptyNode(), + newEmptyNode(), + resultVec + ) + + +proc createMatScalarOperator(matType: typedesc, op: string): NimNode = + result = newStmtList() + + var data = nnkBracket.newTree() + for i in 0 ..< matType.RowCount * matType.ColumnCount: + data.add(infix(nnkBracketExpr.newTree(newDotExpr(ident("a"), ident("data")), newLit(i)), op, ident("b"))) + result.add(newProc( + postfix(nnkAccQuoted.newTree(ident(op)), "*"), + params = [ + ident("auto"), + newIdentDefs(ident("a"), ident(matType.name)), + newIdentDefs(ident("b"), ident("SomeNumber")), + ], + body = nnkObjConstr.newTree(ident(matType.name), nnkExprColonExpr.newTree(ident("data"), data)), + procType = nnkFuncDef, + )) + result.add(newProc( + postfix(nnkAccQuoted.newTree(ident(op)), "*"), + params = [ + ident("auto"), + newIdentDefs(ident("b"), ident("SomeNumber")), + newIdentDefs(ident("a"), ident(matType.name)), + ], + body = nnkObjConstr.newTree(ident(matType.name), nnkExprColonExpr.newTree(ident("data"), data)), + procType = nnkFuncDef, + )) + if op == "-": + var data2 = nnkBracket.newTree() + for i in 0 ..< matType.RowCount * matType.ColumnCount: + data2.add(prefix(nnkBracketExpr.newTree(newDotExpr(ident("a"), ident("data")), newLit(i)), op)) + result.add(newProc( + postfix(nnkAccQuoted.newTree(ident(op)), "*"), + params = [ + ident("auto"), + newIdentDefs(ident("a"), ident(matType.name)), + ], + body = nnkObjConstr.newTree(ident(matType.name), nnkExprColonExpr.newTree(ident("data"), data2)), + procType = nnkFuncDef, + )) + +macro createAllMultiplicationOperators() = + result = newStmtList() + + for op in ["+", "-", "*", "/"]: + result.add(createMatScalarOperator(TMat2, op)) + result.add(createMatScalarOperator(TMat23, op)) + result.add(createMatScalarOperator(TMat32, op)) + result.add(createMatScalarOperator(TMat3, op)) + result.add(createMatScalarOperator(TMat34, op)) + result.add(createMatScalarOperator(TMat43, op)) + result.add(createMatScalarOperator(TMat4, op)) + + result.add(createMatMatMultiplicationOperator(TMat2, TMat2, TMat2)) + result.add(createMatMatMultiplicationOperator(TMat2, TMat23, TMat23)) + result.add(createMatMatMultiplicationOperator(TMat23, TMat32, TMat2)) + result.add(createMatMatMultiplicationOperator(TMat23, TMat3, TMat23)) + result.add(createMatMatMultiplicationOperator(TMat32, TMat2, TMat32)) + result.add(createMatMatMultiplicationOperator(TMat32, TMat23, TMat3)) + result.add(createMatMatMultiplicationOperator(TMat3, TMat32, TMat32)) + result.add(createMatMatMultiplicationOperator(TMat3, TMat3, TMat3)) + result.add(createMatMatMultiplicationOperator(TMat3, TMat34, TMat34)) + result.add(createMatMatMultiplicationOperator(TMat43, TMat3, TMat43)) + result.add(createMatMatMultiplicationOperator(TMat43, TMat34, TMat4)) + result.add(createMatMatMultiplicationOperator(TMat4, TMat43, TMat43)) + result.add(createMatMatMultiplicationOperator(TMat4, TMat4, TMat4)) + + result.add(createMatMatAdditionOperator(TMat2)) + result.add(createMatMatAdditionOperator(TMat23)) + result.add(createMatMatAdditionOperator(TMat32)) + result.add(createMatMatAdditionOperator(TMat3)) + result.add(createMatMatAdditionOperator(TMat34)) + result.add(createMatMatAdditionOperator(TMat43)) + result.add(createMatMatAdditionOperator(TMat4)) + + result.add(createVecMatMultiplicationOperator(TMat2, TVec2)) + result.add(createVecMatMultiplicationOperator(TMat3, TVec3)) + result.add(createVecMatMultiplicationOperator(TMat4, TVec4)) + +createAllMultiplicationOperators() + +func `*`*(mat: Mat4, vec: Vec3f): Vec3f = + (mat * vec.ToVec4(1)).ToVec3 + +func Transposed*[T](m: TMat2[T]): TMat2[T] = TMat2[T](data: [ + m[0, 0], m[1, 0], + m[0, 1], m[1, 1], +]) +func Transposed*[T](m: TMat23[T]): TMat32[T] = TMat32[T](data: [ + m[0, 0], m[1, 0], + m[0, 1], m[1, 1], + m[0, 2], m[1, 2], +]) +func Transposed*[T](m: TMat32[T]): TMat23[T] = TMat23[T](data: [ + m[0, 0], m[1, 0], m[2, 0], + m[0, 1], m[1, 1], m[2, 1], +]) +func Transposed*[T](m: TMat3[T]): TMat3[T] = TMat3[T](data: [ + m[0, 0], m[1, 0], m[2, 0], + m[0, 1], m[1, 1], m[2, 1], + m[0, 2], m[1, 2], m[2, 2], +]) +func Transposed*[T](m: TMat43[T]): TMat34[T] = TMat34[T](data: [ + m[0, 0], m[1, 0], m[2, 0], m[3, 0], + m[0, 1], m[1, 1], m[2, 1], m[3, 1], + m[0, 2], m[1, 2], m[2, 2], m[3, 2], +]) +func Transposed*[T](m: TMat34[T]): TMat43[T] = TMat43[T](data: [ + m[0, 0], m[1, 0], m[2, 0], + m[0, 1], m[1, 1], m[2, 1], + m[0, 2], m[1, 2], m[2, 2], + m[0, 3], m[1, 3], m[2, 3], +]) +func Transposed*[T](m: TMat4[T]): TMat4[T] = TMat4[T](data: [ + m[0, 0], m[1, 0], m[2, 0], m[3, 0], + m[0, 1], m[1, 1], m[2, 1], m[3, 1], + m[0, 2], m[1, 2], m[2, 2], m[3, 2], + m[0, 3], m[1, 3], m[2, 3], m[3, 3], +]) + +func Translate2d*[T](x, y: T): TMat3[T] = TMat3[T](data: [ + T(1), T(0), x, + T(0), T(1), y, + T(0), T(0), T(1), +]) +func Scale2d*[T](sx, sy: T): TMat3[T] = TMat3[T](data: [ + sx, T(0), T(0), + T(0), sy, T(0), + T(0), T(0), T(1), +]) +func Rotate2d*[T](angle: T): TMat3[T] = TMat3[T](data: [ + cos(angle), -sin(angle), T(0), + sin(angle), cos(angle), T(0), + T(0), T(0), T(1), +]) +func Translate*(x = 0'f32, y = 0'f32, z = 0'f32): TMat4[float32] = Mat4(data: [ + 1'f32, 0'f32, 0'f32, x, + 0'f32, 1'f32, 0'f32, y, + 0'f32, 0'f32, 1'f32, z, + 0'f32, 0'f32, 0'f32, 1'f32, +]) +func Translate*[T: TVec3](v: T): TMat4[float32] = Translate(v[0], v[1], v[2]) +func Scale*(x = 1'f32, y = 1'f32, z = 1'f32): Mat4 = Mat4(data: [ + x, 0'f32, 0'f32, 0'f32, + 0'f32, y, 0'f32, 0'f32, + 0'f32, 0'f32, z, 0'f32, + 0'f32, 0'f32, 0'f32, 1'f32, +]) +func Scale*[T: TVec3](v: T): TMat4[float32] = Scale(v[0], v[1], v[2]) +func Rotate*(angle: float32, a: Vec3f): Mat4 = + let + cosa = cos(angle) + sina = sin(angle) + x = a[0] + y = a[1] + z = a[2] + Mat4(data: [ + x * x * (1 - cosa) + cosa, y * x * (1 - cosa) - z * sina, z * x * (1 - cosa) + y * sina, 0'f32, + x * y * (1 - cosa) + z * sina, y * y * (1 - cosa) + cosa, z * y * (1 - cosa) - x * sina, 0'f32, + x * z * (1 - cosa) - y * sina, y * z * (1 - cosa) + x * sina, z * z * (1 - cosa) + cosa, 0'f32, + 0'f32, 0'f32, 0'f32, 1'f32, + ]) + +func asMat3(m: Mat4): auto = + Mat3(data: [ + m[0, 0], m[0, 1], m[0, 2], + m[1, 0], m[1, 1], m[1, 2], + m[2, 0], m[2, 1], m[2, 2], + ]) + + +func Inversed*(m: Mat4): Mat4 = + var m3 = m.asMat3.Transposed + m3[0, 0] = 1'f32 / m3[0, 0] + m3[1, 1] = 1'f32 / m3[1, 1] + m3[2, 2] = 1'f32 / m3[2, 2] + let col3 = -(m3 * m.Col(3).xyz) + return Mat4(data: [ + m3[0, 0], m3[0, 1], m3[0, 2], col3.x, + m3[1, 0], m3[1, 1], m3[1, 2], col3.y, + m3[2, 0], m3[2, 1], m3[2, 2], col3.z, + 0, 0, 0, 1, + ]) + + +# call e.g. TMat32[int]().randomized() to get a random matrix +template makeRandomMatrixInit(mattype: typedesc) = + proc Randomized*[T: SomeInteger](m: mattype[T]): mattype[T] = + for i in 0 ..< result.data.len: + result.data[i] = rand(low(typeof(m.data[0])) .. high(typeof(m.data[0]))) + proc Randomized*[T: SomeFloat](m: mattype[T]): mattype[T] = + for i in 0 ..< result.data.len: + result.data[i] = rand(T(1.0)) + +makeRandomMatrixInit(TMat2) +makeRandomMatrixInit(TMat23) +makeRandomMatrixInit(TMat32) +makeRandomMatrixInit(TMat3) +makeRandomMatrixInit(TMat34) +makeRandomMatrixInit(TMat43) +makeRandomMatrixInit(TMat4) + +func Perspective*(fovy, aspect, zNear, zFar: float32): Mat4 = + let tanHalfFovy = tan(fovy / 2) + return Mat4(data: [ + 1 / (aspect * tanHalfFovy), 0, 0, 0, + 0, 1 / tanHalfFovy, 0, 0, + 0, 0, zFar / (zFar - zNear), -(zFar * zNear) / (zFar - zNear), + 0, 0, 1, 1, + ]) + +func Ortho*(left, right, top, bottom, zNear, zFar: float32): Mat4 = + Mat4(data: [ + 2 / (right - left), 0, 0, -(right + left) / (right - left), + 0, 2 / (bottom - top), 0, -(bottom + top) / (bottom - top), + 0, 0, 1 / (zFar - zNear), zNear / (zFar - zNear), + 0, 0, 0, 1, + ]) + +# create an orthographic perspective that will map from -1 .. 1 on all axis and keep a 1:1 aspect ratio +# the smaller dimension (width or height) will always be 1 and the larger dimension will be larger, to keep the ratio +func OrthoWindowAspect*(windowAspect: float32): Mat4 = + if windowAspect < 1: + let space = 2 * (1 / windowAspect - 1) / 2 + Ortho(-1, 1, -1 - space, 1 + space, 0, 1) + else: + let space = 2 * (windowAspect - 1) / 2 + Ortho(-1 - space, 1 + space, -1, 1, 0, 1) + +func Position*(mat: Mat4): Vec3f {.deprecated.} = + mat.Col(3).ToVec3 + +func Scaling*(mat: Mat4): Vec3f {.deprecated.} = + NewVec4f(mat[0, 0], mat[1, 1], mat[2, 2])