initial mask impl

src/render/mod.rs

@@ -17,6 +17,7 @@ mod primitive;
 mod texture;
 mod util;
 
+pub use data::{Mask, MaskIdx};
 pub use primitive::*;
 
 const SHAPE_SHADER: &str = include_str!("./shader.wgsl");
@@ -33,6 +34,7 @@ pub struct UiRenderer {
     active: Vec<usize>,
     window_buffer: Buffer,
     textures: GpuTextures,
+    masks: ArrBuf<Mask>,
 }
 
 struct RenderLayer {
@@ -99,7 +101,11 @@ impl UiRenderer {
             }
         }
         if self.textures.update(&mut ui.data.textures) {
-            self.rsc_group = Self::rsc_group(device, &self.rsc_layout, &self.textures)
+            self.rsc_group = Self::rsc_group(device, &self.rsc_layout, &self.textures, &self.masks)
+        }
+        if ui.data.masks.changed {
+            ui.data.masks.changed = false;
+            self.masks.update(device, queue, &ui.data.masks[..]);
         }
     }
 
@@ -132,7 +138,7 @@ impl UiRenderer {
         let uniform_layout = device.create_bind_group_layout(&BindGroupLayoutDescriptor {
             entries: &[BindGroupLayoutEntry {
                 binding: 0,
-                visibility: ShaderStages::VERTEX,
+                visibility: ShaderStages::VERTEX | ShaderStages::FRAGMENT,
                 ty: BindingType::Buffer {
                     ty: BufferBindingType::Uniform,
                     has_dynamic_offset: false,
@@ -162,8 +168,14 @@ impl UiRenderer {
         });
 
         let tex_manager = GpuTextures::new(device, queue);
+        let masks = ArrBuf::new(
+            device,
+            BufferUsages::STORAGE | BufferUsages::COPY_DST,
+            "ui masks",
+        );
+
         let rsc_layout = Self::rsc_layout(device, &limits);
-        let rsc_group = Self::rsc_group(device, &rsc_layout, &tex_manager);
+        let rsc_group = Self::rsc_group(device, &rsc_layout, &tex_manager, &masks);
 
         let pipeline_layout = device.create_pipeline_layout(&PipelineLayoutDescriptor {
             label: Some("UI Shape Pipeline Layout"),
@@ -218,6 +230,7 @@ impl UiRenderer {
             layers: HashMap::default(),
             active: Vec::new(),
             textures: tex_manager,
+            masks,
         }
     }
 
@@ -270,6 +283,16 @@ impl UiRenderer {
                     ty: BindingType::Sampler(SamplerBindingType::NonFiltering),
                     count: Some(NonZero::new(limits.max_samplers).unwrap()),
                 },
+                BindGroupLayoutEntry {
+                    binding: 2,
+                    visibility: ShaderStages::FRAGMENT,
+                    ty: BindingType::Buffer {
+                        ty: BufferBindingType::Storage { read_only: true },
+                        has_dynamic_offset: false,
+                        min_binding_size: None,
+                    },
+                    count: None,
+                },
             ],
             label: Some("ui rsc"),
         })
@@ -279,6 +302,7 @@ impl UiRenderer {
         device: &Device,
         layout: &BindGroupLayout,
         tex_manager: &GpuTextures,
+        masks: &ArrBuf<Mask>,
     ) -> BindGroup {
         device.create_bind_group(&BindGroupDescriptor {
             layout,
@@ -291,6 +315,10 @@ impl UiRenderer {
                     binding: 1,
                     resource: BindingResource::SamplerArray(&tex_manager.samplers()),
                 },
+                BindGroupEntry {
+                    binding: 2,
+                    resource: masks.buffer.as_entire_binding(),
+                },
             ],
             label: Some("ui rsc"),
         })