Factor out alignment deduplication into separate function ♻

src/eval/value.rs

@@ -619,8 +619,9 @@ impl From<ValueAny> for Value {
 /// - [`Cast<Value>`](Cast) for `T`.
 ///
 /// # Example
-/// Make a type `FontFamily` that can be cast from a [`Value::Any`] variant
-/// containing a `FontFamily` or from a string.
+/// Allow a type `FontFamily` to be cast from:
+/// - a [`Value::Any`] variant already containing a `FontFamily`
+/// - a string, producing a `FontFamiliy::Named(..)`.
 /// ```
 /// # use typst::typify;
 /// # enum FontFamily { Named(String) }

src/layout/pad.rs

@@ -41,7 +41,7 @@ fn shrink(areas: &Areas, padding: Sides<Linear>) -> Areas {
     }
 }
 
-/// Enlarge the box and move all elements inwards.
+/// Enlarge the frame and move all elements inwards.
 fn pad(frame: &mut Frame, padding: Sides<Linear>) {
     let padding = padding.resolve(frame.size);
     let origin = Point::new(padding.left, padding.top);

src/library/align.rs

@@ -32,17 +32,35 @@ pub fn align(ctx: &mut EvalContext, args: &mut ValueArgs) -> Value {
     Value::template("align", move |ctx| {
         let snapshot = ctx.state.clone();
 
+        let values = first
+            .into_iter()
+            .chain(second.into_iter())
+            .map(|arg: Spanned<AlignValue>| (arg.v.axis(), arg))
+            .chain(hor.into_iter().map(|arg| (Some(SpecAxis::Horizontal), arg)))
+            .chain(ver.into_iter().map(|arg| (Some(SpecAxis::Vertical), arg)));
+
+        apply(ctx, values);
+
+        if ctx.state.aligns.main != snapshot.aligns.main {
+            ctx.push_linebreak();
+        }
+
+        if let Some(body) = &body {
+            body.exec(ctx);
+            ctx.state = snapshot;
+        }
+    })
+}
+
+/// Deduplicate and apply the alignments.
+fn apply(
+    ctx: &mut ExecContext,
+    values: impl Iterator<Item = (Option<SpecAxis>, Spanned<AlignValue>)>,
+) {
     let mut had = Gen::uniform(false);
     let mut had_center = false;
 
-        // Infer the axes alignments belong to.
-        for (axis, Spanned { v: arg, span }) in first
-            .into_iter()
-            .chain(second.into_iter())
-            .map(|arg: Spanned<Alignment>| (arg.v.axis(), arg))
-            .chain(hor.into_iter().map(|arg| (Some(SpecAxis::Horizontal), arg)))
-            .chain(ver.into_iter().map(|arg| (Some(SpecAxis::Vertical), arg)))
-        {
+    for (axis, Spanned { v: arg, span }) in values {
         // Check whether we know which axis this alignment belongs to.
         if let Some(axis) = axis {
             // We know the axis.
@@ -60,7 +78,7 @@ pub fn align(ctx: &mut EvalContext, args: &mut ValueArgs) -> Value {
         } else {
             // We don't know the axis: This has to be a `center` alignment for a
             // positional argument.
-                debug_assert_eq!(arg, Alignment::Center);
+            debug_assert_eq!(arg, AlignValue::Center);
 
             if had.main && had.cross {
                 ctx.diag(error!(span, "duplicate alignment"));
@@ -80,11 +98,10 @@ pub fn align(ctx: &mut EvalContext, args: &mut ValueArgs) -> Value {
         if had_center && (had.main || had.cross) {
             if had.main {
                 ctx.state.aligns.cross = Align::Center;
-                    had.cross = true;
             } else {
                 ctx.state.aligns.main = Align::Center;
-                    had.main = true;
             }
+            had = Gen::uniform(true);
             had_center = false;
         }
     }
@@ -94,21 +111,11 @@ pub fn align(ctx: &mut EvalContext, args: &mut ValueArgs) -> Value {
     if had_center {
         ctx.state.aligns.cross = Align::Center;
     }
-
-        if ctx.state.aligns.main != snapshot.aligns.main {
-            ctx.push_linebreak();
-        }
-
-        if let Some(body) = &body {
-            body.exec(ctx);
-            ctx.state = snapshot;
-        }
-    })
 }
 
-/// An alignment argument.
+/// An alignment value.
 #[derive(Debug, Copy, Clone, Eq, PartialEq, Ord, PartialOrd)]
-pub(super) enum Alignment {
+pub(super) enum AlignValue {
     Left,
     Center,
     Right,
@@ -116,7 +123,7 @@ pub(super) enum Alignment {
     Bottom,
 }
 
-impl Alignment {
+impl AlignValue {
     /// The specific axis this alignment refers to.
     fn axis(self) -> Option<SpecAxis> {
         match self {
@@ -129,7 +136,7 @@ impl Alignment {
     }
 }
 
-impl Switch for Alignment {
+impl Switch for AlignValue {
     type Other = Align;
 
     fn switch(self, dirs: LayoutDirs) -> Self::Other {
@@ -152,7 +159,7 @@ impl Switch for Alignment {
     }
 }
 
-impl Display for Alignment {
+impl Display for AlignValue {
     fn fmt(&self, f: &mut Formatter) -> fmt::Result {
         f.pad(match self {
             Self::Left => "left",
@@ -165,5 +172,5 @@ impl Display for Alignment {
 }
 
 typify! {
-    Alignment: "alignment",
+    AlignValue: "alignment",
 }

src/library/mod.rs

@@ -57,11 +57,11 @@ pub fn new() -> Scope {
     set!(func: "v", v);
 
     // Constants.
-    set!(any: "left", Alignment::Left);
-    set!(any: "center", Alignment::Center);
-    set!(any: "right", Alignment::Right);
-    set!(any: "top", Alignment::Top);
-    set!(any: "bottom", Alignment::Bottom);
+    set!(any: "left", AlignValue::Left);
+    set!(any: "center", AlignValue::Center);
+    set!(any: "right", AlignValue::Right);
+    set!(any: "top", AlignValue::Top);
+    set!(any: "bottom", AlignValue::Bottom);
     set!(any: "ltr", Dir::LTR);
     set!(any: "rtl", Dir::RTL);
     set!(any: "ttb", Dir::TTB);