Skip several gcc.dg/builtin-dynamic-object-size tests on hppa*-*-hpux*
[official-gcc.git] / gcc / fortran / simplify.cc
blob3043483daa90390aef0c17b5f40e85f62a9f0fab
1 /* Simplify intrinsic functions at compile-time.
2 Copyright (C) 2000-2024 Free Software Foundation, Inc.
3 Contributed by Andy Vaught & Katherine Holcomb
5 This file is part of GCC.
7 GCC is free software; you can redistribute it and/or modify it under
8 the terms of the GNU General Public License as published by the Free
9 Software Foundation; either version 3, or (at your option) any later
10 version.
12 GCC is distributed in the hope that it will be useful, but WITHOUT ANY
13 WARRANTY; without even the implied warranty of MERCHANTABILITY or
14 FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
15 for more details.
17 You should have received a copy of the GNU General Public License
18 along with GCC; see the file COPYING3. If not see
19 <http://www.gnu.org/licenses/>. */
21 #include "config.h"
22 #include "system.h"
23 #include "coretypes.h"
24 #include "tm.h" /* For BITS_PER_UNIT. */
25 #include "gfortran.h"
26 #include "arith.h"
27 #include "intrinsic.h"
28 #include "match.h"
29 #include "target-memory.h"
30 #include "constructor.h"
31 #include "version.h" /* For version_string. */
33 /* Prototypes. */
35 static int min_max_choose (gfc_expr *, gfc_expr *, int, bool back_val = false);
37 gfc_expr gfc_bad_expr;
39 static gfc_expr *simplify_size (gfc_expr *, gfc_expr *, int);
42 /* Note that 'simplification' is not just transforming expressions.
43 For functions that are not simplified at compile time, range
44 checking is done if possible.
46 The return convention is that each simplification function returns:
48 A new expression node corresponding to the simplified arguments.
49 The original arguments are destroyed by the caller, and must not
50 be a part of the new expression.
52 NULL pointer indicating that no simplification was possible and
53 the original expression should remain intact.
55 An expression pointer to gfc_bad_expr (a static placeholder)
56 indicating that some error has prevented simplification. The
57 error is generated within the function and should be propagated
58 upwards
60 By the time a simplification function gets control, it has been
61 decided that the function call is really supposed to be the
62 intrinsic. No type checking is strictly necessary, since only
63 valid types will be passed on. On the other hand, a simplification
64 subroutine may have to look at the type of an argument as part of
65 its processing.
67 Array arguments are only passed to these subroutines that implement
68 the simplification of transformational intrinsics.
70 The functions in this file don't have much comment with them, but
71 everything is reasonably straight-forward. The Standard, chapter 13
72 is the best comment you'll find for this file anyway. */
74 /* Range checks an expression node. If all goes well, returns the
75 node, otherwise returns &gfc_bad_expr and frees the node. */
77 static gfc_expr *
78 range_check (gfc_expr *result, const char *name)
80 if (result == NULL)
81 return &gfc_bad_expr;
83 if (result->expr_type != EXPR_CONSTANT)
84 return result;
86 switch (gfc_range_check (result))
88 case ARITH_OK:
89 return result;
91 case ARITH_OVERFLOW:
92 gfc_error ("Result of %s overflows its kind at %L", name,
93 &result->where);
94 break;
96 case ARITH_UNDERFLOW:
97 gfc_error ("Result of %s underflows its kind at %L", name,
98 &result->where);
99 break;
101 case ARITH_NAN:
102 gfc_error ("Result of %s is NaN at %L", name, &result->where);
103 break;
105 default:
106 gfc_error ("Result of %s gives range error for its kind at %L", name,
107 &result->where);
108 break;
111 gfc_free_expr (result);
112 return &gfc_bad_expr;
116 /* A helper function that gets an optional and possibly missing
117 kind parameter. Returns the kind, -1 if something went wrong. */
119 static int
120 get_kind (bt type, gfc_expr *k, const char *name, int default_kind)
122 int kind;
124 if (k == NULL)
125 return default_kind;
127 if (k->expr_type != EXPR_CONSTANT)
129 gfc_error ("KIND parameter of %s at %L must be an initialization "
130 "expression", name, &k->where);
131 return -1;
134 if (gfc_extract_int (k, &kind)
135 || gfc_validate_kind (type, kind, true) < 0)
137 gfc_error ("Invalid KIND parameter of %s at %L", name, &k->where);
138 return -1;
141 return kind;
145 /* Converts an mpz_t signed variable into an unsigned one, assuming
146 two's complement representations and a binary width of bitsize.
147 The conversion is a no-op unless x is negative; otherwise, it can
148 be accomplished by masking out the high bits. */
150 static void
151 convert_mpz_to_unsigned (mpz_t x, int bitsize)
153 mpz_t mask;
155 if (mpz_sgn (x) < 0)
157 /* Confirm that no bits above the signed range are unset if we
158 are doing range checking. */
159 if (flag_range_check != 0)
160 gcc_assert (mpz_scan0 (x, bitsize-1) == ULONG_MAX);
162 mpz_init_set_ui (mask, 1);
163 mpz_mul_2exp (mask, mask, bitsize);
164 mpz_sub_ui (mask, mask, 1);
166 mpz_and (x, x, mask);
168 mpz_clear (mask);
170 else
172 /* Confirm that no bits above the signed range are set if we
173 are doing range checking. */
174 if (flag_range_check != 0)
175 gcc_assert (mpz_scan1 (x, bitsize-1) == ULONG_MAX);
180 /* Converts an mpz_t unsigned variable into a signed one, assuming
181 two's complement representations and a binary width of bitsize.
182 If the bitsize-1 bit is set, this is taken as a sign bit and
183 the number is converted to the corresponding negative number. */
185 void
186 gfc_convert_mpz_to_signed (mpz_t x, int bitsize)
188 mpz_t mask;
190 /* Confirm that no bits above the unsigned range are set if we are
191 doing range checking. */
192 if (flag_range_check != 0)
193 gcc_assert (mpz_scan1 (x, bitsize) == ULONG_MAX);
195 if (mpz_tstbit (x, bitsize - 1) == 1)
197 mpz_init_set_ui (mask, 1);
198 mpz_mul_2exp (mask, mask, bitsize);
199 mpz_sub_ui (mask, mask, 1);
201 /* We negate the number by hand, zeroing the high bits, that is
202 make it the corresponding positive number, and then have it
203 negated by GMP, giving the correct representation of the
204 negative number. */
205 mpz_com (x, x);
206 mpz_add_ui (x, x, 1);
207 mpz_and (x, x, mask);
209 mpz_neg (x, x);
211 mpz_clear (mask);
216 /* Test that the expression is a constant array, simplifying if
217 we are dealing with a parameter array. */
219 static bool
220 is_constant_array_expr (gfc_expr *e)
222 gfc_constructor *c;
223 bool array_OK = true;
224 mpz_t size;
226 if (e == NULL)
227 return true;
229 if (e->expr_type == EXPR_VARIABLE && e->rank > 0
230 && e->symtree->n.sym->attr.flavor == FL_PARAMETER)
231 gfc_simplify_expr (e, 1);
233 if (e->expr_type != EXPR_ARRAY || !gfc_is_constant_expr (e))
234 return false;
236 /* A non-zero-sized constant array shall have a non-empty constructor. */
237 if (e->rank > 0 && e->shape != NULL && e->value.constructor == NULL)
239 mpz_init_set_ui (size, 1);
240 for (int j = 0; j < e->rank; j++)
241 mpz_mul (size, size, e->shape[j]);
242 bool not_size0 = (mpz_cmp_si (size, 0) != 0);
243 mpz_clear (size);
244 if (not_size0)
245 return false;
248 for (c = gfc_constructor_first (e->value.constructor);
249 c; c = gfc_constructor_next (c))
250 if (c->expr->expr_type != EXPR_CONSTANT
251 && c->expr->expr_type != EXPR_STRUCTURE)
253 array_OK = false;
254 break;
257 /* Check and expand the constructor. We do this when either
258 gfc_init_expr_flag is set or for not too large array constructors. */
259 bool expand;
260 expand = (e->rank == 1
261 && e->shape
262 && (mpz_cmp_ui (e->shape[0], flag_max_array_constructor) < 0));
264 if (!array_OK && (gfc_init_expr_flag || expand) && e->rank == 1)
266 bool saved_init_expr_flag = gfc_init_expr_flag;
267 array_OK = gfc_reduce_init_expr (e);
268 /* gfc_reduce_init_expr resets the flag. */
269 gfc_init_expr_flag = saved_init_expr_flag;
271 else
272 return array_OK;
274 /* Recheck to make sure that any EXPR_ARRAYs have gone. */
275 for (c = gfc_constructor_first (e->value.constructor);
276 c; c = gfc_constructor_next (c))
277 if (c->expr->expr_type != EXPR_CONSTANT
278 && c->expr->expr_type != EXPR_STRUCTURE)
279 return false;
281 /* Make sure that the array has a valid shape. */
282 if (e->shape == NULL && e->rank == 1)
284 if (!gfc_array_size(e, &size))
285 return false;
286 e->shape = gfc_get_shape (1);
287 mpz_init_set (e->shape[0], size);
288 mpz_clear (size);
291 return array_OK;
294 bool
295 gfc_is_constant_array_expr (gfc_expr *e)
297 return is_constant_array_expr (e);
301 /* Test for a size zero array. */
302 bool
303 gfc_is_size_zero_array (gfc_expr *array)
306 if (array->rank == 0)
307 return false;
309 if (array->expr_type == EXPR_VARIABLE && array->rank > 0
310 && array->symtree->n.sym->attr.flavor == FL_PARAMETER
311 && array->shape != NULL)
313 for (int i = 0; i < array->rank; i++)
314 if (mpz_cmp_si (array->shape[i], 0) <= 0)
315 return true;
317 return false;
320 if (array->expr_type == EXPR_ARRAY)
321 return array->value.constructor == NULL;
323 return false;
327 /* Initialize a transformational result expression with a given value. */
329 static void
330 init_result_expr (gfc_expr *e, int init, gfc_expr *array)
332 if (e && e->expr_type == EXPR_ARRAY)
334 gfc_constructor *ctor = gfc_constructor_first (e->value.constructor);
335 while (ctor)
337 init_result_expr (ctor->expr, init, array);
338 ctor = gfc_constructor_next (ctor);
341 else if (e && e->expr_type == EXPR_CONSTANT)
343 int i = gfc_validate_kind (e->ts.type, e->ts.kind, false);
344 HOST_WIDE_INT length;
345 gfc_char_t *string;
347 switch (e->ts.type)
349 case BT_LOGICAL:
350 e->value.logical = (init ? 1 : 0);
351 break;
353 case BT_INTEGER:
354 if (init == INT_MIN)
355 mpz_set (e->value.integer, gfc_integer_kinds[i].min_int);
356 else if (init == INT_MAX)
357 mpz_set (e->value.integer, gfc_integer_kinds[i].huge);
358 else
359 mpz_set_si (e->value.integer, init);
360 break;
362 case BT_REAL:
363 if (init == INT_MIN)
365 mpfr_set (e->value.real, gfc_real_kinds[i].huge, GFC_RND_MODE);
366 mpfr_neg (e->value.real, e->value.real, GFC_RND_MODE);
368 else if (init == INT_MAX)
369 mpfr_set (e->value.real, gfc_real_kinds[i].huge, GFC_RND_MODE);
370 else
371 mpfr_set_si (e->value.real, init, GFC_RND_MODE);
372 break;
374 case BT_COMPLEX:
375 mpc_set_si (e->value.complex, init, GFC_MPC_RND_MODE);
376 break;
378 case BT_CHARACTER:
379 if (init == INT_MIN)
381 gfc_expr *len = gfc_simplify_len (array, NULL);
382 gfc_extract_hwi (len, &length);
383 string = gfc_get_wide_string (length + 1);
384 gfc_wide_memset (string, 0, length);
386 else if (init == INT_MAX)
388 gfc_expr *len = gfc_simplify_len (array, NULL);
389 gfc_extract_hwi (len, &length);
390 string = gfc_get_wide_string (length + 1);
391 gfc_wide_memset (string, 255, length);
393 else
395 length = 0;
396 string = gfc_get_wide_string (1);
399 string[length] = '\0';
400 e->value.character.length = length;
401 e->value.character.string = string;
402 break;
404 default:
405 gcc_unreachable();
408 else
409 gcc_unreachable();
413 /* Helper function for gfc_simplify_dot_product() and gfc_simplify_matmul;
414 if conj_a is true, the matrix_a is complex conjugated. */
416 static gfc_expr *
417 compute_dot_product (gfc_expr *matrix_a, int stride_a, int offset_a,
418 gfc_expr *matrix_b, int stride_b, int offset_b,
419 bool conj_a)
421 gfc_expr *result, *a, *b, *c;
423 /* Set result to an INTEGER(1) 0 for numeric types and .false. for
424 LOGICAL. Mixed-mode math in the loop will promote result to the
425 correct type and kind. */
426 if (matrix_a->ts.type == BT_LOGICAL)
427 result = gfc_get_logical_expr (gfc_default_logical_kind, NULL, false);
428 else
429 result = gfc_get_int_expr (1, NULL, 0);
430 result->where = matrix_a->where;
432 a = gfc_constructor_lookup_expr (matrix_a->value.constructor, offset_a);
433 b = gfc_constructor_lookup_expr (matrix_b->value.constructor, offset_b);
434 while (a && b)
436 /* Copying of expressions is required as operands are free'd
437 by the gfc_arith routines. */
438 switch (result->ts.type)
440 case BT_LOGICAL:
441 result = gfc_or (result,
442 gfc_and (gfc_copy_expr (a),
443 gfc_copy_expr (b)));
444 break;
446 case BT_INTEGER:
447 case BT_REAL:
448 case BT_COMPLEX:
449 if (conj_a && a->ts.type == BT_COMPLEX)
450 c = gfc_simplify_conjg (a);
451 else
452 c = gfc_copy_expr (a);
453 result = gfc_add (result, gfc_multiply (c, gfc_copy_expr (b)));
454 break;
456 default:
457 gcc_unreachable();
460 offset_a += stride_a;
461 a = gfc_constructor_lookup_expr (matrix_a->value.constructor, offset_a);
463 offset_b += stride_b;
464 b = gfc_constructor_lookup_expr (matrix_b->value.constructor, offset_b);
467 return result;
471 /* Build a result expression for transformational intrinsics,
472 depending on DIM. */
474 static gfc_expr *
475 transformational_result (gfc_expr *array, gfc_expr *dim, bt type,
476 int kind, locus* where)
478 gfc_expr *result;
479 int i, nelem;
481 if (!dim || array->rank == 1)
482 return gfc_get_constant_expr (type, kind, where);
484 result = gfc_get_array_expr (type, kind, where);
485 result->shape = gfc_copy_shape_excluding (array->shape, array->rank, dim);
486 result->rank = array->rank - 1;
488 /* gfc_array_size() would count the number of elements in the constructor,
489 we have not built those yet. */
490 nelem = 1;
491 for (i = 0; i < result->rank; ++i)
492 nelem *= mpz_get_ui (result->shape[i]);
494 for (i = 0; i < nelem; ++i)
496 gfc_constructor_append_expr (&result->value.constructor,
497 gfc_get_constant_expr (type, kind, where),
498 NULL);
501 return result;
505 typedef gfc_expr* (*transformational_op)(gfc_expr*, gfc_expr*);
507 /* Wrapper function, implements 'op1 += 1'. Only called if MASK
508 of COUNT intrinsic is .TRUE..
510 Interface and implementation mimics arith functions as
511 gfc_add, gfc_multiply, etc. */
513 static gfc_expr *
514 gfc_count (gfc_expr *op1, gfc_expr *op2)
516 gfc_expr *result;
518 gcc_assert (op1->ts.type == BT_INTEGER);
519 gcc_assert (op2->ts.type == BT_LOGICAL);
520 gcc_assert (op2->value.logical);
522 result = gfc_copy_expr (op1);
523 mpz_add_ui (result->value.integer, result->value.integer, 1);
525 gfc_free_expr (op1);
526 gfc_free_expr (op2);
527 return result;
531 /* Transforms an ARRAY with operation OP, according to MASK, to a
532 scalar RESULT. E.g. called if
534 REAL, PARAMETER :: array(n, m) = ...
535 REAL, PARAMETER :: s = SUM(array)
537 where OP == gfc_add(). */
539 static gfc_expr *
540 simplify_transformation_to_scalar (gfc_expr *result, gfc_expr *array, gfc_expr *mask,
541 transformational_op op)
543 gfc_expr *a, *m;
544 gfc_constructor *array_ctor, *mask_ctor;
546 /* Shortcut for constant .FALSE. MASK. */
547 if (mask
548 && mask->expr_type == EXPR_CONSTANT
549 && !mask->value.logical)
550 return result;
552 array_ctor = gfc_constructor_first (array->value.constructor);
553 mask_ctor = NULL;
554 if (mask && mask->expr_type == EXPR_ARRAY)
555 mask_ctor = gfc_constructor_first (mask->value.constructor);
557 while (array_ctor)
559 a = array_ctor->expr;
560 array_ctor = gfc_constructor_next (array_ctor);
562 /* A constant MASK equals .TRUE. here and can be ignored. */
563 if (mask_ctor)
565 m = mask_ctor->expr;
566 mask_ctor = gfc_constructor_next (mask_ctor);
567 if (!m->value.logical)
568 continue;
571 result = op (result, gfc_copy_expr (a));
572 if (!result)
573 return result;
576 return result;
579 /* Transforms an ARRAY with operation OP, according to MASK, to an
580 array RESULT. E.g. called if
582 REAL, PARAMETER :: array(n, m) = ...
583 REAL, PARAMETER :: s(n) = PROD(array, DIM=1)
585 where OP == gfc_multiply().
586 The result might be post processed using post_op. */
588 static gfc_expr *
589 simplify_transformation_to_array (gfc_expr *result, gfc_expr *array, gfc_expr *dim,
590 gfc_expr *mask, transformational_op op,
591 transformational_op post_op)
593 mpz_t size;
594 int done, i, n, arraysize, resultsize, dim_index, dim_extent, dim_stride;
595 gfc_expr **arrayvec, **resultvec, **base, **src, **dest;
596 gfc_constructor *array_ctor, *mask_ctor, *result_ctor;
598 int count[GFC_MAX_DIMENSIONS], extent[GFC_MAX_DIMENSIONS],
599 sstride[GFC_MAX_DIMENSIONS], dstride[GFC_MAX_DIMENSIONS],
600 tmpstride[GFC_MAX_DIMENSIONS];
602 /* Shortcut for constant .FALSE. MASK. */
603 if (mask
604 && mask->expr_type == EXPR_CONSTANT
605 && !mask->value.logical)
606 return result;
608 /* Build an indexed table for array element expressions to minimize
609 linked-list traversal. Masked elements are set to NULL. */
610 gfc_array_size (array, &size);
611 arraysize = mpz_get_ui (size);
612 mpz_clear (size);
614 arrayvec = XCNEWVEC (gfc_expr*, arraysize);
616 array_ctor = gfc_constructor_first (array->value.constructor);
617 mask_ctor = NULL;
618 if (mask && mask->expr_type == EXPR_ARRAY)
619 mask_ctor = gfc_constructor_first (mask->value.constructor);
621 for (i = 0; i < arraysize; ++i)
623 arrayvec[i] = array_ctor->expr;
624 array_ctor = gfc_constructor_next (array_ctor);
626 if (mask_ctor)
628 if (!mask_ctor->expr->value.logical)
629 arrayvec[i] = NULL;
631 mask_ctor = gfc_constructor_next (mask_ctor);
635 /* Same for the result expression. */
636 gfc_array_size (result, &size);
637 resultsize = mpz_get_ui (size);
638 mpz_clear (size);
640 resultvec = XCNEWVEC (gfc_expr*, resultsize);
641 result_ctor = gfc_constructor_first (result->value.constructor);
642 for (i = 0; i < resultsize; ++i)
644 resultvec[i] = result_ctor->expr;
645 result_ctor = gfc_constructor_next (result_ctor);
648 gfc_extract_int (dim, &dim_index);
649 dim_index -= 1; /* zero-base index */
650 dim_extent = 0;
651 dim_stride = 0;
653 for (i = 0, n = 0; i < array->rank; ++i)
655 count[i] = 0;
656 tmpstride[i] = (i == 0) ? 1 : tmpstride[i-1] * mpz_get_si (array->shape[i-1]);
657 if (i == dim_index)
659 dim_extent = mpz_get_si (array->shape[i]);
660 dim_stride = tmpstride[i];
661 continue;
664 extent[n] = mpz_get_si (array->shape[i]);
665 sstride[n] = tmpstride[i];
666 dstride[n] = (n == 0) ? 1 : dstride[n-1] * extent[n-1];
667 n += 1;
670 done = resultsize <= 0;
671 base = arrayvec;
672 dest = resultvec;
673 while (!done)
675 for (src = base, n = 0; n < dim_extent; src += dim_stride, ++n)
676 if (*src)
677 *dest = op (*dest, gfc_copy_expr (*src));
679 if (post_op)
680 *dest = post_op (*dest, *dest);
682 count[0]++;
683 base += sstride[0];
684 dest += dstride[0];
686 n = 0;
687 while (!done && count[n] == extent[n])
689 count[n] = 0;
690 base -= sstride[n] * extent[n];
691 dest -= dstride[n] * extent[n];
693 n++;
694 if (n < result->rank)
696 /* If the nested loop is unrolled GFC_MAX_DIMENSIONS
697 times, we'd warn for the last iteration, because the
698 array index will have already been incremented to the
699 array sizes, and we can't tell that this must make
700 the test against result->rank false, because ranks
701 must not exceed GFC_MAX_DIMENSIONS. */
702 GCC_DIAGNOSTIC_PUSH_IGNORED (-Warray-bounds)
703 count[n]++;
704 base += sstride[n];
705 dest += dstride[n];
706 GCC_DIAGNOSTIC_POP
708 else
709 done = true;
713 /* Place updated expression in result constructor. */
714 result_ctor = gfc_constructor_first (result->value.constructor);
715 for (i = 0; i < resultsize; ++i)
717 result_ctor->expr = resultvec[i];
718 result_ctor = gfc_constructor_next (result_ctor);
721 free (arrayvec);
722 free (resultvec);
723 return result;
727 static gfc_expr *
728 simplify_transformation (gfc_expr *array, gfc_expr *dim, gfc_expr *mask,
729 int init_val, transformational_op op)
731 gfc_expr *result;
732 bool size_zero;
734 size_zero = gfc_is_size_zero_array (array);
736 if (!(is_constant_array_expr (array) || size_zero)
737 || array->shape == NULL
738 || !gfc_is_constant_expr (dim))
739 return NULL;
741 if (mask
742 && !is_constant_array_expr (mask)
743 && mask->expr_type != EXPR_CONSTANT)
744 return NULL;
746 result = transformational_result (array, dim, array->ts.type,
747 array->ts.kind, &array->where);
748 init_result_expr (result, init_val, array);
750 if (size_zero)
751 return result;
753 return !dim || array->rank == 1 ?
754 simplify_transformation_to_scalar (result, array, mask, op) :
755 simplify_transformation_to_array (result, array, dim, mask, op, NULL);
759 /********************** Simplification functions *****************************/
761 gfc_expr *
762 gfc_simplify_abs (gfc_expr *e)
764 gfc_expr *result;
766 if (e->expr_type != EXPR_CONSTANT)
767 return NULL;
769 switch (e->ts.type)
771 case BT_INTEGER:
772 result = gfc_get_constant_expr (BT_INTEGER, e->ts.kind, &e->where);
773 mpz_abs (result->value.integer, e->value.integer);
774 return range_check (result, "IABS");
776 case BT_REAL:
777 result = gfc_get_constant_expr (BT_REAL, e->ts.kind, &e->where);
778 mpfr_abs (result->value.real, e->value.real, GFC_RND_MODE);
779 return range_check (result, "ABS");
781 case BT_COMPLEX:
782 gfc_set_model_kind (e->ts.kind);
783 result = gfc_get_constant_expr (BT_REAL, e->ts.kind, &e->where);
784 mpc_abs (result->value.real, e->value.complex, GFC_RND_MODE);
785 return range_check (result, "CABS");
787 default:
788 gfc_internal_error ("gfc_simplify_abs(): Bad type");
793 static gfc_expr *
794 simplify_achar_char (gfc_expr *e, gfc_expr *k, const char *name, bool ascii)
796 gfc_expr *result;
797 int kind;
798 bool too_large = false;
800 if (e->expr_type != EXPR_CONSTANT)
801 return NULL;
803 kind = get_kind (BT_CHARACTER, k, name, gfc_default_character_kind);
804 if (kind == -1)
805 return &gfc_bad_expr;
807 if (mpz_cmp_si (e->value.integer, 0) < 0)
809 gfc_error ("Argument of %s function at %L is negative", name,
810 &e->where);
811 return &gfc_bad_expr;
814 if (ascii && warn_surprising && mpz_cmp_si (e->value.integer, 127) > 0)
815 gfc_warning (OPT_Wsurprising,
816 "Argument of %s function at %L outside of range [0,127]",
817 name, &e->where);
819 if (kind == 1 && mpz_cmp_si (e->value.integer, 255) > 0)
820 too_large = true;
821 else if (kind == 4)
823 mpz_t t;
824 mpz_init_set_ui (t, 2);
825 mpz_pow_ui (t, t, 32);
826 mpz_sub_ui (t, t, 1);
827 if (mpz_cmp (e->value.integer, t) > 0)
828 too_large = true;
829 mpz_clear (t);
832 if (too_large)
834 gfc_error ("Argument of %s function at %L is too large for the "
835 "collating sequence of kind %d", name, &e->where, kind);
836 return &gfc_bad_expr;
839 result = gfc_get_character_expr (kind, &e->where, NULL, 1);
840 result->value.character.string[0] = mpz_get_ui (e->value.integer);
842 return result;
847 /* We use the processor's collating sequence, because all
848 systems that gfortran currently works on are ASCII. */
850 gfc_expr *
851 gfc_simplify_achar (gfc_expr *e, gfc_expr *k)
853 return simplify_achar_char (e, k, "ACHAR", true);
857 gfc_expr *
858 gfc_simplify_acos (gfc_expr *x)
860 gfc_expr *result;
862 if (x->expr_type != EXPR_CONSTANT)
863 return NULL;
865 switch (x->ts.type)
867 case BT_REAL:
868 if (mpfr_cmp_si (x->value.real, 1) > 0
869 || mpfr_cmp_si (x->value.real, -1) < 0)
871 gfc_error ("Argument of ACOS at %L must be between -1 and 1",
872 &x->where);
873 return &gfc_bad_expr;
875 result = gfc_get_constant_expr (x->ts.type, x->ts.kind, &x->where);
876 mpfr_acos (result->value.real, x->value.real, GFC_RND_MODE);
877 break;
879 case BT_COMPLEX:
880 result = gfc_get_constant_expr (x->ts.type, x->ts.kind, &x->where);
881 mpc_acos (result->value.complex, x->value.complex, GFC_MPC_RND_MODE);
882 break;
884 default:
885 gfc_internal_error ("in gfc_simplify_acos(): Bad type");
888 return range_check (result, "ACOS");
891 gfc_expr *
892 gfc_simplify_acosh (gfc_expr *x)
894 gfc_expr *result;
896 if (x->expr_type != EXPR_CONSTANT)
897 return NULL;
899 switch (x->ts.type)
901 case BT_REAL:
902 if (mpfr_cmp_si (x->value.real, 1) < 0)
904 gfc_error ("Argument of ACOSH at %L must not be less than 1",
905 &x->where);
906 return &gfc_bad_expr;
909 result = gfc_get_constant_expr (x->ts.type, x->ts.kind, &x->where);
910 mpfr_acosh (result->value.real, x->value.real, GFC_RND_MODE);
911 break;
913 case BT_COMPLEX:
914 result = gfc_get_constant_expr (x->ts.type, x->ts.kind, &x->where);
915 mpc_acosh (result->value.complex, x->value.complex, GFC_MPC_RND_MODE);
916 break;
918 default:
919 gfc_internal_error ("in gfc_simplify_acosh(): Bad type");
922 return range_check (result, "ACOSH");
925 gfc_expr *
926 gfc_simplify_adjustl (gfc_expr *e)
928 gfc_expr *result;
929 int count, i, len;
930 gfc_char_t ch;
932 if (e->expr_type != EXPR_CONSTANT)
933 return NULL;
935 len = e->value.character.length;
937 for (count = 0, i = 0; i < len; ++i)
939 ch = e->value.character.string[i];
940 if (ch != ' ')
941 break;
942 ++count;
945 result = gfc_get_character_expr (e->ts.kind, &e->where, NULL, len);
946 for (i = 0; i < len - count; ++i)
947 result->value.character.string[i] = e->value.character.string[count + i];
949 return result;
953 gfc_expr *
954 gfc_simplify_adjustr (gfc_expr *e)
956 gfc_expr *result;
957 int count, i, len;
958 gfc_char_t ch;
960 if (e->expr_type != EXPR_CONSTANT)
961 return NULL;
963 len = e->value.character.length;
965 for (count = 0, i = len - 1; i >= 0; --i)
967 ch = e->value.character.string[i];
968 if (ch != ' ')
969 break;
970 ++count;
973 result = gfc_get_character_expr (e->ts.kind, &e->where, NULL, len);
974 for (i = 0; i < count; ++i)
975 result->value.character.string[i] = ' ';
977 for (i = count; i < len; ++i)
978 result->value.character.string[i] = e->value.character.string[i - count];
980 return result;
984 gfc_expr *
985 gfc_simplify_aimag (gfc_expr *e)
987 gfc_expr *result;
989 if (e->expr_type != EXPR_CONSTANT)
990 return NULL;
992 result = gfc_get_constant_expr (BT_REAL, e->ts.kind, &e->where);
993 mpfr_set (result->value.real, mpc_imagref (e->value.complex), GFC_RND_MODE);
995 return range_check (result, "AIMAG");
999 gfc_expr *
1000 gfc_simplify_aint (gfc_expr *e, gfc_expr *k)
1002 gfc_expr *rtrunc, *result;
1003 int kind;
1005 kind = get_kind (BT_REAL, k, "AINT", e->ts.kind);
1006 if (kind == -1)
1007 return &gfc_bad_expr;
1009 if (e->expr_type != EXPR_CONSTANT)
1010 return NULL;
1012 rtrunc = gfc_copy_expr (e);
1013 mpfr_trunc (rtrunc->value.real, e->value.real);
1015 result = gfc_real2real (rtrunc, kind);
1017 gfc_free_expr (rtrunc);
1019 return range_check (result, "AINT");
1023 gfc_expr *
1024 gfc_simplify_all (gfc_expr *mask, gfc_expr *dim)
1026 return simplify_transformation (mask, dim, NULL, true, gfc_and);
1030 gfc_expr *
1031 gfc_simplify_dint (gfc_expr *e)
1033 gfc_expr *rtrunc, *result;
1035 if (e->expr_type != EXPR_CONSTANT)
1036 return NULL;
1038 rtrunc = gfc_copy_expr (e);
1039 mpfr_trunc (rtrunc->value.real, e->value.real);
1041 result = gfc_real2real (rtrunc, gfc_default_double_kind);
1043 gfc_free_expr (rtrunc);
1045 return range_check (result, "DINT");
1049 gfc_expr *
1050 gfc_simplify_dreal (gfc_expr *e)
1052 gfc_expr *result = NULL;
1054 if (e->expr_type != EXPR_CONSTANT)
1055 return NULL;
1057 result = gfc_get_constant_expr (BT_REAL, e->ts.kind, &e->where);
1058 mpc_real (result->value.real, e->value.complex, GFC_RND_MODE);
1060 return range_check (result, "DREAL");
1064 gfc_expr *
1065 gfc_simplify_anint (gfc_expr *e, gfc_expr *k)
1067 gfc_expr *result;
1068 int kind;
1070 kind = get_kind (BT_REAL, k, "ANINT", e->ts.kind);
1071 if (kind == -1)
1072 return &gfc_bad_expr;
1074 if (e->expr_type != EXPR_CONSTANT)
1075 return NULL;
1077 result = gfc_get_constant_expr (e->ts.type, kind, &e->where);
1078 mpfr_round (result->value.real, e->value.real);
1080 return range_check (result, "ANINT");
1084 gfc_expr *
1085 gfc_simplify_and (gfc_expr *x, gfc_expr *y)
1087 gfc_expr *result;
1088 int kind;
1090 if (x->expr_type != EXPR_CONSTANT || y->expr_type != EXPR_CONSTANT)
1091 return NULL;
1093 kind = x->ts.kind > y->ts.kind ? x->ts.kind : y->ts.kind;
1095 switch (x->ts.type)
1097 case BT_INTEGER:
1098 result = gfc_get_constant_expr (BT_INTEGER, kind, &x->where);
1099 mpz_and (result->value.integer, x->value.integer, y->value.integer);
1100 return range_check (result, "AND");
1102 case BT_LOGICAL:
1103 return gfc_get_logical_expr (kind, &x->where,
1104 x->value.logical && y->value.logical);
1106 default:
1107 gcc_unreachable ();
1112 gfc_expr *
1113 gfc_simplify_any (gfc_expr *mask, gfc_expr *dim)
1115 return simplify_transformation (mask, dim, NULL, false, gfc_or);
1119 gfc_expr *
1120 gfc_simplify_dnint (gfc_expr *e)
1122 gfc_expr *result;
1124 if (e->expr_type != EXPR_CONSTANT)
1125 return NULL;
1127 result = gfc_get_constant_expr (BT_REAL, gfc_default_double_kind, &e->where);
1128 mpfr_round (result->value.real, e->value.real);
1130 return range_check (result, "DNINT");
1134 gfc_expr *
1135 gfc_simplify_asin (gfc_expr *x)
1137 gfc_expr *result;
1139 if (x->expr_type != EXPR_CONSTANT)
1140 return NULL;
1142 switch (x->ts.type)
1144 case BT_REAL:
1145 if (mpfr_cmp_si (x->value.real, 1) > 0
1146 || mpfr_cmp_si (x->value.real, -1) < 0)
1148 gfc_error ("Argument of ASIN at %L must be between -1 and 1",
1149 &x->where);
1150 return &gfc_bad_expr;
1152 result = gfc_get_constant_expr (x->ts.type, x->ts.kind, &x->where);
1153 mpfr_asin (result->value.real, x->value.real, GFC_RND_MODE);
1154 break;
1156 case BT_COMPLEX:
1157 result = gfc_get_constant_expr (x->ts.type, x->ts.kind, &x->where);
1158 mpc_asin (result->value.complex, x->value.complex, GFC_MPC_RND_MODE);
1159 break;
1161 default:
1162 gfc_internal_error ("in gfc_simplify_asin(): Bad type");
1165 return range_check (result, "ASIN");
1169 /* Convert radians to degrees, i.e., x * 180 / pi. */
1171 static void
1172 rad2deg (mpfr_t x)
1174 mpfr_t tmp;
1176 mpfr_init (tmp);
1177 mpfr_const_pi (tmp, GFC_RND_MODE);
1178 mpfr_mul_ui (x, x, 180, GFC_RND_MODE);
1179 mpfr_div (x, x, tmp, GFC_RND_MODE);
1180 mpfr_clear (tmp);
1184 /* Simplify ACOSD(X) where the returned value has units of degree. */
1186 gfc_expr *
1187 gfc_simplify_acosd (gfc_expr *x)
1189 gfc_expr *result;
1191 if (x->expr_type != EXPR_CONSTANT)
1192 return NULL;
1194 if (mpfr_cmp_si (x->value.real, 1) > 0
1195 || mpfr_cmp_si (x->value.real, -1) < 0)
1197 gfc_error ("Argument of ACOSD at %L must be between -1 and 1",
1198 &x->where);
1199 return &gfc_bad_expr;
1202 result = gfc_get_constant_expr (x->ts.type, x->ts.kind, &x->where);
1203 mpfr_acos (result->value.real, x->value.real, GFC_RND_MODE);
1204 rad2deg (result->value.real);
1206 return range_check (result, "ACOSD");
1210 /* Simplify asind (x) where the returned value has units of degree. */
1212 gfc_expr *
1213 gfc_simplify_asind (gfc_expr *x)
1215 gfc_expr *result;
1217 if (x->expr_type != EXPR_CONSTANT)
1218 return NULL;
1220 if (mpfr_cmp_si (x->value.real, 1) > 0
1221 || mpfr_cmp_si (x->value.real, -1) < 0)
1223 gfc_error ("Argument of ASIND at %L must be between -1 and 1",
1224 &x->where);
1225 return &gfc_bad_expr;
1228 result = gfc_get_constant_expr (x->ts.type, x->ts.kind, &x->where);
1229 mpfr_asin (result->value.real, x->value.real, GFC_RND_MODE);
1230 rad2deg (result->value.real);
1232 return range_check (result, "ASIND");
1236 /* Simplify atand (x) where the returned value has units of degree. */
1238 gfc_expr *
1239 gfc_simplify_atand (gfc_expr *x)
1241 gfc_expr *result;
1243 if (x->expr_type != EXPR_CONSTANT)
1244 return NULL;
1246 result = gfc_get_constant_expr (x->ts.type, x->ts.kind, &x->where);
1247 mpfr_atan (result->value.real, x->value.real, GFC_RND_MODE);
1248 rad2deg (result->value.real);
1250 return range_check (result, "ATAND");
1254 gfc_expr *
1255 gfc_simplify_asinh (gfc_expr *x)
1257 gfc_expr *result;
1259 if (x->expr_type != EXPR_CONSTANT)
1260 return NULL;
1262 result = gfc_get_constant_expr (x->ts.type, x->ts.kind, &x->where);
1264 switch (x->ts.type)
1266 case BT_REAL:
1267 mpfr_asinh (result->value.real, x->value.real, GFC_RND_MODE);
1268 break;
1270 case BT_COMPLEX:
1271 mpc_asinh (result->value.complex, x->value.complex, GFC_MPC_RND_MODE);
1272 break;
1274 default:
1275 gfc_internal_error ("in gfc_simplify_asinh(): Bad type");
1278 return range_check (result, "ASINH");
1282 gfc_expr *
1283 gfc_simplify_atan (gfc_expr *x)
1285 gfc_expr *result;
1287 if (x->expr_type != EXPR_CONSTANT)
1288 return NULL;
1290 result = gfc_get_constant_expr (x->ts.type, x->ts.kind, &x->where);
1292 switch (x->ts.type)
1294 case BT_REAL:
1295 mpfr_atan (result->value.real, x->value.real, GFC_RND_MODE);
1296 break;
1298 case BT_COMPLEX:
1299 mpc_atan (result->value.complex, x->value.complex, GFC_MPC_RND_MODE);
1300 break;
1302 default:
1303 gfc_internal_error ("in gfc_simplify_atan(): Bad type");
1306 return range_check (result, "ATAN");
1310 gfc_expr *
1311 gfc_simplify_atanh (gfc_expr *x)
1313 gfc_expr *result;
1315 if (x->expr_type != EXPR_CONSTANT)
1316 return NULL;
1318 switch (x->ts.type)
1320 case BT_REAL:
1321 if (mpfr_cmp_si (x->value.real, 1) >= 0
1322 || mpfr_cmp_si (x->value.real, -1) <= 0)
1324 gfc_error ("Argument of ATANH at %L must be inside the range -1 "
1325 "to 1", &x->where);
1326 return &gfc_bad_expr;
1328 result = gfc_get_constant_expr (x->ts.type, x->ts.kind, &x->where);
1329 mpfr_atanh (result->value.real, x->value.real, GFC_RND_MODE);
1330 break;
1332 case BT_COMPLEX:
1333 result = gfc_get_constant_expr (x->ts.type, x->ts.kind, &x->where);
1334 mpc_atanh (result->value.complex, x->value.complex, GFC_MPC_RND_MODE);
1335 break;
1337 default:
1338 gfc_internal_error ("in gfc_simplify_atanh(): Bad type");
1341 return range_check (result, "ATANH");
1345 gfc_expr *
1346 gfc_simplify_atan2 (gfc_expr *y, gfc_expr *x)
1348 gfc_expr *result;
1350 if (x->expr_type != EXPR_CONSTANT || y->expr_type != EXPR_CONSTANT)
1351 return NULL;
1353 if (mpfr_zero_p (y->value.real) && mpfr_zero_p (x->value.real))
1355 gfc_error ("If first argument of ATAN2 at %L is zero, then the "
1356 "second argument must not be zero", &y->where);
1357 return &gfc_bad_expr;
1360 result = gfc_get_constant_expr (x->ts.type, x->ts.kind, &x->where);
1361 mpfr_atan2 (result->value.real, y->value.real, x->value.real, GFC_RND_MODE);
1363 return range_check (result, "ATAN2");
1367 gfc_expr *
1368 gfc_simplify_bessel_j0 (gfc_expr *x)
1370 gfc_expr *result;
1372 if (x->expr_type != EXPR_CONSTANT)
1373 return NULL;
1375 result = gfc_get_constant_expr (x->ts.type, x->ts.kind, &x->where);
1376 mpfr_j0 (result->value.real, x->value.real, GFC_RND_MODE);
1378 return range_check (result, "BESSEL_J0");
1382 gfc_expr *
1383 gfc_simplify_bessel_j1 (gfc_expr *x)
1385 gfc_expr *result;
1387 if (x->expr_type != EXPR_CONSTANT)
1388 return NULL;
1390 result = gfc_get_constant_expr (x->ts.type, x->ts.kind, &x->where);
1391 mpfr_j1 (result->value.real, x->value.real, GFC_RND_MODE);
1393 return range_check (result, "BESSEL_J1");
1397 gfc_expr *
1398 gfc_simplify_bessel_jn (gfc_expr *order, gfc_expr *x)
1400 gfc_expr *result;
1401 long n;
1403 if (x->expr_type != EXPR_CONSTANT || order->expr_type != EXPR_CONSTANT)
1404 return NULL;
1406 n = mpz_get_si (order->value.integer);
1407 result = gfc_get_constant_expr (x->ts.type, x->ts.kind, &x->where);
1408 mpfr_jn (result->value.real, n, x->value.real, GFC_RND_MODE);
1410 return range_check (result, "BESSEL_JN");
1414 /* Simplify transformational form of JN and YN. */
1416 static gfc_expr *
1417 gfc_simplify_bessel_n2 (gfc_expr *order1, gfc_expr *order2, gfc_expr *x,
1418 bool jn)
1420 gfc_expr *result;
1421 gfc_expr *e;
1422 long n1, n2;
1423 int i;
1424 mpfr_t x2rev, last1, last2;
1426 if (x->expr_type != EXPR_CONSTANT || order1->expr_type != EXPR_CONSTANT
1427 || order2->expr_type != EXPR_CONSTANT)
1428 return NULL;
1430 n1 = mpz_get_si (order1->value.integer);
1431 n2 = mpz_get_si (order2->value.integer);
1432 result = gfc_get_array_expr (x->ts.type, x->ts.kind, &x->where);
1433 result->rank = 1;
1434 result->shape = gfc_get_shape (1);
1435 mpz_init_set_ui (result->shape[0], MAX (n2-n1+1, 0));
1437 if (n2 < n1)
1438 return result;
1440 /* Special case: x == 0; it is J0(0.0) == 1, JN(N > 0, 0.0) == 0; and
1441 YN(N, 0.0) = -Inf. */
1443 if (mpfr_cmp_ui (x->value.real, 0.0) == 0)
1445 if (!jn && flag_range_check)
1447 gfc_error ("Result of BESSEL_YN is -INF at %L", &result->where);
1448 gfc_free_expr (result);
1449 return &gfc_bad_expr;
1452 if (jn && n1 == 0)
1454 e = gfc_get_constant_expr (x->ts.type, x->ts.kind, &x->where);
1455 mpfr_set_ui (e->value.real, 1, GFC_RND_MODE);
1456 gfc_constructor_append_expr (&result->value.constructor, e,
1457 &x->where);
1458 n1++;
1461 for (i = n1; i <= n2; i++)
1463 e = gfc_get_constant_expr (x->ts.type, x->ts.kind, &x->where);
1464 if (jn)
1465 mpfr_set_ui (e->value.real, 0, GFC_RND_MODE);
1466 else
1467 mpfr_set_inf (e->value.real, -1);
1468 gfc_constructor_append_expr (&result->value.constructor, e,
1469 &x->where);
1472 return result;
1475 /* Use the faster but more verbose recurrence algorithm. Bessel functions
1476 are stable for downward recursion and Neumann functions are stable
1477 for upward recursion. It is
1478 x2rev = 2.0/x,
1479 J(N-1, x) = x2rev * N * J(N, x) - J(N+1, x),
1480 Y(N+1, x) = x2rev * N * Y(N, x) - Y(N-1, x).
1481 Cf. http://dlmf.nist.gov/10.74#iv and http://dlmf.nist.gov/10.6#E1 */
1483 gfc_set_model_kind (x->ts.kind);
1485 /* Get first recursion anchor. */
1487 mpfr_init (last1);
1488 if (jn)
1489 mpfr_jn (last1, n2, x->value.real, GFC_RND_MODE);
1490 else
1491 mpfr_yn (last1, n1, x->value.real, GFC_RND_MODE);
1493 e = gfc_get_constant_expr (x->ts.type, x->ts.kind, &x->where);
1494 mpfr_set (e->value.real, last1, GFC_RND_MODE);
1495 if (range_check (e, jn ? "BESSEL_JN" : "BESSEL_YN") == &gfc_bad_expr)
1497 mpfr_clear (last1);
1498 gfc_free_expr (e);
1499 gfc_free_expr (result);
1500 return &gfc_bad_expr;
1502 gfc_constructor_append_expr (&result->value.constructor, e, &x->where);
1504 if (n1 == n2)
1506 mpfr_clear (last1);
1507 return result;
1510 /* Get second recursion anchor. */
1512 mpfr_init (last2);
1513 if (jn)
1514 mpfr_jn (last2, n2-1, x->value.real, GFC_RND_MODE);
1515 else
1516 mpfr_yn (last2, n1+1, x->value.real, GFC_RND_MODE);
1518 e = gfc_get_constant_expr (x->ts.type, x->ts.kind, &x->where);
1519 mpfr_set (e->value.real, last2, GFC_RND_MODE);
1520 if (range_check (e, jn ? "BESSEL_JN" : "BESSEL_YN") == &gfc_bad_expr)
1522 mpfr_clear (last1);
1523 mpfr_clear (last2);
1524 gfc_free_expr (e);
1525 gfc_free_expr (result);
1526 return &gfc_bad_expr;
1528 if (jn)
1529 gfc_constructor_insert_expr (&result->value.constructor, e, &x->where, -2);
1530 else
1531 gfc_constructor_append_expr (&result->value.constructor, e, &x->where);
1533 if (n1 + 1 == n2)
1535 mpfr_clear (last1);
1536 mpfr_clear (last2);
1537 return result;
1540 /* Start actual recursion. */
1542 mpfr_init (x2rev);
1543 mpfr_ui_div (x2rev, 2, x->value.real, GFC_RND_MODE);
1545 for (i = 2; i <= n2-n1; i++)
1547 e = gfc_get_constant_expr (x->ts.type, x->ts.kind, &x->where);
1549 /* Special case: For YN, if the previous N gave -INF, set
1550 also N+1 to -INF. */
1551 if (!jn && !flag_range_check && mpfr_inf_p (last2))
1553 mpfr_set_inf (e->value.real, -1);
1554 gfc_constructor_append_expr (&result->value.constructor, e,
1555 &x->where);
1556 continue;
1559 mpfr_mul_si (e->value.real, x2rev, jn ? (n2-i+1) : (n1+i-1),
1560 GFC_RND_MODE);
1561 mpfr_mul (e->value.real, e->value.real, last2, GFC_RND_MODE);
1562 mpfr_sub (e->value.real, e->value.real, last1, GFC_RND_MODE);
1564 if (range_check (e, jn ? "BESSEL_JN" : "BESSEL_YN") == &gfc_bad_expr)
1566 /* Range_check frees "e" in that case. */
1567 e = NULL;
1568 goto error;
1571 if (jn)
1572 gfc_constructor_insert_expr (&result->value.constructor, e, &x->where,
1573 -i-1);
1574 else
1575 gfc_constructor_append_expr (&result->value.constructor, e, &x->where);
1577 mpfr_set (last1, last2, GFC_RND_MODE);
1578 mpfr_set (last2, e->value.real, GFC_RND_MODE);
1581 mpfr_clear (last1);
1582 mpfr_clear (last2);
1583 mpfr_clear (x2rev);
1584 return result;
1586 error:
1587 mpfr_clear (last1);
1588 mpfr_clear (last2);
1589 mpfr_clear (x2rev);
1590 gfc_free_expr (e);
1591 gfc_free_expr (result);
1592 return &gfc_bad_expr;
1596 gfc_expr *
1597 gfc_simplify_bessel_jn2 (gfc_expr *order1, gfc_expr *order2, gfc_expr *x)
1599 return gfc_simplify_bessel_n2 (order1, order2, x, true);
1603 gfc_expr *
1604 gfc_simplify_bessel_y0 (gfc_expr *x)
1606 gfc_expr *result;
1608 if (x->expr_type != EXPR_CONSTANT)
1609 return NULL;
1611 result = gfc_get_constant_expr (x->ts.type, x->ts.kind, &x->where);
1612 mpfr_y0 (result->value.real, x->value.real, GFC_RND_MODE);
1614 return range_check (result, "BESSEL_Y0");
1618 gfc_expr *
1619 gfc_simplify_bessel_y1 (gfc_expr *x)
1621 gfc_expr *result;
1623 if (x->expr_type != EXPR_CONSTANT)
1624 return NULL;
1626 result = gfc_get_constant_expr (x->ts.type, x->ts.kind, &x->where);
1627 mpfr_y1 (result->value.real, x->value.real, GFC_RND_MODE);
1629 return range_check (result, "BESSEL_Y1");
1633 gfc_expr *
1634 gfc_simplify_bessel_yn (gfc_expr *order, gfc_expr *x)
1636 gfc_expr *result;
1637 long n;
1639 if (x->expr_type != EXPR_CONSTANT || order->expr_type != EXPR_CONSTANT)
1640 return NULL;
1642 n = mpz_get_si (order->value.integer);
1643 result = gfc_get_constant_expr (x->ts.type, x->ts.kind, &x->where);
1644 mpfr_yn (result->value.real, n, x->value.real, GFC_RND_MODE);
1646 return range_check (result, "BESSEL_YN");
1650 gfc_expr *
1651 gfc_simplify_bessel_yn2 (gfc_expr *order1, gfc_expr *order2, gfc_expr *x)
1653 return gfc_simplify_bessel_n2 (order1, order2, x, false);
1657 gfc_expr *
1658 gfc_simplify_bit_size (gfc_expr *e)
1660 int i = gfc_validate_kind (e->ts.type, e->ts.kind, false);
1661 return gfc_get_int_expr (e->ts.kind, &e->where,
1662 gfc_integer_kinds[i].bit_size);
1666 gfc_expr *
1667 gfc_simplify_btest (gfc_expr *e, gfc_expr *bit)
1669 int b;
1671 if (e->expr_type != EXPR_CONSTANT || bit->expr_type != EXPR_CONSTANT)
1672 return NULL;
1674 if (!gfc_check_bitfcn (e, bit))
1675 return &gfc_bad_expr;
1677 if (gfc_extract_int (bit, &b) || b < 0)
1678 return gfc_get_logical_expr (gfc_default_logical_kind, &e->where, false);
1680 return gfc_get_logical_expr (gfc_default_logical_kind, &e->where,
1681 mpz_tstbit (e->value.integer, b));
1685 static int
1686 compare_bitwise (gfc_expr *i, gfc_expr *j)
1688 mpz_t x, y;
1689 int k, res;
1691 gcc_assert (i->ts.type == BT_INTEGER);
1692 gcc_assert (j->ts.type == BT_INTEGER);
1694 mpz_init_set (x, i->value.integer);
1695 k = gfc_validate_kind (i->ts.type, i->ts.kind, false);
1696 convert_mpz_to_unsigned (x, gfc_integer_kinds[k].bit_size);
1698 mpz_init_set (y, j->value.integer);
1699 k = gfc_validate_kind (j->ts.type, j->ts.kind, false);
1700 convert_mpz_to_unsigned (y, gfc_integer_kinds[k].bit_size);
1702 res = mpz_cmp (x, y);
1703 mpz_clear (x);
1704 mpz_clear (y);
1705 return res;
1709 gfc_expr *
1710 gfc_simplify_bge (gfc_expr *i, gfc_expr *j)
1712 if (i->expr_type != EXPR_CONSTANT || j->expr_type != EXPR_CONSTANT)
1713 return NULL;
1715 return gfc_get_logical_expr (gfc_default_logical_kind, &i->where,
1716 compare_bitwise (i, j) >= 0);
1720 gfc_expr *
1721 gfc_simplify_bgt (gfc_expr *i, gfc_expr *j)
1723 if (i->expr_type != EXPR_CONSTANT || j->expr_type != EXPR_CONSTANT)
1724 return NULL;
1726 return gfc_get_logical_expr (gfc_default_logical_kind, &i->where,
1727 compare_bitwise (i, j) > 0);
1731 gfc_expr *
1732 gfc_simplify_ble (gfc_expr *i, gfc_expr *j)
1734 if (i->expr_type != EXPR_CONSTANT || j->expr_type != EXPR_CONSTANT)
1735 return NULL;
1737 return gfc_get_logical_expr (gfc_default_logical_kind, &i->where,
1738 compare_bitwise (i, j) <= 0);
1742 gfc_expr *
1743 gfc_simplify_blt (gfc_expr *i, gfc_expr *j)
1745 if (i->expr_type != EXPR_CONSTANT || j->expr_type != EXPR_CONSTANT)
1746 return NULL;
1748 return gfc_get_logical_expr (gfc_default_logical_kind, &i->where,
1749 compare_bitwise (i, j) < 0);
1753 gfc_expr *
1754 gfc_simplify_ceiling (gfc_expr *e, gfc_expr *k)
1756 gfc_expr *ceil, *result;
1757 int kind;
1759 kind = get_kind (BT_INTEGER, k, "CEILING", gfc_default_integer_kind);
1760 if (kind == -1)
1761 return &gfc_bad_expr;
1763 if (e->expr_type != EXPR_CONSTANT)
1764 return NULL;
1766 ceil = gfc_copy_expr (e);
1767 mpfr_ceil (ceil->value.real, e->value.real);
1769 result = gfc_get_constant_expr (BT_INTEGER, kind, &e->where);
1770 gfc_mpfr_to_mpz (result->value.integer, ceil->value.real, &e->where);
1772 gfc_free_expr (ceil);
1774 return range_check (result, "CEILING");
1778 gfc_expr *
1779 gfc_simplify_char (gfc_expr *e, gfc_expr *k)
1781 return simplify_achar_char (e, k, "CHAR", false);
1785 /* Common subroutine for simplifying CMPLX, COMPLEX and DCMPLX. */
1787 static gfc_expr *
1788 simplify_cmplx (const char *name, gfc_expr *x, gfc_expr *y, int kind)
1790 gfc_expr *result;
1792 if (x->expr_type != EXPR_CONSTANT
1793 || (y != NULL && y->expr_type != EXPR_CONSTANT))
1794 return NULL;
1796 result = gfc_get_constant_expr (BT_COMPLEX, kind, &x->where);
1798 switch (x->ts.type)
1800 case BT_INTEGER:
1801 mpc_set_z (result->value.complex, x->value.integer, GFC_MPC_RND_MODE);
1802 break;
1804 case BT_REAL:
1805 mpc_set_fr (result->value.complex, x->value.real, GFC_RND_MODE);
1806 break;
1808 case BT_COMPLEX:
1809 mpc_set (result->value.complex, x->value.complex, GFC_MPC_RND_MODE);
1810 break;
1812 default:
1813 gfc_internal_error ("gfc_simplify_dcmplx(): Bad type (x)");
1816 if (!y)
1817 return range_check (result, name);
1819 switch (y->ts.type)
1821 case BT_INTEGER:
1822 mpfr_set_z (mpc_imagref (result->value.complex),
1823 y->value.integer, GFC_RND_MODE);
1824 break;
1826 case BT_REAL:
1827 mpfr_set (mpc_imagref (result->value.complex),
1828 y->value.real, GFC_RND_MODE);
1829 break;
1831 default:
1832 gfc_internal_error ("gfc_simplify_dcmplx(): Bad type (y)");
1835 return range_check (result, name);
1839 gfc_expr *
1840 gfc_simplify_cmplx (gfc_expr *x, gfc_expr *y, gfc_expr *k)
1842 int kind;
1844 kind = get_kind (BT_REAL, k, "CMPLX", gfc_default_complex_kind);
1845 if (kind == -1)
1846 return &gfc_bad_expr;
1848 return simplify_cmplx ("CMPLX", x, y, kind);
1852 gfc_expr *
1853 gfc_simplify_complex (gfc_expr *x, gfc_expr *y)
1855 int kind;
1857 if (x->ts.type == BT_INTEGER && y->ts.type == BT_INTEGER)
1858 kind = gfc_default_complex_kind;
1859 else if (x->ts.type == BT_REAL || y->ts.type == BT_INTEGER)
1860 kind = x->ts.kind;
1861 else if (x->ts.type == BT_INTEGER || y->ts.type == BT_REAL)
1862 kind = y->ts.kind;
1863 else if (x->ts.type == BT_REAL && y->ts.type == BT_REAL)
1864 kind = (x->ts.kind > y->ts.kind) ? x->ts.kind : y->ts.kind;
1865 else
1866 gcc_unreachable ();
1868 return simplify_cmplx ("COMPLEX", x, y, kind);
1872 gfc_expr *
1873 gfc_simplify_conjg (gfc_expr *e)
1875 gfc_expr *result;
1877 if (e->expr_type != EXPR_CONSTANT)
1878 return NULL;
1880 result = gfc_copy_expr (e);
1881 mpc_conj (result->value.complex, result->value.complex, GFC_MPC_RND_MODE);
1883 return range_check (result, "CONJG");
1887 /* Simplify atan2d (x) where the unit is degree. */
1889 gfc_expr *
1890 gfc_simplify_atan2d (gfc_expr *y, gfc_expr *x)
1892 gfc_expr *result;
1894 if (x->expr_type != EXPR_CONSTANT || y->expr_type != EXPR_CONSTANT)
1895 return NULL;
1897 if (mpfr_zero_p (y->value.real) && mpfr_zero_p (x->value.real))
1899 gfc_error ("If first argument of ATAN2D at %L is zero, then the "
1900 "second argument must not be zero", &y->where);
1901 return &gfc_bad_expr;
1904 result = gfc_get_constant_expr (x->ts.type, x->ts.kind, &x->where);
1905 mpfr_atan2 (result->value.real, y->value.real, x->value.real, GFC_RND_MODE);
1906 rad2deg (result->value.real);
1908 return range_check (result, "ATAN2D");
1912 gfc_expr *
1913 gfc_simplify_cos (gfc_expr *x)
1915 gfc_expr *result;
1917 if (x->expr_type != EXPR_CONSTANT)
1918 return NULL;
1920 result = gfc_get_constant_expr (x->ts.type, x->ts.kind, &x->where);
1922 switch (x->ts.type)
1924 case BT_REAL:
1925 mpfr_cos (result->value.real, x->value.real, GFC_RND_MODE);
1926 break;
1928 case BT_COMPLEX:
1929 gfc_set_model_kind (x->ts.kind);
1930 mpc_cos (result->value.complex, x->value.complex, GFC_MPC_RND_MODE);
1931 break;
1933 default:
1934 gfc_internal_error ("in gfc_simplify_cos(): Bad type");
1937 return range_check (result, "COS");
1941 static void
1942 deg2rad (mpfr_t x)
1944 mpfr_t d2r;
1946 mpfr_init (d2r);
1947 mpfr_const_pi (d2r, GFC_RND_MODE);
1948 mpfr_div_ui (d2r, d2r, 180, GFC_RND_MODE);
1949 mpfr_mul (x, x, d2r, GFC_RND_MODE);
1950 mpfr_clear (d2r);
1954 /* Simplification routines for SIND, COSD, TAND. */
1955 #include "trigd_fe.inc"
1958 /* Simplify COSD(X) where X has the unit of degree. */
1960 gfc_expr *
1961 gfc_simplify_cosd (gfc_expr *x)
1963 gfc_expr *result;
1965 if (x->expr_type != EXPR_CONSTANT)
1966 return NULL;
1968 result = gfc_get_constant_expr (x->ts.type, x->ts.kind, &x->where);
1969 mpfr_set (result->value.real, x->value.real, GFC_RND_MODE);
1970 simplify_cosd (result->value.real);
1972 return range_check (result, "COSD");
1976 /* Simplify SIND(X) where X has the unit of degree. */
1978 gfc_expr *
1979 gfc_simplify_sind (gfc_expr *x)
1981 gfc_expr *result;
1983 if (x->expr_type != EXPR_CONSTANT)
1984 return NULL;
1986 result = gfc_get_constant_expr (x->ts.type, x->ts.kind, &x->where);
1987 mpfr_set (result->value.real, x->value.real, GFC_RND_MODE);
1988 simplify_sind (result->value.real);
1990 return range_check (result, "SIND");
1994 /* Simplify TAND(X) where X has the unit of degree. */
1996 gfc_expr *
1997 gfc_simplify_tand (gfc_expr *x)
1999 gfc_expr *result;
2001 if (x->expr_type != EXPR_CONSTANT)
2002 return NULL;
2004 result = gfc_get_constant_expr (x->ts.type, x->ts.kind, &x->where);
2005 mpfr_set (result->value.real, x->value.real, GFC_RND_MODE);
2006 simplify_tand (result->value.real);
2008 return range_check (result, "TAND");
2012 /* Simplify COTAND(X) where X has the unit of degree. */
2014 gfc_expr *
2015 gfc_simplify_cotand (gfc_expr *x)
2017 gfc_expr *result;
2019 if (x->expr_type != EXPR_CONSTANT)
2020 return NULL;
2022 /* Implement COTAND = -TAND(x+90).
2023 TAND offers correct exact values for multiples of 30 degrees.
2024 This implementation is also compatible with the behavior of some legacy
2025 compilers. Keep this consistent with gfc_conv_intrinsic_cotand. */
2026 result = gfc_get_constant_expr (x->ts.type, x->ts.kind, &x->where);
2027 mpfr_set (result->value.real, x->value.real, GFC_RND_MODE);
2028 mpfr_add_ui (result->value.real, result->value.real, 90, GFC_RND_MODE);
2029 simplify_tand (result->value.real);
2030 mpfr_neg (result->value.real, result->value.real, GFC_RND_MODE);
2032 return range_check (result, "COTAND");
2036 gfc_expr *
2037 gfc_simplify_cosh (gfc_expr *x)
2039 gfc_expr *result;
2041 if (x->expr_type != EXPR_CONSTANT)
2042 return NULL;
2044 result = gfc_get_constant_expr (x->ts.type, x->ts.kind, &x->where);
2046 switch (x->ts.type)
2048 case BT_REAL:
2049 mpfr_cosh (result->value.real, x->value.real, GFC_RND_MODE);
2050 break;
2052 case BT_COMPLEX:
2053 mpc_cosh (result->value.complex, x->value.complex, GFC_MPC_RND_MODE);
2054 break;
2056 default:
2057 gcc_unreachable ();
2060 return range_check (result, "COSH");
2064 gfc_expr *
2065 gfc_simplify_count (gfc_expr *mask, gfc_expr *dim, gfc_expr *kind)
2067 gfc_expr *result;
2068 bool size_zero;
2070 size_zero = gfc_is_size_zero_array (mask);
2072 if (!(is_constant_array_expr (mask) || size_zero)
2073 || !gfc_is_constant_expr (dim)
2074 || !gfc_is_constant_expr (kind))
2075 return NULL;
2077 result = transformational_result (mask, dim,
2078 BT_INTEGER,
2079 get_kind (BT_INTEGER, kind, "COUNT",
2080 gfc_default_integer_kind),
2081 &mask->where);
2083 init_result_expr (result, 0, NULL);
2085 if (size_zero)
2086 return result;
2088 /* Passing MASK twice, once as data array, once as mask.
2089 Whenever gfc_count is called, '1' is added to the result. */
2090 return !dim || mask->rank == 1 ?
2091 simplify_transformation_to_scalar (result, mask, mask, gfc_count) :
2092 simplify_transformation_to_array (result, mask, dim, mask, gfc_count, NULL);
2095 /* Simplification routine for cshift. This works by copying the array
2096 expressions into a one-dimensional array, shuffling the values into another
2097 one-dimensional array and creating the new array expression from this. The
2098 shuffling part is basically taken from the library routine. */
2100 gfc_expr *
2101 gfc_simplify_cshift (gfc_expr *array, gfc_expr *shift, gfc_expr *dim)
2103 gfc_expr *result;
2104 int which;
2105 gfc_expr **arrayvec, **resultvec;
2106 gfc_expr **rptr, **sptr;
2107 mpz_t size;
2108 size_t arraysize, shiftsize, i;
2109 gfc_constructor *array_ctor, *shift_ctor;
2110 ssize_t *shiftvec, *hptr;
2111 ssize_t shift_val, len;
2112 ssize_t count[GFC_MAX_DIMENSIONS], extent[GFC_MAX_DIMENSIONS],
2113 hs_ex[GFC_MAX_DIMENSIONS + 1],
2114 hstride[GFC_MAX_DIMENSIONS], sstride[GFC_MAX_DIMENSIONS],
2115 a_extent[GFC_MAX_DIMENSIONS], a_stride[GFC_MAX_DIMENSIONS],
2116 h_extent[GFC_MAX_DIMENSIONS],
2117 ss_ex[GFC_MAX_DIMENSIONS + 1];
2118 ssize_t rsoffset;
2119 int d, n;
2120 bool continue_loop;
2121 gfc_expr **src, **dest;
2123 if (!is_constant_array_expr (array))
2124 return NULL;
2126 if (shift->rank > 0)
2127 gfc_simplify_expr (shift, 1);
2129 if (!gfc_is_constant_expr (shift))
2130 return NULL;
2132 /* Make dim zero-based. */
2133 if (dim)
2135 if (!gfc_is_constant_expr (dim))
2136 return NULL;
2137 which = mpz_get_si (dim->value.integer) - 1;
2139 else
2140 which = 0;
2142 if (array->shape == NULL)
2143 return NULL;
2145 gfc_array_size (array, &size);
2146 arraysize = mpz_get_ui (size);
2147 mpz_clear (size);
2149 result = gfc_get_array_expr (array->ts.type, array->ts.kind, &array->where);
2150 result->shape = gfc_copy_shape (array->shape, array->rank);
2151 result->rank = array->rank;
2152 result->ts.u.derived = array->ts.u.derived;
2154 if (arraysize == 0)
2155 return result;
2157 arrayvec = XCNEWVEC (gfc_expr *, arraysize);
2158 array_ctor = gfc_constructor_first (array->value.constructor);
2159 for (i = 0; i < arraysize; i++)
2161 arrayvec[i] = array_ctor->expr;
2162 array_ctor = gfc_constructor_next (array_ctor);
2165 resultvec = XCNEWVEC (gfc_expr *, arraysize);
2167 sstride[0] = 0;
2168 extent[0] = 1;
2169 count[0] = 0;
2171 for (d=0; d < array->rank; d++)
2173 a_extent[d] = mpz_get_si (array->shape[d]);
2174 a_stride[d] = d == 0 ? 1 : a_stride[d-1] * a_extent[d-1];
2177 if (shift->rank > 0)
2179 gfc_array_size (shift, &size);
2180 shiftsize = mpz_get_ui (size);
2181 mpz_clear (size);
2182 shiftvec = XCNEWVEC (ssize_t, shiftsize);
2183 shift_ctor = gfc_constructor_first (shift->value.constructor);
2184 for (d = 0; d < shift->rank; d++)
2186 h_extent[d] = mpz_get_si (shift->shape[d]);
2187 hstride[d] = d == 0 ? 1 : hstride[d-1] * h_extent[d-1];
2190 else
2191 shiftvec = NULL;
2193 /* Shut up compiler */
2194 len = 1;
2195 rsoffset = 1;
2197 n = 0;
2198 for (d=0; d < array->rank; d++)
2200 if (d == which)
2202 rsoffset = a_stride[d];
2203 len = a_extent[d];
2205 else
2207 count[n] = 0;
2208 extent[n] = a_extent[d];
2209 sstride[n] = a_stride[d];
2210 ss_ex[n] = sstride[n] * extent[n];
2211 if (shiftvec)
2212 hs_ex[n] = hstride[n] * extent[n];
2213 n++;
2216 ss_ex[n] = 0;
2217 hs_ex[n] = 0;
2219 if (shiftvec)
2221 for (i = 0; i < shiftsize; i++)
2223 ssize_t val;
2224 val = mpz_get_si (shift_ctor->expr->value.integer);
2225 val = val % len;
2226 if (val < 0)
2227 val += len;
2228 shiftvec[i] = val;
2229 shift_ctor = gfc_constructor_next (shift_ctor);
2231 shift_val = 0;
2233 else
2235 shift_val = mpz_get_si (shift->value.integer);
2236 shift_val = shift_val % len;
2237 if (shift_val < 0)
2238 shift_val += len;
2241 continue_loop = true;
2242 d = array->rank;
2243 rptr = resultvec;
2244 sptr = arrayvec;
2245 hptr = shiftvec;
2247 while (continue_loop)
2249 ssize_t sh;
2250 if (shiftvec)
2251 sh = *hptr;
2252 else
2253 sh = shift_val;
2255 src = &sptr[sh * rsoffset];
2256 dest = rptr;
2257 for (n = 0; n < len - sh; n++)
2259 *dest = *src;
2260 dest += rsoffset;
2261 src += rsoffset;
2263 src = sptr;
2264 for ( n = 0; n < sh; n++)
2266 *dest = *src;
2267 dest += rsoffset;
2268 src += rsoffset;
2270 rptr += sstride[0];
2271 sptr += sstride[0];
2272 if (shiftvec)
2273 hptr += hstride[0];
2274 count[0]++;
2275 n = 0;
2276 while (count[n] == extent[n])
2278 count[n] = 0;
2279 rptr -= ss_ex[n];
2280 sptr -= ss_ex[n];
2281 if (shiftvec)
2282 hptr -= hs_ex[n];
2283 n++;
2284 if (n >= d - 1)
2286 continue_loop = false;
2287 break;
2289 else
2291 count[n]++;
2292 rptr += sstride[n];
2293 sptr += sstride[n];
2294 if (shiftvec)
2295 hptr += hstride[n];
2300 for (i = 0; i < arraysize; i++)
2302 gfc_constructor_append_expr (&result->value.constructor,
2303 gfc_copy_expr (resultvec[i]),
2304 NULL);
2306 return result;
2310 gfc_expr *
2311 gfc_simplify_dcmplx (gfc_expr *x, gfc_expr *y)
2313 return simplify_cmplx ("DCMPLX", x, y, gfc_default_double_kind);
2317 gfc_expr *
2318 gfc_simplify_dble (gfc_expr *e)
2320 gfc_expr *result = NULL;
2321 int tmp1, tmp2;
2323 if (e->expr_type != EXPR_CONSTANT)
2324 return NULL;
2326 /* For explicit conversion, turn off -Wconversion and -Wconversion-extra
2327 warnings. */
2328 tmp1 = warn_conversion;
2329 tmp2 = warn_conversion_extra;
2330 warn_conversion = warn_conversion_extra = 0;
2332 result = gfc_convert_constant (e, BT_REAL, gfc_default_double_kind);
2334 warn_conversion = tmp1;
2335 warn_conversion_extra = tmp2;
2337 if (result == &gfc_bad_expr)
2338 return &gfc_bad_expr;
2340 return range_check (result, "DBLE");
2344 gfc_expr *
2345 gfc_simplify_digits (gfc_expr *x)
2347 int i, digits;
2349 i = gfc_validate_kind (x->ts.type, x->ts.kind, false);
2351 switch (x->ts.type)
2353 case BT_INTEGER:
2354 digits = gfc_integer_kinds[i].digits;
2355 break;
2357 case BT_REAL:
2358 case BT_COMPLEX:
2359 digits = gfc_real_kinds[i].digits;
2360 break;
2362 default:
2363 gcc_unreachable ();
2366 return gfc_get_int_expr (gfc_default_integer_kind, NULL, digits);
2370 gfc_expr *
2371 gfc_simplify_dim (gfc_expr *x, gfc_expr *y)
2373 gfc_expr *result;
2374 int kind;
2376 if (x->expr_type != EXPR_CONSTANT || y->expr_type != EXPR_CONSTANT)
2377 return NULL;
2379 kind = x->ts.kind > y->ts.kind ? x->ts.kind : y->ts.kind;
2380 result = gfc_get_constant_expr (x->ts.type, kind, &x->where);
2382 switch (x->ts.type)
2384 case BT_INTEGER:
2385 if (mpz_cmp (x->value.integer, y->value.integer) > 0)
2386 mpz_sub (result->value.integer, x->value.integer, y->value.integer);
2387 else
2388 mpz_set_ui (result->value.integer, 0);
2390 break;
2392 case BT_REAL:
2393 if (mpfr_cmp (x->value.real, y->value.real) > 0)
2394 mpfr_sub (result->value.real, x->value.real, y->value.real,
2395 GFC_RND_MODE);
2396 else
2397 mpfr_set_ui (result->value.real, 0, GFC_RND_MODE);
2399 break;
2401 default:
2402 gfc_internal_error ("gfc_simplify_dim(): Bad type");
2405 return range_check (result, "DIM");
2409 gfc_expr*
2410 gfc_simplify_dot_product (gfc_expr *vector_a, gfc_expr *vector_b)
2412 /* If vector_a is a zero-sized array, the result is 0 for INTEGER,
2413 REAL, and COMPLEX types and .false. for LOGICAL. */
2414 if (vector_a->shape && mpz_get_si (vector_a->shape[0]) == 0)
2416 if (vector_a->ts.type == BT_LOGICAL)
2417 return gfc_get_logical_expr (gfc_default_logical_kind, NULL, false);
2418 else
2419 return gfc_get_int_expr (gfc_default_integer_kind, NULL, 0);
2422 if (!is_constant_array_expr (vector_a)
2423 || !is_constant_array_expr (vector_b))
2424 return NULL;
2426 return compute_dot_product (vector_a, 1, 0, vector_b, 1, 0, true);
2430 gfc_expr *
2431 gfc_simplify_dprod (gfc_expr *x, gfc_expr *y)
2433 gfc_expr *a1, *a2, *result;
2435 if (x->expr_type != EXPR_CONSTANT || y->expr_type != EXPR_CONSTANT)
2436 return NULL;
2438 a1 = gfc_real2real (x, gfc_default_double_kind);
2439 a2 = gfc_real2real (y, gfc_default_double_kind);
2441 result = gfc_get_constant_expr (BT_REAL, gfc_default_double_kind, &x->where);
2442 mpfr_mul (result->value.real, a1->value.real, a2->value.real, GFC_RND_MODE);
2444 gfc_free_expr (a2);
2445 gfc_free_expr (a1);
2447 return range_check (result, "DPROD");
2451 static gfc_expr *
2452 simplify_dshift (gfc_expr *arg1, gfc_expr *arg2, gfc_expr *shiftarg,
2453 bool right)
2455 gfc_expr *result;
2456 int i, k, size, shift;
2458 if (arg1->expr_type != EXPR_CONSTANT || arg2->expr_type != EXPR_CONSTANT
2459 || shiftarg->expr_type != EXPR_CONSTANT)
2460 return NULL;
2462 k = gfc_validate_kind (BT_INTEGER, arg1->ts.kind, false);
2463 size = gfc_integer_kinds[k].bit_size;
2465 gfc_extract_int (shiftarg, &shift);
2467 /* DSHIFTR(I,J,SHIFT) = DSHIFTL(I,J,SIZE-SHIFT). */
2468 if (right)
2469 shift = size - shift;
2471 result = gfc_get_constant_expr (BT_INTEGER, arg1->ts.kind, &arg1->where);
2472 mpz_set_ui (result->value.integer, 0);
2474 for (i = 0; i < shift; i++)
2475 if (mpz_tstbit (arg2->value.integer, size - shift + i))
2476 mpz_setbit (result->value.integer, i);
2478 for (i = 0; i < size - shift; i++)
2479 if (mpz_tstbit (arg1->value.integer, i))
2480 mpz_setbit (result->value.integer, shift + i);
2482 /* Convert to a signed value. */
2483 gfc_convert_mpz_to_signed (result->value.integer, size);
2485 return result;
2489 gfc_expr *
2490 gfc_simplify_dshiftr (gfc_expr *arg1, gfc_expr *arg2, gfc_expr *shiftarg)
2492 return simplify_dshift (arg1, arg2, shiftarg, true);
2496 gfc_expr *
2497 gfc_simplify_dshiftl (gfc_expr *arg1, gfc_expr *arg2, gfc_expr *shiftarg)
2499 return simplify_dshift (arg1, arg2, shiftarg, false);
2503 gfc_expr *
2504 gfc_simplify_eoshift (gfc_expr *array, gfc_expr *shift, gfc_expr *boundary,
2505 gfc_expr *dim)
2507 bool temp_boundary;
2508 gfc_expr *bnd;
2509 gfc_expr *result;
2510 int which;
2511 gfc_expr **arrayvec, **resultvec;
2512 gfc_expr **rptr, **sptr;
2513 mpz_t size;
2514 size_t arraysize, i;
2515 gfc_constructor *array_ctor, *shift_ctor, *bnd_ctor;
2516 ssize_t shift_val, len;
2517 ssize_t count[GFC_MAX_DIMENSIONS], extent[GFC_MAX_DIMENSIONS],
2518 sstride[GFC_MAX_DIMENSIONS], a_extent[GFC_MAX_DIMENSIONS],
2519 a_stride[GFC_MAX_DIMENSIONS], ss_ex[GFC_MAX_DIMENSIONS + 1];
2520 ssize_t rsoffset;
2521 int d, n;
2522 bool continue_loop;
2523 gfc_expr **src, **dest;
2524 size_t s_len;
2526 if (!is_constant_array_expr (array))
2527 return NULL;
2529 if (shift->rank > 0)
2530 gfc_simplify_expr (shift, 1);
2532 if (!gfc_is_constant_expr (shift))
2533 return NULL;
2535 if (boundary)
2537 if (boundary->rank > 0)
2538 gfc_simplify_expr (boundary, 1);
2540 if (!gfc_is_constant_expr (boundary))
2541 return NULL;
2544 if (dim)
2546 if (!gfc_is_constant_expr (dim))
2547 return NULL;
2548 which = mpz_get_si (dim->value.integer) - 1;
2550 else
2551 which = 0;
2553 s_len = 0;
2554 if (boundary == NULL)
2556 temp_boundary = true;
2557 switch (array->ts.type)
2560 case BT_INTEGER:
2561 bnd = gfc_get_int_expr (array->ts.kind, NULL, 0);
2562 break;
2564 case BT_LOGICAL:
2565 bnd = gfc_get_logical_expr (array->ts.kind, NULL, 0);
2566 break;
2568 case BT_REAL:
2569 bnd = gfc_get_constant_expr (array->ts.type, array->ts.kind, &gfc_current_locus);
2570 mpfr_set_ui (bnd->value.real, 0, GFC_RND_MODE);
2571 break;
2573 case BT_COMPLEX:
2574 bnd = gfc_get_constant_expr (array->ts.type, array->ts.kind, &gfc_current_locus);
2575 mpc_set_ui (bnd->value.complex, 0, GFC_RND_MODE);
2576 break;
2578 case BT_CHARACTER:
2579 s_len = mpz_get_ui (array->ts.u.cl->length->value.integer);
2580 bnd = gfc_get_character_expr (array->ts.kind, &gfc_current_locus, NULL, s_len);
2581 break;
2583 default:
2584 gcc_unreachable();
2588 else
2590 temp_boundary = false;
2591 bnd = boundary;
2594 gfc_array_size (array, &size);
2595 arraysize = mpz_get_ui (size);
2596 mpz_clear (size);
2598 result = gfc_get_array_expr (array->ts.type, array->ts.kind, &array->where);
2599 result->shape = gfc_copy_shape (array->shape, array->rank);
2600 result->rank = array->rank;
2601 result->ts = array->ts;
2603 if (arraysize == 0)
2604 goto final;
2606 if (array->shape == NULL)
2607 goto final;
2609 arrayvec = XCNEWVEC (gfc_expr *, arraysize);
2610 array_ctor = gfc_constructor_first (array->value.constructor);
2611 for (i = 0; i < arraysize; i++)
2613 arrayvec[i] = array_ctor->expr;
2614 array_ctor = gfc_constructor_next (array_ctor);
2617 resultvec = XCNEWVEC (gfc_expr *, arraysize);
2619 extent[0] = 1;
2620 count[0] = 0;
2622 for (d=0; d < array->rank; d++)
2624 a_extent[d] = mpz_get_si (array->shape[d]);
2625 a_stride[d] = d == 0 ? 1 : a_stride[d-1] * a_extent[d-1];
2628 if (shift->rank > 0)
2630 shift_ctor = gfc_constructor_first (shift->value.constructor);
2631 shift_val = 0;
2633 else
2635 shift_ctor = NULL;
2636 shift_val = mpz_get_si (shift->value.integer);
2639 if (bnd->rank > 0)
2640 bnd_ctor = gfc_constructor_first (bnd->value.constructor);
2641 else
2642 bnd_ctor = NULL;
2644 /* Shut up compiler */
2645 len = 1;
2646 rsoffset = 1;
2648 n = 0;
2649 for (d=0; d < array->rank; d++)
2651 if (d == which)
2653 rsoffset = a_stride[d];
2654 len = a_extent[d];
2656 else
2658 count[n] = 0;
2659 extent[n] = a_extent[d];
2660 sstride[n] = a_stride[d];
2661 ss_ex[n] = sstride[n] * extent[n];
2662 n++;
2665 ss_ex[n] = 0;
2667 continue_loop = true;
2668 d = array->rank;
2669 rptr = resultvec;
2670 sptr = arrayvec;
2672 while (continue_loop)
2674 ssize_t sh, delta;
2676 if (shift_ctor)
2677 sh = mpz_get_si (shift_ctor->expr->value.integer);
2678 else
2679 sh = shift_val;
2681 if (( sh >= 0 ? sh : -sh ) > len)
2683 delta = len;
2684 sh = len;
2686 else
2687 delta = (sh >= 0) ? sh: -sh;
2689 if (sh > 0)
2691 src = &sptr[delta * rsoffset];
2692 dest = rptr;
2694 else
2696 src = sptr;
2697 dest = &rptr[delta * rsoffset];
2700 for (n = 0; n < len - delta; n++)
2702 *dest = *src;
2703 dest += rsoffset;
2704 src += rsoffset;
2707 if (sh < 0)
2708 dest = rptr;
2710 n = delta;
2712 if (bnd_ctor)
2714 while (n--)
2716 *dest = gfc_copy_expr (bnd_ctor->expr);
2717 dest += rsoffset;
2720 else
2722 while (n--)
2724 *dest = gfc_copy_expr (bnd);
2725 dest += rsoffset;
2728 rptr += sstride[0];
2729 sptr += sstride[0];
2730 if (shift_ctor)
2731 shift_ctor = gfc_constructor_next (shift_ctor);
2733 if (bnd_ctor)
2734 bnd_ctor = gfc_constructor_next (bnd_ctor);
2736 count[0]++;
2737 n = 0;
2738 while (count[n] == extent[n])
2740 count[n] = 0;
2741 rptr -= ss_ex[n];
2742 sptr -= ss_ex[n];
2743 n++;
2744 if (n >= d - 1)
2746 continue_loop = false;
2747 break;
2749 else
2751 count[n]++;
2752 rptr += sstride[n];
2753 sptr += sstride[n];
2758 for (i = 0; i < arraysize; i++)
2760 gfc_constructor_append_expr (&result->value.constructor,
2761 gfc_copy_expr (resultvec[i]),
2762 NULL);
2765 final:
2766 if (temp_boundary)
2767 gfc_free_expr (bnd);
2769 return result;
2772 gfc_expr *
2773 gfc_simplify_erf (gfc_expr *x)
2775 gfc_expr *result;
2777 if (x->expr_type != EXPR_CONSTANT)
2778 return NULL;
2780 result = gfc_get_constant_expr (x->ts.type, x->ts.kind, &x->where);
2781 mpfr_erf (result->value.real, x->value.real, GFC_RND_MODE);
2783 return range_check (result, "ERF");
2787 gfc_expr *
2788 gfc_simplify_erfc (gfc_expr *x)
2790 gfc_expr *result;
2792 if (x->expr_type != EXPR_CONSTANT)
2793 return NULL;
2795 result = gfc_get_constant_expr (x->ts.type, x->ts.kind, &x->where);
2796 mpfr_erfc (result->value.real, x->value.real, GFC_RND_MODE);
2798 return range_check (result, "ERFC");
2802 /* Helper functions to simplify ERFC_SCALED(x) = ERFC(x) * EXP(X**2). */
2804 #define MAX_ITER 200
2805 #define ARG_LIMIT 12
2807 /* Calculate ERFC_SCALED directly by its definition:
2809 ERFC_SCALED(x) = ERFC(x) * EXP(X**2)
2811 using a large precision for intermediate results. This is used for all
2812 but large values of the argument. */
2813 static void
2814 fullprec_erfc_scaled (mpfr_t res, mpfr_t arg)
2816 mpfr_prec_t prec;
2817 mpfr_t a, b;
2819 prec = mpfr_get_default_prec ();
2820 mpfr_set_default_prec (10 * prec);
2822 mpfr_init (a);
2823 mpfr_init (b);
2825 mpfr_set (a, arg, GFC_RND_MODE);
2826 mpfr_sqr (b, a, GFC_RND_MODE);
2827 mpfr_exp (b, b, GFC_RND_MODE);
2828 mpfr_erfc (a, a, GFC_RND_MODE);
2829 mpfr_mul (a, a, b, GFC_RND_MODE);
2831 mpfr_set (res, a, GFC_RND_MODE);
2832 mpfr_set_default_prec (prec);
2834 mpfr_clear (a);
2835 mpfr_clear (b);
2838 /* Calculate ERFC_SCALED using a power series expansion in 1/arg:
2840 ERFC_SCALED(x) = 1 / (x * sqrt(pi))
2841 * (1 + Sum_n (-1)**n * (1 * 3 * 5 * ... * (2n-1))
2842 / (2 * x**2)**n)
2844 This is used for large values of the argument. Intermediate calculations
2845 are performed with twice the precision. We don't do a fixed number of
2846 iterations of the sum, but stop when it has converged to the required
2847 precision. */
2848 static void
2849 asympt_erfc_scaled (mpfr_t res, mpfr_t arg)
2851 mpfr_t sum, x, u, v, w, oldsum, sumtrunc;
2852 mpz_t num;
2853 mpfr_prec_t prec;
2854 unsigned i;
2856 prec = mpfr_get_default_prec ();
2857 mpfr_set_default_prec (2 * prec);
2859 mpfr_init (sum);
2860 mpfr_init (x);
2861 mpfr_init (u);
2862 mpfr_init (v);
2863 mpfr_init (w);
2864 mpz_init (num);
2866 mpfr_init (oldsum);
2867 mpfr_init (sumtrunc);
2868 mpfr_set_prec (oldsum, prec);
2869 mpfr_set_prec (sumtrunc, prec);
2871 mpfr_set (x, arg, GFC_RND_MODE);
2872 mpfr_set_ui (sum, 1, GFC_RND_MODE);
2873 mpz_set_ui (num, 1);
2875 mpfr_set (u, x, GFC_RND_MODE);
2876 mpfr_sqr (u, u, GFC_RND_MODE);
2877 mpfr_mul_ui (u, u, 2, GFC_RND_MODE);
2878 mpfr_pow_si (u, u, -1, GFC_RND_MODE);
2880 for (i = 1; i < MAX_ITER; i++)
2882 mpfr_set (oldsum, sum, GFC_RND_MODE);
2884 mpz_mul_ui (num, num, 2 * i - 1);
2885 mpz_neg (num, num);
2887 mpfr_set (w, u, GFC_RND_MODE);
2888 mpfr_pow_ui (w, w, i, GFC_RND_MODE);
2890 mpfr_set_z (v, num, GFC_RND_MODE);
2891 mpfr_mul (v, v, w, GFC_RND_MODE);
2893 mpfr_add (sum, sum, v, GFC_RND_MODE);
2895 mpfr_set (sumtrunc, sum, GFC_RND_MODE);
2896 if (mpfr_cmp (sumtrunc, oldsum) == 0)
2897 break;
2900 /* We should have converged by now; otherwise, ARG_LIMIT is probably
2901 set too low. */
2902 gcc_assert (i < MAX_ITER);
2904 /* Divide by x * sqrt(Pi). */
2905 mpfr_const_pi (u, GFC_RND_MODE);
2906 mpfr_sqrt (u, u, GFC_RND_MODE);
2907 mpfr_mul (u, u, x, GFC_RND_MODE);
2908 mpfr_div (sum, sum, u, GFC_RND_MODE);
2910 mpfr_set (res, sum, GFC_RND_MODE);
2911 mpfr_set_default_prec (prec);
2913 mpfr_clears (sum, x, u, v, w, oldsum, sumtrunc, NULL);
2914 mpz_clear (num);
2918 gfc_expr *
2919 gfc_simplify_erfc_scaled (gfc_expr *x)
2921 gfc_expr *result;
2923 if (x->expr_type != EXPR_CONSTANT)
2924 return NULL;
2926 result = gfc_get_constant_expr (x->ts.type, x->ts.kind, &x->where);
2927 if (mpfr_cmp_d (x->value.real, ARG_LIMIT) >= 0)
2928 asympt_erfc_scaled (result->value.real, x->value.real);
2929 else
2930 fullprec_erfc_scaled (result->value.real, x->value.real);
2932 return range_check (result, "ERFC_SCALED");
2935 #undef MAX_ITER
2936 #undef ARG_LIMIT
2939 gfc_expr *
2940 gfc_simplify_epsilon (gfc_expr *e)
2942 gfc_expr *result;
2943 int i;
2945 i = gfc_validate_kind (e->ts.type, e->ts.kind, false);
2947 result = gfc_get_constant_expr (BT_REAL, e->ts.kind, &e->where);
2948 mpfr_set (result->value.real, gfc_real_kinds[i].epsilon, GFC_RND_MODE);
2950 return range_check (result, "EPSILON");
2954 gfc_expr *
2955 gfc_simplify_exp (gfc_expr *x)
2957 gfc_expr *result;
2959 if (x->expr_type != EXPR_CONSTANT)
2960 return NULL;
2962 result = gfc_get_constant_expr (x->ts.type, x->ts.kind, &x->where);
2964 switch (x->ts.type)
2966 case BT_REAL:
2967 mpfr_exp (result->value.real, x->value.real, GFC_RND_MODE);
2968 break;
2970 case BT_COMPLEX:
2971 gfc_set_model_kind (x->ts.kind);
2972 mpc_exp (result->value.complex, x->value.complex, GFC_MPC_RND_MODE);
2973 break;
2975 default:
2976 gfc_internal_error ("in gfc_simplify_exp(): Bad type");
2979 return range_check (result, "EXP");
2983 gfc_expr *
2984 gfc_simplify_exponent (gfc_expr *x)
2986 long int val;
2987 gfc_expr *result;
2989 if (x->expr_type != EXPR_CONSTANT)
2990 return NULL;
2992 result = gfc_get_constant_expr (BT_INTEGER, gfc_default_integer_kind,
2993 &x->where);
2995 /* EXPONENT(inf) = EXPONENT(nan) = HUGE(0) */
2996 if (mpfr_inf_p (x->value.real) || mpfr_nan_p (x->value.real))
2998 int i = gfc_validate_kind (BT_INTEGER, gfc_default_integer_kind, false);
2999 mpz_set (result->value.integer, gfc_integer_kinds[i].huge);
3000 return result;
3003 /* EXPONENT(+/- 0.0) = 0 */
3004 if (mpfr_zero_p (x->value.real))
3006 mpz_set_ui (result->value.integer, 0);
3007 return result;
3010 gfc_set_model (x->value.real);
3012 val = (long int) mpfr_get_exp (x->value.real);
3013 mpz_set_si (result->value.integer, val);
3015 return range_check (result, "EXPONENT");
3019 gfc_expr *
3020 gfc_simplify_failed_or_stopped_images (gfc_expr *team ATTRIBUTE_UNUSED,
3021 gfc_expr *kind)
3023 if (flag_coarray == GFC_FCOARRAY_NONE)
3025 gfc_current_locus = *gfc_current_intrinsic_where;
3026 gfc_fatal_error ("Coarrays disabled at %C, use %<-fcoarray=%> to enable");
3027 return &gfc_bad_expr;
3030 if (flag_coarray == GFC_FCOARRAY_SINGLE)
3032 gfc_expr *result;
3033 int actual_kind;
3034 if (kind)
3035 gfc_extract_int (kind, &actual_kind);
3036 else
3037 actual_kind = gfc_default_integer_kind;
3039 result = gfc_get_array_expr (BT_INTEGER, actual_kind, &gfc_current_locus);
3040 result->rank = 1;
3041 return result;
3044 /* For fcoarray = lib no simplification is possible, because it is not known
3045 what images failed or are stopped at compile time. */
3046 return NULL;
3050 gfc_expr *
3051 gfc_simplify_get_team (gfc_expr *level ATTRIBUTE_UNUSED)
3053 if (flag_coarray == GFC_FCOARRAY_NONE)
3055 gfc_current_locus = *gfc_current_intrinsic_where;
3056 gfc_fatal_error ("Coarrays disabled at %C, use %<-fcoarray=%> to enable");
3057 return &gfc_bad_expr;
3060 if (flag_coarray == GFC_FCOARRAY_SINGLE)
3062 gfc_expr *result;
3063 result = gfc_get_array_expr (BT_INTEGER, gfc_default_integer_kind, &gfc_current_locus);
3064 result->rank = 0;
3065 return result;
3068 /* For fcoarray = lib no simplification is possible, because it is not known
3069 what images failed or are stopped at compile time. */
3070 return NULL;
3074 gfc_expr *
3075 gfc_simplify_float (gfc_expr *a)
3077 gfc_expr *result;
3079 if (a->expr_type != EXPR_CONSTANT)
3080 return NULL;
3082 result = gfc_int2real (a, gfc_default_real_kind);
3084 return range_check (result, "FLOAT");
3088 static bool
3089 is_last_ref_vtab (gfc_expr *e)
3091 gfc_ref *ref;
3092 gfc_component *comp = NULL;
3094 if (e->expr_type != EXPR_VARIABLE)
3095 return false;
3097 for (ref = e->ref; ref; ref = ref->next)
3098 if (ref->type == REF_COMPONENT)
3099 comp = ref->u.c.component;
3101 if (!e->ref || !comp)
3102 return e->symtree->n.sym->attr.vtab;
3104 if (comp->name[0] == '_' && strcmp (comp->name, "_vptr") == 0)
3105 return true;
3107 return false;
3111 gfc_expr *
3112 gfc_simplify_extends_type_of (gfc_expr *a, gfc_expr *mold)
3114 /* Avoid simplification of resolved symbols. */
3115 if (is_last_ref_vtab (a) || is_last_ref_vtab (mold))
3116 return NULL;
3118 if (a->ts.type == BT_DERIVED && mold->ts.type == BT_DERIVED)
3119 return gfc_get_logical_expr (gfc_default_logical_kind, &a->where,
3120 gfc_type_is_extension_of (mold->ts.u.derived,
3121 a->ts.u.derived));
3123 if (UNLIMITED_POLY (a) || UNLIMITED_POLY (mold))
3124 return NULL;
3126 if ((a->ts.type == BT_CLASS && !gfc_expr_attr (a).class_ok)
3127 || (mold->ts.type == BT_CLASS && !gfc_expr_attr (mold).class_ok))
3128 return NULL;
3130 /* Return .false. if the dynamic type can never be an extension. */
3131 if ((a->ts.type == BT_CLASS && mold->ts.type == BT_CLASS
3132 && !gfc_type_is_extension_of
3133 (CLASS_DATA (mold)->ts.u.derived,
3134 CLASS_DATA (a)->ts.u.derived)
3135 && !gfc_type_is_extension_of
3136 (CLASS_DATA (a)->ts.u.derived,
3137 CLASS_DATA (mold)->ts.u.derived))
3138 || (a->ts.type == BT_DERIVED && mold->ts.type == BT_CLASS
3139 && !gfc_type_is_extension_of
3140 (CLASS_DATA (mold)->ts.u.derived,
3141 a->ts.u.derived))
3142 || (a->ts.type == BT_CLASS && mold->ts.type == BT_DERIVED
3143 && !gfc_type_is_extension_of
3144 (mold->ts.u.derived,
3145 CLASS_DATA (a)->ts.u.derived)
3146 && !gfc_type_is_extension_of
3147 (CLASS_DATA (a)->ts.u.derived,
3148 mold->ts.u.derived)))
3149 return gfc_get_logical_expr (gfc_default_logical_kind, &a->where, false);
3151 /* Return .true. if the dynamic type is guaranteed to be an extension. */
3152 if (a->ts.type == BT_CLASS && mold->ts.type == BT_DERIVED
3153 && gfc_type_is_extension_of (mold->ts.u.derived,
3154 CLASS_DATA (a)->ts.u.derived))
3155 return gfc_get_logical_expr (gfc_default_logical_kind, &a->where, true);
3157 return NULL;
3161 gfc_expr *
3162 gfc_simplify_same_type_as (gfc_expr *a, gfc_expr *b)
3164 /* Avoid simplification of resolved symbols. */
3165 if (is_last_ref_vtab (a) || is_last_ref_vtab (b))
3166 return NULL;
3168 /* Return .false. if the dynamic type can never be the
3169 same. */
3170 if (((a->ts.type == BT_CLASS && gfc_expr_attr (a).class_ok)
3171 || (b->ts.type == BT_CLASS && gfc_expr_attr (b).class_ok))
3172 && !gfc_type_compatible (&a->ts, &b->ts)
3173 && !gfc_type_compatible (&b->ts, &a->ts))
3174 return gfc_get_logical_expr (gfc_default_logical_kind, &a->where, false);
3176 if (a->ts.type != BT_DERIVED || b->ts.type != BT_DERIVED)
3177 return NULL;
3179 return gfc_get_logical_expr (gfc_default_logical_kind, &a->where,
3180 gfc_compare_derived_types (a->ts.u.derived,
3181 b->ts.u.derived));
3185 gfc_expr *
3186 gfc_simplify_floor (gfc_expr *e, gfc_expr *k)
3188 gfc_expr *result;
3189 mpfr_t floor;
3190 int kind;
3192 kind = get_kind (BT_INTEGER, k, "FLOOR", gfc_default_integer_kind);
3193 if (kind == -1)
3194 gfc_internal_error ("gfc_simplify_floor(): Bad kind");
3196 if (e->expr_type != EXPR_CONSTANT)
3197 return NULL;
3199 mpfr_init2 (floor, mpfr_get_prec (e->value.real));
3200 mpfr_floor (floor, e->value.real);
3202 result = gfc_get_constant_expr (BT_INTEGER, kind, &e->where);
3203 gfc_mpfr_to_mpz (result->value.integer, floor, &e->where);
3205 mpfr_clear (floor);
3207 return range_check (result, "FLOOR");
3211 gfc_expr *
3212 gfc_simplify_fraction (gfc_expr *x)
3214 gfc_expr *result;
3215 mpfr_exp_t e;
3217 if (x->expr_type != EXPR_CONSTANT)
3218 return NULL;
3220 result = gfc_get_constant_expr (BT_REAL, x->ts.kind, &x->where);
3222 /* FRACTION(inf) = NaN. */
3223 if (mpfr_inf_p (x->value.real))
3225 mpfr_set_nan (result->value.real);
3226 return result;
3229 /* mpfr_frexp() correctly handles zeros and NaNs. */
3230 mpfr_frexp (&e, result->value.real, x->value.real, GFC_RND_MODE);
3232 return range_check (result, "FRACTION");
3236 gfc_expr *
3237 gfc_simplify_gamma (gfc_expr *x)
3239 gfc_expr *result;
3241 if (x->expr_type != EXPR_CONSTANT)
3242 return NULL;
3244 result = gfc_get_constant_expr (x->ts.type, x->ts.kind, &x->where);
3245 mpfr_gamma (result->value.real, x->value.real, GFC_RND_MODE);
3247 return range_check (result, "GAMMA");
3251 gfc_expr *
3252 gfc_simplify_huge (gfc_expr *e)
3254 gfc_expr *result;
3255 int i;
3257 i = gfc_validate_kind (e->ts.type, e->ts.kind, false);
3258 result = gfc_get_constant_expr (e->ts.type, e->ts.kind, &e->where);
3260 switch (e->ts.type)
3262 case BT_INTEGER:
3263 mpz_set (result->value.integer, gfc_integer_kinds[i].huge);
3264 break;
3266 case BT_REAL:
3267 mpfr_set (result->value.real, gfc_real_kinds[i].huge, GFC_RND_MODE);
3268 break;
3270 default:
3271 gcc_unreachable ();
3274 return result;
3278 gfc_expr *
3279 gfc_simplify_hypot (gfc_expr *x, gfc_expr *y)
3281 gfc_expr *result;
3283 if (x->expr_type != EXPR_CONSTANT || y->expr_type != EXPR_CONSTANT)
3284 return NULL;
3286 result = gfc_get_constant_expr (x->ts.type, x->ts.kind, &x->where);
3287 mpfr_hypot (result->value.real, x->value.real, y->value.real, GFC_RND_MODE);
3288 return range_check (result, "HYPOT");
3292 /* We use the processor's collating sequence, because all
3293 systems that gfortran currently works on are ASCII. */
3295 gfc_expr *
3296 gfc_simplify_iachar (gfc_expr *e, gfc_expr *kind)
3298 gfc_expr *result;
3299 gfc_char_t index;
3300 int k;
3302 if (e->expr_type != EXPR_CONSTANT)
3303 return NULL;
3305 if (e->value.character.length != 1)
3307 gfc_error ("Argument of IACHAR at %L must be of length one", &e->where);
3308 return &gfc_bad_expr;
3311 index = e->value.character.string[0];
3313 if (warn_surprising && index > 127)
3314 gfc_warning (OPT_Wsurprising,
3315 "Argument of IACHAR function at %L outside of range 0..127",
3316 &e->where);
3318 k = get_kind (BT_INTEGER, kind, "IACHAR", gfc_default_integer_kind);
3319 if (k == -1)
3320 return &gfc_bad_expr;
3322 result = gfc_get_int_expr (k, &e->where, index);
3324 return range_check (result, "IACHAR");
3328 static gfc_expr *
3329 do_bit_and (gfc_expr *result, gfc_expr *e)
3331 gcc_assert (e->ts.type == BT_INTEGER && e->expr_type == EXPR_CONSTANT);
3332 gcc_assert (result->ts.type == BT_INTEGER
3333 && result->expr_type == EXPR_CONSTANT);
3335 mpz_and (result->value.integer, result->value.integer, e->value.integer);
3336 return result;
3340 gfc_expr *
3341 gfc_simplify_iall (gfc_expr *array, gfc_expr *dim, gfc_expr *mask)
3343 return simplify_transformation (array, dim, mask, -1, do_bit_and);
3347 static gfc_expr *
3348 do_bit_ior (gfc_expr *result, gfc_expr *e)
3350 gcc_assert (e->ts.type == BT_INTEGER && e->expr_type == EXPR_CONSTANT);
3351 gcc_assert (result->ts.type == BT_INTEGER
3352 && result->expr_type == EXPR_CONSTANT);
3354 mpz_ior (result->value.integer, result->value.integer, e->value.integer);
3355 return result;
3359 gfc_expr *
3360 gfc_simplify_iany (gfc_expr *array, gfc_expr *dim, gfc_expr *mask)
3362 return simplify_transformation (array, dim, mask, 0, do_bit_ior);
3366 gfc_expr *
3367 gfc_simplify_iand (gfc_expr *x, gfc_expr *y)
3369 gfc_expr *result;
3371 if (x->expr_type != EXPR_CONSTANT || y->expr_type != EXPR_CONSTANT)
3372 return NULL;
3374 result = gfc_get_constant_expr (BT_INTEGER, x->ts.kind, &x->where);
3375 mpz_and (result->value.integer, x->value.integer, y->value.integer);
3377 return range_check (result, "IAND");
3381 gfc_expr *
3382 gfc_simplify_ibclr (gfc_expr *x, gfc_expr *y)
3384 gfc_expr *result;
3385 int k, pos;
3387 if (x->expr_type != EXPR_CONSTANT || y->expr_type != EXPR_CONSTANT)
3388 return NULL;
3390 if (!gfc_check_bitfcn (x, y))
3391 return &gfc_bad_expr;
3393 gfc_extract_int (y, &pos);
3395 k = gfc_validate_kind (x->ts.type, x->ts.kind, false);
3397 result = gfc_copy_expr (x);
3398 /* Drop any separate memory representation of x to avoid potential
3399 inconsistencies in result. */
3400 if (result->representation.string)
3402 free (result->representation.string);
3403 result->representation.string = NULL;
3406 convert_mpz_to_unsigned (result->value.integer,
3407 gfc_integer_kinds[k].bit_size);
3409 mpz_clrbit (result->value.integer, pos);
3411 gfc_convert_mpz_to_signed (result->value.integer,
3412 gfc_integer_kinds[k].bit_size);
3414 return result;
3418 gfc_expr *
3419 gfc_simplify_ibits (gfc_expr *x, gfc_expr *y, gfc_expr *z)
3421 gfc_expr *result;
3422 int pos, len;
3423 int i, k, bitsize;
3424 int *bits;
3426 if (x->expr_type != EXPR_CONSTANT
3427 || y->expr_type != EXPR_CONSTANT
3428 || z->expr_type != EXPR_CONSTANT)
3429 return NULL;
3431 if (!gfc_check_ibits (x, y, z))
3432 return &gfc_bad_expr;
3434 gfc_extract_int (y, &pos);
3435 gfc_extract_int (z, &len);
3437 k = gfc_validate_kind (BT_INTEGER, x->ts.kind, false);
3439 bitsize = gfc_integer_kinds[k].bit_size;
3441 if (pos + len > bitsize)
3443 gfc_error ("Sum of second and third arguments of IBITS exceeds "
3444 "bit size at %L", &y->where);
3445 return &gfc_bad_expr;
3448 result = gfc_get_constant_expr (x->ts.type, x->ts.kind, &x->where);
3449 convert_mpz_to_unsigned (result->value.integer,
3450 gfc_integer_kinds[k].bit_size);
3452 bits = XCNEWVEC (int, bitsize);
3454 for (i = 0; i < bitsize; i++)
3455 bits[i] = 0;
3457 for (i = 0; i < len; i++)
3458 bits[i] = mpz_tstbit (x->value.integer, i + pos);
3460 for (i = 0; i < bitsize; i++)
3462 if (bits[i] == 0)
3463 mpz_clrbit (result->value.integer, i);
3464 else if (bits[i] == 1)
3465 mpz_setbit (result->value.integer, i);
3466 else
3467 gfc_internal_error ("IBITS: Bad bit");
3470 free (bits);
3472 gfc_convert_mpz_to_signed (result->value.integer,
3473 gfc_integer_kinds[k].bit_size);
3475 return result;
3479 gfc_expr *
3480 gfc_simplify_ibset (gfc_expr *x, gfc_expr *y)
3482 gfc_expr *result;
3483 int k, pos;
3485 if (x->expr_type != EXPR_CONSTANT || y->expr_type != EXPR_CONSTANT)
3486 return NULL;
3488 if (!gfc_check_bitfcn (x, y))
3489 return &gfc_bad_expr;
3491 gfc_extract_int (y, &pos);
3493 k = gfc_validate_kind (x->ts.type, x->ts.kind, false);
3495 result = gfc_copy_expr (x);
3496 /* Drop any separate memory representation of x to avoid potential
3497 inconsistencies in result. */
3498 if (result->representation.string)
3500 free (result->representation.string);
3501 result->representation.string = NULL;
3504 convert_mpz_to_unsigned (result->value.integer,
3505 gfc_integer_kinds[k].bit_size);
3507 mpz_setbit (result->value.integer, pos);
3509 gfc_convert_mpz_to_signed (result->value.integer,
3510 gfc_integer_kinds[k].bit_size);
3512 return result;
3516 gfc_expr *
3517 gfc_simplify_ichar (gfc_expr *e, gfc_expr *kind)
3519 gfc_expr *result;
3520 gfc_char_t index;
3521 int k;
3523 if (e->expr_type != EXPR_CONSTANT)
3524 return NULL;
3526 if (e->value.character.length != 1)
3528 gfc_error ("Argument of ICHAR at %L must be of length one", &e->where);
3529 return &gfc_bad_expr;
3532 index = e->value.character.string[0];
3534 k = get_kind (BT_INTEGER, kind, "ICHAR", gfc_default_integer_kind);
3535 if (k == -1)
3536 return &gfc_bad_expr;
3538 result = gfc_get_int_expr (k, &e->where, index);
3540 return range_check (result, "ICHAR");
3544 gfc_expr *
3545 gfc_simplify_ieor (gfc_expr *x, gfc_expr *y)
3547 gfc_expr *result;
3549 if (x->expr_type != EXPR_CONSTANT || y->expr_type != EXPR_CONSTANT)
3550 return NULL;
3552 result = gfc_get_constant_expr (BT_INTEGER, x->ts.kind, &x->where);
3553 mpz_xor (result->value.integer, x->value.integer, y->value.integer);
3555 return range_check (result, "IEOR");
3559 gfc_expr *
3560 gfc_simplify_index (gfc_expr *x, gfc_expr *y, gfc_expr *b, gfc_expr *kind)
3562 gfc_expr *result;
3563 bool back;
3564 HOST_WIDE_INT len, lensub, start, last, i, index = 0;
3565 int k, delta;
3567 if (x->expr_type != EXPR_CONSTANT || y->expr_type != EXPR_CONSTANT
3568 || ( b != NULL && b->expr_type != EXPR_CONSTANT))
3569 return NULL;
3571 back = (b != NULL && b->value.logical != 0);
3573 k = get_kind (BT_INTEGER, kind, "INDEX", gfc_default_integer_kind);
3574 if (k == -1)
3575 return &gfc_bad_expr;
3577 result = gfc_get_constant_expr (BT_INTEGER, k, &x->where);
3579 len = x->value.character.length;
3580 lensub = y->value.character.length;
3582 if (len < lensub)
3584 mpz_set_si (result->value.integer, 0);
3585 return result;
3588 if (lensub == 0)
3590 if (back)
3591 index = len + 1;
3592 else
3593 index = 1;
3594 goto done;
3597 if (!back)
3599 last = len + 1 - lensub;
3600 start = 0;
3601 delta = 1;
3603 else
3605 last = -1;
3606 start = len - lensub;
3607 delta = -1;
3610 for (; start != last; start += delta)
3612 for (i = 0; i < lensub; i++)
3614 if (x->value.character.string[start + i]
3615 != y->value.character.string[i])
3616 break;
3618 if (i == lensub)
3620 index = start + 1;
3621 goto done;
3625 done:
3626 mpz_set_si (result->value.integer, index);
3627 return range_check (result, "INDEX");
3631 static gfc_expr *
3632 simplify_intconv (gfc_expr *e, int kind, const char *name)
3634 gfc_expr *result = NULL;
3635 int tmp1, tmp2;
3637 /* Convert BOZ to integer, and return without range checking. */
3638 if (e->ts.type == BT_BOZ)
3640 if (!gfc_boz2int (e, kind))
3641 return NULL;
3642 result = gfc_copy_expr (e);
3643 return result;
3646 if (e->expr_type != EXPR_CONSTANT)
3647 return NULL;
3649 /* For explicit conversion, turn off -Wconversion and -Wconversion-extra
3650 warnings. */
3651 tmp1 = warn_conversion;
3652 tmp2 = warn_conversion_extra;
3653 warn_conversion = warn_conversion_extra = 0;
3655 result = gfc_convert_constant (e, BT_INTEGER, kind);
3657 warn_conversion = tmp1;
3658 warn_conversion_extra = tmp2;
3660 if (result == &gfc_bad_expr)
3661 return &gfc_bad_expr;
3663 return range_check (result, name);
3667 gfc_expr *
3668 gfc_simplify_int (gfc_expr *e, gfc_expr *k)
3670 int kind;
3672 kind = get_kind (BT_INTEGER, k, "INT", gfc_default_integer_kind);
3673 if (kind == -1)
3674 return &gfc_bad_expr;
3676 return simplify_intconv (e, kind, "INT");
3679 gfc_expr *
3680 gfc_simplify_int2 (gfc_expr *e)
3682 return simplify_intconv (e, 2, "INT2");
3686 gfc_expr *
3687 gfc_simplify_int8 (gfc_expr *e)
3689 return simplify_intconv (e, 8, "INT8");
3693 gfc_expr *
3694 gfc_simplify_long (gfc_expr *e)
3696 return simplify_intconv (e, 4, "LONG");
3700 gfc_expr *
3701 gfc_simplify_ifix (gfc_expr *e)
3703 gfc_expr *rtrunc, *result;
3705 if (e->expr_type != EXPR_CONSTANT)
3706 return NULL;
3708 rtrunc = gfc_copy_expr (e);
3709 mpfr_trunc (rtrunc->value.real, e->value.real);
3711 result = gfc_get_constant_expr (BT_INTEGER, gfc_default_integer_kind,
3712 &e->where);
3713 gfc_mpfr_to_mpz (result->value.integer, rtrunc->value.real, &e->where);
3715 gfc_free_expr (rtrunc);
3717 return range_check (result, "IFIX");
3721 gfc_expr *
3722 gfc_simplify_idint (gfc_expr *e)
3724 gfc_expr *rtrunc, *result;
3726 if (e->expr_type != EXPR_CONSTANT)
3727 return NULL;
3729 rtrunc = gfc_copy_expr (e);
3730 mpfr_trunc (rtrunc->value.real, e->value.real);
3732 result = gfc_get_constant_expr (BT_INTEGER, gfc_default_integer_kind,
3733 &e->where);
3734 gfc_mpfr_to_mpz (result->value.integer, rtrunc->value.real, &e->where);
3736 gfc_free_expr (rtrunc);
3738 return range_check (result, "IDINT");
3742 gfc_expr *
3743 gfc_simplify_ior (gfc_expr *x, gfc_expr *y)
3745 gfc_expr *result;
3747 if (x->expr_type != EXPR_CONSTANT || y->expr_type != EXPR_CONSTANT)
3748 return NULL;
3750 result = gfc_get_constant_expr (BT_INTEGER, x->ts.kind, &x->where);
3751 mpz_ior (result->value.integer, x->value.integer, y->value.integer);
3753 return range_check (result, "IOR");
3757 static gfc_expr *
3758 do_bit_xor (gfc_expr *result, gfc_expr *e)
3760 gcc_assert (e->ts.type == BT_INTEGER && e->expr_type == EXPR_CONSTANT);
3761 gcc_assert (result->ts.type == BT_INTEGER
3762 && result->expr_type == EXPR_CONSTANT);
3764 mpz_xor (result->value.integer, result->value.integer, e->value.integer);
3765 return result;
3769 gfc_expr *
3770 gfc_simplify_iparity (gfc_expr *array, gfc_expr *dim, gfc_expr *mask)
3772 return simplify_transformation (array, dim, mask, 0, do_bit_xor);
3776 gfc_expr *
3777 gfc_simplify_is_iostat_end (gfc_expr *x)
3779 if (x->expr_type != EXPR_CONSTANT)
3780 return NULL;
3782 return gfc_get_logical_expr (gfc_default_logical_kind, &x->where,
3783 mpz_cmp_si (x->value.integer,
3784 LIBERROR_END) == 0);
3788 gfc_expr *
3789 gfc_simplify_is_iostat_eor (gfc_expr *x)
3791 if (x->expr_type != EXPR_CONSTANT)
3792 return NULL;
3794 return gfc_get_logical_expr (gfc_default_logical_kind, &x->where,
3795 mpz_cmp_si (x->value.integer,
3796 LIBERROR_EOR) == 0);
3800 gfc_expr *
3801 gfc_simplify_isnan (gfc_expr *x)
3803 if (x->expr_type != EXPR_CONSTANT)
3804 return NULL;
3806 return gfc_get_logical_expr (gfc_default_logical_kind, &x->where,
3807 mpfr_nan_p (x->value.real));
3811 /* Performs a shift on its first argument. Depending on the last
3812 argument, the shift can be arithmetic, i.e. with filling from the
3813 left like in the SHIFTA intrinsic. */
3814 static gfc_expr *
3815 simplify_shift (gfc_expr *e, gfc_expr *s, const char *name,
3816 bool arithmetic, int direction)
3818 gfc_expr *result;
3819 int ashift, *bits, i, k, bitsize, shift;
3821 if (e->expr_type != EXPR_CONSTANT || s->expr_type != EXPR_CONSTANT)
3822 return NULL;
3824 gfc_extract_int (s, &shift);
3826 k = gfc_validate_kind (BT_INTEGER, e->ts.kind, false);
3827 bitsize = gfc_integer_kinds[k].bit_size;
3829 result = gfc_get_constant_expr (e->ts.type, e->ts.kind, &e->where);
3831 if (shift == 0)
3833 mpz_set (result->value.integer, e->value.integer);
3834 return result;
3837 if (direction > 0 && shift < 0)
3839 /* Left shift, as in SHIFTL. */
3840 gfc_error ("Second argument of %s is negative at %L", name, &e->where);
3841 return &gfc_bad_expr;
3843 else if (direction < 0)
3845 /* Right shift, as in SHIFTR or SHIFTA. */
3846 if (shift < 0)
3848 gfc_error ("Second argument of %s is negative at %L",
3849 name, &e->where);
3850 return &gfc_bad_expr;
3853 shift = -shift;
3856 ashift = (shift >= 0 ? shift : -shift);
3858 if (ashift > bitsize)
3860 gfc_error ("Magnitude of second argument of %s exceeds bit size "
3861 "at %L", name, &e->where);
3862 return &gfc_bad_expr;
3865 bits = XCNEWVEC (int, bitsize);
3867 for (i = 0; i < bitsize; i++)
3868 bits[i] = mpz_tstbit (e->value.integer, i);
3870 if (shift > 0)
3872 /* Left shift. */
3873 for (i = 0; i < shift; i++)
3874 mpz_clrbit (result->value.integer, i);
3876 for (i = 0; i < bitsize - shift; i++)
3878 if (bits[i] == 0)
3879 mpz_clrbit (result->value.integer, i + shift);
3880 else
3881 mpz_setbit (result->value.integer, i + shift);
3884 else
3886 /* Right shift. */
3887 if (arithmetic && bits[bitsize - 1])
3888 for (i = bitsize - 1; i >= bitsize - ashift; i--)
3889 mpz_setbit (result->value.integer, i);
3890 else
3891 for (i = bitsize - 1; i >= bitsize - ashift; i--)
3892 mpz_clrbit (result->value.integer, i);
3894 for (i = bitsize - 1; i >= ashift; i--)
3896 if (bits[i] == 0)
3897 mpz_clrbit (result->value.integer, i - ashift);
3898 else
3899 mpz_setbit (result->value.integer, i - ashift);
3903 gfc_convert_mpz_to_signed (result->value.integer, bitsize);
3904 free (bits);
3906 return result;
3910 gfc_expr *
3911 gfc_simplify_ishft (gfc_expr *e, gfc_expr *s)
3913 return simplify_shift (e, s, "ISHFT", false, 0);
3917 gfc_expr *
3918 gfc_simplify_lshift (gfc_expr *e, gfc_expr *s)
3920 return simplify_shift (e, s, "LSHIFT", false, 1);
3924 gfc_expr *
3925 gfc_simplify_rshift (gfc_expr *e, gfc_expr *s)
3927 return simplify_shift (e, s, "RSHIFT", true, -1);
3931 gfc_expr *
3932 gfc_simplify_shifta (gfc_expr *e, gfc_expr *s)
3934 return simplify_shift (e, s, "SHIFTA", true, -1);
3938 gfc_expr *
3939 gfc_simplify_shiftl (gfc_expr *e, gfc_expr *s)
3941 return simplify_shift (e, s, "SHIFTL", false, 1);
3945 gfc_expr *
3946 gfc_simplify_shiftr (gfc_expr *e, gfc_expr *s)
3948 return simplify_shift (e, s, "SHIFTR", false, -1);
3952 gfc_expr *
3953 gfc_simplify_ishftc (gfc_expr *e, gfc_expr *s, gfc_expr *sz)
3955 gfc_expr *result;
3956 int shift, ashift, isize, ssize, delta, k;
3957 int i, *bits;
3959 if (e->expr_type != EXPR_CONSTANT || s->expr_type != EXPR_CONSTANT)
3960 return NULL;
3962 gfc_extract_int (s, &shift);
3964 k = gfc_validate_kind (e->ts.type, e->ts.kind, false);
3965 isize = gfc_integer_kinds[k].bit_size;
3967 if (sz != NULL)
3969 if (sz->expr_type != EXPR_CONSTANT)
3970 return NULL;
3972 gfc_extract_int (sz, &ssize);
3974 if (ssize > isize || ssize <= 0)
3975 return &gfc_bad_expr;
3977 else
3978 ssize = isize;
3980 if (shift >= 0)
3981 ashift = shift;
3982 else
3983 ashift = -shift;
3985 if (ashift > ssize)
3987 if (sz == NULL)
3988 gfc_error ("Magnitude of second argument of ISHFTC exceeds "
3989 "BIT_SIZE of first argument at %C");
3990 else
3991 gfc_error ("Absolute value of SHIFT shall be less than or equal "
3992 "to SIZE at %C");
3993 return &gfc_bad_expr;
3996 result = gfc_get_constant_expr (e->ts.type, e->ts.kind, &e->where);
3998 mpz_set (result->value.integer, e->value.integer);
4000 if (shift == 0)
4001 return result;
4003 convert_mpz_to_unsigned (result->value.integer, isize);
4005 bits = XCNEWVEC (int, ssize);
4007 for (i = 0; i < ssize; i++)
4008 bits[i] = mpz_tstbit (e->value.integer, i);
4010 delta = ssize - ashift;
4012 if (shift > 0)
4014 for (i = 0; i < delta; i++)
4016 if (bits[i] == 0)
4017 mpz_clrbit (result->value.integer, i + shift);
4018 else
4019 mpz_setbit (result->value.integer, i + shift);
4022 for (i = delta; i < ssize; i++)
4024 if (bits[i] == 0)
4025 mpz_clrbit (result->value.integer, i - delta);
4026 else
4027 mpz_setbit (result->value.integer, i - delta);
4030 else
4032 for (i = 0; i < ashift; i++)
4034 if (bits[i] == 0)
4035 mpz_clrbit (result->value.integer, i + delta);
4036 else
4037 mpz_setbit (result->value.integer, i + delta);
4040 for (i = ashift; i < ssize; i++)
4042 if (bits[i] == 0)
4043 mpz_clrbit (result->value.integer, i + shift);
4044 else
4045 mpz_setbit (result->value.integer, i + shift);
4049 gfc_convert_mpz_to_signed (result->value.integer, isize);
4051 free (bits);
4052 return result;
4056 gfc_expr *
4057 gfc_simplify_kind (gfc_expr *e)
4059 return gfc_get_int_expr (gfc_default_integer_kind, NULL, e->ts.kind);
4063 static gfc_expr *
4064 simplify_bound_dim (gfc_expr *array, gfc_expr *kind, int d, int upper,
4065 gfc_array_spec *as, gfc_ref *ref, bool coarray)
4067 gfc_expr *l, *u, *result;
4068 int k;
4070 k = get_kind (BT_INTEGER, kind, upper ? "UBOUND" : "LBOUND",
4071 gfc_default_integer_kind);
4072 if (k == -1)
4073 return &gfc_bad_expr;
4075 result = gfc_get_constant_expr (BT_INTEGER, k, &array->where);
4077 /* For non-variables, LBOUND(expr, DIM=n) = 1 and
4078 UBOUND(expr, DIM=n) = SIZE(expr, DIM=n). */
4079 if (!coarray && array->expr_type != EXPR_VARIABLE)
4081 if (upper)
4083 gfc_expr* dim = result;
4084 mpz_set_si (dim->value.integer, d);
4086 result = simplify_size (array, dim, k);
4087 gfc_free_expr (dim);
4088 if (!result)
4089 goto returnNull;
4091 else
4092 mpz_set_si (result->value.integer, 1);
4094 goto done;
4097 /* Otherwise, we have a variable expression. */
4098 gcc_assert (array->expr_type == EXPR_VARIABLE);
4099 gcc_assert (as);
4101 if (!gfc_resolve_array_spec (as, 0))
4102 return NULL;
4104 /* The last dimension of an assumed-size array is special. */
4105 if ((!coarray && d == as->rank && as->type == AS_ASSUMED_SIZE && !upper)
4106 || (coarray && d == as->rank + as->corank
4107 && (!upper || flag_coarray == GFC_FCOARRAY_SINGLE)))
4109 if (as->lower[d-1] && as->lower[d-1]->expr_type == EXPR_CONSTANT)
4111 gfc_free_expr (result);
4112 return gfc_copy_expr (as->lower[d-1]);
4115 goto returnNull;
4118 result = gfc_get_constant_expr (BT_INTEGER, k, &array->where);
4120 /* Then, we need to know the extent of the given dimension. */
4121 if (coarray || (ref->u.ar.type == AR_FULL && !ref->next))
4123 gfc_expr *declared_bound;
4124 int empty_bound;
4125 bool constant_lbound, constant_ubound;
4127 l = as->lower[d-1];
4128 u = as->upper[d-1];
4130 gcc_assert (l != NULL);
4132 constant_lbound = l->expr_type == EXPR_CONSTANT;
4133 constant_ubound = u && u->expr_type == EXPR_CONSTANT;
4135 empty_bound = upper ? 0 : 1;
4136 declared_bound = upper ? u : l;
4138 if ((!upper && !constant_lbound)
4139 || (upper && !constant_ubound))
4140 goto returnNull;
4142 if (!coarray)
4144 /* For {L,U}BOUND, the value depends on whether the array
4145 is empty. We can nevertheless simplify if the declared bound
4146 has the same value as that of an empty array, in which case
4147 the result isn't dependent on the array emptiness. */
4148 if (mpz_cmp_si (declared_bound->value.integer, empty_bound) == 0)
4149 mpz_set_si (result->value.integer, empty_bound);
4150 else if (!constant_lbound || !constant_ubound)
4151 /* Array emptiness can't be determined, we can't simplify. */
4152 goto returnNull;
4153 else if (mpz_cmp (l->value.integer, u->value.integer) > 0)
4154 mpz_set_si (result->value.integer, empty_bound);
4155 else
4156 mpz_set (result->value.integer, declared_bound->value.integer);
4158 else
4159 mpz_set (result->value.integer, declared_bound->value.integer);
4161 else
4163 if (upper)
4165 int d2 = 0, cnt = 0;
4166 for (int idx = 0; idx < ref->u.ar.dimen; ++idx)
4168 if (ref->u.ar.dimen_type[idx] == DIMEN_ELEMENT)
4169 d2++;
4170 else if (cnt < d - 1)
4171 cnt++;
4172 else
4173 break;
4175 if (!gfc_ref_dimen_size (&ref->u.ar, d2 + d - 1, &result->value.integer, NULL))
4176 goto returnNull;
4178 else
4179 mpz_set_si (result->value.integer, (long int) 1);
4182 done:
4183 return range_check (result, upper ? "UBOUND" : "LBOUND");
4185 returnNull:
4186 gfc_free_expr (result);
4187 return NULL;
4191 static gfc_expr *
4192 simplify_bound (gfc_expr *array, gfc_expr *dim, gfc_expr *kind, int upper)
4194 gfc_ref *ref;
4195 gfc_array_spec *as;
4196 ar_type type = AR_UNKNOWN;
4197 int d;
4199 if (array->ts.type == BT_CLASS)
4200 return NULL;
4202 if (array->expr_type != EXPR_VARIABLE)
4204 as = NULL;
4205 ref = NULL;
4206 goto done;
4209 /* Do not attempt to resolve if error has already been issued. */
4210 if (array->symtree->n.sym->error)
4211 return NULL;
4213 /* Follow any component references. */
4214 as = array->symtree->n.sym->as;
4215 for (ref = array->ref; ref; ref = ref->next)
4217 switch (ref->type)
4219 case REF_ARRAY:
4220 type = ref->u.ar.type;
4221 switch (ref->u.ar.type)
4223 case AR_ELEMENT:
4224 as = NULL;
4225 continue;
4227 case AR_FULL:
4228 /* We're done because 'as' has already been set in the
4229 previous iteration. */
4230 goto done;
4232 case AR_UNKNOWN:
4233 return NULL;
4235 case AR_SECTION:
4236 as = ref->u.ar.as;
4237 goto done;
4240 gcc_unreachable ();
4242 case REF_COMPONENT:
4243 as = ref->u.c.component->as;
4244 continue;
4246 case REF_SUBSTRING:
4247 case REF_INQUIRY:
4248 continue;
4252 gcc_unreachable ();
4254 done:
4256 if (as && (as->type == AS_DEFERRED || as->type == AS_ASSUMED_RANK
4257 || (as->type == AS_ASSUMED_SHAPE && upper)))
4258 return NULL;
4260 /* 'array' shall not be an unallocated allocatable variable or a pointer that
4261 is not associated. */
4262 if (array->expr_type == EXPR_VARIABLE
4263 && (gfc_expr_attr (array).allocatable || gfc_expr_attr (array).pointer))
4264 return NULL;
4266 gcc_assert (!as
4267 || (as->type != AS_DEFERRED
4268 && array->expr_type == EXPR_VARIABLE
4269 && !gfc_expr_attr (array).allocatable
4270 && !gfc_expr_attr (array).pointer));
4272 if (dim == NULL)
4274 /* Multi-dimensional bounds. */
4275 gfc_expr *bounds[GFC_MAX_DIMENSIONS];
4276 gfc_expr *e;
4277 int k;
4279 /* UBOUND(ARRAY) is not valid for an assumed-size array. */
4280 if (upper && type == AR_FULL && as && as->type == AS_ASSUMED_SIZE)
4282 /* An error message will be emitted in
4283 check_assumed_size_reference (resolve.cc). */
4284 return &gfc_bad_expr;
4287 /* Simplify the bounds for each dimension. */
4288 for (d = 0; d < array->rank; d++)
4290 bounds[d] = simplify_bound_dim (array, kind, d + 1, upper, as, ref,
4291 false);
4292 if (bounds[d] == NULL || bounds[d] == &gfc_bad_expr)
4294 int j;
4296 for (j = 0; j < d; j++)
4297 gfc_free_expr (bounds[j]);
4299 if (gfc_seen_div0)
4300 return &gfc_bad_expr;
4301 else
4302 return bounds[d];
4306 /* Allocate the result expression. */
4307 k = get_kind (BT_INTEGER, kind, upper ? "UBOUND" : "LBOUND",
4308 gfc_default_integer_kind);
4309 if (k == -1)
4310 return &gfc_bad_expr;
4312 e = gfc_get_array_expr (BT_INTEGER, k, &array->where);
4314 /* The result is a rank 1 array; its size is the rank of the first
4315 argument to {L,U}BOUND. */
4316 e->rank = 1;
4317 e->shape = gfc_get_shape (1);
4318 mpz_init_set_ui (e->shape[0], array->rank);
4320 /* Create the constructor for this array. */
4321 for (d = 0; d < array->rank; d++)
4322 gfc_constructor_append_expr (&e->value.constructor,
4323 bounds[d], &e->where);
4325 return e;
4327 else
4329 /* A DIM argument is specified. */
4330 if (dim->expr_type != EXPR_CONSTANT)
4331 return NULL;
4333 d = mpz_get_si (dim->value.integer);
4335 if ((d < 1 || d > array->rank)
4336 || (d == array->rank && as && as->type == AS_ASSUMED_SIZE && upper))
4338 gfc_error ("DIM argument at %L is out of bounds", &dim->where);
4339 return &gfc_bad_expr;
4342 if (as && as->type == AS_ASSUMED_RANK)
4343 return NULL;
4345 return simplify_bound_dim (array, kind, d, upper, as, ref, false);
4350 static gfc_expr *
4351 simplify_cobound (gfc_expr *array, gfc_expr *dim, gfc_expr *kind, int upper)
4353 gfc_ref *ref;
4354 gfc_array_spec *as;
4355 int d;
4357 if (array->expr_type != EXPR_VARIABLE)
4358 return NULL;
4360 /* Follow any component references. */
4361 as = (array->ts.type == BT_CLASS && CLASS_DATA (array))
4362 ? CLASS_DATA (array)->as
4363 : array->symtree->n.sym->as;
4364 for (ref = array->ref; ref; ref = ref->next)
4366 switch (ref->type)
4368 case REF_ARRAY:
4369 switch (ref->u.ar.type)
4371 case AR_ELEMENT:
4372 if (ref->u.ar.as->corank > 0)
4374 gcc_assert (as == ref->u.ar.as);
4375 goto done;
4377 as = NULL;
4378 continue;
4380 case AR_FULL:
4381 /* We're done because 'as' has already been set in the
4382 previous iteration. */
4383 goto done;
4385 case AR_UNKNOWN:
4386 return NULL;
4388 case AR_SECTION:
4389 as = ref->u.ar.as;
4390 goto done;
4393 gcc_unreachable ();
4395 case REF_COMPONENT:
4396 as = ref->u.c.component->as;
4397 continue;
4399 case REF_SUBSTRING:
4400 case REF_INQUIRY:
4401 continue;
4405 if (!as)
4406 gcc_unreachable ();
4408 done:
4410 if (as->cotype == AS_DEFERRED || as->cotype == AS_ASSUMED_SHAPE)
4411 return NULL;
4413 if (dim == NULL)
4415 /* Multi-dimensional cobounds. */
4416 gfc_expr *bounds[GFC_MAX_DIMENSIONS];
4417 gfc_expr *e;
4418 int k;
4420 /* Simplify the cobounds for each dimension. */
4421 for (d = 0; d < as->corank; d++)
4423 bounds[d] = simplify_bound_dim (array, kind, d + 1 + as->rank,
4424 upper, as, ref, true);
4425 if (bounds[d] == NULL || bounds[d] == &gfc_bad_expr)
4427 int j;
4429 for (j = 0; j < d; j++)
4430 gfc_free_expr (bounds[j]);
4431 return bounds[d];
4435 /* Allocate the result expression. */
4436 e = gfc_get_expr ();
4437 e->where = array->where;
4438 e->expr_type = EXPR_ARRAY;
4439 e->ts.type = BT_INTEGER;
4440 k = get_kind (BT_INTEGER, kind, upper ? "UCOBOUND" : "LCOBOUND",
4441 gfc_default_integer_kind);
4442 if (k == -1)
4444 gfc_free_expr (e);
4445 return &gfc_bad_expr;
4447 e->ts.kind = k;
4449 /* The result is a rank 1 array; its size is the rank of the first
4450 argument to {L,U}COBOUND. */
4451 e->rank = 1;
4452 e->shape = gfc_get_shape (1);
4453 mpz_init_set_ui (e->shape[0], as->corank);
4455 /* Create the constructor for this array. */
4456 for (d = 0; d < as->corank; d++)
4457 gfc_constructor_append_expr (&e->value.constructor,
4458 bounds[d], &e->where);
4459 return e;
4461 else
4463 /* A DIM argument is specified. */
4464 if (dim->expr_type != EXPR_CONSTANT)
4465 return NULL;
4467 d = mpz_get_si (dim->value.integer);
4469 if (d < 1 || d > as->corank)
4471 gfc_error ("DIM argument at %L is out of bounds", &dim->where);
4472 return &gfc_bad_expr;
4475 return simplify_bound_dim (array, kind, d+as->rank, upper, as, ref, true);
4480 gfc_expr *
4481 gfc_simplify_lbound (gfc_expr *array, gfc_expr *dim, gfc_expr *kind)
4483 return simplify_bound (array, dim, kind, 0);
4487 gfc_expr *
4488 gfc_simplify_lcobound (gfc_expr *array, gfc_expr *dim, gfc_expr *kind)
4490 return simplify_cobound (array, dim, kind, 0);
4493 gfc_expr *
4494 gfc_simplify_leadz (gfc_expr *e)
4496 unsigned long lz, bs;
4497 int i;
4499 if (e->expr_type != EXPR_CONSTANT)
4500 return NULL;
4502 i = gfc_validate_kind (e->ts.type, e->ts.kind, false);
4503 bs = gfc_integer_kinds[i].bit_size;
4504 if (mpz_cmp_si (e->value.integer, 0) == 0)
4505 lz = bs;
4506 else if (mpz_cmp_si (e->value.integer, 0) < 0)
4507 lz = 0;
4508 else
4509 lz = bs - mpz_sizeinbase (e->value.integer, 2);
4511 return gfc_get_int_expr (gfc_default_integer_kind, &e->where, lz);
4515 /* Check for constant length of a substring. */
4517 static bool
4518 substring_has_constant_len (gfc_expr *e)
4520 gfc_ref *ref;
4521 HOST_WIDE_INT istart, iend, length;
4522 bool equal_length = false;
4524 if (e->ts.type != BT_CHARACTER)
4525 return false;
4527 for (ref = e->ref; ref; ref = ref->next)
4528 if (ref->type != REF_COMPONENT && ref->type != REF_ARRAY)
4529 break;
4531 if (!ref
4532 || ref->type != REF_SUBSTRING
4533 || !ref->u.ss.start
4534 || ref->u.ss.start->expr_type != EXPR_CONSTANT
4535 || !ref->u.ss.end
4536 || ref->u.ss.end->expr_type != EXPR_CONSTANT)
4537 return false;
4539 /* Basic checks on substring starting and ending indices. */
4540 if (!gfc_resolve_substring (ref, &equal_length))
4541 return false;
4543 istart = gfc_mpz_get_hwi (ref->u.ss.start->value.integer);
4544 iend = gfc_mpz_get_hwi (ref->u.ss.end->value.integer);
4546 if (istart <= iend)
4547 length = iend - istart + 1;
4548 else
4549 length = 0;
4551 /* Fix substring length. */
4552 e->value.character.length = length;
4554 return true;
4558 gfc_expr *
4559 gfc_simplify_len (gfc_expr *e, gfc_expr *kind)
4561 gfc_expr *result;
4562 int k = get_kind (BT_INTEGER, kind, "LEN", gfc_default_integer_kind);
4564 if (k == -1)
4565 return &gfc_bad_expr;
4567 if (e->expr_type == EXPR_CONSTANT
4568 || substring_has_constant_len (e))
4570 result = gfc_get_constant_expr (BT_INTEGER, k, &e->where);
4571 mpz_set_si (result->value.integer, e->value.character.length);
4572 return range_check (result, "LEN");
4574 else if (e->ts.u.cl != NULL && e->ts.u.cl->length != NULL
4575 && e->ts.u.cl->length->expr_type == EXPR_CONSTANT
4576 && e->ts.u.cl->length->ts.type == BT_INTEGER)
4578 result = gfc_get_constant_expr (BT_INTEGER, k, &e->where);
4579 mpz_set (result->value.integer, e->ts.u.cl->length->value.integer);
4580 return range_check (result, "LEN");
4582 else if (e->expr_type == EXPR_VARIABLE && e->ts.type == BT_CHARACTER
4583 && e->symtree->n.sym)
4585 if (e->symtree->n.sym->ts.type != BT_DERIVED
4586 && e->symtree->n.sym->assoc && e->symtree->n.sym->assoc->target
4587 && e->symtree->n.sym->assoc->target->ts.type == BT_DERIVED
4588 && e->symtree->n.sym->assoc->target->symtree->n.sym
4589 && UNLIMITED_POLY (e->symtree->n.sym->assoc->target->symtree->n.sym))
4590 /* The expression in assoc->target points to a ref to the _data
4591 component of the unlimited polymorphic entity. To get the _len
4592 component the last _data ref needs to be stripped and a ref to the
4593 _len component added. */
4594 return gfc_get_len_component (e->symtree->n.sym->assoc->target, k);
4595 else if (e->symtree->n.sym->ts.type == BT_DERIVED
4596 && e->ref && e->ref->type == REF_COMPONENT
4597 && e->ref->u.c.component->attr.pdt_string
4598 && e->ref->u.c.component->ts.type == BT_CHARACTER
4599 && e->ref->u.c.component->ts.u.cl->length)
4601 if (gfc_init_expr_flag)
4603 gfc_expr* tmp;
4604 tmp = gfc_pdt_find_component_copy_initializer (e->symtree->n.sym,
4605 e->ref->u.c
4606 .component->ts.u.cl
4607 ->length->symtree
4608 ->name);
4609 if (tmp)
4610 return tmp;
4612 else
4614 gfc_expr *len_expr = gfc_copy_expr (e);
4615 gfc_free_ref_list (len_expr->ref);
4616 len_expr->ref = NULL;
4617 gfc_find_component (len_expr->symtree->n.sym->ts.u.derived, e->ref
4618 ->u.c.component->ts.u.cl->length->symtree
4619 ->name,
4620 false, true, &len_expr->ref);
4621 len_expr->ts = len_expr->ref->u.c.component->ts;
4622 return len_expr;
4626 return NULL;
4630 gfc_expr *
4631 gfc_simplify_len_trim (gfc_expr *e, gfc_expr *kind)
4633 gfc_expr *result;
4634 size_t count, len, i;
4635 int k = get_kind (BT_INTEGER, kind, "LEN_TRIM", gfc_default_integer_kind);
4637 if (k == -1)
4638 return &gfc_bad_expr;
4640 if (e->expr_type != EXPR_CONSTANT)
4641 return NULL;
4643 len = e->value.character.length;
4644 for (count = 0, i = 1; i <= len; i++)
4645 if (e->value.character.string[len - i] == ' ')
4646 count++;
4647 else
4648 break;
4650 result = gfc_get_int_expr (k, &e->where, len - count);
4651 return range_check (result, "LEN_TRIM");
4654 gfc_expr *
4655 gfc_simplify_lgamma (gfc_expr *x)
4657 gfc_expr *result;
4658 int sg;
4660 if (x->expr_type != EXPR_CONSTANT)
4661 return NULL;
4663 result = gfc_get_constant_expr (x->ts.type, x->ts.kind, &x->where);
4664 mpfr_lgamma (result->value.real, &sg, x->value.real, GFC_RND_MODE);
4666 return range_check (result, "LGAMMA");
4670 gfc_expr *
4671 gfc_simplify_lge (gfc_expr *a, gfc_expr *b)
4673 if (a->expr_type != EXPR_CONSTANT || b->expr_type != EXPR_CONSTANT)
4674 return NULL;
4676 return gfc_get_logical_expr (gfc_default_logical_kind, &a->where,
4677 gfc_compare_string (a, b) >= 0);
4681 gfc_expr *
4682 gfc_simplify_lgt (gfc_expr *a, gfc_expr *b)
4684 if (a->expr_type != EXPR_CONSTANT || b->expr_type != EXPR_CONSTANT)
4685 return NULL;
4687 return gfc_get_logical_expr (gfc_default_logical_kind, &a->where,
4688 gfc_compare_string (a, b) > 0);
4692 gfc_expr *
4693 gfc_simplify_lle (gfc_expr *a, gfc_expr *b)
4695 if (a->expr_type != EXPR_CONSTANT || b->expr_type != EXPR_CONSTANT)
4696 return NULL;
4698 return gfc_get_logical_expr (gfc_default_logical_kind, &a->where,
4699 gfc_compare_string (a, b) <= 0);
4703 gfc_expr *
4704 gfc_simplify_llt (gfc_expr *a, gfc_expr *b)
4706 if (a->expr_type != EXPR_CONSTANT || b->expr_type != EXPR_CONSTANT)
4707 return NULL;
4709 return gfc_get_logical_expr (gfc_default_logical_kind, &a->where,
4710 gfc_compare_string (a, b) < 0);
4714 gfc_expr *
4715 gfc_simplify_log (gfc_expr *x)
4717 gfc_expr *result;
4719 if (x->expr_type != EXPR_CONSTANT)
4720 return NULL;
4722 result = gfc_get_constant_expr (x->ts.type, x->ts.kind, &x->where);
4724 switch (x->ts.type)
4726 case BT_REAL:
4727 if (mpfr_sgn (x->value.real) <= 0)
4729 gfc_error ("Argument of LOG at %L cannot be less than or equal "
4730 "to zero", &x->where);
4731 gfc_free_expr (result);
4732 return &gfc_bad_expr;
4735 mpfr_log (result->value.real, x->value.real, GFC_RND_MODE);
4736 break;
4738 case BT_COMPLEX:
4739 if (mpfr_zero_p (mpc_realref (x->value.complex))
4740 && mpfr_zero_p (mpc_imagref (x->value.complex)))
4742 gfc_error ("Complex argument of LOG at %L cannot be zero",
4743 &x->where);
4744 gfc_free_expr (result);
4745 return &gfc_bad_expr;
4748 gfc_set_model_kind (x->ts.kind);
4749 mpc_log (result->value.complex, x->value.complex, GFC_MPC_RND_MODE);
4750 break;
4752 default:
4753 gfc_internal_error ("gfc_simplify_log: bad type");
4756 return range_check (result, "LOG");
4760 gfc_expr *
4761 gfc_simplify_log10 (gfc_expr *x)
4763 gfc_expr *result;
4765 if (x->expr_type != EXPR_CONSTANT)
4766 return NULL;
4768 if (mpfr_sgn (x->value.real) <= 0)
4770 gfc_error ("Argument of LOG10 at %L cannot be less than or equal "
4771 "to zero", &x->where);
4772 return &gfc_bad_expr;
4775 result = gfc_get_constant_expr (x->ts.type, x->ts.kind, &x->where);
4776 mpfr_log10 (result->value.real, x->value.real, GFC_RND_MODE);
4778 return range_check (result, "LOG10");
4782 gfc_expr *
4783 gfc_simplify_logical (gfc_expr *e, gfc_expr *k)
4785 int kind;
4787 kind = get_kind (BT_LOGICAL, k, "LOGICAL", gfc_default_logical_kind);
4788 if (kind < 0)
4789 return &gfc_bad_expr;
4791 if (e->expr_type != EXPR_CONSTANT)
4792 return NULL;
4794 return gfc_get_logical_expr (kind, &e->where, e->value.logical);
4798 gfc_expr*
4799 gfc_simplify_matmul (gfc_expr *matrix_a, gfc_expr *matrix_b)
4801 gfc_expr *result;
4802 int row, result_rows, col, result_columns;
4803 int stride_a, offset_a, stride_b, offset_b;
4805 if (!is_constant_array_expr (matrix_a)
4806 || !is_constant_array_expr (matrix_b))
4807 return NULL;
4809 /* MATMUL should do mixed-mode arithmetic. Set the result type. */
4810 if (matrix_a->ts.type != matrix_b->ts.type)
4812 gfc_expr e;
4813 e.expr_type = EXPR_OP;
4814 gfc_clear_ts (&e.ts);
4815 e.value.op.op = INTRINSIC_NONE;
4816 e.value.op.op1 = matrix_a;
4817 e.value.op.op2 = matrix_b;
4818 gfc_type_convert_binary (&e, 1);
4819 result = gfc_get_array_expr (e.ts.type, e.ts.kind, &matrix_a->where);
4821 else
4823 result = gfc_get_array_expr (matrix_a->ts.type, matrix_a->ts.kind,
4824 &matrix_a->where);
4827 if (matrix_a->rank == 1 && matrix_b->rank == 2)
4829 result_rows = 1;
4830 result_columns = mpz_get_si (matrix_b->shape[1]);
4831 stride_a = 1;
4832 stride_b = mpz_get_si (matrix_b->shape[0]);
4834 result->rank = 1;
4835 result->shape = gfc_get_shape (result->rank);
4836 mpz_init_set_si (result->shape[0], result_columns);
4838 else if (matrix_a->rank == 2 && matrix_b->rank == 1)
4840 result_rows = mpz_get_si (matrix_a->shape[0]);
4841 result_columns = 1;
4842 stride_a = mpz_get_si (matrix_a->shape[0]);
4843 stride_b = 1;
4845 result->rank = 1;
4846 result->shape = gfc_get_shape (result->rank);
4847 mpz_init_set_si (result->shape[0], result_rows);
4849 else if (matrix_a->rank == 2 && matrix_b->rank == 2)
4851 result_rows = mpz_get_si (matrix_a->shape[0]);
4852 result_columns = mpz_get_si (matrix_b->shape[1]);
4853 stride_a = mpz_get_si (matrix_a->shape[0]);
4854 stride_b = mpz_get_si (matrix_b->shape[0]);
4856 result->rank = 2;
4857 result->shape = gfc_get_shape (result->rank);
4858 mpz_init_set_si (result->shape[0], result_rows);
4859 mpz_init_set_si (result->shape[1], result_columns);
4861 else
4862 gcc_unreachable();
4864 offset_b = 0;
4865 for (col = 0; col < result_columns; ++col)
4867 offset_a = 0;
4869 for (row = 0; row < result_rows; ++row)
4871 gfc_expr *e = compute_dot_product (matrix_a, stride_a, offset_a,
4872 matrix_b, 1, offset_b, false);
4873 gfc_constructor_append_expr (&result->value.constructor,
4874 e, NULL);
4876 offset_a += 1;
4879 offset_b += stride_b;
4882 return result;
4886 gfc_expr *
4887 gfc_simplify_maskr (gfc_expr *i, gfc_expr *kind_arg)
4889 gfc_expr *result;
4890 int kind, arg, k;
4892 if (i->expr_type != EXPR_CONSTANT)
4893 return NULL;
4895 kind = get_kind (BT_INTEGER, kind_arg, "MASKR", gfc_default_integer_kind);
4896 if (kind == -1)
4897 return &gfc_bad_expr;
4898 k = gfc_validate_kind (BT_INTEGER, kind, false);
4900 bool fail = gfc_extract_int (i, &arg);
4901 gcc_assert (!fail);
4903 if (!gfc_check_mask (i, kind_arg))
4904 return &gfc_bad_expr;
4906 result = gfc_get_constant_expr (BT_INTEGER, kind, &i->where);
4908 /* MASKR(n) = 2^n - 1 */
4909 mpz_set_ui (result->value.integer, 1);
4910 mpz_mul_2exp (result->value.integer, result->value.integer, arg);
4911 mpz_sub_ui (result->value.integer, result->value.integer, 1);
4913 gfc_convert_mpz_to_signed (result->value.integer, gfc_integer_kinds[k].bit_size);
4915 return result;
4919 gfc_expr *
4920 gfc_simplify_maskl (gfc_expr *i, gfc_expr *kind_arg)
4922 gfc_expr *result;
4923 int kind, arg, k;
4924 mpz_t z;
4926 if (i->expr_type != EXPR_CONSTANT)
4927 return NULL;
4929 kind = get_kind (BT_INTEGER, kind_arg, "MASKL", gfc_default_integer_kind);
4930 if (kind == -1)
4931 return &gfc_bad_expr;
4932 k = gfc_validate_kind (BT_INTEGER, kind, false);
4934 bool fail = gfc_extract_int (i, &arg);
4935 gcc_assert (!fail);
4937 if (!gfc_check_mask (i, kind_arg))
4938 return &gfc_bad_expr;
4940 result = gfc_get_constant_expr (BT_INTEGER, kind, &i->where);
4942 /* MASKL(n) = 2^bit_size - 2^(bit_size - n) */
4943 mpz_init_set_ui (z, 1);
4944 mpz_mul_2exp (z, z, gfc_integer_kinds[k].bit_size);
4945 mpz_set_ui (result->value.integer, 1);
4946 mpz_mul_2exp (result->value.integer, result->value.integer,
4947 gfc_integer_kinds[k].bit_size - arg);
4948 mpz_sub (result->value.integer, z, result->value.integer);
4949 mpz_clear (z);
4951 gfc_convert_mpz_to_signed (result->value.integer, gfc_integer_kinds[k].bit_size);
4953 return result;
4957 gfc_expr *
4958 gfc_simplify_merge (gfc_expr *tsource, gfc_expr *fsource, gfc_expr *mask)
4960 gfc_expr * result;
4961 gfc_constructor *tsource_ctor, *fsource_ctor, *mask_ctor;
4963 if (mask->expr_type == EXPR_CONSTANT)
4965 /* The standard requires evaluation of all function arguments.
4966 Simplify only when the other dropped argument (FSOURCE or TSOURCE)
4967 is a constant expression. */
4968 if (mask->value.logical)
4970 if (!gfc_is_constant_expr (fsource))
4971 return NULL;
4972 result = gfc_copy_expr (tsource);
4974 else
4976 if (!gfc_is_constant_expr (tsource))
4977 return NULL;
4978 result = gfc_copy_expr (fsource);
4981 /* Parenthesis is needed to get lower bounds of 1. */
4982 result = gfc_get_parentheses (result);
4983 gfc_simplify_expr (result, 1);
4984 return result;
4987 if (!mask->rank || !is_constant_array_expr (mask)
4988 || !is_constant_array_expr (tsource) || !is_constant_array_expr (fsource))
4989 return NULL;
4991 result = gfc_get_array_expr (tsource->ts.type, tsource->ts.kind,
4992 &tsource->where);
4993 if (tsource->ts.type == BT_DERIVED)
4994 result->ts.u.derived = tsource->ts.u.derived;
4995 else if (tsource->ts.type == BT_CHARACTER)
4996 result->ts.u.cl = tsource->ts.u.cl;
4998 tsource_ctor = gfc_constructor_first (tsource->value.constructor);
4999 fsource_ctor = gfc_constructor_first (fsource->value.constructor);
5000 mask_ctor = gfc_constructor_first (mask->value.constructor);
5002 while (mask_ctor)
5004 if (mask_ctor->expr->value.logical)
5005 gfc_constructor_append_expr (&result->value.constructor,
5006 gfc_copy_expr (tsource_ctor->expr),
5007 NULL);
5008 else
5009 gfc_constructor_append_expr (&result->value.constructor,
5010 gfc_copy_expr (fsource_ctor->expr),
5011 NULL);
5012 tsource_ctor = gfc_constructor_next (tsource_ctor);
5013 fsource_ctor = gfc_constructor_next (fsource_ctor);
5014 mask_ctor = gfc_constructor_next (mask_ctor);
5017 result->shape = gfc_get_shape (1);
5018 gfc_array_size (result, &result->shape[0]);
5020 return result;
5024 gfc_expr *
5025 gfc_simplify_merge_bits (gfc_expr *i, gfc_expr *j, gfc_expr *mask_expr)
5027 mpz_t arg1, arg2, mask;
5028 gfc_expr *result;
5030 if (i->expr_type != EXPR_CONSTANT || j->expr_type != EXPR_CONSTANT
5031 || mask_expr->expr_type != EXPR_CONSTANT)
5032 return NULL;
5034 result = gfc_get_constant_expr (BT_INTEGER, i->ts.kind, &i->where);
5036 /* Convert all argument to unsigned. */
5037 mpz_init_set (arg1, i->value.integer);
5038 mpz_init_set (arg2, j->value.integer);
5039 mpz_init_set (mask, mask_expr->value.integer);
5041 /* MERGE_BITS(I,J,MASK) = IOR (IAND (I, MASK), IAND (J, NOT (MASK))). */
5042 mpz_and (arg1, arg1, mask);
5043 mpz_com (mask, mask);
5044 mpz_and (arg2, arg2, mask);
5045 mpz_ior (result->value.integer, arg1, arg2);
5047 mpz_clear (arg1);
5048 mpz_clear (arg2);
5049 mpz_clear (mask);
5051 return result;
5055 /* Selects between current value and extremum for simplify_min_max
5056 and simplify_minval_maxval. */
5057 static int
5058 min_max_choose (gfc_expr *arg, gfc_expr *extremum, int sign, bool back_val)
5060 int ret;
5062 switch (arg->ts.type)
5064 case BT_INTEGER:
5065 if (extremum->ts.kind < arg->ts.kind)
5066 extremum->ts.kind = arg->ts.kind;
5067 ret = mpz_cmp (arg->value.integer,
5068 extremum->value.integer) * sign;
5069 if (ret > 0)
5070 mpz_set (extremum->value.integer, arg->value.integer);
5071 break;
5073 case BT_REAL:
5074 if (extremum->ts.kind < arg->ts.kind)
5075 extremum->ts.kind = arg->ts.kind;
5076 if (mpfr_nan_p (extremum->value.real))
5078 ret = 1;
5079 mpfr_set (extremum->value.real, arg->value.real, GFC_RND_MODE);
5081 else if (mpfr_nan_p (arg->value.real))
5082 ret = -1;
5083 else
5085 ret = mpfr_cmp (arg->value.real, extremum->value.real) * sign;
5086 if (ret > 0)
5087 mpfr_set (extremum->value.real, arg->value.real, GFC_RND_MODE);
5089 break;
5091 case BT_CHARACTER:
5092 #define LENGTH(x) ((x)->value.character.length)
5093 #define STRING(x) ((x)->value.character.string)
5094 if (LENGTH (extremum) < LENGTH(arg))
5096 gfc_char_t *tmp = STRING(extremum);
5098 STRING(extremum) = gfc_get_wide_string (LENGTH(arg) + 1);
5099 memcpy (STRING(extremum), tmp,
5100 LENGTH(extremum) * sizeof (gfc_char_t));
5101 gfc_wide_memset (&STRING(extremum)[LENGTH(extremum)], ' ',
5102 LENGTH(arg) - LENGTH(extremum));
5103 STRING(extremum)[LENGTH(arg)] = '\0'; /* For debugger */
5104 LENGTH(extremum) = LENGTH(arg);
5105 free (tmp);
5107 ret = gfc_compare_string (arg, extremum) * sign;
5108 if (ret > 0)
5110 free (STRING(extremum));
5111 STRING(extremum) = gfc_get_wide_string (LENGTH(extremum) + 1);
5112 memcpy (STRING(extremum), STRING(arg),
5113 LENGTH(arg) * sizeof (gfc_char_t));
5114 gfc_wide_memset (&STRING(extremum)[LENGTH(arg)], ' ',
5115 LENGTH(extremum) - LENGTH(arg));
5116 STRING(extremum)[LENGTH(extremum)] = '\0'; /* For debugger */
5118 #undef LENGTH
5119 #undef STRING
5120 break;
5122 default:
5123 gfc_internal_error ("simplify_min_max(): Bad type in arglist");
5125 if (back_val && ret == 0)
5126 ret = 1;
5128 return ret;
5132 /* This function is special since MAX() can take any number of
5133 arguments. The simplified expression is a rewritten version of the
5134 argument list containing at most one constant element. Other
5135 constant elements are deleted. Because the argument list has
5136 already been checked, this function always succeeds. sign is 1 for
5137 MAX(), -1 for MIN(). */
5139 static gfc_expr *
5140 simplify_min_max (gfc_expr *expr, int sign)
5142 int tmp1, tmp2;
5143 gfc_actual_arglist *arg, *last, *extremum;
5144 gfc_expr *tmp, *ret;
5145 const char *fname;
5147 last = NULL;
5148 extremum = NULL;
5150 arg = expr->value.function.actual;
5152 for (; arg; last = arg, arg = arg->next)
5154 if (arg->expr->expr_type != EXPR_CONSTANT)
5155 continue;
5157 if (extremum == NULL)
5159 extremum = arg;
5160 continue;
5163 min_max_choose (arg->expr, extremum->expr, sign);
5165 /* Delete the extra constant argument. */
5166 last->next = arg->next;
5168 arg->next = NULL;
5169 gfc_free_actual_arglist (arg);
5170 arg = last;
5173 /* If there is one value left, replace the function call with the
5174 expression. */
5175 if (expr->value.function.actual->next != NULL)
5176 return NULL;
5178 /* Handle special cases of specific functions (min|max)1 and
5179 a(min|max)0. */
5181 tmp = expr->value.function.actual->expr;
5182 fname = expr->value.function.isym->name;
5184 if ((tmp->ts.type != BT_INTEGER || tmp->ts.kind != gfc_integer_4_kind)
5185 && (strcmp (fname, "min1") == 0 || strcmp (fname, "max1") == 0))
5187 /* Explicit conversion, turn off -Wconversion and -Wconversion-extra
5188 warnings. */
5189 tmp1 = warn_conversion;
5190 tmp2 = warn_conversion_extra;
5191 warn_conversion = warn_conversion_extra = 0;
5193 ret = gfc_convert_constant (tmp, BT_INTEGER, gfc_integer_4_kind);
5195 warn_conversion = tmp1;
5196 warn_conversion_extra = tmp2;
5198 else if ((tmp->ts.type != BT_REAL || tmp->ts.kind != gfc_real_4_kind)
5199 && (strcmp (fname, "amin0") == 0 || strcmp (fname, "amax0") == 0))
5201 ret = gfc_convert_constant (tmp, BT_REAL, gfc_real_4_kind);
5203 else
5204 ret = gfc_copy_expr (tmp);
5206 return ret;
5211 gfc_expr *
5212 gfc_simplify_min (gfc_expr *e)
5214 return simplify_min_max (e, -1);
5218 gfc_expr *
5219 gfc_simplify_max (gfc_expr *e)
5221 return simplify_min_max (e, 1);
5224 /* Helper function for gfc_simplify_minval. */
5226 static gfc_expr *
5227 gfc_min (gfc_expr *op1, gfc_expr *op2)
5229 min_max_choose (op1, op2, -1);
5230 gfc_free_expr (op1);
5231 return op2;
5234 /* Simplify minval for constant arrays. */
5236 gfc_expr *
5237 gfc_simplify_minval (gfc_expr *array, gfc_expr* dim, gfc_expr *mask)
5239 return simplify_transformation (array, dim, mask, INT_MAX, gfc_min);
5242 /* Helper function for gfc_simplify_maxval. */
5244 static gfc_expr *
5245 gfc_max (gfc_expr *op1, gfc_expr *op2)
5247 min_max_choose (op1, op2, 1);
5248 gfc_free_expr (op1);
5249 return op2;
5253 /* Simplify maxval for constant arrays. */
5255 gfc_expr *
5256 gfc_simplify_maxval (gfc_expr *array, gfc_expr* dim, gfc_expr *mask)
5258 return simplify_transformation (array, dim, mask, INT_MIN, gfc_max);
5262 /* Transform minloc or maxloc of an array, according to MASK,
5263 to the scalar result. This code is mostly identical to
5264 simplify_transformation_to_scalar. */
5266 static gfc_expr *
5267 simplify_minmaxloc_to_scalar (gfc_expr *result, gfc_expr *array, gfc_expr *mask,
5268 gfc_expr *extremum, int sign, bool back_val)
5270 gfc_expr *a, *m;
5271 gfc_constructor *array_ctor, *mask_ctor;
5272 mpz_t count;
5274 mpz_set_si (result->value.integer, 0);
5277 /* Shortcut for constant .FALSE. MASK. */
5278 if (mask
5279 && mask->expr_type == EXPR_CONSTANT
5280 && !mask->value.logical)
5281 return result;
5283 array_ctor = gfc_constructor_first (array->value.constructor);
5284 if (mask && mask->expr_type == EXPR_ARRAY)
5285 mask_ctor = gfc_constructor_first (mask->value.constructor);
5286 else
5287 mask_ctor = NULL;
5289 mpz_init_set_si (count, 0);
5290 while (array_ctor)
5292 mpz_add_ui (count, count, 1);
5293 a = array_ctor->expr;
5294 array_ctor = gfc_constructor_next (array_ctor);
5295 /* A constant MASK equals .TRUE. here and can be ignored. */
5296 if (mask_ctor)
5298 m = mask_ctor->expr;
5299 mask_ctor = gfc_constructor_next (mask_ctor);
5300 if (!m->value.logical)
5301 continue;
5303 if (min_max_choose (a, extremum, sign, back_val) > 0)
5304 mpz_set (result->value.integer, count);
5306 mpz_clear (count);
5307 gfc_free_expr (extremum);
5308 return result;
5311 /* Simplify minloc / maxloc in the absence of a dim argument. */
5313 static gfc_expr *
5314 simplify_minmaxloc_nodim (gfc_expr *result, gfc_expr *extremum,
5315 gfc_expr *array, gfc_expr *mask, int sign,
5316 bool back_val)
5318 ssize_t res[GFC_MAX_DIMENSIONS];
5319 int i, n;
5320 gfc_constructor *result_ctor, *array_ctor, *mask_ctor;
5321 ssize_t count[GFC_MAX_DIMENSIONS], extent[GFC_MAX_DIMENSIONS],
5322 sstride[GFC_MAX_DIMENSIONS];
5323 gfc_expr *a, *m;
5324 bool continue_loop;
5325 bool ma;
5327 for (i = 0; i<array->rank; i++)
5328 res[i] = -1;
5330 /* Shortcut for constant .FALSE. MASK. */
5331 if (mask
5332 && mask->expr_type == EXPR_CONSTANT
5333 && !mask->value.logical)
5334 goto finish;
5336 if (array->shape == NULL)
5337 goto finish;
5339 for (i = 0; i < array->rank; i++)
5341 count[i] = 0;
5342 sstride[i] = (i == 0) ? 1 : sstride[i-1] * mpz_get_si (array->shape[i-1]);
5343 extent[i] = mpz_get_si (array->shape[i]);
5344 if (extent[i] <= 0)
5345 goto finish;
5348 continue_loop = true;
5349 array_ctor = gfc_constructor_first (array->value.constructor);
5350 if (mask && mask->rank > 0)
5351 mask_ctor = gfc_constructor_first (mask->value.constructor);
5352 else
5353 mask_ctor = NULL;
5355 /* Loop over the array elements (and mask), keeping track of
5356 the indices to return. */
5357 while (continue_loop)
5361 a = array_ctor->expr;
5362 if (mask_ctor)
5364 m = mask_ctor->expr;
5365 ma = m->value.logical;
5366 mask_ctor = gfc_constructor_next (mask_ctor);
5368 else
5369 ma = true;
5371 if (ma && min_max_choose (a, extremum, sign, back_val) > 0)
5373 for (i = 0; i<array->rank; i++)
5374 res[i] = count[i];
5376 array_ctor = gfc_constructor_next (array_ctor);
5377 count[0] ++;
5378 } while (count[0] != extent[0]);
5379 n = 0;
5382 /* When we get to the end of a dimension, reset it and increment
5383 the next dimension. */
5384 count[n] = 0;
5385 n++;
5386 if (n >= array->rank)
5388 continue_loop = false;
5389 break;
5391 else
5392 count[n] ++;
5393 } while (count[n] == extent[n]);
5396 finish:
5397 gfc_free_expr (extremum);
5398 result_ctor = gfc_constructor_first (result->value.constructor);
5399 for (i = 0; i<array->rank; i++)
5401 gfc_expr *r_expr;
5402 r_expr = result_ctor->expr;
5403 mpz_set_si (r_expr->value.integer, res[i] + 1);
5404 result_ctor = gfc_constructor_next (result_ctor);
5406 return result;
5409 /* Helper function for gfc_simplify_minmaxloc - build an array
5410 expression with n elements. */
5412 static gfc_expr *
5413 new_array (bt type, int kind, int n, locus *where)
5415 gfc_expr *result;
5416 int i;
5418 result = gfc_get_array_expr (type, kind, where);
5419 result->rank = 1;
5420 result->shape = gfc_get_shape(1);
5421 mpz_init_set_si (result->shape[0], n);
5422 for (i = 0; i < n; i++)
5424 gfc_constructor_append_expr (&result->value.constructor,
5425 gfc_get_constant_expr (type, kind, where),
5426 NULL);
5429 return result;
5432 /* Simplify minloc and maxloc. This code is mostly identical to
5433 simplify_transformation_to_array. */
5435 static gfc_expr *
5436 simplify_minmaxloc_to_array (gfc_expr *result, gfc_expr *array,
5437 gfc_expr *dim, gfc_expr *mask,
5438 gfc_expr *extremum, int sign, bool back_val)
5440 mpz_t size;
5441 int done, i, n, arraysize, resultsize, dim_index, dim_extent, dim_stride;
5442 gfc_expr **arrayvec, **resultvec, **base, **src, **dest;
5443 gfc_constructor *array_ctor, *mask_ctor, *result_ctor;
5445 int count[GFC_MAX_DIMENSIONS], extent[GFC_MAX_DIMENSIONS],
5446 sstride[GFC_MAX_DIMENSIONS], dstride[GFC_MAX_DIMENSIONS],
5447 tmpstride[GFC_MAX_DIMENSIONS];
5449 /* Shortcut for constant .FALSE. MASK. */
5450 if (mask
5451 && mask->expr_type == EXPR_CONSTANT
5452 && !mask->value.logical)
5453 return result;
5455 /* Build an indexed table for array element expressions to minimize
5456 linked-list traversal. Masked elements are set to NULL. */
5457 gfc_array_size (array, &size);
5458 arraysize = mpz_get_ui (size);
5459 mpz_clear (size);
5461 arrayvec = XCNEWVEC (gfc_expr*, arraysize);
5463 array_ctor = gfc_constructor_first (array->value.constructor);
5464 mask_ctor = NULL;
5465 if (mask && mask->expr_type == EXPR_ARRAY)
5466 mask_ctor = gfc_constructor_first (mask->value.constructor);
5468 for (i = 0; i < arraysize; ++i)
5470 arrayvec[i] = array_ctor->expr;
5471 array_ctor = gfc_constructor_next (array_ctor);
5473 if (mask_ctor)
5475 if (!mask_ctor->expr->value.logical)
5476 arrayvec[i] = NULL;
5478 mask_ctor = gfc_constructor_next (mask_ctor);
5482 /* Same for the result expression. */
5483 gfc_array_size (result, &size);
5484 resultsize = mpz_get_ui (size);
5485 mpz_clear (size);
5487 resultvec = XCNEWVEC (gfc_expr*, resultsize);
5488 result_ctor = gfc_constructor_first (result->value.constructor);
5489 for (i = 0; i < resultsize; ++i)
5491 resultvec[i] = result_ctor->expr;
5492 result_ctor = gfc_constructor_next (result_ctor);
5495 gfc_extract_int (dim, &dim_index);
5496 dim_index -= 1; /* zero-base index */
5497 dim_extent = 0;
5498 dim_stride = 0;
5500 for (i = 0, n = 0; i < array->rank; ++i)
5502 count[i] = 0;
5503 tmpstride[i] = (i == 0) ? 1 : tmpstride[i-1] * mpz_get_si (array->shape[i-1]);
5504 if (i == dim_index)
5506 dim_extent = mpz_get_si (array->shape[i]);
5507 dim_stride = tmpstride[i];
5508 continue;
5511 extent[n] = mpz_get_si (array->shape[i]);
5512 sstride[n] = tmpstride[i];
5513 dstride[n] = (n == 0) ? 1 : dstride[n-1] * extent[n-1];
5514 n += 1;
5517 done = resultsize <= 0;
5518 base = arrayvec;
5519 dest = resultvec;
5520 while (!done)
5522 gfc_expr *ex;
5523 ex = gfc_copy_expr (extremum);
5524 for (src = base, n = 0; n < dim_extent; src += dim_stride, ++n)
5526 if (*src && min_max_choose (*src, ex, sign, back_val) > 0)
5527 mpz_set_si ((*dest)->value.integer, n + 1);
5530 count[0]++;
5531 base += sstride[0];
5532 dest += dstride[0];
5533 gfc_free_expr (ex);
5535 n = 0;
5536 while (!done && count[n] == extent[n])
5538 count[n] = 0;
5539 base -= sstride[n] * extent[n];
5540 dest -= dstride[n] * extent[n];
5542 n++;
5543 if (n < result->rank)
5545 /* If the nested loop is unrolled GFC_MAX_DIMENSIONS
5546 times, we'd warn for the last iteration, because the
5547 array index will have already been incremented to the
5548 array sizes, and we can't tell that this must make
5549 the test against result->rank false, because ranks
5550 must not exceed GFC_MAX_DIMENSIONS. */
5551 GCC_DIAGNOSTIC_PUSH_IGNORED (-Warray-bounds)
5552 count[n]++;
5553 base += sstride[n];
5554 dest += dstride[n];
5555 GCC_DIAGNOSTIC_POP
5557 else
5558 done = true;
5562 /* Place updated expression in result constructor. */
5563 result_ctor = gfc_constructor_first (result->value.constructor);
5564 for (i = 0; i < resultsize; ++i)
5566 result_ctor->expr = resultvec[i];
5567 result_ctor = gfc_constructor_next (result_ctor);
5570 free (arrayvec);
5571 free (resultvec);
5572 free (extremum);
5573 return result;
5576 /* Simplify minloc and maxloc for constant arrays. */
5578 static gfc_expr *
5579 gfc_simplify_minmaxloc (gfc_expr *array, gfc_expr *dim, gfc_expr *mask,
5580 gfc_expr *kind, gfc_expr *back, int sign)
5582 gfc_expr *result;
5583 gfc_expr *extremum;
5584 int ikind;
5585 int init_val;
5586 bool back_val = false;
5588 if (!is_constant_array_expr (array)
5589 || !gfc_is_constant_expr (dim))
5590 return NULL;
5592 if (mask
5593 && !is_constant_array_expr (mask)
5594 && mask->expr_type != EXPR_CONSTANT)
5595 return NULL;
5597 if (kind)
5599 if (gfc_extract_int (kind, &ikind, -1))
5600 return NULL;
5602 else
5603 ikind = gfc_default_integer_kind;
5605 if (back)
5607 if (back->expr_type != EXPR_CONSTANT)
5608 return NULL;
5610 back_val = back->value.logical;
5613 if (sign < 0)
5614 init_val = INT_MAX;
5615 else if (sign > 0)
5616 init_val = INT_MIN;
5617 else
5618 gcc_unreachable();
5620 extremum = gfc_get_constant_expr (array->ts.type, array->ts.kind, &array->where);
5621 init_result_expr (extremum, init_val, array);
5623 if (dim)
5625 result = transformational_result (array, dim, BT_INTEGER,
5626 ikind, &array->where);
5627 init_result_expr (result, 0, array);
5629 if (array->rank == 1)
5630 return simplify_minmaxloc_to_scalar (result, array, mask, extremum,
5631 sign, back_val);
5632 else
5633 return simplify_minmaxloc_to_array (result, array, dim, mask, extremum,
5634 sign, back_val);
5636 else
5638 result = new_array (BT_INTEGER, ikind, array->rank, &array->where);
5639 return simplify_minmaxloc_nodim (result, extremum, array, mask,
5640 sign, back_val);
5644 gfc_expr *
5645 gfc_simplify_minloc (gfc_expr *array, gfc_expr *dim, gfc_expr *mask, gfc_expr *kind,
5646 gfc_expr *back)
5648 return gfc_simplify_minmaxloc (array, dim, mask, kind, back, -1);
5651 gfc_expr *
5652 gfc_simplify_maxloc (gfc_expr *array, gfc_expr *dim, gfc_expr *mask, gfc_expr *kind,
5653 gfc_expr *back)
5655 return gfc_simplify_minmaxloc (array, dim, mask, kind, back, 1);
5658 /* Simplify findloc to scalar. Similar to
5659 simplify_minmaxloc_to_scalar. */
5661 static gfc_expr *
5662 simplify_findloc_to_scalar (gfc_expr *result, gfc_expr *array, gfc_expr *value,
5663 gfc_expr *mask, int back_val)
5665 gfc_expr *a, *m;
5666 gfc_constructor *array_ctor, *mask_ctor;
5667 mpz_t count;
5669 mpz_set_si (result->value.integer, 0);
5671 /* Shortcut for constant .FALSE. MASK. */
5672 if (mask
5673 && mask->expr_type == EXPR_CONSTANT
5674 && !mask->value.logical)
5675 return result;
5677 array_ctor = gfc_constructor_first (array->value.constructor);
5678 if (mask && mask->expr_type == EXPR_ARRAY)
5679 mask_ctor = gfc_constructor_first (mask->value.constructor);
5680 else
5681 mask_ctor = NULL;
5683 mpz_init_set_si (count, 0);
5684 while (array_ctor)
5686 mpz_add_ui (count, count, 1);
5687 a = array_ctor->expr;
5688 array_ctor = gfc_constructor_next (array_ctor);
5689 /* A constant MASK equals .TRUE. here and can be ignored. */
5690 if (mask_ctor)
5692 m = mask_ctor->expr;
5693 mask_ctor = gfc_constructor_next (mask_ctor);
5694 if (!m->value.logical)
5695 continue;
5697 if (gfc_compare_expr (a, value, INTRINSIC_EQ) == 0)
5699 /* We have a match. If BACK is true, continue so we find
5700 the last one. */
5701 mpz_set (result->value.integer, count);
5702 if (!back_val)
5703 break;
5706 mpz_clear (count);
5707 return result;
5710 /* Simplify findloc in the absence of a dim argument. Similar to
5711 simplify_minmaxloc_nodim. */
5713 static gfc_expr *
5714 simplify_findloc_nodim (gfc_expr *result, gfc_expr *value, gfc_expr *array,
5715 gfc_expr *mask, bool back_val)
5717 ssize_t res[GFC_MAX_DIMENSIONS];
5718 int i, n;
5719 gfc_constructor *result_ctor, *array_ctor, *mask_ctor;
5720 ssize_t count[GFC_MAX_DIMENSIONS], extent[GFC_MAX_DIMENSIONS],
5721 sstride[GFC_MAX_DIMENSIONS];
5722 gfc_expr *a, *m;
5723 bool continue_loop;
5724 bool ma;
5726 for (i = 0; i < array->rank; i++)
5727 res[i] = -1;
5729 /* Shortcut for constant .FALSE. MASK. */
5730 if (mask
5731 && mask->expr_type == EXPR_CONSTANT
5732 && !mask->value.logical)
5733 goto finish;
5735 for (i = 0; i < array->rank; i++)
5737 count[i] = 0;
5738 sstride[i] = (i == 0) ? 1 : sstride[i-1] * mpz_get_si (array->shape[i-1]);
5739 extent[i] = mpz_get_si (array->shape[i]);
5740 if (extent[i] <= 0)
5741 goto finish;
5744 continue_loop = true;
5745 array_ctor = gfc_constructor_first (array->value.constructor);
5746 if (mask && mask->rank > 0)
5747 mask_ctor = gfc_constructor_first (mask->value.constructor);
5748 else
5749 mask_ctor = NULL;
5751 /* Loop over the array elements (and mask), keeping track of
5752 the indices to return. */
5753 while (continue_loop)
5757 a = array_ctor->expr;
5758 if (mask_ctor)
5760 m = mask_ctor->expr;
5761 ma = m->value.logical;
5762 mask_ctor = gfc_constructor_next (mask_ctor);
5764 else
5765 ma = true;
5767 if (ma && gfc_compare_expr (a, value, INTRINSIC_EQ) == 0)
5769 for (i = 0; i < array->rank; i++)
5770 res[i] = count[i];
5771 if (!back_val)
5772 goto finish;
5774 array_ctor = gfc_constructor_next (array_ctor);
5775 count[0] ++;
5776 } while (count[0] != extent[0]);
5777 n = 0;
5780 /* When we get to the end of a dimension, reset it and increment
5781 the next dimension. */
5782 count[n] = 0;
5783 n++;
5784 if (n >= array->rank)
5786 continue_loop = false;
5787 break;
5789 else
5790 count[n] ++;
5791 } while (count[n] == extent[n]);
5794 finish:
5795 result_ctor = gfc_constructor_first (result->value.constructor);
5796 for (i = 0; i < array->rank; i++)
5798 gfc_expr *r_expr;
5799 r_expr = result_ctor->expr;
5800 mpz_set_si (r_expr->value.integer, res[i] + 1);
5801 result_ctor = gfc_constructor_next (result_ctor);
5803 return result;
5807 /* Simplify findloc to an array. Similar to
5808 simplify_minmaxloc_to_array. */
5810 static gfc_expr *
5811 simplify_findloc_to_array (gfc_expr *result, gfc_expr *array, gfc_expr *value,
5812 gfc_expr *dim, gfc_expr *mask, bool back_val)
5814 mpz_t size;
5815 int done, i, n, arraysize, resultsize, dim_index, dim_extent, dim_stride;
5816 gfc_expr **arrayvec, **resultvec, **base, **src, **dest;
5817 gfc_constructor *array_ctor, *mask_ctor, *result_ctor;
5819 int count[GFC_MAX_DIMENSIONS], extent[GFC_MAX_DIMENSIONS],
5820 sstride[GFC_MAX_DIMENSIONS], dstride[GFC_MAX_DIMENSIONS],
5821 tmpstride[GFC_MAX_DIMENSIONS];
5823 /* Shortcut for constant .FALSE. MASK. */
5824 if (mask
5825 && mask->expr_type == EXPR_CONSTANT
5826 && !mask->value.logical)
5827 return result;
5829 /* Build an indexed table for array element expressions to minimize
5830 linked-list traversal. Masked elements are set to NULL. */
5831 gfc_array_size (array, &size);
5832 arraysize = mpz_get_ui (size);
5833 mpz_clear (size);
5835 arrayvec = XCNEWVEC (gfc_expr*, arraysize);
5837 array_ctor = gfc_constructor_first (array->value.constructor);
5838 mask_ctor = NULL;
5839 if (mask && mask->expr_type == EXPR_ARRAY)
5840 mask_ctor = gfc_constructor_first (mask->value.constructor);
5842 for (i = 0; i < arraysize; ++i)
5844 arrayvec[i] = array_ctor->expr;
5845 array_ctor = gfc_constructor_next (array_ctor);
5847 if (mask_ctor)
5849 if (!mask_ctor->expr->value.logical)
5850 arrayvec[i] = NULL;
5852 mask_ctor = gfc_constructor_next (mask_ctor);
5856 /* Same for the result expression. */
5857 gfc_array_size (result, &size);
5858 resultsize = mpz_get_ui (size);
5859 mpz_clear (size);
5861 resultvec = XCNEWVEC (gfc_expr*, resultsize);
5862 result_ctor = gfc_constructor_first (result->value.constructor);
5863 for (i = 0; i < resultsize; ++i)
5865 resultvec[i] = result_ctor->expr;
5866 result_ctor = gfc_constructor_next (result_ctor);
5869 gfc_extract_int (dim, &dim_index);
5871 dim_index -= 1; /* Zero-base index. */
5872 dim_extent = 0;
5873 dim_stride = 0;
5875 for (i = 0, n = 0; i < array->rank; ++i)
5877 count[i] = 0;
5878 tmpstride[i] = (i == 0) ? 1 : tmpstride[i-1] * mpz_get_si (array->shape[i-1]);
5879 if (i == dim_index)
5881 dim_extent = mpz_get_si (array->shape[i]);
5882 dim_stride = tmpstride[i];
5883 continue;
5886 extent[n] = mpz_get_si (array->shape[i]);
5887 sstride[n] = tmpstride[i];
5888 dstride[n] = (n == 0) ? 1 : dstride[n-1] * extent[n-1];
5889 n += 1;
5892 done = resultsize <= 0;
5893 base = arrayvec;
5894 dest = resultvec;
5895 while (!done)
5897 for (src = base, n = 0; n < dim_extent; src += dim_stride, ++n)
5899 if (*src && gfc_compare_expr (*src, value, INTRINSIC_EQ) == 0)
5901 mpz_set_si ((*dest)->value.integer, n + 1);
5902 if (!back_val)
5903 break;
5907 count[0]++;
5908 base += sstride[0];
5909 dest += dstride[0];
5911 n = 0;
5912 while (!done && count[n] == extent[n])
5914 count[n] = 0;
5915 base -= sstride[n] * extent[n];
5916 dest -= dstride[n] * extent[n];
5918 n++;
5919 if (n < result->rank)
5921 /* If the nested loop is unrolled GFC_MAX_DIMENSIONS
5922 times, we'd warn for the last iteration, because the
5923 array index will have already been incremented to the
5924 array sizes, and we can't tell that this must make
5925 the test against result->rank false, because ranks
5926 must not exceed GFC_MAX_DIMENSIONS. */
5927 GCC_DIAGNOSTIC_PUSH_IGNORED (-Warray-bounds)
5928 count[n]++;
5929 base += sstride[n];
5930 dest += dstride[n];
5931 GCC_DIAGNOSTIC_POP
5933 else
5934 done = true;
5938 /* Place updated expression in result constructor. */
5939 result_ctor = gfc_constructor_first (result->value.constructor);
5940 for (i = 0; i < resultsize; ++i)
5942 result_ctor->expr = resultvec[i];
5943 result_ctor = gfc_constructor_next (result_ctor);
5946 free (arrayvec);
5947 free (resultvec);
5948 return result;
5951 /* Simplify findloc. */
5953 gfc_expr *
5954 gfc_simplify_findloc (gfc_expr *array, gfc_expr *value, gfc_expr *dim,
5955 gfc_expr *mask, gfc_expr *kind, gfc_expr *back)
5957 gfc_expr *result;
5958 int ikind;
5959 bool back_val = false;
5961 if (!is_constant_array_expr (array)
5962 || array->shape == NULL
5963 || !gfc_is_constant_expr (dim))
5964 return NULL;
5966 if (! gfc_is_constant_expr (value))
5967 return 0;
5969 if (mask
5970 && !is_constant_array_expr (mask)
5971 && mask->expr_type != EXPR_CONSTANT)
5972 return NULL;
5974 if (kind)
5976 if (gfc_extract_int (kind, &ikind, -1))
5977 return NULL;
5979 else
5980 ikind = gfc_default_integer_kind;
5982 if (back)
5984 if (back->expr_type != EXPR_CONSTANT)
5985 return NULL;
5987 back_val = back->value.logical;
5990 if (dim)
5992 result = transformational_result (array, dim, BT_INTEGER,
5993 ikind, &array->where);
5994 init_result_expr (result, 0, array);
5996 if (array->rank == 1)
5997 return simplify_findloc_to_scalar (result, array, value, mask,
5998 back_val);
5999 else
6000 return simplify_findloc_to_array (result, array, value, dim, mask,
6001 back_val);
6003 else
6005 result = new_array (BT_INTEGER, ikind, array->rank, &array->where);
6006 return simplify_findloc_nodim (result, value, array, mask, back_val);
6008 return NULL;
6011 gfc_expr *
6012 gfc_simplify_maxexponent (gfc_expr *x)
6014 int i = gfc_validate_kind (BT_REAL, x->ts.kind, false);
6015 return gfc_get_int_expr (gfc_default_integer_kind, &x->where,
6016 gfc_real_kinds[i].max_exponent);
6020 gfc_expr *
6021 gfc_simplify_minexponent (gfc_expr *x)
6023 int i = gfc_validate_kind (BT_REAL, x->ts.kind, false);
6024 return gfc_get_int_expr (gfc_default_integer_kind, &x->where,
6025 gfc_real_kinds[i].min_exponent);
6029 gfc_expr *
6030 gfc_simplify_mod (gfc_expr *a, gfc_expr *p)
6032 gfc_expr *result;
6033 int kind;
6035 /* First check p. */
6036 if (p->expr_type != EXPR_CONSTANT)
6037 return NULL;
6039 /* p shall not be 0. */
6040 switch (p->ts.type)
6042 case BT_INTEGER:
6043 if (mpz_cmp_ui (p->value.integer, 0) == 0)
6045 gfc_error ("Argument %qs of MOD at %L shall not be zero",
6046 "P", &p->where);
6047 return &gfc_bad_expr;
6049 break;
6050 case BT_REAL:
6051 if (mpfr_cmp_ui (p->value.real, 0) == 0)
6053 gfc_error ("Argument %qs of MOD at %L shall not be zero",
6054 "P", &p->where);
6055 return &gfc_bad_expr;
6057 break;
6058 default:
6059 gfc_internal_error ("gfc_simplify_mod(): Bad arguments");
6062 if (a->expr_type != EXPR_CONSTANT)
6063 return NULL;
6065 kind = a->ts.kind > p->ts.kind ? a->ts.kind : p->ts.kind;
6066 result = gfc_get_constant_expr (a->ts.type, kind, &a->where);
6068 if (a->ts.type == BT_INTEGER)
6069 mpz_tdiv_r (result->value.integer, a->value.integer, p->value.integer);
6070 else
6072 gfc_set_model_kind (kind);
6073 mpfr_fmod (result->value.real, a->value.real, p->value.real,
6074 GFC_RND_MODE);
6077 return range_check (result, "MOD");
6081 gfc_expr *
6082 gfc_simplify_modulo (gfc_expr *a, gfc_expr *p)
6084 gfc_expr *result;
6085 int kind;
6087 /* First check p. */
6088 if (p->expr_type != EXPR_CONSTANT)
6089 return NULL;
6091 /* p shall not be 0. */
6092 switch (p->ts.type)
6094 case BT_INTEGER:
6095 if (mpz_cmp_ui (p->value.integer, 0) == 0)
6097 gfc_error ("Argument %qs of MODULO at %L shall not be zero",
6098 "P", &p->where);
6099 return &gfc_bad_expr;
6101 break;
6102 case BT_REAL:
6103 if (mpfr_cmp_ui (p->value.real, 0) == 0)
6105 gfc_error ("Argument %qs of MODULO at %L shall not be zero",
6106 "P", &p->where);
6107 return &gfc_bad_expr;
6109 break;
6110 default:
6111 gfc_internal_error ("gfc_simplify_modulo(): Bad arguments");
6114 if (a->expr_type != EXPR_CONSTANT)
6115 return NULL;
6117 kind = a->ts.kind > p->ts.kind ? a->ts.kind : p->ts.kind;
6118 result = gfc_get_constant_expr (a->ts.type, kind, &a->where);
6120 if (a->ts.type == BT_INTEGER)
6121 mpz_fdiv_r (result->value.integer, a->value.integer, p->value.integer);
6122 else
6124 gfc_set_model_kind (kind);
6125 mpfr_fmod (result->value.real, a->value.real, p->value.real,
6126 GFC_RND_MODE);
6127 if (mpfr_cmp_ui (result->value.real, 0) != 0)
6129 if (mpfr_signbit (a->value.real) != mpfr_signbit (p->value.real))
6130 mpfr_add (result->value.real, result->value.real, p->value.real,
6131 GFC_RND_MODE);
6133 else
6134 mpfr_copysign (result->value.real, result->value.real,
6135 p->value.real, GFC_RND_MODE);
6138 return range_check (result, "MODULO");
6142 gfc_expr *
6143 gfc_simplify_nearest (gfc_expr *x, gfc_expr *s)
6145 gfc_expr *result;
6146 mpfr_exp_t emin, emax;
6147 int kind;
6149 if (x->expr_type != EXPR_CONSTANT || s->expr_type != EXPR_CONSTANT)
6150 return NULL;
6152 result = gfc_copy_expr (x);
6154 /* Save current values of emin and emax. */
6155 emin = mpfr_get_emin ();
6156 emax = mpfr_get_emax ();
6158 /* Set emin and emax for the current model number. */
6159 kind = gfc_validate_kind (BT_REAL, x->ts.kind, 0);
6160 mpfr_set_emin ((mpfr_exp_t) gfc_real_kinds[kind].min_exponent -
6161 mpfr_get_prec(result->value.real) + 1);
6162 mpfr_set_emax ((mpfr_exp_t) gfc_real_kinds[kind].max_exponent);
6163 mpfr_check_range (result->value.real, 0, MPFR_RNDU);
6165 if (mpfr_sgn (s->value.real) > 0)
6167 mpfr_nextabove (result->value.real);
6168 mpfr_subnormalize (result->value.real, 0, MPFR_RNDU);
6170 else
6172 mpfr_nextbelow (result->value.real);
6173 mpfr_subnormalize (result->value.real, 0, MPFR_RNDD);
6176 mpfr_set_emin (emin);
6177 mpfr_set_emax (emax);
6179 /* Only NaN can occur. Do not use range check as it gives an
6180 error for denormal numbers. */
6181 if (mpfr_nan_p (result->value.real) && flag_range_check)
6183 gfc_error ("Result of NEAREST is NaN at %L", &result->where);
6184 gfc_free_expr (result);
6185 return &gfc_bad_expr;
6188 return result;
6192 static gfc_expr *
6193 simplify_nint (const char *name, gfc_expr *e, gfc_expr *k)
6195 gfc_expr *itrunc, *result;
6196 int kind;
6198 kind = get_kind (BT_INTEGER, k, name, gfc_default_integer_kind);
6199 if (kind == -1)
6200 return &gfc_bad_expr;
6202 if (e->expr_type != EXPR_CONSTANT)
6203 return NULL;
6205 itrunc = gfc_copy_expr (e);
6206 mpfr_round (itrunc->value.real, e->value.real);
6208 result = gfc_get_constant_expr (BT_INTEGER, kind, &e->where);
6209 gfc_mpfr_to_mpz (result->value.integer, itrunc->value.real, &e->where);
6211 gfc_free_expr (itrunc);
6213 return range_check (result, name);
6217 gfc_expr *
6218 gfc_simplify_new_line (gfc_expr *e)
6220 gfc_expr *result;
6222 result = gfc_get_character_expr (e->ts.kind, &e->where, NULL, 1);
6223 result->value.character.string[0] = '\n';
6225 return result;
6229 gfc_expr *
6230 gfc_simplify_nint (gfc_expr *e, gfc_expr *k)
6232 return simplify_nint ("NINT", e, k);
6236 gfc_expr *
6237 gfc_simplify_idnint (gfc_expr *e)
6239 return simplify_nint ("IDNINT", e, NULL);
6242 static int norm2_scale;
6244 static gfc_expr *
6245 norm2_add_squared (gfc_expr *result, gfc_expr *e)
6247 mpfr_t tmp;
6249 gcc_assert (e->ts.type == BT_REAL && e->expr_type == EXPR_CONSTANT);
6250 gcc_assert (result->ts.type == BT_REAL
6251 && result->expr_type == EXPR_CONSTANT);
6253 gfc_set_model_kind (result->ts.kind);
6254 int index = gfc_validate_kind (BT_REAL, result->ts.kind, false);
6255 mpfr_exp_t exp;
6256 if (mpfr_regular_p (result->value.real))
6258 exp = mpfr_get_exp (result->value.real);
6259 /* If result is getting close to overflowing, scale down. */
6260 if (exp >= gfc_real_kinds[index].max_exponent - 4
6261 && norm2_scale <= gfc_real_kinds[index].max_exponent - 2)
6263 norm2_scale += 2;
6264 mpfr_div_ui (result->value.real, result->value.real, 16,
6265 GFC_RND_MODE);
6269 mpfr_init (tmp);
6270 if (mpfr_regular_p (e->value.real))
6272 exp = mpfr_get_exp (e->value.real);
6273 /* If e**2 would overflow or close to overflowing, scale down. */
6274 if (exp - norm2_scale >= gfc_real_kinds[index].max_exponent / 2 - 2)
6276 int new_scale = gfc_real_kinds[index].max_exponent / 2 + 4;
6277 mpfr_set_ui (tmp, 1, GFC_RND_MODE);
6278 mpfr_set_exp (tmp, new_scale - norm2_scale);
6279 mpfr_div (result->value.real, result->value.real, tmp, GFC_RND_MODE);
6280 mpfr_div (result->value.real, result->value.real, tmp, GFC_RND_MODE);
6281 norm2_scale = new_scale;
6284 if (norm2_scale)
6286 mpfr_set_ui (tmp, 1, GFC_RND_MODE);
6287 mpfr_set_exp (tmp, norm2_scale);
6288 mpfr_div (tmp, e->value.real, tmp, GFC_RND_MODE);
6290 else
6291 mpfr_set (tmp, e->value.real, GFC_RND_MODE);
6292 mpfr_pow_ui (tmp, tmp, 2, GFC_RND_MODE);
6293 mpfr_add (result->value.real, result->value.real, tmp,
6294 GFC_RND_MODE);
6295 mpfr_clear (tmp);
6297 return result;
6301 static gfc_expr *
6302 norm2_do_sqrt (gfc_expr *result, gfc_expr *e)
6304 gcc_assert (e->ts.type == BT_REAL && e->expr_type == EXPR_CONSTANT);
6305 gcc_assert (result->ts.type == BT_REAL
6306 && result->expr_type == EXPR_CONSTANT);
6308 if (result != e)
6309 mpfr_set (result->value.real, e->value.real, GFC_RND_MODE);
6310 mpfr_sqrt (result->value.real, result->value.real, GFC_RND_MODE);
6311 if (norm2_scale && mpfr_regular_p (result->value.real))
6313 mpfr_t tmp;
6314 mpfr_init (tmp);
6315 mpfr_set_ui (tmp, 1, GFC_RND_MODE);
6316 mpfr_set_exp (tmp, norm2_scale);
6317 mpfr_mul (result->value.real, result->value.real, tmp, GFC_RND_MODE);
6318 mpfr_clear (tmp);
6320 norm2_scale = 0;
6322 return result;
6326 gfc_expr *
6327 gfc_simplify_norm2 (gfc_expr *e, gfc_expr *dim)
6329 gfc_expr *result;
6330 bool size_zero;
6332 size_zero = gfc_is_size_zero_array (e);
6334 if (!(is_constant_array_expr (e) || size_zero)
6335 || (dim != NULL && !gfc_is_constant_expr (dim)))
6336 return NULL;
6338 result = transformational_result (e, dim, e->ts.type, e->ts.kind, &e->where);
6339 init_result_expr (result, 0, NULL);
6341 if (size_zero)
6342 return result;
6344 norm2_scale = 0;
6345 if (!dim || e->rank == 1)
6347 result = simplify_transformation_to_scalar (result, e, NULL,
6348 norm2_add_squared);
6349 mpfr_sqrt (result->value.real, result->value.real, GFC_RND_MODE);
6350 if (norm2_scale && mpfr_regular_p (result->value.real))
6352 mpfr_t tmp;
6353 mpfr_init (tmp);
6354 mpfr_set_ui (tmp, 1, GFC_RND_MODE);
6355 mpfr_set_exp (tmp, norm2_scale);
6356 mpfr_mul (result->value.real, result->value.real, tmp, GFC_RND_MODE);
6357 mpfr_clear (tmp);
6359 norm2_scale = 0;
6361 else
6362 result = simplify_transformation_to_array (result, e, dim, NULL,
6363 norm2_add_squared,
6364 norm2_do_sqrt);
6366 return result;
6370 gfc_expr *
6371 gfc_simplify_not (gfc_expr *e)
6373 gfc_expr *result;
6375 if (e->expr_type != EXPR_CONSTANT)
6376 return NULL;
6378 result = gfc_get_constant_expr (e->ts.type, e->ts.kind, &e->where);
6379 mpz_com (result->value.integer, e->value.integer);
6381 return range_check (result, "NOT");
6385 gfc_expr *
6386 gfc_simplify_null (gfc_expr *mold)
6388 gfc_expr *result;
6390 if (mold)
6392 result = gfc_copy_expr (mold);
6393 result->expr_type = EXPR_NULL;
6395 else
6396 result = gfc_get_null_expr (NULL);
6398 return result;
6402 gfc_expr *
6403 gfc_simplify_num_images (gfc_expr *distance ATTRIBUTE_UNUSED, gfc_expr *failed)
6405 gfc_expr *result;
6407 if (flag_coarray == GFC_FCOARRAY_NONE)
6409 gfc_fatal_error ("Coarrays disabled at %C, use %<-fcoarray=%> to enable");
6410 return &gfc_bad_expr;
6413 if (flag_coarray != GFC_FCOARRAY_SINGLE)
6414 return NULL;
6416 if (failed && failed->expr_type != EXPR_CONSTANT)
6417 return NULL;
6419 /* FIXME: gfc_current_locus is wrong. */
6420 result = gfc_get_constant_expr (BT_INTEGER, gfc_default_integer_kind,
6421 &gfc_current_locus);
6423 if (failed && failed->value.logical != 0)
6424 mpz_set_si (result->value.integer, 0);
6425 else
6426 mpz_set_si (result->value.integer, 1);
6428 return result;
6432 gfc_expr *
6433 gfc_simplify_or (gfc_expr *x, gfc_expr *y)
6435 gfc_expr *result;
6436 int kind;
6438 if (x->expr_type != EXPR_CONSTANT || y->expr_type != EXPR_CONSTANT)
6439 return NULL;
6441 kind = x->ts.kind > y->ts.kind ? x->ts.kind : y->ts.kind;
6443 switch (x->ts.type)
6445 case BT_INTEGER:
6446 result = gfc_get_constant_expr (BT_INTEGER, kind, &x->where);
6447 mpz_ior (result->value.integer, x->value.integer, y->value.integer);
6448 return range_check (result, "OR");
6450 case BT_LOGICAL:
6451 return gfc_get_logical_expr (kind, &x->where,
6452 x->value.logical || y->value.logical);
6453 default:
6454 gcc_unreachable();
6459 gfc_expr *
6460 gfc_simplify_pack (gfc_expr *array, gfc_expr *mask, gfc_expr *vector)
6462 gfc_expr *result;
6463 gfc_constructor *array_ctor, *mask_ctor, *vector_ctor;
6465 if (!is_constant_array_expr (array)
6466 || !is_constant_array_expr (vector)
6467 || (!gfc_is_constant_expr (mask)
6468 && !is_constant_array_expr (mask)))
6469 return NULL;
6471 result = gfc_get_array_expr (array->ts.type, array->ts.kind, &array->where);
6472 if (array->ts.type == BT_DERIVED)
6473 result->ts.u.derived = array->ts.u.derived;
6475 array_ctor = gfc_constructor_first (array->value.constructor);
6476 vector_ctor = vector
6477 ? gfc_constructor_first (vector->value.constructor)
6478 : NULL;
6480 if (mask->expr_type == EXPR_CONSTANT
6481 && mask->value.logical)
6483 /* Copy all elements of ARRAY to RESULT. */
6484 while (array_ctor)
6486 gfc_constructor_append_expr (&result->value.constructor,
6487 gfc_copy_expr (array_ctor->expr),
6488 NULL);
6490 array_ctor = gfc_constructor_next (array_ctor);
6491 vector_ctor = gfc_constructor_next (vector_ctor);
6494 else if (mask->expr_type == EXPR_ARRAY)
6496 /* Copy only those elements of ARRAY to RESULT whose
6497 MASK equals .TRUE.. */
6498 mask_ctor = gfc_constructor_first (mask->value.constructor);
6499 while (mask_ctor && array_ctor)
6501 if (mask_ctor->expr->value.logical)
6503 gfc_constructor_append_expr (&result->value.constructor,
6504 gfc_copy_expr (array_ctor->expr),
6505 NULL);
6506 vector_ctor = gfc_constructor_next (vector_ctor);
6509 array_ctor = gfc_constructor_next (array_ctor);
6510 mask_ctor = gfc_constructor_next (mask_ctor);
6514 /* Append any left-over elements from VECTOR to RESULT. */
6515 while (vector_ctor)
6517 gfc_constructor_append_expr (&result->value.constructor,
6518 gfc_copy_expr (vector_ctor->expr),
6519 NULL);
6520 vector_ctor = gfc_constructor_next (vector_ctor);
6523 result->shape = gfc_get_shape (1);
6524 gfc_array_size (result, &result->shape[0]);
6526 if (array->ts.type == BT_CHARACTER)
6527 result->ts.u.cl = array->ts.u.cl;
6529 return result;
6533 static gfc_expr *
6534 do_xor (gfc_expr *result, gfc_expr *e)
6536 gcc_assert (e->ts.type == BT_LOGICAL && e->expr_type == EXPR_CONSTANT);
6537 gcc_assert (result->ts.type == BT_LOGICAL
6538 && result->expr_type == EXPR_CONSTANT);
6540 result->value.logical = result->value.logical != e->value.logical;
6541 return result;
6545 gfc_expr *
6546 gfc_simplify_is_contiguous (gfc_expr *array)
6548 if (gfc_is_simply_contiguous (array, false, true))
6549 return gfc_get_logical_expr (gfc_default_logical_kind, &array->where, 1);
6551 if (gfc_is_not_contiguous (array))
6552 return gfc_get_logical_expr (gfc_default_logical_kind, &array->where, 0);
6554 return NULL;
6558 gfc_expr *
6559 gfc_simplify_parity (gfc_expr *e, gfc_expr *dim)
6561 return simplify_transformation (e, dim, NULL, 0, do_xor);
6565 gfc_expr *
6566 gfc_simplify_popcnt (gfc_expr *e)
6568 int res, k;
6569 mpz_t x;
6571 if (e->expr_type != EXPR_CONSTANT)
6572 return NULL;
6574 k = gfc_validate_kind (e->ts.type, e->ts.kind, false);
6576 /* Convert argument to unsigned, then count the '1' bits. */
6577 mpz_init_set (x, e->value.integer);
6578 convert_mpz_to_unsigned (x, gfc_integer_kinds[k].bit_size);
6579 res = mpz_popcount (x);
6580 mpz_clear (x);
6582 return gfc_get_int_expr (gfc_default_integer_kind, &e->where, res);
6586 gfc_expr *
6587 gfc_simplify_poppar (gfc_expr *e)
6589 gfc_expr *popcnt;
6590 int i;
6592 if (e->expr_type != EXPR_CONSTANT)
6593 return NULL;
6595 popcnt = gfc_simplify_popcnt (e);
6596 gcc_assert (popcnt);
6598 bool fail = gfc_extract_int (popcnt, &i);
6599 gcc_assert (!fail);
6601 return gfc_get_int_expr (gfc_default_integer_kind, &e->where, i % 2);
6605 gfc_expr *
6606 gfc_simplify_precision (gfc_expr *e)
6608 int i = gfc_validate_kind (e->ts.type, e->ts.kind, false);
6609 return gfc_get_int_expr (gfc_default_integer_kind, &e->where,
6610 gfc_real_kinds[i].precision);
6614 gfc_expr *
6615 gfc_simplify_product (gfc_expr *array, gfc_expr *dim, gfc_expr *mask)
6617 return simplify_transformation (array, dim, mask, 1, gfc_multiply);
6621 gfc_expr *
6622 gfc_simplify_radix (gfc_expr *e)
6624 int i;
6625 i = gfc_validate_kind (e->ts.type, e->ts.kind, false);
6627 switch (e->ts.type)
6629 case BT_INTEGER:
6630 i = gfc_integer_kinds[i].radix;
6631 break;
6633 case BT_REAL:
6634 i = gfc_real_kinds[i].radix;
6635 break;
6637 default:
6638 gcc_unreachable ();
6641 return gfc_get_int_expr (gfc_default_integer_kind, &e->where, i);
6645 gfc_expr *
6646 gfc_simplify_range (gfc_expr *e)
6648 int i;
6649 i = gfc_validate_kind (e->ts.type, e->ts.kind, false);
6651 switch (e->ts.type)
6653 case BT_INTEGER:
6654 i = gfc_integer_kinds[i].range;
6655 break;
6657 case BT_REAL:
6658 case BT_COMPLEX:
6659 i = gfc_real_kinds[i].range;
6660 break;
6662 default:
6663 gcc_unreachable ();
6666 return gfc_get_int_expr (gfc_default_integer_kind, &e->where, i);
6670 gfc_expr *
6671 gfc_simplify_rank (gfc_expr *e)
6673 /* Assumed rank. */
6674 if (e->rank == -1)
6675 return NULL;
6677 return gfc_get_int_expr (gfc_default_integer_kind, &e->where, e->rank);
6681 gfc_expr *
6682 gfc_simplify_real (gfc_expr *e, gfc_expr *k)
6684 gfc_expr *result = NULL;
6685 int kind, tmp1, tmp2;
6687 /* Convert BOZ to real, and return without range checking. */
6688 if (e->ts.type == BT_BOZ)
6690 /* Determine kind for conversion of the BOZ. */
6691 if (k)
6692 gfc_extract_int (k, &kind);
6693 else
6694 kind = gfc_default_real_kind;
6696 if (!gfc_boz2real (e, kind))
6697 return NULL;
6698 result = gfc_copy_expr (e);
6699 return result;
6702 if (e->ts.type == BT_COMPLEX)
6703 kind = get_kind (BT_REAL, k, "REAL", e->ts.kind);
6704 else
6705 kind = get_kind (BT_REAL, k, "REAL", gfc_default_real_kind);
6707 if (kind == -1)
6708 return &gfc_bad_expr;
6710 if (e->expr_type != EXPR_CONSTANT)
6711 return NULL;
6713 /* For explicit conversion, turn off -Wconversion and -Wconversion-extra
6714 warnings. */
6715 tmp1 = warn_conversion;
6716 tmp2 = warn_conversion_extra;
6717 warn_conversion = warn_conversion_extra = 0;
6719 result = gfc_convert_constant (e, BT_REAL, kind);
6721 warn_conversion = tmp1;
6722 warn_conversion_extra = tmp2;
6724 if (result == &gfc_bad_expr)
6725 return &gfc_bad_expr;
6727 return range_check (result, "REAL");
6731 gfc_expr *
6732 gfc_simplify_realpart (gfc_expr *e)
6734 gfc_expr *result;
6736 if (e->expr_type != EXPR_CONSTANT)
6737 return NULL;
6739 result = gfc_get_constant_expr (BT_REAL, e->ts.kind, &e->where);
6740 mpc_real (result->value.real, e->value.complex, GFC_RND_MODE);
6742 return range_check (result, "REALPART");
6745 gfc_expr *
6746 gfc_simplify_repeat (gfc_expr *e, gfc_expr *n)
6748 gfc_expr *result;
6749 gfc_charlen_t len;
6750 mpz_t ncopies;
6751 bool have_length = false;
6753 /* If NCOPIES isn't a constant, there's nothing we can do. */
6754 if (n->expr_type != EXPR_CONSTANT)
6755 return NULL;
6757 /* If NCOPIES is negative, it's an error. */
6758 if (mpz_sgn (n->value.integer) < 0)
6760 gfc_error ("Argument NCOPIES of REPEAT intrinsic is negative at %L",
6761 &n->where);
6762 return &gfc_bad_expr;
6765 /* If we don't know the character length, we can do no more. */
6766 if (e->ts.u.cl && e->ts.u.cl->length
6767 && e->ts.u.cl->length->expr_type == EXPR_CONSTANT)
6769 len = gfc_mpz_get_hwi (e->ts.u.cl->length->value.integer);
6770 have_length = true;
6772 else if (e->expr_type == EXPR_CONSTANT
6773 && (e->ts.u.cl == NULL || e->ts.u.cl->length == NULL))
6775 len = e->value.character.length;
6777 else
6778 return NULL;
6780 /* If the source length is 0, any value of NCOPIES is valid
6781 and everything behaves as if NCOPIES == 0. */
6782 mpz_init (ncopies);
6783 if (len == 0)
6784 mpz_set_ui (ncopies, 0);
6785 else
6786 mpz_set (ncopies, n->value.integer);
6788 /* Check that NCOPIES isn't too large. */
6789 if (len)
6791 mpz_t max, mlen;
6792 int i;
6794 /* Compute the maximum value allowed for NCOPIES: huge(cl) / len. */
6795 mpz_init (max);
6796 i = gfc_validate_kind (BT_INTEGER, gfc_charlen_int_kind, false);
6798 if (have_length)
6800 mpz_tdiv_q (max, gfc_integer_kinds[i].huge,
6801 e->ts.u.cl->length->value.integer);
6803 else
6805 mpz_init (mlen);
6806 gfc_mpz_set_hwi (mlen, len);
6807 mpz_tdiv_q (max, gfc_integer_kinds[i].huge, mlen);
6808 mpz_clear (mlen);
6811 /* The check itself. */
6812 if (mpz_cmp (ncopies, max) > 0)
6814 mpz_clear (max);
6815 mpz_clear (ncopies);
6816 gfc_error ("Argument NCOPIES of REPEAT intrinsic is too large at %L",
6817 &n->where);
6818 return &gfc_bad_expr;
6821 mpz_clear (max);
6823 mpz_clear (ncopies);
6825 /* For further simplification, we need the character string to be
6826 constant. */
6827 if (e->expr_type != EXPR_CONSTANT)
6828 return NULL;
6830 HOST_WIDE_INT ncop;
6831 if (len ||
6832 (e->ts.u.cl->length &&
6833 mpz_sgn (e->ts.u.cl->length->value.integer) != 0))
6835 bool fail = gfc_extract_hwi (n, &ncop);
6836 gcc_assert (!fail);
6838 else
6839 ncop = 0;
6841 if (ncop == 0)
6842 return gfc_get_character_expr (e->ts.kind, &e->where, NULL, 0);
6844 len = e->value.character.length;
6845 gfc_charlen_t nlen = ncop * len;
6847 /* Here's a semi-arbitrary limit. If the string is longer than 1 GB
6848 (2**28 elements * 4 bytes (wide chars) per element) defer to
6849 runtime instead of consuming (unbounded) memory and CPU at
6850 compile time. */
6851 if (nlen > 268435456)
6853 gfc_warning_now (0, "Evaluation of string longer than 2**28 at %L"
6854 " deferred to runtime, expect bugs", &e->where);
6855 return NULL;
6858 result = gfc_get_character_expr (e->ts.kind, &e->where, NULL, nlen);
6859 for (size_t i = 0; i < (size_t) ncop; i++)
6860 for (size_t j = 0; j < (size_t) len; j++)
6861 result->value.character.string[j+i*len]= e->value.character.string[j];
6863 result->value.character.string[nlen] = '\0'; /* For debugger */
6864 return result;
6868 /* This one is a bear, but mainly has to do with shuffling elements. */
6870 gfc_expr *
6871 gfc_simplify_reshape (gfc_expr *source, gfc_expr *shape_exp,
6872 gfc_expr *pad, gfc_expr *order_exp)
6874 int order[GFC_MAX_DIMENSIONS], shape[GFC_MAX_DIMENSIONS];
6875 int i, rank, npad, x[GFC_MAX_DIMENSIONS];
6876 mpz_t index, size;
6877 unsigned long j;
6878 size_t nsource;
6879 gfc_expr *e, *result;
6880 bool zerosize = false;
6882 /* Check that argument expression types are OK. */
6883 if (!is_constant_array_expr (source)
6884 || !is_constant_array_expr (shape_exp)
6885 || !is_constant_array_expr (pad)
6886 || !is_constant_array_expr (order_exp))
6887 return NULL;
6889 if (source->shape == NULL)
6890 return NULL;
6892 /* Proceed with simplification, unpacking the array. */
6894 mpz_init (index);
6895 rank = 0;
6897 for (i = 0; i < GFC_MAX_DIMENSIONS; i++)
6898 x[i] = 0;
6900 for (;;)
6902 e = gfc_constructor_lookup_expr (shape_exp->value.constructor, rank);
6903 if (e == NULL)
6904 break;
6906 gfc_extract_int (e, &shape[rank]);
6908 gcc_assert (rank >= 0 && rank < GFC_MAX_DIMENSIONS);
6909 if (shape[rank] < 0)
6911 gfc_error ("The SHAPE array for the RESHAPE intrinsic at %L has a "
6912 "negative value %d for dimension %d",
6913 &shape_exp->where, shape[rank], rank+1);
6914 mpz_clear (index);
6915 return &gfc_bad_expr;
6918 rank++;
6921 gcc_assert (rank > 0);
6923 /* Now unpack the order array if present. */
6924 if (order_exp == NULL)
6926 for (i = 0; i < rank; i++)
6927 order[i] = i;
6929 else
6931 mpz_t size;
6932 int order_size, shape_size;
6934 if (order_exp->rank != shape_exp->rank)
6936 gfc_error ("Shapes of ORDER at %L and SHAPE at %L are different",
6937 &order_exp->where, &shape_exp->where);
6938 mpz_clear (index);
6939 return &gfc_bad_expr;
6942 gfc_array_size (shape_exp, &size);
6943 shape_size = mpz_get_ui (size);
6944 mpz_clear (size);
6945 gfc_array_size (order_exp, &size);
6946 order_size = mpz_get_ui (size);
6947 mpz_clear (size);
6948 if (order_size != shape_size)
6950 gfc_error ("Sizes of ORDER at %L and SHAPE at %L are different",
6951 &order_exp->where, &shape_exp->where);
6952 mpz_clear (index);
6953 return &gfc_bad_expr;
6956 for (i = 0; i < rank; i++)
6958 e = gfc_constructor_lookup_expr (order_exp->value.constructor, i);
6959 gcc_assert (e);
6961 gfc_extract_int (e, &order[i]);
6963 if (order[i] < 1 || order[i] > rank)
6965 gfc_error ("Element with a value of %d in ORDER at %L must be "
6966 "in the range [1, ..., %d] for the RESHAPE intrinsic "
6967 "near %L", order[i], &order_exp->where, rank,
6968 &shape_exp->where);
6969 mpz_clear (index);
6970 return &gfc_bad_expr;
6973 order[i]--;
6974 if (x[order[i]] != 0)
6976 gfc_error ("ORDER at %L is not a permutation of the size of "
6977 "SHAPE at %L", &order_exp->where, &shape_exp->where);
6978 mpz_clear (index);
6979 return &gfc_bad_expr;
6981 x[order[i]] = 1;
6985 /* Count the elements in the source and padding arrays. */
6987 npad = 0;
6988 if (pad != NULL)
6990 gfc_array_size (pad, &size);
6991 npad = mpz_get_ui (size);
6992 mpz_clear (size);
6995 gfc_array_size (source, &size);
6996 nsource = mpz_get_ui (size);
6997 mpz_clear (size);
6999 /* If it weren't for that pesky permutation we could just loop
7000 through the source and round out any shortage with pad elements.
7001 But no, someone just had to have the compiler do something the
7002 user should be doing. */
7004 for (i = 0; i < rank; i++)
7005 x[i] = 0;
7007 result = gfc_get_array_expr (source->ts.type, source->ts.kind,
7008 &source->where);
7009 if (source->ts.type == BT_DERIVED)
7010 result->ts.u.derived = source->ts.u.derived;
7011 if (source->ts.type == BT_CHARACTER && result->ts.u.cl == NULL)
7012 result->ts = source->ts;
7013 result->rank = rank;
7014 result->shape = gfc_get_shape (rank);
7015 for (i = 0; i < rank; i++)
7017 mpz_init_set_ui (result->shape[i], shape[i]);
7018 if (shape[i] == 0)
7019 zerosize = true;
7022 if (zerosize)
7023 goto sizezero;
7025 while (nsource > 0 || npad > 0)
7027 /* Figure out which element to extract. */
7028 mpz_set_ui (index, 0);
7030 for (i = rank - 1; i >= 0; i--)
7032 mpz_add_ui (index, index, x[order[i]]);
7033 if (i != 0)
7034 mpz_mul_ui (index, index, shape[order[i - 1]]);
7037 if (mpz_cmp_ui (index, INT_MAX) > 0)
7038 gfc_internal_error ("Reshaped array too large at %C");
7040 j = mpz_get_ui (index);
7042 if (j < nsource)
7043 e = gfc_constructor_lookup_expr (source->value.constructor, j);
7044 else
7046 if (npad <= 0)
7048 mpz_clear (index);
7049 if (pad == NULL)
7050 gfc_error ("Without padding, there are not enough elements "
7051 "in the intrinsic RESHAPE source at %L to match "
7052 "the shape", &source->where);
7053 gfc_free_expr (result);
7054 return NULL;
7056 j = j - nsource;
7057 j = j % npad;
7058 e = gfc_constructor_lookup_expr (pad->value.constructor, j);
7060 gcc_assert (e);
7062 gfc_constructor_append_expr (&result->value.constructor,
7063 gfc_copy_expr (e), &e->where);
7065 /* Calculate the next element. */
7066 i = 0;
7068 inc:
7069 if (++x[i] < shape[i])
7070 continue;
7071 x[i++] = 0;
7072 if (i < rank)
7073 goto inc;
7075 break;
7078 sizezero:
7080 mpz_clear (index);
7082 return result;
7086 gfc_expr *
7087 gfc_simplify_rrspacing (gfc_expr *x)
7089 gfc_expr *result;
7090 int i;
7091 long int e, p;
7093 if (x->expr_type != EXPR_CONSTANT)
7094 return NULL;
7096 i = gfc_validate_kind (x->ts.type, x->ts.kind, false);
7098 result = gfc_get_constant_expr (BT_REAL, x->ts.kind, &x->where);
7100 /* RRSPACING(+/- 0.0) = 0.0 */
7101 if (mpfr_zero_p (x->value.real))
7103 mpfr_set_ui (result->value.real, 0, GFC_RND_MODE);
7104 return result;
7107 /* RRSPACING(inf) = NaN */
7108 if (mpfr_inf_p (x->value.real))
7110 mpfr_set_nan (result->value.real);
7111 return result;
7114 /* RRSPACING(NaN) = same NaN */
7115 if (mpfr_nan_p (x->value.real))
7117 mpfr_set (result->value.real, x->value.real, GFC_RND_MODE);
7118 return result;
7121 /* | x * 2**(-e) | * 2**p. */
7122 mpfr_abs (result->value.real, x->value.real, GFC_RND_MODE);
7123 e = - (long int) mpfr_get_exp (x->value.real);
7124 mpfr_mul_2si (result->value.real, result->value.real, e, GFC_RND_MODE);
7126 p = (long int) gfc_real_kinds[i].digits;
7127 mpfr_mul_2si (result->value.real, result->value.real, p, GFC_RND_MODE);
7129 return range_check (result, "RRSPACING");
7133 gfc_expr *
7134 gfc_simplify_scale (gfc_expr *x, gfc_expr *i)
7136 int k, neg_flag, power, exp_range;
7137 mpfr_t scale, radix;
7138 gfc_expr *result;
7140 if (x->expr_type != EXPR_CONSTANT || i->expr_type != EXPR_CONSTANT)
7141 return NULL;
7143 result = gfc_get_constant_expr (BT_REAL, x->ts.kind, &x->where);
7145 if (mpfr_zero_p (x->value.real))
7147 mpfr_set_ui (result->value.real, 0, GFC_RND_MODE);
7148 return result;
7151 k = gfc_validate_kind (BT_REAL, x->ts.kind, false);
7153 exp_range = gfc_real_kinds[k].max_exponent - gfc_real_kinds[k].min_exponent;
7155 /* This check filters out values of i that would overflow an int. */
7156 if (mpz_cmp_si (i->value.integer, exp_range + 2) > 0
7157 || mpz_cmp_si (i->value.integer, -exp_range - 2) < 0)
7159 gfc_error ("Result of SCALE overflows its kind at %L", &result->where);
7160 gfc_free_expr (result);
7161 return &gfc_bad_expr;
7164 /* Compute scale = radix ** power. */
7165 power = mpz_get_si (i->value.integer);
7167 if (power >= 0)
7168 neg_flag = 0;
7169 else
7171 neg_flag = 1;
7172 power = -power;
7175 gfc_set_model_kind (x->ts.kind);
7176 mpfr_init (scale);
7177 mpfr_init (radix);
7178 mpfr_set_ui (radix, gfc_real_kinds[k].radix, GFC_RND_MODE);
7179 mpfr_pow_ui (scale, radix, power, GFC_RND_MODE);
7181 if (neg_flag)
7182 mpfr_div (result->value.real, x->value.real, scale, GFC_RND_MODE);
7183 else
7184 mpfr_mul (result->value.real, x->value.real, scale, GFC_RND_MODE);
7186 mpfr_clears (scale, radix, NULL);
7188 return range_check (result, "SCALE");
7192 /* Variants of strspn and strcspn that operate on wide characters. */
7194 static size_t
7195 wide_strspn (const gfc_char_t *s1, const gfc_char_t *s2)
7197 size_t i = 0;
7198 const gfc_char_t *c;
7200 while (s1[i])
7202 for (c = s2; *c; c++)
7204 if (s1[i] == *c)
7205 break;
7207 if (*c == '\0')
7208 break;
7209 i++;
7212 return i;
7215 static size_t
7216 wide_strcspn (const gfc_char_t *s1, const gfc_char_t *s2)
7218 size_t i = 0;
7219 const gfc_char_t *c;
7221 while (s1[i])
7223 for (c = s2; *c; c++)
7225 if (s1[i] == *c)
7226 break;
7228 if (*c)
7229 break;
7230 i++;
7233 return i;
7237 gfc_expr *
7238 gfc_simplify_scan (gfc_expr *e, gfc_expr *c, gfc_expr *b, gfc_expr *kind)
7240 gfc_expr *result;
7241 int back;
7242 size_t i;
7243 size_t indx, len, lenc;
7244 int k = get_kind (BT_INTEGER, kind, "SCAN", gfc_default_integer_kind);
7246 if (k == -1)
7247 return &gfc_bad_expr;
7249 if (e->expr_type != EXPR_CONSTANT || c->expr_type != EXPR_CONSTANT
7250 || ( b != NULL && b->expr_type != EXPR_CONSTANT))
7251 return NULL;
7253 if (b != NULL && b->value.logical != 0)
7254 back = 1;
7255 else
7256 back = 0;
7258 len = e->value.character.length;
7259 lenc = c->value.character.length;
7261 if (len == 0 || lenc == 0)
7263 indx = 0;
7265 else
7267 if (back == 0)
7269 indx = wide_strcspn (e->value.character.string,
7270 c->value.character.string) + 1;
7271 if (indx > len)
7272 indx = 0;
7274 else
7275 for (indx = len; indx > 0; indx--)
7277 for (i = 0; i < lenc; i++)
7279 if (c->value.character.string[i]
7280 == e->value.character.string[indx - 1])
7281 break;
7283 if (i < lenc)
7284 break;
7288 result = gfc_get_int_expr (k, &e->where, indx);
7289 return range_check (result, "SCAN");
7293 gfc_expr *
7294 gfc_simplify_selected_char_kind (gfc_expr *e)
7296 int kind;
7298 if (e->expr_type != EXPR_CONSTANT)
7299 return NULL;
7301 if (gfc_compare_with_Cstring (e, "ascii", false) == 0
7302 || gfc_compare_with_Cstring (e, "default", false) == 0)
7303 kind = 1;
7304 else if (gfc_compare_with_Cstring (e, "iso_10646", false) == 0)
7305 kind = 4;
7306 else
7307 kind = -1;
7309 return gfc_get_int_expr (gfc_default_integer_kind, &e->where, kind);
7313 gfc_expr *
7314 gfc_simplify_selected_int_kind (gfc_expr *e)
7316 int i, kind, range;
7318 if (e->expr_type != EXPR_CONSTANT || gfc_extract_int (e, &range))
7319 return NULL;
7321 kind = INT_MAX;
7323 for (i = 0; gfc_integer_kinds[i].kind != 0; i++)
7324 if (gfc_integer_kinds[i].range >= range
7325 && gfc_integer_kinds[i].kind < kind)
7326 kind = gfc_integer_kinds[i].kind;
7328 if (kind == INT_MAX)
7329 kind = -1;
7331 return gfc_get_int_expr (gfc_default_integer_kind, &e->where, kind);
7335 gfc_expr *
7336 gfc_simplify_selected_real_kind (gfc_expr *p, gfc_expr *q, gfc_expr *rdx)
7338 int range, precision, radix, i, kind, found_precision, found_range,
7339 found_radix;
7340 locus *loc = &gfc_current_locus;
7342 if (p == NULL)
7343 precision = 0;
7344 else
7346 if (p->expr_type != EXPR_CONSTANT
7347 || gfc_extract_int (p, &precision))
7348 return NULL;
7349 loc = &p->where;
7352 if (q == NULL)
7353 range = 0;
7354 else
7356 if (q->expr_type != EXPR_CONSTANT
7357 || gfc_extract_int (q, &range))
7358 return NULL;
7360 if (!loc)
7361 loc = &q->where;
7364 if (rdx == NULL)
7365 radix = 0;
7366 else
7368 if (rdx->expr_type != EXPR_CONSTANT
7369 || gfc_extract_int (rdx, &radix))
7370 return NULL;
7372 if (!loc)
7373 loc = &rdx->where;
7376 kind = INT_MAX;
7377 found_precision = 0;
7378 found_range = 0;
7379 found_radix = 0;
7381 for (i = 0; gfc_real_kinds[i].kind != 0; i++)
7383 if (gfc_real_kinds[i].precision >= precision)
7384 found_precision = 1;
7386 if (gfc_real_kinds[i].range >= range)
7387 found_range = 1;
7389 if (radix == 0 || gfc_real_kinds[i].radix == radix)
7390 found_radix = 1;
7392 if (gfc_real_kinds[i].precision >= precision
7393 && gfc_real_kinds[i].range >= range
7394 && (radix == 0 || gfc_real_kinds[i].radix == radix)
7395 && gfc_real_kinds[i].kind < kind)
7396 kind = gfc_real_kinds[i].kind;
7399 if (kind == INT_MAX)
7401 if (found_radix && found_range && !found_precision)
7402 kind = -1;
7403 else if (found_radix && found_precision && !found_range)
7404 kind = -2;
7405 else if (found_radix && !found_precision && !found_range)
7406 kind = -3;
7407 else if (found_radix)
7408 kind = -4;
7409 else
7410 kind = -5;
7413 return gfc_get_int_expr (gfc_default_integer_kind, loc, kind);
7417 gfc_expr *
7418 gfc_simplify_set_exponent (gfc_expr *x, gfc_expr *i)
7420 gfc_expr *result;
7421 mpfr_t exp, absv, log2, pow2, frac;
7422 long exp2;
7424 if (x->expr_type != EXPR_CONSTANT || i->expr_type != EXPR_CONSTANT)
7425 return NULL;
7427 result = gfc_get_constant_expr (BT_REAL, x->ts.kind, &x->where);
7429 /* SET_EXPONENT (+/-0.0, I) = +/- 0.0
7430 SET_EXPONENT (NaN) = same NaN */
7431 if (mpfr_zero_p (x->value.real) || mpfr_nan_p (x->value.real))
7433 mpfr_set (result->value.real, x->value.real, GFC_RND_MODE);
7434 return result;
7437 /* SET_EXPONENT (inf) = NaN */
7438 if (mpfr_inf_p (x->value.real))
7440 mpfr_set_nan (result->value.real);
7441 return result;
7444 gfc_set_model_kind (x->ts.kind);
7445 mpfr_init (absv);
7446 mpfr_init (log2);
7447 mpfr_init (exp);
7448 mpfr_init (pow2);
7449 mpfr_init (frac);
7451 mpfr_abs (absv, x->value.real, GFC_RND_MODE);
7452 mpfr_log2 (log2, absv, GFC_RND_MODE);
7454 mpfr_floor (log2, log2);
7455 mpfr_add_ui (exp, log2, 1, GFC_RND_MODE);
7457 /* Old exponent value, and fraction. */
7458 mpfr_ui_pow (pow2, 2, exp, GFC_RND_MODE);
7460 mpfr_div (frac, x->value.real, pow2, GFC_RND_MODE);
7462 /* New exponent. */
7463 exp2 = mpz_get_si (i->value.integer);
7464 mpfr_mul_2si (result->value.real, frac, exp2, GFC_RND_MODE);
7466 mpfr_clears (absv, log2, exp, pow2, frac, NULL);
7468 return range_check (result, "SET_EXPONENT");
7472 gfc_expr *
7473 gfc_simplify_shape (gfc_expr *source, gfc_expr *kind)
7475 mpz_t shape[GFC_MAX_DIMENSIONS];
7476 gfc_expr *result, *e, *f;
7477 gfc_array_ref *ar;
7478 int n;
7479 bool t;
7480 int k = get_kind (BT_INTEGER, kind, "SHAPE", gfc_default_integer_kind);
7482 if (source->rank == -1)
7483 return NULL;
7485 result = gfc_get_array_expr (BT_INTEGER, k, &source->where);
7486 result->shape = gfc_get_shape (1);
7487 mpz_init (result->shape[0]);
7489 if (source->rank == 0)
7490 return result;
7492 if (source->expr_type == EXPR_VARIABLE)
7494 ar = gfc_find_array_ref (source);
7495 t = gfc_array_ref_shape (ar, shape);
7497 else if (source->shape)
7499 t = true;
7500 for (n = 0; n < source->rank; n++)
7502 mpz_init (shape[n]);
7503 mpz_set (shape[n], source->shape[n]);
7506 else
7507 t = false;
7509 for (n = 0; n < source->rank; n++)
7511 e = gfc_get_constant_expr (BT_INTEGER, k, &source->where);
7513 if (t)
7514 mpz_set (e->value.integer, shape[n]);
7515 else
7517 mpz_set_ui (e->value.integer, n + 1);
7519 f = simplify_size (source, e, k);
7520 gfc_free_expr (e);
7521 if (f == NULL)
7523 gfc_free_expr (result);
7524 return NULL;
7526 else
7527 e = f;
7530 if (e == &gfc_bad_expr || range_check (e, "SHAPE") == &gfc_bad_expr)
7532 gfc_free_expr (result);
7533 if (t)
7534 gfc_clear_shape (shape, source->rank);
7535 return &gfc_bad_expr;
7538 gfc_constructor_append_expr (&result->value.constructor, e, NULL);
7541 if (t)
7542 gfc_clear_shape (shape, source->rank);
7544 mpz_set_si (result->shape[0], source->rank);
7546 return result;
7550 static gfc_expr *
7551 simplify_size (gfc_expr *array, gfc_expr *dim, int k)
7553 mpz_t size;
7554 gfc_expr *return_value;
7555 int d;
7556 gfc_ref *ref;
7558 /* For unary operations, the size of the result is given by the size
7559 of the operand. For binary ones, it's the size of the first operand
7560 unless it is scalar, then it is the size of the second. */
7561 if (array->expr_type == EXPR_OP && !array->value.op.uop)
7563 gfc_expr* replacement;
7564 gfc_expr* simplified;
7566 switch (array->value.op.op)
7568 /* Unary operations. */
7569 case INTRINSIC_NOT:
7570 case INTRINSIC_UPLUS:
7571 case INTRINSIC_UMINUS:
7572 case INTRINSIC_PARENTHESES:
7573 replacement = array->value.op.op1;
7574 break;
7576 /* Binary operations. If any one of the operands is scalar, take
7577 the other one's size. If both of them are arrays, it does not
7578 matter -- try to find one with known shape, if possible. */
7579 default:
7580 if (array->value.op.op1->rank == 0)
7581 replacement = array->value.op.op2;
7582 else if (array->value.op.op2->rank == 0)
7583 replacement = array->value.op.op1;
7584 else
7586 simplified = simplify_size (array->value.op.op1, dim, k);
7587 if (simplified)
7588 return simplified;
7590 replacement = array->value.op.op2;
7592 break;
7595 /* Try to reduce it directly if possible. */
7596 simplified = simplify_size (replacement, dim, k);
7598 /* Otherwise, we build a new SIZE call. This is hopefully at least
7599 simpler than the original one. */
7600 if (!simplified)
7602 gfc_expr *kind = gfc_get_int_expr (gfc_default_integer_kind, NULL, k);
7603 simplified = gfc_build_intrinsic_call (gfc_current_ns,
7604 GFC_ISYM_SIZE, "size",
7605 array->where, 3,
7606 gfc_copy_expr (replacement),
7607 gfc_copy_expr (dim),
7608 kind);
7610 return simplified;
7613 for (ref = array->ref; ref; ref = ref->next)
7614 if (ref->type == REF_ARRAY && ref->u.ar.as
7615 && !gfc_resolve_array_spec (ref->u.ar.as, 0))
7616 return NULL;
7618 if (dim == NULL)
7620 if (!gfc_array_size (array, &size))
7621 return NULL;
7623 else
7625 if (dim->expr_type != EXPR_CONSTANT)
7626 return NULL;
7628 if (array->rank == -1)
7629 return NULL;
7631 d = mpz_get_si (dim->value.integer) - 1;
7632 if (d < 0 || d > array->rank - 1)
7634 gfc_error ("DIM argument (%d) to intrinsic SIZE at %L out of range "
7635 "(1:%d)", d+1, &array->where, array->rank);
7636 return &gfc_bad_expr;
7639 if (!gfc_array_dimen_size (array, d, &size))
7640 return NULL;
7643 return_value = gfc_get_constant_expr (BT_INTEGER, k, &array->where);
7644 mpz_set (return_value->value.integer, size);
7645 mpz_clear (size);
7647 return return_value;
7651 gfc_expr *
7652 gfc_simplify_size (gfc_expr *array, gfc_expr *dim, gfc_expr *kind)
7654 gfc_expr *result;
7655 int k = get_kind (BT_INTEGER, kind, "SIZE", gfc_default_integer_kind);
7657 if (k == -1)
7658 return &gfc_bad_expr;
7660 result = simplify_size (array, dim, k);
7661 if (result == NULL || result == &gfc_bad_expr)
7662 return result;
7664 return range_check (result, "SIZE");
7668 /* SIZEOF and C_SIZEOF return the size in bytes of an array element
7669 multiplied by the array size. */
7671 gfc_expr *
7672 gfc_simplify_sizeof (gfc_expr *x)
7674 gfc_expr *result = NULL;
7675 mpz_t array_size;
7676 size_t res_size;
7678 if (x->ts.type == BT_CLASS || x->ts.deferred)
7679 return NULL;
7681 if (x->ts.type == BT_CHARACTER
7682 && (!x->ts.u.cl || !x->ts.u.cl->length
7683 || x->ts.u.cl->length->expr_type != EXPR_CONSTANT))
7684 return NULL;
7686 if (x->rank && x->expr_type != EXPR_ARRAY
7687 && !gfc_array_size (x, &array_size))
7688 return NULL;
7690 result = gfc_get_constant_expr (BT_INTEGER, gfc_index_integer_kind,
7691 &x->where);
7692 gfc_target_expr_size (x, &res_size);
7693 mpz_set_si (result->value.integer, res_size);
7695 return result;
7699 /* STORAGE_SIZE returns the size in bits of a single array element. */
7701 gfc_expr *
7702 gfc_simplify_storage_size (gfc_expr *x,
7703 gfc_expr *kind)
7705 gfc_expr *result = NULL;
7706 int k;
7707 size_t siz;
7709 if (x->ts.type == BT_CLASS || x->ts.deferred)
7710 return NULL;
7712 if (x->ts.type == BT_CHARACTER && x->expr_type != EXPR_CONSTANT
7713 && (!x->ts.u.cl || !x->ts.u.cl->length
7714 || x->ts.u.cl->length->expr_type != EXPR_CONSTANT))
7715 return NULL;
7717 k = get_kind (BT_INTEGER, kind, "STORAGE_SIZE", gfc_default_integer_kind);
7718 if (k == -1)
7719 return &gfc_bad_expr;
7721 result = gfc_get_constant_expr (BT_INTEGER, k, &x->where);
7723 gfc_element_size (x, &siz);
7724 mpz_set_si (result->value.integer, siz);
7725 mpz_mul_ui (result->value.integer, result->value.integer, BITS_PER_UNIT);
7727 return range_check (result, "STORAGE_SIZE");
7731 gfc_expr *
7732 gfc_simplify_sign (gfc_expr *x, gfc_expr *y)
7734 gfc_expr *result;
7736 if (x->expr_type != EXPR_CONSTANT || y->expr_type != EXPR_CONSTANT)
7737 return NULL;
7739 result = gfc_get_constant_expr (x->ts.type, x->ts.kind, &x->where);
7741 switch (x->ts.type)
7743 case BT_INTEGER:
7744 mpz_abs (result->value.integer, x->value.integer);
7745 if (mpz_sgn (y->value.integer) < 0)
7746 mpz_neg (result->value.integer, result->value.integer);
7747 break;
7749 case BT_REAL:
7750 if (flag_sign_zero)
7751 mpfr_copysign (result->value.real, x->value.real, y->value.real,
7752 GFC_RND_MODE);
7753 else
7754 mpfr_setsign (result->value.real, x->value.real,
7755 mpfr_sgn (y->value.real) < 0 ? 1 : 0, GFC_RND_MODE);
7756 break;
7758 default:
7759 gfc_internal_error ("Bad type in gfc_simplify_sign");
7762 return result;
7766 gfc_expr *
7767 gfc_simplify_sin (gfc_expr *x)
7769 gfc_expr *result;
7771 if (x->expr_type != EXPR_CONSTANT)
7772 return NULL;
7774 result = gfc_get_constant_expr (x->ts.type, x->ts.kind, &x->where);
7776 switch (x->ts.type)
7778 case BT_REAL:
7779 mpfr_sin (result->value.real, x->value.real, GFC_RND_MODE);
7780 break;
7782 case BT_COMPLEX:
7783 gfc_set_model (x->value.real);
7784 mpc_sin (result->value.complex, x->value.complex, GFC_MPC_RND_MODE);
7785 break;
7787 default:
7788 gfc_internal_error ("in gfc_simplify_sin(): Bad type");
7791 return range_check (result, "SIN");
7795 gfc_expr *
7796 gfc_simplify_sinh (gfc_expr *x)
7798 gfc_expr *result;
7800 if (x->expr_type != EXPR_CONSTANT)
7801 return NULL;
7803 result = gfc_get_constant_expr (x->ts.type, x->ts.kind, &x->where);
7805 switch (x->ts.type)
7807 case BT_REAL:
7808 mpfr_sinh (result->value.real, x->value.real, GFC_RND_MODE);
7809 break;
7811 case BT_COMPLEX:
7812 mpc_sinh (result->value.complex, x->value.complex, GFC_MPC_RND_MODE);
7813 break;
7815 default:
7816 gcc_unreachable ();
7819 return range_check (result, "SINH");
7823 /* The argument is always a double precision real that is converted to
7824 single precision. TODO: Rounding! */
7826 gfc_expr *
7827 gfc_simplify_sngl (gfc_expr *a)
7829 gfc_expr *result;
7830 int tmp1, tmp2;
7832 if (a->expr_type != EXPR_CONSTANT)
7833 return NULL;
7835 /* For explicit conversion, turn off -Wconversion and -Wconversion-extra
7836 warnings. */
7837 tmp1 = warn_conversion;
7838 tmp2 = warn_conversion_extra;
7839 warn_conversion = warn_conversion_extra = 0;
7841 result = gfc_real2real (a, gfc_default_real_kind);
7843 warn_conversion = tmp1;
7844 warn_conversion_extra = tmp2;
7846 return range_check (result, "SNGL");
7850 gfc_expr *
7851 gfc_simplify_spacing (gfc_expr *x)
7853 gfc_expr *result;
7854 int i;
7855 long int en, ep;
7857 if (x->expr_type != EXPR_CONSTANT)
7858 return NULL;
7860 i = gfc_validate_kind (x->ts.type, x->ts.kind, false);
7861 result = gfc_get_constant_expr (BT_REAL, x->ts.kind, &x->where);
7863 /* SPACING(+/- 0.0) = SPACING(TINY(0.0)) = TINY(0.0) */
7864 if (mpfr_zero_p (x->value.real))
7866 mpfr_set (result->value.real, gfc_real_kinds[i].tiny, GFC_RND_MODE);
7867 return result;
7870 /* SPACING(inf) = NaN */
7871 if (mpfr_inf_p (x->value.real))
7873 mpfr_set_nan (result->value.real);
7874 return result;
7877 /* SPACING(NaN) = same NaN */
7878 if (mpfr_nan_p (x->value.real))
7880 mpfr_set (result->value.real, x->value.real, GFC_RND_MODE);
7881 return result;
7884 /* In the Fortran 95 standard, the result is b**(e - p) where b, e, and p
7885 are the radix, exponent of x, and precision. This excludes the
7886 possibility of subnormal numbers. Fortran 2003 states the result is
7887 b**max(e - p, emin - 1). */
7889 ep = (long int) mpfr_get_exp (x->value.real) - gfc_real_kinds[i].digits;
7890 en = (long int) gfc_real_kinds[i].min_exponent - 1;
7891 en = en > ep ? en : ep;
7893 mpfr_set_ui (result->value.real, 1, GFC_RND_MODE);
7894 mpfr_mul_2si (result->value.real, result->value.real, en, GFC_RND_MODE);
7896 return range_check (result, "SPACING");
7900 gfc_expr *
7901 gfc_simplify_spread (gfc_expr *source, gfc_expr *dim_expr, gfc_expr *ncopies_expr)
7903 gfc_expr *result = NULL;
7904 int nelem, i, j, dim, ncopies;
7905 mpz_t size;
7907 if ((!gfc_is_constant_expr (source)
7908 && !is_constant_array_expr (source))
7909 || !gfc_is_constant_expr (dim_expr)
7910 || !gfc_is_constant_expr (ncopies_expr))
7911 return NULL;
7913 gcc_assert (dim_expr->ts.type == BT_INTEGER);
7914 gfc_extract_int (dim_expr, &dim);
7915 dim -= 1; /* zero-base DIM */
7917 gcc_assert (ncopies_expr->ts.type == BT_INTEGER);
7918 gfc_extract_int (ncopies_expr, &ncopies);
7919 ncopies = MAX (ncopies, 0);
7921 /* Do not allow the array size to exceed the limit for an array
7922 constructor. */
7923 if (source->expr_type == EXPR_ARRAY)
7925 if (!gfc_array_size (source, &size))
7926 gfc_internal_error ("Failure getting length of a constant array.");
7928 else
7929 mpz_init_set_ui (size, 1);
7931 nelem = mpz_get_si (size) * ncopies;
7932 if (nelem > flag_max_array_constructor)
7934 if (gfc_init_expr_flag)
7936 gfc_error ("The number of elements (%d) in the array constructor "
7937 "at %L requires an increase of the allowed %d upper "
7938 "limit. See %<-fmax-array-constructor%> option.",
7939 nelem, &source->where, flag_max_array_constructor);
7940 return &gfc_bad_expr;
7942 else
7943 return NULL;
7946 if (source->expr_type == EXPR_CONSTANT
7947 || source->expr_type == EXPR_STRUCTURE)
7949 gcc_assert (dim == 0);
7951 result = gfc_get_array_expr (source->ts.type, source->ts.kind,
7952 &source->where);
7953 if (source->ts.type == BT_DERIVED)
7954 result->ts.u.derived = source->ts.u.derived;
7955 result->rank = 1;
7956 result->shape = gfc_get_shape (result->rank);
7957 mpz_init_set_si (result->shape[0], ncopies);
7959 for (i = 0; i < ncopies; ++i)
7960 gfc_constructor_append_expr (&result->value.constructor,
7961 gfc_copy_expr (source), NULL);
7963 else if (source->expr_type == EXPR_ARRAY)
7965 int offset, rstride[GFC_MAX_DIMENSIONS], extent[GFC_MAX_DIMENSIONS];
7966 gfc_constructor *source_ctor;
7968 gcc_assert (source->rank < GFC_MAX_DIMENSIONS);
7969 gcc_assert (dim >= 0 && dim <= source->rank);
7971 result = gfc_get_array_expr (source->ts.type, source->ts.kind,
7972 &source->where);
7973 if (source->ts.type == BT_DERIVED)
7974 result->ts.u.derived = source->ts.u.derived;
7975 result->rank = source->rank + 1;
7976 result->shape = gfc_get_shape (result->rank);
7978 for (i = 0, j = 0; i < result->rank; ++i)
7980 if (i != dim)
7981 mpz_init_set (result->shape[i], source->shape[j++]);
7982 else
7983 mpz_init_set_si (result->shape[i], ncopies);
7985 extent[i] = mpz_get_si (result->shape[i]);
7986 rstride[i] = (i == 0) ? 1 : rstride[i-1] * extent[i-1];
7989 offset = 0;
7990 for (source_ctor = gfc_constructor_first (source->value.constructor);
7991 source_ctor; source_ctor = gfc_constructor_next (source_ctor))
7993 for (i = 0; i < ncopies; ++i)
7994 gfc_constructor_insert_expr (&result->value.constructor,
7995 gfc_copy_expr (source_ctor->expr),
7996 NULL, offset + i * rstride[dim]);
7998 offset += (dim == 0 ? ncopies : 1);
8001 else
8003 gfc_error ("Simplification of SPREAD at %C not yet implemented");
8004 return &gfc_bad_expr;
8007 if (source->ts.type == BT_CHARACTER)
8008 result->ts.u.cl = source->ts.u.cl;
8010 return result;
8014 gfc_expr *
8015 gfc_simplify_sqrt (gfc_expr *e)
8017 gfc_expr *result = NULL;
8019 if (e->expr_type != EXPR_CONSTANT)
8020 return NULL;
8022 switch (e->ts.type)
8024 case BT_REAL:
8025 if (mpfr_cmp_si (e->value.real, 0) < 0)
8027 gfc_error ("Argument of SQRT at %L has a negative value",
8028 &e->where);
8029 return &gfc_bad_expr;
8031 result = gfc_get_constant_expr (e->ts.type, e->ts.kind, &e->where);
8032 mpfr_sqrt (result->value.real, e->value.real, GFC_RND_MODE);
8033 break;
8035 case BT_COMPLEX:
8036 gfc_set_model (e->value.real);
8038 result = gfc_get_constant_expr (e->ts.type, e->ts.kind, &e->where);
8039 mpc_sqrt (result->value.complex, e->value.complex, GFC_MPC_RND_MODE);
8040 break;
8042 default:
8043 gfc_internal_error ("invalid argument of SQRT at %L", &e->where);
8046 return range_check (result, "SQRT");
8050 gfc_expr *
8051 gfc_simplify_sum (gfc_expr *array, gfc_expr *dim, gfc_expr *mask)
8053 return simplify_transformation (array, dim, mask, 0, gfc_add);
8057 /* Simplify COTAN(X) where X has the unit of radian. */
8059 gfc_expr *
8060 gfc_simplify_cotan (gfc_expr *x)
8062 gfc_expr *result;
8063 mpc_t swp, *val;
8065 if (x->expr_type != EXPR_CONSTANT)
8066 return NULL;
8068 result = gfc_get_constant_expr (x->ts.type, x->ts.kind, &x->where);
8070 switch (x->ts.type)
8072 case BT_REAL:
8073 mpfr_cot (result->value.real, x->value.real, GFC_RND_MODE);
8074 break;
8076 case BT_COMPLEX:
8077 /* There is no builtin mpc_cot, so compute cot = cos / sin. */
8078 val = &result->value.complex;
8079 mpc_init2 (swp, mpfr_get_default_prec ());
8080 mpc_sin_cos (*val, swp, x->value.complex, GFC_MPC_RND_MODE,
8081 GFC_MPC_RND_MODE);
8082 mpc_div (*val, swp, *val, GFC_MPC_RND_MODE);
8083 mpc_clear (swp);
8084 break;
8086 default:
8087 gcc_unreachable ();
8090 return range_check (result, "COTAN");
8094 gfc_expr *
8095 gfc_simplify_tan (gfc_expr *x)
8097 gfc_expr *result;
8099 if (x->expr_type != EXPR_CONSTANT)
8100 return NULL;
8102 result = gfc_get_constant_expr (x->ts.type, x->ts.kind, &x->where);
8104 switch (x->ts.type)
8106 case BT_REAL:
8107 mpfr_tan (result->value.real, x->value.real, GFC_RND_MODE);
8108 break;
8110 case BT_COMPLEX:
8111 mpc_tan (result->value.complex, x->value.complex, GFC_MPC_RND_MODE);
8112 break;
8114 default:
8115 gcc_unreachable ();
8118 return range_check (result, "TAN");
8122 gfc_expr *
8123 gfc_simplify_tanh (gfc_expr *x)
8125 gfc_expr *result;
8127 if (x->expr_type != EXPR_CONSTANT)
8128 return NULL;
8130 result = gfc_get_constant_expr (x->ts.type, x->ts.kind, &x->where);
8132 switch (x->ts.type)
8134 case BT_REAL:
8135 mpfr_tanh (result->value.real, x->value.real, GFC_RND_MODE);
8136 break;
8138 case BT_COMPLEX:
8139 mpc_tanh (result->value.complex, x->value.complex, GFC_MPC_RND_MODE);
8140 break;
8142 default:
8143 gcc_unreachable ();
8146 return range_check (result, "TANH");
8150 gfc_expr *
8151 gfc_simplify_tiny (gfc_expr *e)
8153 gfc_expr *result;
8154 int i;
8156 i = gfc_validate_kind (BT_REAL, e->ts.kind, false);
8158 result = gfc_get_constant_expr (BT_REAL, e->ts.kind, &e->where);
8159 mpfr_set (result->value.real, gfc_real_kinds[i].tiny, GFC_RND_MODE);
8161 return result;
8165 gfc_expr *
8166 gfc_simplify_trailz (gfc_expr *e)
8168 unsigned long tz, bs;
8169 int i;
8171 if (e->expr_type != EXPR_CONSTANT)
8172 return NULL;
8174 i = gfc_validate_kind (e->ts.type, e->ts.kind, false);
8175 bs = gfc_integer_kinds[i].bit_size;
8176 tz = mpz_scan1 (e->value.integer, 0);
8178 return gfc_get_int_expr (gfc_default_integer_kind,
8179 &e->where, MIN (tz, bs));
8183 gfc_expr *
8184 gfc_simplify_transfer (gfc_expr *source, gfc_expr *mold, gfc_expr *size)
8186 gfc_expr *result;
8187 gfc_expr *mold_element;
8188 size_t source_size;
8189 size_t result_size;
8190 size_t buffer_size;
8191 mpz_t tmp;
8192 unsigned char *buffer;
8193 size_t result_length;
8195 if (!gfc_is_constant_expr (source) || !gfc_is_constant_expr (size))
8196 return NULL;
8198 if (!gfc_resolve_expr (mold))
8199 return NULL;
8200 if (gfc_init_expr_flag && !gfc_is_constant_expr (mold))
8201 return NULL;
8203 if (!gfc_calculate_transfer_sizes (source, mold, size, &source_size,
8204 &result_size, &result_length))
8205 return NULL;
8207 /* Calculate the size of the source. */
8208 if (source->expr_type == EXPR_ARRAY && !gfc_array_size (source, &tmp))
8209 gfc_internal_error ("Failure getting length of a constant array.");
8211 /* Create an empty new expression with the appropriate characteristics. */
8212 result = gfc_get_constant_expr (mold->ts.type, mold->ts.kind,
8213 &source->where);
8214 result->ts = mold->ts;
8216 mold_element = (mold->expr_type == EXPR_ARRAY && mold->value.constructor)
8217 ? gfc_constructor_first (mold->value.constructor)->expr
8218 : mold;
8220 /* Set result character length, if needed. Note that this needs to be
8221 set even for array expressions, in order to pass this information into
8222 gfc_target_interpret_expr. */
8223 if (result->ts.type == BT_CHARACTER && gfc_is_constant_expr (mold_element))
8225 result->value.character.length = mold_element->value.character.length;
8227 /* Let the typespec of the result inherit the string length.
8228 This is crucial if a resulting array has size zero. */
8229 if (mold_element->ts.u.cl->length)
8230 result->ts.u.cl->length = gfc_copy_expr (mold_element->ts.u.cl->length);
8231 else
8232 result->ts.u.cl->length =
8233 gfc_get_int_expr (gfc_charlen_int_kind, NULL,
8234 mold_element->value.character.length);
8237 /* Set the number of elements in the result, and determine its size. */
8239 if (mold->expr_type == EXPR_ARRAY || mold->rank || size)
8241 result->expr_type = EXPR_ARRAY;
8242 result->rank = 1;
8243 result->shape = gfc_get_shape (1);
8244 mpz_init_set_ui (result->shape[0], result_length);
8246 else
8247 result->rank = 0;
8249 /* Allocate the buffer to store the binary version of the source. */
8250 buffer_size = MAX (source_size, result_size);
8251 buffer = (unsigned char*)alloca (buffer_size);
8252 memset (buffer, 0, buffer_size);
8254 /* Now write source to the buffer. */
8255 gfc_target_encode_expr (source, buffer, buffer_size);
8257 /* And read the buffer back into the new expression. */
8258 gfc_target_interpret_expr (buffer, buffer_size, result, false);
8260 return result;
8264 gfc_expr *
8265 gfc_simplify_transpose (gfc_expr *matrix)
8267 int row, matrix_rows, col, matrix_cols;
8268 gfc_expr *result;
8270 if (!is_constant_array_expr (matrix))
8271 return NULL;
8273 gcc_assert (matrix->rank == 2);
8275 if (matrix->shape == NULL)
8276 return NULL;
8278 result = gfc_get_array_expr (matrix->ts.type, matrix->ts.kind,
8279 &matrix->where);
8280 result->rank = 2;
8281 result->shape = gfc_get_shape (result->rank);
8282 mpz_init_set (result->shape[0], matrix->shape[1]);
8283 mpz_init_set (result->shape[1], matrix->shape[0]);
8285 if (matrix->ts.type == BT_CHARACTER)
8286 result->ts.u.cl = matrix->ts.u.cl;
8287 else if (matrix->ts.type == BT_DERIVED)
8288 result->ts.u.derived = matrix->ts.u.derived;
8290 matrix_rows = mpz_get_si (matrix->shape[0]);
8291 matrix_cols = mpz_get_si (matrix->shape[1]);
8292 for (row = 0; row < matrix_rows; ++row)
8293 for (col = 0; col < matrix_cols; ++col)
8295 gfc_expr *e = gfc_constructor_lookup_expr (matrix->value.constructor,
8296 col * matrix_rows + row);
8297 gfc_constructor_insert_expr (&result->value.constructor,
8298 gfc_copy_expr (e), &matrix->where,
8299 row * matrix_cols + col);
8302 return result;
8306 gfc_expr *
8307 gfc_simplify_trim (gfc_expr *e)
8309 gfc_expr *result;
8310 int count, i, len, lentrim;
8312 if (e->expr_type != EXPR_CONSTANT)
8313 return NULL;
8315 len = e->value.character.length;
8316 for (count = 0, i = 1; i <= len; ++i)
8318 if (e->value.character.string[len - i] == ' ')
8319 count++;
8320 else
8321 break;
8324 lentrim = len - count;
8326 result = gfc_get_character_expr (e->ts.kind, &e->where, NULL, lentrim);
8327 for (i = 0; i < lentrim; i++)
8328 result->value.character.string[i] = e->value.character.string[i];
8330 return result;
8334 gfc_expr *
8335 gfc_simplify_image_index (gfc_expr *coarray, gfc_expr *sub)
8337 gfc_expr *result;
8338 gfc_ref *ref;
8339 gfc_array_spec *as;
8340 gfc_constructor *sub_cons;
8341 bool first_image;
8342 int d;
8344 if (!is_constant_array_expr (sub))
8345 return NULL;
8347 /* Follow any component references. */
8348 as = coarray->symtree->n.sym->as;
8349 for (ref = coarray->ref; ref; ref = ref->next)
8350 if (ref->type == REF_COMPONENT)
8351 as = ref->u.ar.as;
8353 if (!as || as->type == AS_DEFERRED)
8354 return NULL;
8356 /* "valid sequence of cosubscripts" are required; thus, return 0 unless
8357 the cosubscript addresses the first image. */
8359 sub_cons = gfc_constructor_first (sub->value.constructor);
8360 first_image = true;
8362 for (d = 1; d <= as->corank; d++)
8364 gfc_expr *ca_bound;
8365 int cmp;
8367 gcc_assert (sub_cons != NULL);
8369 ca_bound = simplify_bound_dim (coarray, NULL, d + as->rank, 0, as,
8370 NULL, true);
8371 if (ca_bound == NULL)
8372 return NULL;
8374 if (ca_bound == &gfc_bad_expr)
8375 return ca_bound;
8377 cmp = mpz_cmp (ca_bound->value.integer, sub_cons->expr->value.integer);
8379 if (cmp == 0)
8381 gfc_free_expr (ca_bound);
8382 sub_cons = gfc_constructor_next (sub_cons);
8383 continue;
8386 first_image = false;
8388 if (cmp > 0)
8390 gfc_error ("Out of bounds in IMAGE_INDEX at %L for dimension %d, "
8391 "SUB has %ld and COARRAY lower bound is %ld)",
8392 &coarray->where, d,
8393 mpz_get_si (sub_cons->expr->value.integer),
8394 mpz_get_si (ca_bound->value.integer));
8395 gfc_free_expr (ca_bound);
8396 return &gfc_bad_expr;
8399 gfc_free_expr (ca_bound);
8401 /* Check whether upperbound is valid for the multi-images case. */
8402 if (d < as->corank)
8404 ca_bound = simplify_bound_dim (coarray, NULL, d + as->rank, 1, as,
8405 NULL, true);
8406 if (ca_bound == &gfc_bad_expr)
8407 return ca_bound;
8409 if (ca_bound && ca_bound->expr_type == EXPR_CONSTANT
8410 && mpz_cmp (ca_bound->value.integer,
8411 sub_cons->expr->value.integer) < 0)
8413 gfc_error ("Out of bounds in IMAGE_INDEX at %L for dimension %d, "
8414 "SUB has %ld and COARRAY upper bound is %ld)",
8415 &coarray->where, d,
8416 mpz_get_si (sub_cons->expr->value.integer),
8417 mpz_get_si (ca_bound->value.integer));
8418 gfc_free_expr (ca_bound);
8419 return &gfc_bad_expr;
8422 if (ca_bound)
8423 gfc_free_expr (ca_bound);
8426 sub_cons = gfc_constructor_next (sub_cons);
8429 gcc_assert (sub_cons == NULL);
8431 if (flag_coarray != GFC_FCOARRAY_SINGLE && !first_image)
8432 return NULL;
8434 result = gfc_get_constant_expr (BT_INTEGER, gfc_default_integer_kind,
8435 &gfc_current_locus);
8436 if (first_image)
8437 mpz_set_si (result->value.integer, 1);
8438 else
8439 mpz_set_si (result->value.integer, 0);
8441 return result;
8444 gfc_expr *
8445 gfc_simplify_image_status (gfc_expr *image, gfc_expr *team ATTRIBUTE_UNUSED)
8447 if (flag_coarray == GFC_FCOARRAY_NONE)
8449 gfc_current_locus = *gfc_current_intrinsic_where;
8450 gfc_fatal_error ("Coarrays disabled at %C, use %<-fcoarray=%> to enable");
8451 return &gfc_bad_expr;
8454 /* Simplification is possible for fcoarray = single only. For all other modes
8455 the result depends on runtime conditions. */
8456 if (flag_coarray != GFC_FCOARRAY_SINGLE)
8457 return NULL;
8459 if (gfc_is_constant_expr (image))
8461 gfc_expr *result;
8462 result = gfc_get_constant_expr (BT_INTEGER, gfc_default_integer_kind,
8463 &image->where);
8464 if (mpz_get_si (image->value.integer) == 1)
8465 mpz_set_si (result->value.integer, 0);
8466 else
8467 mpz_set_si (result->value.integer, GFC_STAT_STOPPED_IMAGE);
8468 return result;
8470 else
8471 return NULL;
8475 gfc_expr *
8476 gfc_simplify_this_image (gfc_expr *coarray, gfc_expr *dim,
8477 gfc_expr *distance ATTRIBUTE_UNUSED)
8479 if (flag_coarray != GFC_FCOARRAY_SINGLE)
8480 return NULL;
8482 /* If no coarray argument has been passed or when the first argument
8483 is actually a distance argument. */
8484 if (coarray == NULL || !gfc_is_coarray (coarray))
8486 gfc_expr *result;
8487 /* FIXME: gfc_current_locus is wrong. */
8488 result = gfc_get_constant_expr (BT_INTEGER, gfc_default_integer_kind,
8489 &gfc_current_locus);
8490 mpz_set_si (result->value.integer, 1);
8491 return result;
8494 /* For -fcoarray=single, this_image(A) is the same as lcobound(A). */
8495 return simplify_cobound (coarray, dim, NULL, 0);
8499 gfc_expr *
8500 gfc_simplify_ubound (gfc_expr *array, gfc_expr *dim, gfc_expr *kind)
8502 return simplify_bound (array, dim, kind, 1);
8505 gfc_expr *
8506 gfc_simplify_ucobound (gfc_expr *array, gfc_expr *dim, gfc_expr *kind)
8508 return simplify_cobound (array, dim, kind, 1);
8512 gfc_expr *
8513 gfc_simplify_unpack (gfc_expr *vector, gfc_expr *mask, gfc_expr *field)
8515 gfc_expr *result, *e;
8516 gfc_constructor *vector_ctor, *mask_ctor, *field_ctor;
8518 if (!is_constant_array_expr (vector)
8519 || !is_constant_array_expr (mask)
8520 || (!gfc_is_constant_expr (field)
8521 && !is_constant_array_expr (field)))
8522 return NULL;
8524 result = gfc_get_array_expr (vector->ts.type, vector->ts.kind,
8525 &vector->where);
8526 if (vector->ts.type == BT_DERIVED)
8527 result->ts.u.derived = vector->ts.u.derived;
8528 result->rank = mask->rank;
8529 result->shape = gfc_copy_shape (mask->shape, mask->rank);
8531 if (vector->ts.type == BT_CHARACTER)
8532 result->ts.u.cl = vector->ts.u.cl;
8534 vector_ctor = gfc_constructor_first (vector->value.constructor);
8535 mask_ctor = gfc_constructor_first (mask->value.constructor);
8536 field_ctor
8537 = field->expr_type == EXPR_ARRAY
8538 ? gfc_constructor_first (field->value.constructor)
8539 : NULL;
8541 while (mask_ctor)
8543 if (mask_ctor->expr->value.logical)
8545 if (vector_ctor)
8547 e = gfc_copy_expr (vector_ctor->expr);
8548 vector_ctor = gfc_constructor_next (vector_ctor);
8550 else
8552 gfc_free_expr (result);
8553 return NULL;
8556 else if (field->expr_type == EXPR_ARRAY)
8558 if (field_ctor)
8559 e = gfc_copy_expr (field_ctor->expr);
8560 else
8562 /* Not enough elements in array FIELD. */
8563 gfc_free_expr (result);
8564 return &gfc_bad_expr;
8567 else
8568 e = gfc_copy_expr (field);
8570 gfc_constructor_append_expr (&result->value.constructor, e, NULL);
8572 mask_ctor = gfc_constructor_next (mask_ctor);
8573 field_ctor = gfc_constructor_next (field_ctor);
8576 return result;
8580 gfc_expr *
8581 gfc_simplify_verify (gfc_expr *s, gfc_expr *set, gfc_expr *b, gfc_expr *kind)
8583 gfc_expr *result;
8584 int back;
8585 size_t index, len, lenset;
8586 size_t i;
8587 int k = get_kind (BT_INTEGER, kind, "VERIFY", gfc_default_integer_kind);
8589 if (k == -1)
8590 return &gfc_bad_expr;
8592 if (s->expr_type != EXPR_CONSTANT || set->expr_type != EXPR_CONSTANT
8593 || ( b != NULL && b->expr_type != EXPR_CONSTANT))
8594 return NULL;
8596 if (b != NULL && b->value.logical != 0)
8597 back = 1;
8598 else
8599 back = 0;
8601 result = gfc_get_constant_expr (BT_INTEGER, k, &s->where);
8603 len = s->value.character.length;
8604 lenset = set->value.character.length;
8606 if (len == 0)
8608 mpz_set_ui (result->value.integer, 0);
8609 return result;
8612 if (back == 0)
8614 if (lenset == 0)
8616 mpz_set_ui (result->value.integer, 1);
8617 return result;
8620 index = wide_strspn (s->value.character.string,
8621 set->value.character.string) + 1;
8622 if (index > len)
8623 index = 0;
8626 else
8628 if (lenset == 0)
8630 mpz_set_ui (result->value.integer, len);
8631 return result;
8633 for (index = len; index > 0; index --)
8635 for (i = 0; i < lenset; i++)
8637 if (s->value.character.string[index - 1]
8638 == set->value.character.string[i])
8639 break;
8641 if (i == lenset)
8642 break;
8646 mpz_set_ui (result->value.integer, index);
8647 return result;
8651 gfc_expr *
8652 gfc_simplify_xor (gfc_expr *x, gfc_expr *y)
8654 gfc_expr *result;
8655 int kind;
8657 if (x->expr_type != EXPR_CONSTANT || y->expr_type != EXPR_CONSTANT)
8658 return NULL;
8660 kind = x->ts.kind > y->ts.kind ? x->ts.kind : y->ts.kind;
8662 switch (x->ts.type)
8664 case BT_INTEGER:
8665 result = gfc_get_constant_expr (BT_INTEGER, kind, &x->where);
8666 mpz_xor (result->value.integer, x->value.integer, y->value.integer);
8667 return range_check (result, "XOR");
8669 case BT_LOGICAL:
8670 return gfc_get_logical_expr (kind, &x->where,
8671 (x->value.logical && !y->value.logical)
8672 || (!x->value.logical && y->value.logical));
8674 default:
8675 gcc_unreachable ();
8680 /****************** Constant simplification *****************/
8682 /* Master function to convert one constant to another. While this is
8683 used as a simplification function, it requires the destination type
8684 and kind information which is supplied by a special case in
8685 do_simplify(). */
8687 gfc_expr *
8688 gfc_convert_constant (gfc_expr *e, bt type, int kind)
8690 gfc_expr *result, *(*f) (gfc_expr *, int);
8691 gfc_constructor *c, *t;
8693 switch (e->ts.type)
8695 case BT_INTEGER:
8696 switch (type)
8698 case BT_INTEGER:
8699 f = gfc_int2int;
8700 break;
8701 case BT_REAL:
8702 f = gfc_int2real;
8703 break;
8704 case BT_COMPLEX:
8705 f = gfc_int2complex;
8706 break;
8707 case BT_LOGICAL:
8708 f = gfc_int2log;
8709 break;
8710 default:
8711 goto oops;
8713 break;
8715 case BT_REAL:
8716 switch (type)
8718 case BT_INTEGER:
8719 f = gfc_real2int;
8720 break;
8721 case BT_REAL:
8722 f = gfc_real2real;
8723 break;
8724 case BT_COMPLEX:
8725 f = gfc_real2complex;
8726 break;
8727 default:
8728 goto oops;
8730 break;
8732 case BT_COMPLEX:
8733 switch (type)
8735 case BT_INTEGER:
8736 f = gfc_complex2int;
8737 break;
8738 case BT_REAL:
8739 f = gfc_complex2real;
8740 break;
8741 case BT_COMPLEX:
8742 f = gfc_complex2complex;
8743 break;
8745 default:
8746 goto oops;
8748 break;
8750 case BT_LOGICAL:
8751 switch (type)
8753 case BT_INTEGER:
8754 f = gfc_log2int;
8755 break;
8756 case BT_LOGICAL:
8757 f = gfc_log2log;
8758 break;
8759 default:
8760 goto oops;
8762 break;
8764 case BT_HOLLERITH:
8765 switch (type)
8767 case BT_INTEGER:
8768 f = gfc_hollerith2int;
8769 break;
8771 case BT_REAL:
8772 f = gfc_hollerith2real;
8773 break;
8775 case BT_COMPLEX:
8776 f = gfc_hollerith2complex;
8777 break;
8779 case BT_CHARACTER:
8780 f = gfc_hollerith2character;
8781 break;
8783 case BT_LOGICAL:
8784 f = gfc_hollerith2logical;
8785 break;
8787 default:
8788 goto oops;
8790 break;
8792 case BT_CHARACTER:
8793 switch (type)
8795 case BT_INTEGER:
8796 f = gfc_character2int;
8797 break;
8799 case BT_REAL:
8800 f = gfc_character2real;
8801 break;
8803 case BT_COMPLEX:
8804 f = gfc_character2complex;
8805 break;
8807 case BT_CHARACTER:
8808 f = gfc_character2character;
8809 break;
8811 case BT_LOGICAL:
8812 f = gfc_character2logical;
8813 break;
8815 default:
8816 goto oops;
8818 break;
8820 default:
8821 oops:
8822 return &gfc_bad_expr;
8825 result = NULL;
8827 switch (e->expr_type)
8829 case EXPR_CONSTANT:
8830 result = f (e, kind);
8831 if (result == NULL)
8832 return &gfc_bad_expr;
8833 break;
8835 case EXPR_ARRAY:
8836 if (!gfc_is_constant_expr (e))
8837 break;
8839 result = gfc_get_array_expr (type, kind, &e->where);
8840 result->shape = gfc_copy_shape (e->shape, e->rank);
8841 result->rank = e->rank;
8843 for (c = gfc_constructor_first (e->value.constructor);
8844 c; c = gfc_constructor_next (c))
8846 gfc_expr *tmp;
8847 if (c->iterator == NULL)
8849 if (c->expr->expr_type == EXPR_ARRAY)
8850 tmp = gfc_convert_constant (c->expr, type, kind);
8851 else if (c->expr->expr_type == EXPR_OP)
8853 if (!gfc_simplify_expr (c->expr, 1))
8854 return &gfc_bad_expr;
8855 tmp = f (c->expr, kind);
8857 else
8858 tmp = f (c->expr, kind);
8860 else
8861 tmp = gfc_convert_constant (c->expr, type, kind);
8863 if (tmp == NULL || tmp == &gfc_bad_expr)
8865 gfc_free_expr (result);
8866 return NULL;
8869 t = gfc_constructor_append_expr (&result->value.constructor,
8870 tmp, &c->where);
8871 if (c->iterator)
8872 t->iterator = gfc_copy_iterator (c->iterator);
8875 break;
8877 default:
8878 break;
8881 return result;
8885 /* Function for converting character constants. */
8886 gfc_expr *
8887 gfc_convert_char_constant (gfc_expr *e, bt type ATTRIBUTE_UNUSED, int kind)
8889 gfc_expr *result;
8890 int i;
8892 if (!gfc_is_constant_expr (e))
8893 return NULL;
8895 if (e->expr_type == EXPR_CONSTANT)
8897 /* Simple case of a scalar. */
8898 result = gfc_get_constant_expr (BT_CHARACTER, kind, &e->where);
8899 if (result == NULL)
8900 return &gfc_bad_expr;
8902 result->value.character.length = e->value.character.length;
8903 result->value.character.string
8904 = gfc_get_wide_string (e->value.character.length + 1);
8905 memcpy (result->value.character.string, e->value.character.string,
8906 (e->value.character.length + 1) * sizeof (gfc_char_t));
8908 /* Check we only have values representable in the destination kind. */
8909 for (i = 0; i < result->value.character.length; i++)
8910 if (!gfc_check_character_range (result->value.character.string[i],
8911 kind))
8913 gfc_error ("Character %qs in string at %L cannot be converted "
8914 "into character kind %d",
8915 gfc_print_wide_char (result->value.character.string[i]),
8916 &e->where, kind);
8917 gfc_free_expr (result);
8918 return &gfc_bad_expr;
8921 return result;
8923 else if (e->expr_type == EXPR_ARRAY)
8925 /* For an array constructor, we convert each constructor element. */
8926 gfc_constructor *c;
8928 result = gfc_get_array_expr (type, kind, &e->where);
8929 result->shape = gfc_copy_shape (e->shape, e->rank);
8930 result->rank = e->rank;
8931 result->ts.u.cl = e->ts.u.cl;
8933 for (c = gfc_constructor_first (e->value.constructor);
8934 c; c = gfc_constructor_next (c))
8936 gfc_expr *tmp = gfc_convert_char_constant (c->expr, type, kind);
8937 if (tmp == &gfc_bad_expr)
8939 gfc_free_expr (result);
8940 return &gfc_bad_expr;
8943 if (tmp == NULL)
8945 gfc_free_expr (result);
8946 return NULL;
8949 gfc_constructor_append_expr (&result->value.constructor,
8950 tmp, &c->where);
8953 return result;
8955 else
8956 return NULL;
8960 gfc_expr *
8961 gfc_simplify_compiler_options (void)
8963 char *str;
8964 gfc_expr *result;
8966 str = gfc_get_option_string ();
8967 result = gfc_get_character_expr (gfc_default_character_kind,
8968 &gfc_current_locus, str, strlen (str));
8969 free (str);
8970 return result;
8974 gfc_expr *
8975 gfc_simplify_compiler_version (void)
8977 char *buffer;
8978 size_t len;
8980 len = strlen ("GCC version ") + strlen (version_string);
8981 buffer = XALLOCAVEC (char, len + 1);
8982 snprintf (buffer, len + 1, "GCC version %s", version_string);
8983 return gfc_get_character_expr (gfc_default_character_kind,
8984 &gfc_current_locus, buffer, len);
8987 /* Simplification routines for intrinsics of IEEE modules. */
8989 gfc_expr *
8990 simplify_ieee_selected_real_kind (gfc_expr *expr)
8992 gfc_actual_arglist *arg;
8993 gfc_expr *p = NULL, *q = NULL, *rdx = NULL;
8995 arg = expr->value.function.actual;
8996 p = arg->expr;
8997 if (arg->next)
8999 q = arg->next->expr;
9000 if (arg->next->next)
9001 rdx = arg->next->next->expr;
9004 /* Currently, if IEEE is supported and this module is built, it means
9005 all our floating-point types conform to IEEE. Hence, we simply handle
9006 IEEE_SELECTED_REAL_KIND like SELECTED_REAL_KIND. */
9007 return gfc_simplify_selected_real_kind (p, q, rdx);
9010 gfc_expr *
9011 simplify_ieee_support (gfc_expr *expr)
9013 /* We consider that if the IEEE modules are loaded, we have full support
9014 for flags, halting and rounding, which are the three functions
9015 (IEEE_SUPPORT_{FLAG,HALTING,ROUNDING}) allowed in constant
9016 expressions. One day, we will need libgfortran to detect support and
9017 communicate it back to us, allowing for partial support. */
9019 return gfc_get_logical_expr (gfc_default_logical_kind, &expr->where,
9020 true);
9023 bool
9024 matches_ieee_function_name (gfc_symbol *sym, const char *name)
9026 int n = strlen(name);
9028 if (!strncmp(sym->name, name, n))
9029 return true;
9031 /* If a generic was used and renamed, we need more work to find out.
9032 Compare the specific name. */
9033 if (sym->generic && !strncmp(sym->generic->sym->name, name, n))
9034 return true;
9036 return false;
9039 gfc_expr *
9040 gfc_simplify_ieee_functions (gfc_expr *expr)
9042 gfc_symbol* sym = expr->symtree->n.sym;
9044 if (matches_ieee_function_name(sym, "ieee_selected_real_kind"))
9045 return simplify_ieee_selected_real_kind (expr);
9046 else if (matches_ieee_function_name(sym, "ieee_support_flag")
9047 || matches_ieee_function_name(sym, "ieee_support_halting")
9048 || matches_ieee_function_name(sym, "ieee_support_rounding"))
9049 return simplify_ieee_support (expr);
9050 else
9051 return NULL;