Skip to main content

wgpu_core/command/
transfer.rs

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/// Error encountered while attempting a data transfer.
41#[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        // this value will be incremented per copied layer
274        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
286/// WebGPU's [validating linear texture data][vltd] algorithm.
287///
288/// Copied with some modifications from WebGPU standard.
289///
290/// If successful, returns a tuple `(bytes, stride, is_contiguous)`, where:
291/// - `bytes` is the number of buffer bytes required for this copy, and
292/// - `stride` number of bytes between array layers.
293/// - `is_contiguous` is true if the linear texture data does not have padding
294///   between rows or between images.
295///
296/// [vltd]: https://gpuweb.github.io/gpuweb/#abstract-opdef-validating-linear-texture-data
297pub(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    // `get_buffer_texture_copy_info()` already proceeded with defaults if these
342    // were not specified, and ensured that the values satisfy the minima if
343    // they were, but now we enforce the WebGPU requirement that they be
344    // specified any time they apply.
345    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    // NOTE: Should never underflow because of our earlier check.
361    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
376/// Validate the source format of a texture copy.
377///
378/// This performs the check from WebGPU's [validating texture buffer copy][vtbc]
379/// algorithm that ensures that the format and aspect form a valid texel copy source
380/// as defined in the [depth-stencil formats][dsf].
381///
382/// [vtbc]: https://gpuweb.github.io/gpuweb/#abstract-opdef-validating-texture-buffer-copy
383/// [dsf]: https://gpuweb.github.io/gpuweb/#depth-formats
384pub(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
399/// Validate the destination format of a texture copy.
400///
401/// This performs the check from WebGPU's [validating texture buffer copy][vtbc]
402/// algorithm that ensures that the format and aspect form a valid texel copy destination
403/// as defined in the [depth-stencil formats][dsf].
404///
405/// [vtbc]: https://gpuweb.github.io/gpuweb/#abstract-opdef-validating-texture-buffer-copy
406/// [dsf]: https://gpuweb.github.io/gpuweb/#depth-formats
407pub(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
424/// Validation for texture/buffer copies.
425///
426/// This implements the following checks from WebGPU's [validating texture buffer copy][vtbc]
427/// algorithm:
428///  * The texture must not be multisampled.
429///  * The copy must be from/to a single aspect of the texture.
430///  * If `aligned` is true, the buffer offset must be aligned appropriately.
431///
432/// And implements the following check from WebGPU's [validating GPUTexelCopyBufferInfo][vtcbi]
433/// algorithm:
434///  * If `aligned` is true, `bytesPerRow` must be a multiple of 256.
435///
436/// Note that the `bytesPerRow` alignment check is enforced whenever
437/// `bytesPerRow` is specified, even if the transfer is not multiple rows and
438/// `bytesPerRow` could have been omitted.
439///
440/// The following steps in [validating texture buffer copy][vtbc] are implemented elsewhere:
441///  * Invocation of other validation algorithms.
442///  * The texture usage (COPY_DST / COPY_SRC) check.
443///  * The check for non-copyable depth/stencil formats. The caller must perform
444///    this check using `validate_texture_copy_src_format` / `validate_texture_copy_dst_format`
445///    before calling this function. This function will panic if
446///    [`wgt::TextureFormat::block_copy_size`] returns `None` due to a
447///    non-copyable format.
448///
449/// [vtbc]: https://gpuweb.github.io/gpuweb/#abstract-opdef-validating-texture-buffer-copy
450/// [vtcbi]: https://www.w3.org/TR/webgpu/#abstract-opdef-validating-gputexelcopybufferinfo
451pub(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        // The case where `block_copy_size` returns `None` is currently
472        // unreachable both for the reason in the expect message, and also
473        // because the currently-defined non-copyable formats are depth/stencil
474        // formats so would take the `if` branch.
475        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
493/// Validate the extent and alignment of a texture copy.
494///
495/// Copied with minor modifications from WebGPU standard. This mostly follows
496/// the [validating GPUTexelCopyTextureInfo][vtcti] and [validating texture copy
497/// range][vtcr] algorithms.
498///
499/// Returns the HAL copy extent and the layer count.
500///
501/// [vtcti]: https://gpuweb.github.io/gpuweb/#abstract-opdef-validating-gputexelcopytextureinfo
502/// [vtcr]: https://gpuweb.github.io/gpuweb/#abstract-opdef-validating-texture-copy-range
503pub(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    // physical size can be larger than the virtual
518    let extent = extent_virtual.physical_size(desc.format);
519
520    // Multisampled and depth-stencil formats do not support partial copies
521    // on x and y dimensions, but do support copying a subset of layers.
522    let requires_exact_size = desc.format.is_depth_stencil_format() || desc.sample_count > 1;
523
524    // Return `Ok` if a run `size` texels long starting at `start_offset` is
525    // valid for `texture_size`. Otherwise, return an appropriate a`Err`.
526    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        // Avoid underflow in the subtraction by checking start_offset against
543        // texture_size first.
544        } 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, // partial copy always allowed on Z/layer dimension
577    )?;
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
606/// Validate a copy within the same texture.
607///
608/// This implements the WebGPU requirement that the [sets of subresources for
609/// texture copy][srtc] of the source and destination be disjoint, i.e. that the
610/// source and destination do not overlap.
611///
612/// This function assumes that the copy ranges have already been validated with
613/// `validate_texture_copy_range`.
614///
615/// [srtc]: https://gpuweb.github.io/gpuweb/#abstract-opdef-set-of-subresources-for-texture-copy
616pub(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        // Copying between different aspects (if it even makes sense), is okay.
626        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        // Copying between non-overlapping layer ranges is okay.
633        return Ok(());
634    }
635
636    if src.mip_level != dst.mip_level {
637        // Copying between different mip levels is okay.
638        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        // Init tracking only considers array layers, not depth/volume slices
659        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    // Register the init action.
673    let immediate_inits = state
674        .texture_memory_actions
675        .register_init_action(&{ init_action });
676
677    // In rare cases we may need to insert an init operation immediately onto the command buffer.
678    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
699/// Prepare a transfer's source texture.
700///
701/// Ensure the source texture of a transfer is in the right initialization
702/// state, and record the state for after the transfer operation.
703fn 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
719/// Prepare a transfer's destination texture.
720///
721/// Ensure the destination texture of a transfer is in the right initialization
722/// state, and record the state for after the transfer operation.
723fn 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    // Attention: If we don't write full texture subresources, we need to a full
730    // clear first since we don't track subrects. This means that in rare cases
731    // even a *destination* texture of a transfer may need an immediate texture
732    // init.
733    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
744/// Handle initialization tracking for a transfer's source or destination buffer.
745///
746/// Ensures that the transfer will not read from uninitialized memory, and updates
747/// the initialization state information to reflect the transfer.
748fn 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        // If the transfer will read the buffer, then the whole region needs to
763        // be initialized.
764        //
765        // If the transfer will not write a contiguous region of the buffer,
766        // then we need to make sure the padding areas are initialized. For now,
767        // initialize the whole region, although this could be improved to
768        // initialize only the necessary parts if doing so is likely to be
769        // faster than initializing the whole thing.
770        //
771        // Adjust the start/end outwards to 4B alignment.
772        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        // If the transfer will write a contiguous region of the buffer, then we
783        // don't need to initialize that region.
784        //
785        // However, if the start and end are not 4B aligned, we need to make
786        // sure that we don't end up trying to initialize non-4B-aligned regions
787        // later.
788        //
789        // Adjust the start/end inwards to 4B alignment, we will handle the
790        // first/last pieces differently.
791        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            // It is possible that `aligned_end + ALIGN_SIZE > dst_buffer.size`,
813            // because `dst_buffer.size` is the user-requested size, not the
814            // final size of the buffer. The final size of the buffer is not
815            // readily available, but was rounded up to COPY_BUFFER_ALIGNMENT,
816            // so no overrun is possible.
817            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    // expecting only a single barrier
1003    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        // NOTE: Should never underflow because of our earlier check.
1028        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    // NOTE: Should never overflow because of our earlier check.
1072    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    // NOTE: Should never underflow because of our earlier check.
1083    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    // NOTE: Should never overflow because of our earlier check.
1093    let destination_end_offset = destination_offset + size;
1094
1095    // This must happen after parameter validation (so that errors are reported
1096    // as required by the spec), but before any side effects.
1097    if size == 0 {
1098        log::trace!("Ignoring copy_buffer_to_buffer of size 0");
1099        return Ok(());
1100    }
1101
1102    // Make sure source is initialized memory and mark dest as initialized.
1103    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, // alignment required for buffer offset
1177    )?;
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    // This must happen after parameter validation (so that errors are reported
1197    // as required by the spec), but before any side effects.
1198    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 texture init *before* dealing with barrier transitions so we
1204    // have an easier time inserting "immediate-inits" that may be required
1205    // by prior discards in rare cases.
1206    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, &regions);
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, // alignment required for buffer offset
1295    )?;
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    // This must happen after parameter validation (so that errors are reported
1320    // as required by the spec), but before any side effects.
1321    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 texture init *before* dealing with barrier transitions so we
1327    // have an easier time inserting "immediate-inits" that may be required
1328    // by prior discards in rare cases.
1329    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            &regions,
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    // src and dst texture format must be copy-compatible
1396    // (https://gpuweb.github.io/gpuweb/#copy-compatible), with an
1397    // extension allowing one plane of a planar source to be copied
1398    // into a single-plane destination of the matching format
1399    // (e.g. NV12 Plane0 -> R8Unorm, NV12 Plane1 -> Rg8Unorm).
1400    //
1401    // When taking this path, `copy_size` and `source.origin` are
1402    // interpreted in *plane* texels, not luma texels: copying NV12
1403    // Plane1 into an Rg8Unorm of size (W/2, H/2) requires
1404    // `copy_size = (W/2, H/2)`. The plane-extent check further down
1405    // enforces this against the subsampled plane extent, so a caller
1406    // passing luma-sized values gets a source-side error pointing at
1407    // the actual mistake rather than an opaque destination overrun.
1408    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    // For planar -> single-plane copies, re-check the source extent
1430    // in plane coordinates. `validate_texture_copy_range` above used
1431    // the full luma extent of the planar source, so it does not
1432    // catch a caller treating `copy_size` / `origin` as luma-sized
1433    // when targeting a subsampled plane (NV12/P010 plane 1).
1434    if planar_split_ok {
1435        // `planar_split_ok` implies `aspect_specific_format(source.aspect)`
1436        // returned `Some`, which is only true for `Plane{0,1,2}`.
1437        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    // `planar_split_ok` already constrains `source.aspect` to a single plane.
1482    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    // This must happen after parameter validation (so that errors are reported
1507    // as required by the spec), but before any side effects.
1508    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 texture init *before* dealing with barrier transitions so we
1514    // have an easier time inserting "immediate-inits" that may be required
1515    // by prior discards in rare cases.
1516    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    //TODO: try to avoid this the collection. It's needed because both
1526    // `src_pending` and `dst_pending` try to hold `trackers.textures` mutably.
1527    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            &regions,
1606        );
1607    }
1608
1609    Ok(())
1610}