typst/tests/typeset.rs

826 lines
25 KiB
Rust
Raw Normal View History

use std::env;
use std::ffi::OsStr;
2022-01-01 14:12:50 +03:00
use std::fs;
use std::path::Path;
use std::rc::Rc;
2020-12-01 00:07:08 +03:00
use image::{GenericImageView, Rgba};
2021-07-08 23:33:44 +03:00
use tiny_skia as sk;
use ttf_parser::{GlyphId, OutlineBuilder};
2021-12-06 16:58:57 +03:00
use usvg::FitTo;
use walkdir::WalkDir;
use typst::diag::Error;
2021-12-30 14:12:19 +03:00
use typst::eval::{Smart, StyleMap, Value};
2021-08-27 14:29:50 +03:00
use typst::font::Face;
2022-01-01 14:12:50 +03:00
use typst::frame::{Element, Frame, Geometry, Shape, Stroke, Text};
use typst::geom::{self, Color, Length, Paint, PathElement, RgbaColor, Size, Transform};
2021-12-06 16:58:57 +03:00
use typst::image::{Image, RasterImage, Svg};
use typst::library::{PageNode, TextNode};
2021-08-09 12:06:37 +03:00
use typst::loading::FsLoader;
use typst::parse::Scanner;
use typst::source::SourceFile;
use typst::syntax::Span;
2021-07-20 21:21:56 +03:00
use typst::Context;
2022-01-01 14:12:50 +03:00
#[cfg(feature = "layout-cache")]
use {
filedescriptor::{FileDescriptor, StdioDescriptor::*},
std::fs::File,
typst::layout::RootNode,
};
2021-02-20 19:53:40 +03:00
const TYP_DIR: &str = "./typ";
const REF_DIR: &str = "./ref";
const PNG_DIR: &str = "./png";
const PDF_DIR: &str = "./pdf";
const FONT_DIR: &str = "../fonts";
fn main() {
env::set_current_dir(env::current_dir().unwrap().join("tests")).unwrap();
let args = Args::new(env::args().skip(1));
let mut filtered = Vec::new();
2022-01-01 14:12:50 +03:00
// Since differents tests can affect each other through the layout cache, a
// deterministic order is very important for reproducibility.
for entry in WalkDir::new(".").sort_by_file_name() {
let entry = entry.unwrap();
if entry.depth() <= 1 {
continue;
}
let src_path = entry.into_path();
if src_path.extension() != Some(OsStr::new("typ")) {
continue;
}
if args.matches(&src_path) {
filtered.push(src_path);
}
}
let len = filtered.len();
2021-01-13 23:33:22 +03:00
if len == 1 {
println!("Running test ...");
2021-01-13 23:33:22 +03:00
} else if len > 1 {
println!("Running {} tests", len);
}
// Set page width to 120pt with 10pt margins, so that the inner page is
// exactly 100pt wide. Page height is unbounded and font size is 10pt so
// that it multiplies to nice round numbers.
2021-12-30 14:12:19 +03:00
let mut styles = StyleMap::new();
styles.set(PageNode::WIDTH, Smart::Custom(Length::pt(120.0)));
styles.set(PageNode::HEIGHT, Smart::Auto);
styles.set(PageNode::LEFT, Smart::Custom(Length::pt(10.0).into()));
styles.set(PageNode::TOP, Smart::Custom(Length::pt(10.0).into()));
styles.set(PageNode::RIGHT, Smart::Custom(Length::pt(10.0).into()));
styles.set(PageNode::BOTTOM, Smart::Custom(Length::pt(10.0).into()));
2021-12-15 13:11:57 +03:00
styles.set(TextNode::SIZE, Length::pt(10.0).into());
2021-07-20 21:21:56 +03:00
// Hook up an assert function into the global scope.
2021-07-21 12:25:49 +03:00
let mut std = typst::library::new();
std.def_func("test", move |_, args| {
let lhs = args.expect::<Value>("left-hand side")?;
let rhs = args.expect::<Value>("right-hand side")?;
if lhs != rhs {
2021-08-12 14:39:33 +03:00
return Err(Error::boxed(
args.span,
2021-08-12 14:39:33 +03:00
format!("Assertion failed: {:?} != {:?}", lhs, rhs),
));
}
Ok(Value::None)
});
2021-07-21 12:25:49 +03:00
// Create loader and context.
let loader = FsLoader::new().with_path(FONT_DIR).wrap();
let mut ctx = Context::builder().std(std).styles(styles).build(loader);
2021-07-21 12:25:49 +03:00
// Run all the tests.
2021-12-15 13:11:57 +03:00
let mut ok = 0;
for src_path in filtered {
let path = src_path.strip_prefix(TYP_DIR).unwrap();
let png_path = Path::new(PNG_DIR).join(path).with_extension("png");
let ref_path = Path::new(REF_DIR).join(path).with_extension("png");
let pdf_path =
args.pdf.then(|| Path::new(PDF_DIR).join(path).with_extension("pdf"));
2021-12-15 13:11:57 +03:00
ok += test(
2021-07-20 21:21:56 +03:00
&mut ctx,
&src_path,
&png_path,
&ref_path,
pdf_path.as_deref(),
args.debug,
2021-12-15 13:11:57 +03:00
) as usize;
}
if len > 1 {
println!("{} / {} tests passed.", ok, len);
2021-01-13 16:07:38 +03:00
}
2021-12-15 13:11:57 +03:00
if ok < len {
std::process::exit(1);
}
}
struct Args {
filter: Vec<String>,
exact: bool,
debug: bool,
pdf: bool,
}
impl Args {
fn new(args: impl Iterator<Item = String>) -> Self {
let mut filter = Vec::new();
let mut exact = false;
let mut debug = false;
let mut pdf = false;
for arg in args {
match arg.as_str() {
// Ignore this, its for cargo.
"--nocapture" => {}
// Match only the exact filename.
"--exact" => exact = true,
// Generate PDFs.
"--pdf" => pdf = true,
// Debug print the layout trees.
"--debug" | "-d" => debug = true,
// Everything else is a file filter.
_ => filter.push(arg),
}
}
Self { filter, pdf, debug, exact }
}
fn matches(&self, path: &Path) -> bool {
if self.exact {
let name = path.file_name().unwrap().to_string_lossy();
self.filter.iter().any(|v| v == &name)
} else {
let path = path.to_string_lossy();
self.filter.is_empty() || self.filter.iter().any(|v| path.contains(v))
}
}
}
fn test(
2021-07-20 21:21:56 +03:00
ctx: &mut Context,
src_path: &Path,
png_path: &Path,
2021-02-20 19:53:40 +03:00
ref_path: &Path,
pdf_path: Option<&Path>,
debug: bool,
) -> bool {
let name = src_path.strip_prefix(TYP_DIR).unwrap_or(src_path);
println!("Testing {}", name.display());
let src = fs::read_to_string(src_path).unwrap();
2021-01-13 16:07:38 +03:00
let mut ok = true;
let mut frames = vec![];
let mut line = 0;
2021-02-01 00:43:11 +03:00
let mut compare_ref = true;
let mut compare_ever = false;
2021-02-01 00:43:11 +03:00
2021-06-29 17:05:05 +03:00
let parts: Vec<_> = src.split("\n---").collect();
for (i, &part) in parts.iter().enumerate() {
2021-02-01 00:43:11 +03:00
let is_header = i == 0
&& parts.len() > 1
&& part
.lines()
.all(|s| s.starts_with("//") || s.chars().all(|c| c.is_whitespace()));
if is_header {
for line in part.lines() {
if line.starts_with("// Ref: false") {
compare_ref = false;
}
}
} else {
let (part_ok, compare_here, part_frames) =
test_part(ctx, src_path, part.into(), i, compare_ref, line, debug);
2021-02-01 00:43:11 +03:00
ok &= part_ok;
compare_ever |= compare_here;
2021-02-01 00:43:11 +03:00
frames.extend(part_frames);
}
line += part.lines().count() + 1;
2021-01-13 16:07:38 +03:00
}
if compare_ever {
if let Some(pdf_path) = pdf_path {
2021-07-20 21:21:56 +03:00
let pdf_data = typst::export::pdf(ctx, &frames);
fs::create_dir_all(&pdf_path.parent().unwrap()).unwrap();
fs::write(pdf_path, pdf_data).unwrap();
}
2021-01-13 16:07:38 +03:00
2021-07-20 21:21:56 +03:00
let canvas = draw(ctx, &frames, 2.0);
fs::create_dir_all(&png_path.parent().unwrap()).unwrap();
2021-03-23 15:17:00 +03:00
canvas.save_png(png_path).unwrap();
2021-01-14 01:19:44 +03:00
2021-07-08 23:33:44 +03:00
if let Ok(ref_pixmap) = sk::Pixmap::load_png(ref_path) {
2021-03-23 15:17:00 +03:00
if canvas != ref_pixmap {
2021-02-20 19:53:40 +03:00
println!(" Does not match reference image. ❌");
2021-01-13 16:07:38 +03:00
ok = false;
}
} else if !frames.is_empty() {
2021-02-20 19:53:40 +03:00
println!(" Failed to open reference image. ❌");
ok = false;
2021-01-13 16:07:38 +03:00
}
}
if ok {
if !debug {
print!("\x1b[1A");
}
println!("Testing {}", name.display());
2021-01-13 16:07:38 +03:00
}
ok
}
2021-02-01 00:43:11 +03:00
fn test_part(
2021-07-20 21:21:56 +03:00
ctx: &mut Context,
2021-08-09 12:06:37 +03:00
src_path: &Path,
src: String,
2021-02-01 00:43:11 +03:00
i: usize,
compare_ref: bool,
line: usize,
debug: bool,
2021-06-18 14:01:55 +03:00
) -> (bool, bool, Vec<Rc<Frame>>) {
2021-08-09 12:06:37 +03:00
let id = ctx.sources.provide(src_path, src);
let source = ctx.sources.get(id);
let (local_compare_ref, mut ref_errors) = parse_metadata(&source);
2021-02-01 00:43:11 +03:00
let compare_ref = local_compare_ref.unwrap_or(compare_ref);
let mut ok = true;
let (frames, mut errors) = match ctx.evaluate(id) {
Ok(module) => {
let tree = module.into_root();
if debug {
println!("{:#?}", tree);
}
let mut frames = tree.layout(ctx);
2021-06-18 14:00:36 +03:00
#[cfg(feature = "layout-cache")]
(ok &= test_incremental(ctx, i, &tree, &frames));
2021-06-27 19:06:39 +03:00
if !compare_ref {
frames.clear();
}
(frames, vec![])
}
Err(errors) => (vec![], *errors),
};
// TODO: Also handle errors from other files.
2021-08-13 13:21:14 +03:00
errors.retain(|error| error.span.source == id);
for error in &mut errors {
error.trace.clear();
}
2021-08-21 17:38:51 +03:00
// The comparison never fails since all spans are from the same source file.
ref_errors.sort_by(|a, b| a.span.partial_cmp(&b.span).unwrap());
errors.sort_by(|a, b| a.span.partial_cmp(&b.span).unwrap());
2021-07-21 12:25:49 +03:00
if errors != ref_errors {
println!(" Subtest {} does not match expected errors. ❌", i);
ok = false;
2021-08-09 12:06:37 +03:00
let source = ctx.sources.get(id);
for error in errors.iter() {
2021-08-13 13:21:14 +03:00
if error.span.source == id && !ref_errors.contains(error) {
print!(" Not annotated | ");
print_error(&source, line, error);
}
}
for error in ref_errors.iter() {
if !errors.contains(error) {
print!(" Not emitted | ");
print_error(&source, line, error);
}
}
}
(ok, compare_ref, frames)
}
#[cfg(feature = "layout-cache")]
fn test_incremental(
ctx: &mut Context,
i: usize,
tree: &RootNode,
frames: &[Rc<Frame>],
) -> bool {
let mut ok = true;
let reference = ctx.layouts.clone();
for level in 0 .. reference.levels() {
ctx.layouts = reference.clone();
ctx.layouts.retain(|x| x == level);
if ctx.layouts.is_empty() {
continue;
}
2021-07-20 21:21:56 +03:00
ctx.layouts.turnaround();
let cached = silenced(|| tree.layout(ctx));
let misses = ctx
.layouts
.entries()
.filter(|e| e.level() == level && !e.hit() && e.age() == 2)
.count();
if misses > 0 {
println!(
" Subtest {} relayout had {} cache misses on level {} of {} ❌",
i,
misses,
level,
reference.levels() - 1,
);
ok = false;
}
if cached != frames {
2021-12-15 13:11:57 +03:00
println!(
" Subtest {} relayout differs from clean pass on level {} ❌",
i, level
);
ok = false;
}
2021-06-27 19:06:39 +03:00
}
ctx.layouts = reference;
ctx.layouts.turnaround();
ok
}
fn parse_metadata(source: &SourceFile) -> (Option<bool>, Vec<Error>) {
2021-02-01 00:43:11 +03:00
let mut compare_ref = None;
let mut errors = vec![];
let lines: Vec<_> = source.src().lines().map(str::trim).collect();
for (i, line) in lines.iter().enumerate() {
2021-02-01 00:43:11 +03:00
if line.starts_with("// Ref: false") {
compare_ref = Some(false);
}
if line.starts_with("// Ref: true") {
compare_ref = Some(true);
}
2021-01-13 16:07:38 +03:00
let rest = if let Some(rest) = line.strip_prefix("// Error: ") {
rest
} else {
continue;
};
fn num(s: &mut Scanner) -> usize {
s.eat_while(|c| c.is_numeric()).parse().unwrap()
}
let comments =
lines[i ..].iter().take_while(|line| line.starts_with("//")).count();
let pos = |s: &mut Scanner| -> usize {
let first = num(s) - 1;
let (delta, column) =
if s.eat_if(':') { (first, num(s) - 1) } else { (0, first) };
let line = (i + comments) + delta;
source.line_column_to_byte(line, column).unwrap()
};
2021-01-13 16:07:38 +03:00
let mut s = Scanner::new(rest);
2021-02-17 23:30:20 +03:00
let start = pos(&mut s);
let end = if s.eat_if('-') { pos(&mut s) } else { start };
2021-08-13 13:21:14 +03:00
let span = Span::new(source.id(), start, end);
2021-02-17 23:30:20 +03:00
2021-08-13 13:21:14 +03:00
errors.push(Error::new(span, s.rest().trim()));
}
(compare_ref, errors)
}
fn print_error(source: &SourceFile, line: usize, error: &Error) {
let start_line = 1 + line + source.byte_to_line(error.span.start).unwrap();
let start_col = 1 + source.byte_to_column(error.span.start).unwrap();
let end_line = 1 + line + source.byte_to_line(error.span.end).unwrap();
let end_col = 1 + source.byte_to_column(error.span.end).unwrap();
println!(
"Error: {}:{}-{}:{}: {}",
start_line, start_col, end_line, end_col, error.message
);
}
2021-08-27 14:29:50 +03:00
fn draw(ctx: &Context, frames: &[Rc<Frame>], dpp: f32) -> sk::Pixmap {
2020-11-25 20:46:47 +03:00
let pad = Length::pt(5.0);
2021-11-26 18:32:06 +03:00
let width = 2.0 * pad + frames.iter().map(|l| l.size.x).max().unwrap_or_default();
let height = pad + frames.iter().map(|l| l.size.y + pad).sum::<Length>();
2020-08-30 23:18:55 +03:00
2021-11-20 17:51:07 +03:00
let pxw = (dpp * width.to_f32()) as u32;
let pxh = (dpp * height.to_f32()) as u32;
2021-11-16 23:32:29 +03:00
if pxw > 4000 || pxh > 4000 {
panic!(
"overlarge image: {} by {} ({:?} x {:?})",
2021-11-16 23:32:29 +03:00
pxw, pxh, width, height,
);
2021-01-14 01:19:44 +03:00
}
2021-11-16 23:32:29 +03:00
let mut canvas = sk::Pixmap::new(pxw, pxh).unwrap();
2021-07-08 23:33:44 +03:00
canvas.fill(sk::Color::BLACK);
2021-11-16 23:32:29 +03:00
let mut mask = sk::ClipMask::new();
let rect = sk::Rect::from_xywh(0.0, 0.0, pxw as f32, pxh as f32).unwrap();
let path = sk::PathBuilder::from_rect(rect);
mask.set_path(pxw, pxh, &path, sk::FillRule::default(), false);
2021-11-20 17:51:07 +03:00
let mut ts =
sk::Transform::from_scale(dpp, dpp).pre_translate(pad.to_f32(), pad.to_f32());
2021-11-16 23:32:29 +03:00
2021-01-03 02:12:09 +03:00
for frame in frames {
2021-11-16 23:32:29 +03:00
let mut background = sk::Paint::default();
background.set_color(sk::Color::WHITE);
2021-11-26 18:32:06 +03:00
let w = frame.size.x.to_f32();
let h = frame.size.y.to_f32();
2021-11-16 23:32:29 +03:00
let rect = sk::Rect::from_xywh(0.0, 0.0, w, h).unwrap();
canvas.fill_rect(rect, &background, ts, None);
2022-01-01 14:12:50 +03:00
draw_frame(&mut canvas, ts, &mask, ctx, frame, true);
2021-11-26 18:32:06 +03:00
ts = ts.pre_translate(0.0, (frame.size.y + pad).to_f32());
}
2020-11-25 20:46:47 +03:00
canvas
}
2021-11-16 23:32:29 +03:00
fn draw_frame(
canvas: &mut sk::Pixmap,
ts: sk::Transform,
mask: &sk::ClipMask,
ctx: &Context,
frame: &Frame,
2022-01-01 14:12:50 +03:00
clip: bool,
2021-11-16 23:32:29 +03:00
) {
2022-01-01 14:12:50 +03:00
let mut storage;
let mut mask = mask;
if clip {
let w = frame.size.x.to_f32();
let h = frame.size.y.to_f32();
let rect = sk::Rect::from_xywh(0.0, 0.0, w, h).unwrap();
let path = sk::PathBuilder::from_rect(rect).transform(ts).unwrap();
let rule = sk::FillRule::default();
storage = mask.clone();
if storage.intersect_path(&path, rule, false).is_none() {
// Fails if clipping rect is empty. In that case we just clip
// everything by returning.
return;
}
mask = &storage;
}
2021-11-16 23:32:29 +03:00
for (pos, element) in &frame.elements {
2021-11-20 17:51:07 +03:00
let x = pos.x.to_f32();
let y = pos.y.to_f32();
2021-11-16 23:32:29 +03:00
let ts = ts.pre_translate(x, y);
match *element {
2021-11-22 17:26:56 +03:00
Element::Group(ref group) => {
2022-01-01 14:12:50 +03:00
let ts = ts.pre_concat(convert_typst_transform(group.transform));
draw_frame(canvas, ts, &mask, ctx, &group.frame, group.clips);
2021-11-22 17:26:56 +03:00
}
2021-11-16 23:32:29 +03:00
Element::Text(ref text) => {
draw_text(canvas, ts, mask, ctx.fonts.get(text.face_id), text);
}
2021-11-20 17:51:07 +03:00
Element::Shape(ref shape) => {
draw_shape(canvas, ts, mask, shape);
2021-11-16 23:32:29 +03:00
}
Element::Image(id, size) => {
draw_image(canvas, ts, mask, ctx.images.get(id), size);
}
Element::Link(_, s) => {
2021-11-20 17:51:07 +03:00
let fill = RgbaColor::new(40, 54, 99, 40).into();
let shape = Shape::filled(Geometry::Rect(s), fill);
draw_shape(canvas, ts, mask, &shape);
2021-11-16 23:32:29 +03:00
}
}
}
}
fn draw_text(
canvas: &mut sk::Pixmap,
ts: sk::Transform,
mask: &sk::ClipMask,
face: &Face,
text: &Text,
) {
2021-08-27 14:29:50 +03:00
let ttf = face.ttf();
2021-11-20 17:51:07 +03:00
let size = text.size.to_f32();
let units_per_em = face.units_per_em as f32;
let pixels_per_em = text.size.to_f32() * ts.sy;
2021-08-27 14:29:50 +03:00
let scale = size / units_per_em;
2021-08-27 14:29:50 +03:00
let mut x = 0.0;
for glyph in &text.glyphs {
2021-08-27 14:29:50 +03:00
let glyph_id = GlyphId(glyph.id);
2021-12-15 13:11:57 +03:00
let offset = x + glyph.x_offset.resolve(text.size).to_f32();
2021-08-27 14:29:50 +03:00
let ts = ts.pre_translate(offset, 0.0);
2020-11-25 20:46:47 +03:00
if let Some(tree) = ttf
2021-08-27 14:29:50 +03:00
.glyph_svg_image(glyph_id)
2021-03-23 17:32:36 +03:00
.and_then(|data| std::str::from_utf8(data).ok())
.map(|svg| {
let viewbox = format!("viewBox=\"0 0 {0} {0}\" xmlns", units_per_em);
svg.replace("xmlns", &viewbox)
})
2021-12-06 16:58:57 +03:00
.and_then(|s| {
usvg::Tree::from_str(&s, &usvg::Options::default().to_ref()).ok()
})
2021-03-23 17:32:36 +03:00
{
for child in tree.root().children() {
if let usvg::NodeKind::Path(node) = &*child.borrow() {
2021-08-27 14:29:50 +03:00
// SVG is already Y-down, no flipping required.
let ts = convert_usvg_transform(node.transform)
2021-08-27 14:29:50 +03:00
.post_scale(scale, scale)
2021-03-23 17:32:36 +03:00
.post_concat(ts);
2021-08-27 14:29:50 +03:00
2021-03-23 17:32:36 +03:00
if let Some(fill) = &node.fill {
2021-08-27 14:29:50 +03:00
let path = convert_usvg_path(&node.data);
2021-03-23 17:32:36 +03:00
let (paint, fill_rule) = convert_usvg_fill(fill);
2021-11-16 23:32:29 +03:00
canvas.fill_path(&path, &paint, fill_rule, ts, Some(mask));
2021-03-23 17:32:36 +03:00
}
}
}
2021-08-27 14:29:50 +03:00
} else if let Some(raster) =
ttf.glyph_raster_image(glyph_id, pixels_per_em as u16)
{
// TODO: Vertical alignment isn't quite right for Apple Color Emoji,
// and maybe also for Noto Color Emoji. And: Is the size calculation
// correct?
2021-12-06 16:58:57 +03:00
let img = RasterImage::parse(&raster.data).unwrap();
2021-08-27 14:29:50 +03:00
let h = text.size;
let w = (img.width() as f64 / img.height() as f64) * h;
let dx = (raster.x as f32) / (img.width() as f32) * size;
let dy = (raster.y as f32) / (img.height() as f32) * size;
let ts = ts.pre_translate(dx, -size - dy);
2021-12-06 16:58:57 +03:00
draw_image(canvas, ts, mask, &Image::Raster(img), Size::new(w, h));
} else {
// Otherwise, draw normal outline.
2021-07-08 23:33:44 +03:00
let mut builder = WrappedPathBuilder(sk::PathBuilder::new());
2021-08-27 14:29:50 +03:00
if ttf.outline_glyph(glyph_id, &mut builder).is_some() {
2021-11-20 17:51:07 +03:00
// Flip vertically because font design coordinate system is Y-up.
2021-08-27 14:29:50 +03:00
let ts = ts.pre_scale(scale, -scale);
let path = builder.0.finish().unwrap();
2021-11-20 17:51:07 +03:00
let paint = convert_typst_paint(text.fill);
2021-11-16 23:32:29 +03:00
canvas.fill_path(&path, &paint, sk::FillRule::default(), ts, Some(mask));
}
2021-03-23 17:32:36 +03:00
}
2021-12-15 13:11:57 +03:00
x += glyph.x_advance.resolve(text.size).to_f32();
}
}
2021-11-20 17:51:07 +03:00
fn draw_shape(
2021-07-08 23:33:44 +03:00
canvas: &mut sk::Pixmap,
ts: sk::Transform,
2021-11-16 23:32:29 +03:00
mask: &sk::ClipMask,
2021-11-20 17:51:07 +03:00
shape: &Shape,
2021-07-08 23:33:44 +03:00
) {
2021-11-20 17:51:07 +03:00
let path = match shape.geometry {
Geometry::Rect(size) => {
2021-11-26 18:32:06 +03:00
let w = size.x.to_f32();
let h = size.y.to_f32();
2021-07-08 23:33:44 +03:00
let rect = sk::Rect::from_xywh(0.0, 0.0, w, h).unwrap();
2021-11-20 17:51:07 +03:00
sk::PathBuilder::from_rect(rect)
2021-03-20 22:19:30 +03:00
}
2021-07-08 23:33:44 +03:00
Geometry::Ellipse(size) => {
2021-11-20 17:51:07 +03:00
let approx = geom::Path::ellipse(size);
convert_typst_path(&approx)
2021-06-11 00:08:52 +03:00
}
2021-11-20 17:51:07 +03:00
Geometry::Line(target) => {
let mut builder = sk::PathBuilder::new();
builder.line_to(target.x.to_f32(), target.y.to_f32());
builder.finish().unwrap()
}
2021-11-20 17:51:07 +03:00
Geometry::Path(ref path) => convert_typst_path(path),
2021-02-06 14:30:44 +03:00
};
2021-11-20 17:51:07 +03:00
if let Some(fill) = shape.fill {
let mut paint = convert_typst_paint(fill);
if matches!(shape.geometry, Geometry::Rect(_)) {
paint.anti_alias = false;
}
let rule = sk::FillRule::default();
canvas.fill_path(&path, &paint, rule, ts, Some(mask));
}
if let Some(Stroke { paint, thickness }) = shape.stroke {
let paint = convert_typst_paint(paint);
let mut stroke = sk::Stroke::default();
stroke.width = thickness.to_f32();
canvas.stroke_path(&path, &paint, &stroke, ts, Some(mask));
}
2021-02-04 23:30:18 +03:00
}
2021-11-16 23:32:29 +03:00
fn draw_image(
canvas: &mut sk::Pixmap,
ts: sk::Transform,
mask: &sk::ClipMask,
img: &Image,
size: Size,
) {
2021-11-26 18:32:06 +03:00
let view_width = size.x.to_f32();
let view_height = size.y.to_f32();
2021-12-06 16:58:57 +03:00
let pixmap = match img {
Image::Raster(img) => {
let w = img.buf.width();
let h = img.buf.height();
let mut pixmap = sk::Pixmap::new(w, h).unwrap();
for ((_, _, src), dest) in img.buf.pixels().zip(pixmap.pixels_mut()) {
let Rgba([r, g, b, a]) = src;
*dest = sk::ColorU8::from_rgba(r, g, b, a).premultiply();
}
pixmap
}
Image::Svg(Svg(tree)) => {
let size = tree.svg_node().size;
let aspect = (size.width() / size.height()) as f32;
let scale = ts.sx.max(ts.sy);
let w = (scale * view_width.max(aspect * view_height)).ceil() as u32;
let h = ((w as f32) / aspect).ceil() as u32;
let mut pixmap = sk::Pixmap::new(w, h).unwrap();
resvg::render(&tree, FitTo::Size(w, h), pixmap.as_mut());
pixmap
}
};
let scale_x = view_width / pixmap.width() as f32;
let scale_y = view_height / pixmap.height() as f32;
2020-11-25 20:46:47 +03:00
2021-07-08 23:33:44 +03:00
let mut paint = sk::Paint::default();
paint.shader = sk::Pattern::new(
2021-03-23 15:17:00 +03:00
pixmap.as_ref(),
2021-07-08 23:33:44 +03:00
sk::SpreadMode::Pad,
sk::FilterQuality::Bilinear,
2020-11-25 20:46:47 +03:00
1.0,
2021-08-27 14:29:50 +03:00
sk::Transform::from_scale(scale_x, scale_y),
2020-11-25 20:46:47 +03:00
);
2021-07-08 23:33:44 +03:00
let rect = sk::Rect::from_xywh(0.0, 0.0, view_width, view_height).unwrap();
2021-11-16 23:32:29 +03:00
canvas.fill_rect(rect, &paint, ts, Some(mask));
}
2021-11-24 00:04:08 +03:00
fn convert_typst_transform(transform: Transform) -> sk::Transform {
let Transform { sx, ky, kx, sy, tx, ty } = transform;
sk::Transform::from_row(
sx.get() as _,
ky.get() as _,
kx.get() as _,
sy.get() as _,
tx.to_f32(),
ty.to_f32(),
)
}
2021-07-08 23:33:44 +03:00
fn convert_typst_paint(paint: Paint) -> sk::Paint<'static> {
2021-11-20 17:51:07 +03:00
let Paint::Solid(Color::Rgba(c)) = paint;
2021-07-08 23:33:44 +03:00
let mut paint = sk::Paint::default();
paint.set_color_rgba8(c.r, c.g, c.b, c.a);
2021-11-20 17:51:07 +03:00
paint.anti_alias = true;
2021-03-20 22:19:30 +03:00
paint
}
2021-07-08 23:33:44 +03:00
fn convert_typst_path(path: &geom::Path) -> sk::Path {
let mut builder = sk::PathBuilder::new();
2021-03-20 22:19:30 +03:00
for elem in &path.0 {
match elem {
2021-07-08 23:33:44 +03:00
PathElement::MoveTo(p) => {
2021-11-20 17:51:07 +03:00
builder.move_to(p.x.to_f32(), p.y.to_f32());
}
2021-07-08 23:33:44 +03:00
PathElement::LineTo(p) => {
2021-11-20 17:51:07 +03:00
builder.line_to(p.x.to_f32(), p.y.to_f32());
}
2021-07-08 23:33:44 +03:00
PathElement::CubicTo(p1, p2, p3) => {
2021-11-20 17:51:07 +03:00
builder.cubic_to(
p1.x.to_f32(),
p1.y.to_f32(),
p2.x.to_f32(),
p2.y.to_f32(),
p3.x.to_f32(),
p3.y.to_f32(),
);
}
2021-07-08 23:33:44 +03:00
PathElement::ClosePath => {
builder.close();
2021-03-23 17:32:36 +03:00
}
2021-03-20 22:19:30 +03:00
};
}
builder.finish().unwrap()
}
2021-07-08 23:33:44 +03:00
fn convert_usvg_transform(transform: usvg::Transform) -> sk::Transform {
let usvg::Transform { a, b, c, d, e, f } = transform;
2021-11-20 17:51:07 +03:00
sk::Transform::from_row(a as _, b as _, c as _, d as _, e as _, f as _)
}
2021-07-08 23:33:44 +03:00
fn convert_usvg_fill(fill: &usvg::Fill) -> (sk::Paint<'static>, sk::FillRule) {
let mut paint = sk::Paint::default();
2021-03-23 17:32:36 +03:00
paint.anti_alias = true;
2021-12-06 16:58:57 +03:00
if let usvg::Paint::Color(usvg::Color { red, green, blue, alpha: _ }) = fill.paint {
2021-11-20 17:51:07 +03:00
paint.set_color_rgba8(red, green, blue, fill.opacity.to_u8())
2021-03-23 17:32:36 +03:00
}
let rule = match fill.rule {
2021-07-08 23:33:44 +03:00
usvg::FillRule::NonZero => sk::FillRule::Winding,
usvg::FillRule::EvenOdd => sk::FillRule::EvenOdd,
2021-03-23 17:32:36 +03:00
};
(paint, rule)
}
2021-07-08 23:33:44 +03:00
fn convert_usvg_path(path: &usvg::PathData) -> sk::Path {
let mut builder = sk::PathBuilder::new();
2021-03-23 17:32:36 +03:00
for seg in path.iter() {
match *seg {
usvg::PathSegment::MoveTo { x, y } => {
2021-11-20 17:51:07 +03:00
builder.move_to(x as _, y as _);
}
2021-03-23 17:32:36 +03:00
usvg::PathSegment::LineTo { x, y } => {
2021-11-20 17:51:07 +03:00
builder.line_to(x as _, y as _);
2021-03-23 17:32:36 +03:00
}
usvg::PathSegment::CurveTo { x1, y1, x2, y2, x, y } => {
2021-11-20 17:51:07 +03:00
builder.cubic_to(x1 as _, y1 as _, x2 as _, y2 as _, x as _, y as _);
}
usvg::PathSegment::ClosePath => {
builder.close();
2021-03-23 17:32:36 +03:00
}
}
}
builder.finish().unwrap()
}
2021-07-08 23:33:44 +03:00
struct WrappedPathBuilder(sk::PathBuilder);
impl OutlineBuilder for WrappedPathBuilder {
fn move_to(&mut self, x: f32, y: f32) {
self.0.move_to(x, y);
}
fn line_to(&mut self, x: f32, y: f32) {
self.0.line_to(x, y);
}
fn quad_to(&mut self, x1: f32, y1: f32, x: f32, y: f32) {
self.0.quad_to(x1, y1, x, y);
}
fn curve_to(&mut self, x1: f32, y1: f32, x2: f32, y2: f32, x: f32, y: f32) {
self.0.cubic_to(x1, y1, x2, y2, x, y);
}
fn close(&mut self) {
self.0.close();
}
}
2021-11-20 17:51:07 +03:00
/// Additional methods for [`Length`].
trait LengthExt {
/// Convert an em length to a number of points.
fn to_f32(self) -> f32;
}
impl LengthExt for Length {
fn to_f32(self) -> f32 {
self.to_pt() as f32
}
}
/// Disable stdout and stderr during execution of `f`.
2022-01-01 14:12:50 +03:00
#[cfg(feature = "layout-cache")]
fn silenced<F, T>(f: F) -> T
where
F: FnOnce() -> T,
{
let path = if cfg!(windows) { "NUL" } else { "/dev/null" };
let null = File::create(path).unwrap();
let stderr = FileDescriptor::redirect_stdio(&null, Stderr).unwrap();
let stdout = FileDescriptor::redirect_stdio(&null, Stdout).unwrap();
let result = f();
FileDescriptor::redirect_stdio(&stderr, Stderr).unwrap();
FileDescriptor::redirect_stdio(&stdout, Stdout).unwrap();
result
}