mirror of
https://github.com/crskycode/GARbro.git
synced 2024-11-23 21:55:34 +08:00
317 lines
12 KiB
C#
317 lines
12 KiB
C#
//! \file ImageMI4.cs
|
|
//! \date Sun Jul 12 15:40:39 2015
|
|
//! \brief ShiinaRio engine image format.
|
|
//
|
|
// 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.ComponentModel.Composition;
|
|
using System.IO;
|
|
using System.Windows.Media;
|
|
|
|
namespace GameRes.Formats.ShiinaRio
|
|
{
|
|
[Export(typeof(ImageFormat))]
|
|
public class Mi4Format : ImageFormat
|
|
{
|
|
public override string Tag { get { return "MI4"; } }
|
|
public override string Description { get { return "ShiinaRio image format"; } }
|
|
public override uint Signature { get { return 0x3449414D; } } // 'MAI4'
|
|
|
|
public override ImageMetaData ReadMetaData (IBinaryStream file)
|
|
{
|
|
file.Position = 8;
|
|
uint width = file.ReadUInt32();
|
|
uint height = file.ReadUInt32();
|
|
return new ImageMetaData
|
|
{
|
|
Width = width,
|
|
Height = height,
|
|
BPP = 24,
|
|
};
|
|
}
|
|
|
|
public override ImageData Read (IBinaryStream stream, ImageMetaData info)
|
|
{
|
|
var reader = new Reader (stream, (int)info.Width, (int)info.Height);
|
|
try
|
|
{
|
|
reader.Unpack (MaiVersion.Second);
|
|
}
|
|
catch
|
|
{
|
|
reader.Unpack (MaiVersion.First);
|
|
}
|
|
return ImageData.Create (info, PixelFormats.Bgr24, null, reader.Data, reader.Stride);
|
|
}
|
|
|
|
public override void Write (Stream file, ImageData image)
|
|
{
|
|
throw new System.NotImplementedException ("Mi4Format.Write not implemented");
|
|
}
|
|
|
|
internal enum MaiVersion
|
|
{
|
|
First, Second
|
|
}
|
|
|
|
internal sealed class Reader
|
|
{
|
|
IBinaryStream m_input;
|
|
byte[] m_output;
|
|
int m_stride;
|
|
|
|
public byte[] Data { get { return m_output; } }
|
|
public int Stride { get { return m_stride; } }
|
|
|
|
public Reader (IBinaryStream file, int width, int height)
|
|
{
|
|
m_input = file;
|
|
m_stride = width * 3;
|
|
m_output = new byte[m_stride*height];
|
|
}
|
|
|
|
public void Unpack (MaiVersion version = MaiVersion.First)
|
|
{
|
|
m_input.Position = 0x10;
|
|
m_bit_count = 0;
|
|
LoadBits();
|
|
if (MaiVersion.First == version)
|
|
UnpackV1();
|
|
else
|
|
UnpackV2();
|
|
}
|
|
|
|
int m_bit_count;
|
|
uint m_bits;
|
|
|
|
void LoadBits ()
|
|
{
|
|
for (int i = 0; i < 4; ++i)
|
|
{
|
|
int b = m_input.ReadByte();
|
|
if (-1 == b)
|
|
break;
|
|
m_bits = (m_bits >> 8) | (uint)(b << 24);
|
|
m_bit_count += 8;
|
|
}
|
|
}
|
|
|
|
uint GetBit ()
|
|
{
|
|
uint bit = m_bits >> 31;
|
|
m_bits <<= 1;
|
|
if (0 == --m_bit_count)
|
|
{
|
|
LoadBits();
|
|
}
|
|
return bit;
|
|
}
|
|
|
|
uint GetBits (int count)
|
|
{
|
|
int avail_bits = Math.Min (count, m_bit_count);
|
|
uint bits = m_bits >> (32 - avail_bits);
|
|
m_bits <<= avail_bits;
|
|
m_bit_count -= avail_bits;
|
|
count -= avail_bits;
|
|
if (0 == m_bit_count)
|
|
{
|
|
LoadBits();
|
|
if (count > 0)
|
|
{
|
|
bits = bits << count | m_bits >> (32 - count);
|
|
m_bits <<= count;
|
|
m_bit_count -= count;
|
|
}
|
|
}
|
|
return bits;
|
|
}
|
|
|
|
void UnpackV1 ()
|
|
{
|
|
int dst = 0;
|
|
byte b = 0, g = 0, r = 0;
|
|
while (dst < m_output.Length)
|
|
{
|
|
if (GetBit() == 0)
|
|
{
|
|
if (GetBit() != 0)
|
|
{
|
|
b = m_input.ReadUInt8();
|
|
g = m_input.ReadUInt8();
|
|
r = m_input.ReadUInt8();
|
|
}
|
|
else if (GetBit() != 0)
|
|
{
|
|
byte v = (byte)GetBits (2);
|
|
if (3 == v)
|
|
{
|
|
b = m_output[dst - m_stride];
|
|
g = m_output[dst - m_stride + 1];
|
|
r = m_output[dst - m_stride + 2];
|
|
}
|
|
else
|
|
{
|
|
b += (byte)(v - 1);
|
|
g += (byte)(GetBits(2) - 1);
|
|
r += (byte)(GetBits(2) - 1);
|
|
}
|
|
}
|
|
else if (GetBit() != 0)
|
|
{
|
|
byte v = (byte)(GetBits(3));
|
|
if (7 == v)
|
|
{
|
|
b = m_output[dst - m_stride + 3];
|
|
g = m_output[dst - m_stride + 4];
|
|
r = m_output[dst - m_stride + 5];
|
|
}
|
|
else
|
|
{
|
|
b += (byte)(v - 3);
|
|
g += (byte)(GetBits(3) - 3);
|
|
r += (byte)(GetBits(3) - 3);
|
|
}
|
|
}
|
|
else if (GetBit() != 0)
|
|
{
|
|
byte v = (byte)GetBits (4);
|
|
if (0xF == v)
|
|
{
|
|
b = m_output[dst - m_stride - 3];
|
|
g = m_output[dst - m_stride - 2];
|
|
r = m_output[dst - m_stride - 1];
|
|
}
|
|
else
|
|
{
|
|
b += (byte)(v - 7);
|
|
g += (byte)(GetBits(4) - 7);
|
|
r += (byte)(GetBits(4) - 7);
|
|
}
|
|
}
|
|
else
|
|
{
|
|
b += (byte)(GetBits(5) - 15);
|
|
g += (byte)(GetBits(5) - 15);
|
|
r += (byte)(GetBits(5) - 15);
|
|
}
|
|
}
|
|
m_output[dst++] = b;
|
|
m_output[dst++] = g;
|
|
m_output[dst++] = r;
|
|
}
|
|
}
|
|
|
|
void UnpackV2 ()
|
|
{
|
|
int dst = 0;
|
|
byte b = 0, g = 0, r = 0;
|
|
while (dst < m_output.Length)
|
|
{
|
|
if (GetBit() == 0)
|
|
{
|
|
if (GetBit() != 0)
|
|
{
|
|
if (m_input.PeekByte() != -1)
|
|
{
|
|
b = m_input.ReadUInt8();
|
|
g = m_input.ReadUInt8();
|
|
r = m_input.ReadUInt8();
|
|
}
|
|
}
|
|
else if (GetBit() != 0)
|
|
{
|
|
byte v = (byte)(GetBits(2));
|
|
if (3 == v)
|
|
{
|
|
b = m_output[dst - m_stride];
|
|
g = m_output[dst - m_stride + 1];
|
|
r = m_output[dst - m_stride + 2];
|
|
}
|
|
else
|
|
{
|
|
b += (byte)(v - 1);
|
|
v = (byte)GetBits (2);
|
|
if (3 == v)
|
|
{
|
|
if (GetBit() != 0)
|
|
{
|
|
b = m_output[dst - m_stride - 3];
|
|
g = m_output[dst - m_stride - 2];
|
|
r = m_output[dst - m_stride - 1];
|
|
}
|
|
else
|
|
{
|
|
b = m_output[dst - m_stride + 3];
|
|
g = m_output[dst - m_stride + 4];
|
|
r = m_output[dst - m_stride + 5];
|
|
}
|
|
}
|
|
else
|
|
{
|
|
g += (byte)(v - 1);
|
|
r += (byte)(GetBits(2) - 1);
|
|
}
|
|
}
|
|
}
|
|
else if (GetBit() != 0)
|
|
{
|
|
byte v = (byte)(GetBits(3));
|
|
if (7 == v)
|
|
{
|
|
b = m_output[dst - m_stride];
|
|
g = m_output[dst - m_stride + 1];
|
|
r = m_output[dst - m_stride + 2];
|
|
b += (byte)(GetBits(3) - 3);
|
|
g += (byte)(GetBits(3) - 3);
|
|
r += (byte)(GetBits(3) - 3);
|
|
}
|
|
else
|
|
{
|
|
b += (byte)(v - 3);
|
|
g += (byte)(GetBits(3) - 3);
|
|
r += (byte)(GetBits(3) - 3);
|
|
}
|
|
}
|
|
else if (GetBit() != 0)
|
|
{
|
|
b += (byte)(GetBits(4) - 7);
|
|
g += (byte)(GetBits(4) - 7);
|
|
r += (byte)(GetBits(4) - 7);
|
|
}
|
|
else
|
|
{
|
|
b += (byte)(GetBits(5) - 15);
|
|
g += (byte)(GetBits(5) - 15);
|
|
r += (byte)(GetBits(5) - 15);
|
|
}
|
|
}
|
|
m_output[dst++] = b;
|
|
m_output[dst++] = g;
|
|
m_output[dst++] = r;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|