libgfortran/m4/eoshift1.m4

   1 `/* Implementation of the EOSHIFT intrinsic
   2    Copyright 2002, 2005, 2007, 2009 Free Software Foundation, Inc.
   3    Contributed by Paul Brook <paul@nowt.org>
   4
   5 This file is part of the GNU Fortran 95 runtime library (libgfortran).
   6
   7 Libgfortran is free software; you can redistribute it and/or
   8 modify it under the terms of the GNU General Public
   9 License as published by the Free Software Foundation; either
  10 version 3 of the License, or (at your option) any later version.
  11
  12 Libgfortran is distributed in the hope that it will be useful,
  13 but WITHOUT ANY WARRANTY; without even the implied warranty of
  14 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15 GNU General Public License for more details.
  16
  17 Under Section 7 of GPL version 3, you are granted additional
  18 permissions described in the GCC Runtime Library Exception, version
  19 3.1, as published by the Free Software Foundation.
  20
  21 You should have received a copy of the GNU General Public License and
  22 a copy of the GCC Runtime Library Exception along with this program;
  23 see the files COPYING3 and COPYING.RUNTIME respectively.  If not, see
  24 <http://www.gnu.org/licenses/>.  */
  25
  26 #include "libgfortran.h"
  27 #include <stdlib.h>
  28 #include <assert.h>
  29 #include <string.h>'
  30
  31 include(iparm.m4)dnl
  32
  33 `#if defined (HAVE_'atype_name`)
  34
  35 static void
  36 eoshift1 (gfc_array_char * const restrict ret,
  37         const gfc_array_char * const restrict array,
  38         const 'atype` * const restrict h,
  39         const char * const restrict pbound,
  40         const 'atype_name` * const restrict pwhich,
  41         index_type size, const char * filler, index_type filler_len)
  42 {
  43   /* r.* indicates the return array.  */
  44   index_type rstride[GFC_MAX_DIMENSIONS];
  45   index_type rstride0;
  46   index_type roffset;
  47   char *rptr;
  48   char * restrict dest;
  49   /* s.* indicates the source array.  */
  50   index_type sstride[GFC_MAX_DIMENSIONS];
  51   index_type sstride0;
  52   index_type soffset;
  53   const char *sptr;
  54   const char *src;
  55   /* h.* indicates the shift array.  */
  56   index_type hstride[GFC_MAX_DIMENSIONS];
  57   index_type hstride0;
  58   const 'atype_name` *hptr;
  59
  60   index_type count[GFC_MAX_DIMENSIONS];
  61   index_type extent[GFC_MAX_DIMENSIONS];
  62   index_type dim;
  63   index_type len;
  64   index_type n;
  65   int which;
  66   'atype_name` sh;
  67   'atype_name` delta;
  68
  69   /* The compiler cannot figure out that these are set, initialize
  70      them to avoid warnings.  */
  71   len = 0;
  72   soffset = 0;
  73   roffset = 0;
  74
  75   if (pwhich)
  76     which = *pwhich - 1;
  77   else
  78     which = 0;
  79
  80   extent[0] = 1;
  81   count[0] = 0;
  82
  83   if (ret->data == NULL)
  84     {
  85       int i;
  86
  87       ret->data = internal_malloc_size (size * size0 ((array_t *)array));
  88       ret->offset = 0;
  89       ret->dtype = array->dtype;
  90       for (i = 0; i < GFC_DESCRIPTOR_RANK (array); i++)
  91         {
  92           ret->dim[i].lbound = 0;
  93           ret->dim[i].ubound = array->dim[i].ubound - array->dim[i].lbound;
  94
  95           if (i == 0)
  96             ret->dim[i].stride = 1;
  97           else
  98             ret->dim[i].stride = (ret->dim[i-1].ubound + 1) * ret->dim[i-1].stride;
  99         }
 100     }
 101   else
 102     {
 103       if (size0 ((array_t *) ret) == 0)
 104         return;
 105     }
 106
 107   n = 0;
 108   for (dim = 0; dim < GFC_DESCRIPTOR_RANK (array); dim++)
 109     {
 110       if (dim == which)
 111         {
 112           roffset = ret->dim[dim].stride * size;
 113           if (roffset == 0)
 114             roffset = size;
 115           soffset = array->dim[dim].stride * size;
 116           if (soffset == 0)
 117             soffset = size;
 118           len = array->dim[dim].ubound + 1 - array->dim[dim].lbound;
 119         }
 120       else
 121         {
 122           count[n] = 0;
 123           extent[n] = array->dim[dim].ubound + 1 - array->dim[dim].lbound;
 124           rstride[n] = ret->dim[dim].stride * size;
 125           sstride[n] = array->dim[dim].stride * size;
 126
 127           hstride[n] = h->dim[n].stride;
 128           n++;
 129         }
 130     }
 131   if (sstride[0] == 0)
 132     sstride[0] = size;
 133   if (rstride[0] == 0)
 134     rstride[0] = size;
 135   if (hstride[0] == 0)
 136     hstride[0] = 1;
 137
 138   dim = GFC_DESCRIPTOR_RANK (array);
 139   rstride0 = rstride[0];
 140   sstride0 = sstride[0];
 141   hstride0 = hstride[0];
 142   rptr = ret->data;
 143   sptr = array->data;
 144   hptr = h->data;
 145
 146   while (rptr)
 147     {
 148       /* Do the shift for this dimension.  */
 149       sh = *hptr;
 150       if (( sh >= 0 ? sh : -sh ) > len)
 151         {
 152           delta = len;
 153           sh = len;
 154         }
 155       else
 156         delta = (sh >= 0) ? sh: -sh;
 157
 158       if (sh > 0)
 159         {
 160           src = &sptr[delta * soffset];
 161           dest = rptr;
 162         }
 163       else
 164         {
 165           src = sptr;
 166           dest = &rptr[delta * roffset];
 167         }
 168       for (n = 0; n < len - delta; n++)
 169         {
 170           memcpy (dest, src, size);
 171           dest += roffset;
 172           src += soffset;
 173         }
 174       if (sh < 0)
 175         dest = rptr;
 176       n = delta;
 177
 178       if (pbound)
 179         while (n--)
 180           {
 181             memcpy (dest, pbound, size);
 182             dest += roffset;
 183           }
 184       else
 185         while (n--)
 186           {
 187             index_type i;
 188
 189             if (filler_len == 1)
 190               memset (dest, filler[0], size);
 191             else
 192               for (i = 0; i < size; i += filler_len)
 193                 memcpy (&dest[i], filler, filler_len);
 194
 195             dest += roffset;
 196           }
 197
 198       /* Advance to the next section.  */
 199       rptr += rstride0;
 200       sptr += sstride0;
 201       hptr += hstride0;
 202       count[0]++;
 203       n = 0;
 204       while (count[n] == extent[n])
 205         {
 206           /* When we get to the end of a dimension, reset it and increment
 207              the next dimension.  */
 208           count[n] = 0;
 209           /* We could precalculate these products, but this is a less
 210              frequently used path so probably not worth it.  */
 211           rptr -= rstride[n] * extent[n];
 212           sptr -= sstride[n] * extent[n];
 213           hptr -= hstride[n] * extent[n];
 214           n++;
 215           if (n >= dim - 1)
 216             {
 217               /* Break out of the loop.  */
 218               rptr = NULL;
 219               break;
 220             }
 221           else
 222             {
 223               count[n]++;
 224               rptr += rstride[n];
 225               sptr += sstride[n];
 226               hptr += hstride[n];
 227             }
 228         }
 229     }
 230 }
 231
 232 void eoshift1_'atype_kind` (gfc_array_char * const restrict,
 233         const gfc_array_char * const restrict,
 234         const 'atype` * const restrict, const char * const restrict,
 235         const 'atype_name` * const restrict);
 236 export_proto(eoshift1_'atype_kind`);
 237
 238 void
 239 eoshift1_'atype_kind` (gfc_array_char * const restrict ret,
 240         const gfc_array_char * const restrict array,
 241         const 'atype` * const restrict h,
 242         const char * const restrict pbound,
 243         const 'atype_name` * const restrict pwhich)
 244 {
 245   eoshift1 (ret, array, h, pbound, pwhich, GFC_DESCRIPTOR_SIZE (array),
 246             "\0", 1);
 247 }
 248
 249
 250 void eoshift1_'atype_kind`_char (gfc_array_char * const restrict,
 251         GFC_INTEGER_4,
 252         const gfc_array_char * const restrict,
 253         const 'atype` * const restrict,
 254         const char * const restrict,
 255         const 'atype_name` * const restrict,
 256         GFC_INTEGER_4, GFC_INTEGER_4);
 257 export_proto(eoshift1_'atype_kind`_char);
 258
 259 void
 260 eoshift1_'atype_kind`_char (gfc_array_char * const restrict ret,
 261         GFC_INTEGER_4 ret_length __attribute__((unused)),
 262         const gfc_array_char * const restrict array,
 263         const 'atype` * const restrict h,
 264         const char *  const restrict pbound,
 265         const 'atype_name` * const restrict pwhich,
 266         GFC_INTEGER_4 array_length,
 267         GFC_INTEGER_4 bound_length __attribute__((unused)))
 268 {
 269   eoshift1 (ret, array, h, pbound, pwhich, array_length, " ", 1);
 270 }
 271
 272
 273 void eoshift1_'atype_kind`_char4 (gfc_array_char * const restrict,
 274         GFC_INTEGER_4,
 275         const gfc_array_char * const restrict,
 276         const 'atype` * const restrict,
 277         const char * const restrict,
 278         const 'atype_name` * const restrict,
 279         GFC_INTEGER_4, GFC_INTEGER_4);
 280 export_proto(eoshift1_'atype_kind`_char4);
 281
 282 void
 283 eoshift1_'atype_kind`_char4 (gfc_array_char * const restrict ret,
 284         GFC_INTEGER_4 ret_length __attribute__((unused)),
 285         const gfc_array_char * const restrict array,
 286         const 'atype` * const restrict h,
 287         const char *  const restrict pbound,
 288         const 'atype_name` * const restrict pwhich,
 289         GFC_INTEGER_4 array_length,
 290         GFC_INTEGER_4 bound_length __attribute__((unused)))
 291 {
 292   static const gfc_char4_t space = (unsigned char) ''` ''`;
 293   eoshift1 (ret, array, h, pbound, pwhich, array_length * sizeof (gfc_char4_t),
 294             (const char *) &space, sizeof (gfc_char4_t));
 295 }
 296
 297 #endif'