prefer compilers earlier in list
[bpt/guile.git] / libguile / array-map.c
index b181068..938f0a7 100644 (file)
@@ -1,6 +1,6 @@
 /* Copyright (C) 1996, 1998, 2000, 2001, 2004, 2005, 2006, 2008, 2009,
  *   2010, 2011, 2012, 2013, 2014 Free Software Foundation, Inc.
- * 
+ *
  * This library is free software; you can redistribute it and/or
  * modify it under the terms of the GNU Lesser General Public License
  * as published by the Free Software Foundation; either version 3 of
@@ -63,11 +63,11 @@ static SCM
 make1array (SCM v, ssize_t inc)
 {
   SCM a = scm_i_make_array (1);
-  SCM_I_ARRAY_BASE (a) = 0;
+  SCM_I_ARRAY_SET_BASE (a, 0);
   SCM_I_ARRAY_DIMS (a)->lbnd = 0;
   SCM_I_ARRAY_DIMS (a)->ubnd = scm_c_array_length (v) - 1;
   SCM_I_ARRAY_DIMS (a)->inc = inc;
-  SCM_I_ARRAY_V (a) = v;
+  SCM_I_ARRAY_SET_V (a, v);
   return a;
 }
 
@@ -102,74 +102,58 @@ cindk (SCM ra, ssize_t *ve, int kend)
 int
 scm_ramapc (void *cproc_ptr, SCM data, SCM ra0, SCM lra, const char *what)
 {
-  SCM z, vra0, lvra, *plvra;
-  ssize_t *vi;
-  int k, kmax, kroll;
   int (*cproc) () = cproc_ptr;
-  int empty = 0;
+  SCM z, va0, lva, *plva;
+  int k, kmax, kroll;
+  ssize_t *vi, inc;
+  size_t len;
 
   /* Prepare reference argument. */
   if (SCM_I_ARRAYP (ra0))
     {
-      k = kmax = SCM_I_ARRAY_NDIM (ra0)-1;
-      vra0 = make1array (SCM_I_ARRAY_V (ra0), SCM_I_ARRAY_DIMS (ra0)[kmax].inc);
+      kmax = SCM_I_ARRAY_NDIM (ra0)-1;
+      inc = kmax < 0 ?  0 : SCM_I_ARRAY_DIMS (ra0)[kmax].inc;
+      va0 = make1array (SCM_I_ARRAY_V (ra0), inc);
 
       /* Find unroll depth */
-      if (k > 0)
+      for (kroll = max(0, kmax); kroll > 0; --kroll)
         {
-          ssize_t inc = SCM_I_ARRAY_DIMS (ra0)[k].inc;
-          do {
-            ssize_t dim = (UBND (ra0, k) - LBND (ra0, k) + 1);
-            empty = empty || (0 == dim);
-            inc *= dim;
-            --k;
-          } while (k >= 0 && inc == SCM_I_ARRAY_DIMS (ra0)[k].inc);
-          kroll = k+1;
+          inc *= (UBND (ra0, kroll) - LBND (ra0, kroll) + 1);
+          if (inc != SCM_I_ARRAY_DIMS (ra0)[kroll-1].inc)
+            break;
         }
-      else
-        kroll = 0;
-
-      /* Check emptiness of not-unrolled axes. */
-      for (; k>=0; --k)
-        empty = empty || (0 == (UBND (ra0, k) - LBND (ra0, k) + 1));
     }
   else
     {
       kroll = kmax = 0;
-      vra0 = ra0 = make1array (ra0, 1);
-      empty = (0 == (UBND (ra0, 0) - LBND (ra0, 0) + 1));
+      va0 = ra0 = make1array (ra0, 1);
     }
 
   /* Prepare rest arguments. */
