hal8174/factorio_blueprint
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Add visualization image output

Browse source
This commit is contained in:
hal8174 2025-05-02 19:29:45 +02:00
parent 8e30f35890
commit 7f563b37a4
1 changed files with 7 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

8
factorio-core/src/visualize/mod.rs
View file

@ -14,13 +14,19 @@ pub trait Visualize {
print::print(v); print::print(v);
} }
fn png_visualization(&self, file: impl AsRef<std::path::Path>) { fn image_visualization(&self) -> RgbaImage {
let v = self.visualize(); let v = self.visualize();
let mut img = RgbaImage::new((v.size.x * 10) as u32, (v.size.y * 10) as u32); let mut img = RgbaImage::new((v.size.x * 10) as u32, (v.size.y * 10) as u32);
image::draw(&v, &mut img); 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 mut f = std::fs::File::create(file).unwrap();
let _ = img.write_to(&mut f, ::image::ImageFormat::Png); let _ = img.write_to(&mut f, ::image::ImageFormat::Png);
} }