toki/numidr
1
0
Fork 0
Code Issues Pull requests Activity

Misc. changes

Browse source
This commit is contained in:
Kiana Sheibani 2022-11-30 12:54:01 -05:00
parent 0a54f607b0
commit 524486bb58
Signed by: toki
GPG key ID: 6CB106C25E86A9F7
5 changed files with 51 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

25
src/Data/NumIdr/Matrix.idr
View file

@ -96,6 +96,23 @@ rotate3DZ : Double -> Matrix' 3 Double
rotate3DZ a = matrix [[cos a, - sin a, 0], [sin a, cos a, 0], [0,0,1]]
export
reflect : {n : _} -> Neg a => Fin n -> Matrix' n a
reflect i = indexSet [i, i] (-1) (repeatDiag 1 0)
export
reflectX : {n : _} -> Neg a => Matrix' (1 + n) a
reflectX = reflect 0
export
reflectY : {n : _} -> Neg a => Matrix' (2 + n) a
reflectY = reflect 1
export
reflectZ : {n : _} -> Neg a => Matrix' (3 + n) a
reflectZ = reflect 2
--------------------------------------------------------------------------------
-- Indexing
--------------------------------------------------------------------------------
@ -225,6 +242,14 @@ trace : Num a => Matrix m n a -> a
trace = sum . diagonal'
||| Construct a matrix that reflects a vector along a hyperplane of the
||| given normal vector. The input does not have to be a unit vector.
export
reflectNormal : (Neg a, Fractional a) => Vector n a -> Matrix' n a
reflectNormal {n} v with (viewShape v)
_ | Shape [n] = repeatDiag 1 0 - (2 / normSq v) *. outer v v
--------------------------------------------------------------------------------
-- Matrix multiplication
--------------------------------------------------------------------------------