// XAML Map Control - https://github.com/ClemensFischer/XAML-Map-Control
// Copyright © 2024 Clemens Fischer
// Licensed under the Microsoft Public License (Ms-PL)

using System;
using System.IO;
using System.Threading.Tasks;
using System.Windows;
using System.Windows.Media;
using System.Windows.Media.Imaging;

namespace MapControl
{
    public static partial class ImageLoader
    {
        public static ImageSource LoadImage(Uri uri)
        {
            return new BitmapImage(uri);
        }

        public static ImageSource LoadImage(Stream stream)
        {
            var image = new BitmapImage();

            image.BeginInit();
            image.CacheOption = BitmapCacheOption.OnLoad;
            image.StreamSource = stream;
            image.EndInit();
            image.Freeze();

            return image;
        }

        public static Task<ImageSource> LoadImageAsync(Stream stream)
        {
            return Task.FromResult(LoadImage(stream));
        }

        public static Task<ImageSource> LoadImageAsync(string path)
        {
            if (!File.Exists(path))
            {
                return Task.FromResult<ImageSource>(null);
            }

            return Task.Run(() =>
            {
                using (var stream = File.OpenRead(path))
                {
                    return LoadImage(stream);
                }
            });
        }

        internal static async Task<ImageSource> LoadMergedImageAsync(Uri uri1, Uri uri2, IProgress<double> progress)
        {
            var images = await LoadImagesAsync(uri1, uri2, progress);

            WriteableBitmap image = null;

            if (images.Length == 2 &&
                images[0] is BitmapSource image1 &&
                images[1] is BitmapSource image2 &&
                image1.PixelHeight == image2.PixelHeight &&
                image1.Format == image2.Format &&
                image1.Format.BitsPerPixel % 8 == 0)
            {
                var format = image1.Format;
                var height = image1.PixelHeight;
                var width1 = image1.PixelWidth;
                var width2 = image2.PixelWidth;
                var stride1 = width1 * format.BitsPerPixel / 8;
                var stride2 = width2 * format.BitsPerPixel / 8;
                var buffer1 = new byte[stride1 * height];
                var buffer2 = new byte[stride2 * height];

                image1.CopyPixels(buffer1, stride1, 0);
                image2.CopyPixels(buffer2, stride2, 0);

                image = new WriteableBitmap(width1 + width2, height, 96, 96, format, null);
                image.WritePixels(new Int32Rect(0, 0, width1, height), buffer1, stride1, 0);
                image.WritePixels(new Int32Rect(width1, 0, width2, height), buffer2, stride2, 0);
                image.Freeze();
            }

            return image;
        }
    }
}