v0.13.0.11
This commit is contained in:
Joonas Rikkonen
@@ -0,0 +1,324 @@
|
||||
using Microsoft.Xna.Framework;
|
||||
using Microsoft.Xna.Framework.Graphics;
|
||||
using System;
|
||||
using System.Collections.Generic;
|
||||
using System.Linq;
|
||||
using System.Text;
|
||||
|
||||
namespace Barotrauma
|
||||
{
|
||||
public class SpriteRecorder : ISpriteBatch, IDisposable
|
||||
{
|
||||
private struct Command
|
||||
{
|
||||
public readonly Texture2D Texture;
|
||||
public readonly VertexPositionColorTexture VertexBL;
|
||||
public readonly VertexPositionColorTexture VertexBR;
|
||||
public readonly VertexPositionColorTexture VertexTL;
|
||||
public readonly VertexPositionColorTexture VertexTR;
|
||||
public readonly float Depth;
|
||||
public readonly Vector2 Min;
|
||||
public readonly Vector2 Max;
|
||||
public readonly int Index;
|
||||
|
||||
public bool Overlaps(Command other)
|
||||
{
|
||||
return
|
||||
Min.X <= other.Max.X && Max.X >= other.Min.X &&
|
||||
Min.Y <= other.Max.Y && Max.Y >= other.Min.Y;
|
||||
}
|
||||
|
||||
public Command(
|
||||
Texture2D texture,
|
||||
Vector2 pos,
|
||||
Rectangle srcRect,
|
||||
Color color,
|
||||
float rotation,
|
||||
Vector2 origin,
|
||||
Vector2 scale,
|
||||
SpriteEffects effects,
|
||||
float depth,
|
||||
int index)
|
||||
{
|
||||
int srcRectLeft = srcRect.Left;
|
||||
int srcRectRight = srcRect.Right;
|
||||
int srcRectTop = srcRect.Top;
|
||||
int srcRectBottom = srcRect.Bottom;
|
||||
if (effects.HasFlag(SpriteEffects.FlipHorizontally))
|
||||
{
|
||||
var temp = srcRectRight;
|
||||
srcRectRight = srcRectLeft;
|
||||
srcRectLeft = temp;
|
||||
}
|
||||
if (effects.HasFlag(SpriteEffects.FlipVertically))
|
||||
{
|
||||
var temp = srcRectBottom;
|
||||
srcRectBottom = srcRectTop;
|
||||
srcRectTop = temp;
|
||||
}
|
||||
|
||||
rotation = MathHelper.ToRadians(rotation);
|
||||
float sin = (float)Math.Sin(rotation);
|
||||
float cos = (float)Math.Cos(rotation);
|
||||
|
||||
var size = srcRect.Size.ToVector2() * scale;
|
||||
|
||||
Vector2 wAdd = new Vector2(size.X * cos, size.X * sin);
|
||||
Vector2 hAdd = new Vector2(-size.Y * sin, size.Y * cos);
|
||||
pos.X -= origin.X * scale.X * cos - origin.Y * scale.Y * sin;
|
||||
pos.Y -= origin.Y * scale.Y * cos + origin.X * scale.X * sin;
|
||||
|
||||
Texture = texture;
|
||||
|
||||
Depth = depth;
|
||||
|
||||
VertexTL.Color = color;
|
||||
VertexTR.Color = color;
|
||||
VertexBL.Color = color;
|
||||
VertexBR.Color = color;
|
||||
|
||||
VertexTL.Position = new Vector3(pos.X, pos.Y, 0f);
|
||||
VertexTR.Position = new Vector3(pos.X + wAdd.X, pos.Y + wAdd.Y, 0f);
|
||||
VertexBL.Position = new Vector3(pos.X + hAdd.X, pos.Y + hAdd.Y, 0f);
|
||||
VertexBR.Position = new Vector3(pos.X + wAdd.X + hAdd.X, pos.Y + wAdd.Y + hAdd.Y, 0f);
|
||||
|
||||
Min = new Vector2(
|
||||
MathUtils.Min
|
||||
(
|
||||
VertexTL.Position.X,
|
||||
VertexTR.Position.X,
|
||||
VertexBL.Position.X,
|
||||
VertexBR.Position.X
|
||||
),
|
||||
MathUtils.Min
|
||||
(
|
||||
VertexTL.Position.Y,
|
||||
VertexTR.Position.Y,
|
||||
VertexBL.Position.Y,
|
||||
VertexBR.Position.Y
|
||||
));
|
||||
|
||||
Max = new Vector2(
|
||||
MathUtils.Max
|
||||
(
|
||||
VertexTL.Position.X,
|
||||
VertexTR.Position.X,
|
||||
VertexBL.Position.X,
|
||||
VertexBR.Position.X
|
||||
),
|
||||
MathUtils.Max
|
||||
(
|
||||
VertexTL.Position.Y,
|
||||
VertexTR.Position.Y,
|
||||
VertexBL.Position.Y,
|
||||
VertexBR.Position.Y
|
||||
));
|
||||
|
||||
VertexTL.TextureCoordinate = new Vector2((float)srcRectLeft / (float)texture.Width, (float)srcRectTop / (float)texture.Height);
|
||||
VertexTR.TextureCoordinate = new Vector2((float)srcRectRight / (float)texture.Width, (float)srcRectTop / (float)texture.Height);
|
||||
VertexBL.TextureCoordinate = new Vector2((float)srcRectLeft / (float)texture.Width, (float)srcRectBottom / (float)texture.Height);
|
||||
VertexBR.TextureCoordinate = new Vector2((float)srcRectRight / (float)texture.Width, (float)srcRectBottom / (float)texture.Height);
|
||||
|
||||
Index = index;
|
||||
|
||||
}
|
||||
}
|
||||
|
||||
private struct RecordedBuffer
|
||||
{
|
||||
public readonly Texture2D Texture;
|
||||
public readonly VertexBuffer VertexBuffer;
|
||||
public readonly int PolyCount;
|
||||
|
||||
public RecordedBuffer(List<Command> commandList, int startIndex, int count)
|
||||
{
|
||||
Texture = commandList[startIndex].Texture;
|
||||
|
||||
VertexBuffer = new VertexBuffer(GameMain.Instance.GraphicsDevice, VertexPositionColorTexture.VertexDeclaration, count * 4, BufferUsage.WriteOnly);
|
||||
VertexPositionColorTexture[] vertices = new VertexPositionColorTexture[count * 4];
|
||||
for (int i = 0; i < count; i++)
|
||||
{
|
||||
vertices[(i * 4) + 0] = commandList[startIndex + i].VertexBL;
|
||||
vertices[(i * 4) + 1] = commandList[startIndex + i].VertexBR;
|
||||
vertices[(i * 4) + 2] = commandList[startIndex + i].VertexTL;
|
||||
vertices[(i * 4) + 3] = commandList[startIndex + i].VertexTR;
|
||||
}
|
||||
VertexBuffer.SetData(vertices);
|
||||
|
||||
PolyCount = count * 2;
|
||||
}
|
||||
}
|
||||
|
||||
public static BasicEffect BasicEffect = null;
|
||||
|
||||
private List<RecordedBuffer> recordedBuffers = new List<RecordedBuffer>();
|
||||
private List<Command> commandList = new List<Command>();
|
||||
private SpriteSortMode currentSortMode;
|
||||
|
||||
private IndexBuffer indexBuffer = null;
|
||||
private int maxSpriteCount = 0;
|
||||
|
||||
public volatile bool ReadyToRender = false;
|
||||
private volatile bool isDisposed = false;
|
||||
|
||||
public Vector2 Min { get; private set; }
|
||||
public Vector2 Max { get; private set; }
|
||||
|
||||
public void Begin(SpriteSortMode sortMode)
|
||||
{
|
||||
ReadyToRender = false;
|
||||
currentSortMode = sortMode;
|
||||
}
|
||||
|
||||
public void Draw(Texture2D texture, Vector2 pos, Rectangle? srcRect, Color color, float rotation, Vector2 origin, Vector2 scale, SpriteEffects effects, float depth)
|
||||
{
|
||||
if (isDisposed) { return; }
|
||||
|
||||
Command command = new Command(texture, pos, srcRect ?? texture.Bounds, color, rotation, origin, scale, effects, depth, commandList?.Count ?? 0);
|
||||
if (commandList.Count == 0) { Min = command.Min; Max = command.Max; }
|
||||
Min = new Vector2(Math.Min(command.Min.X, Min.X), Math.Min(command.Min.Y, Min.Y));
|
||||
Max = new Vector2(Math.Max(command.Max.X, Max.X), Math.Max(command.Max.Y, Max.Y));
|
||||
|
||||
commandList?.Add(command);
|
||||
}
|
||||
|
||||
public void End()
|
||||
{
|
||||
if (isDisposed) { return; }
|
||||
//sort commands according to the sorting
|
||||
//mode given in the last Begin call
|
||||
switch (currentSortMode)
|
||||
{
|
||||
case SpriteSortMode.FrontToBack:
|
||||
commandList.Sort((c1, c2) =>
|
||||
{
|
||||
return c1.Depth < c2.Depth ? -1
|
||||
: c1.Depth > c2.Depth ? 1
|
||||
: c1.Index < c2.Index ? 1
|
||||
: c1.Index > c2.Index ? -1
|
||||
: 0;
|
||||
});
|
||||
break;
|
||||
case SpriteSortMode.BackToFront:
|
||||
commandList.Sort((c1, c2) =>
|
||||
{
|
||||
return c1.Depth < c2.Depth ? 1
|
||||
: c1.Depth > c2.Depth ? -1
|
||||
: c1.Index < c2.Index ? 1
|
||||
: c1.Index > c2.Index ? -1
|
||||
: 0;
|
||||
});
|
||||
break;
|
||||
}
|
||||
|
||||
//try to place commands of the same texture
|
||||
//contiguously for optimal buffer generation
|
||||
//while maintaining the same visual result
|
||||
for (int i=1;i<commandList.Count;i++)
|
||||
{
|
||||
if (commandList[i].Texture != commandList[i-1].Texture)
|
||||
{
|
||||
for (int j = i - 1; j >= 0; j--)
|
||||
{
|
||||
if (commandList[j].Texture == commandList[i].Texture)
|
||||
{
|
||||
//no commands between i and j overlap with
|
||||
//i, therefore we can safely sift i down to
|
||||
//make a contiguous block
|
||||
commandList.SiftElement(i, j + 1);
|
||||
break;
|
||||
}
|
||||
else if (commandList[j].Overlaps(commandList[i]))
|
||||
{
|
||||
//an overlapping command was found, therefore
|
||||
//attempting to sift this one down would change
|
||||
//the visual result
|
||||
break;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
if (isDisposed) { return; }
|
||||
//each contiguous block of commands of the same texture
|
||||
//requires a vertex buffer to be rendered
|
||||
CrossThread.RequestExecutionOnMainThread(() =>
|
||||
{
|
||||
if (commandList.Count == 0) { return; }
|
||||
int startIndex = 0;
|
||||
for (int i = 1; i < commandList.Count; i++)
|
||||
{
|
||||
if (commandList[i].Texture != commandList[startIndex].Texture)
|
||||
{
|
||||
maxSpriteCount = Math.Max(maxSpriteCount, i - startIndex);
|
||||
recordedBuffers.Add(new RecordedBuffer(commandList, startIndex, i - startIndex));
|
||||
startIndex = i;
|
||||
}
|
||||
}
|
||||
recordedBuffers.Add(new RecordedBuffer(commandList, startIndex, commandList.Count - startIndex));
|
||||
maxSpriteCount = Math.Max(maxSpriteCount, commandList.Count - startIndex);
|
||||
});
|
||||
|
||||
commandList.Clear();
|
||||
|
||||
ReadyToRender = true;
|
||||
}
|
||||
|
||||
public void Render(Camera cam)
|
||||
{
|
||||
if (!ReadyToRender) { return; }
|
||||
var gfxDevice = GameMain.Instance.GraphicsDevice;
|
||||
|
||||
BasicEffect ??= new BasicEffect(gfxDevice);
|
||||
BasicEffect.Projection = Matrix.CreateOrthographicOffCenter(new Rectangle(0, 0, cam.Resolution.X, cam.Resolution.Y), -1f, 1f);
|
||||
BasicEffect.View = cam.Transform;
|
||||
BasicEffect.World = Matrix.Identity;
|
||||
BasicEffect.TextureEnabled = true;
|
||||
BasicEffect.VertexColorEnabled = true;
|
||||
BasicEffect.Alpha = 1f;
|
||||
|
||||
int requiredIndexCount = maxSpriteCount * 6;
|
||||
if (requiredIndexCount > 0 && (indexBuffer == null || indexBuffer.IndexCount < requiredIndexCount))
|
||||
{
|
||||
indexBuffer?.Dispose();
|
||||
indexBuffer = new IndexBuffer(gfxDevice, IndexElementSize.SixteenBits, requiredIndexCount * 2, BufferUsage.WriteOnly);
|
||||
ushort[] indices = new ushort[requiredIndexCount * 2];
|
||||
for (int i=0;i<indices.Length;i+=6)
|
||||
{
|
||||
indices[i + 0] = (ushort)((i / 6) * 4 + 1);
|
||||
indices[i + 1] = (ushort)((i / 6) * 4 + 0);
|
||||
indices[i + 2] = (ushort)((i / 6) * 4 + 2);
|
||||
indices[i + 3] = (ushort)((i / 6) * 4 + 1);
|
||||
indices[i + 4] = (ushort)((i / 6) * 4 + 2);
|
||||
indices[i + 5] = (ushort)((i / 6) * 4 + 3);
|
||||
}
|
||||
indexBuffer.SetData(indices);
|
||||
}
|
||||
|
||||
gfxDevice.Indices = indexBuffer;
|
||||
for (int i=0;i<recordedBuffers.Count;i++)
|
||||
{
|
||||
gfxDevice.SetVertexBuffer(recordedBuffers[i].VertexBuffer);
|
||||
BasicEffect.Texture = recordedBuffers[i].Texture;
|
||||
BasicEffect.CurrentTechnique.Passes[0].Apply();
|
||||
gfxDevice.DrawIndexedPrimitives(PrimitiveType.TriangleList, 0, 0, recordedBuffers[i].PolyCount);
|
||||
}
|
||||
}
|
||||
|
||||
public void Dispose()
|
||||
{
|
||||
isDisposed = true;
|
||||
if (recordedBuffers != null)
|
||||
{
|
||||
foreach (var buffer in recordedBuffers)
|
||||
{
|
||||
buffer.VertexBuffer.Dispose();
|
||||
}
|
||||
recordedBuffers.Clear(); recordedBuffers = null;
|
||||
}
|
||||
commandList?.Clear(); commandList = null;
|
||||
indexBuffer?.Dispose(); indexBuffer = null;
|
||||
ReadyToRender = false;
|
||||
}
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user