hjl / source-git / glibc

Forked from source-git/glibc 3 years ago
Clone

Blame stdlib/divrem.c

Packit 6c4009
/* mpn_divrem -- Divide natural numbers, producing both remainder and
Packit 6c4009
   quotient.
Packit 6c4009
Packit 6c4009
Copyright (C) 1993-2018 Free Software Foundation, Inc.
Packit 6c4009
Packit 6c4009
This file is part of the GNU MP Library.
Packit 6c4009
Packit 6c4009
The GNU MP Library is free software; you can redistribute it and/or modify
Packit 6c4009
it under the terms of the GNU Lesser General Public License as published by
Packit 6c4009
the Free Software Foundation; either version 2.1 of the License, or (at your
Packit 6c4009
option) any later version.
Packit 6c4009
Packit 6c4009
The GNU MP Library is distributed in the hope that it will be useful, but
Packit 6c4009
WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
Packit 6c4009
or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU Lesser General Public
Packit 6c4009
License for more details.
Packit 6c4009
Packit 6c4009
You should have received a copy of the GNU Lesser General Public License
Packit 6c4009
along with the GNU MP Library; see the file COPYING.LIB.  If not, see
Packit 6c4009
<http://www.gnu.org/licenses/>.  */
Packit 6c4009
Packit 6c4009
#include <gmp.h>
Packit 6c4009
#include "gmp-impl.h"
Packit 6c4009
#include "longlong.h"
Packit 6c4009
Packit 6c4009
/* Divide num (NP/NSIZE) by den (DP/DSIZE) and write
Packit 6c4009
   the NSIZE-DSIZE least significant quotient limbs at QP
Packit 6c4009
   and the DSIZE long remainder at NP.  If QEXTRA_LIMBS is
Packit 6c4009
   non-zero, generate that many fraction bits and append them after the
Packit 6c4009
   other quotient limbs.
Packit 6c4009
   Return the most significant limb of the quotient, this is always 0 or 1.
Packit 6c4009
Packit 6c4009
   Preconditions:
Packit 6c4009
   0. NSIZE >= DSIZE.
Packit 6c4009
   1. The most significant bit of the divisor must be set.
Packit 6c4009
   2. QP must either not overlap with the input operands at all, or
Packit 6c4009
      QP + DSIZE >= NP must hold true.  (This means that it's
Packit 6c4009
      possible to put the quotient in the high part of NUM, right after the
Packit 6c4009
      remainder in NUM.
Packit 6c4009
   3. NSIZE >= DSIZE, even if QEXTRA_LIMBS is non-zero.  */
Packit 6c4009
Packit 6c4009
mp_limb_t
Packit 6c4009
mpn_divrem (mp_ptr qp, mp_size_t qextra_limbs,
Packit 6c4009
	    mp_ptr np, mp_size_t nsize,
Packit 6c4009
	    mp_srcptr dp, mp_size_t dsize)
