App.xaml.cs

using System;
using System.Diagnostics;
using System.IO;
using System.Runtime.CompilerServices;
using System.Runtime.InteropServices;
using System.Windows;
using System.Windows.Threading;

namespace RSImageProcessor
{
    public partial class App : Application
    {
        [DllImport("kernel32.dll", CharSet = CharSet.Unicode, SetLastError = true)]
        [return: MarshalAs(UnmanagedType.Bool)]
        private static extern bool SetDllDirectory(string lpPathName);

        [DllImport("kernel32.dll", CharSet = CharSet.Unicode, SetLastError = true)]
        [return: MarshalAs(UnmanagedType.Bool)]
        private static extern bool SetEnvironmentVariable(string lpName, string lpValue);

        [DllImport("kernel32.dll", CharSet = CharSet.Unicode, SetLastError = true)]
        private static extern IntPtr LoadLibrary(string lpLibFileName);

        // Module initializer - runs before ANYTHING else
        [ModuleInitializer]
        internal static void InitializeModule()
        {
            SetupEnvironment();
        }

        // Static constructor as backup
        static App()
        {
            SetupEnvironment();
        }

        private static bool _environmentSetup = false;

        private static void SetupEnvironment()
        {
            if (_environmentSetup) return;
            _environmentSetup = true;

            try
            {
                string baseDir = AppDomain.CurrentDomain.BaseDirectory;

                // Remove trailing backslash if present
                baseDir = baseDir.TrimEnd('\\', '/');

                // 1. Set DLL directory
                SetDllDirectory(baseDir);

                // 2. Prepend to PATH
                string currentPath = Environment.GetEnvironmentVariable("PATH") ?? "";
                string newPath = baseDir + ";" + currentPath;
                SetEnvironmentVariable("PATH", newPath);
                Environment.SetEnvironmentVariable("PATH", newPath);

                // 3. Set GDAL_DATA
                string gdalData = Path.Combine(baseDir, "gdal-data");
                if (Directory.Exists(gdalData))
                {
                    SetEnvironmentVariable("GDAL_DATA", gdalData);
                    Environment.SetEnvironmentVariable("GDAL_DATA", gdalData);
                }
                else
                {
                    // Try alternate location
                    gdalData = Path.Combine(baseDir, "maxrev.gdal.core.libshared", "gdal-data");
                    if (Directory.Exists(gdalData))
                    {
                        SetEnvironmentVariable("GDAL_DATA", gdalData);
                        Environment.SetEnvironmentVariable("GDAL_DATA", gdalData);
                    }
                }

                // 4. Set PROJ_LIB
                string projData = Path.Combine(baseDir, "proj-data");
                if (Directory.Exists(projData))
                {
                    SetEnvironmentVariable("PROJ_LIB", projData);
                    Environment.SetEnvironmentVariable("PROJ_LIB", projData);
                }
                else
                {
                    projData = Path.Combine(baseDir, "maxrev.gdal.core.libshared", "proj-data");
                    if (Directory.Exists(projData))
                    {
                        SetEnvironmentVariable("PROJ_LIB", projData);
                        Environment.SetEnvironmentVariable("PROJ_LIB", projData);
                    }
                }

                // 5. Disable plugin search
                SetEnvironmentVariable("GDAL_DRIVER_PATH", "disable");
                Environment.SetEnvironmentVariable("GDAL_DRIVER_PATH", "disable");

                // 6. Pre-load critical DLLs in order
                string[] dllsToLoad = { "gdal.dll", "gdal_wrap.dll" };
                foreach (var dll in dllsToLoad)
                {
                    string dllPath = Path.Combine(baseDir, dll);
                    if (File.Exists(dllPath))
                    {
                        LoadLibrary(dllPath);
                    }
                }
            }
            catch
            {
                // Silently ignore - will fail later with better error message
            }
        }

        protected override void OnStartup(StartupEventArgs e)
        {
            base.OnStartup(e);

            DispatcherUnhandledException += OnDispatcherUnhandledException;
            AppDomain.CurrentDomain.UnhandledException += OnDomainUnhandledException;

            InitializeGdal();
        }

        private void InitializeGdal()
        {
            try
            {
                Core.GdalManager.Initialize();
            }
            catch (Exception ex)
            {
                string errorMsg = ex.Message;
                if (ex.InnerException != null)
                    errorMsg += "\n" + ex.InnerException.Message;

                bool needsVCRuntime = errorMsg.Contains("0x8007007E") ||
                                      errorMsg.Contains("Unable to load DLL") ||
                                      errorMsg.Contains("找不到指定的模块");

                string message;
                if (needsVCRuntime)
                {
                    message = "程序需要 Visual C++ 运行库才能运行。\n\n" +
                              "请下载安装：\n" +
                              "https://aka.ms/vs/17/release/vc_redist.x64.exe\n\n" +
                              "安装后重新运行程序。\n\n" +
                              "点击\"是\"打开下载页面。";
                }
                else
                {
                    message = $"GDAL初始化失败：\n{errorMsg}\n\n程序部分功能可能不可用。";
                }

                var result = MessageBox.Show(
                    message,
                    "初始化错误",
                    needsVCRuntime ? MessageBoxButton.YesNo : MessageBoxButton.OK,
                    MessageBoxImage.Warning);

                if (needsVCRuntime && result == MessageBoxResult.Yes)
                {
                    try
                    {
                        Process.Start(new ProcessStartInfo
                        {
                            FileName = "https://aka.ms/vs/17/release/vc_redist.x64.exe",
                            UseShellExecute = true
                        });
                    }
                    catch { }

                    // Exit after opening download page
                    Environment.Exit(0);
                }

                // Write error log
                WriteErrorLog(ex);
            }
        }

        private void WriteErrorLog(Exception ex)
        {
            try
            {
                string logPath = Path.Combine(AppDomain.CurrentDomain.BaseDirectory, "gdal_error.log");
                string content = Core.GdalManager.GetDiagnosticInfo() + "\n\nException:\n" + ex;
                File.WriteAllText(logPath, content);
            }
            catch { }
        }

        protected override void OnExit(ExitEventArgs e)
        {
            Core.GdalManager.Cleanup();
            base.OnExit(e);
        }

        private void OnDispatcherUnhandledException(object sender, DispatcherUnhandledExceptionEventArgs e)
        {
            MessageBox.Show($"错误：{e.Exception.Message}", "错误", MessageBoxButton.OK, MessageBoxImage.Error);
            e.Handled = true;
        }

        private void OnDomainUnhandledException(object sender, UnhandledExceptionEventArgs e)
        {
            if (e.ExceptionObject is Exception ex)
            {
                MessageBox.Show($"严重错误：{ex.Message}", "错误", MessageBoxButton.OK, MessageBoxImage.Error);
            }
        }
    }
}