From 11da511c784eca003deb90c23570f0873954e0de Mon Sep 17 00:00:00 2001 From: Duncan Wilkie Date: Sat, 18 Nov 2023 06:11:09 -0600 Subject: Initial commit. --- gmp-6.3.0/mpz/rootrem.c | 100 ++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 100 insertions(+) create mode 100644 gmp-6.3.0/mpz/rootrem.c (limited to 'gmp-6.3.0/mpz/rootrem.c') diff --git a/gmp-6.3.0/mpz/rootrem.c b/gmp-6.3.0/mpz/rootrem.c new file mode 100644 index 0000000..67953db --- /dev/null +++ b/gmp-6.3.0/mpz/rootrem.c @@ -0,0 +1,100 @@ +/* mpz_rootrem(root, rem, u, nth) -- Set ROOT to trunc(U^(1/nth)) and + set REM to the remainder. + +Copyright 1999-2003, 2005, 2012 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 either: + + * the GNU Lesser General Public License as published by the Free + Software Foundation; either version 3 of the License, or (at your + option) any later version. + +or + + * 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. + +or both in parallel, as here. + +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 General Public License +for more details. + +You should have received copies of the GNU General Public License and the +GNU Lesser General Public License along with the GNU MP Library. If not, +see https://www.gnu.org/licenses/. */ + +#include /* for NULL */ +#include "gmp-impl.h" + +void +mpz_rootrem (mpz_ptr root, mpz_ptr rem, mpz_srcptr u, unsigned long int nth) +{ + mp_ptr rootp, up, remp; + mp_size_t us, un, rootn, remn; + TMP_DECL; + + us = SIZ(u); + + /* even roots of negatives provoke an exception */ + if (UNLIKELY (us < 0 && (nth & 1) == 0)) + SQRT_OF_NEGATIVE; + + /* root extraction interpreted as c^(1/nth) means a zeroth root should + provoke a divide by zero, do this even if c==0 */ + if (UNLIKELY (nth == 0)) + DIVIDE_BY_ZERO; + + if (us == 0) + { + if (root != NULL) + SIZ(root) = 0; + SIZ(rem) = 0; + return; + } + + un = ABS (us); + rootn = (un - 1) / nth + 1; + + TMP_MARK; + + /* FIXME: Perhaps disallow root == NULL */ + if (root != NULL && u != root) + rootp = MPZ_NEWALLOC (root, rootn); + else + rootp = TMP_ALLOC_LIMBS (rootn); + + if (u != rem) + remp = MPZ_NEWALLOC (rem, un); + else + remp = TMP_ALLOC_LIMBS (un); + + up = PTR(u); + + if (nth == 1) + { + MPN_COPY (rootp, up, un); + remn = 0; + } + else + { + remn = mpn_rootrem (rootp, remp, up, un, (mp_limb_t) nth); + } + + if (root != NULL) + { + SIZ(root) = us >= 0 ? rootn : -rootn; + if (u == root) + MPN_COPY (up, rootp, rootn); + } + + if (u == rem) + MPN_COPY (up, remp, remn); + SIZ(rem) = us >= 0 ? remn : -remn; + TMP_FREE; +} -- cgit v1.2.3