2013-04-21 18:37:05 +04:00
|
|
|
package geom
|
|
|
|
|
|
|
|
import (
|
2013-04-24 00:02:27 +04:00
|
|
|
"bytes"
|
2013-04-21 18:37:05 +04:00
|
|
|
"goposm/element"
|
2013-05-06 21:24:49 +04:00
|
|
|
"goposm/geom/geos"
|
2013-04-21 18:37:05 +04:00
|
|
|
"testing"
|
|
|
|
)
|
|
|
|
|
2013-04-24 00:02:27 +04:00
|
|
|
func TestLineString(t *testing.T) {
|
2013-04-21 18:37:05 +04:00
|
|
|
nodes := make([]element.Node, 2)
|
|
|
|
nodes[0] = element.Node{Lat: 0, Long: 0}
|
|
|
|
nodes[1] = element.Node{Lat: 0, Long: 10}
|
2013-05-06 21:24:49 +04:00
|
|
|
g := geos.NewGEOS()
|
|
|
|
defer g.Finish()
|
2013-05-16 14:17:21 +04:00
|
|
|
geom, err := LineStringWKB(g, nodes)
|
2013-04-24 00:02:27 +04:00
|
|
|
if err != nil {
|
|
|
|
t.Fatal(err)
|
|
|
|
}
|
2013-05-16 14:17:21 +04:00
|
|
|
if bytes.Compare(geom.Wkb[0:2], []byte{0x1, 0x2}) != 0 {
|
|
|
|
t.Errorf("%#v", geom.Wkb)
|
2013-04-21 18:37:05 +04:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2013-04-24 00:02:27 +04:00
|
|
|
func TestPolygon(t *testing.T) {
|
|
|
|
nodes := []element.Node{
|
|
|
|
element.Node{Lat: 0, Long: 0},
|
|
|
|
element.Node{Lat: 0, Long: 10},
|
|
|
|
element.Node{Lat: 10, Long: 10},
|
|
|
|
element.Node{Lat: 0, Long: 0},
|
|
|
|
}
|
2013-05-06 21:24:49 +04:00
|
|
|
g := geos.NewGEOS()
|
|
|
|
defer g.Finish()
|
2013-05-16 14:17:21 +04:00
|
|
|
geom, err := PolygonWKB(g, nodes)
|
2013-04-24 00:02:27 +04:00
|
|
|
if err != nil {
|
|
|
|
t.Fatal(err)
|
|
|
|
}
|
|
|
|
|
2013-05-16 14:17:21 +04:00
|
|
|
if bytes.Compare(geom.Wkb[0:2], []byte{0x1, 0x3}) != 0 {
|
|
|
|
t.Errorf("%#v", geom.Wkb)
|
2013-04-24 00:02:27 +04:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestPolygonNotClosed(t *testing.T) {
|
|
|
|
nodes := []element.Node{
|
|
|
|
element.Node{Lat: 0, Long: 0},
|
|
|
|
element.Node{Lat: 0, Long: 10},
|
|
|
|
element.Node{Lat: 10, Long: 10},
|
|
|
|
}
|
2013-05-06 21:24:49 +04:00
|
|
|
g := geos.NewGEOS()
|
|
|
|
defer g.Finish()
|
|
|
|
_, err := PolygonWKB(g, nodes)
|
2013-04-24 00:02:27 +04:00
|
|
|
if err == nil {
|
|
|
|
t.Fatal("no error")
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2013-04-21 18:37:05 +04:00
|
|
|
func BenchmarkLineString(b *testing.B) {
|
|
|
|
size := 16
|
|
|
|
nodes := make([]element.Node, size)
|
|
|
|
for i := 0; i < size; i++ {
|
|
|
|
nodes[i] = element.Node{Lat: 0, Long: float64(i)}
|
|
|
|
}
|
2013-05-06 21:24:49 +04:00
|
|
|
g := geos.NewGEOS()
|
|
|
|
defer g.Finish()
|
2013-04-21 18:37:05 +04:00
|
|
|
|
|
|
|
for i := 0; i < b.N; i++ {
|
2013-05-06 21:24:49 +04:00
|
|
|
LineStringWKB(g, nodes)
|
2013-04-21 18:37:05 +04:00
|
|
|
}
|
|
|
|
}
|