sqrt.c
上传用户:qaz666999
上传日期:2022-08-06
资源大小:2570k
文件大小:3k
源码类别:

数学计算

开发平台:

Unix_Linux

  1. /* mpf_sqrt -- Compute the square root of a float.
  2. Copyright 1993, 1994, 1996, 2000, 2001, 2004, 2005 Free Software Foundation,
  3. Inc.
  4. This file is part of the GNU MP Library.
  5. The GNU MP Library is free software; you can redistribute it and/or modify
  6. it under the terms of the GNU Lesser General Public License as published by
  7. the Free Software Foundation; either version 3 of the License, or (at your
  8. option) any later version.
  9. The GNU MP Library is distributed in the hope that it will be useful, but
  10. WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
  11. or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU Lesser General Public
  12. License for more details.
  13. You should have received a copy of the GNU Lesser General Public License
  14. along with the GNU MP Library.  If not, see http://www.gnu.org/licenses/.  */
  15. #include <stdio.h> /* for NULL */
  16. #include "gmp.h"
  17. #include "gmp-impl.h"
  18. /* As usual, the aim is to produce PREC(r) limbs of result, with the high
  19.    limb non-zero.  This is accomplished by applying mpn_sqrtrem to either
  20.    2*prec or 2*prec-1 limbs, both such sizes resulting in prec limbs.
  21.    The choice between 2*prec or 2*prec-1 limbs is based on the input
  22.    exponent.  With b=2^GMP_NUMB_BITS the limb base then we can think of
  23.    effectively taking out a factor b^(2k), for suitable k, to get to an
  24.    integer input of the desired size ready for mpn_sqrtrem.  It must be an
  25.    even power taken out, ie. an even number of limbs, so the square root
  26.    gives factor b^k and the radix point is still on a limb boundary.  So if
  27.    EXP(r) is even we'll get an even number of input limbs 2*prec, or if
  28.    EXP(r) is odd we get an odd number 2*prec-1.
  29.    Further limbs below the 2*prec or 2*prec-1 used don't affect the result
  30.    and are simply truncated.  This can be seen by considering an integer x,
  31.    with s=floor(sqrt(x)).  s is the unique integer satisfying s^2 <= x <
  32.    (s+1)^2.  Notice that adding a fraction part to x (ie. some further bits)
  33.    doesn't change the inequality, s remains the unique solution.  Working
  34.    suitable factors of 2 into this argument lets it apply to an intended
  35.    precision at any position for any x, not just the integer binary point.
  36.    If the input is smaller than 2*prec or 2*prec-1, then we just pad with
  37.    zeros, that of course being our usual interpretation of short inputs.
  38.    The effect is to extend the root beyond the size of the input (for
  39.    instance into fractional limbs if u is an integer).  */
  40. void
  41. mpf_sqrt (mpf_ptr r, mpf_srcptr u)
  42. {
  43.   mp_size_t usize;
  44.   mp_ptr up, tp;
  45.   mp_size_t prec, tsize;
  46.   mp_exp_t uexp, expodd;
  47.   TMP_DECL;
  48.   usize = u->_mp_size;
  49.   if (usize <= 0)
  50.     {
  51.       if (usize < 0)
  52.         SQRT_OF_NEGATIVE;
  53.       r->_mp_size = 0;
  54.       r->_mp_exp = 0;
  55.       return;
  56.     }
  57.   TMP_MARK;
  58.   uexp = u->_mp_exp;
  59.   prec = r->_mp_prec;
  60.   up = u->_mp_d;
  61.   expodd = (uexp & 1);
  62.   tsize = 2 * prec - expodd;
  63.   r->_mp_size = prec;
  64.   r->_mp_exp = (uexp + expodd) / 2;    /* ceil(uexp/2) */
  65.   /* root size is ceil(tsize/2), this will be our desired "prec" limbs */
  66.   ASSERT ((tsize + 1) / 2 == prec);
  67.   tp = TMP_ALLOC_LIMBS (tsize);
  68.   if (usize > tsize)
  69.     {
  70.       up += usize - tsize;
  71.       usize = tsize;
  72.       MPN_COPY (tp, up, tsize);
  73.     }
  74.   else
  75.     {
  76.       MPN_ZERO (tp, tsize - usize);
  77.       MPN_COPY (tp + (tsize - usize), up, usize);
  78.     }
  79.   mpn_sqrtrem (r->_mp_d, NULL, tp, tsize);
  80.   TMP_FREE;
  81. }