Skip to content

Commit

Permalink
Adjusted Quads struct to be opaque quad::Layer.
Browse files Browse the repository at this point in the history
  • Loading branch information
@bungoboingo
bungoboingo committed May 26, 2023
1 parent 7bdf20d commit 10a68ac
Show file tree
Hide file tree
Showing 4 changed files with 141 additions and 135 deletions.
2 changes: 1 addition & 1 deletion wgpu/src/backend.rs
View file
Original file line number Diff line number Diff line change
Expand Up @@ -268,7 +268,7 @@ impl Backend {
self.quad_pipeline.render(
quad_layer,
bounds,
&layer.quads.order,
&layer.quads,
&mut render_pass,
);

Expand Down
115 changes: 4 additions & 111 deletions wgpu/src/layer.rs
View file
Original file line number Diff line number Diff line change
Expand Up @@ -12,7 +12,7 @@ pub use text::Text;

use crate::core;
use crate::core::alignment;
use crate::core::{Background, Color, Font, Point, Rectangle, Size, Vector};
use crate::core::{Color, Font, Point, Rectangle, Size, Vector};
use crate::graphics::{Primitive, Viewport};

/// A group of primitives that should be clipped together.
Expand All @@ -22,7 +22,7 @@ pub struct Layer<'a> {
pub bounds: Rectangle,

/// The quads of the [`Layer`].
pub quads: Quads,
pub quads: quad::Layer,

/// The triangle meshes of the [`Layer`].
pub meshes: Vec<Mesh<'a>>,
Expand All @@ -34,35 +34,12 @@ pub struct Layer<'a> {
pub images: Vec<Image>,
}

/// The quads of the [`Layer`].
#[derive(Default, Debug)]
pub struct Quads {
/// The solid quads of the [`Layer`].
pub solids: Vec<quad::Solid>,

/// The gradient quads of the [`Layer`].
pub gradients: Vec<quad::Gradient>,

/// The quad order of the [`Layer`]; stored as a tuple of the quad type & its count.
pub order: Vec<(quad::Order, usize)>,

// The last index of quad ordering.
index: usize,
}

impl Quads {
/// Returns true if there are no quads of any type in [`Quads`].
pub fn is_empty(&self) -> bool {
self.solids.is_empty() && self.gradients.is_empty()
}
}

