1 /* Implementation of the ALL intrinsic
2    Copyright (C) 2002-2016 Free Software Foundation, Inc.
3    Contributed by Paul Brook <paul@nowt.org>
4 
5 This file is part of the GNU Fortran 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 
30 
31 #if defined (HAVE_GFC_LOGICAL_16)
32 
33 
34 extern void all_l16 (gfc_array_l16 * const restrict,
35 	gfc_array_l1 * const restrict, const index_type * const restrict);
36 export_proto(all_l16);
37 
38 void
all_l16(gfc_array_l16 * const restrict retarray,gfc_array_l1 * const restrict array,const index_type * const restrict pdim)39 all_l16 (gfc_array_l16 * const restrict retarray,
40 	gfc_array_l1 * const restrict array,
41 	const index_type * const restrict pdim)
42 {
43   index_type count[GFC_MAX_DIMENSIONS];
44   index_type extent[GFC_MAX_DIMENSIONS];
45   index_type sstride[GFC_MAX_DIMENSIONS];
46   index_type dstride[GFC_MAX_DIMENSIONS];
47   const GFC_LOGICAL_1 * restrict base;
48   GFC_LOGICAL_16 * restrict dest;
49   index_type rank;
50   index_type n;
51   index_type len;
52   index_type delta;
53   index_type dim;
54   int src_kind;
55   int continue_loop;
56 
57   /* Make dim zero based to avoid confusion.  */
58   dim = (*pdim) - 1;
59   rank = GFC_DESCRIPTOR_RANK (array) - 1;
60 
61   src_kind = GFC_DESCRIPTOR_SIZE (array);
62 
63   len = GFC_DESCRIPTOR_EXTENT(array,dim);
64   if (len < 0)
65     len = 0;
66 
67   delta = GFC_DESCRIPTOR_STRIDE_BYTES(array,dim);
68 
69   for (n = 0; n < dim; n++)
70     {
71       sstride[n] = GFC_DESCRIPTOR_STRIDE_BYTES(array,n);
72       extent[n] = GFC_DESCRIPTOR_EXTENT(array,n);
73 
74       if (extent[n] < 0)
75 	extent[n] = 0;
76     }
77   for (n = dim; n < rank; n++)
78     {
79       sstride[n] = GFC_DESCRIPTOR_STRIDE_BYTES(array,n + 1);
80       extent[n] = GFC_DESCRIPTOR_EXTENT(array,n + 1);
81 
82       if (extent[n] < 0)
83 	extent[n] = 0;
84     }
85 
86   if (retarray->base_addr == NULL)
87     {
88       size_t alloc_size, str;
89 
90       for (n = 0; n < rank; n++)
91         {
92           if (n == 0)
93             str = 1;
94           else
95             str = GFC_DESCRIPTOR_STRIDE(retarray,n-1) * extent[n-1];
96 
97 	  GFC_DIMENSION_SET(retarray->dim[n], 0, extent[n] - 1, str);
98 
99         }
100 
101       retarray->offset = 0;
102       retarray->dtype = (array->dtype & ~GFC_DTYPE_RANK_MASK) | rank;
103 
104       alloc_size = GFC_DESCRIPTOR_STRIDE(retarray,rank-1) * extent[rank-1];
105 
106       if (alloc_size == 0)
107 	{
108 	  /* Make sure we have a zero-sized array.  */
109 	  GFC_DIMENSION_SET(retarray->dim[0], 0, -1, 1);
110 	  return;
111 	}
112       else
113 	retarray->base_addr = xmallocarray (alloc_size, sizeof (GFC_LOGICAL_16));
114     }
115   else
116     {
117       if (rank != GFC_DESCRIPTOR_RANK (retarray))
118 	runtime_error ("rank of return array incorrect in"
119 		       " ALL intrinsic: is %ld, should be %ld",
120 		       (long int) GFC_DESCRIPTOR_RANK (retarray),
121 		       (long int) rank);
122 
123       if (unlikely (compile_options.bounds_check))
124 	{
125 	  for (n=0; n < rank; n++)
126 	    {
127 	      index_type ret_extent;
128 
129 	      ret_extent = GFC_DESCRIPTOR_EXTENT(retarray,n);
130 	      if (extent[n] != ret_extent)
131 		runtime_error ("Incorrect extent in return value of"
132 			       " ALL intrinsic in dimension %d:"
133 			       " is %ld, should be %ld", (int) n + 1,
134 			       (long int) ret_extent, (long int) extent[n]);
135 	    }
136 	}
137     }
138 
139   for (n = 0; n < rank; n++)
140     {
141       count[n] = 0;
142       dstride[n] = GFC_DESCRIPTOR_STRIDE(retarray,n);
143       if (extent[n] <= 0)
144 	return;
145     }
146 
147   base = array->base_addr;
148 
149   if (src_kind == 1 || src_kind == 2 || src_kind == 4 || src_kind == 8
150 #ifdef HAVE_GFC_LOGICAL_16
151       || src_kind == 16
152 #endif
153     )
154     {
155       if (base)
156 	base = GFOR_POINTER_TO_L1 (base, src_kind);
157     }
158   else
159     internal_error (NULL, "Funny sized logical array in ALL intrinsic");
160 
161   dest = retarray->base_addr;
162 
163   continue_loop = 1;
164   while (continue_loop)
165     {
166       const GFC_LOGICAL_1 * restrict src;
167       GFC_LOGICAL_16 result;
168       src = base;
169       {
170 
171   /* Return true only if all the elements are set.  */
172   result = 1;
173         if (len <= 0)
174 	  *dest = 1;
175 	else
176 	  {
177 	    for (n = 0; n < len; n++, src += delta)
178 	      {
179 
180   if (! *src)
181     {
182       result = 0;
183       break;
184     }
185           }
186 	    *dest = result;
187 	  }
188       }
189       /* Advance to the next element.  */
190       count[0]++;
191       base += sstride[0];
192       dest += dstride[0];
193       n = 0;
194       while (count[n] == extent[n])
195         {
196           /* When we get to the end of a dimension, reset it and increment
197              the next dimension.  */
198           count[n] = 0;
199           /* We could precalculate these products, but this is a less
200              frequently used path so probably not worth it.  */
201           base -= sstride[n] * extent[n];
202           dest -= dstride[n] * extent[n];
203           n++;
204           if (n == rank)
205             {
206               /* Break out of the look.  */
207               continue_loop = 0;
208               break;
209             }
210           else
211             {
212               count[n]++;
213               base += sstride[n];
214               dest += dstride[n];
215             }
216         }
217     }
218 }
219 
220 #endif
221