GARbro-mirror/ArcFormats/ExHibit/ImageGYU.cs

345 lines
12 KiB
C#
Raw Normal View History

2015-11-02 09:51:04 +08:00
//! \file ImageGYU.cs
//! \date Mon Nov 02 00:38:41 2015
//! \brief ExHIBIT engine image format.
//
2016-08-04 09:05:44 +08:00
// Copyright (C) 2015-2016 by morkt
2015-11-02 09:51:04 +08:00
//
// 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;
2016-08-04 09:05:44 +08:00
using System.Collections.Generic;
2015-11-02 09:51:04 +08:00
using System.IO;
2016-08-04 09:05:44 +08:00
using System.Linq;
2015-11-02 09:51:04 +08:00
using System.Windows.Media.Imaging;
using System.Windows.Media;
using GameRes.Utility;
using GameRes.Compression;
using GameRes.Cryptography;
2016-08-04 09:05:44 +08:00
using GameRes.Formats.Strings;
using System.Collections;
2015-11-02 09:51:04 +08:00
namespace GameRes.Formats.ExHibit
{
internal class GyuMetaData : ImageMetaData
{
public int Flags;
public int CompressionMode;
public uint Key;
public int DataSize;
public int AlphaSize;
public int PaletteSize;
}
2016-08-04 09:05:44 +08:00
[Serializable]
public class GyuMap : ResourceScheme
{
public Dictionary<string, Dictionary<int, uint>> NumericKeys;
public Dictionary<string, Dictionary<string, uint>> StringKeys;
2016-08-04 09:05:44 +08:00
}
2015-11-02 09:51:04 +08:00
[Export(typeof(ImageFormat))]
public class GyuFormat : ImageFormat
{
public override string Tag { get { return "GYU"; } }
public override string Description { get { return "ExHIBIT engine image format"; } }
public override uint Signature { get { return 0x1A555947; } } // 'GYU'
GyuMap DefaultScheme = new GyuMap {
NumericKeys = new Dictionary<string, Dictionary<int, uint>>(),
StringKeys = new Dictionary<string, Dictionary<string, uint>>(),
};
2016-08-04 09:05:44 +08:00
public override ResourceScheme Scheme
{
get { return DefaultScheme; }
set { DefaultScheme = (GyuMap)value; }
2016-08-04 09:05:44 +08:00
}
2016-10-16 13:22:53 +08:00
public override ImageMetaData ReadMetaData (IBinaryStream stream)
2015-11-02 09:51:04 +08:00
{
2016-10-16 13:22:53 +08:00
stream.Position = 4;
return new GyuMetaData
2015-11-02 09:51:04 +08:00
{
2016-10-16 13:22:53 +08:00
Flags = stream.ReadUInt16(),
CompressionMode = stream.ReadUInt16(),
Key = stream.ReadUInt32(),
BPP = stream.ReadInt32(),
Width = stream.ReadUInt32(),
Height = stream.ReadUInt32(),
DataSize = stream.ReadInt32(),
AlphaSize = stream.ReadInt32(),
PaletteSize = stream.ReadInt32(),
};
2015-11-02 09:51:04 +08:00
}
IDictionary CurrentMap = null;
2016-08-04 09:05:44 +08:00
2016-10-16 13:22:53 +08:00
public override ImageData Read (IBinaryStream stream, ImageMetaData info)
2015-11-02 09:51:04 +08:00
{
var meta = (GyuMetaData)info;
if (0 == meta.Key)
2016-08-04 09:05:44 +08:00
{
object token = null;
if (null == CurrentMap)
CurrentMap = QueryScheme();
if (CurrentMap != null)
2016-08-04 09:05:44 +08:00
{
var name = Path.GetFileNameWithoutExtension (meta.FileName);
int num;
if (int.TryParse (name, out num) && CurrentMap.Contains (num))
token = num;
else if (CurrentMap.Contains (name))
token = name;
2016-08-04 09:05:44 +08:00
}
if (null == token)
2016-08-04 09:05:44 +08:00
{
CurrentMap = null;
throw new UnknownEncryptionScheme ("Unknown image encryption key");
}
meta.Key = (uint)CurrentMap[token];
2016-08-04 09:05:44 +08:00
}
2016-10-16 13:22:53 +08:00
var reader = new GyuReader (stream.AsStream, meta);
2015-11-02 09:51:04 +08:00
reader.Unpack();
return ImageData.CreateFlipped (meta, reader.Format, reader.Palette, reader.Data, reader.Stride);
}
public override void Write (Stream file, ImageData image)
{
throw new System.NotImplementedException ("GyuFormat.Write not implemented");
}
2016-08-04 09:05:44 +08:00
private IDictionary QueryScheme ()
2016-08-04 09:05:44 +08:00
{
var options = Query<GyuOptions> (arcStrings.GYUImageEncrypted);
return options.Scheme;
}
public override ResourceOptions GetDefaultOptions ()
{
return new GyuOptions { Scheme = GetScheme (Properties.Settings.Default.GYUTitle) };
2016-08-04 09:05:44 +08:00
}
public override object GetAccessWidget ()
{
var titles = DefaultScheme.NumericKeys.Keys.Concat (DefaultScheme.StringKeys.Keys).OrderBy (x => x);
return new GUI.WidgetGYU (titles);
2016-08-04 09:05:44 +08:00
}
IDictionary GetScheme (string title)
2016-08-04 09:05:44 +08:00
{
Dictionary<int, uint> num_scheme = null;
if (DefaultScheme.NumericKeys.TryGetValue (title, out num_scheme))
return num_scheme;
Dictionary<string, uint> str_scheme = null;
DefaultScheme.StringKeys.TryGetValue (title, out str_scheme);
return str_scheme;
2016-08-04 09:05:44 +08:00
}
2015-11-02 09:51:04 +08:00
}
internal sealed class GyuReader
{
GyuMetaData m_info;
Stream m_input;
byte[] m_output;
int m_width;
int m_height;
public PixelFormat Format { get; private set; }
public BitmapPalette Palette { get; private set; }
public int Stride { get; private set; }
public byte[] Data { get { return m_output; } }
public GyuReader (Stream input, GyuMetaData info)
{
m_info = info;
m_width = (int)info.Width;
m_height = (int)info.Height;
Stride = (m_width * info.BPP / 8 + 3) & ~3;
m_input = input;
if (0 != m_info.AlphaSize)
Format = PixelFormats.Bgra32;
else if (32 == m_info.BPP)
Format = PixelFormats.Bgr32;
else if (24 == m_info.BPP)
Format = PixelFormats.Bgr24;
else if (8 == m_info.BPP)
Format = PixelFormats.Indexed8;
else
throw new NotSupportedException ("Not supported GYU color depth");
if (8 == m_info.BPP && 0 == m_info.PaletteSize)
throw new InvalidFormatException();
}
public void Unpack ()
{
m_input.Position = 0x24;
if (0 != m_info.PaletteSize)
Palette = ImageFormat.ReadPalette (m_input, m_info.PaletteSize);
2015-11-02 09:51:04 +08:00
var packed = new byte[m_info.DataSize];
if (packed.Length != m_input.Read (packed, 0, packed.Length))
throw new EndOfStreamException();
if (m_info.Key != 0xFFFFFFFF)
Deobfuscate (packed, m_info.Key);
if (0x0100 == m_info.CompressionMode)
{
m_output = packed;
}
else
{
m_output = new byte[Stride * m_height];
if (0x0800 == m_info.CompressionMode)
UnpackGyu (packed);
else
UnpackLzss (packed);
}
if (0 != m_info.AlphaSize)
ReadAlpha();
}
void UnpackLzss (byte[] packed)
{
using (var mem = new MemoryStream (packed))
using (var lz = new LzssStream (mem))
if (m_output.Length != lz.Read (m_output, 0, m_output.Length))
throw new EndOfStreamException ();
}
void UnpackGyu (byte[] packed)
{
2015-11-04 08:01:04 +08:00
using (var mem = new MemoryStream (packed, 4, packed.Length-4))
2015-11-02 09:51:04 +08:00
using (var bits = new MsbBitStream (mem))
{
int dst = 0;
m_output[dst++] = (byte)mem.ReadByte();
while (dst < m_output.Length)
{
int b = bits.GetNextBit();
if (-1 == b)
throw new EndOfStreamException();
if (1 == b)
{
m_output[dst++] = (byte)mem.ReadByte();
continue;
}
int count;
int offset;
if (1 == bits.GetNextBit())
{
count = mem.ReadByte() << 8;
count |= mem.ReadByte();
offset = -1 << 13 | count >> 3;
count &= 7;
2015-11-04 08:01:04 +08:00
if (0 != count)
2015-11-02 09:51:04 +08:00
{
++count;
}
else
{
count = mem.ReadByte();
if (0 == count)
break;
}
}
else
{
count = 1 + bits.GetBits (2);
offset = -1 << 8 | mem.ReadByte();
}
Binary.CopyOverlapped (m_output, dst+offset, dst, ++count);
dst += count;
2015-11-02 09:51:04 +08:00
}
}
}
void ReadAlpha ()
{
int alpha_stride = (m_width + 3) & ~3;
Stream alpha_stream;
if (m_info.AlphaSize == alpha_stride * m_height)
alpha_stream = new StreamRegion (m_input, m_input.Position, true);
else
alpha_stream = new LzssStream (m_input, LzssMode.Decompress, true);
using (alpha_stream)
{
int src_stride = Stride;
int new_stride = m_width * 4;
bool extend_alpha = 3 != m_info.Flags;
var alpha_line = new byte[alpha_stride];
var pixels = new byte[new_stride * m_height];
for (int y = 0; y < m_height; ++y)
{
int src = y * src_stride;
int dst = y * new_stride;
if (alpha_line.Length != alpha_stream.Read (alpha_line, 0, alpha_line.Length))
throw new EndOfStreamException();
for (int x = 0; x < m_width; ++x)
{
if (8 == m_info.BPP)
{
var color = Palette.Colors[m_output[src++]];
pixels[dst++] = color.B;
pixels[dst++] = color.G;
pixels[dst++] = color.R;
}
else
{
pixels[dst++] = m_output[src++];
pixels[dst++] = m_output[src++];
pixels[dst++] = m_output[src++];
}
int alpha = alpha_line[x];
if (extend_alpha)
alpha = alpha >= 0x10 ? 0xFF : alpha * 0x10;
pixels[dst++] = (byte)alpha;
}
}
Stride = new_stride;
Palette = null;
m_output = pixels;
}
}
static void Deobfuscate (byte[] data, uint key)
{
var mt = new MersenneTwister (key);
for (int n = 0; n < 10; ++n)
{
uint i1 = mt.Rand() % (uint)data.Length;
uint i2 = mt.Rand() % (uint)data.Length;
var tmp = data[i1];
data[i1] = data[i2];
data[i2] = tmp;
}
}
}
2016-08-04 09:05:44 +08:00
public class GyuOptions : ResourceOptions
{
public IDictionary Scheme;
2016-08-04 09:05:44 +08:00
}
2015-11-02 09:51:04 +08:00
}