add sub dimension

dimension_t/src/dimension_t/dimension_t.c

@@ -10,7 +10,7 @@ long int incr(long int i) { return i + 1; }
 long int decr(long int i) { return i - 1; }
 
 dimension* init_dim(size_t *t, size_t sz){
-  dimension *d = INIT_PERMUTATION_TYPE_SIZE_T(t,sz);
+  dimension *d = INIT_COPY_PERMUTATION_TYPE_SIZE_T(t,sz);
   updateRankDim(d);
   return d;
 }
@@ -19,6 +19,22 @@ create_dim(size_t sz){
   return CREATE_PERMUTATION_TYPE_SIZE_T(sz);
 }
 
+dimension* sub_dim_head(dimension *root, size_t minusSubdim){
+  if(minusSubdim < (root->size)){
+    dimension *d = INIT_PERMUTATION_TYPE_SIZE_T(root->perm, (root->size)-minusSubdim);
+    updateRankDim(d);
+    return d;
+  }
+  return NULL;
+}
+dimension* sub_dim_tail(dimension *root, size_t minusSubdim){
+  if(minusSubdim < (root->size)){
+    dimension *d = INIT_PERMUTATION_TYPE_SIZE_T((root->perm)+minusSubdim, (root->size)-minusSubdim);
+    updateRankDim(d);
+    return d;
+  }
+  return NULL;
+}
 void add_dimension(dimension **d, dimension *d0, dimension *d1) {
     (*d) = create_dim(d0->size + d1->size);
     //printf("d size : %ld\n",(*d)->size);

dimension_t/src/dimension_t/dimension_t.h

@@ -10,6 +10,9 @@ typedef struct PERMUTATION_TYPE_SIZE_T dimension ;
 dimension * create_dim(size_t size);
 dimension* init_dim(size_t *t, size_t sz);
 
+dimension* sub_dim_head(dimension *t, size_t minusSubdim);
+dimension* sub_dim_tail(dimension *t, size_t minusSubdim);
+
 void add_dimension(dimension **d, dimension *d0, dimension *d1);
 void min_dimension(dimension **d, dimension *d0, dimension *d1);
 

dimension_t/test/is_good.c

@@ -34,6 +34,23 @@ TEST(rank){
   EXPECT_EQ(D->rank, 180);
 
 }
+TEST(SubDim){
+  dimension *D=create_dim(4);
+  D->perm[0]=2;
+  D->perm[1]=3;
+  D->perm[2]=5;
+  D->perm[3]=6;
+
+  dimension *d_head2 = sub_dim_head(D,2);
+  
+  
+  EXPECT_EQ(d_head2->rank, 2*3);
+
+  dimension *d_tail2 = sub_dim_tail(D,2);
+  EXPECT_EQ(d_tail2->rank, 5*6);
+
+}
+
 
 TEST(Coord_linear){
   dimension *D=create_dim(4);