view: Cycle through multiple fg colours if supplied

yeslogic · Oct 25, 2024 · 6e9bcb5 · 6e9bcb5
1 parent 146aa99
commit 6e9bcb5
Show file tree

Hide file tree

Showing 3 changed files with 24 additions and 12 deletions.
diff --git a/src/cli.rs b/src/cli.rs
@@ -367,10 +367,10 @@ pub struct ViewOpts {
 
     #[options(
         help = "set the fill colour of the glyphs",
-        meta = "rrggbbaa",
+        meta = "rrggbbaa...",
         no_short
     )]
-    pub fg_colour: Option<Colour>,
+    pub fg_colour: Vec<Colour>,
 
     #[options(
         help = "set the background colour of the generated SVG",
@@ -379,8 +379,8 @@ pub struct ViewOpts {
     )]
     pub bg_colour: Option<Colour>,
 
-    #[options(help = "alias for --fg-colour", meta = "rrggbbaa", no_short)]
-    pub fg_color: Option<Colour>,
+    #[options(help = "alias for --fg-colour", meta = "rrggbbaa...", no_short)]
+    pub fg_color: Vec<Colour>,
 
     #[options(help = "alias for --bg-colour", meta = "rrggbbaa", no_short)]
     pub bg_color: Option<Colour>,

diff --git a/src/view.rs b/src/view.rs
@@ -178,10 +178,15 @@ fn parse_features(features: &str) -> Features {
 
 impl From<&ViewOpts> for SVGMode {
     fn from(opts: &ViewOpts) -> Self {
+        let fg = if opts.fg_colour.is_empty() {
+            &opts.fg_color
+        } else {
+            &opts.fg_colour
+        };
         SVGMode::View {
             mark_origin: opts.mark_origin,
             margin: opts.margin.unwrap_or_default(),
-            fg: opts.fg_colour.or(opts.fg_color),
+            fg: fg.clone(),
             bg: opts.bg_colour.or(opts.bg_color),
         }
     }

diff --git a/src/writer.rs b/src/writer.rs
@@ -1,6 +1,7 @@
 use std::borrow::Cow;
 use std::collections::HashMap;
 use std::fmt::{Display, Formatter};
+use std::iter;
 use std::str::FromStr;
 
 use allsorts::cff::CFF;
@@ -187,7 +188,7 @@ pub enum SVGMode {
     View {
         mark_origin: bool,
         margin: Margin,
-        fg: Option<Colour>,
+        fg: Vec<Colour>,
         bg: Option<Colour>,
     },
 }
@@ -196,6 +197,7 @@ pub struct SVGWriter {
     mode: SVGMode,
     transform: Matrix2x2F,
     usage: Vec<(usize, Vector2F)>,
+    fg_colour: Box<dyn Iterator<Item = Colour>>,
 }
 
 struct Symbols<'info> {
@@ -208,10 +210,18 @@ struct Symbols<'info> {
 
 impl SVGWriter {
     pub fn new(mode: SVGMode, transform: Matrix2x2F) -> Self {
+        let fg_colour = match &mode {
+            SVGMode::TextRenderingTests(_) => {
+                Box::new(iter::empty()) as Box<dyn Iterator<Item = Colour>>
+            }
+            SVGMode::View { fg, .. } => Box::new(fg.clone().into_iter().cycle()),
+        };
+
         SVGWriter {
             mode,
             transform,
             usage: Vec::new(),
+            fg_colour,
         }
     }
 
@@ -300,7 +310,7 @@ impl SVGWriter {
             .push((symbol_index, self.transform * vec2f(x, y)));
     }
 
-    fn end(self, x_max: f32, ascender: i16, descender: i16, symbols: Symbols) -> String {
+    fn end(mut self, x_max: f32, ascender: i16, descender: i16, symbols: Symbols) -> String {
         let mut w = XmlWriter::new(xmlwriter::Options::default());
         w.write_declaration();
         w.start_element("svg");
@@ -414,11 +424,8 @@ impl SVGWriter {
         }
     }
 
-    fn fg_colour(&self) -> Option<Colour> {
-        match self.mode {
-            SVGMode::TextRenderingTests(_) => None,
-            SVGMode::View { fg, .. } => fg,
-        }
+    fn fg_colour(&mut self) -> Option<Colour> {
+        self.fg_colour.next()
     }
 
     fn bg_colour(&self) -> Option<Colour> {