Add main code

scripts/photo_to_kml.py

@@ -0,0 +1,137 @@
+"""Add geotagged images to a kml for viewing in Google Earth.
+
+This script takes an image (or a folder of images), and creates a placemark
+with a thumbnail of that image, based on the GPS coordinates in the EXIF tags.
+
+Examples:
+
+Create a kml for one photo.
+    > python photo_to_kml.py IMG_001100.jpg
+
+Create a kml for a folder of photos, named 'images'.
+    > python photo_to_kml.py images
+
+"""
+
+__author__ = "D. Howe"
+__version__ = "0.1.0"
+__email__ = "d.howe@wrl.unsw.edu.au"
+
+import sys
+import glob
+import simplekml
+from PIL import Image
+from PIL.ExifTags import TAGS, GPSTAGS
+
+
+def get_exif_data(image):
+    """
+    Return a dictionary from the exif data of an PIL Image object, and convert
+    the GPS uags.
+    https://gist.github.com/valgur/2fbed04680864fab1bfc
+    """
+
+    info = image._getexif()
+    if not info:
+        return {}
+    exif_data = {TAGS.get(tag, tag): value for tag, value in info.items()}
+
+    def is_fraction(val):
+        return isinstance(val, tuple) and len(val) == 2 and isinstance(
+            val[0], int) and isinstance(val[1], int)
+
+    def frac_to_dec(frac):
+        return float(frac[0]) / float(frac[1])
+
+    if 'GPSInfo' in exif_data:
+        gpsinfo = {GPSTAGS.get(t, t): v
+                   for t, v in exif_data['GPSInfo'].items()}
+        for tag, value in gpsinfo.items():
+            if is_fraction(value):
+                gpsinfo[tag] = frac_to_dec(value)
+            elif all(is_fraction(x) for x in value):
+                gpsinfo[tag] = tuple(map(frac_to_dec, value))
+        exif_data['GPSInfo'] = gpsinfo
+
+    return exif_data
+
+
+def get_lat_lon(exif_data):
+    """
+    Return the latitude and longitude (if available) from exif_data.
+    https://gist.github.com/valgur/2fbed04680864fab1bfc
+    """
+    lat = None
+    lon = None
+    gps_info = exif_data.get('GPSInfo')
+
+    def convert_to_degrees(value):
+        d, m, s = value
+        return d + (m / 60.0) + (s / 3600.0)
+
+    if gps_info:
+        gps_latitude = gps_info.get('GPSLatitude')
+        gps_latitude_ref = gps_info.get('GPSLatitudeRef')
+        gps_longitude = gps_info.get('GPSLongitude')
+        gps_longitude_ref = gps_info.get('GPSLongitudeRef')
+
+        if (gps_latitude and gps_latitude_ref and gps_longitude and
+                gps_longitude_ref):
+            lat = convert_to_degrees(gps_latitude)
+            if gps_latitude_ref == 'S':
+                lat = -lat
+
+            lon = convert_to_degrees(gps_longitude)
+            if gps_longitude_ref == 'W':
+                lon = -lon
+
+    return lat, lon
+
+
+def export_kml_file(file_names, kml_name):
+    """
+    Create the kml document
+    """
+    kml = simplekml.Kml()
+
+    for file_name in file_names:
+
+        print('Reading ' + file_name + '...')
+
+        with Image.open(file_name) as image:
+            exif_data = get_exif_data(image)
+
+        lat, lon = get_lat_lon(exif_data)
+
+        pnt = kml.newpoint(name=file_name)
+        pnt.coords = [(lon, lat)]
+
+        # Add comtent to popup window
+        pnt.description = (
+            '<![CDATA[ <img src=' + file_name + ' height="500px" />]]>')
+        pnt.stylemap.normalstyle.iconstyle.scale = 1
+        pnt.stylemap.normalstyle.iconstyle.icon.href = (
+            'http://maps.google.com/'
+            'mapfiles/kml/shapes/camera.png')
+        pnt.stylemap.highlightstyle.iconstyle.scale = 2
+        pnt.stylemap.highlightstyle.iconstyle.icon.href = file_name
+
+    kml.save(kml_name)
+
+
+def main():
+    parser = argparse.ArgumentParser(usage=__doc__)
+    parser.add_argument('-f',
+                        '--file',
+                        help='name of parameter file',
+                        default=None)
+
+    if len(sys.argv) == 1:
+        file_names = glob.glob('*.jpg')
+    else:
+        file_names = glob.glob(sys.argv[1])
+    export_kml_file(file_names, 'output.kml')
+
+
+if __name__ == '__main__':
+    main()