implemented the compute_components_per_block function

added .envrc and shell.nix files to load the dependencies via nix and direnv

added .direnv folder to .gitignore

Author: notNN1

.envrc

@@ -0,0 +1,2 @@
+use nix
+clear

.gitignore

@@ -48,3 +48,4 @@ python/tests/unittest
 python/tests/*.so
 python/examples/*.so
 compile_commands.json
+.direnv

mt-kahypar/partition/connected_components/compute_components.cpp

@@ -29,6 +29,8 @@
 namespace mt_kahypar {
 namespace connected_components {
 
+using Bitset = mt_kahypar::ds::Bitset;
+
 template<typename PartitionedHypergraph>
 void compute_components_per_block(const PartitionedHypergraph& phg,
                                   const Context& context,
@@ -38,7 +40,47 @@ void compute_components_per_block(const PartitionedHypergraph& phg,
     list.clear();
   }
 
-  // TODO: compute components and write components of block i to result[i]
+  (void)context;
+
+  Bitset node_colored;
+  node_colored.resize(phg.initialNumNodes());
+  
+  std::queue<HypernodeID> node_queue;
+  PartitionID current_partition;
+
+  for (const HypernodeID& hn : phg.nodes()) {
+    if (node_colored.isSet((size_t) hn)) {
+      continue;
+    }
+
+    current_partition = phg.partID(hn);
+    node_queue.push(hn);
+    
+    ConnectedComponent cc = { };
+
+    while (node_queue.size() > 0) {
+      HypernodeID current = node_queue.front();
+      node_colored.set((size_t) current);
+      cc.nodes.push_back(current);
+      node_queue.pop();
+
+      for (const HyperedgeID& he : phg.incidentEdges(current)) {
+        for (const HypernodeID& incident_hn : phg.pins(he)) {
+          if (node_colored.isSet((size_t) incident_hn)) {
+            continue;
+          }
+          
+          if (phg.partID(incident_hn) != current_partition) {
+            continue;
+          }
+
+          node_queue.push(incident_hn);
+        }
+      }
+    }
+    
+    result[current_partition].push_back(cc);
+  }
 }
 
 

shell.nix

@@ -0,0 +1,11 @@
+{ pkgs ? import <nixpkgs> {}}:
+pkgs.mkShell {
+  packages = with pkgs; [
+    libgcc
+    onetbb
+    hwloc
+    cmake
+  ];
+
+  buildInputs = [ pkgs.bashInteractive ];
+}