pqc/external/flint-2.4.3/fq_poly_factor_templates/test/t-factor.c
2014-05-24 23:16:06 +02:00

279 lines
9.0 KiB
C

/*=============================================================================
This file is part of FLINT.
FLINT is free software; you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation; either version 2 of the License, or
(at your option) any later version.
FLINT is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
You should have received a copy of the GNU General Public License
along with FLINT; if not, write to the Free Software
Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
=============================================================================*/
/******************************************************************************
Copyright (C) 2007 David Howden
Copyright (C) 2007, 2008, 2009, 2010 William Hart
Copyright (C) 2008 Richard Howell-Peak
Copyright (C) 2011 Fredrik Johansson
Copyright (C) 2013 Mike Hansen
******************************************************************************/
#ifdef T
#include "templates.h"
int
main(void)
{
int iter;
FLINT_TEST_INIT(state);
flint_printf("factor....");
fflush(stdout);
/* Default algorithm */
for (iter = 0; iter < flint_test_multiplier(); iter++)
{
int result = 1;
TEMPLATE(T, poly_t) pol1, poly, quot, rem, product;
TEMPLATE(T, poly_factor_t) res;
TEMPLATE(T, ctx_t) ctx;
TEMPLATE(T, t) lead;
slong length, num, i, j;
ulong exp[5], prod1;
TEMPLATE(T, ctx_randtest) (ctx, state);
TEMPLATE(T, poly_init) (pol1, ctx);
TEMPLATE(T, poly_init) (poly, ctx);
TEMPLATE(T, poly_init) (quot, ctx);
TEMPLATE(T, poly_init) (rem, ctx);
TEMPLATE(T, poly_zero) (pol1, ctx);
TEMPLATE(T, poly_one) (pol1, ctx);
length = n_randint(state, 4) + 2;
TEMPLATE(T, poly_randtest_irreducible) (poly, state, length, ctx);
exp[0] = n_randint(state, 3) + 1;
prod1 = exp[0];
for (i = 0; i < exp[0]; i++)
TEMPLATE(T, poly_mul) (pol1, pol1, poly, ctx);
num = n_randint(state, 3) + 1;
for (i = 1; i < num; i++)
{
do
{
length = n_randint(state, 3) + 2;
TEMPLATE(T, poly_randtest_irreducible) (poly, state, length,
ctx);
TEMPLATE(T, poly_divrem) (quot, rem, pol1, poly, ctx);
}
while ((poly->length < 2) || (rem->length == 0));
exp[i] = n_randint(state, 3) + 1;
prod1 *= exp[i];
for (j = 0; j < exp[i]; j++)
TEMPLATE(T, poly_mul) (pol1, pol1, poly, ctx);
}
TEMPLATE(T, poly_factor_init) (res, ctx);
TEMPLATE(T, init) (lead, ctx);
switch (n_randint(state, 4))
{
case 0:
TEMPLATE(T, poly_factor) (res, lead, pol1, ctx);
break;
case 1:
TEMPLATE(T, poly_factor_with_berlekamp) (res, lead, pol1, ctx);
break;
case 2:
if (fmpz_is_even(TEMPLATE(T, ctx_prime) (ctx)))
TEMPLATE(T, poly_factor) (res, lead, pol1, ctx);
else
TEMPLATE(T, poly_factor_with_cantor_zassenhaus) (res, lead,
pol1,
ctx);
break;
case 3:
TEMPLATE(T, poly_factor_with_kaltofen_shoup) (res, lead, pol1,
ctx);
break;
}
fflush(stdout);
result &= (res->num == num);
if (!result)
{
flint_printf("Error: number of factors incorrect, %wd, %wd\n",
res->num, num);
abort();
}
TEMPLATE(T, poly_init) (product, ctx);
TEMPLATE(T, poly_one) (product, ctx);
for (i = 0; i < res->num; i++)
for (j = 0; j < res->exp[i]; j++)
TEMPLATE(T, poly_mul) (product, product, res->poly + i, ctx);
TEMPLATE(T, TEMPLATE(poly_scalar_mul, T)) (product, product, lead,
ctx);
result &= TEMPLATE(T, poly_equal) (pol1, product, ctx);
if (!result)
{
flint_printf
("Error: product of factors does not equal original polynomial\n");
TEMPLATE(T, poly_print_pretty) (pol1, "x", ctx);
flint_printf("\n");
TEMPLATE(T, poly_print_pretty) (product, "x", ctx);
flint_printf("\n");
abort();
}
TEMPLATE(T, poly_clear) (product, ctx);
TEMPLATE(T, poly_clear) (quot, ctx);
TEMPLATE(T, poly_clear) (rem, ctx);
TEMPLATE(T, poly_clear) (pol1, ctx);
TEMPLATE(T, poly_clear) (poly, ctx);
TEMPLATE(T, poly_factor_clear) (res, ctx);
TEMPLATE(T, clear) (lead, ctx);
TEMPLATE(T, ctx_clear) (ctx);
}
/* Test deflation trick */
for (iter = 0; iter < flint_test_multiplier(); iter++)
{
TEMPLATE(T, poly_t) pol1, poly, quot, rem;
TEMPLATE(T, poly_factor_t) res, res2;
TEMPLATE(T, ctx_t) ctx;
TEMPLATE(T, t) lead;
slong length, num, i, j;
slong exp[5], prod1;
ulong inflation;
int found;
TEMPLATE(T, ctx_randtest) (ctx, state);
TEMPLATE(T, poly_init) (pol1, ctx);
TEMPLATE(T, poly_init) (poly, ctx);
TEMPLATE(T, poly_init) (quot, ctx);
TEMPLATE(T, poly_init) (rem, ctx);
TEMPLATE(T, poly_zero) (pol1, ctx);
TEMPLATE(T, poly_one) (pol1, ctx);
inflation = n_randint(state, 7) + 1;
length = n_randint(state, 4) + 2;
TEMPLATE(T, poly_randtest_irreducible) (poly, state, length, ctx);
TEMPLATE(T, poly_inflate) (poly, poly, inflation, ctx);
exp[0] = n_randint(state, 6) + 1;
prod1 = exp[0];
for (i = 0; i < exp[0]; i++)
TEMPLATE(T, poly_mul) (pol1, pol1, poly, ctx);
num = n_randint(state, 5) + 1;
for (i = 1; i < num; i++)
{
do
{
length = n_randint(state, 6) + 2;
TEMPLATE(T, poly_randtest_irreducible) (poly, state, length,
ctx);
TEMPLATE(T, poly_divrem) (quot, rem, pol1, poly, ctx);
}
while ((poly->length < 2) || (rem->length == 0));
exp[i] = n_randint(state, 6) + 1;
prod1 *= exp[i];
TEMPLATE(T, poly_inflate) (poly, poly, inflation, ctx);
for (j = 0; j < exp[i]; j++)
TEMPLATE(T, poly_mul) (pol1, pol1, poly, ctx);
}
TEMPLATE(T, poly_factor_init) (res, ctx);
TEMPLATE(T, poly_factor_init) (res2, ctx);
TEMPLATE(T, init) (lead, ctx);
switch (n_randint(state, 4))
{
case 0:
TEMPLATE(T, poly_factor) (res, lead, pol1, ctx);
break;
case 1:
TEMPLATE(T, poly_factor_with_berlekamp) (res, lead, pol1, ctx);
break;
case 2:
TEMPLATE(T, poly_factor_with_cantor_zassenhaus) (res, lead,
pol1, ctx);
break;
case 3:
TEMPLATE(T, poly_factor_with_kaltofen_shoup) (res, lead, pol1,
ctx);
break;
}
TEMPLATE(T, poly_factor_cantor_zassenhaus) (res2, pol1, ctx);
if (res->num != res2->num)
{
flint_printf("FAIL: different number of factors found\n");
abort();
}
for (i = 0; i < res->num; i++)
{
found = 0;
for (j = 0; j < res2->num; j++)
{
if (TEMPLATE(T, poly_equal)
(res->poly + i, res2->poly + j, ctx)
&& res->exp[i] == res2->exp[j])
{
found = 1;
break;
}
}
if (!found)
{
flint_printf("FAIL: factor not found\n");
abort();
}
}
TEMPLATE(T, poly_clear) (quot, ctx);
TEMPLATE(T, poly_clear) (rem, ctx);
TEMPLATE(T, poly_clear) (pol1, ctx);
TEMPLATE(T, poly_clear) (poly, ctx);
TEMPLATE(T, poly_factor_clear) (res, ctx);
TEMPLATE(T, poly_factor_clear) (res2, ctx);
TEMPLATE(T, clear) (lead, ctx);
TEMPLATE(T, ctx_clear) (ctx);
}
FLINT_TEST_CLEANUP(state);
flint_printf("PASS\n");
return 0;
}
#endif