xref: /netbsd-src/external/lgpl3/gmp/dist/tests/mpn/toom-sqr-shared.h (revision 72c7faa4dbb41dbb0238d6b4a109da0d4b236dd4)
1 /* Test for various Toom squaring functions.
2 
3 Copyright 2009, 2012 Free Software Foundation, Inc.
4 
5 This file is part of the GNU MP Library test suite.
6 
7 The GNU MP Library test suite is free software; you can redistribute it
8 and/or modify it under the terms of the GNU General Public License as
9 published by the Free Software Foundation; either version 3 of the License,
10 or (at your option) any later version.
11 
12 The GNU MP Library test suite is distributed in the hope that it will be
13 useful, but WITHOUT ANY WARRANTY; without even the implied warranty of
14 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General
15 Public License for more details.
16 
17 You should have received a copy of the GNU General Public License along with
18 the GNU MP Library test suite.  If not, see https://www.gnu.org/licenses/.  */
19 
20 
21 #include <stdlib.h>
22 #include <stdio.h>
23 
24 #include "gmp-impl.h"
25 #include "tests.h"
26 
27 /* Main file is expected to define mpn_toomN_mul, mpn_toomN_sqr_itch,
28  * MIN_AN, MAX_AN and then include this file. */
29 
30 #ifndef COUNT
31 #define COUNT 2000
32 #endif
33 
34 int
main(int argc,char ** argv)35 main (int argc, char **argv)
36 {
37   mp_ptr ap, refp, pp, scratch;
38   int count = COUNT;
39   int test;
40   gmp_randstate_ptr rands;
41   TMP_DECL;
42   TMP_MARK;
43 
44   TESTS_REPS (count, argv, argc);
45 
46   tests_start ();
47 
48   if (MAX_AN > MIN_AN) {
49     rands = RANDS;
50 
51     ap = TMP_ALLOC_LIMBS (MAX_AN);
52     refp = TMP_ALLOC_LIMBS (MAX_AN * 2);
53     pp = 1 + TMP_ALLOC_LIMBS (MAX_AN * 2 + 2);
54     scratch
55       = 1+TMP_ALLOC_LIMBS (mpn_toomN_sqr_itch (MAX_AN) + 2);
56 
57     for (test = 0; test < count; test++)
58       {
59 	mp_size_t an;
60 	mp_size_t itch;
61 	mp_limb_t p_before, p_after, s_before, s_after;
62 
63 	an = MIN_AN
64 	  + gmp_urandomm_ui (rands, MAX_AN - MIN_AN);
65 
66 	mpn_random2 (ap, an);
67 	mpn_random2 (pp-1, an * 2 + 2);
68 	p_before = pp[-1];
69 	p_after = pp[an * 2];
70 
71 	itch = mpn_toomN_sqr_itch (an);
72 	ASSERT_ALWAYS (itch <= mpn_toomN_sqr_itch (MAX_AN));
73 	mpn_random2 (scratch-1, itch+2);
74 	s_before = scratch[-1];
75 	s_after = scratch[itch];
76 
77 	mpn_toomN_sqr (pp, ap, an, scratch);
78 	refmpn_mul (refp, ap, an, ap, an);
79 	if (pp[-1] != p_before || pp[an * 2] != p_after
80 	    || scratch[-1] != s_before || scratch[itch] != s_after
81 	    || mpn_cmp (refp, pp, an * 2) != 0)
82 	  {
83 	    printf ("ERROR in test %d, an = %d\n",
84 		    test, (int) an);
85 	    if (pp[-1] != p_before)
86 	      {
87 		printf ("before pp:"); mpn_dump (pp -1, 1);
88 		printf ("keep:   "); mpn_dump (&p_before, 1);
89 	      }
90 	    if (pp[an * 2] != p_after)
91 	      {
92 		printf ("after pp:"); mpn_dump (pp + an * 2, 1);
93 		printf ("keep:   "); mpn_dump (&p_after, 1);
94 	      }
95 	    if (scratch[-1] != s_before)
96 	      {
97 		printf ("before scratch:"); mpn_dump (scratch-1, 1);
98 		printf ("keep:   "); mpn_dump (&s_before, 1);
99 	      }
100 	    if (scratch[itch] != s_after)
101 	      {
102 		printf ("after scratch:"); mpn_dump (scratch + itch, 1);
103 		printf ("keep:   "); mpn_dump (&s_after, 1);
104 	      }
105 	    mpn_dump (ap, an);
106 	    mpn_dump (pp, an * 2);
107 	    mpn_dump (refp, an * 2);
108 
109 	    abort();
110 	  }
111       }
112     TMP_FREE;
113   }
114 
115   tests_end ();
116   return 0;
117 }
118