impl<'a> Layer<'a> {
/// Creates a new [`Layer`] with the given clipping bounds.
pub fn new(bounds: Rectangle) -> Self {
Self {
bounds,
quads: Quads::default(),
quads: quad::Layer::default(),
meshes: Vec::new(),
text: Vec::new(),
images: Vec::new(),
Expand Down Expand Up @@ -179,62 +156,7 @@ impl<'a> Layer<'a> {
border_width: *border_width,
};

let quad_order = match background {
Background::Color(color) => {
layer.quads.solids.push(quad::Solid {
color: color.into_linear(),
quad,
});
quad::Order::Solid
}
Background::Gradient(gradient) => {
let quad = quad::Gradient {
gradient: pack_gradient(
gradient,
Rectangle::new(
quad.position.into(),
quad.size.into(),
),
),
quad,
};

layer.quads.gradients.push(quad);
quad::Order::Gradient
}
};

match (layer.quads.order.get_mut(layer.quads.index), quad_order)
{
(Some((quad_order, count)), quad::Order::Solid) => {
match quad_order {
quad::Order::Solid => {
*count += 1;
}
quad::Order::Gradient => {
layer.quads.order.push((quad::Order::Solid, 1));
layer.quads.index += 1;
}
}
}
(Some((quad_order, count)), quad::Order::Gradient) => {
match quad_order {
quad::Order::Solid => {
layer
.quads
.order
.push((quad::Order::Gradient, 1));
layer.quads.index += 1;
}
quad::Order::Gradient => {
*count += 1;
}
}
}
(None, _) => {
layer.quads.order.push((quad_order, 1));
}
}
layer.quads.add(quad, background);
}
Primitive::Image { handle, bounds } => {
let layer = &mut layers[current_layer];
Expand Down Expand Up @@ -350,32 +272,3 @@ impl<'a> Layer<'a> {
}
}
}

/// Packs the [`Gradient`] for use in shader code.
fn pack_gradient(gradient: &core::Gradient, bounds: Rectangle) -> [f32; 44] {
match gradient {
core::Gradient::Linear(linear) => {
let mut pack: [f32; 44] = [0.0; 44];

for (index, stop) in linear.stops.iter().enumerate() {
let [r, g, b, a] =
stop.map_or(Color::default(), |s| s.color).into_linear();

pack[index * 4] = r;
pack[(index * 4) + 1] = g;
pack[(index * 4) + 2] = b;
pack[(index * 4) + 3] = a;
pack[32 + index] = stop.map_or(2.0, |s| s.offset);
}

let (start, end) = linear.angle.to_distance(&bounds);

pack[40] = start.x;
pack[41] = start.y;
pack[42] = end.x;
pack[43] = end.y;

pack
}
}
}
122 changes: 121 additions & 1 deletion wgpu/src/layer/quad.rs
View file
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
//! A rectangle with certain styled properties.
use crate::core::{Background, Color, Rectangle};
use bytemuck::{Pod, Zeroable};

/// The properties of a quad.
Expand Down Expand Up @@ -58,3 +58,123 @@ pub enum Order {
/// A gradient quad
Gradient,
}

/// A group of [`Quad`]s rendered together.
#[derive(Default, Debug)]
pub struct Layer {
/// The solid quads of the [`Layer`].
solids: Vec<Solid>,

/// The gradient quads of the [`Layer`].
gradients: Vec<Gradient>,

/// The quad order of the [`Layer`]; stored as a tuple of the quad type & its count.
order: Vec<(Order, usize)>,

/// The last index of quad ordering.
index: usize,
}

impl Layer {
/// Returns true if there are no quads of any type in [`Quads`].
pub fn is_empty(&self) -> bool {
self.solids.is_empty() && self.gradients.is_empty()
}

/// The [`Solid`] quads of the [`Layer`].
pub fn solids(&self) -> &[Solid] {
&self.solids
}

/// The [`Gradient`] quads of the [`Layer`].
pub fn gradients(&self) -> &[Gradient] {
&self.gradients
}

/// The order of quads within the [`Layer`], grouped by (type, count) for rendering in batches.
pub fn ordering(&self) -> &[(Order, usize)] {
&self.order
}

/// Adds a [`Quad`] with the provided `Background` type to the quad [`Layer`].
pub fn add(&mut self, quad: Quad, background: &Background) {
let quad_order = match background {
Background::Color(color) => {
self.solids.push(Solid {
color: color.into_linear(),
quad,
});

Order::Solid
}
Background::Gradient(gradient) => {
let quad = Gradient {
gradient: pack_gradient(
gradient,
Rectangle::new(quad.position.into(), quad.size.into()),
),
quad,
};

self.gradients.push(quad);
Order::Gradient
}
};

match (self.order.get_mut(self.index), quad_order) {
(Some((quad_order, count)), Order::Solid) => match quad_order {
Order::Solid => {
*count += 1;
}
Order::Gradient => {
self.order.push((Order::Solid, 1));
self.index += 1;
}
},
(Some((quad_order, count)), Order::Gradient) => match quad_order {
Order::Solid => {
self.order.push((Order::Gradient, 1));
self.index += 1;
}
Order::Gradient => {
*count += 1;
}
},
(None, _) => {
self.order.push((quad_order, 1));
}
}
}
}

/// Packs the [`Gradient`] for use in shader code.
fn pack_gradient(
gradient: &crate::core::Gradient,
bounds: Rectangle,
) -> [f32; 44] {
match gradient {
crate::core::Gradient::Linear(linear) => {
let mut pack: [f32; 44] = [0.0; 44];

for (index, stop) in linear.stops.iter().enumerate() {
let [r, g, b, a] =
stop.map_or(Color::default(), |s| s.color).into_linear();

pack[index * 4] = r;
pack[(index * 4) + 1] = g;
pack[(index * 4) + 2] = b;
pack[(index * 4) + 3] = a;
pack[32 + index] = stop.map_or(2.0, |s| s.offset);
}

let (start, end) = linear.angle.to_distance(&bounds);

pack[40] = start.x;
pack[41] = start.y;
pack[42] = end.x;
pack[43] = end.y;

pack
}
}
}
37 changes: 15 additions & 22 deletions wgpu/src/quad.rs
View file
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
use crate::core::Rectangle;
use crate::graphics::Transformation;
use crate::layer::{self, quad};
use crate::layer::quad;

use std::mem;
use wgpu::util::DeviceExt;
Expand Down Expand Up @@ -69,7 +69,7 @@ impl Pipeline {
&mut self,
device: &wgpu::Device,
queue: &wgpu::Queue,
instances: &layer::Quads,
quads: &quad::Layer,
transformation: Transformation,
scale: f32,
) {
Expand All @@ -78,7 +78,7 @@ impl Pipeline {
}

let layer = &mut self.layers[self.prepare_layer];
layer.prepare(device, queue, instances, transformation, scale);
layer.prepare(device, queue, quads, transformation, scale);

self.prepare_layer += 1;
}
Expand All @@ -87,7 +87,7 @@ impl Pipeline {
&'a self,
layer: usize,
bounds: Rectangle<u32>,
ordering: &Vec<(quad::Order, usize)>,
quads: &quad::Layer,
render_pass: &mut wgpu::RenderPass<'a>,
) {
if let Some(layer) = self.layers.get(layer) {
Expand All @@ -106,7 +106,7 @@ impl Pipeline {
let mut solid_offset = 0;
let mut gradient_offset = 0;

for (quad_order, count) in ordering {
for (quad_order, count) in quads.ordering() {
match quad_order {
quad::Order::Solid => {
render_pass.set_pipeline(&self.solid.pipeline);
Expand Down Expand Up @@ -177,7 +177,7 @@ impl Layer {
&mut self,
device: &wgpu::Device,
queue: &wgpu::Queue,
instances: &layer::Quads,
quads: &quad::Layer,
transformation: Transformation,
scale: f32,
) {
Expand All @@ -192,22 +192,15 @@ impl Layer {
bytemuck::bytes_of(&uniforms),
);

let _ = self.solid.instances.resize(device, instances.solids.len());
let _ = self
.gradient
.instances
.resize(device, instances.gradients.len());
let _ =
self.solid
.instances
.write(queue, 0, instances.solids.as_slice());
self.solid.instance_count = instances.solids.len();
let _ = self.gradient.instances.write(
queue,
0,
instances.gradients.as_slice(),
);
self.gradient.instance_count = instances.gradients.len();
let solids = quads.solids();
let gradients = quads.gradients();

let _ = self.solid.instances.resize(device, solids.len());
let _ = self.gradient.instances.resize(device, gradients.len());
let _ = self.solid.instances.write(queue, 0, solids);
self.solid.instance_count = solids.len();
let _ = self.gradient.instances.write(queue, 0, gradients);
self.gradient.instance_count = gradients.len();
}
}

Expand Down

0 comments on commit 10a68ac

Please sign in to comment.