-  lvra = SCM_EOL;
-  plvra = &lvra;
+  lva = SCM_EOL;
+  plva = &lva;
   for (z = lra; !scm_is_null (z); z = SCM_CDR (z))
     {
-      SCM ra1 = SCM_CAR (z);
-      SCM vra1;
+      SCM va1, ra1 = SCM_CAR (z);
       if (SCM_I_ARRAYP (ra1))
         {
           if (kmax != SCM_I_ARRAY_NDIM (ra1) - 1)
             scm_misc_error (what, "array shape mismatch: ~S", scm_list_1 (ra0));
-          vra1 = make1array (SCM_I_ARRAY_V (ra1), SCM_I_ARRAY_DIMS (ra1)[kmax].inc);
+          inc = kmax < 0 ? 0 : SCM_I_ARRAY_DIMS (ra1)[kmax].inc;
+          va1 = make1array (SCM_I_ARRAY_V (ra1), inc);
 
           /* Check unroll depth. */
-          k = kmax;
-          if (k > kroll)
+          for (k = kmax; k > kroll; --k)
             {
-              ssize_t inc = SCM_I_ARRAY_DIMS (ra1)[k].inc;
-              do {
-                ssize_t l0 = LBND (ra0, k), u0 = UBND (ra0, k);
-                ssize_t l1 = LBND (ra1, k), u1 = UBND (ra1, k);
-                --k;
-                if (l0 == l1 && u0 == u1)
-                  inc *= (u1 - l1 + 1);
-                else if (l0 >= l1 && u0 <= u1)
+              ssize_t l0 = LBND (ra0, k), u0 = UBND (ra0, k);
+              if (l0 < LBND (ra1, k) || u0 > UBND (ra1, k))
+                scm_misc_error (what, "array shape mismatch: ~S", scm_list_1 (ra0));
+              inc *= (u0 - l0 + 1);
+              if (inc != SCM_I_ARRAY_DIMS (ra1)[k-1].inc)
+                {
+                  kroll = k;
                   break;
-                else
-                  scm_misc_error (what, "array shape mismatch: ~S", scm_list_1 (ra0));
-              } while (k >= kroll && inc == SCM_I_ARRAY_DIMS (ra1)[k].inc);
-              kroll = k + 1;
+                }
             }
 
           /* Check matching of not-unrolled axes. */
@@ -181,27 +165,26 @@ scm_ramapc (void *cproc_ptr, SCM data, SCM ra0, SCM lra, const char *what)
         {
           if (kmax != 0)
             scm_misc_error (what, "array shape mismatch: ~S", scm_list_1 (ra0));
-          vra1 = make1array (ra1, 1);
+          va1 = make1array (ra1, 1);
 
-          if (LBND (ra0, 0) < LBND (vra1, 0) || UBND (ra0, 0) > UBND (vra1, 0))
+          if (LBND (ra0, 0) < LBND (va1, 0) || UBND (ra0, 0) > UBND (va1, 0))
             scm_misc_error (what, "array shape mismatch: ~S", scm_list_1 (ra0));
         }
-      *plvra = scm_cons (vra1, SCM_EOL);
-      plvra = SCM_CDRLOC (*plvra);
+      *plva = scm_cons (va1, SCM_EOL);
+      plva = SCM_CDRLOC (*plva);
     }
 
+  /* Check emptiness of not-unrolled axes. */
+  for (k = 0; k < kroll; ++k)
+    if (0 == (UBND (ra0, k) - LBND (ra0, k) + 1))
+      return 1;
+
   /* Set unrolled size. */
-  if (empty)
-    return 1;
-  else
-    {
-      size_t len = 1;
-      for (k = kroll; k <= kmax; ++k)
-        len *= (UBND (ra0, k) - LBND (ra0, k) + 1);
-      UBND (vra0, 0) = len - 1;
-      for (z = lvra; !scm_is_null (z); z = SCM_CDR (z))
-        UBND (SCM_CAR (z), 0) = len - 1;
-    }
+  for (len = 1; k <= kmax; ++k)
+    len *= (UBND (ra0, k) - LBND (ra0, k) + 1);
+  UBND (va0, 0) = len - 1;
+  for (z = lva; !scm_is_null (z); z = SCM_CDR (z))
+    UBND (SCM_CAR (z), 0) = len - 1;
 
   /* Set starting indices and go. */
   vi = scm_gc_malloc_pointerless (sizeof(ssize_t) * kroll, vi_gc_hint);
@@ -212,24 +195,22 @@ scm_ramapc (void *cproc_ptr, SCM data, SCM ra0, SCM lra, const char *what)
       if (k == kroll)
         {
           SCM y = lra;
-          SCM_I_ARRAY_BASE (vra0) = cindk (ra0, vi, kroll);
-          for (z = lvra; !scm_is_null (z); z = SCM_CDR (z), y = SCM_CDR (y))
-            SCM_I_ARRAY_BASE (SCM_CAR (z)) = cindk (SCM_CAR (y), vi, kroll);
-          if (SCM_UNBNDP (data))
-            cproc (vra0, lvra);
-          else
-            cproc (vra0, data, lvra);
-          k--;
+          SCM_I_ARRAY_SET_BASE (va0, cindk (ra0, vi, kroll));
+          for (z = lva; !scm_is_null (z); z = SCM_CDR (z), y = SCM_CDR (y))
+            SCM_I_ARRAY_SET_BASE (SCM_CAR (z), cindk (SCM_CAR (y), vi, kroll));
+          if (! (SCM_UNBNDP (data) ? cproc (va0, lva) : cproc (va0, data, lva)))
+            return 0;
+          --k;
         }
-      else if (vi[k] < SCM_I_ARRAY_DIMS (ra0)[k].ubnd)
+      else if (vi[k] < UBND (ra0, k))
         {
-          vi[k]++;
-          k++;
+          ++vi[k];
+          ++k;
         }
       else
         {
-          vi[k] = SCM_I_ARRAY_DIMS (ra0)[k].lbnd - 1;
-          k--;
+          vi[k] = LBND (ra0, k) - 1;
+          --k;
         }
     }
   while (k >= 0);
