#include "longlong.h"
#include "fmpz.h"
#include "fmpz_vec.h"
#include "fq_nmod.h"
#include "fq_nmod_poly.h"
#include "n_poly.h"
#include "mpoly.h"
#include "fq_nmod_mpoly.h"
#include "fq_nmod_mpoly_factor.h"
static void fq_nmod_mpoly_evals(
slong * Atdeg,
n_fq_poly_struct * out,
const int * ignore,
const fq_nmod_mpoly_t A,
ulong * Amin_exp,
ulong * Amax_exp,
ulong * Astride,
fq_nmod_struct * alpha,
const fq_nmod_mpoly_ctx_t ctx)
{
slong d = fq_nmod_ctx_degree(ctx->fqctx);
slong i, j;
slong nvars = ctx->minfo->nvars;
ulong mask = (-UWORD(1)) >> (FLINT_BITS - A->bits);
slong N = mpoly_words_per_exp_sp(A->bits, ctx->minfo);
slong * offsets = FLINT_ARRAY_ALLOC(2*nvars, slong);
slong * shifts = offsets + nvars;
ulong * varexps = FLINT_ARRAY_ALLOC(nvars, ulong);
ulong varexp;
slong total_degree, lo, hi;
n_poly_struct * caches = FLINT_ARRAY_ALLOC(3*nvars, n_poly_struct);
ulong * t = FLINT_ARRAY_ALLOC(2*d, ulong);
ulong * meval = t + d;
for (j = 0; j < nvars; j++)
{
mpoly_gen_offset_shift_sp(offsets + j, shifts + j, j, A->bits,
ctx->minfo);
n_poly_init(caches + 3*j + 0);
n_poly_init(caches + 3*j + 1);
n_poly_init(caches + 3*j + 2);
n_fq_pow_cache_start_fq_nmod(alpha + j, caches + 3*j + 0,
caches + 3*j + 1, caches + 3*j + 2, ctx->fqctx);
if (ignore[j])
continue;
i = (Astride[j] < 2) ? Amax_exp[j] - Amin_exp[j] :
(Amax_exp[j] - Amin_exp[j])/Astride[j];
n_poly_fit_length(out + j, d*(i + 1));
_nmod_vec_zero(out[j].coeffs, d*(i + 1));
out[j].length = i + 1;
}
total_degree = 0;
for (i = 0; i < A->length; i++)
{
ulong * s = A->coeffs + d*i;
lo = hi = 0;
for (j = 0; j < nvars; j++)
{
varexp = ((A->exps + N*i)[offsets[j]]>>shifts[j])&mask;
FLINT_ASSERT((Astride[j] == 0 && varexp == Amin_exp[j]) ||
(varexp - Amin_exp[j]) % Astride[j] == 0);
varexps[j] = Astride[j] < 2 ? varexp - Amin_exp[j] :
(varexp - Amin_exp[j])/Astride[j];
add_ssaaaa(hi, lo, hi, lo, 0, varexps[j]);
n_fq_pow_cache_mulpow_ui(meval, s, varexps[j], caches + 3*j + 0,
caches + 3*j + 1, caches + 3*j + 2, ctx->fqctx);
s = meval;
}
if (hi == 0 && FLINT_SIGN_EXT(lo) == 0 && total_degree >= 0)
total_degree = FLINT_MAX(total_degree, lo);
else
total_degree = -1;
for (j = 0; j < nvars; j++)
{
varexp = varexps[j];
if (ignore[j])
continue;
FLINT_ASSERT(out[j].alloc >= d*(varexp + 1));
n_fq_pow_cache_mulpow_neg_ui(t, meval, varexp, caches + 3*j + 0,
caches + 3*j + 1, caches + 3*j + 2, ctx->fqctx);
n_fq_add(out[j].coeffs + d*varexp, out[j].coeffs + d*varexp,
t, ctx->fqctx);
}
}
*Atdeg = total_degree;
for (j = 0; j < nvars; j++)
_n_fq_poly_normalise(out + j, d);
for (j = 0; j < 3*nvars; j++)
n_poly_clear(caches + j);
flint_free(offsets);
flint_free(varexps);
flint_free(caches);
flint_free(t);
}
static void fq_nmod_mpoly_evals_lgprime(
slong * Atdeg,
n_fq_poly_struct * out,
const int * ignore,
const fq_nmod_mpoly_t A,
ulong * Amin_exp,
ulong * Amax_exp,
ulong * Astride,
const fq_nmod_mpoly_ctx_t smctx,
fq_nmod_struct * alpha,
const fq_nmod_mpoly_ctx_t lgctx,
const bad_fq_nmod_embed_t emb)
{
slong smd = fq_nmod_ctx_degree(smctx->fqctx);
slong lgd = fq_nmod_ctx_degree(lgctx->fqctx);
slong i, j;
slong nvars = smctx->minfo->nvars;
ulong mask = (-UWORD(1)) >> (FLINT_BITS - A->bits);
slong N = mpoly_words_per_exp_sp(A->bits, smctx->minfo);
slong * offsets = FLINT_ARRAY_ALLOC(2*nvars, slong);
slong * shifts = offsets + nvars;
ulong * varexps = FLINT_ARRAY_ALLOC(nvars, ulong);
ulong varexp, lo, hi;
slong total_degree;
n_poly_struct * caches = FLINT_ARRAY_ALLOC(3*nvars, n_poly_struct);
ulong * t = FLINT_ARRAY_ALLOC(2*lgd, ulong);
ulong * meval = t + lgd;
for (j = 0; j < nvars; j++)
{
mpoly_gen_offset_shift_sp(offsets + j, shifts + j, j, A->bits,
smctx->minfo);
n_poly_init(caches + 3*j + 0);
n_poly_init(caches + 3*j + 1);
n_poly_init(caches + 3*j + 2);
n_fq_pow_cache_start_fq_nmod(alpha + j, caches + 3*j + 0,
caches + 3*j + 1, caches + 3*j + 2, lgctx->fqctx);
if (ignore[j])
continue;
i = (Astride[j] < 2) ? Amax_exp[j] - Amin_exp[j] :
(Amax_exp[j] - Amin_exp[j])/Astride[j];
n_poly_fit_length(out + j, lgd*(i + 1));
_nmod_vec_zero(out[j].coeffs, lgd*(i + 1));
out[j].length = i + 1;
}
total_degree = 0;
for (i = 0; i < A->length; i++)
{
bad_n_fq_embed_sm_elem_to_lg(meval, A->coeffs + smd*i, emb);
hi = lo = 0;
for (j = 0; j < nvars; j++)
{
varexp = ((A->exps + N*i)[offsets[j]]>>shifts[j])&mask;
FLINT_ASSERT((Astride[j] == 0 && varexp == Amin_exp[j]) ||
(varexp - Amin_exp[j]) % Astride[j] == 0);
varexps[j] = Astride[j] < 2 ? varexp - Amin_exp[j] :
(varexp - Amin_exp[j])/Astride[j];
add_ssaaaa(hi, lo, hi, lo, 0, varexps[j]);
n_fq_pow_cache_mulpow_ui(meval, meval, varexps[j], caches + 3*j + 0,
caches + 3*j + 1, caches + 3*j + 2, lgctx->fqctx);
}
if (hi == 0 && FLINT_SIGN_EXT(lo) == 0 && total_degree >= 0)
total_degree = FLINT_MAX(total_degree, (slong) lo);
else
total_degree = -1;
for (j = 0; j < nvars; j++)
{
varexp = varexps[j];
if (ignore[j])
continue;
FLINT_ASSERT(out[j].alloc >= lgd*(varexp + 1));
n_fq_pow_cache_mulpow_neg_ui(t, meval, varexp, caches + 3*j + 0,
caches + 3*j + 1, caches + 3*j + 2, lgctx->fqctx);
n_fq_add(out[j].coeffs + lgd*varexp, out[j].coeffs + lgd*varexp,
t, lgctx->fqctx);
}
}
*Atdeg = total_degree;
for (j = 0; j < nvars; j++)
_n_fq_poly_normalise(out + j, lgd);
for (j = 0; j < 3*nvars; j++)
n_poly_clear(caches + j);
flint_free(offsets);
flint_free(varexps);
flint_free(caches);
flint_free(t);
}
static void mpoly_gcd_info_set_estimates_fq_nmod_mpoly(
mpoly_gcd_info_t I,
const fq_nmod_mpoly_t A,
const fq_nmod_mpoly_t B,
const fq_nmod_mpoly_ctx_t ctx)
{
slong d = fq_nmod_ctx_degree(ctx->fqctx);
int tries_left = 10;
slong nvars = ctx->minfo->nvars;
slong i, j;
n_fq_poly_t Geval;
n_fq_poly_struct * Aevals, * Bevals;
fq_nmod_struct * alpha;
flint_rand_t state;
slong ignore_limit;
int * ignore;
flint_rand_init(state);
ignore = FLINT_ARRAY_ALLOC(nvars, int);
alpha = FLINT_ARRAY_ALLOC(nvars, fq_nmod_struct);
Aevals = FLINT_ARRAY_ALLOC(2*nvars, n_fq_poly_struct);
Bevals = Aevals + nvars;
n_fq_poly_init(Geval);
for (j = 0; j < nvars; j++)
{
fq_nmod_init(alpha + j, ctx->fqctx);
n_fq_poly_init(Aevals + j);
n_fq_poly_init(Bevals + j);
}
ignore_limit = FLINT_MAX(WORD(9999), (A->length + B->length)/4096);
I->Gdeflate_deg_bounds_are_nice = 1;
for (j = 0; j < nvars; j++)
{
if (I->Adeflate_deg[j] > ignore_limit ||
I->Bdeflate_deg[j] > ignore_limit)
{
ignore[j] = 1;
I->Gdeflate_deg_bounds_are_nice = 0;
}
else
{
ignore[j] = 0;
}
}
try_again:
if (--tries_left < 0)
{
I->Gdeflate_deg_bounds_are_nice = 0;
for (j = 0; j < nvars; j++)
{
I->Gdeflate_deg_bound[j] = FLINT_MIN(I->Adeflate_deg[j],
I->Bdeflate_deg[j]);
I->Gterm_count_est[j] = 1 + I->Gdeflate_deg_bound[j]/2;
}
goto cleanup;
}
for (j = 0; j < nvars; j++)
{
fq_nmod_rand(alpha + j, state, ctx->fqctx);
if (fq_nmod_is_zero(alpha + j, ctx->fqctx))
fq_nmod_one(alpha + j, ctx->fqctx);
}
fq_nmod_mpoly_evals(&I->Adeflate_tdeg, Aevals, ignore, A,
I->Amin_exp, I->Amax_exp, I->Gstride, alpha, ctx);
fq_nmod_mpoly_evals(&I->Bdeflate_tdeg, Bevals, ignore, B,
I->Bmin_exp, I->Bmax_exp, I->Gstride, alpha, ctx);
for (j = 0; j < nvars; j++)
{
if (ignore[j])
{
I->Gdeflate_deg_bound[j] = FLINT_MIN(I->Adeflate_deg[j],
I->Bdeflate_deg[j]);
I->Gterm_count_est[j] = 1 + I->Gdeflate_deg_bound[j]/2;
}
else
{
if (I->Adeflate_deg[j] != n_fq_poly_degree(Aevals + j) ||
I->Bdeflate_deg[j] != n_fq_poly_degree(Bevals + j))
{
goto try_again;
}
n_fq_poly_gcd(Geval, Aevals + j, Bevals + j, ctx->fqctx);
I->Gterm_count_est[j] = 0;
I->Gdeflate_deg_bound[j] = n_fq_poly_degree(Geval);
for (i = I->Gdeflate_deg_bound[j]; i >= 0; i--)
I->Gterm_count_est[j] += _n_fq_is_zero(Geval->coeffs + d*i, d);
}
}
cleanup:
n_fq_poly_clear(Geval);
for (j = 0; j < nvars; j++)
{
fq_nmod_clear(alpha + j, ctx->fqctx);
n_fq_poly_clear(Aevals + j);
n_fq_poly_clear(Bevals + j);
}
flint_free(ignore);
flint_free(alpha);
flint_free(Aevals);
flint_rand_clear(state);
return;
}
static void mpoly_gcd_info_set_estimates_fq_nmod_mpoly_lgprime(
mpoly_gcd_info_t I,
const fq_nmod_mpoly_t A,
const fq_nmod_mpoly_t B,
const fq_nmod_mpoly_ctx_t smctx)
{
int tries_left = 10;
slong nvars = smctx->minfo->nvars;
slong i, j;
n_fq_poly_t Geval;
n_fq_poly_struct * Aevals, * Bevals;
fq_nmod_struct * alpha;
flint_rand_t state;
slong ignore_limit;
int * ignore;
fq_nmod_mpoly_ctx_t lgctx;
bad_fq_nmod_mpoly_embed_chooser_t embc;
bad_fq_nmod_embed_struct * cur_emb;
flint_rand_init(state);
cur_emb = bad_fq_nmod_mpoly_embed_chooser_init(embc, lgctx, smctx, state);
ignore = FLINT_ARRAY_ALLOC(nvars, int);
alpha = FLINT_ARRAY_ALLOC(nvars, fq_nmod_struct);
Aevals = FLINT_ARRAY_ALLOC(2*nvars, n_fq_poly_struct);
Bevals = Aevals + nvars;
n_fq_poly_init(Geval);
for (j = 0; j < nvars; j++)
{
fq_nmod_init(alpha + j, lgctx->fqctx);
n_fq_poly_init(Aevals + j);
n_fq_poly_init(Bevals + j);
}
ignore_limit = FLINT_MAX(WORD(9999), (A->length + B->length)/4096);
I->Gdeflate_deg_bounds_are_nice = 1;
for (j = 0; j < nvars; j++)
{
if (I->Adeflate_deg[j] > ignore_limit ||
I->Bdeflate_deg[j] > ignore_limit)
{
ignore[j] = 1;
I->Gdeflate_deg_bounds_are_nice = 0;
}
else
{
ignore[j] = 0;
}
}
try_again:
if (--tries_left < 0 || cur_emb == NULL)
{
I->Gdeflate_deg_bounds_are_nice = 0;
for (j = 0; j < nvars; j++)
{
I->Gdeflate_deg_bound[j] = FLINT_MIN(I->Adeflate_deg[j],
I->Bdeflate_deg[j]);
I->Gterm_count_est[j] = 1 + I->Gdeflate_deg_bound[j]/2;
}
goto cleanup;
}
for (j = 0; j < nvars; j++)
{
fq_nmod_rand(alpha + j, state, lgctx->fqctx);
if (fq_nmod_is_zero(alpha + j, lgctx->fqctx))
fq_nmod_one(alpha + j, lgctx->fqctx);
}
fq_nmod_mpoly_evals_lgprime(&I->Adeflate_tdeg, Aevals, ignore, A,
I->Amin_exp, I->Amax_exp, I->Gstride, smctx, alpha, lgctx, cur_emb);
fq_nmod_mpoly_evals_lgprime(&I->Bdeflate_tdeg, Bevals, ignore, B,
I->Bmin_exp, I->Bmax_exp, I->Gstride, smctx, alpha, lgctx, cur_emb);
for (j = 0; j < nvars; j++)
{
if (ignore[j])
{
I->Gdeflate_deg_bound[j] = FLINT_MIN(I->Adeflate_deg[j],
I->Bdeflate_deg[j]);
I->Gterm_count_est[j] = 1 + I->Gdeflate_deg_bound[j]/2;
}
else
{
slong lgd = fq_nmod_ctx_degree(lgctx->fqctx);
if (I->Adeflate_deg[j] != n_fq_poly_degree(Aevals + j) ||
I->Bdeflate_deg[j] != n_fq_poly_degree(Bevals + j))
{
cur_emb = bad_fq_nmod_mpoly_embed_chooser_next(embc, lgctx, smctx, state);
goto try_again;
}
n_fq_poly_gcd(Geval, Aevals + j, Bevals + j, lgctx->fqctx);
I->Gterm_count_est[j] = 0;
I->Gdeflate_deg_bound[j] = n_fq_poly_degree(Geval);
for (i = I->Gdeflate_deg_bound[j]; i >= 0; i--)
I->Gterm_count_est[j] += _n_fq_is_zero(Geval->coeffs + lgd*i, lgd);
}
}
cleanup:
n_fq_poly_clear(Geval);
for (j = 0; j < nvars; j++)
{
fq_nmod_clear(alpha + j, lgctx->fqctx);
n_fq_poly_clear(Aevals + j);
n_fq_poly_clear(Bevals + j);
}
flint_free(ignore);
flint_free(alpha);
flint_free(Aevals);
bad_fq_nmod_mpoly_embed_chooser_clear(embc, lgctx, smctx, state);
flint_rand_clear(state);
return;
}
static void _parallel_set(
fq_nmod_mpoly_t Abar,
fq_nmod_mpoly_t Bbar,
const fq_nmod_mpoly_t A,
const fq_nmod_mpoly_t B,
const fq_nmod_mpoly_ctx_t ctx)
{
if (Abar == B && Bbar == A)
{
FLINT_ASSERT(Abar != NULL && Bbar != NULL);
fq_nmod_mpoly_set(Abar, B, ctx);
fq_nmod_mpoly_set(Bbar, A, ctx);
fq_nmod_mpoly_swap(Abar, Bbar, ctx);
}
else if (Abar == B && Bbar != A)
{
FLINT_ASSERT(Abar != NULL);
if (Bbar != NULL)
fq_nmod_mpoly_set(Bbar, B, ctx);
fq_nmod_mpoly_set(Abar, A, ctx);
}
else
{
if (Abar != NULL)
fq_nmod_mpoly_set(Abar, A, ctx);
if (Bbar != NULL)
fq_nmod_mpoly_set(Bbar, B, ctx);
}
}
static int _do_trivial(
fq_nmod_mpoly_t G,
fq_nmod_mpoly_t Abar,
fq_nmod_mpoly_t Bbar,
const fq_nmod_mpoly_t A,
const fq_nmod_mpoly_t B,
const mpoly_gcd_info_t I,
const fq_nmod_mpoly_ctx_t ctx)
{
_parallel_set(Abar, Bbar, A, B, ctx);
if (Abar != NULL)
mpoly_monomials_shift_right_ui(Abar->exps, Abar->bits, Abar->length,
I->Gmin_exp, ctx->minfo);
if (Bbar != NULL)
mpoly_monomials_shift_right_ui(Bbar->exps, Bbar->bits, Bbar->length,
I->Gmin_exp, ctx->minfo);
fq_nmod_mpoly_fit_length_reset_bits(G, 1, I->Gbits, ctx);
mpoly_set_monomial_ui(G->exps, I->Gmin_exp, I->Gbits, ctx->minfo);
_n_fq_one(G->coeffs, fq_nmod_ctx_degree(ctx->fqctx));
_fq_nmod_mpoly_set_length(G, 1, ctx);
return 1;
}
static int _do_monomial_gcd(
fq_nmod_mpoly_t G,
fq_nmod_mpoly_t Abar,
fq_nmod_mpoly_t Bbar,
const fq_nmod_mpoly_t A,
const fq_nmod_mpoly_t B,
const fq_nmod_mpoly_ctx_t ctx)
{
slong i;
flint_bitcnt_t Gbits = FLINT_MIN(A->bits, B->bits);
fmpz * minAfields, * minAdegs, * minBdegs;
TMP_INIT;
FLINT_ASSERT(A->length > 0);
FLINT_ASSERT(B->length == 1);
TMP_START;
minAfields = (fmpz *) TMP_ALLOC(ctx->minfo->nfields*sizeof(fmpz));
for (i = 0; i < ctx->minfo->nfields; i++)
fmpz_init(minAfields + i);
mpoly_min_fields_fmpz(minAfields, A->exps, A->length, A->bits, ctx->minfo);
minAdegs = (fmpz *) TMP_ALLOC(ctx->minfo->nvars*sizeof(fmpz));
for (i = 0; i < ctx->minfo->nvars; i++)
fmpz_init(minAdegs + i);
mpoly_get_monomial_ffmpz_unpacked_ffmpz(minAdegs, minAfields, ctx->minfo);
minBdegs = (fmpz *) TMP_ALLOC(ctx->minfo->nvars*sizeof(fmpz));
for (i = 0; i < ctx->minfo->nvars; i++)
fmpz_init(minBdegs + i);
mpoly_get_monomial_ffmpz(minBdegs, B->exps, B->bits, ctx->minfo);
_fmpz_vec_min_inplace(minBdegs, minAdegs, ctx->minfo->nvars);
_parallel_set(Abar, Bbar, A, B, ctx);
if (Abar != NULL)
mpoly_monomials_shift_right_ffmpz(Abar->exps, Abar->bits, Abar->length,
minBdegs, ctx->minfo);
if (Bbar != NULL)
mpoly_monomials_shift_right_ffmpz(Bbar->exps, Bbar->bits, Bbar->length,
minBdegs, ctx->minfo);
fq_nmod_mpoly_fit_length_reset_bits(G, 1, Gbits, ctx);
mpoly_set_monomial_ffmpz(G->exps, minBdegs, Gbits, ctx->minfo);
_n_fq_one(G->coeffs + 0, fq_nmod_ctx_degree(ctx->fqctx));
_fq_nmod_mpoly_set_length(G, 1, ctx);
for (i = 0; i < ctx->minfo->nfields; i++)
{
fmpz_clear(minAfields + i);
}
for (i = 0; i < ctx->minfo->nvars; i++)
{
fmpz_clear(minAdegs + i);
fmpz_clear(minBdegs + i);
}
TMP_END;
return 1;
}
static int _try_monomial_cofactors(
fq_nmod_mpoly_t G,
fq_nmod_mpoly_t Abar,
fq_nmod_mpoly_t Bbar,
const fq_nmod_mpoly_t A,
const fq_nmod_mpoly_t B,
const fq_nmod_mpoly_ctx_t ctx)
{
slong d = fq_nmod_ctx_degree(ctx->fqctx);
int success;
slong i, j;
slong NA, NG;
slong nvars = ctx->minfo->nvars;
fmpz * Abarexps, * Bbarexps, * Texps;
ulong * tmp, * t1, * t2, * a0, * b0;
fq_nmod_mpoly_t T;
flint_bitcnt_t Gbits = FLINT_MIN(A->bits, B->bits);
flint_bitcnt_t Abarbits = A->bits;
flint_bitcnt_t Bbarbits = B->bits;
TMP_INIT;
FLINT_ASSERT(A->length > 0);
FLINT_ASSERT(B->length > 0);
if (A->length != B->length)
return 0;
TMP_START;
tmp = (ulong *) TMP_ALLOC(d*(4 + FLINT_MAX(N_FQ_MUL_ITCH,
N_FQ_INV_ITCH))*sizeof(ulong));
t1 = tmp + d*FLINT_MAX(N_FQ_MUL_ITCH, N_FQ_INV_ITCH);
t2 = t1 + d;
a0 = t2 + d;
b0 = a0 + d;
for (i = A->length - 1; i > 0; i--)
{
_n_fq_mul(t1, A->coeffs + d*0, B->coeffs + d*i, ctx->fqctx, tmp);
_n_fq_mul(t2, B->coeffs + d*0, A->coeffs + d*i, ctx->fqctx, tmp);
success = _n_fq_equal(t1, t2, d);
if (!success)
goto cleanup_less;
}
_n_fq_set(a0, A->coeffs + d*0, d);
_n_fq_set(b0, B->coeffs + d*0, d);
Abarexps = (fmpz *) TMP_ALLOC(3*nvars*sizeof(fmpz));
Bbarexps = Abarexps + 1*nvars;
Texps = Abarexps + 2*nvars;
for (j = 0; j < nvars; j++)
{
fmpz_init(Abarexps + j);
fmpz_init(Bbarexps + j);
fmpz_init(Texps + j);
}
success = mpoly_monomial_cofactors(Abarexps, Bbarexps, A->exps, A->bits,
B->exps, B->bits, A->length, ctx->minfo);
if (!success)
goto cleanup_more;
fq_nmod_mpoly_init3(T, A->length, Gbits, ctx);
NG = mpoly_words_per_exp(Gbits, ctx->minfo);
NA = mpoly_words_per_exp(A->bits, ctx->minfo);
_n_fq_inv(t1, A->coeffs + d*0, ctx->fqctx, tmp);
T->length = A->length;
for (i = 0; i < A->length; i++)
{
mpoly_get_monomial_ffmpz(Texps, A->exps + NA*i, A->bits, ctx->minfo);
_fmpz_vec_sub(Texps, Texps, Abarexps, nvars);
mpoly_set_monomial_ffmpz(T->exps + NG*i, Texps, Gbits, ctx->minfo);
n_fq_mul(T->coeffs + d*i, A->coeffs + d*i, t1, ctx->fqctx);
}
fq_nmod_mpoly_swap(G, T, ctx);
fq_nmod_mpoly_clear(T, ctx);
if (Abar != NULL)
{
fq_nmod_mpoly_fit_length_reset_bits(Abar, 1, Abarbits, ctx);
mpoly_set_monomial_ffmpz(Abar->exps, Abarexps, Abarbits, ctx->minfo);
_n_fq_set(Abar->coeffs + d*0, a0, d);
_fq_nmod_mpoly_set_length(Abar, 1, ctx);
}
if (Bbar != NULL)
{
fq_nmod_mpoly_fit_length_reset_bits(Bbar, 1, Bbarbits, ctx);
mpoly_set_monomial_ffmpz(Bbar->exps, Bbarexps, Bbarbits, ctx->minfo);
_n_fq_set(Bbar->coeffs + d*0, b0, d);
_fq_nmod_mpoly_set_length(Bbar, 1, ctx);
}
success = 1;
cleanup_more:
for (j = 0; j < nvars; j++)
{
fmpz_clear(Abarexps + j);
fmpz_clear(Bbarexps + j);
fmpz_clear(Texps + j);
}
cleanup_less:
TMP_END;
return success;
}
static int _do_univar(
fq_nmod_mpoly_t G,
fq_nmod_mpoly_t Abar,
fq_nmod_mpoly_t Bbar,
const fq_nmod_mpoly_t A,
const fq_nmod_mpoly_t B,
slong v_in_both,
const mpoly_gcd_info_t I,
const fq_nmod_mpoly_ctx_t ctx)
{
fq_nmod_poly_t a, b, g, t, r;
fq_nmod_poly_init(a, ctx->fqctx);
fq_nmod_poly_init(b, ctx->fqctx);
fq_nmod_poly_init(g, ctx->fqctx);
fq_nmod_poly_init(t, ctx->fqctx);
fq_nmod_poly_init(r, ctx->fqctx);
_fq_nmod_mpoly_to_fq_nmod_poly_deflate(a, A, v_in_both,
I->Amin_exp, I->Gstride, ctx);
_fq_nmod_mpoly_to_fq_nmod_poly_deflate(b, B, v_in_both,
I->Bmin_exp, I->Gstride, ctx);
fq_nmod_poly_gcd(g, a, b, ctx->fqctx);
_fq_nmod_mpoly_from_fq_nmod_poly_inflate(G, I->Gbits, g, v_in_both,
I->Gmin_exp, I->Gstride, ctx);
if (Abar != NULL)
{
fq_nmod_poly_divrem(t, r, a, g, ctx->fqctx);
FLINT_ASSERT(fq_nmod_poly_is_zero(r, ctx->fqctx));
_fq_nmod_mpoly_from_fq_nmod_poly_inflate(Abar, I->Abarbits, t,
v_in_both, I->Abarmin_exp, I->Gstride, ctx);
}
if (Bbar != NULL)
{
fq_nmod_poly_divrem(t, r, b, g, ctx->fqctx);
FLINT_ASSERT(fq_nmod_poly_is_zero(r, ctx->fqctx));
_fq_nmod_mpoly_from_fq_nmod_poly_inflate(Bbar, I->Bbarbits, t, v_in_both,
I->Bbarmin_exp, I->Gstride, ctx);
}
fq_nmod_poly_clear(a, ctx->fqctx);
fq_nmod_poly_clear(b, ctx->fqctx);
fq_nmod_poly_clear(g, ctx->fqctx);
fq_nmod_poly_clear(t, ctx->fqctx);
fq_nmod_poly_clear(r, ctx->fqctx);
return 1;
}
static int _try_missing_var(
fq_nmod_mpoly_t G, flint_bitcnt_t Gbits,
fq_nmod_mpoly_t Abar,
fq_nmod_mpoly_t Bbar,
slong var,
const fq_nmod_mpoly_t A, ulong Ashift,
const fq_nmod_mpoly_t B, ulong Bshift,
const fq_nmod_mpoly_ctx_t ctx)
{
int success;
fq_nmod_mpoly_univar_t Au;
fq_nmod_mpoly_univar_init(Au, ctx);
#if FLINT_WANT_ASSERT
fq_nmod_mpoly_to_univar(Au, B, var, ctx);
FLINT_ASSERT(Au->length == 1);
#endif
fq_nmod_mpoly_to_univar(Au, A, var, ctx);
fq_nmod_mpoly_univar_fit_length(Au, Au->length + 1, ctx);
fq_nmod_mpoly_set(Au->coeffs + Au->length, B, ctx);
Au->length++;
if (Abar == NULL && Bbar == NULL)
{
success = _fq_nmod_mpoly_vec_content_mpoly(G, Au->coeffs, Au->length, ctx);
if (!success)
goto cleanup;
fq_nmod_mpoly_repack_bits_inplace(G, Gbits, ctx);
_mpoly_gen_shift_left(G->exps, G->bits, G->length,
var, FLINT_MIN(Ashift, Bshift), ctx->minfo);
}
else
{
fq_nmod_mpoly_t tG, tAbar, tBbar;
fq_nmod_mpoly_init(tG, ctx);
fq_nmod_mpoly_init(tAbar, ctx);
fq_nmod_mpoly_init(tBbar, ctx);
success = _fq_nmod_mpoly_vec_content_mpoly(tG, Au->coeffs, Au->length, ctx);
if (!success)
goto cleanup;
fq_nmod_mpoly_repack_bits_inplace(tG, Gbits, ctx);
_mpoly_gen_shift_left(tG->exps, tG->bits, tG->length,
var, FLINT_MIN(Ashift, Bshift), ctx->minfo);
if (Abar != NULL)
{
success = fq_nmod_mpoly_divides(tAbar, A, tG, ctx);
FLINT_ASSERT(success);
}
if (Bbar != NULL)
{
success = fq_nmod_mpoly_divides(tBbar, B, tG, ctx);
FLINT_ASSERT(success);
}
fq_nmod_mpoly_swap(G, tG, ctx);
if (Abar != NULL)
fq_nmod_mpoly_swap(Abar, tAbar, ctx);
if (Bbar != NULL)
fq_nmod_mpoly_swap(Bbar, tBbar, ctx);
fq_nmod_mpoly_clear(tG, ctx);
fq_nmod_mpoly_clear(tAbar, ctx);
fq_nmod_mpoly_clear(tBbar, ctx);
}
success = 1;
cleanup:
fq_nmod_mpoly_univar_clear(Au, ctx);
return success;
}
static int _try_divides(
fq_nmod_mpoly_t G,
fq_nmod_mpoly_t Abar,
fq_nmod_mpoly_t Bbar,
const fq_nmod_mpoly_t A,
const fq_nmod_mpoly_t BB,
const fq_nmod_mpoly_ctx_t ctx)
{
int success = 0;
fq_nmod_mpoly_t Q, B, M;
fq_nmod_mpoly_init(Q, ctx);
fq_nmod_mpoly_init(B, ctx);
fq_nmod_mpoly_init(M, ctx);
fq_nmod_mpoly_term_content(M, BB, ctx);
fq_nmod_mpoly_divides(B, BB, M, ctx);
if (fq_nmod_mpoly_divides(Q, A, B, ctx))
{
_do_monomial_gcd(G, Abar, Bbar, Q, M, ctx);
fq_nmod_mpoly_mul(G, G, B, ctx);
success = 1;
}
fq_nmod_mpoly_clear(Q, ctx);
fq_nmod_mpoly_clear(B, ctx);
fq_nmod_mpoly_clear(M, ctx);
return success;
}
static int _try_zippel(
fq_nmod_mpoly_t G,
fq_nmod_mpoly_t Abar,
fq_nmod_mpoly_t Bbar,
const fq_nmod_mpoly_t A,
const fq_nmod_mpoly_t B,
const mpoly_gcd_info_t I,
const fq_nmod_mpoly_ctx_t ctx)
{
slong m = I->mvars;
int success;
flint_bitcnt_t wbits;
flint_rand_t randstate;
fq_nmod_mpoly_ctx_t uctx;
fq_nmod_mpolyu_t Au, Bu, Gu, Abaru, Bbaru;
fq_nmod_mpoly_t Ac, Bc, Gc, Abarc, Bbarc;
FLINT_ASSERT(A->bits <= FLINT_BITS);
FLINT_ASSERT(B->bits <= FLINT_BITS);
if (!(I->can_use & MPOLY_GCD_USE_ZIPPEL))
return 0;
FLINT_ASSERT(m >= WORD(2));
FLINT_ASSERT(A->length > 0);
FLINT_ASSERT(B->length > 0);
flint_rand_init(randstate);
fq_nmod_mpoly_ctx_init(uctx, m - 1, ORD_LEX, ctx->fqctx);
wbits = FLINT_MAX(A->bits, B->bits);
fq_nmod_mpolyu_init(Au, wbits, uctx);
fq_nmod_mpolyu_init(Bu, wbits, uctx);
fq_nmod_mpolyu_init(Gu, wbits, uctx);
fq_nmod_mpolyu_init(Abaru, wbits, uctx);
fq_nmod_mpolyu_init(Bbaru, wbits, uctx);
fq_nmod_mpoly_init3(Ac, 0, wbits, uctx);
fq_nmod_mpoly_init3(Bc, 0, wbits, uctx);
fq_nmod_mpoly_init3(Gc, 0, wbits, uctx);
fq_nmod_mpoly_init3(Abarc, 0, wbits, uctx);
fq_nmod_mpoly_init3(Bbarc, 0, wbits, uctx);
fq_nmod_mpoly_to_mpolyu_perm_deflate(Au, uctx, A, ctx,
I->zippel_perm, I->Amin_exp, I->Gstride);
fq_nmod_mpoly_to_mpolyu_perm_deflate(Bu, uctx, B, ctx,
I->zippel_perm, I->Bmin_exp, I->Gstride);
success = fq_nmod_mpolyu_content_mpoly(Ac, Au, uctx) &&
fq_nmod_mpolyu_content_mpoly(Bc, Bu, uctx);
if (!success)
goto cleanup;
fq_nmod_mpolyu_divexact_mpoly_inplace(Au, Ac, uctx);
fq_nmod_mpolyu_divexact_mpoly_inplace(Bu, Bc, uctx);
success = fq_nmod_mpolyu_gcdm_zippel(Gu, Abaru, Bbaru, Au, Bu,
uctx, randstate);
if (!success)
goto cleanup;
if (Abar == NULL && Bbar == NULL)
{
success = fq_nmod_mpoly_gcd(Gc, Ac, Bc, uctx);
if (!success)
goto cleanup;
fq_nmod_mpoly_repack_bits_inplace(Gc, wbits, uctx);
fq_nmod_mpolyu_mul_mpoly_inplace(Gu, Gc, uctx);
fq_nmod_mpoly_from_mpolyu_perm_inflate(G, I->Gbits, ctx, Gu, uctx,
I->zippel_perm, I->Gmin_exp, I->Gstride);
}
else
{
success = fq_nmod_mpoly_gcd_cofactors(Gc, Abarc, Bbarc, Ac, Bc, uctx);
if (!success)
goto cleanup;
fq_nmod_mpoly_repack_bits_inplace(Gc, wbits, uctx);
fq_nmod_mpoly_repack_bits_inplace(Abarc, wbits, uctx);
fq_nmod_mpoly_repack_bits_inplace(Bbarc, wbits, uctx);
fq_nmod_mpolyu_mul_mpoly_inplace(Gu, Gc, uctx);
fq_nmod_mpolyu_mul_mpoly_inplace(Abaru, Abarc, uctx);
fq_nmod_mpolyu_mul_mpoly_inplace(Bbaru, Bbarc, uctx);
fq_nmod_mpoly_from_mpolyu_perm_inflate(G, I->Gbits, ctx, Gu, uctx,
I->zippel_perm, I->Gmin_exp, I->Gstride);
if (Abar != NULL)
fq_nmod_mpoly_from_mpolyu_perm_inflate(Abar, I->Abarbits, ctx,
Abaru, uctx, I->zippel_perm, I->Abarmin_exp, I->Gstride);
if (Bbar != NULL)
fq_nmod_mpoly_from_mpolyu_perm_inflate(Bbar, I->Bbarbits, ctx,
Bbaru, uctx, I->zippel_perm, I->Bbarmin_exp, I->Gstride);
}
success = 1;
cleanup:
fq_nmod_mpolyu_clear(Au, uctx);
fq_nmod_mpolyu_clear(Bu, uctx);
fq_nmod_mpolyu_clear(Gu, uctx);
fq_nmod_mpolyu_clear(Abaru, uctx);
fq_nmod_mpolyu_clear(Bbaru, uctx);
fq_nmod_mpoly_clear(Ac, uctx);
fq_nmod_mpoly_clear(Bc, uctx);
fq_nmod_mpoly_clear(Gc, uctx);
fq_nmod_mpoly_clear(Abarc, uctx);
fq_nmod_mpoly_clear(Bbarc, uctx);
fq_nmod_mpoly_ctx_clear(uctx);
flint_rand_clear(randstate);
return success;
}
static int _try_zippel2(
fq_nmod_mpoly_t G,
fq_nmod_mpoly_t Abar,
fq_nmod_mpoly_t Bbar,
const fq_nmod_mpoly_t A,
const fq_nmod_mpoly_t B,
const mpoly_gcd_info_t I,
const fq_nmod_mpoly_ctx_t ctx)
{
slong i, k;
slong m = I->mvars;
int success;
flint_bitcnt_t wbits;
fq_nmod_mpoly_ctx_t lctx;
fq_nmod_mpoly_t Al, Bl, Gl, Abarl, Bbarl;
fq_nmod_mpoly_t Al_lc, Bl_lc, Ac, Bc, Gc, Abarc, Bbarc, Gamma;
slong * tmp, * Gl_degs, * Al_degs, * Bl_degs, * Gamma_degs, * Gguess;
slong max_degree;
FLINT_ASSERT(A->bits <= FLINT_BITS);
FLINT_ASSERT(B->bits <= FLINT_BITS);
FLINT_ASSERT(A->length > 0);
FLINT_ASSERT(B->length > 0);
if (!(I->can_use & MPOLY_GCD_USE_ZIPPEL2))
return 0;
FLINT_ASSERT(m >= WORD(3));
tmp = FLINT_ARRAY_ALLOC(5*m, slong);
Al_degs = tmp + 1*m;
Bl_degs = tmp + 2*m;
Gl_degs = tmp + 3*m;
Gamma_degs = tmp + 4*m;
fq_nmod_mpoly_ctx_init(lctx, m, ORD_LEX, ctx->fqctx);
max_degree = 0;
for (i = 0; i < m; i++)
{
k = I->zippel2_perm[i];
Gl_degs[i] = I->Gdeflate_deg_bound[k];
Al_degs[i] = I->Adeflate_deg[k];
max_degree = FLINT_MAX(max_degree, Al_degs[i]);
Bl_degs[i] = I->Bdeflate_deg[k];
max_degree = FLINT_MAX(max_degree, Bl_degs[i]);
}
wbits = 1 + FLINT_BIT_COUNT(max_degree);
wbits = mpoly_fix_bits(wbits, lctx->minfo);
FLINT_ASSERT(wbits <= FLINT_BITS);
fq_nmod_mpoly_init3(Al, 0, wbits, lctx);
fq_nmod_mpoly_init3(Bl, 0, wbits, lctx);
fq_nmod_mpoly_init3(Gl, 0, wbits, lctx);
fq_nmod_mpoly_init3(Abarl, 0, wbits, lctx);
fq_nmod_mpoly_init3(Bbarl, 0, wbits, lctx);
fq_nmod_mpoly_init3(Ac, 0, wbits, lctx);
fq_nmod_mpoly_init3(Bc, 0, wbits, lctx);
fq_nmod_mpoly_init3(Gc, 0, wbits, lctx);
fq_nmod_mpoly_init3(Abarc, 0, wbits, lctx);
fq_nmod_mpoly_init3(Bbarc, 0, wbits, lctx);
fq_nmod_mpoly_init3(Gamma, 0, wbits, lctx);
fq_nmod_mpoly_init3(Al_lc, 0, wbits, lctx);
fq_nmod_mpoly_init3(Bl_lc, 0, wbits, lctx);
fq_nmod_mpoly_to_mpolyl_perm_deflate(Al, lctx, A, ctx,
I->zippel2_perm, I->Amin_exp, I->Gstride);
fq_nmod_mpoly_to_mpolyl_perm_deflate(Bl, lctx, B, ctx,
I->zippel2_perm, I->Bmin_exp, I->Gstride);
success = fq_nmod_mpolyl_content(Ac, Al, 2, lctx) &&
fq_nmod_mpolyl_content(Bc, Bl, 2, lctx);
if (!success)
goto cleanup;
if (Abar == NULL && Bbar == NULL)
success = fq_nmod_mpoly_gcd(Gc, Ac, Bc, lctx);
else
success = fq_nmod_mpoly_gcd_cofactors(Gc, Abarc, Bbarc, Ac, Bc, lctx);
if (!success)
goto cleanup;
fq_nmod_mpoly_degrees_si(tmp, Ac, lctx);
for (i = 0; i < m; i++)
Al_degs[i] -= tmp[i];
if (!fq_nmod_mpoly_is_one(Ac, lctx))
{
success = fq_nmod_mpoly_divides(Al, Al, Ac, lctx);
FLINT_ASSERT(success);
}
fq_nmod_mpoly_degrees_si(tmp, Bc, lctx);
for (i = 0; i < m; i++)
Bl_degs[i] -= tmp[i];
if (!fq_nmod_mpoly_is_one(Bc, lctx))
{
success = fq_nmod_mpoly_divides(Bl, Bl, Bc, lctx);
FLINT_ASSERT(success);
}
fq_nmod_mpoly_degrees_si(tmp, Gc, lctx);
for (i = 0; i < m; i++)
Gl_degs[i] -= tmp[i];
fq_nmod_mpoly_repack_bits_inplace(Al, wbits, lctx);
fq_nmod_mpoly_repack_bits_inplace(Bl, wbits, lctx);
fq_nmod_mpolyl_lead_coeff(Al_lc, Al, 2, lctx);
fq_nmod_mpolyl_lead_coeff(Bl_lc, Bl, 2, lctx);
success = fq_nmod_mpoly_gcd(Gamma, Al_lc, Bl_lc, lctx);
if (!success)
goto cleanup;
fq_nmod_mpoly_repack_bits_inplace(Gamma, wbits, lctx);
fq_nmod_mpoly_degrees_si(Gamma_degs, Gamma, lctx);
Gguess = I->Gdeflate_deg_bounds_are_nice ? Gl_degs : NULL;
success = fq_nmod_mpolyl_gcd_zippel_smprime(Gl, Gguess, Abarl, Bbarl,
Al, Al_degs, Bl, Bl_degs, Gamma, Gamma_degs, lctx);
if (!success)
{
success = fq_nmod_mpolyl_gcd_zippel_lgprime(Gl, Gguess, Abarl, Bbarl,
Al, Al_degs, Bl, Bl_degs, Gamma, Gamma_degs, lctx);
if (!success)
goto cleanup;
}
if (!fq_nmod_mpoly_is_one(Gc, lctx))
fq_nmod_mpoly_mul(Gl, Gl, Gc, lctx);
fq_nmod_mpoly_from_mpolyl_perm_inflate(G, I->Gbits, ctx, Gl, lctx,
I->zippel2_perm, I->Gmin_exp, I->Gstride);
if (Abar != NULL)
{
fq_nmod_mpoly_mul(Abarl, Abarl, Abarc, lctx);
fq_nmod_mpoly_from_mpolyl_perm_inflate(Abar, I->Abarbits, ctx, Abarl, lctx,
I->zippel2_perm, I->Abarmin_exp, I->Gstride);
}
if (Bbar != NULL)
{
fq_nmod_mpoly_mul(Bbarl, Bbarl, Bbarc, lctx);
fq_nmod_mpoly_from_mpolyl_perm_inflate(Bbar, I->Bbarbits, ctx, Bbarl, lctx,
I->zippel2_perm, I->Bbarmin_exp, I->Gstride);
}
success = 1;
cleanup:
fq_nmod_mpoly_clear(Al, lctx);
fq_nmod_mpoly_clear(Bl, lctx);
fq_nmod_mpoly_clear(Gl, lctx);
fq_nmod_mpoly_clear(Abarl, lctx);
fq_nmod_mpoly_clear(Bbarl, lctx);
fq_nmod_mpoly_clear(Ac, lctx);
fq_nmod_mpoly_clear(Bc, lctx);
fq_nmod_mpoly_clear(Gc, lctx);
fq_nmod_mpoly_clear(Abarc, lctx);
fq_nmod_mpoly_clear(Bbarc, lctx);
fq_nmod_mpoly_clear(Gamma, lctx);
fq_nmod_mpoly_clear(Al_lc, lctx);
fq_nmod_mpoly_clear(Bl_lc, lctx);
fq_nmod_mpoly_ctx_clear(lctx);
flint_free(tmp);
return success;
}
static int _try_hensel(
fq_nmod_mpoly_t G,
fq_nmod_mpoly_t Abar,
fq_nmod_mpoly_t Bbar,
const fq_nmod_mpoly_t A,
const fq_nmod_mpoly_t B,
const mpoly_gcd_info_t I,
const fq_nmod_mpoly_ctx_t ctx)
{
slong i, k;
slong m = I->mvars;
int success;
flint_bitcnt_t wbits;
fq_nmod_mpoly_ctx_t lctx;
fq_nmod_mpoly_t Al, Bl, Gl, Abarl, Bbarl;
fq_nmod_mpoly_t Ac, Bc, Gc, Abarc, Bbarc;
slong max_deg;
FLINT_ASSERT(A->bits <= FLINT_BITS);
FLINT_ASSERT(B->bits <= FLINT_BITS);
FLINT_ASSERT(A->length > 0);
FLINT_ASSERT(B->length > 0);
if (!(I->can_use & MPOLY_GCD_USE_HENSEL))
return 0;
FLINT_ASSERT(m >= WORD(2));
fq_nmod_mpoly_ctx_init(lctx, m, ORD_LEX, ctx->fqctx);
max_deg = 0;
for (i = 0; i < m; i++)
{
k = I->hensel_perm[i];
max_deg = FLINT_MAX(max_deg, I->Adeflate_deg[k]);
max_deg = FLINT_MAX(max_deg, I->Bdeflate_deg[k]);
}
wbits = 1 + FLINT_BIT_COUNT(max_deg);
wbits = mpoly_fix_bits(wbits, lctx->minfo);
FLINT_ASSERT(wbits <= FLINT_BITS);
fq_nmod_mpoly_init3(Al, 0, wbits, lctx);
fq_nmod_mpoly_init3(Bl, 0, wbits, lctx);
fq_nmod_mpoly_init3(Gl, 0, wbits, lctx);
fq_nmod_mpoly_init3(Abarl, 0, wbits, lctx);
fq_nmod_mpoly_init3(Bbarl, 0, wbits, lctx);
fq_nmod_mpoly_init3(Ac, 0, wbits, lctx);
fq_nmod_mpoly_init3(Bc, 0, wbits, lctx);
fq_nmod_mpoly_init3(Gc, 0, wbits, lctx);
fq_nmod_mpoly_init3(Abarc, 0, wbits, lctx);
fq_nmod_mpoly_init3(Bbarc, 0, wbits, lctx);
fq_nmod_mpoly_to_mpolyl_perm_deflate(Al, lctx, A, ctx,
I->hensel_perm, I->Amin_exp, I->Gstride);
fq_nmod_mpoly_to_mpolyl_perm_deflate(Bl, lctx, B, ctx,
I->hensel_perm, I->Bmin_exp, I->Gstride);
success = fq_nmod_mpolyl_content(Ac, Al, 1, lctx) &&
fq_nmod_mpolyl_content(Bc, Bl, 1, lctx);
if (!success)
goto cleanup;
if (Abar == NULL && Bbar == NULL)
success = fq_nmod_mpoly_gcd(Gc, Ac, Bc, lctx);
else
success = fq_nmod_mpoly_gcd_cofactors(Gc, Abarc, Bbarc, Ac, Bc, lctx);
if (!success)
goto cleanup;
success = fq_nmod_mpoly_divides(Al, Al, Ac, lctx);
FLINT_ASSERT(success);
success = fq_nmod_mpoly_divides(Bl, Bl, Bc, lctx);
FLINT_ASSERT(success);
fq_nmod_mpoly_repack_bits_inplace(Al, wbits, lctx);
fq_nmod_mpoly_repack_bits_inplace(Bl, wbits, lctx);
max_deg = I->Gdeflate_deg_bound[I->hensel_perm[0]];
success = fq_nmod_mpolyl_gcd_hensel_smprime(Gl, max_deg, Abarl, Bbarl, Al, Bl, lctx);
if (!success)
goto cleanup;
fq_nmod_mpoly_mul(Gl, Gl, Gc, lctx);
fq_nmod_mpoly_from_mpolyl_perm_inflate(G, I->Gbits, ctx, Gl, lctx,
I->hensel_perm, I->Gmin_exp, I->Gstride);
if (Abar != NULL)
{
fq_nmod_mpoly_mul(Abarl, Abarl, Abarc, lctx);
fq_nmod_mpoly_from_mpolyl_perm_inflate(Abar, I->Abarbits, ctx, Abarl, lctx,
I->hensel_perm, I->Abarmin_exp, I->Gstride);
}
if (Bbar != NULL)
{
fq_nmod_mpoly_mul(Bbarl, Bbarl, Bbarc, lctx);
fq_nmod_mpoly_from_mpolyl_perm_inflate(Bbar, I->Bbarbits, ctx, Bbarl, lctx,
I->hensel_perm, I->Bbarmin_exp, I->Gstride);
}
success = 1;
cleanup:
fq_nmod_mpoly_clear(Al, lctx);
fq_nmod_mpoly_clear(Bl, lctx);
fq_nmod_mpoly_clear(Gl, lctx);
fq_nmod_mpoly_clear(Abarl, lctx);
fq_nmod_mpoly_clear(Bbarl, lctx);
fq_nmod_mpoly_clear(Ac, lctx);
fq_nmod_mpoly_clear(Bc, lctx);
fq_nmod_mpoly_clear(Gc, lctx);
fq_nmod_mpoly_clear(Abarc, lctx);
fq_nmod_mpoly_clear(Bbarc, lctx);
fq_nmod_mpoly_ctx_clear(lctx);
return success;
}
static int _try_brown(
fq_nmod_mpoly_t G,
fq_nmod_mpoly_t Abar,
fq_nmod_mpoly_t Bbar,
const fq_nmod_mpoly_t A,
const fq_nmod_mpoly_t B,
mpoly_gcd_info_t I,
const fq_nmod_mpoly_ctx_t ctx)
{
int success;
slong m = I->mvars;
flint_bitcnt_t wbits;
fq_nmod_mpoly_ctx_t nctx;
fq_nmod_mpolyn_t An, Bn, Gn, Abarn, Bbarn;
if (!(I->can_use & MPOLY_GCD_USE_BROWN))
return 0;
FLINT_ASSERT(m >= 2);
FLINT_ASSERT(A->bits <= FLINT_BITS);
FLINT_ASSERT(B->bits <= FLINT_BITS);
FLINT_ASSERT(A->length > 0);
FLINT_ASSERT(B->length > 0);
wbits = FLINT_MAX(A->bits, B->bits);
fq_nmod_mpoly_ctx_init(nctx, m, ORD_LEX, ctx->fqctx);
fq_nmod_mpolyn_init(An, wbits, nctx);
fq_nmod_mpolyn_init(Bn, wbits, nctx);
fq_nmod_mpolyn_init(Gn, wbits, nctx);
fq_nmod_mpolyn_init(Abarn, wbits, nctx);
fq_nmod_mpolyn_init(Bbarn, wbits, nctx);
fq_nmod_mpoly_to_mpolyn_perm_deflate(An, nctx, A, ctx,
I->brown_perm, I->Amin_exp, I->Gstride);
fq_nmod_mpoly_to_mpolyn_perm_deflate(Bn, nctx, B, ctx,
I->brown_perm, I->Bmin_exp, I->Gstride);
FLINT_ASSERT(An->bits == wbits);
FLINT_ASSERT(Bn->bits == wbits);
FLINT_ASSERT(An->length > 1);
FLINT_ASSERT(Bn->length > 1);
success = fq_nmod_mpolyn_gcd_brown_smprime(Gn, Abarn, Bbarn, An, Bn,
m - 1, nctx);
if (!success)
{
fq_nmod_mpoly_to_mpolyn_perm_deflate(An, nctx, A, ctx,
I->brown_perm, I->Amin_exp, I->Gstride);
fq_nmod_mpoly_to_mpolyn_perm_deflate(Bn, nctx, B, ctx,
I->brown_perm, I->Bmin_exp, I->Gstride);
success = fq_nmod_mpolyn_gcd_brown_lgprime(Gn, Abarn, Bbarn, An, Bn,
m - 1, nctx);
}
if (!success)
goto cleanup;
fq_nmod_mpoly_from_mpolyn_perm_inflate(G, I->Gbits, ctx, Gn, nctx,
I->brown_perm, I->Gmin_exp, I->Gstride);
if (Abar != NULL)
fq_nmod_mpoly_from_mpolyn_perm_inflate(Abar, I->Abarbits, ctx, Abarn, nctx,
I->brown_perm, I->Abarmin_exp, I->Gstride);
if (Bbar != NULL)
fq_nmod_mpoly_from_mpolyn_perm_inflate(Bbar, I->Bbarbits, ctx, Bbarn, nctx,
I->brown_perm, I->Bbarmin_exp, I->Gstride);
success = 1;
cleanup:
fq_nmod_mpolyn_clear(An, nctx);
fq_nmod_mpolyn_clear(Bn, nctx);
fq_nmod_mpolyn_clear(Gn, nctx);
fq_nmod_mpolyn_clear(Abarn, nctx);
fq_nmod_mpolyn_clear(Bbarn, nctx);
fq_nmod_mpoly_ctx_clear(nctx);
return success;
}
static int _fq_nmod_mpoly_gcd_algo_small(
fq_nmod_mpoly_t G,
fq_nmod_mpoly_t Abar,
fq_nmod_mpoly_t Bbar,
const fq_nmod_mpoly_t A,
const fq_nmod_mpoly_t B,
const fq_nmod_mpoly_ctx_t ctx,
unsigned int algo)
{
int success;
flint_bitcnt_t Gbits = FLINT_MIN(A->bits, B->bits);
flint_bitcnt_t Abarbits = A->bits;
flint_bitcnt_t Bbarbits = B->bits;
slong v_in_both;
slong v_in_either;
slong v_in_A_only;
slong v_in_B_only;
slong j;
slong nvars = ctx->minfo->nvars;
mpoly_gcd_info_t I;
#if FLINT_WANT_ASSERT
fq_nmod_mpoly_t T, Asave, Bsave;
#endif
if (A->length == 1)
return _do_monomial_gcd(G, Bbar, Abar, B, A, ctx);
else if (B->length == 1)
return _do_monomial_gcd(G, Abar, Bbar, A, B, ctx);
#if FLINT_WANT_ASSERT
fq_nmod_mpoly_init(T, ctx);
fq_nmod_mpoly_init(Asave, ctx);
fq_nmod_mpoly_init(Bsave, ctx);
fq_nmod_mpoly_set(Asave, A, ctx);
fq_nmod_mpoly_set(Bsave, B, ctx);
#endif
mpoly_gcd_info_init(I, nvars);
I->Gbits = Gbits;
I->Abarbits = Abarbits;
I->Bbarbits = Bbarbits;
mpoly_gcd_info_limits(I->Amax_exp, I->Amin_exp, I->Alead_count,
I->Atail_count, A->exps, A->bits, A->length, ctx->minfo);
mpoly_gcd_info_limits(I->Bmax_exp, I->Bmin_exp, I->Blead_count,
I->Btail_count, B->exps, B->bits, B->length, ctx->minfo);
for (j = 0; j < nvars; j++)
{
if (I->Amax_exp[j] - I->Amin_exp[j] != I->Bmax_exp[j] - I->Bmin_exp[j])
goto skip_monomial_cofactors;
}
if (_try_monomial_cofactors(G, Abar, Bbar, A, B, ctx))
{
goto successful;
}
skip_monomial_cofactors:
mpoly_gcd_info_stride(I->Gstride,
A->exps, A->bits, A->length, I->Amax_exp, I->Amin_exp,
B->exps, B->bits, B->length, I->Bmax_exp, I->Bmin_exp, ctx->minfo);
for (j = 0; j < nvars; j++)
{
ulong t = I->Gstride[j];
if (t == 0)
{
FLINT_ASSERT(I->Amax_exp[j] == I->Amin_exp[j] ||
I->Bmax_exp[j] == I->Bmin_exp[j]);
}
else
{
FLINT_ASSERT((I->Amax_exp[j] - I->Amin_exp[j]) % t == 0);
FLINT_ASSERT((I->Bmax_exp[j] - I->Bmin_exp[j]) % t == 0);
}
I->Adeflate_deg[j] = t == 0 ? 0 : (I->Amax_exp[j] - I->Amin_exp[j])/t;
I->Bdeflate_deg[j] = t == 0 ? 0 : (I->Bmax_exp[j] - I->Bmin_exp[j])/t;
t = FLINT_MIN(I->Amin_exp[j], I->Bmin_exp[j]);
I->Gmin_exp[j] = t;
I->Abarmin_exp[j] = I->Amin_exp[j] - t;
I->Bbarmin_exp[j] = I->Bmin_exp[j] - t;
}
v_in_both = -WORD(1);
for (j = 0; j < nvars; j++)
{
if (I->Amax_exp[j] > I->Amin_exp[j] && I->Bmax_exp[j] > I->Bmin_exp[j])
{
v_in_both = j;
break;
}
}
if (v_in_both == -WORD(1))
{
_do_trivial(G, Abar, Bbar, A, B, I, ctx);
goto successful;
}
FLINT_ASSERT(0 <= v_in_both && v_in_both < nvars);
v_in_either = -WORD(1);
for (j = 0; j < nvars; j++)
{
if (j == v_in_both)
continue;
if (I->Amax_exp[j] > I->Amin_exp[j] || I->Bmax_exp[j] > I->Bmin_exp[j])
{
v_in_either = j;
break;
}
}
if (v_in_either == -WORD(1))
{
_do_univar(G, Abar, Bbar, A, B, v_in_both, I, ctx);
goto successful;
}
v_in_A_only = -WORD(1);
v_in_B_only = -WORD(1);
for (j = 0; j < nvars; j++)
{
if (I->Amax_exp[j] > I->Amin_exp[j] && I->Bmax_exp[j] == I->Bmin_exp[j])
{
v_in_A_only = j;
break;
}
if (I->Bmax_exp[j] > I->Bmin_exp[j] && I->Amax_exp[j] == I->Amin_exp[j])
{
v_in_B_only = j;
break;
}
}
if (v_in_A_only != -WORD(1))
{
success = _try_missing_var(G, I->Gbits, Abar, Bbar, v_in_A_only,
A, I->Amin_exp[v_in_A_only],
B, I->Bmin_exp[v_in_A_only],
ctx);
goto cleanup;
}
if (v_in_B_only != -WORD(1))
{
success = _try_missing_var(G, I->Gbits, Bbar, Abar, v_in_B_only,
B, I->Bmin_exp[v_in_B_only],
A, I->Amin_exp[v_in_B_only],
ctx);
goto cleanup;
}
mpoly_gcd_info_set_estimates_fq_nmod_mpoly(I, A, B, ctx);
if (!I->Gdeflate_deg_bounds_are_nice)
mpoly_gcd_info_set_estimates_fq_nmod_mpoly_lgprime(I, A, B, ctx);
mpoly_gcd_info_set_perm(I, A->length, B->length, ctx->minfo);
{
int gcd_is_trivial = 1;
int try_a = I->Gdeflate_deg_bounds_are_nice;
int try_b = I->Gdeflate_deg_bounds_are_nice;
for (j = 0; j < nvars; j++)
{
if (I->Gdeflate_deg_bound[j] != 0)
gcd_is_trivial = 0;
if (I->Adeflate_deg[j] != I->Gdeflate_deg_bound[j])
try_a = 0;
if (I->Bdeflate_deg[j] != I->Gdeflate_deg_bound[j])
try_b = 0;
}
if (gcd_is_trivial)
{
_do_trivial(G, Abar, Bbar, A, B, I, ctx);
goto successful;
}
if (try_a && _try_divides(G, Bbar, Abar, B, A, ctx))
goto successful;
if (try_b && _try_divides(G, Abar, Bbar, A, B, ctx))
goto successful;
}
if (algo == MPOLY_GCD_USE_HENSEL)
{
mpoly_gcd_info_measure_hensel(I, A->length, B->length, ctx->minfo);
success = _try_hensel(G, Abar, Bbar, A, B, I, ctx);
goto cleanup;
}
else if (algo == MPOLY_GCD_USE_BROWN)
{
mpoly_gcd_info_measure_brown(I, A->length, B->length, ctx->minfo);
success = _try_brown(G, Abar, Bbar, A, B, I, ctx);
goto cleanup;
}
else if (algo == MPOLY_GCD_USE_ZIPPEL)
{
mpoly_gcd_info_measure_zippel(I, A->length, B->length, ctx->minfo);
success = _try_zippel(G, Abar, Bbar, A, B, I, ctx);
goto cleanup;
}
else if (I->mvars < 3)
{
mpoly_gcd_info_measure_brown(I, A->length, B->length, ctx->minfo);
mpoly_gcd_info_measure_hensel(I, A->length, B->length, ctx->minfo);
algo &= (MPOLY_GCD_USE_BROWN | MPOLY_GCD_USE_HENSEL);
if (algo == MPOLY_GCD_USE_BROWN)
{
success = _try_brown(G, Abar, Bbar, A, B, I, ctx);
}
else if (algo == MPOLY_GCD_USE_HENSEL)
{
success = _try_hensel(G, Abar, Bbar, A, B, I, ctx);
}
else
{
if (I->Adensity + I->Bdensity > 0.05)
{
success = _try_brown(G, Abar, Bbar, A, B, I, ctx) ||
_try_hensel(G, Abar, Bbar, A, B, I, ctx);
}
else
{
success = _try_hensel(G, Abar, Bbar, A, B, I, ctx) ||
_try_brown(G, Abar, Bbar, A, B, I, ctx);
}
}
goto cleanup;
}
else if (algo == MPOLY_GCD_USE_ZIPPEL2)
{
mpoly_gcd_info_measure_zippel2(I, A->length, B->length, ctx->minfo);
success = _try_zippel2(G, Abar, Bbar, A, B, I, ctx);
goto cleanup;
}
else
{
slong k;
double density = I->Adensity + I->Bdensity;
mpoly_gcd_info_measure_hensel(I, A->length, B->length, ctx->minfo);
mpoly_gcd_info_measure_brown(I, A->length, B->length, ctx->minfo);
mpoly_gcd_info_measure_zippel(I, A->length, B->length, ctx->minfo);
mpoly_gcd_info_measure_zippel2(I, A->length, B->length, ctx->minfo);
if (density > 0.08)
{
if (_try_brown(G, Abar, Bbar, A, B, I, ctx))
goto successful;
}
if (I->Adeflate_tdeg > 0 && I->Bdeflate_tdeg > 0)
{
fmpz_t x;
double tdensity;
fmpz_init(x);
fmpz_bin_uiui(x, (ulong)I->Adeflate_tdeg + I->mvars, I->mvars);
tdensity = A->length/fmpz_get_d(x);
fmpz_bin_uiui(x, (ulong)I->Bdeflate_tdeg + I->mvars, I->mvars);
tdensity += B->length/fmpz_get_d(x);
density = FLINT_MAX(density, tdensity);
fmpz_clear(x);
}
if (density > 0.05)
{
if (_try_hensel(G, Abar, Bbar, A, B, I, ctx))
goto successful;
}
k = I->zippel2_perm[1];
k = FLINT_MAX(I->Adeflate_deg[k], I->Bdeflate_deg[k]);
if ((A->length + B->length)/64 < k)
{
if (_try_zippel(G, Abar, Bbar, A, B, I, ctx))
goto successful;
if (_try_zippel2(G, Abar, Bbar, A, B, I, ctx))
goto successful;
}
else
{
if (_try_zippel2(G, Abar, Bbar, A, B, I, ctx))
goto successful;
if (_try_zippel(G, Abar, Bbar, A, B, I, ctx))
goto successful;
}
success = _try_hensel(G, Abar, Bbar, A, B, I, ctx) ||
_try_brown(G, Abar, Bbar, A, B, I, ctx);
goto cleanup;
}
success = 0;
goto cleanup;
successful:
success = 1;
cleanup:
mpoly_gcd_info_clear(I);
if (success)
{
FLINT_ASSERT(G->length > 0);
if (!n_fq_is_one(G->coeffs + 0, ctx->fqctx))
{
if (Abar != NULL)
fq_nmod_mpoly_scalar_mul_n_fq(Abar, Abar, G->coeffs + 0, ctx);
if (Bbar != NULL)
fq_nmod_mpoly_scalar_mul_n_fq(Bbar, Bbar, G->coeffs + 0, ctx);
fq_nmod_mpoly_make_monic(G, G, ctx);
}
FLINT_ASSERT(fq_nmod_mpoly_divides(T, Asave, G, ctx));
FLINT_ASSERT(Abar == NULL || fq_nmod_mpoly_equal(T, Abar, ctx));
FLINT_ASSERT(fq_nmod_mpoly_divides(T, Bsave, G, ctx));
FLINT_ASSERT(Bbar == NULL || fq_nmod_mpoly_equal(T, Bbar, ctx));
}
#if FLINT_WANT_ASSERT
fq_nmod_mpoly_clear(T, ctx);
fq_nmod_mpoly_clear(Asave, ctx);
fq_nmod_mpoly_clear(Bsave, ctx);
#endif
return success;
}
static int _fq_nmod_mpoly_gcd_algo_large(
fq_nmod_mpoly_t G,
fq_nmod_mpoly_t Abar,
fq_nmod_mpoly_t Bbar,
const fq_nmod_mpoly_t A,
const fq_nmod_mpoly_t B,
const fq_nmod_mpoly_ctx_t ctx,
unsigned int algo)
{
int success;
slong k;
fmpz * Ashift, * Astride;
fmpz * Bshift, * Bstride;
fmpz * Gshift, * Gstride;
fq_nmod_mpoly_t Anew, Bnew;
const fq_nmod_mpoly_struct * Ause, * Buse;
if (A->length == 1)
return _do_monomial_gcd(G, Bbar, Abar, B, A, ctx);
if (B->length == 1)
return _do_monomial_gcd(G, Abar, Bbar, A, B, ctx);
if (_try_monomial_cofactors(G, Abar, Bbar, A, B, ctx))
return 1;
fq_nmod_mpoly_init(Anew, ctx);
fq_nmod_mpoly_init(Bnew, ctx);
Ause = A;
if (A->bits > FLINT_BITS)
{
if (!fq_nmod_mpoly_repack_bits(Anew, A, FLINT_BITS, ctx))
goto could_not_repack;
Ause = Anew;
}
Buse = B;
if (B->bits > FLINT_BITS)
{
if (!fq_nmod_mpoly_repack_bits(Bnew, B, FLINT_BITS, ctx))
goto could_not_repack;
Buse = Bnew;
}
success = _fq_nmod_mpoly_gcd_algo_small(G, Abar, Bbar, Ause, Buse, ctx, algo);
goto cleanup;
could_not_repack:
Ashift = _fmpz_vec_init(ctx->minfo->nvars);
Astride = _fmpz_vec_init(ctx->minfo->nvars);
Bshift = _fmpz_vec_init(ctx->minfo->nvars);
Bstride = _fmpz_vec_init(ctx->minfo->nvars);
Gshift = _fmpz_vec_init(ctx->minfo->nvars);
Gstride = _fmpz_vec_init(ctx->minfo->nvars);
fq_nmod_mpoly_deflation(Ashift, Astride, A, ctx);
fq_nmod_mpoly_deflation(Bshift, Bstride, B, ctx);
_fmpz_vec_min(Gshift, Ashift, Bshift, ctx->minfo->nvars);
for (k = 0; k < ctx->minfo->nvars; k++)
{
fmpz_gcd(Gstride + k, Astride + k, Bstride + k);
}
fq_nmod_mpoly_deflate(Anew, A, Ashift, Gstride, ctx);
if (Anew->bits > FLINT_BITS)
{
success = fq_nmod_mpoly_repack_bits(Anew, Anew, FLINT_BITS, ctx);
if (!success)
goto deflate_cleanup;
}
fq_nmod_mpoly_deflate(Bnew, B, Bshift, Gstride, ctx);
if (Bnew->bits > FLINT_BITS)
{
success = fq_nmod_mpoly_repack_bits(Bnew, Bnew, FLINT_BITS, ctx);
if (!success)
goto deflate_cleanup;
}
success = _fq_nmod_mpoly_gcd_algo(G, Abar, Bbar, Anew, Bnew, ctx, algo);
if (!success)
goto deflate_cleanup;
for (k = 0; k < ctx->minfo->nvars; k++)
{
fmpz_sub(Ashift + k, Ashift + k, Gshift + k);
fmpz_sub(Bshift + k, Bshift + k, Gshift + k);
FLINT_ASSERT(fmpz_sgn(Ashift + k) >= 0);
FLINT_ASSERT(fmpz_sgn(Bshift + k) >= 0);
}
fq_nmod_mpoly_inflate(G, G, Gshift, Gstride, ctx);
if (Abar != NULL)
fq_nmod_mpoly_inflate(Abar, Abar, Ashift, Gstride, ctx);
if (Bbar != NULL)
fq_nmod_mpoly_inflate(Bbar, Bbar, Bshift, Gstride, ctx);
FLINT_ASSERT(G->length > 0);
if (!n_fq_is_one(G->coeffs + 0, ctx->fqctx))
{
if (Abar != NULL)
fq_nmod_mpoly_scalar_mul_n_fq(Abar, Abar, G->coeffs + 0, ctx);
if (Bbar != NULL)
fq_nmod_mpoly_scalar_mul_n_fq(Bbar, Bbar, G->coeffs + 0, ctx);
fq_nmod_mpoly_make_monic(G, G, ctx);
}
fq_nmod_mpoly_make_monic(G, G, ctx);
deflate_cleanup:
_fmpz_vec_clear(Ashift, ctx->minfo->nvars);
_fmpz_vec_clear(Astride, ctx->minfo->nvars);
_fmpz_vec_clear(Bshift, ctx->minfo->nvars);
_fmpz_vec_clear(Bstride, ctx->minfo->nvars);
_fmpz_vec_clear(Gshift, ctx->minfo->nvars);
_fmpz_vec_clear(Gstride, ctx->minfo->nvars);
cleanup:
fq_nmod_mpoly_clear(Anew, ctx);
fq_nmod_mpoly_clear(Bnew, ctx);
return success;
}
int _fq_nmod_mpoly_gcd_algo(
fq_nmod_mpoly_t G,
fq_nmod_mpoly_t Abar,
fq_nmod_mpoly_t Bbar,
const fq_nmod_mpoly_t A,
const fq_nmod_mpoly_t B,
const fq_nmod_mpoly_ctx_t ctx,
unsigned int algo)
{
FLINT_ASSERT(!fq_nmod_mpoly_is_zero(A, ctx));
FLINT_ASSERT(!fq_nmod_mpoly_is_zero(B, ctx));
if (A->bits <= FLINT_BITS && B->bits <= FLINT_BITS)
return _fq_nmod_mpoly_gcd_algo_small(G, Abar, Bbar, A, B, ctx, algo);
else
return _fq_nmod_mpoly_gcd_algo_large(G, Abar, Bbar, A, B, ctx, algo);
}
int fq_nmod_mpoly_gcd(
fq_nmod_mpoly_t G,
const fq_nmod_mpoly_t A,
const fq_nmod_mpoly_t B,
const fq_nmod_mpoly_ctx_t ctx)
{
if (fq_nmod_mpoly_is_zero(A, ctx))
{
if (fq_nmod_mpoly_is_zero(B, ctx))
fq_nmod_mpoly_zero(G, ctx);
else
fq_nmod_mpoly_make_monic(G, B, ctx);
return 1;
}
if (fq_nmod_mpoly_is_zero(B, ctx))
{
fq_nmod_mpoly_make_monic(G, A, ctx);
return 1;
}
return _fq_nmod_mpoly_gcd_algo(G, NULL, NULL, A, B, ctx, MPOLY_GCD_USE_ALL);
}