//! \file       ImageEDT.cs
//! \date       Sun Feb 15 10:13:07 2015
//! \brief      Active Soft image format implementation.
//
// Copyright (C) 2015 by morkt
//
// 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.ComponentModel.Composition;
using System.IO;
using System.Text;
using System.Windows;
using System.Windows.Media;
using System.Windows.Media.Imaging;
using GameRes.Utility;

namespace GameRes.Formats.AdPack
{
    internal class EdtMetaData : ImageMetaData
    {
        public uint CompSize;
        public uint ExtraSize;
    }

    internal class Ed8MetaData : ImageMetaData
    {
        public uint PaletteSize;
        public uint CompSize;
    }

    internal class BitReader
    {
        int m_bits = 1;

        protected void ResetBits ()
        {
            m_bits = 1;
        }

        protected int NextBit (Stream input)
        {
            if (1 == m_bits)
            {
                m_bits = input.ReadByte();
                if (-1 == m_bits)
                    throw new InvalidFormatException ("Unexpected end of input");
                m_bits |= 0x100;
            }
            int bit = m_bits & 1;
            m_bits >>= 1;
            return bit;
        }

        protected int ReadBits (int data, int count, Stream input)
        {
            while (count > 0)
            {
                data = (data << 1) | NextBit (input);
                --count;
            }
            return data;
        }

        protected int CountBits (Stream input)
        {
            int bit = 1, count = 0;
            while (count < 0x20 && 1 == bit)
            {
                ++count;
                bit = NextBit (input);
            }
            if (--count != 0)
                return ReadBits (1, count, input);
            else 
                return 1;
        }
    }

    [Export(typeof(ImageFormat))]
    public class EdtFormat : ImageFormat
    {
        public override string         Tag { get { return "EDT"; } }
        public override string Description { get { return "Active Soft RGB image format"; } }
        public override uint     Signature { get { return 0x5552542eu; } } // '.TRU'

        public override void Write (Stream file, ImageData image)
        {
            throw new NotImplementedException ("EdtFormat.Write not implemented");
        }

        public override ImageMetaData ReadMetaData (Stream stream)
        {
            var header = new byte[0x22];
            if (header.Length != stream.Read (header, 0, header.Length))
                return null;
            if (!Binary.AsciiEqual (header, ".TRUE\x8d\x5d\x8c\xcb\x00"))
                return null;
            uint width  = LittleEndian.ToUInt16 (header, 0x0e);
            uint height = LittleEndian.ToUInt16 (header, 0x10);
            uint comp_size  = LittleEndian.ToUInt32 (header, 0x1a);
            uint extra_size = LittleEndian.ToUInt32 (header, 0x1e);
            if (extra_size % 3 != 0 || 0 == extra_size)
                return null;

            return new EdtMetaData
            {
                Width = width,
                Height = height,
                BPP = 24,
                CompSize = comp_size,
                ExtraSize = extra_size,
            };
        }

        public override ImageData Read (Stream stream, ImageMetaData info)
        {
            var meta = info as EdtMetaData;
            if (null == meta)
                throw new ArgumentException ("EdtFormat.Read should be supplied with EdtMetaData", "info");

            stream.Position = 0x22;
            using (var reader = new Reader (stream, meta))
            {
                reader.Unpack();
                byte[] pixels = reader.Data;
                var bitmap = BitmapSource.Create ((int)meta.Width, (int)meta.Height, 96, 96,
                    PixelFormats.Bgr24, null, pixels, (int)meta.Width*3);
                bitmap.Freeze();
                return new ImageData (bitmap, meta);
            }
        }

        internal class Reader : BitReader, IDisposable
        {
            MemoryStream    m_packed;
            MemoryStream    m_extra;
            byte[]          m_output;
            int[]           ShiftTable = new int[32];

            public byte[] Data { get { return m_output; } }

            public Reader (Stream file, EdtMetaData info)
            {
                byte[] packed = new byte[info.CompSize];
                file.Read (packed, 0, packed.Length);
                m_packed = new MemoryStream (packed, false);

                byte[] extra = new byte[info.ExtraSize];
                file.Read (extra, 0, extra.Length);
                m_extra = new MemoryStream (extra, false);
                m_output = new byte[info.Width*info.Height*3];

                int stride = (int)info.Width * 3;
                int offset = stride * -4;
                int i = 0;
                while (offset != 0)
                {
                    int shift = offset - 9;
                    for (int j = 0; j < 7; ++j)
                    {
                        ShiftTable[i++] = shift;
                        shift += 3;
                    }
                    offset += stride;
                }
                offset = -12;
                while (offset != 0)
                {
                    ShiftTable[i++] = offset;
                    offset += 3;
                }
            }

            public void Unpack ()
            {
                int dst = 0;
                if (3 != m_extra.Read (m_output, dst, 3))
                    throw new InvalidFormatException ("Unexpected end of input");
                dst += 3;
                while (dst < m_output.Length)
                {
                    if (1 == NextBit (m_packed))
                    {
                        if (0 == NextBit (m_packed))
                        {
                            int offset = ShiftTable[ReadBits (0, 5, m_packed)];
                            if (dst < -offset)
                                return;
                            int count = CountBits (m_packed) * 3;
                            Binary.CopyOverlapped (m_output, dst + offset, dst, count);
                            dst += count;
                        }
                        else
                        {
                            int offset = -3;
                            if (1 == NextBit (m_packed))
                            {
                                offset = ShiftTable[(0x11191718 >> ((ReadBits (0, 2, m_packed) << 3)) & 0xFF)];
                                if (dst < -offset)
                                    return;
                            }
                            for (int i = 0; i < 3; ++i)
                            {
                                int b = m_output[dst+offset];
                                if (b < 2)
                                    b = 2;
                                else if (b > 0xfd)
                                    b = 0xfd;
                                if (1 == NextBit (m_packed))
                                {
                                    int b2 = 1 + NextBit (m_packed);
                                    if (0 == NextBit (m_packed))
                                        b -= b2;
                                    else
                                        b += b2;
                                }
                                m_output[dst++] = (byte)b;
                            }
                        }
                    }
                    else
                    {
                        if (3 != m_extra.Read (m_output, dst, 3))
                            throw new InvalidFormatException ("Unexpected end of input");
                        dst += 3;
                    }
                }
            }

