Skip to content

Commit

Permalink
Add return values to Frame::with_clip() and Frame::with_save()
Browse files Browse the repository at this point in the history
  • Loading branch information
MichalLebeda authored and hecrj committed Jan 31, 2024
1 parent 5540ac0 commit f1c6339
Show file tree
Hide file tree
Showing 2 changed files with 24 additions and 8 deletions.
16 changes: 12 additions & 4 deletions renderer/src/geometry.rs
Original file line number Diff line number Diff line change
Expand Up @@ -111,12 +111,14 @@ impl Frame {
/// This method is useful to compose transforms and perform drawing
/// operations in different coordinate systems.
#[inline]
pub fn with_save(&mut self, f: impl FnOnce(&mut Frame)) {
pub fn with_save<R>(&mut self, f: impl FnOnce(&mut Frame) -> R) -> R {
delegate!(self, frame, frame.push_transform());

f(self);
let result = f(self);

delegate!(self, frame, frame.pop_transform());

result
}

/// Executes the given drawing operations within a [`Rectangle`] region,
Expand All @@ -126,7 +128,11 @@ impl Frame {
/// This method is useful to perform drawing operations that need to be
/// clipped.
#[inline]
pub fn with_clip(&mut self, region: Rectangle, f: impl FnOnce(&mut Frame)) {
pub fn with_clip<R>(
&mut self,
region: Rectangle,
f: impl FnOnce(&mut Frame) -> R,
) -> R {
let mut frame = match self {
Self::TinySkia(_) => Self::TinySkia(
iced_tiny_skia::geometry::Frame::new(region.size()),
Expand All @@ -137,7 +143,7 @@ impl Frame {
}
};

f(&mut frame);
let result = f(&mut frame);

let origin = Point::new(region.x, region.y);

Expand All @@ -152,6 +158,8 @@ impl Frame {
#[allow(unreachable_patterns)]
_ => unreachable!(),
};

result
}

/// Applies a translation to the current transform of the [`Frame`].
Expand Down
16 changes: 12 additions & 4 deletions wgpu/src/geometry.rs
Original file line number Diff line number Diff line change
Expand Up @@ -388,12 +388,14 @@ impl Frame {
/// This method is useful to compose transforms and perform drawing
/// operations in different coordinate systems.
#[inline]
pub fn with_save(&mut self, f: impl FnOnce(&mut Frame)) {
pub fn with_save<R>(&mut self, f: impl FnOnce(&mut Frame) -> R) -> R {
self.push_transform();

f(self);
let result = f(self);

self.pop_transform();

result
}

/// Pushes the current transform in the transform stack.
Expand All @@ -413,14 +415,20 @@ impl Frame {
/// This method is useful to perform drawing operations that need to be
/// clipped.
#[inline]
pub fn with_clip(&mut self, region: Rectangle, f: impl FnOnce(&mut Frame)) {
pub fn with_clip<R>(
&mut self,
region: Rectangle,
f: impl FnOnce(&mut Frame) -> R,
) -> R {
let mut frame = Frame::new(region.size());

f(&mut frame);
let result = f(&mut frame);

let origin = Point::new(region.x, region.y);

self.clip(frame, origin);

result
}

/// Draws the clipped contents of the given [`Frame`] with origin at the given [`Point`].
Expand Down

0 comments on commit f1c6339

Please sign in to comment.