image_functions.py

# -*- coding: utf-8 -*-

# image_functions.py
# This helper file does image manipulation and display

# Copyright (c) 2022, Harry van der Wolf. all rights reserved.
# This program or module is free software: you can redistribute it and/or
# modify it under the terms of the GNU General Public Licence as published
# by the Free Software Foundation, either version 2 of the Licence, or
# version 3 of the Licence, or (at your option) any later version. It is
# provided for educational purposes and is distributed in the hope that
# it will be useful, but WITHOUT ANY WARRANTY; without even the implied
# warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See
# the GNU General Public Licence for more details.

import io, os, platform, sys
from pathlib import Path
import tkinter as tk
import PySimpleGUI as sg
from PIL import Image, ExifTags  # to manipulate images and read exif
Image.LOAD_TRUNCATED_IMAGES = True
from PIL.ExifTags import TAGS

import file_functions
import ui_actions


def reworkTag(tagtuple, precision):
    '''
    This function returns the value but reformatted in the
    desired numbers after the decimal point

    :param tagtuple:     tuple containing the key and value of the exif tag
    :type tagtuple:      (tuple)
    :param precision:    defines the numbers after the decimal point
    :type precision:     (int)
    '''
    value = round(tagtuple[0] / tagtuple[1], precision)
    # print(v[0], ' # ', v[1], ' => ', value)
    return value


