From 09f0c78c5ec7b8d7bf274beb9622f2281836bcee Mon Sep 17 00:00:00 2001 From: morkt Date: Fri, 23 Nov 2018 07:03:36 +0400 Subject: [PATCH] implemented LNK archives PRT images and WAF audio. --- ArcFormats/ArcFormats.csproj | 4 + ArcFormats/Kid/ArcDAT.cs | 251 +++++++++++++++++++++++++++++++++++ ArcFormats/Kid/AudioWAF.cs | 84 ++++++++++++ ArcFormats/Kid/ImagePRT.cs | 186 ++++++++++++++++++++++++++ 4 files changed, 525 insertions(+) create mode 100644 ArcFormats/Kid/ArcDAT.cs create mode 100644 ArcFormats/Kid/AudioWAF.cs create mode 100644 ArcFormats/Kid/ImagePRT.cs diff --git a/ArcFormats/ArcFormats.csproj b/ArcFormats/ArcFormats.csproj index 8f860b3c..bd71c6cf 100644 --- a/ArcFormats/ArcFormats.csproj +++ b/ArcFormats/ArcFormats.csproj @@ -194,6 +194,9 @@ + + + @@ -551,6 +554,7 @@ + diff --git a/ArcFormats/Kid/ArcDAT.cs b/ArcFormats/Kid/ArcDAT.cs new file mode 100644 index 00000000..140e0a7c --- /dev/null +++ b/ArcFormats/Kid/ArcDAT.cs @@ -0,0 +1,251 @@ +//! \file ArcDAT.cs +//! \date 2018 Nov 17 +//! \brief KID resource archive. +// +// Copyright (C) 2018 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.Security.Cryptography; +using GameRes.Utility; + +namespace GameRes.Formats.Kid +{ + [Export(typeof(ArchiveFormat))] + public class LnkOpener : ArchiveFormat + { + public override string Tag { get { return "DAT/LNK"; } } + public override string Description { get { return "KID resource archive"; } } + public override uint Signature { get { return 0x4B4E4C; } } // 'LNK' + public override bool IsHierarchic { get { return false; } } + public override bool CanWrite { get { return false; } } + + LnkOpener () + { + ContainedFormats = new[] { "PRT", "WAF" }; + } + + public override ArcFile TryOpen (ArcView file) + { + int count = file.View.ReadInt32 (4); + if (!IsSaneCount (count)) + return null; + uint index_offset = 0x10; + long data_offset = index_offset + count * 32; + var dir = new List (count); + for (int i = 0; i < count; ++i) + { + uint offset = file.View.ReadUInt32 (index_offset); + uint size = file.View.ReadUInt32 (index_offset+4); + var name = file.View.ReadString (index_offset+8, 0x18); + if (string.IsNullOrWhiteSpace (name)) + return null; + var entry = Create (name); + entry.Offset = data_offset + offset; + entry.Size = size >> 1; + entry.IsPacked = (size & 1) != 0; + if (!entry.CheckPlacement (file.MaxOffset)) + return null; + dir.Add (entry); + index_offset += 0x20; + } + return new ArcFile (file, this, dir); + } + + public override Stream OpenEntry (ArcFile arc, Entry entry) + { + IBinaryStream input = arc.File.CreateStream (entry.Offset, entry.Size, entry.Name); + var pent = entry as PackedEntry; + if (pent != null && pent.IsPacked && input.Signature == 0x646E6C) // 'lnd' + { + using (var lnd = input) + { + lnd.Position = 8; + pent.UnpackedSize = lnd.ReadUInt32(); + lnd.ReadInt32(); + var data = new byte[pent.UnpackedSize]; + UnpackLnd (lnd, data); + input = new BinMemoryStream (data, entry.Name); + } + } + if (input.Signature == 0x535043) // 'CPS' + { + using (input) + return UnpackCps (input); + } + return input.AsStream; + } + + internal static void UnpackLnd (IBinaryStream input, byte[] output) + { + int unpacked_size = output.Length; + int dst = 0; + while (dst < unpacked_size) + { + int ctl = input.ReadByte(); + if (-1 == ctl) + break; + if ((ctl & 0x80) != 0) + { + if ((ctl & 0x40) != 0) + { + int count = (ctl & 0x1F) + 2; + if ((ctl & 0x20) != 0) + count += input.ReadUInt8() << 5; + count = Math.Min (count, unpacked_size - dst); + byte v = input.ReadUInt8(); + for (int i = 0; i < count; ++i) + output[dst++] = v; + } + else + { + int count = ((ctl >> 2) & 0xF) + 2; + int offset = ((ctl & 3) << 8) + input.ReadUInt8() + 1; + count = Math.Min (count, unpacked_size - dst); + Binary.CopyOverlapped (output, dst - offset, dst, count); + dst += count; + } + } + else if ((ctl & 0x40) != 0) + { + int length = Math.Min ((ctl & 0x3F) + 2, unpacked_size - dst); + int count = input.ReadUInt8(); + input.Read (output, dst, length); + dst += length; + count = Math.Min (count * length, unpacked_size - dst); + if (count > 0) + { + Binary.CopyOverlapped (output, dst - length, dst, count); + dst += count; + } + } + else + { + int count = (ctl & 0x1F) + 1; + if ((ctl & 0x20) != 0) + count += input.ReadUInt8() << 5; + count = Math.Min (count, unpacked_size - dst); + input.Read (output, dst, count); + dst += count; + } + } + } + + void UnpackLnd16 (IBinaryStream input, byte[] output) + { + throw new NotImplementedException ("KID Lnd16 compression not implemented."); + } + + Stream UnpackCps (IBinaryStream input) + { + input.Seek (-4, SeekOrigin.End); + uint key_offset = input.ReadUInt32() - 0x7534682; + input.Position = key_offset; + uint key = input.ReadUInt32() + key_offset + 0x3786425; + + var header = input.ReadHeader (0x10); + int packed_size = header.ToInt32 (4); + int compression = header.ToUInt16 (0xA); + int unpacked_size = header.ToInt32 (0xC); + + var decryptor = new CpsTransform (packed_size, (int)key_offset, key); + using (var decoded = new InputCryptoStream (input.AsStream, decryptor)) + using (var cps = new BinaryStream (decoded, input.Name)) + { + var output = new byte[unpacked_size]; + if ((compression & 1) != 0) + { + cps.ReadInt32(); + UnpackLnd (cps, output); + } + else if ((compression & 2) != 0) + { + UnpackLnd16 (cps, output); + } + else + { + cps.ReadInt32(); + cps.Read (output, 0, unpacked_size); + } + return new BinMemoryStream (output); + } + } + } + + internal sealed class CpsTransform : ICryptoTransform + { + const int BlockSize = 4; + + public bool CanReuseTransform { get { return false; } } + public bool CanTransformMultipleBlocks { get { return true; } } + public int InputBlockSize { get { return BlockSize; } } + public int OutputBlockSize { get { return BlockSize; } } + + int m_data_length; + int m_key_offset; + uint m_key; + int m_position; + + public CpsTransform (int data_length, int key_offset, uint key) + { + m_data_length = data_length; + m_key_offset = key_offset; + m_key = key; + m_position = 0x10; + } + + public int TransformBlock (byte[] inputBuffer, int src, int inputCount, + byte[] outputBuffer, int dst) + { + for (int i = 0; i < inputCount; i += BlockSize) + { + if (m_position == m_data_length - 4) + { + LittleEndian.Pack (0, outputBuffer, dst); + break; + } + uint data = LittleEndian.ToUInt32 (inputBuffer, src); + if (m_position != m_key_offset && m_key_offset != 0) + data -= m_key + (uint)m_data_length; + LittleEndian.Pack (data, outputBuffer, dst); + src += 4; + dst += 4; + m_position += 4; + m_key = 1103515245 * m_key + 39686; + } + return inputCount; + } + + public byte[] TransformFinalBlock (byte[] inputBuffer, int inputOffset, int inputCount) + { + byte[] outputBuffer = new byte[inputCount]; + TransformBlock (inputBuffer, inputOffset, inputCount, outputBuffer, 0); + return outputBuffer; + } + + public void Dispose () + { + } + } +} diff --git a/ArcFormats/Kid/AudioWAF.cs b/ArcFormats/Kid/AudioWAF.cs new file mode 100644 index 00000000..35588d2b --- /dev/null +++ b/ArcFormats/Kid/AudioWAF.cs @@ -0,0 +1,84 @@ +//! \file AudioWAF.cs +//! \date 2018 Nov 18 +//! \brief KID ADPCM audio format. +// +// Copyright (C) 2018 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.Linq; +using System.Text; +using GameRes.Utility; + +namespace GameRes.Formats.Kid +{ + [Export(typeof(AudioFormat))] + public class WafAudio : AudioFormat + { + public override string Tag { get { return "WAF"; } } + public override string Description { get { return "KID ADPCM audio file"; } } + public override uint Signature { get { return 0x464157; } } // 'WAF' + public override bool CanWrite { get { return false; } } + + public override SoundInput TryOpen (IBinaryStream file) + { + var header = file.ReadHeader (0x38); + var format = new WaveFormat { + FormatTag = 2, + Channels = header.ToUInt16 (6), + SamplesPerSecond = header.ToUInt32 (8), + AverageBytesPerSecond = header.ToUInt32 (0xC), + BlockAlign = header.ToUInt16 (0x10), + BitsPerSample = header.ToUInt16 (0x12), + ExtraSize = 0x20, + }; + var codec_data = header.Skip (0x14).Take (format.ExtraSize).ToArray(); + int adpcm_length = header.ToInt32 (0x34); + byte[] wav_header; + using (var wav = new MemoryStream()) + using (var buffer = new BinaryWriter (wav, Encoding.ASCII, true)) + { + buffer.Write (Wav.Signature); + buffer.Write (adpcm_length + 0x46); + buffer.Write (0x45564157); // 'WAVE' + buffer.Write (0x20746d66); // 'fmt ' + buffer.Write (0x32); + buffer.Write (format.FormatTag); + buffer.Write (format.Channels); + buffer.Write (format.SamplesPerSecond); + buffer.Write (format.AverageBytesPerSecond); + buffer.Write (format.BlockAlign); + buffer.Write (format.BitsPerSample); + buffer.Write (format.ExtraSize); + buffer.Write (codec_data); + buffer.Write (0x61746164); // 'data' + buffer.Write (adpcm_length); + buffer.Flush(); + wav_header = wav.ToArray(); + } + Stream input = new StreamRegion (file.AsStream, file.Position); + input = new PrefixStream (wav_header, input); + return new WaveInput (input); + } + } +} diff --git a/ArcFormats/Kid/ImagePRT.cs b/ArcFormats/Kid/ImagePRT.cs new file mode 100644 index 00000000..fef00833 --- /dev/null +++ b/ArcFormats/Kid/ImagePRT.cs @@ -0,0 +1,186 @@ +//! \file ImagePRT.cs +//! \date 2018 Nov 18 +//! \brief KID image format. +// +// Copyright (C) 2018 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 System.Windows.Media.Imaging; + +namespace GameRes.Formats.Kid +{ + internal class PrtMetaData : ImageMetaData + { + public int Version; + public ushort PaletteOffset; + public ushort DataOffset; + public bool HasAlpha; + } + + [Export(typeof(ImageFormat))] + public class PrtFormat : ImageFormat + { + public override string Tag { get { return "PRT"; } } + public override string Description { get { return "KID image format"; } } + public override uint Signature { get { return 0x545250; } } // 'PRT' + + public PrtFormat () + { + Extensions = new[] { "prt", "cps" }; + } + + public override ImageMetaData ReadMetaData (IBinaryStream file) + { + var header = file.ReadHeader (0x14); + int version = header.ToUInt16 (4); + if (version != 101 && version != 102) + return null; + var info = new PrtMetaData { + Width = header.ToUInt16 (0xC), + Height = header.ToUInt16 (0xE), + BPP = header.ToUInt16 (6), + Version = version, + PaletteOffset = header.ToUInt16 (8), + DataOffset = header.ToUInt16 (0xA), + HasAlpha = header.ToInt32 (0x10) != 0, + }; + if (102 == version) + { + info.OffsetX = file.ReadInt32(); + info.OffsetY = file.ReadInt32(); + } + return info; + } + + public override ImageData Read (IBinaryStream file, ImageMetaData info) + { + var reader = new PrtReader (file, (PrtMetaData)info); + return reader.GetImage(); + } + + public override void Write (Stream file, ImageData image) + { + throw new System.NotImplementedException ("PrtFormat.Write not implemented"); + } + } + + internal class PrtReader + { + IBinaryStream m_input; + PrtMetaData m_info; + int m_stride; + + PixelFormat Format { get; set; } + BitmapPalette Palette { get; set; } + + public PrtReader (IBinaryStream input, PrtMetaData info) + { + m_input = input; + m_info = info; + m_stride = ((int)info.Width * (info.BPP / 8) + 3) & ~3; + } + + public ImageData GetImage () + { + if (8 == m_info.BPP) + { + m_input.Position = m_info.PaletteOffset; + Palette = ImageFormat.ReadPalette (m_input.AsStream); + Format = PixelFormats.Indexed8; + } + else if (24 == m_info.BPP) + Format = PixelFormats.Bgr24; + else if (32 == m_info.BPP) + Format = PixelFormats.Bgr32; + else + throw new InvalidFormatException(); + + m_input.Position = m_info.DataOffset; + var pixels = m_input.ReadBytes (m_stride * (int)m_info.Height); + if (!m_info.HasAlpha) + return ImageData.CreateFlipped (m_info, Format, Palette, pixels, m_stride); + + var alpha = m_input.ReadBytes ((int)m_info.Width * (int)m_info.Height); + if (8 == m_info.BPP) + pixels = ApplyAlphaIndexed (pixels, alpha); + else + pixels = ApplyAlphaRgb (pixels, alpha); + return ImageData.Create (m_info, PixelFormats.Bgra32, null, pixels); + } + + byte[] ApplyAlphaIndexed (byte[] input, byte[] alpha) + { + int width = (int)m_info.Width; + int dst_stride = width * 4; + var output = new byte[dst_stride * (int)m_info.Height]; + int dst_row = 0; + int scr_row = input.Length - m_stride; + int asrc = 0; + var colors = Palette.Colors; + while (dst_row < output.Length) + { + int dst = dst_row; + for (int x = 0; x < width; ++x) + { + var color = colors[input[scr_row+x]]; + output[dst++] = color.B; + output[dst++] = color.G; + output[dst++] = color.R; + output[dst++] = alpha[asrc++]; + } + dst_row += dst_stride; + scr_row -= m_stride; + } + return output; + } + + byte[] ApplyAlphaRgb (byte[] input, byte[] alpha) + { + int src_pixel_size = m_info.BPP / 8; + int width = (int)m_info.Width; + int dst_stride = width * 4; + var output = new byte[dst_stride * (int)m_info.Height]; + int dst_row = 0; + int src_row = input.Length - m_stride; + int asrc = 0; + while (dst_row < output.Length) + { + int dst = dst_row; + int src = src_row; + for (int x = 0; x < width; ++x) + { + output[dst++] = input[src ]; + output[dst++] = input[src+1]; + output[dst++] = input[src+2]; + output[dst++] = alpha[asrc++]; + src += src_pixel_size; + } + dst_row += dst_stride; + src_row -= m_stride; + } + return output; + } + } +}