338 lines
11 KiB
Rust
338 lines
11 KiB
Rust
|
use std::collections::HashMap;
|
||
|
|
use std::hash::BuildHasher;
|
||
|
|
use librashader_common::Size;
|
||
|
|
use librashader_preprocess::ShaderParameter;
|
||
|
|
use librashader_reflect::reflect::semantics::{MemberOffset, Semantic, TextureBinding, TextureSemantics, UniformBinding, UniqueSemantics};
|
||
|
|
use crate::uniforms::{BindUniform, NoUniformBinder, UniformStorage};
|
||
|
|
|
||
|
|
/// Trait for input textures used during uniform binding,
|
||
|
|
pub trait TextureInput {
|
||
|
|
/// Gets the size of this input texture.
|
||
|
|
fn size(&self) -> Size<u32>;
|
||
|
|
}
|
||
|
|
|
||
|
|
/// A uniform member offset with context that needs to be resolved.
|
||
|
|
pub trait ContextOffset<H, C>
|
||
|
|
where
|
||
|
|
H: BindUniform<C, f32>,
|
||
|
|
H: BindUniform<C, u32>,
|
||
|
|
H: BindUniform<C, i32>,
|
||
|
|
H: for<'a> BindUniform<C, &'a [f32; 4]>,
|
||
|
|
H: for<'a> BindUniform<C, &'a [f32; 16]>,
|
||
|
|
{
|
||
|
|
/// Gets the `MemberOffset` part of the offset.
|
||
|
|
fn offset(&self) -> MemberOffset;
|
||
|
|
|
||
|
|
/// Gets the context part of the offset.
|
||
|
|
fn context(&self) -> C;
|
||
|
|
}
|
||
|
|
|
||
|
|
impl<H> ContextOffset<H, Option<()>> for MemberOffset
|
||
|
|
where
|
||
|
|
H: BindUniform<Option<()>, f32>,
|
||
|
|
H: BindUniform<Option<()>, u32>,
|
||
|
|
H: BindUniform<Option<()>, i32>,
|
||
|
|
H: for<'a> BindUniform<Option<()>, &'a [f32; 4]>,
|
||
|
|
H: for<'a> BindUniform<Option<()>, &'a [f32; 16]>,
|
||
|
|
{
|
||
|
|
fn offset(&self) -> MemberOffset {
|
||
|
|
*self
|
||
|
|
}
|
||
|
|
|
||
|
|
fn context(&self) -> Option<()> {
|
||
|
|
None
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
/// Trait that abstracts binding of semantics to shader uniforms.
|
||
|
|
pub trait BindSemantics<H=NoUniformBinder, C=Option<()>>
|
||
|
|
where
|
||
|
|
H: BindUniform<C, f32>,
|
||
|
|
H: BindUniform<C, u32>,
|
||
|
|
H: BindUniform<C, i32>,
|
||
|
|
H: for<'b> BindUniform<C, &'b [f32; 4]>,
|
||
|
|
H: for<'b> BindUniform<C, &'b [f32; 16]>
|
||
|
|
{
|
||
|
|
/// The type of the input texture used for semantic binding.
|
||
|
|
type InputTexture: TextureInput;
|
||
|
|
|
||
|
|
/// The set of texture samplers available.
|
||
|
|
type SamplerSet;
|
||
|
|
|
||
|
|
/// The descriptor set or object that holds sampler and texture bindings.
|
||
|
|
type DescriptorSet<'a>;
|
||
|
|
|
||
|
|
/// The device context containing the state of the graphics processor.
|
||
|
|
type DeviceContext;
|
||
|
|
|
||
|
|
/// The type of uniform offsets to use.
|
||
|
|
type UniformOffset: ContextOffset<H, C>;
|
||
|
|
|
||
|
|
|
||
|
|
/// Bind a texture to the input descriptor set
|
||
|
|
fn bind_texture<'a>(
|
||
|
|
descriptors: &mut Self::DescriptorSet<'a>,
|
||
|
|
samplers: &Self::SamplerSet,
|
||
|
|
binding: &TextureBinding,
|
||
|
|
texture: &Self::InputTexture,
|
||
|
|
device: &Self::DeviceContext
|
||
|
|
);
|
||
|
|
|
||
|
|
#[clippy::allow(too_many_arguments)]
|
||
|
|
/// Write uniform and texture semantics to the provided storages.
|
||
|
|
fn bind_semantics<'a>(
|
||
|
|
device: &Self::DeviceContext,
|
||
|
|
sampler_set: &Self::SamplerSet,
|
||
|
|
uniform_storage: &mut UniformStorage<H, C>,
|
||
|
|
descriptor_set: &mut Self::DescriptorSet<'a>,
|
||
|
|
mvp: &[f32; 16],
|
||
|
|
frame_count: u32,
|
||
|
|
frame_direction: i32,
|
||
|
|
framebuffer_size: Size<u32>,
|
||
|
|
viewport_size: Size<u32>,
|
||
|
|
original: &Self::InputTexture,
|
||
|
|
source: &Self::InputTexture,
|
||
|
|
uniform_bindings: &HashMap<UniformBinding, Self::UniformOffset, impl BuildHasher>,
|
||
|
|
texture_meta: &HashMap<Semantic<TextureSemantics>, TextureBinding, impl BuildHasher>,
|
||
|
|
pass_outputs: impl Iterator<Item = Option<impl AsRef<Self::InputTexture>>>,
|
||
|
|
pass_feedback: impl Iterator<Item = Option<impl AsRef<Self::InputTexture>>>,
|
||
|
|
original_history: impl Iterator<Item = Option<impl AsRef<Self::InputTexture>>>,
|
||
|
|
lookup_textures: impl Iterator<Item = (usize, impl AsRef<Self::InputTexture>)>,
|
||
|
|
parameter_defaults: &[ShaderParameter],
|
||
|
|
runtime_parameters: &HashMap<String, f32, impl BuildHasher>,
|
||
|
|
)
|
||
|
|
{
|
||
|
|
// Bind MVP
|
||
|
|
if let Some(offset) = uniform_bindings.get(&UniqueSemantics::MVP.into()) {
|
||
|
|
uniform_storage.bind_mat4(offset.offset(), mvp, offset.context());
|
||
|
|
}
|
||
|
|
|
||
|
|
// Bind OutputSize
|
||
|
|
if let Some(offset) = uniform_bindings.get(&UniqueSemantics::Output.into()) {
|
||
|
|
uniform_storage.bind_vec4(offset.offset(), framebuffer_size, offset.context());
|
||
|
|
}
|
||
|
|
|
||
|
|
// bind FinalViewportSize
|
||
|
|
if let Some(offset) = uniform_bindings
|
||
|
|
.get(&UniqueSemantics::FinalViewport.into())
|
||
|
|
{
|
||
|
|
uniform_storage.bind_vec4(offset.offset(), viewport_size, offset.context());
|
||
|
|
}
|
||
|
|
|
||
|
|
// bind FrameCount
|
||
|
|
if let Some(offset) = uniform_bindings
|
||
|
|
.get(&UniqueSemantics::FrameCount.into())
|
||
|
|
{
|
||
|
|
uniform_storage.bind_scalar(offset.offset(), frame_count, offset.context());
|
||
|
|
}
|
||
|
|
|
||
|
|
// bind FrameDirection
|
||
|
|
if let Some(offset) = uniform_bindings
|
||
|
|
.get(&UniqueSemantics::FrameDirection.into())
|
||
|
|
{
|
||
|
|
uniform_storage
|
||
|
|
.bind_scalar(offset.offset(), frame_direction, offset.context());
|
||
|
|
}
|
||
|
|
|
||
|
|
// bind Original sampler
|
||
|
|
if let Some(binding) = texture_meta
|
||
|
|
.get(&TextureSemantics::Original.semantics(0))
|
||
|
|
{
|
||
|
|
Self::bind_texture(
|
||
|
|
descriptor_set,
|
||
|
|
sampler_set,
|
||
|
|
binding,
|
||
|
|
original,
|
||
|
|
device
|
||
|
|
);
|
||
|
|
}
|
||
|
|
|
||
|
|
// bind OriginalSize
|
||
|
|
if let Some(offset) = uniform_bindings
|
||
|
|
.get(&TextureSemantics::Original.semantics(0).into())
|
||
|
|
{
|
||
|
|
uniform_storage
|
||
|
|
.bind_vec4(offset.offset(), original.size(), offset.context());
|
||
|
|
}
|
||
|
|
|
||
|
|
// bind Source sampler
|
||
|
|
if let Some(binding) = texture_meta
|
||
|
|
.get(&TextureSemantics::Source.semantics(0))
|
||
|
|
{
|
||
|
|
Self::bind_texture(
|
||
|
|
descriptor_set,
|
||
|
|
sampler_set,
|
||
|
|
binding,
|
||
|
|
source,
|
||
|
|
device
|
||
|
|
);
|
||
|
|
}
|
||
|
|
|
||
|
|
// bind SourcelSize
|
||
|
|
if let Some(offset) = uniform_bindings
|
||
|
|
.get(&TextureSemantics::Source.semantics(0).into())
|
||
|
|
{
|
||
|
|
uniform_storage
|
||
|
|
.bind_vec4(offset.offset(), source.size(), offset.context());
|
||
|
|
}
|
||
|
|
|
||
|
|
// OriginalHistory0 aliases OriginalHistory
|
||
|
|
|
||
|
|
// bind OriginalHistory0 sampler
|
||
|
|
if let Some(binding) = texture_meta
|
||
|
|
.get(&TextureSemantics::OriginalHistory.semantics(0))
|
||
|
|
{
|
||
|
|
Self::bind_texture(
|
||
|
|
descriptor_set,
|
||
|
|
sampler_set,
|
||
|
|
binding,
|
||
|
|
original,
|
||
|
|
device
|
||
|
|
);
|
||
|
|
}
|
||
|
|
|
||
|
|
// bind OriginalHistory0Size
|
||
|
|
if let Some(offset) = uniform_bindings
|
||
|
|
.get(&TextureSemantics::OriginalHistory.semantics(0).into())
|
||
|
|
{
|
||
|
|
uniform_storage
|
||
|
|
.bind_vec4(offset.offset(), original.size(), offset.context());
|
||
|
|
}
|
||
|
|
|
||
|
|
// bind OriginalHistory1-..
|
||
|
|
for (index, history) in original_history.enumerate() {
|
||
|
|
let Some(history) = history else {
|
||
|
|
continue;
|
||
|
|
};
|
||
|
|
|
||
|
|
let history = history.as_ref();
|
||
|
|
|
||
|
|
if let Some(binding) = texture_meta
|
||
|
|
.get(&TextureSemantics::OriginalHistory.semantics(index + 1))
|
||
|
|
{
|
||
|
|
Self::bind_texture(
|
||
|
|
descriptor_set,
|
||
|
|
sampler_set,
|
||
|
|
binding,
|
||
|
|
history,
|
||
|
|
device
|
||
|
|
);
|
||
|
|
}
|
||
|
|
|
||
|
|
if let Some(offset) = uniform_bindings.get(
|
||
|
|
&TextureSemantics::OriginalHistory
|
||
|
|
.semantics(index + 1)
|
||
|
|
.into(),
|
||
|
|
) {
|
||
|
|
uniform_storage
|
||
|
|
.bind_vec4(offset.offset(), history.size(), offset.context());
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// bind PassOutput0..
|
||
|
|
// The caller should be responsible for limiting this up to
|
||
|
|
// pass_index
|
||
|
|
for (index, output) in pass_outputs.enumerate() {
|
||
|
|
let Some(output) = output else {
|
||
|
|
continue;
|
||
|
|
};
|
||
|
|
|
||
|
|
let output = output.as_ref();
|
||
|
|
|
||
|
|
if let Some(binding) = texture_meta
|
||
|
|
.get(&TextureSemantics::PassOutput.semantics(index))
|
||
|
|
{
|
||
|
|
Self::bind_texture(
|
||
|
|
descriptor_set,
|
||
|
|
sampler_set,
|
||
|
|
binding,
|
||
|
|
output,
|
||
|
|
device
|
||
|
|
);
|
||
|
|
}
|
||
|
|
|
||
|
|
if let Some(offset) = uniform_bindings.get(
|
||
|
|
&TextureSemantics::PassOutput
|
||
|
|
.semantics(index)
|
||
|
|
.into(),
|
||
|
|
) {
|
||
|
|
uniform_storage
|
||
|
|
.bind_vec4(offset.offset(), output.size(), offset.context());
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// bind PassFeedback0..
|
||
|
|
for (index, feedback) in pass_feedback.enumerate() {
|
||
|
|
let Some(output) = feedback else {
|
||
|
|
continue;
|
||
|
|
};
|
||
|
|
|
||
|
|
let feedback = output.as_ref();
|
||
|
|
|
||
|
|
if let Some(binding) = texture_meta
|
||
|
|
.get(&TextureSemantics::PassFeedback.semantics(index))
|
||
|
|
{
|
||
|
|
Self::bind_texture(
|
||
|
|
descriptor_set,
|
||
|
|
sampler_set,
|
||
|
|
binding,
|
||
|
|
feedback,
|
||
|
|
device
|
||
|
|
);
|
||
|
|
}
|
||
|
|
|
||
|
|
if let Some(offset) = uniform_bindings.get(
|
||
|
|
&TextureSemantics::PassFeedback
|
||
|
|
.semantics(index)
|
||
|
|
.into(),
|
||
|
|
) {
|
||
|
|
uniform_storage
|
||
|
|
.bind_vec4(offset.offset(), feedback.size(), offset.context());
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// bind User parameters
|
||
|
|
for (id, offset) in
|
||
|
|
uniform_bindings
|
||
|
|
.iter()
|
||
|
|
.filter_map(|(binding, value)| match binding {
|
||
|
|
UniformBinding::Parameter(id) => Some((id, value)),
|
||
|
|
_ => None,
|
||
|
|
})
|
||
|
|
{
|
||
|
|
let id = id.as_str();
|
||
|
|
|
||
|
|
let default = parameter_defaults.iter()
|
||
|
|
.find(|&p| p.id == id)
|
||
|
|
.map(|f| f.initial)
|
||
|
|
.unwrap_or(0f32);
|
||
|
|
|
||
|
|
let value = *runtime_parameters.get(id).unwrap_or(&default);
|
||
|
|
|
||
|
|
uniform_storage.bind_scalar(offset.offset(), value, offset.context());
|
||
|
|
}
|
||
|
|
|
||
|
|
// bind luts
|
||
|
|
for (index, lut) in lookup_textures {
|
||
|
|
let lut = lut.as_ref();
|
||
|
|
if let Some(binding) = texture_meta
|
||
|
|
.get(&TextureSemantics::User.semantics(index))
|
||
|
|
{
|
||
|
|
Self::bind_texture(
|
||
|
|
descriptor_set,
|
||
|
|
sampler_set,
|
||
|
|
binding,
|
||
|
|
lut,
|
||
|
|
device
|
||
|
|
);
|
||
|
|
}
|
||
|
|
|
||
|
|
if let Some(offset) = uniform_bindings
|
||
|
|
.get(&TextureSemantics::User.semantics(index).into())
|
||
|
|
{
|
||
|
|
uniform_storage
|
||
|
|
.bind_vec4(offset.offset(), lut.size(), offset.context());
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|