]> git.ipfire.org Git - thirdparty/gcc.git/blobdiff - libgfortran/intrinsics/spread_generic.c
Update copyright years.
[thirdparty/gcc.git] / libgfortran / intrinsics / spread_generic.c
index 9fb4b11b2650581c8a8fe4f3a34003a941d1aed9..3a0c451ab356406f6083cb349ba4ea377a9edfc4 100644 (file)
@@ -1,8 +1,8 @@
 /* Generic implementation of the SPREAD intrinsic
-   Copyright 2002, 2005, 2006, 2007, 2009 Free Software Foundation, Inc.
+   Copyright (C) 2002-2024 Free Software Foundation, Inc.
    Contributed by Paul Brook <paul@nowt.org>
 
-This file is part of the GNU Fortran 95 runtime library (libgfortran).
+This file is part of the GNU Fortran runtime library (libgfortran).
 
 Libgfortran is free software; you can redistribute it and/or
 modify it under the terms of the GNU General Public
@@ -24,14 +24,11 @@ see the files COPYING3 and COPYING.RUNTIME respectively.  If not, see
 <http://www.gnu.org/licenses/>.  */
 
 #include "libgfortran.h"
-#include <stdlib.h>
-#include <assert.h>
 #include <string.h>
 
 static void
 spread_internal (gfc_array_char *ret, const gfc_array_char *source,
-                const index_type *along, const index_type *pncopies,
-                index_type size)
+                const index_type *along, const index_type *pncopies)
 {
   /* r.* indicates the return array.  */
   index_type rstride[GFC_MAX_DIMENSIONS];
@@ -52,9 +49,14 @@ spread_internal (gfc_array_char *ret, const gfc_array_char *source,
   index_type n;
   index_type dim;
   index_type ncopies;
+  index_type size;
+
+  size = GFC_DESCRIPTOR_SIZE(source);
 
   srank = GFC_DESCRIPTOR_RANK(source);
 
+  sstride[0] = 0; /* Avoid warnings if not initialized.  */
+
   rrank = srank + 1;
   if (rrank > GFC_MAX_DIMENSIONS)
     runtime_error ("return rank too large in spread()");
@@ -64,44 +66,45 @@ spread_internal (gfc_array_char *ret, const gfc_array_char *source,
 
   ncopies = *pncopies;
 
-  if (ret->data == NULL)
+  if (ret->base_addr == NULL)
     {
       /* The front end has signalled that we need to populate the
         return array descriptor.  */
-      ret->dtype = (source->dtype & ~GFC_DTYPE_RANK_MASK) | rrank;
+
+      size_t ub, stride;
+
+      ret->dtype.rank = rrank;
+
       dim = 0;
       rs = 1;
       for (n = 0; n < rrank; n++)
        {
-         ret->dim[n].stride = rs;
-         ret->dim[n].lbound = 0;
+         stride = rs;
          if (n == *along - 1)
            {
-             ret->dim[n].ubound = ncopies - 1;
+             ub = 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;
+             extent[dim] = GFC_DESCRIPTOR_EXTENT(source,dim);
+             sstride[dim] = GFC_DESCRIPTOR_STRIDE_BYTES(source,dim);
              rstride[dim] = rs * size;
 
-             ret->dim[n].ubound = extent[dim]-1;
+             ub = extent[dim]-1;
              rs *= extent[dim];
              dim++;
            }
+
+         GFC_DIMENSION_SET(ret->dim[n], 0, ub, stride);
        }
       ret->offset = 0;
-      if (rs > 0)
-        ret->data = internal_malloc_size (rs * size);
-      else
-       {
-         ret->data = internal_malloc_size (1);
-         return;
-       }
+      ret->base_addr = xmallocarray (rs, size);
+
+      if (rs <= 0)
+       return;
     }
   else
     {
@@ -119,10 +122,10 @@ spread_internal (gfc_array_char *ret, const gfc_array_char *source,
            {
              index_type ret_extent;
 
-             ret_extent = ret->dim[n].ubound + 1 - ret->dim[n].lbound;
+             ret_extent = GFC_DESCRIPTOR_EXTENT(ret,n);
              if (n == *along - 1)
                {
-                 rdelta = ret->dim[n].stride * size;
+                 rdelta = GFC_DESCRIPTOR_STRIDE_BYTES(ret,n);
 
                  if (ret_extent != ncopies)
                    runtime_error("Incorrect extent in return value of SPREAD"
@@ -133,8 +136,7 @@ spread_internal (gfc_array_char *ret, const gfc_array_char *source,
              else
                {
                  count[dim] = 0;
-                 extent[dim] = source->dim[dim].ubound + 1
-                   - source->dim[dim].lbound;
+                 extent[dim] = GFC_DESCRIPTOR_EXTENT(source,dim);
                  if (ret_extent != extent[dim])
                    runtime_error("Incorrect extent in return value of SPREAD"
                                  " intrinsic in dimension %ld: is %ld,"
@@ -144,8 +146,8 @@ spread_internal (gfc_array_char *ret, const gfc_array_char *source,
                    
                  if (extent[dim] <= 0)
                    zero_sized = 1;
-                 sstride[dim] = source->dim[dim].stride * size;
-                 rstride[dim] = ret->dim[n].stride * size;
+                 sstride[dim] = GFC_DESCRIPTOR_STRIDE_BYTES(source,dim);
+                 rstride[dim] = GFC_DESCRIPTOR_STRIDE_BYTES(ret,n);
                  dim++;
                }
            }
@@ -156,17 +158,16 @@ spread_internal (gfc_array_char *ret, const gfc_array_char *source,
            {
              if (n == *along - 1)
                {
-                 rdelta = ret->dim[n].stride * size;
+                 rdelta = GFC_DESCRIPTOR_STRIDE_BYTES(ret,n);
                }
              else
                {
                  count[dim] = 0;
-                 extent[dim] = source->dim[dim].ubound + 1
-                   - source->dim[dim].lbound;
+                 extent[dim] = GFC_DESCRIPTOR_EXTENT(source,dim);
                  if (extent[dim] <= 0)
                    zero_sized = 1;
-                 sstride[dim] = source->dim[dim].stride * size;
-                 rstride[dim] = ret->dim[n].stride * size;
+                 sstride[dim] = GFC_DESCRIPTOR_STRIDE_BYTES(source,dim);
+                 rstride[dim] = GFC_DESCRIPTOR_STRIDE_BYTES(ret,n);
                  dim++;
                }
            }
@@ -180,8 +181,8 @@ spread_internal (gfc_array_char *ret, const gfc_array_char *source,
     }
   sstride0 = sstride[0];
   rstride0 = rstride[0];
-  rptr = ret->data;
-  sptr = source->data;
+  rptr = ret->base_addr;
+  sptr = source->base_addr;
 
   while (sptr)
     {
@@ -228,12 +229,14 @@ spread_internal (gfc_array_char *ret, const gfc_array_char *source,
 
 static void
 spread_internal_scalar (gfc_array_char *ret, const char *source,
-                       const index_type *along, const index_type *pncopies,
-                       index_type size)
+                       const index_type *along, const index_type *pncopies)
 {
   int n;
   int ncopies = *pncopies;
   char * dest;
+  size_t size;
+
+  size = GFC_DESCRIPTOR_SIZE(ret);
 
   if (GFC_DESCRIPTOR_RANK (ret) != 1)
     runtime_error ("incorrect destination rank in spread()");
@@ -241,24 +244,22 @@ spread_internal_scalar (gfc_array_char *ret, const char *source,
   if (*along > 1)
     runtime_error ("dim outside of rank in spread()");
 
-  if (ret->data == NULL)
+  if (ret->base_addr == NULL)
     {
-      ret->data = internal_malloc_size (ncopies * size);
+      ret->base_addr = xmallocarray (ncopies, size);
       ret->offset = 0;
-      ret->dim[0].stride = 1;
-      ret->dim[0].lbound = 0;
-      ret->dim[0].ubound = ncopies - 1;
+      GFC_DIMENSION_SET(ret->dim[0], 0, ncopies - 1, 1);
     }
   else
     {
-      if (ncopies - 1 > (ret->dim[0].ubound - ret->dim[0].lbound)
-                          / ret->dim[0].stride)
+      if (ncopies - 1 > (GFC_DESCRIPTOR_EXTENT(ret,0)  - 1)
+                          / GFC_DESCRIPTOR_STRIDE(ret,0))
        runtime_error ("dim too large in spread()");
     }
 
   for (n = 0; n < ncopies; n++)
     {
-      dest = (char*)(ret->data + n*size*ret->dim[0].stride);
+      dest = (char*)(ret->base_addr + n * GFC_DESCRIPTOR_STRIDE_BYTES(ret,0));
       memcpy (dest , source, size);
     }
 }
@@ -276,7 +277,6 @@ spread (gfc_array_char *ret, const gfc_array_char *source,
   type_size = GFC_DTYPE_TYPE_SIZE(ret);
   switch(type_size)
     {
-    case GFC_DTYPE_DERIVED_1:
     case GFC_DTYPE_LOGICAL_1:
     case GFC_DTYPE_INTEGER_1:
       spread_i1 ((gfc_array_i1 *) ret, (gfc_array_i1 *) source,
@@ -319,18 +319,26 @@ spread (gfc_array_char *ret, const gfc_array_char *source,
                 *along, *pncopies);
       return;
 
-#ifdef GFC_HAVE_REAL_10
+/* FIXME: This here is a hack, which will have to be removed when
+   the array descriptor is reworked.  Currently, we don't store the
+   kind value for the type, but only the size.  Because on targets with
+   _Float128, we have sizeof(long double) == sizeof(_Float128),
+   we cannot discriminate here and have to fall back to the generic
+   handling (which is suboptimal).  */
+#if !defined(GFC_REAL_16_IS_FLOAT128)
+# ifdef GFC_HAVE_REAL_10
     case GFC_DTYPE_REAL_10:
       spread_r10 ((gfc_array_r10 *) ret, (gfc_array_r10 *) source,
                 *along, *pncopies);
       return;
-#endif
+# endif
 
-#ifdef GFC_HAVE_REAL_16
+# ifdef GFC_HAVE_REAL_16
     case GFC_DTYPE_REAL_16:
       spread_r16 ((gfc_array_r16 *) ret, (gfc_array_r16 *) source,
                 *along, *pncopies);
       return;
+# endif
 #endif
 
     case GFC_DTYPE_COMPLEX_4:
@@ -343,22 +351,39 @@ spread (gfc_array_char *ret, const gfc_array_char *source,
                 *along, *pncopies);
       return;
 
-#ifdef GFC_HAVE_COMPLEX_10
+/* FIXME: This here is a hack, which will have to be removed when
+   the array descriptor is reworked.  Currently, we don't store the
+   kind value for the type, but only the size.  Because on targets with
+   _Float128, we have sizeof(long double) == sizeof(_Float128),
+   we cannot discriminate here and have to fall back to the generic
+   handling (which is suboptimal).  */
+#if !defined(GFC_REAL_16_IS_FLOAT128)
+# ifdef GFC_HAVE_COMPLEX_10
     case GFC_DTYPE_COMPLEX_10:
       spread_c10 ((gfc_array_c10 *) ret, (gfc_array_c10 *) source,
                 *along, *pncopies);
       return;
-#endif
+# endif
 
-#ifdef GFC_HAVE_COMPLEX_16
+# ifdef GFC_HAVE_COMPLEX_16
     case GFC_DTYPE_COMPLEX_16:
       spread_c16 ((gfc_array_c16 *) ret, (gfc_array_c16 *) source,
                 *along, *pncopies);
       return;
+# endif
 #endif
 
-    case GFC_DTYPE_DERIVED_2:
-      if (GFC_UNALIGNED_2(ret->data) || GFC_UNALIGNED_2(source->data))
+    }
+  
+  switch (GFC_DESCRIPTOR_SIZE (ret))
+    {
+    case 1:
+      spread_i1 ((gfc_array_i1 *) ret, (gfc_array_i1 *) source,
+                *along, *pncopies);
+      return;
+
+    case 2:
+      if (GFC_UNALIGNED_2(ret->base_addr) || GFC_UNALIGNED_2(source->base_addr))
        break;
       else
        {
@@ -367,8 +392,8 @@ spread (gfc_array_char *ret, const gfc_array_char *source,
          return;
        }
 
-    case GFC_DTYPE_DERIVED_4:
-      if (GFC_UNALIGNED_4(ret->data) || GFC_UNALIGNED_4(source->data))
+    case 4:
+      if (GFC_UNALIGNED_4(ret->base_addr) || GFC_UNALIGNED_4(source->base_addr))
        break;
       else
        {
@@ -377,8 +402,8 @@ spread (gfc_array_char *ret, const gfc_array_char *source,
          return;
        }
 
-    case GFC_DTYPE_DERIVED_8:
-      if (GFC_UNALIGNED_8(ret->data) || GFC_UNALIGNED_8(source->data))
+    case 8:
+      if (GFC_UNALIGNED_8(ret->base_addr) || GFC_UNALIGNED_8(source->base_addr))
        break;
       else
        {
@@ -386,21 +411,22 @@ spread (gfc_array_char *ret, const gfc_array_char *source,
                     *along, *pncopies);
          return;
        }
-
 #ifdef HAVE_GFC_INTEGER_16
-    case GFC_DTYPE_DERIVED_16:
-      if (GFC_UNALIGNED_16(ret->data) || GFC_UNALIGNED_16(source->data))
+    case 16:
+      if (GFC_UNALIGNED_16(ret->base_addr)
+         || GFC_UNALIGNED_16(source->base_addr))
        break;
       else
        {
          spread_i16 ((gfc_array_i16 *) ret, (gfc_array_i16 *) source,
                      *along, *pncopies);
          return;
-       }
+           }
 #endif
+
     }
 
-  spread_internal (ret, source, along, pncopies, GFC_DESCRIPTOR_SIZE (source));
+  spread_internal (ret, source, along, pncopies);
 }
 
 
@@ -413,9 +439,10 @@ void
 spread_char (gfc_array_char *ret,
             GFC_INTEGER_4 ret_length __attribute__((unused)),
             const gfc_array_char *source, const index_type *along,
-            const index_type *pncopies, GFC_INTEGER_4 source_length)
+            const index_type *pncopies,
+            GFC_INTEGER_4 source_length __attribute__((unused)))
 {
-  spread_internal (ret, source, along, pncopies, source_length);
+  spread_internal (ret, source, along, pncopies);
 }
 
 
@@ -428,10 +455,10 @@ void
 spread_char4 (gfc_array_char *ret,
              GFC_INTEGER_4 ret_length __attribute__((unused)),
              const gfc_array_char *source, const index_type *along,
-             const index_type *pncopies, GFC_INTEGER_4 source_length)
+             const index_type *pncopies,
+             GFC_INTEGER_4 source_length __attribute__((unused)))
 {
-  spread_internal (ret, source, along, pncopies,
-                  source_length * sizeof (gfc_char4_t));
+  spread_internal (ret, source, along, pncopies);
 }
 
 
@@ -448,13 +475,12 @@ spread_scalar (gfc_array_char *ret, const char *source,
 {
   index_type type_size;
 
-  if (!ret->dtype)
+  if (GFC_DTYPE_IS_UNSET(ret))
     runtime_error ("return array missing descriptor in spread()");
 
   type_size = GFC_DTYPE_TYPE_SIZE(ret);
   switch(type_size)
     {
-    case GFC_DTYPE_DERIVED_1:
     case GFC_DTYPE_LOGICAL_1:
     case GFC_DTYPE_INTEGER_1:
       spread_scalar_i1 ((gfc_array_i1 *) ret, (GFC_INTEGER_1 *) source,
@@ -497,18 +523,26 @@ spread_scalar (gfc_array_char *ret, const char *source,
                        *along, *pncopies);
       return;
 
-#ifdef HAVE_GFC_REAL_10
+/* FIXME: This here is a hack, which will have to be removed when
+   the array descriptor is reworked.  Currently, we don't store the
+   kind value for the type, but only the size.  Because on targets with
+   _Float128, we have sizeof(long double) == sizeof(_Float128),
+   we cannot discriminate here and have to fall back to the generic
+   handling (which is suboptimal).  */
+#if !defined(GFC_REAL_16_IS_FLOAT128)
+# ifdef HAVE_GFC_REAL_10
     case GFC_DTYPE_REAL_10:
       spread_scalar_r10 ((gfc_array_r10 *) ret, (GFC_REAL_10 *) source,
                        *along, *pncopies);
       return;
-#endif
+# endif
 
-#ifdef HAVE_GFC_REAL_16
+# ifdef HAVE_GFC_REAL_16
     case GFC_DTYPE_REAL_16:
       spread_scalar_r16 ((gfc_array_r16 *) ret, (GFC_REAL_16 *) source,
                        *along, *pncopies);
       return;
+# endif
 #endif
 
     case GFC_DTYPE_COMPLEX_4:
@@ -521,22 +555,39 @@ spread_scalar (gfc_array_char *ret, const char *source,
                        *along, *pncopies);
       return;
 
-#ifdef HAVE_GFC_COMPLEX_10
+/* FIXME: This here is a hack, which will have to be removed when
+   the array descriptor is reworked.  Currently, we don't store the
+   kind value for the type, but only the size.  Because on targets with
+   _Float128, we have sizeof(long double) == sizeof(_Float128),
+   we cannot discriminate here and have to fall back to the generic
+   handling (which is suboptimal).  */
+#if !defined(GFC_REAL_16_IS_FLOAT128)
+# ifdef HAVE_GFC_COMPLEX_10
     case GFC_DTYPE_COMPLEX_10:
       spread_scalar_c10 ((gfc_array_c10 *) ret, (GFC_COMPLEX_10 *) source,
                        *along, *pncopies);
       return;
-#endif
+# endif
 
-#ifdef HAVE_GFC_COMPLEX_16
+# ifdef HAVE_GFC_COMPLEX_16
     case GFC_DTYPE_COMPLEX_16:
       spread_scalar_c16 ((gfc_array_c16 *) ret, (GFC_COMPLEX_16 *) source,
                        *along, *pncopies);
       return;
+# endif
 #endif
 
-    case GFC_DTYPE_DERIVED_2:
-      if (GFC_UNALIGNED_2(ret->data) || GFC_UNALIGNED_2(source))
+    }
+
+  switch (GFC_DESCRIPTOR_SIZE(ret))
+    {
+    case 1:
+      spread_scalar_i1 ((gfc_array_i1 *) ret, (GFC_INTEGER_1 *) source,
+                       *along, *pncopies);
+      return;
+
+    case 2:
+      if (GFC_UNALIGNED_2(ret->base_addr) || GFC_UNALIGNED_2(source))
        break;
       else
        {
@@ -545,8 +596,8 @@ spread_scalar (gfc_array_char *ret, const char *source,
          return;
        }
 
-    case GFC_DTYPE_DERIVED_4:
-      if (GFC_UNALIGNED_4(ret->data) || GFC_UNALIGNED_4(source))
+    case 4:
+      if (GFC_UNALIGNED_4(ret->base_addr) || GFC_UNALIGNED_4(source))
        break;
       else
        {
@@ -555,8 +606,8 @@ spread_scalar (gfc_array_char *ret, const char *source,
          return;
        }
 
-    case GFC_DTYPE_DERIVED_8:
-      if (GFC_UNALIGNED_8(ret->data) || GFC_UNALIGNED_8(source))
+    case 8:
+      if (GFC_UNALIGNED_8(ret->base_addr) || GFC_UNALIGNED_8(source))
        break;
       else
        {
@@ -565,8 +616,8 @@ spread_scalar (gfc_array_char *ret, const char *source,
          return;
        }
 #ifdef HAVE_GFC_INTEGER_16
-    case GFC_DTYPE_DERIVED_16:
-      if (GFC_UNALIGNED_16(ret->data) || GFC_UNALIGNED_16(source))
+    case 16:
+      if (GFC_UNALIGNED_16(ret->base_addr) || GFC_UNALIGNED_16(source))
        break;
       else
        {
@@ -575,9 +626,11 @@ spread_scalar (gfc_array_char *ret, const char *source,
          return;
        }
 #endif
+    default:
+      break;
     }
 
-  spread_internal_scalar (ret, source, along, pncopies, GFC_DESCRIPTOR_SIZE (ret));
+  spread_internal_scalar (ret, source, along, pncopies);
 }
 
 
@@ -590,11 +643,12 @@ void
 spread_char_scalar (gfc_array_char *ret,
                    GFC_INTEGER_4 ret_length __attribute__((unused)),
                    const char *source, const index_type *along,
-                   const index_type *pncopies, GFC_INTEGER_4 source_length)
+                   const index_type *pncopies,
+                   GFC_INTEGER_4 source_length __attribute__((unused)))
 {
-  if (!ret->dtype)
+  if (GFC_DTYPE_IS_UNSET(ret))
     runtime_error ("return array missing descriptor in spread()");
-  spread_internal_scalar (ret, source, along, pncopies, source_length);
+  spread_internal_scalar (ret, source, along, pncopies);
 }
 
 
@@ -607,11 +661,12 @@ void
 spread_char4_scalar (gfc_array_char *ret,
                     GFC_INTEGER_4 ret_length __attribute__((unused)),
                     const char *source, const index_type *along,
-                    const index_type *pncopies, GFC_INTEGER_4 source_length)
+                    const index_type *pncopies,
+                    GFC_INTEGER_4 source_length __attribute__((unused)))
 {
-  if (!ret->dtype)
+  if (GFC_DTYPE_IS_UNSET(ret))
     runtime_error ("return array missing descriptor in spread()");
-  spread_internal_scalar (ret, source, along, pncopies,
-                         source_length * sizeof (gfc_char4_t));
+  spread_internal_scalar (ret, source, along, pncopies);
+
 }