bevy/examples/custom_shader.rs

111 lines
4.1 KiB
Rust
Raw Normal View History

2020-03-10 06:08:09 +00:00
use bevy::prelude::*;
2020-02-18 00:33:46 +00:00
2020-02-18 04:23:00 +00:00
#[derive(Uniforms, Default)]
2020-02-18 00:33:46 +00:00
struct MyMaterial {
2020-03-10 06:43:40 +00:00
pub color: Color,
2020-02-18 17:06:24 +00:00
#[uniform(ignore, shader_def)]
pub always_red: bool,
2020-02-18 00:33:46 +00:00
}
fn main() {
App::build()
2020-02-18 00:33:46 +00:00
.add_defaults()
.setup_render_graph(|render_graph_builder| {
render_graph_builder
.add_resource_provider(UniformResourceProvider::<MyMaterial>::new())
.add_pipeline_to_pass(resource_name::pass::MAIN, "MyMaterial", |builder| {
builder
.with_vertex_shader(Shader::from_glsl(
2020-02-18 04:23:00 +00:00
ShaderStage::Vertex,
r#"
2020-02-18 00:33:46 +00:00
#version 450
layout(location = 0) in vec4 Vertex_Position;
2020-02-18 00:33:46 +00:00
layout(location = 0) out vec4 v_Position;
layout(set = 0, binding = 0) uniform Camera {
mat4 ViewProj;
};
layout(set = 1, binding = 0) uniform Object {
mat4 Model;
};
void main() {
v_Position = Model * Vertex_Position;
2020-02-18 00:33:46 +00:00
gl_Position = ViewProj * v_Position;
}
2020-02-18 04:23:00 +00:00
"#,
))
.with_fragment_shader(Shader::from_glsl(
ShaderStage::Fragment,
r#"
2020-02-18 00:33:46 +00:00
#version 450
layout(location = 0) in vec4 v_Position;
layout(location = 0) out vec4 o_Target;
layout(set = 1, binding = 1) uniform MyMaterial_color {
vec4 color;
};
void main() {
o_Target = color;
2020-02-18 17:06:24 +00:00
2020-02-19 06:59:33 +00:00
# ifdef MY_MATERIAL_ALWAYS_RED
2020-02-18 17:06:24 +00:00
o_Target = vec4(0.8, 0.0, 0.0, 1.0);
# endif
2020-02-18 00:33:46 +00:00
}
2020-02-18 04:23:00 +00:00
"#,
))
.with_standard_config();
});
2020-02-18 00:33:46 +00:00
})
.setup_world(setup)
2020-02-18 00:33:46 +00:00
.run();
}
2020-03-09 06:19:07 +00:00
fn setup(world: &mut World, resources: &mut Resources) {
let mut mesh_storage = resources.get_mut::<AssetStorage<Mesh>>().unwrap();
let cube_handle = mesh_storage.add(Mesh::load(MeshType::Cube));
2020-02-18 00:33:46 +00:00
2020-03-20 19:47:33 +00:00
let mut pipeline_storage = resources
.get_mut::<AssetStorage<PipelineDescriptor>>()
.unwrap();
let material_handle = pipeline_storage.get_named("MyMaterial").unwrap();
2020-02-18 00:33:46 +00:00
world
.build()
2020-02-18 04:23:00 +00:00
// cube
2020-03-09 09:02:17 +00:00
.add_entity(MeshMaterialEntity::<MyMaterial> {
2020-02-24 07:50:44 +00:00
mesh: cube_handle,
2020-02-18 17:06:24 +00:00
renderable: Renderable {
pipelines: vec![material_handle],
2020-02-18 17:06:24 +00:00
..Default::default()
},
material: MyMaterial {
color: Color::rgb(0.0, 0.8, 0.0),
2020-02-18 17:06:24 +00:00
always_red: false,
},
2020-03-09 03:29:21 +00:00
translation: Translation::new(-2.0, 0.0, 0.0),
2020-02-18 17:06:24 +00:00
..Default::default()
})
// cube
2020-03-09 09:02:17 +00:00
.add_entity(MeshMaterialEntity::<MyMaterial> {
2020-02-18 00:33:46 +00:00
mesh: cube_handle,
2020-02-18 04:23:00 +00:00
renderable: Renderable {
pipelines: vec![material_handle],
2020-02-18 04:23:00 +00:00
..Default::default()
},
material: MyMaterial {
color: Color::rgb(0.0, 0.0, 0.0),
2020-02-18 17:06:24 +00:00
always_red: true,
2020-02-18 04:23:00 +00:00
},
2020-03-09 03:29:21 +00:00
translation: Translation::new(2.0, 0.0, 0.0),
2020-02-18 03:15:28 +00:00
..Default::default()
2020-02-18 00:33:46 +00:00
})
// camera
2020-03-09 09:02:17 +00:00
.add_entity(CameraEntity {
2020-02-18 00:33:46 +00:00
local_to_world: LocalToWorld(Mat4::look_at_rh(
Vec3::new(3.0, 8.0, 5.0),
Vec3::new(0.0, 0.0, 0.0),
Vec3::new(0.0, 0.0, 1.0),
)),
2020-03-22 04:55:33 +00:00
..Default::default()
2020-02-18 00:33:46 +00:00
})
.build();
}