package/gonum
1
0
Fork 0
mirror of https://github.com/gonum/gonum.git synced 2025-10-06 07:37:03 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

spatial/r2: implement Rotate

Browse Source 
Fixes gonum/gonum#1513.
This commit is contained in:
Sebastien Binet
2020-12-01 12:56:40 +01:00
committed by GitHub
parent 9c06200335
commit d24ce68544
2 changed files with 61 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
spatial/r2/vector_test.go
View File

@@ -240,6 +240,31 @@ func TestCos(t *testing.T) {
}
}
func TestRotate(t *testing.T) {
for _, test := range []struct {
v, q Vec
alpha float64
want Vec
}{
{Vec{1, 0}, Vec{0, 0}, math.Pi / 2, Vec{0, 1}},
{Vec{1, 0}, Vec{0, 0}, 3 * math.Pi / 2, Vec{0, -1}},
{Vec{1, 0}, Vec{0, 1}, 0, Vec{1, 0}},
{Vec{1, 0}, Vec{0, 1}, 2 * math.Pi, Vec{1, 0}},
{Vec{1, 1}, Vec{0, 0}, math.Pi, Vec{-1, -1}},
{Vec{2, 2}, Vec{1, 1}, math.Pi / 2, Vec{0, 2}},
{Vec{2, 2}, Vec{1, 1}, math.Pi, Vec{0, 0}},
{Vec{2, 2}, Vec{2, 0}, math.Pi, Vec{2, -2}},
} {
got := test.v.Rotate(test.alpha, test.q)
if !vecApproxEqual(got, test.want) {
t.Errorf(
"rotate(%v, %v, %v)= %v, want=%v",
test.v, test.alpha, test.q, got, test.want,
)
}
}
}
func vecIsNaN(v Vec) bool {
return math.IsNaN(v.X) && math.IsNaN(v.Y)
}