stgsharp/StgSharp.Common/Mathematics/Linear/MatrixParallel.TaskFactory.cs

//-----------------------------------------------------------------------
// -----------------------------------------------------------------------
// file="MatrixParallel.TaskWrapFactory.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.HighPerformance;
using StgSharp.HighPerformance.Memory;
using System;
using System.Collections.Generic;
using System.Linq;
using System.Runtime.CompilerServices;
using System.Text;
using System.Threading;
using System.Threading.Tasks;

using hlsfAllocator = StgSharp.HighPerformance.Memory.HybridLayerSegregatedFitAllocator;

namespace StgSharp.Mathematics
{
    internal static unsafe class MatrixParallelFactory
    {

        private static hlsfAllocator _intermediateResult;

        private static
#if NET9_0_OR_GREATER
            Lock
#else
            object
#endif
            _hlsfLock = new();

        private static SlabAllocator<ScalarPacket> _scalarAllocator ;

        private static SlabAllocator<MatrixParallelTaskPackage> _wrapAllocator ;

        [MethodImpl(MethodImplOptions.AggressiveInlining)]
        public static MatrixParallelTaskPackage* CreateBaseTask()
        {
            MatrixParallelTaskPackage* p = (MatrixParallelTaskPackage*)_wrapAllocator.Allocate();
            Span<byte> s = new Span<byte>((byte*)p, sizeof(MatrixParallelTaskPackage));
            s.Fill(0);
            return p;
        }

        [MethodImpl(MethodImplOptions.AggressiveInlining)]
        public static ScalarPacket* CreateScalarPacket()
        {
            ScalarPacket* p = (ScalarPacket*)_scalarAllocator.Allocate();
            return p;
        }

<<<<<<< HEAD:StgSharp.Common/Mathematics/Linear/MatrixParallel.TaskWrapFactory.cs
        public static MatrixParallelTaskPackage* FromTaskWrap(MatrixParallelTaskPackage* source, int offset)
        {
            MatrixParallelTaskPackage* p = (MatrixParallelTaskPackage*)_wrapAllocator.Allocate();
            /*
            *p = *source;
            p->Left += offset * p->LeftStride;
            p->Right += offset * p->RightStride;
            p->Result += offset * p->ResultStride;
            /**/
            return p;
        }

=======
        [MethodImpl(MethodImplOptions.AggressiveInlining)]
>>>>>>> 8007eee8461498a85a9515e8fd076f03cb552b64:StgSharp.Common/Mathematics/Linear/MatrixParallel.TaskFactory.cs
        public static void Release(ScalarPacket* source)
        {
            _scalarAllocator.Free((nuint)source);
        }

        [MethodImpl(MethodImplOptions.AggressiveInlining)]
        public static void Release(MatrixParallelTaskPackage* source)
        {
            if (source->Scalar != null) {
                _scalarAllocator.Free((nuint)source->Scalar);
            }
            _wrapAllocator.Free((nuint)source);
        }

        [MethodImpl(MethodImplOptions.AggressiveInlining)]
        internal static void Init()
        {
<<<<<<< HEAD:StgSharp.Common/Mathematics/Linear/MatrixParallel.TaskWrapFactory.cs
            int count = group.Length;
            group[0] = (nint)source;
            int l = source->LeftPrimStride, r = source->RightPrimStride, res = source->ResultPrimStride, c = source->PrimCount / count;
            source->LeftPrimStride *= count;
            source->RightPrimStride *= count;
            source->ResultPrimStride *= count;
            source->PrimCount = c;
            int remain = source->PrimCount - (c * count);
            for (int i = 1; i < count; i++)
            {
                nint handle = (nint)_wrapAllocator.Allocate();
                MatrixParallelTaskPackage* p = (MatrixParallelTaskPackage*)handle;
                *p = *source;
                p->Left += l * i;
                p->Right += r * i;
                p->Result += res * i;
                p->PrimCount = c;
                group[i] = handle;
            }

            // Distribute remaining work to the last few tasks
            for (int i = 0; i < remain; i++)
            {
                int targetIndex = count - 1 - i;
                if (targetIndex >= 0) {
                    ((MatrixParallelTaskPackage*)group[targetIndex])->PrimCount++;
                }
            }
=======
            _scalarAllocator = SlabAllocator<ScalarPacket>.Create(64, SlabBufferLayout.Chunked, true);
            _wrapAllocator = SlabAllocator<MatrixParallelTaskPackage>.Create(64, SlabBufferLayout.Chunked, true);
            _intermediateResult = new hlsfAllocator((nuint)(sizeof(MatrixKernel) * 64));
>>>>>>> 8007eee8461498a85a9515e8fd076f03cb552b64:StgSharp.Common/Mathematics/Linear/MatrixParallel.TaskFactory.cs
        }

    }
}