Sha256: 7f97c3b4d7978dadaa2920ce56dbe74656be0893c1eb9f855f14f02451e26e68
Contents?: true
Size: 1.32 KB
Versions: 3
Compression:
Stored size: 1.32 KB
Contents
<%
n = case name
when /_xrange/; "nx"
when /_yrange/; "ny"
when /_range/; "n"
end
%>
static void
iter_<%=c_func%>(na_loop_t *const lp)
{
size_t n, i;
char *p1, *p2, *p3;
ssize_t s1, s2, s3;
ssize_t x1;
size_t y1;
double y2, y3;
<%=struct%> *w = (<%=struct%>*)(lp->opt_ptr);
INIT_COUNTER(lp, n);
INIT_PTR(lp, 0, p1, s1);
INIT_PTR(lp, 1, p2, s2);
INIT_PTR(lp, 2, p3, s3);
for (i=0; i<n; i++) {
GET_DATA_STRIDE(p1,s1,ssize_t,x1);
if (x1 < 0) {
y1 = x1 + w-><%=n%>;
} else {
y1 = x1;
}
<%=func_name%>(<%=get(:postpose) ? "y1,&y2,&y3,w" : "w,y1,&y2,&y3"%>);
SET_DATA_STRIDE(p2,s2,double,y2);
SET_DATA_STRIDE(p3,s3,double,y3);
}
}
/*
@overload <%=name%>(<%=args[1][1]%>)
@param [Int32/64] <%=args[1][1]%> Int32/64 NArray
@return [Array] array of [[DFloat] <%=args[2][1]%>, [DFloat] <%=args[3][1]%>]
<%= description %>
*/
static VALUE
<%=c_func(1)%>(VALUE self, VALUE v1)
{
<%=struct%> *w;
ndfunc_arg_in_t ain[1] = {{cSSZ,0}};
ndfunc_arg_out_t aout[2] = {{cDF,0},{cDF,0}};
ndfunc_t ndf = {iter_<%=c_func%>, STRIDE_LOOP|NDF_EXTRACT, 1,2, ain,aout};
TypedData_Get_Struct(self, <%=struct%>, &<%=data_type_var%>, w);
return na_ndloop3(&ndf, w, 1, v1);
}
Version data entries
3 entries across 3 versions & 1 rubygems