#include "gr.h"
#include "gr_special.h"
#include "gr_vec.h"
#include "gr_mat.h"
static int
gr_log_jet(gr_ptr res, gr_srcptr x, slong len, gr_ctx_t ctx)
{
slong i;
int status;
slong sz = ctx->sizeof_elem;
if (len <= 0)
return GR_SUCCESS;
status = gr_log(res, x, ctx);
if (status == GR_SUCCESS && len > 1)
{
status |= gr_inv(GR_ENTRY(res, 1, sz), x, ctx);
for (i = 2; i < len; i++)
status |= gr_mul(GR_ENTRY(res, i, sz), GR_ENTRY(res, (i + 1) / 2, sz), GR_ENTRY(res, i / 2, sz), ctx);
for (i = 2; i < len; i++)
status |= gr_div_ui(GR_ENTRY(res, i, sz), GR_ENTRY(res, i, sz), i, ctx);
for (i = 2; i < len; i += 2)
status |= gr_neg(GR_ENTRY(res, i, sz), GR_ENTRY(res, i, sz), ctx);
}
return status;
}
int
gr_mat_log_jordan(gr_mat_t res, const gr_mat_t A, gr_ctx_t ctx)
{
return gr_mat_func_jordan(res, A, (gr_method_vec_op) gr_log_jet, ctx);
}
int
gr_mat_log(gr_mat_t res, const gr_mat_t A, gr_ctx_t ctx)
{
return GR_MAT_UNARY_OP(ctx, MAT_LOG)(res, A, ctx);
}