1 /*
2 * Copyright (c) 2003, 2007-14 Matteo Frigo
3 * Copyright (c) 2003, 2007-14 Massachusetts Institute of Technology
4 *
5 * This program is free software; you can redistribute it and/or modify
6 * it under the terms of the GNU General Public License as published by
7 * the Free Software Foundation; either version 2 of the License, or
8 * (at your option) any later version.
9 *
10 * This program is distributed in the hope that it will be useful,
11 * but WITHOUT ANY WARRANTY; without even the implied warranty of
12 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
13 * GNU General Public License for more details.
14 *
15 * You should have received a copy of the GNU General Public License
16 * along with this program; if not, write to the Free Software
17 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
18 *
19 */
20
21
22 #include "kernel/ifftw.h"
23
dimcpy(iodim * dst,const iodim * src,int rnk)24 static void dimcpy(iodim *dst, const iodim *src, int rnk)
25 {
26 int i;
27 if (FINITE_RNK(rnk))
28 for (i = 0; i < rnk; ++i)
29 dst[i] = src[i];
30 }
31
X(tensor_copy)32 tensor *X(tensor_copy)(const tensor *sz)
33 {
34 tensor *x = X(mktensor)(sz->rnk);
35 dimcpy(x->dims, sz->dims, sz->rnk);
36 return x;
37 }
38
39 /* like X(tensor_copy), but makes strides in-place by
40 setting os = is if k == INPLACE_IS or is = os if k == INPLACE_OS. */
X(tensor_copy_inplace)41 tensor *X(tensor_copy_inplace)(const tensor *sz, inplace_kind k)
42 {
43 tensor *x = X(tensor_copy)(sz);
44 if (FINITE_RNK(x->rnk)) {
45 int i;
46 if (k == INPLACE_OS)
47 for (i = 0; i < x->rnk; ++i)
48 x->dims[i].is = x->dims[i].os;
49 else
50 for (i = 0; i < x->rnk; ++i)
51 x->dims[i].os = x->dims[i].is;
52 }
53 return x;
54 }
55
56 /* Like X(tensor_copy), but copy all of the dimensions *except*
57 except_dim. */
X(tensor_copy_except)58 tensor *X(tensor_copy_except)(const tensor *sz, int except_dim)
59 {
60 tensor *x;
61
62 A(FINITE_RNK(sz->rnk) && sz->rnk >= 1 && except_dim < sz->rnk);
63 x = X(mktensor)(sz->rnk - 1);
64 dimcpy(x->dims, sz->dims, except_dim);
65 dimcpy(x->dims + except_dim, sz->dims + except_dim + 1,
66 x->rnk - except_dim);
67 return x;
68 }
69
70 /* Like X(tensor_copy), but copy only rnk dimensions starting
71 with start_dim. */
X(tensor_copy_sub)72 tensor *X(tensor_copy_sub)(const tensor *sz, int start_dim, int rnk)
73 {
74 tensor *x;
75
76 A(FINITE_RNK(sz->rnk) && start_dim + rnk <= sz->rnk);
77 x = X(mktensor)(rnk);
78 dimcpy(x->dims, sz->dims + start_dim, rnk);
79 return x;
80 }
81
X(tensor_append)82 tensor *X(tensor_append)(const tensor *a, const tensor *b)
83 {
84 if (!FINITE_RNK(a->rnk) || !FINITE_RNK(b->rnk)) {
85 return X(mktensor)(RNK_MINFTY);
86 } else {
87 tensor *x = X(mktensor)(a->rnk + b->rnk);
88 dimcpy(x->dims, a->dims, a->rnk);
89 dimcpy(x->dims + a->rnk, b->dims, b->rnk);
90 return x;
91 }
92 }
93