stgsharp/StgSharp.Common/PipeLine/PipelineNodePort.cs

//-----------------------------------------------------------------------
// -----------------------------------------------------------------------
// file="PipelineNodePort"
// Project: StgSharp
// AuthorGroup: Nitload
// Copyright (c) Nitload. 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 System;
using System.Collections.Generic;
using System.Linq;
using System.Runtime.ExceptionServices;
using System.Text;
using System.Threading;
using System.Threading.Tasks;

namespace StgSharp.PipeLine
{
    public class PipelineNodeInPort
    {

        private HashSet<PipelineNodeOutPort> _linkerSet = new HashSet<PipelineNodeOutPort>();

        public PipelineNodeInPort(PipelineNode container, string label)
        {
            Container = container;
            Label = label;
        }

        public IEnumerable<PipelineNodeOutPort> LinkedPorts => _linkerSet;

        public PipelineNode Container { get; init; }

        public SemaphoreSlim Waiter { get; init; }

        public string Label { get; init; }

        internal HashSet<PipelineNodeOutPort> Linkers => _linkerSet;

        public void Connect(PipelineNodeOutPort port)
        {
            ArgumentNullException.ThrowIfNull(port);
            if (_linkerSet.Add(port))
            {
                port.Linkers.Add(this);
                Container.Previous.Add(port.Container);
                port.Container.Next.Add(Container);
            }
        }

        public IEnumerator<PipelineNode> GetFormerNodes()
        {
            foreach (PipelineNodeOutPort port in _linkerSet) {
                yield return port.Container;
            }
        }

        public void Wait()
        {
            PipelineNodeOutPort.WaitAll(_linkerSet);
        }

        public static void WaitAll(IDictionary<string, PipelineNodeInPort> ports)
        {
            if (ports is null) {
                return;
            }
            foreach (PipelineNodeInPort port in ports.Values) {
                port.Wait();
            }
        }

        public static void WaitAll(IEnumerable<PipelineNodeInPort> ports)
        {
            if (ports is null) {
                return;
            }
            foreach (PipelineNodeInPort port in ports) {
                port.Wait();
            }
        }

    }

    public class PipelineNodeOutPort
    {

        private HashSet<PipelineNodeInPort> _linkerSet = new HashSet<PipelineNodeInPort>();
        private volatile int _passCount;
        private SemaphoreSlim formerCompleteSemaphore;

        public PipelineNodeOutPort(PipelineNode container, string label)
        {
            Container = container;
            Label = label;
            formerCompleteSemaphore = new SemaphoreSlim(initialCount:0, maxCount:1);
        }

        public PipelineNode Container { get; init; }

        public string Label { get; init; }

        internal HashSet<PipelineNodeInPort> Linkers => _linkerSet;

        private SemaphoreSlim Waiter => formerCompleteSemaphore;

        public void Connect(PipelineNodeInPort port)
        {
            ArgumentNullException.ThrowIfNull(port);
            if (_linkerSet.Add(port))
            {
                port.Linkers.Add(this);
                Container.Next.Add(port.Container);
                port.Container.Previous.Add(Container);
            }
        }

        public void Release()
        {
            if (_passCount == 0)
            {
                Waiter.Release();
            } else
            {
                return;
            }
        }

        public static void SkipAll(Dictionary<string, PipelineNodeOutPort> ports)
        {
            if (ports is null) {
                return;
            }
            foreach (PipelineNodeOutPort port in ports.Values) {
                port.Release();
            }
        }

        public void TransmitValue(IPipeLineConnectionPayload value)
        {
            foreach (PipelineNodeInPort item in _linkerSet) { }
        }

        public static void WaitAll(IEnumerable<PipelineNodeOutPort> ports)
        {
            if (ports is null) {
                return;
            }
            foreach (PipelineNodeOutPort port in ports) {
                port.Wait();
            }
        }

        internal void Wait()
        {
            Waiter.Wait();
            if (_passCount >= _linkerSet.Count)
            {
                Interlocked.Exchange(ref _passCount, 0);
            } else
            {
                Interlocked.Increment(ref _passCount);
                Waiter.Release();
            }
        }

    }
}