use super::{BufferId, RenderResourceBinding, SamplerId, TextureId};
use bevy_utils::AHasher;
use std::{
hash::{Hash, Hasher},
ops::Range,
sync::Arc,
};
#[derive(Hash, Eq, PartialEq, Debug, Copy, Clone)]
pub struct BindGroupId(pub u64);
#[derive(Eq, PartialEq, Debug)]
pub struct IndexedBindGroupEntry {
pub index: u32,
pub entry: RenderResourceBinding,
}
#[derive(Clone, Eq, PartialEq, Debug)]
pub struct BindGroup {
pub id: BindGroupId,
pub indexed_bindings: Arc<[IndexedBindGroupEntry]>,
pub dynamic_uniform_indices: Option<Arc<[u32]>>,
}
impl BindGroup {
pub fn build() -> BindGroupBuilder {
BindGroupBuilder::default()
}
}
#[derive(Debug, Default)]
pub struct BindGroupBuilder {
pub indexed_bindings: Vec<IndexedBindGroupEntry>,
pub dynamic_uniform_indices: Vec<u32>,
pub hasher: AHasher,
}
impl BindGroupBuilder {
pub fn add_binding(mut self, index: u32, binding: RenderResourceBinding) -> Self {
if let RenderResourceBinding::Buffer {
dynamic_index: Some(dynamic_index),
..
} = binding
{
self.dynamic_uniform_indices.push(dynamic_index);
}
binding.hash(&mut self.hasher);
self.indexed_bindings.push(IndexedBindGroupEntry {
index,
entry: binding,
});
self
}
pub fn add_texture(self, index: u32, texture: TextureId) -> Self {
self.add_binding(index, RenderResourceBinding::Texture(texture))
}
pub fn add_sampler(self, index: u32, sampler: SamplerId) -> Self {
self.add_binding(index, RenderResourceBinding::Sampler(sampler))
}
pub fn add_buffer(self, index: u32, buffer: BufferId, range: Range<u64>) -> Self {
self.add_binding(
index,
RenderResourceBinding::Buffer {
buffer,
range,
dynamic_index: None,
},
)
}
pub fn add_dynamic_buffer(
self,
index: u32,
buffer: BufferId,
range: Range<u64>,
dynamic_index: u32,
) -> Self {
self.add_binding(
index,
RenderResourceBinding::Buffer {
buffer,
range,
dynamic_index: Some(dynamic_index),
},
)
}
pub fn finish(mut self) -> BindGroup {
self.indexed_bindings.sort_by_key(|i| i.index);
BindGroup {
id: BindGroupId(self.hasher.finish()),
indexed_bindings: self.indexed_bindings.into(),
dynamic_uniform_indices: if self.dynamic_uniform_indices.is_empty() {
None
} else {
Some(self.dynamic_uniform_indices.into())
},
}
}
}