123 lines
2.8 KiB
Go
123 lines
2.8 KiB
Go
package mapping
|
|
|
|
import (
|
|
"imposm3/element"
|
|
)
|
|
|
|
func (m *Mapping) NodeTagFilter() *TagFilter {
|
|
mappings := make(map[string]map[string][]DestTable)
|
|
m.mappings("point", mappings)
|
|
tags := make(map[string]bool)
|
|
m.extraTags("point", tags)
|
|
return &TagFilter{mappings, tags}
|
|
}
|
|
|
|
func (m *Mapping) WayTagFilter() *TagFilter {
|
|
mappings := make(map[string]map[string][]DestTable)
|
|
m.mappings("linestring", mappings)
|
|
m.mappings("polygon", mappings)
|
|
tags := make(map[string]bool)
|
|
m.extraTags("linestring", tags)
|
|
m.extraTags("polygon", tags)
|
|
return &TagFilter{mappings, tags}
|
|
}
|
|
|
|
func (m *Mapping) RelationTagFilter() *RelationTagFilter {
|
|
mappings := make(map[string]map[string][]DestTable)
|
|
m.mappings("linestring", mappings)
|
|
m.mappings("polygon", mappings)
|
|
tags := make(map[string]bool)
|
|
m.extraTags("linestring", tags)
|
|
m.extraTags("polygon", tags)
|
|
// do not filter out type tag
|
|
mappings["type"] = map[string][]DestTable{
|
|
"multipolygon": []DestTable{},
|
|
"boundary": []DestTable{},
|
|
"land_area": []DestTable{},
|
|
}
|
|
return &RelationTagFilter{TagFilter{mappings, tags}}
|
|
}
|
|
|
|
type TagFilter struct {
|
|
mappings map[string]map[string][]DestTable
|
|
extraTags map[string]bool
|
|
}
|
|
|
|
type RelationTagFilter struct {
|
|
TagFilter
|
|
}
|
|
|
|
func (f *TagFilter) Filter(tags *element.Tags) bool {
|
|
if tags == nil {
|
|
return false
|
|
}
|
|
foundMapping := false
|
|
for k, v := range *tags {
|
|
values, ok := f.mappings[k]
|
|
if ok {
|
|
if _, ok := values["__any__"]; ok {
|
|
foundMapping = true
|
|
continue
|
|
} else if _, ok := values[v]; ok {
|
|
foundMapping = true
|
|
continue
|
|
} else if _, ok := f.extraTags[k]; !ok {
|
|
delete(*tags, k)
|
|
}
|
|
} else if _, ok := f.extraTags[k]; !ok {
|
|
delete(*tags, k)
|
|
}
|
|
}
|
|
if foundMapping {
|
|
return true
|
|
} else {
|
|
*tags = nil
|
|
return false
|
|
}
|
|
}
|
|
|
|
func (f *RelationTagFilter) Filter(tags *element.Tags) bool {
|
|
if tags == nil {
|
|
return false
|
|
}
|
|
if t, ok := (*tags)["type"]; ok {
|
|
if t != "multipolygon" && t != "boundary" && t != "land_area" {
|
|
*tags = nil
|
|
return false
|
|
}
|
|
if t == "boundary" {
|
|
if _, ok := (*tags)["boundary"]; !ok {
|
|
// a lot of the boundary relations are not multipolygon
|
|
// only import with boundary tags (e.g. boundary=administrative)
|
|
*tags = nil
|
|
return false
|
|
}
|
|
}
|
|
} else {
|
|
*tags = nil
|
|
return false
|
|
}
|
|
tagCount := len(*tags)
|
|
f.TagFilter.Filter(tags)
|
|
|
|
// we removed tags...
|
|
if len(*tags) < tagCount {
|
|
expectedTags := 0
|
|
if _, ok := (*tags)["name"]; ok {
|
|
expectedTags += 1
|
|
}
|
|
if _, ok := (*tags)["type"]; ok {
|
|
expectedTags += 1
|
|
}
|
|
if len(*tags) == expectedTags {
|
|
// but no tags except name and type are left
|
|
// remove all, otherwise tags from longest
|
|
// way/ring would be used during MP building
|
|
*tags = nil
|
|
return false
|
|
}
|
|
}
|
|
// always return true here since we found a matching type
|
|
return true
|
|
}
|