1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
|
/*
Copyright (C) 2010 William Hart
Copyright (C) 2010 Sebastian Pancratz
This file is part of FLINT.
FLINT is free software: you can redistribute it and/or modify it under
the terms of the GNU Lesser General Public License (LGPL) as published
by the Free Software Foundation; either version 3 of the License, or
(at your option) any later version. See <https://www.gnu.org/licenses/>.
*/
#ifndef FMPZ_VEC_H
#define FMPZ_VEC_H
#ifdef FMPZ_VEC_INLINES_C
#define FMPZ_VEC_INLINE
#else
#define FMPZ_VEC_INLINE static inline
#endif
#include "fmpz_types.h"
#ifdef __cplusplus
extern "C" {
#endif
#define FMPZ_VEC_NORM(vec, i) \
do \
while ((i) && vec[(i) - 1] == WORD(0)) \
(i)--; \
while (0)
#define FMPZ_VEC_SWAP(vec1, len1, vec2, len2) \
do { \
FLINT_SWAP(fmpz *, vec1, vec2); \
FLINT_SWAP(slong, len1, len2); \
} while (0)
/* makes assignment to vec */
#define FMPZ_VEC_TMP_INIT(vec, len) \
do { \
fmpz * __vres; \
slong __vi; \
vec = __vres = TMP_ARRAY_ALLOC(len, fmpz); \
for (__vi = (len); __vi > 0; __vi--) \
fmpz_init(__vres + __vi - 1); \
} while (0)
#define FMPZ_VEC_TMP_CLEAR(vec, len) \
do { \
fmpz * __vres = (vec); \
slong __vi; \
for (__vi = (len); __vi > 0; __vi--) \
fmpz_clear(__vres + __vi - 1); \
} while (0)
/* Memory management *******************************************************/
FMPZ_VEC_INLINE fmpz * _fmpz_vec_init(slong len)
{
return (fmpz *) flint_calloc(len, sizeof(fmpz));
}
void _fmpz_vec_clear(fmpz * vec, slong len);
/* Randomisation ***********************************************************/
void _fmpz_vec_randtest(fmpz * f, flint_rand_t state, slong len, flint_bitcnt_t bits);
void _fmpz_vec_randtest_unsigned(fmpz * f, flint_rand_t state, slong len, flint_bitcnt_t bits);
/* Norms *******************************************************************/
slong _fmpz_vec_max_bits(const fmpz * vec, slong len);
slong _fmpz_vec_max_bits_ref(const fmpz * vec, slong len);
void _fmpz_vec_sum_max_bits(slong * sumabs, slong * maxabs, const fmpz * coeffs, slong length);
slong _fmpz_vec_max_limbs(const fmpz * vec, slong len);
void _fmpz_vec_height(fmpz_t height, const fmpz * vec, slong len);
slong _fmpz_vec_height_index(const fmpz * vec, slong len);
/* Input and output ********************************************************/
#ifdef FLINT_HAVE_FILE
int _fmpz_vec_fprint(FILE * file, const fmpz * vec, slong len);
int _fmpz_vec_fread(FILE * file, fmpz ** vec, slong * len);
#endif
int _fmpz_vec_print(const fmpz * vec, slong len);
int _fmpz_vec_read(fmpz ** vec, slong * len);
/* Conversions *************************************************************/
void _fmpz_vec_get_nmod_vec(nn_ptr res, const fmpz * poly, slong len, nmod_t mod);
void _fmpz_vec_set_nmod_vec(fmpz * res, nn_srcptr poly, slong len, nmod_t mod);
void _fmpz_vec_get_fft(ulong ** coeffs_f, const fmpz * coeffs_m, slong l, slong length);
void _fmpz_vec_set_fft(fmpz * coeffs_m, slong length, const nn_ptr * coeffs_f, slong limbs, slong sign);
slong _fmpz_vec_get_d_vec_2exp(double * appv, const fmpz * vec, slong len);
/* Assignment and basic manipulation ***************************************/
void _fmpz_vec_set(fmpz * vec1, const fmpz * vec2, slong len2);
void _fmpz_vec_swap(fmpz * vec1, fmpz * vec2, slong len2);
void _fmpz_vec_zero(fmpz * vec, slong len);
void _fmpz_vec_scalar_abs(fmpz * vec1, const fmpz * vec2, slong len2);
/* Comparison **************************************************************/
int _fmpz_vec_equal(const fmpz * vec1, const fmpz * vec2, slong len);
int _fmpz_vec_is_zero(const fmpz * vec, slong len);
void _fmpz_vec_min(fmpz * vec1, const fmpz * vec2, const fmpz * vec3, slong len);
void _fmpz_vec_max(fmpz * vec1, const fmpz * vec2, const fmpz * vec3, slong len);
void _fmpz_vec_min_inplace(fmpz * vec1, const fmpz * vec2, slong len);
void _fmpz_vec_max_inplace(fmpz * vec1, const fmpz * vec2, slong len);
/* Sorting ******************************************************************/
void _fmpz_vec_sort(fmpz * vec, slong len);
/* Addition ****************************************************************/
void _fmpz_vec_add(fmpz * res, const fmpz * vec1, const fmpz * vec2, slong len2);
void _fmpz_vec_sub(fmpz * res, const fmpz * vec1, const fmpz * vec2, slong len2);
void _fmpz_vec_neg(fmpz * vec1, const fmpz * vec2, slong len2);
/* Scalar multiplication and division **************************************/
void _fmpz_vec_scalar_mul_si(fmpz * vec1, const fmpz * vec2, slong len2, slong c);
void _fmpz_vec_scalar_mul_ui(fmpz * vec1, const fmpz * vec2, slong len2, ulong c);
void _fmpz_vec_scalar_mul_fmpz(fmpz * vec1, const fmpz * vec2, slong len2, const fmpz_t x);
void _fmpz_vec_scalar_mul_2exp(fmpz * vec1, const fmpz * vec2, slong len2, ulong exp);
void _fmpz_vec_scalar_divexact_si(fmpz * vec1, const fmpz * vec2, slong len2, slong c);
void _fmpz_vec_scalar_divexact_ui(fmpz * vec1, const fmpz * vec2, slong len2, ulong c);
void _fmpz_vec_scalar_divexact_fmpz(fmpz * vec1, const fmpz * vec2, slong len2, const fmpz_t x);
void _fmpz_vec_scalar_fdiv_q_si(fmpz * vec1, const fmpz * vec2, slong len2, slong c);
void _fmpz_vec_scalar_fdiv_q_ui(fmpz * vec1, const fmpz * vec2, slong len2, ulong c);
void _fmpz_vec_scalar_fdiv_q_fmpz(fmpz * vec1, const fmpz * vec2, slong len2, const fmpz_t c);
void _fmpz_vec_scalar_fdiv_q_2exp(fmpz * vec1, const fmpz * vec2, slong len2, ulong exp);
void _fmpz_vec_scalar_fdiv_r_2exp(fmpz * vec1, const fmpz * vec2, slong len2, ulong exp);
void _fmpz_vec_scalar_tdiv_q_si(fmpz * vec1, const fmpz * vec2, slong len2, slong c);
void _fmpz_vec_scalar_tdiv_q_ui(fmpz * vec1, const fmpz * vec2, slong len2, ulong c);
void _fmpz_vec_scalar_tdiv_q_fmpz(fmpz * vec1, const fmpz * vec2, slong len2, const fmpz_t c);
void _fmpz_vec_scalar_tdiv_q_2exp(fmpz * vec1, const fmpz * vec2, slong len2, ulong exp);
void _fmpz_vec_scalar_addmul_si(fmpz * vec1, const fmpz * vec2, slong len2, slong c);
void _fmpz_vec_scalar_addmul_ui(fmpz * vec1, const fmpz * vec2, slong len2, ulong c);
void _fmpz_vec_scalar_addmul_fmpz(fmpz * poly1, const fmpz * poly2, slong len2, const fmpz_t x);
void _fmpz_vec_scalar_addmul_si_2exp(fmpz * vec1, const fmpz * vec2, slong len2, slong c, ulong exp);
void _fmpz_vec_scalar_submul_si(fmpz * vec1, const fmpz * vec2, slong len2, slong c);
void _fmpz_vec_scalar_submul_fmpz(fmpz * vec1, const fmpz * vec2, slong len2, const fmpz_t x);
void _fmpz_vec_scalar_submul_si_2exp(fmpz * vec1, const fmpz * vec2, slong len2, slong c, ulong exp);
/* Vector sum and product **************************************************/
void _fmpz_vec_sum(fmpz_t res, const fmpz * vec, slong len);
void _fmpz_vec_prod(fmpz_t res, const fmpz * vec, slong len);
void _fmpz_ui_vec_prod(fmpz_t res, nn_srcptr vec, slong len);
/* Reduction mod p **********************************************************/
void _fmpz_vec_scalar_mod_fmpz(fmpz *res, const fmpz *vec, slong len, const fmpz_t p);
void _fmpz_vec_scalar_smod_fmpz(fmpz *res, const fmpz *vec, slong len, const fmpz_t p);
/* Gaussian content ********************************************************/
void _fmpz_vec_content(fmpz_t res, const fmpz * vec, slong len);
void _fmpz_vec_content_chained(fmpz_t res, const fmpz * vec, slong len, const fmpz_t inp);
void _fmpz_vec_lcm(fmpz_t res, const fmpz * vec, slong len);
/* Dot product *************************************************************/
void _fmpz_vec_dot_general_naive(fmpz_t res, const fmpz_t initial, int subtract,
const fmpz * a, const fmpz * b, int reverse, slong len);
void _fmpz_vec_dot_general(fmpz_t res, const fmpz_t initial, int subtract,
const fmpz * a, const fmpz * b, int reverse, slong len);
FMPZ_VEC_INLINE
void _fmpz_vec_dot(fmpz_t res, const fmpz * vec1, const fmpz * vec2, slong len2)
{
_fmpz_vec_dot_general(res, NULL, 0, vec1, vec2, 0, len2);
}
#ifdef __cplusplus
}
#endif
#endif
|
