mirror of https://github.com/tidwall/tile38.git
113 lines
2.8 KiB
Go
113 lines
2.8 KiB
Go
package tools
|
|
|
|
import (
|
|
"fmt"
|
|
"strconv"
|
|
"strings"
|
|
)
|
|
|
|
// RTree interface
|
|
type RTree interface {
|
|
Insert(min, max []float64, value interface{})
|
|
Scan(iter func(min, max []float64, value interface{}) bool)
|
|
Search(min, max []float64, iter func(min, max []float64,
|
|
value interface{}) bool)
|
|
Delete(min, max []float64, value interface{})
|
|
Traverse(iter func(min, max []float64, height, level int,
|
|
value interface{}) int)
|
|
Count() int
|
|
TotalOverlapArea() float64
|
|
Nearby(min, max []float64, iter func(min, max []float64,
|
|
item interface{}) bool)
|
|
}
|
|
|
|
func svg(min, max []float64, height int) string {
|
|
var out string
|
|
point := true
|
|
for i := 0; i < 2; i++ {
|
|
if min[i] != max[i] {
|
|
point = false
|
|
break
|
|
}
|
|
}
|
|
if point { // is point
|
|
out += fmt.Sprintf(
|
|
"<rect x=\"%.0f\" y=\"%0.f\" width=\"%0.f\" height=\"%0.f\" "+
|
|
"stroke=\"%s\" fill=\"purple\" "+
|
|
"fill-opacity=\"0\" stroke-opacity=\"1\" "+
|
|
"rx=\"15\" ry=\"15\"/>\n",
|
|
(min[0])*svgScale,
|
|
(min[1])*svgScale,
|
|
(max[0]-min[0]+1/svgScale)*svgScale,
|
|
(max[1]-min[1]+1/svgScale)*svgScale,
|
|
strokes[height%len(strokes)])
|
|
} else { // is rect
|
|
out += fmt.Sprintf(
|
|
"<rect x=\"%.0f\" y=\"%0.f\" width=\"%0.f\" height=\"%0.f\" "+
|
|
"stroke=\"%s\" fill=\"purple\" "+
|
|
"fill-opacity=\"0\" stroke-opacity=\"1\"/>\n",
|
|
(min[0])*svgScale,
|
|
(min[1])*svgScale,
|
|
(max[0]-min[0]+1/svgScale)*svgScale,
|
|
(max[1]-min[1]+1/svgScale)*svgScale,
|
|
strokes[height%len(strokes)])
|
|
}
|
|
return out
|
|
}
|
|
|
|
const (
|
|
// Continue to first child rectangle and/or next sibling.
|
|
Continue = iota
|
|
// Ignore child rectangles but continue to next sibling.
|
|
Ignore
|
|
// Stop iterating
|
|
Stop
|
|
)
|
|
|
|
const svgScale = 4.0
|
|
|
|
var strokes = [...]string{"black", "#cccc00", "green", "red", "purple"}
|
|
|
|
// SVG prints 2D rtree in wgs84 coordinate space
|
|
func SVG(tr RTree) string {
|
|
var out string
|
|
out += fmt.Sprintf("<svg viewBox=\"%.0f %.0f %.0f %.0f\" "+
|
|
"xmlns =\"http://www.w3.org/2000/svg\">\n",
|
|
-190.0*svgScale, -100.0*svgScale,
|
|
380.0*svgScale, 190.0*svgScale)
|
|
|
|
out += fmt.Sprintf("<g transform=\"scale(1,-1)\">\n")
|
|
var outb []byte
|
|
tr.Traverse(func(min, max []float64, height, level int, _ interface{}) int {
|
|
outb = append(outb, svg(min, max, height)...)
|
|
return Continue
|
|
})
|
|
out += string(outb)
|
|
out += fmt.Sprintf("</g>\n")
|
|
out += fmt.Sprintf("</svg>\n")
|
|
return out
|
|
}
|
|
|
|
// Cities returns big list of cities base on json from
|
|
// https://github.com/lutangar/cities.json
|
|
func Cities(bigJSON string) [][2]float64 {
|
|
var out [][2]float64
|
|
s := bigJSON
|
|
for i := 0; ; i++ {
|
|
idx := strings.Index(s, `"lat": "`)
|
|
if idx == -1 {
|
|
break
|
|
}
|
|
s = s[idx+8:]
|
|
idx = strings.IndexByte(s, '"')
|
|
lat, _ := strconv.ParseFloat(s[:idx], 64)
|
|
idx = strings.Index(s, `"lng": "`)
|
|
s = s[idx+8:]
|
|
idx = strings.IndexByte(s, '"')
|
|
lng, _ := strconv.ParseFloat(s[:idx], 64)
|
|
s = s[idx+1:]
|
|
out = append(out, [2]float64{lng, lat})
|
|
}
|
|
return out
|
|
}
|