/**
* \ingroup StringConversions
*
* Sets the rational function \c rop to the value specified by the
* null-terminated string \c s.
*
* This method has now already been somewhat improved and is not very tolerant
* in the handling of malformed input. It expects either legitimate input for
* an \c fmpz_poly_t element, or two such inputs separated by a <tt>/</tt>
* only, in which case it is also assumed that the second polynomial is
* non-zero.
*
* The rational function is brought into canonical form by calling
* #fmpz_poly_q_canonicalize() in this function.
*
* Returns \c 0 if the string represents a valid rational function and
* \c non-zero otherwise.
*/
int fmpz_poly_q_set_str(fmpz_poly_q_t rop, const char *s)
{
int ans, i, m;
size_t len;
char * numstr;
len = strlen(s);
for (m = 0; m < len; m++)
{
if (s[m] == '/')
break;
}
if (m == len)
{
ans = fmpz_poly_set_str(rop->num, s);
fmpz_poly_set_si(rop->den, 1);
return ans;
}
else
{
numstr = flint_malloc(m + 1);
if (!numstr)
{
flint_printf("Exception (fmpz_poly_q_set_str). Memory allocation failed.\n");
abort();
}
for (i = 0; i < m; i++)
numstr[i] = s[i];
numstr[i] = '\0';
ans = fmpz_poly_set_str(rop->num, numstr);
ans |= fmpz_poly_set_str(rop->den, s + (m + 1));
if (ans == 0)
fmpz_poly_q_canonicalise(rop);
else
fmpz_poly_q_zero(rop);
flint_free(numstr);
return ans;
}
}
void
arb_const_e_eval(arb_t s, slong prec)
{
hypgeom_t series;
arb_t t;
arb_init(t);
hypgeom_init(series);
fmpz_poly_set_str(series->A, "1 1");
fmpz_poly_set_str(series->B, "1 1");
fmpz_poly_set_str(series->P, "1 1");
fmpz_poly_set_str(series->Q, "2 0 1");
prec += FLINT_CLOG2(prec);
arb_hypgeom_infsum(s, t, series, prec, prec);
arb_div(s, s, t, prec);
hypgeom_clear(series);
arb_clear(t);
}
void
arb_const_catalan_eval(arb_t s, slong prec)
{
hypgeom_t series;
arb_t t;
arb_init(t);
hypgeom_init(series);
fmpz_poly_set_str(series->A, "3 19 56 40");
fmpz_poly_set_str(series->B, "1 1");
fmpz_poly_set_str(series->P, "5 0 0 0 32 -64");
fmpz_poly_set_str(series->Q, "5 9 96 352 512 256");
prec += FLINT_CLOG2(prec);
arb_hypgeom_infsum(s, t, series, prec, prec);
arb_mul_ui(t, t, 18, prec);
arb_div(s, s, t, prec);
hypgeom_clear(series);
arb_clear(t);
}
void
arb_const_log2_hypgeom_eval(arb_t s, slong prec)
{
hypgeom_t series;
arb_t t;
arb_init(t);
hypgeom_init(series);
fmpz_poly_set_str(series->A, "1 1");
fmpz_poly_set_str(series->B, "1 1");
fmpz_poly_set_str(series->P, "2 0 -1");
fmpz_poly_set_str(series->Q, "2 4 8");
prec += FLINT_CLOG2(prec);
arb_hypgeom_infsum(s, t, series, prec, prec);
arb_mul_ui(s, s, 3, prec);
arb_mul_2exp_si(t, t, 2);
arb_div(s, s, t, prec);
hypgeom_clear(series);
arb_clear(t);
}
int main()
{
printf("get_series....");
fflush(stdout);
{
fmpq_t s;
fmpz_holonomic_t re, de, ans;
fmpq_init(s);
fmpz_holonomic_init(re);
fmpz_holonomic_init(de);
fmpz_holonomic_init(ans);
fmpz_holonomic_fit_length(ans, 4);
fmpq_set_si(s, -7, 3);
/* exp(x) */
fmpz_holonomic_fun_set_exp(de);
fmpz_holonomic_get_series(re, de);
fmpz_poly_set_si(ans->coeffs + 0, -1);
fmpz_poly_set_si2(ans->coeffs + 1, 1, 1);
_fmpz_holonomic_set_length(ans, 2);
check(re, ans, de);
/* sin/cos(x) */
fmpz_holonomic_fun_set_sin_cos(de);
fmpz_holonomic_get_series(re, de);
fmpz_poly_set_str(ans->coeffs + 0, "1 1");
fmpz_poly_set_str(ans->coeffs + 1, "0");
fmpz_poly_set_str(ans->coeffs + 2, "3 2 3 1");
_fmpz_holonomic_set_length(ans, 3);
check(re, ans, de);
/* log(x) */
fmpz_holonomic_fun_set_log(de);
fmpz_holonomic_get_series(re, de);
fmpz_poly_set_si(ans->coeffs + 0, 1);
_fmpz_holonomic_set_length(ans, 1);
check(re, ans, de);
/* log(s+x) */
fmpz_holonomic_shift_fmpq(de, de, s);
fmpz_holonomic_get_series(re, de);
fmpz_poly_set_str(ans->coeffs + 0, "2 0 -3");
fmpz_poly_set_str(ans->coeffs + 1, "2 7 7");
_fmpz_holonomic_set_length(ans, 2);
check(re, ans, de);
/* atan(x) */
fmpz_holonomic_fun_set_atan(de);
fmpz_holonomic_get_series(re, de);
fmpz_poly_set_str(ans->coeffs + 0, "2 0 1");
fmpz_poly_set_str(ans->coeffs + 1, "0");
fmpz_poly_set_str(ans->coeffs + 2, "2 2 1");
_fmpz_holonomic_set_length(ans, 3);
check(re, ans, de);
/* atan(s+x) */
fmpz_holonomic_shift_fmpq(de, de, s);
fmpz_holonomic_get_series(re, de);
fmpz_poly_set_str(ans->coeffs + 0, "2 0 9");
fmpz_poly_set_str(ans->coeffs + 1, "2 -42 -42");
fmpz_poly_set_str(ans->coeffs + 2, "2 116 58");
_fmpz_holonomic_set_length(ans, 3);
check(re, ans, de);
/* erf(x) */
fmpz_holonomic_fun_set_erf(de);
fmpz_holonomic_get_series(re, de);
fmpz_poly_set_str(ans->coeffs + 0, "2 0 2");
fmpz_poly_set_str(ans->coeffs + 1, "0");
fmpz_poly_set_str(ans->coeffs + 2, "3 2 3 1");
_fmpz_holonomic_set_length(ans, 3);
check(re, ans, de);
/* erf(s+x) */
fmpz_holonomic_shift_fmpq(de, de, s);
fmpz_holonomic_get_series(re, de);
fmpz_poly_set_str(ans->coeffs + 0, "2 0 6");
fmpz_poly_set_str(ans->coeffs + 1, "2 -14 -14");
fmpz_poly_set_str(ans->coeffs + 2, "3 6 9 3");
_fmpz_holonomic_set_length(ans, 3);
check(re, ans, de);
/* x^s */
fmpz_holonomic_fun_set_pow_fmpq(de, s);
fmpz_holonomic_get_series(re, de);
fmpz_poly_set_si(ans->coeffs + 0, 1);
_fmpz_holonomic_set_length(ans, 1);
check(re, ans, de);
/* (x+s)^s */
fmpz_holonomic_shift_fmpq(de, de, s);
fmpz_holonomic_normalise_content(de);
fmpz_holonomic_get_series(re, de);
fmpz_poly_set_str(ans->coeffs + 0, "2 -7 -3");
fmpz_poly_set_str(ans->coeffs + 1, "2 7 7");
_fmpz_holonomic_set_length(ans, 2);
check(re, ans, de);
/* x^n */
fmpz_holonomic_fun_set_pow_fmpz(de, fmpq_numref(s));
fmpz_holonomic_get_series(re, de);
fmpz_poly_set_si(ans->coeffs + 0, 1);
_fmpz_holonomic_set_length(ans, 1);
check(re, ans, de);
/* (x+s)^n */
fmpz_holonomic_shift_fmpq(de, de, s);
fmpz_holonomic_normalise_content(de);
fmpz_holonomic_get_series(re, de);
fmpz_poly_set_str(ans->coeffs + 0, "2 -21 -3");
fmpz_poly_set_str(ans->coeffs + 1, "2 7 7");
_fmpz_holonomic_set_length(ans, 2);
check(re, ans, de);
fmpz_holonomic_clear(re);
fmpz_holonomic_clear(de);
fmpz_holonomic_clear(ans);
fmpq_clear(s);
}
flint_cleanup();
printf("PASS\n");
return EXIT_SUCCESS;
}
int
main(void)
{
int i, result;
FLINT_TEST_INIT(state);
flint_printf("resultant_modular_div....");
fflush(stdout);
/* Just one specific test */
{
fmpz_poly_t f, g;
fmpz_t a, b, div;
slong nbits;
fmpz_poly_init(f);
fmpz_poly_init(g);
fmpz_init(a);
fmpz_init(b);
fmpz_init(div);
fmpz_poly_set_str(f, "11 -15 -2 -2 17 0 0 6 0 -5 1 -1");
fmpz_poly_set_str(g, "9 2 1 1 1 1 1 0 -1 -2");
fmpz_set_str(div, "11", 10);
nbits = 42;
fmpz_poly_resultant_modular_div(a, f, g, div, nbits);
/* The result is -44081924855067 = -4007447714097 * 11
* We supply 11 and the missing divisor is less then 2^35 */
fmpz_set_str(b, "-4007447714097", 10);
result = (fmpz_equal(a, b));
if (!result)
{
flint_printf("FAIL:\n");
flint_printf("f(x) = "), fmpz_poly_print_pretty(f, "x"), flint_printf("\n\n");
flint_printf("g(x) = "), fmpz_poly_print_pretty(g, "x"), flint_printf("\n\n");
flint_printf("res(f, h)/div = "), fmpz_print(b), flint_printf("\n\n");
flint_printf("res_mod_div(f, h) = "), fmpz_print(a), flint_printf("\n\n");
flint_printf("divr = "), fmpz_print(div), flint_printf("\n\n");
flint_printf("bitsbound = %wd", nbits), flint_printf("\n\n");
abort();
}
fmpz_poly_clear(f);
fmpz_poly_clear(g);
fmpz_clear(a);
fmpz_clear(b);
fmpz_clear(div);
}
/* Check that R(fg, h) = R(f, h) R(g, h) */
for (i = 0; i < 100; i++)
{
fmpz_t a, b, c, d;
fmpz_poly_t f, g, h, p;
slong nbits;
fmpz_init(a);
fmpz_init(b);
fmpz_init(c);
fmpz_init(d);
fmpz_poly_init(f);
fmpz_poly_init(g);
fmpz_poly_init(h);
fmpz_poly_init(p);
fmpz_poly_randtest(f, state, n_randint(state, 50), 100);
fmpz_poly_randtest(g, state, n_randint(state, 50), 100);
fmpz_poly_randtest(h, state, n_randint(state, 50), 100);
fmpz_poly_resultant_modular(a, f, h);
fmpz_poly_resultant_modular(b, g, h);
if (fmpz_is_zero(b) || fmpz_is_zero(a))
{
fmpz_clear(b);
fmpz_clear(a);
fmpz_poly_clear(f);
fmpz_poly_clear(g);
fmpz_poly_clear(h);
continue;
}
fmpz_mul(c, a, b);
fmpz_poly_mul(p, f, g);
nbits = (slong)fmpz_bits(a) + 1; /* for sign */
fmpz_poly_resultant_modular_div(d, p, h, b, nbits);
result = (fmpz_equal(a, d));
if (!result)
{
flint_printf("FAIL:\n");
flint_printf("p(x) = "), fmpz_poly_print_pretty(p, "x"), flint_printf("\n\n");
flint_printf("h(x) = "), fmpz_poly_print_pretty(h, "x"), flint_printf("\n\n");
flint_printf("res(p, h) = "), fmpz_print(c), flint_printf("\n\n");
flint_printf("res(p, h) = "), fmpz_print(a), flint_printf(" * "), fmpz_print(b), flint_printf("\n\n");
flint_printf("supplied divisor = "), fmpz_print(b), flint_printf("\n\n");
flint_printf("result should be = "), fmpz_print(a), flint_printf("\n\n");
flint_printf("res(p, h)/div = "), fmpz_print(d), flint_printf("\n\n");
flint_printf("bitsbound for result = %wd", nbits), flint_printf("\n\n");
abort();
}
fmpz_clear(a);
fmpz_clear(b);
fmpz_clear(c);
fmpz_clear(d);
fmpz_poly_clear(f);
fmpz_poly_clear(g);
fmpz_poly_clear(h);
fmpz_poly_clear(p);
}
FLINT_TEST_CLEANUP(state);
flint_printf("PASS\n");
return 0;
}
int main(int argc, char *argv[])
{
int ans;
char *str, *strout;
fmpz_poly_t zpoly;
fmpz_poly_q_t qpoly1;
mpz_t mpzzero, mpzone, mpztwo;
mpq_t mpqzero, mpqone, mpqtwo, mpqtwoinv;
FLINT_TEST_INIT(state);
flint_printf("all... ");
fflush(stdout);
/* Accessing numerator and denominator ***********************************/
fmpz_poly_q_init(qpoly1);
fmpz_poly_q_set_str(qpoly1, "2 -1 1/2 0 1");
str = "2 -1 1";
strout = fmpz_poly_get_str(fmpz_poly_q_numref(qpoly1));
ans = !strcmp(str, strout);
if (!ans)
{
flint_printf("test_numref: failed\n");
flint_printf(" Expected \"%s\", got \"%s\"\n", str, strout);
flint_printf(" qpoly1 = \""), fmpz_poly_q_print(qpoly1), flint_printf("\"\n");
abort();
}
fmpz_poly_q_clear(qpoly1);
flint_free(strout);
fmpz_poly_q_init(qpoly1);
fmpz_poly_q_set_str(qpoly1, "2 -1 1/2 0 1");
str = "2 0 1";
strout = fmpz_poly_get_str(fmpz_poly_q_denref(qpoly1));
ans = !strcmp(str, strout);
if (!ans)
{
flint_printf("test_denref: failed\n");
flint_printf(" Expected \"%s\", got \"%s\"\n", str, strout);
abort();
}
fmpz_poly_q_clear(qpoly1);
flint_free(strout);
fmpz_poly_q_init(qpoly1);
fmpz_poly_init(zpoly);
fmpz_poly_q_set_str(qpoly1, "2 -1 1/2 0 1");
fmpz_poly_set(zpoly, fmpz_poly_q_numref(qpoly1));
str = "2 -1 1";
strout = fmpz_poly_get_str(zpoly);
ans = !strcmp(str, strout);
if (!ans)
{
flint_printf("test_get_num: failed\n");
flint_printf(" Expected \"%s\", got \"%s\"\n", str, strout);
abort();
}
fmpz_poly_q_clear(qpoly1);
fmpz_poly_clear(zpoly);
flint_free(strout);
fmpz_poly_q_init(qpoly1);
fmpz_poly_init(zpoly);
fmpz_poly_q_set_str(qpoly1, "2 -1 1/2 0 1");
fmpz_poly_set(zpoly, fmpz_poly_q_denref(qpoly1));
str = "2 0 1";
strout = fmpz_poly_get_str(zpoly);
ans = !strcmp(str, strout);
if (!ans)
{
flint_printf("test_get_den: failed\n");
flint_printf(" Expected \"%s\", got \"%s\"\n", str, strout);
abort();
}
fmpz_poly_q_clear(qpoly1);
fmpz_poly_clear(zpoly);
flint_free(strout);
fmpz_poly_q_init(qpoly1);
fmpz_poly_init(zpoly);
fmpz_poly_q_set_str(qpoly1, "1 1/1 1");
fmpz_poly_set_str(zpoly, "2 0 1");
fmpz_poly_set(fmpz_poly_q_numref(qpoly1), zpoly);
str = "2 0 1";
strout = fmpz_poly_get_str(fmpz_poly_q_numref(qpoly1));
ans = !strcmp(str, strout);
if (!ans)
{
flint_printf("test_set_num: failed\n");
flint_printf(" Expected \"%s\", got \"%s\"\n", str, strout);
abort();
}
fmpz_poly_q_clear(qpoly1);
fmpz_poly_clear(zpoly);
flint_free(strout);
fmpz_poly_q_init(qpoly1);
fmpz_poly_init(zpoly);
fmpz_poly_q_set_str(qpoly1, "1 1/1 1");
fmpz_poly_set_str(zpoly, "2 0 1");
fmpz_poly_set(fmpz_poly_q_denref(qpoly1), zpoly);
str = "2 0 1";
strout = fmpz_poly_get_str(fmpz_poly_q_denref(qpoly1));
ans = !strcmp(str, strout);
if (!ans)
{
flint_printf("test_set_den: failed\n");
flint_printf(" Expected \"%s\", got \"%s\"\n", str, strout);
abort();
}
fmpz_poly_q_clear(qpoly1);
fmpz_poly_clear(zpoly);
flint_free(strout);
/* Canonicalise **********************************************************/
fmpz_poly_q_init(qpoly1);
str = "2 -1 1/2 0 1";
fmpz_poly_q_set_str(qpoly1, str);
strout = fmpz_poly_q_get_str(qpoly1);
ans = !strcmp(str, strout);
if (!ans)
{
flint_printf("test_canonicalize: failed\n");
flint_printf(" Expected \"%s\", got \"%s\"\n", str, strout);
abort();
}
fmpz_poly_q_clear(qpoly1);
flint_free(strout);
fmpz_poly_q_init(qpoly1);
str = "2 -1 -1/2 0 1";
fmpz_poly_q_set_str(qpoly1, "2 1 1/2 0 -1");
strout = fmpz_poly_q_get_str(qpoly1);
ans = !strcmp("2 -1 -1/2 0 1", strout);
if (!ans)
{
flint_printf("test_canonicalize: failed\n");
flint_printf(" Expected \"%s\", got \"%s\"\n", str, strout);
abort();
}
flint_free(strout);
fmpz_poly_q_clear(qpoly1);
/* Initialization, memory management and basic operations ****************/
test_set("0", "0");
test_set("0/1 1", "0");
test_set("3 -1 0 1/2 0 1", "3 -1 0 1/2 0 1");
test_set("3 -1 0 1/2 1 1", "2 -1 1");
test_set_si(-1, "1 -1");
test_set_si(13, "1 13");
test_set_si(0, "0");
test_swap("3 -1 0 1/2 0 1", "1 2/1 3", "1 2/1 3", "3 -1 0 1/2 0 1");
test_zero("0", "0");
test_zero("0/1 1", "0");
test_zero("3 -1 0 1/2 0 1", "0");
test_neg("0", "0");
test_neg("1 1/1 2", "1 -1/1 2");
test_neg("3 -1 0 1/2 0 1", "3 1 0 -1/2 0 1");
test_inv("1 1/1 2", "1 2");
test_inv("3 -1 0 1/2 0 1", "2 0 1/3 -1 0 1");
test_inv("3 -1 0 -1/2 0 1", "2 0 -1/3 1 0 1");
test_inv_inplace("1 1/1 2", "1 2");
test_inv_inplace("3 -1 0 1/2 0 1", "2 0 1/3 -1 0 1");
test_inv_inplace("3 -1 0 -1/2 0 1", "2 0 -1/3 1 0 1");
test_is_zero("0", 1);
test_is_zero("0/1 1", 1);
test_is_zero("3 -1 0 1/2 0 1", 0);
test_is_zero("3 -1 0 1/2 1 1", 0);
test_is_one("0", 0);
test_is_one("0/1 1", 0);
test_is_one("1 1/1 1", 1);
test_is_one("2 1 1/2 1 1", 1);
test_is_one("3 -1 0 1/2 0 1", 0);
test_equal("1 1/1 2", "1 1/1 2", 1);
test_equal("1 1/1 2", "1 1/1 2", 1);
test_equal("3 -1 0 1/2 1 1", "2 -1 1", 1);
test_equal("3 -1 0 1/2 -1 1", "2 -1 1", 0);
/* Addition and subtraction **********************************************/
test_add("3 1 0 1/2 0 1", "2 0 -1/3 1 0 1", "5 1 0 1 0 1/4 0 1 0 1");
test_add("3 -1 0 1/2 1 1", "1 2/2 -1 1", "3 3 -2 1/2 -1 1");
test_add("0/2 1 1", "1 2/1 1", "1 2");
test_add("1 -3/1 4", "0/3 1 0 1", "1 -3/1 4");
test_add("2 1 1/1 1", "2 -1 1/1 1", "2 0 2");
test_add("2 1 1/2 0 1", "2 2 1/2 -1 1", "3 -1 2 2/3 0 -1 1");
test_add("2 -1 1/2 2 1", "3 4 4 1/2 1 1", "4 7 12 7 1/3 2 3 1");
test_add("2 1 1/2 -1 1", "2 1 1", "3 0 1 1/2 -1 1");
test_add("1 1/2 1 1", "2 0 1/2 1 1", "1 1");
test_add("2 1 1/3 4 -4 1", "1 1/2 -2 1", "2 -1 2/3 4 -4 1");
test_add("3 0 1 1/3 1 2 1", "2 0 -1/2 1 1", "0");
test_add("2 1 1/2 0 1", "2 -1 1/2 0 1", "1 2");
test_add("1 1/3 3 5 2", "1 1/3 6 7 2", "1 1/3 2 3 1");
test_add_in_place1("3 1 0 1/2 0 1", "2 0 -1/3 1 0 1", "5 1 0 1 0 1/4 0 1 0 1");
test_add_in_place1("3 -1 0 1/2 1 1", "1 2/2 -1 1", "3 3 -2 1/2 -1 1");
test_add_in_place1("0/2 1 1", "1 2/1 1", "1 2");
test_add_in_place1("1 -3/1 4", "0/3 1 0 1", "1 -3/1 4");
test_add_in_place1("2 1 1/1 1", "2 -1 1/1 1", "2 0 2");
test_add_in_place1("2 1 1/2 0 1", "2 2 1/2 -1 1", "3 -1 2 2/3 0 -1 1");
test_add_in_place1("2 -1 1/2 2 1", "3 4 4 1/2 1 1", "4 7 12 7 1/3 2 3 1");
test_add_in_place1("2 1 1/2 -1 1", "2 1 1", "3 0 1 1/2 -1 1");
test_add_in_place1("1 1/2 1 1", "2 0 1/2 1 1", "1 1");
test_add_in_place1("2 1 1/3 4 -4 1", "1 1/2 -2 1", "2 -1 2/3 4 -4 1");
test_add_in_place1("3 0 1 1/3 1 2 1", "2 0 -1/2 1 1", "0");
test_add_in_place1("2 1 1/2 0 1", "2 -1 1/2 0 1", "1 2");
test_add_in_place2("3 1 0 1/2 0 1", "2 0 -1/3 1 0 1", "5 1 0 1 0 1/4 0 1 0 1");
test_add_in_place2("3 -1 0 1/2 1 1", "1 2/2 -1 1", "3 3 -2 1/2 -1 1");
test_add_in_place2("0/2 1 1", "1 2/1 1", "1 2");
test_add_in_place2("1 -3/1 4", "0/3 1 0 1", "1 -3/1 4");
test_add_in_place2("2 1 1/1 1", "2 -1 1/1 1", "2 0 2");
test_add_in_place2("2 1 1/2 0 1", "2 2 1/2 -1 1", "3 -1 2 2/3 0 -1 1");
test_add_in_place2("2 -1 1/2 2 1", "3 4 4 1/2 1 1", "4 7 12 7 1/3 2 3 1");
test_add_in_place2("2 1 1/2 -1 1", "2 1 1", "3 0 1 1/2 -1 1");
test_add_in_place2("1 1/2 1 1", "2 0 1/2 1 1", "1 1");
test_add_in_place2("2 1 1/3 4 -4 1", "1 1/2 -2 1", "2 -1 2/3 4 -4 1");
test_add_in_place2("3 0 1 1/3 1 2 1", "2 0 -1/2 1 1", "0");
test_add_in_place2("2 1 1/2 0 1", "2 -1 1/2 0 1", "1 2");
test_add_in_place3("2 1 1", "2 2 2");
test_add_in_place3("2 1 1/1 2", "2 1 1");
test_sub("3 1 0 1/2 0 1", "2 0 -1/3 1 0 1", "5 1 0 3 0 1/4 0 1 0 1");
test_sub("3 -1 0 1/2 1 1", "1 2/2 -1 1", "3 -1 -2 1/2 -1 1");
test_sub("0/2 1 1", "1 2/1 1", "1 -2");
test_sub("1 -3/1 4", "0/3 1 0 1", "1 -3/1 4");
test_sub("2 1 1/1 1", "2 -1 1/1 1", "1 2");
test_sub("2 1 1/2 0 1", "2 2 1/2 -1 1", "2 -1 -2/3 0 -1 1");
test_sub("2 -1 1/2 2 1", "3 4 4 1/2 1 1", "4 -9 -12 -5 -1/3 2 3 1");
test_sub("2 -1 1/2 0 1", "1 1", "1 -1/2 0 1");
test_sub("3 1 0 1/2 0 1", "2 0 -1/3 1 0 1", "5 1 0 3 0 1/4 0 1 0 1");
test_sub("3 -1 0 1/2 1 1", "1 2/2 -1 1", "3 -1 -2 1/2 -1 1");
test_sub("0/2 1 1", "1 2/1 1", "1 -2");
test_sub("1 -3/1 4", "0/3 1 0 1", "1 -3/1 4");
test_sub("2 1 1/1 1", "2 -1 1/1 1", "1 2");
test_sub("2 1 1/2 0 1", "2 2 1/2 -1 1", "2 -1 -2/3 0 -1 1");
test_sub("2 -1 1/2 2 1", "3 4 4 1/2 1 1", "4 -9 -12 -5 -1/3 2 3 1");
test_sub("2 1 1/2 -1 1", "2 1 1", "3 2 1 -1/2 -1 1");
test_sub("1 1/2 1 1", "2 0 1/2 1 1", "2 1 -1/2 1 1");
test_sub("2 1 1/3 4 -4 1", "1 1/2 -2 1", "1 3/3 4 -4 1");
test_sub("3 0 1 1/3 1 2 1", "2 0 -1/2 1 1", "2 0 2/2 1 1");
test_sub("2 1 1/2 0 1", "2 -1 1/2 0 1", "1 2/2 0 1");
test_sub("1 1/3 3 5 2", "1 1/3 6 7 2", "1 1/4 6 13 9 2");
test_sub("2 1 1/2 0 2", "2 1 1/2 0 2", "0");
test_sub("2 -1 2/2 0 1", "2 -1 1/2 0 1", "1 1");
test_sub_in_place1("3 1 0 1/2 0 1", "2 0 -1/3 1 0 1", "5 1 0 3 0 1/4 0 1 0 1");
test_sub_in_place1("3 -1 0 1/2 1 1", "1 2/2 -1 1", "3 -1 -2 1/2 -1 1");
test_sub_in_place1("0/2 1 1", "1 2/1 1", "1 -2");
test_sub_in_place1("1 -3/1 4", "0/3 1 0 1", "1 -3/1 4");
test_sub_in_place1("2 1 1/1 1", "2 -1 1/1 1", "1 2");
test_sub_in_place1("2 1 1/2 0 1", "2 2 1/2 -1 1", "2 -1 -2/3 0 -1 1");
test_sub_in_place1("2 -1 1/2 2 1", "3 4 4 1/2 1 1", "4 -9 -12 -5 -1/3 2 3 1");
test_sub_in_place2("3 1 0 1/2 0 1", "2 0 -1/3 1 0 1", "5 1 0 3 0 1/4 0 1 0 1");
test_sub_in_place2("3 -1 0 1/2 1 1", "1 2/2 -1 1", "3 -1 -2 1/2 -1 1");
test_sub_in_place2("0/2 1 1", "1 2/1 1", "1 -2");
test_sub_in_place2("1 -3/1 4", "0/3 1 0 1", "1 -3/1 4");
test_sub_in_place2("2 1 1/1 1", "2 -1 1/1 1", "1 2");
test_sub_in_place2("2 1 1/2 0 1", "2 2 1/2 -1 1", "2 -1 -2/3 0 -1 1");
test_sub_in_place2("2 -1 1/2 2 1", "3 4 4 1/2 1 1", "4 -9 -12 -5 -1/3 2 3 1");
test_sub_in_place3("2 -1 1/2 2 1", "0");
test_addmul("1 1/2 0 2", "2 3 1/1 4", "3 1 0 1/4 -2 0 0 1", "5 -4 3 1 5 1/5 0 -8 0 0 4");
test_submul("1 1/2 0 2", "2 3 1/1 4", "3 1 0 1/4 -2 0 0 1", "5 -4 -3 -1 -1 -1/5 0 -8 0 0 4");
/* Scalar multiplication and devision ************************************/
flint_mpz_init_set_si(mpzzero, 0);
flint_mpz_init_set_si(mpzone, 1);
flint_mpz_init_set_si(mpztwo, 2);
mpq_init(mpqzero); flint_mpq_set_si(mpqzero, 0, 1);
mpq_init(mpqone); flint_mpq_set_si(mpqone, 1, 1);
mpq_init(mpqtwo); flint_mpq_set_si(mpqtwo, 2, 1);
mpq_init(mpqtwoinv); flint_mpq_set_si(mpqtwoinv, 1, 2);
test_scalar_mul_si("0", 1, "0");
test_scalar_mul_si("0", 0, "0");
test_scalar_mul_si("1 2", 0, "0");
test_scalar_mul_si("1 1/1 2", -2, "1 -1");
test_scalar_mul_si("2 1 1/2 -2 3", 5, "2 5 5/2 -2 3");
test_scalar_mul_si("2 1 1/2 -2 2", 3, "2 3 3/2 -2 2");
test_scalar_mul_mpz("0", mpzone, "0");
test_scalar_mul_mpz("0", mpzzero, "0");
test_scalar_mul_mpz("1 2", mpzzero, "0");
test_scalar_mul_mpz("1 1/1 2", mpztwo, "1 1");
test_scalar_mul_mpq("0", mpqone, "0");
test_scalar_mul_mpq("0", mpqzero, "0");
test_scalar_mul_mpq("1 2", mpqzero, "0");
test_scalar_mul_mpq("1 1/1 2", mpqtwo, "1 1");
test_scalar_mul_mpq("1 -2/1 1", mpqtwoinv, "1 -1");
test_scalar_div_si("0", 1, "0");
test_scalar_div_si("1 2", 2, "1 1");
test_scalar_div_si("1 1/1 2", -2, "1 -1/1 4");
test_scalar_div_si("3 -5 0 3/2 1 1", 2, "3 -5 0 3/2 2 2");
test_scalar_div_si("3 2 8 4/2 0 1", 3, "3 2 8 4/2 0 3");
test_scalar_div_si("3 2 8 4/2 0 1", -3, "3 -2 -8 -4/2 0 3");
test_scalar_div_si("3 -27 0 9/2 0 1", -3, "3 9 0 -3/2 0 1");
test_scalar_div_mpz("0", mpzone, "0");
test_scalar_div_mpz("1 2", mpztwo, "1 1");
test_scalar_div_mpz("1 1/1 2", mpztwo, "1 1/1 4");
test_scalar_div_mpq("0", mpqone, "0");
test_scalar_div_mpq("1 2", mpqone, "1 2");
test_scalar_div_mpq("1 1/1 2", mpqtwo, "1 1/1 4");
test_scalar_div_mpq("1 -2/1 1", mpqtwoinv, "1 -4");
mpz_clear(mpzzero);
mpz_clear(mpzone);
mpz_clear(mpztwo);
mpq_clear(mpqzero);
mpq_clear(mpqone);
mpq_clear(mpqtwo);
mpq_clear(mpqtwoinv);
/* Multiplication, division and powing *********************************/
test_mul("3 1 0 1/2 0 1", "2 0 -1/3 1 0 1", "1 -1");
test_mul("3 -1 0 1/2 1 1", "1 2/2 -1 1", "1 2");
test_mul("0/2 1 1", "1 2/1 1", "0");
test_mul("1 -3/1 4", "0/3 1 0 1", "0");
test_mul("2 1 1/1 1", "2 -1 1/1 1", "3 -1 0 1");
test_mul("2 1 1/2 0 1", "2 2 1/2 -1 1", "3 2 3 1/3 0 -1 1");
test_mul("2 -1 1/2 2 1", "3 4 4 1/2 1 1", "3 -2 1 1/2 1 1");
test_mul_in_place1("3 1 0 1/2 0 1", "2 0 -1/3 1 0 1", "1 -1");
test_mul_in_place1("3 -1 0 1/2 1 1", "1 2/2 -1 1", "1 2");
test_mul_in_place1("0/2 1 1", "1 2/1 1", "0");
test_mul_in_place1("1 -3/1 4", "0/3 1 0 1", "0");
test_mul_in_place1("2 1 1/1 1", "2 -1 1/1 1", "3 -1 0 1");
test_mul_in_place1("2 1 1/2 0 1", "2 2 1/2 -1 1", "3 2 3 1/3 0 -1 1");
test_mul_in_place1("2 -1 1/2 2 1", "3 4 4 1/2 1 1", "3 -2 1 1/2 1 1");
test_mul_in_place2("3 1 0 1/2 0 1", "2 0 -1/3 1 0 1", "1 -1");
test_mul_in_place2("3 -1 0 1/2 1 1", "1 2/2 -1 1", "1 2");
test_mul_in_place2("0/2 1 1", "1 2/1 1", "0");
test_mul_in_place2("1 -3/1 4", "0/3 1 0 1", "0");
test_mul_in_place2("2 1 1/1 1", "2 -1 1/1 1", "3 -1 0 1");
test_mul_in_place2("2 1 1/2 0 1", "2 2 1/2 -1 1", "3 2 3 1/3 0 -1 1");
test_mul_in_place2("2 -1 1/2 2 1", "3 4 4 1/2 1 1", "3 -2 1 1/2 1 1");
test_mul_in_place3("2 0 1/2 1 1", "3 0 0 1/3 1 2 1");
test_div("3 -1 0 1/1 2", "2 1 1/1 1", "2 -1 1/1 2");
test_div("0/2 1 1", "2 1 1/1 1", "0");
test_div("3 -1 0 1/1 4", "2 -1 -1/1 2", "2 1 -1/1 2");
test_div("2 1 1", "2 1 -1/2 1 -1", "2 1 1");
test_div("2 1 1/3 4 4 1", "2 -1 1/3 6 5 1", "3 3 4 1/3 -2 1 1");
test_div_in_place1("3 -1 0 1/1 2", "2 1 1/1 1", "2 -1 1/1 2");
test_div_in_place1("0/2 1 1", "2 1 1/1 1", "0");
test_div_in_place1("3 -1 0 1/1 4", "2 -1 -1/1 2", "2 1 -1/1 2");
test_div_in_place1("2 1 1", "2 1 -1/2 1 -1", "2 1 1");
test_div_in_place1("2 1 1/3 4 4 1", "2 -1 1/3 6 5 1", "3 3 4 1/3 -2 1 1");
test_div_in_place1("0", "1 2/2 3 5", "0");
test_div_in_place2("3 -1 0 1/1 2", "2 1 1/1 1", "2 -1 1/1 2");
test_div_in_place2("0/2 1 1", "2 1 1/1 1", "0");
test_div_in_place2("3 -1 0 1/1 4", "2 -1 -1/1 2", "2 1 -1/1 2");
test_div_in_place2("2 1 1", "2 1 -1/2 1 -1", "2 1 1");
test_div_in_place2("2 1 1/3 4 4 1", "2 -1 1/3 6 5 1", "3 3 4 1/3 -2 1 1");
test_div_in_place3("3 -1 0 1/1 2", "1 1");
test_pow("2 0 -1/1 2", 3, "4 0 0 0 -1/1 8");
test_pow("0", 0, "1 1");
test_pow("2 1 -1", 0, "1 1");
test_pow("2 1 1/2 0 1", 0, "1 1");
/* Derivative ************************************************************/
test_derivative("0", "0");
test_derivative("1 2", "0");
test_derivative("1 -1/1 2", "0");
test_derivative("2 0 1", "1 1");
test_derivative("3 1 0 1", "2 0 2");
test_derivative("1 1/2 0 1", "1 -1/3 0 0 1");
test_derivative("2 2 1/2 -1 1", "1 -3/3 1 -2 1");
test_derivative("2 0 1/3 1 2 1", "2 1 -1/4 1 3 3 1");
/* Bug which allowed constant factors */
test_derivative("3 5 1 -2/2 10 2", "3 0 -10 -1/3 25 10 1");
/* Evaluation ************************************************************/
test_evaluate("1 1/1 2", -2, 3, "1/2");
test_evaluate("3 1 0 1/2 0 1", -1, 2, "-5/2");
test_evaluate("2 3 1/2 -1 1", 1, 1, "P");
test_evaluate("2 3 1/2 -1 1", 2, 3, "-11");
test_evaluate("2 3 1/2 -1 2", 1, 2, "P");
test_evaluate("2 1 1/2 -1 1", 2, 1, "3");
/* String methods ********************************************************/
fmpz_poly_q_init(qpoly1);
ans = fmpz_poly_q_set_str(qpoly1, "1 3/xyz");
if ((ans == 0) || !fmpz_poly_q_is_zero(qpoly1))
{
flint_printf("test_set_str: failed\n");
abort();
}
fmpz_poly_q_clear(qpoly1);
fmpz_poly_q_init(qpoly1);
ans = fmpz_poly_q_set_str(qpoly1, "abc/1 3");
if ((ans == 0) || !fmpz_poly_q_is_zero(qpoly1))
{
flint_printf("test_set_str: failed\n");
abort();
}
fmpz_poly_q_clear(qpoly1);
fmpz_poly_q_init(qpoly1);
ans = fmpz_poly_q_set_str(qpoly1, "abc/xyz");
if ((ans == 0) || !fmpz_poly_q_is_zero(qpoly1))
{
flint_printf("test_set_str: failed\n");
abort();
}
fmpz_poly_q_clear(qpoly1);
test_get_str_pretty("1 -3", "-3");
test_get_str_pretty("3 1 2 1", "t^2+2*t+1");
test_get_str_pretty("1 -2/2 1 1", "-2/(t+1)");
test_get_str_pretty("2 1 1/2 -1 1", "(t+1)/(t-1)");
test_get_str_pretty("2 1 1/1 2", "(t+1)/2");
test_get_str_pretty("1 1/1 2", "1/2");
FLINT_TEST_CLEANUP(state);
flint_printf("PASS\n");
return EXIT_SUCCESS;
}
int main(int argc, char *args[])/* setup.txt, ct.txt{ct->lev, ct->row, ct->col, ct}, sk.txt{row, col, poly}*/
{
FILE *fp;
if((fp = fopen(args[1], "r")) == NULL)
{
printf("file read error\n");
exit(0);
}
fmpz_t tmp;
fmpz_init(tmp);
fgets(str, 100, fp);
fmpz_set_str(tmp, str, 10);
long lev, d, i, j;
lev = fmpz_get_si(tmp);
fgets(str, 100, fp);
fmpz_set_str(tmp, str, 10);
d = fmpz_get_si(tmp);
fmpz_poly_t fx;
fmpz_poly_init(fx);
fmpz_poly_set_coeff_si(fx, 0, 1);
fmpz_poly_set_coeff_si(fx, d, 1);
param_node_t *ph, *pr, *ps, *param;
ph = param_node_init(ph);
ps = ph;
for( i = 0 ; i <= lev; i++ ) {
pr = param_node_init(pr);
fgets(str, 100, fp);
fmpz_set_str(pr->q, str, 10);
fgets(str, 100, fp);
fmpz_set_str(tmp, str, 10);
pr->n = fmpz_get_si(tmp);
fgets(str, 100, fp);
fmpz_set_str(tmp, str, 10);
pr->bign = fmpz_get_si(tmp);
ps->next = pr;
ps = pr;
}
ps->next = NULL;
fclose(fp);
param = ph->next;
long row, col, ctlev;
if((fp = fopen(args[2], "r")) == NULL)
{
printf("file read error\n");
exit(0);
}
fgets(str, 30, fp);
fmpz_set_str(tmp, str,10);
ctlev = fmpz_get_si(tmp);
fgets(str, 30, fp);
fmpz_set_str(tmp, str,10);
row = fmpz_get_si(tmp);
fgets(str, 30, fp);
fmpz_set_str(tmp, str,10);
col = fmpz_get_si(tmp);
fmpz_poly_mat_t ct;
fmpz_poly_mat_init(ct, row, col);
for( i = 0 ; i < row ; i++) {
for(j = 0; j < col ; j++) {
fgets(str, 100000, fp);
fmpz_poly_set_str(fmpz_poly_mat_entry(ct, i, j), str);
}
}
fclose(fp);
if((fp = fopen(args[3], "r")) == NULL)
{
printf("file read error\n");
exit(0);
}
long l;
sk_node_t *sh, *ss, *sr;
sh = (sk_node_t *)malloc(sizeof(sk_node_t));
ss = sh;
for(l = 0; l <= lev ; l++){
sr = (sk_node_t *)malloc(sizeof(sk_node_t));
fgets(str, 30, fp);
fmpz_set_str(tmp, str,10);
row = fmpz_get_si(tmp);
fgets(str, 30, fp);
fmpz_set_str(tmp, str,10);
col = fmpz_get_si(tmp);
fmpz_poly_mat_init(sr->sk, row, col);
for( i = 0 ; i < row ; i++) {
for(j = 0; j < col ; j++) {
fgets(str, 100000, fp);
fmpz_poly_set_str(fmpz_poly_mat_entry(sr->sk, i, j), str);
}
}
ss->next = sr;
ss = sr;
}
ss->next = NULL;
fclose(fp);
sh = sh->next;
while(lev > ctlev){
sh = sh->next;
param = param->next;
lev--;
}
fmpz_poly_t ms;
fmpz_poly_init(ms);
e_decrypt(ms, param, sh->sk, ct, fx);
fmpz_poly_print(ms);
printf("\n");
return 0;
}
