// Copyright (c) Xenko contributors (https://xenko.com) and Silicon Studio Corp. (https://www.siliconstudio.co.jp)
// Distributed under the MIT license. See the LICENSE.md file in the project root for more information.

using System;
using System.Collections.Generic;
using System.Runtime.Serialization;

namespace math
{
    /// <summary>
    /// Implementation of a "Guillotine" packer.
    /// More information at http://clb.demon.fi/files/RectangleBinPack.pdf.
    /// </summary>
    public class GuillotinePacker
    {
        private readonly List<Rectangle> freeRectangles = new List<Rectangle>();
        private readonly List<Rectangle> tempFreeRectangles = new List<Rectangle>();

        /// <summary>
        /// A delegate callback used by <see cref="TryInsert"/>
        /// </summary>
        /// <param name="cascadeIndex">The index of the rectangle</param>
        /// <param name="rectangle">The rectangle found</param>
        public delegate void InsertRectangleCallback( int cascadeIndex, ref Rectangle rectangle );

        /// <summary>
        /// Current width used by the packer.
        /// </summary>
        public int Width { get; private set; }

        /// <summary>
        /// Current height used by the packer.
        /// </summary>
        public int Height { get; private set; }

        /// <summary>
        /// Clears the specified region.
        /// </summary>
        /// <param name="width">The width.</param>
        /// <param name="height">The height.</param>
        public void Clear( int width, int height )
        {
            freeRectangles.Clear();
            freeRectangles.Add( new Rectangle { X = 0, Y = 0, Width = width, Height = height } );

            Width = width;
            Height = height;
        }

        /// <summary>
        /// Clears the whole region.
        /// </summary>
        public virtual void Clear()
        {
            Clear( Width, Height );
        }

        /// <summary>
        /// Frees the specified old rectangle.
        /// </summary>
        /// <param name="oldRectangle">The old rectangle.</param>
        public void Free( ref Rectangle oldRectangle )
        {
            freeRectangles.Add( oldRectangle );
        }

        /// <summary>
        /// Tries to fit a single rectangle with the specified width and height.
        /// </summary>
        /// <param name="width">Width requested.</param>
        /// <param name="height">Height requested</param>
        /// <param name="bestRectangle">Fill with the rectangle if it was successfully inserted.</param>
        /// <returns><c>true</c> if it was successfully inserted.</returns>
        public bool Insert( int width, int height, ref Rectangle bestRectangle )
        {
            return Insert( width, height, freeRectangles, ref bestRectangle );
        }

        /// <summary>
        /// Tries to fit multiple rectangle with (width, height).
        /// </summary>
        /// <param name="width">Width requested.</param>
        /// <param name="height">Height requested</param>
        /// <param name="count">The number of rectangle to fit.</param>
        /// <param name="inserted">A callback called for each rectangle successfully fitted.</param>
        /// <returns><c>true</c> if all rectangles were successfully fitted.</returns>
        public bool TryInsert( int width, int height, int count, InsertRectangleCallback inserted )
        {
            var bestRectangle = new Rectangle();
            tempFreeRectangles.Clear();
            foreach( var freeRectangle in freeRectangles )
            {
                tempFreeRectangles.Add( freeRectangle );
            }

            for( var i = 0; i < count; ++i )
            {
                if( !Insert( width, height, tempFreeRectangles, ref bestRectangle ) )
                {
                    tempFreeRectangles.Clear();
                    return false;
                }

                inserted( i, ref bestRectangle );
            }

            // if the insertion went well, use the new configuration
            freeRectangles.Clear();
            foreach( var tempFreeRectangle in tempFreeRectangles )
            {
                freeRectangles.Add( tempFreeRectangle );
            }
            tempFreeRectangles.Clear();

            return true;
        }

        private static bool Insert( int width, int height, List<Rectangle> freeRectanglesList, ref Rectangle bestRectangle )
        {
            // Info on algorithm: http://clb.demon.fi/files/RectangleBinPack.pdf
            int bestScore = int.MaxValue;
            int freeRectangleIndex = -1;

            // Find space for new rectangle
            for( int i = 0; i < freeRectanglesList.Count; ++i )
            {
                var currentFreeRectangle = freeRectanglesList[i];
                if( width == currentFreeRectangle.Width && height == currentFreeRectangle.Height )
                {
                    // Perfect fit
                    bestRectangle.X = currentFreeRectangle.X;
                    bestRectangle.Y = currentFreeRectangle.Y;
                    bestRectangle.Width = width;
                    bestRectangle.Height = height;
                    freeRectangleIndex = i;
                    break;
                }
                if( width <= currentFreeRectangle.Width && height <= currentFreeRectangle.Height )
                {
                    // Can fit inside
                    // Use "BAF" heuristic (best area fit)
                    var score = currentFreeRectangle.Width * currentFreeRectangle.Height - width * height;
                    if( score < bestScore )
                    {
                        bestRectangle.X = currentFreeRectangle.X;
                        bestRectangle.Y = currentFreeRectangle.Y;
                        bestRectangle.Width = width;
                        bestRectangle.Height = height;
                        bestScore = score;
                        freeRectangleIndex = i;
                    }
                }
            }

            // No space could be found
            if( freeRectangleIndex == -1 )
                return false;

            var freeRectangle = freeRectanglesList[freeRectangleIndex];

            // Choose an axis to split (trying to minimize the smaller area "MINAS")
            int w = freeRectangle.Width - bestRectangle.Width;
            int h = freeRectangle.Height - bestRectangle.Height;
            var splitHorizontal = ( bestRectangle.Width * h > w * bestRectangle.Height );

            // Form the two new rectangles.
            var bottom = new Rectangle { X = freeRectangle.X, Y = freeRectangle.Y + bestRectangle.Height, Width = splitHorizontal ? freeRectangle.Width : bestRectangle.Width, Height = h };
            var right = new Rectangle { X = freeRectangle.X + bestRectangle.Width, Y = freeRectangle.Y, Width = w, Height = splitHorizontal ? bestRectangle.Height : freeRectangle.Height };

            if( bottom.Width > 0 && bottom.Height > 0 )
                freeRectanglesList.Add( bottom );
            if( right.Width > 0 && right.Height > 0 )
                freeRectanglesList.Add( right );

            // Remove previously selected freeRectangle
            if( freeRectangleIndex != freeRectanglesList.Count - 1 )
                freeRectanglesList[freeRectangleIndex] = freeRectanglesList[freeRectanglesList.Count - 1];
            freeRectanglesList.RemoveAt( freeRectanglesList.Count - 1 );

            return true;
        }
    }
}