[mtl canvas] triple buffer vertex/index/shape buffers

This commit is contained in:
Martin Fouilleul 2023-03-15 17:48:39 +01:00
parent 0c74e997ae
commit bd8e31c535
3 changed files with 103 additions and 80 deletions

View File

@ -203,8 +203,6 @@ int main()
f32 textX = startX;
f32 textY = startY;
mg_surface_prepare(surface);
/*
mg_set_color_rgba(1, 1, 1, 1);
mg_clear();
@ -270,6 +268,8 @@ int main()
f64 startFlushTime = mp_get_time(MP_CLOCK_MONOTONIC);
mg_surface_prepare(surface);
mg_flush();
f64 startPresentTime = mp_get_time(MP_CLOCK_MONOTONIC);

View File

@ -20,6 +20,8 @@
static const int MG_MTL_CANVAS_DEFAULT_BUFFER_LENGTH = 4<<20;
static const int MG_MTL_MAX_BUFFERS_IN_FLIGHT = 3;
typedef struct mg_mtl_canvas_backend
{
mg_canvas_backend interface;
@ -37,11 +39,17 @@ typedef struct mg_mtl_canvas_backend
mp_rect viewPort;
// triple buffering
u32 bufferIndex;
dispatch_semaphore_t bufferSemaphore;
// textures and buffers
id<MTLTexture> framebuffer;
id<MTLTexture> outTexture;
id<MTLBuffer> shapeBuffer;
id<MTLBuffer> vertexBuffer;
id<MTLBuffer> indexBuffer;
id<MTLBuffer> shapeBuffer[MG_MTL_MAX_BUFFERS_IN_FLIGHT];
id<MTLBuffer> vertexBuffer[MG_MTL_MAX_BUFFERS_IN_FLIGHT];
id<MTLBuffer> indexBuffer[MG_MTL_MAX_BUFFERS_IN_FLIGHT];
id<MTLBuffer> tileCounters;
id<MTLBuffer> tileArrayBuffer;
id<MTLBuffer> triangleArray;
@ -68,9 +76,9 @@ mg_mtl_surface* mg_mtl_canvas_get_surface(mg_mtl_canvas_backend* canvas)
void mg_mtl_canvas_update_vertex_layout(mg_mtl_canvas_backend* backend)
{
char* vertexBase = (char*)[backend->vertexBuffer contents] + backend->vertexBufferOffset;
char* shapeBase = (char*)[backend->shapeBuffer contents] + backend->shapeBufferOffset;
char* indexBase = (char*)[backend->indexBuffer contents] + backend->indexBufferOffset;
char* vertexBase = (char*)[backend->vertexBuffer[backend->bufferIndex] contents] + backend->vertexBufferOffset;
char* shapeBase = (char*)[backend->shapeBuffer[backend->bufferIndex] contents] + backend->shapeBufferOffset;
char* indexBase = (char*)[backend->indexBuffer[backend->bufferIndex] contents] + backend->indexBufferOffset;
backend->interface.vertexLayout = (mg_vertex_layout){
.maxVertexCount = MG_MTL_CANVAS_DEFAULT_BUFFER_LENGTH,
@ -101,18 +109,26 @@ void mg_mtl_canvas_begin(mg_canvas_backend* interface, mg_color clearColor)
{
return;
}
@autoreleasepool
{
if(surface->commandBuffer == nil)
{
mg_mtl_surface_acquire_drawable_and_command_buffer(surface);
}
backend->vertexBufferOffset = 0;
backend->indexBufferOffset = 0;
backend->shapeBufferOffset = 0;
dispatch_semaphore_wait(backend->bufferSemaphore, DISPATCH_TIME_FOREVER);
backend->bufferIndex = (backend->bufferIndex + 1) % MG_MTL_MAX_BUFFERS_IN_FLIGHT;
mg_mtl_canvas_update_vertex_layout(backend);
@autoreleasepool
{
if(surface->commandBuffer == nil || surface->drawable == nil)
{
mg_mtl_surface_acquire_drawable_and_command_buffer(surface);
}
if(surface->drawable != nil)
{
backend->framebuffer = surface->drawable.texture;
MTLClearColor mtlClearColor = MTLClearColorMake(clearColor.r, clearColor.g, clearColor.b, clearColor.a);
MTLRenderPassDescriptor* renderPassDescriptor = [MTLRenderPassDescriptor renderPassDescriptor];
@ -124,24 +140,45 @@ void mg_mtl_canvas_begin(mg_canvas_backend* interface, mg_color clearColor)
id<MTLRenderCommandEncoder> renderEncoder = [surface->commandBuffer renderCommandEncoderWithDescriptor:renderPassDescriptor];
[renderEncoder endEncoding];
}
else
{
backend->framebuffer = nil;
}
}
}
void mg_mtl_canvas_end(mg_canvas_backend* interface)
{}
{
mg_mtl_canvas_backend* backend = (mg_mtl_canvas_backend*)interface;
mg_mtl_surface* surface = mg_mtl_canvas_get_surface(backend);
if(surface && surface->commandBuffer)
{
@autoreleasepool
{
[surface->commandBuffer addCompletedHandler:^(id<MTLCommandBuffer> commandBuffer)
{
dispatch_semaphore_signal(backend->bufferSemaphore);
}
];
}
}
}
void mg_mtl_canvas_draw_batch(mg_canvas_backend* interface, mg_image_data* image, u32 shapeCount, u32 vertexCount, u32 indexCount)
{
mg_mtl_canvas_backend* backend = (mg_mtl_canvas_backend*)interface;
mg_mtl_surface* surface = mg_mtl_canvas_get_surface(backend);
if(!surface)
if(!surface || (backend->framebuffer == nil))
{
return;
}
//TODO: guard against overflowing buffers...
@autoreleasepool
{
ASSERT(indexCount * sizeof(i32) < [backend->indexBuffer length]);
f32 scale = surface->mtlLayer.contentsScale;
vector_uint2 viewportSize = {backend->viewPort.w * scale, backend->viewPort.h * scale};
@ -160,9 +197,9 @@ void mg_mtl_canvas_draw_batch(mg_canvas_backend* interface, mg_image_data* image
id<MTLComputeCommandEncoder> tileEncoder = [surface->commandBuffer computeCommandEncoder];
tileEncoder.label = @"tiling pass";
[tileEncoder setComputePipelineState: backend->tilingPipeline];
[tileEncoder setBuffer: backend->vertexBuffer offset:backend->vertexBufferOffset atIndex: 0];
[tileEncoder setBuffer: backend->indexBuffer offset:backend->indexBufferOffset atIndex: 1];
[tileEncoder setBuffer: backend->shapeBuffer offset:backend->shapeBufferOffset atIndex: 2];
[tileEncoder setBuffer: backend->vertexBuffer[backend->bufferIndex] offset:backend->vertexBufferOffset atIndex: 0];
[tileEncoder setBuffer: backend->indexBuffer[backend->bufferIndex] offset:backend->indexBufferOffset atIndex: 1];
[tileEncoder setBuffer: backend->shapeBuffer[backend->bufferIndex] offset:backend->shapeBufferOffset atIndex: 2];
[tileEncoder setBuffer: backend->tileCounters offset:0 atIndex: 3];
[tileEncoder setBuffer: backend->tileArrayBuffer offset:0 atIndex: 4];
[tileEncoder setBuffer: backend->triangleArray offset:0 atIndex: 5];
@ -238,7 +275,7 @@ void mg_mtl_canvas_draw_batch(mg_canvas_backend* interface, mg_image_data* image
1};
MTLRenderPassDescriptor* renderPassDescriptor = [MTLRenderPassDescriptor renderPassDescriptor];
renderPassDescriptor.colorAttachments[0].texture = surface->drawable.texture;
renderPassDescriptor.colorAttachments[0].texture = backend->framebuffer;
renderPassDescriptor.colorAttachments[0].loadAction = MTLLoadActionLoad;
renderPassDescriptor.colorAttachments[0].storeAction = MTLStoreActionStore;
@ -299,8 +336,15 @@ void mg_mtl_canvas_destroy(mg_canvas_backend* interface)
@autoreleasepool
{
[backend->outTexture release];
[backend->vertexBuffer release];
[backend->indexBuffer release];
for(int i=0; i < MG_MTL_MAX_BUFFERS_IN_FLIGHT; i++)
{
[backend->vertexBuffer[i] release];
[backend->indexBuffer[i] release];
[backend->shapeBuffer[i] release];
}
//NOTE: semaphore does not have a destructor?
[backend->tileArrayBuffer release];
[backend->triangleArray release];
[backend->computePipeline release];
@ -412,20 +456,26 @@ mg_canvas_backend* mg_mtl_canvas_create(mg_surface surface)
//TODO(martin): retain ?
//-----------------------------------------------------------
//NOTE(martin): create buffers for vertex and index
//NOTE(martin): create buffers
//-----------------------------------------------------------
backend->bufferSemaphore = dispatch_semaphore_create(MG_MTL_MAX_BUFFERS_IN_FLIGHT);
backend->bufferIndex = 0;
MTLResourceOptions bufferOptions = MTLResourceCPUCacheModeWriteCombined
| MTLResourceStorageModeShared;
backend->indexBuffer = [metalSurface->device newBufferWithLength: MG_MTL_CANVAS_DEFAULT_BUFFER_LENGTH*sizeof(int)
for(int i=0; i<MG_MTL_MAX_BUFFERS_IN_FLIGHT; i++)
{
backend->indexBuffer[i] = [metalSurface->device newBufferWithLength: MG_MTL_CANVAS_DEFAULT_BUFFER_LENGTH*sizeof(int)
options: bufferOptions];
backend->vertexBuffer = [metalSurface->device newBufferWithLength: MG_MTL_CANVAS_DEFAULT_BUFFER_LENGTH*sizeof(mg_vertex)
backend->vertexBuffer[i] = [metalSurface->device newBufferWithLength: MG_MTL_CANVAS_DEFAULT_BUFFER_LENGTH*sizeof(mg_vertex)
options: bufferOptions];
backend->shapeBuffer = [metalSurface->device newBufferWithLength: MG_MTL_CANVAS_DEFAULT_BUFFER_LENGTH*sizeof(mg_shape)
backend->shapeBuffer[i] = [metalSurface->device newBufferWithLength: MG_MTL_CANVAS_DEFAULT_BUFFER_LENGTH*sizeof(mg_shape)
options: bufferOptions];
}
backend->tileArrayBuffer = [metalSurface->device newBufferWithLength: RENDERER_TILE_BUFFER_SIZE*sizeof(int)*RENDERER_MAX_TILES
options: MTLResourceStorageModePrivate];

View File

@ -17,8 +17,6 @@
#define LOG_SUBSYSTEM "Graphics"
static const u32 MP_MTL_MAX_DRAWABLES_IN_FLIGHT = 3;
typedef struct mg_mtl_surface
{
mg_surface_data interface;
@ -32,8 +30,6 @@ typedef struct mg_mtl_surface
id<CAMetalDrawable> drawable;
id<MTLCommandBuffer> commandBuffer;
dispatch_semaphore_t drawableSemaphore;
} mg_mtl_surface;
void mg_mtl_surface_destroy(mg_surface_data* interface)
@ -59,45 +55,23 @@ void mg_mtl_surface_destroy(mg_surface_data* interface)
void mg_mtl_surface_acquire_drawable_and_command_buffer(mg_mtl_surface* surface)
{@autoreleasepool{
/*WARN(martin): this is super important
When the app is put in the background, it seems that if there are buffers in flight, the drawables to
can be leaked. This causes the gpu to allocate more and more drawables, until the app crashes.
(note: the drawable objects themselves are released once the app comes back to the forefront, but the
memory allocated in the GPU is never freed...)
In background the gpu seems to create drawable if none is available instead of actually
blocking on nextDrawable. These drawable never get freed.
This is not a problem if our shader is fast enough, since a previous drawable becomes
available before we finish the frame. But we want to protect against it anyway
The normal blocking mechanism of nextDrawable seems useless, so we implement our own scheme by
counting the number of drawables available with a semaphore that gets decremented here and
incremented in the presentedHandler of the drawable.
Thus we ensure that we don't consume more drawables than we are able to draw.
//TODO: we _also_ should stop trying to render if we detect that the app is in the background
or occluded, but we can't count only on this because there is a potential race between the
notification of background mode and the rendering.
/*WARN(martin):
//TODO: we should stop trying to render if we detect that the app is in the background
or occluded
//TODO: We should set a reasonable timeout and skip the frame and log an error in case we are stalled
for too long.
*/
dispatch_semaphore_wait(surface->drawableSemaphore, DISPATCH_TIME_FOREVER);
//NOTE: returned drawable could be nil if we stall for more than 1s, although that never seem to happen in practice?
surface->drawable = [surface->mtlLayer nextDrawable];
ASSERT(surface->drawable != nil);
//TODO: make this a weak reference if we use ARC
dispatch_semaphore_t semaphore = surface->drawableSemaphore;
[surface->drawable addPresentedHandler:^(id<MTLDrawable> drawable){
dispatch_semaphore_signal(semaphore);
}];
//NOTE(martin): create a command buffer
surface->commandBuffer = [surface->commandQueue commandBuffer];
[surface->commandBuffer retain];
if(surface->drawable)
{
[surface->drawable retain];
}
surface->commandBuffer = [surface->commandQueue commandBuffer];
[surface->commandBuffer retain];
}}
void mg_mtl_surface_prepare(mg_surface_data* interface)
@ -111,14 +85,15 @@ void mg_mtl_surface_present(mg_surface_data* interface)
mg_mtl_surface* surface = (mg_mtl_surface*)interface;
@autoreleasepool
{
//NOTE(martin): present drawable and commit command buffer
if(surface->drawable != nil)
{
[surface->commandBuffer presentDrawable: surface->drawable];
[surface->commandBuffer commit];
// [surface->commandBuffer waitUntilCompleted];
//TODO: do we really need this?
[surface->drawable release];
surface->drawable = nil;
}
[surface->commandBuffer commit];
// [surface->commandBuffer waitUntilCompleted];
//TODO: do we really need this?
[surface->commandBuffer release];
surface->commandBuffer = nil;
@ -169,8 +144,6 @@ mg_surface_data* mg_mtl_surface_create_for_window(mp_window window)
@autoreleasepool
{
surface->drawableSemaphore = dispatch_semaphore_create(MP_MTL_MAX_DRAWABLES_IN_FLIGHT);
//-----------------------------------------------------------
//NOTE(martin): create a mtl device and a mtl layer and
//-----------------------------------------------------------