contrib/libgmp/mpz/setbit.c

   1 /* mpz_setbit -- set a specified bit.
   2
   3 Copyright (C) 1991, 1993, 1994, 1995 Free Software Foundation, Inc.
   4
   5 This file is part of the GNU MP Library.
   6
   7 The GNU MP Library is free software; you can redistribute it and/or modify
   8 it under the terms of the GNU Library General Public License as published by
   9 the Free Software Foundation; either version 2 of the License, or (at your
  10 option) any later version.
  11
  12 The GNU MP Library is distributed in the hope that it will be useful, but
  13 WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
  14 or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU Library General Public
  15 License for more details.
  16
  17 You should have received a copy of the GNU Library General Public License
  18 along with the GNU MP Library; see the file COPYING.LIB.  If not, write to
  19 the Free Software Foundation, Inc., 59 Temple Place - Suite 330, Boston,
  20 MA 02111-1307, USA. */
  21
  22 #include "gmp.h"
  23 #include "gmp-impl.h"
  24
  25 void
  26 #if __STDC__
  27 mpz_setbit (mpz_ptr d, unsigned long int bit_index)
  28 #else
  29 mpz_setbit (d, bit_index)
  30      mpz_ptr d;
  31      unsigned long int bit_index;
  32 #endif
  33 {
  34   mp_size_t dsize = d->_mp_size;
  35   mp_ptr dp = d->_mp_d;
  36   mp_size_t limb_index;
  37
  38   limb_index = bit_index / BITS_PER_MP_LIMB;
  39   if (dsize >= 0)
  40     {
  41       if (limb_index < dsize)
  42         {
  43           dp[limb_index] |= (mp_limb_t) 1 << (bit_index % BITS_PER_MP_LIMB);
  44           d->_mp_size = dsize;
  45         }
  46       else
  47         {
  48           /* Ugh.  The bit should be set outside of the end of the
  49              number.  We have to increase the size of the number.  */
  50           if (d->_mp_alloc < limb_index + 1)
  51             {
  52               _mpz_realloc (d, limb_index + 1);
  53               dp = d->_mp_d;
  54             }
  55           MPN_ZERO (dp + dsize, limb_index - dsize);
  56           dp[limb_index] = (mp_limb_t) 1 << (bit_index % BITS_PER_MP_LIMB);
  57           d->_mp_size = limb_index + 1;
  58         }
  59     }
  60   else
  61     {
  62       mp_size_t zero_bound;
  63
  64       /* Simulate two's complement arithmetic, i.e. simulate
  65          1. Set OP = ~(OP - 1) [with infinitely many leading ones].
  66          2. set the bit.
  67          3. Set OP = ~OP + 1.  */
  68
  69       dsize = -dsize;
  70
  71       /* No upper bound on this loop, we're sure there's a non-zero limb
  72          sooner ot later.  */
  73       for (zero_bound = 0; ; zero_bound++)
  74         if (dp[zero_bound] != 0)
  75           break;
  76
  77       if (limb_index > zero_bound)
  78         {
  79           if (limb_index < dsize)
  80             dp[limb_index] &= ~((mp_limb_t) 1 << (bit_index % BITS_PER_MP_LIMB));
  81           else
  82             ;
  83         }
  84       else if (limb_index == zero_bound)
  85         {
  86           dp[limb_index] = ((dp[limb_index] - 1)
  87                             & ~((mp_limb_t) 1 << (bit_index % BITS_PER_MP_LIMB))) + 1;
  88           if (dp[limb_index] == 0)
  89             {
  90               mp_size_t i;
  91               for (i = limb_index + 1; i < dsize; i++)
  92                 {
  93                   dp[i] += 1;
  94                   if (dp[i] != 0)
  95                     goto fin;
  96                 }
  97               /* We got carry all way out beyond the end of D.  Increase
  98                  its size (and allocation if necessary).  */
  99               dsize++;
 100               if (d->_mp_alloc < dsize)
 101                 {
 102                   _mpz_realloc (d, dsize);
 103                   dp = d->_mp_d;
 104                 }
 105               dp[i] = 1;
 106               d->_mp_size = -dsize;
 107             fin:;
 108             }
 109         }
 110       else
 111         ;
 112     }
 113 }