// <copyright file="Block.cs" company="James Jackson-South">
// Copyright (c) James Jackson-South and contributors.
// Licensed under the Apache License, Version 2.0.
// </copyright>

using System;
using System.Buffers;

namespace ImageSharp.Formats
{
    /// <summary>
    /// Represents an 8x8 block of coefficients to transform and encode.
    /// </summary>
    internal struct Block : IDisposable
    {
        private static ArrayPool<int> IntArrayPool => ArrayPool<int>.Shared; /*= ArrayPool<int>.Create(BlockSize, 50);*/

        /// <summary>
        /// Gets the size of the block.
        /// </summary>
        public const int BlockSize = 64;

        /// <summary>
        /// The array of block data.
        /// </summary>
        public int[] Data;

        /// <summary>
        /// Initializes a new instance of the <see cref="Block"/> class.
        /// </summary>
        //public Block()
        //{
        //    this.data = new int[BlockSize];
        //}

        public void Init()
        {
            //this.Data = new int[BlockSize];
            this.Data = IntArrayPool.Rent(BlockSize);
        }

        public static Block Create()
        {
            var block = new Block();
            block.Init();
            return block;
        }

        public static Block[] CreateArray(int size)
        {
            Block[] result = new Block[size];
            for (int i = 0; i < result.Length; i++)
            {
                result[i].Init();
            }
            return result;
        }

        public bool IsInitialized => this.Data != null;

        /// <summary>
        /// Gets the pixel data at the given block index.
        /// </summary>
        /// <param name="index">The index of the data to return.</param>
        /// <returns>
        /// The <see cref="int"/>.
        /// </returns>
        public int this[int index]
        {
            get { return this.Data[index]; }
            set { this.Data[index] = value; }
        }

        // TODO: Refactor Block.Dispose() callers to always use 'using' or 'finally' statement!
        public void Dispose()
        {
            if (Data != null)
            {
                IntArrayPool.Return(Data, true);
                Data = null;
            }
        }

        public static void DisposeAll(Block[] blocks)
        {
            for (int i = 0; i < blocks.Length; i++)
            {
                blocks[i].Dispose();
            }
        }
    }
}