mirror of
https://github.com/storytold/vectorcraft.git
synced 2026-10-25 18:27:02 -06:00
Guillem Hermida
d05ccffebc
Symbol instances are placed through a transform as images are, so the Transform effect now moves and copies both (transform_placed), sharing one copy budget (MAX_COPY_OBJECTS). Copies of either stop before one reaches past MAX_REACH, as a path's do (distort::within_reach). Adds end-to-end export tests for a group's image copies in svg and pdf.
884 lines
40 KiB
Rust
884 lines
40 KiB
Rust
use std::sync::Arc;
|
||
|
||
use vectorcraft_color::{BlendMode, Color, Gradient, GradientGeom, GradientKind, GradientPaint, Paint};
|
||
use vectorcraft_doc::{
|
||
Appearance, AppearanceItem, Artboard, CharStyle, Dash, Document, ImageBlob, ImageObject, LineCap, LineJoin, Node, NodeId, NodeKind, TextObject,
|
||
};
|
||
use vectorcraft_geom::{Affine, FillRule, PathData, Point, Rect, shapes};
|
||
|
||
use crate::*;
|
||
|
||
fn doc(w: f64, h: f64) -> Document {
|
||
Document::new(w, h)
|
||
}
|
||
|
||
fn add(d: &mut Document, mut n: Node) -> NodeId {
|
||
n.id = d.alloc_id();
|
||
let layer = d.default_layer().expect("layer");
|
||
let len = d.children(Some(layer)).map_or(0, |c| c.len());
|
||
d.insert(Some(layer), len, n).expect("insert")
|
||
}
|
||
|
||
fn rect_node(r: Rect, fill: Color) -> Node {
|
||
Node::path(NodeId(0), shapes::rectangle(r), Appearance::basic(Paint::solid(fill), Paint::None, 0.0))
|
||
}
|
||
|
||
fn leaves(d: &Document) -> Vec<&Node> {
|
||
let mut out = vec![];
|
||
d.walk(|n| {
|
||
if !n.is_container() && !matches!(n.kind, NodeKind::Path { clipping: true, .. }) {
|
||
out.push(n);
|
||
}
|
||
});
|
||
out
|
||
}
|
||
|
||
fn close(a: Rect, b: Rect, tol: f64) -> bool {
|
||
(a.x0 - b.x0).abs() < tol && (a.y0 - b.y0).abs() < tol && (a.x1 - b.x1).abs() < tol && (a.y1 - b.y1).abs() < tol
|
||
}
|
||
|
||
fn roundtrip(d: &Document) -> Document {
|
||
let bytes = export(d, &PdfOptions::default()).expect("export");
|
||
import(&bytes).expect("import")
|
||
}
|
||
|
||
fn uncompressed(d: &Document) -> String {
|
||
let bytes = export(d, &PdfOptions::uncompressed()).expect("export");
|
||
String::from_utf8_lossy(&bytes).into_owned()
|
||
}
|
||
|
||
fn page_count(bytes: &[u8]) -> usize {
|
||
craft_pdf::syntax::Pdf::new(bytes.to_vec()).expect("parse").pages().len()
|
||
}
|
||
|
||
#[test]
|
||
fn export_is_valid_pdf() {
|
||
let mut d = doc(200.0, 100.0);
|
||
add(&mut d, rect_node(Rect::new(10.0, 10.0, 50.0, 50.0), Color::rgb(1.0, 0.0, 0.0)));
|
||
let bytes = export(&d, &PdfOptions::default()).unwrap();
|
||
assert!(bytes.starts_with(b"%PDF-1.7"));
|
||
let tail = String::from_utf8_lossy(&bytes[bytes.len().saturating_sub(64)..]).into_owned();
|
||
assert!(tail.contains("startxref") && tail.contains("%%EOF"), "{tail}");
|
||
assert!(String::from_utf8_lossy(&bytes).contains("xref"));
|
||
assert_eq!(page_count(&bytes), 1);
|
||
}
|
||
|
||
#[test]
|
||
fn metadata_creator_and_title() {
|
||
let mut d = doc(100.0, 100.0);
|
||
d.title = "Poster".into();
|
||
let s = uncompressed(&d);
|
||
assert!(s.contains("VectorCraft"));
|
||
assert!(s.contains("Poster"));
|
||
}
|
||
|
||
#[test]
|
||
fn one_page_per_artboard_with_sizes() {
|
||
let mut d = doc(200.0, 100.0);
|
||
d.artboards.push(Artboard {
|
||
id: 2,
|
||
name: "B".into(),
|
||
rect: Rect::new(300.0, 0.0, 400.0, 400.0),
|
||
show_center_mark: false,
|
||
show_cross_hairs: false,
|
||
});
|
||
d.artboards.push(Artboard {
|
||
id: 3,
|
||
name: "C".into(),
|
||
rect: Rect::new(0.0, 500.0, 612.0, 1292.0),
|
||
show_center_mark: false,
|
||
show_cross_hairs: false,
|
||
});
|
||
let bytes = export(&d, &PdfOptions::default()).unwrap();
|
||
let pdf = craft_pdf::syntax::Pdf::new(bytes).unwrap();
|
||
let sizes: Vec<(f32, f32)> = pdf.pages().iter().map(|p| p.render_dimensions()).collect();
|
||
assert_eq!(sizes, vec![(200.0, 100.0), (100.0, 400.0), (612.0, 792.0)]);
|
||
}
|
||
|
||
#[test]
|
||
fn artboard_range_and_bad_index() {
|
||
let mut d = doc(200.0, 100.0);
|
||
d.artboards.push(Artboard {
|
||
id: 2,
|
||
name: "B".into(),
|
||
rect: Rect::new(300.0, 0.0, 400.0, 400.0),
|
||
show_center_mark: false,
|
||
show_cross_hairs: false,
|
||
});
|
||
let bytes = export(&d, &PdfOptions { artboards: Some(vec![1]), ..Default::default() }).unwrap();
|
||
let pdf = craft_pdf::syntax::Pdf::new(bytes).unwrap();
|
||
assert_eq!(pdf.pages().len(), 1);
|
||
assert_eq!(pdf.pages()[0].render_dimensions(), (100.0, 400.0));
|
||
assert_eq!(export(&d, &PdfOptions { artboards: Some(vec![5]), ..Default::default() }), Err(PdfError::BadArtboard(5)));
|
||
d.artboards.clear();
|
||
assert_eq!(export(&d, &PdfOptions::default()), Err(PdfError::NoArtboards));
|
||
}
|
||
|
||
#[test]
|
||
fn roundtrip_rect_and_ellipse_bounds() {
|
||
let mut d = doc(300.0, 200.0);
|
||
add(&mut d, rect_node(Rect::new(10.0, 20.0, 110.0, 70.0), Color::rgb(1.0, 0.0, 0.0)));
|
||
let e = Node::path(
|
||
NodeId(0),
|
||
shapes::ellipse(Rect::new(150.0, 40.0, 250.0, 180.0)),
|
||
Appearance::basic(Paint::solid(Color::rgb(0.0, 0.0, 1.0)), Paint::None, 0.0),
|
||
);
|
||
add(&mut d, e);
|
||
let out = roundtrip(&d);
|
||
let l = leaves(&out);
|
||
assert_eq!(l.len(), 2);
|
||
assert!(close(l[0].geometric_bounds().unwrap(), Rect::new(10.0, 20.0, 110.0, 70.0), 0.01), "{:?}", l[0].geometric_bounds());
|
||
assert!(close(l[1].geometric_bounds().unwrap(), Rect::new(150.0, 40.0, 250.0, 180.0), 0.05), "{:?}", l[1].geometric_bounds());
|
||
assert_eq!(out.artboards.len(), 1);
|
||
assert!(close(out.artboards[0].rect, Rect::new(0.0, 0.0, 300.0, 200.0), 1e-6));
|
||
}
|
||
|
||
#[test]
|
||
fn an_instance_whose_art_reaches_the_page_is_written() {
|
||
// The symbol's red rectangle, (100, 50) to (140, 90), drawn from (-50, -50): on the page, while
|
||
// the 20 × 20 square around the origin, moved to (-60, -60) to (-40, -40), is off it.
|
||
let mut d = doc(200.0, 100.0);
|
||
let red = rect_node(Rect::new(100.0, 50.0, 140.0, 90.0), Color::rgb(1.0, 0.0, 0.0));
|
||
let far = vectorcraft_doc::Symbol { name: "Far".into(), art: Arc::new(red) };
|
||
d.symbols.push(far.clone());
|
||
let instance = Node::new(NodeId(0), NodeKind::SymbolInstance { symbol: "Far".into(), xf: Affine::translate((-50.0, -50.0)) });
|
||
add(&mut d, instance.clone());
|
||
let out = roundtrip(&d);
|
||
let l = leaves(&out);
|
||
assert_eq!(l.len(), 1);
|
||
assert!(close(l[0].geometric_bounds().unwrap(), Rect::new(50.0, 0.0, 90.0, 40.0), 0.01), "{:?}", l[0].geometric_bounds());
|
||
// An element of a knockout group, beside a blue rectangle.
|
||
let mut d = doc(200.0, 100.0);
|
||
d.symbols.push(far);
|
||
let blue = rect_node(Rect::new(150.0, 10.0, 170.0, 30.0), Color::rgb(0.0, 0.0, 1.0));
|
||
let mut group = Node::group(NodeId(0), vec![Arc::new(instance), Arc::new(blue)]);
|
||
group.knockout = vectorcraft_doc::Knockout::On;
|
||
add(&mut d, group);
|
||
let fills: Vec<[u8; 4]> = leaves(&roundtrip(&d)).iter().filter_map(|n| Some(n.appearance.fill_paint().color()?.to_rgba8(1.0))).collect();
|
||
assert!(fills.contains(&[255, 0, 0, 255]) && fills.contains(&[0, 0, 255, 255]), "{fills:?}");
|
||
}
|
||
|
||
#[test]
|
||
fn an_instance_whose_brush_art_reaches_the_page_is_written() {
|
||
// The symbol's art: a line from (0, 40) to (0, 60) with a 1 pt stroke painted by a 20 pt round
|
||
// nib. Drawn from (-5, 0), the line lies left of the page, and the nib paints 10 pt either side
|
||
// of it, onto the page.
|
||
let mut d = doc(200.0, 100.0);
|
||
let mut lib = vectorcraft_brush::library(&d);
|
||
let Some(vectorcraft_brush::BrushKind::Calligraphic(c)) = lib.iter_mut().find(|b| b.name == "3 pt. Round").map(|b| &mut b.kind) else {
|
||
panic!("the 3 pt. Round brush")
|
||
};
|
||
c.size = 20.0;
|
||
vectorcraft_brush::store(&mut d, &lib);
|
||
let mut stroke = Appearance::basic(Paint::None, Paint::solid(Color::BLACK), 1.0);
|
||
stroke.stroke_mut().unwrap().brush = Some("3 pt. Round".into());
|
||
let line = Node::path(NodeId(0), shapes::line((0.0, 40.0).into(), (0.0, 60.0).into()), stroke);
|
||
d.symbols.push(vectorcraft_doc::Symbol { name: "Nib".into(), art: Arc::new(line) });
|
||
add(&mut d, Node::new(NodeId(0), NodeKind::SymbolInstance { symbol: "Nib".into(), xf: Affine::translate((-5.0, 0.0)) }));
|
||
let out = roundtrip(&d);
|
||
let l = leaves(&out);
|
||
assert_eq!(l.len(), 1, "the nib's art");
|
||
assert!(close(l[0].geometric_bounds().unwrap(), Rect::new(-15.0, 30.0, 5.0, 70.0), 0.01), "{:?}", l[0].geometric_bounds());
|
||
}
|
||
|
||
#[test]
|
||
fn roundtrip_colours_preserved() {
|
||
let mut d = doc(300.0, 100.0);
|
||
let cols = [Color::rgb8(0x12, 0x34, 0x56), Color::rgb8(0xff, 0x80, 0x00), Color::rgb8(0x00, 0xc0, 0x40)];
|
||
for (i, c) in cols.iter().enumerate() {
|
||
add(&mut d, rect_node(Rect::new(10.0 + 90.0 * i as f64, 10.0, 90.0 + 90.0 * i as f64, 90.0), *c));
|
||
}
|
||
let out = roundtrip(&d);
|
||
let got: Vec<String> = leaves(&out).iter().map(|n| n.appearance.fill_paint().color().unwrap().to_hex()).collect();
|
||
assert_eq!(got, cols.iter().map(|c| c.to_hex()).collect::<Vec<_>>());
|
||
}
|
||
|
||
#[test]
|
||
fn roundtrip_stroke_params_merge_fill_and_stroke() {
|
||
let mut d = doc(200.0, 200.0);
|
||
let mut p = Node::path(
|
||
NodeId(0),
|
||
shapes::rectangle(Rect::new(20.0, 20.0, 120.0, 120.0)),
|
||
Appearance::basic(Paint::solid(Color::WHITE), Paint::solid(Color::BLACK), 4.0),
|
||
);
|
||
{
|
||
let st = p.appearance.stroke_mut().unwrap();
|
||
st.cap = LineCap::Round;
|
||
st.join = LineJoin::Bevel;
|
||
st.miter_limit = 7.0;
|
||
st.dash = Some(Dash { pattern: vec![6.0, 3.0], offset: 1.0, align_corners: false });
|
||
}
|
||
add(&mut d, p);
|
||
let out = roundtrip(&d);
|
||
let l = leaves(&out);
|
||
assert_eq!(l.len(), 1, "fill + stroke of one path should import as one object");
|
||
let st = l[0].appearance.stroke().unwrap();
|
||
assert!((st.width - 4.0).abs() < 1e-3);
|
||
assert_eq!(st.cap, LineCap::Round);
|
||
assert_eq!(st.join, LineJoin::Bevel);
|
||
assert!((st.miter_limit - 7.0).abs() < 1e-3);
|
||
let dash = st.dash.as_ref().unwrap();
|
||
assert_eq!(dash.pattern.len(), 2);
|
||
assert!((dash.pattern[0] - 6.0).abs() < 1e-3 && (dash.offset - 1.0).abs() < 1e-3);
|
||
assert_eq!(l[0].appearance.fill_paint().color().unwrap().to_hex(), "#ffffff");
|
||
}
|
||
|
||
#[test]
|
||
fn roundtrip_open_path_and_even_odd() {
|
||
let mut d = doc(200.0, 200.0);
|
||
let mut bp = vectorcraft_geom::BezPath::new();
|
||
bp.move_to((10.0, 10.0));
|
||
bp.curve_to((50.0, 0.0), (80.0, 100.0), (150.0, 60.0));
|
||
add(&mut d, Node::path(NodeId(0), PathData::from_bezpath(&bp), Appearance::basic(Paint::None, Paint::solid(Color::BLACK), 2.0)));
|
||
let mut ring = shapes::rectangle(Rect::new(20.0, 100.0, 120.0, 190.0));
|
||
ring.subpaths.extend(shapes::rectangle(Rect::new(40.0, 120.0, 100.0, 170.0)).subpaths);
|
||
let mut n = rect_node(Rect::ZERO, Color::BLACK);
|
||
n.kind = NodeKind::Path { path: ring, rule: FillRule::EvenOdd, live: None, clipping: false, guide: false };
|
||
add(&mut d, n);
|
||
let out = roundtrip(&d);
|
||
let l = leaves(&out);
|
||
assert_eq!(l.len(), 2);
|
||
let pb = l[0].path_data().unwrap();
|
||
assert!(!pb.is_closed());
|
||
let b = l[0].geometric_bounds().unwrap();
|
||
let want = bp.bounding_box_exact();
|
||
assert!(close(b, want, 0.05), "{b:?} vs {want:?}");
|
||
assert!(matches!(l[1].kind, NodeKind::Path { rule: FillRule::EvenOdd, .. }));
|
||
assert_eq!(l[1].path_data().unwrap().subpaths.len(), 2);
|
||
}
|
||
|
||
trait Exact {
|
||
fn bounding_box_exact(&self) -> Rect;
|
||
}
|
||
impl Exact for vectorcraft_geom::BezPath {
|
||
fn bounding_box_exact(&self) -> Rect {
|
||
kurbo::Shape::bounding_box(self)
|
||
}
|
||
}
|
||
|
||
#[test]
|
||
fn gradients_exported_as_shadings_and_reimported() {
|
||
let mut d = doc(300.0, 100.0);
|
||
let lin = GradientPaint::new(Gradient::default());
|
||
let mut n = rect_node(Rect::new(0.0, 0.0, 140.0, 100.0), Color::BLACK);
|
||
n.appearance.set_fill(Paint::Gradient(Box::new(lin)));
|
||
add(&mut d, n);
|
||
let rad = GradientPaint::new(Gradient { kind: GradientKind::Radial, ..Gradient::default() });
|
||
let mut n = rect_node(Rect::new(150.0, 0.0, 250.0, 100.0), Color::BLACK);
|
||
n.appearance.set_fill(Paint::Gradient(Box::new(rad)));
|
||
add(&mut d, n);
|
||
let s = uncompressed(&d);
|
||
let s = s.replace(' ', "");
|
||
assert!(s.contains("/ShadingType2"), "axial shading");
|
||
assert!(s.contains("/ShadingType3"), "radial shading");
|
||
let out = roundtrip(&d);
|
||
let l = leaves(&out);
|
||
assert_eq!(l.len(), 2);
|
||
let Paint::Gradient(g) = l[0].appearance.fill_paint() else { panic!("linear gradient lost: {:?}", l[0].appearance) };
|
||
assert_eq!(g.gradient.kind, GradientKind::Linear);
|
||
let geom = g.geom.unwrap();
|
||
assert!((geom.start.x - 0.0).abs() < 0.1 && (geom.end.x - 140.0).abs() < 0.1, "{geom:?}");
|
||
assert_eq!(g.gradient.stops.first().unwrap().color.to_hex(), "#ffffff");
|
||
assert_eq!(g.gradient.stops.last().unwrap().color.to_hex(), "#000000");
|
||
let Paint::Gradient(g) = l[1].appearance.fill_paint() else { panic!("radial gradient lost") };
|
||
assert_eq!(g.gradient.kind, GradientKind::Radial);
|
||
assert!((g.geom.unwrap().start - Point::new(200.0, 50.0)).hypot() < 0.1);
|
||
}
|
||
|
||
#[test]
|
||
fn smooth_revolve_exports_axial_vector_shading_without_images_and_keeps_the_source_live() {
|
||
let mut d = doc(200.0, 200.0);
|
||
let bp = vectorcraft_geom::BezPath::from_svg("M100 30 C140 50 140 120 100 160").unwrap();
|
||
let mut n = Node::path(NodeId(0), PathData::from_bezpath(&bp), Appearance::basic(Paint::solid(Color::rgb(0.2, 0.55, 0.9)), Paint::None, 0.0));
|
||
n.appearance.effects.push(vectorcraft_doc::Effect {
|
||
id: "threeD.revolve".into(),
|
||
visible: true,
|
||
params: serde_json::json!({"offset":20,"rotationX":-25,"cap":true,"segments":16}),
|
||
});
|
||
add(&mut d, n);
|
||
let before = d.clone();
|
||
let pdf = uncompressed(&d).replace(' ', "");
|
||
assert!(pdf.matches("/ShadingType2").count() > 50);
|
||
assert!(!pdf.contains("/Subtype/Image"));
|
||
assert_eq!(d, before);
|
||
let back = roundtrip(&d);
|
||
let strokes: Vec<_> = leaves(&back)
|
||
.into_iter()
|
||
.flat_map(|n| n.appearance.items.iter())
|
||
.filter_map(|i| if let AppearanceItem::Stroke(s) = i { Some(s) } else { None })
|
||
.collect();
|
||
assert!(strokes.len() >= 2, "continuous cap rims remain vector strokes");
|
||
assert!(strokes.iter().all(|s| s.join == LineJoin::Round), "seam strokes keep round joins in PDF");
|
||
}
|
||
|
||
#[test]
|
||
fn hidden_objects_and_template_layers_skipped() {
|
||
let mut d = doc(200.0, 200.0);
|
||
add(&mut d, rect_node(Rect::new(10.0, 10.0, 50.0, 50.0), Color::BLACK));
|
||
let mut hidden = rect_node(Rect::new(60.0, 10.0, 90.0, 50.0), Color::BLACK);
|
||
hidden.visible = false;
|
||
add(&mut d, hidden);
|
||
let t = d.add_layer(Some("Template"));
|
||
if let NodeKind::Layer { template, .. } = &mut d.node_mut(t).unwrap().kind {
|
||
*template = true;
|
||
}
|
||
let mut n = rect_node(Rect::new(100.0, 100.0, 150.0, 150.0), Color::BLACK);
|
||
n.id = d.alloc_id();
|
||
d.insert(Some(t), 0, n).unwrap();
|
||
let mut guide = rect_node(Rect::new(0.0, 0.0, 10.0, 10.0), Color::BLACK);
|
||
if let NodeKind::Path { guide: g, .. } = &mut guide.kind {
|
||
*g = true;
|
||
}
|
||
add(&mut d, guide);
|
||
let out = roundtrip(&d);
|
||
assert_eq!(leaves(&out).len(), 1);
|
||
}
|
||
|
||
#[test]
|
||
fn objects_outside_artboard_are_culled() {
|
||
let mut d = doc(100.0, 100.0);
|
||
add(&mut d, rect_node(Rect::new(10.0, 10.0, 50.0, 50.0), Color::BLACK));
|
||
add(&mut d, rect_node(Rect::new(500.0, 500.0, 550.0, 550.0), Color::BLACK));
|
||
assert_eq!(leaves(&roundtrip(&d)).len(), 1);
|
||
}
|
||
|
||
/// A line from (10, `y`) to (90, `y`).
|
||
fn line(y: f64) -> PathData {
|
||
let mut bp = vectorcraft_geom::BezPath::new();
|
||
bp.move_to((10.0, y));
|
||
bp.line_to((90.0, y));
|
||
PathData::from_bezpath(&bp)
|
||
}
|
||
|
||
/// Art that the PDF would hold a value beyond a billion points for is left out: a path that far,
|
||
/// a gradient placed that far, a stroke so wide that its group's box reaches that far, and an
|
||
/// image placed that far. Each would be written with -2,147,483,648, which the PDF writer panics
|
||
/// on in builds with overflow checks.
|
||
#[test]
|
||
fn art_reaching_beyond_a_billion_points_is_left_out() {
|
||
let far = -2_147_483_649.0;
|
||
let shaded = |kind, start, end| {
|
||
let mut n = rect_node(Rect::new(60.0, 10.0, 90.0, 50.0), Color::BLACK);
|
||
let geom = GradientGeom { start, end, aspect: 1.0, focal: None };
|
||
let g = GradientPaint { geom: Some(geom), ..GradientPaint::new(Gradient { kind, ..Gradient::default() }) };
|
||
n.appearance.set_fill(Paint::Gradient(Box::new(g)));
|
||
n
|
||
};
|
||
let mut wide = Node::path(NodeId(0), line(0.0), Appearance::basic(Paint::None, Paint::solid(Color::BLACK), 4_294_967_296.0));
|
||
wide.opacity = 0.5;
|
||
let image = ImageObject {
|
||
key: "img1".into(),
|
||
width: 4,
|
||
height: 2,
|
||
xf: Affine::new([5.5e8, 0.0, 0.0, 10.0, far, 60.0]),
|
||
link: None,
|
||
placement: Default::default(),
|
||
};
|
||
for (what, n) in [
|
||
("path", rect_node(Rect::new(far, 60.0, 90.0, 90.0), Color::BLACK)),
|
||
("linear gradient", shaded(GradientKind::Linear, Point::new(far, 0.0), Point::new(90.0, 0.0))),
|
||
("radial gradient", shaded(GradientKind::Radial, Point::new(far, 30.0), Point::new(far + 100.0, 30.0))),
|
||
("stroke", wide),
|
||
("image", Node::new(NodeId(0), NodeKind::Image(Box::new(image)))),
|
||
] {
|
||
let mut d = doc(100.0, 100.0);
|
||
d.images.insert("img1".into(), ImageBlob::new("image/png", tiny_png()));
|
||
add(&mut d, rect_node(Rect::new(10.0, 10.0, 50.0, 50.0), Color::BLACK));
|
||
add(&mut d, n);
|
||
assert_eq!(leaves(&roundtrip(&d)).len(), 1, "{what}");
|
||
}
|
||
}
|
||
|
||
/// A dash offset beyond a billion points is written as its remainder after whole periods of the
|
||
/// pattern: -2,147,483,649 (-2,147,483,648 as an f32, which the PDF writer panics on in builds
|
||
/// with overflow checks) is 1 in a pattern of 5 and 5.
|
||
#[test]
|
||
fn dash_offsets_beyond_a_billion_points_wrap_into_their_pattern() {
|
||
let mut d = doc(100.0, 100.0);
|
||
let mut n = Node::path(NodeId(0), line(50.0), Appearance::basic(Paint::None, Paint::solid(Color::BLACK), 2.0));
|
||
n.appearance.stroke_mut().unwrap().dash = Some(Dash { pattern: vec![5.0, 5.0], offset: -2_147_483_649.0, align_corners: false });
|
||
add(&mut d, n);
|
||
assert!(uncompressed(&d).contains("[5 5]1 d"));
|
||
}
|
||
|
||
/// The page of an artboard beyond a billion points is left blank: what is on it would be placed
|
||
/// through -2,147,483,648, which the PDF writer panics on in builds with overflow checks.
|
||
#[test]
|
||
fn artboards_beyond_a_billion_points_export_blank_pages() {
|
||
let mut d = doc(100.0, 100.0);
|
||
d.artboards[0].rect = Rect::new(2_147_483_649.0, 0.0, 2_147_483_749.0, 100.0);
|
||
let mut n = rect_node(Rect::new(2_147_483_659.0, 10.0, 2_147_483_699.0, 50.0), Color::BLACK);
|
||
n.opacity = 0.5;
|
||
add(&mut d, n);
|
||
let bytes = export(&d, &PdfOptions::default()).unwrap();
|
||
assert_eq!(page_count(&bytes), 1);
|
||
assert!(leaves(&import(&bytes).unwrap()).is_empty());
|
||
}
|
||
|
||
#[test]
|
||
fn opacity_and_blend_modes() {
|
||
let mut d = doc(200.0, 200.0);
|
||
let mut n = rect_node(Rect::new(10.0, 10.0, 100.0, 100.0), Color::rgb(1.0, 0.0, 0.0));
|
||
n.opacity = 0.5;
|
||
n.blend = BlendMode::Multiply;
|
||
add(&mut d, n);
|
||
let s = uncompressed(&d);
|
||
let s = s.replace(' ', "");
|
||
assert!(s.contains("/BM/Multiply"), "blend mode in ExtGState");
|
||
assert!(s.contains("/ca0.5"), "opacity in ExtGState");
|
||
assert!(s.contains("/Transparency"), "transparency group");
|
||
let out = roundtrip(&d);
|
||
let l = leaves(&out);
|
||
assert_eq!(l.len(), 1);
|
||
// Opacity/blend end up on the object or an enclosing group.
|
||
let mut op = 1.0;
|
||
let mut bm = BlendMode::Normal;
|
||
out.walk(|n| {
|
||
op *= n.opacity * n.appearance.fill().map_or(1.0, |f| f.opacity);
|
||
if n.blend != BlendMode::Normal {
|
||
bm = n.blend;
|
||
}
|
||
});
|
||
assert!((op - 0.5).abs() < 0.01, "{op}");
|
||
assert_eq!(bm, BlendMode::Multiply);
|
||
}
|
||
|
||
#[test]
|
||
fn clip_group_roundtrip() {
|
||
let mut d = doc(200.0, 200.0);
|
||
let clip = Node::new(
|
||
NodeId(0),
|
||
NodeKind::Path {
|
||
path: shapes::ellipse(Rect::new(20.0, 20.0, 120.0, 120.0)),
|
||
rule: FillRule::NonZero,
|
||
live: None,
|
||
clipping: true,
|
||
guide: false,
|
||
},
|
||
);
|
||
let mut clip = clip;
|
||
clip.id = d.alloc_id();
|
||
let mut a = rect_node(Rect::new(0.0, 0.0, 100.0, 100.0), Color::rgb(0.0, 1.0, 0.0));
|
||
a.id = d.alloc_id();
|
||
let mut b = rect_node(Rect::new(50.0, 50.0, 150.0, 150.0), Color::rgb(0.0, 0.0, 1.0));
|
||
b.id = d.alloc_id();
|
||
let g = Node::new(NodeId(0), NodeKind::Group { children: vec![Arc::new(clip), Arc::new(a), Arc::new(b)], clip: true });
|
||
add(&mut d, g);
|
||
let out = roundtrip(&d);
|
||
let mut groups = vec![];
|
||
out.walk(|n| {
|
||
if let NodeKind::Group { clip: true, children } = &n.kind {
|
||
groups.push(children.len());
|
||
}
|
||
});
|
||
assert_eq!(groups, vec![3]);
|
||
let cg = {
|
||
let mut r = None;
|
||
out.walk(|n| {
|
||
if matches!(n.kind, NodeKind::Group { clip: true, .. }) {
|
||
r = n.geometric_bounds();
|
||
}
|
||
});
|
||
r.unwrap()
|
||
};
|
||
assert!(close(cg, Rect::new(20.0, 20.0, 120.0, 120.0), 0.05), "{cg:?}");
|
||
}
|
||
|
||
#[test]
|
||
fn text_exported_as_outlines() {
|
||
let mut d = doc(300.0, 100.0);
|
||
let t = TextObject::point(Point::new(10.0, 50.0), "Hello", CharStyle { size: 24.0, ..CharStyle::default() });
|
||
add(&mut d, Node::new(NodeId(0), NodeKind::Text(Box::new(t))));
|
||
let mut outlined = PdfOptions::uncompressed();
|
||
outlined.settings.advanced.outline_text = true;
|
||
let s = String::from_utf8_lossy(&export(&d, &outlined).unwrap()).into_owned();
|
||
assert!(!s.contains("/Font"), "no fonts when outlining text");
|
||
// Real text in an embedded font by default, as a PDF from Illustrator has it (#733).
|
||
assert!(uncompressed(&d).contains("/FontFile"), "a font embedded by default");
|
||
let r = import_with_report(&export(&d, &outlined).unwrap(), &ImportOptions::default()).unwrap();
|
||
let l = leaves(&r.document);
|
||
assert!(!l.is_empty());
|
||
assert!(l.iter().all(|n| matches!(n.kind, NodeKind::Path { .. })));
|
||
let b = l.iter().fold(None, |acc, n| vectorcraft_geom::union_opt(acc, n.geometric_bounds())).unwrap();
|
||
assert!(b.x0 >= 9.0 && b.x1 < 120.0 && b.y1 <= 56.0 && b.y0 > 25.0, "{b:?}");
|
||
}
|
||
|
||
fn tiny_png() -> Vec<u8> {
|
||
let mut img = image::RgbaImage::new(4, 2);
|
||
for (x, _, p) in img.enumerate_pixels_mut() {
|
||
*p = image::Rgba([if x < 2 { 255 } else { 0 }, 0, 0, 255]);
|
||
}
|
||
let mut out = vec![];
|
||
img.write_to(&mut std::io::Cursor::new(&mut out), image::ImageFormat::Png).unwrap();
|
||
out
|
||
}
|
||
|
||
#[test]
|
||
fn image_roundtrip() {
|
||
let mut d = doc(200.0, 200.0);
|
||
d.images.insert("img1".into(), ImageBlob::new("image/png", tiny_png()));
|
||
let im = ImageObject {
|
||
key: "img1".into(),
|
||
width: 4,
|
||
height: 2,
|
||
xf: Affine::translate((20.0, 30.0)) * Affine::scale(10.0),
|
||
link: None,
|
||
placement: Default::default(),
|
||
};
|
||
add(&mut d, Node::new(NodeId(0), NodeKind::Image(Box::new(im))));
|
||
let out = roundtrip(&d);
|
||
let l = leaves(&out);
|
||
assert_eq!(l.len(), 1);
|
||
let NodeKind::Image(im) = &l[0].kind else { panic!("expected image, got {:?}", l[0].kind_label()) };
|
||
assert_eq!((im.width, im.height), (4, 2));
|
||
assert!(close(l[0].geometric_bounds().unwrap(), Rect::new(20.0, 30.0, 60.0, 50.0), 0.01), "{:?}", l[0].geometric_bounds());
|
||
let blob = &out.images[&im.key];
|
||
let decoded = image::load_from_memory(&blob.bytes).unwrap().to_rgba8();
|
||
assert_eq!(decoded.get_pixel(0, 0).0, [255, 0, 0, 255]);
|
||
assert_eq!(decoded.get_pixel(3, 0).0, [0, 0, 0, 255]);
|
||
}
|
||
|
||
/// #2023: a group's Transform effect copies an image among its members as it does the paths: the
|
||
/// PDF draws the image three times, where the effect put it.
|
||
#[test]
|
||
fn a_group_transform_effect_exports_its_image_copies() {
|
||
let mut d = doc(200.0, 100.0);
|
||
d.images.insert("img1".into(), ImageBlob::new("image/png", tiny_png()));
|
||
let im = ImageObject {
|
||
key: "img1".into(),
|
||
width: 4,
|
||
height: 2,
|
||
xf: Affine::translate((20.0, 30.0)) * Affine::scale(5.0),
|
||
link: None,
|
||
placement: Default::default(),
|
||
};
|
||
let image = Arc::new(Node::new(d.alloc_id(), NodeKind::Image(Box::new(im))));
|
||
let mut square = rect_node(Rect::new(20.0, 60.0, 30.0, 70.0), Color::BLACK);
|
||
square.id = d.alloc_id();
|
||
let mut g = Node::group(NodeId(0), vec![Arc::new(square), image]);
|
||
g.appearance.effects.push(vectorcraft_doc::Effect {
|
||
id: "distort.transform".into(),
|
||
params: serde_json::json!({"moveH": 50, "copies": 2}),
|
||
visible: true,
|
||
});
|
||
add(&mut d, g);
|
||
let out = roundtrip(&d);
|
||
let lefts: Vec<f64> =
|
||
leaves(&out).iter().filter(|n| matches!(n.kind, NodeKind::Image(_))).filter_map(|n| n.geometric_bounds()).map(|b| b.x0.round()).collect();
|
||
assert_eq!(lefts, [20.0, 70.0, 120.0]);
|
||
}
|
||
|
||
/// Build a PDF with a correct xref table from object bodies.
|
||
fn handmade_pdf(content: &str, media: &str) -> Vec<u8> {
|
||
let objs = [
|
||
"<< /Type /Catalog /Pages 2 0 R >>".to_string(),
|
||
"<< /Type /Pages /Kids [3 0 R] /Count 1 >>".to_string(),
|
||
format!("<< /Type /Page /Parent 2 0 R /MediaBox {media} /Contents 4 0 R /Resources << >> >>"),
|
||
format!("<< /Length {} >>\nstream\n{content}\nendstream", content.len() + 1),
|
||
];
|
||
let mut out = b"%PDF-1.4\n".to_vec();
|
||
let mut offs = vec![];
|
||
for (i, o) in objs.iter().enumerate() {
|
||
offs.push(out.len());
|
||
out.extend(format!("{} 0 obj\n{o}\nendobj\n", i + 1).bytes());
|
||
}
|
||
let xref = out.len();
|
||
out.extend(format!("xref\n0 {}\n0000000000 65535 f \n", objs.len() + 1).bytes());
|
||
for o in offs {
|
||
out.extend(format!("{o:010} 00000 n \n").bytes());
|
||
}
|
||
out.extend(format!("trailer\n<< /Size {} /Root 1 0 R >>\nstartxref\n{xref}\n%%EOF\n", objs.len() + 1).bytes());
|
||
out
|
||
}
|
||
|
||
#[test]
|
||
fn import_handmade_pdf() {
|
||
let content = "1 0 0 RG 0 0 1 rg 3 w 10 10 50 30 re B \
|
||
0 1 1 0 k 100 100 20 20 re f \
|
||
0.5 g q 1 0 0 1 150 0 cm 0 0 m 40 0 l 20 30 l h f Q";
|
||
let bytes = handmade_pdf(content, "[0 0 200 300]");
|
||
let r = import_with_report(&bytes, &ImportOptions::default()).unwrap();
|
||
let d = r.document;
|
||
assert!(close(d.artboards[0].rect, Rect::new(0.0, 0.0, 200.0, 300.0), 1e-9));
|
||
let l = leaves(&d);
|
||
assert_eq!(l.len(), 3);
|
||
// y-up PDF rect (10,10)-(60,40) → y-down (10,260)-(60,290).
|
||
assert!(close(l[0].geometric_bounds().unwrap(), Rect::new(10.0, 260.0, 60.0, 290.0), 1e-6), "{:?}", l[0].geometric_bounds());
|
||
assert_eq!(l[0].appearance.fill_paint().color().unwrap().to_hex(), "#0000ff");
|
||
assert_eq!(l[0].appearance.stroke_paint().color().unwrap().to_hex(), "#ff0000");
|
||
assert!((l[0].appearance.stroke_width() - 3.0).abs() < 1e-6);
|
||
// CMYK red-ish (0,1,1,0) → approximately red through the CMYK profile.
|
||
let [r_, g_, b_] = l[1].appearance.fill_paint().color().unwrap().to_rgb();
|
||
assert!(r_ > 0.8 && g_ < 0.3 && b_ < 0.3, "{r_} {g_} {b_}");
|
||
// Gray 0.5 and the `cm` translation.
|
||
assert_eq!(l[2].appearance.fill_paint().color().unwrap().to_hex(), "#808080");
|
||
assert!(close(l[2].geometric_bounds().unwrap(), Rect::new(150.0, 270.0, 190.0, 300.0), 1e-6));
|
||
}
|
||
|
||
/// Art off the page opens on the pasteboard round its artboard, not cut away (#472).
|
||
#[test]
|
||
fn import_keeps_art_outside_the_page() {
|
||
let content = "0 0 1 rg 10 10 50 30 re f 0 1 0 rg -300 -300 50 50 re f 1 0 0 rg 500 500 20 20 re f";
|
||
let d = import(&handmade_pdf(content, "[0 0 200 300]")).unwrap();
|
||
assert!(close(d.artboards[0].rect, Rect::new(0.0, 0.0, 200.0, 300.0), 1e-9));
|
||
let bounds: Vec<Rect> = leaves(&d).iter().map(|n| n.geometric_bounds().unwrap()).collect();
|
||
let want = [Rect::new(10.0, 260.0, 60.0, 290.0), Rect::new(-300.0, 550.0, -250.0, 600.0), Rect::new(500.0, -220.0, 520.0, -200.0)];
|
||
assert_eq!(bounds.len(), want.len(), "{bounds:?}");
|
||
for (b, w) in bounds.iter().zip(want) {
|
||
assert!(close(*b, w, 1e-6), "{b:?} vs {w:?}");
|
||
}
|
||
}
|
||
|
||
#[test]
|
||
fn import_invalid_dash_array_strokes_solid() {
|
||
// A dash array with a negative value is invalid (PDF 32000-1 §8.4.3.6): the line is solid. A valid
|
||
// array next to it stays dashed.
|
||
let content = "0 0 1 RG 4 w [-5 3] 0 d 10 10 m 90 10 l S [6 2] 1 d 10 50 m 90 50 l S";
|
||
let d = import(&handmade_pdf(content, "[0 0 100 100]")).unwrap();
|
||
let l = leaves(&d);
|
||
assert_eq!(l.len(), 2);
|
||
let dash = |n: &Node| match &n.appearance.items[..] {
|
||
[.., AppearanceItem::Stroke(st)] => st.dash.clone(),
|
||
items => panic!("no stroke: {items:?}"),
|
||
};
|
||
assert_eq!(dash(l[0]), None);
|
||
assert_eq!(dash(l[1]).map(|d| d.pattern), Some(vec![6.0, 2.0]));
|
||
// And the exporter never writes an invalid array back.
|
||
let mut d2 = doc(100.0, 100.0);
|
||
let mut n = rect_node(Rect::new(10.0, 10.0, 90.0, 90.0), Color::rgb(1.0, 0.0, 0.0));
|
||
n.appearance = Appearance::basic(Paint::None, Paint::solid(Color::BLACK), 2.0).into();
|
||
if let Some(AppearanceItem::Stroke(st)) = n.appearance.items.last_mut() {
|
||
st.dash = Some(Dash { pattern: vec![-5.0, 3.0], offset: 0.0, align_corners: false });
|
||
}
|
||
add(&mut d2, n);
|
||
let s = uncompressed(&d2);
|
||
assert!(!s.contains("-5 3]") && !s.contains("[0 3]"), "invalid dash written:\n{s}");
|
||
}
|
||
|
||
#[test]
|
||
fn import_clip_from_handmade_pdf() {
|
||
let content = "q 0 0 50 50 re W n 1 0 0 rg 0 0 100 100 re f Q";
|
||
let d = import(&handmade_pdf(content, "[0 0 100 100]")).unwrap();
|
||
let mut clip_groups = 0;
|
||
d.walk(|n| {
|
||
if matches!(n.kind, NodeKind::Group { clip: true, .. }) {
|
||
clip_groups += 1;
|
||
assert!(close(n.geometric_bounds().unwrap(), Rect::new(0.0, 50.0, 50.0, 100.0), 1e-6));
|
||
}
|
||
});
|
||
assert_eq!(clip_groups, 1);
|
||
}
|
||
|
||
#[test]
|
||
fn import_max_pages_and_multi_page() {
|
||
let mut d = doc(200.0, 100.0);
|
||
d.artboards.push(Artboard {
|
||
id: 2,
|
||
name: "B".into(),
|
||
rect: Rect::new(300.0, 0.0, 400.0, 400.0),
|
||
show_center_mark: false,
|
||
show_cross_hairs: false,
|
||
});
|
||
add(&mut d, rect_node(Rect::new(10.0, 10.0, 50.0, 50.0), Color::BLACK));
|
||
add(&mut d, rect_node(Rect::new(310.0, 10.0, 350.0, 50.0), Color::BLACK));
|
||
let bytes = export(&d, &PdfOptions::default()).unwrap();
|
||
let all = import(&bytes).unwrap();
|
||
assert_eq!(all.artboards.len(), 2);
|
||
assert_eq!(all.layers.len(), 2);
|
||
assert!((all.artboards[1].rect.width() - 100.0).abs() < 1e-9 && (all.artboards[1].rect.height() - 400.0).abs() < 1e-9);
|
||
// The second page's rect lands at the same place relative to its artboard.
|
||
let second = leaves(&all)[1].geometric_bounds().unwrap();
|
||
let ab = all.artboards[1].rect;
|
||
assert!(close(second, Rect::new(ab.x0 + 10.0, 10.0, ab.x0 + 50.0, 50.0), 0.01), "{second:?}");
|
||
let one = import_with_report(&bytes, &ImportOptions { max_pages: Some(1), ..Default::default() }).unwrap();
|
||
assert_eq!(one.document.artboards.len(), 1);
|
||
assert_eq!(leaves(&one.document).len(), 1);
|
||
}
|
||
|
||
#[test]
|
||
fn import_rejects_garbage() {
|
||
assert!(matches!(import(b"not a pdf at all"), Err(PdfError::Parse(_)) | Err(PdfError::NoPages)));
|
||
}
|
||
|
||
/// Options with `settings` changed by `f`.
|
||
fn with(f: impl FnOnce(&mut PdfSettings)) -> PdfOptions {
|
||
let mut o = PdfOptions::default();
|
||
f(&mut o.settings);
|
||
o
|
||
}
|
||
|
||
#[test]
|
||
fn compatibility_levels() {
|
||
let mut d = doc(100.0, 100.0);
|
||
add(&mut d, rect_node(Rect::new(10.0, 10.0, 50.0, 50.0), Color::rgb(1.0, 0.0, 0.0)));
|
||
for (c, header) in [(Compatibility::Pdf14, "%PDF-1.4"), (Compatibility::Pdf16, "%PDF-1.6"), (Compatibility::Pdf20, "%PDF-2.0")] {
|
||
let b = export(&d, &with(|s| s.compatibility = c)).unwrap();
|
||
assert!(b.starts_with(header.as_bytes()), "{c:?}");
|
||
}
|
||
let a = export(&d, &with(|s| s.standard = Standard::PdfA2b)).unwrap();
|
||
assert!(String::from_utf8_lossy(&a).contains("pdfaid"), "PDF/A identification in XMP");
|
||
// PDF/X files are PDF 1.3 (PDF/X-1a, PDF/X-3) or at most 1.6 (PDF/X-4): not the default 1.7.
|
||
for (x, header) in [(Standard::PdfX1a, "%PDF-1.3"), (Standard::PdfX3, "%PDF-1.3"), (Standard::PdfX4, "%PDF-1.6")] {
|
||
assert!(matches!(export(&d, &with(|s| s.standard = x)), Err(PdfError::BadSetting(_))), "{x:?}");
|
||
let b = export(&d, &with(|s| (s.standard, s.compatibility) = (x, x.version()))).unwrap();
|
||
assert!(b.starts_with(header.as_bytes()), "{x:?}");
|
||
}
|
||
// PDF/A-2b is a PDF 1.7 standard.
|
||
let e = export(&d, &with(|s| (s.standard, s.compatibility) = (Standard::PdfA2b, Compatibility::Pdf20)));
|
||
assert!(matches!(e, Err(PdfError::BadSetting(_))), "{e:?}");
|
||
}
|
||
|
||
#[test]
|
||
fn compress_option_changes_output() {
|
||
let mut d = doc(100.0, 100.0);
|
||
for i in 0..20 {
|
||
add(&mut d, rect_node(Rect::new(i as f64, i as f64, 50.0 + i as f64, 50.0), Color::BLACK));
|
||
}
|
||
let c = export(&d, &PdfOptions::default()).unwrap();
|
||
let u = export(&d, &PdfOptions::uncompressed()).unwrap();
|
||
assert!(u.len() > c.len());
|
||
assert!(String::from_utf8_lossy(&u).contains(" re") || String::from_utf8_lossy(&u).contains(" l"));
|
||
}
|
||
|
||
#[test]
|
||
fn stroke_alignment_and_arrowheads_export() {
|
||
let mut d = doc(200.0, 200.0);
|
||
let mut p = Node::path(
|
||
NodeId(0),
|
||
shapes::rectangle(Rect::new(20.0, 20.0, 120.0, 120.0)),
|
||
Appearance::basic(Paint::None, Paint::solid(Color::BLACK), 6.0),
|
||
);
|
||
p.appearance.stroke_mut().unwrap().align = vectorcraft_doc::StrokeAlign::Inside;
|
||
add(&mut d, p);
|
||
let mut line = Node::path(
|
||
NodeId(0),
|
||
shapes::line(Point::new(10.0, 150.0), Point::new(150.0, 150.0)),
|
||
Appearance::basic(Paint::None, Paint::solid(Color::BLACK), 2.0),
|
||
);
|
||
line.appearance.stroke_mut().unwrap().end_arrow = Some(vectorcraft_doc::Arrowhead::Triangle);
|
||
add(&mut d, line);
|
||
let r = export_with_report(&d, &PdfOptions::default()).unwrap();
|
||
let out = import(&r.bytes).unwrap();
|
||
let mut widths = vec![];
|
||
out.walk(|n| {
|
||
if let Some(AppearanceItem::Stroke(s)) = n.appearance.items.iter().find(|i| matches!(i, AppearanceItem::Stroke(_))) {
|
||
widths.push(s.width);
|
||
}
|
||
});
|
||
assert!(widths.iter().any(|w| (w - 12.0).abs() < 1e-3), "inside stroke doubled and clipped: {widths:?}");
|
||
// The arrowhead is an extra filled object.
|
||
assert!(leaves(&out).len() >= 3);
|
||
}
|
||
|
||
#[test]
|
||
fn creation_date_written() {
|
||
let d = doc(100.0, 100.0);
|
||
// 2024-02-29 12:34:56 UTC
|
||
let bytes = export(&d, &PdfOptions { created: Some(1_709_210_096), ..PdfOptions::uncompressed() }).unwrap();
|
||
assert!(String::from_utf8_lossy(&bytes).contains("D:20240229123456"));
|
||
}
|
||
|
||
#[test]
|
||
fn opacity_mask_exports_as_luminosity_soft_mask() {
|
||
let mut d = doc(100.0, 100.0);
|
||
let mut n = rect_node(Rect::new(10.0, 10.0, 90.0, 90.0), Color::rgb(1.0, 0.0, 0.0));
|
||
let art = rect_node(Rect::new(10.0, 10.0, 50.0, 90.0), Color::WHITE);
|
||
n.extras_mut().mask = Some(Box::new(vectorcraft_doc::OpacityMask::new(art, true)));
|
||
add(&mut d, n);
|
||
let bytes = export(&d, &PdfOptions::uncompressed()).unwrap();
|
||
let text = String::from_utf8_lossy(&bytes);
|
||
assert!(text.contains("/SMask"), "soft mask in an ExtGState");
|
||
assert!(text.contains("/Luminosity"));
|
||
}
|
||
|
||
#[test]
|
||
fn empty_and_populated_file_info_preserve_key_presence() {
|
||
let mut d = doc(100.0, 100.0);
|
||
d.title.clear();
|
||
let s = uncompressed(&d);
|
||
for key in ["/Title", "/Author", "/Keywords", "<dc:title>", "<dc:creator>", "<pdf:Keywords>"] {
|
||
assert!(!s.contains(key), "unexpected empty metadata {key}");
|
||
}
|
||
d.title = "Synthetic title".into();
|
||
d.metadata.author = " Test author ".into();
|
||
d.metadata.keywords = vec!["test keyword".into()];
|
||
let bytes = export(&d, &PdfOptions { created: Some(1_700_000_000), ..PdfOptions::uncompressed() }).unwrap();
|
||
let s = String::from_utf8_lossy(&bytes);
|
||
for key in ["/Title", "/Author", "/Keywords", "/CreationDate", "<dc:title>", "<dc:creator>", "<pdf:Keywords>"] {
|
||
assert!(s.contains(key), "missing populated metadata {key}");
|
||
}
|
||
assert!(s.contains("Synthetic title") && s.contains("Test author") && s.contains("test keyword"));
|
||
}
|
||
|
||
/// A function or colour space built on itself is refused before the interpreter builds it (by
|
||
/// recursion, with no cap: a stack overflow), and those of real files are read.
|
||
#[test]
|
||
fn functions_and_colour_spaces_built_on_themselves_are_refused() {
|
||
use vectorcraft_testkit::pdf::{PdfPage, first_extra, pdf_with};
|
||
let first = first_extra(1);
|
||
let (me, next) = (format!("{first} 0 R"), format!("{} 0 R", first + 1));
|
||
let shaded = PdfPage {
|
||
resources: format!("/Shading << /S << /ShadingType 2 /ColorSpace /DeviceRGB /Coords [0 0 100 0] /Function {me} >> >>"),
|
||
..PdfPage::new(100.0, 100.0, "/S sh")
|
||
};
|
||
let filled = PdfPage { resources: format!("/ColorSpace << /C {me} >>"), ..PdfPage::new(100.0, 100.0, "/C cs 0 sc 0 0 50 50 re f") };
|
||
let stitch = |on: &str| format!("<< /FunctionType 3 /Domain [0 1] /Functions [{on}] /Bounds [] /Encode [0 1] >>");
|
||
let indexed = |on: &str| format!("[/Indexed {on} 1 <000000ffffff>]");
|
||
let ramp = "<< /FunctionType 2 /Domain [0 1] /C0 [1 0 0] /C1 [0 0 1] /N 1 >>";
|
||
// `n` functions, each stitching the next, then a plain one.
|
||
let chain = |n: usize| (1..=n).map(|i| stitch(&format!("{} 0 R", first + i))).chain([ramp.to_string()]).collect::<Vec<_>>();
|
||
let calculator =
|
||
|program: &str| format!("<< /FunctionType 4 /Domain [0 1] /Range [0 1 0 1 0 1] /Length {} >>\nstream\n{program}\nendstream", program.len());
|
||
for (what, page, objects) in [
|
||
("a function stitching itself", &shaded, vec![stitch(&me)]),
|
||
("two functions stitching each other", &shaded, vec![stitch(&next), stitch(&me)]),
|
||
("twenty functions, each stitching the next", &shaded, chain(20)),
|
||
("procedures in procedures", &shaded, vec![calculator(&format!("{}{}", "{".repeat(100), "}".repeat(100)))]),
|
||
("a colour space on itself", &filled, vec![indexed(&me)]),
|
||
("two colour spaces on each other", &filled, vec![indexed(&next), indexed(&me)]),
|
||
// Its leading items are read by the interpreter all the same: an item that can't be read
|
||
// (a stream of an external file) after them doesn't hide them.
|
||
("a function stitching itself, then an item that can't be read", &shaded, vec![stitch(&format!("{me} << /F (x) >> stream"))]),
|
||
("a colour space on itself, then an item that can't be read", &filled, vec![format!("[/Indexed {me} 1 <000000ffffff> << /F (x) >> stream]")]),
|
||
] {
|
||
let objects: Vec<&str> = objects.iter().map(String::as_str).collect();
|
||
let e = import(&pdf_with(std::slice::from_ref(page), &objects, None)).err().map(|e| e.to_string());
|
||
// A damaged array is refused as damaged: what it holds can't be looked at whole.
|
||
let refused = |e: &str| e.contains("built on one another without end") || e.contains("is damaged and can't be read");
|
||
assert!(e.as_deref().is_some_and(refused), "{what}: {e:?}");
|
||
}
|
||
for (what, page, objects) in [
|
||
("a function stitching one", &shaded, vec![stitch(&next), ramp.to_string()]),
|
||
("three functions, each stitching the next", &shaded, chain(3)),
|
||
("a calculator function", &shaded, vec![calculator("{ dup 1 exch sub 0 }")]),
|
||
("an indexed colour space", &filled, vec![indexed("/DeviceRGB")]),
|
||
] {
|
||
let objects: Vec<&str> = objects.iter().map(String::as_str).collect();
|
||
let d = import(&pdf_with(std::slice::from_ref(page), &objects, None)).unwrap_or_else(|e| panic!("{what}: {e}"));
|
||
assert!(d.layers.iter().any(|l| l.children().is_some_and(|c| !c.is_empty())), "{what}: its art is read");
|
||
}
|
||
}
|
||
|
||
/// A dictionary value the reader can't parse (a name with a broken `#` escape, a number with a
|
||
/// stray `-`) is left out of the nesting check, which goes on with the rest of the file: a file
|
||
/// whose shading has a plain function is read, and one whose function stitches itself is refused.
|
||
#[test]
|
||
fn dictionary_values_the_reader_cannot_parse_are_skipped() {
|
||
use vectorcraft_testkit::pdf::{PdfPage, first_extra, pdf_with};
|
||
let me = format!("{} 0 R", first_extra(1));
|
||
let shaded = PdfPage {
|
||
resources: format!("/Shading << /S << /ShadingType 2 /ColorSpace /DeviceRGB /Coords [0 0 100 0] /Function {me} >> >>"),
|
||
..PdfPage::new(100.0, 100.0, "/S sh")
|
||
};
|
||
let ramp = "<< /FunctionType 2 /Domain [0 1] /C0 [1 0 0] /C1 [0 0 1] /N 1 >>";
|
||
let stitch = format!("<< /FunctionType 3 /Domain [0 1] /Functions [{me}] /Bounds [] /Encode [0 1] >>");
|
||
for junk in ["<< /A /#zz >>", "<< /A 12-.5 >>"] {
|
||
let [plain, looped] = [ramp, stitch.as_str()].map(|f| pdf_with(std::slice::from_ref(&shaded), &[f, junk], None));
|
||
assert_eq!(info(&plain, None).map(|i| i.pages.len()).ok(), Some(1), "{junk}");
|
||
let d = import(&plain).unwrap_or_else(|e| panic!("{junk}: {e}"));
|
||
assert!(d.layers.iter().any(|l| l.children().is_some_and(|c| !c.is_empty())), "{junk}: its art is read");
|
||
let e = import(&looped).err().map(|e| e.to_string());
|
||
assert!(e.as_deref().is_some_and(|e| e.contains("built on one another without end")), "{junk}: {e:?}");
|
||
}
|
||
}
|