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

数学计算

开发平台:

Unix_Linux

  1. /* mpz_congruent_2exp_p -- test congruence of mpz mod 2^n.
  2. Copyright 2001, 2002 Free Software Foundation, Inc.
  3. This file is part of the GNU MP Library.
  4. The GNU MP Library is free software; you can redistribute it and/or modify
  5. it under the terms of the GNU Lesser General Public License as published by
  6. the Free Software Foundation; either version 3 of the License, or (at your
  7. option) any later version.
  8. The GNU MP Library is distributed in the hope that it will be useful, but
  9. WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
  10. or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU Lesser General Public
  11. License for more details.
  12. You should have received a copy of the GNU Lesser General Public License
  13. along with the GNU MP Library.  If not, see http://www.gnu.org/licenses/.  */
  14. #include "gmp.h"
  15. #include "gmp-impl.h"
  16. int
  17. mpz_congruent_2exp_p (mpz_srcptr a, mpz_srcptr c, mp_bitcnt_t d)
  18. {
  19.   mp_size_t      i, dlimbs;
  20.   unsigned       dbits;
  21.   mp_ptr         ap, cp;
  22.   mp_limb_t      dmask, alimb, climb, sum;
  23.   mp_size_t      asize_signed, csize_signed, asize, csize;
  24.   if (ABSIZ(a) < ABSIZ(c))
  25.     MPZ_SRCPTR_SWAP (a, c);
  26.   dlimbs = d / GMP_NUMB_BITS;
  27.   dbits = d % GMP_NUMB_BITS;
  28.   dmask = (CNST_LIMB(1) << dbits) - 1;
  29.   ap = PTR(a);
  30.   cp = PTR(c);
  31.   asize_signed = SIZ(a);
  32.   asize = ABS(asize_signed);
  33.   csize_signed = SIZ(c);
  34.   csize = ABS(csize_signed);
  35.   if (csize_signed == 0)
  36.     goto a_zeros;
  37.   if ((asize_signed ^ csize_signed) >= 0)
  38.     {
  39.       /* same signs, direct comparison */
  40.       /* a==c for limbs in common */
  41.       if (mpn_cmp (ap, cp, MIN (csize, dlimbs)) != 0)
  42.         return 0;
  43.       /* if that's all of dlimbs, then a==c for remaining bits */
  44.       if (csize > dlimbs)
  45.         return ((ap[dlimbs]-cp[dlimbs]) & dmask) == 0;
  46.     a_zeros:
  47.       /* a remains, need all zero bits */
  48.       /* if d covers all of a and c, then must be exactly equal */
  49.       if (asize <= dlimbs)
  50.         return asize == csize;
  51.       /* whole limbs zero */
  52.       for (i = csize; i < dlimbs; i++)
  53.         if (ap[i] != 0)
  54.           return 0;
  55.       /* partial limb zero */
  56.       return (ap[dlimbs] & dmask) == 0;
  57.     }
  58.   else
  59.     {
  60.       /* different signs, negated comparison */
  61.       /* common low zero limbs, stopping at first non-zeros, which must
  62.          match twos complement */
  63.       i = 0;
  64.       for (;;)
  65.         {
  66.           ASSERT (i < csize);  /* always have a non-zero limb on c */
  67.           alimb = ap[i];
  68.           climb = cp[i];
  69.           sum = (alimb + climb) & GMP_NUMB_MASK;
  70.           if (i >= dlimbs)
  71.             return (sum & dmask) == 0;
  72.           i++;
  73.           /* require both zero, or first non-zeros as twos-complements */
  74.           if (sum != 0)
  75.             return 0;
  76.           if (alimb != 0)
  77.             break;
  78.         }
  79.       /* further limbs matching as ones-complement */
  80.       for (;;)
  81.         {
  82.           if (i >= csize)
  83.             break;
  84.           alimb = ap[i];
  85.           climb = cp[i];
  86.           sum = (alimb + climb + 1) & GMP_NUMB_MASK;
  87.           if (i >= dlimbs)
  88.             return (sum & dmask) == 0;
  89.           if (sum != 0)
  90.             return 0;
  91.           i++;
  92.         }
  93.       /* no more c, so require all 1 bits in a */
  94.       if (asize < dlimbs)
  95.         return 0;   /* not enough a */
  96.       /* whole limbs */
  97.       for ( ; i < dlimbs; i++)
  98.         if (ap[i] != GMP_NUMB_MAX)
  99.           return 0;
  100.       /* if only whole limbs, no further fetches from a */
  101.       if (dbits == 0)
  102.         return 1;
  103.       /* need enough a */
  104.       if (asize == dlimbs)
  105.         return 0;
  106.       return ((ap[dlimbs]+1) & dmask) == 0;
  107.     }
  108. }