mirror of
https://github.com/gfx-rs/wgpu.git
synced 2026-04-22 03:02:01 -04:00
879 lines
34 KiB
Rust
879 lines
34 KiB
Rust
/* This Source Code Form is subject to the terms of the Mozilla Public
|
|
* License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
|
|
|
|
#[cfg(feature = "trace")]
|
|
use crate::device::trace::Action;
|
|
use crate::{
|
|
command::{
|
|
extract_texture_selector, validate_linear_texture_data, validate_texture_copy_range,
|
|
CommandBuffer, CopySide, ImageCopyTexture, TransferError,
|
|
},
|
|
conv,
|
|
device::{DeviceError, WaitIdleError},
|
|
hub::{Global, GlobalIdentityHandlerFactory, HalApi, Storage, Token},
|
|
id,
|
|
memory_init_tracker::{MemoryInitKind, MemoryInitTrackerAction},
|
|
resource::{Buffer, BufferAccessError, BufferMapState},
|
|
FastHashMap, FastHashSet,
|
|
};
|
|
|
|
use hal::{CommandEncoder as _, Device as _, Queue as _};
|
|
use parking_lot::Mutex;
|
|
use smallvec::SmallVec;
|
|
use std::{iter, mem, num::NonZeroU32, ops::Range, ptr};
|
|
use thiserror::Error;
|
|
|
|
/// Number of command buffers that we generate from the same pool
|
|
/// for the write_xxx commands, before the pool is recycled.
|
|
///
|
|
/// If we don't stop at some point, the pool will grow forever,
|
|
/// without a concrete moment of when it can be cleared.
|
|
const WRITE_COMMAND_BUFFERS_PER_POOL: usize = 64;
|
|
|
|
struct StagingData<A: hal::Api> {
|
|
buffer: A::Buffer,
|
|
}
|
|
|
|
impl<A: hal::Api> StagingData<A> {
|
|
unsafe fn write(
|
|
&self,
|
|
device: &A::Device,
|
|
offset: wgt::BufferAddress,
|
|
data: &[u8],
|
|
) -> Result<(), hal::DeviceError> {
|
|
let mapping = device.map_buffer(&self.buffer, offset..offset + data.len() as u64)?;
|
|
ptr::copy_nonoverlapping(data.as_ptr(), mapping.ptr.as_ptr(), data.len());
|
|
if !mapping.is_coherent {
|
|
device
|
|
.flush_mapped_ranges(&self.buffer, iter::once(offset..offset + data.len() as u64));
|
|
}
|
|
device.unmap_buffer(&self.buffer)?;
|
|
Ok(())
|
|
}
|
|
}
|
|
|
|
#[derive(Debug)]
|
|
pub enum TempResource<A: hal::Api> {
|
|
Buffer(A::Buffer),
|
|
Texture(A::Texture),
|
|
}
|
|
|
|
/// A queue execution for a particular command encoder.
|
|
pub(super) struct EncoderInFlight<A: hal::Api> {
|
|
raw: A::CommandEncoder,
|
|
cmd_buffers: Vec<A::CommandBuffer>,
|
|
}
|
|
|
|
impl<A: hal::Api> EncoderInFlight<A> {
|
|
pub(super) unsafe fn land(mut self) -> A::CommandEncoder {
|
|
self.raw.reset_all(self.cmd_buffers.into_iter());
|
|
self.raw
|
|
}
|
|
}
|
|
|
|
#[derive(Debug)]
|
|
pub(crate) struct PendingWrites<A: hal::Api> {
|
|
pub command_encoder: A::CommandEncoder,
|
|
pub is_active: bool,
|
|
pub temp_resources: Vec<TempResource<A>>,
|
|
pub dst_buffers: FastHashSet<id::BufferId>,
|
|
pub dst_textures: FastHashSet<id::TextureId>,
|
|
pub executing_command_buffers: Vec<A::CommandBuffer>,
|
|
}
|
|
|
|
impl<A: hal::Api> PendingWrites<A> {
|
|
pub fn new(command_encoder: A::CommandEncoder) -> Self {
|
|
Self {
|
|
command_encoder,
|
|
is_active: false,
|
|
temp_resources: Vec::new(),
|
|
dst_buffers: FastHashSet::default(),
|
|
dst_textures: FastHashSet::default(),
|
|
executing_command_buffers: Vec::new(),
|
|
}
|
|
}
|
|
|
|
pub fn dispose(mut self, device: &A::Device) {
|
|
unsafe {
|
|
if self.is_active {
|
|
self.command_encoder.discard_encoding();
|
|
}
|
|
self.command_encoder
|
|
.reset_all(self.executing_command_buffers.into_iter());
|
|
device.destroy_command_encoder(self.command_encoder);
|
|
}
|
|
|
|
for resource in self.temp_resources {
|
|
match resource {
|
|
TempResource::Buffer(buffer) => unsafe {
|
|
device.destroy_buffer(buffer);
|
|
},
|
|
TempResource::Texture(texture) => unsafe {
|
|
device.destroy_texture(texture);
|
|
},
|
|
}
|
|
}
|
|
}
|
|
|
|
pub fn consume_temp(&mut self, resource: TempResource<A>) {
|
|
self.temp_resources.push(resource);
|
|
}
|
|
|
|
fn consume(&mut self, stage: StagingData<A>) {
|
|
self.temp_resources.push(TempResource::Buffer(stage.buffer));
|
|
}
|
|
|
|
#[must_use]
|
|
fn pre_submit(&mut self) -> Option<&A::CommandBuffer> {
|
|
self.dst_buffers.clear();
|
|
self.dst_textures.clear();
|
|
if self.is_active {
|
|
let cmd_buf = unsafe { self.command_encoder.end_encoding().unwrap() };
|
|
self.is_active = false;
|
|
self.executing_command_buffers.push(cmd_buf);
|
|
self.executing_command_buffers.last()
|
|
} else {
|
|
None
|
|
}
|
|
}
|
|
|
|
#[must_use]
|
|
fn post_submit(
|
|
&mut self,
|
|
command_allocator: &Mutex<super::CommandAllocator<A>>,
|
|
device: &A::Device,
|
|
queue: &A::Queue,
|
|
) -> Option<EncoderInFlight<A>> {
|
|
if self.executing_command_buffers.len() >= WRITE_COMMAND_BUFFERS_PER_POOL {
|
|
let new_encoder = command_allocator
|
|
.lock()
|
|
.acquire_encoder(device, queue)
|
|
.unwrap();
|
|
Some(EncoderInFlight {
|
|
raw: mem::replace(&mut self.command_encoder, new_encoder),
|
|
cmd_buffers: mem::take(&mut self.executing_command_buffers),
|
|
})
|
|
} else {
|
|
None
|
|
}
|
|
}
|
|
|
|
pub fn activate(&mut self) -> &mut A::CommandEncoder {
|
|
if !self.is_active {
|
|
unsafe {
|
|
self.command_encoder
|
|
.begin_encoding(Some("_PendingWrites"))
|
|
.unwrap();
|
|
}
|
|
self.is_active = true;
|
|
}
|
|
&mut self.command_encoder
|
|
}
|
|
}
|
|
|
|
#[derive(Default)]
|
|
struct RequiredBufferInits {
|
|
map: FastHashMap<id::BufferId, Vec<Range<wgt::BufferAddress>>>,
|
|
}
|
|
|
|
impl RequiredBufferInits {
|
|
fn add<A: hal::Api>(
|
|
&mut self,
|
|
buffer_memory_init_actions: &[MemoryInitTrackerAction<id::BufferId>],
|
|
buffer_guard: &mut Storage<Buffer<A>, id::BufferId>,
|
|
) -> Result<(), QueueSubmitError> {
|
|
for buffer_use in buffer_memory_init_actions.iter() {
|
|
let buffer = buffer_guard
|
|
.get_mut(buffer_use.id)
|
|
.map_err(|_| QueueSubmitError::DestroyedBuffer(buffer_use.id))?;
|
|
|
|
let uninitialized_ranges = buffer.initialization_status.drain(buffer_use.range.clone());
|
|
match buffer_use.kind {
|
|
MemoryInitKind::ImplicitlyInitialized => {
|
|
uninitialized_ranges.for_each(drop);
|
|
}
|
|
MemoryInitKind::NeedsInitializedMemory => {
|
|
self.map
|
|
.entry(buffer_use.id)
|
|
.or_default()
|
|
.extend(uninitialized_ranges);
|
|
}
|
|
}
|
|
}
|
|
Ok(())
|
|
}
|
|
}
|
|
|
|
impl<A: hal::Api> super::Device<A> {
|
|
fn prepare_stage(&mut self, size: wgt::BufferAddress) -> Result<StagingData<A>, DeviceError> {
|
|
profiling::scope!("prepare_stage");
|
|
let stage_desc = hal::BufferDescriptor {
|
|
label: Some("_Staging"),
|
|
size,
|
|
usage: hal::BufferUse::MAP_WRITE | hal::BufferUse::COPY_SRC,
|
|
memory_flags: hal::MemoryFlag::TRANSIENT,
|
|
};
|
|
let buffer = unsafe { self.raw.create_buffer(&stage_desc)? };
|
|
Ok(StagingData { buffer })
|
|
}
|
|
|
|
fn initialize_buffer_memory(
|
|
&mut self,
|
|
mut required_buffer_inits: RequiredBufferInits,
|
|
buffer_guard: &mut Storage<Buffer<A>, id::BufferId>,
|
|
) -> Result<(), QueueSubmitError> {
|
|
self.pending_writes
|
|
.dst_buffers
|
|
.extend(required_buffer_inits.map.keys());
|
|
|
|
let encoder = self.pending_writes.activate();
|
|
let mut trackers = self.trackers.lock();
|
|
|
|
for (buffer_id, mut ranges) in required_buffer_inits.map.drain() {
|
|
// Collapse touching ranges. We can't do this any earlier since we only now gathered ranges from several different command buffers!
|
|
ranges.sort_by(|a, b| a.start.cmp(&b.start));
|
|
for i in (1..ranges.len()).rev() {
|
|
assert!(ranges[i - 1].end <= ranges[i].start); // The memory init tracker made sure of this!
|
|
if ranges[i].start == ranges[i - 1].end {
|
|
ranges[i - 1].end = ranges[i].end;
|
|
ranges.swap_remove(i); // Ordering not important at this point
|
|
}
|
|
}
|
|
|
|
// Don't do use_replace since the buffer may already no longer have a ref_count.
|
|
// However, we *know* that it is currently in use, so the tracker must already know about it.
|
|
let transition = trackers.buffers.change_replace_tracked(
|
|
id::Valid(buffer_id),
|
|
(),
|
|
hal::BufferUse::COPY_DST,
|
|
);
|
|
let buffer = buffer_guard.get(buffer_id).unwrap();
|
|
let raw_buf = buffer
|
|
.raw
|
|
.as_ref()
|
|
.ok_or(QueueSubmitError::DestroyedBuffer(buffer_id))?;
|
|
unsafe {
|
|
encoder.transition_buffers(transition.map(|pending| pending.into_hal(buffer)));
|
|
}
|
|
|
|
for range in ranges {
|
|
assert!(range.start % 4 == 0, "Buffer {:?} has an uninitialized range with a start not aligned to 4 (start was {})", raw_buf, range.start);
|
|
assert!(range.end % 4 == 0, "Buffer {:?} has an uninitialized range with an end not aligned to 4 (end was {})", raw_buf, range.end);
|
|
|
|
unsafe {
|
|
encoder.fill_buffer(raw_buf, range, 0);
|
|
}
|
|
}
|
|
}
|
|
|
|
Ok(())
|
|
}
|
|
}
|
|
|
|
#[derive(Clone, Debug, Error)]
|
|
#[error("queue is invalid")]
|
|
pub struct InvalidQueue;
|
|
|
|
#[derive(Clone, Debug, Error)]
|
|
pub enum QueueWriteError {
|
|
#[error(transparent)]
|
|
Queue(#[from] DeviceError),
|
|
#[error(transparent)]
|
|
Transfer(#[from] TransferError),
|
|
}
|
|
|
|
#[derive(Clone, Debug, Error)]
|
|
pub enum QueueSubmitError {
|
|
#[error(transparent)]
|
|
Queue(#[from] DeviceError),
|
|
#[error("buffer {0:?} is destroyed")]
|
|
DestroyedBuffer(id::BufferId),
|
|
#[error("texture {0:?} is destroyed")]
|
|
DestroyedTexture(id::TextureId),
|
|
#[error(transparent)]
|
|
Unmap(#[from] BufferAccessError),
|
|
#[error("swap chain output was dropped before the command buffer got submitted")]
|
|
SwapChainOutputDropped,
|
|
#[error("GPU got stuck :(")]
|
|
StuckGpu,
|
|
}
|
|
|
|
//TODO: move out common parts of write_xxx.
|
|
|
|
impl<G: GlobalIdentityHandlerFactory> Global<G> {
|
|
pub fn queue_write_buffer<A: HalApi>(
|
|
&self,
|
|
queue_id: id::QueueId,
|
|
buffer_id: id::BufferId,
|
|
buffer_offset: wgt::BufferAddress,
|
|
data: &[u8],
|
|
) -> Result<(), QueueWriteError> {
|
|
profiling::scope!("write_buffer", "Queue");
|
|
|
|
let hub = A::hub(self);
|
|
let mut token = Token::root();
|
|
let (mut device_guard, mut token) = hub.devices.write(&mut token);
|
|
let device = device_guard
|
|
.get_mut(queue_id)
|
|
.map_err(|_| DeviceError::Invalid)?;
|
|
let (buffer_guard, _) = hub.buffers.read(&mut token);
|
|
|
|
#[cfg(feature = "trace")]
|
|
if let Some(ref trace) = device.trace {
|
|
let mut trace = trace.lock();
|
|
let data_path = trace.make_binary("bin", data);
|
|
trace.add(Action::WriteBuffer {
|
|
id: buffer_id,
|
|
data: data_path,
|
|
range: buffer_offset..buffer_offset + data.len() as wgt::BufferAddress,
|
|
queued: true,
|
|
});
|
|
}
|
|
|
|
let data_size = data.len() as wgt::BufferAddress;
|
|
if data_size == 0 {
|
|
log::trace!("Ignoring write_buffer of size 0");
|
|
return Ok(());
|
|
}
|
|
|
|
let stage = device.prepare_stage(data_size)?;
|
|
unsafe { stage.write(&device.raw, 0, data) }.map_err(DeviceError::from)?;
|
|
|
|
let mut trackers = device.trackers.lock();
|
|
let (dst, transition) = trackers
|
|
.buffers
|
|
.use_replace(&*buffer_guard, buffer_id, (), hal::BufferUse::COPY_DST)
|
|
.map_err(TransferError::InvalidBuffer)?;
|
|
let dst_raw = dst
|
|
.raw
|
|
.as_ref()
|
|
.ok_or(TransferError::InvalidBuffer(buffer_id))?;
|
|
if !dst.usage.contains(wgt::BufferUsage::COPY_DST) {
|
|
return Err(TransferError::MissingCopyDstUsageFlag(Some(buffer_id), None).into());
|
|
}
|
|
dst.life_guard.use_at(device.active_submission_index + 1);
|
|
|
|
if data_size % wgt::COPY_BUFFER_ALIGNMENT != 0 {
|
|
return Err(TransferError::UnalignedCopySize(data_size).into());
|
|
}
|
|
if buffer_offset % wgt::COPY_BUFFER_ALIGNMENT != 0 {
|
|
return Err(TransferError::UnalignedBufferOffset(buffer_offset).into());
|
|
}
|
|
if buffer_offset + data_size > dst.size {
|
|
return Err(TransferError::BufferOverrun {
|
|
start_offset: buffer_offset,
|
|
end_offset: buffer_offset + data_size,
|
|
buffer_size: dst.size,
|
|
side: CopySide::Destination,
|
|
}
|
|
.into());
|
|
}
|
|
|
|
let region = wgt::BufferSize::new(data.len() as u64).map(|size| hal::BufferCopy {
|
|
src_offset: 0,
|
|
dst_offset: buffer_offset,
|
|
size,
|
|
});
|
|
let barriers = iter::once(hal::BufferBarrier {
|
|
buffer: &stage.buffer,
|
|
usage: hal::BufferUse::MAP_WRITE..hal::BufferUse::COPY_SRC,
|
|
})
|
|
.chain(transition.map(|pending| pending.into_hal(dst)));
|
|
let encoder = device.pending_writes.activate();
|
|
unsafe {
|
|
encoder.transition_buffers(barriers);
|
|
encoder.copy_buffer_to_buffer(&stage.buffer, dst_raw, region.into_iter());
|
|
}
|
|
|
|
device.pending_writes.consume(stage);
|
|
device.pending_writes.dst_buffers.insert(buffer_id);
|
|
|
|
// Ensure the overwritten bytes are marked as initialized so they don't need to be nulled prior to mapping or binding.
|
|
{
|
|
drop(buffer_guard);
|
|
let (mut buffer_guard, _) = hub.buffers.write(&mut token);
|
|
|
|
let dst = buffer_guard.get_mut(buffer_id).unwrap();
|
|
dst.initialization_status
|
|
.clear(buffer_offset..(buffer_offset + data_size));
|
|
}
|
|
|
|
Ok(())
|
|
}
|
|
|
|
pub fn queue_write_texture<A: HalApi>(
|
|
&self,
|
|
queue_id: id::QueueId,
|
|
destination: &ImageCopyTexture,
|
|
data: &[u8],
|
|
data_layout: &wgt::ImageDataLayout,
|
|
size: &wgt::Extent3d,
|
|
) -> Result<(), QueueWriteError> {
|
|
profiling::scope!("write_texture", "Queue");
|
|
|
|
let hub = A::hub(self);
|
|
let mut token = Token::root();
|
|
let (mut device_guard, mut token) = hub.devices.write(&mut token);
|
|
let device = device_guard
|
|
.get_mut(queue_id)
|
|
.map_err(|_| DeviceError::Invalid)?;
|
|
|
|
#[cfg(feature = "trace")]
|
|
if let Some(ref trace) = device.trace {
|
|
let mut trace = trace.lock();
|
|
let data_path = trace.make_binary("bin", data);
|
|
trace.add(Action::WriteTexture {
|
|
to: destination.clone(),
|
|
data: data_path,
|
|
layout: *data_layout,
|
|
size: *size,
|
|
});
|
|
}
|
|
|
|
if size.width == 0 || size.height == 0 || size.depth_or_array_layers == 0 {
|
|
log::trace!("Ignoring write_texture of size 0");
|
|
return Ok(());
|
|
}
|
|
|
|
let (texture_guard, _) = hub.textures.read(&mut token);
|
|
let (selector, texture_base, texture_format) =
|
|
extract_texture_selector(destination, size, &*texture_guard)?;
|
|
let format_desc = texture_format.describe();
|
|
validate_linear_texture_data(
|
|
data_layout,
|
|
texture_format,
|
|
data.len() as wgt::BufferAddress,
|
|
CopySide::Source,
|
|
format_desc.block_size as wgt::BufferAddress,
|
|
size,
|
|
false,
|
|
)?;
|
|
|
|
if !conv::is_valid_copy_dst_texture_format(texture_format) {
|
|
return Err(TransferError::CopyToForbiddenTextureFormat(texture_format).into());
|
|
}
|
|
let (block_width, block_height) = format_desc.block_dimensions;
|
|
let width_blocks = size.width / block_width as u32;
|
|
let height_blocks = size.height / block_height as u32;
|
|
|
|
let block_rows_per_image = match data_layout.rows_per_image {
|
|
Some(rows_per_image) => rows_per_image.get(),
|
|
None => {
|
|
// doesn't really matter because we need this only if we copy more than one layer, and then we validate for this being not None
|
|
size.height
|
|
}
|
|
};
|
|
|
|
let bytes_per_row_alignment = get_lowest_common_denom(
|
|
device.alignments.buffer_copy_pitch.get() as u32,
|
|
format_desc.block_size as u32,
|
|
);
|
|
let stage_bytes_per_row = align_to(
|
|
format_desc.block_size as u32 * width_blocks,
|
|
bytes_per_row_alignment,
|
|
);
|
|
|
|
let block_rows_in_copy =
|
|
(size.depth_or_array_layers - 1) * block_rows_per_image + height_blocks;
|
|
let stage_size = stage_bytes_per_row as u64 * block_rows_in_copy as u64;
|
|
let stage = device.prepare_stage(stage_size)?;
|
|
|
|
let mut trackers = device.trackers.lock();
|
|
let (dst, transition) = trackers
|
|
.textures
|
|
.use_replace(
|
|
&*texture_guard,
|
|
destination.texture,
|
|
selector,
|
|
hal::TextureUse::COPY_DST,
|
|
)
|
|
.unwrap();
|
|
let dst_raw = dst
|
|
.raw
|
|
.as_ref()
|
|
.ok_or(TransferError::InvalidTexture(destination.texture))?;
|
|
|
|
if !dst.desc.usage.contains(wgt::TextureUsage::COPY_DST) {
|
|
return Err(
|
|
TransferError::MissingCopyDstUsageFlag(None, Some(destination.texture)).into(),
|
|
);
|
|
}
|
|
let max_image_extent =
|
|
validate_texture_copy_range(destination, &dst.desc, CopySide::Destination, size)?;
|
|
dst.life_guard.use_at(device.active_submission_index + 1);
|
|
|
|
let bytes_per_row = if let Some(bytes_per_row) = data_layout.bytes_per_row {
|
|
bytes_per_row.get()
|
|
} else {
|
|
width_blocks * format_desc.block_size as u32
|
|
};
|
|
|
|
let mapping = unsafe { device.raw.map_buffer(&stage.buffer, 0..stage_size) }
|
|
.map_err(DeviceError::from)?;
|
|
unsafe {
|
|
profiling::scope!("copy");
|
|
if stage_bytes_per_row == bytes_per_row {
|
|
// Fast path if the data isalready being aligned optimally.
|
|
ptr::copy_nonoverlapping(data.as_ptr(), mapping.ptr.as_ptr(), stage_size as usize);
|
|
} else {
|
|
// Copy row by row into the optimal alignment.
|
|
let copy_bytes_per_row = stage_bytes_per_row.min(bytes_per_row) as usize;
|
|
for layer in 0..size.depth_or_array_layers {
|
|
let rows_offset = layer * block_rows_per_image;
|
|
for row in 0..height_blocks {
|
|
ptr::copy_nonoverlapping(
|
|
data.as_ptr()
|
|
.offset((rows_offset + row) as isize * bytes_per_row as isize),
|
|
mapping.ptr.as_ptr().offset(
|
|
(rows_offset + row) as isize * stage_bytes_per_row as isize,
|
|
),
|
|
copy_bytes_per_row,
|
|
);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
unsafe {
|
|
if !mapping.is_coherent {
|
|
device
|
|
.raw
|
|
.flush_mapped_ranges(&stage.buffer, iter::once(0..stage_size));
|
|
}
|
|
device
|
|
.raw
|
|
.unmap_buffer(&stage.buffer)
|
|
.map_err(DeviceError::from)?;
|
|
}
|
|
|
|
// WebGPU uses the physical size of the texture for copies whereas vulkan uses
|
|
// the virtual size. We have passed validation, so it's safe to use the
|
|
// image extent data directly. We want the provided copy size to be no larger than
|
|
// the virtual size.
|
|
let region = hal::BufferTextureCopy {
|
|
buffer_layout: wgt::ImageDataLayout {
|
|
offset: 0,
|
|
bytes_per_row: NonZeroU32::new(stage_bytes_per_row),
|
|
rows_per_image: NonZeroU32::new(block_rows_per_image),
|
|
},
|
|
texture_base,
|
|
size: wgt::Extent3d {
|
|
width: size.width.min(max_image_extent.width),
|
|
height: size.height.min(max_image_extent.height),
|
|
depth_or_array_layers: size.depth_or_array_layers,
|
|
},
|
|
};
|
|
|
|
let barrier = hal::BufferBarrier {
|
|
buffer: &stage.buffer,
|
|
usage: hal::BufferUse::MAP_WRITE..hal::BufferUse::COPY_SRC,
|
|
};
|
|
let encoder = device.pending_writes.activate();
|
|
unsafe {
|
|
encoder.transition_buffers(iter::once(barrier));
|
|
encoder.transition_textures(transition.map(|pending| pending.into_hal(dst)));
|
|
encoder.copy_buffer_to_texture(&stage.buffer, dst_raw, iter::once(region));
|
|
}
|
|
|
|
device.pending_writes.consume(stage);
|
|
device
|
|
.pending_writes
|
|
.dst_textures
|
|
.insert(destination.texture);
|
|
|
|
Ok(())
|
|
}
|
|
|
|
pub fn queue_submit<A: HalApi>(
|
|
&self,
|
|
queue_id: id::QueueId,
|
|
command_buffer_ids: &[id::CommandBufferId],
|
|
) -> Result<(), QueueSubmitError> {
|
|
profiling::scope!("submit", "Queue");
|
|
|
|
let hub = A::hub(self);
|
|
let mut token = Token::root();
|
|
|
|
let callbacks = {
|
|
let (mut device_guard, mut token) = hub.devices.write(&mut token);
|
|
let device = device_guard
|
|
.get_mut(queue_id)
|
|
.map_err(|_| DeviceError::Invalid)?;
|
|
device.temp_suspected.clear();
|
|
device.active_submission_index += 1;
|
|
let submit_index = device.active_submission_index;
|
|
let mut active_executions = Vec::new();
|
|
|
|
{
|
|
let mut signal_swapchain_semaphores = SmallVec::<[_; 1]>::new();
|
|
let (mut swap_chain_guard, mut token) = hub.swap_chains.write(&mut token);
|
|
let (mut command_buffer_guard, mut token) = hub.command_buffers.write(&mut token);
|
|
|
|
if !command_buffer_ids.is_empty() {
|
|
profiling::scope!("prepare");
|
|
|
|
let (render_bundle_guard, mut token) = hub.render_bundles.read(&mut token);
|
|
let (_, mut token) = hub.pipeline_layouts.read(&mut token);
|
|
let (bind_group_guard, mut token) = hub.bind_groups.read(&mut token);
|
|
let (compute_pipe_guard, mut token) = hub.compute_pipelines.read(&mut token);
|
|
let (render_pipe_guard, mut token) = hub.render_pipelines.read(&mut token);
|
|
let (mut buffer_guard, mut token) = hub.buffers.write(&mut token);
|
|
let (texture_guard, mut token) = hub.textures.write(&mut token);
|
|
let (texture_view_guard, mut token) = hub.texture_views.read(&mut token);
|
|
let (sampler_guard, _) = hub.samplers.read(&mut token);
|
|
|
|
let mut required_buffer_inits = RequiredBufferInits::default();
|
|
//Note: locking the trackers has to be done after the storages
|
|
let mut trackers = device.trackers.lock();
|
|
|
|
//TODO: if multiple command buffers are submitted, we can re-use the last
|
|
// native command buffer of the previous chain instead of always creating
|
|
// a temporary one, since the chains are not finished.
|
|
|
|
// finish all the command buffers first
|
|
for &cmb_id in command_buffer_ids {
|
|
let mut cmdbuf = match hub
|
|
.command_buffers
|
|
.unregister_locked(cmb_id, &mut *command_buffer_guard)
|
|
{
|
|
Some(cmdbuf) => cmdbuf,
|
|
None => continue,
|
|
};
|
|
#[cfg(feature = "trace")]
|
|
if let Some(ref trace) = device.trace {
|
|
trace.lock().add(Action::Submit(
|
|
submit_index,
|
|
cmdbuf.commands.take().unwrap(),
|
|
));
|
|
}
|
|
if !cmdbuf.is_finished() {
|
|
device.destroy_command_buffer(cmdbuf);
|
|
continue;
|
|
}
|
|
|
|
required_buffer_inits
|
|
.add(&cmdbuf.buffer_memory_init_actions, &mut *buffer_guard)?;
|
|
// optimize the tracked states
|
|
cmdbuf.trackers.optimize();
|
|
|
|
for sc_id in cmdbuf.used_swap_chains.drain(..) {
|
|
let sc = &mut swap_chain_guard[sc_id.value];
|
|
if sc.acquired_texture.is_none() {
|
|
return Err(QueueSubmitError::SwapChainOutputDropped);
|
|
}
|
|
if sc.active_submission_index != submit_index {
|
|
sc.active_submission_index = submit_index;
|
|
// Only add a signal if this is the first time for this swapchain
|
|
// to be used in the submission.
|
|
signal_swapchain_semaphores.push(sc_id.value);
|
|
}
|
|
}
|
|
|
|
// update submission IDs
|
|
for id in cmdbuf.trackers.buffers.used() {
|
|
let buffer = &mut buffer_guard[id];
|
|
let raw_buf = match buffer.raw {
|
|
Some(ref raw) => raw,
|
|
None => {
|
|
return Err(QueueSubmitError::DestroyedBuffer(id.0));
|
|
}
|
|
};
|
|
if !buffer.life_guard.use_at(submit_index) {
|
|
if let BufferMapState::Active { .. } = buffer.map_state {
|
|
log::warn!("Dropped buffer has a pending mapping.");
|
|
unsafe { device.raw.unmap_buffer(raw_buf) }
|
|
.map_err(DeviceError::from)?;
|
|
}
|
|
device.temp_suspected.buffers.push(id);
|
|
} else {
|
|
match buffer.map_state {
|
|
BufferMapState::Idle => (),
|
|
_ => panic!("Buffer {:?} is still mapped", id),
|
|
}
|
|
}
|
|
}
|
|
for id in cmdbuf.trackers.textures.used() {
|
|
let texture = &texture_guard[id];
|
|
if texture.raw.is_none() {
|
|
return Err(QueueSubmitError::DestroyedTexture(id.0));
|
|
}
|
|
if !texture.life_guard.use_at(submit_index) {
|
|
device.temp_suspected.textures.push(id);
|
|
}
|
|
}
|
|
for id in cmdbuf.trackers.views.used() {
|
|
if !texture_view_guard[id].life_guard.use_at(submit_index) {
|
|
device.temp_suspected.texture_views.push(id);
|
|
}
|
|
}
|
|
for id in cmdbuf.trackers.bind_groups.used() {
|
|
if !bind_group_guard[id].life_guard.use_at(submit_index) {
|
|
device.temp_suspected.bind_groups.push(id);
|
|
}
|
|
}
|
|
for id in cmdbuf.trackers.samplers.used() {
|
|
if !sampler_guard[id].life_guard.use_at(submit_index) {
|
|
device.temp_suspected.samplers.push(id);
|
|
}
|
|
}
|
|
for id in cmdbuf.trackers.compute_pipes.used() {
|
|
if !compute_pipe_guard[id].life_guard.use_at(submit_index) {
|
|
device.temp_suspected.compute_pipelines.push(id);
|
|
}
|
|
}
|
|
for id in cmdbuf.trackers.render_pipes.used() {
|
|
if !render_pipe_guard[id].life_guard.use_at(submit_index) {
|
|
device.temp_suspected.render_pipelines.push(id);
|
|
}
|
|
}
|
|
for id in cmdbuf.trackers.bundles.used() {
|
|
if !render_bundle_guard[id].life_guard.use_at(submit_index) {
|
|
device.temp_suspected.render_bundles.push(id);
|
|
}
|
|
}
|
|
|
|
let mut baked = cmdbuf.into_baked();
|
|
|
|
// execute resource transitions
|
|
unsafe {
|
|
baked
|
|
.encoder
|
|
.begin_encoding(Some("_Transit"))
|
|
.map_err(DeviceError::from)?
|
|
};
|
|
log::trace!("Stitching command buffer {:?} before submission", cmb_id);
|
|
trackers.merge_extend_stateless(&baked.trackers);
|
|
CommandBuffer::insert_barriers(
|
|
&mut baked.encoder,
|
|
&mut *trackers,
|
|
&baked.trackers.buffers,
|
|
&baked.trackers.textures,
|
|
&*buffer_guard,
|
|
&*texture_guard,
|
|
);
|
|
let transit = unsafe { baked.encoder.end_encoding().unwrap() };
|
|
baked.list.insert(0, transit);
|
|
active_executions.push(EncoderInFlight {
|
|
raw: baked.encoder,
|
|
cmd_buffers: baked.list,
|
|
});
|
|
}
|
|
|
|
log::trace!("Device after submission {}: {:#?}", submit_index, trackers);
|
|
drop(trackers);
|
|
if !required_buffer_inits.map.is_empty() {
|
|
device
|
|
.initialize_buffer_memory(required_buffer_inits, &mut *buffer_guard)?;
|
|
}
|
|
}
|
|
|
|
let super::Device {
|
|
ref mut pending_writes,
|
|
ref mut queue,
|
|
ref mut fence,
|
|
..
|
|
} = *device;
|
|
let refs = pending_writes
|
|
.pre_submit()
|
|
.into_iter()
|
|
.chain(
|
|
active_executions
|
|
.iter()
|
|
.flat_map(|pool_execution| pool_execution.cmd_buffers.iter()),
|
|
)
|
|
.collect::<Vec<_>>();
|
|
unsafe {
|
|
queue
|
|
.submit(&refs, Some((fence, submit_index)))
|
|
.map_err(DeviceError::from)?;
|
|
}
|
|
}
|
|
|
|
let callbacks = match device.maintain(&hub, false, &mut token) {
|
|
Ok(callbacks) => callbacks,
|
|
Err(WaitIdleError::Device(err)) => return Err(QueueSubmitError::Queue(err)),
|
|
Err(WaitIdleError::StuckGpu) => return Err(QueueSubmitError::StuckGpu),
|
|
};
|
|
|
|
profiling::scope!("cleanup");
|
|
if let Some(pending_execution) = device.pending_writes.post_submit(
|
|
&device.command_allocator,
|
|
&device.raw,
|
|
&device.queue,
|
|
) {
|
|
active_executions.push(pending_execution);
|
|
}
|
|
super::Device::lock_life_internal(&device.life_tracker, &mut token).track_submission(
|
|
submit_index,
|
|
device.pending_writes.temp_resources.drain(..),
|
|
active_executions,
|
|
);
|
|
|
|
callbacks
|
|
};
|
|
|
|
// the map callbacks should execute with nothing locked!
|
|
drop(token);
|
|
super::fire_map_callbacks(callbacks);
|
|
|
|
Ok(())
|
|
}
|
|
|
|
pub fn queue_get_timestamp_period<A: HalApi>(
|
|
&self,
|
|
queue_id: id::QueueId,
|
|
) -> Result<f32, InvalidQueue> {
|
|
let hub = A::hub(self);
|
|
let mut token = Token::root();
|
|
let (device_guard, _) = hub.devices.read(&mut token);
|
|
match device_guard.get(queue_id) {
|
|
Ok(_device) => Ok(1.0), //TODO?
|
|
Err(_) => Err(InvalidQueue),
|
|
}
|
|
}
|
|
}
|
|
|
|
fn get_lowest_common_denom(a: u32, b: u32) -> u32 {
|
|
let gcd = if a >= b {
|
|
get_greatest_common_divisor(a, b)
|
|
} else {
|
|
get_greatest_common_divisor(b, a)
|
|
};
|
|
a * b / gcd
|
|
}
|
|
|
|
fn get_greatest_common_divisor(mut a: u32, mut b: u32) -> u32 {
|
|
assert!(a >= b);
|
|
loop {
|
|
let c = a % b;
|
|
if c == 0 {
|
|
return b;
|
|
} else {
|
|
a = b;
|
|
b = c;
|
|
}
|
|
}
|
|
}
|
|
|
|
fn align_to(value: u32, alignment: u32) -> u32 {
|
|
match value % alignment {
|
|
0 => value,
|
|
other => value - other + alignment,
|
|
}
|
|
}
|
|
|
|
#[test]
|
|
fn test_lcd() {
|
|
assert_eq!(get_lowest_common_denom(2, 2), 2);
|
|
assert_eq!(get_lowest_common_denom(2, 3), 6);
|
|
assert_eq!(get_lowest_common_denom(6, 4), 12);
|
|
}
|
|
|
|
#[test]
|
|
fn test_gcd() {
|
|
assert_eq!(get_greatest_common_divisor(5, 1), 1);
|
|
assert_eq!(get_greatest_common_divisor(4, 2), 2);
|
|
assert_eq!(get_greatest_common_divisor(6, 4), 2);
|
|
assert_eq!(get_greatest_common_divisor(7, 7), 7);
|
|
}
|