set up buffer estimation for Linux

David Mayerich
1 parent 5e1b4869
Showing 1 changed file with 8 additions and 1 deletions Show diff stats
stim/envi/binary.h
@@ -7,9 +7,12 @@
 #include "../math/vector.h"
 #include <fstream>
 #include <sys/stat.h>
+#include <cstring>
 #ifdef _WIN32
 #include <Windows.h>
+#else
+#include <unistd.h>
 #endif
 namespace stim{
@@ -38,7 +41,7 @@ protected:
 	/// Private initialization function used to set default parameters in the data structure.
 	void init(){
-		memset(R, 0, sizeof(unsigned long long) * D);		//initialize the resolution to zero
+		std::memset(R, 0, sizeof(unsigned long long) * D);		//initialize the resolution to zero
 		header = 0;											//initialize the header size to zero
 		mask = NULL;
@@ -131,6 +134,10 @@ public:
 		statex.dwLength = sizeof (statex);
 		GlobalMemoryStatusEx (&statex);
 		buffer_size = (size_t)(statex.ullAvailPhys * mem_frac);
+#else
+		size_t pages = sysconf(_SC_PHYS_PAGES);
+		size_t page_size = sysconf(_SC_PAGE_SIZE);
+		buffer_size = (size_t)(pages * page_size *  mem_frac);
 #endif
 	}