3
0
Fork 0
mirror of https://github.com/Z3Prover/z3 synced 2025-04-11 11:43:36 +00:00
z3/test/object_allocator.cpp
Leonardo de Moura 68269c43a6 other components
Signed-off-by: Leonardo de Moura <leonardo@microsoft.com>
2012-10-02 11:48:48 -07:00

122 lines
3.2 KiB
C++

/*++
Copyright (c) 2006 Microsoft Corporation
Module Name:
object_allocator.cpp
Abstract:
<abstract>
Author:
Leonardo de Moura (leonardo) 2010-06-09.
Revision History:
--*/
#include"rational.h"
#include"object_allocator.h"
struct cell {
rational m_coeff;
unsigned m_i;
unsigned m_j;
cell * m_next_row;
cell * m_next_col;
public:
static unsigned g_num_allocated_cells;
static unsigned g_num_deallocated_cells;
static unsigned g_num_recycled_cells;
cell() {
g_num_allocated_cells++;
}
~cell() {
g_num_deallocated_cells++;
}
void reset() {
m_coeff.reset();
g_num_recycled_cells++;
}
};
unsigned cell::g_num_allocated_cells = 0;
unsigned cell::g_num_deallocated_cells = 0;
unsigned cell::g_num_recycled_cells = 0;
typedef object_allocator<cell, true, simple_reset_proc<cell> > cell_allocator;
static void tst1() {
cell_allocator m;
cell * c1 = m.allocate<true>();
cell * c2 = m.allocate<true>();
c1->m_coeff = rational(10);
m.recycle(c1);
cell * c3 = m.allocate<true>();
SASSERT(c3->m_coeff.is_zero());
}
static void tst2() {
cell_allocator m;
SASSERT(m.capacity() >= 2);
cell_allocator::worker_object_allocator m1 = m.get_worker_allocator(0);
cell_allocator::worker_object_allocator m2 = m.get_worker_allocator(1);
m.enable_concurrent(true);
vector<std::pair<cell *, int> > object_coeff_pairs;
unsigned num_resets = 0;
for (unsigned i = 0; i < 100000; i++) {
unsigned idx = rand() % 6;
if (idx < 4) {
cell * c;
if (idx < 2)
c = m1.allocate<true>();
else
c = m2.allocate<true>();
SASSERT(c->m_coeff.is_zero());
int val = rand();
c->m_coeff = rational(val);
object_coeff_pairs.push_back(std::make_pair(c, val));
}
else {
if (!object_coeff_pairs.empty()) {
unsigned idx = rand() % object_coeff_pairs.size();
cell * c = object_coeff_pairs[idx].first;
CTRACE("object_allocator", c->m_coeff != rational(object_coeff_pairs[idx].second), tout << c->m_coeff << " != " << rational(object_coeff_pairs[idx].second) << "\n";);
SASSERT(c->m_coeff == rational(object_coeff_pairs[idx].second));
if (idx < 5)
m1.recycle(c);
else
m2.recycle(c);
object_coeff_pairs.erase(object_coeff_pairs.begin() + idx);
}
}
if (rand() % 5000 == 0) {
m.enable_concurrent(false);
m.reset();
object_coeff_pairs.reset();
m.enable_concurrent(true);
num_resets++;
}
}
TRACE("object_allocator", tout << "num. resets: " << num_resets << "\n";);
}
void tst_object_allocator() {
tst1();
tst2();
TRACE("object_allocator", tout << "num. allocated cells: " << cell::g_num_allocated_cells << "\nnum. deallocated cells: " << cell::g_num_deallocated_cells <<
"\nnum. recycled cells: " << cell::g_num_recycled_cells << "\n";);
SASSERT(cell::g_num_allocated_cells == cell::g_num_deallocated_cells);
}