void
acb_mat_pow_ui(acb_mat_t B, const acb_mat_t A, ulong exp, long prec)
{
long d = acb_mat_nrows(A);
if (exp <= 2 || d <= 1)
{
if (exp == 0 || d == 0)
{
acb_mat_one(B);
}
else if (d == 1)
{
acb_pow_ui(acb_mat_entry(B, 0, 0),
acb_mat_entry(A, 0, 0), exp, prec);
}
else if (exp == 1)
{
acb_mat_set(B, A);
}
else if (exp == 2)
{
acb_mat_mul(B, A, A, prec); /* todo: sqr */
}
}
else
{
acb_mat_t T, U;
long i;
acb_mat_init(T, d, d);
acb_mat_set(T, A);
acb_mat_init(U, d, d);
for (i = ((long) FLINT_BIT_COUNT(exp)) - 2; i >= 0; i--)
{
acb_mat_mul(U, T, T, prec); /* todo: sqr */
if (exp & (1L << i))
acb_mat_mul(T, U, A, prec);
else
acb_mat_swap(T, U);
}
acb_mat_swap(B, T);
acb_mat_clear(T);
acb_mat_clear(U);
}
}
void
acb_mat_mul(acb_mat_t C, const acb_mat_t A, const acb_mat_t B, slong prec)
{
slong ar, ac, br, bc, i, j, k;
ar = acb_mat_nrows(A);
ac = acb_mat_ncols(A);
br = acb_mat_nrows(B);
bc = acb_mat_ncols(B);
if (ac != br || ar != acb_mat_nrows(C) || bc != acb_mat_ncols(C))
{
flint_printf("acb_mat_mul: incompatible dimensions\n");
abort();
}
if (br == 0)
{
acb_mat_zero(C);
return;
}
if (A == C || B == C)
{
acb_mat_t T;
acb_mat_init(T, ar, bc);
acb_mat_mul(T, A, B, prec);
acb_mat_swap(T, C);
acb_mat_clear(T);
return;
}
for (i = 0; i < ar; i++)
{
for (j = 0; j < bc; j++)
{
acb_mul(acb_mat_entry(C, i, j),
acb_mat_entry(A, i, 0),
acb_mat_entry(B, 0, j), prec);
for (k = 1; k < br; k++)
{
acb_addmul(acb_mat_entry(C, i, j),
acb_mat_entry(A, i, k),
acb_mat_entry(B, k, j), prec);
}
}
}
}
int
acb_mat_inv(acb_mat_t X, const acb_mat_t A, long prec)
{
if (X == A)
{
int r;
acb_mat_t T;
acb_mat_init(T, acb_mat_nrows(A), acb_mat_ncols(A));
r = acb_mat_inv(T, A, prec);
acb_mat_swap(T, X);
acb_mat_clear(T);
return r;
}
acb_mat_one(X);
return acb_mat_solve(X, A, X, prec);
}
