Home | History | Annotate | Line # | Download | only in dist 
      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