pqc/external/flint-2.4.3/padic_polyxx.h
2014-05-24 23:16:06 +02:00

399 lines
14 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) 2013 Tom Bachmann
******************************************************************************/
#ifndef PADIC_POLYXX_H
#define PADIC_POLYXX_H
#include "padic_poly.h"
#include "padicxx.h"
#include "fmpz_polyxx.h"
#include "fmpq_polyxx.h"
#include "flintxx/stdmath.h"
// TODO input and output
namespace flint {
FLINT_DEFINE_BINOP(padic_polyxx_get_coeff)
FLINT_DEFINE_BINOP(compose_pow)
namespace detail {
template<class Padic>
struct padic_poly_traits
{
typedef slong prec_ref_t;
typedef slong prec_srcref_t;
typedef slong val_ref_t;
typedef slong val_srcref_t;
static slong prec(const Padic& p) {return tools::padic_output_prec(p);}
static slong val(const Padic& p) {return padic_poly_val(p.evaluate()._poly());}
};
} //detail
template<class Operation, class Data>
class padic_polyxx_expression
: public expression<derived_wrapper<padic_polyxx_expression>, Operation, Data>
{
public:
typedef expression<derived_wrapper< ::flint::padic_polyxx_expression>,
Operation, Data> base_t;
FLINTXX_DEFINE_BASICS(padic_polyxx_expression)
FLINTXX_DEFINE_CTORS(padic_polyxx_expression)
FLINTXX_DEFINE_C_REF(padic_polyxx_expression, padic_poly_struct, _poly)
public:
typedef detail::padic_poly_traits<padic_polyxx_expression> traits_t;
PADICXX_DEFINE_STD
static padic_polyxx_expression zero(padicxx_ctx_srcref ctx)
{return padic_polyxx_expression(ctx);}
static padic_polyxx_expression zero(padicxx_ctx_srcref ctx, slong N)
{return padic_polyxx_expression(ctx, N);}
static padic_polyxx_expression one(padicxx_ctx_srcref ctx)
{
padic_polyxx_expression res(ctx);
res.set_one();
return res;
}
static padic_polyxx_expression one(padicxx_ctx_srcref ctx, slong N)
{
padic_polyxx_expression res(ctx, N);
res.set_one();
return res;
}
template<class T>
static padic_polyxx_expression from_QQ(const T& q, padicxx_ctx_srcref ctx,
typename mp::enable_if<mp::or_<traits::is_fmpqxx<T>,
traits::is_fmpzxx<T>, traits::is_integer<T> > >::type* = 0)
{
padic_polyxx_expression res(ctx);
res = q;
return res;
}
template<class T>
static padic_polyxx_expression from_QQ(const T& q, padicxx_ctx_srcref ctx,
slong N,
typename mp::enable_if<mp::or_<traits::is_fmpqxx<T>,
traits::is_fmpzxx<T>, traits::is_integer<T> > >::type* = 0)
{
padic_polyxx_expression res(ctx, N);
res = q;
return res;
}
template<class T>
static padic_polyxx_expression from_QQX(const T& q, padicxx_ctx_srcref ctx,
typename mp::enable_if<mp::or_<traits::is_fmpq_polyxx<T>,
traits::is_fmpz_polyxx<T> > >::type* = 0)
{
padic_polyxx_expression res(ctx);
res = q;
return res;
}
template<class T>
static padic_polyxx_expression from_QQX(const T& q, padicxx_ctx_srcref ctx,
slong N,
typename mp::enable_if<mp::or_<traits::is_fmpq_polyxx<T>,
traits::is_fmpz_polyxx<T> > >::type* = 0)
{
padic_polyxx_expression res(ctx, N);
res = q;
return res;
}
template<class T>
static padic_polyxx_expression _from_ground(const T& q)
{
padic_polyxx_expression res(q.get_ctx(), q.prec());
res = q;
return res;
}
template<class T>
static padic_polyxx_expression from_ground(const T& q,
typename mp::enable_if<traits::is_padicxx<T> >::type* = 0)
{
return _from_ground(q.evaluate());
}
// Create a temporary. The context will be estimated, and the precision
// will be the maximum of all subexpressions.
evaluated_t create_temporary() const
{
return evaluated_t(estimate_ctx(), prec());
}
// static methods which only make sense with padicxx
static padic_polyxx_expression randtest(frandxx& state,
slong len,
padicxx_ctx_srcref ctx, slong prec = PADIC_DEFAULT_PREC)
{
padic_polyxx_expression res(ctx, prec);
padic_poly_randtest(res._poly(), state._data(), len, ctx._ctx());
return res;
}
static padic_polyxx_expression randtest_not_zero(frandxx& state,
slong len,
padicxx_ctx_srcref ctx, slong prec = PADIC_DEFAULT_PREC)
{
padic_polyxx_expression res(ctx, prec);
padic_poly_randtest_not_zero(res._poly(), state._data(), len, ctx._ctx());
return res;
}
static padic_polyxx_expression randtest_val(frandxx& state,
slong val, slong len,
padicxx_ctx_srcref ctx, slong prec = PADIC_DEFAULT_PREC)
{
padic_polyxx_expression res(ctx, prec);
padic_poly_randtest_val(res._poly(), state._data(), val, len, ctx._ctx());
return res;
}
// These only make sense with immediates
void reduce() {padic_poly_reduce(_poly(), _ctx());}
void realloc(slong alloc) {padic_poly_realloc(_poly(), alloc);}
void fit_length(slong len) {padic_poly_fit_length(_poly(), len);}
void _normalise() {_padic_poly_normalise(_poly());}
void _set_length(slong len) {_padic_poly_set_length(_poly(), len);}
void set_zero() {padic_poly_zero(_poly());}
void set_one() {padic_poly_one(_poly());}
void truncate(slong n) {fmpz_poly_truncate(_poly(), n);}
void canonicalise() {padic_poly_canonicalise(_poly());}
bool is_canonical() const {return padic_poly_is_canonical(_poly());}
bool is_reduced() const {return padic_poly_is_reduced(_poly());}
template<class Padic>
void set_coeff(slong n, const Padic& p,
typename mp::enable_if<traits::is_padicxx<Padic> >::type* = 0)
{padic_poly_set_coeff_padic(_poly(), n, p._padic(), _ctx());}
// these cause evaluation
bool is_zero() const {return padic_poly_is_zero(this->evaluate()._poly());}
bool is_one() const {return padic_poly_is_one(this->evaluate()._poly());}
slong length() const {return padic_poly_length(this->evaluate()._poly());}
slong degree() const {return padic_poly_degree(this->evaluate()._poly());}
// forwarding of lazy functions
FLINTXX_DEFINE_MEMBER_BINOP_(get_coeff, padic_polyxx_get_coeff)
FLINTXX_DEFINE_MEMBER_BINOP_(operator(), compeval)
FLINTXX_DEFINE_MEMBER_BINOP(pow)
FLINTXX_DEFINE_MEMBER_BINOP(compose_pow)
FLINTXX_DEFINE_MEMBER_BINOP(inv_series)
FLINTXX_DEFINE_MEMBER_BINOP(shift_left)
FLINTXX_DEFINE_MEMBER_BINOP(shift_right)
FLINTXX_DEFINE_MEMBER_UNOP(derivative)
};
namespace detail {
struct padic_poly_data;
}
typedef padic_polyxx_expression<operations::immediate,
detail::padic_poly_data> padic_polyxx;
typedef padic_polyxx_expression<operations::immediate,
flint_classes::ref_data<padic_polyxx, padic_poly_struct> >
padic_polyxx_ref;
typedef padic_polyxx_expression<operations::immediate, flint_classes::srcref_data<
padic_polyxx, padic_polyxx_ref, padic_poly_struct> > padic_polyxx_srcref;
namespace traits {
template<> struct has_padicxx_ctx<padic_polyxx> : mp::true_ { };
template<> struct has_padicxx_ctx<padic_polyxx_ref> : mp::true_ { };
template<> struct has_padicxx_ctx<padic_polyxx_srcref> : mp::true_ { };
} // traits
namespace detail {
template<>
struct padic_poly_traits<padic_polyxx_srcref>
{
typedef slong prec_ref_t;
typedef slong prec_srcref_t;
typedef slong val_ref_t;
typedef slong val_srcref_t;
template<class P>
static slong prec(P p) {return p._data().N;}
template<class P>
static slong val(P p) {return padic_poly_val(p._poly());}
};
template<>
struct padic_poly_traits<padic_polyxx_ref>
: padic_poly_traits<padic_polyxx_srcref>
{
typedef slong& prec_ref_t;
typedef slong& val_ref_t;
template<class P>
static slong& prec(P& p) {return padic_poly_prec(p._poly());}
template<class P>
static slong prec(const P& p) {return padic_poly_prec(p._poly());}
template<class P>
static slong& val(P& p) {return padic_poly_val(p._poly());}
template<class P>
static slong val(const P& p) {return padic_poly_val(p._poly());}
};
template<>
struct padic_poly_traits<padic_polyxx>
: padic_poly_traits<padic_polyxx_ref> { };
} // detail
PADICXX_DEFINE_REF_STRUCTS(padic_polyxx, padic_poly_struct, padic_poly_prec)
namespace detail {
struct padic_poly_data
{
typedef padic_poly_t& data_ref_t;
typedef const padic_poly_t& data_srcref_t;
padicxx_ctx_srcref ctx;
padic_poly_t inner;
padic_poly_data(padicxx_ctx_srcref c)
: ctx(c)
{
padic_poly_init(inner);
}
padic_poly_data(padicxx_ctx_srcref c, slong N, slong alloc = 0)
: ctx(c)
{
padic_poly_init2(inner, alloc, N);
}
padic_poly_data(const padic_poly_data& o)
: ctx(o.ctx)
{
padic_poly_init2(inner, padic_poly_length(o.inner),
padic_poly_prec(o.inner));
padic_poly_set(inner, o.inner, ctx._ctx());
}
~padic_poly_data() {padic_poly_clear(inner);}
padic_poly_data(padic_polyxx_srcref c)
: ctx(c.get_ctx())
{
padic_poly_init2(inner, c.length(), c.prec());
padic_poly_set(inner, c._poly(), ctx._ctx());
}
};
} // detail
#define PADIC_POLYXX_COND_S FLINTXX_COND_S(padic_polyxx)
#define PADIC_POLYXX_COND_T FLINTXX_COND_T(padic_polyxx)
namespace rules {
FLINT_DEFINE_DOIT_COND2(assignment, PADIC_POLYXX_COND_T, PADIC_POLYXX_COND_S,
padic_poly_set(to._poly(), from._poly(), to._ctx()))
FLINT_DEFINE_DOIT_COND2(assignment, PADIC_POLYXX_COND_T, PADICXX_COND_S,
padic_poly_set_padic(to._poly(), from._padic(), to._ctx()))
FLINT_DEFINE_DOIT_COND2(assignment, PADIC_POLYXX_COND_T, traits::is_signed_integer,
padic_poly_set_si(to._poly(), from, to._ctx()))
FLINT_DEFINE_DOIT_COND2(assignment, PADIC_POLYXX_COND_T, traits::is_unsigned_integer,
padic_poly_set_ui(to._poly(), from, to._ctx()))
FLINT_DEFINE_DOIT_COND2(assignment, PADIC_POLYXX_COND_T, FMPZXX_COND_S,
padic_poly_set_fmpz(to._poly(), from._fmpz(), to._ctx()))
FLINT_DEFINE_DOIT_COND2(assignment, PADIC_POLYXX_COND_T, FMPQXX_COND_S,
padic_poly_set_fmpq(to._poly(), from._fmpq(), to._ctx()))
FLINT_DEFINE_DOIT_COND2(assignment, PADIC_POLYXX_COND_T, FMPZ_POLYXX_COND_S,
padic_poly_set_fmpz_poly(to._poly(), from._poly(), to._ctx()))
FLINT_DEFINE_DOIT_COND2(assignment, PADIC_POLYXX_COND_T, FMPQ_POLYXX_COND_S,
padic_poly_set_fmpq_poly(to._poly(), from._poly(), to._ctx()))
FLINTXX_DEFINE_SWAP(padic_polyxx, padic_poly_swap(e1._poly(), e2._poly()))
FLINTXX_DEFINE_EQUALS(padic_polyxx, padic_poly_equal(e1._poly(), e2._poly()))
FLINT_DEFINE_PRINT_COND(PADIC_POLYXX_COND_S,
padic_poly_fprint(to, from._poly(), from._ctx()))
FLINT_DEFINE_PRINT_PRETTY_COND_2(PADIC_POLYXX_COND_S, const char*,
padic_poly_fprint_pretty(to, from._poly(), extra, from._ctx()))
FLINTXX_DEFINE_CONVERSION_TMP(fmpz_polyxx, padic_polyxx,
execution_check(padic_poly_get_fmpz_poly(
to._poly(), from._poly(), from._ctx()),
"to<fmpz_polyxx>", "padic_polyxx"))
FLINTXX_DEFINE_CONVERSION_TMP(fmpq_polyxx, padic_polyxx,
padic_poly_get_fmpq_poly(to._poly(), from._poly(), from._ctx()))
FLINT_DEFINE_BINARY_EXPR_COND2(padic_polyxx_get_coeff_op, padicxx,
PADIC_POLYXX_COND_S, traits::fits_into_slong,
padic_poly_get_coeff_padic(to._padic(), e1._poly(), e2, to._ctx()))
FLINT_DEFINE_CBINARY_EXPR_COND2(plus, padic_polyxx,
PADIC_POLYXX_COND_S, PADIC_POLYXX_COND_S,
padic_poly_add(to._poly(), e1._poly(), e2._poly(), to._ctx()))
FLINT_DEFINE_BINARY_EXPR_COND2(minus, padic_polyxx,
PADIC_POLYXX_COND_S, PADIC_POLYXX_COND_S,
padic_poly_sub(to._poly(), e1._poly(), e2._poly(), to._ctx()))
FLINT_DEFINE_UNARY_EXPR_COND(negate, padic_polyxx, PADIC_POLYXX_COND_S,
padic_poly_neg(to._poly(), from._poly(), to._ctx()))
FLINT_DEFINE_CBINARY_EXPR_COND2(times, padic_polyxx,
PADIC_POLYXX_COND_S, PADICXX_COND_S,
padic_poly_scalar_mul_padic(to._poly(), e1._poly(), e2._padic(), to._ctx()))
FLINT_DEFINE_CBINARY_EXPR_COND2(times, padic_polyxx,
PADIC_POLYXX_COND_S, PADIC_POLYXX_COND_S,
padic_poly_mul(to._poly(), e1._poly(), e2._poly(), to._ctx()))
FLINT_DEFINE_BINARY_EXPR_COND2(pow_op, padic_polyxx, PADIC_POLYXX_COND_S,
traits::is_unsigned_integer,
padic_poly_pow(to._poly(), e1._poly(), e2, to._ctx()))
FLINT_DEFINE_BINARY_EXPR_COND2(inv_series_op, padic_polyxx,
PADIC_POLYXX_COND_S, traits::fits_into_slong,
padic_poly_inv_series(to._poly(), e1._poly(), e2, to._ctx()))
FLINT_DEFINE_UNARY_EXPR_COND(derivative_op, padic_polyxx, PADIC_POLYXX_COND_S,
padic_poly_derivative(to._poly(), from._poly(), to._ctx()))
FLINT_DEFINE_BINARY_EXPR_COND2(shift_left_op, padic_polyxx,
PADIC_POLYXX_COND_S, traits::fits_into_slong,
padic_poly_shift_left(to._poly(), e1._poly(), e2, to._ctx()))
FLINT_DEFINE_BINARY_EXPR_COND2(shift_right_op, padic_polyxx,
PADIC_POLYXX_COND_S, traits::fits_into_slong,
padic_poly_shift_right(to._poly(), e1._poly(), e2, to._ctx()))
FLINT_DEFINE_BINARY_EXPR_COND2(evaluate_op, padicxx,
PADIC_POLYXX_COND_S, PADICXX_COND_S,
padic_poly_evaluate_padic(to._padic(), e1._poly(), e2._padic(), to._ctx()))
FLINT_DEFINE_BINARY_EXPR_COND2(compose_op, padic_polyxx,
PADIC_POLYXX_COND_S, PADIC_POLYXX_COND_S,
padic_poly_compose(to._poly(), e1._poly(), e2._poly(), to._ctx()))
FLINT_DEFINE_BINARY_EXPR_COND2(compose_pow_op, padic_polyxx,
PADIC_POLYXX_COND_S, traits::fits_into_slong,
padic_poly_compose_pow(to._poly(), e1._poly(), e2, to._ctx()))
} // rules
} // flint
#endif