-
Notifications
You must be signed in to change notification settings - Fork 32
/
bboxtagfilter.go
108 lines (94 loc) · 2.19 KB
/
bboxtagfilter.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
package bboxtagfilter
import (
"bbox"
"gopnik"
"plugins"
json "github.com/orofarne/strict-json"
)
type filterItem struct {
BBoxes []bbox.BBox
Drop []string
Add []string
DropOtherwise []string
AddOtherwise []string
}
type bBoxTagFilterPluginConfig struct {
Rules []filterItem
}
type BBoxTagFilterPlugin struct {
rules []filterItem
}
func strArrDrop(arr []string, elem string) []string {
for i := 0; i < len(arr); i++ {
if arr[i] == elem {
// Remove i-th element
arr = arr[:i+copy(arr[i:], arr[i+1:])]
}
}
return arr
}
func strArrAdd(arr []string, elem string) []string {
// Search if elem is also present in the slice
for i := 0; i < len(arr); i++ {
if arr[i] == elem {
return arr
}
}
return append(arr, elem)
}
func (self *BBoxTagFilterPlugin) applyFilterItem(coord *gopnik.TileCoord, fi *filterItem, inBb bool) {
if inBb {
for _, tag := range fi.Drop {
coord.Tags = strArrDrop(coord.Tags, tag)
}
for _, tag := range fi.Add {
coord.Tags = strArrAdd(coord.Tags, tag)
}
} else {
for _, tag := range fi.DropOtherwise {
coord.Tags = strArrDrop(coord.Tags, tag)
}
for _, tag := range fi.AddOtherwise {
coord.Tags = strArrAdd(coord.Tags, tag)
}
}
}
func (self *BBoxTagFilterPlugin) Filter(coord gopnik.TileCoord) (gopnik.TileCoord, error) {
L:
for _, rule := range self.rules {
// Check bboxes
for _, bb := range rule.BBoxes {
if bb.Crosses(coord) {
self.applyFilterItem(&coord, &rule, true)
continue L
}
}
self.applyFilterItem(&coord, &rule, false)
}
return coord, nil
}
func (self *BBoxTagFilterPlugin) Configure(cfg json.RawMessage) error {
var config bBoxTagFilterPluginConfig
err := json.Unmarshal(cfg, &config)
if err != nil {
return err
}
self.rules = config.Rules
return nil
}
type BBoxTagFilterPluginFactory struct {
}
func (self *BBoxTagFilterPluginFactory) Name() string {
return "BBoxTagFilterPlugin"
}
func (self *BBoxTagFilterPluginFactory) New(cfg json.RawMessage) (interface{}, error) {
res := new(BBoxTagFilterPlugin)
err := res.Configure(cfg)
if err != nil {
return nil, err
}
return res, nil
}
func init() {
plugins.DefaultPluginStore.AddPlugin(new(BBoxTagFilterPluginFactory))
}