PR rtl-optimization/43520
[official-gcc.git] / gcc / fortran / expr.c
blob92454f6536f0999bcdbb0aba41b215accec4bcc6
1 /* Routines for manipulation of expression nodes.
2 Copyright (C) 2000, 2001, 2002, 2003, 2004, 2005, 2006, 2007, 2008,
3 2009, 2010
4 Free Software Foundation, Inc.
5 Contributed by Andy Vaught
7 This file is part of GCC.
9 GCC is free software; you can redistribute it and/or modify it under
10 the terms of the GNU General Public License as published by the Free
11 Software Foundation; either version 3, or (at your option) any later
12 version.
14 GCC is distributed in the hope that it will be useful, but WITHOUT ANY
15 WARRANTY; without even the implied warranty of MERCHANTABILITY or
16 FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
17 for more details.
19 You should have received a copy of the GNU General Public License
20 along with GCC; see the file COPYING3. If not see
21 <http://www.gnu.org/licenses/>. */
23 #include "config.h"
24 #include "system.h"
25 #include "gfortran.h"
26 #include "arith.h"
27 #include "match.h"
28 #include "target-memory.h" /* for gfc_convert_boz */
29 #include "constructor.h"
32 /* The following set of functions provide access to gfc_expr* of
33 various types - actual all but EXPR_FUNCTION and EXPR_VARIABLE.
35 There are two functions available elsewhere that provide
36 slightly different flavours of variables. Namely:
37 expr.c (gfc_get_variable_expr)
38 symbol.c (gfc_lval_expr_from_sym)
39 TODO: Merge these functions, if possible. */
41 /* Get a new expression node. */
43 gfc_expr *
44 gfc_get_expr (void)
46 gfc_expr *e;
48 e = XCNEW (gfc_expr);
49 gfc_clear_ts (&e->ts);
50 e->shape = NULL;
51 e->ref = NULL;
52 e->symtree = NULL;
53 return e;
57 /* Get a new expression node that is an array constructor
58 of given type and kind. */
60 gfc_expr *
61 gfc_get_array_expr (bt type, int kind, locus *where)
63 gfc_expr *e;
65 e = gfc_get_expr ();
66 e->expr_type = EXPR_ARRAY;
67 e->value.constructor = NULL;
68 e->rank = 1;
69 e->shape = NULL;
71 e->ts.type = type;
72 e->ts.kind = kind;
73 if (where)
74 e->where = *where;
76 return e;
80 /* Get a new expression node that is the NULL expression. */
82 gfc_expr *
83 gfc_get_null_expr (locus *where)
85 gfc_expr *e;
87 e = gfc_get_expr ();
88 e->expr_type = EXPR_NULL;
89 e->ts.type = BT_UNKNOWN;
91 if (where)
92 e->where = *where;
94 return e;
98 /* Get a new expression node that is an operator expression node. */
100 gfc_expr *
101 gfc_get_operator_expr (locus *where, gfc_intrinsic_op op,
102 gfc_expr *op1, gfc_expr *op2)
104 gfc_expr *e;
106 e = gfc_get_expr ();
107 e->expr_type = EXPR_OP;
108 e->value.op.op = op;
109 e->value.op.op1 = op1;
110 e->value.op.op2 = op2;
112 if (where)
113 e->where = *where;
115 return e;
119 /* Get a new expression node that is an structure constructor
120 of given type and kind. */
122 gfc_expr *
123 gfc_get_structure_constructor_expr (bt type, int kind, locus *where)
125 gfc_expr *e;
127 e = gfc_get_expr ();
128 e->expr_type = EXPR_STRUCTURE;
129 e->value.constructor = NULL;
131 e->ts.type = type;
132 e->ts.kind = kind;
133 if (where)
134 e->where = *where;
136 return e;
140 /* Get a new expression node that is an constant of given type and kind. */
142 gfc_expr *
143 gfc_get_constant_expr (bt type, int kind, locus *where)
145 gfc_expr *e;
147 if (!where)
148 gfc_internal_error ("gfc_get_constant_expr(): locus 'where' cannot be NULL");
150 e = gfc_get_expr ();
152 e->expr_type = EXPR_CONSTANT;
153 e->ts.type = type;
154 e->ts.kind = kind;
155 e->where = *where;
157 switch (type)
159 case BT_INTEGER:
160 mpz_init (e->value.integer);
161 break;
163 case BT_REAL:
164 gfc_set_model_kind (kind);
165 mpfr_init (e->value.real);
166 break;
168 case BT_COMPLEX:
169 gfc_set_model_kind (kind);
170 mpc_init2 (e->value.complex, mpfr_get_default_prec());
171 break;
173 default:
174 break;
177 return e;
181 /* Get a new expression node that is an string constant.
182 If no string is passed, a string of len is allocated,
183 blanked and null-terminated. */
185 gfc_expr *
186 gfc_get_character_expr (int kind, locus *where, const char *src, int len)
188 gfc_expr *e;
189 gfc_char_t *dest;
191 if (!src)
193 dest = gfc_get_wide_string (len + 1);
194 gfc_wide_memset (dest, ' ', len);
195 dest[len] = '\0';
197 else
198 dest = gfc_char_to_widechar (src);
200 e = gfc_get_constant_expr (BT_CHARACTER, kind,
201 where ? where : &gfc_current_locus);
202 e->value.character.string = dest;
203 e->value.character.length = len;
205 return e;
209 /* Get a new expression node that is an integer constant. */
211 gfc_expr *
212 gfc_get_int_expr (int kind, locus *where, int value)
214 gfc_expr *p;
215 p = gfc_get_constant_expr (BT_INTEGER, kind,
216 where ? where : &gfc_current_locus);
218 mpz_init_set_si (p->value.integer, value);
220 return p;
224 /* Get a new expression node that is a logical constant. */
226 gfc_expr *
227 gfc_get_logical_expr (int kind, locus *where, bool value)
229 gfc_expr *p;
230 p = gfc_get_constant_expr (BT_LOGICAL, kind,
231 where ? where : &gfc_current_locus);
233 p->value.logical = value;
235 return p;
239 gfc_expr *
240 gfc_get_iokind_expr (locus *where, io_kind k)
242 gfc_expr *e;
244 /* Set the types to something compatible with iokind. This is needed to
245 get through gfc_free_expr later since iokind really has no Basic Type,
246 BT, of its own. */
248 e = gfc_get_expr ();
249 e->expr_type = EXPR_CONSTANT;
250 e->ts.type = BT_LOGICAL;
251 e->value.iokind = k;
252 e->where = *where;
254 return e;
258 /* Given an expression pointer, return a copy of the expression. This
259 subroutine is recursive. */
261 gfc_expr *
262 gfc_copy_expr (gfc_expr *p)
264 gfc_expr *q;
265 gfc_char_t *s;
266 char *c;
268 if (p == NULL)
269 return NULL;
271 q = gfc_get_expr ();
272 *q = *p;
274 switch (q->expr_type)
276 case EXPR_SUBSTRING:
277 s = gfc_get_wide_string (p->value.character.length + 1);
278 q->value.character.string = s;
279 memcpy (s, p->value.character.string,
280 (p->value.character.length + 1) * sizeof (gfc_char_t));
281 break;
283 case EXPR_CONSTANT:
284 /* Copy target representation, if it exists. */
285 if (p->representation.string)
287 c = XCNEWVEC (char, p->representation.length + 1);
288 q->representation.string = c;
289 memcpy (c, p->representation.string, (p->representation.length + 1));
292 /* Copy the values of any pointer components of p->value. */
293 switch (q->ts.type)
295 case BT_INTEGER:
296 mpz_init_set (q->value.integer, p->value.integer);
297 break;
299 case BT_REAL:
300 gfc_set_model_kind (q->ts.kind);
301 mpfr_init (q->value.real);
302 mpfr_set (q->value.real, p->value.real, GFC_RND_MODE);
303 break;
305 case BT_COMPLEX:
306 gfc_set_model_kind (q->ts.kind);
307 mpc_init2 (q->value.complex, mpfr_get_default_prec());
308 mpc_set (q->value.complex, p->value.complex, GFC_MPC_RND_MODE);
309 break;
311 case BT_CHARACTER:
312 if (p->representation.string)
313 q->value.character.string
314 = gfc_char_to_widechar (q->representation.string);
315 else
317 s = gfc_get_wide_string (p->value.character.length + 1);
318 q->value.character.string = s;
320 /* This is the case for the C_NULL_CHAR named constant. */
321 if (p->value.character.length == 0
322 && (p->ts.is_c_interop || p->ts.is_iso_c))
324 *s = '\0';
325 /* Need to set the length to 1 to make sure the NUL
326 terminator is copied. */
327 q->value.character.length = 1;
329 else
330 memcpy (s, p->value.character.string,
331 (p->value.character.length + 1) * sizeof (gfc_char_t));
333 break;
335 case BT_HOLLERITH:
336 case BT_LOGICAL:
337 case BT_DERIVED:
338 case BT_CLASS:
339 break; /* Already done. */
341 case BT_PROCEDURE:
342 case BT_VOID:
343 /* Should never be reached. */
344 case BT_UNKNOWN:
345 gfc_internal_error ("gfc_copy_expr(): Bad expr node");
346 /* Not reached. */
349 break;
351 case EXPR_OP:
352 switch (q->value.op.op)
354 case INTRINSIC_NOT:
355 case INTRINSIC_PARENTHESES:
356 case INTRINSIC_UPLUS:
357 case INTRINSIC_UMINUS:
358 q->value.op.op1 = gfc_copy_expr (p->value.op.op1);
359 break;
361 default: /* Binary operators. */
362 q->value.op.op1 = gfc_copy_expr (p->value.op.op1);
363 q->value.op.op2 = gfc_copy_expr (p->value.op.op2);
364 break;
367 break;
369 case EXPR_FUNCTION:
370 q->value.function.actual =
371 gfc_copy_actual_arglist (p->value.function.actual);
372 break;
374 case EXPR_COMPCALL:
375 case EXPR_PPC:
376 q->value.compcall.actual =
377 gfc_copy_actual_arglist (p->value.compcall.actual);
378 q->value.compcall.tbp = p->value.compcall.tbp;
379 break;
381 case EXPR_STRUCTURE:
382 case EXPR_ARRAY:
383 q->value.constructor = gfc_constructor_copy (p->value.constructor);
384 break;
386 case EXPR_VARIABLE:
387 case EXPR_NULL:
388 break;
391 q->shape = gfc_copy_shape (p->shape, p->rank);
393 q->ref = gfc_copy_ref (p->ref);
395 return q;
399 /* Workhorse function for gfc_free_expr() that frees everything
400 beneath an expression node, but not the node itself. This is
401 useful when we want to simplify a node and replace it with
402 something else or the expression node belongs to another structure. */
404 static void
405 free_expr0 (gfc_expr *e)
407 int n;
409 switch (e->expr_type)
411 case EXPR_CONSTANT:
412 /* Free any parts of the value that need freeing. */
413 switch (e->ts.type)
415 case BT_INTEGER:
416 mpz_clear (e->value.integer);
417 break;
419 case BT_REAL:
420 mpfr_clear (e->value.real);
421 break;
423 case BT_CHARACTER:
424 gfc_free (e->value.character.string);
425 break;
427 case BT_COMPLEX:
428 mpc_clear (e->value.complex);
429 break;
431 default:
432 break;
435 /* Free the representation. */
436 if (e->representation.string)
437 gfc_free (e->representation.string);
439 break;
441 case EXPR_OP:
442 if (e->value.op.op1 != NULL)
443 gfc_free_expr (e->value.op.op1);
444 if (e->value.op.op2 != NULL)
445 gfc_free_expr (e->value.op.op2);
446 break;
448 case EXPR_FUNCTION:
449 gfc_free_actual_arglist (e->value.function.actual);
450 break;
452 case EXPR_COMPCALL:
453 case EXPR_PPC:
454 gfc_free_actual_arglist (e->value.compcall.actual);
455 break;
457 case EXPR_VARIABLE:
458 break;
460 case EXPR_ARRAY:
461 case EXPR_STRUCTURE:
462 gfc_constructor_free (e->value.constructor);
463 break;
465 case EXPR_SUBSTRING:
466 gfc_free (e->value.character.string);
467 break;
469 case EXPR_NULL:
470 break;
472 default:
473 gfc_internal_error ("free_expr0(): Bad expr type");
476 /* Free a shape array. */
477 if (e->shape != NULL)
479 for (n = 0; n < e->rank; n++)
480 mpz_clear (e->shape[n]);
482 gfc_free (e->shape);
485 gfc_free_ref_list (e->ref);
487 memset (e, '\0', sizeof (gfc_expr));
491 /* Free an expression node and everything beneath it. */
493 void
494 gfc_free_expr (gfc_expr *e)
496 if (e == NULL)
497 return;
498 free_expr0 (e);
499 gfc_free (e);
503 /* Free an argument list and everything below it. */
505 void
506 gfc_free_actual_arglist (gfc_actual_arglist *a1)
508 gfc_actual_arglist *a2;
510 while (a1)
512 a2 = a1->next;
513 gfc_free_expr (a1->expr);
514 gfc_free (a1);
515 a1 = a2;
520 /* Copy an arglist structure and all of the arguments. */
522 gfc_actual_arglist *
523 gfc_copy_actual_arglist (gfc_actual_arglist *p)
525 gfc_actual_arglist *head, *tail, *new_arg;
527 head = tail = NULL;
529 for (; p; p = p->next)
531 new_arg = gfc_get_actual_arglist ();
532 *new_arg = *p;
534 new_arg->expr = gfc_copy_expr (p->expr);
535 new_arg->next = NULL;
537 if (head == NULL)
538 head = new_arg;
539 else
540 tail->next = new_arg;
542 tail = new_arg;
545 return head;
549 /* Free a list of reference structures. */
551 void
552 gfc_free_ref_list (gfc_ref *p)
554 gfc_ref *q;
555 int i;
557 for (; p; p = q)
559 q = p->next;
561 switch (p->type)
563 case REF_ARRAY:
564 for (i = 0; i < GFC_MAX_DIMENSIONS; i++)
566 gfc_free_expr (p->u.ar.start[i]);
567 gfc_free_expr (p->u.ar.end[i]);
568 gfc_free_expr (p->u.ar.stride[i]);
571 break;
573 case REF_SUBSTRING:
574 gfc_free_expr (p->u.ss.start);
575 gfc_free_expr (p->u.ss.end);
576 break;
578 case REF_COMPONENT:
579 break;
582 gfc_free (p);
587 /* Graft the *src expression onto the *dest subexpression. */
589 void
590 gfc_replace_expr (gfc_expr *dest, gfc_expr *src)
592 free_expr0 (dest);
593 *dest = *src;
594 gfc_free (src);
598 /* Try to extract an integer constant from the passed expression node.
599 Returns an error message or NULL if the result is set. It is
600 tempting to generate an error and return SUCCESS or FAILURE, but
601 failure is OK for some callers. */
603 const char *
604 gfc_extract_int (gfc_expr *expr, int *result)
606 if (expr->expr_type != EXPR_CONSTANT)
607 return _("Constant expression required at %C");
609 if (expr->ts.type != BT_INTEGER)
610 return _("Integer expression required at %C");
612 if ((mpz_cmp_si (expr->value.integer, INT_MAX) > 0)
613 || (mpz_cmp_si (expr->value.integer, INT_MIN) < 0))
615 return _("Integer value too large in expression at %C");
618 *result = (int) mpz_get_si (expr->value.integer);
620 return NULL;
624 /* Recursively copy a list of reference structures. */
626 gfc_ref *
627 gfc_copy_ref (gfc_ref *src)
629 gfc_array_ref *ar;
630 gfc_ref *dest;
632 if (src == NULL)
633 return NULL;
635 dest = gfc_get_ref ();
636 dest->type = src->type;
638 switch (src->type)
640 case REF_ARRAY:
641 ar = gfc_copy_array_ref (&src->u.ar);
642 dest->u.ar = *ar;
643 gfc_free (ar);
644 break;
646 case REF_COMPONENT:
647 dest->u.c = src->u.c;
648 break;
650 case REF_SUBSTRING:
651 dest->u.ss = src->u.ss;
652 dest->u.ss.start = gfc_copy_expr (src->u.ss.start);
653 dest->u.ss.end = gfc_copy_expr (src->u.ss.end);
654 break;
657 dest->next = gfc_copy_ref (src->next);
659 return dest;
663 /* Detect whether an expression has any vector index array references. */
666 gfc_has_vector_index (gfc_expr *e)
668 gfc_ref *ref;
669 int i;
670 for (ref = e->ref; ref; ref = ref->next)
671 if (ref->type == REF_ARRAY)
672 for (i = 0; i < ref->u.ar.dimen; i++)
673 if (ref->u.ar.dimen_type[i] == DIMEN_VECTOR)
674 return 1;
675 return 0;
679 /* Insert a reference to the component of the given name.
680 Only to be used with CLASS containers. */
682 void
683 gfc_add_component_ref (gfc_expr *e, const char *name)
685 gfc_ref **tail = &(e->ref);
686 gfc_ref *next = NULL;
687 gfc_symbol *derived = e->symtree->n.sym->ts.u.derived;
688 while (*tail != NULL)
690 if ((*tail)->type == REF_COMPONENT)
691 derived = (*tail)->u.c.component->ts.u.derived;
692 if ((*tail)->type == REF_ARRAY && (*tail)->next == NULL)
693 break;
694 tail = &((*tail)->next);
696 if (*tail != NULL && strcmp (name, "$data") == 0)
697 next = *tail;
698 (*tail) = gfc_get_ref();
699 (*tail)->next = next;
700 (*tail)->type = REF_COMPONENT;
701 (*tail)->u.c.sym = derived;
702 (*tail)->u.c.component = gfc_find_component (derived, name, true, true);
703 gcc_assert((*tail)->u.c.component);
704 if (!next)
705 e->ts = (*tail)->u.c.component->ts;
709 /* Copy a shape array. */
711 mpz_t *
712 gfc_copy_shape (mpz_t *shape, int rank)
714 mpz_t *new_shape;
715 int n;
717 if (shape == NULL)
718 return NULL;
720 new_shape = gfc_get_shape (rank);
722 for (n = 0; n < rank; n++)
723 mpz_init_set (new_shape[n], shape[n]);
725 return new_shape;
729 /* Copy a shape array excluding dimension N, where N is an integer
730 constant expression. Dimensions are numbered in fortran style --
731 starting with ONE.
733 So, if the original shape array contains R elements
734 { s1 ... sN-1 sN sN+1 ... sR-1 sR}
735 the result contains R-1 elements:
736 { s1 ... sN-1 sN+1 ... sR-1}
738 If anything goes wrong -- N is not a constant, its value is out
739 of range -- or anything else, just returns NULL. */
741 mpz_t *
742 gfc_copy_shape_excluding (mpz_t *shape, int rank, gfc_expr *dim)
744 mpz_t *new_shape, *s;
745 int i, n;
747 if (shape == NULL
748 || rank <= 1
749 || dim == NULL
750 || dim->expr_type != EXPR_CONSTANT
751 || dim->ts.type != BT_INTEGER)
752 return NULL;
754 n = mpz_get_si (dim->value.integer);
755 n--; /* Convert to zero based index. */
756 if (n < 0 || n >= rank)
757 return NULL;
759 s = new_shape = gfc_get_shape (rank - 1);
761 for (i = 0; i < rank; i++)
763 if (i == n)
764 continue;
765 mpz_init_set (*s, shape[i]);
766 s++;
769 return new_shape;
773 /* Return the maximum kind of two expressions. In general, higher
774 kind numbers mean more precision for numeric types. */
777 gfc_kind_max (gfc_expr *e1, gfc_expr *e2)
779 return (e1->ts.kind > e2->ts.kind) ? e1->ts.kind : e2->ts.kind;
783 /* Returns nonzero if the type is numeric, zero otherwise. */
785 static int
786 numeric_type (bt type)
788 return type == BT_COMPLEX || type == BT_REAL || type == BT_INTEGER;
792 /* Returns nonzero if the typespec is a numeric type, zero otherwise. */
795 gfc_numeric_ts (gfc_typespec *ts)
797 return numeric_type (ts->type);
801 /* Return an expression node with an optional argument list attached.
802 A variable number of gfc_expr pointers are strung together in an
803 argument list with a NULL pointer terminating the list. */
805 gfc_expr *
806 gfc_build_conversion (gfc_expr *e)
808 gfc_expr *p;
810 p = gfc_get_expr ();
811 p->expr_type = EXPR_FUNCTION;
812 p->symtree = NULL;
813 p->value.function.actual = NULL;
815 p->value.function.actual = gfc_get_actual_arglist ();
816 p->value.function.actual->expr = e;
818 return p;
822 /* Given an expression node with some sort of numeric binary
823 expression, insert type conversions required to make the operands
824 have the same type. Conversion warnings are disabled if wconversion
825 is set to 0.
827 The exception is that the operands of an exponential don't have to
828 have the same type. If possible, the base is promoted to the type
829 of the exponent. For example, 1**2.3 becomes 1.0**2.3, but
830 1.0**2 stays as it is. */
832 void
833 gfc_type_convert_binary (gfc_expr *e, int wconversion)
835 gfc_expr *op1, *op2;
837 op1 = e->value.op.op1;
838 op2 = e->value.op.op2;
840 if (op1->ts.type == BT_UNKNOWN || op2->ts.type == BT_UNKNOWN)
842 gfc_clear_ts (&e->ts);
843 return;
846 /* Kind conversions of same type. */
847 if (op1->ts.type == op2->ts.type)
849 if (op1->ts.kind == op2->ts.kind)
851 /* No type conversions. */
852 e->ts = op1->ts;
853 goto done;
856 if (op1->ts.kind > op2->ts.kind)
857 gfc_convert_type_warn (op2, &op1->ts, 2, wconversion);
858 else
859 gfc_convert_type_warn (op1, &op2->ts, 2, wconversion);
861 e->ts = op1->ts;
862 goto done;
865 /* Integer combined with real or complex. */
866 if (op2->ts.type == BT_INTEGER)
868 e->ts = op1->ts;
870 /* Special case for ** operator. */
871 if (e->value.op.op == INTRINSIC_POWER)
872 goto done;
874 gfc_convert_type_warn (e->value.op.op2, &e->ts, 2, wconversion);
875 goto done;
878 if (op1->ts.type == BT_INTEGER)
880 e->ts = op2->ts;
881 gfc_convert_type_warn (e->value.op.op1, &e->ts, 2, wconversion);
882 goto done;
885 /* Real combined with complex. */
886 e->ts.type = BT_COMPLEX;
887 if (op1->ts.kind > op2->ts.kind)
888 e->ts.kind = op1->ts.kind;
889 else
890 e->ts.kind = op2->ts.kind;
891 if (op1->ts.type != BT_COMPLEX || op1->ts.kind != e->ts.kind)
892 gfc_convert_type_warn (e->value.op.op1, &e->ts, 2, wconversion);
893 if (op2->ts.type != BT_COMPLEX || op2->ts.kind != e->ts.kind)
894 gfc_convert_type_warn (e->value.op.op2, &e->ts, 2, wconversion);
896 done:
897 return;
901 static match
902 check_specification_function (gfc_expr *e)
904 gfc_symbol *sym;
906 if (!e->symtree)
907 return MATCH_NO;
909 sym = e->symtree->n.sym;
911 /* F95, 7.1.6.2; F2003, 7.1.7 */
912 if (sym
913 && sym->attr.function
914 && sym->attr.pure
915 && !sym->attr.intrinsic
916 && !sym->attr.recursive
917 && sym->attr.proc != PROC_INTERNAL
918 && sym->attr.proc != PROC_ST_FUNCTION
919 && sym->attr.proc != PROC_UNKNOWN
920 && sym->formal == NULL)
921 return MATCH_YES;
923 return MATCH_NO;
926 /* Function to determine if an expression is constant or not. This
927 function expects that the expression has already been simplified. */
930 gfc_is_constant_expr (gfc_expr *e)
932 gfc_constructor *c;
933 gfc_actual_arglist *arg;
935 if (e == NULL)
936 return 1;
938 switch (e->expr_type)
940 case EXPR_OP:
941 return (gfc_is_constant_expr (e->value.op.op1)
942 && (e->value.op.op2 == NULL
943 || gfc_is_constant_expr (e->value.op.op2)));
945 case EXPR_VARIABLE:
946 return 0;
948 case EXPR_FUNCTION:
949 case EXPR_PPC:
950 case EXPR_COMPCALL:
951 /* Specification functions are constant. */
952 if (check_specification_function (e) == MATCH_YES)
953 return 1;
955 /* Call to intrinsic with at least one argument. */
956 if (e->value.function.isym && e->value.function.actual)
958 for (arg = e->value.function.actual; arg; arg = arg->next)
959 if (!gfc_is_constant_expr (arg->expr))
960 return 0;
962 return 1;
964 else
965 return 0;
967 case EXPR_CONSTANT:
968 case EXPR_NULL:
969 return 1;
971 case EXPR_SUBSTRING:
972 return e->ref == NULL || (gfc_is_constant_expr (e->ref->u.ss.start)
973 && gfc_is_constant_expr (e->ref->u.ss.end));
975 case EXPR_STRUCTURE:
976 for (c = gfc_constructor_first (e->value.constructor);
977 c; c = gfc_constructor_next (c))
978 if (!gfc_is_constant_expr (c->expr))
979 return 0;
981 return 1;
983 case EXPR_ARRAY:
984 return gfc_constant_ac (e);
986 default:
987 gfc_internal_error ("gfc_is_constant_expr(): Unknown expression type");
988 return 0;
993 /* Is true if an array reference is followed by a component or substring
994 reference. */
995 bool
996 is_subref_array (gfc_expr * e)
998 gfc_ref * ref;
999 bool seen_array;
1001 if (e->expr_type != EXPR_VARIABLE)
1002 return false;
1004 if (e->symtree->n.sym->attr.subref_array_pointer)
1005 return true;
1007 seen_array = false;
1008 for (ref = e->ref; ref; ref = ref->next)
1010 if (ref->type == REF_ARRAY
1011 && ref->u.ar.type != AR_ELEMENT)
1012 seen_array = true;
1014 if (seen_array
1015 && ref->type != REF_ARRAY)
1016 return seen_array;
1018 return false;
1022 /* Try to collapse intrinsic expressions. */
1024 static gfc_try
1025 simplify_intrinsic_op (gfc_expr *p, int type)
1027 gfc_intrinsic_op op;
1028 gfc_expr *op1, *op2, *result;
1030 if (p->value.op.op == INTRINSIC_USER)
1031 return SUCCESS;
1033 op1 = p->value.op.op1;
1034 op2 = p->value.op.op2;
1035 op = p->value.op.op;
1037 if (gfc_simplify_expr (op1, type) == FAILURE)
1038 return FAILURE;
1039 if (gfc_simplify_expr (op2, type) == FAILURE)
1040 return FAILURE;
1042 if (!gfc_is_constant_expr (op1)
1043 || (op2 != NULL && !gfc_is_constant_expr (op2)))
1044 return SUCCESS;
1046 /* Rip p apart. */
1047 p->value.op.op1 = NULL;
1048 p->value.op.op2 = NULL;
1050 switch (op)
1052 case INTRINSIC_PARENTHESES:
1053 result = gfc_parentheses (op1);
1054 break;
1056 case INTRINSIC_UPLUS:
1057 result = gfc_uplus (op1);
1058 break;
1060 case INTRINSIC_UMINUS:
1061 result = gfc_uminus (op1);
1062 break;
1064 case INTRINSIC_PLUS:
1065 result = gfc_add (op1, op2);
1066 break;
1068 case INTRINSIC_MINUS:
1069 result = gfc_subtract (op1, op2);
1070 break;
1072 case INTRINSIC_TIMES:
1073 result = gfc_multiply (op1, op2);
1074 break;
1076 case INTRINSIC_DIVIDE:
1077 result = gfc_divide (op1, op2);
1078 break;
1080 case INTRINSIC_POWER:
1081 result = gfc_power (op1, op2);
1082 break;
1084 case INTRINSIC_CONCAT:
1085 result = gfc_concat (op1, op2);
1086 break;
1088 case INTRINSIC_EQ:
1089 case INTRINSIC_EQ_OS:
1090 result = gfc_eq (op1, op2, op);
1091 break;
1093 case INTRINSIC_NE:
1094 case INTRINSIC_NE_OS:
1095 result = gfc_ne (op1, op2, op);
1096 break;
1098 case INTRINSIC_GT:
1099 case INTRINSIC_GT_OS:
1100 result = gfc_gt (op1, op2, op);
1101 break;
1103 case INTRINSIC_GE:
1104 case INTRINSIC_GE_OS:
1105 result = gfc_ge (op1, op2, op);
1106 break;
1108 case INTRINSIC_LT:
1109 case INTRINSIC_LT_OS:
1110 result = gfc_lt (op1, op2, op);
1111 break;
1113 case INTRINSIC_LE:
1114 case INTRINSIC_LE_OS:
1115 result = gfc_le (op1, op2, op);
1116 break;
1118 case INTRINSIC_NOT:
1119 result = gfc_not (op1);
1120 break;
1122 case INTRINSIC_AND:
1123 result = gfc_and (op1, op2);
1124 break;
1126 case INTRINSIC_OR:
1127 result = gfc_or (op1, op2);
1128 break;
1130 case INTRINSIC_EQV:
1131 result = gfc_eqv (op1, op2);
1132 break;
1134 case INTRINSIC_NEQV:
1135 result = gfc_neqv (op1, op2);
1136 break;
1138 default:
1139 gfc_internal_error ("simplify_intrinsic_op(): Bad operator");
1142 if (result == NULL)
1144 gfc_free_expr (op1);
1145 gfc_free_expr (op2);
1146 return FAILURE;
1149 result->rank = p->rank;
1150 result->where = p->where;
1151 gfc_replace_expr (p, result);
1153 return SUCCESS;
1157 /* Subroutine to simplify constructor expressions. Mutually recursive
1158 with gfc_simplify_expr(). */
1160 static gfc_try
1161 simplify_constructor (gfc_constructor_base base, int type)
1163 gfc_constructor *c;
1164 gfc_expr *p;
1166 for (c = gfc_constructor_first (base); c; c = gfc_constructor_next (c))
1168 if (c->iterator
1169 && (gfc_simplify_expr (c->iterator->start, type) == FAILURE
1170 || gfc_simplify_expr (c->iterator->end, type) == FAILURE
1171 || gfc_simplify_expr (c->iterator->step, type) == FAILURE))
1172 return FAILURE;
1174 if (c->expr)
1176 /* Try and simplify a copy. Replace the original if successful
1177 but keep going through the constructor at all costs. Not
1178 doing so can make a dog's dinner of complicated things. */
1179 p = gfc_copy_expr (c->expr);
1181 if (gfc_simplify_expr (p, type) == FAILURE)
1183 gfc_free_expr (p);
1184 continue;
1187 gfc_replace_expr (c->expr, p);
1191 return SUCCESS;
1195 /* Pull a single array element out of an array constructor. */
1197 static gfc_try
1198 find_array_element (gfc_constructor_base base, gfc_array_ref *ar,
1199 gfc_constructor **rval)
1201 unsigned long nelemen;
1202 int i;
1203 mpz_t delta;
1204 mpz_t offset;
1205 mpz_t span;
1206 mpz_t tmp;
1207 gfc_constructor *cons;
1208 gfc_expr *e;
1209 gfc_try t;
1211 t = SUCCESS;
1212 e = NULL;
1214 mpz_init_set_ui (offset, 0);
1215 mpz_init (delta);
1216 mpz_init (tmp);
1217 mpz_init_set_ui (span, 1);
1218 for (i = 0; i < ar->dimen; i++)
1220 if (gfc_reduce_init_expr (ar->as->lower[i]) == FAILURE
1221 || gfc_reduce_init_expr (ar->as->upper[i]) == FAILURE)
1223 t = FAILURE;
1224 cons = NULL;
1225 goto depart;
1228 e = gfc_copy_expr (ar->start[i]);
1229 if (e->expr_type != EXPR_CONSTANT)
1231 cons = NULL;
1232 goto depart;
1235 gcc_assert (ar->as->upper[i]->expr_type == EXPR_CONSTANT
1236 && ar->as->lower[i]->expr_type == EXPR_CONSTANT);
1238 /* Check the bounds. */
1239 if ((ar->as->upper[i]
1240 && mpz_cmp (e->value.integer,
1241 ar->as->upper[i]->value.integer) > 0)
1242 || (mpz_cmp (e->value.integer,
1243 ar->as->lower[i]->value.integer) < 0))
1245 gfc_error ("Index in dimension %d is out of bounds "
1246 "at %L", i + 1, &ar->c_where[i]);
1247 cons = NULL;
1248 t = FAILURE;
1249 goto depart;
1252 mpz_sub (delta, e->value.integer, ar->as->lower[i]->value.integer);
1253 mpz_mul (delta, delta, span);
1254 mpz_add (offset, offset, delta);
1256 mpz_set_ui (tmp, 1);
1257 mpz_add (tmp, tmp, ar->as->upper[i]->value.integer);
1258 mpz_sub (tmp, tmp, ar->as->lower[i]->value.integer);
1259 mpz_mul (span, span, tmp);
1262 for (cons = gfc_constructor_first (base), nelemen = mpz_get_ui (offset);
1263 cons && nelemen > 0; cons = gfc_constructor_next (cons), nelemen--)
1265 if (cons->iterator)
1267 cons = NULL;
1268 goto depart;
1272 depart:
1273 mpz_clear (delta);
1274 mpz_clear (offset);
1275 mpz_clear (span);
1276 mpz_clear (tmp);
1277 if (e)
1278 gfc_free_expr (e);
1279 *rval = cons;
1280 return t;
1284 /* Find a component of a structure constructor. */
1286 static gfc_constructor *
1287 find_component_ref (gfc_constructor_base base, gfc_ref *ref)
1289 gfc_component *comp;
1290 gfc_component *pick;
1291 gfc_constructor *c = gfc_constructor_first (base);
1293 comp = ref->u.c.sym->components;
1294 pick = ref->u.c.component;
1295 while (comp != pick)
1297 comp = comp->next;
1298 c = gfc_constructor_next (c);
1301 return c;
1305 /* Replace an expression with the contents of a constructor, removing
1306 the subobject reference in the process. */
1308 static void
1309 remove_subobject_ref (gfc_expr *p, gfc_constructor *cons)
1311 gfc_expr *e;
1313 if (cons)
1315 e = cons->expr;
1316 cons->expr = NULL;
1318 else
1319 e = gfc_copy_expr (p);
1320 e->ref = p->ref->next;
1321 p->ref->next = NULL;
1322 gfc_replace_expr (p, e);
1326 /* Pull an array section out of an array constructor. */
1328 static gfc_try
1329 find_array_section (gfc_expr *expr, gfc_ref *ref)
1331 int idx;
1332 int rank;
1333 int d;
1334 int shape_i;
1335 int limit;
1336 long unsigned one = 1;
1337 bool incr_ctr;
1338 mpz_t start[GFC_MAX_DIMENSIONS];
1339 mpz_t end[GFC_MAX_DIMENSIONS];
1340 mpz_t stride[GFC_MAX_DIMENSIONS];
1341 mpz_t delta[GFC_MAX_DIMENSIONS];
1342 mpz_t ctr[GFC_MAX_DIMENSIONS];
1343 mpz_t delta_mpz;
1344 mpz_t tmp_mpz;
1345 mpz_t nelts;
1346 mpz_t ptr;
1347 gfc_constructor_base base;
1348 gfc_constructor *cons, *vecsub[GFC_MAX_DIMENSIONS];
1349 gfc_expr *begin;
1350 gfc_expr *finish;
1351 gfc_expr *step;
1352 gfc_expr *upper;
1353 gfc_expr *lower;
1354 gfc_try t;
1356 t = SUCCESS;
1358 base = expr->value.constructor;
1359 expr->value.constructor = NULL;
1361 rank = ref->u.ar.as->rank;
1363 if (expr->shape == NULL)
1364 expr->shape = gfc_get_shape (rank);
1366 mpz_init_set_ui (delta_mpz, one);
1367 mpz_init_set_ui (nelts, one);
1368 mpz_init (tmp_mpz);
1370 /* Do the initialization now, so that we can cleanup without
1371 keeping track of where we were. */
1372 for (d = 0; d < rank; d++)
1374 mpz_init (delta[d]);
1375 mpz_init (start[d]);
1376 mpz_init (end[d]);
1377 mpz_init (ctr[d]);
1378 mpz_init (stride[d]);
1379 vecsub[d] = NULL;
1382 /* Build the counters to clock through the array reference. */
1383 shape_i = 0;
1384 for (d = 0; d < rank; d++)
1386 /* Make this stretch of code easier on the eye! */
1387 begin = ref->u.ar.start[d];
1388 finish = ref->u.ar.end[d];
1389 step = ref->u.ar.stride[d];
1390 lower = ref->u.ar.as->lower[d];
1391 upper = ref->u.ar.as->upper[d];
1393 if (ref->u.ar.dimen_type[d] == DIMEN_VECTOR) /* Vector subscript. */
1395 gfc_constructor *ci;
1396 gcc_assert (begin);
1398 if (begin->expr_type != EXPR_ARRAY || !gfc_is_constant_expr (begin))
1400 t = FAILURE;
1401 goto cleanup;
1404 gcc_assert (begin->rank == 1);
1405 /* Zero-sized arrays have no shape and no elements, stop early. */
1406 if (!begin->shape)
1408 mpz_init_set_ui (nelts, 0);
1409 break;
1412 vecsub[d] = gfc_constructor_first (begin->value.constructor);
1413 mpz_set (ctr[d], vecsub[d]->expr->value.integer);
1414 mpz_mul (nelts, nelts, begin->shape[0]);
1415 mpz_set (expr->shape[shape_i++], begin->shape[0]);
1417 /* Check bounds. */
1418 for (ci = vecsub[d]; ci; ci = gfc_constructor_next (ci))
1420 if (mpz_cmp (ci->expr->value.integer, upper->value.integer) > 0
1421 || mpz_cmp (ci->expr->value.integer,
1422 lower->value.integer) < 0)
1424 gfc_error ("index in dimension %d is out of bounds "
1425 "at %L", d + 1, &ref->u.ar.c_where[d]);
1426 t = FAILURE;
1427 goto cleanup;
1431 else
1433 if ((begin && begin->expr_type != EXPR_CONSTANT)
1434 || (finish && finish->expr_type != EXPR_CONSTANT)
1435 || (step && step->expr_type != EXPR_CONSTANT))
1437 t = FAILURE;
1438 goto cleanup;
1441 /* Obtain the stride. */
1442 if (step)
1443 mpz_set (stride[d], step->value.integer);
1444 else
1445 mpz_set_ui (stride[d], one);
1447 if (mpz_cmp_ui (stride[d], 0) == 0)
1448 mpz_set_ui (stride[d], one);
1450 /* Obtain the start value for the index. */
1451 if (begin)
1452 mpz_set (start[d], begin->value.integer);
1453 else
1454 mpz_set (start[d], lower->value.integer);
1456 mpz_set (ctr[d], start[d]);
1458 /* Obtain the end value for the index. */
1459 if (finish)
1460 mpz_set (end[d], finish->value.integer);
1461 else
1462 mpz_set (end[d], upper->value.integer);
1464 /* Separate 'if' because elements sometimes arrive with
1465 non-null end. */
1466 if (ref->u.ar.dimen_type[d] == DIMEN_ELEMENT)
1467 mpz_set (end [d], begin->value.integer);
1469 /* Check the bounds. */
1470 if (mpz_cmp (ctr[d], upper->value.integer) > 0
1471 || mpz_cmp (end[d], upper->value.integer) > 0
1472 || mpz_cmp (ctr[d], lower->value.integer) < 0
1473 || mpz_cmp (end[d], lower->value.integer) < 0)
1475 gfc_error ("index in dimension %d is out of bounds "
1476 "at %L", d + 1, &ref->u.ar.c_where[d]);
1477 t = FAILURE;
1478 goto cleanup;
1481 /* Calculate the number of elements and the shape. */
1482 mpz_set (tmp_mpz, stride[d]);
1483 mpz_add (tmp_mpz, end[d], tmp_mpz);
1484 mpz_sub (tmp_mpz, tmp_mpz, ctr[d]);
1485 mpz_div (tmp_mpz, tmp_mpz, stride[d]);
1486 mpz_mul (nelts, nelts, tmp_mpz);
1488 /* An element reference reduces the rank of the expression; don't
1489 add anything to the shape array. */
1490 if (ref->u.ar.dimen_type[d] != DIMEN_ELEMENT)
1491 mpz_set (expr->shape[shape_i++], tmp_mpz);
1494 /* Calculate the 'stride' (=delta) for conversion of the
1495 counter values into the index along the constructor. */
1496 mpz_set (delta[d], delta_mpz);
1497 mpz_sub (tmp_mpz, upper->value.integer, lower->value.integer);
1498 mpz_add_ui (tmp_mpz, tmp_mpz, one);
1499 mpz_mul (delta_mpz, delta_mpz, tmp_mpz);
1502 mpz_init (ptr);
1503 cons = gfc_constructor_first (base);
1505 /* Now clock through the array reference, calculating the index in
1506 the source constructor and transferring the elements to the new
1507 constructor. */
1508 for (idx = 0; idx < (int) mpz_get_si (nelts); idx++)
1510 if (ref->u.ar.offset)
1511 mpz_set (ptr, ref->u.ar.offset->value.integer);
1512 else
1513 mpz_init_set_ui (ptr, 0);
1515 incr_ctr = true;
1516 for (d = 0; d < rank; d++)
1518 mpz_set (tmp_mpz, ctr[d]);
1519 mpz_sub (tmp_mpz, tmp_mpz, ref->u.ar.as->lower[d]->value.integer);
1520 mpz_mul (tmp_mpz, tmp_mpz, delta[d]);
1521 mpz_add (ptr, ptr, tmp_mpz);
1523 if (!incr_ctr) continue;
1525 if (ref->u.ar.dimen_type[d] == DIMEN_VECTOR) /* Vector subscript. */
1527 gcc_assert(vecsub[d]);
1529 if (!gfc_constructor_next (vecsub[d]))
1530 vecsub[d] = gfc_constructor_first (ref->u.ar.start[d]->value.constructor);
1531 else
1533 vecsub[d] = gfc_constructor_next (vecsub[d]);
1534 incr_ctr = false;
1536 mpz_set (ctr[d], vecsub[d]->expr->value.integer);
1538 else
1540 mpz_add (ctr[d], ctr[d], stride[d]);
1542 if (mpz_cmp_ui (stride[d], 0) > 0
1543 ? mpz_cmp (ctr[d], end[d]) > 0
1544 : mpz_cmp (ctr[d], end[d]) < 0)
1545 mpz_set (ctr[d], start[d]);
1546 else
1547 incr_ctr = false;
1551 limit = mpz_get_ui (ptr);
1552 if (limit >= gfc_option.flag_max_array_constructor)
1554 gfc_error ("The number of elements in the array constructor "
1555 "at %L requires an increase of the allowed %d "
1556 "upper limit. See -fmax-array-constructor "
1557 "option", &expr->where,
1558 gfc_option.flag_max_array_constructor);
1559 return FAILURE;
1562 cons = gfc_constructor_lookup (base, limit);
1563 gcc_assert (cons);
1564 gfc_constructor_append_expr (&expr->value.constructor,
1565 gfc_copy_expr (cons->expr), NULL);
1568 mpz_clear (ptr);
1570 cleanup:
1572 mpz_clear (delta_mpz);
1573 mpz_clear (tmp_mpz);
1574 mpz_clear (nelts);
1575 for (d = 0; d < rank; d++)
1577 mpz_clear (delta[d]);
1578 mpz_clear (start[d]);
1579 mpz_clear (end[d]);
1580 mpz_clear (ctr[d]);
1581 mpz_clear (stride[d]);
1583 gfc_constructor_free (base);
1584 return t;
1587 /* Pull a substring out of an expression. */
1589 static gfc_try
1590 find_substring_ref (gfc_expr *p, gfc_expr **newp)
1592 int end;
1593 int start;
1594 int length;
1595 gfc_char_t *chr;
1597 if (p->ref->u.ss.start->expr_type != EXPR_CONSTANT
1598 || p->ref->u.ss.end->expr_type != EXPR_CONSTANT)
1599 return FAILURE;
1601 *newp = gfc_copy_expr (p);
1602 gfc_free ((*newp)->value.character.string);
1604 end = (int) mpz_get_ui (p->ref->u.ss.end->value.integer);
1605 start = (int) mpz_get_ui (p->ref->u.ss.start->value.integer);
1606 length = end - start + 1;
1608 chr = (*newp)->value.character.string = gfc_get_wide_string (length + 1);
1609 (*newp)->value.character.length = length;
1610 memcpy (chr, &p->value.character.string[start - 1],
1611 length * sizeof (gfc_char_t));
1612 chr[length] = '\0';
1613 return SUCCESS;
1618 /* Simplify a subobject reference of a constructor. This occurs when
1619 parameter variable values are substituted. */
1621 static gfc_try
1622 simplify_const_ref (gfc_expr *p)
1624 gfc_constructor *cons, *c;
1625 gfc_expr *newp;
1626 gfc_ref *last_ref;
1628 while (p->ref)
1630 switch (p->ref->type)
1632 case REF_ARRAY:
1633 switch (p->ref->u.ar.type)
1635 case AR_ELEMENT:
1636 /* <type/kind spec>, parameter :: x(<int>) = scalar_expr
1637 will generate this. */
1638 if (p->expr_type != EXPR_ARRAY)
1640 remove_subobject_ref (p, NULL);
1641 break;
1643 if (find_array_element (p->value.constructor, &p->ref->u.ar,
1644 &cons) == FAILURE)
1645 return FAILURE;
1647 if (!cons)
1648 return SUCCESS;
1650 remove_subobject_ref (p, cons);
1651 break;
1653 case AR_SECTION:
1654 if (find_array_section (p, p->ref) == FAILURE)
1655 return FAILURE;
1656 p->ref->u.ar.type = AR_FULL;
1658 /* Fall through. */
1660 case AR_FULL:
1661 if (p->ref->next != NULL
1662 && (p->ts.type == BT_CHARACTER || p->ts.type == BT_DERIVED))
1664 for (c = gfc_constructor_first (p->value.constructor);
1665 c; c = gfc_constructor_next (c))
1667 c->expr->ref = gfc_copy_ref (p->ref->next);
1668 if (simplify_const_ref (c->expr) == FAILURE)
1669 return FAILURE;
1672 if (p->ts.type == BT_DERIVED
1673 && p->ref->next
1674 && (c = gfc_constructor_first (p->value.constructor)))
1676 /* There may have been component references. */
1677 p->ts = c->expr->ts;
1680 last_ref = p->ref;
1681 for (; last_ref->next; last_ref = last_ref->next) {};
1683 if (p->ts.type == BT_CHARACTER
1684 && last_ref->type == REF_SUBSTRING)
1686 /* If this is a CHARACTER array and we possibly took
1687 a substring out of it, update the type-spec's
1688 character length according to the first element
1689 (as all should have the same length). */
1690 int string_len;
1691 if ((c = gfc_constructor_first (p->value.constructor)))
1693 const gfc_expr* first = c->expr;
1694 gcc_assert (first->expr_type == EXPR_CONSTANT);
1695 gcc_assert (first->ts.type == BT_CHARACTER);
1696 string_len = first->value.character.length;
1698 else
1699 string_len = 0;
1701 if (!p->ts.u.cl)
1702 p->ts.u.cl = gfc_new_charlen (p->symtree->n.sym->ns,
1703 NULL);
1704 else
1705 gfc_free_expr (p->ts.u.cl->length);
1707 p->ts.u.cl->length
1708 = gfc_get_int_expr (gfc_default_integer_kind,
1709 NULL, string_len);
1712 gfc_free_ref_list (p->ref);
1713 p->ref = NULL;
1714 break;
1716 default:
1717 return SUCCESS;
1720 break;
1722 case REF_COMPONENT:
1723 cons = find_component_ref (p->value.constructor, p->ref);
1724 remove_subobject_ref (p, cons);
1725 break;
1727 case REF_SUBSTRING:
1728 if (find_substring_ref (p, &newp) == FAILURE)
1729 return FAILURE;
1731 gfc_replace_expr (p, newp);
1732 gfc_free_ref_list (p->ref);
1733 p->ref = NULL;
1734 break;
1738 return SUCCESS;
1742 /* Simplify a chain of references. */
1744 static gfc_try
1745 simplify_ref_chain (gfc_ref *ref, int type)
1747 int n;
1749 for (; ref; ref = ref->next)
1751 switch (ref->type)
1753 case REF_ARRAY:
1754 for (n = 0; n < ref->u.ar.dimen; n++)
1756 if (gfc_simplify_expr (ref->u.ar.start[n], type) == FAILURE)
1757 return FAILURE;
1758 if (gfc_simplify_expr (ref->u.ar.end[n], type) == FAILURE)
1759 return FAILURE;
1760 if (gfc_simplify_expr (ref->u.ar.stride[n], type) == FAILURE)
1761 return FAILURE;
1763 break;
1765 case REF_SUBSTRING:
1766 if (gfc_simplify_expr (ref->u.ss.start, type) == FAILURE)
1767 return FAILURE;
1768 if (gfc_simplify_expr (ref->u.ss.end, type) == FAILURE)
1769 return FAILURE;
1770 break;
1772 default:
1773 break;
1776 return SUCCESS;
1780 /* Try to substitute the value of a parameter variable. */
1782 static gfc_try
1783 simplify_parameter_variable (gfc_expr *p, int type)
1785 gfc_expr *e;
1786 gfc_try t;
1788 e = gfc_copy_expr (p->symtree->n.sym->value);
1789 if (e == NULL)
1790 return FAILURE;
1792 e->rank = p->rank;
1794 /* Do not copy subobject refs for constant. */
1795 if (e->expr_type != EXPR_CONSTANT && p->ref != NULL)
1796 e->ref = gfc_copy_ref (p->ref);
1797 t = gfc_simplify_expr (e, type);
1799 /* Only use the simplification if it eliminated all subobject references. */
1800 if (t == SUCCESS && !e->ref)
1801 gfc_replace_expr (p, e);
1802 else
1803 gfc_free_expr (e);
1805 return t;
1808 /* Given an expression, simplify it by collapsing constant
1809 expressions. Most simplification takes place when the expression
1810 tree is being constructed. If an intrinsic function is simplified
1811 at some point, we get called again to collapse the result against
1812 other constants.
1814 We work by recursively simplifying expression nodes, simplifying
1815 intrinsic functions where possible, which can lead to further
1816 constant collapsing. If an operator has constant operand(s), we
1817 rip the expression apart, and rebuild it, hoping that it becomes
1818 something simpler.
1820 The expression type is defined for:
1821 0 Basic expression parsing
1822 1 Simplifying array constructors -- will substitute
1823 iterator values.
1824 Returns FAILURE on error, SUCCESS otherwise.
1825 NOTE: Will return SUCCESS even if the expression can not be simplified. */
1827 gfc_try
1828 gfc_simplify_expr (gfc_expr *p, int type)
1830 gfc_actual_arglist *ap;
1832 if (p == NULL)
1833 return SUCCESS;
1835 switch (p->expr_type)
1837 case EXPR_CONSTANT:
1838 case EXPR_NULL:
1839 break;
1841 case EXPR_FUNCTION:
1842 for (ap = p->value.function.actual; ap; ap = ap->next)
1843 if (gfc_simplify_expr (ap->expr, type) == FAILURE)
1844 return FAILURE;
1846 if (p->value.function.isym != NULL
1847 && gfc_intrinsic_func_interface (p, 1) == MATCH_ERROR)
1848 return FAILURE;
1850 break;
1852 case EXPR_SUBSTRING:
1853 if (simplify_ref_chain (p->ref, type) == FAILURE)
1854 return FAILURE;
1856 if (gfc_is_constant_expr (p))
1858 gfc_char_t *s;
1859 int start, end;
1861 start = 0;
1862 if (p->ref && p->ref->u.ss.start)
1864 gfc_extract_int (p->ref->u.ss.start, &start);
1865 start--; /* Convert from one-based to zero-based. */
1868 end = p->value.character.length;
1869 if (p->ref && p->ref->u.ss.end)
1870 gfc_extract_int (p->ref->u.ss.end, &end);
1872 s = gfc_get_wide_string (end - start + 2);
1873 memcpy (s, p->value.character.string + start,
1874 (end - start) * sizeof (gfc_char_t));
1875 s[end - start + 1] = '\0'; /* TODO: C-style string. */
1876 gfc_free (p->value.character.string);
1877 p->value.character.string = s;
1878 p->value.character.length = end - start;
1879 p->ts.u.cl = gfc_new_charlen (gfc_current_ns, NULL);
1880 p->ts.u.cl->length = gfc_get_int_expr (gfc_default_integer_kind,
1881 NULL,
1882 p->value.character.length);
1883 gfc_free_ref_list (p->ref);
1884 p->ref = NULL;
1885 p->expr_type = EXPR_CONSTANT;
1887 break;
1889 case EXPR_OP:
1890 if (simplify_intrinsic_op (p, type) == FAILURE)
1891 return FAILURE;
1892 break;
1894 case EXPR_VARIABLE:
1895 /* Only substitute array parameter variables if we are in an
1896 initialization expression, or we want a subsection. */
1897 if (p->symtree->n.sym->attr.flavor == FL_PARAMETER
1898 && (gfc_init_expr || p->ref
1899 || p->symtree->n.sym->value->expr_type != EXPR_ARRAY))
1901 if (simplify_parameter_variable (p, type) == FAILURE)
1902 return FAILURE;
1903 break;
1906 if (type == 1)
1908 gfc_simplify_iterator_var (p);
1911 /* Simplify subcomponent references. */
1912 if (simplify_ref_chain (p->ref, type) == FAILURE)
1913 return FAILURE;
1915 break;
1917 case EXPR_STRUCTURE:
1918 case EXPR_ARRAY:
1919 if (simplify_ref_chain (p->ref, type) == FAILURE)
1920 return FAILURE;
1922 if (simplify_constructor (p->value.constructor, type) == FAILURE)
1923 return FAILURE;
1925 if (p->expr_type == EXPR_ARRAY && p->ref && p->ref->type == REF_ARRAY
1926 && p->ref->u.ar.type == AR_FULL)
1927 gfc_expand_constructor (p);
1929 if (simplify_const_ref (p) == FAILURE)
1930 return FAILURE;
1932 break;
1934 case EXPR_COMPCALL:
1935 case EXPR_PPC:
1936 gcc_unreachable ();
1937 break;
1940 return SUCCESS;
1944 /* Returns the type of an expression with the exception that iterator
1945 variables are automatically integers no matter what else they may
1946 be declared as. */
1948 static bt
1949 et0 (gfc_expr *e)
1951 if (e->expr_type == EXPR_VARIABLE && gfc_check_iter_variable (e) == SUCCESS)
1952 return BT_INTEGER;
1954 return e->ts.type;
1958 /* Check an intrinsic arithmetic operation to see if it is consistent
1959 with some type of expression. */
1961 static gfc_try check_init_expr (gfc_expr *);
1964 /* Scalarize an expression for an elemental intrinsic call. */
1966 static gfc_try
1967 scalarize_intrinsic_call (gfc_expr *e)
1969 gfc_actual_arglist *a, *b;
1970 gfc_constructor_base ctor;
1971 gfc_constructor *args[5];
1972 gfc_constructor *ci, *new_ctor;
1973 gfc_expr *expr, *old;
1974 int n, i, rank[5], array_arg;
1976 /* Find which, if any, arguments are arrays. Assume that the old
1977 expression carries the type information and that the first arg
1978 that is an array expression carries all the shape information.*/
1979 n = array_arg = 0;
1980 a = e->value.function.actual;
1981 for (; a; a = a->next)
1983 n++;
1984 if (a->expr->expr_type != EXPR_ARRAY)
1985 continue;
1986 array_arg = n;
1987 expr = gfc_copy_expr (a->expr);
1988 break;
1991 if (!array_arg)
1992 return FAILURE;
1994 old = gfc_copy_expr (e);
1996 gfc_constructor_free (expr->value.constructor);
1997 expr->value.constructor = NULL;
1998 expr->ts = old->ts;
1999 expr->where = old->where;
2000 expr->expr_type = EXPR_ARRAY;
2002 /* Copy the array argument constructors into an array, with nulls
2003 for the scalars. */
2004 n = 0;
2005 a = old->value.function.actual;
2006 for (; a; a = a->next)
2008 /* Check that this is OK for an initialization expression. */
2009 if (a->expr && check_init_expr (a->expr) == FAILURE)
2010 goto cleanup;
2012 rank[n] = 0;
2013 if (a->expr && a->expr->rank && a->expr->expr_type == EXPR_VARIABLE)
2015 rank[n] = a->expr->rank;
2016 ctor = a->expr->symtree->n.sym->value->value.constructor;
2017 args[n] = gfc_constructor_first (ctor);
2019 else if (a->expr && a->expr->expr_type == EXPR_ARRAY)
2021 if (a->expr->rank)
2022 rank[n] = a->expr->rank;
2023 else
2024 rank[n] = 1;
2025 ctor = gfc_constructor_copy (a->expr->value.constructor);
2026 args[n] = gfc_constructor_first (ctor);
2028 else
2029 args[n] = NULL;
2031 n++;
2035 /* Using the array argument as the master, step through the array
2036 calling the function for each element and advancing the array
2037 constructors together. */
2038 for (ci = args[array_arg - 1]; ci; ci = gfc_constructor_next (ci))
2040 new_ctor = gfc_constructor_append_expr (&expr->value.constructor,
2041 gfc_copy_expr (old), NULL);
2043 gfc_free_actual_arglist (new_ctor->expr->value.function.actual);
2044 a = NULL;
2045 b = old->value.function.actual;
2046 for (i = 0; i < n; i++)
2048 if (a == NULL)
2049 new_ctor->expr->value.function.actual
2050 = a = gfc_get_actual_arglist ();
2051 else
2053 a->next = gfc_get_actual_arglist ();
2054 a = a->next;
2057 if (args[i])
2058 a->expr = gfc_copy_expr (args[i]->expr);
2059 else
2060 a->expr = gfc_copy_expr (b->expr);
2062 b = b->next;
2065 /* Simplify the function calls. If the simplification fails, the
2066 error will be flagged up down-stream or the library will deal
2067 with it. */
2068 gfc_simplify_expr (new_ctor->expr, 0);
2070 for (i = 0; i < n; i++)
2071 if (args[i])
2072 args[i] = gfc_constructor_next (args[i]);
2074 for (i = 1; i < n; i++)
2075 if (rank[i] && ((args[i] != NULL && args[array_arg - 1] == NULL)
2076 || (args[i] == NULL && args[array_arg - 1] != NULL)))
2077 goto compliance;
2080 free_expr0 (e);
2081 *e = *expr;
2082 gfc_free_expr (old);
2083 return SUCCESS;
2085 compliance:
2086 gfc_error_now ("elemental function arguments at %C are not compliant");
2088 cleanup:
2089 gfc_free_expr (expr);
2090 gfc_free_expr (old);
2091 return FAILURE;
2095 static gfc_try
2096 check_intrinsic_op (gfc_expr *e, gfc_try (*check_function) (gfc_expr *))
2098 gfc_expr *op1 = e->value.op.op1;
2099 gfc_expr *op2 = e->value.op.op2;
2101 if ((*check_function) (op1) == FAILURE)
2102 return FAILURE;
2104 switch (e->value.op.op)
2106 case INTRINSIC_UPLUS:
2107 case INTRINSIC_UMINUS:
2108 if (!numeric_type (et0 (op1)))
2109 goto not_numeric;
2110 break;
2112 case INTRINSIC_EQ:
2113 case INTRINSIC_EQ_OS:
2114 case INTRINSIC_NE:
2115 case INTRINSIC_NE_OS:
2116 case INTRINSIC_GT:
2117 case INTRINSIC_GT_OS:
2118 case INTRINSIC_GE:
2119 case INTRINSIC_GE_OS:
2120 case INTRINSIC_LT:
2121 case INTRINSIC_LT_OS:
2122 case INTRINSIC_LE:
2123 case INTRINSIC_LE_OS:
2124 if ((*check_function) (op2) == FAILURE)
2125 return FAILURE;
2127 if (!(et0 (op1) == BT_CHARACTER && et0 (op2) == BT_CHARACTER)
2128 && !(numeric_type (et0 (op1)) && numeric_type (et0 (op2))))
2130 gfc_error ("Numeric or CHARACTER operands are required in "
2131 "expression at %L", &e->where);
2132 return FAILURE;
2134 break;
2136 case INTRINSIC_PLUS:
2137 case INTRINSIC_MINUS:
2138 case INTRINSIC_TIMES:
2139 case INTRINSIC_DIVIDE:
2140 case INTRINSIC_POWER:
2141 if ((*check_function) (op2) == FAILURE)
2142 return FAILURE;
2144 if (!numeric_type (et0 (op1)) || !numeric_type (et0 (op2)))
2145 goto not_numeric;
2147 break;
2149 case INTRINSIC_CONCAT:
2150 if ((*check_function) (op2) == FAILURE)
2151 return FAILURE;
2153 if (et0 (op1) != BT_CHARACTER || et0 (op2) != BT_CHARACTER)
2155 gfc_error ("Concatenation operator in expression at %L "
2156 "must have two CHARACTER operands", &op1->where);
2157 return FAILURE;
2160 if (op1->ts.kind != op2->ts.kind)
2162 gfc_error ("Concat operator at %L must concatenate strings of the "
2163 "same kind", &e->where);
2164 return FAILURE;
2167 break;
2169 case INTRINSIC_NOT:
2170 if (et0 (op1) != BT_LOGICAL)
2172 gfc_error (".NOT. operator in expression at %L must have a LOGICAL "
2173 "operand", &op1->where);
2174 return FAILURE;
2177 break;
2179 case INTRINSIC_AND:
2180 case INTRINSIC_OR:
2181 case INTRINSIC_EQV:
2182 case INTRINSIC_NEQV:
2183 if ((*check_function) (op2) == FAILURE)
2184 return FAILURE;
2186 if (et0 (op1) != BT_LOGICAL || et0 (op2) != BT_LOGICAL)
2188 gfc_error ("LOGICAL operands are required in expression at %L",
2189 &e->where);
2190 return FAILURE;
2193 break;
2195 case INTRINSIC_PARENTHESES:
2196 break;
2198 default:
2199 gfc_error ("Only intrinsic operators can be used in expression at %L",
2200 &e->where);
2201 return FAILURE;
2204 return SUCCESS;
2206 not_numeric:
2207 gfc_error ("Numeric operands are required in expression at %L", &e->where);
2209 return FAILURE;
2212 /* F2003, 7.1.7 (3): In init expression, allocatable components
2213 must not be data-initialized. */
2214 static gfc_try
2215 check_alloc_comp_init (gfc_expr *e)
2217 gfc_component *comp;
2218 gfc_constructor *ctor;
2220 gcc_assert (e->expr_type == EXPR_STRUCTURE);
2221 gcc_assert (e->ts.type == BT_DERIVED);
2223 for (comp = e->ts.u.derived->components,
2224 ctor = gfc_constructor_first (e->value.constructor);
2225 comp; comp = comp->next, ctor = gfc_constructor_next (ctor))
2227 if (comp->attr.allocatable
2228 && ctor->expr->expr_type != EXPR_NULL)
2230 gfc_error("Invalid initialization expression for ALLOCATABLE "
2231 "component '%s' in structure constructor at %L",
2232 comp->name, &ctor->expr->where);
2233 return FAILURE;
2237 return SUCCESS;
2240 static match
2241 check_init_expr_arguments (gfc_expr *e)
2243 gfc_actual_arglist *ap;
2245 for (ap = e->value.function.actual; ap; ap = ap->next)
2246 if (check_init_expr (ap->expr) == FAILURE)
2247 return MATCH_ERROR;
2249 return MATCH_YES;
2252 static gfc_try check_restricted (gfc_expr *);
2254 /* F95, 7.1.6.1, Initialization expressions, (7)
2255 F2003, 7.1.7 Initialization expression, (8) */
2257 static match
2258 check_inquiry (gfc_expr *e, int not_restricted)
2260 const char *name;
2261 const char *const *functions;
2263 static const char *const inquiry_func_f95[] = {
2264 "lbound", "shape", "size", "ubound",
2265 "bit_size", "len", "kind",
2266 "digits", "epsilon", "huge", "maxexponent", "minexponent",
2267 "precision", "radix", "range", "tiny",
2268 NULL
2271 static const char *const inquiry_func_f2003[] = {
2272 "lbound", "shape", "size", "ubound",
2273 "bit_size", "len", "kind",
2274 "digits", "epsilon", "huge", "maxexponent", "minexponent",
2275 "precision", "radix", "range", "tiny",
2276 "new_line", NULL
2279 int i;
2280 gfc_actual_arglist *ap;
2282 if (!e->value.function.isym
2283 || !e->value.function.isym->inquiry)
2284 return MATCH_NO;
2286 /* An undeclared parameter will get us here (PR25018). */
2287 if (e->symtree == NULL)
2288 return MATCH_NO;
2290 name = e->symtree->n.sym->name;
2292 functions = (gfc_option.warn_std & GFC_STD_F2003)
2293 ? inquiry_func_f2003 : inquiry_func_f95;
2295 for (i = 0; functions[i]; i++)
2296 if (strcmp (functions[i], name) == 0)
2297 break;
2299 if (functions[i] == NULL)
2300 return MATCH_ERROR;
2302 /* At this point we have an inquiry function with a variable argument. The
2303 type of the variable might be undefined, but we need it now, because the
2304 arguments of these functions are not allowed to be undefined. */
2306 for (ap = e->value.function.actual; ap; ap = ap->next)
2308 if (!ap->expr)
2309 continue;
2311 if (ap->expr->ts.type == BT_UNKNOWN)
2313 if (ap->expr->symtree->n.sym->ts.type == BT_UNKNOWN
2314 && gfc_set_default_type (ap->expr->symtree->n.sym, 0, gfc_current_ns)
2315 == FAILURE)
2316 return MATCH_NO;
2318 ap->expr->ts = ap->expr->symtree->n.sym->ts;
2321 /* Assumed character length will not reduce to a constant expression
2322 with LEN, as required by the standard. */
2323 if (i == 5 && not_restricted
2324 && ap->expr->symtree->n.sym->ts.type == BT_CHARACTER
2325 && ap->expr->symtree->n.sym->ts.u.cl->length == NULL)
2327 gfc_error ("Assumed character length variable '%s' in constant "
2328 "expression at %L", e->symtree->n.sym->name, &e->where);
2329 return MATCH_ERROR;
2331 else if (not_restricted && check_init_expr (ap->expr) == FAILURE)
2332 return MATCH_ERROR;
2334 if (not_restricted == 0
2335 && ap->expr->expr_type != EXPR_VARIABLE
2336 && check_restricted (ap->expr) == FAILURE)
2337 return MATCH_ERROR;
2340 return MATCH_YES;
2344 /* F95, 7.1.6.1, Initialization expressions, (5)
2345 F2003, 7.1.7 Initialization expression, (5) */
2347 static match
2348 check_transformational (gfc_expr *e)
2350 static const char * const trans_func_f95[] = {
2351 "repeat", "reshape", "selected_int_kind",
2352 "selected_real_kind", "transfer", "trim", NULL
2355 static const char * const trans_func_f2003[] = {
2356 "all", "any", "count", "dot_product", "matmul", "null", "pack",
2357 "product", "repeat", "reshape", "selected_char_kind", "selected_int_kind",
2358 "selected_real_kind", "spread", "sum", "transfer", "transpose",
2359 "trim", "unpack", NULL
2362 int i;
2363 const char *name;
2364 const char *const *functions;
2366 if (!e->value.function.isym
2367 || !e->value.function.isym->transformational)
2368 return MATCH_NO;
2370 name = e->symtree->n.sym->name;
2372 functions = (gfc_option.allow_std & GFC_STD_F2003)
2373 ? trans_func_f2003 : trans_func_f95;
2375 /* NULL() is dealt with below. */
2376 if (strcmp ("null", name) == 0)
2377 return MATCH_NO;
2379 for (i = 0; functions[i]; i++)
2380 if (strcmp (functions[i], name) == 0)
2381 break;
2383 if (functions[i] == NULL)
2385 gfc_error("transformational intrinsic '%s' at %L is not permitted "
2386 "in an initialization expression", name, &e->where);
2387 return MATCH_ERROR;
2390 return check_init_expr_arguments (e);
2394 /* F95, 7.1.6.1, Initialization expressions, (6)
2395 F2003, 7.1.7 Initialization expression, (6) */
2397 static match
2398 check_null (gfc_expr *e)
2400 if (strcmp ("null", e->symtree->n.sym->name) != 0)
2401 return MATCH_NO;
2403 return check_init_expr_arguments (e);
2407 static match
2408 check_elemental (gfc_expr *e)
2410 if (!e->value.function.isym
2411 || !e->value.function.isym->elemental)
2412 return MATCH_NO;
2414 if (e->ts.type != BT_INTEGER
2415 && e->ts.type != BT_CHARACTER
2416 && gfc_notify_std (GFC_STD_F2003, "Extension: Evaluation of "
2417 "nonstandard initialization expression at %L",
2418 &e->where) == FAILURE)
2419 return MATCH_ERROR;
2421 return check_init_expr_arguments (e);
2425 static match
2426 check_conversion (gfc_expr *e)
2428 if (!e->value.function.isym
2429 || !e->value.function.isym->conversion)
2430 return MATCH_NO;
2432 return check_init_expr_arguments (e);
2436 /* Verify that an expression is an initialization expression. A side
2437 effect is that the expression tree is reduced to a single constant
2438 node if all goes well. This would normally happen when the
2439 expression is constructed but function references are assumed to be
2440 intrinsics in the context of initialization expressions. If
2441 FAILURE is returned an error message has been generated. */
2443 static gfc_try
2444 check_init_expr (gfc_expr *e)
2446 match m;
2447 gfc_try t;
2449 if (e == NULL)
2450 return SUCCESS;
2452 switch (e->expr_type)
2454 case EXPR_OP:
2455 t = check_intrinsic_op (e, check_init_expr);
2456 if (t == SUCCESS)
2457 t = gfc_simplify_expr (e, 0);
2459 break;
2461 case EXPR_FUNCTION:
2462 t = FAILURE;
2465 gfc_intrinsic_sym* isym;
2466 gfc_symbol* sym;
2468 sym = e->symtree->n.sym;
2469 if (!gfc_is_intrinsic (sym, 0, e->where)
2470 || (m = gfc_intrinsic_func_interface (e, 0)) != MATCH_YES)
2472 gfc_error ("Function '%s' in initialization expression at %L "
2473 "must be an intrinsic function",
2474 e->symtree->n.sym->name, &e->where);
2475 break;
2478 if ((m = check_conversion (e)) == MATCH_NO
2479 && (m = check_inquiry (e, 1)) == MATCH_NO
2480 && (m = check_null (e)) == MATCH_NO
2481 && (m = check_transformational (e)) == MATCH_NO
2482 && (m = check_elemental (e)) == MATCH_NO)
2484 gfc_error ("Intrinsic function '%s' at %L is not permitted "
2485 "in an initialization expression",
2486 e->symtree->n.sym->name, &e->where);
2487 m = MATCH_ERROR;
2490 /* Try to scalarize an elemental intrinsic function that has an
2491 array argument. */
2492 isym = gfc_find_function (e->symtree->n.sym->name);
2493 if (isym && isym->elemental
2494 && (t = scalarize_intrinsic_call (e)) == SUCCESS)
2495 break;
2498 if (m == MATCH_YES)
2499 t = gfc_simplify_expr (e, 0);
2501 break;
2503 case EXPR_VARIABLE:
2504 t = SUCCESS;
2506 if (gfc_check_iter_variable (e) == SUCCESS)
2507 break;
2509 if (e->symtree->n.sym->attr.flavor == FL_PARAMETER)
2511 /* A PARAMETER shall not be used to define itself, i.e.
2512 REAL, PARAMETER :: x = transfer(0, x)
2513 is invalid. */
2514 if (!e->symtree->n.sym->value)
2516 gfc_error("PARAMETER '%s' is used at %L before its definition "
2517 "is complete", e->symtree->n.sym->name, &e->where);
2518 t = FAILURE;
2520 else
2521 t = simplify_parameter_variable (e, 0);
2523 break;
2526 if (gfc_in_match_data ())
2527 break;
2529 t = FAILURE;
2531 if (e->symtree->n.sym->as)
2533 switch (e->symtree->n.sym->as->type)
2535 case AS_ASSUMED_SIZE:
2536 gfc_error ("Assumed size array '%s' at %L is not permitted "
2537 "in an initialization expression",
2538 e->symtree->n.sym->name, &e->where);
2539 break;
2541 case AS_ASSUMED_SHAPE:
2542 gfc_error ("Assumed shape array '%s' at %L is not permitted "
2543 "in an initialization expression",
2544 e->symtree->n.sym->name, &e->where);
2545 break;
2547 case AS_DEFERRED:
2548 gfc_error ("Deferred array '%s' at %L is not permitted "
2549 "in an initialization expression",
2550 e->symtree->n.sym->name, &e->where);
2551 break;
2553 case AS_EXPLICIT:
2554 gfc_error ("Array '%s' at %L is a variable, which does "
2555 "not reduce to a constant expression",
2556 e->symtree->n.sym->name, &e->where);
2557 break;
2559 default:
2560 gcc_unreachable();
2563 else
2564 gfc_error ("Parameter '%s' at %L has not been declared or is "
2565 "a variable, which does not reduce to a constant "
2566 "expression", e->symtree->n.sym->name, &e->where);
2568 break;
2570 case EXPR_CONSTANT:
2571 case EXPR_NULL:
2572 t = SUCCESS;
2573 break;
2575 case EXPR_SUBSTRING:
2576 t = check_init_expr (e->ref->u.ss.start);
2577 if (t == FAILURE)
2578 break;
2580 t = check_init_expr (e->ref->u.ss.end);
2581 if (t == SUCCESS)
2582 t = gfc_simplify_expr (e, 0);
2584 break;
2586 case EXPR_STRUCTURE:
2587 t = e->ts.is_iso_c ? SUCCESS : FAILURE;
2588 if (t == SUCCESS)
2589 break;
2591 t = check_alloc_comp_init (e);
2592 if (t == FAILURE)
2593 break;
2595 t = gfc_check_constructor (e, check_init_expr);
2596 if (t == FAILURE)
2597 break;
2599 break;
2601 case EXPR_ARRAY:
2602 t = gfc_check_constructor (e, check_init_expr);
2603 if (t == FAILURE)
2604 break;
2606 t = gfc_expand_constructor (e);
2607 if (t == FAILURE)
2608 break;
2610 t = gfc_check_constructor_type (e);
2611 break;
2613 default:
2614 gfc_internal_error ("check_init_expr(): Unknown expression type");
2617 return t;
2620 /* Reduces a general expression to an initialization expression (a constant).
2621 This used to be part of gfc_match_init_expr.
2622 Note that this function doesn't free the given expression on FAILURE. */
2624 gfc_try
2625 gfc_reduce_init_expr (gfc_expr *expr)
2627 gfc_try t;
2629 gfc_init_expr = 1;
2630 t = gfc_resolve_expr (expr);
2631 if (t == SUCCESS)
2632 t = check_init_expr (expr);
2633 gfc_init_expr = 0;
2635 if (t == FAILURE)
2636 return FAILURE;
2638 if (expr->expr_type == EXPR_ARRAY)
2640 if (gfc_check_constructor_type (expr) == FAILURE)
2641 return FAILURE;
2642 if (gfc_expand_constructor (expr) == FAILURE)
2643 return FAILURE;
2646 return SUCCESS;
2650 /* Match an initialization expression. We work by first matching an
2651 expression, then reducing it to a constant. The reducing it to
2652 constant part requires a global variable to flag the prohibition
2653 of a non-integer exponent in -std=f95 mode. */
2655 bool init_flag = false;
2657 match
2658 gfc_match_init_expr (gfc_expr **result)
2660 gfc_expr *expr;
2661 match m;
2662 gfc_try t;
2664 expr = NULL;
2666 init_flag = true;
2668 m = gfc_match_expr (&expr);
2669 if (m != MATCH_YES)
2671 init_flag = false;
2672 return m;
2675 t = gfc_reduce_init_expr (expr);
2676 if (t != SUCCESS)
2678 gfc_free_expr (expr);
2679 init_flag = false;
2680 return MATCH_ERROR;
2683 *result = expr;
2684 init_flag = false;
2686 return MATCH_YES;
2690 /* Given an actual argument list, test to see that each argument is a
2691 restricted expression and optionally if the expression type is
2692 integer or character. */
2694 static gfc_try
2695 restricted_args (gfc_actual_arglist *a)
2697 for (; a; a = a->next)
2699 if (check_restricted (a->expr) == FAILURE)
2700 return FAILURE;
2703 return SUCCESS;
2707 /************* Restricted/specification expressions *************/
2710 /* Make sure a non-intrinsic function is a specification function. */
2712 static gfc_try
2713 external_spec_function (gfc_expr *e)
2715 gfc_symbol *f;
2717 f = e->value.function.esym;
2719 if (f->attr.proc == PROC_ST_FUNCTION)
2721 gfc_error ("Specification function '%s' at %L cannot be a statement "
2722 "function", f->name, &e->where);
2723 return FAILURE;
2726 if (f->attr.proc == PROC_INTERNAL)
2728 gfc_error ("Specification function '%s' at %L cannot be an internal "
2729 "function", f->name, &e->where);
2730 return FAILURE;
2733 if (!f->attr.pure && !f->attr.elemental)
2735 gfc_error ("Specification function '%s' at %L must be PURE", f->name,
2736 &e->where);
2737 return FAILURE;
2740 if (f->attr.recursive)
2742 gfc_error ("Specification function '%s' at %L cannot be RECURSIVE",
2743 f->name, &e->where);
2744 return FAILURE;
2747 return restricted_args (e->value.function.actual);
2751 /* Check to see that a function reference to an intrinsic is a
2752 restricted expression. */
2754 static gfc_try
2755 restricted_intrinsic (gfc_expr *e)
2757 /* TODO: Check constraints on inquiry functions. 7.1.6.2 (7). */
2758 if (check_inquiry (e, 0) == MATCH_YES)
2759 return SUCCESS;
2761 return restricted_args (e->value.function.actual);
2765 /* Check the expressions of an actual arglist. Used by check_restricted. */
2767 static gfc_try
2768 check_arglist (gfc_actual_arglist* arg, gfc_try (*checker) (gfc_expr*))
2770 for (; arg; arg = arg->next)
2771 if (checker (arg->expr) == FAILURE)
2772 return FAILURE;
2774 return SUCCESS;
2778 /* Check the subscription expressions of a reference chain with a checking
2779 function; used by check_restricted. */
2781 static gfc_try
2782 check_references (gfc_ref* ref, gfc_try (*checker) (gfc_expr*))
2784 int dim;
2786 if (!ref)
2787 return SUCCESS;
2789 switch (ref->type)
2791 case REF_ARRAY:
2792 for (dim = 0; dim != ref->u.ar.dimen; ++dim)
2794 if (checker (ref->u.ar.start[dim]) == FAILURE)
2795 return FAILURE;
2796 if (checker (ref->u.ar.end[dim]) == FAILURE)
2797 return FAILURE;
2798 if (checker (ref->u.ar.stride[dim]) == FAILURE)
2799 return FAILURE;
2801 break;
2803 case REF_COMPONENT:
2804 /* Nothing needed, just proceed to next reference. */
2805 break;
2807 case REF_SUBSTRING:
2808 if (checker (ref->u.ss.start) == FAILURE)
2809 return FAILURE;
2810 if (checker (ref->u.ss.end) == FAILURE)
2811 return FAILURE;
2812 break;
2814 default:
2815 gcc_unreachable ();
2816 break;
2819 return check_references (ref->next, checker);
2823 /* Verify that an expression is a restricted expression. Like its
2824 cousin check_init_expr(), an error message is generated if we
2825 return FAILURE. */
2827 static gfc_try
2828 check_restricted (gfc_expr *e)
2830 gfc_symbol* sym;
2831 gfc_try t;
2833 if (e == NULL)
2834 return SUCCESS;
2836 switch (e->expr_type)
2838 case EXPR_OP:
2839 t = check_intrinsic_op (e, check_restricted);
2840 if (t == SUCCESS)
2841 t = gfc_simplify_expr (e, 0);
2843 break;
2845 case EXPR_FUNCTION:
2846 if (e->value.function.esym)
2848 t = check_arglist (e->value.function.actual, &check_restricted);
2849 if (t == SUCCESS)
2850 t = external_spec_function (e);
2852 else
2854 if (e->value.function.isym && e->value.function.isym->inquiry)
2855 t = SUCCESS;
2856 else
2857 t = check_arglist (e->value.function.actual, &check_restricted);
2859 if (t == SUCCESS)
2860 t = restricted_intrinsic (e);
2862 break;
2864 case EXPR_VARIABLE:
2865 sym = e->symtree->n.sym;
2866 t = FAILURE;
2868 /* If a dummy argument appears in a context that is valid for a
2869 restricted expression in an elemental procedure, it will have
2870 already been simplified away once we get here. Therefore we
2871 don't need to jump through hoops to distinguish valid from
2872 invalid cases. */
2873 if (sym->attr.dummy && sym->ns == gfc_current_ns
2874 && sym->ns->proc_name && sym->ns->proc_name->attr.elemental)
2876 gfc_error ("Dummy argument '%s' not allowed in expression at %L",
2877 sym->name, &e->where);
2878 break;
2881 if (sym->attr.optional)
2883 gfc_error ("Dummy argument '%s' at %L cannot be OPTIONAL",
2884 sym->name, &e->where);
2885 break;
2888 if (sym->attr.intent == INTENT_OUT)
2890 gfc_error ("Dummy argument '%s' at %L cannot be INTENT(OUT)",
2891 sym->name, &e->where);
2892 break;
2895 /* Check reference chain if any. */
2896 if (check_references (e->ref, &check_restricted) == FAILURE)
2897 break;
2899 /* gfc_is_formal_arg broadcasts that a formal argument list is being
2900 processed in resolve.c(resolve_formal_arglist). This is done so
2901 that host associated dummy array indices are accepted (PR23446).
2902 This mechanism also does the same for the specification expressions
2903 of array-valued functions. */
2904 if (e->error
2905 || sym->attr.in_common
2906 || sym->attr.use_assoc
2907 || sym->attr.dummy
2908 || sym->attr.implied_index
2909 || sym->attr.flavor == FL_PARAMETER
2910 || (sym->ns && sym->ns == gfc_current_ns->parent)
2911 || (sym->ns && gfc_current_ns->parent
2912 && sym->ns == gfc_current_ns->parent->parent)
2913 || (sym->ns->proc_name != NULL
2914 && sym->ns->proc_name->attr.flavor == FL_MODULE)
2915 || (gfc_is_formal_arg () && (sym->ns == gfc_current_ns)))
2917 t = SUCCESS;
2918 break;
2921 gfc_error ("Variable '%s' cannot appear in the expression at %L",
2922 sym->name, &e->where);
2923 /* Prevent a repetition of the error. */
2924 e->error = 1;
2925 break;
2927 case EXPR_NULL:
2928 case EXPR_CONSTANT:
2929 t = SUCCESS;
2930 break;
2932 case EXPR_SUBSTRING:
2933 t = gfc_specification_expr (e->ref->u.ss.start);
2934 if (t == FAILURE)
2935 break;
2937 t = gfc_specification_expr (e->ref->u.ss.end);
2938 if (t == SUCCESS)
2939 t = gfc_simplify_expr (e, 0);
2941 break;
2943 case EXPR_STRUCTURE:
2944 t = gfc_check_constructor (e, check_restricted);
2945 break;
2947 case EXPR_ARRAY:
2948 t = gfc_check_constructor (e, check_restricted);
2949 break;
2951 default:
2952 gfc_internal_error ("check_restricted(): Unknown expression type");
2955 return t;
2959 /* Check to see that an expression is a specification expression. If
2960 we return FAILURE, an error has been generated. */
2962 gfc_try
2963 gfc_specification_expr (gfc_expr *e)
2965 gfc_component *comp;
2967 if (e == NULL)
2968 return SUCCESS;
2970 if (e->ts.type != BT_INTEGER)
2972 gfc_error ("Expression at %L must be of INTEGER type, found %s",
2973 &e->where, gfc_basic_typename (e->ts.type));
2974 return FAILURE;
2977 if (e->expr_type == EXPR_FUNCTION
2978 && !e->value.function.isym
2979 && !e->value.function.esym
2980 && !gfc_pure (e->symtree->n.sym)
2981 && (!gfc_is_proc_ptr_comp (e, &comp)
2982 || !comp->attr.pure))
2984 gfc_error ("Function '%s' at %L must be PURE",
2985 e->symtree->n.sym->name, &e->where);
2986 /* Prevent repeat error messages. */
2987 e->symtree->n.sym->attr.pure = 1;
2988 return FAILURE;
2991 if (e->rank != 0)
2993 gfc_error ("Expression at %L must be scalar", &e->where);
2994 return FAILURE;
2997 if (gfc_simplify_expr (e, 0) == FAILURE)
2998 return FAILURE;
3000 return check_restricted (e);
3004 /************** Expression conformance checks. *************/
3006 /* Given two expressions, make sure that the arrays are conformable. */
3008 gfc_try
3009 gfc_check_conformance (gfc_expr *op1, gfc_expr *op2, const char *optype_msgid, ...)
3011 int op1_flag, op2_flag, d;
3012 mpz_t op1_size, op2_size;
3013 gfc_try t;
3015 va_list argp;
3016 char buffer[240];
3018 if (op1->rank == 0 || op2->rank == 0)
3019 return SUCCESS;
3021 va_start (argp, optype_msgid);
3022 vsnprintf (buffer, 240, optype_msgid, argp);
3023 va_end (argp);
3025 if (op1->rank != op2->rank)
3027 gfc_error ("Incompatible ranks in %s (%d and %d) at %L", _(buffer),
3028 op1->rank, op2->rank, &op1->where);
3029 return FAILURE;
3032 t = SUCCESS;
3034 for (d = 0; d < op1->rank; d++)
3036 op1_flag = gfc_array_dimen_size (op1, d, &op1_size) == SUCCESS;
3037 op2_flag = gfc_array_dimen_size (op2, d, &op2_size) == SUCCESS;
3039 if (op1_flag && op2_flag && mpz_cmp (op1_size, op2_size) != 0)
3041 gfc_error ("Different shape for %s at %L on dimension %d "
3042 "(%d and %d)", _(buffer), &op1->where, d + 1,
3043 (int) mpz_get_si (op1_size),
3044 (int) mpz_get_si (op2_size));
3046 t = FAILURE;
3049 if (op1_flag)
3050 mpz_clear (op1_size);
3051 if (op2_flag)
3052 mpz_clear (op2_size);
3054 if (t == FAILURE)
3055 return FAILURE;
3058 return SUCCESS;
3062 /* Given an assignable expression and an arbitrary expression, make
3063 sure that the assignment can take place. */
3065 gfc_try
3066 gfc_check_assign (gfc_expr *lvalue, gfc_expr *rvalue, int conform)
3068 gfc_symbol *sym;
3069 gfc_ref *ref;
3070 int has_pointer;
3072 sym = lvalue->symtree->n.sym;
3074 /* Check INTENT(IN), unless the object itself is the component or
3075 sub-component of a pointer. */
3076 has_pointer = sym->attr.pointer;
3078 for (ref = lvalue->ref; ref; ref = ref->next)
3079 if (ref->type == REF_COMPONENT && ref->u.c.component->attr.pointer)
3081 has_pointer = 1;
3082 break;
3085 if (!has_pointer && sym->attr.intent == INTENT_IN)
3087 gfc_error ("Cannot assign to INTENT(IN) variable '%s' at %L",
3088 sym->name, &lvalue->where);
3089 return FAILURE;
3092 /* 12.5.2.2, Note 12.26: The result variable is very similar to any other
3093 variable local to a function subprogram. Its existence begins when
3094 execution of the function is initiated and ends when execution of the
3095 function is terminated...
3096 Therefore, the left hand side is no longer a variable, when it is: */
3097 if (sym->attr.flavor == FL_PROCEDURE && sym->attr.proc != PROC_ST_FUNCTION
3098 && !sym->attr.external)
3100 bool bad_proc;
3101 bad_proc = false;
3103 /* (i) Use associated; */
3104 if (sym->attr.use_assoc)
3105 bad_proc = true;
3107 /* (ii) The assignment is in the main program; or */
3108 if (gfc_current_ns->proc_name->attr.is_main_program)
3109 bad_proc = true;
3111 /* (iii) A module or internal procedure... */
3112 if ((gfc_current_ns->proc_name->attr.proc == PROC_INTERNAL
3113 || gfc_current_ns->proc_name->attr.proc == PROC_MODULE)
3114 && gfc_current_ns->parent
3115 && (!(gfc_current_ns->parent->proc_name->attr.function
3116 || gfc_current_ns->parent->proc_name->attr.subroutine)
3117 || gfc_current_ns->parent->proc_name->attr.is_main_program))
3119 /* ... that is not a function... */
3120 if (!gfc_current_ns->proc_name->attr.function)
3121 bad_proc = true;
3123 /* ... or is not an entry and has a different name. */
3124 if (!sym->attr.entry && sym->name != gfc_current_ns->proc_name->name)
3125 bad_proc = true;
3128 /* (iv) Host associated and not the function symbol or the
3129 parent result. This picks up sibling references, which
3130 cannot be entries. */
3131 if (!sym->attr.entry
3132 && sym->ns == gfc_current_ns->parent
3133 && sym != gfc_current_ns->proc_name
3134 && sym != gfc_current_ns->parent->proc_name->result)
3135 bad_proc = true;
3137 if (bad_proc)
3139 gfc_error ("'%s' at %L is not a VALUE", sym->name, &lvalue->where);
3140 return FAILURE;
3144 if (rvalue->rank != 0 && lvalue->rank != rvalue->rank)
3146 gfc_error ("Incompatible ranks %d and %d in assignment at %L",
3147 lvalue->rank, rvalue->rank, &lvalue->where);
3148 return FAILURE;
3151 if (lvalue->ts.type == BT_UNKNOWN)
3153 gfc_error ("Variable type is UNKNOWN in assignment at %L",
3154 &lvalue->where);
3155 return FAILURE;
3158 if (rvalue->expr_type == EXPR_NULL)
3160 if (has_pointer && (ref == NULL || ref->next == NULL)
3161 && lvalue->symtree->n.sym->attr.data)
3162 return SUCCESS;
3163 else
3165 gfc_error ("NULL appears on right-hand side in assignment at %L",
3166 &rvalue->where);
3167 return FAILURE;
3171 /* This is possibly a typo: x = f() instead of x => f(). */
3172 if (gfc_option.warn_surprising
3173 && rvalue->expr_type == EXPR_FUNCTION
3174 && rvalue->symtree->n.sym->attr.pointer)
3175 gfc_warning ("POINTER valued function appears on right-hand side of "
3176 "assignment at %L", &rvalue->where);
3178 /* Check size of array assignments. */
3179 if (lvalue->rank != 0 && rvalue->rank != 0
3180 && gfc_check_conformance (lvalue, rvalue, "array assignment") != SUCCESS)
3181 return FAILURE;
3183 if (rvalue->is_boz && lvalue->ts.type != BT_INTEGER
3184 && lvalue->symtree->n.sym->attr.data
3185 && gfc_notify_std (GFC_STD_GNU, "Extension: BOZ literal at %L used to "
3186 "initialize non-integer variable '%s'",
3187 &rvalue->where, lvalue->symtree->n.sym->name)
3188 == FAILURE)
3189 return FAILURE;
3190 else if (rvalue->is_boz && !lvalue->symtree->n.sym->attr.data
3191 && gfc_notify_std (GFC_STD_GNU, "Extension: BOZ literal at %L outside "
3192 "a DATA statement and outside INT/REAL/DBLE/CMPLX",
3193 &rvalue->where) == FAILURE)
3194 return FAILURE;
3196 /* Handle the case of a BOZ literal on the RHS. */
3197 if (rvalue->is_boz && lvalue->ts.type != BT_INTEGER)
3199 int rc;
3200 if (gfc_option.warn_surprising)
3201 gfc_warning ("BOZ literal at %L is bitwise transferred "
3202 "non-integer symbol '%s'", &rvalue->where,
3203 lvalue->symtree->n.sym->name);
3204 if (!gfc_convert_boz (rvalue, &lvalue->ts))
3205 return FAILURE;
3206 if ((rc = gfc_range_check (rvalue)) != ARITH_OK)
3208 if (rc == ARITH_UNDERFLOW)
3209 gfc_error ("Arithmetic underflow of bit-wise transferred BOZ at %L"
3210 ". This check can be disabled with the option "
3211 "-fno-range-check", &rvalue->where);
3212 else if (rc == ARITH_OVERFLOW)
3213 gfc_error ("Arithmetic overflow of bit-wise transferred BOZ at %L"
3214 ". This check can be disabled with the option "
3215 "-fno-range-check", &rvalue->where);
3216 else if (rc == ARITH_NAN)
3217 gfc_error ("Arithmetic NaN of bit-wise transferred BOZ at %L"
3218 ". This check can be disabled with the option "
3219 "-fno-range-check", &rvalue->where);
3220 return FAILURE;
3224 if (gfc_compare_types (&lvalue->ts, &rvalue->ts))
3225 return SUCCESS;
3227 /* Only DATA Statements come here. */
3228 if (!conform)
3230 /* Numeric can be converted to any other numeric. And Hollerith can be
3231 converted to any other type. */
3232 if ((gfc_numeric_ts (&lvalue->ts) && gfc_numeric_ts (&rvalue->ts))
3233 || rvalue->ts.type == BT_HOLLERITH)
3234 return SUCCESS;
3236 if (lvalue->ts.type == BT_LOGICAL && rvalue->ts.type == BT_LOGICAL)
3237 return SUCCESS;
3239 gfc_error ("Incompatible types in DATA statement at %L; attempted "
3240 "conversion of %s to %s", &lvalue->where,
3241 gfc_typename (&rvalue->ts), gfc_typename (&lvalue->ts));
3243 return FAILURE;
3246 /* Assignment is the only case where character variables of different
3247 kind values can be converted into one another. */
3248 if (lvalue->ts.type == BT_CHARACTER && rvalue->ts.type == BT_CHARACTER)
3250 if (lvalue->ts.kind != rvalue->ts.kind)
3251 gfc_convert_chartype (rvalue, &lvalue->ts);
3253 return SUCCESS;
3256 return gfc_convert_type (rvalue, &lvalue->ts, 1);
3260 /* Check that a pointer assignment is OK. We first check lvalue, and
3261 we only check rvalue if it's not an assignment to NULL() or a
3262 NULLIFY statement. */
3264 gfc_try
3265 gfc_check_pointer_assign (gfc_expr *lvalue, gfc_expr *rvalue)
3267 symbol_attribute attr;
3268 gfc_ref *ref;
3269 int is_pure;
3270 int pointer, check_intent_in, proc_pointer;
3272 if (lvalue->symtree->n.sym->ts.type == BT_UNKNOWN
3273 && !lvalue->symtree->n.sym->attr.proc_pointer)
3275 gfc_error ("Pointer assignment target is not a POINTER at %L",
3276 &lvalue->where);
3277 return FAILURE;
3280 if (lvalue->symtree->n.sym->attr.flavor == FL_PROCEDURE
3281 && lvalue->symtree->n.sym->attr.use_assoc
3282 && !lvalue->symtree->n.sym->attr.proc_pointer)
3284 gfc_error ("'%s' in the pointer assignment at %L cannot be an "
3285 "l-value since it is a procedure",
3286 lvalue->symtree->n.sym->name, &lvalue->where);
3287 return FAILURE;
3291 /* Check INTENT(IN), unless the object itself is the component or
3292 sub-component of a pointer. */
3293 check_intent_in = 1;
3294 pointer = lvalue->symtree->n.sym->attr.pointer;
3295 proc_pointer = lvalue->symtree->n.sym->attr.proc_pointer;
3297 for (ref = lvalue->ref; ref; ref = ref->next)
3299 if (pointer)
3300 check_intent_in = 0;
3302 if (ref->type == REF_COMPONENT)
3304 pointer = ref->u.c.component->attr.pointer;
3305 proc_pointer = ref->u.c.component->attr.proc_pointer;
3308 if (ref->type == REF_ARRAY && ref->next == NULL)
3310 if (ref->u.ar.type == AR_FULL)
3311 break;
3313 if (ref->u.ar.type != AR_SECTION)
3315 gfc_error ("Expected bounds specification for '%s' at %L",
3316 lvalue->symtree->n.sym->name, &lvalue->where);
3317 return FAILURE;
3320 if (gfc_notify_std (GFC_STD_F2003,"Fortran 2003: Bounds "
3321 "specification for '%s' in pointer assignment "
3322 "at %L", lvalue->symtree->n.sym->name,
3323 &lvalue->where) == FAILURE)
3324 return FAILURE;
3326 gfc_error ("Pointer bounds remapping at %L is not yet implemented "
3327 "in gfortran", &lvalue->where);
3328 /* TODO: See PR 29785. Add checks that all lbounds are specified and
3329 either never or always the upper-bound; strides shall not be
3330 present. */
3331 return FAILURE;
3335 if (check_intent_in && lvalue->symtree->n.sym->attr.intent == INTENT_IN)
3337 gfc_error ("Cannot assign to INTENT(IN) variable '%s' at %L",
3338 lvalue->symtree->n.sym->name, &lvalue->where);
3339 return FAILURE;
3342 if (!pointer && !proc_pointer
3343 && !(lvalue->ts.type == BT_CLASS
3344 && lvalue->ts.u.derived->components->attr.pointer))
3346 gfc_error ("Pointer assignment to non-POINTER at %L", &lvalue->where);
3347 return FAILURE;
3350 is_pure = gfc_pure (NULL);
3352 if (is_pure && gfc_impure_variable (lvalue->symtree->n.sym)
3353 && lvalue->symtree->n.sym->value != rvalue)
3355 gfc_error ("Bad pointer object in PURE procedure at %L", &lvalue->where);
3356 return FAILURE;
3359 /* If rvalue is a NULL() or NULLIFY, we're done. Otherwise the type,
3360 kind, etc for lvalue and rvalue must match, and rvalue must be a
3361 pure variable if we're in a pure function. */
3362 if (rvalue->expr_type == EXPR_NULL && rvalue->ts.type == BT_UNKNOWN)
3363 return SUCCESS;
3365 /* F2008, C723 (pointer) and C726 (proc-pointer); for PURE also C1283. */
3366 if (lvalue->expr_type == EXPR_VARIABLE
3367 && gfc_is_coindexed (lvalue))
3369 gfc_ref *ref;
3370 for (ref = lvalue->ref; ref; ref = ref->next)
3371 if (ref->type == REF_ARRAY && ref->u.ar.codimen)
3373 gfc_error ("Pointer object at %L shall not have a coindex",
3374 &lvalue->where);
3375 return FAILURE;
3379 /* Checks on rvalue for procedure pointer assignments. */
3380 if (proc_pointer)
3382 char err[200];
3383 gfc_symbol *s1,*s2;
3384 gfc_component *comp;
3385 const char *name;
3387 attr = gfc_expr_attr (rvalue);
3388 if (!((rvalue->expr_type == EXPR_NULL)
3389 || (rvalue->expr_type == EXPR_FUNCTION && attr.proc_pointer)
3390 || (rvalue->expr_type == EXPR_VARIABLE && attr.proc_pointer)
3391 || (rvalue->expr_type == EXPR_VARIABLE
3392 && attr.flavor == FL_PROCEDURE)))
3394 gfc_error ("Invalid procedure pointer assignment at %L",
3395 &rvalue->where);
3396 return FAILURE;
3398 if (attr.abstract)
3400 gfc_error ("Abstract interface '%s' is invalid "
3401 "in procedure pointer assignment at %L",
3402 rvalue->symtree->name, &rvalue->where);
3403 return FAILURE;
3405 /* Check for C727. */
3406 if (attr.flavor == FL_PROCEDURE)
3408 if (attr.proc == PROC_ST_FUNCTION)
3410 gfc_error ("Statement function '%s' is invalid "
3411 "in procedure pointer assignment at %L",
3412 rvalue->symtree->name, &rvalue->where);
3413 return FAILURE;
3415 if (attr.proc == PROC_INTERNAL &&
3416 gfc_notify_std (GFC_STD_F2008, "Internal procedure '%s' is "
3417 "invalid in procedure pointer assignment at %L",
3418 rvalue->symtree->name, &rvalue->where) == FAILURE)
3419 return FAILURE;
3422 /* Ensure that the calling convention is the same. As other attributes
3423 such as DLLEXPORT may differ, one explicitly only tests for the
3424 calling conventions. */
3425 if (rvalue->expr_type == EXPR_VARIABLE
3426 && lvalue->symtree->n.sym->attr.ext_attr
3427 != rvalue->symtree->n.sym->attr.ext_attr)
3429 symbol_attribute calls;
3431 calls.ext_attr = 0;
3432 gfc_add_ext_attribute (&calls, EXT_ATTR_CDECL, NULL);
3433 gfc_add_ext_attribute (&calls, EXT_ATTR_STDCALL, NULL);
3434 gfc_add_ext_attribute (&calls, EXT_ATTR_FASTCALL, NULL);
3436 if ((calls.ext_attr & lvalue->symtree->n.sym->attr.ext_attr)
3437 != (calls.ext_attr & rvalue->symtree->n.sym->attr.ext_attr))
3439 gfc_error ("Mismatch in the procedure pointer assignment "
3440 "at %L: mismatch in the calling convention",
3441 &rvalue->where);
3442 return FAILURE;
3446 if (gfc_is_proc_ptr_comp (lvalue, &comp))
3447 s1 = comp->ts.interface;
3448 else
3449 s1 = lvalue->symtree->n.sym;
3451 if (gfc_is_proc_ptr_comp (rvalue, &comp))
3453 s2 = comp->ts.interface;
3454 name = comp->name;
3456 else if (rvalue->expr_type == EXPR_FUNCTION)
3458 s2 = rvalue->symtree->n.sym->result;
3459 name = rvalue->symtree->n.sym->result->name;
3461 else
3463 s2 = rvalue->symtree->n.sym;
3464 name = rvalue->symtree->n.sym->name;
3467 if (s1 && s2 && !gfc_compare_interfaces (s1, s2, name, 0, 1,
3468 err, sizeof(err)))
3470 gfc_error ("Interface mismatch in procedure pointer assignment "
3471 "at %L: %s", &rvalue->where, err);
3472 return FAILURE;
3475 return SUCCESS;
3478 if (!gfc_compare_types (&lvalue->ts, &rvalue->ts))
3480 gfc_error ("Different types in pointer assignment at %L; attempted "
3481 "assignment of %s to %s", &lvalue->where,
3482 gfc_typename (&rvalue->ts), gfc_typename (&lvalue->ts));
3483 return FAILURE;
3486 if (lvalue->ts.type != BT_CLASS && lvalue->ts.kind != rvalue->ts.kind)
3488 gfc_error ("Different kind type parameters in pointer "
3489 "assignment at %L", &lvalue->where);
3490 return FAILURE;
3493 if (lvalue->rank != rvalue->rank)
3495 gfc_error ("Different ranks in pointer assignment at %L",
3496 &lvalue->where);
3497 return FAILURE;
3500 /* Now punt if we are dealing with a NULLIFY(X) or X = NULL(X). */
3501 if (rvalue->expr_type == EXPR_NULL)
3502 return SUCCESS;
3504 if (lvalue->ts.type == BT_CHARACTER)
3506 gfc_try t = gfc_check_same_strlen (lvalue, rvalue, "pointer assignment");
3507 if (t == FAILURE)
3508 return FAILURE;
3511 if (rvalue->expr_type == EXPR_VARIABLE && is_subref_array (rvalue))
3512 lvalue->symtree->n.sym->attr.subref_array_pointer = 1;
3514 attr = gfc_expr_attr (rvalue);
3515 if (!attr.target && !attr.pointer)
3517 gfc_error ("Pointer assignment target is neither TARGET "
3518 "nor POINTER at %L", &rvalue->where);
3519 return FAILURE;
3522 if (is_pure && gfc_impure_variable (rvalue->symtree->n.sym))
3524 gfc_error ("Bad target in pointer assignment in PURE "
3525 "procedure at %L", &rvalue->where);
3528 if (gfc_has_vector_index (rvalue))
3530 gfc_error ("Pointer assignment with vector subscript "
3531 "on rhs at %L", &rvalue->where);
3532 return FAILURE;
3535 if (attr.is_protected && attr.use_assoc
3536 && !(attr.pointer || attr.proc_pointer))
3538 gfc_error ("Pointer assignment target has PROTECTED "
3539 "attribute at %L", &rvalue->where);
3540 return FAILURE;
3543 /* F2008, C725. For PURE also C1283. */
3544 if (rvalue->expr_type == EXPR_VARIABLE
3545 && gfc_is_coindexed (rvalue))
3547 gfc_ref *ref;
3548 for (ref = rvalue->ref; ref; ref = ref->next)
3549 if (ref->type == REF_ARRAY && ref->u.ar.codimen)
3551 gfc_error ("Data target at %L shall not have a coindex",
3552 &rvalue->where);
3553 return FAILURE;
3557 return SUCCESS;
3561 /* Relative of gfc_check_assign() except that the lvalue is a single
3562 symbol. Used for initialization assignments. */
3564 gfc_try
3565 gfc_check_assign_symbol (gfc_symbol *sym, gfc_expr *rvalue)
3567 gfc_expr lvalue;
3568 gfc_try r;
3570 memset (&lvalue, '\0', sizeof (gfc_expr));
3572 lvalue.expr_type = EXPR_VARIABLE;
3573 lvalue.ts = sym->ts;
3574 if (sym->as)
3575 lvalue.rank = sym->as->rank;
3576 lvalue.symtree = (gfc_symtree *) gfc_getmem (sizeof (gfc_symtree));
3577 lvalue.symtree->n.sym = sym;
3578 lvalue.where = sym->declared_at;
3580 if (sym->attr.pointer || sym->attr.proc_pointer
3581 || (sym->ts.type == BT_CLASS
3582 && sym->ts.u.derived->components->attr.pointer
3583 && rvalue->expr_type == EXPR_NULL))
3584 r = gfc_check_pointer_assign (&lvalue, rvalue);
3585 else
3586 r = gfc_check_assign (&lvalue, rvalue, 1);
3588 gfc_free (lvalue.symtree);
3590 return r;
3594 /* Get an expression for a default initializer. */
3596 gfc_expr *
3597 gfc_default_initializer (gfc_typespec *ts)
3599 gfc_expr *init;
3600 gfc_component *comp;
3602 /* See if we have a default initializer. */
3603 for (comp = ts->u.derived->components; comp; comp = comp->next)
3604 if (comp->initializer || comp->attr.allocatable)
3605 break;
3607 if (!comp)
3608 return NULL;
3610 init = gfc_get_structure_constructor_expr (ts->type, ts->kind,
3611 &ts->u.derived->declared_at);
3612 init->ts = *ts;
3614 for (comp = ts->u.derived->components; comp; comp = comp->next)
3616 gfc_constructor *ctor = gfc_constructor_get();
3618 if (comp->initializer)
3619 ctor->expr = gfc_copy_expr (comp->initializer);
3621 if (comp->attr.allocatable)
3623 ctor->expr = gfc_get_expr ();
3624 ctor->expr->expr_type = EXPR_NULL;
3625 ctor->expr->ts = comp->ts;
3628 gfc_constructor_append (&init->value.constructor, ctor);
3631 return init;
3635 /* Given a symbol, create an expression node with that symbol as a
3636 variable. If the symbol is array valued, setup a reference of the
3637 whole array. */
3639 gfc_expr *
3640 gfc_get_variable_expr (gfc_symtree *var)
3642 gfc_expr *e;
3644 e = gfc_get_expr ();
3645 e->expr_type = EXPR_VARIABLE;
3646 e->symtree = var;
3647 e->ts = var->n.sym->ts;
3649 if (var->n.sym->as != NULL)
3651 e->rank = var->n.sym->as->rank;
3652 e->ref = gfc_get_ref ();
3653 e->ref->type = REF_ARRAY;
3654 e->ref->u.ar.type = AR_FULL;
3657 return e;
3661 /* Returns the array_spec of a full array expression. A NULL is
3662 returned otherwise. */
3663 gfc_array_spec *
3664 gfc_get_full_arrayspec_from_expr (gfc_expr *expr)
3666 gfc_array_spec *as;
3667 gfc_ref *ref;
3669 if (expr->rank == 0)
3670 return NULL;
3672 /* Follow any component references. */
3673 if (expr->expr_type == EXPR_VARIABLE
3674 || expr->expr_type == EXPR_CONSTANT)
3676 as = expr->symtree->n.sym->as;
3677 for (ref = expr->ref; ref; ref = ref->next)
3679 switch (ref->type)
3681 case REF_COMPONENT:
3682 as = ref->u.c.component->as;
3683 continue;
3685 case REF_SUBSTRING:
3686 continue;
3688 case REF_ARRAY:
3690 switch (ref->u.ar.type)
3692 case AR_ELEMENT:
3693 case AR_SECTION:
3694 case AR_UNKNOWN:
3695 as = NULL;
3696 continue;
3698 case AR_FULL:
3699 break;
3701 break;
3706 else
3707 as = NULL;
3709 return as;
3713 /* General expression traversal function. */
3715 bool
3716 gfc_traverse_expr (gfc_expr *expr, gfc_symbol *sym,
3717 bool (*func)(gfc_expr *, gfc_symbol *, int*),
3718 int f)
3720 gfc_array_ref ar;
3721 gfc_ref *ref;
3722 gfc_actual_arglist *args;
3723 gfc_constructor *c;
3724 int i;
3726 if (!expr)
3727 return false;
3729 if ((*func) (expr, sym, &f))
3730 return true;
3732 if (expr->ts.type == BT_CHARACTER
3733 && expr->ts.u.cl
3734 && expr->ts.u.cl->length
3735 && expr->ts.u.cl->length->expr_type != EXPR_CONSTANT
3736 && gfc_traverse_expr (expr->ts.u.cl->length, sym, func, f))
3737 return true;
3739 switch (expr->expr_type)
3741 case EXPR_PPC:
3742 case EXPR_COMPCALL:
3743 case EXPR_FUNCTION:
3744 for (args = expr->value.function.actual; args; args = args->next)
3746 if (gfc_traverse_expr (args->expr, sym, func, f))
3747 return true;
3749 break;
3751 case EXPR_VARIABLE:
3752 case EXPR_CONSTANT:
3753 case EXPR_NULL:
3754 case EXPR_SUBSTRING:
3755 break;
3757 case EXPR_STRUCTURE:
3758 case EXPR_ARRAY:
3759 for (c = gfc_constructor_first (expr->value.constructor);
3760 c; c = gfc_constructor_next (c))
3762 if (gfc_traverse_expr (c->expr, sym, func, f))
3763 return true;
3764 if (c->iterator)
3766 if (gfc_traverse_expr (c->iterator->var, sym, func, f))
3767 return true;
3768 if (gfc_traverse_expr (c->iterator->start, sym, func, f))
3769 return true;
3770 if (gfc_traverse_expr (c->iterator->end, sym, func, f))
3771 return true;
3772 if (gfc_traverse_expr (c->iterator->step, sym, func, f))
3773 return true;
3776 break;
3778 case EXPR_OP:
3779 if (gfc_traverse_expr (expr->value.op.op1, sym, func, f))
3780 return true;
3781 if (gfc_traverse_expr (expr->value.op.op2, sym, func, f))
3782 return true;
3783 break;
3785 default:
3786 gcc_unreachable ();
3787 break;
3790 ref = expr->ref;
3791 while (ref != NULL)
3793 switch (ref->type)
3795 case REF_ARRAY:
3796 ar = ref->u.ar;
3797 for (i = 0; i < GFC_MAX_DIMENSIONS; i++)
3799 if (gfc_traverse_expr (ar.start[i], sym, func, f))
3800 return true;
3801 if (gfc_traverse_expr (ar.end[i], sym, func, f))
3802 return true;
3803 if (gfc_traverse_expr (ar.stride[i], sym, func, f))
3804 return true;
3806 break;
3808 case REF_SUBSTRING:
3809 if (gfc_traverse_expr (ref->u.ss.start, sym, func, f))
3810 return true;
3811 if (gfc_traverse_expr (ref->u.ss.end, sym, func, f))
3812 return true;
3813 break;
3815 case REF_COMPONENT:
3816 if (ref->u.c.component->ts.type == BT_CHARACTER
3817 && ref->u.c.component->ts.u.cl
3818 && ref->u.c.component->ts.u.cl->length
3819 && ref->u.c.component->ts.u.cl->length->expr_type
3820 != EXPR_CONSTANT
3821 && gfc_traverse_expr (ref->u.c.component->ts.u.cl->length,
3822 sym, func, f))
3823 return true;
3825 if (ref->u.c.component->as)
3826 for (i = 0; i < ref->u.c.component->as->rank
3827 + ref->u.c.component->as->corank; i++)
3829 if (gfc_traverse_expr (ref->u.c.component->as->lower[i],
3830 sym, func, f))
3831 return true;
3832 if (gfc_traverse_expr (ref->u.c.component->as->upper[i],
3833 sym, func, f))
3834 return true;
3836 break;
3838 default:
3839 gcc_unreachable ();
3841 ref = ref->next;
3843 return false;
3846 /* Traverse expr, marking all EXPR_VARIABLE symbols referenced. */
3848 static bool
3849 expr_set_symbols_referenced (gfc_expr *expr,
3850 gfc_symbol *sym ATTRIBUTE_UNUSED,
3851 int *f ATTRIBUTE_UNUSED)
3853 if (expr->expr_type != EXPR_VARIABLE)
3854 return false;
3855 gfc_set_sym_referenced (expr->symtree->n.sym);
3856 return false;
3859 void
3860 gfc_expr_set_symbols_referenced (gfc_expr *expr)
3862 gfc_traverse_expr (expr, NULL, expr_set_symbols_referenced, 0);
3866 /* Determine if an expression is a procedure pointer component. If yes, the
3867 argument 'comp' will point to the component (provided that 'comp' was
3868 provided). */
3870 bool
3871 gfc_is_proc_ptr_comp (gfc_expr *expr, gfc_component **comp)
3873 gfc_ref *ref;
3874 bool ppc = false;
3876 if (!expr || !expr->ref)
3877 return false;
3879 ref = expr->ref;
3880 while (ref->next)
3881 ref = ref->next;
3883 if (ref->type == REF_COMPONENT)
3885 ppc = ref->u.c.component->attr.proc_pointer;
3886 if (ppc && comp)
3887 *comp = ref->u.c.component;
3890 return ppc;
3894 /* Walk an expression tree and check each variable encountered for being typed.
3895 If strict is not set, a top-level variable is tolerated untyped in -std=gnu
3896 mode as is a basic arithmetic expression using those; this is for things in
3897 legacy-code like:
3899 INTEGER :: arr(n), n
3900 INTEGER :: arr(n + 1), n
3902 The namespace is needed for IMPLICIT typing. */
3904 static gfc_namespace* check_typed_ns;
3906 static bool
3907 expr_check_typed_help (gfc_expr* e, gfc_symbol* sym ATTRIBUTE_UNUSED,
3908 int* f ATTRIBUTE_UNUSED)
3910 gfc_try t;
3912 if (e->expr_type != EXPR_VARIABLE)
3913 return false;
3915 gcc_assert (e->symtree);
3916 t = gfc_check_symbol_typed (e->symtree->n.sym, check_typed_ns,
3917 true, e->where);
3919 return (t == FAILURE);
3922 gfc_try
3923 gfc_expr_check_typed (gfc_expr* e, gfc_namespace* ns, bool strict)
3925 bool error_found;
3927 /* If this is a top-level variable or EXPR_OP, do the check with strict given
3928 to us. */
3929 if (!strict)
3931 if (e->expr_type == EXPR_VARIABLE && !e->ref)
3932 return gfc_check_symbol_typed (e->symtree->n.sym, ns, strict, e->where);
3934 if (e->expr_type == EXPR_OP)
3936 gfc_try t = SUCCESS;
3938 gcc_assert (e->value.op.op1);
3939 t = gfc_expr_check_typed (e->value.op.op1, ns, strict);
3941 if (t == SUCCESS && e->value.op.op2)
3942 t = gfc_expr_check_typed (e->value.op.op2, ns, strict);
3944 return t;
3948 /* Otherwise, walk the expression and do it strictly. */
3949 check_typed_ns = ns;
3950 error_found = gfc_traverse_expr (e, NULL, &expr_check_typed_help, 0);
3952 return error_found ? FAILURE : SUCCESS;
3955 /* Walk an expression tree and replace all symbols with a corresponding symbol
3956 in the formal_ns of "sym". Needed for copying interfaces in PROCEDURE
3957 statements. The boolean return value is required by gfc_traverse_expr. */
3959 static bool
3960 replace_symbol (gfc_expr *expr, gfc_symbol *sym, int *i ATTRIBUTE_UNUSED)
3962 if ((expr->expr_type == EXPR_VARIABLE
3963 || (expr->expr_type == EXPR_FUNCTION
3964 && !gfc_is_intrinsic (expr->symtree->n.sym, 0, expr->where)))
3965 && expr->symtree->n.sym->ns == sym->ts.interface->formal_ns)
3967 gfc_symtree *stree;
3968 gfc_namespace *ns = sym->formal_ns;
3969 /* Don't use gfc_get_symtree as we prefer to fail badly if we don't find
3970 the symtree rather than create a new one (and probably fail later). */
3971 stree = gfc_find_symtree (ns ? ns->sym_root : gfc_current_ns->sym_root,
3972 expr->symtree->n.sym->name);
3973 gcc_assert (stree);
3974 stree->n.sym->attr = expr->symtree->n.sym->attr;
3975 expr->symtree = stree;
3977 return false;
3980 void
3981 gfc_expr_replace_symbols (gfc_expr *expr, gfc_symbol *dest)
3983 gfc_traverse_expr (expr, dest, &replace_symbol, 0);
3986 /* The following is analogous to 'replace_symbol', and needed for copying
3987 interfaces for procedure pointer components. The argument 'sym' must formally
3988 be a gfc_symbol, so that the function can be passed to gfc_traverse_expr.
3989 However, it gets actually passed a gfc_component (i.e. the procedure pointer
3990 component in whose formal_ns the arguments have to be). */
3992 static bool
3993 replace_comp (gfc_expr *expr, gfc_symbol *sym, int *i ATTRIBUTE_UNUSED)
3995 gfc_component *comp;
3996 comp = (gfc_component *)sym;
3997 if ((expr->expr_type == EXPR_VARIABLE
3998 || (expr->expr_type == EXPR_FUNCTION
3999 && !gfc_is_intrinsic (expr->symtree->n.sym, 0, expr->where)))
4000 && expr->symtree->n.sym->ns == comp->ts.interface->formal_ns)
4002 gfc_symtree *stree;
4003 gfc_namespace *ns = comp->formal_ns;
4004 /* Don't use gfc_get_symtree as we prefer to fail badly if we don't find
4005 the symtree rather than create a new one (and probably fail later). */
4006 stree = gfc_find_symtree (ns ? ns->sym_root : gfc_current_ns->sym_root,
4007 expr->symtree->n.sym->name);
4008 gcc_assert (stree);
4009 stree->n.sym->attr = expr->symtree->n.sym->attr;
4010 expr->symtree = stree;
4012 return false;
4015 void
4016 gfc_expr_replace_comp (gfc_expr *expr, gfc_component *dest)
4018 gfc_traverse_expr (expr, (gfc_symbol *)dest, &replace_comp, 0);
4022 bool
4023 gfc_is_coindexed (gfc_expr *e)
4025 gfc_ref *ref;
4027 for (ref = e->ref; ref; ref = ref->next)
4028 if (ref->type == REF_ARRAY && ref->u.ar.codimen > 0)
4029 return true;
4031 return false;
4035 /* Check whether the expression has an ultimate allocatable component.
4036 Being itself allocatable does not count. */
4037 bool
4038 gfc_has_ultimate_allocatable (gfc_expr *e)
4040 gfc_ref *ref, *last = NULL;
4042 if (e->expr_type != EXPR_VARIABLE)
4043 return false;
4045 for (ref = e->ref; ref; ref = ref->next)
4046 if (ref->type == REF_COMPONENT)
4047 last = ref;
4049 if (last && last->u.c.component->ts.type == BT_CLASS)
4050 return last->u.c.component->ts.u.derived->components->attr.alloc_comp;
4051 else if (last && last->u.c.component->ts.type == BT_DERIVED)
4052 return last->u.c.component->ts.u.derived->attr.alloc_comp;
4053 else if (last)
4054 return false;
4056 if (e->ts.type == BT_CLASS)
4057 return e->ts.u.derived->components->attr.alloc_comp;
4058 else if (e->ts.type == BT_DERIVED)
4059 return e->ts.u.derived->attr.alloc_comp;
4060 else
4061 return false;
4065 /* Check whether the expression has an pointer component.
4066 Being itself a pointer does not count. */
4067 bool
4068 gfc_has_ultimate_pointer (gfc_expr *e)
4070 gfc_ref *ref, *last = NULL;
4072 if (e->expr_type != EXPR_VARIABLE)
4073 return false;
4075 for (ref = e->ref; ref; ref = ref->next)
4076 if (ref->type == REF_COMPONENT)
4077 last = ref;
4079 if (last && last->u.c.component->ts.type == BT_CLASS)
4080 return last->u.c.component->ts.u.derived->components->attr.pointer_comp;
4081 else if (last && last->u.c.component->ts.type == BT_DERIVED)
4082 return last->u.c.component->ts.u.derived->attr.pointer_comp;
4083 else if (last)
4084 return false;
4086 if (e->ts.type == BT_CLASS)
4087 return e->ts.u.derived->components->attr.pointer_comp;
4088 else if (e->ts.type == BT_DERIVED)
4089 return e->ts.u.derived->attr.pointer_comp;
4090 else
4091 return false;