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

105 lines
2.8 KiB
C++

#include "expr_rand.h"
#include "ast_pp.h"
#include "bv_decl_plugin.h"
#include "array_decl_plugin.h"
#include "arith_decl_plugin.h"
#include "ast_smt_pp.h"
#include <iostream>
#include <sstream>
static unsigned rand_seed = 1;
void tst_expr_arith(unsigned num_files) {
ast_manager m;
m.register_decl_plugins();
expr_rand er(m);
er.seed(rand_seed);
er.initialize_arith(20);
family_id fid = m.get_family_id("arith");
sort* int_ty = m.mk_sort(fid, INT_SORT, 0, 0);
sort* real_ty = m.mk_sort(fid, REAL_SORT, 0, 0);
er.initialize_array(3, int_ty, int_ty);
er.initialize_array(3, int_ty, real_ty);
er.initialize_basic(20);
for (unsigned i = 0; i < num_files; ++i) {
expr_ref e(m);
er.get_next(m.mk_bool_sort(), e);
ast_smt_pp pp(m);
pp.set_logic("QF_AUFLIA");
std::ostringstream buffer;
buffer << "random_arith_" << i << ".smt";
std::cout << buffer.str() << "\n";
std::ofstream file(buffer.str().c_str());
pp.display(file, e.get());
file.close();
}
}
void tst_expr_rand(unsigned num_files) {
ast_manager m;
m.register_plugin(symbol("bv"), alloc(bv_decl_plugin));
m.register_plugin(symbol("array"), alloc(array_decl_plugin));
expr_rand er(m);
er.initialize_bv(20);
er.seed(rand_seed);
parameter p1(1);
parameter p2(2);
parameter p8(8);
parameter p32(32);
family_id bvfid = m.get_family_id("bv");
sort* bv1 = m.mk_sort(bvfid, BV_SORT, 1, &p1);
sort* bv2 = m.mk_sort(bvfid, BV_SORT, 1, &p2);
sort* bv8 = m.mk_sort(bvfid, BV_SORT, 1, &p8);
sort* bv32 = m.mk_sort(bvfid, BV_SORT, 1, &p32);
er.initialize_array(3, bv8, bv32);
er.initialize_array(3, bv1, bv1);
er.initialize_array(3, bv1, bv2);
er.initialize_array(3, bv2, bv1);
er.initialize_array(3, bv2, bv2);
er.initialize_array(3, bv8, bv8);
er.initialize_array(3, bv32, bv32);
er.initialize_basic(20);
for (unsigned i = 0; i < num_files; ++i) {
expr_ref e(m);
er.get_next(m.mk_bool_sort(), e);
ast_smt_pp pp(m);
pp.set_logic("QF_AUFBV");
std::ostringstream buffer;
buffer << "random_bv_" << i << ".smt";
std::cout << buffer.str() << "\n";
std::ofstream file(buffer.str().c_str());
pp.display(file, e.get());
file.close();
}
}
void tst_expr_rand(char** argv, int argc, int& i) {
if (i + 1 < argc) {
unsigned num_files = atol(argv[i+1]);
i += 1;
if (i + 1 < argc && 0 == strncmp(argv[i+1],"/rs:",3)) {
rand_seed = atol(argv[i+1]+4);
std::cout << "random seed:" << rand_seed << "\n";
i += 1;
}
if (i + 1 < argc && 0 == strcmp(argv[i+1],"/arith")) {
tst_expr_arith(num_files);
return;
}
tst_expr_rand(num_files);
}
}