1 files changed, 95 insertions, 0 deletions
diff --git a/Carpet/CarpetLib/src/limits.cc b/Carpet/CarpetLib/src/limits.cc
new file mode 100644
index 000000000..f3c5ed650
--- /dev/null
+++ b/Carpet/CarpetLib/src/limits.cc
@@ -0,0 +1,95 @@
+#include <cctk.h>
+#include <cctk_Parameters.h>
+
+#include <algorithm>
+#include <cassert>
+#include <iostream>
+#include <sys/resource.h>
+
+#include "defs.hh"
+
+#include "limits.hh"
+
+namespace CarpetLib {
+  
+  using namespace std;
+  
+  
+  
+  static
+  void
+  set_limit (int resource, char const * name, CCTK_INT value);
+  
+  static
+  ostream &
+  operator<< (ostream & s, struct rlimit const & limit);
+  
+  static
+  void
+  output (ostream & s, rlim_t const & value);
+  
+  
+  
+  void
+  set_system_limits ()
+  {
+    DECLARE_CCTK_PARAMETERS;
+    set_limit (RLIMIT_CORE, "core file size", max_core_size_MB);
+    set_limit (RLIMIT_AS,   "memory size",    max_memory_size_MB);
+  }
+  
+  
+  
+  void
+  set_limit (int const resource, char const * const name, CCTK_INT const value)
+  {
+    struct rlimit limit;
+    check (not getrlimit (resource, & limit));
+    
+    if (value == -2 ) {
+      // Only show limit
+      cout << "Current " << name << " limit: " << limit << endl;
+      return;
+    }
+    
+    cout << "Old " << name << " limit: " << limit << endl;
+    
+    if (value == -1) {
+      limit.rlim_cur = limit.rlim_max;
+    } else {
+      limit.rlim_cur = min ((rlim_t) value * 1024 * 1024, limit.rlim_max);
+    }
+    
+    check (not setrlimit (resource, & limit));
+    check (not getrlimit (resource, & limit));
+    
+    cout << "New " << name << " limit: " << limit << endl;
+  }
+  
+  
+  
+  static
+  ostream &
+  operator<< (ostream & s, struct rlimit const & limit)
+  {
+    s << "hard=";
+    output (s, limit.rlim_max);
+    s << ", soft=";
+    output (s, limit.rlim_cur);
+    return s;
+  }
+  
+  
+  
+  static
+  void
+  output (ostream & s, rlim_t const & value)
+  {
+    if (value == RLIM_INFINITY) {
+      s << "[unlimited]";
+    } else {
+      s << (value / CCTK_REAL (1024*1024)) << " MB";
+    }
+  }
+  
+} // namespace Carpet