Blame mpz/cong_ui.c

Packit 5c3484
/* mpz_congruent_ui_p -- test congruence of mpz and ulong.
Packit 5c3484
Packit 5c3484
Copyright 2000-2002, 2012 Free Software Foundation, Inc.
Packit 5c3484
Packit 5c3484
This file is part of the GNU MP Library.
Packit 5c3484
Packit 5c3484
The GNU MP Library is free software; you can redistribute it and/or modify
Packit 5c3484
it under the terms of either:
Packit 5c3484
Packit 5c3484
  * the GNU Lesser General Public License as published by the Free
Packit 5c3484
    Software Foundation; either version 3 of the License, or (at your
Packit 5c3484
    option) any later version.
Packit 5c3484
Packit 5c3484
or
Packit 5c3484
Packit 5c3484
  * the GNU General Public License as published by the Free Software
Packit 5c3484
    Foundation; either version 2 of the License, or (at your option) any
Packit 5c3484
    later version.
Packit 5c3484
Packit 5c3484
or both in parallel, as here.
Packit 5c3484
Packit 5c3484
The GNU MP Library is distributed in the hope that it will be useful, but
Packit 5c3484
WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
Packit 5c3484
or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
Packit 5c3484
for more details.
Packit 5c3484
Packit 5c3484
You should have received copies of the GNU General Public License and the
Packit 5c3484
GNU Lesser General Public License along with the GNU MP Library.  If not,
Packit 5c3484
see https://www.gnu.org/licenses/.  */
Packit 5c3484
Packit 5c3484
#include "gmp.h"
Packit 5c3484
#include "gmp-impl.h"
Packit 5c3484
#include "longlong.h"
Packit 5c3484
Packit 5c3484
Packit 5c3484
/* There's some explicit checks for c
Packit 5c3484
   application might use that in a test.
Packit 5c3484
Packit 5c3484
   Hopefully the compiler can generate something good for r==(c%d), though
Packit 5c3484
   if modexact is being used exclusively then that's not reached.  */
Packit 5c3484
Packit 5c3484
int
Packit 5c3484
mpz_congruent_ui_p (mpz_srcptr a, unsigned long cu, unsigned long du)
Packit 5c3484
{
Packit 5c3484
  mp_srcptr  ap;
Packit 5c3484
  mp_size_t  asize;
Packit 5c3484
  mp_limb_t  c, d, r;
Packit 5c3484
Packit 5c3484
  if (UNLIKELY (du == 0))
Packit 5c3484
    return (mpz_cmp_ui (a, cu) == 0);
Packit 5c3484
Packit 5c3484
  asize = SIZ(a);
Packit 5c3484
  if (asize == 0)
Packit 5c3484
    {
Packit 5c3484
      if (cu < du)
Packit 5c3484
	return cu == 0;
Packit 5c3484
      else
Packit 5c3484
	return (cu % du) == 0;
Packit 5c3484
    }
Packit 5c3484
Packit 5c3484
  /* For nails don't try to be clever if c or d is bigger than a limb, just
Packit 5c3484
     fake up some mpz_t's and go to the main mpz_congruent_p.  */
Packit 5c3484
  if (du > GMP_NUMB_MAX || cu > GMP_NUMB_MAX)
Packit 5c3484
    {
Packit 5c3484
      mp_limb_t  climbs[2], dlimbs[2];
Packit 5c3484
      mpz_t      cz, dz;
Packit 5c3484
Packit 5c3484
      ALLOC(cz) = 2;
Packit 5c3484
      PTR(cz) = climbs;
Packit 5c3484
      ALLOC(dz) = 2;
Packit 5c3484
      PTR(dz) = dlimbs;
Packit 5c3484
Packit 5c3484
      mpz_set_ui (cz, cu);
Packit 5c3484
      mpz_set_ui (dz, du);
Packit 5c3484
      return mpz_congruent_p (a, cz, dz);
Packit 5c3484
    }
Packit 5c3484
Packit 5c3484
  /* NEG_MOD works on limbs, so convert ulong to limb */
Packit 5c3484
  c = cu;
Packit 5c3484
  d = du;
Packit 5c3484
Packit 5c3484
  if (asize < 0)
Packit 5c3484
    {
Packit 5c3484
      asize = -asize;
Packit 5c3484
      NEG_MOD (c, c, d);
Packit 5c3484
    }
Packit 5c3484
Packit 5c3484
  ap = PTR (a);
Packit 5c3484
Packit 5c3484
  if (ABOVE_THRESHOLD (asize, BMOD_1_TO_MOD_1_THRESHOLD))
Packit 5c3484
    {
Packit 5c3484
      r = mpn_mod_1 (ap, asize, d);
Packit 5c3484
      if (c < d)
Packit 5c3484
	return r == c;
Packit 5c3484
      else
Packit 5c3484
	return r == (c % d);
Packit 5c3484
    }
Packit 5c3484
Packit 5c3484
  if ((d & 1) == 0)
Packit 5c3484
    {
Packit 5c3484
      /* Strip low zero bits to get odd d required by modexact.  If
Packit 5c3484
	 d==e*2^n then a==c mod d if and only if both a==c mod 2^n
Packit 5c3484
	 and a==c mod e.  */
Packit 5c3484
Packit 5c3484
      unsigned	twos;
Packit 5c3484
Packit 5c3484
      if ((ap[0]-c) & LOW_ZEROS_MASK (d))
Packit 5c3484
	return 0;
Packit 5c3484
Packit 5c3484
      count_trailing_zeros (twos, d);
Packit 5c3484
      d >>= twos;
Packit 5c3484
    }
Packit 5c3484
Packit 5c3484
  r = mpn_modexact_1c_odd (ap, asize, d, c);
Packit 5c3484
  return r == 0 || r == d;
Packit 5c3484
}