DisplayMagician/HeliosPlus/UIForms/ApplyingProfileForm.cs

using System;
using System.Collections.Generic;
using System.ComponentModel;
using System.Data;
using System.Diagnostics;
using System.Drawing;
using System.Linq;
using System.Text;
using System.Threading.Tasks;
using System.Windows.Forms;
using WinFormAnimation;

namespace HeliosPlus.UIForms
{
    public partial class ApplyingProfileForm : Form
    {
        private readonly Bitmap _progressImage;
        private readonly List<Point> _progressPositions = new List<Point>();
        private int _countdownCounter;
        private readonly int _displayChangeMaxDelta = -1;
        private int _displayChangeDelta;
        private int _lastCount;
        private bool _isClosing;

        public ApplyingProfileForm()
        {
            InitializeComponent();
            _progressImage = new Bitmap(progressPanel.Width, progressPanel.Height);
            Controls.Remove(progressPanel);
            progressPanel.BackColor = BackColor;
            progressBar.Invalidated += (sender, args) => Invalidate();
            progressPanel.Invalidated += (sender, args) => Invalidate();
            Reposition();
        }

        public ApplyingProfileForm(Task taskToRun = null, int countdown = 0, string title = null, string message = null, Color progressColor = default(Color), bool cancellable = false, int displayChangeMaxDelta = 5) : this()
        {
            _countdownCounter = countdown;
            _lastCount = _countdownCounter;
            _displayChangeMaxDelta = displayChangeMaxDelta;
            Cancellable = cancellable;
            TaskToRun = taskToRun;
            if (progressColor.Equals(default(Color)))
                progressColor = Color.OrangeRed;
            ProgressColor = progressColor;
            if (!string.IsNullOrEmpty(title)) Title = title;
            if (!string.IsNullOrEmpty(message)) Message = message;
        }

        public string Title { get; set; } = "Please wait...";
        public string Message { get; set; } = "It won't be long now!";
        public Color ProgressColor { get; set; } = Color.OrangeRed;
        public bool Cancellable{ get; set; } = false;
        public Task TaskToRun { get; set; } = null;

        protected override void OnKeyDown(KeyEventArgs e)
        {
            e.Handled = true;
        }

        protected override void OnPaint(PaintEventArgs e)
        {
            lock (_progressPositions)
            {
                progressPanel.DrawToBitmap(_progressImage, new Rectangle(Point.Empty, progressPanel.Size));

                foreach (var position in _progressPositions)
                {
                    e.Graphics.DrawImage(_progressImage, new Rectangle(position, progressPanel.Size));
                }
            }

            base.OnPaint(e);
        }

        protected override bool ProcessCmdKey(ref Message msg, Keys keyData)
        {
            if (keyData != Keys.Escape)
            {
                return base.ProcessCmdKey(ref msg, keyData);
            }

            if (Cancellable)
            {
                t_countdown.Stop();
                DialogResult = DialogResult.Cancel;
                Close();

                return true;
            }

            return true;
        }

        private void HandleDisplayChangeDelta()
        {
            if (_displayChangeMaxDelta <= -1) return;
            _displayChangeDelta = _lastCount - _countdownCounter;
            if (_displayChangeDelta > _displayChangeMaxDelta)
            {
                Debug.Print("_displayChangeDelta > _displayChangeMaxDelta! " + _displayChangeDelta + " > " + _displayChangeMaxDelta);
                DialogResult = DialogResult.OK;
                Close();
            }
        }

        private void DoCountdown()
        {
            if (_countdownCounter > 0)
            {
                lbl_message.Text = Title;
                lbl_sub_message.Text = Message;
                progressBar.ProgressColor = ProgressColor;
                progressBar.Maximum = _countdownCounter;
                progressBar.Value = _countdownCounter;
                progressBar.Text = _countdownCounter.ToString();
                t_countdown.Start();
                if (TaskToRun is Task)
                    TaskToRun.Start();

            }
            else
            {
                progressBar.Style = ProgressBarStyle.Marquee;
                progressBar.Text = "";
                progressBar.Maximum = 100;
                progressBar.Value = 50;
                progressBar.Style = ProgressBarStyle.Marquee;
                if (TaskToRun is Task)
                    TaskToRun.Start();
                DialogResult = DialogResult.OK;
                Close();
            }

            HandleDisplayChangeDelta();
        }

