1 /* Implementation of the IPARITY intrinsic
2 Copyright (C) 2010-2018 Free Software Foundation, Inc.
3 Contributed by Tobias Burnus <burnus@net-b.de>
5 This file is part of the GNU Fortran runtime library (libgfortran).
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.
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.
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.
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/>. */
26 #include "libgfortran.h"
29 #if defined (HAVE_GFC_INTEGER_4) && defined (HAVE_GFC_INTEGER_4)
32 extern void iparity_i4 (gfc_array_i4
* const restrict
,
33 gfc_array_i4
* const restrict
, const index_type
* const restrict
);
34 export_proto(iparity_i4
);
37 iparity_i4 (gfc_array_i4
* const restrict retarray
,
38 gfc_array_i4
* const restrict array
,
39 const index_type
* const restrict pdim
)
41 index_type count
[GFC_MAX_DIMENSIONS
];
42 index_type extent
[GFC_MAX_DIMENSIONS
];
43 index_type sstride
[GFC_MAX_DIMENSIONS
];
44 index_type dstride
[GFC_MAX_DIMENSIONS
];
45 const GFC_INTEGER_4
* restrict base
;
46 GFC_INTEGER_4
* restrict dest
;
54 /* Make dim zero based to avoid confusion. */
55 rank
= GFC_DESCRIPTOR_RANK (array
) - 1;
58 if (unlikely (dim
< 0 || dim
> rank
))
60 runtime_error ("Dim argument incorrect in IPARITY intrinsic: "
61 "is %ld, should be between 1 and %ld",
62 (long int) dim
+ 1, (long int) rank
+ 1);
65 len
= GFC_DESCRIPTOR_EXTENT(array
,dim
);
68 delta
= GFC_DESCRIPTOR_STRIDE(array
,dim
);
70 for (n
= 0; n
< dim
; n
++)
72 sstride
[n
] = GFC_DESCRIPTOR_STRIDE(array
,n
);
73 extent
[n
] = GFC_DESCRIPTOR_EXTENT(array
,n
);
78 for (n
= dim
; n
< rank
; n
++)
80 sstride
[n
] = GFC_DESCRIPTOR_STRIDE(array
, n
+ 1);
81 extent
[n
] = GFC_DESCRIPTOR_EXTENT(array
, n
+ 1);
87 if (retarray
->base_addr
== NULL
)
89 size_t alloc_size
, str
;
91 for (n
= 0; n
< rank
; n
++)
96 str
= GFC_DESCRIPTOR_STRIDE(retarray
,n
-1) * extent
[n
-1];
98 GFC_DIMENSION_SET(retarray
->dim
[n
], 0, extent
[n
] - 1, str
);
102 retarray
->offset
= 0;
103 GFC_DTYPE_COPY_SETRANK(retarray
,array
,rank
);
105 alloc_size
= GFC_DESCRIPTOR_STRIDE(retarray
,rank
-1) * extent
[rank
-1];
107 retarray
->base_addr
= xmallocarray (alloc_size
, sizeof (GFC_INTEGER_4
));
110 /* Make sure we have a zero-sized array. */
111 GFC_DIMENSION_SET(retarray
->dim
[0], 0, -1, 1);
118 if (rank
!= GFC_DESCRIPTOR_RANK (retarray
))
119 runtime_error ("rank of return array incorrect in"
120 " IPARITY intrinsic: is %ld, should be %ld",
121 (long int) (GFC_DESCRIPTOR_RANK (retarray
)),
124 if (unlikely (compile_options
.bounds_check
))
125 bounds_ifunction_return ((array_t
*) retarray
, extent
,
126 "return value", "IPARITY");
129 for (n
= 0; n
< rank
; n
++)
132 dstride
[n
] = GFC_DESCRIPTOR_STRIDE(retarray
,n
);
137 base
= array
->base_addr
;
138 dest
= retarray
->base_addr
;
141 while (continue_loop
)
143 const GFC_INTEGER_4
* restrict src
;
144 GFC_INTEGER_4 result
;
153 for (n
= 0; n
< len
; n
++, src
+= delta
)
162 /* Advance to the next element. */
167 while (count
[n
] == extent
[n
])
169 /* When we get to the end of a dimension, reset it and increment
170 the next dimension. */
172 /* We could precalculate these products, but this is a less
173 frequently used path so probably not worth it. */
174 base
-= sstride
[n
] * extent
[n
];
175 dest
-= dstride
[n
] * extent
[n
];
179 /* Break out of the loop. */
194 extern void miparity_i4 (gfc_array_i4
* const restrict
,
195 gfc_array_i4
* const restrict
, const index_type
* const restrict
,
196 gfc_array_l1
* const restrict
);
197 export_proto(miparity_i4
);
200 miparity_i4 (gfc_array_i4
* const restrict retarray
,
201 gfc_array_i4
* const restrict array
,
202 const index_type
* const restrict pdim
,
203 gfc_array_l1
* const restrict mask
)
205 index_type count
[GFC_MAX_DIMENSIONS
];
206 index_type extent
[GFC_MAX_DIMENSIONS
];
207 index_type sstride
[GFC_MAX_DIMENSIONS
];
208 index_type dstride
[GFC_MAX_DIMENSIONS
];
209 index_type mstride
[GFC_MAX_DIMENSIONS
];
210 GFC_INTEGER_4
* restrict dest
;
211 const GFC_INTEGER_4
* restrict base
;
212 const GFC_LOGICAL_1
* restrict mbase
;
222 rank
= GFC_DESCRIPTOR_RANK (array
) - 1;
225 if (unlikely (dim
< 0 || dim
> rank
))
227 runtime_error ("Dim argument incorrect in IPARITY intrinsic: "
228 "is %ld, should be between 1 and %ld",
229 (long int) dim
+ 1, (long int) rank
+ 1);
232 len
= GFC_DESCRIPTOR_EXTENT(array
,dim
);
236 mbase
= mask
->base_addr
;
238 mask_kind
= GFC_DESCRIPTOR_SIZE (mask
);
240 if (mask_kind
== 1 || mask_kind
== 2 || mask_kind
== 4 || mask_kind
== 8
241 #ifdef HAVE_GFC_LOGICAL_16
245 mbase
= GFOR_POINTER_TO_L1 (mbase
, mask_kind
);
247 runtime_error ("Funny sized logical array");
249 delta
= GFC_DESCRIPTOR_STRIDE(array
,dim
);
250 mdelta
= GFC_DESCRIPTOR_STRIDE_BYTES(mask
,dim
);
252 for (n
= 0; n
< dim
; n
++)
254 sstride
[n
] = GFC_DESCRIPTOR_STRIDE(array
,n
);
255 mstride
[n
] = GFC_DESCRIPTOR_STRIDE_BYTES(mask
,n
);
256 extent
[n
] = GFC_DESCRIPTOR_EXTENT(array
,n
);
262 for (n
= dim
; n
< rank
; n
++)
264 sstride
[n
] = GFC_DESCRIPTOR_STRIDE(array
,n
+ 1);
265 mstride
[n
] = GFC_DESCRIPTOR_STRIDE_BYTES(mask
, n
+ 1);
266 extent
[n
] = GFC_DESCRIPTOR_EXTENT(array
, n
+ 1);
272 if (retarray
->base_addr
== NULL
)
274 size_t alloc_size
, str
;
276 for (n
= 0; n
< rank
; n
++)
281 str
= GFC_DESCRIPTOR_STRIDE(retarray
,n
-1) * extent
[n
-1];
283 GFC_DIMENSION_SET(retarray
->dim
[n
], 0, extent
[n
] - 1, str
);
287 alloc_size
= GFC_DESCRIPTOR_STRIDE(retarray
,rank
-1) * extent
[rank
-1];
289 retarray
->offset
= 0;
290 GFC_DTYPE_COPY_SETRANK(retarray
,array
,rank
);
294 /* Make sure we have a zero-sized array. */
295 GFC_DIMENSION_SET(retarray
->dim
[0], 0, -1, 1);
299 retarray
->base_addr
= xmallocarray (alloc_size
, sizeof (GFC_INTEGER_4
));
304 if (rank
!= GFC_DESCRIPTOR_RANK (retarray
))
305 runtime_error ("rank of return array incorrect in IPARITY intrinsic");
307 if (unlikely (compile_options
.bounds_check
))
309 bounds_ifunction_return ((array_t
*) retarray
, extent
,
310 "return value", "IPARITY");
311 bounds_equal_extents ((array_t
*) mask
, (array_t
*) array
,
312 "MASK argument", "IPARITY");
316 for (n
= 0; n
< rank
; n
++)
319 dstride
[n
] = GFC_DESCRIPTOR_STRIDE(retarray
,n
);
324 dest
= retarray
->base_addr
;
325 base
= array
->base_addr
;
329 const GFC_INTEGER_4
* restrict src
;
330 const GFC_LOGICAL_1
* restrict msrc
;
331 GFC_INTEGER_4 result
;
337 for (n
= 0; n
< len
; n
++, src
+= delta
, msrc
+= mdelta
)
345 /* Advance to the next element. */
351 while (count
[n
] == extent
[n
])
353 /* When we get to the end of a dimension, reset it and increment
354 the next dimension. */
356 /* We could precalculate these products, but this is a less
357 frequently used path so probably not worth it. */
358 base
-= sstride
[n
] * extent
[n
];
359 mbase
-= mstride
[n
] * extent
[n
];
360 dest
-= dstride
[n
] * extent
[n
];
364 /* Break out of the loop. */
380 extern void siparity_i4 (gfc_array_i4
* const restrict
,
381 gfc_array_i4
* const restrict
, const index_type
* const restrict
,
383 export_proto(siparity_i4
);
386 siparity_i4 (gfc_array_i4
* const restrict retarray
,
387 gfc_array_i4
* const restrict array
,
388 const index_type
* const restrict pdim
,
389 GFC_LOGICAL_4
* mask
)
391 index_type count
[GFC_MAX_DIMENSIONS
];
392 index_type extent
[GFC_MAX_DIMENSIONS
];
393 index_type dstride
[GFC_MAX_DIMENSIONS
];
394 GFC_INTEGER_4
* restrict dest
;
402 iparity_i4 (retarray
, array
, pdim
);
405 /* Make dim zero based to avoid confusion. */
407 rank
= GFC_DESCRIPTOR_RANK (array
) - 1;
409 if (unlikely (dim
< 0 || dim
> rank
))
411 runtime_error ("Dim argument incorrect in IPARITY intrinsic: "
412 "is %ld, should be between 1 and %ld",
413 (long int) dim
+ 1, (long int) rank
+ 1);
416 for (n
= 0; n
< dim
; n
++)
418 extent
[n
] = GFC_DESCRIPTOR_EXTENT(array
,n
);
424 for (n
= dim
; n
< rank
; n
++)
427 GFC_DESCRIPTOR_EXTENT(array
,n
+ 1);
433 if (retarray
->base_addr
== NULL
)
435 size_t alloc_size
, str
;
437 for (n
= 0; n
< rank
; n
++)
442 str
= GFC_DESCRIPTOR_STRIDE(retarray
,n
-1) * extent
[n
-1];
444 GFC_DIMENSION_SET(retarray
->dim
[n
], 0, extent
[n
] - 1, str
);
448 retarray
->offset
= 0;
449 GFC_DTYPE_COPY_SETRANK(retarray
,array
,rank
);
451 alloc_size
= GFC_DESCRIPTOR_STRIDE(retarray
,rank
-1) * extent
[rank
-1];
455 /* Make sure we have a zero-sized array. */
456 GFC_DIMENSION_SET(retarray
->dim
[0], 0, -1, 1);
460 retarray
->base_addr
= xmallocarray (alloc_size
, sizeof (GFC_INTEGER_4
));
464 if (rank
!= GFC_DESCRIPTOR_RANK (retarray
))
465 runtime_error ("rank of return array incorrect in"
466 " IPARITY intrinsic: is %ld, should be %ld",
467 (long int) (GFC_DESCRIPTOR_RANK (retarray
)),
470 if (unlikely (compile_options
.bounds_check
))
472 for (n
=0; n
< rank
; n
++)
474 index_type ret_extent
;
476 ret_extent
= GFC_DESCRIPTOR_EXTENT(retarray
,n
);
477 if (extent
[n
] != ret_extent
)
478 runtime_error ("Incorrect extent in return value of"
479 " IPARITY intrinsic in dimension %ld:"
480 " is %ld, should be %ld", (long int) n
+ 1,
481 (long int) ret_extent
, (long int) extent
[n
]);
486 for (n
= 0; n
< rank
; n
++)
489 dstride
[n
] = GFC_DESCRIPTOR_STRIDE(retarray
,n
);
492 dest
= retarray
->base_addr
;
500 while (count
[n
] == extent
[n
])
502 /* When we get to the end of a dimension, reset it and increment
503 the next dimension. */
505 /* We could precalculate these products, but this is a less
506 frequently used path so probably not worth it. */
507 dest
-= dstride
[n
] * extent
[n
];