Contents

static VALUE cLinearResult;

static void
iter_<%=c_func%>(na_loop_t *const lp)
{
    gsl_matrix X;
    gsl_vector y;
    gsl_vector c;
    gsl_matrix cov;
    double *chisq;
    <%=struct%> *ws = (<%=struct%>*)(lp->opt_ptr);

    SET_GSL_MATRIX_FROM_LOOPT(lp,0,X);
    SET_GSL_VECTOR_FROM_LOOPT(lp,1,y);
    SET_GSL_VECTOR_FROM_LOOPT(lp,2,c);
    SET_GSL_MATRIX_FROM_LOOPT(lp,3,cov);
    chisq = (double*)NDL_PTR(lp,4);

    <%=func_name%>(&X,&y,&c,&cov,chisq,ws);
}

/*
  @overload <%=name%>(<%=args[0][1]%>,<%=args[1][1]%>)
  @param  [DFloat]    <%=args[0][1]%> (input matrix) predictor variables
  @param  [DFloat]    <%=args[1][1]%> (input vector) observation
  @return [GSL::Multifit::LinearResult] result Struct with members: <%=args[2][1]%>, <%=args[3][1]%>, <%=args[4][1]%>.

  <%=desc%>

*/
static VALUE
<%=c_func(2)%>(VALUE mod, VALUE v1, VALUE v2)
{
    size_t c_shape[1], cov_shape[2];
    ndfunc_arg_in_t ain[2] = {{cDF,2},{cDF,1}};
    ndfunc_arg_out_t aout[3] = {{cDF,1,c_shape},{cDF,2,cov_shape},{cDF,0}};
    ndfunc_t ndf = { iter_<%=c_func%>, NO_LOOP|NDF_EXTRACT,
                     2, 3, ain, aout };
    VALUE vws, r, result;
    narray_t *X, *y;
    size_t n, p;
    <%=struct%> *ws;

    GetNArray(v1,X);
    GetNArray(v2,y);
    CHECK_GE_2D(X);
    CHECK_GE_1D(y);
    n = MAT_SIZE1(X);
    p = MAT_SIZE2(X);
    CHECK_SIZE_EQ(n,VEC_SIZE(y),"y size does not match X column size");

    c_shape[0] = cov_shape[0] = cov_shape[1] = p;

    ws = gsl_multifit_linear_alloc(n,p);
    if (!ws)
        rb_raise(rb_eNoMemError,"fail to allocate struct");
    vws = TypedData_Wrap_Struct(<%=class_var%>, &<%=data_type_var%>, (void*)ws);

    r = na_ndloop3(&ndf, ws, 2, v1, v2);

    result = rb_class_new_instance(3, RARRAY_PTR(r), cLinearResult);

    RB_GC_GUARD(vws);
    RB_GC_GUARD(r);
    return result;
}

Version data entries

3 entries across 3 versions & 1 rubygems

Version	Path
numo-gsl-0.1.2	ext/numo/gsl/multifit/tmpl/multifit_linear.c
numo-gsl-0.1.1	ext/numo/gsl/multifit/tmpl/multifit_linear.c
numo-gsl-0.1.0	ext/numo/gsl/multifit/tmpl/multifit_linear.c