Z3 sources

Signed-off-by: Leonardo de Moura <leonardo@microsoft.com>

lib/memory_manager.h

@@ -0,0 +1,105 @@
+/*++
+Copyright (c) 2006 Microsoft Corporation
+
+Module Name:
+
+    memory_manager.h
+
+Abstract:
+
+    Custom memory layer.
+
+Author:
+
+    Nikolaj Bjorner (nbjorner) 2007-07-24
+
+Revision History:
+
+--*/
+#ifndef _MEMORY_H_
+#define _MEMORY_H_
+
+#include<cstdlib>
+#include<ostream>
+#include"z3_exception.h"
+
+class out_of_memory_error : public z3_error {
+public:
+    out_of_memory_error();
+};
+
+class memory {
+public:
+    static bool is_out_of_memory();
+    static void initialize(size_t max_size);
+    static void set_high_watermark(size_t watermak);
+    static bool above_high_watermark();
+    static void set_max_size(size_t max_size);
+    static void finalize();
+    static void display_max_usage(std::ostream& os);
+    static void display_i_max_usage(std::ostream& os);
+    static void deallocate(void* p);
+    static void* allocate(size_t s);
+    static void deallocate(char const* file, int line, void* p);
+    static void* allocate(char const* file, int line, char const* obj, size_t s);
+    static unsigned long long get_allocation_size();
+    static unsigned long long get_max_used_memory();
+    // temporary hack to avoid out-of-memory crash in z3.exe
+    static void exit_when_out_of_memory(bool flag, char const * msg);
+};
+
+
+#if _DEBUG
+
+#define alloc(T,...) new (memory::allocate(__FILE__,__LINE__,#T, sizeof(T))) T(__VA_ARGS__)
+#define dealloc(_ptr_) deallocf(__FILE__,__LINE__,_ptr_)
+
+template<typename T>
+void deallocf(char const* file, int line, T * ptr) {
+    if (ptr == 0) return;
+    ptr->~T();
+    memory::deallocate(file, line, ptr);
+}
+
+#else 
+
+#define alloc(T,...) new (memory::allocate(sizeof(T))) T(__VA_ARGS__)
+
+template<typename T>
+void dealloc(T * ptr) {
+    if (ptr == 0) return;
+    ptr->~T();
+    memory::deallocate(ptr);
+}
+
+#endif
+
+template<typename T>
+T * alloc_vect(unsigned sz) {
+    T * r = static_cast<T*>(memory::allocate(sizeof(T) * sz));
+    T * curr = r;
+    for (unsigned i = 0; i < sz; i++, curr++) 
+        new (curr) T();
+    return r;
+}
+
+template<typename T>
+void dealloc_vect(T * ptr, unsigned sz) {
+    if (ptr == 0) return;
+    T * curr = ptr;
+    for (unsigned i = 0; i < sz; i++, curr++)
+        curr->~T();
+    memory::deallocate(ptr);
+}
+
+#define alloc_svect(T, sz) static_cast<T*>(memory::allocate(sizeof(T) * sz))
+
+template<typename T>
+void dealloc_svect(T * ptr) {
+    if (ptr == 0) return;
+    memory::deallocate(ptr);
+}
+
+
+#endif /* _MEMORY_H_ */
+