GARbro-mirror/ArcFormats/BlackRainbow/ImageBMD.cs
2016-10-16 09:22:53 +04:00

271 lines
9.3 KiB
C#

//! \file ImageBMD.cs
//! \date Wed Mar 25 09:35:06 2015
//! \brief Black Rainbow BMD image format implementation.
//
// Copyright (C) 2015 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.Diagnostics;
using System.IO;
using System.Text;
using System.Windows;
using System.Windows.Media;
using System.Windows.Media.Imaging;
using GameRes.Compression;
using GameRes.Utility;
namespace GameRes.Formats.BlackRainbow
{
internal class BmdMetaData : ImageMetaData
{
public uint PackedSize;
public int Flag;
}
[Export(typeof(ImageFormat))]
public class BmdFormat : ImageFormat
{
public override string Tag { get { return "BMD"; } }
public override string Description { get { return "Black Rainbow bitmap format"; } }
public override uint Signature { get { return 0x444d425fu; } } // '_BMD'
public override bool CanWrite { get { return true; } }
public override ImageMetaData ReadMetaData (IBinaryStream stream)
{
var header = stream.ReadHeader (0x14);
return new BmdMetaData
{
Width = header.ToUInt32 (8),
Height = header.ToUInt32 (12),
BPP = 32,
PackedSize = header.ToUInt32 (4),
Flag = header.ToInt32 (0x10),
};
}
public override ImageData Read (IBinaryStream stream, ImageMetaData info)
{
var meta = (BmdMetaData)info;
stream.Position = 0x14;
int image_size = (int)(meta.Width*meta.Height*4);
using (var reader = new LzssReader (stream.AsStream, (int)meta.PackedSize, image_size))
{
PixelFormat format = meta.Flag != 0 ? PixelFormats.Bgra32 : PixelFormats.Bgr32;
reader.Unpack();
return ImageData.Create (meta, format, null, reader.Data, (int)meta.Width*4);
}
}
public override void Write (Stream file, ImageData image)
{
using (var output = new BinaryWriter (file, Encoding.ASCII, true))
using (var writer = new Writer (image.Bitmap))
{
writer.Pack();
output.Write (Signature);
output.Write (writer.Size);
output.Write (image.Width);
output.Write (image.Height);
output.Write (writer.HasAlpha ? 1 : 0);
output.Write (writer.Data, 0, (int)writer.Size);
}
}
internal class Writer : IDisposable
{
const int MinChunkSize = 3;
const int MaxChunkSize = MinChunkSize+0xf;
const int FrameSize = 0x1000;
byte[] m_input;
MemoryStream m_output;
bool m_has_alpha = false;
byte[] m_frame = new byte[FrameSize];
public byte[] Data { get { return m_output.GetBuffer(); } }
public uint Size { get { return (uint)m_output.Length; } }
public bool HasAlpha { get { return m_has_alpha; } }
public Writer (BitmapSource bitmap)
{
if (bitmap.Format != PixelFormats.Bgra32)
bitmap = new FormatConvertedBitmap (bitmap, PixelFormats.Bgra32, null, 0);
m_input = new byte[bitmap.PixelWidth*bitmap.PixelHeight*4];
bitmap.CopyPixels (m_input, bitmap.PixelWidth*4, 0);
for (int i = 3; i < m_input.Length; i += 4)
{
if (0xff != m_input[i])
{
m_has_alpha = true;
break;
}
}
if (!m_has_alpha)
for (int i = 3; i < m_input.Length; i += 4)
m_input[i] = 0;
m_output = new MemoryStream();
}
public void Pack ()
{
int frame_pos = 0x1000 - 18;
int src = 0;
while (src < m_input.Length)
{
int chunk_size;
int offset = FindChunk (src, out chunk_size);
if (-1 == offset)
{
PutByte (m_input[src]);
chunk_size = 1;
}
else
PutChunk (offset, chunk_size);
for (int i = 0; i < chunk_size; ++i)
{
m_frame[frame_pos++] = m_input[src++];
frame_pos &= 0xfff;
}
}
Flush();
}
struct Chunk
{
public short Offset;
public byte Data;
public Chunk (byte b)
{
Offset = -1;
Data = b;
}
public Chunk (int offset, int count)
{
Debug.Assert (offset < 0x1000 && count >= MinChunkSize && count <= MaxChunkSize);
Offset = (short)offset;
Data = (byte)((count - MinChunkSize) & 0x0f);
}
}
List<Chunk> m_queue = new List<Chunk> (8);
void PutByte (byte b)
{
m_queue.Add (new Chunk (b));
if (8 == m_queue.Count)
Flush();
}
void PutChunk (int offset, int size)
{
m_queue.Add (new Chunk (offset, size));
if (8 == m_queue.Count)
Flush();
}
void Flush ()
{
if (0 == m_queue.Count)
return;
int ctl = 0;
int bit = 1;
for (int i = 0; i < m_queue.Count; ++i)
{
if (m_queue[i].Offset < 0)
ctl |= bit;
bit <<= 1;
}
m_output.WriteByte ((byte)ctl);
for (int i = 0; i < m_queue.Count; ++i)
{
var chunk = m_queue[i];
if (chunk.Offset >= 0)
{
byte lo = (byte)(chunk.Offset & 0xff);
byte hi = (byte)((chunk.Offset & 0xf00) >> 4);
hi |= chunk.Data;
m_output.WriteByte (lo);
m_output.WriteByte (hi);
}
else
m_output.WriteByte (chunk.Data);
}
m_queue.Clear();
}
private int FindChunk (int pos, out int size)
{
size = 0;
int chunk_limit = Math.Min (MaxChunkSize, m_input.Length-pos);
if (chunk_limit < MinChunkSize)
return -1;
int offset = -1;
for (int i = 0; i < m_frame.Length; )
{
int first = Array.IndexOf (m_frame, m_input[pos], i);
if (-1 == first)
break;
int j = 1;
while (j < chunk_limit && m_frame[(first+j)&0xfff] == m_input[pos+j])
++j;
if (j > size && j >= MinChunkSize)
{
offset = first;
size = j;
if (chunk_limit == j)
break;
}
i = first + 1;
}
return offset;
}
#region IDisposable Members
bool disposed = false;
public void Dispose ()
{
Dispose (true);
GC.SuppressFinalize (this);
}
protected virtual void Dispose (bool disposing)
{
if (!disposed)
{
if (disposing)
{
m_output.Dispose();
}
disposed = true;
}
}
#endregion
}
}
}