arb/fmpz_holonomic/get_nth_fmpz.c

68 lines
2 KiB
C

/*=============================================================================
This file is part of ARB.
ARB 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.
ARB 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 ARB; if not, write to the Free Software
Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
=============================================================================*/
/******************************************************************************
Copyright (C) 2012 Fredrik Johansson
******************************************************************************/
#include "fmpz_holonomic.h"
void
fmpz_holonomic_get_nth_fmpz(fmpz_t res, const fmpz_holonomic_t op, const fmpz * initial, long n0, long n)
{
long r = fmpz_holonomic_order(op);
if (r == 0)
{
fmpz_zero(res);
return;
}
else if (n < n0)
{
printf("not implemented\n");
abort();
}
else if (n - n0 < r)
{
fmpz_set(res, initial + n - n0);
return;
}
else
{
fmpz_mat_t M;
long i;
fmpz_t Q;
fmpz_mat_init(M, r, r);
fmpz_init(Q);
fmpz_holonomic_forward_fmpz_mat(M, Q, op, n0, n - n0 - r + 1);
fmpz_mul(fmpz_mat_entry(M, r - 1, 0), fmpz_mat_entry(M, r - 1, 0), initial + 0);
for (i = 1; i < r; i++)
fmpz_addmul(fmpz_mat_entry(M, r - 1, 0), fmpz_mat_entry(M, r - 1, i), initial + i);
fmpz_divexact(res, fmpz_mat_entry(M, r - 1, 0), Q);
fmpz_mat_clear(M);
fmpz_clear(Q);
}
}