using System;
using System.Collections.Generic;
using System.Drawing;
using System.Drawing.Imaging;
using System.IO;
using System.Windows;
using CommunityToolkit.Mvvm.ComponentModel;
using CommunityToolkit.Mvvm.Input;
using Microsoft.Win32;
using System.Runtime.InteropServices;
using System.Windows.Interop;
using System.Windows.Media.Imaging;

namespace RevitLess
{
    public partial class MainViewModel : ObservableObject
    {
        [ObservableProperty]
        public partial List<RevitFileAssist.FamilyTypeDefinition> SymbolTypes { get; set; }
        [ObservableProperty]
        public partial RevitFileAssist.FamilyTypeDefinition SelectedSymbolType { get; set; }
        [ObservableProperty]
        public partial System.Windows.Media.ImageSource PreviewImage { get; set; }

        [ObservableProperty]
        public partial string FileInfo { get; set; }

        [DllImport("gdi32.dll")]
        static extern bool DeleteObject(IntPtr hObject);

        public static System.Windows.Media.ImageSource ToImageSource(System.Drawing.Image image)
        {
            if (image == null)
            {
                return null;
            }

            // 始终创建一个与输入 image 无关的 Bitmap 副本（深拷贝），
            // 然后从该副本获取 HBITMAP 并创建 WPF BitmapSource。
            // 这样可以确保底层 GDI+ 资源在返回后仍然有效（WPF 会复制像素数据）。
            using (System.Drawing.Bitmap clone = new System.Drawing.Bitmap(image))
            {
                IntPtr hBitmap = clone.GetHbitmap();
                try
                {
                    System.Windows.Media.Imaging.BitmapSource bitmapSource = Imaging.CreateBitmapSourceFromHBitmap(
                        hBitmap,
                        IntPtr.Zero,
                        Int32Rect.Empty,
                        BitmapSizeOptions.FromEmptyOptions());
                    bitmapSource.Freeze();
                    return bitmapSource;
                }
                finally
                {
                    // 释放由 GetHbitmap 创建的 GDI 对象
                    DeleteObject(hBitmap);
                }
            }
        }

        private static System.Windows.Media.ImageSource ConvertImageToImageSource(System.Drawing.Image image)
        {
            using (var ms = new MemoryStream())
            {
                image.Save(ms, ImageFormat.Png);
                ms.Seek(0, SeekOrigin.Begin);
                var bitmap = new System.Windows.Media.Imaging.BitmapImage();
                bitmap.BeginInit();
                bitmap.CacheOption = System.Windows.Media.Imaging.BitmapCacheOption.OnLoad;
                bitmap.StreamSource = ms;
                bitmap.EndInit();
                bitmap.Freeze();
                return bitmap;
            }
        }
        [RelayCommand]
        private void OpenFile()
        {
            OpenFileDialog dialog = new OpenFileDialog();
            dialog.Filter = "Revit文件|*.rvt;*.rfa";
            if (dialog.ShowDialog() == true)
            {
                Storage storage = new Storage(dialog.FileName);
                //var image = storage.ThumbnailImage.GetPreviewAsImage();
                var info = new RevitFileAssist(dialog.FileName);
                PreviewImage = ToImageSource(info.PreviewImage);
                SymbolTypes = info.SymbolTypes;
                //FileInfo = info.FileInfo.ToString();
                //FileInfo = storage.BasicInfo.ToString();
                //SaveImage(dialog.FileName);
            }
        }
        private static void SaveImage(string filePath)
        {
            var fi = new System.IO.FileInfo(filePath);
            Storage storage = new Storage(fi.FullName);
            var b = storage.ThumbnailImage.GetPreviewAsImage();
            MessageBox.Show(storage.BasicInfo.ToString());

            //Bitmap bm = new Bitmap(b.Width, b.Height, PixelFormat.Format24bppRgb);
            //Graphics g = Graphics.FromImage(bm);
            //g.DrawImage(b, new Point(0, 0));
            //g.Dispose();

            //bm.Save(Path.Combine(fi.Directory.FullName, "_Images", fi.Name.Remove(fi.Name.Length - 4) + ".png"), System.Drawing.Imaging.ImageFormat.Png);
        }
    }
 
}