Refactor UI renderer to pass node information via a uniform buffer

bevyengine · Feb 10, 2024 · f93938d · f93938d
1 parent ed435aa
commit f93938d
Showing 1 changed file with 21 additions and 0 deletions.
diff --git a/crates/bevy_ui/src/render/render_pass.rs b/crates/bevy_ui/src/render/render_pass.rs
@@ -128,6 +128,7 @@ pub type DrawUi = (
  SetItemPipeline,
  SetUiViewBindGroup<0>,
  SetUiTextureBindGroup<1>,
+ SetUiUniformBindGroup<2>,
  DrawUiNode,
 );
 
@@ -171,6 +172,26 @@ impl<P: PhaseItem, const I: usize> RenderCommand<P> for SetUiTextureBindGroup<I>
  RenderCommandResult::Success
  }
 }
+pub struct SetUiUniformBindGroup<const I: usize>;
+impl<const I: usize> EntityRenderCommand for SetUiUniformBindGroup<I> {
+ type Param = (SRes<UiMeta>, SQuery<Read<UiBatch>>);
+
+ fn render<'w>(
+ _view: Entity,
+ item: Entity,
+ (ui_meta, query_batch): SystemParamItem<'w, '_, Self::Param>,
+ pass: &mut TrackedRenderPass<'w>,
+ ) -> RenderCommandResult {
+ let batch = query_batch.get(item).unwrap();
+
+ pass.set_bind_group(
+ I,
+ ui_meta.into_inner().ui_uniform_bind_group.as_ref().unwrap(),
+ &[batch.ui_uniform_offset],
+ );
+ RenderCommandResult::Success
+ }
+}
 pub struct DrawUiNode;
 impl<P: PhaseItem> RenderCommand<P> for DrawUiNode {
  type Param = SRes<UiMeta>;