2022-11-12 13:52:22 +11:00
|
|
|
mod hello_triangle;
|
2022-11-14 16:14:05 +11:00
|
|
|
mod filter;
|
2022-11-14 17:49:51 +11:00
|
|
|
mod filter_pass;
|
|
|
|
mod util;
|
|
|
|
mod framebuffer;
|
2022-11-12 13:52:22 +11:00
|
|
|
|
2022-11-09 17:11:25 +11:00
|
|
|
use std::collections::HashMap;
|
|
|
|
use std::error::Error;
|
2022-11-14 16:14:05 +11:00
|
|
|
use std::iter::Filter;
|
2022-11-09 17:11:25 +11:00
|
|
|
use std::path::Path;
|
2022-11-14 16:14:05 +11:00
|
|
|
use gl::types::{GLenum, GLint, GLsizei, GLsizeiptr, GLuint};
|
|
|
|
use glfw::Key::P;
|
2022-11-09 17:11:25 +11:00
|
|
|
use rustc_hash::FxHashMap;
|
2022-11-12 17:23:49 +11:00
|
|
|
use spirv_cross::spirv::Decoration;
|
2022-11-14 17:49:51 +11:00
|
|
|
use filter_pass::FilterPass;
|
|
|
|
use framebuffer::Framebuffer;
|
2022-11-09 17:51:10 +11:00
|
|
|
|
2022-11-14 17:49:51 +11:00
|
|
|
use librashader::{FilterMode, ShaderFormat, ShaderSource, WrapMode};
|
2022-11-14 16:14:05 +11:00
|
|
|
use librashader_presets::{ShaderPassConfig, ShaderPreset};
|
2022-11-12 17:23:49 +11:00
|
|
|
use librashader_reflect::back::{CompileShader, ShaderCompilerOutput};
|
|
|
|
use librashader_reflect::back::cross::{GlslangGlslContext, GlVersion};
|
2022-11-11 17:44:41 +11:00
|
|
|
use librashader_reflect::back::targets::{FromCompilation, GLSL};
|
2022-11-09 17:11:25 +11:00
|
|
|
use librashader_reflect::front::shaderc::GlslangCompilation;
|
2022-11-11 17:44:41 +11:00
|
|
|
use librashader_reflect::reflect::cross::CrossReflect;
|
2022-11-12 17:23:49 +11:00
|
|
|
use librashader_reflect::reflect::{ReflectSemantics, ReflectShader, ShaderReflection, UniformSemantic};
|
|
|
|
use librashader_reflect::reflect::semantics::{MemberOffset, SemanticMap, TextureSemantics, VariableMeta, VariableSemantics};
|
2022-11-09 17:51:10 +11:00
|
|
|
use librashader_reflect::reflect::{TextureSemanticMap, VariableSemanticMap};
|
2022-11-14 17:49:51 +11:00
|
|
|
use util::{Location, VariableLocation, RingBuffer, Size, Texture, TextureMeta, Viewport};
|
2022-11-09 17:11:25 +11:00
|
|
|
|
2022-11-14 16:14:05 +11:00
|
|
|
unsafe fn gl_compile_shader(stage: GLenum, source: &str) -> GLuint {
|
|
|
|
let shader = gl::CreateShader(stage);
|
|
|
|
gl::ShaderSource(shader, 1, &source.as_bytes().as_ptr().cast(), std::ptr::null());
|
|
|
|
gl::CompileShader(shader);
|
|
|
|
|
|
|
|
let mut compile_status = 0;
|
|
|
|
gl::GetShaderiv(shader, gl::COMPILE_STATUS, &mut compile_status);
|
|
|
|
|
|
|
|
if compile_status == 0 {
|
|
|
|
panic!("failed to compile")
|
|
|
|
}
|
|
|
|
shader
|
|
|
|
}
|
|
|
|
|
2022-11-14 17:49:51 +11:00
|
|
|
impl FilterChain {
|
|
|
|
fn load_pass_semantics(uniform_semantics: &mut FxHashMap<String, UniformSemantic>, texture_semantics: &mut FxHashMap<String, SemanticMap<TextureSemantics>>,
|
2022-11-09 17:11:25 +11:00
|
|
|
config: &ShaderPassConfig) {
|
2022-11-14 17:49:51 +11:00
|
|
|
let Some(alias) = &config.alias else {
|
|
|
|
return;
|
|
|
|
};
|
2022-11-14 16:14:05 +11:00
|
|
|
|
2022-11-14 17:49:51 +11:00
|
|
|
// Ignore empty aliases
|
|
|
|
if alias.trim().is_empty() {
|
|
|
|
return;
|
2022-11-14 16:14:05 +11:00
|
|
|
}
|
|
|
|
|
2022-11-14 17:49:51 +11:00
|
|
|
let index = config.id as u32;
|
|
|
|
|
|
|
|
// PassOutput
|
|
|
|
texture_semantics.insert(alias.clone(), SemanticMap {
|
|
|
|
semantics: TextureSemantics::PassOutput,
|
|
|
|
index
|
|
|
|
});
|
|
|
|
uniform_semantics.insert(format!("{alias}Size"), UniformSemantic::Texture(SemanticMap {
|
|
|
|
semantics: TextureSemantics::PassOutput,
|
|
|
|
index
|
|
|
|
}));
|
|
|
|
|
|
|
|
// PassFeedback
|
|
|
|
texture_semantics.insert(format!("{alias}Feedback"), SemanticMap {
|
|
|
|
semantics: TextureSemantics::PassFeedback,
|
|
|
|
index
|
|
|
|
});
|
|
|
|
uniform_semantics.insert(format!("{alias}FeedbackSize"), UniformSemantic::Texture(SemanticMap {
|
|
|
|
semantics: TextureSemantics::PassFeedback,
|
|
|
|
index
|
|
|
|
}));
|
2022-11-14 16:14:05 +11:00
|
|
|
}
|
|
|
|
|
2022-11-14 17:49:51 +11:00
|
|
|
fn reflect_parameter(pipeline: GLuint, meta: &VariableMeta) -> VariableLocation {
|
|
|
|
// todo: support both ubo and pushco
|
|
|
|
// todo: fix this.
|
|
|
|
match meta.offset {
|
|
|
|
MemberOffset::Ubo(_) => {
|
|
|
|
let vert_name = format!("RARCH_UBO_VERTEX_INSTANCE.{}\0", meta.id);
|
|
|
|
let frag_name = format!("RARCH_UBO_FRAGMENT_INSTANCE.{}\0", meta.id);
|
|
|
|
unsafe {
|
|
|
|
let vertex = gl::GetUniformLocation(pipeline, vert_name.as_ptr().cast());
|
|
|
|
let fragment = gl::GetUniformLocation(pipeline, frag_name.as_ptr().cast());
|
2022-11-14 16:14:05 +11:00
|
|
|
|
2022-11-14 17:49:51 +11:00
|
|
|
VariableLocation::Ubo(Location {
|
|
|
|
vertex,
|
|
|
|
fragment
|
|
|
|
})
|
|
|
|
}
|
2022-11-14 16:14:05 +11:00
|
|
|
}
|
2022-11-14 17:49:51 +11:00
|
|
|
MemberOffset::PushConstant(_) => {
|
|
|
|
let vert_name = format!("RARCH_PUSH_VERTEX_INSTANCE.{}\0", meta.id);
|
|
|
|
let frag_name = format!("RARCH_PUSH_FRAGMENT_INSTANCE.{}\0", meta.id);
|
|
|
|
unsafe {
|
|
|
|
let vertex = gl::GetUniformLocation(pipeline, vert_name.as_ptr().cast());
|
|
|
|
let fragment = gl::GetUniformLocation(pipeline, frag_name.as_ptr().cast());
|
2022-11-14 16:14:05 +11:00
|
|
|
|
2022-11-14 17:49:51 +11:00
|
|
|
VariableLocation::Push(Location {
|
|
|
|
vertex,
|
|
|
|
fragment
|
|
|
|
})
|
2022-11-14 16:14:05 +11:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
}
|
2022-11-14 17:49:51 +11:00
|
|
|
|
2022-11-12 17:23:49 +11:00
|
|
|
pub struct FilterChain {
|
2022-11-14 16:14:05 +11:00
|
|
|
passes: Vec<FilterPass>,
|
|
|
|
semantics: ReflectSemantics,
|
|
|
|
preset: ShaderPreset,
|
|
|
|
original_history: Vec<Framebuffer>,
|
|
|
|
history: Vec<Texture>,
|
|
|
|
feedback: Vec<Texture>
|
2022-11-12 17:23:49 +11:00
|
|
|
}
|
|
|
|
|
2022-11-13 18:05:49 +11:00
|
|
|
|
2022-11-14 16:14:05 +11:00
|
|
|
impl FilterChain {
|
|
|
|
pub fn load(path: impl AsRef<Path>) -> Result<FilterChain, Box<dyn Error>> {
|
|
|
|
let preset = librashader_presets::ShaderPreset::try_parse(path)?;
|
|
|
|
let mut uniform_semantics: FxHashMap<String, UniformSemantic> = Default::default();
|
|
|
|
let mut texture_semantics: FxHashMap<String, SemanticMap<TextureSemantics>> = Default::default();
|
|
|
|
|
|
|
|
let mut passes: Vec<(&ShaderPassConfig, ShaderSource, _)> = preset.shaders.iter()
|
|
|
|
.map(|shader| {
|
|
|
|
eprintln!("[gl] loading {}", &shader.name.display());
|
|
|
|
let source: ShaderSource = librashader_preprocess::load_shader_source(&shader.name)
|
|
|
|
.unwrap();
|
|
|
|
|
|
|
|
let spirv = librashader_reflect::front::shaderc::compile_spirv(&source)
|
|
|
|
.unwrap();
|
|
|
|
let mut reflect = GLSL::from_compilation(spirv).unwrap();
|
|
|
|
|
|
|
|
for parameter in source.parameters.iter() {
|
|
|
|
uniform_semantics.insert(parameter.id.clone(), UniformSemantic::Variable(SemanticMap {
|
|
|
|
semantics: VariableSemantics::FloatParameter,
|
|
|
|
index: ()
|
|
|
|
}));
|
|
|
|
}
|
2022-11-09 17:11:25 +11:00
|
|
|
|
2022-11-14 16:14:05 +11:00
|
|
|
(shader, source, reflect)
|
|
|
|
}).collect();
|
2022-11-09 17:11:25 +11:00
|
|
|
|
2022-11-14 16:14:05 +11:00
|
|
|
// todo: this can probably be extracted out.
|
2022-11-09 17:11:25 +11:00
|
|
|
|
2022-11-14 16:14:05 +11:00
|
|
|
for details in &passes {
|
2022-11-14 17:49:51 +11:00
|
|
|
FilterChain::load_pass_semantics(&mut uniform_semantics, &mut texture_semantics, details.0)
|
2022-11-14 16:14:05 +11:00
|
|
|
}
|
2022-11-09 17:11:25 +11:00
|
|
|
|
2022-11-14 16:14:05 +11:00
|
|
|
// add lut params
|
|
|
|
for (index, texture) in preset.textures.iter().enumerate() {
|
|
|
|
texture_semantics.insert(texture.name.clone(), SemanticMap {
|
|
|
|
semantics: TextureSemantics::User,
|
|
|
|
index: index as u32
|
|
|
|
});
|
|
|
|
|
|
|
|
uniform_semantics.insert(format!("{}Size", texture.name), UniformSemantic::Texture(SemanticMap {
|
|
|
|
semantics: TextureSemantics::User,
|
|
|
|
index: index as u32
|
|
|
|
}));
|
|
|
|
}
|
2022-11-09 17:11:25 +11:00
|
|
|
|
2022-11-14 16:14:05 +11:00
|
|
|
let semantics = ReflectSemantics {
|
|
|
|
uniform_semantics,
|
|
|
|
non_uniform_semantics: texture_semantics
|
|
|
|
};
|
2022-11-09 17:11:25 +11:00
|
|
|
|
2022-11-14 16:14:05 +11:00
|
|
|
let mut filters = Vec::new();
|
2022-11-13 17:57:47 +11:00
|
|
|
|
2022-11-14 16:14:05 +11:00
|
|
|
// initialize passes
|
|
|
|
for (index, (config, source, mut reflect)) in passes.into_iter().enumerate() {
|
|
|
|
let mut semantics = semantics.clone();
|
|
|
|
|
|
|
|
let reflection = reflect.reflect(index as u32, &semantics)?;
|
|
|
|
let glsl = reflect.compile(GlVersion::V4_60)?;
|
2022-11-13 17:57:47 +11:00
|
|
|
|
2022-11-14 16:14:05 +11:00
|
|
|
let vertex_resources = glsl.context.compiler.vertex.get_shader_resources()?;
|
2022-11-13 17:57:47 +11:00
|
|
|
|
2022-11-14 16:14:05 +11:00
|
|
|
// todo: split this out.
|
|
|
|
let (program, ubo_location) = unsafe {
|
|
|
|
let vertex = gl_compile_shader(gl::VERTEX_SHADER, glsl.vertex.as_str());
|
|
|
|
let fragment = gl_compile_shader(gl::FRAGMENT_SHADER, glsl.fragment.as_str());
|
2022-11-09 17:11:25 +11:00
|
|
|
|
2022-11-14 16:14:05 +11:00
|
|
|
let program = gl::CreateProgram();
|
|
|
|
gl::AttachShader(program, vertex);
|
|
|
|
gl::AttachShader(program, fragment);
|
2022-11-12 17:23:49 +11:00
|
|
|
|
2022-11-14 16:14:05 +11:00
|
|
|
for res in &vertex_resources.stage_inputs {
|
|
|
|
let loc = glsl.context.compiler.vertex.get_decoration(res.id, Decoration::Location)?;
|
|
|
|
let loc_name = format!("RARCH_ATTRIBUTE_{loc}");
|
|
|
|
gl::BindAttribLocation(program, loc, loc_name.as_str().as_ptr().cast())
|
|
|
|
}
|
|
|
|
gl::LinkProgram(program);
|
|
|
|
gl::DeleteShader(vertex);
|
|
|
|
gl::DeleteShader(fragment);
|
|
|
|
|
|
|
|
let mut status = 0;
|
|
|
|
gl::GetProgramiv(program, gl::LINK_STATUS, &mut status);
|
|
|
|
if status != 1 {
|
|
|
|
panic!("failed to link program")
|
|
|
|
}
|
2022-11-12 17:23:49 +11:00
|
|
|
|
2022-11-14 16:14:05 +11:00
|
|
|
for binding in &glsl.context.texture_fixups {
|
|
|
|
let loc_name = format!("RARCH_TEXTURE_{}", *binding);
|
|
|
|
unsafe {
|
|
|
|
let location = gl::GetUniformLocation(program, loc_name.as_str().as_ptr().cast());
|
|
|
|
if location >= 0 {
|
|
|
|
gl::Uniform1i(location, *binding as GLint);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
2022-11-12 17:23:49 +11:00
|
|
|
|
2022-11-14 16:14:05 +11:00
|
|
|
unsafe {
|
|
|
|
gl::UseProgram(0);
|
|
|
|
(program, Location {
|
|
|
|
vertex: gl::GetUniformBlockIndex(program, b"RARCH_UBO_VERTEX\0".as_ptr().cast()),
|
|
|
|
fragment: gl::GetUniformBlockIndex(program, b"RARCH_UBO_FRAGMENT\0".as_ptr().cast()),
|
|
|
|
})
|
|
|
|
}
|
|
|
|
};
|
2022-11-12 17:23:49 +11:00
|
|
|
|
2022-11-14 16:14:05 +11:00
|
|
|
let ubo_ring = if let Some(ubo) = &reflection.ubo {
|
|
|
|
let size = ubo.size;
|
|
|
|
let mut ring: RingBuffer<GLuint, 16> = RingBuffer::new();
|
2022-11-12 17:23:49 +11:00
|
|
|
unsafe {
|
2022-11-14 17:49:51 +11:00
|
|
|
gl::GenBuffers(16, ring.items_mut().as_mut_ptr());
|
|
|
|
for buffer in ring.items() {
|
2022-11-14 16:14:05 +11:00
|
|
|
gl::BindBuffer(gl::UNIFORM_BUFFER, *buffer);
|
|
|
|
gl::BufferData(gl::UNIFORM_BUFFER, size as GLsizeiptr, std::ptr::null(), gl::STREAM_DRAW);
|
2022-11-12 17:23:49 +11:00
|
|
|
}
|
2022-11-14 16:14:05 +11:00
|
|
|
gl::BindBuffer(gl::UNIFORM_BUFFER, 0);
|
2022-11-12 17:23:49 +11:00
|
|
|
}
|
2022-11-14 16:14:05 +11:00
|
|
|
Some(ring)
|
|
|
|
} else {
|
|
|
|
None
|
|
|
|
};
|
|
|
|
|
2022-11-14 17:49:51 +11:00
|
|
|
let uniform_buffer = vec![0; reflection.ubo.as_ref().map(|ubo| ubo.size as usize).unwrap_or(0)].into_boxed_slice();
|
|
|
|
let push_buffer = vec![0; reflection.push_constant.as_ref().map(|push| push.size as usize).unwrap_or(0)].into_boxed_slice();
|
2022-11-14 16:14:05 +11:00
|
|
|
|
|
|
|
// todo: reflect indexed parameters
|
|
|
|
let mut locations = FxHashMap::default();
|
|
|
|
for param in reflection.meta.parameter_meta.values() {
|
2022-11-14 17:49:51 +11:00
|
|
|
locations.insert(param.id.clone(), FilterChain::reflect_parameter(program, param));
|
2022-11-12 17:23:49 +11:00
|
|
|
}
|
|
|
|
|
2022-11-14 16:14:05 +11:00
|
|
|
for param in reflection.meta.variable_meta.values() {
|
2022-11-14 17:49:51 +11:00
|
|
|
locations.insert(param.id.clone(), FilterChain::reflect_parameter(program, param));
|
2022-11-12 17:23:49 +11:00
|
|
|
}
|
|
|
|
|
|
|
|
|
2022-11-14 16:14:05 +11:00
|
|
|
// eprintln!("{:#?}", semantics);
|
|
|
|
eprintln!("{:#?}", reflection.meta);
|
|
|
|
eprintln!("{:#?}", locations);
|
|
|
|
eprintln!("{:#?}", reflection.push_constant);
|
|
|
|
// eprintln!("====fragment====");
|
|
|
|
// eprintln!("{:#}", glsl.fragment);
|
|
|
|
// eprintln!("====vertex====");
|
|
|
|
// eprintln!("{:#}", glsl.vertex);
|
|
|
|
|
|
|
|
filters.push(FilterPass {
|
|
|
|
reflection,
|
|
|
|
compiled: glsl,
|
|
|
|
program,
|
|
|
|
ubo_location,
|
|
|
|
ubo_ring,
|
|
|
|
uniform_buffer,
|
|
|
|
push_buffer,
|
|
|
|
locations,
|
|
|
|
|
|
|
|
// no idea if this works.
|
|
|
|
// retroarch checks if feedback frames are used but we'll just init it tbh.
|
|
|
|
framebuffer: Framebuffer::new(1),
|
|
|
|
feedback_framebuffer: Framebuffer::new(1)
|
|
|
|
});
|
2022-11-12 17:23:49 +11:00
|
|
|
}
|
|
|
|
|
2022-11-14 16:14:05 +11:00
|
|
|
eprintln!("{:?}", filters.iter().map(|f| f.program).collect::<Vec<_>>());
|
|
|
|
// let mut glprogram: Vec<GLuint> = Vec::new();
|
|
|
|
// for compilation in &compiled {
|
|
|
|
// // compilation.context.compiler.vertex
|
|
|
|
// }
|
2022-11-12 17:23:49 +11:00
|
|
|
|
2022-11-14 16:14:05 +11:00
|
|
|
// eprintln!("{:#?}", reflections);
|
2022-11-09 17:11:25 +11:00
|
|
|
|
2022-11-14 16:14:05 +11:00
|
|
|
// eprintln!("{:#?}", compiled./);
|
|
|
|
// eprintln!("{:?}", preset);
|
|
|
|
// eprintln!("{:?}", reflect.reflect(&ReflectOptions {
|
|
|
|
// pass_number: i as u32,
|
|
|
|
// uniform_semantics,
|
|
|
|
// non_uniform_semantics: Default::default(),
|
|
|
|
// }));
|
|
|
|
|
|
|
|
// todo: apply shader pass
|
|
|
|
// gl3.cpp: 1942
|
|
|
|
|
|
|
|
|
|
|
|
Ok(FilterChain {
|
|
|
|
passes: filters,
|
|
|
|
semantics,
|
|
|
|
preset,
|
|
|
|
original_history: vec![],
|
|
|
|
history: vec![],
|
|
|
|
feedback: vec![]
|
|
|
|
})
|
|
|
|
}
|
2022-11-09 17:11:25 +11:00
|
|
|
|
|
|
|
|
2022-11-14 16:14:05 +11:00
|
|
|
// how much info do we actually need?
|
2022-11-14 17:49:51 +11:00
|
|
|
// fn frame(&mut self, count: u64, vp: &Viewport, input: &Texture, clear: bool) {
|
|
|
|
//
|
|
|
|
// // todo: make copy
|
|
|
|
//
|
|
|
|
// let original = Texture {
|
|
|
|
// handle: input.handle,
|
|
|
|
// format: self.preset.shaders.first().,
|
|
|
|
// size: Size {},
|
|
|
|
// padded_size: Size {}
|
|
|
|
// };
|
|
|
|
// // todo: deal with the mess that is frame history
|
|
|
|
// }
|
2022-11-09 17:11:25 +11:00
|
|
|
|
2022-11-14 17:49:51 +11:00
|
|
|
fn do_final_pass(&mut self, count: u64, vp: &Viewport, input: Texture, clear: bool, mvp: &[f32]) {
|
2022-11-09 17:11:25 +11:00
|
|
|
|
2022-11-14 17:49:51 +11:00
|
|
|
// todo: make copy
|
|
|
|
|
|
|
|
// todo: get filter info from pass data.
|
|
|
|
let original = TextureMeta {
|
|
|
|
texture: input,
|
|
|
|
filter: gl::LINEAR,
|
|
|
|
mip_filter: gl::LINEAR_MIPMAP_LINEAR,
|
|
|
|
wrap_mode: Default::default()
|
|
|
|
};
|
2022-11-11 18:26:57 +11:00
|
|
|
|
|
|
|
|
2022-11-14 17:49:51 +11:00
|
|
|
// todo: deal with the mess that is frame history
|
|
|
|
}
|
2022-11-11 18:26:57 +11:00
|
|
|
}
|
|
|
|
|
2022-11-14 16:14:05 +11:00
|
|
|
|
2022-11-09 17:11:25 +11:00
|
|
|
#[cfg(test)]
|
|
|
|
mod tests {
|
|
|
|
use super::*;
|
|
|
|
|
2022-11-12 13:52:22 +11:00
|
|
|
#[test]
|
|
|
|
fn triangle() {
|
|
|
|
let (glfw, window, events, shader, vao) = hello_triangle::setup();
|
2022-11-14 16:14:05 +11:00
|
|
|
FilterChain::load("../test/basic.slangp").unwrap();
|
|
|
|
// FilterChain::load("../test/slang-shaders/crt/crt-royale.slangp").unwrap();
|
|
|
|
|
2022-11-12 13:52:22 +11:00
|
|
|
hello_triangle::do_loop(glfw, window, events, shader, vao);
|
|
|
|
}
|
|
|
|
|
2022-11-12 17:23:49 +11:00
|
|
|
// #[test]
|
|
|
|
// fn load_preset() {
|
|
|
|
//
|
|
|
|
// load("../test/basic.slangp")
|
|
|
|
// .unwrap();
|
|
|
|
// }
|
2022-11-09 17:11:25 +11:00
|
|
|
}
|