def reworkExposure(tagtuple):
    value = "1/" + str(tagtuple[1] // 10)
    return str(value)

# This is just an example function which I might need
def get_identifier_tag(filename):
    im = Image.open(filename)
    exifd = im.getexif()
    keys = list(exifd.keys())
    for k, v in im.getexif().items():
        tag = TAGS.get(k)
        print(k, '\t', tag)
    im.close()

# Read all the exif info from the loaded images, if available
def get_all_exif_info(filename):
    '''
    This function reads all the exif info from the loaded images, if available

    :param filename:   the full path of the filename to retrieve the info from
    :type filename:    (str)
    '''
    exif_dictionary = {}
    reference_image = ""
    read_error = ""
    try:
        # We need a try here in case we have an image we can't open (corrupt, not an image, direct RAW image)
        img = Image.open(filename)
        try:
            exif = img.getexif()
            if exif is not None:
                print('image: ', filename)
                exif_dictionary['filename'] = img.filename
                exif_dictionary['width x height'] = img.size
                for k, v in img.getexif().items():
                    tag = TAGS.get(k)
                    exif_dictionary[tag] = v
                    if (tag == 'ExposureBiasValue'):
                        print(TAGS.get(k), ' : ', v)
                        if v == '0.0' or v == '0' or v == 0.0 or v == 0:
                            reference_image = filename
                            print('reference_image: ', reference_image)
                # On pillow after jan 2nd 2022 we need additionally the exif.get_ifd(0x8769)
                for k, v in img.getexif().get_ifd(0x8769).items():
                    tag = TAGS.get(k)
                    exif_dictionary[tag] = v
                    if (tag == 'ExposureBiasValue'):
                        print(TAGS.get(k), ' : ', v)
                        if v == '0.0' or v == '0' or v == 0.0 or v == 0:
                            reference_image = filename
                            print('reference_image: ', reference_image)

            else:
                exif_dictionary['no exif'] = 'no exif'
        except Exception as e:
            exif_dictionary['no exif'] = 'no exif'
            pass

        img.close()
    except Exception as e:
        print("Error opening this image/file: " + filename)
        read_error = "Error opening this image/file: " + filename
        pass
    #exif_dictionary = {k: str(v).encode("utf-8") for k,v in exif_dictionary.items()}
    return reference_image, exif_dictionary, read_error

def get_relevant_exif_info(filename):
    exif_dictionary={}
    reference_image = ""

    img=Image.open(filename)
    exifd = img.getexif()
    keys = list(exifd.keys())
    #exif_dictionary['filename'] = img.filename
    exif_dictionary['width x height'] = img.size
    for k, v in exifd.items():
        tag = TAGS.get(k)
        print(tag)
        if (TAGS.get(k) == 'ISOSpeedRatings'): #ID=0x8827
            print(TAGS.get(k), " : ", v)
            tag=TAGS.get(k)
            exif_dictionary[tag]=v
        if (TAGS.get(k) == 'ExposureTime'): # id=0x829a
            print(TAGS.get(k), " : ", v)
            tag=TAGS.get(k)
            exif_dictionary[tag]=v
        if (TAGS.get(k) == 'ExposureBiasValue'): # id=0x9204
            tag=TAGS.get(k)
            exif_dictionary[tag]=v
            if v == '0.0' or v == '0' or v == 0.0 or v == 0:
                reference_image = filename
                print('reference_image: ', reference_image)
        if (TAGS.get(k) == 'FNumber'): # id=0x829d
            print(TAGS.get(k), " : ", v)
            tag=TAGS.get(k)
            exif_dictionary[tag]=v

    return reference_image, exif_dictionary



'''
# Read some basic exif info from a file
def get_basic_exif_info_from_file(filename, output):
    exif_dictionary = {}
    exif_id_dictionary = {}
    reference_image =""   # This image contains the image with Exposure bias value/Exposure compensation 0

    img = Image.open(filename)
    exifd = img.getexif()
    keys = list(exifd.keys())
    exif_dictionary['filename'] = img.filename
    exif_dictionary['width x height'] = img.size
    for k, v in img.getexif().items():
        if (TAGS.get(k) == 'ISOSpeedRatings'):  # ID=0x8827
            # print(TAGS.get(k), " : ", v)
            exif_id_dictionary[k] = v
            tag = TAGS.get(k)
            exif_dictionary[tag] = v
        elif (TAGS.get(k) == 'DateTimeOriginal'):  # id=0x9003
            # print(TAGS.get(k), " : ", v)
            exif_id_dictionary[k] = v
            tag = TAGS.get(k)
            exif_dictionary[tag] = v
        elif (TAGS.get(k) == 'MaxApertureValue'):  # ID=0x9205
            # print(TAGS.get(k), " : ", reworkTag(v, 1))
            exif_id_dictionary[k] = v
            tag = TAGS.get(k)
            exif_dictionary[tag] = v
        elif (TAGS.get(k) == 'FocalLength'):  # id=0x920a
            # print(TAGS.get(k), " : ", reworkTag(v, 1))
            exif_id_dictionary[k] = v
            tag = TAGS.get(k)
            exif_dictionary[tag] = v
        elif (TAGS.get(k) == 'FocalLengthIn35mmFilm'):  # id=0xa405
            # print(TAGS.get(k), " : ", v)
            exif_id_dictionary[k] = v
            tag = TAGS.get(k)
            exif_dictionary[tag] = v
            # Field of View (FOV) http://www.bobatkins.com/photography/technical/field_of_view.html
            # FOV(rectilinear) = 2 * arctan(framesize / (focal length * 2)) # framesize 36mm, focal length in 35 mm
            FOV = 2 * (math.atan(36 / (v * 2)))
            exif_dictionary['FOV'] = FOV
        elif (TAGS.get(k) == 'ExposureTime'):  # id=0x829a
            # print(TAGS.get(k), " : ", reworkExposure(v))
            exif_id_dictionary[k] = v
            tag = TAGS.get(k)
            exif_dictionary[tag] = v
        elif (TAGS.get(k) == 'ExposureBiasValue'):  # id=0x9204
            exif_id_dictionary[k] = v
            tag = TAGS.get(k)
            exif_dictionary[tag] = v
            #print("ExposureCompensation\t: ", v)
            if v == '0.0' or v == '0':
                reference_image = filename
                print('reference_image ', reference_image)
        elif (TAGS.get(k) == 'FNumber'):  # id=0x829d
            # print(TAGS.get(k), " : ", reworkTag(v, 1))
            exif_id_dictionary[k] = v
            tag = TAGS.get(k)
            exif_dictionary[tag] = v
        elif (TAGS.get(k) == 'Orientation'):  # id=0x274d
            exif_id_dictionary[k] = v
            tag = TAGS.get(k)
            exif_dictionary[tag] = v
            #print(TAGS.get(k), " : ", v)
    img.close()
    
    if (output == 'print'):
        print('\n\n', exif_dictionary)
        # print("\n\n",exif_dictionary['ExposureBiasValue'])

    return reference_image, exif_dictionary, exif_id_dictionary
'''

def get_basic_exif_info_from_file(filename):
    exif_id_dictionary = {}
    img = Image.open(filename)
    exifd = img.getexif()
    keys = list(exifd.keys())
    for k, v in img.getexif().items():
        if (TAGS.get(k) == 'ISOSpeedRatings'):  # ID=0x8827
            exif_id_dictionary['"0x'+ k + '"'] = str(v)
        elif (TAGS.get(k) == 'DateTimeOriginal'):  # id=0x9003
            exif_id_dictionary['"0x'+ k + '"'] = str(v)
        elif (TAGS.get(k) == 'MaxApertureValue'):  # ID=0x9205
            exif_id_dictionary['"0x'+ k + '"'] = str(v)
        elif (TAGS.get(k) == 'FocalLength'):  # id=0x920a
            exif_id_dictionary['"0x'+ k + '"'] = str(v)
        elif (TAGS.get(k) == 'FocalLengthIn35mmFilm'):  # id=0xa405
            exif_id_dictionary['"0x'+ k + '"'] = str(v)
        elif (TAGS.get(k) == 'ExposureTime'):  # id=0x829a
            exif_id_dictionary['"0x'+ k + '"'] = str(v)
        elif (TAGS.get(k) == 'ExposureBiasValue'):  # id=0x9204
            exif_id_dictionary['"0x'+ k + '"'] = str(v)
        elif (TAGS.get(k) == 'FNumber'):  # id=0x829d
            exif_id_dictionary['"0x'+ k + '"'] = str(v)
        elif (TAGS.get(k) == 'Orientation'):  # id=0x274d
            exif_id_dictionary['"0x'+ k + '"'] = str(v)
    img.close()

    return exif_id_dictionary



# This function resizes the original selected images when
# the user clicks on "create preview" or "create enfused image"
def resizetopreview(all_values, folder, tmpfolder):
    '''
    This function resizes the original selected images when
    the user clicks on "create preview" or "create fused image"

    :param all_values:    The values from the Gui elements
    :type all_values:     (Dict[Any, Any]) - {Element_key : value}
    :param folder:        The folder holding the original images
    :type folder:         (str)
    :param tmpfolder:     The dynamically created work folder in the OS temp folder
    :type tmpfolder:      (str)
    '''
    img = ""
    failed = ""
    images = []

    files = all_values['-FILE LIST-']
    for file in files:
        nfile = os.path.join(folder, file)
        previewfile = os.path.join(tmpfolder, file)

        if not os.path.exists(previewfile):  # This means that the preview file does not exist yet
            print("previewfile: ", previewfile, " does not exist yet")
            if platform.system() == 'Windows':
                img += "\"" + nfile.replace("/", "\\") + "\" "
            else:
                img += "\"" + nfile + "\" "
            try:
                preview_img = Image.open(nfile)
                exifd = preview_img.getexif()  # Get all exif data from original image
                sg.user_settings_filename(path=Path.home())
                longestSide = int(sg.user_settings_get_entry('last_size_chosen', '480'))
                preview_img.thumbnail((longestSide, longestSide), Image.ANTIALIAS)
                preview_img.save(previewfile, "JPEG", exif=exifd)  # Save all exif data from original to resized image
            except Exception as e:
                failed += ui_actions.logger(e) + '\n'  # get the error from the logger
                preview_img.save(previewfile, "JPEG")
                preview_img.close()
                # pass
            images.append(previewfile)
        else:  # They do exist but we still need to populate or images list
            images.append(previewfile)
    return failed, images


def resizesingletopreview(folder, tmpfolder, image):
    img = ""

    previewfile = os.path.join(tmpfolder, image)
    orgfile = os.path.join(folder, image)
    #print('previewfile ' + previewfile + '; orgfile ' + orgfile)
    if not os.path.exists(previewfile):  # This means that the preview file does not exist yet
        print("previewfile: ", previewfile, " does not exist yet")
        if platform.system() == 'Windows':
            img += "\"" + orgfile.replace("/", "\\") + "\" "
        else:
            img += "\"" + orgfile + "\" "
        try:
            org_img = Image.open(orgfile)
            exifd = org_img.getexif()  # Get all exif data from original image
            sg.user_settings_filename(path=Path.home())
            longestSide = int(sg.user_settings_get_entry('last_size_chosen', '480'))
            org_img.thumbnail((longestSide, longestSide), Image.ANTIALIAS)
            if (".tif" in previewfile.lower()) or (".tiff" in previewfile.lower()):
                print("we have a tif(f) file being " + previewfile)
                preview_without_extension = os.path.splitext(previewfile)[0]
                org_img.save(preview_without_extension + ".jpg", "JPEG", exif=exifd)  # Save all exif data from original to resized image
                previewfile = preview_without_extension + ".jpg"
            else:
                org_img.save(previewfile, "JPEG", exif=exifd)  # Save all exif data from original to resized image
            org_img.close()

        except Exception as e:
            ui_actions.logger(e)
            # pass

    return previewfile

def resizesingletothumb(folder, tmpfolder, image):
    img = ""

    thumbfile = os.path.join(tmpfolder, 'thumb-' + image)
    orgfile = os.path.join(folder, image)
    #print('thumbfile ' + thumbfile + '; orgfile ' + orgfile)
    if not os.path.exists(thumbfile):  # This means that the thumb file does not exist yet
        print("thumbfile: ", thumbfile, " does not exist yet")
        if platform.system() == 'Windows':
            img += "\"" + orgfile.replace("/", "\\") + "\" "
        else:
            img += "\"" + orgfile + "\" "
        try:
            org_img = Image.open(orgfile)
            exifd = org_img.getexif()  # Get all exif data from original image
            org_img.thumbnail((240, 240), Image.ANTIALIAS)
            # we can have tiffs and jpgs, but we always convert to jpg meaning we might need to change the file extension
            if (".tif" in thumbfile.lower()) or (".tiff" in thumbfile.lower()):
                print("we have a tif(f) file being " + thumbfile)
                thumb_without_extension = os.path.splitext(thumbfile)[0]
                print("thumb_without_extension " + thumb_without_extension)
                org_img.save(thumb_without_extension + ".jpg", "JPEG", exif=exifd)  # Save all exif data from original to resized image
                thumbfile = thumb_without_extension + ".jpg"
            else:
                org_img.save(thumbfile, "JPEG", exif=exifd)  # Save all exif data from original to resized image

            org_img.close()
        except Exception as e:
            ui_actions.logger(e)
            # pass

    return thumbfile

def create_dcraw_command(filename):
    '''
    This function creates the dcraw command for coverting raw images

    :filename:   This is the path/filename of the original RAW image
    '''
    cmd_string = ""
    cmd_list = []

    dcraw_bin = sg.user_settings_get_entry('dcrawlocation', '')
    if len(dcraw_bin) > 0:
        dcraw_params = sg.user_settings_get_entry('dcrawparams', '-v -w -H 2 -T')
        cmd_string = dcraw_bin

        if platform.system() == 'Windows':
            cmd_list = dcraw_params.split(" ")
            cmd_list.append(filename.replace("/", "\\"))
        else:
            cmd_string += " " + dcraw_params + " \"" + filename + "\" "

    return cmd_string, cmd_list

def get_curr_screen_geometry():
    root = tk.Tk()
    width = root.winfo_screenwidth()
    height = root.winfo_screenheight()
    root.destroy()  # do not forget to delete the tk window
    return int(width), int(height)


def display_preview(mainwindow, imgfile):
    print("imgfile to display ", imgfile)
    try:
        image = Image.open(str(imgfile))
        try:
            image = reorient_img(image)
        except Exception as e:
            print("No image orentation data. Mostly due to tiffs created from RAW")
            pass
        sg.user_settings_filename(path=Path.home())
        longestSide = int(sg.user_settings_get_entry('last_size_chosen', '480'))
        image.thumbnail((longestSide, longestSide), Image.ANTIALIAS)
        bio = io.BytesIO()
        image.save(bio, format='PNG')
        mainwindow['-IMAGE-'].update(data=bio.getvalue())
        image.close()
    except Exception as e:
        print("Something went wrong converting ", imgfile)
        pass


def display_thumb(mainwindow, imgfile):
    print("thumbfile to display ", imgfile)
    try:
        image = Image.open(str(imgfile))
        try:
            image = reorient_img(image)
        except Exception as e:
            print("No image orentation data. Mostly due to tiffs created from RAW")
            pass
        image.thumbnail((240, 240), Image.ANTIALIAS)
        bio = io.BytesIO()
        image.save(bio, format='PNG')
        mainwindow['-THUMB-'].update(data=bio.getvalue())
        image.close()
    except Exception as e:
        print("Something went wrong converting ", imgfile)
        pass


# This displays the final image in the default viewer
def displayImage(imgpath):
    rawimgpath = str(imgpath)
    newImg = Image.open(rawimgpath)
    newImg.show()
    newImg.close()

def reorient_img(pil_img):
    img_exif = pil_img.getexif()

    if len(img_exif):
        if img_exif[274] == 3:
            pil_img = pil_img.transpose(Image.ROTATE_180)
        elif img_exif[274] == 6:
            pil_img = pil_img.transpose(Image.ROTATE_270)
        elif img_exif[274] == 8:
            pil_img = pil_img.transpose(Image.ROTATE_90)

    return pil_img


def displayImageWindow(imgpath):
    '''
    This function displays the created final image in a python window.
    On a screen size >= 2K and < 4k, it will use a slightly smaller window

    :param imgpath:     the complete path that leads to the created image
    :type imgpath:      (str)
    '''
    tmpImg = Image.open(str(imgpath))
    newImg = reorient_img(tmpImg)
    # imgsize = newImg.size
    scrwidth, scrheight = get_curr_screen_geometry()
    # print('scrwidth x scrheight: ' + str(scrwidth) + 'x' + str(scrheight))
    # 4k is 3840 x 2160, HD is 1920 x 18080
    if scrwidth >= 1920 and scrwidth < 3840:
        scrwidth = 1860
        scrheight = 1026
    newImg.thumbnail((scrwidth, scrheight), Image.ANTIALIAS)
    bio = io.BytesIO()
    newImg.save(bio, format='PNG')
    layout = [
        [sg.Image(bio.getvalue(), key='-IMAGE-')],
        [sg.Button('Exit', visible=False, key='-exit-')]  # invisible button necessary to allow windows read
    ]
    window = sg.Window('Your image: ' + imgpath, layout, no_titlebar=False, location=(0, 0), size=(scrwidth, scrheight),
                       keep_on_top=True, icon=get_icon())

    tmpImg.close()
    newImg.close()

    while True:
        event, values = window.read()
        if event in (sg.WINDOW_CLOSED, '-exit-'):
            break
    window.close


def get_icon():
    '''
    This function adds the program icon to the top-left of the displayed windows and popups
    '''
    '''
    if platform.system() == 'Windows':
        wicon = os.path.join(os.path.dirname(os.path.abspath(__file__)), 'images', 'logo.ico')
    else:
        wicon = os.path.join(os.path.dirname(os.path.abspath(__file__)), 'images', 'logo.png')
    '''
    if platform.system() == 'Linux' or platform.system() == 'Darwin':
        testpyinstaller = getattr(sys, '_MEIPASS', 'NotRunningInPyInstaller')
        prefix = os.path.realpath('.')
        if testpyinstaller != 'NotRunningInPyInstaller':
            prefix = testpyinstaller;
        wicon = os.path.join(prefix, 'images', 'logo.png')
    else: # Windows
        wicon = os.path.join(os.path.dirname(os.path.abspath(__file__)), 'images', 'logo.ico')

    return wicon


def get_filename_images(values, folder):
    '''
    This functions requests the filename and possible folder where to save
    the to be created image file.
    Next to that the path will be derived from the folder where the images reside

    :param values:          This is the dictionary containing all UI key variables
    :type values:           (Dict[Any, Any]) - {Element_key : value}
    :param folder:          The folder that contains the full size original images
    :type folder:           (list)
    '''
    full_images = []

    if values['_saveToSource_']:
        folderFileName = file_functions.getFileName(folder)  # user wants to save image to same folder
    else:
        folderFileName = file_functions.getFileName('')  # user wants to select new folder
    if folderFileName[0] != '' and folderFileName[0] != folder:
        folder = folderFileName[0]
    if folderFileName[1] == '':
        print('User pressed OK without providing a filename')
        sg.popup('Error! no filename provided!', 'You did not provide a filename.\n',
                 'The program can\'t create an image without a filename.', icon=get_icon(), keep_on_top=True)
    elif folderFileName[1] == 'Cancel':
        print('user Cancelled')
    else:
        files = values['-FILE LIST-']
        for file in files:
            full_images.append(os.path.join(folder, file))

    return folderFileName[1], full_images

def copy_exif_info(reference_image, imagepath):
    print('reference_image ', reference_image)
    print('imagepath ', imagepath)
    ref_img = Image.open(reference_image)
    ref_exifd = ref_img.getexif()
    #print("ref_exifd ", str(ref_exifd))
    # ref_exifd = ref_img.info['exif']
    ref_img.close()
    pil_img = Image.open(imagepath)
    dcraw_converted = False
    for tag, value in ref_exifd.items():
        #print(tag, value)
        if 'dcraw' in str(value).lower():
            # If we have a tiff converted from a RAW, we do not have exif info
            dcraw_converted = True

    if dcraw_converted:
        pil_img.save(imagepath, "JPEG")
    else:
        pil_img.save(imagepath, "JPEG", exif=ref_exifd)

    pil_img.close()
    '''
    try:
        print('reference_image ', reference_image)
        print('imagepath ', imagepath)
        ref_img = Image.open(reference_image)
        ref_exifd = ref_img.getexif()
        #ref_exifd = ref_img.info['exif']
        ref_img.close()
    except Exception as e:
        print("something went wrong copying the exif info from " + reference_image)
        pass


    pil_img = Image.open(imagepath)
    # first make a backup in case of error
    backup = pil_img.copy()
    try:
        backuppath = imagepath.replace('preview', 'backup-preview')
        backup.save(backuppath, "JPEG")
        pil_img.save(imagepath, "JPEG", exif=ref_exifd)
        pil_img.close()
        os.remove(backuppath)
    except Exception as e:
        print("something went wrong saving the exif info to " + imagepath)
        os.replace(backuppath, imagepath)
        pass
    '''