Adjust import settings

Author: dec4234

Cargo.toml

@@ -4,4 +4,7 @@ version = "0.1.0"
 edition = "2024"
 
 [dependencies]
+targetgen-lib = { path = "targetgen-lib"}
 clap = {version = "4.5.20", features = ["default"]}
+simple_logger = "5.0.0"
+log = "0.4.22"

src/main.rs

@@ -1,3 +1,23 @@
+use log::LevelFilter;
+use simple_logger::SimpleLogger;
+use targetgen_lib::generator;
+use targetgen_lib::generator::TargetGenerator;
+
 fn main() {
 
+}
+
+#[ignore]
+#[test]
+fn create_live_test() {
+	SimpleLogger::new().with_level(LevelFilter::Debug).init().unwrap();
+
+	generator::util::cleanup_output("output").unwrap();
+
+	let mut tg = TargetGenerator::new("backgrounds", "objects", "output/annotations.json").unwrap();
+	tg.config.permit_collisions = false;
+	tg.config.do_random_rotation = true;
+	tg.generate_targets(1, ..6, "output").unwrap();
+
+	tg.close();
 }

targetgen-lib/src/generator/mod.rs

@@ -20,7 +20,7 @@ use util::STANDARD_PPM;
 
 pub mod coco;
 pub mod error;
-pub(crate) mod util;
+pub mod util;
 pub mod config;
 
 const COLLISION_ATTEMPTS: u32 = 15;
@@ -30,7 +30,7 @@ pub struct TargetGenerator {
 	pub object_manager: ObjectManager,
 	background_loader: BackgroundLoader,
 	coco_generator: Arc<Mutex<CocoGenerator>>,
-	pub(crate) config: TargetGeneratorConfig,
+	pub config: TargetGeneratorConfig,
 	resized_cache: Cache<String, DynamicImage>,
 }
 

targetgen-lib/src/generator/util.rs

@@ -1,3 +1,5 @@
+use std::fs;
+use std::path::Path;
 use crate::generator::error::GenerationError;
 use image::metadata::Orientation;
 use image::DynamicImage;
@@ -67,6 +69,18 @@ pub fn is_image_type(path: &str) -> bool {
 	path.ends_with(".png") || path.ends_with(".jpg") || path.ends_with(".jpeg")
 }
 
+pub fn cleanup_output<P: AsRef<Path>>(folder_path: P) -> std::io::Result<()> {
+	for entry in fs::read_dir(folder_path)? {
+		let entry = entry?;
+		let path = entry.path();
+		
+		if entry.file_type()?.is_file() && (is_image_type(path.to_str().unwrap()) || path.extension().unwrap() == "json") {
+			fs::remove_file(entry.path())?;
+		}
+	}
+	Ok(())
+}
+
 #[test]
 fn test_is_image_type() {
 	assert_eq!(is_image_type("test.png"), true);

targetgen-lib/src/lib.rs

@@ -1,5 +1,5 @@
 #![allow(dead_code)] // allows unused struct members
 
-mod backgrounds;
-mod objects;
-mod generator;
+pub mod backgrounds;
+pub mod objects;
+pub mod generator;