expose save_text

kgraph.cpp

@@ -334,6 +334,21 @@ namespace kgraph {
         }
 
         virtual void save (char const *path, int format) const {
+            if (format == FORMAT_TEXT) {
+                std::cerr << "Saving to text file; you won't be able to load text file." << std::endl;
+                ofstream os(path);
+                os << graph.size() << endl;
+                for (unsigned i = 0; i < graph.size(); ++i) {
+                    auto const &knn = graph[i];
+                    uint32_t K = knn.size();
+                    os << K;
+                    for (unsigned k = 0; k < K; ++k) {
+                        os << ' ' << knn[k].id << ' ' << knn[k].dist;
+                    }
+                    os << endl;
+                }
+                return;
+            }
             ofstream os(path, ios::binary);
             uint32_t N = graph.size();
             os.write(KGRAPH_MAGIC, KGRAPH_MAGIC_SIZE);

kgraph.h

@@ -127,7 +127,8 @@ namespace kgraph {
 
         enum {
             FORMAT_DEFAULT = 0,
-            FORMAT_NO_DIST = 1
+            FORMAT_NO_DIST = 1,
+            FORMAT_TEXT = 128
         };
 
         /// Information and statistics of the indexing algorithm.

python/pykgraph.cpp

@@ -269,6 +269,9 @@ namespace {
         void save (char const *path) const {
             index->save(path);
         }
+        void save_text (char const *path) const {
+            index->save(path, kgraph::KGraph::FORMAT_TEXT);
+        }
     };
 
     template <typename DATA_TYPE, typename METRIC_TYPE>
@@ -400,6 +403,10 @@ class KGraph {
         impl->save(path);
     }
 
+    void save_text (char const *path) const {
+        impl->save_text(path);
+    }
+
     void build (unsigned iterations,
                unsigned L,
                unsigned K,
@@ -468,6 +475,7 @@ BOOST_PYTHON_MODULE(pykgraph)
             //  (python::arg("data"), python::arg("metric") = "euclidean")))
         .def("load", &KGraph::load)
         .def("save", &KGraph::save)
+        .def("save_text", &KGraph::save_text)
         .def("build", &KGraph::build,
              (python::arg("data"),
               python::arg("iterations") = kgraph::default_iterations,