updating rendered version

Author: ChrisBarker-NOAA

.buildinfo

@@ -1,4 +1,4 @@
 # Sphinx build info version 1
 # This file hashes the configuration used when building these files. When it is not found, a full rebuild will be done.
-config: cbd7e031f47c4eeff9f0ceaf8bd09f7a
+config: 991000b9a46c5b7dfd2ded2cef37faea
 tags: 645f666f9bcd5a90fca523b33c5a78b7

_downloads/0c9b35f0d7e64ab95fd1077dd8c59150/mem_check.py

@@ -0,0 +1,109 @@
+"""
+mem_check.py
+
+functions for getting memoroy use of the current python process
+
+Windows and *nix versions
+
+USAGE:
+
+amount = get_mem_use(units='MB') # options are KB, MB, GB
+
+"""
+
+import sys
+
+div = {'GB': 1024*1024*1024,
+       'MB': 1024*1024,
+       'KB': 1024,
+       }
+
+if sys.platform.startswith('win'):
+
+    """
+
+    Functions for getting memory usage of Windows processes.
+
+    from:
+
+    http://code.activestate.com/recipes/578513-get-memory-usage-of-windows-processes-using-getpro/
+
+    get_mem_use(units='MB') is the one to get memory use for the current process.
+
+
+    """
+    import ctypes
+    from ctypes import wintypes
+
+    GetCurrentProcess = ctypes.windll.kernel32.GetCurrentProcess
+    GetCurrentProcess.argtypes = []
+    GetCurrentProcess.restype = wintypes.HANDLE
+
+    SIZE_T = ctypes.c_size_t
+
+    class PROCESS_MEMORY_COUNTERS_EX(ctypes.Structure):
+        _fields_ = [
+            ('cb', wintypes.DWORD),
+            ('PageFaultCount', wintypes.DWORD),
+            ('PeakWorkingSetSize', SIZE_T),
+            ('WorkingSetSize', SIZE_T),
+            ('QuotaPeakPagedPoolUsage', SIZE_T),
+            ('QuotaPagedPoolUsage', SIZE_T),
+            ('QuotaPeakNonPagedPoolUsage', SIZE_T),
+            ('QuotaNonPagedPoolUsage', SIZE_T),
+            ('PagefileUsage', SIZE_T),
+            ('PeakPagefileUsage', SIZE_T),
+            ('PrivateUsage', SIZE_T),
+        ]
+
+    GetProcessMemoryInfo = ctypes.windll.psapi.GetProcessMemoryInfo
+    GetProcessMemoryInfo.argtypes = [
+        wintypes.HANDLE,
+        ctypes.POINTER(PROCESS_MEMORY_COUNTERS_EX),
+        wintypes.DWORD,
+    ]
+    GetProcessMemoryInfo.restype = wintypes.BOOL
+
+    def get_current_process():
+        """Return handle to current process."""
+        return GetCurrentProcess()
+
+    def get_memory_info(process=None):
+        """Return Win32 process memory counters structure as a dict."""
+        if process is None:
+            process = get_current_process()
+        counters = PROCESS_MEMORY_COUNTERS_EX()
+        ret = GetProcessMemoryInfo(process, ctypes.byref(counters),
+                                   ctypes.sizeof(counters))
+        if not ret:
+            raise ctypes.WinError()
+        info = dict((name, getattr(counters, name))
+                    for name, _ in counters._fields_)
+        return info
+
+    def get_mem_use(units='MB'):
+        """
+        returns the total memory use of the current python process
+
+        :param units='MB': the units you want the reslut in. Options are:
+                           'GB', 'MB', 'KB'
+        """
+        info = get_memory_info()
+        return info['PrivateUsage'] / float(div[units])
+
+
+else: # for posix systems only tested on OS-X for now
+    def get_mem_use(units='MB'):
+        """
+        returns the total memory use of the current python process
+
+        :param units='MB': the units you want the reslut in. Options are:
+                           'GB', 'MB', 'KB'
+        """
+        import resource
+        #useage = resource.getrusage(resource.RUSAGE_SELF).ru_maxrss
+        div = {'GB': 1024*1024*1024,
+               'MB': 1024*1024,
+               'KB': 1024,
+               }
+        return resource.getrusage(resource.RUSAGE_SELF).ru_maxrss / float(div[units])