/*
* Test binary operations with b1 and b2
*/
static void test_ops(bvarith_buffer_t *b1, bvarith_buffer_t *b2) {
bvarith_buffer_t b;
uint32_t n;
assert(b1->bitsize == b2->bitsize);
printf("b1: ");
print_bvarith_buffer(stdout, b1);
printf("\nb2: ");
print_bvarith_buffer(stdout, b2);
printf("\n");
printf("Equality test: ");
if (bvarith_buffer_equal(b1, b2)) {
printf("yes\n");
} else {
printf("no\n");
}
n = b1->bitsize;
init_bvarith_buffer(&b, &prod_table, &store);
bvarith_buffer_prepare(&b, n);
bvarith_buffer_add_buffer(&b, b1);
bvarith_buffer_add_buffer(&b, b2);
bvarith_buffer_normalize(&b);
printf(" b1 + b2: ");
print_bvarith_buffer(stdout, &b);
printf("\n");
bvarith_buffer_prepare(&b, n);
bvarith_buffer_add_buffer(&b, b1);
bvarith_buffer_sub_buffer(&b, b2);
bvarith_buffer_normalize(&b);
printf(" b1 - b2: ");
print_bvarith_buffer(stdout, &b);
printf("\n");
bvarith_buffer_prepare(&b, n);
bvarith_buffer_add_buffer(&b, b2);
bvarith_buffer_sub_buffer(&b, b1);
bvarith_buffer_normalize(&b);
printf(" b2 - b1: ");
print_bvarith_buffer(stdout, &b);
printf("\n");
bvarith_buffer_prepare(&b, n);
bvarith_buffer_add_buffer(&b, b1);
bvarith_buffer_mul_buffer(&b, b2);
bvarith_buffer_normalize(&b);
printf(" b1 * b2: ");
print_bvarith_buffer(stdout, &b);
printf("\n");
bvarith_buffer_prepare(&b, n);
bvarith_buffer_add_buffer(&b, b2);
bvarith_buffer_mul_buffer(&b, b1);
bvarith_buffer_normalize(&b);
printf(" b2 * b1: ");
print_bvarith_buffer(stdout, &b);
printf("\n");
bvarith_buffer_prepare(&b, n);
bvarith_buffer_add_buffer_times_buffer(&b, b1, b2);
bvarith_buffer_normalize(&b);
printf(" b1 * b2: ");
print_bvarith_buffer(stdout, &b);
printf("\n");
bvarith_buffer_prepare(&b, n);
bvarith_buffer_sub_buffer_times_buffer(&b, b1, b2);
bvarith_buffer_normalize(&b);
printf("- b1 * b2: ");
print_bvarith_buffer(stdout, &b);
printf("\n");
delete_bvarith_buffer(&b);
printf("----\n");
}
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;
}
