214 lines
8.5 KiB
C
214 lines
8.5 KiB
C
/* mpz expression evaluation, simple part
|
|
|
|
Copyright 2000, 2001, 2002 Free Software Foundation, Inc.
|
|
|
|
This file is part of the GNU MP Library.
|
|
|
|
The GNU MP Library is free software; you can redistribute it and/or modify
|
|
it under the terms of the GNU Lesser General Public License as published by
|
|
the Free Software Foundation; either version 2.1 of the License, or (at your
|
|
option) any later version.
|
|
|
|
The GNU MP Library 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 Lesser General Public
|
|
License for more details.
|
|
|
|
You should have received a copy of the GNU Lesser General Public License
|
|
along with the GNU MP Library; see the file COPYING.LIB. If not, write to
|
|
the Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
|
|
MA 02110-1301, USA. */
|
|
|
|
#include <ctype.h>
|
|
#include <stdio.h>
|
|
#include <string.h>
|
|
#include "gmp.h"
|
|
#include "expr-impl.h"
|
|
|
|
|
|
/* Change this to "#define TRACE(x) x" to get some traces. */
|
|
#define TRACE(x)
|
|
|
|
|
|
/* These are macros, so need function wrappers. */
|
|
static int
|
|
e_mpz_sgn (mpz_srcptr x)
|
|
{
|
|
return mpz_sgn (x);
|
|
}
|
|
static int
|
|
e_mpz_odd_p (mpz_srcptr x)
|
|
{
|
|
return mpz_odd_p (x);
|
|
}
|
|
static int
|
|
e_mpz_even_p (mpz_srcptr x)
|
|
{
|
|
return mpz_even_p (x);
|
|
}
|
|
|
|
/* These wrapped because MPEXPR_TYPE_I_ functions are expected to return
|
|
"int" whereas these return "unsigned long". */
|
|
static void
|
|
e_mpz_hamdist (mpz_ptr w, mpz_srcptr x, mpz_srcptr y)
|
|
{
|
|
mpz_set_ui (w, mpz_hamdist (x, y));
|
|
}
|
|
static void
|
|
e_mpz_popcount (mpz_ptr w, mpz_srcptr x)
|
|
{
|
|
mpz_set_ui (w, mpz_popcount (x));
|
|
}
|
|
static void
|
|
e_mpz_scan0 (mpz_ptr w, mpz_srcptr x, unsigned long start)
|
|
{
|
|
mpz_set_ui (w, mpz_scan0 (x, start));
|
|
}
|
|
static void
|
|
e_mpz_scan1 (mpz_ptr w, mpz_srcptr x, unsigned long start)
|
|
{
|
|
mpz_set_ui (w, mpz_scan1 (x, start));
|
|
}
|
|
|
|
/* These wrapped because they're in-place whereas MPEXPR_TYPE_BINARY_UI
|
|
expects a separate source and destination. Actually the parser will
|
|
normally pass w==x anyway. */
|
|
static void
|
|
e_mpz_setbit (mpz_ptr w, mpz_srcptr x, unsigned long n)
|
|
{
|
|
if (w != x)
|
|
mpz_set (w, x);
|
|
mpz_setbit (w, n);
|
|
}
|
|
static void
|
|
e_mpz_clrbit (mpz_ptr w, mpz_srcptr x, unsigned long n)
|
|
{
|
|
if (w != x)
|
|
mpz_set (w, x);
|
|
mpz_clrbit (w, n);
|
|
}
|
|
|
|
static __gmp_const struct mpexpr_operator_t _mpz_expr_standard_table[] = {
|
|
|
|
{ "**", (mpexpr_fun_t) mpz_pow_ui,
|
|
MPEXPR_TYPE_BINARY_UI | MPEXPR_TYPE_RIGHTASSOC, 220 },
|
|
|
|
{ "~", (mpexpr_fun_t) mpz_com,
|
|
MPEXPR_TYPE_UNARY | MPEXPR_TYPE_PREFIX, 210 },
|
|
{ "!", (mpexpr_fun_t) e_mpz_sgn,
|
|
MPEXPR_TYPE_LOGICAL_NOT | MPEXPR_TYPE_PREFIX, 210 },
|
|
{ "-", (mpexpr_fun_t) mpz_neg,
|
|
MPEXPR_TYPE_UNARY | MPEXPR_TYPE_PREFIX, 210 },
|
|
|
|
{ "*", (mpexpr_fun_t) mpz_mul, MPEXPR_TYPE_BINARY, 200 },
|
|
{ "/", (mpexpr_fun_t) mpz_tdiv_q, MPEXPR_TYPE_BINARY, 200 },
|
|
{ "%", (mpexpr_fun_t) mpz_tdiv_r, MPEXPR_TYPE_BINARY, 200 },
|
|
|
|
{ "+", (mpexpr_fun_t) mpz_add, MPEXPR_TYPE_BINARY, 190 },
|
|
{ "-", (mpexpr_fun_t) mpz_sub, MPEXPR_TYPE_BINARY, 190 },
|
|
|
|
{ "<<", (mpexpr_fun_t) mpz_mul_2exp, MPEXPR_TYPE_BINARY_UI, 180 },
|
|
{ ">>", (mpexpr_fun_t) mpz_tdiv_q_2exp, MPEXPR_TYPE_BINARY_UI, 180 },
|
|
|
|
{ "<=", (mpexpr_fun_t) mpz_cmp, MPEXPR_TYPE_CMP_LE, 170 },
|
|
{ "<", (mpexpr_fun_t) mpz_cmp, MPEXPR_TYPE_CMP_LT, 170 },
|
|
{ ">=", (mpexpr_fun_t) mpz_cmp, MPEXPR_TYPE_CMP_GE, 170 },
|
|
{ ">", (mpexpr_fun_t) mpz_cmp, MPEXPR_TYPE_CMP_GT, 170 },
|
|
|
|
{ "==", (mpexpr_fun_t) mpz_cmp, MPEXPR_TYPE_CMP_EQ, 160 },
|
|
{ "!=", (mpexpr_fun_t) mpz_cmp, MPEXPR_TYPE_CMP_NE, 160 },
|
|
|
|
{ "&", (mpexpr_fun_t) mpz_and, MPEXPR_TYPE_BINARY, 150 },
|
|
{ "^", (mpexpr_fun_t) mpz_xor, MPEXPR_TYPE_BINARY, 140 },
|
|
{ "|", (mpexpr_fun_t) mpz_ior, MPEXPR_TYPE_BINARY, 130 },
|
|
{ "&&", (mpexpr_fun_t) e_mpz_sgn, MPEXPR_TYPE_LOGICAL_AND, 120 },
|
|
{ "||", (mpexpr_fun_t) e_mpz_sgn, MPEXPR_TYPE_LOGICAL_OR, 110 },
|
|
|
|
{ ":", NULL, MPEXPR_TYPE_COLON, 101 },
|
|
{ "?", (mpexpr_fun_t) e_mpz_sgn, MPEXPR_TYPE_QUESTION, 100 },
|
|
|
|
{ ")", NULL, MPEXPR_TYPE_CLOSEPAREN, 4 },
|
|
{ "(", NULL, MPEXPR_TYPE_OPENPAREN, 3 },
|
|
{ ",", NULL, MPEXPR_TYPE_ARGSEP, 2 },
|
|
{ "$", NULL, MPEXPR_TYPE_VARIABLE, 1 },
|
|
|
|
{ "abs", (mpexpr_fun_t) mpz_abs, MPEXPR_TYPE_UNARY },
|
|
{ "bin", (mpexpr_fun_t) mpz_bin_ui, MPEXPR_TYPE_BINARY_UI },
|
|
{ "clrbit", (mpexpr_fun_t) e_mpz_clrbit, MPEXPR_TYPE_BINARY_UI },
|
|
{ "cmp", (mpexpr_fun_t) mpz_cmp, MPEXPR_TYPE_I_BINARY },
|
|
{ "cmpabs", (mpexpr_fun_t) mpz_cmpabs, MPEXPR_TYPE_I_BINARY },
|
|
{ "congruent_p",(mpexpr_fun_t)mpz_congruent_p, MPEXPR_TYPE_I_TERNARY },
|
|
{ "divisible_p",(mpexpr_fun_t)mpz_divisible_p, MPEXPR_TYPE_I_BINARY },
|
|
{ "even_p", (mpexpr_fun_t) e_mpz_even_p, MPEXPR_TYPE_I_UNARY },
|
|
{ "fib", (mpexpr_fun_t) mpz_fib_ui, MPEXPR_TYPE_UNARY_UI },
|
|
{ "fac", (mpexpr_fun_t) mpz_fac_ui, MPEXPR_TYPE_UNARY_UI },
|
|
{ "gcd", (mpexpr_fun_t) mpz_gcd, MPEXPR_TYPE_BINARY
|
|
| MPEXPR_TYPE_PAIRWISE },
|
|
{ "hamdist", (mpexpr_fun_t) e_mpz_hamdist, MPEXPR_TYPE_BINARY },
|
|
{ "invert", (mpexpr_fun_t) mpz_invert, MPEXPR_TYPE_BINARY },
|
|
{ "jacobi", (mpexpr_fun_t) mpz_jacobi, MPEXPR_TYPE_I_BINARY },
|
|
{ "kronecker", (mpexpr_fun_t) mpz_kronecker, MPEXPR_TYPE_I_BINARY },
|
|
{ "lcm", (mpexpr_fun_t) mpz_lcm, MPEXPR_TYPE_BINARY
|
|
| MPEXPR_TYPE_PAIRWISE },
|
|
{ "lucnum", (mpexpr_fun_t) mpz_lucnum_ui, MPEXPR_TYPE_UNARY_UI },
|
|
{ "max", (mpexpr_fun_t) mpz_cmp, MPEXPR_TYPE_MAX
|
|
| MPEXPR_TYPE_PAIRWISE },
|
|
{ "min", (mpexpr_fun_t) mpz_cmp, MPEXPR_TYPE_MIN
|
|
| MPEXPR_TYPE_PAIRWISE },
|
|
{ "nextprime", (mpexpr_fun_t) mpz_nextprime, MPEXPR_TYPE_UNARY },
|
|
{ "odd_p", (mpexpr_fun_t) e_mpz_odd_p, MPEXPR_TYPE_I_UNARY },
|
|
{ "perfect_power_p", (mpexpr_fun_t)mpz_perfect_power_p, MPEXPR_TYPE_I_UNARY},
|
|
{ "perfect_square_p",(mpexpr_fun_t)mpz_perfect_square_p,MPEXPR_TYPE_I_UNARY},
|
|
{ "popcount", (mpexpr_fun_t) e_mpz_popcount, MPEXPR_TYPE_UNARY },
|
|
{ "powm", (mpexpr_fun_t) mpz_powm, MPEXPR_TYPE_TERNARY },
|
|
{ "probab_prime_p", (mpexpr_fun_t)mpz_probab_prime_p, MPEXPR_TYPE_I_UNARY},
|
|
{ "root", (mpexpr_fun_t) mpz_root, MPEXPR_TYPE_BINARY_UI },
|
|
{ "scan0", (mpexpr_fun_t) e_mpz_scan0, MPEXPR_TYPE_BINARY_UI },
|
|
{ "scan1", (mpexpr_fun_t) e_mpz_scan1, MPEXPR_TYPE_BINARY_UI },
|
|
{ "setbit", (mpexpr_fun_t) e_mpz_setbit, MPEXPR_TYPE_BINARY_UI },
|
|
{ "tstbit", (mpexpr_fun_t) mpz_tstbit, MPEXPR_TYPE_I_BINARY_UI },
|
|
{ "sgn", (mpexpr_fun_t) e_mpz_sgn, MPEXPR_TYPE_I_UNARY },
|
|
{ "sqrt", (mpexpr_fun_t) mpz_sqrt, MPEXPR_TYPE_UNARY },
|
|
{ NULL }
|
|
};
|
|
|
|
/* The table is available globally only through a pointer, so the table size
|
|
can change without breaking binary compatibility. */
|
|
__gmp_const struct mpexpr_operator_t * __gmp_const mpz_expr_standard_table
|
|
= _mpz_expr_standard_table;
|
|
|
|
|
|
int
|
|
#if HAVE_STDARG
|
|
mpz_expr (mpz_ptr res, int base, __gmp_const char *e, ...)
|
|
#else
|
|
mpz_expr (va_alist)
|
|
va_dcl
|
|
#endif
|
|
{
|
|
mpz_srcptr var[MPEXPR_VARIABLES];
|
|
va_list ap;
|
|
int ret;
|
|
#if HAVE_STDARG
|
|
va_start (ap, e);
|
|
#else
|
|
mpz_ptr res;
|
|
int base;
|
|
__gmp_const char *e;
|
|
va_start (ap);
|
|
res = va_arg (ap, mpz_ptr);
|
|
base = va_arg (ap, int);
|
|
e = va_arg (ap, __gmp_const char *);
|
|
#endif
|
|
|
|
TRACE (printf ("mpz_expr(): base %d, %s\n", base, e));
|
|
ret = mpexpr_va_to_var ((void **) var, ap);
|
|
va_end (ap);
|
|
|
|
if (ret != MPEXPR_RESULT_OK)
|
|
return ret;
|
|
|
|
return mpz_expr_a (mpz_expr_standard_table, res, base, e, strlen(e), var);
|
|
}
|
|
|