@@ -593,12 +574,12 @@ ramap (SCM ra0, SCM proc, SCM ras)
   scm_t_array_handle h0;
   size_t n, i0;
   ssize_t i, inc0;
-  scm_array_get_handle (SCM_I_ARRAY_V (ra0), &h0);
   i0 = SCM_I_ARRAY_BASE (ra0);
   inc0 = SCM_I_ARRAY_DIMS (ra0)->inc;
   i = SCM_I_ARRAY_DIMS (ra0)->lbnd;
   n = SCM_I_ARRAY_DIMS (ra0)->ubnd - i + 1;
   ra0 = SCM_I_ARRAY_V (ra0);
+  scm_array_get_handle (ra0, &h0);
   if (scm_is_null (ras))
     for (; n--; i0 += inc0)
       h0.vset (h0.vector, i0, scm_call_0 (proc));
@@ -608,11 +589,11 @@ ramap (SCM ra0, SCM proc, SCM ras)
       scm_t_array_handle h1;
       size_t i1;
       ssize_t inc1;
-      scm_array_get_handle (SCM_I_ARRAY_V (ra1), &h1);
       i1 = SCM_I_ARRAY_BASE (ra1);
       inc1 = SCM_I_ARRAY_DIMS (ra1)->inc;
       ras = SCM_CDR (ras);
       ra1 = SCM_I_ARRAY_V (ra1);
+      scm_array_get_handle (ra1, &h1);
       if (scm_is_null (ras))
         for (; n--; i0 += inc0, i1 += inc1)
           h0.vset (h0.vector, i0, scm_call_1 (proc, h1.vref (h1.vector, i1)));
@@ -670,9 +651,10 @@ rafe (SCM ra0, SCM proc, SCM ras)
   scm_t_array_handle h0;
   size_t i0;
   ssize_t inc0;
-  scm_array_get_handle (SCM_I_ARRAY_V (ra0), &h0);
   i0 = SCM_I_ARRAY_BASE (ra0);
   inc0 = SCM_I_ARRAY_DIMS (ra0)->inc;
+  ra0 = SCM_I_ARRAY_V (ra0);
+  scm_array_get_handle (ra0, &h0);
   if (scm_is_null (ras))
     for (; n--; i0 += inc0)
       scm_call_1 (proc, h0.vref (h0.vector, i0));
