install.texi (mips-*-*): Recommend binutils 2.18.
[official-gcc.git] / gcc / fortran / expr.c
blobfb1886e94b612d57d85ca642c9c2d4d331c13089
1 /* Routines for manipulation of expression nodes.
2 Copyright (C) 2000, 2001, 2002, 2003, 2004, 2005, 2006, 2007
3 Free Software Foundation, Inc.
4 Contributed by Andy Vaught
6 This file is part of GCC.
8 GCC is free software; you can redistribute it and/or modify it under
9 the terms of the GNU General Public License as published by the Free
10 Software Foundation; either version 3, or (at your option) any later
11 version.
13 GCC is distributed in the hope that it will be useful, but WITHOUT ANY
14 WARRANTY; without even the implied warranty of MERCHANTABILITY or
15 FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
16 for more details.
18 You should have received a copy of the GNU General Public License
19 along with GCC; see the file COPYING3. If not see
20 <http://www.gnu.org/licenses/>. */
22 #include "config.h"
23 #include "system.h"
24 #include "gfortran.h"
25 #include "arith.h"
26 #include "match.h"
27 #include "target-memory.h" /* for gfc_convert_boz */
29 /* Get a new expr node. */
31 gfc_expr *
32 gfc_get_expr (void)
34 gfc_expr *e;
36 e = gfc_getmem (sizeof (gfc_expr));
37 gfc_clear_ts (&e->ts);
38 e->shape = NULL;
39 e->ref = NULL;
40 e->symtree = NULL;
41 e->con_by_offset = NULL;
42 return e;
46 /* Free an argument list and everything below it. */
48 void
49 gfc_free_actual_arglist (gfc_actual_arglist *a1)
51 gfc_actual_arglist *a2;
53 while (a1)
55 a2 = a1->next;
56 gfc_free_expr (a1->expr);
57 gfc_free (a1);
58 a1 = a2;
63 /* Copy an arglist structure and all of the arguments. */
65 gfc_actual_arglist *
66 gfc_copy_actual_arglist (gfc_actual_arglist *p)
68 gfc_actual_arglist *head, *tail, *new;
70 head = tail = NULL;
72 for (; p; p = p->next)
74 new = gfc_get_actual_arglist ();
75 *new = *p;
77 new->expr = gfc_copy_expr (p->expr);
78 new->next = NULL;
80 if (head == NULL)
81 head = new;
82 else
83 tail->next = new;
85 tail = new;
88 return head;
92 /* Free a list of reference structures. */
94 void
95 gfc_free_ref_list (gfc_ref *p)
97 gfc_ref *q;
98 int i;
100 for (; p; p = q)
102 q = p->next;
104 switch (p->type)
106 case REF_ARRAY:
107 for (i = 0; i < GFC_MAX_DIMENSIONS; i++)
109 gfc_free_expr (p->u.ar.start[i]);
110 gfc_free_expr (p->u.ar.end[i]);
111 gfc_free_expr (p->u.ar.stride[i]);
114 break;
116 case REF_SUBSTRING:
117 gfc_free_expr (p->u.ss.start);
118 gfc_free_expr (p->u.ss.end);
119 break;
121 case REF_COMPONENT:
122 break;
125 gfc_free (p);
130 /* Workhorse function for gfc_free_expr() that frees everything
131 beneath an expression node, but not the node itself. This is
132 useful when we want to simplify a node and replace it with
133 something else or the expression node belongs to another structure. */
135 static void
136 free_expr0 (gfc_expr *e)
138 int n;
140 switch (e->expr_type)
142 case EXPR_CONSTANT:
143 /* Free any parts of the value that need freeing. */
144 switch (e->ts.type)
146 case BT_INTEGER:
147 mpz_clear (e->value.integer);
148 break;
150 case BT_REAL:
151 mpfr_clear (e->value.real);
152 break;
154 case BT_CHARACTER:
155 gfc_free (e->value.character.string);
156 break;
158 case BT_COMPLEX:
159 mpfr_clear (e->value.complex.r);
160 mpfr_clear (e->value.complex.i);
161 break;
163 default:
164 break;
167 /* Free the representation, except in character constants where it
168 is the same as value.character.string and thus already freed. */
169 if (e->representation.string && e->ts.type != BT_CHARACTER)
170 gfc_free (e->representation.string);
172 break;
174 case EXPR_OP:
175 if (e->value.op.op1 != NULL)
176 gfc_free_expr (e->value.op.op1);
177 if (e->value.op.op2 != NULL)
178 gfc_free_expr (e->value.op.op2);
179 break;
181 case EXPR_FUNCTION:
182 gfc_free_actual_arglist (e->value.function.actual);
183 break;
185 case EXPR_VARIABLE:
186 break;
188 case EXPR_ARRAY:
189 case EXPR_STRUCTURE:
190 gfc_free_constructor (e->value.constructor);
191 break;
193 case EXPR_SUBSTRING:
194 gfc_free (e->value.character.string);
195 break;
197 case EXPR_NULL:
198 break;
200 default:
201 gfc_internal_error ("free_expr0(): Bad expr type");
204 /* Free a shape array. */
205 if (e->shape != NULL)
207 for (n = 0; n < e->rank; n++)
208 mpz_clear (e->shape[n]);
210 gfc_free (e->shape);
213 gfc_free_ref_list (e->ref);
215 memset (e, '\0', sizeof (gfc_expr));
219 /* Free an expression node and everything beneath it. */
221 void
222 gfc_free_expr (gfc_expr *e)
224 if (e == NULL)
225 return;
226 if (e->con_by_offset)
227 splay_tree_delete (e->con_by_offset);
228 free_expr0 (e);
229 gfc_free (e);
233 /* Graft the *src expression onto the *dest subexpression. */
235 void
236 gfc_replace_expr (gfc_expr *dest, gfc_expr *src)
238 free_expr0 (dest);
239 *dest = *src;
240 gfc_free (src);
244 /* Try to extract an integer constant from the passed expression node.
245 Returns an error message or NULL if the result is set. It is
246 tempting to generate an error and return SUCCESS or FAILURE, but
247 failure is OK for some callers. */
249 const char *
250 gfc_extract_int (gfc_expr *expr, int *result)
252 if (expr->expr_type != EXPR_CONSTANT)
253 return _("Constant expression required at %C");
255 if (expr->ts.type != BT_INTEGER)
256 return _("Integer expression required at %C");
258 if ((mpz_cmp_si (expr->value.integer, INT_MAX) > 0)
259 || (mpz_cmp_si (expr->value.integer, INT_MIN) < 0))
261 return _("Integer value too large in expression at %C");
264 *result = (int) mpz_get_si (expr->value.integer);
266 return NULL;
270 /* Recursively copy a list of reference structures. */
272 static gfc_ref *
273 copy_ref (gfc_ref *src)
275 gfc_array_ref *ar;
276 gfc_ref *dest;
278 if (src == NULL)
279 return NULL;
281 dest = gfc_get_ref ();
282 dest->type = src->type;
284 switch (src->type)
286 case REF_ARRAY:
287 ar = gfc_copy_array_ref (&src->u.ar);
288 dest->u.ar = *ar;
289 gfc_free (ar);
290 break;
292 case REF_COMPONENT:
293 dest->u.c = src->u.c;
294 break;
296 case REF_SUBSTRING:
297 dest->u.ss = src->u.ss;
298 dest->u.ss.start = gfc_copy_expr (src->u.ss.start);
299 dest->u.ss.end = gfc_copy_expr (src->u.ss.end);
300 break;
303 dest->next = copy_ref (src->next);
305 return dest;
309 /* Detect whether an expression has any vector index array references. */
312 gfc_has_vector_index (gfc_expr *e)
314 gfc_ref *ref;
315 int i;
316 for (ref = e->ref; ref; ref = ref->next)
317 if (ref->type == REF_ARRAY)
318 for (i = 0; i < ref->u.ar.dimen; i++)
319 if (ref->u.ar.dimen_type[i] == DIMEN_VECTOR)
320 return 1;
321 return 0;
325 /* Copy a shape array. */
327 mpz_t *
328 gfc_copy_shape (mpz_t *shape, int rank)
330 mpz_t *new_shape;
331 int n;
333 if (shape == NULL)
334 return NULL;
336 new_shape = gfc_get_shape (rank);
338 for (n = 0; n < rank; n++)
339 mpz_init_set (new_shape[n], shape[n]);
341 return new_shape;
345 /* Copy a shape array excluding dimension N, where N is an integer
346 constant expression. Dimensions are numbered in fortran style --
347 starting with ONE.
349 So, if the original shape array contains R elements
350 { s1 ... sN-1 sN sN+1 ... sR-1 sR}
351 the result contains R-1 elements:
352 { s1 ... sN-1 sN+1 ... sR-1}
354 If anything goes wrong -- N is not a constant, its value is out
355 of range -- or anything else, just returns NULL. */
357 mpz_t *
358 gfc_copy_shape_excluding (mpz_t *shape, int rank, gfc_expr *dim)
360 mpz_t *new_shape, *s;
361 int i, n;
363 if (shape == NULL
364 || rank <= 1
365 || dim == NULL
366 || dim->expr_type != EXPR_CONSTANT
367 || dim->ts.type != BT_INTEGER)
368 return NULL;
370 n = mpz_get_si (dim->value.integer);
371 n--; /* Convert to zero based index. */
372 if (n < 0 || n >= rank)
373 return NULL;
375 s = new_shape = gfc_get_shape (rank - 1);
377 for (i = 0; i < rank; i++)
379 if (i == n)
380 continue;
381 mpz_init_set (*s, shape[i]);
382 s++;
385 return new_shape;
389 /* Given an expression pointer, return a copy of the expression. This
390 subroutine is recursive. */
392 gfc_expr *
393 gfc_copy_expr (gfc_expr *p)
395 gfc_expr *q;
396 char *s;
398 if (p == NULL)
399 return NULL;
401 q = gfc_get_expr ();
402 *q = *p;
404 switch (q->expr_type)
406 case EXPR_SUBSTRING:
407 s = gfc_getmem (p->value.character.length + 1);
408 q->value.character.string = s;
410 memcpy (s, p->value.character.string, p->value.character.length + 1);
411 break;
413 case EXPR_CONSTANT:
414 /* Copy target representation, if it exists. */
415 if (p->representation.string)
417 s = gfc_getmem (p->representation.length + 1);
418 q->representation.string = s;
420 memcpy (s, p->representation.string, p->representation.length + 1);
423 /* Copy the values of any pointer components of p->value. */
424 switch (q->ts.type)
426 case BT_INTEGER:
427 mpz_init_set (q->value.integer, p->value.integer);
428 break;
430 case BT_REAL:
431 gfc_set_model_kind (q->ts.kind);
432 mpfr_init (q->value.real);
433 mpfr_set (q->value.real, p->value.real, GFC_RND_MODE);
434 break;
436 case BT_COMPLEX:
437 gfc_set_model_kind (q->ts.kind);
438 mpfr_init (q->value.complex.r);
439 mpfr_init (q->value.complex.i);
440 mpfr_set (q->value.complex.r, p->value.complex.r, GFC_RND_MODE);
441 mpfr_set (q->value.complex.i, p->value.complex.i, GFC_RND_MODE);
442 break;
444 case BT_CHARACTER:
445 if (p->representation.string)
446 q->value.character.string = q->representation.string;
447 else
449 s = gfc_getmem (p->value.character.length + 1);
450 q->value.character.string = s;
452 /* This is the case for the C_NULL_CHAR named constant. */
453 if (p->value.character.length == 0
454 && (p->ts.is_c_interop || p->ts.is_iso_c))
456 *s = '\0';
457 /* Need to set the length to 1 to make sure the NUL
458 terminator is copied. */
459 q->value.character.length = 1;
461 else
462 memcpy (s, p->value.character.string,
463 p->value.character.length + 1);
465 break;
467 case BT_HOLLERITH:
468 case BT_LOGICAL:
469 case BT_DERIVED:
470 break; /* Already done. */
472 case BT_PROCEDURE:
473 case BT_VOID:
474 /* Should never be reached. */
475 case BT_UNKNOWN:
476 gfc_internal_error ("gfc_copy_expr(): Bad expr node");
477 /* Not reached. */
480 break;
482 case EXPR_OP:
483 switch (q->value.op.operator)
485 case INTRINSIC_NOT:
486 case INTRINSIC_PARENTHESES:
487 case INTRINSIC_UPLUS:
488 case INTRINSIC_UMINUS:
489 q->value.op.op1 = gfc_copy_expr (p->value.op.op1);
490 break;
492 default: /* Binary operators. */
493 q->value.op.op1 = gfc_copy_expr (p->value.op.op1);
494 q->value.op.op2 = gfc_copy_expr (p->value.op.op2);
495 break;
498 break;
500 case EXPR_FUNCTION:
501 q->value.function.actual =
502 gfc_copy_actual_arglist (p->value.function.actual);
503 break;
505 case EXPR_STRUCTURE:
506 case EXPR_ARRAY:
507 q->value.constructor = gfc_copy_constructor (p->value.constructor);
508 break;
510 case EXPR_VARIABLE:
511 case EXPR_NULL:
512 break;
515 q->shape = gfc_copy_shape (p->shape, p->rank);
517 q->ref = copy_ref (p->ref);
519 return q;
523 /* Return the maximum kind of two expressions. In general, higher
524 kind numbers mean more precision for numeric types. */
527 gfc_kind_max (gfc_expr *e1, gfc_expr *e2)
529 return (e1->ts.kind > e2->ts.kind) ? e1->ts.kind : e2->ts.kind;
533 /* Returns nonzero if the type is numeric, zero otherwise. */
535 static int
536 numeric_type (bt type)
538 return type == BT_COMPLEX || type == BT_REAL || type == BT_INTEGER;
542 /* Returns nonzero if the typespec is a numeric type, zero otherwise. */
545 gfc_numeric_ts (gfc_typespec *ts)
547 return numeric_type (ts->type);
551 /* Returns an expression node that is an integer constant. */
553 gfc_expr *
554 gfc_int_expr (int i)
556 gfc_expr *p;
558 p = gfc_get_expr ();
560 p->expr_type = EXPR_CONSTANT;
561 p->ts.type = BT_INTEGER;
562 p->ts.kind = gfc_default_integer_kind;
564 p->where = gfc_current_locus;
565 mpz_init_set_si (p->value.integer, i);
567 return p;
571 /* Returns an expression node that is a logical constant. */
573 gfc_expr *
574 gfc_logical_expr (int i, locus *where)
576 gfc_expr *p;
578 p = gfc_get_expr ();
580 p->expr_type = EXPR_CONSTANT;
581 p->ts.type = BT_LOGICAL;
582 p->ts.kind = gfc_default_logical_kind;
584 if (where == NULL)
585 where = &gfc_current_locus;
586 p->where = *where;
587 p->value.logical = i;
589 return p;
593 /* Return an expression node with an optional argument list attached.
594 A variable number of gfc_expr pointers are strung together in an
595 argument list with a NULL pointer terminating the list. */
597 gfc_expr *
598 gfc_build_conversion (gfc_expr *e)
600 gfc_expr *p;
602 p = gfc_get_expr ();
603 p->expr_type = EXPR_FUNCTION;
604 p->symtree = NULL;
605 p->value.function.actual = NULL;
607 p->value.function.actual = gfc_get_actual_arglist ();
608 p->value.function.actual->expr = e;
610 return p;
614 /* Given an expression node with some sort of numeric binary
615 expression, insert type conversions required to make the operands
616 have the same type.
618 The exception is that the operands of an exponential don't have to
619 have the same type. If possible, the base is promoted to the type
620 of the exponent. For example, 1**2.3 becomes 1.0**2.3, but
621 1.0**2 stays as it is. */
623 void
624 gfc_type_convert_binary (gfc_expr *e)
626 gfc_expr *op1, *op2;
628 op1 = e->value.op.op1;
629 op2 = e->value.op.op2;
631 if (op1->ts.type == BT_UNKNOWN || op2->ts.type == BT_UNKNOWN)
633 gfc_clear_ts (&e->ts);
634 return;
637 /* Kind conversions of same type. */
638 if (op1->ts.type == op2->ts.type)
640 if (op1->ts.kind == op2->ts.kind)
642 /* No type conversions. */
643 e->ts = op1->ts;
644 goto done;
647 if (op1->ts.kind > op2->ts.kind)
648 gfc_convert_type (op2, &op1->ts, 2);
649 else
650 gfc_convert_type (op1, &op2->ts, 2);
652 e->ts = op1->ts;
653 goto done;
656 /* Integer combined with real or complex. */
657 if (op2->ts.type == BT_INTEGER)
659 e->ts = op1->ts;
661 /* Special case for ** operator. */
662 if (e->value.op.operator == INTRINSIC_POWER)
663 goto done;
665 gfc_convert_type (e->value.op.op2, &e->ts, 2);
666 goto done;
669 if (op1->ts.type == BT_INTEGER)
671 e->ts = op2->ts;
672 gfc_convert_type (e->value.op.op1, &e->ts, 2);
673 goto done;
676 /* Real combined with complex. */
677 e->ts.type = BT_COMPLEX;
678 if (op1->ts.kind > op2->ts.kind)
679 e->ts.kind = op1->ts.kind;
680 else
681 e->ts.kind = op2->ts.kind;
682 if (op1->ts.type != BT_COMPLEX || op1->ts.kind != e->ts.kind)
683 gfc_convert_type (e->value.op.op1, &e->ts, 2);
684 if (op2->ts.type != BT_COMPLEX || op2->ts.kind != e->ts.kind)
685 gfc_convert_type (e->value.op.op2, &e->ts, 2);
687 done:
688 return;
692 static match
693 check_specification_function (gfc_expr *e)
695 gfc_symbol *sym;
697 if (!e->symtree)
698 return MATCH_NO;
700 sym = e->symtree->n.sym;
702 /* F95, 7.1.6.2; F2003, 7.1.7 */
703 if (sym
704 && sym->attr.function
705 && sym->attr.pure
706 && !sym->attr.intrinsic
707 && !sym->attr.recursive
708 && sym->attr.proc != PROC_INTERNAL
709 && sym->attr.proc != PROC_ST_FUNCTION
710 && sym->attr.proc != PROC_UNKNOWN
711 && sym->formal == NULL)
712 return MATCH_YES;
714 return MATCH_NO;
717 /* Function to determine if an expression is constant or not. This
718 function expects that the expression has already been simplified. */
721 gfc_is_constant_expr (gfc_expr *e)
723 gfc_constructor *c;
724 gfc_actual_arglist *arg;
725 int rv;
727 if (e == NULL)
728 return 1;
730 switch (e->expr_type)
732 case EXPR_OP:
733 rv = (gfc_is_constant_expr (e->value.op.op1)
734 && (e->value.op.op2 == NULL
735 || gfc_is_constant_expr (e->value.op.op2)));
736 break;
738 case EXPR_VARIABLE:
739 rv = 0;
740 break;
742 case EXPR_FUNCTION:
743 /* Specification functions are constant. */
744 if (check_specification_function (e) == MATCH_YES)
746 rv = 1;
747 break;
750 /* Call to intrinsic with at least one argument. */
751 rv = 0;
752 if (e->value.function.isym && e->value.function.actual)
754 for (arg = e->value.function.actual; arg; arg = arg->next)
756 if (!gfc_is_constant_expr (arg->expr))
757 break;
759 if (arg == NULL)
760 rv = 1;
762 break;
764 case EXPR_CONSTANT:
765 case EXPR_NULL:
766 rv = 1;
767 break;
769 case EXPR_SUBSTRING:
770 rv = e->ref == NULL || (gfc_is_constant_expr (e->ref->u.ss.start)
771 && gfc_is_constant_expr (e->ref->u.ss.end));
772 break;
774 case EXPR_STRUCTURE:
775 rv = 0;
776 for (c = e->value.constructor; c; c = c->next)
777 if (!gfc_is_constant_expr (c->expr))
778 break;
780 if (c == NULL)
781 rv = 1;
782 break;
784 case EXPR_ARRAY:
785 rv = gfc_constant_ac (e);
786 break;
788 default:
789 gfc_internal_error ("gfc_is_constant_expr(): Unknown expression type");
792 return rv;
796 /* Is true if an array reference is followed by a component or substring
797 reference. */
798 bool
799 is_subref_array (gfc_expr * e)
801 gfc_ref * ref;
802 bool seen_array;
804 if (e->expr_type != EXPR_VARIABLE)
805 return false;
807 if (e->symtree->n.sym->attr.subref_array_pointer)
808 return true;
810 seen_array = false;
811 for (ref = e->ref; ref; ref = ref->next)
813 if (ref->type == REF_ARRAY
814 && ref->u.ar.type != AR_ELEMENT)
815 seen_array = true;
817 if (seen_array
818 && ref->type != REF_ARRAY)
819 return seen_array;
821 return false;
825 /* Try to collapse intrinsic expressions. */
827 static try
828 simplify_intrinsic_op (gfc_expr *p, int type)
830 gfc_intrinsic_op op;
831 gfc_expr *op1, *op2, *result;
833 if (p->value.op.operator == INTRINSIC_USER)
834 return SUCCESS;
836 op1 = p->value.op.op1;
837 op2 = p->value.op.op2;
838 op = p->value.op.operator;
840 if (gfc_simplify_expr (op1, type) == FAILURE)
841 return FAILURE;
842 if (gfc_simplify_expr (op2, type) == FAILURE)
843 return FAILURE;
845 if (!gfc_is_constant_expr (op1)
846 || (op2 != NULL && !gfc_is_constant_expr (op2)))
847 return SUCCESS;
849 /* Rip p apart. */
850 p->value.op.op1 = NULL;
851 p->value.op.op2 = NULL;
853 switch (op)
855 case INTRINSIC_PARENTHESES:
856 result = gfc_parentheses (op1);
857 break;
859 case INTRINSIC_UPLUS:
860 result = gfc_uplus (op1);
861 break;
863 case INTRINSIC_UMINUS:
864 result = gfc_uminus (op1);
865 break;
867 case INTRINSIC_PLUS:
868 result = gfc_add (op1, op2);
869 break;
871 case INTRINSIC_MINUS:
872 result = gfc_subtract (op1, op2);
873 break;
875 case INTRINSIC_TIMES:
876 result = gfc_multiply (op1, op2);
877 break;
879 case INTRINSIC_DIVIDE:
880 result = gfc_divide (op1, op2);
881 break;
883 case INTRINSIC_POWER:
884 result = gfc_power (op1, op2);
885 break;
887 case INTRINSIC_CONCAT:
888 result = gfc_concat (op1, op2);
889 break;
891 case INTRINSIC_EQ:
892 case INTRINSIC_EQ_OS:
893 result = gfc_eq (op1, op2, op);
894 break;
896 case INTRINSIC_NE:
897 case INTRINSIC_NE_OS:
898 result = gfc_ne (op1, op2, op);
899 break;
901 case INTRINSIC_GT:
902 case INTRINSIC_GT_OS:
903 result = gfc_gt (op1, op2, op);
904 break;
906 case INTRINSIC_GE:
907 case INTRINSIC_GE_OS:
908 result = gfc_ge (op1, op2, op);
909 break;
911 case INTRINSIC_LT:
912 case INTRINSIC_LT_OS:
913 result = gfc_lt (op1, op2, op);
914 break;
916 case INTRINSIC_LE:
917 case INTRINSIC_LE_OS:
918 result = gfc_le (op1, op2, op);
919 break;
921 case INTRINSIC_NOT:
922 result = gfc_not (op1);
923 break;
925 case INTRINSIC_AND:
926 result = gfc_and (op1, op2);
927 break;
929 case INTRINSIC_OR:
930 result = gfc_or (op1, op2);
931 break;
933 case INTRINSIC_EQV:
934 result = gfc_eqv (op1, op2);
935 break;
937 case INTRINSIC_NEQV:
938 result = gfc_neqv (op1, op2);
939 break;
941 default:
942 gfc_internal_error ("simplify_intrinsic_op(): Bad operator");
945 if (result == NULL)
947 gfc_free_expr (op1);
948 gfc_free_expr (op2);
949 return FAILURE;
952 result->rank = p->rank;
953 result->where = p->where;
954 gfc_replace_expr (p, result);
956 return SUCCESS;
960 /* Subroutine to simplify constructor expressions. Mutually recursive
961 with gfc_simplify_expr(). */
963 static try
964 simplify_constructor (gfc_constructor *c, int type)
966 gfc_expr *p;
968 for (; c; c = c->next)
970 if (c->iterator
971 && (gfc_simplify_expr (c->iterator->start, type) == FAILURE
972 || gfc_simplify_expr (c->iterator->end, type) == FAILURE
973 || gfc_simplify_expr (c->iterator->step, type) == FAILURE))
974 return FAILURE;
976 if (c->expr)
978 /* Try and simplify a copy. Replace the original if successful
979 but keep going through the constructor at all costs. Not
980 doing so can make a dog's dinner of complicated things. */
981 p = gfc_copy_expr (c->expr);
983 if (gfc_simplify_expr (p, type) == FAILURE)
985 gfc_free_expr (p);
986 continue;
989 gfc_replace_expr (c->expr, p);
993 return SUCCESS;
997 /* Pull a single array element out of an array constructor. */
999 static try
1000 find_array_element (gfc_constructor *cons, gfc_array_ref *ar,
1001 gfc_constructor **rval)
1003 unsigned long nelemen;
1004 int i;
1005 mpz_t delta;
1006 mpz_t offset;
1007 mpz_t span;
1008 mpz_t tmp;
1009 gfc_expr *e;
1010 try t;
1012 t = SUCCESS;
1013 e = NULL;
1015 mpz_init_set_ui (offset, 0);
1016 mpz_init (delta);
1017 mpz_init (tmp);
1018 mpz_init_set_ui (span, 1);
1019 for (i = 0; i < ar->dimen; i++)
1021 e = gfc_copy_expr (ar->start[i]);
1022 if (e->expr_type != EXPR_CONSTANT)
1024 cons = NULL;
1025 goto depart;
1027 /* Check the bounds. */
1028 if ((ar->as->upper[i]
1029 && ar->as->upper[i]->expr_type == EXPR_CONSTANT
1030 && mpz_cmp (e->value.integer,
1031 ar->as->upper[i]->value.integer) > 0)
1033 (ar->as->lower[i]->expr_type == EXPR_CONSTANT
1034 && mpz_cmp (e->value.integer,
1035 ar->as->lower[i]->value.integer) < 0))
1037 gfc_error ("Index in dimension %d is out of bounds "
1038 "at %L", i + 1, &ar->c_where[i]);
1039 cons = NULL;
1040 t = FAILURE;
1041 goto depart;
1044 mpz_sub (delta, e->value.integer, ar->as->lower[i]->value.integer);
1045 mpz_mul (delta, delta, span);
1046 mpz_add (offset, offset, delta);
1048 mpz_set_ui (tmp, 1);
1049 mpz_add (tmp, tmp, ar->as->upper[i]->value.integer);
1050 mpz_sub (tmp, tmp, ar->as->lower[i]->value.integer);
1051 mpz_mul (span, span, tmp);
1054 if (cons)
1056 for (nelemen = mpz_get_ui (offset); nelemen > 0; nelemen--)
1058 if (cons->iterator)
1060 cons = NULL;
1061 goto depart;
1063 cons = cons->next;
1067 depart:
1068 mpz_clear (delta);
1069 mpz_clear (offset);
1070 mpz_clear (span);
1071 mpz_clear (tmp);
1072 if (e)
1073 gfc_free_expr (e);
1074 *rval = cons;
1075 return t;
1079 /* Find a component of a structure constructor. */
1081 static gfc_constructor *
1082 find_component_ref (gfc_constructor *cons, gfc_ref *ref)
1084 gfc_component *comp;
1085 gfc_component *pick;
1087 comp = ref->u.c.sym->components;
1088 pick = ref->u.c.component;
1089 while (comp != pick)
1091 comp = comp->next;
1092 cons = cons->next;
1095 return cons;
1099 /* Replace an expression with the contents of a constructor, removing
1100 the subobject reference in the process. */
1102 static void
1103 remove_subobject_ref (gfc_expr *p, gfc_constructor *cons)
1105 gfc_expr *e;
1107 e = cons->expr;
1108 cons->expr = NULL;
1109 e->ref = p->ref->next;
1110 p->ref->next = NULL;
1111 gfc_replace_expr (p, e);
1115 /* Pull an array section out of an array constructor. */
1117 static try
1118 find_array_section (gfc_expr *expr, gfc_ref *ref)
1120 int idx;
1121 int rank;
1122 int d;
1123 int shape_i;
1124 long unsigned one = 1;
1125 bool incr_ctr;
1126 mpz_t start[GFC_MAX_DIMENSIONS];
1127 mpz_t end[GFC_MAX_DIMENSIONS];
1128 mpz_t stride[GFC_MAX_DIMENSIONS];
1129 mpz_t delta[GFC_MAX_DIMENSIONS];
1130 mpz_t ctr[GFC_MAX_DIMENSIONS];
1131 mpz_t delta_mpz;
1132 mpz_t tmp_mpz;
1133 mpz_t nelts;
1134 mpz_t ptr;
1135 mpz_t index;
1136 gfc_constructor *cons;
1137 gfc_constructor *base;
1138 gfc_expr *begin;
1139 gfc_expr *finish;
1140 gfc_expr *step;
1141 gfc_expr *upper;
1142 gfc_expr *lower;
1143 gfc_constructor *vecsub[GFC_MAX_DIMENSIONS], *c;
1144 try t;
1146 t = SUCCESS;
1148 base = expr->value.constructor;
1149 expr->value.constructor = NULL;
1151 rank = ref->u.ar.as->rank;
1153 if (expr->shape == NULL)
1154 expr->shape = gfc_get_shape (rank);
1156 mpz_init_set_ui (delta_mpz, one);
1157 mpz_init_set_ui (nelts, one);
1158 mpz_init (tmp_mpz);
1160 /* Do the initialization now, so that we can cleanup without
1161 keeping track of where we were. */
1162 for (d = 0; d < rank; d++)
1164 mpz_init (delta[d]);
1165 mpz_init (start[d]);
1166 mpz_init (end[d]);
1167 mpz_init (ctr[d]);
1168 mpz_init (stride[d]);
1169 vecsub[d] = NULL;
1172 /* Build the counters to clock through the array reference. */
1173 shape_i = 0;
1174 for (d = 0; d < rank; d++)
1176 /* Make this stretch of code easier on the eye! */
1177 begin = ref->u.ar.start[d];
1178 finish = ref->u.ar.end[d];
1179 step = ref->u.ar.stride[d];
1180 lower = ref->u.ar.as->lower[d];
1181 upper = ref->u.ar.as->upper[d];
1183 if (ref->u.ar.dimen_type[d] == DIMEN_VECTOR) /* Vector subscript. */
1185 gcc_assert (begin);
1187 if (begin->expr_type != EXPR_ARRAY || !gfc_is_constant_expr (begin))
1189 t = FAILURE;
1190 goto cleanup;
1193 gcc_assert (begin->rank == 1);
1194 gcc_assert (begin->shape);
1196 vecsub[d] = begin->value.constructor;
1197 mpz_set (ctr[d], vecsub[d]->expr->value.integer);
1198 mpz_mul (nelts, nelts, begin->shape[0]);
1199 mpz_set (expr->shape[shape_i++], begin->shape[0]);
1201 /* Check bounds. */
1202 for (c = vecsub[d]; c; c = c->next)
1204 if (mpz_cmp (c->expr->value.integer, upper->value.integer) > 0
1205 || mpz_cmp (c->expr->value.integer,
1206 lower->value.integer) < 0)
1208 gfc_error ("index in dimension %d is out of bounds "
1209 "at %L", d + 1, &ref->u.ar.c_where[d]);
1210 t = FAILURE;
1211 goto cleanup;
1215 else
1217 if ((begin && begin->expr_type != EXPR_CONSTANT)
1218 || (finish && finish->expr_type != EXPR_CONSTANT)
1219 || (step && step->expr_type != EXPR_CONSTANT))
1221 t = FAILURE;
1222 goto cleanup;
1225 /* Obtain the stride. */
1226 if (step)
1227 mpz_set (stride[d], step->value.integer);
1228 else
1229 mpz_set_ui (stride[d], one);
1231 if (mpz_cmp_ui (stride[d], 0) == 0)
1232 mpz_set_ui (stride[d], one);
1234 /* Obtain the start value for the index. */
1235 if (begin)
1236 mpz_set (start[d], begin->value.integer);
1237 else
1238 mpz_set (start[d], lower->value.integer);
1240 mpz_set (ctr[d], start[d]);
1242 /* Obtain the end value for the index. */
1243 if (finish)
1244 mpz_set (end[d], finish->value.integer);
1245 else
1246 mpz_set (end[d], upper->value.integer);
1248 /* Separate 'if' because elements sometimes arrive with
1249 non-null end. */
1250 if (ref->u.ar.dimen_type[d] == DIMEN_ELEMENT)
1251 mpz_set (end [d], begin->value.integer);
1253 /* Check the bounds. */
1254 if (mpz_cmp (ctr[d], upper->value.integer) > 0
1255 || mpz_cmp (end[d], upper->value.integer) > 0
1256 || mpz_cmp (ctr[d], lower->value.integer) < 0
1257 || mpz_cmp (end[d], lower->value.integer) < 0)
1259 gfc_error ("index in dimension %d is out of bounds "
1260 "at %L", d + 1, &ref->u.ar.c_where[d]);
1261 t = FAILURE;
1262 goto cleanup;
1265 /* Calculate the number of elements and the shape. */
1266 mpz_set (tmp_mpz, stride[d]);
1267 mpz_add (tmp_mpz, end[d], tmp_mpz);
1268 mpz_sub (tmp_mpz, tmp_mpz, ctr[d]);
1269 mpz_div (tmp_mpz, tmp_mpz, stride[d]);
1270 mpz_mul (nelts, nelts, tmp_mpz);
1272 /* An element reference reduces the rank of the expression; don't
1273 add anything to the shape array. */
1274 if (ref->u.ar.dimen_type[d] != DIMEN_ELEMENT)
1275 mpz_set (expr->shape[shape_i++], tmp_mpz);
1278 /* Calculate the 'stride' (=delta) for conversion of the
1279 counter values into the index along the constructor. */
1280 mpz_set (delta[d], delta_mpz);
1281 mpz_sub (tmp_mpz, upper->value.integer, lower->value.integer);
1282 mpz_add_ui (tmp_mpz, tmp_mpz, one);
1283 mpz_mul (delta_mpz, delta_mpz, tmp_mpz);
1286 mpz_init (index);
1287 mpz_init (ptr);
1288 cons = base;
1290 /* Now clock through the array reference, calculating the index in
1291 the source constructor and transferring the elements to the new
1292 constructor. */
1293 for (idx = 0; idx < (int) mpz_get_si (nelts); idx++)
1295 if (ref->u.ar.offset)
1296 mpz_set (ptr, ref->u.ar.offset->value.integer);
1297 else
1298 mpz_init_set_ui (ptr, 0);
1300 incr_ctr = true;
1301 for (d = 0; d < rank; d++)
1303 mpz_set (tmp_mpz, ctr[d]);
1304 mpz_sub (tmp_mpz, tmp_mpz, ref->u.ar.as->lower[d]->value.integer);
1305 mpz_mul (tmp_mpz, tmp_mpz, delta[d]);
1306 mpz_add (ptr, ptr, tmp_mpz);
1308 if (!incr_ctr) continue;
1310 if (ref->u.ar.dimen_type[d] == DIMEN_VECTOR) /* Vector subscript. */
1312 gcc_assert(vecsub[d]);
1314 if (!vecsub[d]->next)
1315 vecsub[d] = ref->u.ar.start[d]->value.constructor;
1316 else
1318 vecsub[d] = vecsub[d]->next;
1319 incr_ctr = false;
1321 mpz_set (ctr[d], vecsub[d]->expr->value.integer);
1323 else
1325 mpz_add (ctr[d], ctr[d], stride[d]);
1327 if (mpz_cmp_ui (stride[d], 0) > 0
1328 ? mpz_cmp (ctr[d], end[d]) > 0
1329 : mpz_cmp (ctr[d], end[d]) < 0)
1330 mpz_set (ctr[d], start[d]);
1331 else
1332 incr_ctr = false;
1336 /* There must be a better way of dealing with negative strides
1337 than resetting the index and the constructor pointer! */
1338 if (mpz_cmp (ptr, index) < 0)
1340 mpz_set_ui (index, 0);
1341 cons = base;
1344 while (mpz_cmp (ptr, index) > 0)
1346 mpz_add_ui (index, index, one);
1347 cons = cons->next;
1350 gfc_append_constructor (expr, gfc_copy_expr (cons->expr));
1353 mpz_clear (ptr);
1354 mpz_clear (index);
1356 cleanup:
1358 mpz_clear (delta_mpz);
1359 mpz_clear (tmp_mpz);
1360 mpz_clear (nelts);
1361 for (d = 0; d < rank; d++)
1363 mpz_clear (delta[d]);
1364 mpz_clear (start[d]);
1365 mpz_clear (end[d]);
1366 mpz_clear (ctr[d]);
1367 mpz_clear (stride[d]);
1369 gfc_free_constructor (base);
1370 return t;
1373 /* Pull a substring out of an expression. */
1375 static try
1376 find_substring_ref (gfc_expr *p, gfc_expr **newp)
1378 int end;
1379 int start;
1380 int length;
1381 char *chr;
1383 if (p->ref->u.ss.start->expr_type != EXPR_CONSTANT
1384 || p->ref->u.ss.end->expr_type != EXPR_CONSTANT)
1385 return FAILURE;
1387 *newp = gfc_copy_expr (p);
1388 gfc_free ((*newp)->value.character.string);
1390 end = (int) mpz_get_ui (p->ref->u.ss.end->value.integer);
1391 start = (int) mpz_get_ui (p->ref->u.ss.start->value.integer);
1392 length = end - start + 1;
1394 chr = (*newp)->value.character.string = gfc_getmem (length + 1);
1395 (*newp)->value.character.length = length;
1396 memcpy (chr, &p->value.character.string[start - 1], length);
1397 chr[length] = '\0';
1398 return SUCCESS;
1403 /* Simplify a subobject reference of a constructor. This occurs when
1404 parameter variable values are substituted. */
1406 static try
1407 simplify_const_ref (gfc_expr *p)
1409 gfc_constructor *cons;
1410 gfc_expr *newp;
1412 while (p->ref)
1414 switch (p->ref->type)
1416 case REF_ARRAY:
1417 switch (p->ref->u.ar.type)
1419 case AR_ELEMENT:
1420 if (find_array_element (p->value.constructor, &p->ref->u.ar,
1421 &cons) == FAILURE)
1422 return FAILURE;
1424 if (!cons)
1425 return SUCCESS;
1427 remove_subobject_ref (p, cons);
1428 break;
1430 case AR_SECTION:
1431 if (find_array_section (p, p->ref) == FAILURE)
1432 return FAILURE;
1433 p->ref->u.ar.type = AR_FULL;
1435 /* Fall through. */
1437 case AR_FULL:
1438 if (p->ref->next != NULL
1439 && (p->ts.type == BT_CHARACTER || p->ts.type == BT_DERIVED))
1441 cons = p->value.constructor;
1442 for (; cons; cons = cons->next)
1444 cons->expr->ref = copy_ref (p->ref->next);
1445 simplify_const_ref (cons->expr);
1448 gfc_free_ref_list (p->ref);
1449 p->ref = NULL;
1450 break;
1452 default:
1453 return SUCCESS;
1456 break;
1458 case REF_COMPONENT:
1459 cons = find_component_ref (p->value.constructor, p->ref);
1460 remove_subobject_ref (p, cons);
1461 break;
1463 case REF_SUBSTRING:
1464 if (find_substring_ref (p, &newp) == FAILURE)
1465 return FAILURE;
1467 gfc_replace_expr (p, newp);
1468 gfc_free_ref_list (p->ref);
1469 p->ref = NULL;
1470 break;
1474 return SUCCESS;
1478 /* Simplify a chain of references. */
1480 static try
1481 simplify_ref_chain (gfc_ref *ref, int type)
1483 int n;
1485 for (; ref; ref = ref->next)
1487 switch (ref->type)
1489 case REF_ARRAY:
1490 for (n = 0; n < ref->u.ar.dimen; n++)
1492 if (gfc_simplify_expr (ref->u.ar.start[n], type) == FAILURE)
1493 return FAILURE;
1494 if (gfc_simplify_expr (ref->u.ar.end[n], type) == FAILURE)
1495 return FAILURE;
1496 if (gfc_simplify_expr (ref->u.ar.stride[n], type) == FAILURE)
1497 return FAILURE;
1499 break;
1501 case REF_SUBSTRING:
1502 if (gfc_simplify_expr (ref->u.ss.start, type) == FAILURE)
1503 return FAILURE;
1504 if (gfc_simplify_expr (ref->u.ss.end, type) == FAILURE)
1505 return FAILURE;
1506 break;
1508 default:
1509 break;
1512 return SUCCESS;
1516 /* Try to substitute the value of a parameter variable. */
1518 static try
1519 simplify_parameter_variable (gfc_expr *p, int type)
1521 gfc_expr *e;
1522 try t;
1524 e = gfc_copy_expr (p->symtree->n.sym->value);
1525 if (e == NULL)
1526 return FAILURE;
1528 e->rank = p->rank;
1530 /* Do not copy subobject refs for constant. */
1531 if (e->expr_type != EXPR_CONSTANT && p->ref != NULL)
1532 e->ref = copy_ref (p->ref);
1533 t = gfc_simplify_expr (e, type);
1535 /* Only use the simplification if it eliminated all subobject references. */
1536 if (t == SUCCESS && !e->ref)
1537 gfc_replace_expr (p, e);
1538 else
1539 gfc_free_expr (e);
1541 return t;
1544 /* Given an expression, simplify it by collapsing constant
1545 expressions. Most simplification takes place when the expression
1546 tree is being constructed. If an intrinsic function is simplified
1547 at some point, we get called again to collapse the result against
1548 other constants.
1550 We work by recursively simplifying expression nodes, simplifying
1551 intrinsic functions where possible, which can lead to further
1552 constant collapsing. If an operator has constant operand(s), we
1553 rip the expression apart, and rebuild it, hoping that it becomes
1554 something simpler.
1556 The expression type is defined for:
1557 0 Basic expression parsing
1558 1 Simplifying array constructors -- will substitute
1559 iterator values.
1560 Returns FAILURE on error, SUCCESS otherwise.
1561 NOTE: Will return SUCCESS even if the expression can not be simplified. */
1564 gfc_simplify_expr (gfc_expr *p, int type)
1566 gfc_actual_arglist *ap;
1568 if (p == NULL)
1569 return SUCCESS;
1571 switch (p->expr_type)
1573 case EXPR_CONSTANT:
1574 case EXPR_NULL:
1575 break;
1577 case EXPR_FUNCTION:
1578 for (ap = p->value.function.actual; ap; ap = ap->next)
1579 if (gfc_simplify_expr (ap->expr, type) == FAILURE)
1580 return FAILURE;
1582 if (p->value.function.isym != NULL
1583 && gfc_intrinsic_func_interface (p, 1) == MATCH_ERROR)
1584 return FAILURE;
1586 break;
1588 case EXPR_SUBSTRING:
1589 if (simplify_ref_chain (p->ref, type) == FAILURE)
1590 return FAILURE;
1592 if (gfc_is_constant_expr (p))
1594 char *s;
1595 int start, end;
1597 if (p->ref && p->ref->u.ss.start)
1599 gfc_extract_int (p->ref->u.ss.start, &start);
1600 start--; /* Convert from one-based to zero-based. */
1602 else
1603 start = 0;
1605 if (p->ref && p->ref->u.ss.end)
1606 gfc_extract_int (p->ref->u.ss.end, &end);
1607 else
1608 end = p->value.character.length;
1610 s = gfc_getmem (end - start + 2);
1611 memcpy (s, p->value.character.string + start, end - start);
1612 s[end - start + 1] = '\0'; /* TODO: C-style string. */
1613 gfc_free (p->value.character.string);
1614 p->value.character.string = s;
1615 p->value.character.length = end - start;
1616 p->ts.cl = gfc_get_charlen ();
1617 p->ts.cl->next = gfc_current_ns->cl_list;
1618 gfc_current_ns->cl_list = p->ts.cl;
1619 p->ts.cl->length = gfc_int_expr (p->value.character.length);
1620 gfc_free_ref_list (p->ref);
1621 p->ref = NULL;
1622 p->expr_type = EXPR_CONSTANT;
1624 break;
1626 case EXPR_OP:
1627 if (simplify_intrinsic_op (p, type) == FAILURE)
1628 return FAILURE;
1629 break;
1631 case EXPR_VARIABLE:
1632 /* Only substitute array parameter variables if we are in an
1633 initialization expression, or we want a subsection. */
1634 if (p->symtree->n.sym->attr.flavor == FL_PARAMETER
1635 && (gfc_init_expr || p->ref
1636 || p->symtree->n.sym->value->expr_type != EXPR_ARRAY))
1638 if (simplify_parameter_variable (p, type) == FAILURE)
1639 return FAILURE;
1640 break;
1643 if (type == 1)
1645 gfc_simplify_iterator_var (p);
1648 /* Simplify subcomponent references. */
1649 if (simplify_ref_chain (p->ref, type) == FAILURE)
1650 return FAILURE;
1652 break;
1654 case EXPR_STRUCTURE:
1655 case EXPR_ARRAY:
1656 if (simplify_ref_chain (p->ref, type) == FAILURE)
1657 return FAILURE;
1659 if (simplify_constructor (p->value.constructor, type) == FAILURE)
1660 return FAILURE;
1662 if (p->expr_type == EXPR_ARRAY && p->ref && p->ref->type == REF_ARRAY
1663 && p->ref->u.ar.type == AR_FULL)
1664 gfc_expand_constructor (p);
1666 if (simplify_const_ref (p) == FAILURE)
1667 return FAILURE;
1669 break;
1672 return SUCCESS;
1676 /* Returns the type of an expression with the exception that iterator
1677 variables are automatically integers no matter what else they may
1678 be declared as. */
1680 static bt
1681 et0 (gfc_expr *e)
1683 if (e->expr_type == EXPR_VARIABLE && gfc_check_iter_variable (e) == SUCCESS)
1684 return BT_INTEGER;
1686 return e->ts.type;
1690 /* Check an intrinsic arithmetic operation to see if it is consistent
1691 with some type of expression. */
1693 static try check_init_expr (gfc_expr *);
1696 /* Scalarize an expression for an elemental intrinsic call. */
1698 static try
1699 scalarize_intrinsic_call (gfc_expr *e)
1701 gfc_actual_arglist *a, *b;
1702 gfc_constructor *args[5], *ctor, *new_ctor;
1703 gfc_expr *expr, *old;
1704 int n, i, rank[5];
1706 old = gfc_copy_expr (e);
1708 /* Assume that the old expression carries the type information and
1709 that the first arg carries all the shape information. */
1710 expr = gfc_copy_expr (old->value.function.actual->expr);
1711 gfc_free_constructor (expr->value.constructor);
1712 expr->value.constructor = NULL;
1714 expr->ts = old->ts;
1715 expr->expr_type = EXPR_ARRAY;
1717 /* Copy the array argument constructors into an array, with nulls
1718 for the scalars. */
1719 n = 0;
1720 a = old->value.function.actual;
1721 for (; a; a = a->next)
1723 /* Check that this is OK for an initialization expression. */
1724 if (a->expr && check_init_expr (a->expr) == FAILURE)
1725 goto cleanup;
1727 rank[n] = 0;
1728 if (a->expr && a->expr->rank && a->expr->expr_type == EXPR_VARIABLE)
1730 rank[n] = a->expr->rank;
1731 ctor = a->expr->symtree->n.sym->value->value.constructor;
1732 args[n] = gfc_copy_constructor (ctor);
1734 else if (a->expr && a->expr->expr_type == EXPR_ARRAY)
1736 if (a->expr->rank)
1737 rank[n] = a->expr->rank;
1738 else
1739 rank[n] = 1;
1740 args[n] = gfc_copy_constructor (a->expr->value.constructor);
1742 else
1743 args[n] = NULL;
1744 n++;
1747 for (i = 1; i < n; i++)
1748 if (rank[i] && rank[i] != rank[0])
1749 goto compliance;
1751 /* Using the first argument as the master, step through the array
1752 calling the function for each element and advancing the array
1753 constructors together. */
1754 ctor = args[0];
1755 new_ctor = NULL;
1756 for (; ctor; ctor = ctor->next)
1758 if (expr->value.constructor == NULL)
1759 expr->value.constructor
1760 = new_ctor = gfc_get_constructor ();
1761 else
1763 new_ctor->next = gfc_get_constructor ();
1764 new_ctor = new_ctor->next;
1766 new_ctor->expr = gfc_copy_expr (old);
1767 gfc_free_actual_arglist (new_ctor->expr->value.function.actual);
1768 a = NULL;
1769 b = old->value.function.actual;
1770 for (i = 0; i < n; i++)
1772 if (a == NULL)
1773 new_ctor->expr->value.function.actual
1774 = a = gfc_get_actual_arglist ();
1775 else
1777 a->next = gfc_get_actual_arglist ();
1778 a = a->next;
1780 if (args[i])
1781 a->expr = gfc_copy_expr (args[i]->expr);
1782 else
1783 a->expr = gfc_copy_expr (b->expr);
1785 b = b->next;
1788 /* Simplify the function calls. */
1789 if (gfc_simplify_expr (new_ctor->expr, 0) == FAILURE)
1790 goto cleanup;
1792 for (i = 0; i < n; i++)
1793 if (args[i])
1794 args[i] = args[i]->next;
1796 for (i = 1; i < n; i++)
1797 if (rank[i] && ((args[i] != NULL && args[0] == NULL)
1798 || (args[i] == NULL && args[0] != NULL)))
1799 goto compliance;
1802 free_expr0 (e);
1803 *e = *expr;
1804 gfc_free_expr (old);
1805 return SUCCESS;
1807 compliance:
1808 gfc_error_now ("elemental function arguments at %C are not compliant");
1810 cleanup:
1811 gfc_free_expr (expr);
1812 gfc_free_expr (old);
1813 return FAILURE;
1817 static try
1818 check_intrinsic_op (gfc_expr *e, try (*check_function) (gfc_expr *))
1820 gfc_expr *op1 = e->value.op.op1;
1821 gfc_expr *op2 = e->value.op.op2;
1823 if ((*check_function) (op1) == FAILURE)
1824 return FAILURE;
1826 switch (e->value.op.operator)
1828 case INTRINSIC_UPLUS:
1829 case INTRINSIC_UMINUS:
1830 if (!numeric_type (et0 (op1)))
1831 goto not_numeric;
1832 break;
1834 case INTRINSIC_EQ:
1835 case INTRINSIC_EQ_OS:
1836 case INTRINSIC_NE:
1837 case INTRINSIC_NE_OS:
1838 case INTRINSIC_GT:
1839 case INTRINSIC_GT_OS:
1840 case INTRINSIC_GE:
1841 case INTRINSIC_GE_OS:
1842 case INTRINSIC_LT:
1843 case INTRINSIC_LT_OS:
1844 case INTRINSIC_LE:
1845 case INTRINSIC_LE_OS:
1846 if ((*check_function) (op2) == FAILURE)
1847 return FAILURE;
1849 if (!(et0 (op1) == BT_CHARACTER && et0 (op2) == BT_CHARACTER)
1850 && !(numeric_type (et0 (op1)) && numeric_type (et0 (op2))))
1852 gfc_error ("Numeric or CHARACTER operands are required in "
1853 "expression at %L", &e->where);
1854 return FAILURE;
1856 break;
1858 case INTRINSIC_PLUS:
1859 case INTRINSIC_MINUS:
1860 case INTRINSIC_TIMES:
1861 case INTRINSIC_DIVIDE:
1862 case INTRINSIC_POWER:
1863 if ((*check_function) (op2) == FAILURE)
1864 return FAILURE;
1866 if (!numeric_type (et0 (op1)) || !numeric_type (et0 (op2)))
1867 goto not_numeric;
1869 if (e->value.op.operator == INTRINSIC_POWER
1870 && check_function == check_init_expr && et0 (op2) != BT_INTEGER)
1872 if (gfc_notify_std (GFC_STD_F2003,"Fortran 2003: Noninteger "
1873 "exponent in an initialization "
1874 "expression at %L", &op2->where)
1875 == FAILURE)
1876 return FAILURE;
1879 break;
1881 case INTRINSIC_CONCAT:
1882 if ((*check_function) (op2) == FAILURE)
1883 return FAILURE;
1885 if (et0 (op1) != BT_CHARACTER || et0 (op2) != BT_CHARACTER)
1887 gfc_error ("Concatenation operator in expression at %L "
1888 "must have two CHARACTER operands", &op1->where);
1889 return FAILURE;
1892 if (op1->ts.kind != op2->ts.kind)
1894 gfc_error ("Concat operator at %L must concatenate strings of the "
1895 "same kind", &e->where);
1896 return FAILURE;
1899 break;
1901 case INTRINSIC_NOT:
1902 if (et0 (op1) != BT_LOGICAL)
1904 gfc_error (".NOT. operator in expression at %L must have a LOGICAL "
1905 "operand", &op1->where);
1906 return FAILURE;
1909 break;
1911 case INTRINSIC_AND:
1912 case INTRINSIC_OR:
1913 case INTRINSIC_EQV:
1914 case INTRINSIC_NEQV:
1915 if ((*check_function) (op2) == FAILURE)
1916 return FAILURE;
1918 if (et0 (op1) != BT_LOGICAL || et0 (op2) != BT_LOGICAL)
1920 gfc_error ("LOGICAL operands are required in expression at %L",
1921 &e->where);
1922 return FAILURE;
1925 break;
1927 case INTRINSIC_PARENTHESES:
1928 break;
1930 default:
1931 gfc_error ("Only intrinsic operators can be used in expression at %L",
1932 &e->where);
1933 return FAILURE;
1936 return SUCCESS;
1938 not_numeric:
1939 gfc_error ("Numeric operands are required in expression at %L", &e->where);
1941 return FAILURE;
1945 static match
1946 check_init_expr_arguments (gfc_expr *e)
1948 gfc_actual_arglist *ap;
1950 for (ap = e->value.function.actual; ap; ap = ap->next)
1951 if (check_init_expr (ap->expr) == FAILURE)
1952 return MATCH_ERROR;
1954 return MATCH_YES;
1957 /* F95, 7.1.6.1, Initialization expressions, (7)
1958 F2003, 7.1.7 Initialization expression, (8) */
1960 static match
1961 check_inquiry (gfc_expr *e, int not_restricted)
1963 const char *name;
1964 const char *const *functions;
1966 static const char *const inquiry_func_f95[] = {
1967 "lbound", "shape", "size", "ubound",
1968 "bit_size", "len", "kind",
1969 "digits", "epsilon", "huge", "maxexponent", "minexponent",
1970 "precision", "radix", "range", "tiny",
1971 NULL
1974 static const char *const inquiry_func_f2003[] = {
1975 "lbound", "shape", "size", "ubound",
1976 "bit_size", "len", "kind",
1977 "digits", "epsilon", "huge", "maxexponent", "minexponent",
1978 "precision", "radix", "range", "tiny",
1979 "new_line", NULL
1982 int i;
1983 gfc_actual_arglist *ap;
1985 if (!e->value.function.isym
1986 || !e->value.function.isym->inquiry)
1987 return MATCH_NO;
1989 /* An undeclared parameter will get us here (PR25018). */
1990 if (e->symtree == NULL)
1991 return MATCH_NO;
1993 name = e->symtree->n.sym->name;
1995 functions = (gfc_option.warn_std & GFC_STD_F2003)
1996 ? inquiry_func_f2003 : inquiry_func_f95;
1998 for (i = 0; functions[i]; i++)
1999 if (strcmp (functions[i], name) == 0)
2000 break;
2002 if (functions[i] == NULL)
2003 return MATCH_ERROR;
2005 /* At this point we have an inquiry function with a variable argument. The
2006 type of the variable might be undefined, but we need it now, because the
2007 arguments of these functions are not allowed to be undefined. */
2009 for (ap = e->value.function.actual; ap; ap = ap->next)
2011 if (!ap->expr)
2012 continue;
2014 if (ap->expr->ts.type == BT_UNKNOWN)
2016 if (ap->expr->symtree->n.sym->ts.type == BT_UNKNOWN
2017 && gfc_set_default_type (ap->expr->symtree->n.sym, 0, gfc_current_ns)
2018 == FAILURE)
2019 return MATCH_NO;
2021 ap->expr->ts = ap->expr->symtree->n.sym->ts;
2024 /* Assumed character length will not reduce to a constant expression
2025 with LEN, as required by the standard. */
2026 if (i == 5 && not_restricted
2027 && ap->expr->symtree->n.sym->ts.type == BT_CHARACTER
2028 && ap->expr->symtree->n.sym->ts.cl->length == NULL)
2030 gfc_error ("Assumed character length variable '%s' in constant "
2031 "expression at %L", e->symtree->n.sym->name, &e->where);
2032 return MATCH_ERROR;
2034 else if (not_restricted && check_init_expr (ap->expr) == FAILURE)
2035 return MATCH_ERROR;
2038 return MATCH_YES;
2042 /* F95, 7.1.6.1, Initialization expressions, (5)
2043 F2003, 7.1.7 Initialization expression, (5) */
2045 static match
2046 check_transformational (gfc_expr *e)
2048 static const char * const trans_func_f95[] = {
2049 "repeat", "reshape", "selected_int_kind",
2050 "selected_real_kind", "transfer", "trim", NULL
2053 int i;
2054 const char *name;
2056 if (!e->value.function.isym
2057 || !e->value.function.isym->transformational)
2058 return MATCH_NO;
2060 name = e->symtree->n.sym->name;
2062 /* NULL() is dealt with below. */
2063 if (strcmp ("null", name) == 0)
2064 return MATCH_NO;
2066 for (i = 0; trans_func_f95[i]; i++)
2067 if (strcmp (trans_func_f95[i], name) == 0)
2068 break;
2070 /* FIXME, F2003: implement translation of initialization
2071 expressions before enabling this check. For F95, error
2072 out if the transformational function is not in the list. */
2073 #if 0
2074 if (trans_func_f95[i] == NULL
2075 && gfc_notify_std (GFC_STD_F2003,
2076 "transformational intrinsic '%s' at %L is not permitted "
2077 "in an initialization expression", name, &e->where) == FAILURE)
2078 return MATCH_ERROR;
2079 #else
2080 if (trans_func_f95[i] == NULL)
2082 gfc_error("transformational intrinsic '%s' at %L is not permitted "
2083 "in an initialization expression", name, &e->where);
2084 return MATCH_ERROR;
2086 #endif
2088 return check_init_expr_arguments (e);
2092 /* F95, 7.1.6.1, Initialization expressions, (6)
2093 F2003, 7.1.7 Initialization expression, (6) */
2095 static match
2096 check_null (gfc_expr *e)
2098 if (strcmp ("null", e->symtree->n.sym->name) != 0)
2099 return MATCH_NO;
2101 return check_init_expr_arguments (e);
2105 static match
2106 check_elemental (gfc_expr *e)
2108 if (!e->value.function.isym
2109 || !e->value.function.isym->elemental)
2110 return MATCH_NO;
2112 if (e->ts.type != BT_INTEGER
2113 && e->ts.type != BT_CHARACTER
2114 && gfc_notify_std (GFC_STD_F2003, "Extension: Evaluation of "
2115 "nonstandard initialization expression at %L",
2116 &e->where) == FAILURE)
2117 return MATCH_ERROR;
2119 return check_init_expr_arguments (e);
2123 static match
2124 check_conversion (gfc_expr *e)
2126 if (!e->value.function.isym
2127 || !e->value.function.isym->conversion)
2128 return MATCH_NO;
2130 return check_init_expr_arguments (e);
2134 /* Verify that an expression is an initialization expression. A side
2135 effect is that the expression tree is reduced to a single constant
2136 node if all goes well. This would normally happen when the
2137 expression is constructed but function references are assumed to be
2138 intrinsics in the context of initialization expressions. If
2139 FAILURE is returned an error message has been generated. */
2141 static try
2142 check_init_expr (gfc_expr *e)
2144 match m;
2145 try t;
2146 gfc_intrinsic_sym *isym;
2148 if (e == NULL)
2149 return SUCCESS;
2151 switch (e->expr_type)
2153 case EXPR_OP:
2154 t = check_intrinsic_op (e, check_init_expr);
2155 if (t == SUCCESS)
2156 t = gfc_simplify_expr (e, 0);
2158 break;
2160 case EXPR_FUNCTION:
2161 t = FAILURE;
2163 if ((m = check_specification_function (e)) != MATCH_YES)
2165 if ((m = gfc_intrinsic_func_interface (e, 0)) != MATCH_YES)
2167 gfc_error ("Function '%s' in initialization expression at %L "
2168 "must be an intrinsic or a specification function",
2169 e->symtree->n.sym->name, &e->where);
2170 break;
2173 if ((m = check_conversion (e)) == MATCH_NO
2174 && (m = check_inquiry (e, 1)) == MATCH_NO
2175 && (m = check_null (e)) == MATCH_NO
2176 && (m = check_transformational (e)) == MATCH_NO
2177 && (m = check_elemental (e)) == MATCH_NO)
2179 gfc_error ("Intrinsic function '%s' at %L is not permitted "
2180 "in an initialization expression",
2181 e->symtree->n.sym->name, &e->where);
2182 m = MATCH_ERROR;
2185 /* Try to scalarize an elemental intrinsic function that has an
2186 array argument. */
2187 isym = gfc_find_function (e->symtree->n.sym->name);
2188 if (isym && isym->elemental
2189 && e->value.function.actual->expr->expr_type == EXPR_ARRAY)
2191 if ((t = scalarize_intrinsic_call (e)) == SUCCESS)
2192 break;
2196 if (m == MATCH_YES)
2197 t = gfc_simplify_expr (e, 0);
2199 break;
2201 case EXPR_VARIABLE:
2202 t = SUCCESS;
2204 if (gfc_check_iter_variable (e) == SUCCESS)
2205 break;
2207 if (e->symtree->n.sym->attr.flavor == FL_PARAMETER)
2209 /* A PARAMETER shall not be used to define itself, i.e.
2210 REAL, PARAMETER :: x = transfer(0, x)
2211 is invalid. */
2212 if (!e->symtree->n.sym->value)
2214 gfc_error("PARAMETER '%s' is used at %L before its definition "
2215 "is complete", e->symtree->n.sym->name, &e->where);
2216 t = FAILURE;
2218 else
2219 t = simplify_parameter_variable (e, 0);
2221 break;
2224 if (gfc_in_match_data ())
2225 break;
2227 t = FAILURE;
2229 if (e->symtree->n.sym->as)
2231 switch (e->symtree->n.sym->as->type)
2233 case AS_ASSUMED_SIZE:
2234 gfc_error ("Assumed size array '%s' at %L is not permitted "
2235 "in an initialization expression",
2236 e->symtree->n.sym->name, &e->where);
2237 break;
2239 case AS_ASSUMED_SHAPE:
2240 gfc_error ("Assumed shape array '%s' at %L is not permitted "
2241 "in an initialization expression",
2242 e->symtree->n.sym->name, &e->where);
2243 break;
2245 case AS_DEFERRED:
2246 gfc_error ("Deferred array '%s' at %L is not permitted "
2247 "in an initialization expression",
2248 e->symtree->n.sym->name, &e->where);
2249 break;
2251 case AS_EXPLICIT:
2252 gfc_error ("Array '%s' at %L is a variable, which does "
2253 "not reduce to a constant expression",
2254 e->symtree->n.sym->name, &e->where);
2255 break;
2257 default:
2258 gcc_unreachable();
2261 else
2262 gfc_error ("Parameter '%s' at %L has not been declared or is "
2263 "a variable, which does not reduce to a constant "
2264 "expression", e->symtree->n.sym->name, &e->where);
2266 break;
2268 case EXPR_CONSTANT:
2269 case EXPR_NULL:
2270 t = SUCCESS;
2271 break;
2273 case EXPR_SUBSTRING:
2274 t = check_init_expr (e->ref->u.ss.start);
2275 if (t == FAILURE)
2276 break;
2278 t = check_init_expr (e->ref->u.ss.end);
2279 if (t == SUCCESS)
2280 t = gfc_simplify_expr (e, 0);
2282 break;
2284 case EXPR_STRUCTURE:
2285 if (e->ts.is_iso_c)
2286 t = SUCCESS;
2287 else
2288 t = gfc_check_constructor (e, check_init_expr);
2289 break;
2291 case EXPR_ARRAY:
2292 t = gfc_check_constructor (e, check_init_expr);
2293 if (t == FAILURE)
2294 break;
2296 t = gfc_expand_constructor (e);
2297 if (t == FAILURE)
2298 break;
2300 t = gfc_check_constructor_type (e);
2301 break;
2303 default:
2304 gfc_internal_error ("check_init_expr(): Unknown expression type");
2307 return t;
2311 /* Match an initialization expression. We work by first matching an
2312 expression, then reducing it to a constant. */
2314 match
2315 gfc_match_init_expr (gfc_expr **result)
2317 gfc_expr *expr;
2318 match m;
2319 try t;
2321 m = gfc_match_expr (&expr);
2322 if (m != MATCH_YES)
2323 return m;
2325 gfc_init_expr = 1;
2326 t = gfc_resolve_expr (expr);
2327 if (t == SUCCESS)
2328 t = check_init_expr (expr);
2329 gfc_init_expr = 0;
2331 if (t == FAILURE)
2333 gfc_free_expr (expr);
2334 return MATCH_ERROR;
2337 if (expr->expr_type == EXPR_ARRAY
2338 && (gfc_check_constructor_type (expr) == FAILURE
2339 || gfc_expand_constructor (expr) == FAILURE))
2341 gfc_free_expr (expr);
2342 return MATCH_ERROR;
2345 /* Not all inquiry functions are simplified to constant expressions
2346 so it is necessary to call check_inquiry again. */
2347 if (!gfc_is_constant_expr (expr) && check_inquiry (expr, 1) != MATCH_YES
2348 && !gfc_in_match_data ())
2350 gfc_error ("Initialization expression didn't reduce %C");
2351 return MATCH_ERROR;
2354 *result = expr;
2356 return MATCH_YES;
2360 static try check_restricted (gfc_expr *);
2362 /* Given an actual argument list, test to see that each argument is a
2363 restricted expression and optionally if the expression type is
2364 integer or character. */
2366 static try
2367 restricted_args (gfc_actual_arglist *a)
2369 for (; a; a = a->next)
2371 if (check_restricted (a->expr) == FAILURE)
2372 return FAILURE;
2375 return SUCCESS;
2379 /************* Restricted/specification expressions *************/
2382 /* Make sure a non-intrinsic function is a specification function. */
2384 static try
2385 external_spec_function (gfc_expr *e)
2387 gfc_symbol *f;
2389 f = e->value.function.esym;
2391 if (f->attr.proc == PROC_ST_FUNCTION)
2393 gfc_error ("Specification function '%s' at %L cannot be a statement "
2394 "function", f->name, &e->where);
2395 return FAILURE;
2398 if (f->attr.proc == PROC_INTERNAL)
2400 gfc_error ("Specification function '%s' at %L cannot be an internal "
2401 "function", f->name, &e->where);
2402 return FAILURE;
2405 if (!f->attr.pure && !f->attr.elemental)
2407 gfc_error ("Specification function '%s' at %L must be PURE", f->name,
2408 &e->where);
2409 return FAILURE;
2412 if (f->attr.recursive)
2414 gfc_error ("Specification function '%s' at %L cannot be RECURSIVE",
2415 f->name, &e->where);
2416 return FAILURE;
2419 return restricted_args (e->value.function.actual);
2423 /* Check to see that a function reference to an intrinsic is a
2424 restricted expression. */
2426 static try
2427 restricted_intrinsic (gfc_expr *e)
2429 /* TODO: Check constraints on inquiry functions. 7.1.6.2 (7). */
2430 if (check_inquiry (e, 0) == MATCH_YES)
2431 return SUCCESS;
2433 return restricted_args (e->value.function.actual);
2437 /* Verify that an expression is a restricted expression. Like its
2438 cousin check_init_expr(), an error message is generated if we
2439 return FAILURE. */
2441 static try
2442 check_restricted (gfc_expr *e)
2444 gfc_symbol *sym;
2445 try t;
2447 if (e == NULL)
2448 return SUCCESS;
2450 switch (e->expr_type)
2452 case EXPR_OP:
2453 t = check_intrinsic_op (e, check_restricted);
2454 if (t == SUCCESS)
2455 t = gfc_simplify_expr (e, 0);
2457 break;
2459 case EXPR_FUNCTION:
2460 t = e->value.function.esym ? external_spec_function (e)
2461 : restricted_intrinsic (e);
2462 break;
2464 case EXPR_VARIABLE:
2465 sym = e->symtree->n.sym;
2466 t = FAILURE;
2468 /* If a dummy argument appears in a context that is valid for a
2469 restricted expression in an elemental procedure, it will have
2470 already been simplified away once we get here. Therefore we
2471 don't need to jump through hoops to distinguish valid from
2472 invalid cases. */
2473 if (sym->attr.dummy && sym->ns == gfc_current_ns
2474 && sym->ns->proc_name && sym->ns->proc_name->attr.elemental)
2476 gfc_error ("Dummy argument '%s' not allowed in expression at %L",
2477 sym->name, &e->where);
2478 break;
2481 if (sym->attr.optional)
2483 gfc_error ("Dummy argument '%s' at %L cannot be OPTIONAL",
2484 sym->name, &e->where);
2485 break;
2488 if (sym->attr.intent == INTENT_OUT)
2490 gfc_error ("Dummy argument '%s' at %L cannot be INTENT(OUT)",
2491 sym->name, &e->where);
2492 break;
2495 /* gfc_is_formal_arg broadcasts that a formal argument list is being
2496 processed in resolve.c(resolve_formal_arglist). This is done so
2497 that host associated dummy array indices are accepted (PR23446).
2498 This mechanism also does the same for the specification expressions
2499 of array-valued functions. */
2500 if (sym->attr.in_common
2501 || sym->attr.use_assoc
2502 || sym->attr.dummy
2503 || sym->attr.implied_index
2504 || sym->ns != gfc_current_ns
2505 || (sym->ns->proc_name != NULL
2506 && sym->ns->proc_name->attr.flavor == FL_MODULE)
2507 || (gfc_is_formal_arg () && (sym->ns == gfc_current_ns)))
2509 t = SUCCESS;
2510 break;
2513 gfc_error ("Variable '%s' cannot appear in the expression at %L",
2514 sym->name, &e->where);
2516 break;
2518 case EXPR_NULL:
2519 case EXPR_CONSTANT:
2520 t = SUCCESS;
2521 break;
2523 case EXPR_SUBSTRING:
2524 t = gfc_specification_expr (e->ref->u.ss.start);
2525 if (t == FAILURE)
2526 break;
2528 t = gfc_specification_expr (e->ref->u.ss.end);
2529 if (t == SUCCESS)
2530 t = gfc_simplify_expr (e, 0);
2532 break;
2534 case EXPR_STRUCTURE:
2535 t = gfc_check_constructor (e, check_restricted);
2536 break;
2538 case EXPR_ARRAY:
2539 t = gfc_check_constructor (e, check_restricted);
2540 break;
2542 default:
2543 gfc_internal_error ("check_restricted(): Unknown expression type");
2546 return t;
2550 /* Check to see that an expression is a specification expression. If
2551 we return FAILURE, an error has been generated. */
2554 gfc_specification_expr (gfc_expr *e)
2557 if (e == NULL)
2558 return SUCCESS;
2560 if (e->ts.type != BT_INTEGER)
2562 gfc_error ("Expression at %L must be of INTEGER type", &e->where);
2563 return FAILURE;
2566 if (e->expr_type == EXPR_FUNCTION
2567 && !e->value.function.isym
2568 && !e->value.function.esym
2569 && !gfc_pure (e->symtree->n.sym))
2571 gfc_error ("Function '%s' at %L must be PURE",
2572 e->symtree->n.sym->name, &e->where);
2573 /* Prevent repeat error messages. */
2574 e->symtree->n.sym->attr.pure = 1;
2575 return FAILURE;
2578 if (e->rank != 0)
2580 gfc_error ("Expression at %L must be scalar", &e->where);
2581 return FAILURE;
2584 if (gfc_simplify_expr (e, 0) == FAILURE)
2585 return FAILURE;
2587 return check_restricted (e);
2591 /************** Expression conformance checks. *************/
2593 /* Given two expressions, make sure that the arrays are conformable. */
2596 gfc_check_conformance (const char *optype_msgid, gfc_expr *op1, gfc_expr *op2)
2598 int op1_flag, op2_flag, d;
2599 mpz_t op1_size, op2_size;
2600 try t;
2602 if (op1->rank == 0 || op2->rank == 0)
2603 return SUCCESS;
2605 if (op1->rank != op2->rank)
2607 gfc_error ("Incompatible ranks in %s (%d and %d) at %L", _(optype_msgid),
2608 op1->rank, op2->rank, &op1->where);
2609 return FAILURE;
2612 t = SUCCESS;
2614 for (d = 0; d < op1->rank; d++)
2616 op1_flag = gfc_array_dimen_size (op1, d, &op1_size) == SUCCESS;
2617 op2_flag = gfc_array_dimen_size (op2, d, &op2_size) == SUCCESS;
2619 if (op1_flag && op2_flag && mpz_cmp (op1_size, op2_size) != 0)
2621 gfc_error ("Different shape for %s at %L on dimension %d "
2622 "(%d and %d)", _(optype_msgid), &op1->where, d + 1,
2623 (int) mpz_get_si (op1_size),
2624 (int) mpz_get_si (op2_size));
2626 t = FAILURE;
2629 if (op1_flag)
2630 mpz_clear (op1_size);
2631 if (op2_flag)
2632 mpz_clear (op2_size);
2634 if (t == FAILURE)
2635 return FAILURE;
2638 return SUCCESS;
2642 /* Given an assignable expression and an arbitrary expression, make
2643 sure that the assignment can take place. */
2646 gfc_check_assign (gfc_expr *lvalue, gfc_expr *rvalue, int conform)
2648 gfc_symbol *sym;
2649 gfc_ref *ref;
2650 int has_pointer;
2652 sym = lvalue->symtree->n.sym;
2654 /* Check INTENT(IN), unless the object itself is the component or
2655 sub-component of a pointer. */
2656 has_pointer = sym->attr.pointer;
2658 for (ref = lvalue->ref; ref; ref = ref->next)
2659 if (ref->type == REF_COMPONENT && ref->u.c.component->pointer)
2661 has_pointer = 1;
2662 break;
2665 if (!has_pointer && sym->attr.intent == INTENT_IN)
2667 gfc_error ("Cannot assign to INTENT(IN) variable '%s' at %L",
2668 sym->name, &lvalue->where);
2669 return FAILURE;
2672 /* 12.5.2.2, Note 12.26: The result variable is very similar to any other
2673 variable local to a function subprogram. Its existence begins when
2674 execution of the function is initiated and ends when execution of the
2675 function is terminated...
2676 Therefore, the left hand side is no longer a variable, when it is: */
2677 if (sym->attr.flavor == FL_PROCEDURE && sym->attr.proc != PROC_ST_FUNCTION
2678 && !sym->attr.external)
2680 bool bad_proc;
2681 bad_proc = false;
2683 /* (i) Use associated; */
2684 if (sym->attr.use_assoc)
2685 bad_proc = true;
2687 /* (ii) The assignment is in the main program; or */
2688 if (gfc_current_ns->proc_name->attr.is_main_program)
2689 bad_proc = true;
2691 /* (iii) A module or internal procedure... */
2692 if ((gfc_current_ns->proc_name->attr.proc == PROC_INTERNAL
2693 || gfc_current_ns->proc_name->attr.proc == PROC_MODULE)
2694 && gfc_current_ns->parent
2695 && (!(gfc_current_ns->parent->proc_name->attr.function
2696 || gfc_current_ns->parent->proc_name->attr.subroutine)
2697 || gfc_current_ns->parent->proc_name->attr.is_main_program))
2699 /* ... that is not a function... */
2700 if (!gfc_current_ns->proc_name->attr.function)
2701 bad_proc = true;
2703 /* ... or is not an entry and has a different name. */
2704 if (!sym->attr.entry && sym->name != gfc_current_ns->proc_name->name)
2705 bad_proc = true;
2708 if (bad_proc)
2710 gfc_error ("'%s' at %L is not a VALUE", sym->name, &lvalue->where);
2711 return FAILURE;
2715 if (rvalue->rank != 0 && lvalue->rank != rvalue->rank)
2717 gfc_error ("Incompatible ranks %d and %d in assignment at %L",
2718 lvalue->rank, rvalue->rank, &lvalue->where);
2719 return FAILURE;
2722 if (lvalue->ts.type == BT_UNKNOWN)
2724 gfc_error ("Variable type is UNKNOWN in assignment at %L",
2725 &lvalue->where);
2726 return FAILURE;
2729 if (rvalue->expr_type == EXPR_NULL)
2731 if (lvalue->symtree->n.sym->attr.pointer
2732 && lvalue->symtree->n.sym->attr.data)
2733 return SUCCESS;
2734 else
2736 gfc_error ("NULL appears on right-hand side in assignment at %L",
2737 &rvalue->where);
2738 return FAILURE;
2742 if (sym->attr.cray_pointee
2743 && lvalue->ref != NULL
2744 && lvalue->ref->u.ar.type == AR_FULL
2745 && lvalue->ref->u.ar.as->cp_was_assumed)
2747 gfc_error ("Vector assignment to assumed-size Cray Pointee at %L "
2748 "is illegal", &lvalue->where);
2749 return FAILURE;
2752 /* This is possibly a typo: x = f() instead of x => f(). */
2753 if (gfc_option.warn_surprising
2754 && rvalue->expr_type == EXPR_FUNCTION
2755 && rvalue->symtree->n.sym->attr.pointer)
2756 gfc_warning ("POINTER valued function appears on right-hand side of "
2757 "assignment at %L", &rvalue->where);
2759 /* Check size of array assignments. */
2760 if (lvalue->rank != 0 && rvalue->rank != 0
2761 && gfc_check_conformance ("array assignment", lvalue, rvalue) != SUCCESS)
2762 return FAILURE;
2764 if (rvalue->is_boz && lvalue->ts.type != BT_INTEGER
2765 && lvalue->symtree->n.sym->attr.data
2766 && gfc_notify_std (GFC_STD_GNU, "Extension: BOZ literal at %L used to "
2767 "initialize non-integer variable '%s'",
2768 &rvalue->where, lvalue->symtree->n.sym->name)
2769 == FAILURE)
2770 return FAILURE;
2771 else if (rvalue->is_boz && !lvalue->symtree->n.sym->attr.data
2772 && gfc_notify_std (GFC_STD_GNU, "Extension: BOZ literal at %L outside "
2773 "a DATA statement and outside INT/REAL/DBLE/CMPLX",
2774 &rvalue->where) == FAILURE)
2775 return FAILURE;
2777 /* Handle the case of a BOZ literal on the RHS. */
2778 if (rvalue->is_boz && lvalue->ts.type != BT_INTEGER)
2780 int rc;
2781 if (gfc_option.warn_surprising)
2782 gfc_warning ("BOZ literal at %L is bitwise transferred "
2783 "non-integer symbol '%s'", &rvalue->where,
2784 lvalue->symtree->n.sym->name);
2785 if (!gfc_convert_boz (rvalue, &lvalue->ts))
2786 return FAILURE;
2787 if ((rc = gfc_range_check (rvalue)) != ARITH_OK)
2789 if (rc == ARITH_UNDERFLOW)
2790 gfc_error ("Arithmetic underflow of bit-wise transferred BOZ at %L"
2791 ". This check can be disabled with the option "
2792 "-fno-range-check", &rvalue->where);
2793 else if (rc == ARITH_OVERFLOW)
2794 gfc_error ("Arithmetic overflow of bit-wise transferred BOZ at %L"
2795 ". This check can be disabled with the option "
2796 "-fno-range-check", &rvalue->where);
2797 else if (rc == ARITH_NAN)
2798 gfc_error ("Arithmetic NaN of bit-wise transferred BOZ at %L"
2799 ". This check can be disabled with the option "
2800 "-fno-range-check", &rvalue->where);
2801 return FAILURE;
2805 if (gfc_compare_types (&lvalue->ts, &rvalue->ts))
2806 return SUCCESS;
2808 if (!conform)
2810 /* Numeric can be converted to any other numeric. And Hollerith can be
2811 converted to any other type. */
2812 if ((gfc_numeric_ts (&lvalue->ts) && gfc_numeric_ts (&rvalue->ts))
2813 || rvalue->ts.type == BT_HOLLERITH)
2814 return SUCCESS;
2816 if (lvalue->ts.type == BT_LOGICAL && rvalue->ts.type == BT_LOGICAL)
2817 return SUCCESS;
2819 gfc_error ("Incompatible types in assignment at %L, %s to %s",
2820 &rvalue->where, gfc_typename (&rvalue->ts),
2821 gfc_typename (&lvalue->ts));
2823 return FAILURE;
2826 return gfc_convert_type (rvalue, &lvalue->ts, 1);
2830 /* Check that a pointer assignment is OK. We first check lvalue, and
2831 we only check rvalue if it's not an assignment to NULL() or a
2832 NULLIFY statement. */
2835 gfc_check_pointer_assign (gfc_expr *lvalue, gfc_expr *rvalue)
2837 symbol_attribute attr;
2838 gfc_ref *ref;
2839 int is_pure;
2840 int pointer, check_intent_in;
2842 if (lvalue->symtree->n.sym->ts.type == BT_UNKNOWN)
2844 gfc_error ("Pointer assignment target is not a POINTER at %L",
2845 &lvalue->where);
2846 return FAILURE;
2849 if (lvalue->symtree->n.sym->attr.flavor == FL_PROCEDURE
2850 && lvalue->symtree->n.sym->attr.use_assoc)
2852 gfc_error ("'%s' in the pointer assignment at %L cannot be an "
2853 "l-value since it is a procedure",
2854 lvalue->symtree->n.sym->name, &lvalue->where);
2855 return FAILURE;
2859 /* Check INTENT(IN), unless the object itself is the component or
2860 sub-component of a pointer. */
2861 check_intent_in = 1;
2862 pointer = lvalue->symtree->n.sym->attr.pointer;
2864 for (ref = lvalue->ref; ref; ref = ref->next)
2866 if (pointer)
2867 check_intent_in = 0;
2869 if (ref->type == REF_COMPONENT && ref->u.c.component->pointer)
2870 pointer = 1;
2873 if (check_intent_in && lvalue->symtree->n.sym->attr.intent == INTENT_IN)
2875 gfc_error ("Cannot assign to INTENT(IN) variable '%s' at %L",
2876 lvalue->symtree->n.sym->name, &lvalue->where);
2877 return FAILURE;
2880 if (!pointer)
2882 gfc_error ("Pointer assignment to non-POINTER at %L", &lvalue->where);
2883 return FAILURE;
2886 is_pure = gfc_pure (NULL);
2888 if (is_pure && gfc_impure_variable (lvalue->symtree->n.sym)
2889 && lvalue->symtree->n.sym->value != rvalue)
2891 gfc_error ("Bad pointer object in PURE procedure at %L", &lvalue->where);
2892 return FAILURE;
2895 /* If rvalue is a NULL() or NULLIFY, we're done. Otherwise the type,
2896 kind, etc for lvalue and rvalue must match, and rvalue must be a
2897 pure variable if we're in a pure function. */
2898 if (rvalue->expr_type == EXPR_NULL && rvalue->ts.type == BT_UNKNOWN)
2899 return SUCCESS;
2901 if (!gfc_compare_types (&lvalue->ts, &rvalue->ts))
2903 gfc_error ("Different types in pointer assignment at %L",
2904 &lvalue->where);
2905 return FAILURE;
2908 if (lvalue->ts.kind != rvalue->ts.kind)
2910 gfc_error ("Different kind type parameters in pointer "
2911 "assignment at %L", &lvalue->where);
2912 return FAILURE;
2915 if (lvalue->rank != rvalue->rank)
2917 gfc_error ("Different ranks in pointer assignment at %L",
2918 &lvalue->where);
2919 return FAILURE;
2922 /* Now punt if we are dealing with a NULLIFY(X) or X = NULL(X). */
2923 if (rvalue->expr_type == EXPR_NULL)
2924 return SUCCESS;
2926 if (lvalue->ts.type == BT_CHARACTER
2927 && lvalue->ts.cl && rvalue->ts.cl
2928 && lvalue->ts.cl->length && rvalue->ts.cl->length
2929 && abs (gfc_dep_compare_expr (lvalue->ts.cl->length,
2930 rvalue->ts.cl->length)) == 1)
2932 gfc_error ("Different character lengths in pointer "
2933 "assignment at %L", &lvalue->where);
2934 return FAILURE;
2937 if (rvalue->expr_type == EXPR_VARIABLE && is_subref_array (rvalue))
2938 lvalue->symtree->n.sym->attr.subref_array_pointer = 1;
2940 attr = gfc_expr_attr (rvalue);
2941 if (!attr.target && !attr.pointer)
2943 gfc_error ("Pointer assignment target is neither TARGET "
2944 "nor POINTER at %L", &rvalue->where);
2945 return FAILURE;
2948 if (is_pure && gfc_impure_variable (rvalue->symtree->n.sym))
2950 gfc_error ("Bad target in pointer assignment in PURE "
2951 "procedure at %L", &rvalue->where);
2954 if (gfc_has_vector_index (rvalue))
2956 gfc_error ("Pointer assignment with vector subscript "
2957 "on rhs at %L", &rvalue->where);
2958 return FAILURE;
2961 if (attr.protected && attr.use_assoc)
2963 gfc_error ("Pointer assigment target has PROTECTED "
2964 "attribute at %L", &rvalue->where);
2965 return FAILURE;
2968 return SUCCESS;
2972 /* Relative of gfc_check_assign() except that the lvalue is a single
2973 symbol. Used for initialization assignments. */
2976 gfc_check_assign_symbol (gfc_symbol *sym, gfc_expr *rvalue)
2978 gfc_expr lvalue;
2979 try r;
2981 memset (&lvalue, '\0', sizeof (gfc_expr));
2983 lvalue.expr_type = EXPR_VARIABLE;
2984 lvalue.ts = sym->ts;
2985 if (sym->as)
2986 lvalue.rank = sym->as->rank;
2987 lvalue.symtree = (gfc_symtree *) gfc_getmem (sizeof (gfc_symtree));
2988 lvalue.symtree->n.sym = sym;
2989 lvalue.where = sym->declared_at;
2991 if (sym->attr.pointer)
2992 r = gfc_check_pointer_assign (&lvalue, rvalue);
2993 else
2994 r = gfc_check_assign (&lvalue, rvalue, 1);
2996 gfc_free (lvalue.symtree);
2998 return r;
3002 /* Get an expression for a default initializer. */
3004 gfc_expr *
3005 gfc_default_initializer (gfc_typespec *ts)
3007 gfc_constructor *tail;
3008 gfc_expr *init;
3009 gfc_component *c;
3011 /* See if we have a default initializer. */
3012 for (c = ts->derived->components; c; c = c->next)
3013 if (c->initializer || c->allocatable)
3014 break;
3016 if (!c)
3017 return NULL;
3019 /* Build the constructor. */
3020 init = gfc_get_expr ();
3021 init->expr_type = EXPR_STRUCTURE;
3022 init->ts = *ts;
3023 init->where = ts->derived->declared_at;
3025 tail = NULL;
3026 for (c = ts->derived->components; c; c = c->next)
3028 if (tail == NULL)
3029 init->value.constructor = tail = gfc_get_constructor ();
3030 else
3032 tail->next = gfc_get_constructor ();
3033 tail = tail->next;
3036 if (c->initializer)
3037 tail->expr = gfc_copy_expr (c->initializer);
3039 if (c->allocatable)
3041 tail->expr = gfc_get_expr ();
3042 tail->expr->expr_type = EXPR_NULL;
3043 tail->expr->ts = c->ts;
3046 return init;
3050 /* Given a symbol, create an expression node with that symbol as a
3051 variable. If the symbol is array valued, setup a reference of the
3052 whole array. */
3054 gfc_expr *
3055 gfc_get_variable_expr (gfc_symtree *var)
3057 gfc_expr *e;
3059 e = gfc_get_expr ();
3060 e->expr_type = EXPR_VARIABLE;
3061 e->symtree = var;
3062 e->ts = var->n.sym->ts;
3064 if (var->n.sym->as != NULL)
3066 e->rank = var->n.sym->as->rank;
3067 e->ref = gfc_get_ref ();
3068 e->ref->type = REF_ARRAY;
3069 e->ref->u.ar.type = AR_FULL;
3072 return e;
3076 /* General expression traversal function. */
3078 bool
3079 gfc_traverse_expr (gfc_expr *expr, gfc_symbol *sym,
3080 bool (*func)(gfc_expr *, gfc_symbol *, int*),
3081 int f)
3083 gfc_array_ref ar;
3084 gfc_ref *ref;
3085 gfc_actual_arglist *args;
3086 gfc_constructor *c;
3087 int i;
3089 if (!expr)
3090 return false;
3092 if ((*func) (expr, sym, &f))
3093 return true;
3095 if (expr->ts.type == BT_CHARACTER
3096 && expr->ts.cl
3097 && expr->ts.cl->length
3098 && expr->ts.cl->length->expr_type != EXPR_CONSTANT
3099 && gfc_traverse_expr (expr->ts.cl->length, sym, func, f))
3100 return true;
3102 switch (expr->expr_type)
3104 case EXPR_FUNCTION:
3105 for (args = expr->value.function.actual; args; args = args->next)
3107 if (gfc_traverse_expr (args->expr, sym, func, f))
3108 return true;
3110 break;
3112 case EXPR_VARIABLE:
3113 case EXPR_CONSTANT:
3114 case EXPR_NULL:
3115 case EXPR_SUBSTRING:
3116 break;
3118 case EXPR_STRUCTURE:
3119 case EXPR_ARRAY:
3120 for (c = expr->value.constructor; c; c = c->next)
3122 if (gfc_traverse_expr (c->expr, sym, func, f))
3123 return true;
3124 if (c->iterator)
3126 if (gfc_traverse_expr (c->iterator->var, sym, func, f))
3127 return true;
3128 if (gfc_traverse_expr (c->iterator->start, sym, func, f))
3129 return true;
3130 if (gfc_traverse_expr (c->iterator->end, sym, func, f))
3131 return true;
3132 if (gfc_traverse_expr (c->iterator->step, sym, func, f))
3133 return true;
3136 break;
3138 case EXPR_OP:
3139 if (gfc_traverse_expr (expr->value.op.op1, sym, func, f))
3140 return true;
3141 if (gfc_traverse_expr (expr->value.op.op2, sym, func, f))
3142 return true;
3143 break;
3145 default:
3146 gcc_unreachable ();
3147 break;
3150 ref = expr->ref;
3151 while (ref != NULL)
3153 switch (ref->type)
3155 case REF_ARRAY:
3156 ar = ref->u.ar;
3157 for (i = 0; i < GFC_MAX_DIMENSIONS; i++)
3159 if (gfc_traverse_expr (ar.start[i], sym, func, f))
3160 return true;
3161 if (gfc_traverse_expr (ar.end[i], sym, func, f))
3162 return true;
3163 if (gfc_traverse_expr (ar.stride[i], sym, func, f))
3164 return true;
3166 break;
3168 case REF_SUBSTRING:
3169 if (gfc_traverse_expr (ref->u.ss.start, sym, func, f))
3170 return true;
3171 if (gfc_traverse_expr (ref->u.ss.end, sym, func, f))
3172 return true;
3173 break;
3175 case REF_COMPONENT:
3176 if (ref->u.c.component->ts.type == BT_CHARACTER
3177 && ref->u.c.component->ts.cl
3178 && ref->u.c.component->ts.cl->length
3179 && ref->u.c.component->ts.cl->length->expr_type
3180 != EXPR_CONSTANT
3181 && gfc_traverse_expr (ref->u.c.component->ts.cl->length,
3182 sym, func, f))
3183 return true;
3185 if (ref->u.c.component->as)
3186 for (i = 0; i < ref->u.c.component->as->rank; i++)
3188 if (gfc_traverse_expr (ref->u.c.component->as->lower[i],
3189 sym, func, f))
3190 return true;
3191 if (gfc_traverse_expr (ref->u.c.component->as->upper[i],
3192 sym, func, f))
3193 return true;
3195 break;
3197 default:
3198 gcc_unreachable ();
3200 ref = ref->next;
3202 return false;
3205 /* Traverse expr, marking all EXPR_VARIABLE symbols referenced. */
3207 static bool
3208 expr_set_symbols_referenced (gfc_expr *expr,
3209 gfc_symbol *sym ATTRIBUTE_UNUSED,
3210 int *f ATTRIBUTE_UNUSED)
3212 if (expr->expr_type != EXPR_VARIABLE)
3213 return false;
3214 gfc_set_sym_referenced (expr->symtree->n.sym);
3215 return false;
3218 void
3219 gfc_expr_set_symbols_referenced (gfc_expr *expr)
3221 gfc_traverse_expr (expr, NULL, expr_set_symbols_referenced, 0);