primera subida

lib/convert.py

@@ -0,0 +1,92 @@
+# 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}")