@@ -723,45 +705,46 @@ array_index_map_1 (SCM ra, SCM proc)
 static void
 array_index_map_n (SCM ra, SCM proc)
 {
+  scm_t_array_handle h;
   size_t i;
-  int j, k, kmax = SCM_I_ARRAY_NDIM (ra) - 1;
+  int k, kmax = SCM_I_ARRAY_NDIM (ra) - 1;
   ssize_t *vi;
+  SCM **si;
+  SCM args = SCM_EOL;
+  SCM *p = &args;
 
   vi = scm_gc_malloc_pointerless (sizeof(ssize_t) * (kmax + 1), vi_gc_hint);
+  si = scm_gc_malloc_pointerless (sizeof(SCM *) * (kmax + 1), vi_gc_hint);
 
   for (k = 0; k <= kmax; k++)
     {
       vi[k] = SCM_I_ARRAY_DIMS (ra)[k].lbnd;
       if (vi[k] > SCM_I_ARRAY_DIMS (ra)[k].ubnd)
         return;
+      *p = scm_cons (scm_from_ssize_t (vi[k]), SCM_EOL);
+      si[k] = SCM_CARLOC (*p);
+      p = SCM_CDRLOC (*p);
     }
 
+  scm_array_get_handle (ra, &h);
   k = kmax;
   do
     {
       if (k == kmax)
         {
-          SCM args = SCM_EOL;
-          SCM *p = &args, *q;
           vi[kmax] = SCM_I_ARRAY_DIMS (ra)[kmax].lbnd;
           i = cindk (ra, vi, kmax+1);
-          for (j = 0; j<=kmax; ++j)
+          for (; vi[kmax] <= SCM_I_ARRAY_DIMS (ra)[kmax].ubnd; ++vi[kmax])
             {
-              *p = scm_cons (scm_from_ssize_t (vi[j]), SCM_EOL);
-              q = SCM_CARLOC (*p);
-              p = SCM_CDRLOC (*p);
-            }
-          for (; vi[kmax] <= SCM_I_ARRAY_DIMS (ra)[kmax].ubnd;
-               *q = scm_from_ssize_t (++vi[kmax]))
-            {
-              ASET (SCM_I_ARRAY_V (ra), i, scm_apply_0 (proc, args));
+              *(si[kmax]) = scm_from_ssize_t (vi[kmax]);
+              h.vset (h.vector, i, scm_apply_0 (proc, args));
               i += SCM_I_ARRAY_DIMS (ra)[kmax].inc;
             }
           k--;
         }
       else if (vi[k] < SCM_I_ARRAY_DIMS (ra)[k].ubnd)
         {
-          vi[k]++;
+          *(si[k]) = scm_from_ssize_t (++vi[k]);
           k++;
         }
       else
@@ -771,6 +754,7 @@ array_index_map_n (SCM ra, SCM proc)
         }
     }
   while (k >= 0);
+  scm_array_handle_release (&h);
 }
 
 SCM_DEFINE (scm_array_index_map_x, "array-index-map!", 2, 0, 0,
@@ -831,7 +815,7 @@ array_compare (scm_t_array_handle *hx, scm_t_array_handle *hy,
         return 0;
 
       i = hx->dims[dim].ubnd - hx->dims[dim].lbnd + 1;
-      
+
       incx = hx->dims[dim].inc;
       incy = hy->dims[dim].inc;
       posx += (i - 1) * incx;
@@ -848,11 +832,11 @@ SCM
 scm_array_equal_p (SCM x, SCM y)
 {
   scm_t_array_handle hx, hy;
-  SCM res;  
-  
+  SCM res;
+
   scm_array_get_handle (x, &hx);
   scm_array_get_handle (y, &hy);
-  
+
   res = scm_from_bool (hx.ndims == hy.ndims
                        && hx.element_type == hy.element_type);
 
@@ -876,7 +860,7 @@ SCM_DEFINE (scm_i_array_equal_p, "array-equal?", 0, 2, 1,
 {
   if (SCM_UNBNDP (ra0) || SCM_UNBNDP (ra1))
     return SCM_BOOL_T;
-  
+
   while (!scm_is_null (rest))
     { if (scm_is_false (scm_array_equal_p (ra0, ra1)))
         return SCM_BOOL_F;