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

数学计算

开发平台:

Unix_Linux

  1. /* mpn_toom43_mul -- Multiply {ap,an} and {bp,bn} where an is nominally 4/3
  2.    times as large as bn.  Or more accurately, bn < an < 2 bn.
  3.    Contributed to the GNU project by Marco Bodrato.
  4.    The idea of applying toom to unbalanced multiplication is due to Marco
  5.    Bodrato and Alberto Zanoni.
  6.    THE FUNCTION IN THIS FILE IS INTERNAL WITH A MUTABLE INTERFACE.  IT IS ONLY
  7.    SAFE TO REACH IT THROUGH DOCUMENTED INTERFACES.  IN FACT, IT IS ALMOST
  8.    GUARANTEED THAT IT WILL CHANGE OR DISAPPEAR IN A FUTURE GNU MP RELEASE.
  9. Copyright 2009 Free Software Foundation, Inc.
  10. This file is part of the GNU MP Library.
  11. The GNU MP Library is free software; you can redistribute it and/or modify
  12. it under the terms of the GNU Lesser General Public License as published by
  13. the Free Software Foundation; either version 3 of the License, or (at your
  14. option) any later version.
  15. The GNU MP Library is distributed in the hope that it will be useful, but
  16. WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
  17. or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU Lesser General Public
  18. License for more details.
  19. You should have received a copy of the GNU Lesser General Public License
  20. along with the GNU MP Library.  If not, see http://www.gnu.org/licenses/.  */
  21. #include "gmp.h"
  22. #include "gmp-impl.h"
  23. /* Evaluate in: -2, -1, 0, +1, +2, +inf
  24.   <-s-><--n--><--n--><--n-->
  25.    ___ ______ ______ ______
  26.   |a3_|___a2_|___a1_|___a0_|
  27. |_b2_|___b1_|___b0_|
  28. <-t--><--n--><--n-->
  29.   v0  =  a0             * b0          #   A(0)*B(0)
  30.   v1  = (a0+ a1+ a2+ a3)*(b0+ b1+ b2) #   A(1)*B(1)      ah  <= 3  bh <= 2
  31.   vm1 = (a0- a1+ a2- a3)*(b0- b1+ b2) #  A(-1)*B(-1)    |ah| <= 1 |bh|<= 1
  32.   v2  = (a0+2a1+4a2+8a3)*(b0+2b1+4b2) #   A(2)*B(2)      ah  <= 14 bh <= 6
  33.   vm2 = (a0-2a1+4a2-8a3)*(b0-2b1+4b2) #  A(-2)*B(-2)    |ah| <= 9 |bh|<= 4
  34.   vinf=              a3 *         b2  # A(inf)*B(inf)
  35. */
  36. void
  37. mpn_toom43_mul (mp_ptr pp,
  38. mp_srcptr ap, mp_size_t an,
  39. mp_srcptr bp, mp_size_t bn, mp_ptr scratch)
  40. {
  41.   mp_size_t n, s, t;
  42.   enum toom6_flags flags;
  43.   mp_limb_t cy;
  44. #define a0  ap
  45. #define a1  (ap + n)
  46. #define a2  (ap + 2 * n)
  47. #define a3  (ap + 3 * n)
  48. #define b0  bp
  49. #define b1  (bp + n)
  50. #define b2  (bp + 2 * n)
  51.   n = 1 + (3 * an >= 4 * bn ? (an - 1) >> 2 : (bn - 1) / (size_t) 3);
  52.   s = an - 3 * n;
  53.   t = bn - 2 * n;
  54.   ASSERT (0 < s && s <= n);
  55.   ASSERT (0 < t && t <= n);
  56.   /* This is true whenever an >= 25 or bn >= 19, I think. It
  57.      guarantees that we can fit 5 values of size n+1 in the product
  58.      area. */
  59.   ASSERT (s+t >= 5);
  60. #define v0    pp /* 2n */
  61. #define vm1   (scratch) /* 2n+1 */
  62. #define v1    (pp + 2*n) /* 2n+1 */
  63. #define vm2   (scratch + 2 * n + 1) /* 2n+1 */
  64. #define v2    (scratch + 4 * n + 2) /* 2n+1 */
  65. #define vinf  (pp + 5 * n) /* s+t */
  66. #define bs1    pp /* n+1 */
  67. #define bsm1  (scratch + 2 * n + 2) /* n+1 */
  68. #define asm1  (scratch + 3 * n + 3) /* n+1 */
  69. #define asm2  (scratch + 4 * n + 4) /* n+1 */
  70. #define bsm2  (pp + n + 1) /* n+1 */
  71. #define bs2   (pp + 2 * n + 2) /* n+1 */
  72. #define as2   (pp + 3 * n + 3) /* n+1 */
  73. #define as1   (pp + 4 * n + 4) /* n+1 */
  74.   /* Total sccratch need is 6 * n + 3 + 1; we allocate one extra
  75.      limb, because products will overwrite 2n+2 limbs. */
  76. #define a0a2  scratch
  77. #define b0b2  scratch
  78. #define a1a3  asm1
  79. #define b1d   bsm1
  80.   /* Compute as2 and asm2.  */
  81.   flags = toom6_vm2_neg & mpn_toom_eval_dgr3_pm2 (as2, asm2, ap, n, s, a1a3);
  82.   /* Compute bs2 and bsm2.  */
  83.   b1d[n] = mpn_lshift (b1d, b1, n, 1); /*       2b1      */
  84.   cy  = mpn_lshift (b0b2, b2, t, 2); /*  4b2           */
  85.   cy += mpn_add_n (b0b2, b0b2, b0, t); /*  4b2      + b0 */
  86.   if (t != n)
  87.     cy = mpn_add_1 (b0b2 + t, b0 + t, n - t, cy);
  88.   b0b2[n] = cy;
  89. #if HAVE_NATIVE_mpn_add_n_sub_n
  90.   if (mpn_cmp (b0b2, b1d, n+1) < 0)
  91.     {
  92.       mpn_add_n_sub_n (bs2, bsm2, b1d, b0b2, n+1);
  93.       flags ^= toom6_vm2_neg;
  94.     }
  95.   else
  96.     {
  97.       mpn_add_n_sub_n (bs2, bsm2, b0b2, b1d, n+1);
  98.     }
  99. #else
  100.   mpn_add_n (bs2, b0b2, b1d, n+1);
  101.   if (mpn_cmp (b0b2, b1d, n+1) < 0)
  102.     {
  103.       mpn_sub_n (bsm2, b1d, b0b2, n+1);
  104.       flags ^= toom6_vm2_neg;
  105.     }
  106.   else
  107.     {
  108.       mpn_sub_n (bsm2, b0b2, b1d, n+1);
  109.     }
  110. #endif
  111.   /* Compute as1 and asm1.  */
  112.   flags ^= toom6_vm1_neg & mpn_toom_eval_dgr3_pm1 (as1, asm1, ap, n, s, a0a2);
  113.   /* Compute bs1 and bsm1.  */
  114.   bsm1[n] = mpn_add (bsm1, b0, n, b2, t);
  115. #if HAVE_NATIVE_mpn_add_n_sub_n
  116.   if (bsm1[n] == 0 && mpn_cmp (bsm1, b1, n) < 0)
  117.     {
  118.       cy = mpn_add_n_sub_n (bs1, bsm1, b1, bsm1, n);
  119.       bs1[n] = cy >> 1;
  120.       flags ^= toom6_vm1_neg;
  121.     }
  122.   else
  123.     {
  124.       cy = mpn_add_n_sub_n (bs1, bsm1, bsm1, b1, n);
  125.       bs1[n] = bsm1[n] + (cy >> 1);
  126.       bsm1[n]-= cy & 1;
  127.     }
  128. #else
  129.   bs1[n] = bsm1[n] + mpn_add_n (bs1, bsm1, b1, n);
  130.   if (bsm1[n] == 0 && mpn_cmp (bsm1, b1, n) < 0)
  131.     {
  132.       mpn_sub_n (bsm1, b1, bsm1, n);
  133.       flags ^= toom6_vm1_neg;
  134.     }
  135.   else
  136.     {
  137.       bsm1[n] -= mpn_sub_n (bsm1, bsm1, b1, n);
  138.     }
  139. #endif
  140.   ASSERT (as1[n] <= 3);
  141.   ASSERT (bs1[n] <= 2);
  142.   ASSERT (asm1[n] <= 1);
  143.   ASSERT (bsm1[n] <= 1);
  144.   ASSERT (as2[n] <=14);
  145.   ASSERT (bs2[n] <= 6);
  146.   ASSERT (asm2[n] <= 9);
  147.   ASSERT (bsm2[n] <= 4);
  148.   /* vm1, 2n+1 limbs */
  149.   mpn_mul_n (vm1, asm1, bsm1, n+1);  /* W4 */
  150.   /* vm2, 2n+1 limbs */
  151.   mpn_mul_n (vm2, asm2, bsm2, n+1);  /* W2 */
  152.   /* v2, 2n+1 limbs */
  153.   mpn_mul_n (v2, as2, bs2, n+1);  /* W1 */
  154.   /* v1, 2n+1 limbs */
  155.   mpn_mul_n (v1, as1, bs1, n+1);  /* W3 */
  156.   /* vinf, s+t limbs */   /* W0 */
  157.   if (s > t)  mpn_mul (vinf, a3, s, b2, t);
  158.   else        mpn_mul (vinf, b2, t, a3, s);
  159.   /* v0, 2n limbs */
  160.   mpn_mul_n (v0, ap, bp, n);  /* W5 */
  161.   mpn_toom_interpolate_6pts (pp, n, flags, vm1, vm2, v2, t + s);
  162. #undef v0
  163. #undef vm1
  164. #undef v1
  165. #undef vm2
  166. #undef v2
  167. #undef vinf
  168. #undef bs1
  169. #undef bs2
  170. #undef bsm1
  171. #undef bsm2
  172. #undef asm1
  173. #undef asm2
  174. /* #undef as1 */
  175. /* #undef as2 */
  176. #undef a0a2
  177. #undef b0b2
  178. #undef a1a3
  179. #undef b1d
  180. #undef a0
  181. #undef a1
  182. #undef a2
  183. #undef a3
  184. #undef b0
  185. #undef b1
  186. #undef b2
  187. }