Fix compilation on windows

Windows has no getrusage(), so use GetProcessMemoryInfo() on Windows instead.

Author: hyperair

Makefile.am

@@ -10,7 +10,12 @@ fptree_SOURCES =				\
 	src/parser.hh				\
 	src/parser.cc				\
 	src/fptree.hh				\
-	src/fptree.cc
+	src/fptree.cc				\
+	src/meminfo.hh				\
+	src/meminfo.cc
 
 fptree_CXXFLAGS = -std=c++0x
 fptree_CPPFLAGS = -I$(top_srcdir)/src
+if WINDOWS
+fptree_LDADD = -lpsapi
+endif

configure.ac

@@ -3,6 +3,7 @@
 
 AC_INIT([fptree], [0.1], [[email protected]])
 AC_CONFIG_SRCDIR([src/transactions.hh])
+AC_CANONICAL_HOST
 
 AM_INIT_AUTOMAKE([foreign dist-zip])
 
@@ -12,5 +13,12 @@ AC_PROG_CXX
 
 AC_TYPE_SIZE_T
 
+case $host_os in
+    *mingw32*|*cygwin*)
+        is_windows=yes;
+esac
+
+AM_CONDITIONAL([WINDOWS], [test "$is_windows" = "yes"])
+
 AC_CONFIG_FILES([Makefile])
 AC_OUTPUT

src/main.cc

@@ -3,14 +3,13 @@
 #include <iostream>
 #include <fstream>
 #include <chrono>
-#include <sys/time.h>
-#include <sys/resource.h>
 
 #include <itemset.hh>
 #include <transactions.hh>
 #include <types.hh>
 #include <parser.hh>
 #include <fptree.hh>
+#include <meminfo.hh>
 
 int main (int argc, char **argv)
 {
@@ -39,20 +38,17 @@ int main (int argc, char **argv)
 
     std::string dump_filename = filename + ".fptree";
 
-    rusage res_usage;
-    getrusage (RUSAGE_SELF, &res_usage);
-
-    std::cout << "File:             " << filename             << std::endl
-              << "FPtree dump file: " << dump_filename        << std::endl
-              << "Threshold:        " << min_support          << std::endl
-              << "Number of nodes:  " << stats.size           << std::endl
-              << "Number of leaves: " << stats.n_leaves       << std::endl
-              << "Height:           " << stats.height         << std::endl
-              << "Minimum fanout:   " << stats.min_fanout     << std::endl
-              << "Average fanout:   " << stats.average_fanout << std::endl
-              << "Maximum fanout:   " << stats.max_fanout     << std::endl
-              << "Runtime:          " << duration << "s"      << std::endl
-              << "Max. memory:      " << res_usage.ru_maxrss << "kB"
+    std::cout << "File:             " << filename               << std::endl
+              << "FPtree dump file: " << dump_filename          << std::endl
+              << "Threshold:        " << min_support            << std::endl
+              << "Number of nodes:  " << stats.size             << std::endl
+              << "Number of leaves: " << stats.n_leaves         << std::endl
+              << "Height:           " << stats.height           << std::endl
+              << "Minimum fanout:   " << stats.min_fanout       << std::endl
+              << "Average fanout:   " << stats.average_fanout   << std::endl
+              << "Maximum fanout:   " << stats.max_fanout       << std::endl
+              << "Runtime:          " << duration << "s"        << std::endl
+              << "Max. memory:      " << fpt::get_peakmem_kb () << "kB"
               << std::endl;
 
     {

src/meminfo.cc

@@ -0,0 +1,29 @@
+#include <meminfo.hh>
+
+#ifdef __WIN32__
+#   define PSAPI_VERSION 1
+#   include <windows.h>
+#   include <psapi.h>
+
+long fpt::get_peakmem_kb ()
+{
+    PROCESS_MEMORY_COUNTERS pmc;
+
+    GetProcessMemoryInfo (GetCurrentProcess (), &pmc, sizeof (pmc));
+
+    return pmc.PeakWorkingSetSize / 1024;
+}
+
+#else
+#   include <sys/time.h>
+#   include <sys/resource.h>
+
+long fpt::get_peakmem_kb ()
+{
+    rusage res_usage;
+    getrusage (RUSAGE_SELF, &res_usage);
+
+    return res_usage.ru_maxrss;
+}
+
+#endif

src/meminfo.hh

@@ -0,0 +1,9 @@
+#ifndef MEMINFO_HH
+#define MEMINFO_HH
+
+namespace fpt
+{
+    long get_peakmem_kb ();
+}
+
+#endif  // MEMINFO_HH