93 lines
3.4 KiB
Python
93 lines
3.4 KiB
Python
|
# Copyright Jonah Lefkoff
|
||
|
|
|
||
|
|
import xml.etree.ElementTree as ET
|
||
|
|
import re
|
||
|
|
import os
|
||
|
|
import argparse
|
||
|
|
import zipfile
|
||
|
|
from geopy.distance import distance
|
||
|
|
import json
|
||
|
|
|
||
|
|
|
||
|
|
# arg parser
|
||
|
|
parser = argparse.ArgumentParser(
|
||
|
|
description='Converts KMZ radar video maps to GeoJSON.')
|
||
|
|
parser.add_argument('kmz_path', metavar='kmz_path', type=str,
|
||
|
|
help='The path to a folder of KMZ files to be converted.')
|
||
|
|
parser.add_argument('json_out', metavar='json_out', type=str,
|
||
|
|
help='The path to output the GeoJSON files.')
|
||
|
|
parser.add_argument('-r', '--radius',
|
||
|
|
help='The maximum radius of the map in nautical miles. Defaults to 100.', type=int, required=False)
|
||
|
|
args = parser.parse_args()
|
||
|
|
|
||
|
|
# takes a file path to a KMZ file and converts it to a kml to pass to the kml_to_geojson function
|
||
|
|
def kmz_to_kml(fname):
|
||
|
|
zf = zipfile.ZipFile(fname,'r')
|
||
|
|
for fn in zf.namelist():
|
||
|
|
if fn.endswith('.kml'):
|
||
|
|
content = zf.read(fn)
|
||
|
|
if args.radius:
|
||
|
|
parse_kml_to_geojson(content, fname, args.radius)
|
||
|
|
else:
|
||
|
|
parse_kml_to_geojson(content, fname)
|
||
|
|
else:
|
||
|
|
print("no kml file")
|
||
|
|
|
||
|
|
# takes a kml file and converts it to a GeoJSON file and writes it to the json_out path
|
||
|
|
|
||
|
|
def parse_kml_to_geojson(kml, fname, max_radius=100):
|
||
|
|
root = ET.fromstring(kml)
|
||
|
|
|
||
|
|
features = []
|
||
|
|
|
||
|
|
look_at_coordinates = (
|
||
|
|
float(root.find('.//{http://earth.google.com/kml/2.0}LookAt/{http://earth.google.com/kml/2.0}longitude').text),
|
||
|
|
float(root.find('.//{http://earth.google.com/kml/2.0}LookAt/{http://earth.google.com/kml/2.0}latitude').text)
|
||
|
|
)
|
||
|
|
|
||
|
|
for line_string in root.findall('.//{http://earth.google.com/kml/2.0}LineString'):
|
||
|
|
coordinates = line_string.find('{http://earth.google.com/kml/2.0}coordinates').text.strip().split()
|
||
|
|
|
||
|
|
coordinates_list = []
|
||
|
|
for coord in coordinates:
|
||
|
|
lon, lat, _ = coord.split(',')
|
||
|
|
coordinates_list.append([float(lon), float(lat)])
|
||
|
|
|
||
|
|
# Calculate distance between each point in LineString and LookAt coordinates, needs to be lat long not long lat
|
||
|
|
distances = []
|
||
|
|
for coord in coordinates_list:
|
||
|
|
distances.append(distance([coord[1],coord[0]], [look_at_coordinates[1],look_at_coordinates[0]]).nautical)
|
||
|
|
|
||
|
|
# Filter LineStrings based on the maximum radius
|
||
|
|
if max(distances) <= max_radius:
|
||
|
|
feature = {
|
||
|
|
"type": "Feature",
|
||
|
|
"properties": {
|
||
|
|
"color": "#ffffff",
|
||
|
|
"style": "solid",
|
||
|
|
"thickness": "1",
|
||
|
|
},
|
||
|
|
"geometry": {
|
||
|
|
"type": "LineString",
|
||
|
|
"coordinates": coordinates_list
|
||
|
|
}
|
||
|
|
}
|
||
|
|
features.append(feature)
|
||
|
|
|
||
|
|
geojson_data = {
|
||
|
|
"type": "FeatureCollection",
|
||
|
|
"features": features
|
||
|
|
}
|
||
|
|
|
||
|
|
# get the map name from the file name
|
||
|
|
map_name = re.search(r'([a-zA-Z0-9]+)\.kmz', fname).group(1)
|
||
|
|
# create the output file named after the input file
|
||
|
|
with open(f"{args.json_out}/{map_name}.geojson", "w+") as output_file:
|
||
|
|
# write to the file
|
||
|
|
json.dump(geojson_data, output_file, indent=2)
|
||
|
|
|
||
|
|
|
||
|
|
# loop through each file in the kmz_path and run the kmz_to_kml function on it
|
||
|
|
for fname in os.listdir(args.kmz_path):
|
||
|
|
kmz_to_kml(f"{args.kmz_path}/{fname}")
|