62 lines
2.3 KiB
Rust
62 lines
2.3 KiB
Rust
|
use librashader_common::{FilterMode, WrapMode};
|
||
|
|
use crate::error;
|
||
|
|
use rustc_hash::FxHashMap;
|
||
|
|
use std::sync::Arc;
|
||
|
|
use wgpu::{Sampler, SamplerBorderColor, SamplerDescriptor};
|
||
|
|
|
||
|
|
pub struct SamplerSet {
|
||
|
|
// todo: may need to deal with differences in mip filter.
|
||
|
|
samplers: FxHashMap<(WrapMode, FilterMode, FilterMode), Sampler>,
|
||
|
|
}
|
||
|
|
|
||
|
|
impl SamplerSet {
|
||
|
|
#[inline(always)]
|
||
|
|
pub fn get(&self, wrap: WrapMode, filter: FilterMode, mipmap: FilterMode) -> &Sampler {
|
||
|
|
// eprintln!("{wrap}, {filter}, {mip}");
|
||
|
|
// SAFETY: the sampler set is complete for the matrix
|
||
|
|
// wrap x filter x mipmap
|
||
|
|
unsafe {
|
||
|
|
self.samplers
|
||
|
|
.get(&(wrap, filter, mipmap))
|
||
|
|
.unwrap_unchecked()
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
pub fn new(device: &wgpu::Device) -> error::Result<SamplerSet> {
|
||
|
|
let mut samplers = FxHashMap::default();
|
||
|
|
let wrap_modes = &[
|
||
|
|
WrapMode::ClampToBorder,
|
||
|
|
WrapMode::ClampToEdge,
|
||
|
|
WrapMode::Repeat,
|
||
|
|
WrapMode::MirroredRepeat,
|
||
|
|
];
|
||
|
|
for wrap_mode in wrap_modes {
|
||
|
|
for filter_mode in &[FilterMode::Linear, FilterMode::Nearest] {
|
||
|
|
for mipmap_filter in &[FilterMode::Linear, FilterMode::Nearest] {
|
||
|
|
samplers.insert(
|
||
|
|
(*wrap_mode, *filter_mode, *mipmap_filter),
|
||
|
|
device.create_sampler(&SamplerDescriptor {
|
||
|
|
label: None,
|
||
|
|
address_mode_u: wrap_mode.into(),
|
||
|
|
address_mode_v: wrap_mode.into(),
|
||
|
|
address_mode_w: wrap_mode.into(),
|
||
|
|
mag_filter: filter_mode.into(),
|
||
|
|
min_filter: filter_mode.into(),
|
||
|
|
mipmap_filter: mipmap_filter.into(),
|
||
|
|
lod_min_clamp: 0.0,
|
||
|
|
lod_max_clamp: 1000.0,
|
||
|
|
compare: None,
|
||
|
|
anisotropy_clamp: 1,
|
||
|
|
border_color: Some(SamplerBorderColor::TransparentBlack),
|
||
|
|
})
|
||
|
|
);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// assert all samplers were created.
|
||
|
|
assert_eq!(samplers.len(), wrap_modes.len() * 2 * 2);
|
||
|
|
Ok(SamplerSet { samplers })
|
||
|
|
}
|
||
|
|
}
|