mirror of
https://github.com/crskycode/GARbro.git
synced 2024-12-29 22:34:13 +08:00
404 lines
14 KiB
C#
404 lines
14 KiB
C#
//! \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;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|