232 lines
10 KiB
C#
232 lines
10 KiB
C#
|
//#define PACKING_DEBUG
|
||
|
|
|
||
|
|
using System;
|
||
|
|
using Unity.Burst;
|
||
|
|
using Unity.Mathematics;
|
||
|
|
using UnityEngine;
|
||
|
|
using Unity.Collections;
|
||
|
|
using Unity.Collections.LowLevel.Unsafe;
|
||
|
|
|
||
|
|
namespace UnityEditor.U2D.Common
|
||
|
|
{
|
||
|
|
[BurstCompile]
|
||
|
|
internal static class ImagePacker
|
||
|
|
{
|
||
|
|
/// <summary>
|
||
|
|
/// Given an array of rects, the method returns an array of rects arranged within outPackedWidth and outPackedHeight
|
||
|
|
/// </summary>
|
||
|
|
/// <param name="rects">Rects to pack</param>
|
||
|
|
/// <param name="padding">Padding between each rect</param>
|
||
|
|
/// <param name="outPackedRects">Rects arranged within outPackedWidth and outPackedHeight</param>
|
||
|
|
/// <param name="outPackedWidth">Width of the packed rects</param>
|
||
|
|
/// <param name="outPackedHeight">Height of the packed rects</param>
|
||
|
|
public static void Pack(RectInt[] rects, int padding, out RectInt[] outPackedRects, out int outPackedWidth, out int outPackedHeight)
|
||
|
|
{
|
||
|
|
var packNode = InternalPack(rects, padding);
|
||
|
|
outPackedWidth = packNode.rect.width;
|
||
|
|
outPackedHeight = packNode.rect.height;
|
||
|
|
var visitor = new CollectPackNodePositionVisitor();
|
||
|
|
packNode.AcceptVisitor(visitor);
|
||
|
|
|
||
|
|
outPackedRects = new RectInt[rects.Length];
|
||
|
|
for (int i = 0; i < rects.Length; ++i)
|
||
|
|
outPackedRects[i] = new RectInt(visitor.positions[i].x + padding, visitor.positions[i].y + padding, rects[i].width, rects[i].height);
|
||
|
|
#if PACKING_DEBUG
|
||
|
|
var emptyNodeCollector = new CollectEmptyNodePositionVisitor();
|
||
|
|
packNode.AcceptVisitor(emptyNodeCollector);
|
||
|
|
Array.Resize(ref outPackedRects, rects.Length + emptyNodeCollector.emptyAreas.Count);
|
||
|
|
for (int i = rects.Length; i < outPackedRects.Length; ++i)
|
||
|
|
outPackedRects[i] = emptyNodeCollector.emptyAreas[i - rects.Length];
|
||
|
|
#endif
|
||
|
|
}
|
||
|
|
|
||
|
|
/// <summary>
|
||
|
|
/// Packs image buffer into a single buffer. Image buffers are assumed to be 4 bytes per pixel in RGBA format
|
||
|
|
/// </summary>
|
||
|
|
/// <param name="buffers">Image buffers to pack</param>
|
||
|
|
/// <param name="width">Image buffers width</param>
|
||
|
|
/// <param name="height">Image buffers height</param>
|
||
|
|
/// <param name="padding">Padding between each packed image</param>
|
||
|
|
/// <param name="outPackedBuffer">Packed image buffer</param>
|
||
|
|
/// <param name="outPackedBufferWidth">Packed image buffer's width</param>
|
||
|
|
/// <param name="outPackedBufferHeight">Packed image buffer's height</param>
|
||
|
|
/// <param name="outPackedRect">Location of each image buffers in the packed buffer</param>
|
||
|
|
/// <param name="outUVTransform">Translation data from image original buffer to packed buffer</param>
|
||
|
|
public static void Pack(NativeArray<Color32>[] buffers, int[] width, int[] height, int padding, out NativeArray<Color32> outPackedBuffer, out int outPackedBufferWidth, out int outPackedBufferHeight, out RectInt[] outPackedRect, out Vector2Int[] outUVTransform)
|
||
|
|
{
|
||
|
|
UnityEngine.Profiling.Profiler.BeginSample("Pack");
|
||
|
|
// Determine the area that contains data in the buffer
|
||
|
|
outPackedBuffer = default(NativeArray<Color32>);
|
||
|
|
try
|
||
|
|
{
|
||
|
|
var tightRects = FindTightRectJob.Execute(buffers, width, height);
|
||
|
|
var tightRectArea = new RectInt[tightRects.Length];
|
||
|
|
for (var i = 0; i < tightRects.Length; ++i)
|
||
|
|
{
|
||
|
|
var t = tightRects[i];
|
||
|
|
t.width = tightRects[i].width;
|
||
|
|
t.height = tightRects[i].height;
|
||
|
|
tightRectArea[i] = t;
|
||
|
|
}
|
||
|
|
Pack(tightRectArea, padding, out outPackedRect, out outPackedBufferWidth, out outPackedBufferHeight);
|
||
|
|
var packBufferSize = (ulong)outPackedBufferWidth * (ulong)outPackedBufferHeight;
|
||
|
|
|
||
|
|
if (packBufferSize < 0 || packBufferSize >= int.MaxValue)
|
||
|
|
{
|
||
|
|
throw new ArgumentException("Unable to create pack texture. Image size is too big to pack.");
|
||
|
|
}
|
||
|
|
outUVTransform = new Vector2Int[tightRectArea.Length];
|
||
|
|
for (var i = 0; i < outUVTransform.Length; ++i)
|
||
|
|
{
|
||
|
|
outUVTransform[i] = new Vector2Int(outPackedRect[i].x - tightRects[i].x, outPackedRect[i].y - tightRects[i].y);
|
||
|
|
}
|
||
|
|
outPackedBuffer = new NativeArray<Color32>(outPackedBufferWidth * outPackedBufferHeight, Allocator.Persistent);
|
||
|
|
|
||
|
|
Blit(outPackedBuffer, outPackedRect, outPackedBufferWidth, buffers, tightRects, width, padding);
|
||
|
|
}
|
||
|
|
catch (Exception ex)
|
||
|
|
{
|
||
|
|
if (outPackedBuffer.IsCreated)
|
||
|
|
outPackedBuffer.Dispose();
|
||
|
|
throw ex;
|
||
|
|
}
|
||
|
|
finally
|
||
|
|
{
|
||
|
|
UnityEngine.Profiling.Profiler.EndSample();
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
static ImagePackNode InternalPack(RectInt[] rects, int padding)
|
||
|
|
{
|
||
|
|
if (rects == null || rects.Length == 0)
|
||
|
|
return new ImagePackNode() { rect = new RectInt(0, 0, 0, 0)};
|
||
|
|
var sortedRects = new ImagePackRect[rects.Length];
|
||
|
|
for (int i = 0; i < rects.Length; ++i)
|
||
|
|
{
|
||
|
|
sortedRects[i] = new ImagePackRect();
|
||
|
|
sortedRects[i].rect = rects[i];
|
||
|
|
sortedRects[i].index = i;
|
||
|
|
}
|
||
|
|
var initialHeight = (int)NextPowerOfTwo((ulong)rects[0].height);
|
||
|
|
Array.Sort<ImagePackRect>(sortedRects);
|
||
|
|
var root = new ImagePackNode();
|
||
|
|
root.rect = new RectInt(0, 0, (int)NextPowerOfTwo((ulong)rects[0].width), initialHeight);
|
||
|
|
|
||
|
|
for (int i = 0; i < rects.Length; ++i)
|
||
|
|
{
|
||
|
|
if (!root.Insert(sortedRects[i], padding)) // we can't fit
|
||
|
|
{
|
||
|
|
int newWidth = root.rect.width , newHeight = root.rect.height;
|
||
|
|
if (root.rect.width < root.rect.height)
|
||
|
|
{
|
||
|
|
newWidth = (int)NextPowerOfTwo((ulong)root.rect.width + 1);
|
||
|
|
// Every time height changes, we reset height to grow again.
|
||
|
|
newHeight = initialHeight;
|
||
|
|
}
|
||
|
|
else
|
||
|
|
newHeight = (int)NextPowerOfTwo((ulong)root.rect.height + 1);
|
||
|
|
// Reset all packing and try again
|
||
|
|
root = new ImagePackNode();
|
||
|
|
root.rect = new RectInt(0, 0, newWidth, newHeight);
|
||
|
|
i = -1;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return root;
|
||
|
|
}
|
||
|
|
|
||
|
|
public static void Blit(NativeArray<Color32> buffer, RectInt[] blitToArea, int bufferBytesPerRow, NativeArray<Color32>[] originalBuffer, RectInt[] blitFromArea, int[] bytesPerRow, int padding)
|
||
|
|
{
|
||
|
|
UnityEngine.Profiling.Profiler.BeginSample("Blit");
|
||
|
|
|
||
|
|
for (var bufferIndex = 0; bufferIndex < blitToArea.Length && bufferIndex < originalBuffer.Length && bufferIndex < blitFromArea.Length; ++bufferIndex)
|
||
|
|
{
|
||
|
|
var fromArea = new int4(blitFromArea[bufferIndex].x, blitFromArea[bufferIndex].y, blitFromArea[bufferIndex].width, blitFromArea[bufferIndex].height);
|
||
|
|
var toArea = new int4(blitToArea[bufferIndex].x, blitToArea[bufferIndex].y, blitToArea[bufferIndex].width, blitToArea[bufferIndex].height);
|
||
|
|
|
||
|
|
unsafe
|
||
|
|
{
|
||
|
|
var originalBufferPtr = (Color32*) originalBuffer[bufferIndex].GetUnsafeReadOnlyPtr();
|
||
|
|
var outputBufferPtr = (Color32*) buffer.GetUnsafePtr();
|
||
|
|
BurstedBlit(originalBufferPtr, in fromArea, in toArea, bytesPerRow[bufferIndex], bufferBytesPerRow, outputBufferPtr);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
#if PACKING_DEBUG
|
||
|
|
var emptyColors = new Color32[]
|
||
|
|
{
|
||
|
|
new Color32((byte)255, (byte)0, (byte)0, (byte)255),
|
||
|
|
new Color32((byte)255, (byte)255, (byte)0, (byte)255),
|
||
|
|
new Color32((byte)255, (byte)0, (byte)255, (byte)255),
|
||
|
|
new Color32((byte)255, (byte)255, (byte)255, (byte)255),
|
||
|
|
new Color32((byte)0, (byte)255, (byte)0, (byte)255),
|
||
|
|
new Color32((byte)0, (byte)0, (byte)255, (byte)255)
|
||
|
|
};
|
||
|
|
|
||
|
|
for (int k = originalBuffer.Length; k < blitToArea.Length; ++k)
|
||
|
|
{
|
||
|
|
var rectFrom = blitToArea[k];
|
||
|
|
for (int i = 0; i < rectFrom.height; ++i)
|
||
|
|
{
|
||
|
|
for (int j = 0; j < rectFrom.width; ++j)
|
||
|
|
{
|
||
|
|
c[((rectFrom.y + i) * bufferbytesPerRow) + rectFrom.x + j] =
|
||
|
|
emptyColors[k % emptyColors.Length];
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
#endif
|
||
|
|
UnityEngine.Profiling.Profiler.EndSample();
|
||
|
|
}
|
||
|
|
|
||
|
|
[BurstCompile]
|
||
|
|
static unsafe void BurstedBlit(Color32* originalBuffer, in int4 rectFrom, in int4 rectTo, int bytesPerRow, int bufferBytesPerRow, Color32* outputBuffer)
|
||
|
|
{
|
||
|
|
var c = outputBuffer;
|
||
|
|
var b = originalBuffer;
|
||
|
|
var toXStart = (int)(rectTo.z * 0.5f - rectFrom.z * 0.5f);
|
||
|
|
var toYStart = (int)(rectTo.w * 0.5f - rectFrom.w * 0.5f);
|
||
|
|
toXStart = toXStart <= 0 ? rectTo.x : toXStart + rectTo.x;
|
||
|
|
toYStart = toYStart <= 0 ? rectTo.y : toYStart + rectTo.y;
|
||
|
|
for (var i = 0; i < rectFrom.w && i < rectTo.w; ++i)
|
||
|
|
{
|
||
|
|
for (var j = 0; j < rectFrom.z && j < rectTo.z; ++j)
|
||
|
|
{
|
||
|
|
var cc = b[(rectFrom.y + i) * bytesPerRow + rectFrom.x + j];
|
||
|
|
c[((toYStart + i) * bufferBytesPerRow) + toXStart + j] = cc;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
internal static ulong NextPowerOfTwo(ulong v)
|
||
|
|
{
|
||
|
|
v -= 1;
|
||
|
|
v |= v >> 16;
|
||
|
|
v |= v >> 8;
|
||
|
|
v |= v >> 4;
|
||
|
|
v |= v >> 2;
|
||
|
|
v |= v >> 1;
|
||
|
|
return v + 1;
|
||
|
|
}
|
||
|
|
|
||
|
|
internal class ImagePackRect : IComparable<ImagePackRect>
|
||
|
|
{
|
||
|
|
public RectInt rect;
|
||
|
|
public int index;
|
||
|
|
|
||
|
|
public int CompareTo(ImagePackRect obj)
|
||
|
|
{
|
||
|
|
var lhsArea = rect.width * rect.height;
|
||
|
|
var rhsArea = obj.rect.width * obj.rect.height;
|
||
|
|
if (lhsArea > rhsArea)
|
||
|
|
return -1;
|
||
|
|
if (lhsArea < rhsArea)
|
||
|
|
return 1;
|
||
|
|
if (index < obj.index)
|
||
|
|
return -1;
|
||
|
|
|
||
|
|
return 1;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|