Add Layout trait

factorio-layout/src/valid_layout.rs

@@ -0,0 +1,44 @@
+use factorio_core::prelude::{Position, PositionType};
+use factorio_pathfinding::Pathfinder;
+use rand::Rng;
+
+use crate::{
+    LayoutResult, Layouter,
+    misc::{initally_set_blocks, path_input_from_blocks_positions},
+};
+
+pub struct ValidLayout {
+    pub max_tries: usize,
+    pub retries: usize,
+    pub start_size: Position,
+    pub growth: Position,
+}
+
+impl Layouter for ValidLayout {
+    fn layout<R: Rng, P: Pathfinder>(
+        &self,
+        input: &crate::LayoutInput,
+        pathfinder: &P,
+        rng: &mut R,
+    ) -> Option<LayoutResult> {
+        for i in 0..self.max_tries {
+            let size = self.start_size + i as PositionType * self.growth;
+
+            if let Some(blocks) = initally_set_blocks(input, size, self.retries, rng) {
+                let (connections, map) = path_input_from_blocks_positions(input, size, &blocks);
+
+                if let Some(paths) = pathfinder.find_paths(factorio_pathfinding::PathInput {
+                    connections: &connections,
+                    map,
+                }) {
+                    return Some(LayoutResult {
+                        positions: blocks,
+                        path_result: paths,
+                    });
+                }
+            }
+        }
+
+        None
+    }
+}