render pass begin/end

This commit is contained in:
Dzmitry Malyshau 2018-09-30 23:36:29 -04:00
parent 6b940bc446
commit 71b170979d
8 changed files with 127 additions and 66 deletions

View File

@ -51,9 +51,9 @@ impl<B: hal::Backend> CommandAllocator<B> {
return cmd_buf; return cmd_buf;
} }
for raw in pool.raw.allocate(20, hal::command::RawLevel::Primary) { for cmbuf in pool.raw.allocate(20, hal::command::RawLevel::Primary) {
pool.available.push(CommandBuffer { pool.available.push(CommandBuffer {
raw, raw: Some(cmbuf),
fence: device.create_fence(false), fence: device.create_fence(false),
recorded_thread_id: thread_id, recorded_thread_id: thread_id,
}); });
@ -66,7 +66,7 @@ impl<B: hal::Backend> CommandAllocator<B> {
} }
pub fn recycle(&self, mut cmd_buf: CommandBuffer<B>) { pub fn recycle(&self, mut cmd_buf: CommandBuffer<B>) {
cmd_buf.raw.reset(false); cmd_buf.raw.as_mut().unwrap().reset(false);
self.inner self.inner
.lock() .lock()
.unwrap() .unwrap()

View File

@ -12,6 +12,7 @@ use {
BufferId, Color, CommandBufferId, ComputePassId, Origin3d, RenderPassId, TextureId, BufferId, Color, CommandBufferId, ComputePassId, Origin3d, RenderPassId, TextureId,
TextureViewId, TextureViewId,
}; };
use registry::{self, Items, Registry};
use std::thread::ThreadId; use std::thread::ThreadId;
@ -71,7 +72,7 @@ pub struct TextureCopyView {
} }
pub struct CommandBuffer<B: hal::Backend> { pub struct CommandBuffer<B: hal::Backend> {
pub(crate) raw: B::CommandBuffer, pub(crate) raw: Option<B::CommandBuffer>,
fence: B::Fence, fence: B::Fence,
recorded_thread_id: ThreadId, recorded_thread_id: ThreadId,
} }
@ -81,9 +82,28 @@ pub struct CommandBufferDescriptor {}
#[no_mangle] #[no_mangle]
pub extern "C" fn wgpu_command_buffer_begin_render_pass( pub extern "C" fn wgpu_command_buffer_begin_render_pass(
_command_buffer: CommandBufferId, command_buffer_id: CommandBufferId,
_descriptor: RenderPassDescriptor,
) -> RenderPassId { ) -> RenderPassId {
unimplemented!() let raw = registry::COMMAND_BUFFER_REGISTRY
.lock()
.get_mut(command_buffer_id)
.raw
.take()
.unwrap();
/*TODO:
raw.begin_render_pass(
render_pass: &B::RenderPass,
framebuffer: &B::Framebuffer,
render_area: pso::Rect,
clear_values: T,
hal::SubpassContents::Inline,
);*/
registry::RENDER_PASS_REGISTRY
.lock()
.register(RenderPass::new(raw, command_buffer_id))
} }
#[no_mangle] #[no_mangle]

View File

@ -1,7 +1,39 @@
use hal; use hal;
use hal::command::RawCommandBuffer;
//use {CommandBuffer, CommandBufferId, RenderPassId}; use registry::{self, Items, Registry};
use {CommandBufferId, RenderPassId};
pub struct RenderPass<B: hal::Backend> { pub struct RenderPass<B: hal::Backend> {
raw: B::CommandBuffer, raw: B::CommandBuffer,
cmb_id: CommandBufferId,
}
// This is needed for `cmb_id` - would be great to remove.
#[cfg(not(feature = "remote"))]
unsafe impl<B: hal::Backend> Sync for RenderPass<B> {}
impl<B: hal::Backend> RenderPass<B> {
pub fn new(raw: B::CommandBuffer, cmb_id: CommandBufferId) -> Self {
RenderPass {
raw,
cmb_id,
}
}
}
#[no_mangle]
pub extern "C" fn wgpu_render_pass_end_pass(
render_pass_id: RenderPassId,
) -> CommandBufferId {
let mut rp = registry::RENDER_PASS_REGISTRY
.lock()
.take(render_pass_id);
rp.raw.end_render_pass();
registry::COMMAND_BUFFER_REGISTRY
.lock()
.get_mut(rp.cmb_id)
.raw = Some(rp.raw);
rp.cmb_id
} }

View File

@ -189,7 +189,7 @@ fn map_stencil_face(
hal::pso::StencilFace { hal::pso::StencilFace {
fun: map_compare_function(stencil_state_face_desc.compare), fun: map_compare_function(stencil_state_face_desc.compare),
mask_read: hal::pso::State::Static(stencil_read_mask), // TODO dynamic? mask_read: hal::pso::State::Static(stencil_read_mask), // TODO dynamic?
mask_write: hal::pso::State::Static(stencil_read_mask), // TODO dynamic? mask_write: hal::pso::State::Static(stencil_write_mask), // TODO dynamic?
op_fail: map_stencil_operation(stencil_state_face_desc.stencil_fail_op), op_fail: map_stencil_operation(stencil_state_face_desc.stencil_fail_op),
op_depth_fail: map_stencil_operation(stencil_state_face_desc.depth_fail_op), op_depth_fail: map_stencil_operation(stencil_state_face_desc.depth_fail_op),
op_pass: map_stencil_operation(stencil_state_face_desc.pass_op), op_pass: map_stencil_operation(stencil_state_face_desc.pass_op),

View File

@ -11,7 +11,7 @@ use {
}; };
pub struct Device<B: hal::Backend> { pub struct Device<B: hal::Backend> {
device: B::Device, raw: B::Device,
queue_group: hal::QueueGroup<B, hal::General>, queue_group: hal::QueueGroup<B, hal::General>,
mem_allocator: memory::SmartAllocator<B>, mem_allocator: memory::SmartAllocator<B>,
com_allocator: command::CommandAllocator<B>, com_allocator: command::CommandAllocator<B>,
@ -19,12 +19,12 @@ pub struct Device<B: hal::Backend> {
impl<B: hal::Backend> Device<B> { impl<B: hal::Backend> Device<B> {
pub(crate) fn new( pub(crate) fn new(
device: B::Device, raw: B::Device,
queue_group: hal::QueueGroup<B, hal::General>, queue_group: hal::QueueGroup<B, hal::General>,
mem_props: hal::MemoryProperties, mem_props: hal::MemoryProperties,
) -> Self { ) -> Self {
Device { Device {
device, raw,
mem_allocator: memory::SmartAllocator::new(mem_props, 1, 1, 1, 1), mem_allocator: memory::SmartAllocator::new(mem_props, 1, 1, 1, 1),
com_allocator: command::CommandAllocator::new(queue_group.family()), com_allocator: command::CommandAllocator::new(queue_group.family()),
queue_group, queue_group,
@ -44,7 +44,7 @@ pub extern "C" fn wgpu_device_create_bind_group_layout(
let bindings = unsafe { slice::from_raw_parts(desc.bindings, desc.bindings_length) }; let bindings = unsafe { slice::from_raw_parts(desc.bindings, desc.bindings_length) };
let device_guard = registry::DEVICE_REGISTRY.lock(); let device_guard = registry::DEVICE_REGISTRY.lock();
let device = device_guard.get(device_id); let device = device_guard.get(device_id);
let descriptor_set_layout = device.device.create_descriptor_set_layout( let descriptor_set_layout = device.raw.create_descriptor_set_layout(
bindings.iter().map(|binding| { bindings.iter().map(|binding| {
hal::pso::DescriptorSetLayoutBinding { hal::pso::DescriptorSetLayoutBinding {
binding: binding.binding, binding: binding.binding,
@ -75,7 +75,7 @@ pub extern "C" fn wgpu_device_create_pipeline_layout(
.map(|id| bind_group_layout_guard.get(id.clone())) .map(|id| bind_group_layout_guard.get(id.clone()))
.collect::<Vec<_>>(); .collect::<Vec<_>>();
let device_guard = registry::DEVICE_REGISTRY.lock(); let device_guard = registry::DEVICE_REGISTRY.lock();
let device = &device_guard.get(device_id).device; let device = &device_guard.get(device_id).raw;
let pipeline_layout = let pipeline_layout =
device.create_pipeline_layout(descriptor_set_layouts.iter().map(|d| &d.raw), &[]); // TODO: push constants device.create_pipeline_layout(descriptor_set_layouts.iter().map(|d| &d.raw), &[]); // TODO: push constants
registry::PIPELINE_LAYOUT_REGISTRY registry::PIPELINE_LAYOUT_REGISTRY
@ -99,7 +99,7 @@ pub extern "C" fn wgpu_device_create_blend_state(
#[no_mangle] #[no_mangle]
pub extern "C" fn wgpu_device_create_depth_stencil_state( pub extern "C" fn wgpu_device_create_depth_stencil_state(
device_id: DeviceId, _device_id: DeviceId,
desc: pipeline::DepthStencilStateDescriptor, desc: pipeline::DepthStencilStateDescriptor,
) -> DepthStencilStateId { ) -> DepthStencilStateId {
registry::DEPTH_STENCIL_STATE_REGISTRY registry::DEPTH_STENCIL_STATE_REGISTRY
@ -115,7 +115,7 @@ pub extern "C" fn wgpu_device_create_shader_module(
desc: pipeline::ShaderModuleDescriptor, desc: pipeline::ShaderModuleDescriptor,
) -> ShaderModuleId { ) -> ShaderModuleId {
let device_guard = registry::DEVICE_REGISTRY.lock(); let device_guard = registry::DEVICE_REGISTRY.lock();
let device = &device_guard.get(device_id).device; let device = &device_guard.get(device_id).raw;
let shader = device let shader = device
.create_shader_module(unsafe { slice::from_raw_parts(desc.code.bytes, desc.code.length) }) .create_shader_module(unsafe { slice::from_raw_parts(desc.code.bytes, desc.code.length) })
.unwrap(); .unwrap();
@ -131,8 +131,8 @@ pub extern "C" fn wgpu_device_create_command_buffer(
) -> CommandBufferId { ) -> CommandBufferId {
let mut device_guard = registry::DEVICE_REGISTRY.lock(); let mut device_guard = registry::DEVICE_REGISTRY.lock();
let device = device_guard.get_mut(device_id); let device = device_guard.get_mut(device_id);
let mut cmd_buf = device.com_allocator.allocate(&device.device); let mut cmd_buf = device.com_allocator.allocate(&device.raw);
cmd_buf.raw.begin( cmd_buf.raw.as_mut().unwrap().begin(
hal::command::CommandBufferFlags::ONE_TIME_SUBMIT, hal::command::CommandBufferFlags::ONE_TIME_SUBMIT,
hal::command::CommandBufferInheritanceInfo::default(), hal::command::CommandBufferInheritanceInfo::default(),
); );
@ -158,10 +158,11 @@ pub extern "C" fn wgpu_queue_submit(
let mut command_buffer_guard = registry::COMMAND_BUFFER_REGISTRY.lock(); let mut command_buffer_guard = registry::COMMAND_BUFFER_REGISTRY.lock();
for &cmb_id in command_buffer_ids { for &cmb_id in command_buffer_ids {
let mut cmd_buf = command_buffer_guard.take(cmb_id); let mut cmd_buf = command_buffer_guard.take(cmb_id);
cmd_buf.raw.finish();
{ {
let mut raw = cmd_buf.raw.as_mut().unwrap();
raw.finish();
let submission = hal::queue::RawSubmission { let submission = hal::queue::RawSubmission {
cmd_buffers: iter::once(&cmd_buf.raw), cmd_buffers: iter::once(raw),
wait_semaphores: &[], wait_semaphores: &[],
signal_semaphores: &[], signal_semaphores: &[],
}; };
@ -180,29 +181,52 @@ pub extern "C" fn wgpu_device_create_attachment_state(
device_id: DeviceId, device_id: DeviceId,
desc: pipeline::AttachmentStateDescriptor, desc: pipeline::AttachmentStateDescriptor,
) -> AttachmentStateId { ) -> AttachmentStateId {
// TODO: Assume that `AttachmentStateDescriptor` contains multiple attachments. let device_guard = registry::DEVICE_REGISTRY.lock();
let attachments = unsafe { slice::from_raw_parts(desc.formats, desc.formats_length) } let device = &device_guard.get(device_id).raw;
let color_formats = unsafe {
slice::from_raw_parts(desc.formats, desc.formats_length)
};
let color_formats: Vec<_> = color_formats
.iter() .iter()
.map(|format| { .cloned()
hal::pass::Attachment { .map(conv::map_texture_format)
format: Some(conv::map_texture_format(*format)), .collect();
samples: 1, // TODO map let depth_stencil_format = None;
ops: hal::pass::AttachmentOps {
// TODO map let base_pass = {
load: hal::pass::AttachmentLoadOp::Clear, let attachments = color_formats.iter().map(|cf| hal::pass::Attachment {
store: hal::pass::AttachmentStoreOp::Store, format: Some(*cf),
}, samples: 1,
stencil_ops: hal::pass::AttachmentOps { ops: hal::pass::AttachmentOps::DONT_CARE,
// TODO map stencil_ops: hal::pass::AttachmentOps::DONT_CARE,
load: hal::pass::AttachmentLoadOp::DontCare, layouts: hal::image::Layout::General .. hal::image::Layout::General,
store: hal::pass::AttachmentStoreOp::DontCare, });
},
layouts: hal::image::Layout::Undefined..hal::image::Layout::Present, // TODO map let subpass = hal::pass::SubpassDesc {
} colors: &[(0, hal::image::Layout::ColorAttachmentOptimal)],
}).collect(); depth_stencil: None,
inputs: &[],
resolves: &[],
preserves: &[],
};
device.create_render_pass(
attachments,
&[subpass],
&[],
)
};
let at_state = pipeline::AttachmentState {
base_pass,
color_formats,
depth_stencil_format,
};
registry::ATTACHMENT_STATE_REGISTRY registry::ATTACHMENT_STATE_REGISTRY
.lock() .lock()
.register(pipeline::AttachmentState { raw: attachments }) .register(at_state)
} }
#[no_mangle] #[no_mangle]
@ -217,7 +241,7 @@ pub extern "C" fn wgpu_device_create_render_pipeline(
}; };
let device_guard = registry::DEVICE_REGISTRY.lock(); let device_guard = registry::DEVICE_REGISTRY.lock();
let device = &device_guard.get(device_id).device; let device = &device_guard.get(device_id).raw;
let pipeline_layout_guard = registry::PIPELINE_LAYOUT_REGISTRY.lock(); let pipeline_layout_guard = registry::PIPELINE_LAYOUT_REGISTRY.lock();
let layout = &pipeline_layout_guard.get(desc.layout).raw; let layout = &pipeline_layout_guard.get(desc.layout).raw;
let pipeline_stages = unsafe { slice::from_raw_parts(desc.stages, desc.stages_length) }; let pipeline_stages = unsafe { slice::from_raw_parts(desc.stages, desc.stages_length) };
@ -318,33 +342,12 @@ pub extern "C" fn wgpu_device_create_render_pipeline(
}; };
let attachment_state_guard = registry::ATTACHMENT_STATE_REGISTRY.lock(); let attachment_state_guard = registry::ATTACHMENT_STATE_REGISTRY.lock();
let attachments = &attachment_state_guard.get(desc.attachment_state).raw; let attachment_state = attachment_state_guard.get(desc.attachment_state);
// TODO
let subpass = hal::pass::SubpassDesc {
colors: &[(0, hal::image::Layout::ColorAttachmentOptimal)],
depth_stencil: None,
inputs: &[],
resolves: &[],
preserves: &[],
};
// TODO
let subpass_dependency = hal::pass::SubpassDependency {
passes: hal::pass::SubpassRef::External..hal::pass::SubpassRef::Pass(0),
stages: hal::pso::PipelineStage::COLOR_ATTACHMENT_OUTPUT
..hal::pso::PipelineStage::COLOR_ATTACHMENT_OUTPUT,
accesses: hal::image::Access::empty()
..(hal::image::Access::COLOR_ATTACHMENT_READ
| hal::image::Access::COLOR_ATTACHMENT_WRITE),
};
let main_pass = &device.create_render_pass(&attachments[..], &[subpass], &[subpass_dependency]);
// TODO // TODO
let subpass = hal::pass::Subpass { let subpass = hal::pass::Subpass {
index: 0, index: 0,
main_pass, main_pass: &attachment_state.base_pass,
}; };
// TODO // TODO

View File

@ -97,7 +97,7 @@ pub type InputStateId = Id;
pub type ShaderModuleId = Id; pub type ShaderModuleId = Id;
type ShaderModuleHandle = ShaderModule<B>; type ShaderModuleHandle = ShaderModule<B>;
pub type AttachmentStateId = Id; pub type AttachmentStateId = Id;
type AttachmentStateHandle = AttachmentState; type AttachmentStateHandle = AttachmentState<B>;
pub type ComputePipelineId = Id; pub type ComputePipelineId = Id;
pub type RenderPipelineId = Id; pub type RenderPipelineId = Id;
type RenderPipelineHandle = RenderPipeline<B>; type RenderPipelineHandle = RenderPipeline<B>;
@ -105,4 +105,5 @@ type RenderPipelineHandle = RenderPipeline<B>;
pub type CommandBufferId = Id; pub type CommandBufferId = Id;
type CommandBufferHandle = CommandBuffer<B>; type CommandBufferHandle = CommandBuffer<B>;
pub type RenderPassId = Id; pub type RenderPassId = Id;
type RenderPassHandle = RenderPass<B>;
pub type ComputePassId = Id; pub type ComputePassId = Id;

View File

@ -200,8 +200,10 @@ pub struct AttachmentStateDescriptor {
pub formats_length: usize, pub formats_length: usize,
} }
pub(crate) struct AttachmentState { pub(crate) struct AttachmentState<B: hal::Backend> {
pub raw: Vec<hal::pass::Attachment>, pub base_pass: B::RenderPass,
pub color_formats: Vec<hal::format::Format>,
pub depth_stencil_format: Option<hal::format::Format>,
} }
#[repr(C)] #[repr(C)]

View File

@ -12,6 +12,7 @@ use std::sync::Arc;
use { use {
AdapterHandle, AttachmentStateHandle, BindGroupLayoutHandle, BlendStateHandle, AdapterHandle, AttachmentStateHandle, BindGroupLayoutHandle, BlendStateHandle,
CommandBufferHandle, DepthStencilStateHandle, DeviceHandle, InstanceHandle, CommandBufferHandle, DepthStencilStateHandle, DeviceHandle, InstanceHandle,
RenderPassHandle,
PipelineLayoutHandle, RenderPipelineHandle, ShaderModuleHandle, PipelineLayoutHandle, RenderPipelineHandle, ShaderModuleHandle,
}; };
@ -175,4 +176,6 @@ lazy_static! {
ConcreteRegistry::new(); ConcreteRegistry::new();
pub(crate) static ref SHADER_MODULE_REGISTRY: ConcreteRegistry<ShaderModuleHandle> = pub(crate) static ref SHADER_MODULE_REGISTRY: ConcreteRegistry<ShaderModuleHandle> =
ConcreteRegistry::new(); ConcreteRegistry::new();
pub(crate) static ref RENDER_PASS_REGISTRY: ConcreteRegistry<RenderPassHandle> =
ConcreteRegistry::new();
} }