summaryrefslogtreecommitdiff
path: root/src/library/image.rs
blob: cd6a97d11b1aae30d611b79e848fa6dca37c18ac (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
use ::image::GenericImageView;

use super::*;
use crate::image::ImageId;
use crate::layout::{AnyNode, Element, Frame, Layout, LayoutContext, Regions};

/// `image`: An image.
///
/// Supports PNG and JPEG files.
///
/// # Positional parameters
/// - Path to image file: of type `string`.
///
/// # Return value
/// A template that inserts an image.
pub fn image(ctx: &mut EvalContext, args: &mut FuncArgs) -> Value {
    let path = args.eat_expect::<Spanned<String>>(ctx, "path to image file");
    let width = args.eat_named(ctx, "width");
    let height = args.eat_named(ctx, "height");

    let mut node = None;
    if let Some(path) = &path {
        if let Some(id) = ctx.cache.image.load(ctx.loader, &path.v) {
            let img = ctx.cache.image.get(id);
            let dimensions = img.buf.dimensions();
            node = Some(ImageNode { id, dimensions, width, height });
        } else {
            ctx.diag(error!(path.span, "failed to load image"));
        }
    }

    Value::template("image", move |ctx| {
        if let Some(node) = node {
            ctx.push(node);
        }
    })
}

/// An image node.
#[derive(Debug, Copy, Clone, PartialEq, Hash)]
struct ImageNode {
    /// The id of the image file.
    id: ImageId,
    /// The pixel dimensions of the image.
    dimensions: (u32, u32),
    /// The fixed width, if any.
    width: Option<Linear>,
    /// The fixed height, if any.
    height: Option<Linear>,
}

impl Layout for ImageNode {
    fn layout(&self, _: &mut LayoutContext, regions: &Regions) -> Vec<Frame> {
        let Regions { current, base, .. } = regions;
        let width = self.width.map(|w| w.resolve(base.width));
        let height = self.height.map(|w| w.resolve(base.height));

        let pixel_width = self.dimensions.0 as f64;
        let pixel_height = self.dimensions.1 as f64;
        let pixel_ratio = pixel_width / pixel_height;

        let size = match (width, height) {
            (Some(width), Some(height)) => Size::new(width, height),
            (Some(width), None) => Size::new(width, width / pixel_ratio),
            (None, Some(height)) => Size::new(height * pixel_ratio, height),
            (None, None) => {
                let ratio = current.width / current.height;
                if ratio < pixel_ratio && current.width.is_finite() {
                    Size::new(current.width, current.width / pixel_ratio)
                } else if current.height.is_finite() {
                    // TODO: Fix issue with line spacing.
                    Size::new(current.height * pixel_ratio, current.height)
                } else {
                    // Totally unbounded region, we have to make up something.
                    Size::new(Length::pt(pixel_width), Length::pt(pixel_height))
                }
            }
        };

        let mut frame = Frame::new(size, size.height);
        frame.push(Point::zero(), Element::Image(self.id, size));
        vec![frame]
    }
}

impl From<ImageNode> for AnyNode {
    fn from(image: ImageNode) -> Self {
        Self::new(image)
    }
}