mirror of
https://github.com/crskycode/GARbro.git
synced 2026-07-08 01:30:18 +08:00
implemented Will co. archives creation.
This commit is contained in:
@@ -23,8 +23,12 @@
|
||||
// IN THE SOFTWARE.
|
||||
//
|
||||
|
||||
using System;
|
||||
using System.Collections.Generic;
|
||||
using System.ComponentModel.Composition;
|
||||
using System.IO;
|
||||
using GameRes.Formats.Properties;
|
||||
using GameRes.Formats.Strings;
|
||||
|
||||
namespace GameRes.Formats.Will
|
||||
{
|
||||
@@ -35,14 +39,24 @@ namespace GameRes.Formats.Will
|
||||
public uint DirOffset;
|
||||
}
|
||||
|
||||
public class ArcOptions : ResourceOptions
|
||||
{
|
||||
public int NameLength { get; set; }
|
||||
}
|
||||
|
||||
[Export(typeof(ArchiveFormat))]
|
||||
public class ArcOpener : ArchiveFormat
|
||||
{
|
||||
public override string Tag { get { return "ARC"; } }
|
||||
public override string Tag { get { return "WARC"; } }
|
||||
public override string Description { get { return "Will Co. game engine resource archive"; } }
|
||||
public override uint Signature { get { return 0; } }
|
||||
public override bool IsHierarchic { get { return false; } }
|
||||
public override bool CanCreate { get { return false; } }
|
||||
public override bool CanCreate { get { return true; } }
|
||||
|
||||
ArcOpener ()
|
||||
{
|
||||
Extensions = new string[] { "arc" };
|
||||
}
|
||||
|
||||
public override ArcFile TryOpen (ArcView file)
|
||||
{
|
||||
@@ -52,16 +66,14 @@ namespace GameRes.Formats.Will
|
||||
|
||||
uint dir_offset = 4;
|
||||
var ext_list = new List<ExtRecord> (ext_count);
|
||||
int file_count = 0;
|
||||
for (int i = 0; i < ext_count; ++i)
|
||||
{
|
||||
string ext = file.View.ReadString (dir_offset, 4).ToLowerInvariant();
|
||||
int count = file.View.ReadInt32 (dir_offset+4);
|
||||
uint offset = file.View.ReadUInt32 (dir_offset+8);
|
||||
if (count <= 0 || count > 0xffff || offset <= dir_offset)
|
||||
if (count <= 0 || count > 0xffff || offset <= dir_offset || offset > file.MaxOffset)
|
||||
return null;
|
||||
ext_list.Add (new ExtRecord { Extension = ext, FileCount = count, DirOffset = offset });
|
||||
file_count += count;
|
||||
dir_offset += 12;
|
||||
}
|
||||
var dir = ReadFileList (file, ext_list, 9);
|
||||
@@ -77,10 +89,14 @@ namespace GameRes.Formats.Will
|
||||
var dir = new List<Entry>();
|
||||
foreach (var ext in ext_list)
|
||||
{
|
||||
dir.Capacity = dir.Count + ext.FileCount;
|
||||
uint dir_offset = ext.DirOffset;
|
||||
for (int i = 0; i < ext.FileCount; ++i)
|
||||
{
|
||||
string name = file.View.ReadString (dir_offset, name_size).ToLowerInvariant()+'.'+ext.Extension;
|
||||
string name = file.View.ReadString (dir_offset, name_size);
|
||||
if (string.IsNullOrEmpty (name))
|
||||
return null;
|
||||
name = name.ToLowerInvariant()+'.'+ext.Extension;
|
||||
var entry = FormatCatalog.Instance.CreateEntry (name);
|
||||
entry.Size = file.View.ReadUInt32 (dir_offset+name_size);
|
||||
entry.Offset = file.View.ReadUInt32 (dir_offset+name_size+4);
|
||||
@@ -92,5 +108,119 @@ namespace GameRes.Formats.Will
|
||||
}
|
||||
return dir;
|
||||
}
|
||||
|
||||
public override ResourceOptions GetDefaultOptions ()
|
||||
{
|
||||
return new ArcOptions { NameLength = Settings.Default.WARCNameLength };
|
||||
}
|
||||
|
||||
public override object GetCreationWidget ()
|
||||
{
|
||||
return new GUI.CreateWARCWidget();
|
||||
}
|
||||
|
||||
internal class ArcEntry : Entry
|
||||
{
|
||||
public byte[] RawName;
|
||||
}
|
||||
|
||||
internal class ArcDirectory
|
||||
{
|
||||
public byte[] Extension;
|
||||
public uint DirOffset;
|
||||
public List<ArcEntry> Files;
|
||||
}
|
||||
|
||||
public override void Create (Stream output, IEnumerable<Entry> list, ResourceOptions options,
|
||||
EntryCallback callback)
|
||||
{
|
||||
var arc_options = GetOptions<ArcOptions> (options);
|
||||
var encoding = Encodings.cp932.WithFatalFallback();
|
||||
|
||||
int file_count = 0;
|
||||
var file_table = new SortedDictionary<string, ArcDirectory>();
|
||||
foreach (var entry in list)
|
||||
{
|
||||
string ext = Path.GetExtension (entry.Name).TrimStart ('.').ToUpperInvariant();
|
||||
if (string.IsNullOrEmpty (ext))
|
||||
throw new InvalidFileName (entry.Name, arcStrings.MsgNoExtension);
|
||||
if (ext.Length > 3)
|
||||
throw new InvalidFileName (entry.Name, arcStrings.MsgExtensionTooLong);
|
||||
string name = Path.GetFileNameWithoutExtension (entry.Name).ToUpperInvariant();
|
||||
byte[] raw_name = encoding.GetBytes (name);
|
||||
if (raw_name.Length > arc_options.NameLength)
|
||||
throw new InvalidFileName (entry.Name, arcStrings.MsgFileNameTooLong);
|
||||
|
||||
ArcDirectory dir;
|
||||
if (!file_table.TryGetValue (ext, out dir))
|
||||
{
|
||||
byte[] raw_ext = encoding.GetBytes (ext);
|
||||
if (raw_ext.Length > 3)
|
||||
throw new InvalidFileName (entry.Name, arcStrings.MsgExtensionTooLong);
|
||||
dir = new ArcDirectory { Extension = raw_ext, Files = new List<ArcEntry>() };
|
||||
file_table[ext] = dir;
|
||||
}
|
||||
dir.Files.Add (new ArcEntry { Name = entry.Name, RawName = raw_name });
|
||||
++file_count;
|
||||
}
|
||||
if (null != callback)
|
||||
callback (file_count+1, null, null);
|
||||
|
||||
int callback_count = 0;
|
||||
long dir_offset = 4 + file_table.Count * 12;
|
||||
long data_offset = dir_offset + (arc_options.NameLength + 9) * file_count;
|
||||
output.Position = data_offset;
|
||||
foreach (var ext in file_table.Keys)
|
||||
{
|
||||
var dir = file_table[ext];
|
||||
dir.DirOffset = (uint)dir_offset;
|
||||
dir_offset += (arc_options.NameLength + 9) * dir.Files.Count;
|
||||
foreach (var entry in dir.Files)
|
||||
{
|
||||
if (null != callback)
|
||||
callback (callback_count++, entry, arcStrings.MsgAddingFile);
|
||||
entry.Offset = data_offset;
|
||||
using (var input = File.OpenRead (entry.Name))
|
||||
{
|
||||
var size = input.Length;
|
||||
if (size > uint.MaxValue || data_offset + size > uint.MaxValue)
|
||||
throw new FileSizeException();
|
||||
data_offset += size;
|
||||
entry.Size = (uint)size;
|
||||
input.CopyTo (output);
|
||||
}
|
||||
}
|
||||
}
|
||||
if (null != callback)
|
||||
callback (callback_count++, null, arcStrings.MsgWritingIndex);
|
||||
|
||||
output.Position = 0;
|
||||
using (var header = new BinaryWriter (output, encoding, true))
|
||||
{
|
||||
byte[] buffer = new byte[arc_options.NameLength+1];
|
||||
header.Write (file_table.Count);
|
||||
foreach (var ext in file_table)
|
||||
{
|
||||
Array.Copy (ext.Value.Extension, buffer, ext.Value.Extension.Length);
|
||||
for (int i = ext.Value.Extension.Length; i < 4; ++i)
|
||||
buffer[i] = 0;
|
||||
header.Write (buffer, 0, 4);
|
||||
header.Write (ext.Value.Files.Count);
|
||||
header.Write (ext.Value.DirOffset);
|
||||
}
|
||||
foreach (var ext in file_table)
|
||||
{
|
||||
foreach (var entry in ext.Value.Files)
|
||||
{
|
||||
Array.Copy (entry.RawName, buffer, entry.RawName.Length);
|
||||
for (int i = entry.RawName.Length; i < buffer.Length; ++i)
|
||||
buffer[i] = 0;
|
||||
header.Write (buffer);
|
||||
header.Write (entry.Size);
|
||||
header.Write ((uint)entry.Offset);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user