2017-07-15 06:08:51 +04:00

343 lines
12 KiB
C#

//! \file ImageGBC.cs
//! \date Mon Oct 03 04:16:11 2016
//! \brief Primel the Adventure System resource archive.
//
// Copyright (C) 2016-2017 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.ComponentModel.Composition;
using System.IO;
using System.Windows.Media;
using GameRes.Utility;
namespace GameRes.Formats.Primel
{
internal class GbcMetaData : ImageMetaData
{
public int Flags;
}
[Export(typeof(ImageFormat))]
public class GbcFormat : ImageFormat
{
public override string Tag { get { return "GBC"; } }
public override string Description { get { return "Primel Adventure System image format"; } }
public override uint Signature { get { return 0x46434247; } } // 'GBCF'
public override ImageMetaData ReadMetaData (IBinaryStream stream)
{
var header = stream.ReadHeader (0x14);
return new GbcMetaData
{
Width = header.ToUInt32 (8),
Height = header.ToUInt32 (0xC),
BPP = header.ToUInt16 (0x10),
Flags = header.ToUInt16 (0x12),
};
}
public override ImageData Read (IBinaryStream stream, ImageMetaData info)
{
using (var reader = new GbcReader (stream.AsStream, (GbcMetaData)info))
{
reader.Unpack();
return ImageData.Create (info, reader.Format, null, reader.Pixels);
}
}
public override void Write (Stream file, ImageData image)
{
throw new System.NotImplementedException ("GbcFormat.Write not implemented");
}
}
internal sealed class GbcReader : IDisposable
{
MsbBitStream m_input;
byte[] m_output;
GbcMetaData m_info;
int m_stride;
public byte[] Pixels { get { return m_output; } }
public PixelFormat Format { get; private set; }
public GbcReader (Stream input, GbcMetaData info)
{
if (32 == info.BPP)
Format = PixelFormats.Bgra32;
else if (24 == info.BPP)
Format = PixelFormats.Bgr24;
else if (8 == info.BPP)
Format = PixelFormats.Gray8;
else
throw new InvalidFormatException();
m_input = new MsbBitStream (input, true);
m_info = info;
m_stride = (int)m_info.Width * m_info.BPP / 8;
m_output = new byte[m_stride * (int)m_info.Height];
}
public void Unpack ()
{
if (0x800 == (m_info.Flags & 0xFF00))
UnpackV2();
else
UnpackV1();
}
void UnpackV1 ()
{
m_input.Input.Position = 0x30;
int pixel_size = m_info.BPP / 8;
int blocks_w = (int)(m_info.Width + 7) / 8;
int blocks_h = (int)(m_info.Height + 7) / 8;
short[,] block = new short[pixel_size, 64];
for (int by = 0; by < blocks_h; ++by)
{
int dst_line = by * 8 * m_stride;
for (int bx = 0; bx < blocks_w; ++bx)
{
int dst_block = dst_line + bx * 8 * pixel_size;
short last = 0;
for (int i = 0; i < 64; i++)
{
last += (short)GetInt();
block[0, ZigzagOrder[i]] = last;
}
for (int j = 1; j < pixel_size; ++j)
{
for (int i = 0; i < 64; ++i)
block[j, ZigzagOrder[i]] = (short)GetInt();
RestoreBlock (block, j);
}
for (int y = 0; y < 8; ++y)
{
if (by + 1 == blocks_h && (by * 8 + y) >= m_info.Height)
break;
int src = y * 8;
int dst = dst_block + y * m_stride;
for (int x = 0; x < 8; ++x)
{
if (bx + 1 == blocks_w && (bx * 8 + x) >= m_info.Width)
break;
int p = dst + (x + 1) * pixel_size - 1;
for (int j = 0; j < pixel_size; ++j)
{
m_output[p--] = (byte)(block[j, src] - 128);
}
++src;
}
}
}
}
}
void UnpackV2 ()
{
m_input.Input.Position = 0x30;
int pixel_size = m_info.BPP / 8;
int blocks_w = (int)(m_info.Width + 7) / 8;
int blocks_h = (int)(m_info.Height + 7) / 8;
short[,] block = new short[pixel_size, 64];
for (int by = 0; by < blocks_h; ++by)
{
int dst_line = by * 8 * m_stride;
for (int bx = 0; bx < blocks_w; ++bx)
{
int dst_block = dst_line + bx * 8 * pixel_size;
for (int i = 0; i < pixel_size; ++i)
{
for (int j = 0; j < 64; ++j)
block[i,j] = 0;
RestoreBlockV2 (block, i);
}
for (int y = 0; y < 8; ++y)
{
if (by + 1 == blocks_h && (by * 8 + y) >= m_info.Height)
break;
int src = y * 8;
int dst = dst_block + y * m_stride;
for (int x = 0; x < 8; ++x)
{
if (bx + 1 == blocks_w && (bx * 8 + x) >= m_info.Width)
break;
if (4 == pixel_size)
{
m_output[dst + x * 4 + 2] = (byte)block[0, src+x];
m_output[dst + x * 4 + 1] = (byte)block[1, src+x];
m_output[dst + x * 4] = (byte)block[2, src+x];
m_output[dst + x * 4 + 3] = (byte)block[3, src+x];
}
else if (3 == pixel_size)
{
m_output[dst + x * 3 + 2] = (byte)block[0, src+x];
m_output[dst + x * 3 + 1] = (byte)block[1, src+x];
m_output[dst + x * 3] = (byte)block[2, src+x];
}
else
{
var val = block[0, src+x];
m_output[dst + x] = (byte)val;
}
}
}
}
}
}
void RestoreBlock (short[,] block, int n)
{
int row = 8;
for (int col = 1; col < 8; ++col)
{
block[n, col] += block[n, col - 1];
block[n, row] += block[n, row - 8];
row += 8;
}
row = 8;
for (int y = 1; y < 8; ++y)
{
for (int x = 1; x < 8; ++x)
{
block[n, row + x] += block[n, row + x - 9];
}
row += 8;
}
}
void RestoreBlockV2 (short[,] block, int plane)
{
int skip;
for (int i = 0; i < 64; ++i)
{
int n = GetIntV2 (out skip);
if (n != 0)
block[plane, ZigzagOrder[i]] = (short)n;
else if (0 == skip)
break;
else
i += skip - 1;
}
for (int row = 0; row < 64; row += 8)
for (int x = 1; x < 8; ++x)
{
block[plane, row+x] += block[plane, row+x-1];
}
for (int row = 8; row < 64; row += 8)
for (int x = 0; x < 8; ++x)
{
block[plane, row+x] += block[plane, row-8+x];
}
}
int GetInt ()
{
int count = m_input.GetBits (4);
switch (count)
{
case 0: return 0;
case 1: return 1;
case 2:
case 3:
case 4:
case 5:
case 6:
case 7:
return m_input.GetBits (count - 1) + (1 << (count - 1));
case 8: return -1;
case 9: return -2;
default:
return m_input.GetBits (count - 9) - (2 << (count - 9));
case -1: throw new EndOfStreamException();
}
}
int GetIntV2 (out int repeat)
{
int count = m_input.GetBits (4);
repeat = 0;
switch (count)
{
case 0:
repeat = 1;
while (repeat < 16 && 1 == m_input.GetNextBit())
++repeat;
if (16 == repeat)
repeat = 0;
return 0;
case 1: return 1;
case 2:
case 3:
case 4:
case 5:
case 6:
case 7:
return m_input.GetBits (count - 1) + (1 << (count - 1));
case 8: return -1;
case 9: return -2;
default:
return m_input.GetBits (count - 9) - (2 << (count - 9));
case -1: throw new EndOfStreamException();
}
}
bool _disposed = false;
public void Dispose ()
{
if (!_disposed)
{
m_input.Dispose();
_disposed = true;
}
}
readonly static byte[] ZigzagOrder = {
0x00, 0x01, 0x08, 0x10, 0x09, 0x02, 0x03, 0x0A,
0x11, 0x18, 0x20, 0x19, 0x12, 0x0B, 0x04, 0x05,
0x0C, 0x13, 0x1A, 0x21, 0x28, 0x30, 0x29, 0x22,
0x1B, 0x14, 0x0D, 0x06, 0x07, 0x0E, 0x15, 0x1C,
0x23, 0x2A, 0x31, 0x38, 0x39, 0x32, 0x2B, 0x24,
0x1D, 0x16, 0x0F, 0x17, 0x1E, 0x25, 0x2C, 0x33,
0x3A, 0x3B, 0x34, 0x2D, 0x26, 0x1F, 0x27, 0x2E,
0x35, 0x3C, 0x3D, 0x36, 0x2F, 0x37, 0x3E, 0x3F,
};
}
}