85 lines
2.3 KiB
Python
85 lines
2.3 KiB
Python
import fiona
|
|
from shapely.geometry import shape
|
|
import numpy as np
|
|
from tqdm import tqdm
|
|
import zipfile
|
|
from io import BytesIO
|
|
#from urllib.request import urlopen
|
|
import requests
|
|
import argparse
|
|
import os
|
|
|
|
def convertLinestring(linestring):
|
|
outlist = []
|
|
|
|
pointx = linestring.coords.xy[0]
|
|
pointy = linestring.coords.xy[1]
|
|
|
|
for j in range(len(pointx)):
|
|
outlist.extend([float(pointx[j]),float(pointy[j])])
|
|
|
|
outlist.extend([0,0])
|
|
return outlist
|
|
|
|
def extractLines(shapefile, tolerance):
|
|
print("Extracting map lines")
|
|
outlist = []
|
|
|
|
for i in tqdm(range(len(shapefile))):
|
|
|
|
simplified = shape(shapefile[i]['geometry'])#.simplify(tolerance, preserve_topology=False)
|
|
|
|
if(simplified.geom_type == "LineString"):
|
|
outlist.extend(convertLinestring(simplified))
|
|
|
|
elif(simplified.geom_type == "MultiPolygon" or simplified.geom_type == "Polygon"):
|
|
|
|
if(simplified.boundary.geom_type == "MultiLineString"):
|
|
for boundary in simplified.boundary:
|
|
outlist.extend(convertLinestring(boundary))
|
|
else:
|
|
outlist.extend(convertLinestring(simplified.boundary))
|
|
|
|
else:
|
|
print("Unsupported type: " + simplified.geom_type)
|
|
|
|
|
|
|
|
return outlist
|
|
|
|
parser = argparse.ArgumentParser(description='viz1090 Natural Earth Data Map Converter')
|
|
parser.add_argument("--tolerance", default=0.001, type=float, help="map simplification tolerance")
|
|
parser.add_argument("--scale", default="10m", choices=["10m","50m","110m"], type=str, help="map file scale")
|
|
parser.add_argument("mapfile", type=str, help="shapefile to load (e.g., from https://www.naturalearthdata.com/downloads/")
|
|
|
|
args = parser.parse_args()
|
|
|
|
shapefile = fiona.open(args.mapfile)
|
|
|
|
outlist = extractLines(shapefile, args.tolerance)
|
|
|
|
bin_file = open("airportdata.bin", "wb")
|
|
np.asarray(outlist).astype(np.single).tofile(bin_file)
|
|
bin_file.close()
|
|
|
|
print("Wrote %d points" % (len(outlist) / 2))
|
|
|
|
bin_file = open("airportnames", "w")
|
|
|
|
shapefile = fiona.open("ne_10m_airports.shp")
|
|
|
|
count = 0
|
|
|
|
for i in tqdm(range(len(shapefile))):
|
|
|
|
xcoord = shapefile[i]['geometry']['coordinates'][0]
|
|
ycoord = shapefile[i]['geometry']['coordinates'][1]
|
|
name = shapefile[i]['properties']['iata_code']
|
|
|
|
outstring = "{0} {1} {2}\n".format(xcoord, ycoord, name)
|
|
bin_file.write(outstring)
|
|
count = count + 1
|
|
|
|
bin_file.close()
|
|
|
|
print("Wrote %d airport names" % count) |