From 4a17c06f22d3236da6f30c397695ef3771a9d393 Mon Sep 17 00:00:00 2001
From: metamuffin <metamuffin@disroot.org>
Date: Wed, 20 Sep 2023 17:23:45 +0200
Subject: support for different vector metrics

---
 src/main.rs | 16 +++++++++++++---
 1 file changed, 13 insertions(+), 3 deletions(-)

(limited to 'src/main.rs')

diff --git a/src/main.rs b/src/main.rs
index 474532c..2dda2da 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -19,7 +19,9 @@ enum Embedder {
     Brightness,
     Hue,
     Color,
-    Content,
+    ContentEuclidean,
+    ContentCosineSim,
+    ContentManhatten,
 }
 
 #[derive(Debug, Clone, Copy, clap::ValueEnum)]
@@ -32,7 +34,7 @@ enum TspAlg {
 #[derive(Debug, Parser)]
 struct Args {
     /// Characteristic to sort by
-    #[arg(short, long, default_value = "content")]
+    #[arg(short, long, default_value = "content-euclidean")]
     embedder: Embedder,
 
     /// Symlink the sorted images into this directory
@@ -175,7 +177,15 @@ fn main() -> Result<()> {
         Embedder::Brightness => process_embedder(BrightnessEmbedder, &args, &cfg),
         Embedder::Hue => process_embedder(HueEmbedder, &args, &cfg),
         Embedder::Color => process_embedder(ColorEmbedder, &args, &cfg),
-        Embedder::Content => process_embedder(ContentEmbedder::new(&cfg), &args, &cfg),
+        Embedder::ContentCosineSim => {
+            process_embedder(ContentEmbedder::<CosineSimilarity>::new(&cfg), &args, &cfg)
+        }
+        Embedder::ContentEuclidean => {
+            process_embedder(ContentEmbedder::<EuclidianDistance>::new(&cfg), &args, &cfg)
+        }
+        Embedder::ContentManhatten => {
+            process_embedder(ContentEmbedder::<ManhattenDistance>::new(&cfg), &args, &cfg)
+        }
     }?;
 
     if args.benchmark {
-- 
cgit v1.2.3-70-g09d2