92 lines
1.9 KiB
Go
92 lines
1.9 KiB
Go
package writer
|
|
|
|
import (
|
|
"github.com/omniscale/imposm3/cache"
|
|
"github.com/omniscale/imposm3/database"
|
|
"github.com/omniscale/imposm3/element"
|
|
"github.com/omniscale/imposm3/geom"
|
|
"github.com/omniscale/imposm3/geom/geos"
|
|
"github.com/omniscale/imposm3/mapping"
|
|
"github.com/omniscale/imposm3/stats"
|
|
"sync"
|
|
)
|
|
|
|
type NodeWriter struct {
|
|
OsmElemWriter
|
|
nodes chan *element.Node
|
|
pointMatcher mapping.NodeMatcher
|
|
}
|
|
|
|
func NewNodeWriter(
|
|
osmCache *cache.OSMCache,
|
|
nodes chan *element.Node,
|
|
inserter database.Inserter,
|
|
progress *stats.Statistics,
|
|
matcher mapping.NodeMatcher,
|
|
srid int,
|
|
) *OsmElemWriter {
|
|
nw := NodeWriter{
|
|
OsmElemWriter: OsmElemWriter{
|
|
osmCache: osmCache,
|
|
progress: progress,
|
|
wg: &sync.WaitGroup{},
|
|
inserter: inserter,
|
|
srid: srid,
|
|
},
|
|
pointMatcher: matcher,
|
|
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 matches := nw.pointMatcher.MatchNode(n); len(matches) > 0 {
|
|
nw.NodeToSrid(n)
|
|
if nw.expireor != nil {
|
|
nw.expireor.Expire(n.Long, n.Lat)
|
|
}
|
|
point, err := geom.Point(geos, *n)
|
|
if err != nil {
|
|
if errl, ok := err.(ErrorLevel); !ok || errl.Level() > 0 {
|
|
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 {
|
|
if err := nw.inserter.InsertPoint(n.OSMElem, matches); err != nil {
|
|
log.Warn(err)
|
|
continue
|
|
}
|
|
}
|
|
} else {
|
|
if err := nw.inserter.InsertPoint(n.OSMElem, matches); err != nil {
|
|
log.Warn(err)
|
|
continue
|
|
}
|
|
}
|
|
|
|
}
|
|
}
|
|
nw.wg.Done()
|
|
}
|