Chris@42: /* Chris@42: * Copyright (c) 2003, 2007-14 Matteo Frigo Chris@42: * Copyright (c) 2003, 2007-14 Massachusetts Institute of Technology Chris@42: * Chris@42: * This program is free software; you can redistribute it and/or modify Chris@42: * it under the terms of the GNU General Public License as published by Chris@42: * the Free Software Foundation; either version 2 of the License, or Chris@42: * (at your option) any later version. Chris@42: * Chris@42: * This program is distributed in the hope that it will be useful, Chris@42: * but WITHOUT ANY WARRANTY; without even the implied warranty of Chris@42: * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the Chris@42: * GNU General Public License for more details. Chris@42: * Chris@42: * You should have received a copy of the GNU General Public License Chris@42: * along with this program; if not, write to the Free Software Chris@42: * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA Chris@42: * Chris@42: */ Chris@42: Chris@42: Chris@42: #include "dft.h" Chris@42: Chris@42: /* fill a complex array with zeros. */ Chris@42: static void recur(const iodim *dims, int rnk, R *ri, R *ii) Chris@42: { Chris@42: if (rnk == RNK_MINFTY) Chris@42: return; Chris@42: else if (rnk == 0) Chris@42: ri[0] = ii[0] = K(0.0); Chris@42: else if (rnk > 0) { Chris@42: INT i, n = dims[0].n; Chris@42: INT is = dims[0].is; Chris@42: Chris@42: if (rnk == 1) { Chris@42: /* this case is redundant but faster */ Chris@42: for (i = 0; i < n; ++i) Chris@42: ri[i * is] = ii[i * is] = K(0.0); Chris@42: } else { Chris@42: for (i = 0; i < n; ++i) Chris@42: recur(dims + 1, rnk - 1, ri + i * is, ii + i * is); Chris@42: } Chris@42: } Chris@42: } Chris@42: Chris@42: Chris@42: void X(dft_zerotens)(tensor *sz, R *ri, R *ii) Chris@42: { Chris@42: recur(sz->dims, sz->rnk, ri, ii); Chris@42: }