public override void DoEncoded(IStreamEncoder encoder, Action action)
{
// Encode the data into a stream
Stream encoded = null;
using (MemoryStream memStream = new MemoryStream()) {
// Stream key
string key = CurrentPointer.ToString() + "_decoded";
// Add the stream
StreamFile sf = new StreamFile(key, memStream, Context)
{
Endianness = currentFile.Endianness
};
Context.AddFile(sf);
DoAt(sf.StartPointer, () => {
action();
memStream.Position = 0;
encoded = encoder.EncodeStream(memStream);
});
Context.RemoveFile(sf);
}
// Turn stream into array & write bytes
if (encoded != null)
{
using (MemoryStream ms = new MemoryStream()) {
encoded.CopyTo(ms);
writer.Write(ms.ToArray());
}
encoded.Close();
}
}
private void button1_Click(object sender, EventArgs e)
{
_offset = DateTimeOffset.UtcNow;
Task.Run(() =>
{
_service = new DesktopDuplicationCaptureService();
var screenInfo = _service.Initialize(new ComponentOptions("desktopduplication"));
_encoder = new TestEncoder(screenInfo)
{
SetImage = ImportData
};
while (true)
{
_service.Capture(_encoder);
_fps++;
var diff = (DateTimeOffset.UtcNow - _offset);
if (diff.TotalSeconds >= 1)
{
var fps = _fps;
var data = _data;
_data = 0;
_fps = 0;
_offset = DateTimeOffset.UtcNow;
BeginInvoke(new Action(() => { this.Text = $"{fps} FPS ({data / 1024} KiB/s)"; }));
}
}
});
}
public Utility_Decoders_TypeViewModel(LocalizedString name, IStreamEncoder encoder, Func <string> getFileFilter, Games?game = null)
{
Name = name;
Encoder = encoder;
GetFileFilter = getFileFilter;
Game = game;
}
public void Capture(IStreamEncoder encoder)
{
if (!RetrieveFrame())
{
return;
}
try
{
var mapSource = _captureDevice.ImmediateContext.MapSubresource(_desktopImageTexture, 0, MapMode.Read, MapFlags.None);
try
{
// Get the desktop capture texture
encoder.CodeImage(mapSource.DataPointer, PixelFormat.Format32bppArgb);
}
finally
{
_captureDevice.ImmediateContext.UnmapSubresource(_desktopImageTexture, 0);
}
}
finally
{
ReleaseFrame();
}
}
public override void Initialize()
{
base.Initialize();
var captureCommandLine = MazeContext.Request.Headers["capture"];
var encoderCommandLine = MazeContext.Request.Headers[HeaderNames.AcceptEncoding];
var captureOptions = ComponentOptions.Parse(captureCommandLine);
var encoderOptions = ComponentOptions.Parse(encoderCommandLine);
_captureService = ResolveService(captureOptions, _captureServices);
_streamEncoder = ResolveService(encoderOptions, _streamEncoders);
var screenInfo = _captureService.Initialize(captureOptions);
_streamEncoder.Initialize(screenInfo, this, encoderOptions);
}
public static void ExportAllCompressedData(Context context, Pointer offset, IStreamEncoder encoder, byte[] header, string outputDir, int alignment = 4, int minDecompSize = 33)
{
var s = context.Deserializer;
s.Goto(offset);
// Keep track of blocks
var blocks = new List <Tuple <long, long, int> >();
// Enumerate every byte
for (int i = 0; i < s.CurrentLength; i += alignment)
{
// Go to the offset
s.Goto(offset + i);
// Check for compression header
if (s.SerializeArray <byte>(default, header.Length).SequenceEqual(header))
public void Capture(IStreamEncoder encoder)
{
using (var graphics = Graphics.FromImage(_currentImage))
{
var deviceContext = graphics.GetHdc();
NativeMethods.BitBlt(deviceContext, 0, 0, _screenBounds.Width, _screenBounds.Height, _screenDeviceContext, _screenBounds.X,
_screenBounds.Y, SRCCOPY);
graphics.ReleaseHdc(deviceContext);
var bitmapData = _currentImage.LockBits(new Rectangle(0, 0, _currentImage.Width, _currentImage.Height), ImageLockMode.ReadOnly,
_currentImage.PixelFormat);
try
{
encoder.CodeImage(bitmapData.Scan0, bitmapData.PixelFormat);
}
finally
{
_currentImage.UnlockBits(bitmapData);
}
}
}
/// <summary>
/// Processes a file
/// </summary>
/// <param name="inputFiles">The input files to process</param>
/// <param name="outputDir">The output directory</param>
/// <param name="shouldDecode">True if the files should be decoded, or false to encode them</param>
private void ProcessFile(IEnumerable <FileSystemPath> inputFiles, FileSystemPath outputDir, bool shouldDecode)
{
if (IsLoading)
{
return;
}
try
{
IsLoading = true;
// Get the encoder
IStreamEncoder encoder = SelectedType.Encoder;
// Process every file
foreach (FileSystemPath file in inputFiles)
{
// Open the input file
using FileStream inputStream = File.OpenRead(file);
// Open and create the destination file
using FileStream outputStream = File.OpenWrite((outputDir + file.Name).GetNonExistingFileName());
// Process the file data
if (shouldDecode)
{
encoder.DecodeStream(inputStream, outputStream);
}
else
{
encoder.EncodeStream(inputStream, outputStream);
}
}
}
finally
{
IsLoading = false;
}
}
public override void DoEncoded(IStreamEncoder encoder, Action action)
{
}
public override void SerializeBlock(SerializerObject s)
{
if (s.GameSettings.EngineVersion == EngineVersion.GBA_BatmanVengeance)
{
if (StructType != Type.Collision)
{
Unk_02 = s.Serialize <byte>(Unk_02, name: nameof(Unk_02));
Unk_03 = s.Serialize <byte>(Unk_03, name: nameof(Unk_03));
LayerID = s.Serialize <byte>(LayerID, name: nameof(LayerID));
ClusterIndex = s.Serialize <byte>(ClusterIndex, name: nameof(ClusterIndex));
// TODO: figure out what this is. One of these
UnkBytes = s.SerializeArray <byte>(UnkBytes, 5, name: nameof(UnkBytes));
ShouldSetBGAlphaBlending = s.Serialize <bool>(ShouldSetBGAlphaBlending, name: nameof(ShouldSetBGAlphaBlending));
Unk_0E = s.Serialize <byte>(Unk_0E, name: nameof(Unk_0E));
ColorMode = s.Serialize <GBA_ColorMode>(ColorMode, name: nameof(ColorMode));
}
}
else if (s.GameSettings.GBA_IsMilan)
{
ColorMode = GBA_ColorMode.Color4bpp;
// Serialize cluster
Cluster = s.DoAt(ShanghaiOffsetTable.GetPointer(0), () => s.SerializeObject <GBA_Cluster>(Cluster, name: nameof(Cluster)));
IStreamEncoder encoder = null;
switch (Cluster.Milan_MapCompressionType)
{
case GBA_Cluster.Milan_CompressionType.Collision_RL:
case GBA_Cluster.Milan_CompressionType.Map_RL:
encoder = new GBA_RLEEncoder();
break;
case GBA_Cluster.Milan_CompressionType.Collsiion_LZSS:
case GBA_Cluster.Milan_CompressionType.Map_LZSS:
encoder = new GBA_LZSSEncoder();
break;
}
// Go to the map data
s.Goto(ShanghaiOffsetTable.GetPointer(2));
if (StructType == Type.Layer2D)
{
s.DoEncodedIf(encoder, encoder != null, () => Shanghai_MapIndices_16 = s.SerializeArray <ushort>(Shanghai_MapIndices_16, ((Width + Width % 2) * (Height + Height % 2)) / 4, name: nameof(Shanghai_MapIndices_16)));
var end = s.CurrentPointer;
// Go to the map tiles
s.Goto(ShanghaiOffsetTable.GetPointer(1));
s.DoEncodedIf(encoder, encoder != null, () => Shanghai_MapTiles = s.SerializeObjectArray <MapTile>(Shanghai_MapTiles, (Shanghai_MapIndices_16.Max(x => BitHelpers.ExtractBits(x, 12, 0)) + 1) * 4, name: nameof(Shanghai_MapTiles)));
s.Goto(end);
}
else // Collision
{
s.DoEncodedIf(encoder, encoder != null, () => SerializeTileMap(s));
}
s.Align();
// Return to avoid serializing the map tile array normally
return;
}
else if (s.GameSettings.GBA_IsShanghai)
{
ColorMode = GBA_ColorMode.Color4bpp;
IsCompressed = false;
if (StructType == Type.Layer2D)
{
// Serialize cluster
Cluster = s.DoAt(ShanghaiOffsetTable.GetPointer(0), () => s.SerializeObject <GBA_Cluster>(Cluster, name: nameof(Cluster)));
// Go to the map data
s.Goto(ShanghaiOffsetTable.GetPointer(2));
// If the map tile size is not 0 the map is split into a 32x8 index array and 8x8 map tiles
if (Cluster.Shanghai_MapTileSize != 0)
{
var indexArrayLength = Mathf.CeilToInt(Width / 4f) * Height;
// The tile size is either 1 or 2 bytes
if (Cluster.Shanghai_MapTileSize == 1)
{
Shanghai_MapIndices_8 = s.SerializeArray <byte>(Shanghai_MapIndices_8, indexArrayLength, name: nameof(Shanghai_MapIndices_8));
}
else
{
Shanghai_MapIndices_16 = s.SerializeArray <ushort>(Shanghai_MapIndices_16, indexArrayLength, name: nameof(Shanghai_MapIndices_16));
}
var indexArray = Cluster.Shanghai_MapTileSize == 1 ? Shanghai_MapIndices_8.Select(x => (ushort)x).ToArray() : Shanghai_MapIndices_16;
Shanghai_MapTiles = s.SerializeObjectArray <MapTile>(Shanghai_MapTiles, (indexArray.Max() + 1) * 4, name: nameof(Shanghai_MapTiles));
// Return to avoid serializing the map tile array normally
return;
}
}
else
{
// Go to the collision data
s.Goto(ShanghaiOffsetTable.GetPointer(0));
// Serialize header properties
Shanghai_CollisionValue1 = s.Serialize <ushort>(Shanghai_CollisionValue1, name: nameof(Shanghai_CollisionValue1));
Width = s.Serialize <ushort>(Width, name: nameof(Width));
Height = s.Serialize <ushort>(Height, name: nameof(Height));
Shanghai_CollisionValue2 = s.Serialize <ushort>(Shanghai_CollisionValue2, name: nameof(Shanghai_CollisionValue2));
}
}
else
{
StructType = s.Serialize <Type>(StructType, name: nameof(StructType));
if (StructType != Type.TextLayerMode7)
{
IsCompressed = s.Serialize <bool>(IsCompressed, name: nameof(IsCompressed));
}
else
{
Unk_01 = s.Serialize <bool>(Unk_01, name: nameof(Unk_01));
}
Unk_02 = s.Serialize <byte>(Unk_02, name: nameof(Unk_02));
Unk_03 = s.Serialize <byte>(Unk_03, name: nameof(Unk_03));
Width = s.Serialize <ushort>(Width, name: nameof(Width));
Height = s.Serialize <ushort>(Height, name: nameof(Height));
switch (StructType)
{
case Type.TextLayerMode7:
LayerID = s.Serialize <byte>(LayerID, name: nameof(LayerID));
ShouldSetBGAlphaBlending = s.Serialize <bool>(ShouldSetBGAlphaBlending, name: nameof(ShouldSetBGAlphaBlending));
AlphaBlending_Coeff = s.Serialize <sbyte>(AlphaBlending_Coeff, name: nameof(AlphaBlending_Coeff));
UnkBytes = s.SerializeArray <byte>(UnkBytes, 0x14, name: nameof(UnkBytes));
ColorMode = s.Serialize <GBA_ColorMode>(ColorMode, name: nameof(ColorMode));
// 21 bytes
// Prio is 0x1D
// ColorMode is 0x1F
// Width & height seems duplicates again (is it actually width and height?)
break;
case Type.RotscaleLayerMode7:
LayerID = s.Serialize <byte>(LayerID, name: nameof(LayerID));
ShouldSetBGAlphaBlending = s.Serialize <bool>(ShouldSetBGAlphaBlending, name: nameof(ShouldSetBGAlphaBlending));
AlphaBlending_Coeff = s.Serialize <sbyte>(AlphaBlending_Coeff, name: nameof(AlphaBlending_Coeff));
// The game hard-codes the color mode
ColorMode = GBA_ColorMode.Color8bpp;
Unk_0C = s.Serialize <byte>(Unk_0C, name: nameof(Unk_0C));
Unk_0D = s.Serialize <byte>(Unk_0D, name: nameof(Unk_0D));
Unk_0E = s.Serialize <byte>(Unk_0E, name: nameof(Unk_0E));
Unk_0F = s.Serialize <byte>(Unk_0F, name: nameof(Unk_0F));
Mode7_10 = s.Serialize <byte>(Mode7_10, name: nameof(Mode7_10));
Mode7_11 = s.Serialize <byte>(Mode7_11, name: nameof(Mode7_11));
Mode7_12 = s.Serialize <byte>(Mode7_12, name: nameof(Mode7_12));
Mode7_13 = s.Serialize <byte>(Mode7_13, name: nameof(Mode7_13));
Mode7_14 = s.Serialize <byte>(Mode7_14, name: nameof(Mode7_14));
break;
case Type.SplinterCellZoom:
if (LayerID < 2)
{
Unk_0C = s.Serialize <byte>(Unk_0C, name: nameof(Unk_0C));
Unk_0D = s.Serialize <byte>(Unk_0D, name: nameof(Unk_0D));
Unk_0E = s.Serialize <byte>(Unk_0E, name: nameof(Unk_0E));
Unk_0F = s.Serialize <byte>(Unk_0F, name: nameof(Unk_0F));
if (s.GameSettings.EngineVersion == EngineVersion.GBA_SplinterCell_NGage)
{
ColorMode = GBA_ColorMode.Color8bpp;
}
else
{
ColorMode = GBA_ColorMode.Color4bpp;
}
}
else
{
ColorMode = GBA_ColorMode.Color8bpp;
}
UsesTileKitDirectly = true;
break;
case Type.PoP:
Unk_0C = s.Serialize <byte>(Unk_0C, name: nameof(Unk_0C));
Unk_0D = s.Serialize <byte>(Unk_0D, name: nameof(Unk_0D));
Unk_0E = s.Serialize <byte>(Unk_0E, name: nameof(Unk_0E));
Unk_0F = s.Serialize <byte>(Unk_0F, name: nameof(Unk_0F));
ColorMode = GBA_ColorMode.Color8bpp;
break;
case Type.Layer2D:
LayerID = s.Serialize <byte>(LayerID, name: nameof(LayerID));
ClusterIndex = s.Serialize <byte>(ClusterIndex, name: nameof(ClusterIndex));
ShouldSetBGAlphaBlending = s.Serialize <bool>(ShouldSetBGAlphaBlending, name: nameof(ShouldSetBGAlphaBlending));
AlphaBlending_Coeff = s.Serialize <sbyte>(AlphaBlending_Coeff, name: nameof(AlphaBlending_Coeff));
UsesTileKitDirectly = s.Serialize <bool>(UsesTileKitDirectly, name: nameof(UsesTileKitDirectly));
ColorMode = s.Serialize <GBA_ColorMode>(ColorMode, name: nameof(ColorMode));
TileKitIndex = s.Serialize <byte>(TileKitIndex, name: nameof(TileKitIndex));
Unk_0F = s.Serialize <byte>(Unk_0F, name: nameof(Unk_0F));
break;
}
}
if (StructType != Type.TextLayerMode7)
{
if (!IsCompressed)
{
SerializeTileMap(s);
}
else if (s.GameSettings.EngineVersion >= EngineVersion.GBA_PrinceOfPersia && StructType != Type.PoP)
{
s.DoEncoded(new GBA_Huffman4Encoder(), () => s.DoEncoded(new GBA_LZSSEncoder(), () => SerializeTileMap(s)));
}
else
{
s.DoEncoded(new GBA_LZSSEncoder(), () => SerializeTileMap(s));
}
s.Align();
}
}
public override void DoEncoded(IStreamEncoder encoder, Action action, BinaryFile.Endian?endianness = null)
{
action();
}
public override void DoEncoded(IStreamEncoder encoder, Action action)
{
SwitchSerializer(Deserializer);
CurrentSerializer.DoEncoded(encoder, action);
}
public override void DoEncoded(IStreamEncoder encoder, Action action, BinaryFile.Endian?endianness = null)
{
SwitchSerializer(Deserializer);
CurrentSerializer.DoEncoded(encoder, action, endianness);
}
