diagram native svg output

1 files changed, 148 insertions, 0 deletions

diff --git a/server/tools/src/diagram_svg.rs b/server/tools/src/diagram_svg.rs
new file mode 100644
index 00000000..a5d09316
--- /dev/null
+++ b/server/tools/src/diagram_svg.rs
@@ -0,0 +1,148 @@
+/*
+    Hurry Curry! - a game about cooking
+    Copyright (C) 2025 Hurry Curry! Contributors
+
+    This program is free software: you can redistribute it and/or modify
+    it under the terms of the GNU Affero General Public License as published by
+    the Free Software Foundation, version 3 of the License only.
+
+    This program is distributed in the hope that it will be useful,
+    but WITHOUT ANY WARRANTY; without even the implied warranty of
+    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+    GNU Affero General Public License for more details.
+
+    You should have received a copy of the GNU Affero General Public License
+    along with this program.  If not, see <https://www.gnu.org/licenses/>.
+
+*/
+
+use anyhow::Result;
+use hurrycurry_protocol::{
+    Gamedata, Message,
+    book::{Diagram, DiagramNode, NodeStyle},
+    glam::Vec2,
+};
+use std::fmt::Write;
+
+const SIZE: f32 = 64.;
+const HSIZE: f32 = SIZE / 2.;
+
+pub fn diagram_svg(data: &Gamedata, diagram: &Diagram) -> Result<String> {
+    let mut out = String::new();
+
+    let (vb_min, vb_max) = diagram
+        .nodes
+        .iter()
+        .map(|n| (n.position, n.position))
+        .reduce(|(xa, xb), (ya, yb)| (xa.min(ya), xb.max(yb)))
+        .unwrap();
+
+    writeln!(
+        out,
+        r#"<svg xmlns="http://www.w3.org/2000/svg" viewBox="{} {} {} {}" width="{}" height="{}">"#,
+        vb_min.x - HSIZE,
+        vb_min.y - HSIZE,
+        vb_max.x + HSIZE,
+        vb_max.y + HSIZE,
+        vb_max.x - vb_min.x + SIZE,
+        vb_max.y - vb_min.y + SIZE,
+    )?;
+
+    for node in &diagram.nodes {
+        match node.label {
+            Message::Translation { .. } => {}
+            Message::Text(_) => {}
+            Message::Item(item_index) => {
+                image_node(
+                    &mut out,
+                    node,
+                    format!("items/{}.png", data.item_name(item_index)),
+                )?;
+            }
+            Message::Tile(tile_index) => {
+                image_node(
+                    &mut out,
+                    node,
+                    format!("tiles/{}.png", data.tile_name(tile_index)),
+                )?;
+            }
+        }
+    }
+    for edge in &diagram.edges {
+        let src_node = &diagram.nodes[edge.src];
+        let dst_node = &diagram.nodes[edge.dst];
+        let src = node_edge_connect_pos(src_node, dst_node);
+        let dst = node_edge_connect_pos(dst_node, src_node);
+        writeln!(
+            out,
+            r#"<path fill="none" stroke="black" stroke-width="2" d="M{} {} L{} {}" />"#,
+            src.x, src.y, dst.x, dst.y,
+        )?;
+
+        // Array tip
+        let dir = (src - dst).normalize_or_zero();
+        let c1 = dst + (dir + dir.perp() * 0.5) * 10.;
+        let c2 = dst + (dir + dir.perp() * -0.5) * 10.;
+        writeln!(
+            out,
+            r#"<path fill="black" stroke="none" d="M{} {} L{} {} L{} {} Z" />"#,
+            dst.x, dst.y, c1.x, c1.y, c2.x, c2.y
+        )?;
+    }
+
+    writeln!(out, "</svg>")?;
+    Ok(out)
+}
+
+fn node_edge_connect_pos(src: &DiagramNode, dst: &DiagramNode) -> Vec2 {
+    let dir = (dst.position - src.position).normalize_or_zero();
+    if matches!(
+        src.style,
+        NodeStyle::FinalProduct | NodeStyle::IntermediateProduct
+    ) {
+        src.position + dir * HSIZE // circle
+    } else {
+        src.position + dir / dir.y.abs() * HSIZE // square (only +Y and -Y sides)
+    }
+}
+
+fn node_color(node: &DiagramNode) -> &'static str {
+    match node.style {
+        NodeStyle::FinalProduct => "#555",
+        NodeStyle::IntermediateProduct => "#333",
+        NodeStyle::ProcessActive => "#47c42b",
+        NodeStyle::ProcessPassive => "#c4a32b",
+        NodeStyle::ProcessInstant => "#5452d8",
+    }
+}
+
+fn image_node(out: &mut String, node: &DiagramNode, path: String) -> Result<()> {
+    if matches!(
+        node.style,
+        NodeStyle::FinalProduct | NodeStyle::IntermediateProduct
+    ) {
+        writeln!(
+            out,
+            r#"<circle cx="{}" cy="{}" r="{}" stroke="none" fill="{}" />"#,
+            node.position.x,
+            node.position.y,
+            HSIZE,
+            node_color(node)
+        )?;
+    } else {
+        writeln!(
+            out,
+            r#"<rect x="{}" y="{}" width="{SIZE}" height="{SIZE}" stroke="none" fill="{}" />"#,
+            node.position.x - HSIZE,
+            node.position.y - HSIZE,
+            node_color(node)
+        )?;
+    }
+    writeln!(
+        out,
+        r#"<image href="{path}" x="{}" y="{}" width="{SIZE}" height="{SIZE}" />"#,
+        node.position.x - HSIZE,
+        node.position.y - HSIZE,
+    )?;
+    Ok(())
+}