aboutsummaryrefslogtreecommitdiff
path: root/src/main.rs
diff options
context:
space:
mode:
Diffstat (limited to 'src/main.rs')
-rw-r--r--src/main.rs6
1 files changed, 5 insertions, 1 deletions
diff --git a/src/main.rs b/src/main.rs
index f9bf4fc..71e5d00 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -61,6 +61,10 @@ struct Args {
#[arg(long, default_value = "christofides")]
tsp_approx: TspAlg,
+ /// Seed for hashing. Random by default.
+ #[arg(long)]
+ hash_seed: Option<u64>,
+
images: Vec<PathBuf>,
}
@@ -138,7 +142,7 @@ where
}
let embeds: Vec<_> = embeds.into_iter().map(|e| e.unwrap()).collect();
- let (tsp_path, total_dist) = tsp(&embeds, &args.tsp_approx);
+ let (tsp_path, total_dist) = tsp(&embeds, &args.tsp_approx, &args.hash_seed);
Ok((
tsp_path.iter().map(|i| args.images[*i].clone()).collect(),