/*
* Multiply b by t
* - t must be defined in table and be a bitvector term of same bitsize as b
* - b->ptbl must be the same as table->pprods
*/
void bvarith_buffer_mul_term(bvarith_buffer_t *b, term_table_t *table, term_t t) {
pprod_t **v;
bvpoly_t *p;
int32_t i;
assert(b->ptbl == table->pprods);
assert(pos_term(t) && good_term(table, t) && is_bitvector_term(table, t) &&
term_bitsize(table, t) == b->bitsize);
i = index_of(t);
switch (table->kind[i]) {
case POWER_PRODUCT:
bvarith_buffer_mul_pp(b, pprod_for_idx(table, i));
break;
case BV_CONSTANT:
bvarith_buffer_mul_const(b, bvconst_for_idx(table, i)->data);
break;
case BV_POLY:
p = bvpoly_for_idx(table, i);
v = pprods_for_bvpoly(table, p);
bvarith_buffer_mul_bvpoly(b, p, v);
term_table_reset_pbuffer(table);
break;
default:
bvarith_buffer_mul_var(b, t);
break;
}
}
/*
* Tests: one buffer
* - n = bitsize
*/
static void test1(uint32_t n) {
bvarith_buffer_t buffer;
uint32_t q0[4];
assert(0 < n && n <= 128);
init_bvarith_buffer(&buffer, &prod_table, &store);
bvarith_buffer_prepare(&buffer, n);
printf("Empty buffer\n");
test_buffer(&buffer);
printf("x_0 + x_1\n");
bvarith_buffer_add_var(&buffer, 0);
bvarith_buffer_add_var(&buffer, 1);
bvarith_buffer_normalize(&buffer);
test_buffer(&buffer);
printf("After reset\n");
bvarith_buffer_prepare(&buffer, n);
test_buffer(&buffer);
printf("x_2 - x_0\n");
bvarith_buffer_add_var(&buffer, 2);
bvarith_buffer_sub_var(&buffer, 0);
bvarith_buffer_normalize(&buffer);
test_buffer(&buffer);
printf("x_2 - x_0 + x_1 + x_0\n");
bvarith_buffer_prepare(&buffer, n);
bvarith_buffer_add_var(&buffer, 2);
bvarith_buffer_sub_var(&buffer, 0);
bvarith_buffer_add_var(&buffer, 1);
bvarith_buffer_add_var(&buffer, 0);
bvarith_buffer_normalize(&buffer);
test_buffer(&buffer);
printf("Adding 3\n");
bvconst_set32(q0, 4, 3);
bvarith_buffer_add_const(&buffer, q0);
bvarith_buffer_normalize(&buffer);
test_buffer(&buffer);
printf("Negating\n");
bvarith_buffer_negate(&buffer);
bvarith_buffer_normalize(&buffer);
test_buffer(&buffer);
printf("Negating again\n");
bvarith_buffer_negate(&buffer);
bvarith_buffer_normalize(&buffer);
test_buffer(&buffer);
printf("Multiplying by 2 x_4\n");
bvconst_set32(q0, 4, 2);
bvarith_buffer_mul_varmono(&buffer, q0, 4);
bvarith_buffer_normalize(&buffer);
test_buffer(&buffer);
printf("Multiplying by x_1^2\n");
bvarith_buffer_mul_var(&buffer, 1);
bvarith_buffer_mul_var(&buffer, 1);
bvarith_buffer_normalize(&buffer);
test_buffer(&buffer);
printf("Multiplying by 0\n");
bvconst_clear(q0, 4);
bvarith_buffer_mul_const(&buffer, q0);
bvarith_buffer_normalize(&buffer);
test_buffer(&buffer);
printf("x_1 + 1 - x_2\n");
bvarith_buffer_prepare(&buffer, n);
bvarith_buffer_add_var(&buffer, 1);
bvconst_set32(q0, 4, 1);
bvarith_buffer_add_const(&buffer, q0);
bvarith_buffer_sub_var(&buffer, 2);
bvarith_buffer_normalize(&buffer);
test_buffer(&buffer);
printf("Squaring\n");
bvarith_buffer_square(&buffer);
bvarith_buffer_normalize(&buffer);
test_buffer(&buffer);
printf("Squaring\n");
bvarith_buffer_square(&buffer);
bvarith_buffer_normalize(&buffer);
test_buffer(&buffer);
printf("Squaring\n");
bvarith_buffer_square(&buffer);
bvarith_buffer_normalize(&buffer);
test_buffer(&buffer);
delete_bvarith_buffer(&buffer);
}
int main(void) {
thvar_t x, y, z;
init_bvconstants();
init_bv_vartable(&vtbl);
init_bvexp_table(&table, &vtbl);
bvexp_init_buffer64(&table, &b1);
bvexp_init_buffer64(&table, &b2);
bvexp_init_buffer(&table, &c1);
bvexp_init_buffer(&table, &c2);
printf("=== Initial table ===\n");
print_bvexp_table(stdout, &table);
printf("\n");
x = make_bvvar(&vtbl, 10);
y = make_bvvar(&vtbl, 10);
z = make_bvvar(&vtbl, 10);
// 2 + x + y
bvarith64_buffer_prepare(&b1, 10);
bvarith64_buffer_add_const(&b1, 2);
bvarith64_buffer_add_var(&b1, x);
bvarith64_buffer_add_var(&b1, y);
bvarith64_buffer_normalize(&b1);
test_buffer64(&table, &b1);
bvarith64_buffer_prepare(&b1, 10);
bvarith64_buffer_add_const(&b1, 2);
bvarith64_buffer_add_var(&b1, x);
bvarith64_buffer_add_var(&b1, y);
bvarith64_buffer_normalize(&b1);
test_buffer64(&table, &b1);
// (x + z) * (y - z)
bvarith64_buffer_prepare(&b1, 10);
bvarith64_buffer_add_var(&b1, x);
bvarith64_buffer_add_var(&b1, z);
bvarith64_buffer_prepare(&b2, 10);
bvarith64_buffer_add_var(&b2, y);
bvarith64_buffer_sub_var(&b2, z);
bvarith64_buffer_mul_buffer(&b1, &b2);
bvarith64_buffer_normalize(&b1);
test_buffer64(&table, &b1);
bvarith64_buffer_prepare(&b1, 10);
bvarith64_buffer_add_var(&b1, x);
bvarith64_buffer_add_var(&b1, z);
bvarith64_buffer_prepare(&b2, 10);
bvarith64_buffer_add_var(&b2, y);
bvarith64_buffer_sub_var(&b2, z);
bvarith64_buffer_mul_buffer(&b1, &b2);
bvarith64_buffer_normalize(&b1);
test_buffer64(&table, &b1);
// x * y * z^2
bvarith64_buffer_prepare(&b1, 10);
bvarith64_buffer_add_var(&b1, z);
bvarith64_buffer_square(&b1);
bvarith64_buffer_mul_var(&b1, y);
bvarith64_buffer_mul_var(&b1, x);
bvarith64_buffer_normalize(&b1);
test_buffer64(&table, &b1);
bvarith64_buffer_prepare(&b1, 10);
bvarith64_buffer_add_var(&b1, z);
bvarith64_buffer_square(&b1);
bvarith64_buffer_mul_var(&b1, y);
bvarith64_buffer_mul_var(&b1, x);
bvarith64_buffer_normalize(&b1);
test_buffer64(&table, &b1);
// Large coefficients
x = make_bvvar(&vtbl, 100);
y = make_bvvar(&vtbl, 100);
z = make_bvvar(&vtbl, 100);
// 1 + x + y
bvarith_buffer_prepare(&c1, 100);
bvarith_buffer_set_one(&c1);
bvarith_buffer_add_var(&c1, x);
bvarith_buffer_add_var(&c1, y);
bvarith_buffer_normalize(&c1);
test_buffer(&table, &c1);
bvarith_buffer_prepare(&c1, 100);
bvarith_buffer_set_one(&c1);
bvarith_buffer_add_var(&c1, x);
bvarith_buffer_add_var(&c1, y);
bvarith_buffer_normalize(&c1);
test_buffer(&table, &c1);
// (x + z) * (y - z)
bvarith_buffer_prepare(&c1, 100);
bvarith_buffer_add_var(&c1, x);
bvarith_buffer_add_var(&c1, z);
bvarith_buffer_prepare(&c2, 100);
bvarith_buffer_add_var(&c2, y);
bvarith_buffer_sub_var(&c2, z);
bvarith_buffer_mul_buffer(&c1, &c2);
bvarith_buffer_normalize(&c1);
test_buffer(&table, &c1);
bvarith_buffer_prepare(&c1, 100);
bvarith_buffer_add_var(&c1, x);
bvarith_buffer_add_var(&c1, z);
bvarith_buffer_prepare(&c2, 100);
bvarith_buffer_add_var(&c2, y);
bvarith_buffer_sub_var(&c2, z);
bvarith_buffer_mul_buffer(&c1, &c2);
bvarith_buffer_normalize(&c1);
test_buffer(&table, &c1);
// x * y * z^2
bvarith_buffer_prepare(&c1, 100);
bvarith_buffer_add_var(&c1, z);
bvarith_buffer_square(&c1);
bvarith_buffer_mul_var(&c1, y);
bvarith_buffer_mul_var(&c1, x);
bvarith_buffer_normalize(&c1);
test_buffer(&table, &c1);
bvarith_buffer_prepare(&c1, 100);
bvarith_buffer_add_var(&c1, z);
bvarith_buffer_square(&c1);
bvarith_buffer_mul_var(&c1, y);
bvarith_buffer_mul_var(&c1, x);
bvarith_buffer_normalize(&c1);
test_buffer(&table, &c1);
printf("=== Final table ===\n");
print_bvexp_table(stdout, &table);
printf("\n");
// remove two variables
bvexp_table_remove_vars(&table, 11);
printf("=== After removing two variables ===\n");
print_bvexp_table(stdout, &table);
printf("\n");
// recheck: 1 + x + y
bvarith_buffer_prepare(&c1, 100);
bvarith_buffer_set_one(&c1);
bvarith_buffer_add_var(&c1, x);
bvarith_buffer_add_var(&c1, y);
bvarith_buffer_normalize(&c1);
test_buffer(&table, &c1);
// recheck: (x + z) * (y - z)
bvarith_buffer_prepare(&c1, 100);
bvarith_buffer_add_var(&c1, x);
bvarith_buffer_add_var(&c1, z);
bvarith_buffer_prepare(&c2, 100);
bvarith_buffer_add_var(&c2, y);
bvarith_buffer_sub_var(&c2, z);
bvarith_buffer_mul_buffer(&c1, &c2);
bvarith_buffer_normalize(&c1);
test_buffer(&table, &c1);
bvarith_buffer_prepare(&c1, 100);
bvarith_buffer_add_var(&c1, x);
bvarith_buffer_add_var(&c1, z);
bvarith_buffer_prepare(&c2, 100);
bvarith_buffer_add_var(&c2, y);
bvarith_buffer_sub_var(&c2, z);
bvarith_buffer_mul_buffer(&c1, &c2);
bvarith_buffer_normalize(&c1);
test_buffer(&table, &c1);
printf("=== Final table ===\n");
print_bvexp_table(stdout, &table);
printf("\n");
// cleanup
delete_bvarith64_buffer(&b1);
delete_bvarith64_buffer(&b2);
delete_bvarith_buffer(&c1);
delete_bvarith_buffer(&c2);
delete_bvexp_table(&table);
delete_bv_vartable(&vtbl);
cleanup_bvconstants();
return 0;
}
