GARbro-mirror/ArcFormats/Ankh/ArcGRP.cs

577 lines
21 KiB
C#
Raw Normal View History

2016-03-20 13:42:36 +08:00
//! \file ArcGRP.cs
//! \date Sun Mar 20 02:07:17 2016
//! \brief Ankh resource archive.
//
// Copyright (C) 2016 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 GameRes.Utility;
namespace GameRes.Formats.Ankh
{
[Export(typeof(ArchiveFormat))]
public class GrpOpener : ArchiveFormat
{
public override string Tag { get { return "GRP/ICE"; } }
public override string Description { get { return "Ice Soft resource archive"; } }
2016-03-20 13:42:36 +08:00
public override uint Signature { get { return 0; } }
public override bool IsHierarchic { get { return false; } }
public override bool CanWrite { get { return false; } }
2016-03-20 13:42:36 +08:00
public GrpOpener ()
{
Extensions = new string[] { "grp", "bin" };
2016-03-20 13:42:36 +08:00
}
public override ArcFile TryOpen (ArcView file)
{
uint first_offset = file.View.ReadUInt32 (0);
if (first_offset < 8 || first_offset >= file.MaxOffset || 0 != (first_offset & 3))
2016-03-20 13:42:36 +08:00
return null;
int count = (int)(first_offset - 4) / 4;
2016-03-20 13:42:36 +08:00
if (!IsSaneCount (count))
return null;
var base_name = Path.GetFileNameWithoutExtension (file.Name);
uint index_offset = 0;
uint next_offset = first_offset;
var dir = new List<Entry> (count);
for (int i = 0; i < count && next_offset < file.MaxOffset; ++i)
2016-03-20 13:42:36 +08:00
{
var entry = new PackedEntry { Offset = next_offset };
index_offset += 4;
next_offset = file.View.ReadUInt32 (index_offset);
if (next_offset < entry.Offset)
return null;
entry.Size = (uint)(next_offset - entry.Offset);
entry.UnpackedSize = entry.Size;
if (entry.Size != 0)
{
if (!entry.CheckPlacement (file.MaxOffset))
return null;
entry.Name = string.Format ("{0}#{1:D4}", base_name, i);
dir.Add (entry);
}
}
if (0 == dir.Count)
return null;
DetectFileTypes (file, dir);
return new ArcFile (file, this, dir);
}
void DetectFileTypes (ArcView file, List<Entry> dir)
{
foreach (PackedEntry entry in dir)
2016-03-20 13:42:36 +08:00
{
if (entry.Size <= 8)
2016-03-20 13:42:36 +08:00
continue;
if (file.View.AsciiEqual (entry.Offset, "TPW"))
{
entry.IsPacked = file.View.ReadByte (entry.Offset+3) != 0;
long start_offset = entry.Offset+4;
if (entry.IsPacked)
{
entry.UnpackedSize = file.View.ReadUInt32 (start_offset);
start_offset = entry.Offset+11;
}
else
{
entry.Offset = start_offset;
entry.Size -= 4;
}
if (file.View.AsciiEqual (start_offset, "BM"))
entry.ChangeType (ImageFormat.Bmp);
}
else if (file.View.AsciiEqual (entry.Offset+4, "HDJ\0"))
2016-03-20 13:42:36 +08:00
{
if (file.View.AsciiEqual (entry.Offset+12, "BM"))
entry.ChangeType (ImageFormat.Bmp);
else if (file.View.AsciiEqual (entry.Offset+12, "MThd"))
entry.Name = Path.ChangeExtension (entry.Name, "mid");
entry.UnpackedSize = file.View.ReadUInt32 (entry.Offset);
2016-03-20 13:42:36 +08:00
entry.IsPacked = true;
}
else if (file.View.AsciiEqual (entry.Offset+4, "OggS"))
{
entry.ChangeType (OggAudio.Instance);
entry.Offset += 4;
entry.Size -= 4;
}
2016-03-20 13:42:36 +08:00
else if (entry.Size > 12 && file.View.AsciiEqual (entry.Offset+8, "RIFF"))
{
entry.ChangeType (AudioFormat.Wav);
entry.UnpackedSize = file.View.ReadUInt32 (entry.Offset);
2016-03-20 13:42:36 +08:00
entry.IsPacked = true;
}
else if (file.View.AsciiEqual (entry.Offset, "BM"))
2016-03-20 13:42:36 +08:00
{
entry.ChangeType (ImageFormat.Bmp);
2016-03-20 13:42:36 +08:00
}
else if (file.View.AsciiEqual (entry.Offset, "RIFF"))
{
entry.ChangeType (AudioFormat.Wav);
}
else if (file.View.AsciiEqual (entry.Offset, "\x89PNG"))
{
entry.ChangeType (ImageFormat.Png);
}
else if (entry.Size > 0x16 && IsAudioEntry (file, entry))
{
entry.Type = "audio";
}
2016-03-20 13:42:36 +08:00
}
}
bool IsAudioEntry (ArcView file, Entry entry)
{
uint signature = file.View.ReadUInt32 (entry.Offset);
if (signature != 0x010001 && signature != 0x020001)
return false;
int extra = file.View.ReadUInt16 (entry.Offset+0x10);
if (extra != 0)
return false;
uint size = file.View.ReadUInt32 (entry.Offset+0x12);
return 0x16 + size == entry.Size;
2016-03-20 13:42:36 +08:00
}
public override Stream OpenEntry (ArcFile arc, Entry entry)
{
var pent = entry as PackedEntry;
if (pent != null && pent.IsPacked && pent.Size > 8)
{
try
{
if (arc.File.View.AsciiEqual (entry.Offset, "TPW"))
return OpenTpw (arc, pent);
if (arc.File.View.AsciiEqual (entry.Offset+4, "HDJ\0"))
return OpenImage (arc, pent);
if (entry.Size > 12 && 'W' == arc.File.View.ReadByte (entry.Offset+4)
&& arc.File.View.AsciiEqual (entry.Offset+8, "RIFF"))
return OpenAudio (arc, entry);
}
catch (Exception X)
{
System.Diagnostics.Trace.WriteLine (X.Message, "[GRP]");
}
}
return base.OpenEntry (arc, entry);
2016-03-20 13:42:36 +08:00
}
Stream OpenImage (ArcFile arc, PackedEntry entry)
2016-03-20 13:42:36 +08:00
{
using (var packed = arc.File.CreateStream (entry.Offset+8, entry.Size-8))
using (var reader = new GrpUnpacker (packed))
{
var unpacked = new byte[entry.UnpackedSize];
2016-03-20 13:42:36 +08:00
reader.UnpackHDJ (unpacked, 0);
2016-10-16 13:22:53 +08:00
return new BinMemoryStream (unpacked, entry.Name);
2016-03-20 13:42:36 +08:00
}
}
Stream OpenAudio (ArcFile arc, Entry entry)
{
int unpacked_size = arc.File.View.ReadInt32 (entry.Offset);
byte pack_type = arc.File.View.ReadByte (entry.Offset+5);
byte channels = arc.File.View.ReadByte (entry.Offset+6);
byte header_size = arc.File.View.ReadByte (entry.Offset+7);
if (unpacked_size <= 0 || header_size > unpacked_size
|| !('A' == pack_type || 'S' == pack_type))
return base.OpenEntry (arc, entry);
var unpacked = new byte[unpacked_size];
arc.File.View.Read (entry.Offset+8, unpacked, 0, header_size);
uint packed_size = entry.Size - 8 - header_size;
using (var packed = arc.File.CreateStream (entry.Offset+8+header_size, packed_size))
using (var reader = new GrpUnpacker (packed))
{
if ('A' == pack_type)
reader.UnpackA (unpacked, header_size, channels);
else
reader.UnpackS (unpacked, header_size, channels);
2016-10-16 13:22:53 +08:00
return new BinMemoryStream (unpacked, entry.Name);
2016-03-20 13:42:36 +08:00
}
}
Stream OpenTpw (ArcFile arc, PackedEntry entry)
{
var output = new byte[entry.UnpackedSize];
using (var input = arc.File.CreateStream (entry.Offset, entry.Size))
{
input.Position = 8;
var offsets = new int[4];
offsets[0] = input.ReadUInt16();
offsets[1] = offsets[0] * 2;
offsets[2] = offsets[0] * 3;
offsets[3] = offsets[0] * 4;
int dst = 0;
while (dst < output.Length)
{
byte ctl = input.ReadUInt8();
if (0 == ctl)
break;
int count;
if (ctl < 0x40)
{
input.Read (output, dst, ctl);
dst += ctl;
}
else if (ctl <= 0x6F)
{
if (0x6F == ctl)
count = input.ReadUInt16();
else
count = (ctl + 0xC3) & 0xFF;
byte v = input.ReadUInt8();
while (count --> 0)
output[dst++] = v;
}
else if (ctl <= 0x9F)
{
if (ctl == 0x9F)
count = input.ReadUInt16();
else
count = (ctl + 0x92) & 0xFF;
byte v1 = input.ReadUInt8();
byte v2 = input.ReadUInt8();
while (count --> 0)
{
output[dst++] = v1;
output[dst++] = v2;
}
}
else if (ctl <= 0xBF)
{
if (ctl == 0xBF)
count = input.ReadUInt16();
else
count = ((ctl + 0x62) & 0xFF);
input.Read (output, dst, 3);
if (count > 0)
{
count *= 3;
Binary.CopyOverlapped (output, dst, dst+3, count-3);
dst += count;
}
}
else
{
count = (ctl & 0x3F) + 3;
int offset = input.ReadUInt8();
offset = (offset & 0x3F) - offsets[offset >> 6];
Binary.CopyOverlapped (output, dst+offset, dst, count);
dst += count;
}
}
2016-10-16 13:22:53 +08:00
return new BinMemoryStream (output, entry.Name);
}
}
2016-03-20 13:42:36 +08:00
}
internal sealed class GrpUnpacker : IDisposable
{
IBinaryStream m_input;
2016-03-20 13:42:36 +08:00
uint m_bits;
int m_cached_bits;
public GrpUnpacker (IBinaryStream input)
2016-03-20 13:42:36 +08:00
{
m_input = input;
2016-03-20 13:42:36 +08:00
}
// Different games have slightly different formats using exact same headers,
// so have to invent some flawed format recognition here.
enum GrpVariant { Default, BoD };
static GrpVariant LastUsedMethod = GrpVariant.Default;
static GrpVariant GetOppositeVariant ()
{
return GrpVariant.Default == LastUsedMethod ? GrpVariant.BoD : GrpVariant.Default;
}
2016-03-20 13:42:36 +08:00
public void UnpackHDJ (byte[] output, int dst)
{
try
{
if (UnpackHDJVariant (output, dst, LastUsedMethod))
return;
}
catch { /* ignore unpack errors */ }
var method = GetOppositeVariant();
m_input.Position = 0;
if (!UnpackHDJVariant (output, dst, method))
throw new InvalidFormatException();
LastUsedMethod = method;
}
private bool UnpackHDJVariant (byte[] output, int dst, GrpVariant method)
2016-03-20 13:42:36 +08:00
{
ResetBits();
int word_count = 0;
int byte_count = 0;
uint next_byte = 0;
uint next_word = 0;
while (dst < output.Length)
{
if (GetNextBit() != 0)
{
int count = 0;
bool long_count = false;
int offset;
if (GetNextBit() != 0)
{
if (0 == word_count)
{
next_word = m_input.ReadUInt32();
word_count = 2;
}
count = (int)((next_word >> 13) & 7) + 3;
offset = (int)(next_word | 0xFFFFE000);
next_word >>= 16;
--word_count;
long_count = 10 == count;
}
else
{
if (method == GrpVariant.Default)
count = GetBits (2);
2016-03-20 13:42:36 +08:00
if (0 == byte_count)
{
next_byte = m_input.ReadUInt32();
byte_count = 4;
}
if (method != GrpVariant.Default)
count = GetBits (2);
count += 2;
long_count = 5 == count;
2016-03-20 13:42:36 +08:00
offset = (int)(next_byte | 0xFFFFFF00);
next_byte >>= 8;
--byte_count;
}
if (long_count)
{
int n = 0;
while (GetNextBit() != 0)
++n;
if (n != 0)
count += GetBits (n) + 1;
}
int src = dst + offset;
if (src < 0 || src >= dst || dst + count > output.Length)
return false;
Binary.CopyOverlapped (output, src, dst, count);
2016-03-20 13:42:36 +08:00
dst += count;
}
else
{
if (0 == byte_count)
{
next_byte = m_input.ReadUInt32();
byte_count = 4;
}
output[dst++] = (byte)next_byte;
next_byte >>= 8;
--byte_count;
}
}
return true;
2016-03-20 13:42:36 +08:00
}
public void UnpackS (byte[] output, int dst, int channels)
{
try
{
if (UnpackSVariant (output, dst, channels, LastUsedMethod))
return;
}
catch { /* ignore parse errors */ }
var method = GetOppositeVariant();
m_input.Position = 0;
if (UnpackSVariant (output, dst, channels, method))
LastUsedMethod = method;
}
bool UnpackSVariant (byte[] output, int dst, int channels, GrpVariant method)
{
if (GrpVariant.Default == method)
UnpackSv2 (output, dst, channels);
else
UnpackSv1 (output, dst, channels);
return m_input.PeekByte() == -1; // rather loose test, but whatever
}
void UnpackSv1 (byte[] output, int dst, int channels)
{
ResetBits();
short last_word = 0;
while (dst < output.Length)
{
int word;
if (GetNextBit() != 0)
{
if (GetNextBit() != 0)
word = GetBits (10) << 6;
else
word = 0;
}
else
{
int adjust = GetBits (5) << 6;
if (0 != (adjust & 0x400))
adjust = -(adjust & 0x3FF);
word = last_word + adjust;
}
last_word = (short)word;
LittleEndian.Pack (last_word, output, dst);
dst += 2;
}
}
void UnpackSv2 (byte[] output, int dst, int channels)
2016-03-20 13:42:36 +08:00
{
if (channels != 1)
m_input.Seek ((channels-1) * 4, SeekOrigin.Current);
2016-03-20 13:42:36 +08:00
int step = channels * 2;
for (int i = 0; i < channels; ++i)
{
ResetBits();
int pos = dst;
short last_word = 0;
while (pos < output.Length)
{
int word;
if (GetNextBit() != 0)
{
if (GetNextBit() != 0)
{
word = GetBits (10) << 6;
}
else
{
int repeat;
if (GetNextBit() != 0)
{
int bit_length = 0;
do
{
++bit_length;
}
while (GetNextBit() != 0);
repeat = GetBits (bit_length) + 4;
}
else
{
repeat = GetBits (2);
}
word = 0;
while (repeat --> 0)
{
output[pos] = 0;
output[pos+1] = 0;
pos += step;
}
}
}
else
{
int adjust = (short)(GetBits (5) << 11) >> 5;
word = last_word + adjust;
}
LittleEndian.Pack ((short)word, output, pos);
last_word = (short)word;
pos += step;
}
dst += 2;
}
}
public void UnpackA (byte[] output, int dst, int channels)
{
if (channels != 1)
m_input.Seek ((channels-1) * 4, SeekOrigin.Current);
2016-03-20 13:42:36 +08:00
int step = 2 * channels;
for (int i = 0; i < channels; ++i)
{
int pos = dst;
ResetBits();
while (pos < output.Length)
{
int word = GetBits (10) << 6;;
LittleEndian.Pack ((short)word, output, pos);
pos += step;
}
dst += 2;
}
}
void ResetBits ()
{
m_cached_bits = 0;
}
int GetNextBit ()
{
return GetBits (1);
}
int GetBits (int count)
{
if (0 == m_cached_bits)
{
m_bits = m_input.ReadUInt32();
m_cached_bits = 32;
}
uint val;
if (m_cached_bits < count)
{
uint next_bits = m_input.ReadUInt32();
val = (m_bits | (next_bits >> m_cached_bits)) >> (32 - count);
m_bits = next_bits << (count - m_cached_bits);
m_cached_bits = 32 - (count - m_cached_bits);
}
else
{
val = m_bits >> (32 - count);
m_bits <<= count;
m_cached_bits -= count;
}
return (int)val;
}
#region IDisposable Members
bool _disposed = false;
public void Dispose ()
{
if (!_disposed)
{
m_input.Dispose();
_disposed = true;
}
}
#endregion
}
}