/*
    Copyright (C) 2009, 2010 William Hart
    Copyright (C) 2009, 2010 Andy Novocin
    Copyright (C) 2014 Abhinav Baid

    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/>.
*/

#include "double_extras.h"
#include "fmpz.h"
#include "fmpz_vec.h"
#include "fmpz_mat.h"
#include "fmpz_lll.h"

#ifdef FUNC_HEAD
#undef FUNC_HEAD
#endif

#ifdef LIMIT
#undef LIMIT
#endif

#ifdef COMPUTE
#undef COMPUTE
#endif

#ifdef TYPE
#undef TYPE
#endif

#define FUNC_HEAD int fmpz_lll_check_babai_heuristic_d(int kappa, fmpz_mat_t B, fmpz_mat_t U, d_mat_t mu, d_mat_t r, double *s, \
       d_mat_t appB, int *expo, fmpz_gram_t A, \
       int a, int zeros, int kappamax, int n, const fmpz_lll_t fl)
#define LIMIT kappa
#define COMPUTE(G, I, J, C)                                         \
do {                                                                \
    d_mat_entry(G, I, J) =                                          \
            fmpz_lll_heuristic_dot(appB->rows[I], appB->rows[J], C, \
                                   B, I, J, expo[I] + expo[J]);     \
} while (0)
#define TYPE 1
#include "babai.c"
#undef FUNC_HEAD
#undef LIMIT
#undef COMPUTE
#undef TYPE
