GARbro-mirror/ArcFormats/Aoi/ImageAGF.cs

200 lines
7.2 KiB
C#
Raw Normal View History

2016-06-25 01:58:47 +08:00
//! \file ImageAGF.cs
//! \date Thu Jun 23 16:23:03 2016
//! \brief Aoi engine image format.
//
// 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.ComponentModel.Composition;
using System.Diagnostics;
using System.IO;
using System.Text;
using System.Windows.Media;
using System.Windows.Media.Imaging;
using GameRes.Utility;
namespace GameRes.Formats.Aoi
{
internal class AgfMetaData : ImageMetaData
{
public int Version;
public uint DataOffset;
public uint Flags;
public uint BaseNameOffset;
}
[Export(typeof(ImageFormat))]
public class AgfFormat : ImageFormat
{
public override string Tag { get { return "AGF/AOI"; } }
public override string Description { get { return "Aoi engine image format"; } }
public override uint Signature { get { return 0x464741; } } // 'AGF'
public AgfFormat ()
{
Extensions = new string[] { "agf" };
}
2016-10-15 16:21:12 +08:00
public override ImageMetaData ReadMetaData (IBinaryStream stream)
2016-06-25 01:58:47 +08:00
{
2016-10-15 16:21:12 +08:00
var header = stream.ReadHeader (0x80);
int version = header.ToInt32 (4);
2016-06-25 01:58:47 +08:00
if (version != 1 && version != 2)
return null;
var info = new AgfMetaData
{
2016-10-15 16:21:12 +08:00
Width = header.ToUInt32 (0x1C),
Height = header.ToUInt32 (0x20),
2016-06-25 01:58:47 +08:00
BPP = 32,
Version = version,
};
if (1 == version)
{
2016-10-15 16:21:12 +08:00
info.DataOffset = header.ToUInt32 (0x0C);
2016-06-25 01:58:47 +08:00
}
else
{
2016-10-15 16:21:12 +08:00
info.DataOffset = header.ToUInt32 (0x10);
info.Flags = header.ToUInt32 (0x54);
info.BaseNameOffset = header.ToUInt32 (0x6C);
2016-06-25 01:58:47 +08:00
}
return info;
}
2016-10-15 16:21:12 +08:00
public override ImageData Read (IBinaryStream input, ImageMetaData info)
2016-06-25 01:58:47 +08:00
{
var meta = (AgfMetaData)info;
var pixels = new byte[meta.Width * meta.Height * 4];
int dst = 0;
2016-10-15 16:21:12 +08:00
input.Position = meta.DataOffset;
while (dst < pixels.Length)
2016-06-25 01:58:47 +08:00
{
2016-10-15 16:21:12 +08:00
uint op = input.ReadUInt32();
int count = (int)(op >> 8);
switch (op & 0xFF)
2016-06-25 01:58:47 +08:00
{
2016-10-15 16:21:12 +08:00
case 1:
count *= 4;
input.Read (pixels, dst, count);
break;
2016-06-25 01:58:47 +08:00
2016-10-15 16:21:12 +08:00
case 2:
input.Read (pixels, dst, 4);
count *= 4;
Binary.CopyOverlapped (pixels, dst, dst+4, count - 4);
break;
2016-06-25 01:58:47 +08:00
2016-10-15 16:21:12 +08:00
case 3:
int chunk_size = (count >> 8) * 4;
count = (count & 0xFF) * chunk_size;
input.Read (pixels, dst, chunk_size);
Binary.CopyOverlapped (pixels, dst, dst + chunk_size, count - chunk_size);
break;
2016-06-25 01:58:47 +08:00
2016-10-15 16:21:12 +08:00
case 4:
int offset = (count & 0xFFF) * 4;
count = (count >> 12) * 4;
Binary.CopyOverlapped (pixels, dst - offset, dst, count);
break;
2016-06-25 01:58:47 +08:00
2016-10-15 16:21:12 +08:00
case 5:
count = (count >> 8) & 0xFF;
2016-10-16 13:22:53 +08:00
input.Seek ((count - count / 4) * 4, SeekOrigin.Current);
2016-10-15 16:21:12 +08:00
count *= 4;
break;
2016-06-25 01:58:47 +08:00
2016-10-15 16:21:12 +08:00
default:
throw new InvalidFormatException();
2016-06-25 01:58:47 +08:00
}
2016-10-15 16:21:12 +08:00
dst += count;
}
if (0 != (meta.Flags & 0x10) && 0 != meta.BaseNameOffset)
{
try
2016-06-25 01:58:47 +08:00
{
2016-10-15 16:21:12 +08:00
var base_name = ReadBaseName (input, meta);
if (VFS.FileExists (base_name))
2016-06-25 01:58:47 +08:00
{
2016-10-16 13:22:53 +08:00
using (var base_file = VFS.OpenBinaryStream (base_name))
2016-06-25 01:58:47 +08:00
{
2016-10-16 13:22:53 +08:00
var base_image = Read (base_file);
2016-10-15 16:21:12 +08:00
BlendImage (meta, pixels, base_image.Bitmap);
2016-06-25 01:58:47 +08:00
}
}
2016-10-15 16:21:12 +08:00
}
catch (Exception X)
{
Trace.WriteLine (string.Format ("{0}: baseline image read error: {1}",
meta.FileName, X.Message), "[AGF]");
2016-06-25 01:58:47 +08:00
}
}
return ImageData.Create (info, PixelFormats.Bgra32, null, pixels);
}
2016-10-15 16:21:12 +08:00
string ReadBaseName (IBinaryStream input, AgfMetaData info)
2016-06-25 01:58:47 +08:00
{
2016-10-16 13:22:53 +08:00
input.Position = info.DataOffset + info.BaseNameOffset;
2016-10-15 16:21:12 +08:00
using (var reader = new BinaryReader (input.AsStream, Encoding.Unicode, true))
2016-06-25 01:58:47 +08:00
{
2016-10-15 16:21:12 +08:00
var name = new StringBuilder();
for (;;)
{
2016-10-16 13:22:53 +08:00
char c = reader.ReadChar();
2016-10-15 16:21:12 +08:00
if (0 == c)
break;
name.Append (c);
}
var dir_name = VFS.GetDirectoryName (info.FileName);
return VFS.CombinePath (dir_name, name.ToString());
2016-06-25 01:58:47 +08:00
}
}
void BlendImage (AgfMetaData info, byte[] overlay, BitmapSource bitmap)
{
if (bitmap.PixelWidth != info.Width || bitmap.PixelHeight != info.Height)
return;
if (bitmap.Format.BitsPerPixel != 32)
bitmap = new FormatConvertedBitmap (bitmap, PixelFormats.Bgra32, null, 0);
int stride = bitmap.PixelWidth * 4;
byte[] pixels = new byte[stride * bitmap.PixelHeight];
bitmap.CopyPixels (pixels, stride, 0);
for (int i = 0; i < pixels.Length; i += 4)
{
int pix = overlay[i] | overlay[i+1] | overlay[i+2] | overlay[i+3];
if (0 == pix)
{
overlay[i ] = pixels[i ];
overlay[i+1] = pixels[i+1];
overlay[i+2] = pixels[i+2];
overlay[i+3] = pixels[i+3];
}
}
}
public override void Write (Stream file, ImageData image)
{
throw new System.NotImplementedException ("AgfFormat.Write not implemented");
}
}
}