docgen: Support pikchr diagrams

Signed-off-by: Daniel Silverstone <dsilvers@digital-scurf.org>

1 files changed, 14 insertions, 0 deletions

diff --git a/src/visitor/typesetting.rs b/src/visitor/typesetting.rs
index deffcd9..f59f81d 100644
--- a/src/visitor/typesetting.rs
+++ b/src/visitor/typesetting.rs
@@ -39,6 +39,20 @@ impl<'a> MutVisitor for TypesettingVisitor<'a> {
                         *block = typeset::plantuml_to_block(s)
                     } else if is_class(attr, "roadmap") {
                         *block = typeset::roadmap_to_block(s)
+                    } else if is_class(attr, "pikchr") {
+                        let other_classes: Vec<_> = attr
+                            .1
+                            .iter()
+                            .map(String::as_str)
+                            .filter(|s| *s != "pikchr")
+                            .collect();
+                        let class = if other_classes.is_empty() {
+                            None
+                        } else {
+                            Some(other_classes.join(" "))
+                        };
+                        let class = class.as_deref();
+                        *block = typeset::pikchr_to_block(s, class)
                     }
                 }
                 _ => {