simplified matrix type

This commit is contained in:
Thomas Friedel 2019-03-28 16:39:59 +01:00
parent 0e16a8261b
commit b2502f5472
2 changed files with 49 additions and 63 deletions

View file

@ -161,8 +161,7 @@ func (cv *Canvas) Height() int {
func (cv *Canvas) Size() (int, int) { return cv.b.Size() }
func (cv *Canvas) tf(v vec) vec {
v, _ = v.mulMat(cv.state.transform)
return v
return v.mulMat(cv.state.transform)
}
const alphaTexSize = 2048
@ -418,12 +417,12 @@ func (cv *Canvas) Rotate(angle float64) {
// Transform updates the current transformation with the given matrix
func (cv *Canvas) Transform(a, b, c, d, e, f float64) {
cv.state.transform = mat{a, b, 0, c, d, 0, e, f, 1}.mul(cv.state.transform)
cv.state.transform = mat{a, b, c, d, e, f}.mul(cv.state.transform)
}
// SetTransform replaces the current transformation with the given matrix
func (cv *Canvas) SetTransform(a, b, c, d, e, f float64) {
cv.state.transform = mat{a, b, 0, c, d, 0, e, f, 1}
cv.state.transform = mat{a, b, c, d, e, f}
}
// SetShadowColor sets the color of the shadow. If it is fully transparent (default)

105
math.go
View file

@ -11,43 +11,42 @@ func (v vec) String() string {
return fmt.Sprintf("[%f,%f]", v[0], v[1])
}
func (v1 vec) add(v2 vec) vec {
return vec{v1[0] + v2[0], v1[1] + v2[1]}
func (v vec) add(v2 vec) vec {
return vec{v[0] + v2[0], v[1] + v2[1]}
}
func (v1 vec) sub(v2 vec) vec {
return vec{v1[0] - v2[0], v1[1] - v2[1]}
func (v vec) sub(v2 vec) vec {
return vec{v[0] - v2[0], v[1] - v2[1]}
}
func (v1 vec) mul(v2 vec) vec {
return vec{v1[0] * v2[0], v1[1] * v2[1]}
func (v vec) mul(v2 vec) vec {
return vec{v[0] * v2[0], v[1] * v2[1]}
}
func (v vec) mulf(f float64) vec {
return vec{v[0] * f, v[1] * f}
}
func (v vec) mulMat(m mat) (vec, float64) {
func (v vec) mulMat(m mat) vec {
return vec{
m[0]*v[0] + m[3]*v[1] + m[6],
m[1]*v[0] + m[4]*v[1] + m[7]},
m[2]*v[0] + m[5]*v[1] + m[8]
m[0]*v[0] + m[2]*v[1] + m[4],
m[1]*v[0] + m[3]*v[1] + m[5]}
}
func (v vec) mulMat2(m mat2) vec {
return vec{m[0]*v[0] + m[2]*v[1], m[1]*v[0] + m[3]*v[1]}
}
func (v1 vec) div(v2 vec) vec {
return vec{v1[0] / v2[0], v1[1] / v2[1]}
func (v vec) div(v2 vec) vec {
return vec{v[0] / v2[0], v[1] / v2[1]}
}
func (v vec) divf(f float64) vec {
return vec{v[0] / f, v[1] / f}
}
func (v1 vec) dot(v2 vec) float64 {
return v1[0]*v2[0] + v1[1]*v2[1]
func (v vec) dot(v2 vec) float64 {
return v[0]*v2[0] + v[1]*v2[1]
}
func (v vec) len() float64 {
@ -74,82 +73,70 @@ func (v vec) angleTo(v2 vec) float64 {
return math.Acos(v.norm().dot(v2.norm()))
}
type mat [9]float64
type mat [6]float64
func (m *mat) String() string {
return fmt.Sprintf("[%f,%f,%f,\n %f,%f,%f,\n %f,%f,%f,]", m[0], m[3], m[6], m[1], m[4], m[7], m[2], m[5], m[8])
return fmt.Sprintf("[%f,%f,0,\n %f,%f,0,\n %f,%f,1,]", m[0], m[2], m[4], m[1], m[3], m[5])
}
func matIdentity() mat {
return mat{
1, 0, 0,
0, 1, 0,
0, 0, 1}
1, 0,
0, 1,
0, 0}
}
func matTranslate(v vec) mat {
return mat{
1, 0, 0,
0, 1, 0,
v[0], v[1], 1}
1, 0,
0, 1,
v[0], v[1]}
}
func matScale(v vec) mat {
return mat{
v[0], 0, 0,
0, v[1], 0,
0, 0, 1}
v[0], 0,
0, v[1],
0, 0}
}
func matRotate(radians float64) mat {
s, c := math.Sincos(radians)
return mat{
c, s, 0,
-s, c, 0,
0, 0, 1}
c, s,
-s, c,
0, 0}
}
func (m1 mat) mul(m2 mat) mat {
func (m mat) mul(m2 mat) mat {
return mat{
m1[0]*m2[0] + m1[1]*m2[3] + m1[2]*m2[6],
m1[0]*m2[1] + m1[1]*m2[4] + m1[2]*m2[7],
m1[0]*m2[2] + m1[1]*m2[5] + m1[2]*m2[8],
m1[3]*m2[0] + m1[4]*m2[3] + m1[5]*m2[6],
m1[3]*m2[1] + m1[4]*m2[4] + m1[5]*m2[7],
m1[3]*m2[2] + m1[4]*m2[5] + m1[5]*m2[8],
m1[6]*m2[0] + m1[7]*m2[3] + m1[8]*m2[6],
m1[6]*m2[1] + m1[7]*m2[4] + m1[8]*m2[7],
m1[6]*m2[2] + m1[7]*m2[5] + m1[8]*m2[8]}
m[0]*m2[0] + m[1]*m2[2],
m[0]*m2[1] + m[1]*m2[3],
m[2]*m2[0] + m[3]*m2[2],
m[2]*m2[1] + m[3]*m2[3],
m[4]*m2[0] + m[5]*m2[2] + m2[4],
m[4]*m2[1] + m[5]*m2[3] + m2[5]}
}
func (m mat) invert() mat {
var identity float64 = 1.0 / (m[0]*m[4]*m[8] + m[3]*m[7]*m[2] + m[6]*m[1]*m[5] - m[6]*m[4]*m[2] - m[3]*m[1]*m[8] - m[0]*m[7]*m[5])
identity := 1.0 / (m[0]*m[3] - m[2]*m[1])
return mat{
(m[4]*m[8] - m[5]*m[7]) * identity,
(m[2]*m[7] - m[1]*m[8]) * identity,
(m[1]*m[5] - m[2]*m[4]) * identity,
(m[5]*m[6] - m[3]*m[8]) * identity,
(m[0]*m[8] - m[2]*m[6]) * identity,
(m[2]*m[3] - m[0]*m[5]) * identity,
(m[3]*m[7] - m[4]*m[6]) * identity,
(m[1]*m[6] - m[0]*m[7]) * identity,
(m[0]*m[4] - m[1]*m[3]) * identity}
}
func (m mat) f32() [9]float32 {
return [9]float32{
float32(m[0]), float32(m[1]), float32(m[2]),
float32(m[3]), float32(m[4]), float32(m[5]),
float32(m[6]), float32(m[7]), float32(m[8])}
}
func (m mat) mat2() mat2 {
return mat2{m[0], m[1], m[3], m[4]}
m[3] * identity,
-m[1] * identity,
-m[2] * identity,
m[0] * identity,
(m[2]*m[5] - m[3]*m[4]) * identity,
(m[1]*m[4] - m[0]*m[5]) * identity,
}
}
type mat2 [4]float64
func (m mat) mat2() mat2 {
return mat2{m[0], m[1], m[2], m[3]}
}
func (m *mat2) String() string {
return fmt.Sprintf("[%f,%f,\n %f,%f]", m[0], m[2], m[1], m[3])
}