        private void Reposition()
        {
            lock (_progressPositions)
            {
                // Screen.Allscreens is a static variable and won't be updated while the program is running.
                // SO we use reflection to wipe the value and force it to be updated afresh.
                typeof(Screen).GetField("screens", System.Reflection.BindingFlags.Static | System.Reflection.BindingFlags.NonPublic).SetValue(null, null);
                Screen[] screens = Screen.AllScreens;
                Size = screens.Select(screen => screen.Bounds)
                                            .Aggregate(Rectangle.Union)
                                            .Size;

                _progressPositions.Clear();
                _progressPositions.AddRange(
                    screens.Select(
                        screen =>
                            new Point(screen.Bounds.X + ((screen.Bounds.Width - progressPanel.Width) / 2),
                                screen.Bounds.Y + ((screen.Bounds.Height - progressPanel.Height) / 2))
                    )
                );
            }
            Invalidate();
        }

        private void ApplyingProfileForm_FormClosing(object sender, FormClosingEventArgs e)
        {
            if (_isClosing)
            {
                return;
            }

            _isClosing = true;
            e.Cancel = true;
            var dialogResult = DialogResult;
            new Animator(new Path((float)Opacity, 0, 200))
                .Play(new SafeInvoker<float>(f =>
                {
                    try
                    {
                        Opacity = f;
                    }
                    catch (Exception ex)
                    {
                        Console.WriteLine($"ApplyingProfileForm/ApplyingProfileForm_FormClosing exception: {ex.Message}: {ex.StackTrace} - {ex.InnerException}");
                        // ignored
                    }
                }, this), new SafeInvoker(() =>
                {
                    DialogResult = dialogResult;
                    Close();
                }, this));
        }

        private void ApplyingProfileForm_Reposition(object sender, EventArgs e)
        {
            Reposition();
        }

        private void ApplyingProfileForm_Shown(object sender, EventArgs e)
        {
            Reposition();
            new Animator(new Path((float)Opacity, 0.97f, 200))
                .Play(new SafeInvoker<float>(f =>
                {
                    try
                    {
                        Opacity = f;
                    }
                    catch (Exception ex)
                    {
                        Console.WriteLine($"ApplyingProfileForm/ApplyingProfileForm_Shown exception: {ex.Message}: {ex.StackTrace} - {ex.InnerException}");
                        // ignored
                    }
                }, this), new SafeInvoker(DoCountdown, this));
        }

        private void t_countdown_Tick(object sender, EventArgs e)
        {
            HandleDisplayChangeDelta();
            progressBar.Value = _countdownCounter;
            progressBar.Text = progressBar.Value.ToString();

            if (_countdownCounter <= 0)
            {
                _countdownCounter = 0;
                t_countdown.Stop();
                DialogResult = DialogResult.OK;
                Close();

                return;
            }
            Reposition();
            _countdownCounter--;
        }

        protected override void WndProc(ref Message m)
        {
            const int WM_SETTINGCHANGE = 0x001A;
            const int SPI_SETWORKAREA = 0x02F;
            const int WM_DISPLAYCHANGE = 0x007E;

            const int x_bitshift = 0;
            const int y_bitshift = 16;
            const int xy_mask = 0xFFFF;

            bool displayChange = false;

            switch (m.Msg)
            {
                case WM_SETTINGCHANGE:
                    Debug.Print("Message: " + m.ToString());
                    Debug.Print("WM_SETTINGCHANGE");
                    switch ((int)m.WParam)
                    {
                        case SPI_SETWORKAREA:
                            Debug.Print("SPI_SETWORKAREA");
                            displayChange = true;
                            break;
                    }
                    break;
                case WM_DISPLAYCHANGE:
                    int cxScreen = (xy_mask & ((int)m.LParam) >> x_bitshift);
                    int cyScreen = (xy_mask & ((int)m.LParam) >> y_bitshift);
                    Debug.Print("Message: " + m.ToString());
                    Debug.Print("WM_DISPLAYCHANGE");
                    Debug.Print("cxScreen: " + cxScreen + " cyScreen: " + cyScreen);
                    displayChange = true;
                    break;
            }
            if (displayChange)
            {
                _displayChangeDelta = _lastCount - _countdownCounter;
                _lastCount = _countdownCounter;
                Debug.Print("Display Change Detected at t " + _lastCount + " difference between changes is " + _displayChangeDelta);
            }

            base.WndProc(ref m);
        }

    }
}