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 */ 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