"""Add geotagged images to a kml for viewing in Google Earth.

This script takes a folder of images, and creates a placemark with a thumbnail
of each image, based on the GPS coordinates in the EXIF tags.

Examples:

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 os
import argparse
from PIL import Image
from PIL.ExifTags import TAGS, GPSTAGS
import simplekml


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(dirname, fnames, kml_name):
    """
    Create the kml document
    """
    kml = simplekml.Kml()

    for fname in fnames:
        print('Reading {}...'.format(fname))
        with Image.open(os.path.join(dirname, fname)) as image:
            exif_data = get_exif_data(image)

        lat, lon = get_lat_lon(exif_data)
        pnt = kml.newpoint(name=fname)
        pnt.coords = [(lon, lat)]

        # Add content to popup window
        pnt.description = ('<![CDATA[ <img src={} height="500px" />]]>'.format(
            os.path.join(dirname, fname)))
        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 = os.path.join(
            dirname, fname)

    kml.save(kml_name)


def main():
    parser = argparse.ArgumentParser(usage=__doc__)
    parser.add_argument(
        'folder', nargs='?', help='name of input folder', default=None)
    args = parser.parse_args()

    # Get files in image directory
    ext = ['.jpg', '.jpeg', '.tif', '.tiff']
    dirname = args.folder
    fnames = [
        f for f in os.listdir(dirname)
        if os.path.splitext(f)[-1].lower() in ext
    ]

    # Create kml file
    kml_name = dirname + '.kml'
    export_kml_file(dirname, fnames, kml_name)


if __name__ == '__main__':
    main()