Packit 6c4009
{
Packit 6c4009
  mp_limb_t most_significant_q_limb = 0;
Packit 6c4009
Packit 6c4009
  switch (dsize)
Packit 6c4009
    {
Packit 6c4009
    case 0:
Packit 6c4009
      /* We are asked to divide by zero, so go ahead and do it!  (To make
Packit 6c4009
	 the compiler not remove this statement, return the value.)  */
Packit 6c4009
      return 1 / dsize;
Packit 6c4009
Packit 6c4009
    case 1:
Packit 6c4009
      {
Packit 6c4009
	mp_size_t i;
Packit 6c4009
	mp_limb_t n1;
Packit 6c4009
	mp_limb_t d;
Packit 6c4009
Packit 6c4009
	d = dp[0];
Packit 6c4009
	n1 = np[nsize - 1];
Packit 6c4009
Packit 6c4009
	if (n1 >= d)
Packit 6c4009
	  {
Packit 6c4009
	    n1 -= d;
Packit 6c4009
	    most_significant_q_limb = 1;
Packit 6c4009
	  }
Packit 6c4009
Packit 6c4009
	qp += qextra_limbs;
Packit 6c4009
	for (i = nsize - 2; i >= 0; i--)
Packit 6c4009
	  udiv_qrnnd (qp[i], n1, n1, np[i], d);
Packit 6c4009
	qp -= qextra_limbs;
Packit 6c4009
Packit 6c4009
	for (i = qextra_limbs - 1; i >= 0; i--)
Packit 6c4009
	  udiv_qrnnd (qp[i], n1, n1, 0, d);
Packit 6c4009
Packit 6c4009
	np[0] = n1;
Packit 6c4009
      }
Packit 6c4009
      break;
Packit 6c4009
Packit 6c4009
    case 2:
Packit 6c4009
      {
Packit 6c4009
	mp_size_t i;
Packit 6c4009
	mp_limb_t n1, n0, n2;
Packit 6c4009
	mp_limb_t d1, d0;
Packit 6c4009
Packit 6c4009
	np += nsize - 2;
Packit 6c4009
	d1 = dp[1];
Packit 6c4009
	d0 = dp[0];
Packit 6c4009
	n1 = np[1];
Packit 6c4009
	n0 = np[0];
Packit 6c4009
Packit 6c4009
	if (n1 >= d1 && (n1 > d1 || n0 >= d0))
Packit 6c4009
	  {
Packit 6c4009
	    sub_ddmmss (n1, n0, n1, n0, d1, d0);
Packit 6c4009
	    most_significant_q_limb = 1;
Packit 6c4009
	  }
Packit 6c4009
Packit 6c4009
	for (i = qextra_limbs + nsize - 2 - 1; i >= 0; i--)
Packit 6c4009
	  {
Packit 6c4009
	    mp_limb_t q;
Packit 6c4009
	    mp_limb_t r;
Packit 6c4009
Packit 6c4009
	    if (i >= qextra_limbs)
Packit 6c4009
	      np--;
Packit 6c4009
	    else
Packit 6c4009
	      np[0] = 0;
Packit 6c4009
Packit 6c4009
	    if (n1 == d1)
Packit 6c4009
	      {
Packit 6c4009
		/* Q should be either 111..111 or 111..110.  Need special
Packit 6c4009
		   treatment of this rare case as normal division would
Packit 6c4009
		   give overflow.  */
Packit 6c4009
		q = ~(mp_limb_t) 0;
Packit 6c4009
Packit 6c4009
		r = n0 + d1;
Packit 6c4009
		if (r < d1)	/* Carry in the addition? */
Packit 6c4009
		  {
Packit 6c4009
		    add_ssaaaa (n1, n0, r - d0, np[0], 0, d0);
Packit 6c4009
		    qp[i] = q;
Packit 6c4009
		    continue;
Packit 6c4009
		  }
Packit 6c4009
		n1 = d0 - (d0 != 0);
Packit 6c4009
		n0 = -d0;
Packit 6c4009
	      }
Packit 6c4009
	    else
Packit 6c4009
	      {
Packit 6c4009
		udiv_qrnnd (q, r, n1, n0, d1);
Packit 6c4009
		umul_ppmm (n1, n0, d0, q);
Packit 6c4009
	      }
Packit 6c4009
Packit 6c4009
	    n2 = np[0];
Packit 6c4009
	  q_test:
Packit 6c4009
	    if (n1 > r || (n1 == r && n0 > n2))
Packit 6c4009
	      {
Packit 6c4009
		/* The estimated Q was too large.  */
Packit 6c4009
		q--;
Packit 6c4009
Packit 6c4009
		sub_ddmmss (n1, n0, n1, n0, 0, d0);
Packit 6c4009
		r += d1;
Packit 6c4009
		if (r >= d1)	/* If not carry, test Q again.  */
Packit 6c4009
		  goto q_test;
Packit 6c4009
	      }
Packit 6c4009
Packit 6c4009
	    qp[i] = q;
Packit 6c4009
	    sub_ddmmss (n1, n0, r, n2, n1, n0);
Packit 6c4009
	  }
Packit 6c4009
	np[1] = n1;
Packit 6c4009
	np[0] = n0;
Packit 6c4009
      }
Packit 6c4009
      break;
Packit 6c4009
Packit 6c4009
    default:
Packit 6c4009
      {
Packit 6c4009
	mp_size_t i;
Packit 6c4009
	mp_limb_t dX, d1, n0;
Packit 6c4009
Packit 6c4009
	np += nsize - dsize;
Packit 6c4009
	dX = dp[dsize - 1];
Packit 6c4009
	d1 = dp[dsize - 2];
Packit 6c4009
	n0 = np[dsize - 1];
Packit 6c4009
Packit 6c4009
	if (n0 >= dX)
Packit 6c4009
	  {
Packit 6c4009
	    if (n0 > dX || mpn_cmp (np, dp, dsize - 1) >= 0)
Packit 6c4009
	      {
Packit 6c4009
		mpn_sub_n (np, np, dp, dsize);
Packit 6c4009
		n0 = np[dsize - 1];
Packit 6c4009
		most_significant_q_limb = 1;
Packit 6c4009
	      }
Packit 6c4009
	  }
Packit 6c4009
Packit 6c4009
	for (i = qextra_limbs + nsize - dsize - 1; i >= 0; i--)
Packit 6c4009
	  {
Packit 6c4009
	    mp_limb_t q;
Packit 6c4009
	    mp_limb_t n1, n2;
Packit 6c4009
	    mp_limb_t cy_limb;
Packit 6c4009
Packit 6c4009
	    if (i >= qextra_limbs)
Packit 6c4009
	      {
Packit 6c4009
		np--;
Packit 6c4009
		n2 = np[dsize];
Packit 6c4009
	      }
Packit 6c4009
	    else
Packit 6c4009
	      {
Packit 6c4009
		n2 = np[dsize - 1];
Packit 6c4009
		MPN_COPY_DECR (np + 1, np, dsize);
Packit 6c4009
		np[0] = 0;
Packit 6c4009
	      }
Packit 6c4009
Packit 6c4009
	    if (n0 == dX)
Packit 6c4009
	      /* This might over-estimate q, but it's probably not worth
Packit 6c4009
		 the extra code here to find out.  */
Packit 6c4009
	      q = ~(mp_limb_t) 0;
Packit 6c4009
	    else
Packit 6c4009
	      {
Packit 6c4009
		mp_limb_t r;
Packit 6c4009
Packit 6c4009
		udiv_qrnnd (q, r, n0, np[dsize - 1], dX);
Packit 6c4009
		umul_ppmm (n1, n0, d1, q);
Packit 6c4009
Packit 6c4009
		while (n1 > r || (n1 == r && n0 > np[dsize - 2]))
Packit 6c4009
		  {
Packit 6c4009
		    q--;
Packit 6c4009
		    r += dX;
Packit 6c4009
		    if (r < dX)	/* I.e. "carry in previous addition?"  */
Packit 6c4009
		      break;
Packit 6c4009
		    n1 -= n0 < d1;
Packit 6c4009
		    n0 -= d1;
Packit 6c4009
		  }
Packit 6c4009
	      }
Packit 6c4009
Packit 6c4009
	    /* Possible optimization: We already have (q * n0) and (1 * n1)
Packit 6c4009
	       after the calculation of q.  Taking advantage of that, we
Packit 6c4009
	       could make this loop make two iterations less.  */
Packit 6c4009
Packit 6c4009
	    cy_limb = mpn_submul_1 (np, dp, dsize, q);
Packit 6c4009
Packit 6c4009
	    if (n2 != cy_limb)
Packit 6c4009
	      {
Packit 6c4009
		mpn_add_n (np, np, dp, dsize);
Packit 6c4009
		q--;
Packit 6c4009
	      }
Packit 6c4009
Packit 6c4009
	    qp[i] = q;
Packit 6c4009
	    n0 = np[dsize - 1];
Packit 6c4009
	  }
Packit 6c4009
      }
Packit 6c4009
    }
Packit 6c4009
Packit 6c4009
  return most_significant_q_limb;
Packit 6c4009
}