import cv2
import os
from opendm import log
from opendm.thermal_tools import dji_unpack
from opendm.exiftool import extract_raw_thermal_image_data
from opendm.thermal_tools.thermal_utils import sensor_vals_to_temp

def resize_to_match(image, match_photo = None):
    """
    Resize images to match the dimension of another photo
    :param image  numpy array containing image data to resize
    :param match_photo ODM_Photo whose dimensions should be used for resize
    :return numpy array with resized image data
    """
    if match_photo is not None:
        h, w, _ = image.shape
        if w != match_photo.width or h != match_photo.height:
            image = cv2.resize(image, None, 
                    fx=match_photo.width/w, 
                    fy=match_photo.height/h,
                    interpolation=cv2.INTER_LANCZOS4)
    return image

def dn_to_temperature(photo, image, images_path):
    """
    Convert Digital Number values to temperature (C) values
    :param photo ODM_Photo
    :param image numpy array containing image data
    :param images_path path to original source image to read data using PIL for DJI thermal photos
    :return numpy array with temperature (C) image values
    """

    # Handle thermal bands
    if photo.is_thermal():
        # Every camera stores thermal information differently
        # The following will work for MicaSense Altum cameras
        # but not necessarily for others
        if photo.camera_make == "MicaSense" and photo.camera_model[:5] == "Altum":
            image = image.astype("float32")
            image -= (273.15 * 100.0) # Convert Kelvin to Celsius
            image *= 0.01
            return image
        elif photo.camera_make == "DJI" and photo.camera_model == "ZH20T":
            filename, file_extension = os.path.splitext(photo.filename)
            # DJI H20T high gain mode supports measurement of -40~150 celsius degrees
            if file_extension.lower() in [".tif", ".tiff"] and image.min() >= 23315: # Calibrated grayscale tif
                image = image.astype("float32")
                image -= (273.15 * 100.0) # Convert Kelvin to Celsius
                image *= 0.01
                return image
            else:
                return image
        elif photo.camera_make == "DJI" and photo.camera_model == "MAVIC2-ENTERPRISE-ADVANCED":
            image = dji_unpack.extract_temperatures_dji(photo, image, images_path)
            image = image.astype("float32")
            return image
        else:
            try:
                params, image = extract_raw_thermal_image_data(os.path.join(images_path, photo.filename))
                image = sensor_vals_to_temp(image, **params)
            except Exception as e:
                log.ODM_WARNING("Cannot radiometrically calibrate %s: %s" % (photo.filename, str(e)))

            image = image.astype("float32")
            return image
    else:
        image = image.astype("float32")
        log.ODM_WARNING("Tried to radiometrically calibrate a non-thermal image with temperature values (%s)" % photo.filename)
        return image