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

数学计算

开发平台:

Unix_Linux

  1. /* mpn_remove -- divide out all multiples of odd mpn number from another mpn
  2.    number.
  3.    Contributed to the GNU project by Torbjorn Granlund.
  4.    THE FUNCTION IN THIS FILE IS INTERNAL WITH A MUTABLE INTERFACE.  IT IS ONLY
  5.    SAFE TO REACH IT THROUGH DOCUMENTED INTERFACES.  IN FACT, IT IS ALMOST
  6.    GUARANTEED THAT IT WILL CHANGE OR DISAPPEAR IN A FUTURE GMP RELEASE.
  7. Copyright 2009 Free Software Foundation, Inc.
  8. This file is part of the GNU MP Library.
  9. The GNU MP Library is free software; you can redistribute it and/or modify
  10. it under the terms of the GNU Lesser General Public License as published by
  11. the Free Software Foundation; either version 3 of the License, or (at your
  12. option) any later version.
  13. The GNU MP Library is distributed in the hope that it will be useful, but
  14. WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
  15. or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU Lesser General Public
  16. License for more details.
  17. You should have received a copy of the GNU Lesser General Public License
  18. along with the GNU MP Library.  If not, see http://www.gnu.org/licenses/.  */
  19. #include "gmp.h"
  20. #include "gmp-impl.h"
  21. #if GMP_LIMB_BITS > 50
  22. #define LOG 50
  23. #else
  24. #define LOG GMP_LIMB_BITS
  25. #endif
  26. /* Input: U = {up,un}, V = {vp,vn} must be odd, cap
  27.    Ouput  W = {wp,*wn} allocation need is exactly *wn
  28.    Set W = U / V^k, where k is the largest integer <= cap such that the
  29.    division yields an integer.
  30.    FIXME: We currently allow any operand overlap.  This is quite non mpn-ish
  31.    and might be changed, since it cost significant temporary space.
  32.    * If we require W to have space for un limbs, we could save qp or qp2 (but
  33.      we will still need to copy things into wp 50% of the time).
  34.    * If we allow ourselves to clobber U, we could save the other of qp and qp2.
  35. */
  36. mp_bitcnt_t
  37. mpn_remove (mp_ptr wp, mp_size_t *wn,
  38.     mp_ptr up, mp_size_t un, mp_ptr vp, mp_size_t vn,
  39.     mp_bitcnt_t cap)
  40. {
  41.   mp_ptr    pwpsp[LOG];
  42.   mp_size_t pwpsn[LOG];
  43.   mp_size_t npowers;
  44.   mp_ptr tp, qp, np, pp, qp2, scratch_out;
  45.   mp_size_t pn, nn, qn, i;
  46.   mp_bitcnt_t pwr;
  47.   TMP_DECL;
  48.   ASSERT (un > 0);
  49.   ASSERT (vn > 0);
  50.   ASSERT (vp[0] % 2 != 0); /* 2-adic division wants odd numbers */
  51.   ASSERT (vn > 1 || vp[0] > 1); /* else we would loop indefinitely */
  52.   TMP_MARK;
  53.   tp = TMP_ALLOC_LIMBS ((un + vn) / 2); /* remainder */
  54.   qp = TMP_ALLOC_LIMBS (un); /* quotient, alternating */
  55.   qp2 = TMP_ALLOC_LIMBS (un); /* quotient, alternating */
  56.   np = TMP_ALLOC_LIMBS (un + LOG); /* powers of V */
  57.   pp = vp;
  58.   pn = vn;
  59.   /* FIXME: This allocation need indicate a flaw in the current itch mechanism:
  60.      Which operands not greater than un,un will incur the worst itch?  We need
  61.      a parallel foo_maxitch set of functions.  */
  62.   scratch_out = TMP_ALLOC_LIMBS (mpn_bdiv_qr_itch (un, un >> 1));
  63.   MPN_COPY (qp, up, un);
  64.   qn = un;
  65.   npowers = 0;
  66.   while (qn >= pn)
  67.     {
  68.       mpn_bdiv_qr (qp2, tp, qp, qn, pp, pn, scratch_out);
  69.       if (!mpn_zero_p (tp, pn))
  70. break; /* could not divide by V^npowers */
  71.       MP_PTR_SWAP (qp, qp2);
  72.       qn = qn - pn;
  73.       qn += qp[qn] != 0;
  74.       pwpsp[npowers] = pp;
  75.       pwpsn[npowers] = pn;
  76.       npowers++;
  77.       if (((mp_bitcnt_t) 2 << npowers) - 1 > cap)
  78. break;
  79.       nn = 2 * pn - 1; /* next power will be at least this many limbs */
  80.       if (nn > qn)
  81. break; /* next power would be overlarge */
  82.       mpn_sqr (np, pp, pn);
  83.       nn += np[nn] != 0;
  84.       pp = np;
  85.       pn = nn;
  86.       np += nn;
  87.     }
  88.   pwr = ((mp_bitcnt_t) 1 << npowers) - 1;
  89.   for (i = npowers - 1; i >= 0; i--)
  90.     {
  91.       pp = pwpsp[i];
  92.       pn = pwpsn[i];
  93.       if (qn < pn)
  94. continue;
  95.       if (pwr + ((mp_bitcnt_t) 1 << i) > cap)
  96. continue; /* V^i would bring us past cap */
  97.       mpn_bdiv_qr (qp2, tp, qp, qn, pp, pn, scratch_out);
  98.       if (!mpn_zero_p (tp, pn))
  99. continue; /* could not divide by V^i */
  100.       MP_PTR_SWAP (qp, qp2);
  101.       qn = qn - pn;
  102.       qn += qp[qn] != 0;
  103.       pwr += (mp_bitcnt_t) 1 << i;
  104.     }
  105.   MPN_COPY (wp, qp, qn);
  106.   *wn = qn;
  107.   TMP_FREE;
  108.   return pwr;
  109. }