Avalonia/src/Avalonia.Native/PlatformThreadingInterface.cs

// Copyright (c) The Avalonia Project. All rights reserved.
// Licensed under the MIT license. See licence.md file in the project root for full license information.

using System;
using System.Threading;
using Avalonia.Native.Interop;
using Avalonia.Platform;
using Avalonia.Threading;
using SharpGen.Runtime;

namespace Avalonia.Native
{
    public class PlatformThreadingInterface : IPlatformThreadingInterface
    {
        class TimerCallback : CallbackBase, IAvnActionCallback
        {
            readonly Action _tick;

            public TimerCallback(Action tick)
            {
                _tick = tick;
            }

            public void Run()
            {
                _tick();
            }
        }

        class SignaledCallback : CallbackBase, IAvnSignaledCallback
        {
            readonly PlatformThreadingInterface _parent;

            public SignaledCallback(PlatformThreadingInterface parent)
            {
                _parent = parent;
            }

            public void Signaled(int priority, bool priorityContainsMeaningfulValue)
            {
                _parent.Signaled?.Invoke(priorityContainsMeaningfulValue ? (DispatcherPriority?)priority : null);
            }
        }

        readonly IAvnPlatformThreadingInterface _native;

        public PlatformThreadingInterface(IAvnPlatformThreadingInterface native)
        {
            _native = native;
            using (var cb = new SignaledCallback(this))
                _native.SignaledCallback = cb;
        }

        public bool CurrentThreadIsLoopThread => _native.CurrentThreadIsLoopThread;

        public event Action<DispatcherPriority?> Signaled;

        public void RunLoop(CancellationToken cancellationToken)
        {
            if (cancellationToken.CanBeCanceled == false)
                _native.RunLoop(null);
            else
            {
                var l = new object();
                var cancellation = _native.CreateLoopCancellation();
                cancellationToken.Register(() =>
                {
                    lock (l)
                    {
                        cancellation?.Cancel();
                        cancellation?.Dispose();
                        cancellation = null;
                    }
                });
                try
                {
                    _native.RunLoop(cancellation);
                }
                finally
                {
                    lock(l)
                    {
                        cancellation?.Dispose();
                        cancellation = null;
                    }
                }
            }
        }

        public void Signal(DispatcherPriority priority)
        {
            _native.Signal((int)priority);
        }

        public IDisposable StartTimer(DispatcherPriority priority, TimeSpan interval, Action tick)
        {
            using (var cb = new TimerCallback(tick))
                return _native.StartTimer((int)priority, (int)interval.TotalMilliseconds, cb);
        }
    }
}