OpenHuntingData/build-gl-style.py at master · OpenBounds/OpenHuntingData · GitHub

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
#!/usr/bin/env python

import json
import logging
from optparse import OptionParser
import os
import sys
from shapely.geometry import shape

source_key_max_length = 100 #from gaiacloud/gaia/mapsources/models.py

def change_source_url(style, source_id, new_url):
    sources = style["sources"]
    for key, source in sources.iteritems():
        if key == source_id:
            source["tiles"] = [new_url]

    return style


def generate_style(template, source, destination, name=None, center=None):
    with open(template, "rb") as f:
        style_template = json.load(f)

    if name is None:
        name = source
    style_template["id"] = source
    style_template["name"] = name

    if center is not None:
        style_template["center"] = center

    server_style = dict(style_template)
    change_source_url(server_style, "data",
        "https://s3.amazonaws.com/data.openbounds.org/USAHunting/vector/" + \
        source + "/{z}/{x}/{y}.pbf")

    with open(os.path.join(destination), "wb") as f:
        json.dump(server_style, f, sort_keys=True, indent=4, separators=(',', ': '))


def generate_styles_from_catalog(template, catalog_path, destination_dir):
    with open(catalog_path, "rb") as f:
        catalog = json.load(f)
    tileserver_config = {}

    for feature in catalog['features']:
        # Data extraction
        feature_geometry = shape(feature['geometry'])
        path = feature['properties']['path'].split('.')[0]
        filename = path.replace("/", "_")[:source_key_max_length] + ".json"
        try:
            center = feature_geometry.centroid.coords[0]
        except:
            logging.error(feature)
            logging.error("Failed to get centroid for file " + filename)

        generate_style(template, path,
            os.path.join(destination_dir, filename),
            name=feature['properties']['name'],
            center=center)
        tileserver_config["/OpenHuntingData/" + path] = {
            "source": "gl:///opt/styles/OpenHuntingData/" + filename,
            "sourceMaxZoom": 13,
            "headers": {},
        }
    print json.dumps(tileserver_config, sort_keys=True, indent=4, separators=(',', ': '))

def _main():
    usage = "usage: %prog destination source"
    parser = OptionParser(usage=usage,
                          description="")
    parser.add_option("-d", "--debug", action="store_true", dest="debug",
                      help="Turn on debug logging")
    parser.add_option("-q", "--quiet", action="store_true", dest="quiet",
                      help="turn off all logging")

    (options, args) = parser.parse_args()

    logging.basicConfig(level=logging.DEBUG if options.debug else
    (logging.ERROR if options.quiet else logging.INFO))

    if len(args) != 2:
        logging.error("Error: wrong number of arguements, see --help")
        sys.exit(-1)

    destination, source = args

    if not os.path.exists(destination):
        logging.error("Destination directory does not exist")
        sys.exit(-1)

    if not os.path.isdir(destination):
        logging.error("destination is not a directory")
        sys.exit(-1)

    if os.path.basename(source) == "catalog.geojson":
        generate_styles_from_catalog("StyleTemplate.json", source, destination)
    else:
        source = source.strip("/")
        generate_style("StyleTemplate.json", source, destination)

if __name__ == "__main__":
    _main()