Improve boundaries between `quad` submodules in `iced_wgpu`

1 files changed, 32 insertions, 18 deletions

diff --git a/wgpu/src/quad/gradient.rs b/wgpu/src/quad/gradient.rs
index 2729afdf..d3884ce1 100644
--- a/wgpu/src/quad/gradient.rs
+++ b/wgpu/src/quad/gradient.rs
@@ -24,13 +24,13 @@ unsafe impl Zeroable for Gradient {}
 
 #[derive(Debug)]
 pub struct Pipeline {
-    pub pipeline: wgpu::RenderPipeline,
+    pipeline: wgpu::RenderPipeline,
 }
 
 #[derive(Debug)]
 pub struct Layer {
-    pub instances: Buffer<Gradient>,
-    pub instance_count: usize,
+    instances: Buffer<Gradient>,
+    instance_count: usize,
 }
 
 impl Layer {
@@ -48,23 +48,16 @@ impl Layer {
         }
     }
 
-    pub fn draw<'a>(
-        &'a self,
-        constants: &'a wgpu::BindGroup,
-        render_pass: &mut wgpu::RenderPass<'a>,
-        range: Range<usize>,
+    pub fn prepare(
+        &mut self,
+        device: &wgpu::Device,
+        queue: &wgpu::Queue,
+        instances: &[Gradient],
     ) {
-        #[cfg(feature = "tracing")]
-        let _ = tracing::info_span!("Wgpu::Quad::Gradient", "DRAW").entered();
-
-        render_pass.set_bind_group(0, constants, &[]);
-        render_pass.set_vertex_buffer(1, self.instances.slice(..));
+        let _ = self.instances.resize(device, instances.len());
+        let _ = self.instances.write(queue, 0, instances);
 
-        render_pass.draw_indexed(
-            0..quad::INDICES.len() as u32,
-            0,
-            range.start as u32..range.end as u32,
-        );
+        self.instance_count = instances.len();
     }
 }
 
@@ -158,4 +151,25 @@ impl Pipeline {
 
         Self { pipeline }
     }
+
+    pub fn render<'a>(
+        &'a self,
+        render_pass: &mut wgpu::RenderPass<'a>,
+        constants: &'a wgpu::BindGroup,
+        layer: &'a Layer,
+        range: Range<usize>,
+    ) {
+        #[cfg(feature = "tracing")]
+        let _ = tracing::info_span!("Wgpu::Quad::Gradient", "DRAW").entered();
+
+        render_pass.set_pipeline(&self.pipeline);
+        render_pass.set_bind_group(0, constants, &[]);
+        render_pass.set_vertex_buffer(1, layer.instances.slice(..));
+
+        render_pass.draw_indexed(
+            0..quad::INDICES.len() as u32,
+            0,
+            range.start as u32..range.end as u32,
+        );
+    }
 }