RenderCommand

bevyengine · Mar 3, 2021 · 105c7ab · 105c7ab
1 parent 655053d
commit 105c7ab
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 0 deletions.
diff --git a/crates/bevy_render/src/draw.rs b/crates/bevy_render/src/draw.rs
@@ -13,6 +13,7 @@ use bevy_ecs::{Query, Res, ResMut, SystemParam};
 use bevy_reflect::{Reflect, ReflectComponent};
 use std::{ops::Range, sync::Arc};
 use thiserror::Error;
+use crate::pipeline::BindingShaderStage;
 
 /// A queued command for the renderer
 #[derive(Debug, Clone, Eq, PartialEq)]
@@ -35,6 +36,11 @@ pub enum RenderCommand {
  bind_group: BindGroupId,
  dynamic_uniform_indices: Option<Arc<[u32]>>,
  },
+ SetPushConstants {
+ stages: BindingShaderStage,
+ offset: u32,
+ data: Arc<[u8]>
+ },
  DrawIndexed {
  indices: Range<u32>,
  base_vertex: i32,
@@ -106,6 +112,14 @@ impl Draw {
  });
  }
 
+ pub fn set_push_constants(&mut self, stages: BindingShaderStage, offset: u32, data: Arc<[u8]>) {
+ self.render_command(RenderCommand::SetPushConstants {
+ stages,
+ offset,
+ data,
+ });
+ }
+
  pub fn set_bind_group(&mut self, index: u32, bind_group: &BindGroup) {
  self.render_command(RenderCommand::SetBindGroup {
  index,

diff --git a/crates/bevy_render/src/render_graph/nodes/pass_node.rs b/crates/bevy_render/src/render_graph/nodes/pass_node.rs
@@ -329,6 +329,13 @@ where
  );
  draw_state.set_bind_group(*index, *bind_group);
  }
+ RenderCommand::SetPushConstants {
+ stages,
+ offset,
+ data,
+ } => {
+ render_pass.set_push_constants(*stages, *offset, data);
+ }
  }
  }
  }