//-----------------------------------------------------------------------
// -----------------------------------------------------------------------
// file="HLSFAllocator.SegmentEntry"
// 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.Runtime.CompilerServices;
using System.Runtime.InteropServices;

namespace StgSharp.HighPerformance.Memory
{
    public unsafe partial class HybridLayerSegregatedFitAllocator
    {

        /// <summary>
        ///   Check if two Entries are adjacent in memory
        /// </summary>
        [MethodImpl(MethodImplOptions.AggressiveInlining)]
        private static bool AssertAdjacentTo(Entry* current, Entry* other)
        {
            // Only a predicate; never throw here. Callers rely on false to skip merging.
            return other->Position + (nuint)other->Size == current->Position ||
                   current->Position + (nuint)current->Size == other->Position;
        }

        /// <summary>
        ///   check out if a bucket is within the same higher-level boundary as currentOffset
        /// </summary>
        private bool AssertBoundary(Entry* previous, Entry* current, int level)
        {
            level = int.Min(level, MaxLevel);
            int nextLevelSize = levelSizeArray[level] * 4;
            ulong prevHead = previous->Position - (ulong)m_Buffer,
                currHead = current->Position - (ulong)m_Buffer;
            return (currHead / (ulong)nextLevelSize) == (prevHead / (ulong)nextLevelSize);
        }

        /// <summary>
        ///   Determine segment index based on size and originLevel
        /// </summary>
        [MethodImpl(MethodImplOptions.AggressiveInlining)]
        private static int DetermineSegmentIndex(long size, int level)
        {
            level = int.Min(level, MaxLevel);
            return (int)((size - 1) / levelSizeArray[level]);
        }

        [MethodImpl(MethodImplOptions.AggressiveInlining)]
        private static void InsertEntryAfter(Entry* entry, Entry* position)
        {
            Entry* next = position->NextNear;
            entry->NextNear = next;
            entry->PreviousNear = position;
            next->PreviousNear = entry;
            position->NextNear = entry;
        }

        [MethodImpl(MethodImplOptions.AggressiveInlining)]
        private static void InsertEntryBefore(Entry* entry, Entry* position)
        {
            Entry* prev = position->PreviousNear;
            entry->NextNear = position;
            entry->PreviousNear = prev;
            prev->NextNear = entry;
            position->PreviousNear = entry;
        }

        /// <summary>
        ///   Used in Free operations, efficiently remove from buckets and merge adjacent blocks.
        ///   Current Entry will serve as the final merged block. The Merge will be in lazy
        ///   strategy.
        /// </summary>
        private void MergeAndRemoveFromBuckets(Entry* entry)
        {
            if (entry == null) {
                return;
            }

            /*
            Console.WriteLine("============ Merge start =======================");
            try
            {
                Console.WriteLine("forward from entry");
                Entry* __cur = entry;
                int __cnt = 0;
                while (__cur != null && __cur != _spareMemory && __cnt < 512)
                {
                    Console.WriteLine(EntryToString(__cur));
                    __cur = __cur->NextNear;
                    __cnt++;
                    if (__cur == entry)
                    {
                        Console.WriteLine("(looped back to start)");
                        break;
                    }
                }
                Console.WriteLine(EntryToString(__cur));
                Console.WriteLine("MergeAndRemoveFromBuckets: backward from entry");
                __cur = entry;
                __cnt = 0;
                while (__cur != null && __cur != _spareMemory && __cnt < 512)
                {
                    Console.WriteLine(EntryToString(__cur));
                    __cur = __cur->PreviousNear;
                    __cnt++;
                    if (__cur == entry)
                    {
                        Console.WriteLine("(looped back to start)");
                        break;
                    }
                }
                Console.WriteLine(EntryToString(__cur));
            }
            catch { }
            finally
            {
                Console.WriteLine("==========================================");
            }
            /**/

            long size = entry->Size;
            int originLevel = entry->Level;
            int segment = DetermineSegmentIndex(size, originLevel);
            do
            {
                // previous direction merge
                Entry* p = entry->PreviousNear;
                if (p == null ||
                    p == _spareMemory ||
                    p->State != EntryState.Empty ||
                    p->Level != entry->Level ||
                    !AssertBoundary(p, entry, originLevel))
                {
                    break;
                }

                // Console.WriteLine($"{(ulong)p->Size} merged to {(ulong)entry->Size} previous");
                entry->Size += p->Size;
                entry->Position = p->Position;
                RemoveFromBucket(originLevel, DetermineSegmentIndex(p->Size, originLevel), (BucketNode*)p->Position);
                RemoveFromPositionChain(p);
            } while (true);

            do
            {
                // next direction merge
                Entry* n = entry->NextNear;
                if (n == null ||
                    n == _spareMemory ||
                    n->State != EntryState.Empty ||
                    n->Level != entry->Level ||
                    !AssertBoundary(n, entry, originLevel))
                {
                    break;
                }

                // Console.WriteLine($"{(ulong)n->Size} merged to {(ulong)entry->Size} next");
                entry->Size += n->Size;
                RemoveFromBucket(originLevel, DetermineSegmentIndex(n->Size, originLevel), (BucketNode*)n->Position);
                RemoveFromPositionChain(n);
            } while (true);
            originLevel = int.Min(9, originLevel);
            entry->Level += entry->Size >= levelSizeArray[originLevel] * 4 ? 1 : 0;
            entry->Level &= 15;

            BucketNode* bucket = (BucketNode*)entry->Position;
            bucket->EntryRef = entry;
        }

