From 7f563b37a4e89a132a4555e71fc10f9c35b1dd69 Mon Sep 17 00:00:00 2001
From: hal8174 <henning.lindemann@pm.me>
Date: Fri, 2 May 2025 19:29:45 +0200
Subject: [PATCH] Add visualization image output

---
 factorio-core/src/visualize/mod.rs | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/factorio-core/src/visualize/mod.rs b/factorio-core/src/visualize/mod.rs
index e083a00..c96cac8 100644
--- a/factorio-core/src/visualize/mod.rs
+++ b/factorio-core/src/visualize/mod.rs
@@ -14,13 +14,19 @@ pub trait Visualize {
         print::print(v);
     }
 
-    fn png_visualization(&self, file: impl AsRef<std::path::Path>) {
+    fn image_visualization(&self) -> RgbaImage {
         let v = self.visualize();
 
         let mut img = RgbaImage::new((v.size.x * 10) as u32, (v.size.y * 10) as u32);
 
         image::draw(&v, &mut img);
 
+        img
+    }
+
+    fn png_visualization(&self, file: impl AsRef<std::path::Path>) {
+        let img = self.image_visualization();
+
         let mut f = std::fs::File::create(file).unwrap();
         let _ = img.write_to(&mut f, ::image::ImageFormat::Png);
     }