* gcc.dg/const-elim-1.c: Remove xfail for xtensa-*-*.

[pf3gnuchains/gcc-fork.git] / libgfortran / intrinsics / spread_generic.c

diff --git a/libgfortran/intrinsics/spread_generic.c b/libgfortran/intrinsics/spread_generic.c
index e40739e..8d39d30 100644 (file)
--- a/libgfortran/intrinsics/spread_generic.c
+++ b/libgfortran/intrinsics/spread_generic.c
@@ -34,60 +34,112 @@ Boston, MA 02111-1307, USA.  */
 #include <string.h>
 #include "libgfortran.h"
 
-extern void spread (const gfc_array_char *, const gfc_array_char *,
+extern void spread (gfc_array_char *, const gfc_array_char *,
                    const index_type *, const index_type *);
 export_proto(spread);
 
 void
-spread (const gfc_array_char *ret, const gfc_array_char *source,
+spread (gfc_array_char *ret, const gfc_array_char *source,
        const index_type *along, const index_type *pncopies)
 {
   /* r.* indicates the return array.  */
-  index_type rstride[GFC_MAX_DIMENSIONS - 1];
+  index_type rstride[GFC_MAX_DIMENSIONS];
   index_type rstride0;
   index_type rdelta;
+  index_type rrank;
+  index_type rs;
   char *rptr;
   char *dest;
   /* s.* indicates the source array.  */
-  index_type sstride[GFC_MAX_DIMENSIONS - 1];
+  index_type sstride[GFC_MAX_DIMENSIONS];
   index_type sstride0;
+  index_type srank;
   const char *sptr;
 
-  index_type count[GFC_MAX_DIMENSIONS - 1];
-  index_type extent[GFC_MAX_DIMENSIONS - 1];
+  index_type count[GFC_MAX_DIMENSIONS];
+  index_type extent[GFC_MAX_DIMENSIONS];
   index_type n;
   index_type dim;
   index_type size;
   index_type ncopies;
 
+  srank = GFC_DESCRIPTOR_RANK(source);
+
+  rrank = srank + 1;
+  if (rrank > GFC_MAX_DIMENSIONS)
+    runtime_error ("return rank too large in spread()");
+
+  if (*along > rrank)
+      runtime_error ("dim outside of rank in spread()");
+
+  ncopies = *pncopies;
+
   size = GFC_DESCRIPTOR_SIZE (source);
-  dim = 0;
-  for (n = 0; n < GFC_DESCRIPTOR_RANK (ret); n++)
+  if (ret->data == NULL)
     {
-      if (n == *along - 1)
-        {
-          rdelta = ret->dim[n].stride * size;
-        }
-      else
-        {
-          count[dim] = 0;
-          extent[dim] = source->dim[dim].ubound + 1 - source->dim[dim].lbound;
-          sstride[dim] = source->dim[dim].stride * size;
-          rstride[dim] = ret->dim[n].stride * size;
-          dim++;
-        }
+      /* The front end has signalled that we need to populate the
+        return array descriptor.  */
+      ret->dtype = (source->dtype & ~GFC_DTYPE_RANK_MASK) | rrank;
+      dim = 0;
+      rs = 1;
+      for (n = 0; n < rrank; n++)
+       {
+         ret->dim[n].stride = rs;
+         ret->dim[n].lbound = 0;
+         if (n == *along - 1)
+           {
+             ret->dim[n].ubound = ncopies - 1;
+             rdelta = rs * size;
+             rs *= ncopies;
+           }
+         else
+           {
+             count[dim] = 0;
+             extent[dim] = source->dim[dim].ubound + 1
+               - source->dim[dim].lbound;
+             sstride[dim] = source->dim[dim].stride * size;
+             rstride[dim] = rs * size;
+
+             ret->dim[n].ubound = extent[dim]-1;
+             rs *= extent[dim];
+             dim++;
+           }
+       }
+      ret->base = 0;
+      ret->data = internal_malloc_size (rs * size);
     }
-  dim = GFC_DESCRIPTOR_RANK (source);
-  if (sstride[0] == 0)
-    sstride[0] = size;
-  if (rstride[0] == 0)
-    rstride[0] = size;
+  else
+    {
+      dim = 0;
+      if (GFC_DESCRIPTOR_RANK(ret) != rrank)
+       runtime_error ("rank mismatch in spread()");
 
+      if (ret->dim[0].stride == 0)
+       ret->dim[0].stride = 1;
+
+      for (n = 0; n < rrank; n++)
+       {
+         if (n == *along - 1)
+           {
+             rdelta = ret->dim[n].stride * size;
+           }
+         else
+           {
+             count[dim] = 0;
+             extent[dim] = source->dim[dim].ubound + 1
+               - source->dim[dim].lbound;
+             sstride[dim] = source->dim[dim].stride * size;
+             rstride[dim] = ret->dim[n].stride * size;
+             dim++;
+           }
+       }
+      if (sstride[0] == 0)
+       sstride[0] = size;
+    }
   sstride0 = sstride[0];
   rstride0 = rstride[0];
   rptr = ret->data;
   sptr = source->data;
-  ncopies = *pncopies;
 
   while (sptr)
     {
@@ -113,7 +165,7 @@ spread (const gfc_array_char *ret, const gfc_array_char *source,
           sptr -= sstride[n] * extent[n];
           rptr -= rstride[n] * extent[n];
           n++;
-          if (n >= dim)
+          if (n >= srank)
             {
               /* Break out of the loop.  */
               sptr = NULL;