1use alloc::{format, string::String, sync::Arc, vec::Vec};
2
3use arrayvec::ArrayVec;
4use thiserror::Error;
5use wgt::{
6 error::{ErrorType, WebGpuError},
7 BufferAddress, BufferTextureCopyInfoError, BufferUsages, Extent3d, TextureSelector,
8 TextureUsages,
9};
10
11use crate::{
12 api_log,
13 command::{
14 clear_texture, encoder::EncodingState, ArcCommand, CommandEncoderError, EncoderStateError,
15 },
16 device::MissingDownlevelFlags,
17 global::Global,
18 id::{BufferId, CommandEncoderId, TextureId},
19 init_tracker::{
20 has_copy_partial_init_tracker_coverage, MemoryInitKind, TextureInitRange,
21 TextureInitTrackerAction,
22 },
23 resource::{
24 Buffer, MissingBufferUsageError, MissingTextureUsageError, ParentDevice, RawResourceAccess,
25 Texture, TextureErrorDimension,
26 },
27};
28
29use super::ClearError;
30
31type TexelCopyBufferInfo = wgt::TexelCopyBufferInfo<BufferId>;
32type TexelCopyTextureInfo = wgt::TexelCopyTextureInfo<Arc<Texture>>;
33
34#[derive(Clone, Copy, Debug, Eq, PartialEq)]
35pub enum CopySide {
36 Source,
37 Destination,
38}
39
40#[derive(Clone, Debug, Error)]
42#[non_exhaustive]
43pub enum TransferError {
44 #[error("Source and destination cannot be the same buffer")]
45 SameSourceDestinationBuffer,
46 #[error(transparent)]
47 MissingBufferUsage(#[from] MissingBufferUsageError),
48 #[error(transparent)]
49 MissingTextureUsage(#[from] MissingTextureUsageError),
50 #[error(
51 "Copy at offset {start_offset} bytes would end up overrunning the bounds of the {side:?} buffer of size {buffer_size}"
52 )]
53 BufferStartOffsetOverrun {
54 start_offset: BufferAddress,
55 buffer_size: BufferAddress,
56 side: CopySide,
57 },
58 #[error(
59 "Copy at offset {start_offset} for {size} bytes would end up overrunning the bounds of the {side:?} buffer of size {buffer_size}"
60 )]
61 BufferEndOffsetOverrun {
62 start_offset: BufferAddress,
63 size: BufferAddress,
64 buffer_size: BufferAddress,
65 side: CopySide,
66 },
67 #[error("Copy of {dimension:?} {start_offset}..{end_offset} would end up overrunning the bounds of the {side:?} texture of {dimension:?} size {texture_size}")]
68 TextureOverrun {
69 start_offset: u32,
70 end_offset: u32,
71 texture_size: u32,
72 dimension: TextureErrorDimension,
73 side: CopySide,
74 },
75 #[error("Partial copy of {start_offset}..{end_offset} on {dimension:?} dimension with size {texture_size} \
76 is not supported for the {side:?} texture format {format:?} with {sample_count} samples")]
77 UnsupportedPartialTransfer {
78 format: wgt::TextureFormat,
79 sample_count: u32,
80 start_offset: u32,
81 end_offset: u32,
82 texture_size: u32,
83 dimension: TextureErrorDimension,
84 side: CopySide,
85 },
86 #[error(
87 "Copying{} layers {}..{} to{} layers {}..{} of the same texture is not allowed",
88 if *src_aspects == wgt::TextureAspect::All { String::new() } else { format!(" {src_aspects:?}") },
89 src_origin_z,
90 src_origin_z + array_layer_count,
91 if *dst_aspects == wgt::TextureAspect::All { String::new() } else { format!(" {dst_aspects:?}") },
92 dst_origin_z,
93 dst_origin_z + array_layer_count,
94 )]
95 InvalidCopyWithinSameTexture {
96 src_aspects: wgt::TextureAspect,
97 dst_aspects: wgt::TextureAspect,
98 src_origin_z: u32,
99 dst_origin_z: u32,
100 array_layer_count: u32,
101 },
102 #[error("Unable to select texture aspect {aspect:?} from format {format:?}")]
103 InvalidTextureAspect {
104 format: wgt::TextureFormat,
105 aspect: wgt::TextureAspect,
106 },
107 #[error("Unable to select texture mip level {level} out of {total}")]
108 InvalidTextureMipLevel { level: u32, total: u32 },
109 #[error("Texture dimension must be 2D when copying from an external texture")]
110 InvalidDimensionExternal,
111 #[error("Buffer offset {0} is not aligned to block size or `COPY_BUFFER_ALIGNMENT`")]
112 UnalignedBufferOffset(BufferAddress),
113 #[error("Copy size {0} does not respect `COPY_BUFFER_ALIGNMENT`")]
114 UnalignedCopySize(BufferAddress),
115 #[error("Copy width is not a multiple of block width")]
116 UnalignedCopyWidth,
117 #[error("Copy height is not a multiple of block height")]
118 UnalignedCopyHeight,
119 #[error("Copy origin's x component is not a multiple of block width")]
120 UnalignedCopyOriginX,
121 #[error("Copy origin's y component is not a multiple of block height")]
122 UnalignedCopyOriginY,
123 #[error("Bytes per row does not respect `COPY_BYTES_PER_ROW_ALIGNMENT`")]
124 UnalignedBytesPerRow,
125 #[error("Number of bytes per row needs to be specified since more than one row is copied")]
126 UnspecifiedBytesPerRow,
127 #[error("Number of rows per image needs to be specified since more than one image is copied")]
128 UnspecifiedRowsPerImage,
129 #[error("Number of bytes per row is less than the number of bytes in a complete row")]
130 InvalidBytesPerRow,
131 #[error("Number of rows per image is invalid")]
132 InvalidRowsPerImage,
133 #[error("Overflow while computing the size of the copy")]
134 SizeOverflow,
135 #[error("Copy source aspects must refer to all aspects of the source texture format")]
136 CopySrcMissingAspects,
137 #[error(
138 "Copy destination aspects must refer to all aspects of the destination texture format"
139 )]
140 CopyDstMissingAspects,
141 #[error("Copy aspect must refer to a single aspect of texture format")]
142 CopyAspectNotOne,
143 #[error("Copying from textures with format {0:?} is forbidden")]
144 CopyFromForbiddenTextureFormat(wgt::TextureFormat),
145 #[error("Copying from textures with format {format:?} and aspect {aspect:?} is forbidden")]
146 CopyFromForbiddenTextureFormatAspect {
147 format: wgt::TextureFormat,
148 aspect: wgt::TextureAspect,
149 },
150 #[error("Copying to textures with format {0:?} is forbidden")]
151 CopyToForbiddenTextureFormat(wgt::TextureFormat),
152 #[error("Copying to textures with format {format:?} and aspect {aspect:?} is forbidden")]
153 CopyToForbiddenTextureFormatAspect {
154 format: wgt::TextureFormat,
155 aspect: wgt::TextureAspect,
156 },
157 #[error(
158 "Copying to textures with format {0:?} is forbidden when copying from external texture"
159 )]
160 ExternalCopyToForbiddenTextureFormat(wgt::TextureFormat),
161 #[error(
162 "Source format ({src_format:?}) and destination format ({dst_format:?}) are not copy-compatible (they may only differ in srgb-ness)"
163 )]
164 TextureFormatsNotCopyCompatible {
165 src_format: wgt::TextureFormat,
166 dst_format: wgt::TextureFormat,
167 },
168 #[error(transparent)]
169 MemoryInitFailure(#[from] ClearError),
170 #[error("Cannot encode this copy because of a missing downelevel flag")]
171 MissingDownlevelFlags(#[from] MissingDownlevelFlags),
172 #[error("Source texture sample count must be 1, got {sample_count}")]
173 InvalidSampleCount { sample_count: u32 },
174 #[error(
175 "Source sample count ({src_sample_count:?}) and destination sample count ({dst_sample_count:?}) are not equal"
176 )]
177 SampleCountNotEqual {
178 src_sample_count: u32,
179 dst_sample_count: u32,
180 },
181 #[error("Requested mip level {requested} does not exist (count: {count})")]
182 InvalidMipLevel { requested: u32, count: u32 },
183 #[error("Buffer is expected to be unmapped, but was not")]
184 BufferNotAvailable,
185}
186
187impl WebGpuError for TransferError {
188 fn webgpu_error_type(&self) -> ErrorType {
189 match self {
190 Self::MissingBufferUsage(e) => e.webgpu_error_type(),
191 Self::MissingTextureUsage(e) => e.webgpu_error_type(),
192 Self::MemoryInitFailure(e) => e.webgpu_error_type(),
193
194 Self::BufferEndOffsetOverrun { .. }
195 | Self::TextureOverrun { .. }
196 | Self::BufferStartOffsetOverrun { .. }
197 | Self::UnsupportedPartialTransfer { .. }
198 | Self::InvalidCopyWithinSameTexture { .. }
199 | Self::InvalidTextureAspect { .. }
200 | Self::InvalidTextureMipLevel { .. }
201 | Self::InvalidDimensionExternal
202 | Self::UnalignedBufferOffset(..)
203 | Self::UnalignedCopySize(..)
204 | Self::UnalignedCopyWidth
205 | Self::UnalignedCopyHeight
206 | Self::UnalignedCopyOriginX
207 | Self::UnalignedCopyOriginY
208 | Self::UnalignedBytesPerRow
209 | Self::UnspecifiedBytesPerRow
210 | Self::UnspecifiedRowsPerImage
211 | Self::InvalidBytesPerRow
212 | Self::InvalidRowsPerImage
213 | Self::SizeOverflow
214 | Self::CopySrcMissingAspects
215 | Self::CopyDstMissingAspects
216 | Self::CopyAspectNotOne
217 | Self::CopyFromForbiddenTextureFormat(..)
218 | Self::CopyFromForbiddenTextureFormatAspect { .. }
219 | Self::CopyToForbiddenTextureFormat(..)
220 | Self::CopyToForbiddenTextureFormatAspect { .. }
221 | Self::ExternalCopyToForbiddenTextureFormat(..)
222 | Self::TextureFormatsNotCopyCompatible { .. }
223 | Self::MissingDownlevelFlags(..)
224 | Self::InvalidSampleCount { .. }
225 | Self::SampleCountNotEqual { .. }
226 | Self::InvalidMipLevel { .. }
227 | Self::SameSourceDestinationBuffer
228 | Self::BufferNotAvailable => ErrorType::Validation,
229 }
230 }
231}
232
233impl From<BufferTextureCopyInfoError> for TransferError {
234 fn from(value: BufferTextureCopyInfoError) -> Self {
235 match value {
236 BufferTextureCopyInfoError::InvalidBytesPerRow => Self::InvalidBytesPerRow,
237 BufferTextureCopyInfoError::InvalidRowsPerImage => Self::InvalidRowsPerImage,
238 BufferTextureCopyInfoError::ImageStrideOverflow
239 | BufferTextureCopyInfoError::ImageBytesOverflow(_)
240 | BufferTextureCopyInfoError::ArraySizeOverflow(_) => Self::SizeOverflow,
241 }
242 }
243}
244
245pub(crate) fn extract_texture_selector<T>(
246 copy_texture: &wgt::TexelCopyTextureInfo<T>,
247 copy_size: &Extent3d,
248 texture: &Texture,
249) -> Result<(TextureSelector, hal::TextureCopyBase), TransferError> {
250 let format = texture.desc.format;
251 let copy_aspect = hal::FormatAspects::new(format, copy_texture.aspect);
252 if copy_aspect.is_empty() {
253 return Err(TransferError::InvalidTextureAspect {
254 format,
255 aspect: copy_texture.aspect,
256 });
257 }
258
259 let (layers, origin_z) = match texture.desc.dimension {
260 wgt::TextureDimension::D1 => (0..1, 0),
261 wgt::TextureDimension::D2 => (
262 copy_texture.origin.z..copy_texture.origin.z + copy_size.depth_or_array_layers,
263 0,
264 ),
265 wgt::TextureDimension::D3 => (0..1, copy_texture.origin.z),
266 };
267 let base = hal::TextureCopyBase {
268 origin: wgt::Origin3d {
269 x: copy_texture.origin.x,
270 y: copy_texture.origin.y,
271 z: origin_z,
272 },
273 array_layer: layers.start,
275 mip_level: copy_texture.mip_level,
276 aspect: copy_aspect,
277 };
278 let selector = TextureSelector {
279 mips: copy_texture.mip_level..copy_texture.mip_level + 1,
280 layers,
281 };
282
283 Ok((selector, base))
284}
285
286pub(crate) fn validate_linear_texture_data(
298 layout: &wgt::TexelCopyBufferLayout,
299 format: wgt::TextureFormat,
300 aspect: wgt::TextureAspect,
301 buffer_size: BufferAddress,
302 buffer_side: CopySide,
303 copy_size: &Extent3d,
304) -> Result<(BufferAddress, BufferAddress, bool), TransferError> {
305 let wgt::BufferTextureCopyInfo {
306 copy_width,
307 copy_height,
308 depth_or_array_layers,
309
310 offset,
311
312 block_size_bytes: _,
313 block_width_texels,
314 block_height_texels,
315
316 width_blocks: _,
317 height_blocks,
318
319 row_bytes_dense,
320 row_stride_bytes,
321
322 image_stride_rows: _,
323 image_stride_bytes,
324
325 image_rows_dense: _,
326 image_bytes_dense,
327
328 bytes_in_copy,
329 } = layout.get_buffer_texture_copy_info(format, aspect, copy_size)?;
330
331 if !copy_width.is_multiple_of(block_width_texels) {
332 return Err(TransferError::UnalignedCopyWidth);
333 }
334 if !copy_height.is_multiple_of(block_height_texels) {
335 return Err(TransferError::UnalignedCopyHeight);
336 }
337
338 let requires_multiple_rows = depth_or_array_layers > 1 || height_blocks > 1;
339 let requires_multiple_images = depth_or_array_layers > 1;
340
341 if layout.bytes_per_row.is_none() && requires_multiple_rows {
346 return Err(TransferError::UnspecifiedBytesPerRow);
347 }
348
349 if layout.rows_per_image.is_none() && requires_multiple_images {
350 return Err(TransferError::UnspecifiedRowsPerImage);
351 };
352
353 if offset > buffer_size {
354 return Err(TransferError::BufferStartOffsetOverrun {
355 start_offset: offset,
356 buffer_size,
357 side: buffer_side,
358 });
359 }
360 if bytes_in_copy > buffer_size - offset {
362 return Err(TransferError::BufferEndOffsetOverrun {
363 start_offset: offset,
364 size: bytes_in_copy,
365 buffer_size,
366 side: buffer_side,
367 });
368 }
369
370 let is_contiguous = (row_stride_bytes == row_bytes_dense || !requires_multiple_rows)
371 && (image_stride_bytes == image_bytes_dense || !requires_multiple_images);
372
373 Ok((bytes_in_copy, image_stride_bytes, is_contiguous))
374}
375
376pub(crate) fn validate_texture_copy_src_format(
385 format: wgt::TextureFormat,
386 aspect: wgt::TextureAspect,
387) -> Result<(), TransferError> {
388 use wgt::TextureAspect as Ta;
389 use wgt::TextureFormat as Tf;
390 match (format, aspect) {
391 (Tf::Depth24Plus, _) => Err(TransferError::CopyFromForbiddenTextureFormat(format)),
392 (Tf::Depth24PlusStencil8, Ta::DepthOnly) => {
393 Err(TransferError::CopyFromForbiddenTextureFormatAspect { format, aspect })
394 }
395 _ => Ok(()),
396 }
397}
398
399pub(crate) fn validate_texture_copy_dst_format(
408 format: wgt::TextureFormat,
409 aspect: wgt::TextureAspect,
410) -> Result<(), TransferError> {
411 use wgt::TextureAspect as Ta;
412 use wgt::TextureFormat as Tf;
413 match (format, aspect) {
414 (Tf::Depth24Plus | Tf::Depth32Float, _) => {
415 Err(TransferError::CopyToForbiddenTextureFormat(format))
416 }
417 (Tf::Depth24PlusStencil8 | Tf::Depth32FloatStencil8, Ta::DepthOnly) => {
418 Err(TransferError::CopyToForbiddenTextureFormatAspect { format, aspect })
419 }
420 _ => Ok(()),
421 }
422}
423
424pub(crate) fn validate_texture_buffer_copy<T>(
452 texture_copy_view: &wgt::TexelCopyTextureInfo<T>,
453 aspect: hal::FormatAspects,
454 desc: &wgt::TextureDescriptor<String, Vec<wgt::TextureFormat>>,
455 layout: &wgt::TexelCopyBufferLayout,
456 aligned: bool,
457) -> Result<(), TransferError> {
458 if desc.sample_count != 1 {
459 return Err(TransferError::InvalidSampleCount {
460 sample_count: desc.sample_count,
461 });
462 }
463
464 if !aspect.is_one() {
465 return Err(TransferError::CopyAspectNotOne);
466 }
467
468 let offset_alignment = if desc.format.is_depth_stencil_format() {
469 4
470 } else {
471 desc.format
476 .block_copy_size(Some(texture_copy_view.aspect))
477 .expect("non-copyable formats should have been rejected previously")
478 };
479
480 if aligned && !layout.offset.is_multiple_of(u64::from(offset_alignment)) {
481 return Err(TransferError::UnalignedBufferOffset(layout.offset));
482 }
483
484 if let Some(bytes_per_row) = layout.bytes_per_row {
485 if aligned && bytes_per_row % wgt::COPY_BYTES_PER_ROW_ALIGNMENT != 0 {
486 return Err(TransferError::UnalignedBytesPerRow);
487 }
488 }
489
490 Ok(())
491}
492
493pub(crate) fn validate_texture_copy_range<T>(
504 texture_copy_view: &wgt::TexelCopyTextureInfo<T>,
505 desc: &wgt::TextureDescriptor<String, Vec<wgt::TextureFormat>>,
506 texture_side: CopySide,
507 copy_size: &Extent3d,
508) -> Result<(hal::CopyExtent, u32), TransferError> {
509 let (block_width, block_height) = desc.format.block_dimensions();
510
511 let extent_virtual = desc.mip_level_size(texture_copy_view.mip_level).ok_or(
512 TransferError::InvalidTextureMipLevel {
513 level: texture_copy_view.mip_level,
514 total: desc.mip_level_count,
515 },
516 )?;
517 let extent = extent_virtual.physical_size(desc.format);
519
520 let requires_exact_size = desc.format.is_depth_stencil_format() || desc.sample_count > 1;
523
524 let check_dimension = |dimension: TextureErrorDimension,
527 start_offset: u32,
528 size: u32,
529 texture_size: u32,
530 requires_exact_size: bool|
531 -> Result<(), TransferError> {
532 if requires_exact_size && (start_offset != 0 || size != texture_size) {
533 Err(TransferError::UnsupportedPartialTransfer {
534 format: desc.format,
535 sample_count: desc.sample_count,
536 start_offset,
537 end_offset: start_offset.wrapping_add(size),
538 texture_size,
539 dimension,
540 side: texture_side,
541 })
542 } else if start_offset > texture_size || texture_size - start_offset < size {
545 Err(TransferError::TextureOverrun {
546 start_offset,
547 end_offset: start_offset.wrapping_add(size),
548 texture_size,
549 dimension,
550 side: texture_side,
551 })
552 } else {
553 Ok(())
554 }
555 };
556
557 check_dimension(
558 TextureErrorDimension::X,
559 texture_copy_view.origin.x,
560 copy_size.width,
561 extent.width,
562 requires_exact_size,
563 )?;
564 check_dimension(
565 TextureErrorDimension::Y,
566 texture_copy_view.origin.y,
567 copy_size.height,
568 extent.height,
569 requires_exact_size,
570 )?;
571 check_dimension(
572 TextureErrorDimension::Z,
573 texture_copy_view.origin.z,
574 copy_size.depth_or_array_layers,
575 extent.depth_or_array_layers,
576 false, )?;
578
579 if !texture_copy_view.origin.x.is_multiple_of(block_width) {
580 return Err(TransferError::UnalignedCopyOriginX);
581 }
582 if !texture_copy_view.origin.y.is_multiple_of(block_height) {
583 return Err(TransferError::UnalignedCopyOriginY);
584 }
585 if !copy_size.width.is_multiple_of(block_width) {
586 return Err(TransferError::UnalignedCopyWidth);
587 }
588 if !copy_size.height.is_multiple_of(block_height) {
589 return Err(TransferError::UnalignedCopyHeight);
590 }
591
592 let (depth, array_layer_count) = match desc.dimension {
593 wgt::TextureDimension::D1 => (1, 1),
594 wgt::TextureDimension::D2 => (1, copy_size.depth_or_array_layers),
595 wgt::TextureDimension::D3 => (copy_size.depth_or_array_layers, 1),
596 };
597
598 let copy_extent = hal::CopyExtent {
599 width: copy_size.width,
600 height: copy_size.height,
601 depth,
602 };
603 Ok((copy_extent, array_layer_count))
604}
605
606pub(crate) fn validate_copy_within_same_texture<T>(
617 src: &wgt::TexelCopyTextureInfo<T>,
618 dst: &wgt::TexelCopyTextureInfo<T>,
619 format: wgt::TextureFormat,
620 array_layer_count: u32,
621) -> Result<(), TransferError> {
622 let src_aspects = hal::FormatAspects::new(format, src.aspect);
623 let dst_aspects = hal::FormatAspects::new(format, dst.aspect);
624 if (src_aspects & dst_aspects).is_empty() {
625 return Ok(());
627 }
628
629 if src.origin.z >= dst.origin.z + array_layer_count
630 || dst.origin.z >= src.origin.z + array_layer_count
631 {
632 return Ok(());
634 }
635
636 if src.mip_level != dst.mip_level {
637 return Ok(());
639 }
640
641 Err(TransferError::InvalidCopyWithinSameTexture {
642 src_aspects: src.aspect,
643 dst_aspects: dst.aspect,
644 src_origin_z: src.origin.z,
645 dst_origin_z: dst.origin.z,
646 array_layer_count,
647 })
648}
649
650fn handle_texture_init(
651 state: &mut EncodingState,
652 init_kind: MemoryInitKind,
653 copy_texture: &TexelCopyTextureInfo,
654 copy_size: &Extent3d,
655 texture: &Arc<Texture>,
656) -> Result<(), ClearError> {
657 let init_layer_range = if texture.desc.dimension == wgt::TextureDimension::D3 {
658 0..1
660 } else {
661 copy_texture.origin.z..copy_texture.origin.z + copy_size.depth_or_array_layers
662 };
663 let init_action = TextureInitTrackerAction {
664 texture: texture.clone(),
665 range: TextureInitRange {
666 mip_range: copy_texture.mip_level..copy_texture.mip_level + 1,
667 layer_range: init_layer_range,
668 },
669 kind: init_kind,
670 };
671
672 let immediate_inits = state
674 .texture_memory_actions
675 .register_init_action(&{ init_action });
676
677 if !immediate_inits.is_empty() {
679 for init in immediate_inits {
680 clear_texture(
681 &init.texture,
682 TextureInitRange {
683 mip_range: init.mip_level..(init.mip_level + 1),
684 layer_range: init.layer..(init.layer + 1),
685 },
686 state.raw_encoder,
687 &mut state.tracker.textures,
688 &state.device.alignments,
689 state.device.zero_buffer.as_ref(),
690 state.snatch_guard,
691 state.device.instance_flags,
692 )?;
693 }
694 }
695
696 Ok(())
697}
698
699fn handle_src_texture_init(
704 state: &mut EncodingState,
705 source: &TexelCopyTextureInfo,
706 copy_size: &Extent3d,
707 texture: &Arc<Texture>,
708) -> Result<(), TransferError> {
709 handle_texture_init(
710 state,
711 MemoryInitKind::NeedsInitializedMemory,
712 source,
713 copy_size,
714 texture,
715 )?;
716 Ok(())
717}
718
719fn handle_dst_texture_init(
724 state: &mut EncodingState,
725 destination: &wgt::TexelCopyTextureInfo<Arc<Texture>>,
726 copy_size: &Extent3d,
727 texture: &Arc<Texture>,
728) -> Result<(), TransferError> {
729 let dst_init_kind =
734 if has_copy_partial_init_tracker_coverage(copy_size, destination, &texture.desc) {
735 MemoryInitKind::NeedsInitializedMemory
736 } else {
737 MemoryInitKind::ImplicitlyInitialized
738 };
739
740 handle_texture_init(state, dst_init_kind, destination, copy_size, texture)?;
741 Ok(())
742}
743
744fn handle_buffer_init(
749 state: &mut EncodingState,
750 info: &wgt::TexelCopyBufferInfo<Arc<Buffer>>,
751 direction: CopySide,
752 required_buffer_bytes_in_copy: BufferAddress,
753 is_contiguous: bool,
754) {
755 const ALIGN_SIZE: BufferAddress = wgt::COPY_BUFFER_ALIGNMENT;
756 const ALIGN_MASK: BufferAddress = wgt::COPY_BUFFER_ALIGNMENT - 1;
757
758 let buffer = &info.buffer;
759 let start = info.layout.offset;
760 let end = info.layout.offset + required_buffer_bytes_in_copy;
761 if !is_contiguous || direction == CopySide::Source {
762 let aligned_start = start & !ALIGN_MASK;
773 let aligned_end = (end + ALIGN_MASK) & !ALIGN_MASK;
774 state
775 .buffer_memory_init_actions
776 .extend(buffer.initialization_status.read().create_action(
777 buffer,
778 aligned_start..aligned_end,
779 MemoryInitKind::NeedsInitializedMemory,
780 ));
781 } else {
782 let aligned_start = (start + ALIGN_MASK) & !ALIGN_MASK;
792 let aligned_end = end & !ALIGN_MASK;
793 if aligned_start != start {
794 state.buffer_memory_init_actions.extend(
795 buffer.initialization_status.read().create_action(
796 buffer,
797 aligned_start - ALIGN_SIZE..aligned_start,
798 MemoryInitKind::NeedsInitializedMemory,
799 ),
800 );
801 }
802 if aligned_start != aligned_end {
803 state.buffer_memory_init_actions.extend(
804 buffer.initialization_status.read().create_action(
805 buffer,
806 aligned_start..aligned_end,
807 MemoryInitKind::ImplicitlyInitialized,
808 ),
809 );
810 }
811 if aligned_end != end {
812 state.buffer_memory_init_actions.extend(
818 buffer.initialization_status.read().create_action(
819 buffer,
820 aligned_end..aligned_end + ALIGN_SIZE,
821 MemoryInitKind::NeedsInitializedMemory,
822 ),
823 );
824 }
825 }
826}
827
828impl Global {
829 pub fn command_encoder_copy_buffer_to_buffer(
830 &self,
831 command_encoder_id: CommandEncoderId,
832 source: BufferId,
833 source_offset: BufferAddress,
834 destination: BufferId,
835 destination_offset: BufferAddress,
836 size: Option<BufferAddress>,
837 ) -> Result<(), EncoderStateError> {
838 profiling::scope!("CommandEncoder::copy_buffer_to_buffer");
839 api_log!(
840 "CommandEncoder::copy_buffer_to_buffer {source:?} -> {destination:?} {size:?}bytes"
841 );
842
843 let hub = &self.hub;
844
845 let cmd_enc = hub.command_encoders.get(command_encoder_id);
846 let mut cmd_buf_data = cmd_enc.data.lock();
847
848 cmd_buf_data.push_with(|| -> Result<_, CommandEncoderError> {
849 Ok(ArcCommand::CopyBufferToBuffer {
850 src: self.resolve_buffer_id(source)?,
851 src_offset: source_offset,
852 dst: self.resolve_buffer_id(destination)?,
853 dst_offset: destination_offset,
854 size,
855 })
856 })
857 }
858
859 pub fn command_encoder_copy_buffer_to_texture(
860 &self,
861 command_encoder_id: CommandEncoderId,
862 source: &TexelCopyBufferInfo,
863 destination: &wgt::TexelCopyTextureInfo<TextureId>,
864 copy_size: &Extent3d,
865 ) -> Result<(), EncoderStateError> {
866 profiling::scope!("CommandEncoder::copy_buffer_to_texture");
867 api_log!(
868 "CommandEncoder::copy_buffer_to_texture {:?} -> {:?} {copy_size:?}",
869 source.buffer,
870 destination.texture
871 );
872
873 let cmd_enc = self.hub.command_encoders.get(command_encoder_id);
874
875 let mut cmd_buf_data = cmd_enc.data.lock();
876
877 cmd_buf_data.push_with(|| -> Result<_, CommandEncoderError> {
878 let texture = self.resolve_texture_id(destination.texture);
879 texture.check_valid(&cmd_enc.device.snatchable_lock.read())?;
880 Ok(ArcCommand::CopyBufferToTexture {
881 src: wgt::TexelCopyBufferInfo::<Arc<Buffer>> {
882 buffer: self.resolve_buffer_id(source.buffer)?,
883 layout: source.layout,
884 },
885 dst: wgt::TexelCopyTextureInfo::<Arc<Texture>> {
886 texture,
887 mip_level: destination.mip_level,
888 origin: destination.origin,
889 aspect: destination.aspect,
890 },
891 size: *copy_size,
892 })
893 })
894 }
895
896 pub fn command_encoder_copy_texture_to_buffer(
897 &self,
898 command_encoder_id: CommandEncoderId,
899 source: &wgt::TexelCopyTextureInfo<TextureId>,
900 destination: &TexelCopyBufferInfo,
901 copy_size: &Extent3d,
902 ) -> Result<(), EncoderStateError> {
903 profiling::scope!("CommandEncoder::copy_texture_to_buffer");
904 api_log!(
905 "CommandEncoder::copy_texture_to_buffer {:?} -> {:?} {copy_size:?}",
906 source.texture,
907 destination.buffer
908 );
909
910 let cmd_enc = self.hub.command_encoders.get(command_encoder_id);
911
912 let mut cmd_buf_data = cmd_enc.data.lock();
913
914 cmd_buf_data.push_with(|| -> Result<_, CommandEncoderError> {
915 let texture = self.resolve_texture_id(source.texture);
916 texture.check_valid(&cmd_enc.device.snatchable_lock.read())?;
917 Ok(ArcCommand::CopyTextureToBuffer {
918 src: wgt::TexelCopyTextureInfo::<Arc<Texture>> {
919 texture,
920 mip_level: source.mip_level,
921 origin: source.origin,
922 aspect: source.aspect,
923 },
924 dst: wgt::TexelCopyBufferInfo::<Arc<Buffer>> {
925 buffer: self.resolve_buffer_id(destination.buffer)?,
926 layout: destination.layout,
927 },
928 size: *copy_size,
929 })
930 })
931 }
932
933 pub fn command_encoder_copy_texture_to_texture(
934 &self,
935 command_encoder_id: CommandEncoderId,
936 source: &wgt::TexelCopyTextureInfo<TextureId>,
937 destination: &wgt::TexelCopyTextureInfo<TextureId>,
938 copy_size: &Extent3d,
939 ) -> Result<(), EncoderStateError> {
940 profiling::scope!("CommandEncoder::copy_texture_to_texture");
941 api_log!(
942 "CommandEncoder::copy_texture_to_texture {:?} -> {:?} {copy_size:?}",
943 source.texture,
944 destination.texture
945 );
946
947 let cmd_enc = self.hub.command_encoders.get(command_encoder_id);
948
949 let mut cmd_buf_data = cmd_enc.data.lock();
950
951 cmd_buf_data.push_with(|| -> Result<_, CommandEncoderError> {
952 let src_texture = self.resolve_texture_id(source.texture);
953 let dst_texture = self.resolve_texture_id(destination.texture);
954 {
955 let snatch_guard = cmd_enc.device.snatchable_lock.read();
956 src_texture.check_valid(&snatch_guard)?;
957 dst_texture.check_valid(&snatch_guard)?;
958 }
959 Ok(ArcCommand::CopyTextureToTexture {
960 src: wgt::TexelCopyTextureInfo {
961 texture: src_texture,
962 mip_level: source.mip_level,
963 origin: source.origin,
964 aspect: source.aspect,
965 },
966 dst: wgt::TexelCopyTextureInfo {
967 texture: dst_texture,
968 mip_level: destination.mip_level,
969 origin: destination.origin,
970 aspect: destination.aspect,
971 },
972 size: *copy_size,
973 })
974 })
975 }
976}
977
978#[inline(never)]
979pub(super) fn copy_buffer_to_buffer(
980 state: &mut EncodingState,
981 src_buffer: &Arc<Buffer>,
982 source_offset: BufferAddress,
983 dst_buffer: &Arc<Buffer>,
984 destination_offset: BufferAddress,
985 size: Option<BufferAddress>,
986) -> Result<(), CommandEncoderError> {
987 if src_buffer.is_equal(dst_buffer) {
988 return Err(TransferError::SameSourceDestinationBuffer.into());
989 }
990
991 src_buffer.same_device(state.device)?;
992
993 let src_pending = state
994 .tracker
995 .buffers
996 .set_single(src_buffer, wgt::BufferUses::COPY_SRC);
997
998 let src_raw = src_buffer.try_raw(state.snatch_guard)?;
999 src_buffer
1000 .check_usage(BufferUsages::COPY_SRC)
1001 .map_err(TransferError::MissingBufferUsage)?;
1002 let src_barrier = src_pending.map(|pending| pending.into_hal(src_buffer, state.snatch_guard));
1004
1005 dst_buffer.same_device(state.device)?;
1006
1007 let dst_pending = state
1008 .tracker
1009 .buffers
1010 .set_single(dst_buffer, wgt::BufferUses::COPY_DST);
1011
1012 let dst_raw = dst_buffer.try_raw(state.snatch_guard)?;
1013 dst_buffer
1014 .check_usage(BufferUsages::COPY_DST)
1015 .map_err(TransferError::MissingBufferUsage)?;
1016 let dst_barrier = dst_pending.map(|pending| pending.into_hal(dst_buffer, state.snatch_guard));
1017
1018 if source_offset > src_buffer.size {
1019 return Err(TransferError::BufferStartOffsetOverrun {
1020 start_offset: source_offset,
1021 buffer_size: src_buffer.size,
1022 side: CopySide::Source,
1023 }
1024 .into());
1025 }
1026 let size = size.unwrap_or_else(|| {
1027 src_buffer.size - source_offset
1029 });
1030
1031 if !size.is_multiple_of(wgt::COPY_BUFFER_ALIGNMENT) {
1032 return Err(TransferError::UnalignedCopySize(size).into());
1033 }
1034 if !source_offset.is_multiple_of(wgt::COPY_BUFFER_ALIGNMENT) {
1035 return Err(TransferError::UnalignedBufferOffset(source_offset).into());
1036 }
1037 if !destination_offset.is_multiple_of(wgt::COPY_BUFFER_ALIGNMENT) {
1038 return Err(TransferError::UnalignedBufferOffset(destination_offset).into());
1039 }
1040 if !state
1041 .device
1042 .downlevel
1043 .flags
1044 .contains(wgt::DownlevelFlags::UNRESTRICTED_INDEX_BUFFER)
1045 && (src_buffer.usage.contains(BufferUsages::INDEX)
1046 || dst_buffer.usage.contains(BufferUsages::INDEX))
1047 {
1048 let forbidden_usages = BufferUsages::VERTEX
1049 | BufferUsages::UNIFORM
1050 | BufferUsages::INDIRECT
1051 | BufferUsages::STORAGE;
1052 if src_buffer.usage.intersects(forbidden_usages)
1053 || dst_buffer.usage.intersects(forbidden_usages)
1054 {
1055 return Err(TransferError::MissingDownlevelFlags(MissingDownlevelFlags(
1056 wgt::DownlevelFlags::UNRESTRICTED_INDEX_BUFFER,
1057 ))
1058 .into());
1059 }
1060 }
1061
1062 if size > src_buffer.size - source_offset {
1063 return Err(TransferError::BufferEndOffsetOverrun {
1064 start_offset: source_offset,
1065 size,
1066 buffer_size: src_buffer.size,
1067 side: CopySide::Source,
1068 }
1069 .into());
1070 }
1071 let source_end_offset = source_offset + size;
1073
1074 if destination_offset > dst_buffer.size {
1075 return Err(TransferError::BufferStartOffsetOverrun {
1076 start_offset: destination_offset,
1077 buffer_size: dst_buffer.size,
1078 side: CopySide::Destination,
1079 }
1080 .into());
1081 }
1082 if size > dst_buffer.size - destination_offset {
1084 return Err(TransferError::BufferEndOffsetOverrun {
1085 start_offset: destination_offset,
1086 size,
1087 buffer_size: dst_buffer.size,
1088 side: CopySide::Destination,
1089 }
1090 .into());
1091 }
1092 let destination_end_offset = destination_offset + size;
1094
1095 if size == 0 {
1098 log::trace!("Ignoring copy_buffer_to_buffer of size 0");
1099 return Ok(());
1100 }
1101
1102 state
1104 .buffer_memory_init_actions
1105 .extend(dst_buffer.initialization_status.read().create_action(
1106 dst_buffer,
1107 destination_offset..destination_end_offset,
1108 MemoryInitKind::ImplicitlyInitialized,
1109 ));
1110 state
1111 .buffer_memory_init_actions
1112 .extend(src_buffer.initialization_status.read().create_action(
1113 src_buffer,
1114 source_offset..source_end_offset,
1115 MemoryInitKind::NeedsInitializedMemory,
1116 ));
1117
1118 let region = hal::BufferCopy {
1119 src_offset: source_offset,
1120 dst_offset: destination_offset,
1121 size: wgt::BufferSize::new(size).unwrap(),
1122 };
1123 let barriers = src_barrier
1124 .into_iter()
1125 .chain(dst_barrier)
1126 .collect::<Vec<_>>();
1127 unsafe {
1128 state.raw_encoder.transition_buffers(&barriers);
1129 state
1130 .raw_encoder
1131 .copy_buffer_to_buffer(src_raw, dst_raw, &[region]);
1132 }
1133
1134 Ok(())
1135}
1136
1137#[inline(never)]
1138pub(super) fn copy_buffer_to_texture(
1139 state: &mut EncodingState,
1140 source: &wgt::TexelCopyBufferInfo<Arc<Buffer>>,
1141 destination: &wgt::TexelCopyTextureInfo<Arc<Texture>>,
1142 copy_size: &Extent3d,
1143) -> Result<(), CommandEncoderError> {
1144 let dst_texture = &destination.texture;
1145 let src_buffer = &source.buffer;
1146
1147 dst_texture.same_device(state.device)?;
1148 src_buffer.same_device(state.device)?;
1149
1150 let (hal_copy_size, array_layer_count) = validate_texture_copy_range(
1151 destination,
1152 &dst_texture.desc,
1153 CopySide::Destination,
1154 copy_size,
1155 )?;
1156
1157 let (dst_range, dst_base) = extract_texture_selector(destination, copy_size, dst_texture)?;
1158
1159 let src_raw = src_buffer.try_raw(state.snatch_guard)?;
1160 src_buffer
1161 .check_usage(BufferUsages::COPY_SRC)
1162 .map_err(TransferError::MissingBufferUsage)?;
1163
1164 let dst_raw = dst_texture.try_inner(state.snatch_guard)?.raw();
1165 dst_texture
1166 .check_usage(TextureUsages::COPY_DST)
1167 .map_err(TransferError::MissingTextureUsage)?;
1168
1169 validate_texture_copy_dst_format(dst_texture.desc.format, destination.aspect)?;
1170
1171 validate_texture_buffer_copy(
1172 destination,
1173 dst_base.aspect,
1174 &dst_texture.desc,
1175 &source.layout,
1176 true, )?;
1178
1179 let (required_buffer_bytes_in_copy, bytes_per_array_layer, is_contiguous) =
1180 validate_linear_texture_data(
1181 &source.layout,
1182 dst_texture.desc.format,
1183 destination.aspect,
1184 src_buffer.size,
1185 CopySide::Source,
1186 copy_size,
1187 )?;
1188
1189 if dst_texture.desc.format.is_depth_stencil_format() {
1190 state
1191 .device
1192 .require_downlevel_flags(wgt::DownlevelFlags::DEPTH_TEXTURE_AND_BUFFER_COPIES)
1193 .map_err(TransferError::from)?;
1194 }
1195
1196 if copy_size.width == 0 || copy_size.height == 0 || copy_size.depth_or_array_layers == 0 {
1199 log::trace!("Ignoring copy_buffer_to_texture of size 0");
1200 return Ok(());
1201 }
1202
1203 handle_dst_texture_init(state, destination, copy_size, dst_texture)?;
1207
1208 let src_pending = state
1209 .tracker
1210 .buffers
1211 .set_single(src_buffer, wgt::BufferUses::COPY_SRC);
1212 let src_barrier = src_pending.map(|pending| pending.into_hal(src_buffer, state.snatch_guard));
1213
1214 let dst_pending =
1215 state
1216 .tracker
1217 .textures
1218 .set_single(dst_texture, dst_range, wgt::TextureUses::COPY_DST);
1219 let dst_barrier = dst_pending
1220 .map(|pending| pending.into_hal(dst_raw))
1221 .collect::<Vec<_>>();
1222
1223 handle_buffer_init(
1224 state,
1225 source,
1226 CopySide::Source,
1227 required_buffer_bytes_in_copy,
1228 is_contiguous,
1229 );
1230
1231 let regions = (0..array_layer_count)
1232 .map(|rel_array_layer| {
1233 let mut texture_base = dst_base.clone();
1234 texture_base.array_layer += rel_array_layer;
1235 let mut buffer_layout = source.layout;
1236 buffer_layout.offset += rel_array_layer as u64 * bytes_per_array_layer;
1237 hal::BufferTextureCopy {
1238 buffer_layout,
1239 texture_base,
1240 size: hal_copy_size,
1241 }
1242 })
1243 .collect::<Vec<_>>();
1244
1245 unsafe {
1246 state.raw_encoder.transition_textures(&dst_barrier);
1247 state.raw_encoder.transition_buffers(src_barrier.as_slice());
1248 state
1249 .raw_encoder
1250 .copy_buffer_to_texture(src_raw, dst_raw, ®ions);
1251 }
1252
1253 Ok(())
1254}
1255
1256#[inline(never)]
1257pub(super) fn copy_texture_to_buffer(
1258 state: &mut EncodingState,
1259 source: &TexelCopyTextureInfo,
1260 destination: &wgt::TexelCopyBufferInfo<Arc<Buffer>>,
1261 copy_size: &Extent3d,
1262) -> Result<(), CommandEncoderError> {
1263 let src_texture = &source.texture;
1264 let dst_buffer = &destination.buffer;
1265
1266 src_texture.same_device(state.device)?;
1267 dst_buffer.same_device(state.device)?;
1268
1269 let (hal_copy_size, array_layer_count) =
1270 validate_texture_copy_range(source, &src_texture.desc, CopySide::Source, copy_size)?;
1271
1272 let (src_range, src_base) = extract_texture_selector(source, copy_size, src_texture)?;
1273
1274 let src_raw = src_texture.try_inner(state.snatch_guard)?.raw();
1275 src_texture
1276 .check_usage(TextureUsages::COPY_SRC)
1277 .map_err(TransferError::MissingTextureUsage)?;
1278
1279 if source.mip_level >= src_texture.desc.mip_level_count {
1280 return Err(TransferError::InvalidMipLevel {
1281 requested: source.mip_level,
1282 count: src_texture.desc.mip_level_count,
1283 }
1284 .into());
1285 }
1286
1287 validate_texture_copy_src_format(src_texture.desc.format, source.aspect)?;
1288
1289 validate_texture_buffer_copy(
1290 source,
1291 src_base.aspect,
1292 &src_texture.desc,
1293 &destination.layout,
1294 true, )?;
1296
1297 let (required_buffer_bytes_in_copy, bytes_per_array_layer, is_contiguous) =
1298 validate_linear_texture_data(
1299 &destination.layout,
1300 src_texture.desc.format,
1301 source.aspect,
1302 dst_buffer.size,
1303 CopySide::Destination,
1304 copy_size,
1305 )?;
1306
1307 if src_texture.desc.format.is_depth_stencil_format() {
1308 state
1309 .device
1310 .require_downlevel_flags(wgt::DownlevelFlags::DEPTH_TEXTURE_AND_BUFFER_COPIES)
1311 .map_err(TransferError::from)?;
1312 }
1313
1314 let dst_raw = dst_buffer.try_raw(state.snatch_guard)?;
1315 dst_buffer
1316 .check_usage(BufferUsages::COPY_DST)
1317 .map_err(TransferError::MissingBufferUsage)?;
1318
1319 if copy_size.width == 0 || copy_size.height == 0 || copy_size.depth_or_array_layers == 0 {
1322 log::trace!("Ignoring copy_texture_to_buffer of size 0");
1323 return Ok(());
1324 }
1325
1326 handle_src_texture_init(state, source, copy_size, src_texture)?;
1330
1331 let src_pending =
1332 state
1333 .tracker
1334 .textures
1335 .set_single(src_texture, src_range, wgt::TextureUses::COPY_SRC);
1336 let src_barrier = src_pending
1337 .map(|pending| pending.into_hal(src_raw))
1338 .collect::<Vec<_>>();
1339
1340 let dst_pending = state
1341 .tracker
1342 .buffers
1343 .set_single(dst_buffer, wgt::BufferUses::COPY_DST);
1344
1345 let dst_barrier = dst_pending.map(|pending| pending.into_hal(dst_buffer, state.snatch_guard));
1346
1347 handle_buffer_init(
1348 state,
1349 destination,
1350 CopySide::Destination,
1351 required_buffer_bytes_in_copy,
1352 is_contiguous,
1353 );
1354
1355 let regions = (0..array_layer_count)
1356 .map(|rel_array_layer| {
1357 let mut texture_base = src_base.clone();
1358 texture_base.array_layer += rel_array_layer;
1359 let mut buffer_layout = destination.layout;
1360 buffer_layout.offset += rel_array_layer as u64 * bytes_per_array_layer;
1361 hal::BufferTextureCopy {
1362 buffer_layout,
1363 texture_base,
1364 size: hal_copy_size,
1365 }
1366 })
1367 .collect::<Vec<_>>();
1368 unsafe {
1369 state.raw_encoder.transition_buffers(dst_barrier.as_slice());
1370 state.raw_encoder.transition_textures(&src_barrier);
1371 state.raw_encoder.copy_texture_to_buffer(
1372 src_raw,
1373 wgt::TextureUses::COPY_SRC,
1374 dst_raw,
1375 ®ions,
1376 );
1377 }
1378
1379 Ok(())
1380}
1381
1382#[inline(never)]
1383pub(super) fn copy_texture_to_texture(
1384 state: &mut EncodingState,
1385 source: &TexelCopyTextureInfo,
1386 destination: &TexelCopyTextureInfo,
1387 copy_size: &Extent3d,
1388) -> Result<(), CommandEncoderError> {
1389 let src_texture = &source.texture;
1390 let dst_texture = &destination.texture;
1391
1392 src_texture.same_device(state.device)?;
1393 dst_texture.same_device(state.device)?;
1394
1395 let src_fmt_no_srgb = src_texture.desc.format.remove_srgb_suffix();
1409 let dst_fmt_no_srgb = dst_texture.desc.format.remove_srgb_suffix();
1410 let planar_split_ok = src_fmt_no_srgb.is_multi_planar_format()
1411 && src_fmt_no_srgb.aspect_specific_format(source.aspect) == Some(dst_fmt_no_srgb);
1412 if src_fmt_no_srgb != dst_fmt_no_srgb && !planar_split_ok {
1413 return Err(TransferError::TextureFormatsNotCopyCompatible {
1414 src_format: src_texture.desc.format,
1415 dst_format: dst_texture.desc.format,
1416 }
1417 .into());
1418 }
1419
1420 let (src_copy_size, array_layer_count) =
1421 validate_texture_copy_range(source, &src_texture.desc, CopySide::Source, copy_size)?;
1422 let (dst_copy_size, _) = validate_texture_copy_range(
1423 destination,
1424 &dst_texture.desc,
1425 CopySide::Destination,
1426 copy_size,
1427 )?;
1428
1429 if planar_split_ok {
1435 let plane = source.aspect.to_plane().expect("planar_split_ok aspect");
1438 let plane_extent = src_texture
1439 .desc
1440 .compute_render_extent(source.mip_level, Some(plane));
1441 let check = |dimension, start: u32, size: u32, plane_size: u32| {
1442 if start > plane_size || plane_size - start < size {
1443 Err(TransferError::TextureOverrun {
1444 start_offset: start,
1445 end_offset: start.wrapping_add(size),
1446 texture_size: plane_size,
1447 dimension,
1448 side: CopySide::Source,
1449 })
1450 } else {
1451 Ok(())
1452 }
1453 };
1454 check(
1455 TextureErrorDimension::X,
1456 source.origin.x,
1457 copy_size.width,
1458 plane_extent.width,
1459 )?;
1460 check(
1461 TextureErrorDimension::Y,
1462 source.origin.y,
1463 copy_size.height,
1464 plane_extent.height,
1465 )?;
1466 }
1467
1468 if Arc::as_ptr(src_texture) == Arc::as_ptr(dst_texture) {
1469 validate_copy_within_same_texture(
1470 source,
1471 destination,
1472 src_texture.desc.format,
1473 array_layer_count,
1474 )?;
1475 }
1476
1477 let (src_range, src_tex_base) = extract_texture_selector(source, copy_size, src_texture)?;
1478 let (dst_range, dst_tex_base) = extract_texture_selector(destination, copy_size, dst_texture)?;
1479 let src_texture_aspects = hal::FormatAspects::from(src_texture.desc.format);
1480 let dst_texture_aspects = hal::FormatAspects::from(dst_texture.desc.format);
1481 if src_tex_base.aspect != src_texture_aspects && !planar_split_ok {
1483 return Err(TransferError::CopySrcMissingAspects.into());
1484 }
1485 if dst_tex_base.aspect != dst_texture_aspects {
1486 return Err(TransferError::CopyDstMissingAspects.into());
1487 }
1488
1489 if src_texture.desc.sample_count != dst_texture.desc.sample_count {
1490 return Err(TransferError::SampleCountNotEqual {
1491 src_sample_count: src_texture.desc.sample_count,
1492 dst_sample_count: dst_texture.desc.sample_count,
1493 }
1494 .into());
1495 }
1496
1497 let src_raw = src_texture.try_inner(state.snatch_guard)?.raw();
1498 src_texture
1499 .check_usage(TextureUsages::COPY_SRC)
1500 .map_err(TransferError::MissingTextureUsage)?;
1501 let dst_raw = dst_texture.try_inner(state.snatch_guard)?.raw();
1502 dst_texture
1503 .check_usage(TextureUsages::COPY_DST)
1504 .map_err(TransferError::MissingTextureUsage)?;
1505
1506 if copy_size.width == 0 || copy_size.height == 0 || copy_size.depth_or_array_layers == 0 {
1509 log::trace!("Ignoring copy_texture_to_texture of size 0");
1510 return Ok(());
1511 }
1512
1513 handle_src_texture_init(state, source, copy_size, src_texture)?;
1517 handle_dst_texture_init(state, destination, copy_size, dst_texture)?;
1518
1519 let src_pending =
1520 state
1521 .tracker
1522 .textures
1523 .set_single(src_texture, src_range, wgt::TextureUses::COPY_SRC);
1524
1525 let mut barriers: ArrayVec<_, 2> = src_pending
1528 .map(|pending| pending.into_hal(src_raw))
1529 .collect();
1530
1531 let dst_pending =
1532 state
1533 .tracker
1534 .textures
1535 .set_single(dst_texture, dst_range, wgt::TextureUses::COPY_DST);
1536 barriers.extend(dst_pending.map(|pending| pending.into_hal(dst_raw)));
1537
1538 let hal_copy_size = hal::CopyExtent {
1539 width: src_copy_size.width.min(dst_copy_size.width),
1540 height: src_copy_size.height.min(dst_copy_size.height),
1541 depth: src_copy_size.depth.min(dst_copy_size.depth),
1542 };
1543
1544 let dst_format = dst_texture.desc.format;
1545
1546 let regions = (0..array_layer_count).map(|rel_array_layer| {
1547 let mut src_base = src_tex_base.clone();
1548 let mut dst_base = dst_tex_base.clone();
1549 src_base.array_layer += rel_array_layer;
1550 dst_base.array_layer += rel_array_layer;
1551 hal::TextureCopy {
1552 src_base,
1553 dst_base,
1554 size: hal_copy_size,
1555 }
1556 });
1557
1558 let regions = if dst_tex_base.aspect == hal::FormatAspects::DEPTH_STENCIL {
1559 regions
1560 .flat_map(|region| {
1561 let (mut depth, mut stencil) = (region.clone(), region);
1562 depth.src_base.aspect = hal::FormatAspects::DEPTH;
1563 depth.dst_base.aspect = hal::FormatAspects::DEPTH;
1564 stencil.src_base.aspect = hal::FormatAspects::STENCIL;
1565 stencil.dst_base.aspect = hal::FormatAspects::STENCIL;
1566 [depth, stencil]
1567 })
1568 .collect::<Vec<_>>()
1569 } else if let Some(plane_count) = dst_format.planes() {
1570 regions
1571 .into_iter()
1572 .flat_map(|region| {
1573 (0..plane_count).map(move |plane| {
1574 let mut plane_region = region.clone();
1575
1576 let plane_aspect = wgt::TextureAspect::from_plane(plane)
1577 .expect("expected texture aspect to exist for the plane");
1578 let plane_aspect = hal::FormatAspects::new(dst_format, plane_aspect);
1579 plane_region.src_base.aspect = plane_aspect;
1580 plane_region.dst_base.aspect = plane_aspect;
1581
1582 let (w_subsampling, h_subsampling) =
1583 dst_format.subsampling_factors(Some(plane));
1584 plane_region.src_base.origin.x /= w_subsampling;
1585 plane_region.src_base.origin.y /= h_subsampling;
1586 plane_region.dst_base.origin.x /= w_subsampling;
1587 plane_region.dst_base.origin.y /= h_subsampling;
1588
1589 plane_region.size.width /= w_subsampling;
1590 plane_region.size.height /= h_subsampling;
1591
1592 plane_region
1593 })
1594 })
1595 .collect::<Vec<_>>()
1596 } else {
1597 regions.collect::<Vec<_>>()
1598 };
1599 unsafe {
1600 state.raw_encoder.transition_textures(&barriers);
1601 state.raw_encoder.copy_texture_to_texture(
1602 src_raw,
1603 wgt::TextureUses::COPY_SRC,
1604 dst_raw,
1605 ®ions,
1606 );
1607 }
1608
1609 Ok(())
1610}