66 lines
1.9 KiB
Python
66 lines
1.9 KiB
Python
from lxml import etree as ElementTree
|
|
import numpy as np
|
|
import sys
|
|
from tqdm import tqdm
|
|
import argparse
|
|
|
|
parser = argparse.ArgumentParser(description='viz1090 SVG Map Converter')
|
|
parser.add_argument("--resolution", default=250, type=int, nargs=1, help="downsample resolution")
|
|
parser.add_argument("file", nargs=1, help="filename")
|
|
|
|
args = parser.parse_args()
|
|
|
|
|
|
if(len(args.file) == 0):
|
|
print("No input filename given")
|
|
exit()
|
|
|
|
parser = ElementTree.XMLParser(recover=True)
|
|
tree = ElementTree.parse(args.file[0], parser)
|
|
polys = tree.xpath('//polygon')
|
|
|
|
bin_file = open("mapdata.bin", "wb")
|
|
|
|
outlist = []
|
|
|
|
|
|
resolution = args.resolution[0]
|
|
|
|
print("Reading points")
|
|
for i in tqdm(range(len(polys))):
|
|
#for i in range(40):
|
|
p = polys[i]
|
|
currentPoints = (p.attrib['points']).replace(","," ").split()
|
|
|
|
if(len(currentPoints) == 14): #remove little circles in the McCurley maps
|
|
continue
|
|
|
|
prevx = 0
|
|
prevy = 0
|
|
|
|
temp = []
|
|
|
|
for i in range(int(len(currentPoints)/2)):
|
|
#currentPoints[2 * i + 0] = "%.*f" % (precision, float(currentPoints[2 * i + 0]))
|
|
#currentPoints[2 * i + 1] = "%.*f" % (precision, float(currentPoints[2 * i + 1]))
|
|
|
|
currentPoints[2 * i + 0] = float(int(resolution * float(currentPoints[2 * i + 0]))) / resolution
|
|
currentPoints[2 * i + 1] = float(int(resolution * float(currentPoints[2 * i + 1]))) / resolution
|
|
|
|
if(currentPoints[2 * i + 0] != prevx or currentPoints[2 * i + 1] != prevy):
|
|
temp.extend([currentPoints[2 * i + 0],currentPoints[2 * i + 1]])
|
|
|
|
prevx = currentPoints[2 * i + 0]
|
|
prevy = currentPoints[2 * i + 1]
|
|
|
|
if(len(currentPoints) > 6): #must be at least a triangle
|
|
outlist.extend(temp)
|
|
#outlist.extend([temp[0],temp[1]])
|
|
outlist.extend(["0","0"])
|
|
|
|
|
|
np.asarray(outlist).astype(np.single).tofile(bin_file)
|
|
bin_file.close()
|
|
|
|
print("Wrote %d points" % (len(outlist) / 2))
|