Add support for custom stage typesHEADmain

1 files changed, 17 insertions, 4 deletions

diff --git a/examples/generate_map.rs b/examples/generate_map.rs
index 2d83b0b..1728d52 100644
--- a/examples/generate_map.rs
+++ b/examples/generate_map.rs
@@ -9,27 +9,40 @@ fn main() -> Result<(), Box<dyn std::error::Error>> {
 
     // Generate PNG with smart positioning
     println!("Generating PNG with smart positioning...");
-    let png_data = render_to_png(&map, StageType::Activities, &config)?;
+    let png_data = render_to_png(&map, &StageType::Activities, &config)?;
     fs::write("example.png", png_data)?;
     println!("Generated example.png");
 
     // Generate SVG with smart positioning
     println!("Generating SVG with smart positioning...");
-    let svg_data = render_to_svg(&map, StageType::Activities, &config)?;
+    let svg_data = render_to_svg(&map, &StageType::Activities, &config)?;
     fs::write("example.svg", svg_data)?;
     println!("Generated example.svg");
 
     // Generate with different stage type
     println!("Generating with publication types stage type...");
-    let png_publication_types = render_to_png(&map, StageType::PublicationTypes, &config)?;
+    let png_publication_types = render_to_png(&map, &StageType::PublicationTypes, &config)?;
     fs::write("example_publication_types.png", png_publication_types)?;
     println!("Generated example_publication_types.png");
 
+    // Generate with a custom stage type
+    println!("Generating with publication types stage type...");
+    let stage_type = StageType::Custom {
+        name: "My Stage Type".to_string(),
+        i: "Earlier".to_string(),
+        ii: "Early".to_string(),
+        iii: "Late".to_string(),
+        iv: "Later".to_string(),
+    };
+    let png_publication_types = render_to_svg(&map, &stage_type, &config)?;
+    fs::write("example_custom_stages.svg", png_publication_types)?;
+    println!("Generated example_publication_types.png");
+
     // Generate without smart positioning (simple right-side positioning)
     println!("Generating without smart positioning...");
     let mut config_no_smart = Configuration::default();
     config_no_smart.options.smart_label_positioning = false;
-    let png_no_smart = render_to_png(&map, StageType::Activities, &config_no_smart)?;
+    let png_no_smart = render_to_png(&map, &StageType::Activities, &config_no_smart)?;
     fs::write("example_no_smart_positioning.png", png_no_smart)?;
     println!("Generated example_no_smart_positioning.png");