verify partition: support metis format

Author: N-Maas

tools/verify_partition.cc

@@ -32,7 +32,9 @@
 #include <string>
 
 #include "mt-kahypar/macros.h"
+#include "mt-kahypar/datastructures/static_graph.h"
 #include "mt-kahypar/datastructures/static_hypergraph.h"
+#include "mt-kahypar/datastructures/partitioned_graph.h"
 #include "mt-kahypar/datastructures/partitioned_hypergraph.h"
 #include "mt-kahypar/datastructures/connectivity_info.h"
 #include "mt-kahypar/partition/context.h"
@@ -46,9 +48,12 @@ using namespace mt_kahypar;
 namespace po = boost::program_options;
 
 using Hypergraph = ds::StaticHypergraph;
+using StaticGraph = ds::StaticGraph;
 using PartitionedHypergraph = ds::PartitionedHypergraph<Hypergraph, ds::ConnectivityInfo>;
+using PartitionedGraph = ds::PartitionedGraph<StaticGraph>;
 
-bool readPartitionFile(const std::string& partition_file, PartitionedHypergraph& hypergraph) {
+template<typename PHG>
+bool readPartitionFile(const std::string& partition_file, PHG& hypergraph) {
   bool success = true;
   std::vector<PartitionID> partition;
   mt_kahypar::io::readPartitionFile(partition_file, hypergraph.initialNumNodes(), partition);
@@ -67,6 +72,44 @@ bool readPartitionFile(const std::string& partition_file, PartitionedHypergraph&
   return success;
 }
 
+template<typename PHG>
+bool computeResults(PHG& phg, Context& context) {
+  // Setup Context
+  context.setupPartWeights(phg.totalWeight());
+
+  // Read Partition File
+  bool success = readPartitionFile(context.partition.graph_partition_filename, phg);
+
+  for ( PartitionID i = 0; i < context.partition.k; ++i ) {
+    if ( !(phg.partWeight(i) <= context.partition.max_part_weights[i]) ) {
+      LOG << RED << "[ERROR]" << END << "Block" << (i + 1) << "has weight"
+          << phg.partWeight(i) << ", but maximum allowed block weight is"
+          << context.partition.max_part_weights[i] << END;
+      success = false;
+    }
+  }
+
+  // Check fixed vertices
+  if ( phg.hasFixedVertices() ) {
+    for ( const HypernodeID& hn : phg.nodes() ) {
+      if ( phg.isFixed(hn) && phg.fixedVertexBlock(hn) != phg.partID(hn) ) {
+        LOG << RED << "Node" << hn << "is fixed to block" << phg.fixedVertexBlock(hn)
+            << ", but assigned to block" << phg.partID(hn) << END;
+        success = false;
+      }
+    }
+  }
+
+  std::cout << "***********************" << context.partition.k
+            << "-way Partition Result************************" << std::endl;
+  std::cout << "cut =" << metrics::quality(phg, Objective::cut) << std::endl;
+  std::cout << "soed =" << metrics::quality(phg, Objective::soed) << std::endl;
+  std::cout << "km1 = " << metrics::quality(phg, Objective::km1) << std::endl;
+  std::cout << "imbalance = " << metrics::imbalance(phg, context) << std::endl;
+
+  return success;
+}
+
 int main(int argc, char* argv[]) {
   Context context;
 
@@ -78,6 +121,17 @@ int main(int argc, char* argv[]) {
           ("partition-file,b",
            po::value<std::string>(&context.partition.graph_partition_filename)->value_name("<string>")->required(),
            "Partition Filename")
+          ("input-file-format",
+            po::value<std::string>()->value_name("<string>")->notifier([&](const std::string& s) {
+              if (s == "hmetis") {
+                context.partition.file_format = FileFormat::hMetis;
+              } else if (s == "metis") {
+                context.partition.file_format = FileFormat::Metis;
+              }
+            }),
+            "Input file format: \n"
+            " - hmetis : hMETIS hypergraph file format \n"
+            " - metis : METIS graph file format")
           ("blocks,k",
            po::value<PartitionID>(&context.partition.k)->value_name("<int>")->required(),
            "Number of Blocks")
@@ -97,48 +151,32 @@ int main(int argc, char* argv[]) {
     mt_kahypar::io::readInputFile(
       context.partition.graph_filename, PresetType::default_preset,
       InstanceType::hypergraph, FileFormat::hMetis, true);
-  Hypergraph& hg = utils::cast<Hypergraph>(hypergraph);
-  PartitionedHypergraph phg(context.partition.k, hg, parallel_tag_t());
 
-  // Add fixed vertices
-  if ( context.partition.fixed_vertex_filename != "" ) {
-    mt_kahypar::io::addFixedVerticesFromFile(
-      hypergraph, context.partition.fixed_vertex_filename, context.partition.k);
-  }
+  bool success;
+  if (context.partition.file_format == FileFormat::hMetis) {
+    Hypergraph& hg = utils::cast<Hypergraph>(hypergraph);
+    PartitionedHypergraph phg(context.partition.k, hg, parallel_tag_t());
 
-  // Setup Context
-  context.setupPartWeights(hg.totalWeight());
+    // Add fixed vertices
+    if ( context.partition.fixed_vertex_filename != "" ) {
+      mt_kahypar::io::addFixedVerticesFromFile(
+        hypergraph, context.partition.fixed_vertex_filename, context.partition.k);
+    }
 
-  // Read Partition File
-  bool success = readPartitionFile(context.partition.graph_partition_filename, phg);
+    success = computeResults(phg, context);
+  } else {
+    StaticGraph& hg = utils::cast<StaticGraph>(hypergraph);
+    PartitionedGraph phg(context.partition.k, hg, parallel_tag_t());
 
-  for ( PartitionID i = 0; i < context.partition.k; ++i ) {
-    if ( !(phg.partWeight(i) <= context.partition.max_part_weights[i]) ) {
-      LOG << RED << "[ERROR]" << END << "Block" << (i + 1) << "has weight"
-          << phg.partWeight(i) << ", but maximum allowed block weight is"
-          << context.partition.max_part_weights[i] << END;
-      success = false;
+    // Add fixed vertices
+    if ( context.partition.fixed_vertex_filename != "" ) {
+      mt_kahypar::io::addFixedVerticesFromFile(
+        hypergraph, context.partition.fixed_vertex_filename, context.partition.k);
     }
-  }
 
-  // Check fixed vertices
-  if ( phg.hasFixedVertices() ) {
-    for ( const HypernodeID& hn : phg.nodes() ) {
-      if ( phg.isFixed(hn) && phg.fixedVertexBlock(hn) != phg.partID(hn) ) {
-        LOG << RED << "Node" << hn << "is fixed to block" << phg.fixedVertexBlock(hn)
-            << ", but assigned to block" << phg.partID(hn) << END;
-        success = false;
-      }
-    }
+    success = computeResults(phg, context);
   }
 
-  std::cout << "***********************" << context.partition.k
-            << "-way Partition Result************************" << std::endl;
-  std::cout << "cut =" << metrics::quality(phg, Objective::cut) << std::endl;
-  std::cout << "soed =" << metrics::quality(phg, Objective::soed) << std::endl;
-  std::cout << "km1 = " << metrics::quality(phg, Objective::km1) << std::endl;
-  std::cout << "imbalance = " << metrics::imbalance(phg, context) << std::endl;
-
   utils::delete_hypergraph(hypergraph);
 
   return success ? 0 : -1;