Sha256: d71ded74127066b588f84eac3833df9f97baca519d75eec22030d09405e1a52e
Contents?: true
Size: 1.75 KB
Versions: 7
Compression:
Stored size: 1.75 KB
Contents
<% (is_float ? ["_ignan","_prnan"] : [""]).each do |j| %>
static void
<%=c_iter%><%=j%>(na_loop_t *const lp)
{
size_t n;
char *p1, *p2;
dtype *buf;
INIT_COUNTER(lp, n);
p1 = (lp->args[0]).ptr + (lp->args[0].iter[0]).pos;
p2 = (lp->args[1]).ptr + (lp->args[1].iter[0]).pos;
buf = (dtype*)p1;
<%=type_name%>_qsort<%=j%>(buf, n, sizeof(dtype));
<% if is_float %>
for (; n; n--) {
if (!isnan(buf[n-1])) break;
}
<% end %>
if (n==0) {
*(dtype*)p2 = buf[0];
}
else if (n%2==0) {
*(dtype*)p2 = (buf[n/2-1]+buf[n/2])/2;
}
else {
*(dtype*)p2 = buf[(n-1)/2];
}
}
<% end %>
/*
<%=name%> of self.
<% if is_float %>
@overload <%=name%>(axis:nil, keepdims:false, nan:false)
@param [TrueClass] nan (keyword) If true, propagete NaN. If false, ignore NaN.
<% else %>
@overload <%=name%>(axis:nil, keepdims:false)
<% end %>
@param [Numeric,Array,Range] axis (keyword) Affected dimensions.
@param [TrueClass] keepdims (keyword) If true, the reduced axes are left in the result array as dimensions with size one.
@return [Numo::<%=class_name%>] returns <%=name%> of self.
*/
static VALUE
<%=c_func(-1)%>(int argc, VALUE *argv, VALUE self)
{
VALUE reduce;
ndfunc_arg_in_t ain[2] = {{OVERWRITE,0},{sym_reduce,0}};
ndfunc_arg_out_t aout[1] = {{INT2FIX(0),0}};
ndfunc_t ndf = {0, NDF_HAS_LOOP|NDF_FLAT_REDUCE, 2,1, ain,aout};
self = na_copy(self); // as temporary buffer
<% if is_float %>
ndf.func = <%=c_iter%>_ignan;
reduce = na_reduce_dimension(argc, argv, 1, &self, &ndf, <%=c_iter%>_prnan);
<% else %>
ndf.func = <%=c_iter%>;
reduce = na_reduce_dimension(argc, argv, 1, &self, &ndf, 0);
<% end %>
return na_ndloop(&ndf, 2, self, reduce);
}
Version data entries
7 entries across 7 versions & 1 rubygems