82 lines
1.5 KiB
Go
82 lines
1.5 KiB
Go
package writer
|
|
|
|
import (
|
|
"imposm3/cache"
|
|
"imposm3/database"
|
|
"imposm3/element"
|
|
"imposm3/geom"
|
|
"imposm3/geom/geos"
|
|
"imposm3/proj"
|
|
"imposm3/stats"
|
|
"sync"
|
|
)
|
|
|
|
type NodeWriter struct {
|
|
OsmElemWriter
|
|
nodes chan *element.Node
|
|
}
|
|
|
|
func NewNodeWriter(osmCache *cache.OSMCache, nodes chan *element.Node,
|
|
inserter database.Inserter, progress *stats.Statistics,
|
|
srid int) *OsmElemWriter {
|
|
nw := NodeWriter{
|
|
OsmElemWriter: OsmElemWriter{
|
|
osmCache: osmCache,
|
|
progress: progress,
|
|
wg: &sync.WaitGroup{},
|
|
inserter: inserter,
|
|
srid: srid,
|
|
},
|
|
nodes: nodes,
|
|
}
|
|
nw.OsmElemWriter.writer = &nw
|
|
return &nw.OsmElemWriter
|
|
}
|
|
|
|
func (nw *NodeWriter) loop() {
|
|
geos := geos.NewGeos()
|
|
geos.SetHandleSrid(nw.srid)
|
|
defer geos.Finish()
|
|
|
|
for n := range nw.nodes {
|
|
nw.progress.AddNodes(1)
|
|
if ok, matches := nw.inserter.ProbePoint(n.OSMElem); ok {
|
|
proj.NodeToMerc(n)
|
|
if nw.expireTiles != nil {
|
|
nw.expireTiles.ExpireFromNodes([]element.Node{*n})
|
|
}
|
|
point, err := geom.Point(geos, *n)
|
|
if err != nil {
|
|
if err, ok := err.(ErrorLevel); ok {
|
|
if err.Level() <= 0 {
|
|
continue
|
|
}
|
|
}
|
|
log.Warn(err)
|
|
continue
|
|
}
|
|
|
|
n.Geom, err = geom.AsGeomElement(geos, point)
|
|
if err != nil {
|
|
log.Warn(err)
|
|
continue
|
|
}
|
|
|
|
if nw.limiter != nil {
|
|
parts, err := nw.limiter.Clip(n.Geom.Geom)
|
|
if err != nil {
|
|
log.Warn(err)
|
|
continue
|
|
}
|
|
if len(parts) >= 1 {
|
|
nw.inserter.InsertPoint(n.OSMElem, matches)
|
|
}
|
|
} else {
|
|
nw.inserter.InsertPoint(n.OSMElem, matches)
|
|
}
|
|
|
|
}
|
|
}
|
|
nw.wg.Done()
|
|
}
|