mirror of
https://github.com/nestriness/nestri.git
synced 2026-09-19 17:25:19 +03:00
6164e0c636
A Vulkan implicit layer that captures frames from inside the workload's own process and encodes them on the GPU they were drawn on. Fourth and last of this batch, imported as a tree from `nestrilabs/nescapture` on the same terms. Filed under `apps/` rather than `crates/` despite building a cdylib. The rule here is what a thing *is*, not what it compiles to: this is a finished artefact that gets installed into an image beside its layer manifest, not a library another crate in this tree depends on. `crates/` is for the latter, and putting this there would make the distinction useless the first time someone looked. Wired to the workspace, `nesprotocol` by path. Its description named the transport component; that reads better as what it actually is — where the frames go — so it says that instead. Whole workspace builds and tests: 21 across four members. Co-Authored-By: Claude Opus 5 (1M context) <noreply@anthropic.com>
364 lines
14 KiB
Rust
364 lines
14 KiB
Rust
// ─────────────────────────────────────────────────────────────────────────────
|
|
// dispatch.rs — raw Vulkan function-pointer type aliases and dispatch tables
|
|
// ─────────────────────────────────────────────────────────────────────────────
|
|
|
|
use ash::vk;
|
|
use std::os::raw::c_char;
|
|
|
|
pub type RawFn = unsafe extern "system" fn();
|
|
|
|
// ── Instance-level ────────────────────────────────────────────────────────────
|
|
|
|
pub type PFN_vkGetInstanceProcAddr =
|
|
unsafe extern "system" fn(vk::Instance, *const c_char) -> Option<RawFn>;
|
|
pub type PFN_vkGetDeviceProcAddr =
|
|
unsafe extern "system" fn(vk::Device, *const c_char) -> Option<RawFn>;
|
|
|
|
pub type PFN_vkCreateInstance = unsafe extern "system" fn(
|
|
*const vk::InstanceCreateInfo,
|
|
*const vk::AllocationCallbacks,
|
|
*mut vk::Instance,
|
|
) -> vk::Result;
|
|
pub type PFN_vkDestroyInstance =
|
|
unsafe extern "system" fn(vk::Instance, *const vk::AllocationCallbacks);
|
|
|
|
pub type PFN_vkGetPhysicalDeviceMemoryProperties =
|
|
unsafe extern "system" fn(vk::PhysicalDevice, *mut vk::PhysicalDeviceMemoryProperties);
|
|
|
|
pub type PFN_vkCreateDevice = unsafe extern "system" fn(
|
|
vk::PhysicalDevice,
|
|
*const vk::DeviceCreateInfo,
|
|
*const vk::AllocationCallbacks,
|
|
*mut vk::Device,
|
|
) -> vk::Result;
|
|
|
|
// ── Device infrastructure ─────────────────────────────────────────────────────
|
|
|
|
pub type PFN_vkDestroyDevice =
|
|
unsafe extern "system" fn(vk::Device, *const vk::AllocationCallbacks);
|
|
|
|
pub type PFN_vkGetDeviceQueue = unsafe extern "system" fn(vk::Device, u32, u32, *mut vk::Queue);
|
|
|
|
pub type PFN_vkQueuePresentKHR =
|
|
unsafe extern "system" fn(vk::Queue, *const vk::PresentInfoKHR) -> vk::Result;
|
|
|
|
// ── Phase 1: Shader modules ───────────────────────────────────────────────────
|
|
|
|
pub type PFN_vkCreateShaderModule = unsafe extern "system" fn(
|
|
vk::Device,
|
|
*const vk::ShaderModuleCreateInfo,
|
|
*const vk::AllocationCallbacks,
|
|
*mut vk::ShaderModule,
|
|
) -> vk::Result;
|
|
|
|
pub type PFN_vkDestroyShaderModule =
|
|
unsafe extern "system" fn(vk::Device, vk::ShaderModule, *const vk::AllocationCallbacks);
|
|
|
|
// ── Phase 1: Graphics pipelines ──────────────────────────────────────────────
|
|
|
|
pub type PFN_vkCreateGraphicsPipelines = unsafe extern "system" fn(
|
|
vk::Device,
|
|
vk::PipelineCache,
|
|
u32,
|
|
*const vk::GraphicsPipelineCreateInfo,
|
|
*const vk::AllocationCallbacks,
|
|
*mut vk::Pipeline,
|
|
) -> vk::Result;
|
|
|
|
pub type PFN_vkDestroyPipeline =
|
|
unsafe extern "system" fn(vk::Device, vk::Pipeline, *const vk::AllocationCallbacks);
|
|
|
|
// ── Phase 2: Image views and framebuffers ─────────────────────────────────────
|
|
|
|
pub type PFN_vkCreateImageView = unsafe extern "system" fn(
|
|
vk::Device,
|
|
*const vk::ImageViewCreateInfo,
|
|
*const vk::AllocationCallbacks,
|
|
*mut vk::ImageView,
|
|
) -> vk::Result;
|
|
|
|
pub type PFN_vkDestroyImageView =
|
|
unsafe extern "system" fn(vk::Device, vk::ImageView, *const vk::AllocationCallbacks);
|
|
|
|
pub type PFN_vkCreateFramebuffer = unsafe extern "system" fn(
|
|
vk::Device,
|
|
*const vk::FramebufferCreateInfo,
|
|
*const vk::AllocationCallbacks,
|
|
*mut vk::Framebuffer,
|
|
) -> vk::Result;
|
|
|
|
pub type PFN_vkDestroyFramebuffer =
|
|
unsafe extern "system" fn(vk::Device, vk::Framebuffer, *const vk::AllocationCallbacks);
|
|
|
|
pub type PFN_vkAllocateCommandBuffers = unsafe extern "system" fn(
|
|
vk::Device,
|
|
*const vk::CommandBufferAllocateInfo,
|
|
*mut vk::CommandBuffer,
|
|
) -> vk::Result;
|
|
|
|
pub type PFN_vkFreeCommandBuffers =
|
|
unsafe extern "system" fn(vk::Device, vk::CommandPool, u32, *const vk::CommandBuffer);
|
|
|
|
// ── Phase 2: Render pass and rendering ───────────────────────────────────────
|
|
|
|
pub type PFN_vkCmdBindPipeline =
|
|
unsafe extern "system" fn(vk::CommandBuffer, vk::PipelineBindPoint, vk::Pipeline);
|
|
|
|
pub type PFN_vkCmdBeginRenderPass = unsafe extern "system" fn(
|
|
vk::CommandBuffer,
|
|
*const vk::RenderPassBeginInfo,
|
|
vk::SubpassContents,
|
|
);
|
|
|
|
pub type PFN_vkCmdEndRenderPass = unsafe extern "system" fn(vk::CommandBuffer);
|
|
|
|
pub type PFN_vkCmdBeginRenderingKHR =
|
|
unsafe extern "system" fn(vk::CommandBuffer, *const vk::RenderingInfo);
|
|
|
|
pub type PFN_vkCmdEndRenderingKHR = unsafe extern "system" fn(vk::CommandBuffer);
|
|
|
|
// ── Phase 4: Capture images and memory ───────────────────────────────────────
|
|
|
|
pub type PFN_vkCreateImage = unsafe extern "system" fn(
|
|
vk::Device,
|
|
*const vk::ImageCreateInfo,
|
|
*const vk::AllocationCallbacks,
|
|
*mut vk::Image,
|
|
) -> vk::Result;
|
|
|
|
pub type PFN_vkDestroyImage =
|
|
unsafe extern "system" fn(vk::Device, vk::Image, *const vk::AllocationCallbacks);
|
|
|
|
pub type PFN_vkAllocateMemory = unsafe extern "system" fn(
|
|
vk::Device,
|
|
*const vk::MemoryAllocateInfo,
|
|
*const vk::AllocationCallbacks,
|
|
*mut vk::DeviceMemory,
|
|
) -> vk::Result;
|
|
|
|
pub type PFN_vkFreeMemory =
|
|
unsafe extern "system" fn(vk::Device, vk::DeviceMemory, *const vk::AllocationCallbacks);
|
|
|
|
pub type PFN_vkBindImageMemory = unsafe extern "system" fn(
|
|
vk::Device,
|
|
vk::Image,
|
|
vk::DeviceMemory,
|
|
vk::DeviceSize,
|
|
) -> vk::Result;
|
|
|
|
pub type PFN_vkGetImageMemoryRequirements =
|
|
unsafe extern "system" fn(vk::Device, vk::Image, *mut vk::MemoryRequirements);
|
|
|
|
pub type PFN_vkMapMemory = unsafe extern "system" fn(
|
|
vk::Device,
|
|
vk::DeviceMemory,
|
|
vk::DeviceSize,
|
|
vk::DeviceSize,
|
|
vk::MemoryMapFlags,
|
|
*mut *mut std::os::raw::c_void,
|
|
) -> vk::Result;
|
|
|
|
pub type PFN_vkUnmapMemory = unsafe extern "system" fn(vk::Device, vk::DeviceMemory);
|
|
|
|
pub type PFN_vkCmdPipelineBarrier = unsafe extern "system" fn(
|
|
vk::CommandBuffer,
|
|
vk::PipelineStageFlags,
|
|
vk::PipelineStageFlags,
|
|
vk::DependencyFlags,
|
|
u32,
|
|
*const vk::MemoryBarrier,
|
|
u32,
|
|
*const vk::BufferMemoryBarrier,
|
|
u32,
|
|
*const vk::ImageMemoryBarrier,
|
|
);
|
|
|
|
pub type PFN_vkCmdCopyImage = unsafe extern "system" fn(
|
|
vk::CommandBuffer,
|
|
vk::Image,
|
|
vk::ImageLayout,
|
|
vk::Image,
|
|
vk::ImageLayout,
|
|
u32,
|
|
*const vk::ImageCopy,
|
|
);
|
|
|
|
pub type PFN_vkGetImageSubresourceLayout = unsafe extern "system" fn(
|
|
vk::Device,
|
|
vk::Image,
|
|
*const vk::ImageSubresource,
|
|
*mut vk::SubresourceLayout,
|
|
);
|
|
|
|
// DMA-BUF fd export (used to share final_image with pixelforge zero-copy)
|
|
pub type PFN_vkGetMemoryFdKHR = unsafe extern "system" fn(
|
|
vk::Device,
|
|
*const vk::MemoryGetFdInfoKHR,
|
|
*mut std::os::raw::c_int,
|
|
) -> vk::Result;
|
|
|
|
// ── Phase 4: Synchronisation ─────────────────────────────────────────────────
|
|
|
|
pub type PFN_vkCreateFence = unsafe extern "system" fn(
|
|
vk::Device,
|
|
*const vk::FenceCreateInfo,
|
|
*const vk::AllocationCallbacks,
|
|
*mut vk::Fence,
|
|
) -> vk::Result;
|
|
|
|
pub type PFN_vkDestroyFence =
|
|
unsafe extern "system" fn(vk::Device, vk::Fence, *const vk::AllocationCallbacks);
|
|
|
|
pub type PFN_vkCreateCommandPool = unsafe extern "system" fn(
|
|
vk::Device,
|
|
*const vk::CommandPoolCreateInfo,
|
|
*const vk::AllocationCallbacks,
|
|
*mut vk::CommandPool,
|
|
) -> vk::Result;
|
|
|
|
pub type PFN_vkDestroyCommandPool =
|
|
unsafe extern "system" fn(vk::Device, vk::CommandPool, *const vk::AllocationCallbacks);
|
|
|
|
pub type PFN_vkResetCommandPool =
|
|
unsafe extern "system" fn(vk::Device, vk::CommandPool, vk::CommandPoolResetFlags) -> vk::Result;
|
|
|
|
pub type PFN_vkBeginCommandBuffer =
|
|
unsafe extern "system" fn(vk::CommandBuffer, *const vk::CommandBufferBeginInfo) -> vk::Result;
|
|
|
|
pub type PFN_vkEndCommandBuffer = unsafe extern "system" fn(vk::CommandBuffer) -> vk::Result;
|
|
|
|
pub type PFN_vkQueueSubmit =
|
|
unsafe extern "system" fn(vk::Queue, u32, *const vk::SubmitInfo, vk::Fence) -> vk::Result;
|
|
|
|
pub type PFN_vkWaitForFences =
|
|
unsafe extern "system" fn(vk::Device, u32, *const vk::Fence, vk::Bool32, u64) -> vk::Result;
|
|
|
|
pub type PFN_vkResetFences =
|
|
unsafe extern "system" fn(vk::Device, u32, *const vk::Fence) -> vk::Result;
|
|
|
|
// ── Phase 4: Swapchain tracking ──────────────────────────────────────────────
|
|
|
|
pub type PFN_vkCreateSwapchainKHR = unsafe extern "system" fn(
|
|
vk::Device,
|
|
*const vk::SwapchainCreateInfoKHR,
|
|
*const vk::AllocationCallbacks,
|
|
*mut vk::SwapchainKHR,
|
|
) -> vk::Result;
|
|
|
|
pub type PFN_vkDestroySwapchainKHR =
|
|
unsafe extern "system" fn(vk::Device, vk::SwapchainKHR, *const vk::AllocationCallbacks);
|
|
|
|
pub type PFN_vkGetSwapchainImagesKHR =
|
|
unsafe extern "system" fn(vk::Device, vk::SwapchainKHR, *mut u32, *mut vk::Image) -> vk::Result;
|
|
|
|
// ── Phase 6: Draw commands ───────────────────────────────────────────────────
|
|
|
|
pub type PFN_vkCmdDraw = unsafe extern "system" fn(vk::CommandBuffer, u32, u32, u32, u32);
|
|
pub type PFN_vkCmdDrawIndexed =
|
|
unsafe extern "system" fn(vk::CommandBuffer, u32, u32, u32, i32, u32);
|
|
pub type PFN_vkCmdDrawIndirect =
|
|
unsafe extern "system" fn(vk::CommandBuffer, vk::Buffer, vk::DeviceSize, u32, u32);
|
|
pub type PFN_vkCmdDrawIndexedIndirect =
|
|
unsafe extern "system" fn(vk::CommandBuffer, vk::Buffer, vk::DeviceSize, u32, u32);
|
|
pub type PFN_vkCmdDrawIndirectCount = unsafe extern "system" fn(
|
|
vk::CommandBuffer,
|
|
vk::Buffer,
|
|
vk::DeviceSize,
|
|
vk::Buffer,
|
|
vk::DeviceSize,
|
|
u32,
|
|
u32,
|
|
);
|
|
pub type PFN_vkCmdDrawIndexedIndirectCount = unsafe extern "system" fn(
|
|
vk::CommandBuffer,
|
|
vk::Buffer,
|
|
vk::DeviceSize,
|
|
vk::Buffer,
|
|
vk::DeviceSize,
|
|
u32,
|
|
u32,
|
|
);
|
|
|
|
pub type PFN_vkResetCommandBuffer =
|
|
unsafe extern "system" fn(vk::CommandBuffer, vk::CommandBufferResetFlags) -> vk::Result;
|
|
|
|
// ── Dispatch table structs ────────────────────────────────────────────────────
|
|
|
|
pub struct NextInstanceFn {
|
|
pub get_instance_proc_addr: PFN_vkGetInstanceProcAddr,
|
|
pub destroy_instance: PFN_vkDestroyInstance,
|
|
pub get_physical_device_memory_properties: PFN_vkGetPhysicalDeviceMemoryProperties,
|
|
pub create_device: PFN_vkCreateDevice,
|
|
}
|
|
|
|
#[derive(Clone, Copy)]
|
|
pub struct NextDeviceFn {
|
|
// Infrastructure
|
|
pub get_device_proc_addr: PFN_vkGetDeviceProcAddr,
|
|
pub destroy_device: PFN_vkDestroyDevice,
|
|
pub get_device_queue: PFN_vkGetDeviceQueue,
|
|
pub queue_present_khr: Option<PFN_vkQueuePresentKHR>,
|
|
|
|
// Phase 1
|
|
pub create_shader_module: PFN_vkCreateShaderModule,
|
|
pub destroy_shader_module: PFN_vkDestroyShaderModule,
|
|
pub create_graphics_pipelines: PFN_vkCreateGraphicsPipelines,
|
|
pub destroy_pipeline: PFN_vkDestroyPipeline,
|
|
|
|
// Phase 2
|
|
pub create_image_view: PFN_vkCreateImageView,
|
|
pub destroy_image_view: PFN_vkDestroyImageView,
|
|
pub create_framebuffer: PFN_vkCreateFramebuffer,
|
|
pub destroy_framebuffer: PFN_vkDestroyFramebuffer,
|
|
pub allocate_command_buffers: PFN_vkAllocateCommandBuffers,
|
|
pub free_command_buffers: PFN_vkFreeCommandBuffers,
|
|
pub cmd_bind_pipeline: PFN_vkCmdBindPipeline,
|
|
pub cmd_begin_render_pass: PFN_vkCmdBeginRenderPass,
|
|
pub cmd_end_render_pass: PFN_vkCmdEndRenderPass,
|
|
pub cmd_begin_rendering_khr: Option<PFN_vkCmdBeginRenderingKHR>,
|
|
pub cmd_end_rendering_khr: Option<PFN_vkCmdEndRenderingKHR>,
|
|
|
|
// Phase 4 — capture images
|
|
pub create_image: PFN_vkCreateImage,
|
|
pub destroy_image: PFN_vkDestroyImage,
|
|
pub allocate_memory: PFN_vkAllocateMemory,
|
|
pub free_memory: PFN_vkFreeMemory,
|
|
pub bind_image_memory: PFN_vkBindImageMemory,
|
|
pub get_image_memory_requirements: PFN_vkGetImageMemoryRequirements,
|
|
pub map_memory: PFN_vkMapMemory,
|
|
pub unmap_memory: PFN_vkUnmapMemory,
|
|
pub cmd_pipeline_barrier: PFN_vkCmdPipelineBarrier,
|
|
pub cmd_copy_image: PFN_vkCmdCopyImage,
|
|
pub get_image_subresource_layout: PFN_vkGetImageSubresourceLayout,
|
|
/// `None` when `VK_KHR_external_memory_fd` is unavailable.
|
|
/// Required for DMA-BUF export to pixelforge's VkDevice.
|
|
pub get_memory_fd_khr: Option<PFN_vkGetMemoryFdKHR>,
|
|
|
|
// Phase 4 — synchronisation
|
|
pub create_fence: PFN_vkCreateFence,
|
|
pub destroy_fence: PFN_vkDestroyFence,
|
|
pub create_command_pool: PFN_vkCreateCommandPool,
|
|
pub destroy_command_pool: PFN_vkDestroyCommandPool,
|
|
pub reset_command_pool: PFN_vkResetCommandPool,
|
|
pub begin_command_buffer: PFN_vkBeginCommandBuffer,
|
|
pub end_command_buffer: PFN_vkEndCommandBuffer,
|
|
pub queue_submit: PFN_vkQueueSubmit,
|
|
pub wait_for_fences: PFN_vkWaitForFences,
|
|
pub reset_fences: PFN_vkResetFences,
|
|
|
|
// Phase 4 — swapchain
|
|
pub create_swapchain_khr: Option<PFN_vkCreateSwapchainKHR>,
|
|
pub destroy_swapchain_khr: Option<PFN_vkDestroySwapchainKHR>,
|
|
pub get_swapchain_images_khr: Option<PFN_vkGetSwapchainImagesKHR>,
|
|
|
|
// Phase 6 — draw commands
|
|
pub cmd_draw: PFN_vkCmdDraw,
|
|
pub cmd_draw_indexed: PFN_vkCmdDrawIndexed,
|
|
pub cmd_draw_indirect: PFN_vkCmdDrawIndirect,
|
|
pub cmd_draw_indexed_indirect: PFN_vkCmdDrawIndexedIndirect,
|
|
pub cmd_draw_indirect_count: Option<PFN_vkCmdDrawIndirectCount>,
|
|
pub cmd_draw_indexed_indirect_count: Option<PFN_vkCmdDrawIndexedIndirectCount>,
|
|
|
|
pub reset_command_buffer: PFN_vkResetCommandBuffer,
|
|
}
|