78 lines
2.1 KiB
Rust
78 lines
2.1 KiB
Rust
use crate::{Image, TextureFormatPixelInfo};
|
|
use bevy_asset::{io::Reader, AssetLoader, LoadContext, RenderAssetUsages};
|
|
use image::ImageDecoder;
|
|
use serde::{Deserialize, Serialize};
|
|
use thiserror::Error;
|
|
use wgpu_types::{Extent3d, TextureDimension, TextureFormat};
|
|
|
|
/// Loads EXR textures as Texture assets
|
|
#[derive(Clone, Default)]
|
|
#[cfg(feature = "exr")]
|
|
pub struct ExrTextureLoader;
|
|
|
|
#[derive(Serialize, Deserialize, Default, Debug)]
|
|
#[cfg(feature = "exr")]
|
|
pub struct ExrTextureLoaderSettings {
|
|
pub asset_usage: RenderAssetUsages,
|
|
}
|
|
|
|
/// Possible errors that can be produced by [`ExrTextureLoader`]
|
|
#[non_exhaustive]
|
|
#[derive(Debug, Error)]
|
|
#[cfg(feature = "exr")]
|
|
pub enum ExrTextureLoaderError {
|
|
#[error(transparent)]
|
|
Io(#[from] std::io::Error),
|
|
#[error(transparent)]
|
|
ImageError(#[from] image::ImageError),
|
|
}
|
|
|
|
impl AssetLoader for ExrTextureLoader {
|
|
type Asset = Image;
|
|
type Settings = ExrTextureLoaderSettings;
|
|
type Error = ExrTextureLoaderError;
|
|
|
|
async fn load(
|
|
&self,
|
|
reader: &mut dyn Reader,
|
|
settings: &Self::Settings,
|
|
_load_context: &mut LoadContext<'_>,
|
|
) -> Result<Image, Self::Error> {
|
|
let format = TextureFormat::Rgba32Float;
|
|
debug_assert_eq!(
|
|
format.pixel_size(),
|
|
4 * 4,
|
|
"Format should have 32bit x 4 size"
|
|
);
|
|
|
|
let mut bytes = Vec::new();
|
|
reader.read_to_end(&mut bytes).await?;
|
|
let decoder = image::codecs::openexr::OpenExrDecoder::with_alpha_preference(
|
|
std::io::Cursor::new(bytes),
|
|
Some(true),
|
|
)?;
|
|
let (width, height) = decoder.dimensions();
|
|
|
|
let total_bytes = decoder.total_bytes() as usize;
|
|
|
|
let mut buf = vec![0u8; total_bytes];
|
|
decoder.read_image(buf.as_mut_slice())?;
|
|
|
|
Ok(Image::new(
|
|
Extent3d {
|
|
width,
|
|
height,
|
|
depth_or_array_layers: 1,
|
|
},
|
|
TextureDimension::D2,
|
|
buf,
|
|
format,
|
|
settings.asset_usage,
|
|
))
|
|
}
|
|
|
|
fn extensions(&self) -> &[&str] {
|
|
&["exr"]
|
|
}
|
|
}
|