diff --git a/src/main.rs b/src/main.rs index 0075f30..c316356 100644 --- a/src/main.rs +++ b/src/main.rs @@ -250,7 +250,11 @@ fn main() -> Result<(), Box> { let raw_conf = open_file(matches.value_of("file").unwrap_or("two_robots.txt"))?; let mut robot_pool: Vec = Vec::new(); - let world: world::World = parse_config(raw_conf, &mut robot_pool)?; + let world: world::World = match matches.value_of("random-world") { + None => parse_config(raw_conf, &mut robot_pool)?, + Some(_) => world::random_world(), + }; + let mut hash: HashMap = HashMap::new(); create_hash_map(&robot_pool, &mut hash); println!("World {{ x_max = {}; y_max = {} }}", world.x, world.y);