1 files changed, 195 insertions, 0 deletions
diff --git a/src/Tangential_complex/benchmark/XML_exporter.h b/src/Tangential_complex/benchmark/XML_exporter.h
new file mode 100644
index 00000000..16b62eb6
--- /dev/null
+++ b/src/Tangential_complex/benchmark/XML_exporter.h
@@ -0,0 +1,195 @@
+/*    This file is part of the Gudhi Library - https://gudhi.inria.fr/ - which is released under MIT.
+ *    See file LICENSE or go to https://gudhi.inria.fr/licensing/ for full license details.
+ *    Author(s):       Clement Jamin
+ *
+ *    Copyright (C) 2016 Inria
+ *
+ *    Modification(s):
+ *      - YYYY/MM Author: Description of the modification
+ */
+
+#include <string>
+#include <vector>
+#include <iostream>
+#include <fstream>
+#include <ctime>
+
+template<typename value_type = std::string>
+class Simple_XML_exporter {
+ public:
+  typedef value_type Value_type;
+  typedef std::vector<value_type> Element;
+  typedef std::map<std::string, value_type> Element_with_map;
+  typedef std::vector<Element> List_of_elements;
+
+  Simple_XML_exporter(
+                      const std::string &list_name,
+                      const std::string &element_name,
+                      const std::vector<std::string> &subelement_names,
+                      bool add_timestamp = true)
+      : m_list_name(list_name),
+      m_element_name(element_name),
+      m_subelement_names(subelement_names),
+      m_add_timestamp(add_timestamp) { }
+
+  bool add_element(const Element &element) {
+    if (element.size() == m_subelement_names.size()) {
+      m_list_of_elements.push_back(element);
+      return true;
+    } else {
+      std::cerr << "ERROR: element.size() == m_subelement_names.size()" << std::endl;
+      return false;
+    }
+  }
+
+  bool add_element(Element_with_map &element) {
+    Element elt;
+
+    std::vector<std::string>::const_iterator
+    it_subelement_name = m_subelement_names.begin();
+    std::vector<std::string>::const_iterator
+    it_subelement_name_end = m_subelement_names.end();
+    for (; it_subelement_name != it_subelement_name_end; ++it_subelement_name) {
+      elt.push_back(element[*it_subelement_name]);
+    }
+
+    return add_element(elt);
+  }
+
+  bool export_to_xml(const std::string &filename) const {
+    std::ofstream xmlfile;
+    xmlfile.open(filename.c_str());
+    xmlfile << "<?xml version='1.0'?>" << std::endl;
+    xmlfile << "<" << m_list_name << ">" << std::endl;
+
+    typename List_of_elements::const_iterator it_element = m_list_of_elements.begin();
+    typename List_of_elements::const_iterator it_element_end = m_list_of_elements.end();
+    for (int id = 1; it_element != it_element_end; ++it_element, ++id) {
+      xmlfile << "  <" << m_element_name << ">" << std::endl;
+      std::vector<std::string>::const_iterator
+      it_subelement_name = m_subelement_names.begin();
+      std::vector<std::string>::const_iterator
+      it_subelement_name_end = m_subelement_names.end();
+
+      if (m_add_timestamp)
+        xmlfile << "    <id> " << time(NULL) << " </id>" << std::endl;
+
+      for (int i = 0;
+           it_subelement_name != it_subelement_name_end;
+           ++it_subelement_name, ++i) {
+        xmlfile
+            << "    <" << *it_subelement_name << "> "
+            << (*it_element)[i]
+            << " </" << *it_subelement_name << ">" << std::endl;
+      }
+      xmlfile << "  </" << m_element_name << ">" << std::endl;
+    }
+
+    xmlfile << "</" << m_list_name << ">" << std::endl;
+    xmlfile.close();
+    return 0;
+
+  }
+
+ protected:
+  std::string m_list_name;
+  std::string m_element_name;
+  std::vector<std::string> m_subelement_names;
+  List_of_elements m_list_of_elements;
+  bool m_add_timestamp;
+};
+
+template<typename value_type = std::string>
+class Streaming_XML_exporter {
+ public:
+  typedef value_type Value_type;
+  typedef std::vector<value_type> Element;
+  typedef std::map<std::string, value_type> Element_with_map;
+  typedef std::vector<Element> List_of_elements;
+
+  Streaming_XML_exporter(
+                         const std::string &filename,
+                         const std::string &list_name,
+                         const std::string &element_name,
+                         const std::vector<std::string> &subelement_names,
+                         bool add_timestamp = true)
+      : m_list_name(list_name),
+      m_element_name(element_name),
+      m_subelement_names(subelement_names),
+      m_add_timestamp(add_timestamp) {
+    m_xml_fstream.open(filename.c_str());
+    if (m_xml_fstream.good()) {
+      m_xml_fstream << "<?xml version='1.0'?>" << std::endl;
+      m_xml_fstream << "<" << m_list_name << ">" << std::endl;
+    } else {
+      std::cerr << "Could not open file '" << filename << "'." << std::endl;
+    }
+  }
+
+  virtual ~Streaming_XML_exporter() {
+    close_file();
+  }
+
+  void close_file() {
+    m_xml_fstream.close();
+  }
+
+  bool add_element(const Element &element) {
+    if (element.size() == m_subelement_names.size()) {
+      m_xml_fstream << "  <" << m_element_name << ">" << std::endl;
+      std::vector<std::string>::const_iterator
+      it_subelement_name = m_subelement_names.begin();
+      std::vector<std::string>::const_iterator
+      it_subelement_name_end = m_subelement_names.end();
+
+      if (m_add_timestamp) {
+        m_xml_fstream << "    <id> " << time(NULL) << " </id>" << std::endl;
+      }
+
+      for (int i = 0;
+           it_subelement_name != it_subelement_name_end;
+           ++it_subelement_name, ++i) {
+        m_xml_fstream
+            << "    <" << *it_subelement_name << "> "
+            << element[i]
+            << " </" << *it_subelement_name << ">" << std::endl;
+      }
+      m_xml_fstream << "  </" << m_element_name << ">" << std::endl;
+
+      // Save current pointer position
+      std::ofstream::streampos pos = m_xml_fstream.tellp();
+      // Close the XML file (temporarily) so that the XML file is always correct
+      m_xml_fstream << "</" << m_list_name << ">" << std::endl;
+      // Restore the pointer position so that the next "add_element" will overwrite
+      // the end of the file
+      m_xml_fstream.seekp(pos);
+
+      m_xml_fstream.flush();
+      return true;
+    } else {
+      std::cerr << "ERROR: element.size() == m_subelement_names.size()" << std::endl;
+      return false;
+    }
+  }
+
+  bool add_element(Element_with_map &element) {
+    Element elt;
+
+    std::vector<std::string>::const_iterator
+    it_subelement_name = m_subelement_names.begin();
+    std::vector<std::string>::const_iterator
+    it_subelement_name_end = m_subelement_names.end();
+    for (; it_subelement_name != it_subelement_name_end; ++it_subelement_name) {
+      elt.push_back(element[*it_subelement_name]);
+    }
+
+    return add_element(elt);
+  }
+
+ protected:
+  std::ofstream m_xml_fstream;
+  std::string m_list_name;
+  std::string m_element_name;
+  std::vector<std::string> m_subelement_names;
+  bool m_add_timestamp;
+};