Chris@10: /* Chris@10: * Copyright (c) 2003, 2007-11 Matteo Frigo Chris@10: * Copyright (c) 2003, 2007-11 Massachusetts Institute of Technology Chris@10: * Chris@10: * This program is free software; you can redistribute it and/or modify Chris@10: * it under the terms of the GNU General Public License as published by Chris@10: * the Free Software Foundation; either version 2 of the License, or Chris@10: * (at your option) any later version. Chris@10: * Chris@10: * This program is distributed in the hope that it will be useful, Chris@10: * but WITHOUT ANY WARRANTY; without even the implied warranty of Chris@10: * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the Chris@10: * GNU General Public License for more details. Chris@10: * Chris@10: * You should have received a copy of the GNU General Public License Chris@10: * along with this program; if not, write to the Free Software Chris@10: * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA Chris@10: * Chris@10: */ Chris@10: Chris@10: Chris@10: #include "rdft.h" Chris@10: Chris@10: /* like X(tensor_max_index), but takes into account the special n/2+1 Chris@10: final dimension for the complex output/input of an R2HC/HC2R transform. */ Chris@10: INT X(rdft2_tensor_max_index)(const tensor *sz, rdft_kind k) Chris@10: { Chris@10: int i; Chris@10: INT n = 0; Chris@10: Chris@10: A(FINITE_RNK(sz->rnk)); Chris@10: for (i = 0; i + 1 < sz->rnk; ++i) { Chris@10: const iodim *p = sz->dims + i; Chris@10: n += (p->n - 1) * X(imax)(X(iabs)(p->is), X(iabs)(p->os)); Chris@10: } Chris@10: if (i < sz->rnk) { Chris@10: const iodim *p = sz->dims + i; Chris@10: INT is, os; Chris@10: X(rdft2_strides)(k, p, &is, &os); Chris@10: n += X(imax)((p->n - 1) * X(iabs)(is), (p->n/2) * X(iabs)(os)); Chris@10: } Chris@10: return n; Chris@10: }