        /// <summary>
        ///   Remove node from position linked list, this method does not recycle a BucketNode from
        ///   size bucket
        /// </summary>
        private void RemoveFromPositionChain(Entry* entry)
        {
            Entry* prev = entry->PreviousNear;
            Entry* next = entry->NextNear;

            prev->NextNear = next;
            next->PreviousNear = prev;

            // Clear node pointers
            entry->NextNear = null;
            entry->PreviousNear = null;

            // Console.WriteLine("entry recycled.");
            _entries.Free((nuint)entry);
        }

        [MethodImpl(MethodImplOptions.AggressiveInlining)]
        private static void SetEntryLevel(Entry* entry, nuint position, long size)
        {
            entry ->Position = position;
            entry->Size = size;
            entry->Level = GetLevelFromSize(size);
        }

        /// <summary>
        ///   Entry structure - streamlined version (only position and state information) Stored in
        ///   external SLAB
        /// </summary>
        [StructLayout(LayoutKind.Explicit, Size = 40)]
        internal struct Entry
        {

            [FieldOffset(0)]  internal Entry* NextNear;      // Position linked list: next pointer
            [FieldOffset(8)] internal Entry* PreviousNear;  // Position linked list: previous pointer

            [FieldOffset(24)] internal int _Level;           // Level of the block in the allocator
            [FieldOffset(16)] internal long Size;           // Block size information
            [FieldOffset(36)] public EntryState State;
            [FieldOffset(28)]  public nuint Position;        // Memory block position

            public int Level
            {
                [MethodImpl(MethodImplOptions.AggressiveInlining)]
                get => _Level;
                [MethodImpl(MethodImplOptions.AggressiveInlining)]
                set
                {
                    int size = 64 * (1 << (value * 2));
                    if (Size / size > 3 && Size < 64 * 1024 * 1024L) {
                        throw new HLSFPositionChainBreakException();
                    }
                    _Level = value;
                }
            }

        }

        internal enum EntryState : uint
        {

            Spare,
            Empty,
            Alloc,
            Large

        }

        #region Entry Operations

        [MethodImpl(MethodImplOptions.AggressiveInlining)]
        private static bool IsNullOrEmptyEntryIndex(ulong index)
        {
            return index == 0;
        }

        [MethodImpl(MethodImplOptions.AggressiveInlining)]
        private static bool IsNullOrEmptyEntryIndex(Entry* entry)
        {
            return entry == null;
        }

        #endregion
    }
}