bendns' repos / tracer
small racing game im working on
Diffstat (limited to 'lib/matrix.gd')
-rw-r--r--lib/matrix.gd148
1 files changed, 148 insertions, 0 deletions
diff --git a/lib/matrix.gd b/lib/matrix.gd
new file mode 100644
index 0000000..f33103a
--- /dev/null
+++ b/lib/matrix.gd
@@ -0,0 +1,148 @@
+## Matrix calculations.
+##
+## This class allows to perform all the basic operations with matrices.
+## (add, sub, multiply, scalar, dot product)
+## It is also possible to import and export matrices from an array.
+class_name Matrix
+
+var rows: int
+var cols: int
+
+var data = []
+
+
+func _init(_rows: int,_cols: int,value: float = 0.0):
+ randomize()
+ rows = _rows
+ cols = _cols
+ for row in range(rows):
+ data.insert(row , [])
+ for col in range(cols):
+ data[row].insert(col, value)
+
+
+static func from_array(arr: Array[float]) -> Matrix:
+ var result = Matrix.new(arr.size(), 1)
+ for row in range(result.rows):
+ result.data[row][0] = arr[row]
+
+ return result
+
+
+static func to_array(matrix: Matrix) -> Array[float]:
+ var result = []
+ for row in range(matrix.rows):
+ for col in range(matrix.cols):
+ result.append(matrix.data[row][col])
+
+ return result
+
+
+static func rand(matrix: Matrix) -> Matrix:
+ randomize()
+
+ var result = Matrix.new(matrix.rows, matrix.cols)
+
+ for row in range(result.rows):
+ for col in range(result.cols):
+ result.data[row][col] = randf_range(-1, 1)
+
+ return result
+
+
+static func add(a: Matrix, b: Matrix) -> Matrix:
+ assert(a.rows == b.rows and a.cols == b.cols)
+
+ var result = Matrix.new(a.rows, a.cols)
+
+ for row in range(result.rows):
+ for col in range(result.cols):
+ result.data[row][col] = a.data[row][col] + b.data[row][col]
+
+ return result
+
+
+static func subtract(a: Matrix, b: Matrix) -> Matrix:
+ assert(a.rows == b.rows and a.cols == b.cols)
+
+ var result = Matrix.new(a.rows, a.cols)
+
+ for row in range(result.rows):
+ for col in range(result.cols):
+ result.data[row][col] = a.data[row][col] - b.data[row][col]
+
+ return result
+
+
+static func scalar(matrix: Matrix, value: float) -> Matrix:
+ var result = Matrix.new(matrix.rows, matrix.cols)
+
+ for row in range(result.rows):
+ for col in range(result.cols):
+ result.data[row][col] = matrix.data[row][col] * value
+
+ return result
+
+
+static func product(a: Matrix, b: Matrix) -> Matrix:
+ assert(a.cols == b.rows)
+
+ var result = Matrix.new(a.rows, b.cols)
+
+ for row in range(result.rows):
+ for col in range(result.cols):
+ result.data[row][col] = 0.0
+ for k in range(a.cols):
+ result.data[row][col] += a.data[row][k] * b.data[k][col]
+
+ return result
+
+
+static func multiply(a: Matrix, b: Matrix) -> Matrix:
+ assert(a.rows == b.rows and a.cols == b.cols)
+
+ var result = Matrix.new(a.rows, a.cols)
+
+ for row in range(result.rows):
+ for col in range(result.cols):
+ result.data[row][col] = a.data[row][col] * b.data[row][col]
+
+ return result
+
+
+static func random(a: Matrix, b: Matrix) -> Matrix:
+ randomize()
+ var result = Matrix.new(a.rows, a.cols)
+ for row in range(result.rows):
+ for col in range(result.cols):
+ result.data[row][col] = a.data[row][col] if randf_range(0, 1) > 0.5 else b.data[row][col]
+
+ return result
+
+
+static func transpose(matrix: Matrix) -> Matrix:
+ var result = Matrix.new(matrix.cols, matrix.rows)
+
+ for row in range(result.rows):
+ for col in range(result.cols):
+ result.data[row][col] = matrix.data[col][row]
+
+ return result
+
+
+static func copy(matrix: Matrix) -> Matrix:
+ var result = Matrix.new(matrix.rows, matrix.cols)
+ for row in range(result.rows):
+ for col in range(result.cols):
+ result.data[row][col] = matrix.data[row][col]
+ return result
+
+
+static func map(matrix: Matrix, callback) -> Matrix:
+ var result = Matrix.new(matrix.rows, matrix.cols)
+
+ for row in range(result.rows):
+ for col in range(result.cols):
+ result.data[row][col] = callback.call(matrix.data[row][col], row, col)
+
+ return result