147 lines
3.1 KiB
Go
147 lines
3.1 KiB
Go
package writer
|
|
|
|
import (
|
|
"fmt"
|
|
"imposm3/cache"
|
|
"imposm3/database"
|
|
"imposm3/element"
|
|
"imposm3/geom"
|
|
"imposm3/geom/geos"
|
|
"imposm3/mapping"
|
|
"imposm3/proj"
|
|
"imposm3/stats"
|
|
"log"
|
|
"sync"
|
|
)
|
|
|
|
type RelationWriter struct {
|
|
OsmElemWriter
|
|
rel chan *element.Relation
|
|
tagMatcher *mapping.TagMatcher
|
|
}
|
|
|
|
func NewRelationWriter(osmCache *cache.OSMCache, diffCache *cache.DiffCache, rel chan *element.Relation,
|
|
insertBuffer database.RowInserter, tagMatcher *mapping.TagMatcher, progress *stats.Statistics,
|
|
srid int) *OsmElemWriter {
|
|
rw := RelationWriter{
|
|
OsmElemWriter: OsmElemWriter{
|
|
osmCache: osmCache,
|
|
diffCache: diffCache,
|
|
progress: progress,
|
|
wg: &sync.WaitGroup{},
|
|
insertBuffer: insertBuffer,
|
|
srid: srid,
|
|
},
|
|
rel: rel,
|
|
tagMatcher: tagMatcher,
|
|
}
|
|
rw.OsmElemWriter.writer = &rw
|
|
return &rw.OsmElemWriter
|
|
}
|
|
|
|
func (rw *RelationWriter) loop() {
|
|
geos := geos.NewGeos()
|
|
geos.SetHandleSrid(rw.srid)
|
|
defer geos.Finish()
|
|
|
|
NextRel:
|
|
for r := range rw.rel {
|
|
rw.progress.AddRelations(1)
|
|
err := rw.osmCache.Ways.FillMembers(r.Members)
|
|
if err != nil {
|
|
if err != cache.NotFound {
|
|
fmt.Println(err)
|
|
}
|
|
continue NextRel
|
|
}
|
|
for _, m := range r.Members {
|
|
if m.Way == nil {
|
|
continue
|
|
}
|
|
err := rw.osmCache.Coords.FillWay(m.Way)
|
|
if err != nil {
|
|
if err != cache.NotFound {
|
|
fmt.Println(err)
|
|
}
|
|
continue NextRel
|
|
}
|
|
proj.NodesToMerc(m.Way.Nodes)
|
|
}
|
|
|
|
// BuildRelation updates r.Members but we need all of them
|
|
// for the diffCache
|
|
allMembers := r.Members
|
|
|
|
// prepare relation first (build rings and compute actual
|
|
// relation tags)
|
|
prepedRel, err := geom.PrepareRelation(r, rw.srid)
|
|
if err != nil {
|
|
if err, ok := err.(ErrorLevel); ok {
|
|
if err.Level() <= 0 {
|
|
continue NextRel
|
|
}
|
|
}
|
|
log.Println(err)
|
|
continue NextRel
|
|
}
|
|
|
|
// check for matches befor building the geometry
|
|
matches := rw.tagMatcher.Match(&r.Tags)
|
|
if len(matches) == 0 {
|
|
continue NextRel
|
|
}
|
|
|
|
// build the multipolygon
|
|
r, err = prepedRel.Build()
|
|
if err != nil {
|
|
if r.Geom != nil && r.Geom.Geom != nil {
|
|
geos.Destroy(r.Geom.Geom)
|
|
}
|
|
if err, ok := err.(ErrorLevel); ok {
|
|
if err.Level() <= 0 {
|
|
continue NextRel
|
|
|
|
}
|
|
}
|
|
log.Println(err)
|
|
continue NextRel
|
|
}
|
|
|
|
if rw.limiter != nil {
|
|
parts, err := rw.limiter.Clip(r.Geom.Geom)
|
|
if err != nil {
|
|
log.Println(err)
|
|
continue NextRel
|
|
}
|
|
for _, g := range parts {
|
|
rel := element.Relation(*r)
|
|
rel.Geom = &element.Geometry{g, geos.AsEwkbHex(g)}
|
|
rw.insertMatches(&rel.OSMElem, matches)
|
|
}
|
|
} else {
|
|
rw.insertMatches(&r.OSMElem, matches)
|
|
}
|
|
err = rw.osmCache.InsertedWays.PutMembers(r.Members)
|
|
if err != nil {
|
|
fmt.Println(err)
|
|
}
|
|
if rw.diffCache != nil {
|
|
rw.diffCache.Ways.AddFromMembers(r.Id, allMembers)
|
|
for _, member := range allMembers {
|
|
if member.Way != nil {
|
|
rw.diffCache.Coords.AddFromWay(member.Way)
|
|
}
|
|
}
|
|
}
|
|
if rw.expireTiles != nil {
|
|
for _, m := range allMembers {
|
|
if m.Way != nil {
|
|
rw.expireTiles.ExpireFromNodes(m.Way.Nodes)
|
|
}
|
|
}
|
|
}
|
|
geos.Destroy(r.Geom.Geom)
|
|
}
|
|
rw.wg.Done()
|
|
}
|