libgfortran/m4/eoshift1.m4

   1 `/* Implementation of the EOSHIFT intrinsic
   2    Copyright 2002, 2005, 2007 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 2 of the License, or (at your option) any later version.
  11
  12 In addition to the permissions in the GNU General Public License, the
  13 Free Software Foundation gives you unlimited permission to link the
  14 compiled version of this file into combinations with other programs,
  15 and to distribute those combinations without any restriction coming
  16 from the use of this file.  (The General Public License restrictions
  17 do apply in other respects; for example, they cover modification of
  18 the file, and distribution when not linked into a combine
  19 executable.)
  20
  21 Libgfortran is distributed in the hope that it will be useful,
  22 but WITHOUT ANY WARRANTY; without even the implied warranty of
  23 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  24 GNU General Public License for more details.
  25
  26 You should have received a copy of the GNU General Public
  27 License along with libgfortran; see the file COPYING.  If not,
  28 write to the Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor,
  29 Boston, MA 02110-1301, USA.  */
  30
  31 #include "libgfortran.h"
  32 #include <stdlib.h>
  33 #include <assert.h>
  34 #include <string.h>'
  35
  36 include(iparm.m4)dnl
  37
  38 `#if defined (HAVE_'atype_name`)
  39
  40 static void
  41 eoshift1 (gfc_array_char * const restrict ret,
  42         const gfc_array_char * const restrict array,
  43         const 'atype` * const restrict h,
  44         const char * const restrict pbound,
  45         const 'atype_name` * const restrict pwhich,
  46         index_type size, char filler)
  47 {
  48   /* r.* indicates the return array.  */
  49   index_type rstride[GFC_MAX_DIMENSIONS];
  50   index_type rstride0;
  51   index_type roffset;
  52   char *rptr;
  53   char *dest;
  54   /* s.* indicates the source array.  */
  55   index_type sstride[GFC_MAX_DIMENSIONS];
  56   index_type sstride0;
  57   index_type soffset;
  58   const char *sptr;
  59   const char *src;
  60   /* h.* indicates the shift array.  */
  61   index_type hstride[GFC_MAX_DIMENSIONS];
  62   index_type hstride0;
  63   const 'atype_name` *hptr;
  64
  65   index_type count[GFC_MAX_DIMENSIONS];
  66   index_type extent[GFC_MAX_DIMENSIONS];
  67   index_type dim;
  68   index_type len;
  69   index_type n;
  70   int which;
  71   'atype_name` sh;
  72   'atype_name` delta;
  73
  74   /* The compiler cannot figure out that these are set, initialize
  75      them to avoid warnings.  */
  76   len = 0;
  77   soffset = 0;
  78   roffset = 0;
  79
  80   if (pwhich)
  81     which = *pwhich - 1;
  82   else
  83     which = 0;
  84
  85   extent[0] = 1;
  86   count[0] = 0;
  87
  88   if (ret->data == NULL)
  89     {
  90       int i;
  91
  92       ret->data = internal_malloc_size (size * size0 ((array_t *)array));
  93       ret->offset = 0;
  94       ret->dtype = array->dtype;
  95       for (i = 0; i < GFC_DESCRIPTOR_RANK (array); i++)
  96         {
  97           ret->dim[i].lbound = 0;
  98           ret->dim[i].ubound = array->dim[i].ubound - array->dim[i].lbound;
  99
 100           if (i == 0)
 101             ret->dim[i].stride = 1;
 102           else
 103             ret->dim[i].stride = (ret->dim[i-1].ubound + 1) * ret->dim[i-1].stride;
 104         }
 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             memset (dest, filler, size);
 188             dest += roffset;
 189           }
 190
 191       /* Advance to the next section.  */
 192       rptr += rstride0;
 193       sptr += sstride0;
 194       hptr += hstride0;
 195       count[0]++;
 196       n = 0;
 197       while (count[n] == extent[n])
 198         {
 199           /* When we get to the end of a dimension, reset it and increment
 200              the next dimension.  */
 201           count[n] = 0;
 202           /* We could precalculate these products, but this is a less
 203              frequently used path so probably not worth it.  */
 204           rptr -= rstride[n] * extent[n];
 205           sptr -= sstride[n] * extent[n];
 206           hptr -= hstride[n] * extent[n];
 207           n++;
 208           if (n >= dim - 1)
 209             {
 210               /* Break out of the loop.  */
 211               rptr = NULL;
 212               break;
 213             }
 214           else
 215             {
 216               count[n]++;
 217               rptr += rstride[n];
 218               sptr += sstride[n];
 219               hptr += hstride[n];
 220             }
 221         }
 222     }
 223 }
 224
 225 void eoshift1_'atype_kind` (gfc_array_char * const restrict,
 226         const gfc_array_char * const restrict,
 227         const 'atype` * const restrict, const char * const restrict,
 228         const 'atype_name` * const restrict);
 229 export_proto(eoshift1_'atype_kind`);
 230
 231 void
 232 eoshift1_'atype_kind` (gfc_array_char * const restrict ret,
 233         const gfc_array_char * const restrict array,
 234         const 'atype` * const restrict h,
 235         const char * const restrict pbound,
 236         const 'atype_name` * const restrict pwhich)
 237 {
 238   eoshift1 (ret, array, h, pbound, pwhich, GFC_DESCRIPTOR_SIZE (array), 0);
 239 }
 240
 241 void eoshift1_'atype_kind`_char (gfc_array_char * const restrict,
 242         GFC_INTEGER_4,
 243         const gfc_array_char * const restrict,
 244         const 'atype` * const restrict,
 245         const char * const restrict,
 246         const 'atype_name` * const restrict,
 247         GFC_INTEGER_4, GFC_INTEGER_4);
 248 export_proto(eoshift1_'atype_kind`_char);
 249
 250 void
 251 eoshift1_'atype_kind`_char (gfc_array_char * const restrict ret,
 252         GFC_INTEGER_4 ret_length __attribute__((unused)),
 253         const gfc_array_char * const restrict array,
 254         const 'atype` * const restrict h,
 255         const char *  const restrict pbound,
 256         const 'atype_name` * const restrict pwhich,
 257         GFC_INTEGER_4 array_length,
 258         GFC_INTEGER_4 bound_length __attribute__((unused)))
 259 {
 260   eoshift1 (ret, array, h, pbound, pwhich, array_length, ''` ''`);
 261 }
 262
 263 #endif'