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

数学计算

开发平台:

Unix_Linux

  1. /* mpz_import -- set mpz from word data.
  2. Copyright 2002 Free Software Foundation, Inc.
  3. This file is part of the GNU MP Library.
  4. The GNU MP Library is free software; you can redistribute it and/or modify
  5. it under the terms of the GNU Lesser General Public License as published by
  6. the Free Software Foundation; either version 3 of the License, or (at your
  7. option) any later version.
  8. The GNU MP Library is distributed in the hope that it will be useful, but
  9. WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
  10. or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU Lesser General Public
  11. License for more details.
  12. You should have received a copy of the GNU Lesser General Public License
  13. along with the GNU MP Library.  If not, see http://www.gnu.org/licenses/.  */
  14. #include <stdio.h>
  15. #include "gmp.h"
  16. #include "gmp-impl.h"
  17. #if HAVE_LIMB_BIG_ENDIAN
  18. #define HOST_ENDIAN     1
  19. #endif
  20. #if HAVE_LIMB_LITTLE_ENDIAN
  21. #define HOST_ENDIAN     (-1)
  22. #endif
  23. #ifndef HOST_ENDIAN
  24. static const mp_limb_t  endian_test = (CNST_LIMB(1) << (GMP_LIMB_BITS-7)) - 1;
  25. #define HOST_ENDIAN     (* (signed char *) &endian_test)
  26. #endif
  27. void
  28. mpz_import (mpz_ptr z, size_t count, int order,
  29.             size_t size, int endian, size_t nail, const void *data)
  30. {
  31.   mp_size_t  zsize;
  32.   mp_ptr     zp;
  33.   ASSERT (order == 1 || order == -1);
  34.   ASSERT (endian == 1 || endian == 0 || endian == -1);
  35.   ASSERT (nail <= 8*size);
  36.   zsize = (count * (8*size - nail) + GMP_NUMB_BITS-1) / GMP_NUMB_BITS;
  37.   MPZ_REALLOC (z, zsize);
  38.   zp = PTR(z);
  39.   if (endian == 0)
  40.     endian = HOST_ENDIAN;
  41.   /* Can't use these special cases with nails currently, since they don't
  42.      mask out the nail bits in the input data.  */
  43.   if (nail == 0 && GMP_NAIL_BITS == 0)
  44.     {
  45.       unsigned  align = ((char *) data - (char *) NULL) % sizeof (mp_limb_t);
  46.       if (order == -1
  47.           && size == sizeof (mp_limb_t)
  48.           && endian == HOST_ENDIAN
  49.           && align == 0)
  50.         {
  51.           MPN_COPY (zp, (mp_srcptr) data, (mp_size_t) count);
  52.           goto done;
  53.         }
  54.       if (order == -1
  55.           && size == sizeof (mp_limb_t)
  56.           && endian == - HOST_ENDIAN
  57.           && align == 0)
  58.         {
  59.           MPN_BSWAP (zp, (mp_srcptr) data, (mp_size_t) count);
  60.           goto done;
  61.         }
  62.       if (order == 1
  63.           && size == sizeof (mp_limb_t)
  64.           && endian == HOST_ENDIAN
  65.           && align == 0)
  66.         {
  67.           MPN_REVERSE (zp, (mp_srcptr) data, (mp_size_t) count);
  68.           goto done;
  69.         }
  70.     }
  71.   {
  72.     mp_limb_t      limb, byte, wbitsmask;
  73.     size_t         i, j, numb, wbytes;
  74.     mp_size_t      woffset;
  75.     unsigned char  *dp;
  76.     int            lbits, wbits;
  77.     numb = size * 8 - nail;
  78.     /* whole bytes to process */
  79.     wbytes = numb / 8;
  80.     /* partial byte to process */
  81.     wbits = numb % 8;
  82.     wbitsmask = (CNST_LIMB(1) << wbits) - 1;
  83.     /* offset to get to the next word after processing wbytes and wbits */
  84.     woffset = (numb + 7) / 8;
  85.     woffset = (endian >= 0 ? woffset : -woffset)
  86.       + (order < 0 ? size : - (mp_size_t) size);
  87.     /* least significant byte */
  88.     dp = (unsigned char *) data
  89.       + (order >= 0 ? (count-1)*size : 0) + (endian >= 0 ? size-1 : 0);
  90. #define ACCUMULATE(N)                                   
  91.     do {                                                
  92.       ASSERT (lbits < GMP_NUMB_BITS);                   
  93.       ASSERT (limb <= (CNST_LIMB(1) << lbits) - 1);     
  94.                                                         
  95.       limb |= (mp_limb_t) byte << lbits;                
  96.       lbits += (N);                                     
  97.       if (lbits >= GMP_NUMB_BITS)                       
  98.         {                                               
  99.           *zp++ = limb & GMP_NUMB_MASK;                 
  100.           lbits -= GMP_NUMB_BITS;                       
  101.           ASSERT (lbits < (N));                         
  102.           limb = byte >> ((N) - lbits);                 
  103.         }                                               
  104.     } while (0)
  105.     limb = 0;
  106.     lbits = 0;
  107.     for (i = 0; i < count; i++)
  108.       {
  109.         for (j = 0; j < wbytes; j++)
  110.           {
  111.             byte = *dp;
  112.             dp -= endian;
  113.             ACCUMULATE (8);
  114.           }
  115.         if (wbits != 0)
  116.           {
  117.             byte = *dp & wbitsmask;
  118.             dp -= endian;
  119.             ACCUMULATE (wbits);
  120.           }
  121.         dp += woffset;
  122.       }
  123.     if (lbits != 0)
  124.       {
  125.         ASSERT (lbits <= GMP_NUMB_BITS);
  126.         ASSERT_LIMB (limb);
  127.         *zp++ = limb;
  128.       }
  129.     ASSERT (zp == PTR(z) + zsize);
  130.     /* low byte of word after most significant */
  131.     ASSERT (dp == (unsigned char *) data
  132.             + (order < 0 ? count*size : - (mp_size_t) size)
  133.             + (endian >= 0 ? (mp_size_t) size - 1 : 0));
  134.   }
  135.  done:
  136.   zp = PTR(z);
  137.   MPN_NORMALIZE (zp, zsize);
  138.   SIZ(z) = zsize;
  139. }