From d6e411b20d269a8aa52523d82f0951e53fbcc6bc Mon Sep 17 00:00:00 2001
From: Dominik Rosch <dominik.rosch@student.kit.edu>
Date: Mon, 3 Jun 2024 13:04:01 +0200
Subject: [PATCH] add cli option ofr kN with spanning tree

---
 kaminpar-shm/context_io.cc | 4 +++-
 kaminpar-shm/factories.cc  | 2 ++
 2 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/kaminpar-shm/context_io.cc b/kaminpar-shm/context_io.cc
index 34c7d4b8..aa0e5d6b 100644
--- a/kaminpar-shm/context_io.cc
+++ b/kaminpar-shm/context_io.cc
@@ -78,7 +78,9 @@ std::unordered_map<std::string, SparsificationAlgorithm> get_sparsification_algo
       {"random", SparsificationAlgorithm::UNIFORM_RANDOM_SAMPLING},
       {"rn", SparsificationAlgorithm::UNIFORM_RANDOM_SAMPLING},
       {"k-neighbour", SparsificationAlgorithm::K_NEIGHBOUR},
-      {"kn", SparsificationAlgorithm::K_NEIGHBOUR}
+      {"kn", SparsificationAlgorithm::K_NEIGHBOUR},
+      {"k-neighbour-spanning-tree", SparsificationAlgorithm::K_NEIGHBOUR_SPANNING_TREE},
+      {"kn-st", SparsificationAlgorithm::K_NEIGHBOUR_SPANNING_TREE}
   };
 }
 
diff --git a/kaminpar-shm/factories.cc b/kaminpar-shm/factories.cc
index 20910cfa..cdfd9008 100644
--- a/kaminpar-shm/factories.cc
+++ b/kaminpar-shm/factories.cc
@@ -100,6 +100,8 @@ std::unique_ptr<sparsification::Sampler> create_sampler(const Context &ctx) {
     return std::make_unique<sparsification::UniformRandomSampler>();
   case SparsificationAlgorithm::K_NEIGHBOUR:
     return std::make_unique<sparsification::kNeighbourSampler>();
+  case SparsificationAlgorithm::K_NEIGHBOUR_SPANNING_TREE:
+    return std::make_unique<sparsification::kNeighbourSampler>(true);
   }
 
   __builtin_unreachable();