Sha256: c199d33f4471a9e535d7e86d4607472f52b29e0ac75965edd6dd492dbf80d3b9

Contents?: true

Size: 1.77 KB

Versions: 3

Compression:

Stored size: 1.77 KB

Contents

<% [64,32].each do |i| %>
#define idx_t int<%=i%>_t
static void
iter_<%=c_func%>_index<%=i%>(na_loop_t *const lp)
{
    size_t   n, idx;
    char    *d_ptr, *i_ptr, *o_ptr;
    ssize_t  d_step, i_step;

    INIT_COUNTER(lp, n);
    INIT_PTR(lp, 0, d_ptr, d_step);

    idx = <%=func_name%>((double*)d_ptr,d_step/sizeof(double),n);

    INIT_PTR(lp, 1, i_ptr, i_step);
    o_ptr = NDL_PTR(lp,2);
    *(idx_t*)o_ptr = *(idx_t*)(i_ptr + i_step * idx);
}
#undef idx_t
<% end %>

/*
 <%=desc%>
  @overload <%=name%>() => Integer
  @overload <%=name%>(axis:nil, keepdims:false) => Integer or Numo::Int32/64
  <% desc_param.each do |x|%>
  <%=x%><% end %>
  @param [Numeric,Array,Range] axis (keyword) Axes along which the operation is performed.
  @param [TrueClass] keepdims (keyword) If true, the reduced axes are left in th*/
static VALUE
<%=c_func%>(int argc, VALUE *argv, VALUE mod)
{
    narray_t *na;
    VALUE idx, reduce;
    ndfunc_arg_in_t ain[3] = {{cDF,0},{Qnil,0},{sym_reduce,0}};
    ndfunc_arg_out_t aout[1] = {{0,0,0}};
    ndfunc_t ndf = {0, STRIDE_LOOP_NIP|NDF_FLAT_REDUCE|NDF_EXTRACT, 3,1, ain,aout};

    if (argc<1) {
        rb_raise(rb_eArgError,"wrong number of argument (%d for >=1)",argc);
    }
    GetNArray(argv[0],na);
    if (na->ndim==0) {
        return INT2FIX(0);
    }
    if (na->size > (~(u_int32_t)0)) {
        aout[0].type = numo_cInt64;
        idx = rb_narray_new(numo_cInt64, na->ndim, na->shape);
        ndf.func = iter_<%=c_func%>_index64;
    } else {
        aout[0].type = numo_cInt32;
        idx = rb_narray_new(numo_cInt32, na->ndim, na->shape);
        ndf.func = iter_<%=c_func%>_index32;
    }
    rb_funcall(idx, rb_intern("seq"), 0);

    reduce = nary_reduce_dimension(argc-1, argv+1, 1, argv, &ndf, 0);

    return na_ndloop(&ndf, 3, argv[0], idx, reduce);
}

Version data entries

3 entries across 3 versions & 1 rubygems

Version Path
numo-gsl-0.1.2 ext/numo/gsl/stats/tmpl/stats_1a0p_index.c
numo-gsl-0.1.1 ext/numo/gsl/stats/tmpl/stats_1a0p_index.c
numo-gsl-0.1.0 ext/numo/gsl/stats/tmpl/stats_1a0p_index.c