ghc/runtime/gmp/mpn_lshift.c

   1 /* mpn_lshift -- Shift left low level.
   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 left
  25    and store the USIZE least significant digits of the result at WP.
  26    Return the bits shifted out from the most significant digit.
  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_limb
  35 #ifdef __STDC__
  36 mpn_lshift (mp_ptr wp,
  37             mp_srcptr up, mp_size usize,
  38             unsigned cnt)
  39 #else
  40 mpn_lshift (wp, up, usize, cnt)
  41      mp_ptr wp;
  42      mp_srcptr up;
  43      mp_size usize;
  44      unsigned cnt;
  45 #endif
  46 {
  47   mp_limb high_limb, low_limb;
  48   unsigned sh_1, sh_2;
  49   mp_size i;
  50   mp_limb retval;
  51
  52   if (usize == 0)
  53     return 0;
  54
  55   sh_1 = cnt;
  56   if (sh_1 == 0)
  57     {
  58       if (wp != up)
  59         {
  60           /* Copy from high end to low end, to allow specified input/output
  61              overlapping.  */
  62           for (i = usize - 1; i >= 0; i--)
  63             wp[i] = up[i];
  64         }
  65       return 0;
  66     }
  67
  68   wp += 1;
  69   sh_2 = BITS_PER_MP_LIMB - sh_1;
  70   i = usize - 1;
  71   low_limb = up[i];
  72   retval = low_limb >> sh_2;
  73   high_limb = low_limb;
  74   while (--i >= 0)
  75     {
  76       low_limb = up[i];
  77       wp[i] = (high_limb << sh_1) | (low_limb >> sh_2);
  78       high_limb = low_limb;
  79     }
  80   wp[i] = high_limb << sh_1;
  81
  82   return retval;
  83 }