pqc/external/flint-2.4.3/fmpz_mat.h
2014-05-24 23:16:06 +02:00

317 lines
10 KiB
C

/*=============================================================================
This file is part of FLINT.
FLINT is free software; you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation; either version 2 of the License, or
(at your option) any later version.
FLINT is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
You should have received a copy of the GNU General Public License
along with FLINT; if not, write to the Free Software
Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
=============================================================================*/
/******************************************************************************
Copyright (C) 2010 William Hart
Copyright (C) 2010 Fredrik Johansson
******************************************************************************/
#ifndef FMPZ_MAT_H
#define FMPZ_MAT_H
#undef ulong
#define ulong ulongxx /* interferes with system includes */
#include <stdio.h>
#undef ulong
#include <gmp.h>
#define ulong mp_limb_t
#include "flint.h"
#include "fmpz.h"
#include "nmod_mat.h"
#include "fmpz_poly.h"
#ifdef __cplusplus
extern "C" {
#endif
typedef struct
{
fmpz * entries;
slong r;
slong c;
fmpz ** rows;
} fmpz_mat_struct;
typedef fmpz_mat_struct fmpz_mat_t[1];
/* Memory management ********************************************************/
#define fmpz_mat_entry(mat,i,j) ((mat)->rows[i] + (j))
#define fmpz_mat_nrows(mat) ((mat)->r)
#define fmpz_mat_ncols(mat) ((mat)->c)
void fmpz_mat_init(fmpz_mat_t mat, slong rows, slong cols);
void fmpz_mat_init_set(fmpz_mat_t mat, const fmpz_mat_t src);
void fmpz_mat_swap(fmpz_mat_t mat1, fmpz_mat_t mat2);
void fmpz_mat_set(fmpz_mat_t mat1, const fmpz_mat_t mat2);
void fmpz_mat_clear(fmpz_mat_t mat);
int fmpz_mat_equal(const fmpz_mat_t mat1, const fmpz_mat_t mat2);
int fmpz_mat_is_zero(const fmpz_mat_t mat);
static __inline__ int
fmpz_mat_is_empty(const fmpz_mat_t mat)
{
return (mat->r == 0) || (mat->c == 0);
}
static __inline__ int
fmpz_mat_is_square(const fmpz_mat_t mat)
{
return (mat->r == mat->c);
}
void fmpz_mat_zero(fmpz_mat_t mat);
void fmpz_mat_one(fmpz_mat_t mat);
/* Input and output *********************************************************/
int fmpz_mat_fprint(FILE * file, const fmpz_mat_t mat);
int fmpz_mat_fprint_pretty(FILE * file, const fmpz_mat_t mat);
static __inline__
int fmpz_mat_print(const fmpz_mat_t mat)
{
return fmpz_mat_fprint(stdout, mat);
}
static __inline__
int fmpz_mat_print_pretty(const fmpz_mat_t mat)
{
return fmpz_mat_fprint_pretty(stdout, mat);
}
int fmpz_mat_fread(FILE* file, fmpz_mat_t mat);
static __inline__
int fmpz_mat_read(fmpz_mat_t mat)
{
return fmpz_mat_fread(stdin, mat);
}
/* Random matrix generation *************************************************/
void fmpz_mat_randbits(fmpz_mat_t mat, flint_rand_t state, mp_bitcnt_t bits);
void fmpz_mat_randtest(fmpz_mat_t mat, flint_rand_t state, mp_bitcnt_t bits);
void fmpz_mat_randtest_unsigned(fmpz_mat_t mat, flint_rand_t state, mp_bitcnt_t bits);
void fmpz_mat_randintrel(fmpz_mat_t mat, flint_rand_t state, mp_bitcnt_t bits);
void fmpz_mat_randsimdioph(fmpz_mat_t mat, flint_rand_t state, mp_bitcnt_t bits, mp_bitcnt_t bits2);
void fmpz_mat_randntrulike(fmpz_mat_t mat, flint_rand_t state, mp_bitcnt_t bits, ulong q);
void fmpz_mat_randntrulike2(fmpz_mat_t mat, flint_rand_t state, mp_bitcnt_t bits, ulong q);
void fmpz_mat_randajtai(fmpz_mat_t mat, flint_rand_t state, double alpha);
void fmpz_mat_randrank(fmpz_mat_t mat, flint_rand_t state, slong rank, mp_bitcnt_t bits);
void fmpz_mat_randdet(fmpz_mat_t mat, flint_rand_t state, const fmpz_t det);
void fmpz_mat_randops(fmpz_mat_t mat, flint_rand_t state, slong count);
int fmpz_mat_randpermdiag(fmpz_mat_t mat, flint_rand_t state, const fmpz * diag, slong n);
/* Norms */
slong fmpz_mat_max_bits(const fmpz_mat_t mat);
/* Transpose */
void fmpz_mat_transpose(fmpz_mat_t B, const fmpz_mat_t A);
/* Addition and subtraction */
void fmpz_mat_add(fmpz_mat_t C, const fmpz_mat_t A, const fmpz_mat_t B);
void fmpz_mat_sub(fmpz_mat_t C, const fmpz_mat_t A, const fmpz_mat_t B);
void fmpz_mat_neg(fmpz_mat_t B, const fmpz_mat_t A);
/* Scalar operations */
void fmpz_mat_scalar_mul_fmpz(fmpz_mat_t B, const fmpz_mat_t A, const fmpz_t c);
void fmpz_mat_scalar_mul_si(fmpz_mat_t B, const fmpz_mat_t A, slong c);
void fmpz_mat_scalar_mul_ui(fmpz_mat_t B, const fmpz_mat_t A, ulong c);
void fmpz_mat_scalar_addmul_fmpz(fmpz_mat_t B, const fmpz_mat_t A, const fmpz_t c);
void fmpz_mat_scalar_addmul_si(fmpz_mat_t B, const fmpz_mat_t A, slong c);
void fmpz_mat_scalar_addmul_ui(fmpz_mat_t B, const fmpz_mat_t A, ulong c);
void fmpz_mat_scalar_submul_fmpz(fmpz_mat_t B, const fmpz_mat_t A, const fmpz_t c);
void fmpz_mat_scalar_submul_si(fmpz_mat_t B, const fmpz_mat_t A, slong c);
void fmpz_mat_scalar_submul_ui(fmpz_mat_t B, const fmpz_mat_t A, ulong c);
void fmpz_mat_scalar_addmul_nmod_mat_fmpz(fmpz_mat_t B, const nmod_mat_t A, const fmpz_t c);
void fmpz_mat_scalar_addmul_nmod_mat_ui(fmpz_mat_t B, const nmod_mat_t A, ulong c);
void fmpz_mat_scalar_divexact_fmpz(fmpz_mat_t B, const fmpz_mat_t A, const fmpz_t c);
void fmpz_mat_scalar_divexact_si(fmpz_mat_t B, const fmpz_mat_t A, slong c);
void fmpz_mat_scalar_divexact_ui(fmpz_mat_t B, const fmpz_mat_t A, ulong c);
void fmpz_mat_scalar_mod_fmpz(fmpz_mat_t B, const fmpz_mat_t A, const fmpz_t m);
/* Multiplication */
void fmpz_mat_mul(fmpz_mat_t C, const fmpz_mat_t A, const fmpz_mat_t B);
void fmpz_mat_mul_classical(fmpz_mat_t C, const fmpz_mat_t A,
const fmpz_mat_t B);
void fmpz_mat_mul_classical_inline(fmpz_mat_t C, const fmpz_mat_t A,
const fmpz_mat_t B);
void _fmpz_mat_mul_multi_mod(fmpz_mat_t C, const fmpz_mat_t A,
const fmpz_mat_t B, mp_bitcnt_t bits);
void fmpz_mat_mul_multi_mod(fmpz_mat_t C, const fmpz_mat_t A,
const fmpz_mat_t B);
void fmpz_mat_sqr(fmpz_mat_t B, const fmpz_mat_t A);
void fmpz_mat_pow(fmpz_mat_t B, const fmpz_mat_t A, ulong exp);
/* Permutations */
static __inline__ void
fmpz_mat_swap_rows(fmpz_mat_t mat, slong * perm, slong r, slong s)
{
if (r != s)
{
fmpz * u;
slong t;
if (perm)
{
t = perm[s];
perm[s] = perm[r];
perm[r] = t;
}
u = mat->rows[s];
mat->rows[s] = mat->rows[r];
mat->rows[r] = u;
}
}
/* Gaussian elimination *****************************************************/
slong fmpz_mat_find_pivot_any(const fmpz_mat_t mat,
slong start_row, slong end_row, slong c);
slong fmpz_mat_fflu(fmpz_mat_t B, fmpz_t den, slong * perm,
const fmpz_mat_t A, int rank_check);
slong fmpz_mat_rref(fmpz_mat_t B, fmpz_t den, const fmpz_mat_t A);
/* Modular gaussian elimination *********************************************/
slong
fmpz_mat_rref_mod(slong * perm, fmpz_mat_t A, const fmpz_t p);
/* Trace ********************************************************************/
void fmpz_mat_trace(fmpz_t trace, const fmpz_mat_t mat);
/* Determinant **************************************************************/
void fmpz_mat_det(fmpz_t det, const fmpz_mat_t A);
void fmpz_mat_det_cofactor(fmpz_t det, const fmpz_mat_t A);
void _fmpz_mat_det_cofactor_2x2(fmpz_t det, fmpz ** const x);
void _fmpz_mat_det_cofactor_3x3(fmpz_t det, fmpz ** const x);
void _fmpz_mat_det_cofactor_4x4(fmpz_t det, fmpz ** const x);
void fmpz_mat_det_bareiss(fmpz_t det, const fmpz_mat_t A);
void fmpz_mat_det_modular(fmpz_t det, const fmpz_mat_t A, int proved);
void fmpz_mat_det_modular_accelerated(fmpz_t det,
const fmpz_mat_t A, int proved);
void fmpz_mat_det_modular_given_divisor(fmpz_t det, const fmpz_mat_t A,
const fmpz_t d, int proved);
void fmpz_mat_det_bound(fmpz_t bound, const fmpz_mat_t A);
void fmpz_mat_det_divisor(fmpz_t d, const fmpz_mat_t A);
/* Characteristic polynomial ************************************************/
void _fmpz_mat_charpoly(fmpz *cp, const fmpz_mat_t mat);
void fmpz_mat_charpoly(fmpz_poly_t cp, const fmpz_mat_t mat);
/* Rank *********************************************************************/
slong fmpz_mat_rank(const fmpz_mat_t A);
/* Nonsingular solving ******************************************************/
void fmpz_mat_solve_bound(fmpz_t N, fmpz_t D,
const fmpz_mat_t A, const fmpz_mat_t B);
int fmpz_mat_solve(fmpz_mat_t X, fmpz_t den,
const fmpz_mat_t A, const fmpz_mat_t B);
int fmpz_mat_solve_cramer(fmpz_mat_t X, fmpz_t den,
const fmpz_mat_t A, const fmpz_mat_t B);
int fmpz_mat_solve_fflu(fmpz_mat_t X, fmpz_t den,
const fmpz_mat_t A, const fmpz_mat_t B);
void fmpz_mat_solve_fflu_precomp(fmpz_mat_t X, const slong * perm,
const fmpz_mat_t FFLU, const fmpz_mat_t B);
int fmpz_mat_solve_dixon(fmpz_mat_t X, fmpz_t mod,
const fmpz_mat_t A, const fmpz_mat_t B);
/* Nullspace ****************************************************************/
slong fmpz_mat_nullspace(fmpz_mat_t res, const fmpz_mat_t mat);
/* Inverse ******************************************************************/
int fmpz_mat_inv(fmpz_mat_t B, fmpz_t den, const fmpz_mat_t A);
/* Modular reduction and reconstruction *************************************/
void fmpz_mat_set_nmod_mat(fmpz_mat_t A, const nmod_mat_t Amod);
void fmpz_mat_set_nmod_mat_unsigned(fmpz_mat_t A, const nmod_mat_t Amod);
void fmpz_mat_get_nmod_mat(nmod_mat_t Amod, const fmpz_mat_t A);
void fmpz_mat_CRT_ui(fmpz_mat_t res, const fmpz_mat_t mat1,
const fmpz_t m1, const nmod_mat_t mat2, int sign);
void
fmpz_mat_multi_mod_ui_precomp(nmod_mat_t * residues, slong nres,
const fmpz_mat_t mat, const fmpz_comb_t comb, fmpz_comb_temp_t temp);
void
fmpz_mat_multi_mod_ui(nmod_mat_t * residues, slong nres, const fmpz_mat_t mat);
void
fmpz_mat_multi_CRT_ui_precomp(fmpz_mat_t mat,
nmod_mat_t * const residues, slong nres,
const fmpz_comb_t comb, fmpz_comb_temp_t temp, int sign);
void fmpz_mat_multi_CRT_ui(fmpz_mat_t mat, nmod_mat_t * const residues,
slong nres, int sign);
#ifdef __cplusplus
}
#endif
#endif