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

数学计算

开发平台:

Unix_Linux

  1. /* mpz_urandomm (rop, state, n) -- Generate a uniform pseudorandom
  2.    integer in the range 0 to N-1, using STATE as the random state
  3.    previously initialized by a call to gmp_randinit().
  4. Copyright 2000, 2002  Free Software Foundation, Inc.
  5. This file is part of the GNU MP Library.
  6. The GNU MP Library is free software; you can redistribute it and/or modify
  7. it under the terms of the GNU Lesser General Public License as published by
  8. the Free Software Foundation; either version 3 of the License, or (at your
  9. option) any later version.
  10. The GNU MP Library is distributed in the hope that it will be useful, but
  11. WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
  12. or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU Lesser General Public
  13. License for more details.
  14. You should have received a copy of the GNU Lesser General Public License
  15. along with the GNU MP Library.  If not, see http://www.gnu.org/licenses/.  */
  16. #include "gmp.h"
  17. #include "gmp-impl.h"
  18. #include "longlong.h" /* for count_leading_zeros */
  19. #define MAX_URANDOMM_ITER  80
  20. void
  21. mpz_urandomm (mpz_ptr rop, gmp_randstate_t rstate, mpz_srcptr n)
  22. {
  23.   mp_ptr rp, np, nlast;
  24.   mp_size_t nbits, size;
  25.   int count;
  26.   int pow2;
  27.   int cmp;
  28.   TMP_DECL;
  29.   size = ABSIZ (n);
  30.   if (size == 0)
  31.     DIVIDE_BY_ZERO;
  32.   nlast = &PTR (n)[size - 1];
  33.   /* Detect whether n is a power of 2.  */
  34.   pow2 = POW2_P (*nlast);
  35.   if (pow2 != 0)
  36.     for (np = PTR (n); np < nlast; np++)
  37.       if (*np != 0)
  38. {
  39.   pow2 = 0; /* Mark n as `not a power of two'.  */
  40.   break;
  41. }
  42.   count_leading_zeros (count, *nlast);
  43.   nbits = size * GMP_NUMB_BITS - (count - GMP_NAIL_BITS) - pow2;
  44.   if (nbits == 0) /* nbits == 0 means that n was == 1.  */
  45.     {
  46.       SIZ (rop) = 0;
  47.       return;
  48.     }
  49.   TMP_MARK;
  50.   np = PTR (n);
  51.   if (rop == n)
  52.     {
  53.       mp_ptr tp;
  54.       tp = TMP_ALLOC_LIMBS (size);
  55.       MPN_COPY (tp, np, size);
  56.       np = tp;
  57.     }
  58.   /* Here the allocated size can be one too much if n is a power of
  59.      (2^GMP_NUMB_BITS) but it's convenient for using mpn_cmp below.  */
  60.   rp = MPZ_REALLOC (rop, size);
  61.   /* Clear last limb to prevent the case in which size is one too much.  */
  62.   rp[size - 1] = 0;
  63.   count = MAX_URANDOMM_ITER; /* Set iteration count limit.  */
  64.   do
  65.     {
  66.       _gmp_rand (rp, rstate, nbits);
  67.       MPN_CMP (cmp, rp, np, size);
  68.     }
  69.   while (cmp >= 0 && --count != 0);
  70.   if (count == 0)
  71.     /* Too many iterations; return result mod n == result - n */
  72.     mpn_sub_n (rp, rp, np, size);
  73.   MPN_NORMALIZE (rp, size);
  74.   SIZ (rop) = size;
  75.   TMP_FREE;
  76. }