mod text;
use std::borrow::Cow;
use std::fmt;
use cairo::{BorrowError, Context, Filter, Format, ImageSurface, Matrix, Status, SurfacePattern};
use piet::kurbo::{Affine, PathEl, Point, QuadBez, Rect, Shape, Size};
use piet::{
Color, Error, FixedGradient, ImageFormat, InterpolationMode, IntoBrush, LineCap, LineJoin,
RenderContext, StrokeStyle,
};
pub use crate::text::{
CairoFont, CairoFontBuilder, CairoText, CairoTextLayout, CairoTextLayoutBuilder,
};
pub struct CairoRenderContext<'a> {
ctx: &'a mut Context,
text: CairoText<'a>,
transform_stack: Vec<Affine>,
}
impl<'a> CairoRenderContext<'a> {
pub fn new(ctx: &mut Context) -> CairoRenderContext {
CairoRenderContext {
ctx,
text: CairoText::new(),
transform_stack: Vec::new(),
}
}
}
#[derive(Clone)]
pub enum Brush {
Solid(u32),
Linear(cairo::LinearGradient),
Radial(cairo::RadialGradient),
}
#[derive(Debug)]
struct WrappedStatus(Status);
impl fmt::Display for WrappedStatus {
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
write!(f, "Cairo error: {:?}", self.0)
}
}
impl std::error::Error for WrappedStatus {}
trait WrapError<T> {
fn wrap(self) -> Result<T, Error>;
}
impl<T> WrapError<T> for Result<T, BorrowError> {
fn wrap(self) -> Result<T, Error> {
self.map_err(|e| {
let e: Box<dyn std::error::Error> = Box::new(e);
e.into()
})
}
}
impl<T> WrapError<T> for Result<T, Status> {
fn wrap(self) -> Result<T, Error> {
self.map_err(|e| {
let e: Box<dyn std::error::Error> = Box::new(WrappedStatus(e));
e.into()
})
}
}
macro_rules! set_gradient_stops {
($dst: expr, $stops: expr) => {
for stop in $stops {
let rgba = stop.color.as_rgba_u32();
$dst.add_color_stop_rgba(
stop.pos as f64,
byte_to_frac(rgba >> 24),
byte_to_frac(rgba >> 16),
byte_to_frac(rgba >> 8),
byte_to_frac(rgba),
);
}
};
}
impl<'a> RenderContext for CairoRenderContext<'a> {
type Brush = Brush;
type Text = CairoText<'a>;
type TextLayout = CairoTextLayout;
type Image = ImageSurface;
fn status(&mut self) -> Result<(), Error> {
let status = self.ctx.status();
if status == Status::Success {
Ok(())
} else {
let e: Box<dyn std::error::Error> = Box::new(WrappedStatus(status));
Err(e.into())
}
}
fn clear(&mut self, color: Color) {
let rgba = color.as_rgba_u32();
self.ctx.set_source_rgb(
byte_to_frac(rgba >> 24),
byte_to_frac(rgba >> 16),
byte_to_frac(rgba >> 8),
);
self.ctx.paint();
}
fn solid_brush(&mut self, color: Color) -> Brush {
Brush::Solid(color.as_rgba_u32())
}
fn gradient(&mut self, gradient: impl Into<FixedGradient>) -> Result<Brush, Error> {
match gradient.into() {
FixedGradient::Linear(linear) => {
let (x0, y0) = (linear.start.x, linear.start.y);
let (x1, y1) = (linear.end.x, linear.end.y);
let lg = cairo::LinearGradient::new(x0, y0, x1, y1);
set_gradient_stops!(&lg, &linear.stops);
Ok(Brush::Linear(lg))
}
FixedGradient::Radial(radial) => {
let (xc, yc) = (radial.center.x, radial.center.y);
let (xo, yo) = (radial.origin_offset.x, radial.origin_offset.y);
let r = radial.radius;
let rg = cairo::RadialGradient::new(xc + xo, yc + yo, 0.0, xc, yc, r);
set_gradient_stops!(&rg, &radial.stops);
Ok(Brush::Radial(rg))
}
}
}
fn fill(&mut self, shape: impl Shape, brush: &impl IntoBrush<Self>) {
let brush = brush.make_brush(self, || shape.bounding_box());
self.set_path(shape);
self.set_brush(&*brush);
self.ctx.set_fill_rule(cairo::FillRule::Winding);
self.ctx.fill();
}
fn fill_even_odd(&mut self, shape: impl Shape, brush: &impl IntoBrush<Self>) {
let brush = brush.make_brush(self, || shape.bounding_box());
self.set_path(shape);
self.set_brush(&*brush);
self.ctx.set_fill_rule(cairo::FillRule::EvenOdd);
self.ctx.fill();
}
fn clip(&mut self, shape: impl Shape) {
self.set_path(shape);
self.ctx.set_fill_rule(cairo::FillRule::Winding);
self.ctx.clip();
}
fn stroke(&mut self, shape: impl Shape, brush: &impl IntoBrush<Self>, width: f64) {
let brush = brush.make_brush(self, || shape.bounding_box());
self.set_path(shape);
self.set_stroke(width, None);
self.set_brush(&*brush);
self.ctx.stroke();
}
fn stroke_styled(
&mut self,
shape: impl Shape,
brush: &impl IntoBrush<Self>,
width: f64,
style: &StrokeStyle,
) {
let brush = brush.make_brush(self, || shape.bounding_box());
self.set_path(shape);
self.set_stroke(width, Some(style));
self.set_brush(&*brush);
self.ctx.stroke();
}
fn text(&mut self) -> &mut Self::Text {
&mut self.text
}
fn draw_text(
&mut self,
layout: &Self::TextLayout,
pos: impl Into<Point>,
brush: &impl IntoBrush<Self>,
) {
let brush = brush.make_brush(self, || Rect::ZERO);
self.ctx.set_scaled_font(&layout.font);
self.set_brush(&*brush);
let pos = pos.into();
for lm in &layout.line_metrics {
self.ctx
.move_to(pos.x, pos.y + lm.cumulative_height - lm.height);
self.ctx
.show_text(&layout.text[lm.start_offset..lm.end_offset]);
}
}
fn save(&mut self) -> Result<(), Error> {
self.ctx.save();
let state = self.transform_stack.last().copied().unwrap_or_default();
self.transform_stack.push(state);
self.status()
}
fn restore(&mut self) -> Result<(), Error> {
if self.transform_stack.pop().is_some() {
self.ctx.restore();
self.status()
} else {
Err(Error::StackUnbalance)
}
}
fn finish(&mut self) -> Result<(), Error> {
self.status()
}
fn transform(&mut self, transform: Affine) {
if let Some(last) = self.transform_stack.last_mut() {
*last *= transform;
} else {
self.transform_stack.push(transform);
}
self.ctx.transform(affine_to_matrix(transform));
}
fn current_transform(&self) -> Affine {
self.transform_stack.last().copied().unwrap_or_default()
}
#[allow(clippy::identity_op)]
fn make_image(
&mut self,
width: usize,
height: usize,
buf: &[u8],
format: ImageFormat,
) -> Result<Self::Image, Error> {
let cairo_fmt = match format {
ImageFormat::Rgb => Format::Rgb24,
ImageFormat::RgbaSeparate | ImageFormat::RgbaPremul => Format::ARgb32,
_ => return Err(Error::NotSupported),
};
let mut image = ImageSurface::create(cairo_fmt, width as i32, height as i32).wrap()?;
let bytes_per_pixel = format.bytes_per_pixel();
let bytes_per_row = width * bytes_per_pixel;
let stride = image.get_stride() as usize;
{
let mut data = image.get_data().wrap()?;
for y in 0..height {
let src_off = y * bytes_per_row;
let dst_off = y * stride;
match format {
ImageFormat::Rgb => {
for x in 0..width {
data[dst_off + x * 4 + 0] = buf[src_off + x * 3 + 2];
data[dst_off + x * 4 + 1] = buf[src_off + x * 3 + 1];
data[dst_off + x * 4 + 2] = buf[src_off + x * 3 + 0];
}
}
ImageFormat::RgbaPremul => {
for x in 0..width {
data[dst_off + x * 4 + 0] = buf[src_off + x * 4 + 2];
data[dst_off + x * 4 + 1] = buf[src_off + x * 4 + 1];
data[dst_off + x * 4 + 2] = buf[src_off + x * 4 + 0];
data[dst_off + x * 4 + 3] = buf[src_off + x * 4 + 3];
}
}
ImageFormat::RgbaSeparate => {
fn premul(x: u8, a: u8) -> u8 {
let y = (x as u16) * (a as u16);
((y + (y >> 8) + 0x80) >> 8) as u8
}
for x in 0..width {
let a = buf[src_off + x * 4 + 3];
data[dst_off + x * 4 + 0] = premul(buf[src_off + x * 4 + 2], a);
data[dst_off + x * 4 + 1] = premul(buf[src_off + x * 4 + 1], a);
data[dst_off + x * 4 + 2] = premul(buf[src_off + x * 4 + 0], a);
data[dst_off + x * 4 + 3] = a;
}
}
_ => return Err(Error::NotSupported),
}
}
}
Ok(image)
}
#[inline]
fn draw_image(
&mut self,
image: &Self::Image,
dst_rect: impl Into<Rect>,
interp: InterpolationMode,
) {
draw_image(self, image, None, dst_rect.into(), interp);
}
#[inline]
fn draw_image_area(
&mut self,
image: &Self::Image,
src_rect: impl Into<Rect>,
dst_rect: impl Into<Rect>,
interp: InterpolationMode,
) {
draw_image(self, image, Some(src_rect.into()), dst_rect.into(), interp);
}
fn blurred_rect(&mut self, rect: Rect, blur_radius: f64, brush: &impl IntoBrush<Self>) {
let brush = brush.make_brush(self, || rect);
let (image, origin) = compute_blurred_rect(rect, blur_radius);
self.set_brush(&*brush);
self.ctx.mask_surface(&image, origin.x, origin.y);
}
}
fn draw_image<'a>(
ctx: &mut CairoRenderContext<'a>,
image: &<CairoRenderContext<'a> as RenderContext>::Image,
src_rect: Option<Rect>,
dst_rect: Rect,
interp: InterpolationMode,
) {
let _ = ctx.with_save(|rc| {
let surface_pattern = SurfacePattern::create(image);
let filter = match interp {
InterpolationMode::NearestNeighbor => Filter::Nearest,
InterpolationMode::Bilinear => Filter::Bilinear,
};
surface_pattern.set_filter(filter);
let src_rect = match src_rect {
Some(src_rect) => src_rect,
None => Size::new(image.get_width() as f64, image.get_height() as f64).to_rect(),
};
let scale_x = dst_rect.width() / src_rect.width();
let scale_y = dst_rect.height() / src_rect.height();
rc.clip(dst_rect);
rc.ctx.translate(
dst_rect.x0 - scale_x * src_rect.x0,
dst_rect.y0 - scale_y * src_rect.y0,
);
rc.ctx.scale(scale_x, scale_y);
rc.ctx.set_source(&surface_pattern);
rc.ctx.paint();
Ok(())
});
}
impl<'a> IntoBrush<CairoRenderContext<'a>> for Brush {
fn make_brush<'b>(
&'b self,
_piet: &mut CairoRenderContext,
_bbox: impl FnOnce() -> Rect,
) -> std::borrow::Cow<'b, Brush> {
Cow::Borrowed(self)
}
}
fn convert_line_cap(line_cap: LineCap) -> cairo::LineCap {
match line_cap {
LineCap::Butt => cairo::LineCap::Butt,
LineCap::Round => cairo::LineCap::Round,
LineCap::Square => cairo::LineCap::Square,
}
}
fn convert_line_join(line_join: LineJoin) -> cairo::LineJoin {
match line_join {
LineJoin::Miter => cairo::LineJoin::Miter,
LineJoin::Round => cairo::LineJoin::Round,
LineJoin::Bevel => cairo::LineJoin::Bevel,
}
}
impl<'a> CairoRenderContext<'a> {
fn set_brush(&mut self, brush: &Brush) {
match *brush {
Brush::Solid(rgba) => self.ctx.set_source_rgba(
byte_to_frac(rgba >> 24),
byte_to_frac(rgba >> 16),
byte_to_frac(rgba >> 8),
byte_to_frac(rgba),
),
Brush::Linear(ref linear) => self.ctx.set_source(linear),
Brush::Radial(ref radial) => self.ctx.set_source(radial),
}
}
fn set_stroke(&mut self, width: f64, style: Option<&StrokeStyle>) {
self.ctx.set_line_width(width);
let line_join = style
.and_then(|style| style.line_join)
.unwrap_or(LineJoin::Miter);
self.ctx.set_line_join(convert_line_join(line_join));
let line_cap = style
.and_then(|style| style.line_cap)
.unwrap_or(LineCap::Butt);
self.ctx.set_line_cap(convert_line_cap(line_cap));
let miter_limit = style.and_then(|style| style.miter_limit).unwrap_or(10.0);
self.ctx.set_miter_limit(miter_limit);
match style.and_then(|style| style.dash.as_ref()) {
None => self.ctx.set_dash(&[], 0.0),
Some((dashes, offset)) => self.ctx.set_dash(dashes, *offset),
}
}
fn set_path(&mut self, shape: impl Shape) {
self.ctx.new_path();
let mut last = Point::ZERO;
for el in shape.to_bez_path(1e-3) {
match el {
PathEl::MoveTo(p) => {
self.ctx.move_to(p.x, p.y);
last = p;
}
PathEl::LineTo(p) => {
self.ctx.line_to(p.x, p.y);
last = p;
}
PathEl::QuadTo(p1, p2) => {
let q = QuadBez::new(last, p1, p2);
let c = q.raise();
self.ctx
.curve_to(c.p1.x, c.p1.y, c.p2.x, c.p2.y, p2.x, p2.y);
last = p2;
}
PathEl::CurveTo(p1, p2, p3) => {
self.ctx.curve_to(p1.x, p1.y, p2.x, p2.y, p3.x, p3.y);
last = p3;
}
PathEl::ClosePath => self.ctx.close_path(),
}
}
}
}
fn byte_to_frac(byte: u32) -> f64 {
((byte & 255) as f64) * (1.0 / 255.0)
}
fn affine_to_matrix(affine: Affine) -> Matrix {
let a = affine.as_coeffs();
Matrix {
xx: a[0],
yx: a[1],
xy: a[2],
yy: a[3],
x0: a[4],
y0: a[5],
}
}
fn compute_blurred_rect(rect: Rect, radius: f64) -> (ImageSurface, Point) {
let size = piet::util::size_for_blurred_rect(rect, radius);
let mut image =
ImageSurface::create(Format::A8, size.width as i32, size.height as i32).unwrap();
let stride = image.get_stride() as usize;
let mut data = image.get_data().unwrap();
let rect_exp = piet::util::compute_blurred_rect(rect, radius, stride, &mut *data);
std::mem::drop(data);
let origin = rect_exp.origin();
(image, origin)
}