mirror of
https://github.com/crskycode/GARbro.git
synced 2024-11-23 21:55:34 +08:00
331 lines
13 KiB
C#
331 lines
13 KiB
C#
//! \file ArcDAT.cs
|
|
//! \date 2018 Aug 30
|
|
//! \brief Unity GX4Lib 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.Linq;
|
|
using System.Windows.Media;
|
|
using GameRes.Gx4Lib;
|
|
using GameRes.Utility;
|
|
|
|
namespace GameRes.Formats.Unity.Gx4Lib
|
|
{
|
|
internal class VisualDiffData
|
|
{
|
|
public string BaseFileName;
|
|
public int PosX;
|
|
public int PosY;
|
|
}
|
|
|
|
[Export(typeof(ArchiveFormat))]
|
|
public class DatOpener : ArchiveFormat
|
|
{
|
|
public override string Tag { get { return "DAT/GX4LIB"; } }
|
|
public override string Description { get { return "Unity GX4Lib resource archive"; } }
|
|
public override uint Signature { get { return 0; } }
|
|
public override bool IsHierarchic { get { return false; } }
|
|
public override bool CanWrite { get { return false; } }
|
|
|
|
Lazy<Dictionary<string, VisualDiffData>> DefaultVisualMap = new Lazy<Dictionary<string, VisualDiffData>> (ReadVisualData);
|
|
|
|
public override ArcFile TryOpen (ArcView file)
|
|
{
|
|
uint index_size = file.View.ReadUInt32 (0);
|
|
if (index_size <= 12 || index_size >= file.MaxOffset)
|
|
return null;
|
|
if (file.View.ReadInt32 (5) != 1 || file.View.ReadInt32 (9) != -1)
|
|
return null;
|
|
using (var hstream = file.CreateStream (4, index_size))
|
|
{
|
|
var pf = new PackageFile();
|
|
var index = pf.Deserialize (hstream);
|
|
if (null == index)
|
|
return null;
|
|
string type = "";
|
|
if (index is PFAudioHeaders)
|
|
type = "audio";
|
|
else if (index is PFImageHeaders)
|
|
type = "image";
|
|
uint data_offset = index_size + 4;
|
|
var dir = index.headers.Select (h => new PackedEntry {
|
|
Name = h.FileName,
|
|
Type = type,
|
|
Offset = h.readStartBytePos + data_offset,
|
|
Size = (uint)h.ByteLength
|
|
} as Entry).ToList();
|
|
return new ArcFile (file, this, dir);
|
|
}
|
|
}
|
|
|
|
public override Stream OpenEntry (ArcFile arc, Entry entry)
|
|
{
|
|
if (arc.File.View.AsciiEqual (entry.Offset, "UnityRaw"))
|
|
return OpenUnityRaw (arc, entry);
|
|
var pent = (PackedEntry)entry;
|
|
if (!pent.IsPacked)
|
|
{
|
|
byte id = arc.File.View.ReadByte (entry.Offset);
|
|
uint packed_size = arc.File.View.ReadUInt32 (entry.Offset+1);
|
|
if ((id & ~1) != 0x5E || packed_size != entry.Size)
|
|
return base.OpenEntry (arc, entry);
|
|
pent.IsPacked = 0 != (id & 1);
|
|
pent.UnpackedSize = arc.File.View.ReadUInt32 (entry.Offset+5);
|
|
}
|
|
if (!pent.IsPacked)
|
|
return arc.File.CreateStream (pent.Offset+9, pent.UnpackedSize);
|
|
var data = new byte[pent.UnpackedSize];
|
|
using (var input = arc.File.CreateStream (pent.Offset+9, pent.Size))
|
|
QlzUnpack (input, data);
|
|
return new BinMemoryStream (data, entry.Name);
|
|
}
|
|
|
|
public override IImageDecoder OpenImage (ArcFile arc, Entry entry)
|
|
{
|
|
var input = arc.OpenBinaryEntry (entry);
|
|
int length = (int)(input.Length - 0x10);
|
|
if ((length & 3) != 0)
|
|
return ImageFormatDecoder.Create (input);
|
|
uint width = input.ReadUInt32();
|
|
uint height = input.ReadUInt32();
|
|
int buf_width = input.ReadInt32();
|
|
int buf_height = input.ReadInt32();
|
|
var info = new ImageMetaData { Width = width, Height = height, BPP = 32 };
|
|
if (DefaultVisualMap.Value.ContainsKey (entry.Name))
|
|
{
|
|
var diff_info = DefaultVisualMap.Value[entry.Name];
|
|
if (VFS.FileExists (diff_info.BaseFileName))
|
|
{
|
|
var base_entry = VFS.FindFile (diff_info.BaseFileName);
|
|
using (var visbase = VFS.OpenImage (base_entry))
|
|
{
|
|
var base_decoder = visbase as Gx4ImageDecoder;
|
|
if (base_decoder != null)
|
|
{
|
|
info.OffsetX = diff_info.PosX;
|
|
info.OffsetY = diff_info.PosY;
|
|
var pixels = base_decoder.ReadPixels();
|
|
return new Gx4OverlayDecoder (pixels, base_decoder.Info, input, info);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
return new Gx4ImageDecoder (input, info);
|
|
}
|
|
|
|
Stream OpenUnityRaw (ArcFile arc, Entry entry)
|
|
{
|
|
using (var stream = arc.File.CreateStream (entry.Offset, entry.Size))
|
|
using (var input = new AssetReader (stream))
|
|
{
|
|
var signature = input.ReadCString();
|
|
if (signature != "UnityRaw")
|
|
return base.OpenEntry (arc, entry);
|
|
int format = input.ReadInt32();
|
|
input.ReadCString();
|
|
input.ReadCString();
|
|
uint file_size = input.ReadUInt32();
|
|
if (file_size != entry.Size)
|
|
return base.OpenEntry (arc, entry);
|
|
uint header_size = input.ReadUInt32();
|
|
int entry_count = input.ReadInt32();
|
|
int bundle_count = input.ReadInt32();
|
|
if (entry_count != 1 || bundle_count != 1)
|
|
return base.OpenEntry (arc, entry);
|
|
|
|
input.Position = header_size;
|
|
int count = input.ReadInt32();
|
|
long asset_pos = input.Position;
|
|
input.ReadCString(); // asset_name
|
|
header_size = input.ReadUInt32();
|
|
uint asset_size = input.ReadUInt32();
|
|
long base_pos = asset_pos + header_size - 4;
|
|
|
|
input.Position = base_pos;
|
|
var index = new ResourcesAssetsDeserializer (arc.File.Name);
|
|
var dir = index.Parse (input, base_pos);
|
|
if (null == dir || 0 == dir.Count)
|
|
return base.OpenEntry (arc, entry);;
|
|
return arc.File.CreateStream (entry.Offset + dir[0].Offset, dir[0].Size);
|
|
}
|
|
}
|
|
|
|
void QlzUnpack (IBinaryStream input, byte[] output)
|
|
{
|
|
int dst = 0;
|
|
uint bits = 1;
|
|
int output_last = output.Length - 11;
|
|
while (dst < output.Length)
|
|
{
|
|
if (1 == bits)
|
|
{
|
|
bits = input.ReadUInt32();
|
|
}
|
|
if ((bits & 1) == 1)
|
|
{
|
|
int ctl = input.PeekByte();
|
|
int offset, count = 3;
|
|
if ((ctl & 3) == 0)
|
|
{
|
|
offset = input.ReadUInt8() >> 2;
|
|
}
|
|
else if ((ctl & 2) == 0)
|
|
{
|
|
offset = input.ReadUInt16() >> 2;
|
|
}
|
|
else if ((ctl & 1) == 0)
|
|
{
|
|
offset = input.ReadUInt16() >> 6;
|
|
count += ((ctl >> 2) & 0xF);
|
|
}
|
|
else if ((ctl & 0x7F) != 3)
|
|
{
|
|
offset = (input.ReadInt24() >> 7) & 0x1FFFF;
|
|
count += ((ctl >> 2) & 0x1F) - 1;
|
|
}
|
|
else
|
|
{
|
|
uint v = input.ReadUInt32();
|
|
offset = (int)(v >> 15);
|
|
count += (int)((v >> 7) & 0xFF);
|
|
}
|
|
Binary.CopyOverlapped (output, dst-offset, dst, count);
|
|
dst += count;
|
|
}
|
|
else
|
|
{
|
|
if (dst > output_last)
|
|
break;
|
|
output[dst++] = input.ReadUInt8();
|
|
}
|
|
bits >>= 1;
|
|
}
|
|
while (dst < output.Length)
|
|
{
|
|
if (1 == bits)
|
|
{
|
|
input.Seek (4, SeekOrigin.Current);
|
|
bits = 0x80000000u;
|
|
}
|
|
output[dst++] = input.ReadUInt8();
|
|
bits >>= 1;
|
|
}
|
|
}
|
|
|
|
internal static Dictionary<string, VisualDiffData> ReadVisualData ()
|
|
{
|
|
var map = new Dictionary<string, VisualDiffData>();
|
|
try
|
|
{
|
|
FormatCatalog.Instance.ReadFileList ("gx4_bsz_visual.lst", (line) => {
|
|
var parts = line.Split (':', ',');
|
|
if (parts.Length > 3)
|
|
{
|
|
var diff = new VisualDiffData { BaseFileName = parts[1] };
|
|
if (int.TryParse (parts[2], out diff.PosX) && int.TryParse (parts[3], out diff.PosY))
|
|
map[parts[0]] = diff;
|
|
}
|
|
});
|
|
}
|
|
catch { /* ignore errors */ }
|
|
return map;
|
|
}
|
|
}
|
|
|
|
internal class Gx4ImageDecoder : BinaryImageDecoder
|
|
{
|
|
public int Stride { get; private set; }
|
|
|
|
public Gx4ImageDecoder (IBinaryStream input, ImageMetaData info) : base (input, info)
|
|
{
|
|
Stride = (int)info.Width * 4;
|
|
}
|
|
|
|
public byte[] ReadPixels ()
|
|
{
|
|
m_input.Position = 0x10;
|
|
var pixels = m_input.ReadBytes (Stride * (int)Info.Height);
|
|
for (int i = 0; i < pixels.Length; i += 4)
|
|
{
|
|
byte t = pixels[i];
|
|
pixels[i] = pixels[i+2];
|
|
pixels[i+2] = t;
|
|
}
|
|
return pixels;
|
|
}
|
|
|
|
protected override ImageData GetImageData ()
|
|
{
|
|
var pixels = ReadPixels();
|
|
return ImageData.CreateFlipped (Info, PixelFormats.Bgra32, null, pixels, Stride);
|
|
}
|
|
}
|
|
|
|
internal class Gx4OverlayDecoder : BinaryImageDecoder
|
|
{
|
|
byte[] m_pixels;
|
|
|
|
ImageMetaData OverlayInfo { get; set; }
|
|
|
|
public Gx4OverlayDecoder (byte[] pixels, ImageMetaData info, IBinaryStream input, ImageMetaData overlay_info)
|
|
: base (input, info)
|
|
{
|
|
m_pixels = pixels;
|
|
OverlayInfo = overlay_info;
|
|
}
|
|
|
|
protected override ImageData GetImageData ()
|
|
{
|
|
m_input.Position = 0x10;
|
|
int base_stride = (int)Info.Width * 4;
|
|
int diff_stride = (int)OverlayInfo.Width * 4;
|
|
var diff = m_input.ReadBytes (diff_stride * (int)OverlayInfo.Height);
|
|
int dst_line = m_pixels.Length - base_stride * (OverlayInfo.OffsetY + 1) + OverlayInfo.OffsetX * 4;
|
|
int src_line = diff.Length - diff_stride;
|
|
for (int y = 0; y < (int)OverlayInfo.Height; ++y)
|
|
{
|
|
int dst = dst_line;
|
|
int src = src_line;
|
|
for (int x = 0; x < (int)OverlayInfo.Width; ++x)
|
|
{
|
|
if (!(0 == diff[src] && 0 == diff[src+1] && 0 == diff[src+2] && 0xFF == diff[src+3]))
|
|
{
|
|
m_pixels[dst ] = diff[src+2];
|
|
m_pixels[dst+1] = diff[src+1];
|
|
m_pixels[dst+2] = diff[src ];
|
|
m_pixels[dst+3] = diff[src+3];
|
|
}
|
|
dst += 4;
|
|
src += 4;
|
|
}
|
|
dst_line -= base_stride;
|
|
src_line -= diff_stride;
|
|
}
|
|
return ImageData.CreateFlipped (Info, PixelFormats.Bgra32, null, m_pixels, base_stride);
|
|
}
|
|
}
|
|
}
|