1 /*
2 * Copyright 2012 Ecole Normale Superieure
3 *
4 * Use of this software is governed by the MIT license
5 *
6 * Written by Sven Verdoolaege,
7 * Ecole Normale Superieure, 45 rue d'Ulm, 75230 Paris, France
8 */
9
10 #include <isl/ctx.h>
11 #include <isl/space.h>
12
13 #include <isl_multi_macro.h>
14
15 /* Given two multi expressions, "multi1"
16 *
17 * [A1 A2] -> [B1 B2]
18 *
19 * where A2 starts at position "in_pos" and B2 starts at position "out_pos",
20 * and "multi2"
21 *
22 * [C] -> [D]
23 *
24 * return the multi expression
25 *
26 * [A1 C A2] -> [B1 D B2]
27 *
28 * We first insert input dimensions to obtain
29 *
30 * [A1 C A2] -> [B1 B2]
31 *
32 * and
33 *
34 * [A1 C A2] -> [D]
35 *
36 * and then apply range_splice.
37 */
MULTI(BASE)38 __isl_give MULTI(BASE) *FN(MULTI(BASE),splice)(
39 __isl_take MULTI(BASE) *multi1, unsigned in_pos, unsigned out_pos,
40 __isl_take MULTI(BASE) *multi2)
41 {
42 isl_size n_in1;
43 isl_size n_in2;
44
45 n_in1 = FN(MULTI(BASE),dim)(multi1, isl_dim_in);
46 n_in2 = FN(MULTI(BASE),dim)(multi2, isl_dim_in);
47 if (n_in1 < 0 || n_in2 < 0)
48 goto error;
49
50 if (FN(MULTI(BASE),check_range)(multi1, isl_dim_in, in_pos, 0) < 0)
51 goto error;
52
53 multi1 = FN(MULTI(BASE),insert_dims)(multi1, isl_dim_in, in_pos, n_in2);
54 multi2 = FN(MULTI(BASE),insert_dims)(multi2, isl_dim_in, n_in2,
55 n_in1 - in_pos);
56 multi2 = FN(MULTI(BASE),insert_dims)(multi2, isl_dim_in, 0, in_pos);
57
58 return FN(MULTI(BASE),range_splice)(multi1, out_pos, multi2);
59 error:
60 FN(MULTI(BASE),free)(multi1);
61 FN(MULTI(BASE),free)(multi2);
62 return NULL;
63 }
64