Added TriangleF.

Renamed Add to Extend on PolygonF{,32}.
Add now translates a polygon.
This commit is contained in:
Sander Schobers 2021-08-09 01:37:16 +02:00
parent 88ed955ce4
commit e01aa3242d
4 changed files with 222 additions and 4 deletions

View File

@ -1,5 +1,7 @@
package geom
import "log"
// PointsF is a set of points.
type PointsF []PointF
@ -11,10 +13,97 @@ type PolygonF struct {
Points PointsF
}
// Add creates a new polyqon based on p with one or more extra points q.
func (p PolygonF) Add(q ...PointF) PolygonF {
// Add adds q as a vector to all points in p.
func (p PolygonF) Add(q PointF) PolygonF {
var r = p.copy()
for i, pt := range r.Points {
r.Points[i] = pt.Add(q)
}
return r
}
func (p PolygonF) copy() PolygonF {
var q = PolygonF{make(PointsF, len(p.Points))}
copy(q.Points, p.Points)
return q
}
// Extend creates a new polygon based on p with one or more extra points q.
func (p PolygonF) Extend(q ...PointF) PolygonF {
var t = PolygonF{make(PointsF, len(p.Points)+len(q))}
copy(t.Points, p.Points)
copy(t.Points[len(p.Points):], q)
return t
}
// Mul multiplies the points of polygon p with t and returns the result.
func (p PolygonF) Mul(t float64) PolygonF {
var q = p.copy()
for i, pt := range q.Points {
q.Points[i] = pt.Mul(t)
}
return q
}
// Triangulate triangulates the polygon p.
func (p PolygonF) Triangulate() []TriangleF {
var triangles []TriangleF
points := p.copy().Points[:]
n := len(points)
if n > 0 && points[0] == points[n-1] { // remove first point if polygon is closed
points = points[1:]
n--
}
triangle := func(i int) TriangleF {
return TrF(points[(i+n-1)%n], points[i], points[(i+1)%n])
}
ear := func(i int) (TriangleF, bool) {
t := triangle(i)
if t.Winding() == WindingCounterClockwise {
return TriangleF{}, false
}
for j := 0; j < n-3; j++ {
p := points[(i+2+j)%n]
if t.Contains(p) {
return TriangleF{}, false
}
}
return t, true
}
for n >= 3 {
leastSharp := -1
var leastSharpAngle float64
for i := range points {
if t, ok := ear(i); ok {
sharpAngle := t.SmallestAngle()
if leastSharp < 0 || sharpAngle > leastSharpAngle {
leastSharp = i
leastSharpAngle = sharpAngle
}
}
}
if leastSharp < 0 {
if n >= 3 {
log.Println("not fully triangulated")
}
break
}
triangles = append(triangles, triangle(leastSharp))
points = append(points[:leastSharp], points[leastSharp+1:]...)
n = len(points)
}
return triangles
}
// Reverse reverses the order of the points of polygon p.
func (p PolygonF) Reverse() PolygonF {
n := len(p.Points)
var q = PolygonF{make(PointsF, n)}
for i := 0; i < n/2; i++ {
q.Points[i], q.Points[n-i-1] = p.Points[n-i-1], p.Points[i]
}
return q
}

View File

@ -11,10 +11,34 @@ type PolygonF32 struct {
Points PointsF32
}
// Add creates a new polyqon based on p with one or more extra points q.
func (p PolygonF32) Add(q ...PointF32) PolygonF32 {
// Add adds q as a vector to all points in p.
func (p PolygonF32) Add(q PointF32) PolygonF32 {
var r = p.copy()
for i, pt := range r.Points {
r.Points[i] = pt.Add(q)
}
return r
}
func (p PolygonF32) copy() PolygonF32 {
var q = PolygonF32{make(PointsF32, len(p.Points))}
copy(q.Points, p.Points)
return q
}
// Extend creates a new polygon based on p with one or more extra points q.
func (p PolygonF32) Extend(q ...PointF32) PolygonF32 {
var t = PolygonF32{make(PointsF32, len(p.Points)+len(q))}
copy(t.Points, p.Points)
copy(t.Points[len(p.Points):], q)
return t
}
// Mul multiplies the points of polygon p with t and returns the result.
func (p PolygonF32) Mul(t float32) PolygonF32 {
var q = p.copy()
for i, pt := range q.Points {
q.Points[i] = pt.Mul(t)
}
return q
}

93
trianglef.go Normal file
View File

@ -0,0 +1,93 @@
package geom
import (
"math"
)
// TriangleF is defined by three points that describe a 2D triangle.
type TriangleF struct {
Points [3]PointF
}
// TrF is a shorthand method to create a TriangleF.
func TrF(q, r, s PointF) TriangleF {
return TriangleF{Points: [3]PointF{q, r, s}}
}
// Add translates the triangle by point p.
func (t TriangleF) Add(p PointF) TriangleF {
return TriangleF{Points: [3]PointF{
t.Points[0].Add(p),
t.Points[1].Add(p),
t.Points[2].Add(p),
}}
}
// Center gives the average of the three points.
func (t TriangleF) Center() PointF {
return t.Points[0].Add(t.Points[1]).Add(t.Points[2]).Mul(1. / 3)
}
// Contains tests if point p lies in the triangle.
func (t TriangleF) Contains(p PointF) bool {
u, v, w := t.Points[0], t.Points[1], t.Points[2]
const eps = 1e-9
q := 1 / (-v.Y*w.X + u.Y*(-v.X+w.X) + u.X*(v.Y-w.Y) + v.X*w.Y)
r := (u.Y*w.X - u.X*w.Y + (w.Y-u.Y)*p.X + (u.X-w.X)*p.Y) * q
s := (u.X*v.Y - u.Y*v.X + (u.Y-v.Y)*p.X + (v.X-u.X)*p.Y) * q
if r < -eps || r > 1+eps || s < -eps || s > 1+eps || r+s > 1+eps {
return false
}
return true
}
// Inset insets all the points of the triangle towards the center of the triangle with distance f.
func (t TriangleF) Inset(f float64) TriangleF {
center := t.Center()
inset := func(p PointF) PointF {
centered := p.Sub(center)
length := p.Distance(center)
factor := (length - f) / length
return center.Add(centered.Mul(factor))
}
return TriangleF{Points: [3]PointF{
inset(t.Points[0]),
inset(t.Points[1]),
inset(t.Points[2]),
}}
}
// Mul multiplies all the points of the triangle with f.
func (t TriangleF) Mul(f float64) TriangleF {
return TriangleF{Points: [3]PointF{
t.Points[0].Mul(f),
t.Points[1].Mul(f),
t.Points[2].Mul(f),
}}
}
// SmallestAngle returns the smallest/sharpest angle of the triangle.
func (t TriangleF) SmallestAngle() float64 {
u, v, w := t.Points[0], t.Points[1], t.Points[2]
q := math.Acos(w.Sub(u).Norm().Dot(v.Sub(u).Norm()))
r := math.Acos(u.Sub(v).Norm().Dot(w.Sub(v).Norm()))
s := math.Acos(v.Sub(w).Norm().Dot(u.Sub(w).Norm()))
return math.Min(q, math.Min(r, s))
}
// Winding determines the winding of the triangle.
func (t TriangleF) Winding() Winding {
u, v := t.Points[1].Sub(t.Points[0]), t.Points[2].Sub(t.Points[1])
if u.X*v.Y-u.Y*v.X > 0 {
return WindingClockwise
}
return WindingCounterClockwise
}
// Winding describes the order of points.
type Winding bool
const (
WindingClockwise Winding = false
WindingCounterClockwise Winding = true
)

12
trianglef_test.go Normal file
View File

@ -0,0 +1,12 @@
package geom
import (
"testing"
"github.com/stretchr/testify/assert"
)
func TestTriangleFSmallestAngle(t *testing.T) {
triangle := TrF(PtF(0, 0), PtF(0, 1), PtF(1, 1))
assert.InEpsilon(t, .25*Pi, triangle.SmallestAngle(), 1e-9)
}