ghc/runtime/gmp/mpn_rshiftci.c

   1 /* mpn_rshiftci -- Shift a low level natural-number integer with carry in.
   2
   3 Copyright (C) 1991 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 General Public License as published by
   9 the Free Software Foundation; either version 2, or (at your option)
  10 any later version.
  11
  12 The GNU MP Library is distributed in the hope that it will be useful,
  13 but WITHOUT ANY WARRANTY; without even the implied warranty of
  14 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15 GNU General Public License for more details.
  16
  17 You should have received a copy of the GNU General Public License
  18 along with the GNU MP Library; see the file COPYING.  If not, write to
  19 the Free Software Foundation, 675 Mass Ave, Cambridge, MA 02139, USA.  */
  20
  21 #include "gmp.h"
  22 #include "gmp-impl.h"
  23
  24 /* Shift U (pointed to by UP and USIZE digits long) CNT bits to the right
  25    and store the USIZE least significant digits of the result at WP.
  26    Return the size of the result.
  27
  28    Argument constraints:
  29    0. U must be normalized (i.e. it's most significant digit != 0).
  30    1. 0 <= CNT < BITS_PER_MP_LIMB
  31    2. If the result is to be written over the input, WP must be <= UP.
  32 */
  33
  34 mp_size
  35 #ifdef __STDC__
  36 mpn_rshiftci (mp_ptr wp,
  37                mp_srcptr up, mp_size usize,
  38                unsigned cnt,
  39                mp_limb carry_in)
  40 #else
  41 mpn_rshiftci (wp, up, usize, cnt, carry_in)
  42      mp_ptr wp;
  43      mp_srcptr up;
  44      mp_size usize;
  45      unsigned cnt;
  46      mp_limb carry_in;
  47 #endif
  48 {
  49   mp_limb high_limb, low_limb;
  50   unsigned sh_1, sh_2;
  51   mp_size i;
  52
  53   if (usize <= 0)
  54     return 0;
  55
  56   sh_1 = cnt;
  57   if (sh_1 == 0)
  58     {
  59       if (wp != up)
  60         {
  61           /* Copy from low end to high end, to allow specified input/output
  62              overlapping.  */
  63           for (i = 0; i < usize; i++)
  64             wp[i] = up[i];
  65         }
  66       return usize;
  67     }
  68
  69   wp -= 1;
  70   sh_2 = BITS_PER_MP_LIMB - sh_1;
  71   low_limb = up[0];
  72   for (i = 1; i < usize; i++)
  73     {
  74       high_limb = up[i];
  75       wp[i] = (low_limb >> sh_1) | (high_limb << sh_2);
  76       low_limb = high_limb;
  77     }
  78   low_limb = (low_limb >> sh_1) | (carry_in << sh_2);
  79   if (low_limb != 0)
  80     {
  81       wp[i] = low_limb;
  82       return usize;
  83     }
  84
  85   return usize - 1;
  86 }