2011-04-04 Tobias Burnus <burnus@net-b.de>
[official-gcc.git] / gcc / fortran / expr.c
blob287a2a24e3c4cbee6681ae4b772daff1bc401b18
1 /* Routines for manipulation of expression nodes.
2 Copyright (C) 2000, 2001, 2002, 2003, 2004, 2005, 2006, 2007, 2008,
3 2009, 2010, 2011
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_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 /* Copy a shape array. */
681 mpz_t *
682 gfc_copy_shape (mpz_t *shape, int rank)
684 mpz_t *new_shape;
685 int n;
687 if (shape == NULL)
688 return NULL;
690 new_shape = gfc_get_shape (rank);
692 for (n = 0; n < rank; n++)
693 mpz_init_set (new_shape[n], shape[n]);
695 return new_shape;
699 /* Copy a shape array excluding dimension N, where N is an integer
700 constant expression. Dimensions are numbered in fortran style --
701 starting with ONE.
703 So, if the original shape array contains R elements
704 { s1 ... sN-1 sN sN+1 ... sR-1 sR}
705 the result contains R-1 elements:
706 { s1 ... sN-1 sN+1 ... sR-1}
708 If anything goes wrong -- N is not a constant, its value is out
709 of range -- or anything else, just returns NULL. */
711 mpz_t *
712 gfc_copy_shape_excluding (mpz_t *shape, int rank, gfc_expr *dim)
714 mpz_t *new_shape, *s;
715 int i, n;
717 if (shape == NULL
718 || rank <= 1
719 || dim == NULL
720 || dim->expr_type != EXPR_CONSTANT
721 || dim->ts.type != BT_INTEGER)
722 return NULL;
724 n = mpz_get_si (dim->value.integer);
725 n--; /* Convert to zero based index. */
726 if (n < 0 || n >= rank)
727 return NULL;
729 s = new_shape = gfc_get_shape (rank - 1);
731 for (i = 0; i < rank; i++)
733 if (i == n)
734 continue;
735 mpz_init_set (*s, shape[i]);
736 s++;
739 return new_shape;
743 /* Return the maximum kind of two expressions. In general, higher
744 kind numbers mean more precision for numeric types. */
747 gfc_kind_max (gfc_expr *e1, gfc_expr *e2)
749 return (e1->ts.kind > e2->ts.kind) ? e1->ts.kind : e2->ts.kind;
753 /* Returns nonzero if the type is numeric, zero otherwise. */
755 static int
756 numeric_type (bt type)
758 return type == BT_COMPLEX || type == BT_REAL || type == BT_INTEGER;
762 /* Returns nonzero if the typespec is a numeric type, zero otherwise. */
765 gfc_numeric_ts (gfc_typespec *ts)
767 return numeric_type (ts->type);
771 /* Return an expression node with an optional argument list attached.
772 A variable number of gfc_expr pointers are strung together in an
773 argument list with a NULL pointer terminating the list. */
775 gfc_expr *
776 gfc_build_conversion (gfc_expr *e)
778 gfc_expr *p;
780 p = gfc_get_expr ();
781 p->expr_type = EXPR_FUNCTION;
782 p->symtree = NULL;
783 p->value.function.actual = NULL;
785 p->value.function.actual = gfc_get_actual_arglist ();
786 p->value.function.actual->expr = e;
788 return p;
792 /* Given an expression node with some sort of numeric binary
793 expression, insert type conversions required to make the operands
794 have the same type. Conversion warnings are disabled if wconversion
795 is set to 0.
797 The exception is that the operands of an exponential don't have to
798 have the same type. If possible, the base is promoted to the type
799 of the exponent. For example, 1**2.3 becomes 1.0**2.3, but
800 1.0**2 stays as it is. */
802 void
803 gfc_type_convert_binary (gfc_expr *e, int wconversion)
805 gfc_expr *op1, *op2;
807 op1 = e->value.op.op1;
808 op2 = e->value.op.op2;
810 if (op1->ts.type == BT_UNKNOWN || op2->ts.type == BT_UNKNOWN)
812 gfc_clear_ts (&e->ts);
813 return;
816 /* Kind conversions of same type. */
817 if (op1->ts.type == op2->ts.type)
819 if (op1->ts.kind == op2->ts.kind)
821 /* No type conversions. */
822 e->ts = op1->ts;
823 goto done;
826 if (op1->ts.kind > op2->ts.kind)
827 gfc_convert_type_warn (op2, &op1->ts, 2, wconversion);
828 else
829 gfc_convert_type_warn (op1, &op2->ts, 2, wconversion);
831 e->ts = op1->ts;
832 goto done;
835 /* Integer combined with real or complex. */
836 if (op2->ts.type == BT_INTEGER)
838 e->ts = op1->ts;
840 /* Special case for ** operator. */
841 if (e->value.op.op == INTRINSIC_POWER)
842 goto done;
844 gfc_convert_type_warn (e->value.op.op2, &e->ts, 2, wconversion);
845 goto done;
848 if (op1->ts.type == BT_INTEGER)
850 e->ts = op2->ts;
851 gfc_convert_type_warn (e->value.op.op1, &e->ts, 2, wconversion);
852 goto done;
855 /* Real combined with complex. */
856 e->ts.type = BT_COMPLEX;
857 if (op1->ts.kind > op2->ts.kind)
858 e->ts.kind = op1->ts.kind;
859 else
860 e->ts.kind = op2->ts.kind;
861 if (op1->ts.type != BT_COMPLEX || op1->ts.kind != e->ts.kind)
862 gfc_convert_type_warn (e->value.op.op1, &e->ts, 2, wconversion);
863 if (op2->ts.type != BT_COMPLEX || op2->ts.kind != e->ts.kind)
864 gfc_convert_type_warn (e->value.op.op2, &e->ts, 2, wconversion);
866 done:
867 return;
871 /* Function to determine if an expression is constant or not. This
872 function expects that the expression has already been simplified. */
875 gfc_is_constant_expr (gfc_expr *e)
877 gfc_constructor *c;
878 gfc_actual_arglist *arg;
879 gfc_symbol *sym;
881 if (e == NULL)
882 return 1;
884 switch (e->expr_type)
886 case EXPR_OP:
887 return (gfc_is_constant_expr (e->value.op.op1)
888 && (e->value.op.op2 == NULL
889 || gfc_is_constant_expr (e->value.op.op2)));
891 case EXPR_VARIABLE:
892 return 0;
894 case EXPR_FUNCTION:
895 case EXPR_PPC:
896 case EXPR_COMPCALL:
897 /* Call to intrinsic with at least one argument. */
898 if (e->value.function.isym && e->value.function.actual)
900 for (arg = e->value.function.actual; arg; arg = arg->next)
901 if (!gfc_is_constant_expr (arg->expr))
902 return 0;
905 /* Make sure we have a symbol. */
906 gcc_assert (e->symtree);
908 sym = e->symtree->n.sym;
910 /* Specification functions are constant. */
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 1;
923 if (e->value.function.isym
924 && (e->value.function.isym->elemental
925 || e->value.function.isym->pure
926 || e->value.function.isym->inquiry
927 || e->value.function.isym->transformational))
928 return 1;
930 return 0;
932 case EXPR_CONSTANT:
933 case EXPR_NULL:
934 return 1;
936 case EXPR_SUBSTRING:
937 return e->ref == NULL || (gfc_is_constant_expr (e->ref->u.ss.start)
938 && gfc_is_constant_expr (e->ref->u.ss.end));
940 case EXPR_ARRAY:
941 case EXPR_STRUCTURE:
942 c = gfc_constructor_first (e->value.constructor);
943 if ((e->expr_type == EXPR_ARRAY) && c && c->iterator)
944 return gfc_constant_ac (e);
946 for (; c; c = gfc_constructor_next (c))
947 if (!gfc_is_constant_expr (c->expr))
948 return 0;
950 return 1;
953 default:
954 gfc_internal_error ("gfc_is_constant_expr(): Unknown expression type");
955 return 0;
960 /* Is true if an array reference is followed by a component or substring
961 reference. */
962 bool
963 is_subref_array (gfc_expr * e)
965 gfc_ref * ref;
966 bool seen_array;
968 if (e->expr_type != EXPR_VARIABLE)
969 return false;
971 if (e->symtree->n.sym->attr.subref_array_pointer)
972 return true;
974 seen_array = false;
975 for (ref = e->ref; ref; ref = ref->next)
977 if (ref->type == REF_ARRAY
978 && ref->u.ar.type != AR_ELEMENT)
979 seen_array = true;
981 if (seen_array
982 && ref->type != REF_ARRAY)
983 return seen_array;
985 return false;
989 /* Try to collapse intrinsic expressions. */
991 static gfc_try
992 simplify_intrinsic_op (gfc_expr *p, int type)
994 gfc_intrinsic_op op;
995 gfc_expr *op1, *op2, *result;
997 if (p->value.op.op == INTRINSIC_USER)
998 return SUCCESS;
1000 op1 = p->value.op.op1;
1001 op2 = p->value.op.op2;
1002 op = p->value.op.op;
1004 if (gfc_simplify_expr (op1, type) == FAILURE)
1005 return FAILURE;
1006 if (gfc_simplify_expr (op2, type) == FAILURE)
1007 return FAILURE;
1009 if (!gfc_is_constant_expr (op1)
1010 || (op2 != NULL && !gfc_is_constant_expr (op2)))
1011 return SUCCESS;
1013 /* Rip p apart. */
1014 p->value.op.op1 = NULL;
1015 p->value.op.op2 = NULL;
1017 switch (op)
1019 case INTRINSIC_PARENTHESES:
1020 result = gfc_parentheses (op1);
1021 break;
1023 case INTRINSIC_UPLUS:
1024 result = gfc_uplus (op1);
1025 break;
1027 case INTRINSIC_UMINUS:
1028 result = gfc_uminus (op1);
1029 break;
1031 case INTRINSIC_PLUS:
1032 result = gfc_add (op1, op2);
1033 break;
1035 case INTRINSIC_MINUS:
1036 result = gfc_subtract (op1, op2);
1037 break;
1039 case INTRINSIC_TIMES:
1040 result = gfc_multiply (op1, op2);
1041 break;
1043 case INTRINSIC_DIVIDE:
1044 result = gfc_divide (op1, op2);
1045 break;
1047 case INTRINSIC_POWER:
1048 result = gfc_power (op1, op2);
1049 break;
1051 case INTRINSIC_CONCAT:
1052 result = gfc_concat (op1, op2);
1053 break;
1055 case INTRINSIC_EQ:
1056 case INTRINSIC_EQ_OS:
1057 result = gfc_eq (op1, op2, op);
1058 break;
1060 case INTRINSIC_NE:
1061 case INTRINSIC_NE_OS:
1062 result = gfc_ne (op1, op2, op);
1063 break;
1065 case INTRINSIC_GT:
1066 case INTRINSIC_GT_OS:
1067 result = gfc_gt (op1, op2, op);
1068 break;
1070 case INTRINSIC_GE:
1071 case INTRINSIC_GE_OS:
1072 result = gfc_ge (op1, op2, op);
1073 break;
1075 case INTRINSIC_LT:
1076 case INTRINSIC_LT_OS:
1077 result = gfc_lt (op1, op2, op);
1078 break;
1080 case INTRINSIC_LE:
1081 case INTRINSIC_LE_OS:
1082 result = gfc_le (op1, op2, op);
1083 break;
1085 case INTRINSIC_NOT:
1086 result = gfc_not (op1);
1087 break;
1089 case INTRINSIC_AND:
1090 result = gfc_and (op1, op2);
1091 break;
1093 case INTRINSIC_OR:
1094 result = gfc_or (op1, op2);
1095 break;
1097 case INTRINSIC_EQV:
1098 result = gfc_eqv (op1, op2);
1099 break;
1101 case INTRINSIC_NEQV:
1102 result = gfc_neqv (op1, op2);
1103 break;
1105 default:
1106 gfc_internal_error ("simplify_intrinsic_op(): Bad operator");
1109 if (result == NULL)
1111 gfc_free_expr (op1);
1112 gfc_free_expr (op2);
1113 return FAILURE;
1116 result->rank = p->rank;
1117 result->where = p->where;
1118 gfc_replace_expr (p, result);
1120 return SUCCESS;
1124 /* Subroutine to simplify constructor expressions. Mutually recursive
1125 with gfc_simplify_expr(). */
1127 static gfc_try
1128 simplify_constructor (gfc_constructor_base base, int type)
1130 gfc_constructor *c;
1131 gfc_expr *p;
1133 for (c = gfc_constructor_first (base); c; c = gfc_constructor_next (c))
1135 if (c->iterator
1136 && (gfc_simplify_expr (c->iterator->start, type) == FAILURE
1137 || gfc_simplify_expr (c->iterator->end, type) == FAILURE
1138 || gfc_simplify_expr (c->iterator->step, type) == FAILURE))
1139 return FAILURE;
1141 if (c->expr)
1143 /* Try and simplify a copy. Replace the original if successful
1144 but keep going through the constructor at all costs. Not
1145 doing so can make a dog's dinner of complicated things. */
1146 p = gfc_copy_expr (c->expr);
1148 if (gfc_simplify_expr (p, type) == FAILURE)
1150 gfc_free_expr (p);
1151 continue;
1154 gfc_replace_expr (c->expr, p);
1158 return SUCCESS;
1162 /* Pull a single array element out of an array constructor. */
1164 static gfc_try
1165 find_array_element (gfc_constructor_base base, gfc_array_ref *ar,
1166 gfc_constructor **rval)
1168 unsigned long nelemen;
1169 int i;
1170 mpz_t delta;
1171 mpz_t offset;
1172 mpz_t span;
1173 mpz_t tmp;
1174 gfc_constructor *cons;
1175 gfc_expr *e;
1176 gfc_try t;
1178 t = SUCCESS;
1179 e = NULL;
1181 mpz_init_set_ui (offset, 0);
1182 mpz_init (delta);
1183 mpz_init (tmp);
1184 mpz_init_set_ui (span, 1);
1185 for (i = 0; i < ar->dimen; i++)
1187 if (gfc_reduce_init_expr (ar->as->lower[i]) == FAILURE
1188 || gfc_reduce_init_expr (ar->as->upper[i]) == FAILURE)
1190 t = FAILURE;
1191 cons = NULL;
1192 goto depart;
1195 e = gfc_copy_expr (ar->start[i]);
1196 if (e->expr_type != EXPR_CONSTANT)
1198 cons = NULL;
1199 goto depart;
1202 gcc_assert (ar->as->upper[i]->expr_type == EXPR_CONSTANT
1203 && ar->as->lower[i]->expr_type == EXPR_CONSTANT);
1205 /* Check the bounds. */
1206 if ((ar->as->upper[i]
1207 && mpz_cmp (e->value.integer,
1208 ar->as->upper[i]->value.integer) > 0)
1209 || (mpz_cmp (e->value.integer,
1210 ar->as->lower[i]->value.integer) < 0))
1212 gfc_error ("Index in dimension %d is out of bounds "
1213 "at %L", i + 1, &ar->c_where[i]);
1214 cons = NULL;
1215 t = FAILURE;
1216 goto depart;
1219 mpz_sub (delta, e->value.integer, ar->as->lower[i]->value.integer);
1220 mpz_mul (delta, delta, span);
1221 mpz_add (offset, offset, delta);
1223 mpz_set_ui (tmp, 1);
1224 mpz_add (tmp, tmp, ar->as->upper[i]->value.integer);
1225 mpz_sub (tmp, tmp, ar->as->lower[i]->value.integer);
1226 mpz_mul (span, span, tmp);
1229 for (cons = gfc_constructor_first (base), nelemen = mpz_get_ui (offset);
1230 cons && nelemen > 0; cons = gfc_constructor_next (cons), nelemen--)
1232 if (cons->iterator)
1234 cons = NULL;
1235 goto depart;
1239 depart:
1240 mpz_clear (delta);
1241 mpz_clear (offset);
1242 mpz_clear (span);
1243 mpz_clear (tmp);
1244 if (e)
1245 gfc_free_expr (e);
1246 *rval = cons;
1247 return t;
1251 /* Find a component of a structure constructor. */
1253 static gfc_constructor *
1254 find_component_ref (gfc_constructor_base base, gfc_ref *ref)
1256 gfc_component *comp;
1257 gfc_component *pick;
1258 gfc_constructor *c = gfc_constructor_first (base);
1260 comp = ref->u.c.sym->components;
1261 pick = ref->u.c.component;
1262 while (comp != pick)
1264 comp = comp->next;
1265 c = gfc_constructor_next (c);
1268 return c;
1272 /* Replace an expression with the contents of a constructor, removing
1273 the subobject reference in the process. */
1275 static void
1276 remove_subobject_ref (gfc_expr *p, gfc_constructor *cons)
1278 gfc_expr *e;
1280 if (cons)
1282 e = cons->expr;
1283 cons->expr = NULL;
1285 else
1286 e = gfc_copy_expr (p);
1287 e->ref = p->ref->next;
1288 p->ref->next = NULL;
1289 gfc_replace_expr (p, e);
1293 /* Pull an array section out of an array constructor. */
1295 static gfc_try
1296 find_array_section (gfc_expr *expr, gfc_ref *ref)
1298 int idx;
1299 int rank;
1300 int d;
1301 int shape_i;
1302 int limit;
1303 long unsigned one = 1;
1304 bool incr_ctr;
1305 mpz_t start[GFC_MAX_DIMENSIONS];
1306 mpz_t end[GFC_MAX_DIMENSIONS];
1307 mpz_t stride[GFC_MAX_DIMENSIONS];
1308 mpz_t delta[GFC_MAX_DIMENSIONS];
1309 mpz_t ctr[GFC_MAX_DIMENSIONS];
1310 mpz_t delta_mpz;
1311 mpz_t tmp_mpz;
1312 mpz_t nelts;
1313 mpz_t ptr;
1314 gfc_constructor_base base;
1315 gfc_constructor *cons, *vecsub[GFC_MAX_DIMENSIONS];
1316 gfc_expr *begin;
1317 gfc_expr *finish;
1318 gfc_expr *step;
1319 gfc_expr *upper;
1320 gfc_expr *lower;
1321 gfc_try t;
1323 t = SUCCESS;
1325 base = expr->value.constructor;
1326 expr->value.constructor = NULL;
1328 rank = ref->u.ar.as->rank;
1330 if (expr->shape == NULL)
1331 expr->shape = gfc_get_shape (rank);
1333 mpz_init_set_ui (delta_mpz, one);
1334 mpz_init_set_ui (nelts, one);
1335 mpz_init (tmp_mpz);
1337 /* Do the initialization now, so that we can cleanup without
1338 keeping track of where we were. */
1339 for (d = 0; d < rank; d++)
1341 mpz_init (delta[d]);
1342 mpz_init (start[d]);
1343 mpz_init (end[d]);
1344 mpz_init (ctr[d]);
1345 mpz_init (stride[d]);
1346 vecsub[d] = NULL;
1349 /* Build the counters to clock through the array reference. */
1350 shape_i = 0;
1351 for (d = 0; d < rank; d++)
1353 /* Make this stretch of code easier on the eye! */
1354 begin = ref->u.ar.start[d];
1355 finish = ref->u.ar.end[d];
1356 step = ref->u.ar.stride[d];
1357 lower = ref->u.ar.as->lower[d];
1358 upper = ref->u.ar.as->upper[d];
1360 if (ref->u.ar.dimen_type[d] == DIMEN_VECTOR) /* Vector subscript. */
1362 gfc_constructor *ci;
1363 gcc_assert (begin);
1365 if (begin->expr_type != EXPR_ARRAY || !gfc_is_constant_expr (begin))
1367 t = FAILURE;
1368 goto cleanup;
1371 gcc_assert (begin->rank == 1);
1372 /* Zero-sized arrays have no shape and no elements, stop early. */
1373 if (!begin->shape)
1375 mpz_init_set_ui (nelts, 0);
1376 break;
1379 vecsub[d] = gfc_constructor_first (begin->value.constructor);
1380 mpz_set (ctr[d], vecsub[d]->expr->value.integer);
1381 mpz_mul (nelts, nelts, begin->shape[0]);
1382 mpz_set (expr->shape[shape_i++], begin->shape[0]);
1384 /* Check bounds. */
1385 for (ci = vecsub[d]; ci; ci = gfc_constructor_next (ci))
1387 if (mpz_cmp (ci->expr->value.integer, upper->value.integer) > 0
1388 || mpz_cmp (ci->expr->value.integer,
1389 lower->value.integer) < 0)
1391 gfc_error ("index in dimension %d is out of bounds "
1392 "at %L", d + 1, &ref->u.ar.c_where[d]);
1393 t = FAILURE;
1394 goto cleanup;
1398 else
1400 if ((begin && begin->expr_type != EXPR_CONSTANT)
1401 || (finish && finish->expr_type != EXPR_CONSTANT)
1402 || (step && step->expr_type != EXPR_CONSTANT))
1404 t = FAILURE;
1405 goto cleanup;
1408 /* Obtain the stride. */
1409 if (step)
1410 mpz_set (stride[d], step->value.integer);
1411 else
1412 mpz_set_ui (stride[d], one);
1414 if (mpz_cmp_ui (stride[d], 0) == 0)
1415 mpz_set_ui (stride[d], one);
1417 /* Obtain the start value for the index. */
1418 if (begin)
1419 mpz_set (start[d], begin->value.integer);
1420 else
1421 mpz_set (start[d], lower->value.integer);
1423 mpz_set (ctr[d], start[d]);
1425 /* Obtain the end value for the index. */
1426 if (finish)
1427 mpz_set (end[d], finish->value.integer);
1428 else
1429 mpz_set (end[d], upper->value.integer);
1431 /* Separate 'if' because elements sometimes arrive with
1432 non-null end. */
1433 if (ref->u.ar.dimen_type[d] == DIMEN_ELEMENT)
1434 mpz_set (end [d], begin->value.integer);
1436 /* Check the bounds. */
1437 if (mpz_cmp (ctr[d], upper->value.integer) > 0
1438 || mpz_cmp (end[d], upper->value.integer) > 0
1439 || mpz_cmp (ctr[d], lower->value.integer) < 0
1440 || mpz_cmp (end[d], lower->value.integer) < 0)
1442 gfc_error ("index in dimension %d is out of bounds "
1443 "at %L", d + 1, &ref->u.ar.c_where[d]);
1444 t = FAILURE;
1445 goto cleanup;
1448 /* Calculate the number of elements and the shape. */
1449 mpz_set (tmp_mpz, stride[d]);
1450 mpz_add (tmp_mpz, end[d], tmp_mpz);
1451 mpz_sub (tmp_mpz, tmp_mpz, ctr[d]);
1452 mpz_div (tmp_mpz, tmp_mpz, stride[d]);
1453 mpz_mul (nelts, nelts, tmp_mpz);
1455 /* An element reference reduces the rank of the expression; don't
1456 add anything to the shape array. */
1457 if (ref->u.ar.dimen_type[d] != DIMEN_ELEMENT)
1458 mpz_set (expr->shape[shape_i++], tmp_mpz);
1461 /* Calculate the 'stride' (=delta) for conversion of the
1462 counter values into the index along the constructor. */
1463 mpz_set (delta[d], delta_mpz);
1464 mpz_sub (tmp_mpz, upper->value.integer, lower->value.integer);
1465 mpz_add_ui (tmp_mpz, tmp_mpz, one);
1466 mpz_mul (delta_mpz, delta_mpz, tmp_mpz);
1469 mpz_init (ptr);
1470 cons = gfc_constructor_first (base);
1472 /* Now clock through the array reference, calculating the index in
1473 the source constructor and transferring the elements to the new
1474 constructor. */
1475 for (idx = 0; idx < (int) mpz_get_si (nelts); idx++)
1477 if (ref->u.ar.offset)
1478 mpz_set (ptr, ref->u.ar.offset->value.integer);
1479 else
1480 mpz_init_set_ui (ptr, 0);
1482 incr_ctr = true;
1483 for (d = 0; d < rank; d++)
1485 mpz_set (tmp_mpz, ctr[d]);
1486 mpz_sub (tmp_mpz, tmp_mpz, ref->u.ar.as->lower[d]->value.integer);
1487 mpz_mul (tmp_mpz, tmp_mpz, delta[d]);
1488 mpz_add (ptr, ptr, tmp_mpz);
1490 if (!incr_ctr) continue;
1492 if (ref->u.ar.dimen_type[d] == DIMEN_VECTOR) /* Vector subscript. */
1494 gcc_assert(vecsub[d]);
1496 if (!gfc_constructor_next (vecsub[d]))
1497 vecsub[d] = gfc_constructor_first (ref->u.ar.start[d]->value.constructor);
1498 else
1500 vecsub[d] = gfc_constructor_next (vecsub[d]);
1501 incr_ctr = false;
1503 mpz_set (ctr[d], vecsub[d]->expr->value.integer);
1505 else
1507 mpz_add (ctr[d], ctr[d], stride[d]);
1509 if (mpz_cmp_ui (stride[d], 0) > 0
1510 ? mpz_cmp (ctr[d], end[d]) > 0
1511 : mpz_cmp (ctr[d], end[d]) < 0)
1512 mpz_set (ctr[d], start[d]);
1513 else
1514 incr_ctr = false;
1518 limit = mpz_get_ui (ptr);
1519 if (limit >= gfc_option.flag_max_array_constructor)
1521 gfc_error ("The number of elements in the array constructor "
1522 "at %L requires an increase of the allowed %d "
1523 "upper limit. See -fmax-array-constructor "
1524 "option", &expr->where,
1525 gfc_option.flag_max_array_constructor);
1526 return FAILURE;
1529 cons = gfc_constructor_lookup (base, limit);
1530 gcc_assert (cons);
1531 gfc_constructor_append_expr (&expr->value.constructor,
1532 gfc_copy_expr (cons->expr), NULL);
1535 mpz_clear (ptr);
1537 cleanup:
1539 mpz_clear (delta_mpz);
1540 mpz_clear (tmp_mpz);
1541 mpz_clear (nelts);
1542 for (d = 0; d < rank; d++)
1544 mpz_clear (delta[d]);
1545 mpz_clear (start[d]);
1546 mpz_clear (end[d]);
1547 mpz_clear (ctr[d]);
1548 mpz_clear (stride[d]);
1550 gfc_constructor_free (base);
1551 return t;
1554 /* Pull a substring out of an expression. */
1556 static gfc_try
1557 find_substring_ref (gfc_expr *p, gfc_expr **newp)
1559 int end;
1560 int start;
1561 int length;
1562 gfc_char_t *chr;
1564 if (p->ref->u.ss.start->expr_type != EXPR_CONSTANT
1565 || p->ref->u.ss.end->expr_type != EXPR_CONSTANT)
1566 return FAILURE;
1568 *newp = gfc_copy_expr (p);
1569 gfc_free ((*newp)->value.character.string);
1571 end = (int) mpz_get_ui (p->ref->u.ss.end->value.integer);
1572 start = (int) mpz_get_ui (p->ref->u.ss.start->value.integer);
1573 length = end - start + 1;
1575 chr = (*newp)->value.character.string = gfc_get_wide_string (length + 1);
1576 (*newp)->value.character.length = length;
1577 memcpy (chr, &p->value.character.string[start - 1],
1578 length * sizeof (gfc_char_t));
1579 chr[length] = '\0';
1580 return SUCCESS;
1585 /* Simplify a subobject reference of a constructor. This occurs when
1586 parameter variable values are substituted. */
1588 static gfc_try
1589 simplify_const_ref (gfc_expr *p)
1591 gfc_constructor *cons, *c;
1592 gfc_expr *newp;
1593 gfc_ref *last_ref;
1595 while (p->ref)
1597 switch (p->ref->type)
1599 case REF_ARRAY:
1600 switch (p->ref->u.ar.type)
1602 case AR_ELEMENT:
1603 /* <type/kind spec>, parameter :: x(<int>) = scalar_expr
1604 will generate this. */
1605 if (p->expr_type != EXPR_ARRAY)
1607 remove_subobject_ref (p, NULL);
1608 break;
1610 if (find_array_element (p->value.constructor, &p->ref->u.ar,
1611 &cons) == FAILURE)
1612 return FAILURE;
1614 if (!cons)
1615 return SUCCESS;
1617 remove_subobject_ref (p, cons);
1618 break;
1620 case AR_SECTION:
1621 if (find_array_section (p, p->ref) == FAILURE)
1622 return FAILURE;
1623 p->ref->u.ar.type = AR_FULL;
1625 /* Fall through. */
1627 case AR_FULL:
1628 if (p->ref->next != NULL
1629 && (p->ts.type == BT_CHARACTER || p->ts.type == BT_DERIVED))
1631 for (c = gfc_constructor_first (p->value.constructor);
1632 c; c = gfc_constructor_next (c))
1634 c->expr->ref = gfc_copy_ref (p->ref->next);
1635 if (simplify_const_ref (c->expr) == FAILURE)
1636 return FAILURE;
1639 if (p->ts.type == BT_DERIVED
1640 && p->ref->next
1641 && (c = gfc_constructor_first (p->value.constructor)))
1643 /* There may have been component references. */
1644 p->ts = c->expr->ts;
1647 last_ref = p->ref;
1648 for (; last_ref->next; last_ref = last_ref->next) {};
1650 if (p->ts.type == BT_CHARACTER
1651 && last_ref->type == REF_SUBSTRING)
1653 /* If this is a CHARACTER array and we possibly took
1654 a substring out of it, update the type-spec's
1655 character length according to the first element
1656 (as all should have the same length). */
1657 int string_len;
1658 if ((c = gfc_constructor_first (p->value.constructor)))
1660 const gfc_expr* first = c->expr;
1661 gcc_assert (first->expr_type == EXPR_CONSTANT);
1662 gcc_assert (first->ts.type == BT_CHARACTER);
1663 string_len = first->value.character.length;
1665 else
1666 string_len = 0;
1668 if (!p->ts.u.cl)
1669 p->ts.u.cl = gfc_new_charlen (p->symtree->n.sym->ns,
1670 NULL);
1671 else
1672 gfc_free_expr (p->ts.u.cl->length);
1674 p->ts.u.cl->length
1675 = gfc_get_int_expr (gfc_default_integer_kind,
1676 NULL, string_len);
1679 gfc_free_ref_list (p->ref);
1680 p->ref = NULL;
1681 break;
1683 default:
1684 return SUCCESS;
1687 break;
1689 case REF_COMPONENT:
1690 cons = find_component_ref (p->value.constructor, p->ref);
1691 remove_subobject_ref (p, cons);
1692 break;
1694 case REF_SUBSTRING:
1695 if (find_substring_ref (p, &newp) == FAILURE)
1696 return FAILURE;
1698 gfc_replace_expr (p, newp);
1699 gfc_free_ref_list (p->ref);
1700 p->ref = NULL;
1701 break;
1705 return SUCCESS;
1709 /* Simplify a chain of references. */
1711 static gfc_try
1712 simplify_ref_chain (gfc_ref *ref, int type)
1714 int n;
1716 for (; ref; ref = ref->next)
1718 switch (ref->type)
1720 case REF_ARRAY:
1721 for (n = 0; n < ref->u.ar.dimen; n++)
1723 if (gfc_simplify_expr (ref->u.ar.start[n], type) == FAILURE)
1724 return FAILURE;
1725 if (gfc_simplify_expr (ref->u.ar.end[n], type) == FAILURE)
1726 return FAILURE;
1727 if (gfc_simplify_expr (ref->u.ar.stride[n], type) == FAILURE)
1728 return FAILURE;
1730 break;
1732 case REF_SUBSTRING:
1733 if (gfc_simplify_expr (ref->u.ss.start, type) == FAILURE)
1734 return FAILURE;
1735 if (gfc_simplify_expr (ref->u.ss.end, type) == FAILURE)
1736 return FAILURE;
1737 break;
1739 default:
1740 break;
1743 return SUCCESS;
1747 /* Try to substitute the value of a parameter variable. */
1749 static gfc_try
1750 simplify_parameter_variable (gfc_expr *p, int type)
1752 gfc_expr *e;
1753 gfc_try t;
1755 e = gfc_copy_expr (p->symtree->n.sym->value);
1756 if (e == NULL)
1757 return FAILURE;
1759 e->rank = p->rank;
1761 /* Do not copy subobject refs for constant. */
1762 if (e->expr_type != EXPR_CONSTANT && p->ref != NULL)
1763 e->ref = gfc_copy_ref (p->ref);
1764 t = gfc_simplify_expr (e, type);
1766 /* Only use the simplification if it eliminated all subobject references. */
1767 if (t == SUCCESS && !e->ref)
1768 gfc_replace_expr (p, e);
1769 else
1770 gfc_free_expr (e);
1772 return t;
1775 /* Given an expression, simplify it by collapsing constant
1776 expressions. Most simplification takes place when the expression
1777 tree is being constructed. If an intrinsic function is simplified
1778 at some point, we get called again to collapse the result against
1779 other constants.
1781 We work by recursively simplifying expression nodes, simplifying
1782 intrinsic functions where possible, which can lead to further
1783 constant collapsing. If an operator has constant operand(s), we
1784 rip the expression apart, and rebuild it, hoping that it becomes
1785 something simpler.
1787 The expression type is defined for:
1788 0 Basic expression parsing
1789 1 Simplifying array constructors -- will substitute
1790 iterator values.
1791 Returns FAILURE on error, SUCCESS otherwise.
1792 NOTE: Will return SUCCESS even if the expression can not be simplified. */
1794 gfc_try
1795 gfc_simplify_expr (gfc_expr *p, int type)
1797 gfc_actual_arglist *ap;
1799 if (p == NULL)
1800 return SUCCESS;
1802 switch (p->expr_type)
1804 case EXPR_CONSTANT:
1805 case EXPR_NULL:
1806 break;
1808 case EXPR_FUNCTION:
1809 for (ap = p->value.function.actual; ap; ap = ap->next)
1810 if (gfc_simplify_expr (ap->expr, type) == FAILURE)
1811 return FAILURE;
1813 if (p->value.function.isym != NULL
1814 && gfc_intrinsic_func_interface (p, 1) == MATCH_ERROR)
1815 return FAILURE;
1817 break;
1819 case EXPR_SUBSTRING:
1820 if (simplify_ref_chain (p->ref, type) == FAILURE)
1821 return FAILURE;
1823 if (gfc_is_constant_expr (p))
1825 gfc_char_t *s;
1826 int start, end;
1828 start = 0;
1829 if (p->ref && p->ref->u.ss.start)
1831 gfc_extract_int (p->ref->u.ss.start, &start);
1832 start--; /* Convert from one-based to zero-based. */
1835 end = p->value.character.length;
1836 if (p->ref && p->ref->u.ss.end)
1837 gfc_extract_int (p->ref->u.ss.end, &end);
1839 s = gfc_get_wide_string (end - start + 2);
1840 memcpy (s, p->value.character.string + start,
1841 (end - start) * sizeof (gfc_char_t));
1842 s[end - start + 1] = '\0'; /* TODO: C-style string. */
1843 gfc_free (p->value.character.string);
1844 p->value.character.string = s;
1845 p->value.character.length = end - start;
1846 p->ts.u.cl = gfc_new_charlen (gfc_current_ns, NULL);
1847 p->ts.u.cl->length = gfc_get_int_expr (gfc_default_integer_kind,
1848 NULL,
1849 p->value.character.length);
1850 gfc_free_ref_list (p->ref);
1851 p->ref = NULL;
1852 p->expr_type = EXPR_CONSTANT;
1854 break;
1856 case EXPR_OP:
1857 if (simplify_intrinsic_op (p, type) == FAILURE)
1858 return FAILURE;
1859 break;
1861 case EXPR_VARIABLE:
1862 /* Only substitute array parameter variables if we are in an
1863 initialization expression, or we want a subsection. */
1864 if (p->symtree->n.sym->attr.flavor == FL_PARAMETER
1865 && (gfc_init_expr_flag || p->ref
1866 || p->symtree->n.sym->value->expr_type != EXPR_ARRAY))
1868 if (simplify_parameter_variable (p, type) == FAILURE)
1869 return FAILURE;
1870 break;
1873 if (type == 1)
1875 gfc_simplify_iterator_var (p);
1878 /* Simplify subcomponent references. */
1879 if (simplify_ref_chain (p->ref, type) == FAILURE)
1880 return FAILURE;
1882 break;
1884 case EXPR_STRUCTURE:
1885 case EXPR_ARRAY:
1886 if (simplify_ref_chain (p->ref, type) == FAILURE)
1887 return FAILURE;
1889 if (simplify_constructor (p->value.constructor, type) == FAILURE)
1890 return FAILURE;
1892 if (p->expr_type == EXPR_ARRAY && p->ref && p->ref->type == REF_ARRAY
1893 && p->ref->u.ar.type == AR_FULL)
1894 gfc_expand_constructor (p, false);
1896 if (simplify_const_ref (p) == FAILURE)
1897 return FAILURE;
1899 break;
1901 case EXPR_COMPCALL:
1902 case EXPR_PPC:
1903 gcc_unreachable ();
1904 break;
1907 return SUCCESS;
1911 /* Returns the type of an expression with the exception that iterator
1912 variables are automatically integers no matter what else they may
1913 be declared as. */
1915 static bt
1916 et0 (gfc_expr *e)
1918 if (e->expr_type == EXPR_VARIABLE && gfc_check_iter_variable (e) == SUCCESS)
1919 return BT_INTEGER;
1921 return e->ts.type;
1925 /* Check an intrinsic arithmetic operation to see if it is consistent
1926 with some type of expression. */
1928 static gfc_try check_init_expr (gfc_expr *);
1931 /* Scalarize an expression for an elemental intrinsic call. */
1933 static gfc_try
1934 scalarize_intrinsic_call (gfc_expr *e)
1936 gfc_actual_arglist *a, *b;
1937 gfc_constructor_base ctor;
1938 gfc_constructor *args[5];
1939 gfc_constructor *ci, *new_ctor;
1940 gfc_expr *expr, *old;
1941 int n, i, rank[5], array_arg;
1943 /* Find which, if any, arguments are arrays. Assume that the old
1944 expression carries the type information and that the first arg
1945 that is an array expression carries all the shape information.*/
1946 n = array_arg = 0;
1947 a = e->value.function.actual;
1948 for (; a; a = a->next)
1950 n++;
1951 if (a->expr->expr_type != EXPR_ARRAY)
1952 continue;
1953 array_arg = n;
1954 expr = gfc_copy_expr (a->expr);
1955 break;
1958 if (!array_arg)
1959 return FAILURE;
1961 old = gfc_copy_expr (e);
1963 gfc_constructor_free (expr->value.constructor);
1964 expr->value.constructor = NULL;
1965 expr->ts = old->ts;
1966 expr->where = old->where;
1967 expr->expr_type = EXPR_ARRAY;
1969 /* Copy the array argument constructors into an array, with nulls
1970 for the scalars. */
1971 n = 0;
1972 a = old->value.function.actual;
1973 for (; a; a = a->next)
1975 /* Check that this is OK for an initialization expression. */
1976 if (a->expr && check_init_expr (a->expr) == FAILURE)
1977 goto cleanup;
1979 rank[n] = 0;
1980 if (a->expr && a->expr->rank && a->expr->expr_type == EXPR_VARIABLE)
1982 rank[n] = a->expr->rank;
1983 ctor = a->expr->symtree->n.sym->value->value.constructor;
1984 args[n] = gfc_constructor_first (ctor);
1986 else if (a->expr && a->expr->expr_type == EXPR_ARRAY)
1988 if (a->expr->rank)
1989 rank[n] = a->expr->rank;
1990 else
1991 rank[n] = 1;
1992 ctor = gfc_constructor_copy (a->expr->value.constructor);
1993 args[n] = gfc_constructor_first (ctor);
1995 else
1996 args[n] = NULL;
1998 n++;
2002 /* Using the array argument as the master, step through the array
2003 calling the function for each element and advancing the array
2004 constructors together. */
2005 for (ci = args[array_arg - 1]; ci; ci = gfc_constructor_next (ci))
2007 new_ctor = gfc_constructor_append_expr (&expr->value.constructor,
2008 gfc_copy_expr (old), NULL);
2010 gfc_free_actual_arglist (new_ctor->expr->value.function.actual);
2011 a = NULL;
2012 b = old->value.function.actual;
2013 for (i = 0; i < n; i++)
2015 if (a == NULL)
2016 new_ctor->expr->value.function.actual
2017 = a = gfc_get_actual_arglist ();
2018 else
2020 a->next = gfc_get_actual_arglist ();
2021 a = a->next;
2024 if (args[i])
2025 a->expr = gfc_copy_expr (args[i]->expr);
2026 else
2027 a->expr = gfc_copy_expr (b->expr);
2029 b = b->next;
2032 /* Simplify the function calls. If the simplification fails, the
2033 error will be flagged up down-stream or the library will deal
2034 with it. */
2035 gfc_simplify_expr (new_ctor->expr, 0);
2037 for (i = 0; i < n; i++)
2038 if (args[i])
2039 args[i] = gfc_constructor_next (args[i]);
2041 for (i = 1; i < n; i++)
2042 if (rank[i] && ((args[i] != NULL && args[array_arg - 1] == NULL)
2043 || (args[i] == NULL && args[array_arg - 1] != NULL)))
2044 goto compliance;
2047 free_expr0 (e);
2048 *e = *expr;
2049 gfc_free_expr (old);
2050 return SUCCESS;
2052 compliance:
2053 gfc_error_now ("elemental function arguments at %C are not compliant");
2055 cleanup:
2056 gfc_free_expr (expr);
2057 gfc_free_expr (old);
2058 return FAILURE;
2062 static gfc_try
2063 check_intrinsic_op (gfc_expr *e, gfc_try (*check_function) (gfc_expr *))
2065 gfc_expr *op1 = e->value.op.op1;
2066 gfc_expr *op2 = e->value.op.op2;
2068 if ((*check_function) (op1) == FAILURE)
2069 return FAILURE;
2071 switch (e->value.op.op)
2073 case INTRINSIC_UPLUS:
2074 case INTRINSIC_UMINUS:
2075 if (!numeric_type (et0 (op1)))
2076 goto not_numeric;
2077 break;
2079 case INTRINSIC_EQ:
2080 case INTRINSIC_EQ_OS:
2081 case INTRINSIC_NE:
2082 case INTRINSIC_NE_OS:
2083 case INTRINSIC_GT:
2084 case INTRINSIC_GT_OS:
2085 case INTRINSIC_GE:
2086 case INTRINSIC_GE_OS:
2087 case INTRINSIC_LT:
2088 case INTRINSIC_LT_OS:
2089 case INTRINSIC_LE:
2090 case INTRINSIC_LE_OS:
2091 if ((*check_function) (op2) == FAILURE)
2092 return FAILURE;
2094 if (!(et0 (op1) == BT_CHARACTER && et0 (op2) == BT_CHARACTER)
2095 && !(numeric_type (et0 (op1)) && numeric_type (et0 (op2))))
2097 gfc_error ("Numeric or CHARACTER operands are required in "
2098 "expression at %L", &e->where);
2099 return FAILURE;
2101 break;
2103 case INTRINSIC_PLUS:
2104 case INTRINSIC_MINUS:
2105 case INTRINSIC_TIMES:
2106 case INTRINSIC_DIVIDE:
2107 case INTRINSIC_POWER:
2108 if ((*check_function) (op2) == FAILURE)
2109 return FAILURE;
2111 if (!numeric_type (et0 (op1)) || !numeric_type (et0 (op2)))
2112 goto not_numeric;
2114 break;
2116 case INTRINSIC_CONCAT:
2117 if ((*check_function) (op2) == FAILURE)
2118 return FAILURE;
2120 if (et0 (op1) != BT_CHARACTER || et0 (op2) != BT_CHARACTER)
2122 gfc_error ("Concatenation operator in expression at %L "
2123 "must have two CHARACTER operands", &op1->where);
2124 return FAILURE;
2127 if (op1->ts.kind != op2->ts.kind)
2129 gfc_error ("Concat operator at %L must concatenate strings of the "
2130 "same kind", &e->where);
2131 return FAILURE;
2134 break;
2136 case INTRINSIC_NOT:
2137 if (et0 (op1) != BT_LOGICAL)
2139 gfc_error (".NOT. operator in expression at %L must have a LOGICAL "
2140 "operand", &op1->where);
2141 return FAILURE;
2144 break;
2146 case INTRINSIC_AND:
2147 case INTRINSIC_OR:
2148 case INTRINSIC_EQV:
2149 case INTRINSIC_NEQV:
2150 if ((*check_function) (op2) == FAILURE)
2151 return FAILURE;
2153 if (et0 (op1) != BT_LOGICAL || et0 (op2) != BT_LOGICAL)
2155 gfc_error ("LOGICAL operands are required in expression at %L",
2156 &e->where);
2157 return FAILURE;
2160 break;
2162 case INTRINSIC_PARENTHESES:
2163 break;
2165 default:
2166 gfc_error ("Only intrinsic operators can be used in expression at %L",
2167 &e->where);
2168 return FAILURE;
2171 return SUCCESS;
2173 not_numeric:
2174 gfc_error ("Numeric operands are required in expression at %L", &e->where);
2176 return FAILURE;
2179 /* F2003, 7.1.7 (3): In init expression, allocatable components
2180 must not be data-initialized. */
2181 static gfc_try
2182 check_alloc_comp_init (gfc_expr *e)
2184 gfc_component *comp;
2185 gfc_constructor *ctor;
2187 gcc_assert (e->expr_type == EXPR_STRUCTURE);
2188 gcc_assert (e->ts.type == BT_DERIVED);
2190 for (comp = e->ts.u.derived->components,
2191 ctor = gfc_constructor_first (e->value.constructor);
2192 comp; comp = comp->next, ctor = gfc_constructor_next (ctor))
2194 if (comp->attr.allocatable
2195 && ctor->expr->expr_type != EXPR_NULL)
2197 gfc_error("Invalid initialization expression for ALLOCATABLE "
2198 "component '%s' in structure constructor at %L",
2199 comp->name, &ctor->expr->where);
2200 return FAILURE;
2204 return SUCCESS;
2207 static match
2208 check_init_expr_arguments (gfc_expr *e)
2210 gfc_actual_arglist *ap;
2212 for (ap = e->value.function.actual; ap; ap = ap->next)
2213 if (check_init_expr (ap->expr) == FAILURE)
2214 return MATCH_ERROR;
2216 return MATCH_YES;
2219 static gfc_try check_restricted (gfc_expr *);
2221 /* F95, 7.1.6.1, Initialization expressions, (7)
2222 F2003, 7.1.7 Initialization expression, (8) */
2224 static match
2225 check_inquiry (gfc_expr *e, int not_restricted)
2227 const char *name;
2228 const char *const *functions;
2230 static const char *const inquiry_func_f95[] = {
2231 "lbound", "shape", "size", "ubound",
2232 "bit_size", "len", "kind",
2233 "digits", "epsilon", "huge", "maxexponent", "minexponent",
2234 "precision", "radix", "range", "tiny",
2235 NULL
2238 static const char *const inquiry_func_f2003[] = {
2239 "lbound", "shape", "size", "ubound",
2240 "bit_size", "len", "kind",
2241 "digits", "epsilon", "huge", "maxexponent", "minexponent",
2242 "precision", "radix", "range", "tiny",
2243 "new_line", NULL
2246 int i;
2247 gfc_actual_arglist *ap;
2249 if (!e->value.function.isym
2250 || !e->value.function.isym->inquiry)
2251 return MATCH_NO;
2253 /* An undeclared parameter will get us here (PR25018). */
2254 if (e->symtree == NULL)
2255 return MATCH_NO;
2257 name = e->symtree->n.sym->name;
2259 functions = (gfc_option.warn_std & GFC_STD_F2003)
2260 ? inquiry_func_f2003 : inquiry_func_f95;
2262 for (i = 0; functions[i]; i++)
2263 if (strcmp (functions[i], name) == 0)
2264 break;
2266 if (functions[i] == NULL)
2267 return MATCH_ERROR;
2269 /* At this point we have an inquiry function with a variable argument. The
2270 type of the variable might be undefined, but we need it now, because the
2271 arguments of these functions are not allowed to be undefined. */
2273 for (ap = e->value.function.actual; ap; ap = ap->next)
2275 if (!ap->expr)
2276 continue;
2278 if (ap->expr->ts.type == BT_UNKNOWN)
2280 if (ap->expr->symtree->n.sym->ts.type == BT_UNKNOWN
2281 && gfc_set_default_type (ap->expr->symtree->n.sym, 0, gfc_current_ns)
2282 == FAILURE)
2283 return MATCH_NO;
2285 ap->expr->ts = ap->expr->symtree->n.sym->ts;
2288 /* Assumed character length will not reduce to a constant expression
2289 with LEN, as required by the standard. */
2290 if (i == 5 && not_restricted
2291 && ap->expr->symtree->n.sym->ts.type == BT_CHARACTER
2292 && (ap->expr->symtree->n.sym->ts.u.cl->length == NULL
2293 || ap->expr->symtree->n.sym->ts.deferred))
2295 gfc_error ("Assumed or deferred character length variable '%s' "
2296 " in constant expression at %L",
2297 ap->expr->symtree->n.sym->name,
2298 &ap->expr->where);
2299 return MATCH_ERROR;
2301 else if (not_restricted && check_init_expr (ap->expr) == FAILURE)
2302 return MATCH_ERROR;
2304 if (not_restricted == 0
2305 && ap->expr->expr_type != EXPR_VARIABLE
2306 && check_restricted (ap->expr) == FAILURE)
2307 return MATCH_ERROR;
2309 if (not_restricted == 0
2310 && ap->expr->expr_type == EXPR_VARIABLE
2311 && ap->expr->symtree->n.sym->attr.dummy
2312 && ap->expr->symtree->n.sym->attr.optional)
2313 return MATCH_NO;
2316 return MATCH_YES;
2320 /* F95, 7.1.6.1, Initialization expressions, (5)
2321 F2003, 7.1.7 Initialization expression, (5) */
2323 static match
2324 check_transformational (gfc_expr *e)
2326 static const char * const trans_func_f95[] = {
2327 "repeat", "reshape", "selected_int_kind",
2328 "selected_real_kind", "transfer", "trim", NULL
2331 static const char * const trans_func_f2003[] = {
2332 "all", "any", "count", "dot_product", "matmul", "null", "pack",
2333 "product", "repeat", "reshape", "selected_char_kind", "selected_int_kind",
2334 "selected_real_kind", "spread", "sum", "transfer", "transpose",
2335 "trim", "unpack", NULL
2338 int i;
2339 const char *name;
2340 const char *const *functions;
2342 if (!e->value.function.isym
2343 || !e->value.function.isym->transformational)
2344 return MATCH_NO;
2346 name = e->symtree->n.sym->name;
2348 functions = (gfc_option.allow_std & GFC_STD_F2003)
2349 ? trans_func_f2003 : trans_func_f95;
2351 /* NULL() is dealt with below. */
2352 if (strcmp ("null", name) == 0)
2353 return MATCH_NO;
2355 for (i = 0; functions[i]; i++)
2356 if (strcmp (functions[i], name) == 0)
2357 break;
2359 if (functions[i] == NULL)
2361 gfc_error("transformational intrinsic '%s' at %L is not permitted "
2362 "in an initialization expression", name, &e->where);
2363 return MATCH_ERROR;
2366 return check_init_expr_arguments (e);
2370 /* F95, 7.1.6.1, Initialization expressions, (6)
2371 F2003, 7.1.7 Initialization expression, (6) */
2373 static match
2374 check_null (gfc_expr *e)
2376 if (strcmp ("null", e->symtree->n.sym->name) != 0)
2377 return MATCH_NO;
2379 return check_init_expr_arguments (e);
2383 static match
2384 check_elemental (gfc_expr *e)
2386 if (!e->value.function.isym
2387 || !e->value.function.isym->elemental)
2388 return MATCH_NO;
2390 if (e->ts.type != BT_INTEGER
2391 && e->ts.type != BT_CHARACTER
2392 && gfc_notify_std (GFC_STD_F2003, "Extension: Evaluation of "
2393 "nonstandard initialization expression at %L",
2394 &e->where) == FAILURE)
2395 return MATCH_ERROR;
2397 return check_init_expr_arguments (e);
2401 static match
2402 check_conversion (gfc_expr *e)
2404 if (!e->value.function.isym
2405 || !e->value.function.isym->conversion)
2406 return MATCH_NO;
2408 return check_init_expr_arguments (e);
2412 /* Verify that an expression is an initialization expression. A side
2413 effect is that the expression tree is reduced to a single constant
2414 node if all goes well. This would normally happen when the
2415 expression is constructed but function references are assumed to be
2416 intrinsics in the context of initialization expressions. If
2417 FAILURE is returned an error message has been generated. */
2419 static gfc_try
2420 check_init_expr (gfc_expr *e)
2422 match m;
2423 gfc_try t;
2425 if (e == NULL)
2426 return SUCCESS;
2428 switch (e->expr_type)
2430 case EXPR_OP:
2431 t = check_intrinsic_op (e, check_init_expr);
2432 if (t == SUCCESS)
2433 t = gfc_simplify_expr (e, 0);
2435 break;
2437 case EXPR_FUNCTION:
2438 t = FAILURE;
2441 gfc_intrinsic_sym* isym;
2442 gfc_symbol* sym;
2444 sym = e->symtree->n.sym;
2445 if (!gfc_is_intrinsic (sym, 0, e->where)
2446 || (m = gfc_intrinsic_func_interface (e, 0)) != MATCH_YES)
2448 gfc_error ("Function '%s' in initialization expression at %L "
2449 "must be an intrinsic function",
2450 e->symtree->n.sym->name, &e->where);
2451 break;
2454 if ((m = check_conversion (e)) == MATCH_NO
2455 && (m = check_inquiry (e, 1)) == MATCH_NO
2456 && (m = check_null (e)) == MATCH_NO
2457 && (m = check_transformational (e)) == MATCH_NO
2458 && (m = check_elemental (e)) == MATCH_NO)
2460 gfc_error ("Intrinsic function '%s' at %L is not permitted "
2461 "in an initialization expression",
2462 e->symtree->n.sym->name, &e->where);
2463 m = MATCH_ERROR;
2466 /* Try to scalarize an elemental intrinsic function that has an
2467 array argument. */
2468 isym = gfc_find_function (e->symtree->n.sym->name);
2469 if (isym && isym->elemental
2470 && (t = scalarize_intrinsic_call (e)) == SUCCESS)
2471 break;
2474 if (m == MATCH_YES)
2475 t = gfc_simplify_expr (e, 0);
2477 break;
2479 case EXPR_VARIABLE:
2480 t = SUCCESS;
2482 if (gfc_check_iter_variable (e) == SUCCESS)
2483 break;
2485 if (e->symtree->n.sym->attr.flavor == FL_PARAMETER)
2487 /* A PARAMETER shall not be used to define itself, i.e.
2488 REAL, PARAMETER :: x = transfer(0, x)
2489 is invalid. */
2490 if (!e->symtree->n.sym->value)
2492 gfc_error("PARAMETER '%s' is used at %L before its definition "
2493 "is complete", e->symtree->n.sym->name, &e->where);
2494 t = FAILURE;
2496 else
2497 t = simplify_parameter_variable (e, 0);
2499 break;
2502 if (gfc_in_match_data ())
2503 break;
2505 t = FAILURE;
2507 if (e->symtree->n.sym->as)
2509 switch (e->symtree->n.sym->as->type)
2511 case AS_ASSUMED_SIZE:
2512 gfc_error ("Assumed size array '%s' at %L is not permitted "
2513 "in an initialization expression",
2514 e->symtree->n.sym->name, &e->where);
2515 break;
2517 case AS_ASSUMED_SHAPE:
2518 gfc_error ("Assumed shape array '%s' at %L is not permitted "
2519 "in an initialization expression",
2520 e->symtree->n.sym->name, &e->where);
2521 break;
2523 case AS_DEFERRED:
2524 gfc_error ("Deferred array '%s' at %L is not permitted "
2525 "in an initialization expression",
2526 e->symtree->n.sym->name, &e->where);
2527 break;
2529 case AS_EXPLICIT:
2530 gfc_error ("Array '%s' at %L is a variable, which does "
2531 "not reduce to a constant expression",
2532 e->symtree->n.sym->name, &e->where);
2533 break;
2535 default:
2536 gcc_unreachable();
2539 else
2540 gfc_error ("Parameter '%s' at %L has not been declared or is "
2541 "a variable, which does not reduce to a constant "
2542 "expression", e->symtree->n.sym->name, &e->where);
2544 break;
2546 case EXPR_CONSTANT:
2547 case EXPR_NULL:
2548 t = SUCCESS;
2549 break;
2551 case EXPR_SUBSTRING:
2552 t = check_init_expr (e->ref->u.ss.start);
2553 if (t == FAILURE)
2554 break;
2556 t = check_init_expr (e->ref->u.ss.end);
2557 if (t == SUCCESS)
2558 t = gfc_simplify_expr (e, 0);
2560 break;
2562 case EXPR_STRUCTURE:
2563 t = e->ts.is_iso_c ? SUCCESS : FAILURE;
2564 if (t == SUCCESS)
2565 break;
2567 t = check_alloc_comp_init (e);
2568 if (t == FAILURE)
2569 break;
2571 t = gfc_check_constructor (e, check_init_expr);
2572 if (t == FAILURE)
2573 break;
2575 break;
2577 case EXPR_ARRAY:
2578 t = gfc_check_constructor (e, check_init_expr);
2579 if (t == FAILURE)
2580 break;
2582 t = gfc_expand_constructor (e, true);
2583 if (t == FAILURE)
2584 break;
2586 t = gfc_check_constructor_type (e);
2587 break;
2589 default:
2590 gfc_internal_error ("check_init_expr(): Unknown expression type");
2593 return t;
2596 /* Reduces a general expression to an initialization expression (a constant).
2597 This used to be part of gfc_match_init_expr.
2598 Note that this function doesn't free the given expression on FAILURE. */
2600 gfc_try
2601 gfc_reduce_init_expr (gfc_expr *expr)
2603 gfc_try t;
2605 gfc_init_expr_flag = true;
2606 t = gfc_resolve_expr (expr);
2607 if (t == SUCCESS)
2608 t = check_init_expr (expr);
2609 gfc_init_expr_flag = false;
2611 if (t == FAILURE)
2612 return FAILURE;
2614 if (expr->expr_type == EXPR_ARRAY)
2616 if (gfc_check_constructor_type (expr) == FAILURE)
2617 return FAILURE;
2618 if (gfc_expand_constructor (expr, true) == FAILURE)
2619 return FAILURE;
2622 return SUCCESS;
2626 /* Match an initialization expression. We work by first matching an
2627 expression, then reducing it to a constant. */
2629 match
2630 gfc_match_init_expr (gfc_expr **result)
2632 gfc_expr *expr;
2633 match m;
2634 gfc_try t;
2636 expr = NULL;
2638 gfc_init_expr_flag = true;
2640 m = gfc_match_expr (&expr);
2641 if (m != MATCH_YES)
2643 gfc_init_expr_flag = false;
2644 return m;
2647 t = gfc_reduce_init_expr (expr);
2648 if (t != SUCCESS)
2650 gfc_free_expr (expr);
2651 gfc_init_expr_flag = false;
2652 return MATCH_ERROR;
2655 *result = expr;
2656 gfc_init_expr_flag = false;
2658 return MATCH_YES;
2662 /* Given an actual argument list, test to see that each argument is a
2663 restricted expression and optionally if the expression type is
2664 integer or character. */
2666 static gfc_try
2667 restricted_args (gfc_actual_arglist *a)
2669 for (; a; a = a->next)
2671 if (check_restricted (a->expr) == FAILURE)
2672 return FAILURE;
2675 return SUCCESS;
2679 /************* Restricted/specification expressions *************/
2682 /* Make sure a non-intrinsic function is a specification function. */
2684 static gfc_try
2685 external_spec_function (gfc_expr *e)
2687 gfc_symbol *f;
2689 f = e->value.function.esym;
2691 if (f->attr.proc == PROC_ST_FUNCTION)
2693 gfc_error ("Specification function '%s' at %L cannot be a statement "
2694 "function", f->name, &e->where);
2695 return FAILURE;
2698 if (f->attr.proc == PROC_INTERNAL)
2700 gfc_error ("Specification function '%s' at %L cannot be an internal "
2701 "function", f->name, &e->where);
2702 return FAILURE;
2705 if (!f->attr.pure && !f->attr.elemental)
2707 gfc_error ("Specification function '%s' at %L must be PURE", f->name,
2708 &e->where);
2709 return FAILURE;
2712 if (f->attr.recursive)
2714 gfc_error ("Specification function '%s' at %L cannot be RECURSIVE",
2715 f->name, &e->where);
2716 return FAILURE;
2719 return restricted_args (e->value.function.actual);
2723 /* Check to see that a function reference to an intrinsic is a
2724 restricted expression. */
2726 static gfc_try
2727 restricted_intrinsic (gfc_expr *e)
2729 /* TODO: Check constraints on inquiry functions. 7.1.6.2 (7). */
2730 if (check_inquiry (e, 0) == MATCH_YES)
2731 return SUCCESS;
2733 return restricted_args (e->value.function.actual);
2737 /* Check the expressions of an actual arglist. Used by check_restricted. */
2739 static gfc_try
2740 check_arglist (gfc_actual_arglist* arg, gfc_try (*checker) (gfc_expr*))
2742 for (; arg; arg = arg->next)
2743 if (checker (arg->expr) == FAILURE)
2744 return FAILURE;
2746 return SUCCESS;
2750 /* Check the subscription expressions of a reference chain with a checking
2751 function; used by check_restricted. */
2753 static gfc_try
2754 check_references (gfc_ref* ref, gfc_try (*checker) (gfc_expr*))
2756 int dim;
2758 if (!ref)
2759 return SUCCESS;
2761 switch (ref->type)
2763 case REF_ARRAY:
2764 for (dim = 0; dim != ref->u.ar.dimen; ++dim)
2766 if (checker (ref->u.ar.start[dim]) == FAILURE)
2767 return FAILURE;
2768 if (checker (ref->u.ar.end[dim]) == FAILURE)
2769 return FAILURE;
2770 if (checker (ref->u.ar.stride[dim]) == FAILURE)
2771 return FAILURE;
2773 break;
2775 case REF_COMPONENT:
2776 /* Nothing needed, just proceed to next reference. */
2777 break;
2779 case REF_SUBSTRING:
2780 if (checker (ref->u.ss.start) == FAILURE)
2781 return FAILURE;
2782 if (checker (ref->u.ss.end) == FAILURE)
2783 return FAILURE;
2784 break;
2786 default:
2787 gcc_unreachable ();
2788 break;
2791 return check_references (ref->next, checker);
2795 /* Verify that an expression is a restricted expression. Like its
2796 cousin check_init_expr(), an error message is generated if we
2797 return FAILURE. */
2799 static gfc_try
2800 check_restricted (gfc_expr *e)
2802 gfc_symbol* sym;
2803 gfc_try t;
2805 if (e == NULL)
2806 return SUCCESS;
2808 switch (e->expr_type)
2810 case EXPR_OP:
2811 t = check_intrinsic_op (e, check_restricted);
2812 if (t == SUCCESS)
2813 t = gfc_simplify_expr (e, 0);
2815 break;
2817 case EXPR_FUNCTION:
2818 if (e->value.function.esym)
2820 t = check_arglist (e->value.function.actual, &check_restricted);
2821 if (t == SUCCESS)
2822 t = external_spec_function (e);
2824 else
2826 if (e->value.function.isym && e->value.function.isym->inquiry)
2827 t = SUCCESS;
2828 else
2829 t = check_arglist (e->value.function.actual, &check_restricted);
2831 if (t == SUCCESS)
2832 t = restricted_intrinsic (e);
2834 break;
2836 case EXPR_VARIABLE:
2837 sym = e->symtree->n.sym;
2838 t = FAILURE;
2840 /* If a dummy argument appears in a context that is valid for a
2841 restricted expression in an elemental procedure, it will have
2842 already been simplified away once we get here. Therefore we
2843 don't need to jump through hoops to distinguish valid from
2844 invalid cases. */
2845 if (sym->attr.dummy && sym->ns == gfc_current_ns
2846 && sym->ns->proc_name && sym->ns->proc_name->attr.elemental)
2848 gfc_error ("Dummy argument '%s' not allowed in expression at %L",
2849 sym->name, &e->where);
2850 break;
2853 if (sym->attr.optional)
2855 gfc_error ("Dummy argument '%s' at %L cannot be OPTIONAL",
2856 sym->name, &e->where);
2857 break;
2860 if (sym->attr.intent == INTENT_OUT)
2862 gfc_error ("Dummy argument '%s' at %L cannot be INTENT(OUT)",
2863 sym->name, &e->where);
2864 break;
2867 /* Check reference chain if any. */
2868 if (check_references (e->ref, &check_restricted) == FAILURE)
2869 break;
2871 /* gfc_is_formal_arg broadcasts that a formal argument list is being
2872 processed in resolve.c(resolve_formal_arglist). This is done so
2873 that host associated dummy array indices are accepted (PR23446).
2874 This mechanism also does the same for the specification expressions
2875 of array-valued functions. */
2876 if (e->error
2877 || sym->attr.in_common
2878 || sym->attr.use_assoc
2879 || sym->attr.dummy
2880 || sym->attr.implied_index
2881 || sym->attr.flavor == FL_PARAMETER
2882 || (sym->ns && sym->ns == gfc_current_ns->parent)
2883 || (sym->ns && gfc_current_ns->parent
2884 && sym->ns == gfc_current_ns->parent->parent)
2885 || (sym->ns->proc_name != NULL
2886 && sym->ns->proc_name->attr.flavor == FL_MODULE)
2887 || (gfc_is_formal_arg () && (sym->ns == gfc_current_ns)))
2889 t = SUCCESS;
2890 break;
2893 gfc_error ("Variable '%s' cannot appear in the expression at %L",
2894 sym->name, &e->where);
2895 /* Prevent a repetition of the error. */
2896 e->error = 1;
2897 break;
2899 case EXPR_NULL:
2900 case EXPR_CONSTANT:
2901 t = SUCCESS;
2902 break;
2904 case EXPR_SUBSTRING:
2905 t = gfc_specification_expr (e->ref->u.ss.start);
2906 if (t == FAILURE)
2907 break;
2909 t = gfc_specification_expr (e->ref->u.ss.end);
2910 if (t == SUCCESS)
2911 t = gfc_simplify_expr (e, 0);
2913 break;
2915 case EXPR_STRUCTURE:
2916 t = gfc_check_constructor (e, check_restricted);
2917 break;
2919 case EXPR_ARRAY:
2920 t = gfc_check_constructor (e, check_restricted);
2921 break;
2923 default:
2924 gfc_internal_error ("check_restricted(): Unknown expression type");
2927 return t;
2931 /* Check to see that an expression is a specification expression. If
2932 we return FAILURE, an error has been generated. */
2934 gfc_try
2935 gfc_specification_expr (gfc_expr *e)
2937 gfc_component *comp;
2939 if (e == NULL)
2940 return SUCCESS;
2942 if (e->ts.type != BT_INTEGER)
2944 gfc_error ("Expression at %L must be of INTEGER type, found %s",
2945 &e->where, gfc_basic_typename (e->ts.type));
2946 return FAILURE;
2949 if (e->expr_type == EXPR_FUNCTION
2950 && !e->value.function.isym
2951 && !e->value.function.esym
2952 && !gfc_pure (e->symtree->n.sym)
2953 && (!gfc_is_proc_ptr_comp (e, &comp)
2954 || !comp->attr.pure))
2956 gfc_error ("Function '%s' at %L must be PURE",
2957 e->symtree->n.sym->name, &e->where);
2958 /* Prevent repeat error messages. */
2959 e->symtree->n.sym->attr.pure = 1;
2960 return FAILURE;
2963 if (e->rank != 0)
2965 gfc_error ("Expression at %L must be scalar", &e->where);
2966 return FAILURE;
2969 if (gfc_simplify_expr (e, 0) == FAILURE)
2970 return FAILURE;
2972 return check_restricted (e);
2976 /************** Expression conformance checks. *************/
2978 /* Given two expressions, make sure that the arrays are conformable. */
2980 gfc_try
2981 gfc_check_conformance (gfc_expr *op1, gfc_expr *op2, const char *optype_msgid, ...)
2983 int op1_flag, op2_flag, d;
2984 mpz_t op1_size, op2_size;
2985 gfc_try t;
2987 va_list argp;
2988 char buffer[240];
2990 if (op1->rank == 0 || op2->rank == 0)
2991 return SUCCESS;
2993 va_start (argp, optype_msgid);
2994 vsnprintf (buffer, 240, optype_msgid, argp);
2995 va_end (argp);
2997 if (op1->rank != op2->rank)
2999 gfc_error ("Incompatible ranks in %s (%d and %d) at %L", _(buffer),
3000 op1->rank, op2->rank, &op1->where);
3001 return FAILURE;
3004 t = SUCCESS;
3006 for (d = 0; d < op1->rank; d++)
3008 op1_flag = gfc_array_dimen_size (op1, d, &op1_size) == SUCCESS;
3009 op2_flag = gfc_array_dimen_size (op2, d, &op2_size) == SUCCESS;
3011 if (op1_flag && op2_flag && mpz_cmp (op1_size, op2_size) != 0)
3013 gfc_error ("Different shape for %s at %L on dimension %d "
3014 "(%d and %d)", _(buffer), &op1->where, d + 1,
3015 (int) mpz_get_si (op1_size),
3016 (int) mpz_get_si (op2_size));
3018 t = FAILURE;
3021 if (op1_flag)
3022 mpz_clear (op1_size);
3023 if (op2_flag)
3024 mpz_clear (op2_size);
3026 if (t == FAILURE)
3027 return FAILURE;
3030 return SUCCESS;
3034 /* Given an assignable expression and an arbitrary expression, make
3035 sure that the assignment can take place. */
3037 gfc_try
3038 gfc_check_assign (gfc_expr *lvalue, gfc_expr *rvalue, int conform)
3040 gfc_symbol *sym;
3041 gfc_ref *ref;
3042 int has_pointer;
3044 sym = lvalue->symtree->n.sym;
3046 /* See if this is the component or subcomponent of a pointer. */
3047 has_pointer = sym->attr.pointer;
3048 for (ref = lvalue->ref; ref; ref = ref->next)
3049 if (ref->type == REF_COMPONENT && ref->u.c.component->attr.pointer)
3051 has_pointer = 1;
3052 break;
3055 /* 12.5.2.2, Note 12.26: The result variable is very similar to any other
3056 variable local to a function subprogram. Its existence begins when
3057 execution of the function is initiated and ends when execution of the
3058 function is terminated...
3059 Therefore, the left hand side is no longer a variable, when it is: */
3060 if (sym->attr.flavor == FL_PROCEDURE && sym->attr.proc != PROC_ST_FUNCTION
3061 && !sym->attr.external)
3063 bool bad_proc;
3064 bad_proc = false;
3066 /* (i) Use associated; */
3067 if (sym->attr.use_assoc)
3068 bad_proc = true;
3070 /* (ii) The assignment is in the main program; or */
3071 if (gfc_current_ns->proc_name->attr.is_main_program)
3072 bad_proc = true;
3074 /* (iii) A module or internal procedure... */
3075 if ((gfc_current_ns->proc_name->attr.proc == PROC_INTERNAL
3076 || gfc_current_ns->proc_name->attr.proc == PROC_MODULE)
3077 && gfc_current_ns->parent
3078 && (!(gfc_current_ns->parent->proc_name->attr.function
3079 || gfc_current_ns->parent->proc_name->attr.subroutine)
3080 || gfc_current_ns->parent->proc_name->attr.is_main_program))
3082 /* ... that is not a function... */
3083 if (!gfc_current_ns->proc_name->attr.function)
3084 bad_proc = true;
3086 /* ... or is not an entry and has a different name. */
3087 if (!sym->attr.entry && sym->name != gfc_current_ns->proc_name->name)
3088 bad_proc = true;
3091 /* (iv) Host associated and not the function symbol or the
3092 parent result. This picks up sibling references, which
3093 cannot be entries. */
3094 if (!sym->attr.entry
3095 && sym->ns == gfc_current_ns->parent
3096 && sym != gfc_current_ns->proc_name
3097 && sym != gfc_current_ns->parent->proc_name->result)
3098 bad_proc = true;
3100 if (bad_proc)
3102 gfc_error ("'%s' at %L is not a VALUE", sym->name, &lvalue->where);
3103 return FAILURE;
3107 if (rvalue->rank != 0 && lvalue->rank != rvalue->rank)
3109 gfc_error ("Incompatible ranks %d and %d in assignment at %L",
3110 lvalue->rank, rvalue->rank, &lvalue->where);
3111 return FAILURE;
3114 if (lvalue->ts.type == BT_UNKNOWN)
3116 gfc_error ("Variable type is UNKNOWN in assignment at %L",
3117 &lvalue->where);
3118 return FAILURE;
3121 if (rvalue->expr_type == EXPR_NULL)
3123 if (has_pointer && (ref == NULL || ref->next == NULL)
3124 && lvalue->symtree->n.sym->attr.data)
3125 return SUCCESS;
3126 else
3128 gfc_error ("NULL appears on right-hand side in assignment at %L",
3129 &rvalue->where);
3130 return FAILURE;
3134 /* This is possibly a typo: x = f() instead of x => f(). */
3135 if (gfc_option.warn_surprising
3136 && rvalue->expr_type == EXPR_FUNCTION
3137 && rvalue->symtree->n.sym->attr.pointer)
3138 gfc_warning ("POINTER valued function appears on right-hand side of "
3139 "assignment at %L", &rvalue->where);
3141 /* Check size of array assignments. */
3142 if (lvalue->rank != 0 && rvalue->rank != 0
3143 && gfc_check_conformance (lvalue, rvalue, "array assignment") != SUCCESS)
3144 return FAILURE;
3146 if (rvalue->is_boz && lvalue->ts.type != BT_INTEGER
3147 && lvalue->symtree->n.sym->attr.data
3148 && gfc_notify_std (GFC_STD_GNU, "Extension: BOZ literal at %L used to "
3149 "initialize non-integer variable '%s'",
3150 &rvalue->where, lvalue->symtree->n.sym->name)
3151 == FAILURE)
3152 return FAILURE;
3153 else if (rvalue->is_boz && !lvalue->symtree->n.sym->attr.data
3154 && gfc_notify_std (GFC_STD_GNU, "Extension: BOZ literal at %L outside "
3155 "a DATA statement and outside INT/REAL/DBLE/CMPLX",
3156 &rvalue->where) == FAILURE)
3157 return FAILURE;
3159 /* Handle the case of a BOZ literal on the RHS. */
3160 if (rvalue->is_boz && lvalue->ts.type != BT_INTEGER)
3162 int rc;
3163 if (gfc_option.warn_surprising)
3164 gfc_warning ("BOZ literal at %L is bitwise transferred "
3165 "non-integer symbol '%s'", &rvalue->where,
3166 lvalue->symtree->n.sym->name);
3167 if (!gfc_convert_boz (rvalue, &lvalue->ts))
3168 return FAILURE;
3169 if ((rc = gfc_range_check (rvalue)) != ARITH_OK)
3171 if (rc == ARITH_UNDERFLOW)
3172 gfc_error ("Arithmetic underflow of bit-wise transferred BOZ at %L"
3173 ". This check can be disabled with the option "
3174 "-fno-range-check", &rvalue->where);
3175 else if (rc == ARITH_OVERFLOW)
3176 gfc_error ("Arithmetic overflow of bit-wise transferred BOZ at %L"
3177 ". This check can be disabled with the option "
3178 "-fno-range-check", &rvalue->where);
3179 else if (rc == ARITH_NAN)
3180 gfc_error ("Arithmetic NaN of bit-wise transferred BOZ at %L"
3181 ". This check can be disabled with the option "
3182 "-fno-range-check", &rvalue->where);
3183 return FAILURE;
3187 if (gfc_compare_types (&lvalue->ts, &rvalue->ts))
3188 return SUCCESS;
3190 /* Only DATA Statements come here. */
3191 if (!conform)
3193 /* Numeric can be converted to any other numeric. And Hollerith can be
3194 converted to any other type. */
3195 if ((gfc_numeric_ts (&lvalue->ts) && gfc_numeric_ts (&rvalue->ts))
3196 || rvalue->ts.type == BT_HOLLERITH)
3197 return SUCCESS;
3199 if (lvalue->ts.type == BT_LOGICAL && rvalue->ts.type == BT_LOGICAL)
3200 return SUCCESS;
3202 gfc_error ("Incompatible types in DATA statement at %L; attempted "
3203 "conversion of %s to %s", &lvalue->where,
3204 gfc_typename (&rvalue->ts), gfc_typename (&lvalue->ts));
3206 return FAILURE;
3209 /* Assignment is the only case where character variables of different
3210 kind values can be converted into one another. */
3211 if (lvalue->ts.type == BT_CHARACTER && rvalue->ts.type == BT_CHARACTER)
3213 if (lvalue->ts.kind != rvalue->ts.kind)
3214 gfc_convert_chartype (rvalue, &lvalue->ts);
3216 return SUCCESS;
3219 return gfc_convert_type (rvalue, &lvalue->ts, 1);
3223 /* Check that a pointer assignment is OK. We first check lvalue, and
3224 we only check rvalue if it's not an assignment to NULL() or a
3225 NULLIFY statement. */
3227 gfc_try
3228 gfc_check_pointer_assign (gfc_expr *lvalue, gfc_expr *rvalue)
3230 symbol_attribute attr;
3231 gfc_ref *ref;
3232 bool is_pure, is_implicit_pure, rank_remap;
3233 int proc_pointer;
3235 if (lvalue->symtree->n.sym->ts.type == BT_UNKNOWN
3236 && !lvalue->symtree->n.sym->attr.proc_pointer)
3238 gfc_error ("Pointer assignment target is not a POINTER at %L",
3239 &lvalue->where);
3240 return FAILURE;
3243 if (lvalue->symtree->n.sym->attr.flavor == FL_PROCEDURE
3244 && lvalue->symtree->n.sym->attr.use_assoc
3245 && !lvalue->symtree->n.sym->attr.proc_pointer)
3247 gfc_error ("'%s' in the pointer assignment at %L cannot be an "
3248 "l-value since it is a procedure",
3249 lvalue->symtree->n.sym->name, &lvalue->where);
3250 return FAILURE;
3253 proc_pointer = lvalue->symtree->n.sym->attr.proc_pointer;
3255 rank_remap = false;
3256 for (ref = lvalue->ref; ref; ref = ref->next)
3258 if (ref->type == REF_COMPONENT)
3259 proc_pointer = ref->u.c.component->attr.proc_pointer;
3261 if (ref->type == REF_ARRAY && ref->next == NULL)
3263 int dim;
3265 if (ref->u.ar.type == AR_FULL)
3266 break;
3268 if (ref->u.ar.type != AR_SECTION)
3270 gfc_error ("Expected bounds specification for '%s' at %L",
3271 lvalue->symtree->n.sym->name, &lvalue->where);
3272 return FAILURE;
3275 if (gfc_notify_std (GFC_STD_F2003,"Fortran 2003: Bounds "
3276 "specification for '%s' in pointer assignment "
3277 "at %L", lvalue->symtree->n.sym->name,
3278 &lvalue->where) == FAILURE)
3279 return FAILURE;
3281 /* When bounds are given, all lbounds are necessary and either all
3282 or none of the upper bounds; no strides are allowed. If the
3283 upper bounds are present, we may do rank remapping. */
3284 for (dim = 0; dim < ref->u.ar.dimen; ++dim)
3286 if (!ref->u.ar.start[dim])
3288 gfc_error ("Lower bound has to be present at %L",
3289 &lvalue->where);
3290 return FAILURE;
3292 if (ref->u.ar.stride[dim])
3294 gfc_error ("Stride must not be present at %L",
3295 &lvalue->where);
3296 return FAILURE;
3299 if (dim == 0)
3300 rank_remap = (ref->u.ar.end[dim] != NULL);
3301 else
3303 if ((rank_remap && !ref->u.ar.end[dim])
3304 || (!rank_remap && ref->u.ar.end[dim]))
3306 gfc_error ("Either all or none of the upper bounds"
3307 " must be specified at %L", &lvalue->where);
3308 return FAILURE;
3315 is_pure = gfc_pure (NULL);
3316 is_implicit_pure = gfc_implicit_pure (NULL);
3318 /* If rvalue is a NULL() or NULLIFY, we're done. Otherwise the type,
3319 kind, etc for lvalue and rvalue must match, and rvalue must be a
3320 pure variable if we're in a pure function. */
3321 if (rvalue->expr_type == EXPR_NULL && rvalue->ts.type == BT_UNKNOWN)
3322 return SUCCESS;
3324 /* F2008, C723 (pointer) and C726 (proc-pointer); for PURE also C1283. */
3325 if (lvalue->expr_type == EXPR_VARIABLE
3326 && gfc_is_coindexed (lvalue))
3328 gfc_ref *ref;
3329 for (ref = lvalue->ref; ref; ref = ref->next)
3330 if (ref->type == REF_ARRAY && ref->u.ar.codimen)
3332 gfc_error ("Pointer object at %L shall not have a coindex",
3333 &lvalue->where);
3334 return FAILURE;
3338 /* Checks on rvalue for procedure pointer assignments. */
3339 if (proc_pointer)
3341 char err[200];
3342 gfc_symbol *s1,*s2;
3343 gfc_component *comp;
3344 const char *name;
3346 attr = gfc_expr_attr (rvalue);
3347 if (!((rvalue->expr_type == EXPR_NULL)
3348 || (rvalue->expr_type == EXPR_FUNCTION && attr.proc_pointer)
3349 || (rvalue->expr_type == EXPR_VARIABLE && attr.proc_pointer)
3350 || (rvalue->expr_type == EXPR_VARIABLE
3351 && attr.flavor == FL_PROCEDURE)))
3353 gfc_error ("Invalid procedure pointer assignment at %L",
3354 &rvalue->where);
3355 return FAILURE;
3357 if (attr.abstract)
3359 gfc_error ("Abstract interface '%s' is invalid "
3360 "in procedure pointer assignment at %L",
3361 rvalue->symtree->name, &rvalue->where);
3362 return FAILURE;
3364 /* Check for C727. */
3365 if (attr.flavor == FL_PROCEDURE)
3367 if (attr.proc == PROC_ST_FUNCTION)
3369 gfc_error ("Statement function '%s' is invalid "
3370 "in procedure pointer assignment at %L",
3371 rvalue->symtree->name, &rvalue->where);
3372 return FAILURE;
3374 if (attr.proc == PROC_INTERNAL &&
3375 gfc_notify_std (GFC_STD_F2008, "Internal procedure '%s' is "
3376 "invalid in procedure pointer assignment at %L",
3377 rvalue->symtree->name, &rvalue->where) == FAILURE)
3378 return FAILURE;
3381 /* Ensure that the calling convention is the same. As other attributes
3382 such as DLLEXPORT may differ, one explicitly only tests for the
3383 calling conventions. */
3384 if (rvalue->expr_type == EXPR_VARIABLE
3385 && lvalue->symtree->n.sym->attr.ext_attr
3386 != rvalue->symtree->n.sym->attr.ext_attr)
3388 symbol_attribute calls;
3390 calls.ext_attr = 0;
3391 gfc_add_ext_attribute (&calls, EXT_ATTR_CDECL, NULL);
3392 gfc_add_ext_attribute (&calls, EXT_ATTR_STDCALL, NULL);
3393 gfc_add_ext_attribute (&calls, EXT_ATTR_FASTCALL, NULL);
3395 if ((calls.ext_attr & lvalue->symtree->n.sym->attr.ext_attr)
3396 != (calls.ext_attr & rvalue->symtree->n.sym->attr.ext_attr))
3398 gfc_error ("Mismatch in the procedure pointer assignment "
3399 "at %L: mismatch in the calling convention",
3400 &rvalue->where);
3401 return FAILURE;
3405 if (gfc_is_proc_ptr_comp (lvalue, &comp))
3406 s1 = comp->ts.interface;
3407 else
3408 s1 = lvalue->symtree->n.sym;
3410 if (gfc_is_proc_ptr_comp (rvalue, &comp))
3412 s2 = comp->ts.interface;
3413 name = comp->name;
3415 else if (rvalue->expr_type == EXPR_FUNCTION)
3417 s2 = rvalue->symtree->n.sym->result;
3418 name = rvalue->symtree->n.sym->result->name;
3420 else
3422 s2 = rvalue->symtree->n.sym;
3423 name = rvalue->symtree->n.sym->name;
3426 if (s1 && s2 && !gfc_compare_interfaces (s1, s2, name, 0, 1,
3427 err, sizeof(err)))
3429 gfc_error ("Interface mismatch in procedure pointer assignment "
3430 "at %L: %s", &rvalue->where, err);
3431 return FAILURE;
3434 return SUCCESS;
3437 if (!gfc_compare_types (&lvalue->ts, &rvalue->ts))
3439 gfc_error ("Different types in pointer assignment at %L; attempted "
3440 "assignment of %s to %s", &lvalue->where,
3441 gfc_typename (&rvalue->ts), gfc_typename (&lvalue->ts));
3442 return FAILURE;
3445 if (lvalue->ts.type != BT_CLASS && lvalue->ts.kind != rvalue->ts.kind)
3447 gfc_error ("Different kind type parameters in pointer "
3448 "assignment at %L", &lvalue->where);
3449 return FAILURE;
3452 if (lvalue->rank != rvalue->rank && !rank_remap)
3454 gfc_error ("Different ranks in pointer assignment at %L", &lvalue->where);
3455 return FAILURE;
3458 if (lvalue->ts.type == BT_CLASS && rvalue->ts.type == BT_DERIVED)
3459 /* Make sure the vtab is present. */
3460 gfc_find_derived_vtab (rvalue->ts.u.derived);
3462 /* Check rank remapping. */
3463 if (rank_remap)
3465 mpz_t lsize, rsize;
3467 /* If this can be determined, check that the target must be at least as
3468 large as the pointer assigned to it is. */
3469 if (gfc_array_size (lvalue, &lsize) == SUCCESS
3470 && gfc_array_size (rvalue, &rsize) == SUCCESS
3471 && mpz_cmp (rsize, lsize) < 0)
3473 gfc_error ("Rank remapping target is smaller than size of the"
3474 " pointer (%ld < %ld) at %L",
3475 mpz_get_si (rsize), mpz_get_si (lsize),
3476 &lvalue->where);
3477 return FAILURE;
3480 /* The target must be either rank one or it must be simply contiguous
3481 and F2008 must be allowed. */
3482 if (rvalue->rank != 1)
3484 if (!gfc_is_simply_contiguous (rvalue, true))
3486 gfc_error ("Rank remapping target must be rank 1 or"
3487 " simply contiguous at %L", &rvalue->where);
3488 return FAILURE;
3490 if (gfc_notify_std (GFC_STD_F2008, "Fortran 2008: Rank remapping"
3491 " target is not rank 1 at %L", &rvalue->where)
3492 == FAILURE)
3493 return FAILURE;
3497 /* Now punt if we are dealing with a NULLIFY(X) or X = NULL(X). */
3498 if (rvalue->expr_type == EXPR_NULL)
3499 return SUCCESS;
3501 if (lvalue->ts.type == BT_CHARACTER)
3503 gfc_try t = gfc_check_same_strlen (lvalue, rvalue, "pointer assignment");
3504 if (t == FAILURE)
3505 return FAILURE;
3508 if (rvalue->expr_type == EXPR_VARIABLE && is_subref_array (rvalue))
3509 lvalue->symtree->n.sym->attr.subref_array_pointer = 1;
3511 attr = gfc_expr_attr (rvalue);
3513 if (rvalue->expr_type == EXPR_FUNCTION && !attr.pointer)
3515 gfc_error ("Target expression in pointer assignment "
3516 "at %L must deliver a pointer result",
3517 &rvalue->where);
3518 return FAILURE;
3521 if (!attr.target && !attr.pointer)
3523 gfc_error ("Pointer assignment target is neither TARGET "
3524 "nor POINTER at %L", &rvalue->where);
3525 return FAILURE;
3528 if (is_pure && gfc_impure_variable (rvalue->symtree->n.sym))
3530 gfc_error ("Bad target in pointer assignment in PURE "
3531 "procedure at %L", &rvalue->where);
3534 if (is_implicit_pure && gfc_impure_variable (rvalue->symtree->n.sym))
3535 gfc_current_ns->proc_name->attr.implicit_pure = 0;
3538 if (gfc_has_vector_index (rvalue))
3540 gfc_error ("Pointer assignment with vector subscript "
3541 "on rhs at %L", &rvalue->where);
3542 return FAILURE;
3545 if (attr.is_protected && attr.use_assoc
3546 && !(attr.pointer || attr.proc_pointer))
3548 gfc_error ("Pointer assignment target has PROTECTED "
3549 "attribute at %L", &rvalue->where);
3550 return FAILURE;
3553 /* F2008, C725. For PURE also C1283. */
3554 if (rvalue->expr_type == EXPR_VARIABLE
3555 && gfc_is_coindexed (rvalue))
3557 gfc_ref *ref;
3558 for (ref = rvalue->ref; ref; ref = ref->next)
3559 if (ref->type == REF_ARRAY && ref->u.ar.codimen)
3561 gfc_error ("Data target at %L shall not have a coindex",
3562 &rvalue->where);
3563 return FAILURE;
3567 return SUCCESS;
3571 /* Relative of gfc_check_assign() except that the lvalue is a single
3572 symbol. Used for initialization assignments. */
3574 gfc_try
3575 gfc_check_assign_symbol (gfc_symbol *sym, gfc_expr *rvalue)
3577 gfc_expr lvalue;
3578 gfc_try r;
3580 memset (&lvalue, '\0', sizeof (gfc_expr));
3582 lvalue.expr_type = EXPR_VARIABLE;
3583 lvalue.ts = sym->ts;
3584 if (sym->as)
3585 lvalue.rank = sym->as->rank;
3586 lvalue.symtree = (gfc_symtree *) gfc_getmem (sizeof (gfc_symtree));
3587 lvalue.symtree->n.sym = sym;
3588 lvalue.where = sym->declared_at;
3590 if (sym->attr.pointer || sym->attr.proc_pointer
3591 || (sym->ts.type == BT_CLASS && CLASS_DATA (sym)->attr.class_pointer
3592 && rvalue->expr_type == EXPR_NULL))
3593 r = gfc_check_pointer_assign (&lvalue, rvalue);
3594 else
3595 r = gfc_check_assign (&lvalue, rvalue, 1);
3597 gfc_free (lvalue.symtree);
3599 if (r == FAILURE)
3600 return r;
3602 if (sym->attr.pointer && rvalue->expr_type != EXPR_NULL)
3604 /* F08:C461. Additional checks for pointer initialization. */
3605 symbol_attribute attr;
3606 attr = gfc_expr_attr (rvalue);
3607 if (attr.allocatable)
3609 gfc_error ("Pointer initialization target at %C "
3610 "must not be ALLOCATABLE ");
3611 return FAILURE;
3613 if (!attr.target || attr.pointer)
3615 gfc_error ("Pointer initialization target at %C "
3616 "must have the TARGET attribute");
3617 return FAILURE;
3619 if (!attr.save)
3621 gfc_error ("Pointer initialization target at %C "
3622 "must have the SAVE attribute");
3623 return FAILURE;
3627 if (sym->attr.proc_pointer && rvalue->expr_type != EXPR_NULL)
3629 /* F08:C1220. Additional checks for procedure pointer initialization. */
3630 symbol_attribute attr = gfc_expr_attr (rvalue);
3631 if (attr.proc_pointer)
3633 gfc_error ("Procedure pointer initialization target at %L "
3634 "may not be a procedure pointer", &rvalue->where);
3635 return FAILURE;
3639 return SUCCESS;
3643 /* Check for default initializer; sym->value is not enough
3644 as it is also set for EXPR_NULL of allocatables. */
3646 bool
3647 gfc_has_default_initializer (gfc_symbol *der)
3649 gfc_component *c;
3651 gcc_assert (der->attr.flavor == FL_DERIVED);
3652 for (c = der->components; c; c = c->next)
3653 if (c->ts.type == BT_DERIVED)
3655 if (!c->attr.pointer
3656 && gfc_has_default_initializer (c->ts.u.derived))
3657 return true;
3659 else
3661 if (c->initializer)
3662 return true;
3665 return false;
3668 /* Get an expression for a default initializer. */
3670 gfc_expr *
3671 gfc_default_initializer (gfc_typespec *ts)
3673 gfc_expr *init;
3674 gfc_component *comp;
3676 /* See if we have a default initializer in this, but not in nested
3677 types (otherwise we could use gfc_has_default_initializer()). */
3678 for (comp = ts->u.derived->components; comp; comp = comp->next)
3679 if (comp->initializer || comp->attr.allocatable
3680 || (comp->ts.type == BT_CLASS && CLASS_DATA (comp)->attr.allocatable))
3681 break;
3683 if (!comp)
3684 return NULL;
3686 init = gfc_get_structure_constructor_expr (ts->type, ts->kind,
3687 &ts->u.derived->declared_at);
3688 init->ts = *ts;
3690 for (comp = ts->u.derived->components; comp; comp = comp->next)
3692 gfc_constructor *ctor = gfc_constructor_get();
3694 if (comp->initializer)
3695 ctor->expr = gfc_copy_expr (comp->initializer);
3697 if (comp->attr.allocatable
3698 || (comp->ts.type == BT_CLASS && CLASS_DATA (comp)->attr.allocatable))
3700 ctor->expr = gfc_get_expr ();
3701 ctor->expr->expr_type = EXPR_NULL;
3702 ctor->expr->ts = comp->ts;
3705 gfc_constructor_append (&init->value.constructor, ctor);
3708 return init;
3712 /* Given a symbol, create an expression node with that symbol as a
3713 variable. If the symbol is array valued, setup a reference of the
3714 whole array. */
3716 gfc_expr *
3717 gfc_get_variable_expr (gfc_symtree *var)
3719 gfc_expr *e;
3721 e = gfc_get_expr ();
3722 e->expr_type = EXPR_VARIABLE;
3723 e->symtree = var;
3724 e->ts = var->n.sym->ts;
3726 if (var->n.sym->as != NULL)
3728 e->rank = var->n.sym->as->rank;
3729 e->ref = gfc_get_ref ();
3730 e->ref->type = REF_ARRAY;
3731 e->ref->u.ar.type = AR_FULL;
3734 return e;
3738 gfc_expr *
3739 gfc_lval_expr_from_sym (gfc_symbol *sym)
3741 gfc_expr *lval;
3742 lval = gfc_get_expr ();
3743 lval->expr_type = EXPR_VARIABLE;
3744 lval->where = sym->declared_at;
3745 lval->ts = sym->ts;
3746 lval->symtree = gfc_find_symtree (sym->ns->sym_root, sym->name);
3748 /* It will always be a full array. */
3749 lval->rank = sym->as ? sym->as->rank : 0;
3750 if (lval->rank)
3752 lval->ref = gfc_get_ref ();
3753 lval->ref->type = REF_ARRAY;
3754 lval->ref->u.ar.type = AR_FULL;
3755 lval->ref->u.ar.dimen = lval->rank;
3756 lval->ref->u.ar.where = sym->declared_at;
3757 lval->ref->u.ar.as = sym->as;
3760 return lval;
3764 /* Returns the array_spec of a full array expression. A NULL is
3765 returned otherwise. */
3766 gfc_array_spec *
3767 gfc_get_full_arrayspec_from_expr (gfc_expr *expr)
3769 gfc_array_spec *as;
3770 gfc_ref *ref;
3772 if (expr->rank == 0)
3773 return NULL;
3775 /* Follow any component references. */
3776 if (expr->expr_type == EXPR_VARIABLE
3777 || expr->expr_type == EXPR_CONSTANT)
3779 as = expr->symtree->n.sym->as;
3780 for (ref = expr->ref; ref; ref = ref->next)
3782 switch (ref->type)
3784 case REF_COMPONENT:
3785 as = ref->u.c.component->as;
3786 continue;
3788 case REF_SUBSTRING:
3789 continue;
3791 case REF_ARRAY:
3793 switch (ref->u.ar.type)
3795 case AR_ELEMENT:
3796 case AR_SECTION:
3797 case AR_UNKNOWN:
3798 as = NULL;
3799 continue;
3801 case AR_FULL:
3802 break;
3804 break;
3809 else
3810 as = NULL;
3812 return as;
3816 /* General expression traversal function. */
3818 bool
3819 gfc_traverse_expr (gfc_expr *expr, gfc_symbol *sym,
3820 bool (*func)(gfc_expr *, gfc_symbol *, int*),
3821 int f)
3823 gfc_array_ref ar;
3824 gfc_ref *ref;
3825 gfc_actual_arglist *args;
3826 gfc_constructor *c;
3827 int i;
3829 if (!expr)
3830 return false;
3832 if ((*func) (expr, sym, &f))
3833 return true;
3835 if (expr->ts.type == BT_CHARACTER
3836 && expr->ts.u.cl
3837 && expr->ts.u.cl->length
3838 && expr->ts.u.cl->length->expr_type != EXPR_CONSTANT
3839 && gfc_traverse_expr (expr->ts.u.cl->length, sym, func, f))
3840 return true;
3842 switch (expr->expr_type)
3844 case EXPR_PPC:
3845 case EXPR_COMPCALL:
3846 case EXPR_FUNCTION:
3847 for (args = expr->value.function.actual; args; args = args->next)
3849 if (gfc_traverse_expr (args->expr, sym, func, f))
3850 return true;
3852 break;
3854 case EXPR_VARIABLE:
3855 case EXPR_CONSTANT:
3856 case EXPR_NULL:
3857 case EXPR_SUBSTRING:
3858 break;
3860 case EXPR_STRUCTURE:
3861 case EXPR_ARRAY:
3862 for (c = gfc_constructor_first (expr->value.constructor);
3863 c; c = gfc_constructor_next (c))
3865 if (gfc_traverse_expr (c->expr, sym, func, f))
3866 return true;
3867 if (c->iterator)
3869 if (gfc_traverse_expr (c->iterator->var, sym, func, f))
3870 return true;
3871 if (gfc_traverse_expr (c->iterator->start, sym, func, f))
3872 return true;
3873 if (gfc_traverse_expr (c->iterator->end, sym, func, f))
3874 return true;
3875 if (gfc_traverse_expr (c->iterator->step, sym, func, f))
3876 return true;
3879 break;
3881 case EXPR_OP:
3882 if (gfc_traverse_expr (expr->value.op.op1, sym, func, f))
3883 return true;
3884 if (gfc_traverse_expr (expr->value.op.op2, sym, func, f))
3885 return true;
3886 break;
3888 default:
3889 gcc_unreachable ();
3890 break;
3893 ref = expr->ref;
3894 while (ref != NULL)
3896 switch (ref->type)
3898 case REF_ARRAY:
3899 ar = ref->u.ar;
3900 for (i = 0; i < GFC_MAX_DIMENSIONS; i++)
3902 if (gfc_traverse_expr (ar.start[i], sym, func, f))
3903 return true;
3904 if (gfc_traverse_expr (ar.end[i], sym, func, f))
3905 return true;
3906 if (gfc_traverse_expr (ar.stride[i], sym, func, f))
3907 return true;
3909 break;
3911 case REF_SUBSTRING:
3912 if (gfc_traverse_expr (ref->u.ss.start, sym, func, f))
3913 return true;
3914 if (gfc_traverse_expr (ref->u.ss.end, sym, func, f))
3915 return true;
3916 break;
3918 case REF_COMPONENT:
3919 if (ref->u.c.component->ts.type == BT_CHARACTER
3920 && ref->u.c.component->ts.u.cl
3921 && ref->u.c.component->ts.u.cl->length
3922 && ref->u.c.component->ts.u.cl->length->expr_type
3923 != EXPR_CONSTANT
3924 && gfc_traverse_expr (ref->u.c.component->ts.u.cl->length,
3925 sym, func, f))
3926 return true;
3928 if (ref->u.c.component->as)
3929 for (i = 0; i < ref->u.c.component->as->rank
3930 + ref->u.c.component->as->corank; i++)
3932 if (gfc_traverse_expr (ref->u.c.component->as->lower[i],
3933 sym, func, f))
3934 return true;
3935 if (gfc_traverse_expr (ref->u.c.component->as->upper[i],
3936 sym, func, f))
3937 return true;
3939 break;
3941 default:
3942 gcc_unreachable ();
3944 ref = ref->next;
3946 return false;
3949 /* Traverse expr, marking all EXPR_VARIABLE symbols referenced. */
3951 static bool
3952 expr_set_symbols_referenced (gfc_expr *expr,
3953 gfc_symbol *sym ATTRIBUTE_UNUSED,
3954 int *f ATTRIBUTE_UNUSED)
3956 if (expr->expr_type != EXPR_VARIABLE)
3957 return false;
3958 gfc_set_sym_referenced (expr->symtree->n.sym);
3959 return false;
3962 void
3963 gfc_expr_set_symbols_referenced (gfc_expr *expr)
3965 gfc_traverse_expr (expr, NULL, expr_set_symbols_referenced, 0);
3969 /* Determine if an expression is a procedure pointer component. If yes, the
3970 argument 'comp' will point to the component (provided that 'comp' was
3971 provided). */
3973 bool
3974 gfc_is_proc_ptr_comp (gfc_expr *expr, gfc_component **comp)
3976 gfc_ref *ref;
3977 bool ppc = false;
3979 if (!expr || !expr->ref)
3980 return false;
3982 ref = expr->ref;
3983 while (ref->next)
3984 ref = ref->next;
3986 if (ref->type == REF_COMPONENT)
3988 ppc = ref->u.c.component->attr.proc_pointer;
3989 if (ppc && comp)
3990 *comp = ref->u.c.component;
3993 return ppc;
3997 /* Walk an expression tree and check each variable encountered for being typed.
3998 If strict is not set, a top-level variable is tolerated untyped in -std=gnu
3999 mode as is a basic arithmetic expression using those; this is for things in
4000 legacy-code like:
4002 INTEGER :: arr(n), n
4003 INTEGER :: arr(n + 1), n
4005 The namespace is needed for IMPLICIT typing. */
4007 static gfc_namespace* check_typed_ns;
4009 static bool
4010 expr_check_typed_help (gfc_expr* e, gfc_symbol* sym ATTRIBUTE_UNUSED,
4011 int* f ATTRIBUTE_UNUSED)
4013 gfc_try t;
4015 if (e->expr_type != EXPR_VARIABLE)
4016 return false;
4018 gcc_assert (e->symtree);
4019 t = gfc_check_symbol_typed (e->symtree->n.sym, check_typed_ns,
4020 true, e->where);
4022 return (t == FAILURE);
4025 gfc_try
4026 gfc_expr_check_typed (gfc_expr* e, gfc_namespace* ns, bool strict)
4028 bool error_found;
4030 /* If this is a top-level variable or EXPR_OP, do the check with strict given
4031 to us. */
4032 if (!strict)
4034 if (e->expr_type == EXPR_VARIABLE && !e->ref)
4035 return gfc_check_symbol_typed (e->symtree->n.sym, ns, strict, e->where);
4037 if (e->expr_type == EXPR_OP)
4039 gfc_try t = SUCCESS;
4041 gcc_assert (e->value.op.op1);
4042 t = gfc_expr_check_typed (e->value.op.op1, ns, strict);
4044 if (t == SUCCESS && e->value.op.op2)
4045 t = gfc_expr_check_typed (e->value.op.op2, ns, strict);
4047 return t;
4051 /* Otherwise, walk the expression and do it strictly. */
4052 check_typed_ns = ns;
4053 error_found = gfc_traverse_expr (e, NULL, &expr_check_typed_help, 0);
4055 return error_found ? FAILURE : SUCCESS;
4058 /* Walk an expression tree and replace all symbols with a corresponding symbol
4059 in the formal_ns of "sym". Needed for copying interfaces in PROCEDURE
4060 statements. The boolean return value is required by gfc_traverse_expr. */
4062 static bool
4063 replace_symbol (gfc_expr *expr, gfc_symbol *sym, int *i ATTRIBUTE_UNUSED)
4065 if ((expr->expr_type == EXPR_VARIABLE
4066 || (expr->expr_type == EXPR_FUNCTION
4067 && !gfc_is_intrinsic (expr->symtree->n.sym, 0, expr->where)))
4068 && expr->symtree->n.sym->ns == sym->ts.interface->formal_ns)
4070 gfc_symtree *stree;
4071 gfc_namespace *ns = sym->formal_ns;
4072 /* Don't use gfc_get_symtree as we prefer to fail badly if we don't find
4073 the symtree rather than create a new one (and probably fail later). */
4074 stree = gfc_find_symtree (ns ? ns->sym_root : gfc_current_ns->sym_root,
4075 expr->symtree->n.sym->name);
4076 gcc_assert (stree);
4077 stree->n.sym->attr = expr->symtree->n.sym->attr;
4078 expr->symtree = stree;
4080 return false;
4083 void
4084 gfc_expr_replace_symbols (gfc_expr *expr, gfc_symbol *dest)
4086 gfc_traverse_expr (expr, dest, &replace_symbol, 0);
4089 /* The following is analogous to 'replace_symbol', and needed for copying
4090 interfaces for procedure pointer components. The argument 'sym' must formally
4091 be a gfc_symbol, so that the function can be passed to gfc_traverse_expr.
4092 However, it gets actually passed a gfc_component (i.e. the procedure pointer
4093 component in whose formal_ns the arguments have to be). */
4095 static bool
4096 replace_comp (gfc_expr *expr, gfc_symbol *sym, int *i ATTRIBUTE_UNUSED)
4098 gfc_component *comp;
4099 comp = (gfc_component *)sym;
4100 if ((expr->expr_type == EXPR_VARIABLE
4101 || (expr->expr_type == EXPR_FUNCTION
4102 && !gfc_is_intrinsic (expr->symtree->n.sym, 0, expr->where)))
4103 && expr->symtree->n.sym->ns == comp->ts.interface->formal_ns)
4105 gfc_symtree *stree;
4106 gfc_namespace *ns = comp->formal_ns;
4107 /* Don't use gfc_get_symtree as we prefer to fail badly if we don't find
4108 the symtree rather than create a new one (and probably fail later). */
4109 stree = gfc_find_symtree (ns ? ns->sym_root : gfc_current_ns->sym_root,
4110 expr->symtree->n.sym->name);
4111 gcc_assert (stree);
4112 stree->n.sym->attr = expr->symtree->n.sym->attr;
4113 expr->symtree = stree;
4115 return false;
4118 void
4119 gfc_expr_replace_comp (gfc_expr *expr, gfc_component *dest)
4121 gfc_traverse_expr (expr, (gfc_symbol *)dest, &replace_comp, 0);
4125 bool
4126 gfc_is_coindexed (gfc_expr *e)
4128 gfc_ref *ref;
4130 for (ref = e->ref; ref; ref = ref->next)
4131 if (ref->type == REF_ARRAY && ref->u.ar.codimen > 0)
4133 int n;
4134 for (n = ref->u.ar.dimen; n < ref->u.ar.dimen + ref->u.ar.codimen; n++)
4135 if (ref->u.ar.dimen_type[n] != DIMEN_THIS_IMAGE)
4136 return true;
4139 return false;
4143 bool
4144 gfc_get_corank (gfc_expr *e)
4146 int corank;
4147 gfc_ref *ref;
4148 corank = e->symtree->n.sym->as ? e->symtree->n.sym->as->corank : 0;
4149 for (ref = e->ref; ref; ref = ref->next)
4151 if (ref->type == REF_ARRAY)
4152 corank = ref->u.ar.as->corank;
4153 gcc_assert (ref->type != REF_SUBSTRING);
4155 return corank;
4159 /* Check whether the expression has an ultimate allocatable component.
4160 Being itself allocatable does not count. */
4161 bool
4162 gfc_has_ultimate_allocatable (gfc_expr *e)
4164 gfc_ref *ref, *last = NULL;
4166 if (e->expr_type != EXPR_VARIABLE)
4167 return false;
4169 for (ref = e->ref; ref; ref = ref->next)
4170 if (ref->type == REF_COMPONENT)
4171 last = ref;
4173 if (last && last->u.c.component->ts.type == BT_CLASS)
4174 return CLASS_DATA (last->u.c.component)->attr.alloc_comp;
4175 else if (last && last->u.c.component->ts.type == BT_DERIVED)
4176 return last->u.c.component->ts.u.derived->attr.alloc_comp;
4177 else if (last)
4178 return false;
4180 if (e->ts.type == BT_CLASS)
4181 return CLASS_DATA (e)->attr.alloc_comp;
4182 else if (e->ts.type == BT_DERIVED)
4183 return e->ts.u.derived->attr.alloc_comp;
4184 else
4185 return false;
4189 /* Check whether the expression has an pointer component.
4190 Being itself a pointer does not count. */
4191 bool
4192 gfc_has_ultimate_pointer (gfc_expr *e)
4194 gfc_ref *ref, *last = NULL;
4196 if (e->expr_type != EXPR_VARIABLE)
4197 return false;
4199 for (ref = e->ref; ref; ref = ref->next)
4200 if (ref->type == REF_COMPONENT)
4201 last = ref;
4203 if (last && last->u.c.component->ts.type == BT_CLASS)
4204 return CLASS_DATA (last->u.c.component)->attr.pointer_comp;
4205 else if (last && last->u.c.component->ts.type == BT_DERIVED)
4206 return last->u.c.component->ts.u.derived->attr.pointer_comp;
4207 else if (last)
4208 return false;
4210 if (e->ts.type == BT_CLASS)
4211 return CLASS_DATA (e)->attr.pointer_comp;
4212 else if (e->ts.type == BT_DERIVED)
4213 return e->ts.u.derived->attr.pointer_comp;
4214 else
4215 return false;
4219 /* Check whether an expression is "simply contiguous", cf. F2008, 6.5.4.
4220 Note: A scalar is not regarded as "simply contiguous" by the standard.
4221 if bool is not strict, some futher checks are done - for instance,
4222 a "(::1)" is accepted. */
4224 bool
4225 gfc_is_simply_contiguous (gfc_expr *expr, bool strict)
4227 bool colon;
4228 int i;
4229 gfc_array_ref *ar = NULL;
4230 gfc_ref *ref, *part_ref = NULL;
4232 if (expr->expr_type == EXPR_FUNCTION)
4233 return expr->value.function.esym
4234 ? expr->value.function.esym->result->attr.contiguous : false;
4235 else if (expr->expr_type != EXPR_VARIABLE)
4236 return false;
4238 if (expr->rank == 0)
4239 return false;
4241 for (ref = expr->ref; ref; ref = ref->next)
4243 if (ar)
4244 return false; /* Array shall be last part-ref. */
4246 if (ref->type == REF_COMPONENT)
4247 part_ref = ref;
4248 else if (ref->type == REF_SUBSTRING)
4249 return false;
4250 else if (ref->u.ar.type != AR_ELEMENT)
4251 ar = &ref->u.ar;
4254 if ((part_ref && !part_ref->u.c.component->attr.contiguous
4255 && part_ref->u.c.component->attr.pointer)
4256 || (!part_ref && !expr->symtree->n.sym->attr.contiguous
4257 && (expr->symtree->n.sym->attr.pointer
4258 || expr->symtree->n.sym->as->type == AS_ASSUMED_SHAPE)))
4259 return false;
4261 if (!ar || ar->type == AR_FULL)
4262 return true;
4264 gcc_assert (ar->type == AR_SECTION);
4266 /* Check for simply contiguous array */
4267 colon = true;
4268 for (i = 0; i < ar->dimen; i++)
4270 if (ar->dimen_type[i] == DIMEN_VECTOR)
4271 return false;
4273 if (ar->dimen_type[i] == DIMEN_ELEMENT)
4275 colon = false;
4276 continue;
4279 gcc_assert (ar->dimen_type[i] == DIMEN_RANGE);
4282 /* If the previous section was not contiguous, that's an error,
4283 unless we have effective only one element and checking is not
4284 strict. */
4285 if (!colon && (strict || !ar->start[i] || !ar->end[i]
4286 || ar->start[i]->expr_type != EXPR_CONSTANT
4287 || ar->end[i]->expr_type != EXPR_CONSTANT
4288 || mpz_cmp (ar->start[i]->value.integer,
4289 ar->end[i]->value.integer) != 0))
4290 return false;
4292 /* Following the standard, "(::1)" or - if known at compile time -
4293 "(lbound:ubound)" are not simply contigous; if strict
4294 is false, they are regarded as simply contiguous. */
4295 if (ar->stride[i] && (strict || ar->stride[i]->expr_type != EXPR_CONSTANT
4296 || ar->stride[i]->ts.type != BT_INTEGER
4297 || mpz_cmp_si (ar->stride[i]->value.integer, 1) != 0))
4298 return false;
4300 if (ar->start[i]
4301 && (strict || ar->start[i]->expr_type != EXPR_CONSTANT
4302 || !ar->as->lower[i]
4303 || ar->as->lower[i]->expr_type != EXPR_CONSTANT
4304 || mpz_cmp (ar->start[i]->value.integer,
4305 ar->as->lower[i]->value.integer) != 0))
4306 colon = false;
4308 if (ar->end[i]
4309 && (strict || ar->end[i]->expr_type != EXPR_CONSTANT
4310 || !ar->as->upper[i]
4311 || ar->as->upper[i]->expr_type != EXPR_CONSTANT
4312 || mpz_cmp (ar->end[i]->value.integer,
4313 ar->as->upper[i]->value.integer) != 0))
4314 colon = false;
4317 return true;
4321 /* Build call to an intrinsic procedure. The number of arguments has to be
4322 passed (rather than ending the list with a NULL value) because we may
4323 want to add arguments but with a NULL-expression. */
4325 gfc_expr*
4326 gfc_build_intrinsic_call (const char* name, locus where, unsigned numarg, ...)
4328 gfc_expr* result;
4329 gfc_actual_arglist* atail;
4330 gfc_intrinsic_sym* isym;
4331 va_list ap;
4332 unsigned i;
4334 isym = gfc_find_function (name);
4335 gcc_assert (isym);
4337 result = gfc_get_expr ();
4338 result->expr_type = EXPR_FUNCTION;
4339 result->ts = isym->ts;
4340 result->where = where;
4341 result->value.function.name = name;
4342 result->value.function.isym = isym;
4344 va_start (ap, numarg);
4345 atail = NULL;
4346 for (i = 0; i < numarg; ++i)
4348 if (atail)
4350 atail->next = gfc_get_actual_arglist ();
4351 atail = atail->next;
4353 else
4354 atail = result->value.function.actual = gfc_get_actual_arglist ();
4356 atail->expr = va_arg (ap, gfc_expr*);
4358 va_end (ap);
4360 return result;
4364 /* Check if an expression may appear in a variable definition context
4365 (F2008, 16.6.7) or pointer association context (F2008, 16.6.8).
4366 This is called from the various places when resolving
4367 the pieces that make up such a context.
4369 Optionally, a possible error message can be suppressed if context is NULL
4370 and just the return status (SUCCESS / FAILURE) be requested. */
4372 gfc_try
4373 gfc_check_vardef_context (gfc_expr* e, bool pointer, const char* context)
4375 gfc_symbol* sym;
4376 bool is_pointer;
4377 bool check_intentin;
4378 bool ptr_component;
4379 symbol_attribute attr;
4380 gfc_ref* ref;
4382 if (!pointer && e->expr_type == EXPR_FUNCTION
4383 && e->symtree->n.sym->result->attr.pointer)
4385 if (!(gfc_option.allow_std & GFC_STD_F2008))
4387 if (context)
4388 gfc_error ("Fortran 2008: Pointer functions in variable definition"
4389 " context (%s) at %L", context, &e->where);
4390 return FAILURE;
4393 else if (e->expr_type != EXPR_VARIABLE)
4395 if (context)
4396 gfc_error ("Non-variable expression in variable definition context (%s)"
4397 " at %L", context, &e->where);
4398 return FAILURE;
4401 gcc_assert (e->symtree);
4402 sym = e->symtree->n.sym;
4404 if (!pointer && sym->attr.flavor == FL_PARAMETER)
4406 if (context)
4407 gfc_error ("Named constant '%s' in variable definition context (%s)"
4408 " at %L", sym->name, context, &e->where);
4409 return FAILURE;
4411 if (!pointer && sym->attr.flavor != FL_VARIABLE
4412 && !(sym->attr.flavor == FL_PROCEDURE && sym == sym->result)
4413 && !(sym->attr.flavor == FL_PROCEDURE && sym->attr.proc_pointer))
4415 if (context)
4416 gfc_error ("'%s' in variable definition context (%s) at %L is not"
4417 " a variable", sym->name, context, &e->where);
4418 return FAILURE;
4421 /* Find out whether the expr is a pointer; this also means following
4422 component references to the last one. */
4423 attr = gfc_expr_attr (e);
4424 is_pointer = (attr.pointer || attr.proc_pointer);
4425 if (pointer && !is_pointer)
4427 if (context)
4428 gfc_error ("Non-POINTER in pointer association context (%s)"
4429 " at %L", context, &e->where);
4430 return FAILURE;
4433 /* INTENT(IN) dummy argument. Check this, unless the object itself is
4434 the component of sub-component of a pointer. Obviously,
4435 procedure pointers are of no interest here. */
4436 check_intentin = true;
4437 ptr_component = sym->attr.pointer;
4438 for (ref = e->ref; ref && check_intentin; ref = ref->next)
4440 if (ptr_component && ref->type == REF_COMPONENT)
4441 check_intentin = false;
4442 if (ref->type == REF_COMPONENT && ref->u.c.component->attr.pointer)
4443 ptr_component = true;
4445 if (check_intentin && sym->attr.intent == INTENT_IN)
4447 if (pointer && is_pointer)
4449 if (context)
4450 gfc_error ("Dummy argument '%s' with INTENT(IN) in pointer"
4451 " association context (%s) at %L",
4452 sym->name, context, &e->where);
4453 return FAILURE;
4455 if (!pointer && !is_pointer)
4457 if (context)
4458 gfc_error ("Dummy argument '%s' with INTENT(IN) in variable"
4459 " definition context (%s) at %L",
4460 sym->name, context, &e->where);
4461 return FAILURE;
4465 /* PROTECTED and use-associated. */
4466 if (sym->attr.is_protected && sym->attr.use_assoc && check_intentin)
4468 if (pointer && is_pointer)
4470 if (context)
4471 gfc_error ("Variable '%s' is PROTECTED and can not appear in a"
4472 " pointer association context (%s) at %L",
4473 sym->name, context, &e->where);
4474 return FAILURE;
4476 if (!pointer && !is_pointer)
4478 if (context)
4479 gfc_error ("Variable '%s' is PROTECTED and can not appear in a"
4480 " variable definition context (%s) at %L",
4481 sym->name, context, &e->where);
4482 return FAILURE;
4486 /* Variable not assignable from a PURE procedure but appears in
4487 variable definition context. */
4488 if (!pointer && gfc_pure (NULL) && gfc_impure_variable (sym))
4490 if (context)
4491 gfc_error ("Variable '%s' can not appear in a variable definition"
4492 " context (%s) at %L in PURE procedure",
4493 sym->name, context, &e->where);
4494 return FAILURE;
4497 if (!pointer && gfc_implicit_pure (NULL) && gfc_impure_variable (sym))
4498 gfc_current_ns->proc_name->attr.implicit_pure = 0;
4500 /* Check variable definition context for associate-names. */
4501 if (!pointer && sym->assoc)
4503 const char* name;
4504 gfc_association_list* assoc;
4506 gcc_assert (sym->assoc->target);
4508 /* If this is a SELECT TYPE temporary (the association is used internally
4509 for SELECT TYPE), silently go over to the target. */
4510 if (sym->attr.select_type_temporary)
4512 gfc_expr* t = sym->assoc->target;
4514 gcc_assert (t->expr_type == EXPR_VARIABLE);
4515 name = t->symtree->name;
4517 if (t->symtree->n.sym->assoc)
4518 assoc = t->symtree->n.sym->assoc;
4519 else
4520 assoc = sym->assoc;
4522 else
4524 name = sym->name;
4525 assoc = sym->assoc;
4527 gcc_assert (name && assoc);
4529 /* Is association to a valid variable? */
4530 if (!assoc->variable)
4532 if (context)
4534 if (assoc->target->expr_type == EXPR_VARIABLE)
4535 gfc_error ("'%s' at %L associated to vector-indexed target can"
4536 " not be used in a variable definition context (%s)",
4537 name, &e->where, context);
4538 else
4539 gfc_error ("'%s' at %L associated to expression can"
4540 " not be used in a variable definition context (%s)",
4541 name, &e->where, context);
4543 return FAILURE;
4546 /* Target must be allowed to appear in a variable definition context. */
4547 if (gfc_check_vardef_context (assoc->target, pointer, NULL) == FAILURE)
4549 if (context)
4550 gfc_error ("Associate-name '%s' can not appear in a variable"
4551 " definition context (%s) at %L because its target"
4552 " at %L can not, either",
4553 name, context, &e->where,
4554 &assoc->target->where);
4555 return FAILURE;
4559 return SUCCESS;