flint-sys 0.9.0

Bindings to the FLINT C library
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126

/*
    Copyright (C) 2018 Daniel Schultz

    This file is part of FLINT.

    FLINT is free software: you can redistribute it and/or modify it under
    the terms of the GNU Lesser General Public License (LGPL) as published
    by the Free Software Foundation; either version 3 of the License, or
    (at your option) any later version.  See <https://www.gnu.org/licenses/>.
*/

#include "mpoly.h"
#include "fmpq_mpoly.h"

void _fmpq_mpoly_set_coeff_fmpq_fmpz(fmpq_mpoly_t qpoly,
                  const fmpq_t c, const fmpz * exp, const fmpq_mpoly_ctx_t qctx)
{
    flint_bitcnt_t exp_bits;
    slong i, N, index;
    ulong * cmpmask;
    ulong * packed_exp;
    int exists;
    fmpz_mpoly_struct * poly = qpoly->zpoly;
    const fmpz_mpoly_ctx_struct * ctx = qctx->zctx;
    TMP_INIT;

    TMP_START;

    exp_bits = mpoly_exp_bits_required_ffmpz(exp, ctx->minfo);
    exp_bits = mpoly_fix_bits(exp_bits, ctx->minfo);
    fmpz_mpoly_fit_bits(poly, exp_bits, ctx);

    N = mpoly_words_per_exp(poly->bits, ctx->minfo);
    cmpmask = (ulong*) TMP_ALLOC(N*sizeof(ulong));
    mpoly_get_cmpmask(cmpmask, N, poly->bits, ctx->minfo);

    packed_exp = (ulong *) TMP_ALLOC(N*sizeof(ulong));

    mpoly_set_monomial_ffmpz(packed_exp, exp, poly->bits, ctx->minfo);
    exists = mpoly_monomial_exists(&index, poly->exps,
                                  packed_exp, poly->length, N, cmpmask);

    if (!exists)
    {
        if (!fmpq_is_zero(c)) /* make new term only if coeff is nonzero*/
        {
            fmpz_mpoly_fit_length(poly, poly->length + 1, ctx);
            if (poly->length > 0)
            {
                fmpz_t prod;
                fmpz_init(prod);

                fmpz_mul(prod, fmpq_numref(qpoly->content), fmpq_denref(c));
                fmpz_mpoly_scalar_mul_fmpz(poly, poly, prod, ctx);

                for (i = poly->length; i >= index + 1; i--)
                {
                    fmpz_set(poly->coeffs + i, poly->coeffs + i - 1);
                    mpoly_monomial_set(poly->exps + N*i, poly->exps + N*(i - 1), N);
                }
                fmpz_mul(poly->coeffs + index, fmpq_denref(qpoly->content), fmpq_numref(c));

                fmpq_div_fmpz(qpoly->content, qpoly->content, prod);
                fmpz_clear(prod);

            } else
            {
                index = 0;
                fmpz_one(poly->coeffs + index);
                fmpq_set(qpoly->content, c);
            }

            mpoly_monomial_set(poly->exps + N*index, packed_exp, N);
            poly->length++; /* safe because length is increasing */
        }
    } else if (fmpq_is_zero(c)) /* zero coeff, remove term */
    {
        for (i = index; i < poly->length - 1; i++)
        {
            fmpz_set(poly->coeffs + i, poly->coeffs + i + 1);
            mpoly_monomial_set(poly->exps + N*i, poly->exps + N*(i + 1), N);
        }

        _fmpz_mpoly_set_length(poly, poly->length - 1, ctx);

    } else /* term with that monomial exists, coeff is nonzero */
    {
        fmpz_t prod;
        fmpz_init(prod);

        fmpz_mul(prod, fmpq_numref(qpoly->content), fmpq_denref(c));
        fmpz_mpoly_scalar_mul_fmpz(poly, poly, prod, ctx);

        fmpz_mul(poly->coeffs + index, fmpq_denref(qpoly->content), fmpq_numref(c));
        fmpq_div_fmpz(qpoly->content, qpoly->content, prod);
        fmpz_clear(prod);
    }

    fmpq_mpoly_reduce(qpoly, qctx);

    TMP_END;
}


void fmpq_mpoly_set_coeff_fmpq_fmpz(fmpq_mpoly_t poly,
                const fmpq_t c, fmpz * const * exp, const fmpq_mpoly_ctx_t ctx)
{
    slong i, nvars = ctx->zctx->minfo->nvars;
    fmpz * newexp;
    TMP_INIT;

    TMP_START;
    newexp = (nvars > 0) ? TMP_ALLOC(nvars*sizeof(fmpz)) : NULL;
    for (i = 0; i < nvars; i++)
    {
        fmpz_init(newexp + i);
        fmpz_set(newexp + i, exp[i]);
    }

    _fmpq_mpoly_set_coeff_fmpq_fmpz(poly, c, newexp, ctx);

    for (i = 0; i < nvars; i++)
        fmpz_clear(newexp + i);

    TMP_END;
}