            #region IDisposable Members
            bool disposed = false;

            public void Dispose ()
            {
                Dispose (true);
                GC.SuppressFinalize (this);
            }

            protected virtual void Dispose (bool disposing)
            {
                if (!disposed)
                {
                    if (disposing)
                    {
                        m_packed.Dispose();
                        m_extra.Dispose();
                    }
                    disposed = true;
                }
            }
            #endregion
        }
    }

    [Export(typeof(ImageFormat))]
    public class Ed8Format : ImageFormat
    {
        public override string         Tag { get { return "ED8"; } }
        public override string Description { get { return "Active Soft indexed image format"; } }
        public override uint     Signature { get { return 0x6942382eu; } } // '.8Bi'

        public Ed8Format ()
        {
            Extensions = new string[] { "ed8", "sal" };
        }

        public override void Write (Stream file, ImageData image)
        {
            throw new NotImplementedException ("Ed8Format.Write not implemented");
        }

        public override ImageMetaData ReadMetaData (Stream stream)
        {
            var header = new byte[0x1a];
            if (header.Length != stream.Read (header, 0, header.Length))
                return null;
            if (!Binary.AsciiEqual (header, ".8Bit\x8d\x5d\x8c\xcb\x00"))
                return null;
            uint width  = LittleEndian.ToUInt16 (header, 0x0e);
            uint height = LittleEndian.ToUInt16 (header, 0x10);
            uint palette_size = LittleEndian.ToUInt32 (header, 0x12);
            uint comp_size  = LittleEndian.ToUInt32 (header, 0x16);
            if (palette_size > 0x100)
                return null;

            return new Ed8MetaData
            {
                Width = width,
                Height = height,
                BPP = 8,
                PaletteSize = palette_size,
                CompSize = comp_size,
            };
        }

        public override ImageData Read (Stream stream, ImageMetaData info)
        {
            var meta = info as Ed8MetaData;
            if (null == meta)
                throw new ArgumentException ("Ed8Format.Read should be supplied with Ed8MetaData", "info");

            stream.Position = 0x1a;
            var reader = new Reader (stream, meta);
            reader.Unpack();
            byte[] pixels = reader.Data;
            var palette = new BitmapPalette (reader.Palette);
            var bitmap = BitmapSource.Create ((int)info.Width, (int)info.Height, 96, 96,
                PixelFormats.Indexed8, palette, pixels, (int)info.Width);
            bitmap.Freeze();
            return new ImageData (bitmap, info);
        }

        internal class Reader : BitReader
        {
            Stream          m_input;
            byte[]          m_data;
            Color[]         m_palette;
            int             m_width;

            public Color[] Palette { get { return m_palette; } }
            public byte[]     Data { get { return m_data; } }

            private static readonly sbyte[] ShiftTable = new sbyte[] {
            //  (-1,0) (0,1)  (-2,0) (-1,1)  (1,1) (0,2)  (-2,1) (2,1)
                -0x10,  0x01, -0x20, -0x0F,  0x11,  0x02, -0x1F,  0x21,
            //  (-2,2) (-1,2) (1,2)  (2,2)   (0,3) (-1,3)
                -0x1E, -0x0E,  0x12,  0x22,  0x03, -0x0D
            };

            public Reader (Stream file, Ed8MetaData info)
            {
                m_width = (int)info.Width;
                int palette_size = (int)info.PaletteSize*3;
                var palette_data = new byte[Math.Max (0x300, palette_size)];
                if (palette_size != file.Read (palette_data, 0, palette_size))
                    throw new InvalidFormatException();
                m_palette = new Color[0x100];
                for (int i = 0; i < m_palette.Length; ++i)
                {
                    m_palette[i] = Color.FromRgb (palette_data[i*3+2], palette_data[i*3+1], palette_data[i*3]);
                }
                m_input = file;
                m_data = new byte[info.Width * info.Height];
            }

            public void Unpack ()
            {
                int data_pos = 0;
                while (data_pos < m_data.Length)
                {
                    m_data[data_pos++] = (byte)ReadBits (0, 8, m_input);
                    if (m_data.Length == data_pos)
                        break;
                    if (1 == NextBit (m_input))
                        continue;
                    int prev_code = -1;
                    while (data_pos < m_data.Length)
                    {
                        int code = 0;
                        if (1 == NextBit (m_input))
                        {
                            if (1 == NextBit (m_input))
                                code = NextBit (m_input) + 1;
                            code = (code << 1) + NextBit (m_input) + 1;
                        }
                        code = (code << 1) + NextBit (m_input);
                        if (code == prev_code)
                            break;
                        prev_code = code;
                        int count = CountBits (m_input);
                        if (prev_code >= 2)
                           ++count;
                        if (data_pos + count > m_data.Length)
                            throw new InvalidFormatException();
                        int shift = ShiftTable[prev_code];
                        int offset = (shift >> 4) - (shift & 0xf) * m_width;
                        Binary.CopyOverlapped (m_data, data_pos+offset, data_pos, count);
                        data_pos += count;
                    }
                }
            }
        }
    }
}