parsec-vdd/app/App.xaml.cs

using System;
using System.Threading.Tasks;
using System.Threading;
using System.Windows;
using System.Linq;
using System.Windows.Interop;
using System.Windows.Media;
using System.Reflection;
using System.Collections.Generic;
using System.Collections;
using System.Globalization;
using System.Resources;

namespace ParsecVDisplay
{
    public partial class App : Application
    {
        const string ID = "QpHOX8IBUHBznGtqk9xm1";
        public const string NAME = "ParsecVDisplay";
        public const string VERSION = "0.45.2";
        public const string GITHUB_REPO = "nomi-san/parsec-vdd";

        public static bool Silent { get; private set; }
        public static string[] Languages => LanguageDicts.Keys.ToArray();
        static Dictionary<string, ResourceDictionary> LanguageDicts;

        protected override void OnStartup(StartupEventArgs e)
        {
            if (e.Args.Length >= 2 && e.Args[0] == "-custom")
            {
                var modes = Display.ParseModes(e.Args[1]);
                ParsecVDD.SetCustomDisplayModes(modes);

                if (e.Args.Length >= 3)
                {
                    if (Enum.TryParse<ParsecVDD.ParentGPU>(e.Args[2], true, out var kind))
                    {
                        ParsecVDD.SetParentGPU(kind);
                    }
                }

                Shutdown();
                return;
            }

            Silent = e.Args.Contains("-silent");

            var signal = new EventWaitHandle(false,
                EventResetMode.AutoReset, ID, out var isOwned);

            if (!isOwned)
            {
                signal.Set();
                Shutdown();
                return;
            }

            var status = ParsecVDD.QueryStatus();
            if (status != Device.Status.OK)
            {
                if (status == Device.Status.RESTART_REQUIRED)
                {
                    MessageBox.Show("You must restart your PC to complete the driver setup.",
                        NAME, MessageBoxButton.OK, MessageBoxImage.Warning);
                }
                else if (status == Device.Status.DISABLED)
                {
                    MessageBox.Show($"{ParsecVDD.ADAPTER} is disabled, please enable it.",
                        NAME, MessageBoxButton.OK, MessageBoxImage.Warning);
                }
                else if (status == Device.Status.NOT_INSTALLED)
                {
                    MessageBox.Show("Please install the driver first.",
                        NAME, MessageBoxButton.OK, MessageBoxImage.Warning);
                }
                else
                {
                    MessageBox.Show($"The driver is not OK, please check again. Current status: {status}.",
                        NAME, MessageBoxButton.OK, MessageBoxImage.Warning);
                }

                Shutdown();
                return;
            }

            if (ParsecVDD.Init() == false)
            {
                MessageBox.Show("Failed to obtain the device handle, please check the driver installation again.",
                    NAME, MessageBoxButton.OK, MessageBoxImage.Warning);

                Shutdown();
                return;
            }

            Task.Run(() =>
            {
                while (signal.WaitOne())
                {
                    Tray.ShowApp();
                }
            });

            base.OnStartup(e);
            LoadLanguages();

            // Disable GPU to prevent flickering when adding display
            RenderOptions.ProcessRenderMode = RenderMode.SoftwareOnly;
        }

        protected override void OnExit(ExitEventArgs e)
        {
            ParsecVDD.Uninit();
            base.OnExit(e);
        }

        static void LoadLanguages()
        {
            LanguageDicts = new Dictionary<string, ResourceDictionary>();

            var assembly = Assembly.GetExecutingAssembly();
            var rm = new ResourceManager(assembly.GetName().Name + ".g", assembly);
            try
            {
                var list = rm.GetResourceSet(CultureInfo.CurrentCulture, true, true);
                foreach (DictionaryEntry item in list)
                {
                    if (item.Key is string key
                        && key.StartsWith("languages/"))
                    {
                        var source = key
                            .Replace("languages/", "Languages/")
                            .Replace(".baml", ".xaml");

                        try
                        {
                            var dict = new ResourceDictionary();
                            dict.Source = new Uri(source, UriKind.Relative);

                            var name = dict["lang_name"].ToString();
                            LanguageDicts.Add(name, dict);
                        }
                        catch
                        {
                        }
                    }
                }
            }
            finally
            {
                rm.ReleaseAllResources();
            }

            SetLanguage(Config.Language, saveConfig: false);
        }

        public static void SetLanguage(string name, bool saveConfig = true)
        {
            if (LanguageDicts.TryGetValue(name, out var dict))
            {
                Current.Resources.MergedDictionaries.Add(dict);

                if (saveConfig)
                {
                    Config.Language = name;
                }
            }
        }

        public static string GetTranslation(string key)
        {
            try
            {
                var trans = Current.FindResource(key);
                if (trans != null && trans is string)
                    return trans as string;
            }
            catch
            {
            }

            return string.Format("{{{{{0}}}}}", key);
        }
    }
}