long wavelet_coeffs(unsigned int N, unsigned int flags, const long dims[N], const long min[N], const long flen)
{
unsigned int levels = wavelet_num_levels(N, flags, dims, min, flen);
assert(levels > 0);
return wavelet_coeffs_r(levels - 1, N, flags, dims, min, flen);
}
static long wavelet_coeffs_r(unsigned int levels, unsigned int N, unsigned int flags, const long dims[N], const long min[N], const long flen)
{
long wdims[2 * N];
wavelet_dims(N, flags, wdims, dims, flen);
long coeffs = md_calc_size(N, wdims);
long bands = md_calc_size(N, wdims + N);
assert((0 == flags) == (0 == levels));
if (0 == flags)
return bands * coeffs;
return coeffs * (bands - 1) + wavelet_coeffs_r(levels - 1, N, wavelet_filter_flags(N, flags, wdims, min), wdims, min, flen);
}
void wavelet_coeffs2(unsigned int N, unsigned int flags, long odims[N], const long dims[N], const long min[N], const long flen)
{
md_select_dims(N, ~flags, odims, dims);
if (0 == flags)
return;
unsigned int levels = wavelet_num_levels(N, flags, dims, min, flen);
assert(levels > 0);
long wdims[N];
md_select_dims(N, flags, wdims, dims); // remove unmodified dims
unsigned int b = ffs(flags) - 1;
odims[b] = wavelet_coeffs_r(levels - 1, N, flags, wdims, min, flen);
}
