108 lines
3.1 KiB
Rust
108 lines
3.1 KiB
Rust
use image::GenericImageView;
|
|
use thiserror::Error;
|
|
#[derive(Debug, Error)]
|
|
pub enum TextureError {
|
|
#[error("Failed to load image")]
|
|
ImageError(
|
|
#[from]
|
|
#[source]
|
|
image::ImageError,
|
|
),
|
|
}
|
|
pub struct Texture {
|
|
#[allow(unused)]
|
|
pub texture: wgpu::Texture,
|
|
pub view: wgpu::TextureView,
|
|
pub sampler: wgpu::Sampler,
|
|
}
|
|
impl Texture {
|
|
pub fn from_bytes(
|
|
device: &wgpu::Device,
|
|
queue: &wgpu::Queue,
|
|
bytes: &[u8],
|
|
label: &str,
|
|
) -> Result<Self, TextureError> {
|
|
let img = image::load_from_memory(bytes)?;
|
|
Self::from_image(device, queue, &img, Some(label))
|
|
}
|
|
|
|
pub fn from_image(
|
|
device: &wgpu::Device,
|
|
queue: &wgpu::Queue,
|
|
img: &image::DynamicImage,
|
|
label: Option<&str>,
|
|
) -> Result<Self, TextureError> {
|
|
let rgba = img.to_rgba8();
|
|
let dimensions = img.dimensions();
|
|
|
|
let size = wgpu::Extent3d {
|
|
width: dimensions.0,
|
|
height: dimensions.1,
|
|
depth_or_array_layers: 1,
|
|
};
|
|
let texture = device.create_texture(&wgpu::TextureDescriptor {
|
|
label,
|
|
size,
|
|
mip_level_count: 1,
|
|
sample_count: 1,
|
|
dimension: wgpu::TextureDimension::D2,
|
|
format: wgpu::TextureFormat::Rgba8UnormSrgb,
|
|
usage: wgpu::TextureUsages::TEXTURE_BINDING | wgpu::TextureUsages::COPY_DST,
|
|
view_formats: &[],
|
|
});
|
|
|
|
queue.write_texture(
|
|
wgpu::TexelCopyTextureInfo {
|
|
aspect: wgpu::TextureAspect::All,
|
|
texture: &texture,
|
|
mip_level: 0,
|
|
origin: wgpu::Origin3d::ZERO,
|
|
},
|
|
&rgba,
|
|
wgpu::TexelCopyBufferLayout {
|
|
offset: 0,
|
|
bytes_per_row: Some(4 * dimensions.0),
|
|
rows_per_image: Some(dimensions.1),
|
|
},
|
|
size,
|
|
);
|
|
|
|
let view = texture.create_view(&wgpu::TextureViewDescriptor::default());
|
|
let sampler = device.create_sampler(&wgpu::SamplerDescriptor {
|
|
address_mode_u: wgpu::AddressMode::ClampToEdge,
|
|
address_mode_v: wgpu::AddressMode::ClampToEdge,
|
|
address_mode_w: wgpu::AddressMode::ClampToEdge,
|
|
mag_filter: wgpu::FilterMode::Linear,
|
|
min_filter: wgpu::FilterMode::Nearest,
|
|
mipmap_filter: wgpu::FilterMode::Nearest,
|
|
..Default::default()
|
|
});
|
|
|
|
Ok(Self {
|
|
texture,
|
|
view,
|
|
sampler,
|
|
})
|
|
}
|
|
}
|
|
|
|
pub fn create_checkerboard() -> image::DynamicImage {
|
|
let size = 512;
|
|
let tile_size = 24;
|
|
let mut img = image::RgbaImage::new(size, size);
|
|
|
|
for y in 0..size {
|
|
for x in 0..size {
|
|
let tile_x = x / tile_size;
|
|
let tile_y = y / tile_size;
|
|
let color = if (tile_x + tile_y) % 2 == 0 {
|
|
[0, 0, 0, 255]
|
|
} else {
|
|
[255, 5, 255, 255]
|
|
};
|
|
img.put_pixel(x, y, image::Rgba(color));
|
|
}
|
|
}
|
|
|
|
image::DynamicImage::ImageRgba8(img)
|
|
}
|