1 #include <stdlib.h>
2 #include <string.h>
3
4 #include <isl_int.h>
5
6 extern int isl_sioimath_decode(isl_sioimath val, int32_t *small, mp_int *big);
7 extern int isl_sioimath_decode_big(isl_sioimath val, mp_int *big);
8 extern int isl_sioimath_decode_small(isl_sioimath val, int32_t *small);
9
10 extern isl_sioimath isl_sioimath_encode_small(int32_t val);
11 extern isl_sioimath isl_sioimath_encode_big(mp_int val);
12 extern int isl_sioimath_is_small(isl_sioimath val);
13 extern int isl_sioimath_is_big(isl_sioimath val);
14 extern int32_t isl_sioimath_get_small(isl_sioimath val);
15 extern mp_int isl_sioimath_get_big(isl_sioimath val);
16
17 extern void isl_siomath_uint32_to_digits(uint32_t num, mp_digit *digits,
18 mp_size *used);
19 extern void isl_siomath_ulong_to_digits(unsigned long num, mp_digit *digits,
20 mp_size *used);
21 extern void isl_siomath_uint64_to_digits(uint64_t num, mp_digit *digits,
22 mp_size *used);
23
24 extern mp_int isl_sioimath_bigarg_src(isl_sioimath arg,
25 isl_sioimath_scratchspace_t *scratch);
26 extern mp_int isl_sioimath_siarg_src(signed long arg,
27 isl_sioimath_scratchspace_t *scratch);
28 extern mp_int isl_sioimath_si64arg_src(int64_t arg,
29 isl_sioimath_scratchspace_t *scratch);
30 extern mp_int isl_sioimath_uiarg_src(unsigned long arg,
31 isl_sioimath_scratchspace_t *scratch);
32 extern mp_int isl_sioimath_reinit_big(isl_sioimath_ptr ptr);
33 extern void isl_sioimath_set_small(isl_sioimath_ptr ptr, int32_t val);
34 extern void isl_sioimath_set_int32(isl_sioimath_ptr ptr, int32_t val);
35 extern void isl_sioimath_set_int64(isl_sioimath_ptr ptr, int64_t val);
36 extern void isl_sioimath_promote(isl_sioimath_ptr dst);
37 extern void isl_sioimath_try_demote(isl_sioimath_ptr dst);
38
39 extern void isl_sioimath_init(isl_sioimath_ptr dst);
40 extern void isl_sioimath_clear(isl_sioimath_ptr dst);
41 extern void isl_sioimath_set(isl_sioimath_ptr dst, isl_sioimath_src val);
42 extern void isl_sioimath_set_si(isl_sioimath_ptr dst, long val);
43 extern void isl_sioimath_set_ui(isl_sioimath_ptr dst, unsigned long val);
44 extern int isl_sioimath_fits_slong(isl_sioimath_src val);
45 extern long isl_sioimath_get_si(isl_sioimath_src val);
46 extern int isl_sioimath_fits_ulong(isl_sioimath_src val);
47 extern unsigned long isl_sioimath_get_ui(isl_sioimath_src val);
48 extern double isl_sioimath_get_d(isl_sioimath_src val);
49 extern char *isl_sioimath_get_str(isl_sioimath_src val);
50 extern void isl_sioimath_abs(isl_sioimath_ptr dst, isl_sioimath_src arg);
51 extern void isl_sioimath_neg(isl_sioimath_ptr dst, isl_sioimath_src arg);
52 extern void isl_sioimath_swap(isl_sioimath_ptr lhs, isl_sioimath_ptr rhs);
53 extern void isl_sioimath_add_ui(isl_sioimath_ptr dst, isl_sioimath lhs,
54 unsigned long rhs);
55 extern void isl_sioimath_sub_ui(isl_sioimath_ptr dst, isl_sioimath lhs,
56 unsigned long rhs);
57
58 extern void isl_sioimath_add(isl_sioimath_ptr dst, isl_sioimath_src lhs,
59 isl_sioimath_src rhs);
60 extern void isl_sioimath_sub(isl_sioimath_ptr dst, isl_sioimath_src lhs,
61 isl_sioimath_src rhs);
62 extern void isl_sioimath_mul(isl_sioimath_ptr dst, isl_sioimath_src lhs,
63 isl_sioimath_src rhs);
64 extern void isl_sioimath_mul_2exp(isl_sioimath_ptr dst, isl_sioimath lhs,
65 unsigned long rhs);
66 extern void isl_sioimath_mul_si(isl_sioimath_ptr dst, isl_sioimath lhs,
67 signed long rhs);
68 extern void isl_sioimath_mul_ui(isl_sioimath_ptr dst, isl_sioimath lhs,
69 unsigned long rhs);
70 extern void isl_sioimath_pow_ui(isl_sioimath_ptr dst, isl_sioimath_src lhs,
71 unsigned long rhs);
72 extern void isl_sioimath_addmul(isl_sioimath_ptr dst, isl_sioimath_src lhs,
73 isl_sioimath_src rhs);
74 extern void isl_sioimath_addmul_ui(isl_sioimath_ptr dst, isl_sioimath_src lhs,
75 unsigned long rhs);
76 extern void isl_sioimath_submul(isl_sioimath_ptr dst, isl_sioimath_src lhs,
77 isl_sioimath_src rhs);
78 extern void isl_sioimath_submul_ui(isl_sioimath_ptr dst, isl_sioimath_src lhs,
79 unsigned long rhs);
80
81 /* Implements the Euclidean algorithm to compute the greatest common divisor of
82 * two values in small representation.
83 */
isl_sioimath_smallgcd(int32_t lhs,int32_t rhs)84 static uint32_t isl_sioimath_smallgcd(int32_t lhs, int32_t rhs)
85 {
86 uint32_t dividend, divisor, remainder;
87
88 dividend = labs(lhs);
89 divisor = labs(rhs);
90 while (divisor) {
91 remainder = dividend % divisor;
92 dividend = divisor;
93 divisor = remainder;
94 }
95
96 return dividend;
97 }
98
99 /* Compute the greatest common divisor.
100 *
101 * Per GMP convention, gcd(0,0)==0 and otherwise always positive.
102 */
isl_sioimath_gcd(isl_sioimath_ptr dst,isl_sioimath_src lhs,isl_sioimath_src rhs)103 void isl_sioimath_gcd(isl_sioimath_ptr dst, isl_sioimath_src lhs,
104 isl_sioimath_src rhs)
105 {
106 int32_t lhssmall, rhssmall;
107 uint32_t smallgcd;
108 isl_sioimath_scratchspace_t scratchlhs, scratchrhs;
109
110 if (isl_sioimath_decode_small(lhs, &lhssmall) &&
111 isl_sioimath_decode_small(rhs, &rhssmall)) {
112 smallgcd = isl_sioimath_smallgcd(lhssmall, rhssmall);
113 isl_sioimath_set_small(dst, smallgcd);
114 return;
115 }
116
117 impz_gcd(isl_sioimath_reinit_big(dst),
118 isl_sioimath_bigarg_src(lhs, &scratchlhs),
119 isl_sioimath_bigarg_src(rhs, &scratchrhs));
120 isl_sioimath_try_demote(dst);
121 }
122
123 /* Compute the lowest common multiple of two numbers.
124 */
isl_sioimath_lcm(isl_sioimath_ptr dst,isl_sioimath_src lhs,isl_sioimath_src rhs)125 void isl_sioimath_lcm(isl_sioimath_ptr dst, isl_sioimath_src lhs,
126 isl_sioimath_src rhs)
127 {
128 int32_t lhssmall, rhssmall;
129 uint32_t smallgcd;
130 uint64_t multiple;
131 isl_sioimath_scratchspace_t scratchlhs, scratchrhs;
132
133 if (isl_sioimath_decode_small(lhs, &lhssmall) &&
134 isl_sioimath_decode_small(rhs, &rhssmall)) {
135 if (lhssmall == 0 || rhssmall == 0) {
136 isl_sioimath_set_small(dst, 0);
137 return;
138 }
139 smallgcd = isl_sioimath_smallgcd(lhssmall, rhssmall);
140 multiple = (uint64_t) abs(lhssmall) * (uint64_t) abs(rhssmall);
141 isl_sioimath_set_int64(dst, multiple / smallgcd);
142 return;
143 }
144
145 impz_lcm(isl_sioimath_reinit_big(dst),
146 isl_sioimath_bigarg_src(lhs, &scratchlhs),
147 isl_sioimath_bigarg_src(rhs, &scratchrhs));
148 isl_sioimath_try_demote(dst);
149 }
150
151 extern void isl_sioimath_tdiv_q(isl_sioimath_ptr dst, isl_sioimath_src lhs,
152 isl_sioimath_src rhs);
153 extern void isl_sioimath_tdiv_q_ui(isl_sioimath_ptr dst, isl_sioimath_src lhs,
154 unsigned long rhs);
155 extern void isl_sioimath_cdiv_q(isl_sioimath_ptr dst, isl_sioimath_src lhs,
156 isl_sioimath_src rhs);
157 extern void isl_sioimath_cdiv_q_ui(isl_sioimath_ptr dst, isl_sioimath_src lhs,
158 unsigned long rhs);
159 extern void isl_sioimath_fdiv_q(isl_sioimath_ptr dst, isl_sioimath_src lhs,
160 isl_sioimath_src rhs);
161 extern void isl_sioimath_fdiv_q_ui(isl_sioimath_ptr dst, isl_sioimath_src lhs,
162 unsigned long rhs);
163 extern void isl_sioimath_fdiv_r(isl_sioimath_ptr dst, isl_sioimath_src lhs,
164 isl_sioimath_src rhs);
165
166 /* Parse a number from a string.
167 * If it has less than 10 characters then it will fit into the small
168 * representation (i.e. strlen("2147483647")). Otherwise, let IMath parse it.
169 */
isl_sioimath_read(isl_sioimath_ptr dst,const char * str)170 void isl_sioimath_read(isl_sioimath_ptr dst, const char *str)
171 {
172 int32_t small;
173
174 if (strlen(str) < 10) {
175 small = strtol(str, NULL, 10);
176 isl_sioimath_set_small(dst, small);
177 return;
178 }
179
180 mp_int_read_string(isl_sioimath_reinit_big(dst), 10, str);
181 isl_sioimath_try_demote(dst);
182 }
183
184 extern int isl_sioimath_sgn(isl_sioimath_src arg);
185 extern int isl_sioimath_cmp(isl_sioimath_src lhs, isl_sioimath_src rhs);
186 extern int isl_sioimath_cmp_si(isl_sioimath_src lhs, signed long rhs);
187 extern int isl_sioimath_abs_cmp(isl_sioimath_src lhs, isl_sioimath_src rhs);
188 extern int isl_sioimath_is_divisible_by(isl_sioimath_src lhs,
189 isl_sioimath_src rhs);
190
191 extern uint32_t isl_sioimath_hash(isl_sioimath_src arg, uint32_t hash);
192 extern size_t isl_sioimath_sizeinbase(isl_sioimath_src arg, int base);
193 extern void isl_sioimath_print(FILE *out, isl_sioimath_src i, int width);
194
195 /* Print an isl_int to FILE*. Adds space padding to the left until at least
196 * width characters are printed.
197 */
isl_sioimath_print(FILE * out,isl_sioimath_src i,int width)198 void isl_sioimath_print(FILE *out, isl_sioimath_src i, int width)
199 {
200 size_t len;
201 int32_t small;
202 mp_int big;
203 char *buf;
204
205 if (isl_sioimath_decode_small(i, &small)) {
206 fprintf(out, "%*" PRIi32, width, small);
207 return;
208 }
209
210 big = isl_sioimath_get_big(i);
211 len = mp_int_string_len(big, 10);
212 buf = malloc(len);
213 mp_int_to_string(big, 10, buf, len);
214 fprintf(out, "%*s", width, buf);
215 free(buf);
216 }
217
218 /* Print a number to stdout. Meant for debugging.
219 */
isl_sioimath_dump(isl_sioimath_src arg)220 void isl_sioimath_dump(isl_sioimath_src arg)
221 {
222 isl_sioimath_print(stdout, arg, 0);
223 }
224