1use alloc::borrow::ToOwned as _;
2
3use wgt::TextureDataOrder;
4
5#[derive(Clone, Debug, PartialEq, Eq, Hash)]
7pub struct BufferInitDescriptor<'a> {
8 pub label: crate::Label<'a>,
10 pub contents: &'a [u8],
12 pub usage: wgt::BufferUsages,
15}
16
17pub trait DeviceExt {
19 fn create_buffer_init(&self, desc: &BufferInitDescriptor<'_>) -> crate::Buffer;
21
22 fn create_texture_with_data(
31 &self,
32 queue: &crate::Queue,
33 desc: &crate::TextureDescriptor<'_>,
34 order: TextureDataOrder,
35 data: &[u8],
36 ) -> crate::Texture;
37}
38
39impl DeviceExt for crate::Device {
40 fn create_buffer_init(&self, descriptor: &BufferInitDescriptor<'_>) -> crate::Buffer {
41 if descriptor.contents.is_empty() {
43 let wgt_descriptor = crate::BufferDescriptor {
44 label: descriptor.label,
45 size: 0,
46 usage: descriptor.usage,
47 mapped_at_creation: false,
48 };
49
50 self.create_buffer(&wgt_descriptor)
51 } else {
52 let unpadded_size = descriptor.contents.len() as crate::BufferAddress;
53 let align_mask = crate::COPY_BUFFER_ALIGNMENT - 1;
58 let padded_size =
59 ((unpadded_size + align_mask) & !align_mask).max(crate::COPY_BUFFER_ALIGNMENT);
60
61 let wgt_descriptor = crate::BufferDescriptor {
62 label: descriptor.label,
63 size: padded_size,
64 usage: descriptor.usage,
65 mapped_at_creation: true,
66 };
67
68 let buffer = self.create_buffer(&wgt_descriptor);
69
70 buffer.slice(..).get_mapped_range_mut()[..unpadded_size as usize]
71 .copy_from_slice(descriptor.contents);
72 buffer.unmap();
73
74 buffer
75 }
76 }
77
78 fn create_texture_with_data(
79 &self,
80 queue: &crate::Queue,
81 desc: &crate::TextureDescriptor<'_>,
82 order: TextureDataOrder,
83 data: &[u8],
84 ) -> crate::Texture {
85 let mut desc = desc.to_owned();
87 desc.usage |= crate::TextureUsages::COPY_DST;
88 let texture = self.create_texture(&desc);
89
90 let block_size = desc.format.block_copy_size(None).unwrap_or(4);
94 let (block_width, block_height) = desc.format.block_dimensions();
95 let layer_iterations = desc.array_layer_count();
96
97 let outer_iteration;
98 let inner_iteration;
99 match order {
100 TextureDataOrder::LayerMajor => {
101 outer_iteration = layer_iterations;
102 inner_iteration = desc.mip_level_count;
103 }
104 TextureDataOrder::MipMajor => {
105 outer_iteration = desc.mip_level_count;
106 inner_iteration = layer_iterations;
107 }
108 }
109
110 let mut binary_offset = 0;
111 for outer in 0..outer_iteration {
112 for inner in 0..inner_iteration {
113 let (layer, mip) = match order {
114 TextureDataOrder::LayerMajor => (outer, inner),
115 TextureDataOrder::MipMajor => (inner, outer),
116 };
117
118 let mut mip_size = desc.mip_level_size(mip).unwrap();
119 if desc.dimension != wgt::TextureDimension::D3 {
121 mip_size.depth_or_array_layers = 1;
122 }
123
124 let mip_physical = mip_size.physical_size(desc.format);
128
129 let width_blocks = mip_physical.width / block_width;
132 let height_blocks = mip_physical.height / block_height;
133
134 let bytes_per_row = width_blocks * block_size;
135 let data_size = bytes_per_row * height_blocks * mip_size.depth_or_array_layers;
136
137 let end_offset = binary_offset + data_size as usize;
138
139 queue.write_texture(
140 crate::TexelCopyTextureInfo {
141 texture: &texture,
142 mip_level: mip,
143 origin: crate::Origin3d {
144 x: 0,
145 y: 0,
146 z: layer,
147 },
148 aspect: wgt::TextureAspect::All,
149 },
150 &data[binary_offset..end_offset],
151 crate::TexelCopyBufferLayout {
152 offset: 0,
153 bytes_per_row: Some(bytes_per_row),
154 rows_per_image: Some(height_blocks),
155 },
156 mip_physical,
157 );
158
159 binary_offset = end_offset;
160 }
161 }
162
163 texture
164 }
165}