stgsharp/源代码/StgSharp/Logic/TimeSpanProvider.cs

//-----------------------------------------------------------------------
//-----------------------------------------------------------------------
//     file="TimeSpanProvider.cs"
//     Project: StgSharp
//     AuthorGroup: Nitload Space
//     Copyright (c) Nitload Space. All rights reserved.
//     
//     Permission is hereby granted, free of charge, to any person 
//     obtaining a copy of this software and associated documentation 
//     files (the “Software”), to deal in the Software without restriction, 
//     including without limitation the rights to use, copy, modify, merge,
//     publish, distribute, sublicense, and/or sell copies of the Software, 
//     and to permit persons to whom the Software is furnished to do so, 
//     subject to the following conditions:
//     
//     The above copyright notice and 
//     this permission notice shall be included in all copies 
//     or substantial portions of the Software.
//     
//     THE SOFTWARE IS PROVIDED “AS IS”, 
//     WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, 
//     INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, 
//     FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. 
//     IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, 
//     DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, 
//     ARISING FROM, OUT OF OR IN CONNECTION WITH 
//     THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
//     
//-----------------------------------------------------------------------
//-----------------------------------------------------------------------
using StgSharp.Logic;

using System;
using System.Linq;
using System.Threading;

namespace StgSharp
{

    /// <summary>
    /// A microsecond-accuracy timer. It will refresh every a certain time span.
    /// </summary>
    public class TimeSpanProvider
    {

        private Action _frameUpdateMissCallback = null;

        private readonly int _maxSpanCount;
        private Action _refreshCallback;
        private SemaphoreSlim _refreshSemaphore = new SemaphoreSlim(0, 1);
        private long _spanBegin;
        private readonly long _spanLength;
        private bool _subscribed;
        private Counter<int> timeSpanCount = new Counter<int>(-1, 1, 1);

        /// <summary>
        /// Create a new <see cref="TimeSpanProvider"/> and define the length of time span. Length of a single time span is defined in microseconds.
        /// </summary>
        /// <param name="spanMicroSeconds"></param>
        /// <param name="provider"></param>
        public TimeSpanProvider(long spanMicroSeconds, TimeSourceProviderBase provider)
        {
            _spanLength = spanMicroSeconds;
            _maxSpanCount = int.MaxValue;
            provider.AddSubscriber(this);
            _subscribed = true;
            _refreshCallback = new Action(() => { });
        }

        /// <summary>
        /// Create a new <see cref="TimeSpanProvider"/> and define the length of time span. 
        /// Length of a single time span is defined in seconds.
        /// </summary>
        /// <param name="spanSeconds"></param>
        /// <param name="provider"></param>
        public TimeSpanProvider(double spanSeconds, TimeSourceProviderBase provider)
        {
            _spanLength = (long)(spanSeconds * 1000L * 1000L);
            _spanBegin = -_spanLength;
            _maxSpanCount = int.MaxValue;
            provider.AddSubscriber(this);
            _subscribed = true;
            _refreshCallback = new Action(() => { });
        }

        /// <summary>
        /// Create a new <see cref="TimeSpanProvider"/> and define the length of time span. Length of a single time span is defined in microseconds.
        /// </summary>
        /// <param name="spanLength"></param>
        /// <param name="maxSpan"></param>
        /// <param name="provider"></param>
        public TimeSpanProvider(long spanLength, int maxSpan, TimeSourceProviderBase provider)
        {
            if (maxSpan <= 0)
            {
                maxSpan = int.MaxValue;
            }
            _spanLength = spanLength; _spanBegin = -_spanLength;
            provider.AddSubscriber(this);
            _subscribed = true;
        }

        public int CurrentSpan
=> timeSpanCount.Value;

        /// <summary>
        /// Time source provider provided in <see cref="StgSharp"/> internal framework.
        /// </summary>
        public static TimeSourceProviderBase DefaultProvider => StgSharpTime.OnlyInstance;

        public Action FrameUpdateMissCallback
        {
            get => _frameUpdateMissCallback;
            set => _frameUpdateMissCallback = value;
        }

        /// <summary>
        /// The method this <see cref="TimeSpanProvider"/> will call every time a new time span started.
        /// </summary>
        public Action SpanRefreshCallback
        {
            get => _refreshCallback;
            set => _refreshCallback = value;
        }

        public void SubscribeToTimeSource(TimeSourceProviderBase serviceHandler)
        {
            if (_subscribed)
            {
                return;
            }
            _subscribed = true;
            serviceHandler.AddSubscriber(this);
        }

        public void WaitNextSpan()
        {
            _refreshSemaphore.Wait();
        }

        internal bool CheckTime(long microseconds)
        {
            if (microseconds - _spanBegin < _spanLength)
            {
                return true;
            }
            
            _spanBegin = microseconds;
            timeSpanCount++;

            if (_refreshCallback != null)
            {
                _refreshCallback();
            }
            if (_refreshSemaphore.CurrentCount == 0)
            {
                _refreshSemaphore.Release();
            }
            else
            {
                if (_frameUpdateMissCallback != null)
                {
                    _frameUpdateMissCallback();
                }
            }
            if (timeSpanCount > _maxSpanCount)
            {
                return false;
            }
            return true;
        }

        public static implicit operator int(TimeSpanProvider provider)
        {
            return provider.timeSpanCount.Value;
        }

    }
}