PR rtl-optimization/43520
[official-gcc.git] / gcc / fortran / trans-array.c
blobb03cc9400c92649410a1a2baed89325791680d61
1 /* Array translation routines
2 Copyright (C) 2002, 2003, 2004, 2005, 2006, 2007, 2008, 2009, 2010
3 Free Software Foundation, Inc.
4 Contributed by Paul Brook <paul@nowt.org>
5 and Steven Bosscher <s.bosscher@student.tudelft.nl>
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 /* trans-array.c-- Various array related code, including scalarization,
24 allocation, initialization and other support routines. */
26 /* How the scalarizer works.
27 In gfortran, array expressions use the same core routines as scalar
28 expressions.
29 First, a Scalarization State (SS) chain is built. This is done by walking
30 the expression tree, and building a linear list of the terms in the
31 expression. As the tree is walked, scalar subexpressions are translated.
33 The scalarization parameters are stored in a gfc_loopinfo structure.
34 First the start and stride of each term is calculated by
35 gfc_conv_ss_startstride. During this process the expressions for the array
36 descriptors and data pointers are also translated.
38 If the expression is an assignment, we must then resolve any dependencies.
39 In fortran all the rhs values of an assignment must be evaluated before
40 any assignments take place. This can require a temporary array to store the
41 values. We also require a temporary when we are passing array expressions
42 or vector subscripts as procedure parameters.
44 Array sections are passed without copying to a temporary. These use the
45 scalarizer to determine the shape of the section. The flag
46 loop->array_parameter tells the scalarizer that the actual values and loop
47 variables will not be required.
49 The function gfc_conv_loop_setup generates the scalarization setup code.
50 It determines the range of the scalarizing loop variables. If a temporary
51 is required, this is created and initialized. Code for scalar expressions
52 taken outside the loop is also generated at this time. Next the offset and
53 scaling required to translate from loop variables to array indices for each
54 term is calculated.
56 A call to gfc_start_scalarized_body marks the start of the scalarized
57 expression. This creates a scope and declares the loop variables. Before
58 calling this gfc_make_ss_chain_used must be used to indicate which terms
59 will be used inside this loop.
61 The scalar gfc_conv_* functions are then used to build the main body of the
62 scalarization loop. Scalarization loop variables and precalculated scalar
63 values are automatically substituted. Note that gfc_advance_se_ss_chain
64 must be used, rather than changing the se->ss directly.
66 For assignment expressions requiring a temporary two sub loops are
67 generated. The first stores the result of the expression in the temporary,
68 the second copies it to the result. A call to
69 gfc_trans_scalarized_loop_boundary marks the end of the main loop code and
70 the start of the copying loop. The temporary may be less than full rank.
72 Finally gfc_trans_scalarizing_loops is called to generate the implicit do
73 loops. The loops are added to the pre chain of the loopinfo. The post
74 chain may still contain cleanup code.
76 After the loop code has been added into its parent scope gfc_cleanup_loop
77 is called to free all the SS allocated by the scalarizer. */
79 #include "config.h"
80 #include "system.h"
81 #include "coretypes.h"
82 #include "tree.h"
83 #include "gimple.h"
84 #include "ggc.h"
85 #include "toplev.h"
86 #include "real.h"
87 #include "flags.h"
88 #include "gfortran.h"
89 #include "constructor.h"
90 #include "trans.h"
91 #include "trans-stmt.h"
92 #include "trans-types.h"
93 #include "trans-array.h"
94 #include "trans-const.h"
95 #include "dependency.h"
97 static gfc_ss *gfc_walk_subexpr (gfc_ss *, gfc_expr *);
98 static bool gfc_get_array_constructor_size (mpz_t *, gfc_constructor_base);
100 /* The contents of this structure aren't actually used, just the address. */
101 static gfc_ss gfc_ss_terminator_var;
102 gfc_ss * const gfc_ss_terminator = &gfc_ss_terminator_var;
105 static tree
106 gfc_array_dataptr_type (tree desc)
108 return (GFC_TYPE_ARRAY_DATAPTR_TYPE (TREE_TYPE (desc)));
112 /* Build expressions to access the members of an array descriptor.
113 It's surprisingly easy to mess up here, so never access
114 an array descriptor by "brute force", always use these
115 functions. This also avoids problems if we change the format
116 of an array descriptor.
118 To understand these magic numbers, look at the comments
119 before gfc_build_array_type() in trans-types.c.
121 The code within these defines should be the only code which knows the format
122 of an array descriptor.
124 Any code just needing to read obtain the bounds of an array should use
125 gfc_conv_array_* rather than the following functions as these will return
126 know constant values, and work with arrays which do not have descriptors.
128 Don't forget to #undef these! */
130 #define DATA_FIELD 0
131 #define OFFSET_FIELD 1
132 #define DTYPE_FIELD 2
133 #define DIMENSION_FIELD 3
135 #define STRIDE_SUBFIELD 0
136 #define LBOUND_SUBFIELD 1
137 #define UBOUND_SUBFIELD 2
139 /* This provides READ-ONLY access to the data field. The field itself
140 doesn't have the proper type. */
142 tree
143 gfc_conv_descriptor_data_get (tree desc)
145 tree field, type, t;
147 type = TREE_TYPE (desc);
148 gcc_assert (GFC_DESCRIPTOR_TYPE_P (type));
150 field = TYPE_FIELDS (type);
151 gcc_assert (DATA_FIELD == 0);
153 t = fold_build3 (COMPONENT_REF, TREE_TYPE (field), desc, field, NULL_TREE);
154 t = fold_convert (GFC_TYPE_ARRAY_DATAPTR_TYPE (type), t);
156 return t;
159 /* This provides WRITE access to the data field.
161 TUPLES_P is true if we are generating tuples.
163 This function gets called through the following macros:
164 gfc_conv_descriptor_data_set
165 gfc_conv_descriptor_data_set. */
167 void
168 gfc_conv_descriptor_data_set (stmtblock_t *block, tree desc, tree value)
170 tree field, type, t;
172 type = TREE_TYPE (desc);
173 gcc_assert (GFC_DESCRIPTOR_TYPE_P (type));
175 field = TYPE_FIELDS (type);
176 gcc_assert (DATA_FIELD == 0);
178 t = fold_build3 (COMPONENT_REF, TREE_TYPE (field), desc, field, NULL_TREE);
179 gfc_add_modify (block, t, fold_convert (TREE_TYPE (field), value));
183 /* This provides address access to the data field. This should only be
184 used by array allocation, passing this on to the runtime. */
186 tree
187 gfc_conv_descriptor_data_addr (tree desc)
189 tree field, type, t;
191 type = TREE_TYPE (desc);
192 gcc_assert (GFC_DESCRIPTOR_TYPE_P (type));
194 field = TYPE_FIELDS (type);
195 gcc_assert (DATA_FIELD == 0);
197 t = fold_build3 (COMPONENT_REF, TREE_TYPE (field), desc, field, NULL_TREE);
198 return gfc_build_addr_expr (NULL_TREE, t);
201 static tree
202 gfc_conv_descriptor_offset (tree desc)
204 tree type;
205 tree field;
207 type = TREE_TYPE (desc);
208 gcc_assert (GFC_DESCRIPTOR_TYPE_P (type));
210 field = gfc_advance_chain (TYPE_FIELDS (type), OFFSET_FIELD);
211 gcc_assert (field != NULL_TREE && TREE_TYPE (field) == gfc_array_index_type);
213 return fold_build3 (COMPONENT_REF, TREE_TYPE (field),
214 desc, field, NULL_TREE);
217 tree
218 gfc_conv_descriptor_offset_get (tree desc)
220 return gfc_conv_descriptor_offset (desc);
223 void
224 gfc_conv_descriptor_offset_set (stmtblock_t *block, tree desc,
225 tree value)
227 tree t = gfc_conv_descriptor_offset (desc);
228 gfc_add_modify (block, t, fold_convert (TREE_TYPE (t), value));
232 tree
233 gfc_conv_descriptor_dtype (tree desc)
235 tree field;
236 tree type;
238 type = TREE_TYPE (desc);
239 gcc_assert (GFC_DESCRIPTOR_TYPE_P (type));
241 field = gfc_advance_chain (TYPE_FIELDS (type), DTYPE_FIELD);
242 gcc_assert (field != NULL_TREE && TREE_TYPE (field) == gfc_array_index_type);
244 return fold_build3 (COMPONENT_REF, TREE_TYPE (field),
245 desc, field, NULL_TREE);
248 static tree
249 gfc_conv_descriptor_dimension (tree desc, tree dim)
251 tree field;
252 tree type;
253 tree tmp;
255 type = TREE_TYPE (desc);
256 gcc_assert (GFC_DESCRIPTOR_TYPE_P (type));
258 field = gfc_advance_chain (TYPE_FIELDS (type), DIMENSION_FIELD);
259 gcc_assert (field != NULL_TREE
260 && TREE_CODE (TREE_TYPE (field)) == ARRAY_TYPE
261 && TREE_CODE (TREE_TYPE (TREE_TYPE (field))) == RECORD_TYPE);
263 tmp = fold_build3 (COMPONENT_REF, TREE_TYPE (field),
264 desc, field, NULL_TREE);
265 tmp = gfc_build_array_ref (tmp, dim, NULL);
266 return tmp;
269 static tree
270 gfc_conv_descriptor_stride (tree desc, tree dim)
272 tree tmp;
273 tree field;
275 tmp = gfc_conv_descriptor_dimension (desc, dim);
276 field = TYPE_FIELDS (TREE_TYPE (tmp));
277 field = gfc_advance_chain (field, STRIDE_SUBFIELD);
278 gcc_assert (field != NULL_TREE && TREE_TYPE (field) == gfc_array_index_type);
280 tmp = fold_build3 (COMPONENT_REF, TREE_TYPE (field),
281 tmp, field, NULL_TREE);
282 return tmp;
285 tree
286 gfc_conv_descriptor_stride_get (tree desc, tree dim)
288 tree type = TREE_TYPE (desc);
289 gcc_assert (GFC_DESCRIPTOR_TYPE_P (type));
290 if (integer_zerop (dim)
291 && GFC_TYPE_ARRAY_AKIND (type) == GFC_ARRAY_ALLOCATABLE)
292 return gfc_index_one_node;
294 return gfc_conv_descriptor_stride (desc, dim);
297 void
298 gfc_conv_descriptor_stride_set (stmtblock_t *block, tree desc,
299 tree dim, tree value)
301 tree t = gfc_conv_descriptor_stride (desc, dim);
302 gfc_add_modify (block, t, fold_convert (TREE_TYPE (t), value));
305 static tree
306 gfc_conv_descriptor_lbound (tree desc, tree dim)
308 tree tmp;
309 tree field;
311 tmp = gfc_conv_descriptor_dimension (desc, dim);
312 field = TYPE_FIELDS (TREE_TYPE (tmp));
313 field = gfc_advance_chain (field, LBOUND_SUBFIELD);
314 gcc_assert (field != NULL_TREE && TREE_TYPE (field) == gfc_array_index_type);
316 tmp = fold_build3 (COMPONENT_REF, TREE_TYPE (field),
317 tmp, field, NULL_TREE);
318 return tmp;
321 tree
322 gfc_conv_descriptor_lbound_get (tree desc, tree dim)
324 return gfc_conv_descriptor_lbound (desc, dim);
327 void
328 gfc_conv_descriptor_lbound_set (stmtblock_t *block, tree desc,
329 tree dim, tree value)
331 tree t = gfc_conv_descriptor_lbound (desc, dim);
332 gfc_add_modify (block, t, fold_convert (TREE_TYPE (t), value));
335 static tree
336 gfc_conv_descriptor_ubound (tree desc, tree dim)
338 tree tmp;
339 tree field;
341 tmp = gfc_conv_descriptor_dimension (desc, dim);
342 field = TYPE_FIELDS (TREE_TYPE (tmp));
343 field = gfc_advance_chain (field, UBOUND_SUBFIELD);
344 gcc_assert (field != NULL_TREE && TREE_TYPE (field) == gfc_array_index_type);
346 tmp = fold_build3 (COMPONENT_REF, TREE_TYPE (field),
347 tmp, field, NULL_TREE);
348 return tmp;
351 tree
352 gfc_conv_descriptor_ubound_get (tree desc, tree dim)
354 return gfc_conv_descriptor_ubound (desc, dim);
357 void
358 gfc_conv_descriptor_ubound_set (stmtblock_t *block, tree desc,
359 tree dim, tree value)
361 tree t = gfc_conv_descriptor_ubound (desc, dim);
362 gfc_add_modify (block, t, fold_convert (TREE_TYPE (t), value));
365 /* Build a null array descriptor constructor. */
367 tree
368 gfc_build_null_descriptor (tree type)
370 tree field;
371 tree tmp;
373 gcc_assert (GFC_DESCRIPTOR_TYPE_P (type));
374 gcc_assert (DATA_FIELD == 0);
375 field = TYPE_FIELDS (type);
377 /* Set a NULL data pointer. */
378 tmp = build_constructor_single (type, field, null_pointer_node);
379 TREE_CONSTANT (tmp) = 1;
380 /* All other fields are ignored. */
382 return tmp;
386 /* Cleanup those #defines. */
388 #undef DATA_FIELD
389 #undef OFFSET_FIELD
390 #undef DTYPE_FIELD
391 #undef DIMENSION_FIELD
392 #undef STRIDE_SUBFIELD
393 #undef LBOUND_SUBFIELD
394 #undef UBOUND_SUBFIELD
397 /* Mark a SS chain as used. Flags specifies in which loops the SS is used.
398 flags & 1 = Main loop body.
399 flags & 2 = temp copy loop. */
401 void
402 gfc_mark_ss_chain_used (gfc_ss * ss, unsigned flags)
404 for (; ss != gfc_ss_terminator; ss = ss->next)
405 ss->useflags = flags;
408 static void gfc_free_ss (gfc_ss *);
411 /* Free a gfc_ss chain. */
413 static void
414 gfc_free_ss_chain (gfc_ss * ss)
416 gfc_ss *next;
418 while (ss != gfc_ss_terminator)
420 gcc_assert (ss != NULL);
421 next = ss->next;
422 gfc_free_ss (ss);
423 ss = next;
428 /* Free a SS. */
430 static void
431 gfc_free_ss (gfc_ss * ss)
433 int n;
435 switch (ss->type)
437 case GFC_SS_SECTION:
438 for (n = 0; n < GFC_MAX_DIMENSIONS; n++)
440 if (ss->data.info.subscript[n])
441 gfc_free_ss_chain (ss->data.info.subscript[n]);
443 break;
445 default:
446 break;
449 gfc_free (ss);
453 /* Free all the SS associated with a loop. */
455 void
456 gfc_cleanup_loop (gfc_loopinfo * loop)
458 gfc_ss *ss;
459 gfc_ss *next;
461 ss = loop->ss;
462 while (ss != gfc_ss_terminator)
464 gcc_assert (ss != NULL);
465 next = ss->loop_chain;
466 gfc_free_ss (ss);
467 ss = next;
472 /* Associate a SS chain with a loop. */
474 void
475 gfc_add_ss_to_loop (gfc_loopinfo * loop, gfc_ss * head)
477 gfc_ss *ss;
479 if (head == gfc_ss_terminator)
480 return;
482 ss = head;
483 for (; ss && ss != gfc_ss_terminator; ss = ss->next)
485 if (ss->next == gfc_ss_terminator)
486 ss->loop_chain = loop->ss;
487 else
488 ss->loop_chain = ss->next;
490 gcc_assert (ss == gfc_ss_terminator);
491 loop->ss = head;
495 /* Generate an initializer for a static pointer or allocatable array. */
497 void
498 gfc_trans_static_array_pointer (gfc_symbol * sym)
500 tree type;
502 gcc_assert (TREE_STATIC (sym->backend_decl));
503 /* Just zero the data member. */
504 type = TREE_TYPE (sym->backend_decl);
505 DECL_INITIAL (sym->backend_decl) = gfc_build_null_descriptor (type);
509 /* If the bounds of SE's loop have not yet been set, see if they can be
510 determined from array spec AS, which is the array spec of a called
511 function. MAPPING maps the callee's dummy arguments to the values
512 that the caller is passing. Add any initialization and finalization
513 code to SE. */
515 void
516 gfc_set_loop_bounds_from_array_spec (gfc_interface_mapping * mapping,
517 gfc_se * se, gfc_array_spec * as)
519 int n, dim;
520 gfc_se tmpse;
521 tree lower;
522 tree upper;
523 tree tmp;
525 if (as && as->type == AS_EXPLICIT)
526 for (dim = 0; dim < se->loop->dimen; dim++)
528 n = se->loop->order[dim];
529 if (se->loop->to[n] == NULL_TREE)
531 /* Evaluate the lower bound. */
532 gfc_init_se (&tmpse, NULL);
533 gfc_apply_interface_mapping (mapping, &tmpse, as->lower[dim]);
534 gfc_add_block_to_block (&se->pre, &tmpse.pre);
535 gfc_add_block_to_block (&se->post, &tmpse.post);
536 lower = fold_convert (gfc_array_index_type, tmpse.expr);
538 /* ...and the upper bound. */
539 gfc_init_se (&tmpse, NULL);
540 gfc_apply_interface_mapping (mapping, &tmpse, as->upper[dim]);
541 gfc_add_block_to_block (&se->pre, &tmpse.pre);
542 gfc_add_block_to_block (&se->post, &tmpse.post);
543 upper = fold_convert (gfc_array_index_type, tmpse.expr);
545 /* Set the upper bound of the loop to UPPER - LOWER. */
546 tmp = fold_build2 (MINUS_EXPR, gfc_array_index_type, upper, lower);
547 tmp = gfc_evaluate_now (tmp, &se->pre);
548 se->loop->to[n] = tmp;
554 /* Generate code to allocate an array temporary, or create a variable to
555 hold the data. If size is NULL, zero the descriptor so that the
556 callee will allocate the array. If DEALLOC is true, also generate code to
557 free the array afterwards.
559 If INITIAL is not NULL, it is packed using internal_pack and the result used
560 as data instead of allocating a fresh, unitialized area of memory.
562 Initialization code is added to PRE and finalization code to POST.
563 DYNAMIC is true if the caller may want to extend the array later
564 using realloc. This prevents us from putting the array on the stack. */
566 static void
567 gfc_trans_allocate_array_storage (stmtblock_t * pre, stmtblock_t * post,
568 gfc_ss_info * info, tree size, tree nelem,
569 tree initial, bool dynamic, bool dealloc)
571 tree tmp;
572 tree desc;
573 bool onstack;
575 desc = info->descriptor;
576 info->offset = gfc_index_zero_node;
577 if (size == NULL_TREE || integer_zerop (size))
579 /* A callee allocated array. */
580 gfc_conv_descriptor_data_set (pre, desc, null_pointer_node);
581 onstack = FALSE;
583 else
585 /* Allocate the temporary. */
586 onstack = !dynamic && initial == NULL_TREE
587 && gfc_can_put_var_on_stack (size);
589 if (onstack)
591 /* Make a temporary variable to hold the data. */
592 tmp = fold_build2 (MINUS_EXPR, TREE_TYPE (nelem), nelem,
593 gfc_index_one_node);
594 tmp = build_range_type (gfc_array_index_type, gfc_index_zero_node,
595 tmp);
596 tmp = build_array_type (gfc_get_element_type (TREE_TYPE (desc)),
597 tmp);
598 tmp = gfc_create_var (tmp, "A");
599 tmp = gfc_build_addr_expr (NULL_TREE, tmp);
600 gfc_conv_descriptor_data_set (pre, desc, tmp);
602 else
604 /* Allocate memory to hold the data or call internal_pack. */
605 if (initial == NULL_TREE)
607 tmp = gfc_call_malloc (pre, NULL, size);
608 tmp = gfc_evaluate_now (tmp, pre);
610 else
612 tree packed;
613 tree source_data;
614 tree was_packed;
615 stmtblock_t do_copying;
617 tmp = TREE_TYPE (initial); /* Pointer to descriptor. */
618 gcc_assert (TREE_CODE (tmp) == POINTER_TYPE);
619 tmp = TREE_TYPE (tmp); /* The descriptor itself. */
620 tmp = gfc_get_element_type (tmp);
621 gcc_assert (tmp == gfc_get_element_type (TREE_TYPE (desc)));
622 packed = gfc_create_var (build_pointer_type (tmp), "data");
624 tmp = build_call_expr_loc (input_location,
625 gfor_fndecl_in_pack, 1, initial);
626 tmp = fold_convert (TREE_TYPE (packed), tmp);
627 gfc_add_modify (pre, packed, tmp);
629 tmp = build_fold_indirect_ref_loc (input_location,
630 initial);
631 source_data = gfc_conv_descriptor_data_get (tmp);
633 /* internal_pack may return source->data without any allocation
634 or copying if it is already packed. If that's the case, we
635 need to allocate and copy manually. */
637 gfc_start_block (&do_copying);
638 tmp = gfc_call_malloc (&do_copying, NULL, size);
639 tmp = fold_convert (TREE_TYPE (packed), tmp);
640 gfc_add_modify (&do_copying, packed, tmp);
641 tmp = gfc_build_memcpy_call (packed, source_data, size);
642 gfc_add_expr_to_block (&do_copying, tmp);
644 was_packed = fold_build2 (EQ_EXPR, boolean_type_node,
645 packed, source_data);
646 tmp = gfc_finish_block (&do_copying);
647 tmp = build3_v (COND_EXPR, was_packed, tmp,
648 build_empty_stmt (input_location));
649 gfc_add_expr_to_block (pre, tmp);
651 tmp = fold_convert (pvoid_type_node, packed);
654 gfc_conv_descriptor_data_set (pre, desc, tmp);
657 info->data = gfc_conv_descriptor_data_get (desc);
659 /* The offset is zero because we create temporaries with a zero
660 lower bound. */
661 gfc_conv_descriptor_offset_set (pre, desc, gfc_index_zero_node);
663 if (dealloc && !onstack)
665 /* Free the temporary. */
666 tmp = gfc_conv_descriptor_data_get (desc);
667 tmp = gfc_call_free (fold_convert (pvoid_type_node, tmp));
668 gfc_add_expr_to_block (post, tmp);
673 /* Generate code to create and initialize the descriptor for a temporary
674 array. This is used for both temporaries needed by the scalarizer, and
675 functions returning arrays. Adjusts the loop variables to be
676 zero-based, and calculates the loop bounds for callee allocated arrays.
677 Allocate the array unless it's callee allocated (we have a callee
678 allocated array if 'callee_alloc' is true, or if loop->to[n] is
679 NULL_TREE for any n). Also fills in the descriptor, data and offset
680 fields of info if known. Returns the size of the array, or NULL for a
681 callee allocated array.
683 PRE, POST, INITIAL, DYNAMIC and DEALLOC are as for
684 gfc_trans_allocate_array_storage.
687 tree
688 gfc_trans_create_temp_array (stmtblock_t * pre, stmtblock_t * post,
689 gfc_loopinfo * loop, gfc_ss_info * info,
690 tree eltype, tree initial, bool dynamic,
691 bool dealloc, bool callee_alloc, locus * where)
693 tree type;
694 tree desc;
695 tree tmp;
696 tree size;
697 tree nelem;
698 tree cond;
699 tree or_expr;
700 int n;
701 int dim;
703 gcc_assert (info->dimen > 0);
705 if (gfc_option.warn_array_temp && where)
706 gfc_warning ("Creating array temporary at %L", where);
708 /* Set the lower bound to zero. */
709 for (dim = 0; dim < info->dimen; dim++)
711 n = loop->order[dim];
712 /* Callee allocated arrays may not have a known bound yet. */
713 if (loop->to[n])
714 loop->to[n] = gfc_evaluate_now (fold_build2 (MINUS_EXPR,
715 gfc_array_index_type,
716 loop->to[n], loop->from[n]), pre);
717 loop->from[n] = gfc_index_zero_node;
719 info->delta[dim] = gfc_index_zero_node;
720 info->start[dim] = gfc_index_zero_node;
721 info->end[dim] = gfc_index_zero_node;
722 info->stride[dim] = gfc_index_one_node;
723 info->dim[dim] = dim;
726 /* Initialize the descriptor. */
727 type =
728 gfc_get_array_type_bounds (eltype, info->dimen, loop->from, loop->to, 1,
729 GFC_ARRAY_UNKNOWN, true);
730 desc = gfc_create_var (type, "atmp");
731 GFC_DECL_PACKED_ARRAY (desc) = 1;
733 info->descriptor = desc;
734 size = gfc_index_one_node;
736 /* Fill in the array dtype. */
737 tmp = gfc_conv_descriptor_dtype (desc);
738 gfc_add_modify (pre, tmp, gfc_get_dtype (TREE_TYPE (desc)));
741 Fill in the bounds and stride. This is a packed array, so:
743 size = 1;
744 for (n = 0; n < rank; n++)
746 stride[n] = size
747 delta = ubound[n] + 1 - lbound[n];
748 size = size * delta;
750 size = size * sizeof(element);
753 or_expr = NULL_TREE;
755 /* If there is at least one null loop->to[n], it is a callee allocated
756 array. */
757 for (n = 0; n < info->dimen; n++)
758 if (loop->to[n] == NULL_TREE)
760 size = NULL_TREE;
761 break;
764 for (n = 0; n < info->dimen; n++)
766 if (size == NULL_TREE)
768 /* For a callee allocated array express the loop bounds in terms
769 of the descriptor fields. */
770 tmp =
771 fold_build2 (MINUS_EXPR, gfc_array_index_type,
772 gfc_conv_descriptor_ubound_get (desc, gfc_rank_cst[n]),
773 gfc_conv_descriptor_lbound_get (desc, gfc_rank_cst[n]));
774 loop->to[n] = tmp;
775 continue;
778 /* Store the stride and bound components in the descriptor. */
779 gfc_conv_descriptor_stride_set (pre, desc, gfc_rank_cst[n], size);
781 gfc_conv_descriptor_lbound_set (pre, desc, gfc_rank_cst[n],
782 gfc_index_zero_node);
784 gfc_conv_descriptor_ubound_set (pre, desc, gfc_rank_cst[n], loop->to[n]);
786 tmp = fold_build2 (PLUS_EXPR, gfc_array_index_type,
787 loop->to[n], gfc_index_one_node);
789 /* Check whether the size for this dimension is negative. */
790 cond = fold_build2 (LE_EXPR, boolean_type_node, tmp,
791 gfc_index_zero_node);
792 cond = gfc_evaluate_now (cond, pre);
794 if (n == 0)
795 or_expr = cond;
796 else
797 or_expr = fold_build2 (TRUTH_OR_EXPR, boolean_type_node, or_expr, cond);
799 size = fold_build2 (MULT_EXPR, gfc_array_index_type, size, tmp);
800 size = gfc_evaluate_now (size, pre);
803 /* Get the size of the array. */
805 if (size && !callee_alloc)
807 /* If or_expr is true, then the extent in at least one
808 dimension is zero and the size is set to zero. */
809 size = fold_build3 (COND_EXPR, gfc_array_index_type,
810 or_expr, gfc_index_zero_node, size);
812 nelem = size;
813 size = fold_build2 (MULT_EXPR, gfc_array_index_type, size,
814 fold_convert (gfc_array_index_type,
815 TYPE_SIZE_UNIT (gfc_get_element_type (type))));
817 else
819 nelem = size;
820 size = NULL_TREE;
823 gfc_trans_allocate_array_storage (pre, post, info, size, nelem, initial,
824 dynamic, dealloc);
826 if (info->dimen > loop->temp_dim)
827 loop->temp_dim = info->dimen;
829 return size;
833 /* Generate code to transpose array EXPR by creating a new descriptor
834 in which the dimension specifications have been reversed. */
836 void
837 gfc_conv_array_transpose (gfc_se * se, gfc_expr * expr)
839 tree dest, src, dest_index, src_index;
840 gfc_loopinfo *loop;
841 gfc_ss_info *dest_info;
842 gfc_ss *dest_ss, *src_ss;
843 gfc_se src_se;
844 int n;
846 loop = se->loop;
848 src_ss = gfc_walk_expr (expr);
849 dest_ss = se->ss;
851 dest_info = &dest_ss->data.info;
852 gcc_assert (dest_info->dimen == 2);
854 /* Get a descriptor for EXPR. */
855 gfc_init_se (&src_se, NULL);
856 gfc_conv_expr_descriptor (&src_se, expr, src_ss);
857 gfc_add_block_to_block (&se->pre, &src_se.pre);
858 gfc_add_block_to_block (&se->post, &src_se.post);
859 src = src_se.expr;
861 /* Allocate a new descriptor for the return value. */
862 dest = gfc_create_var (TREE_TYPE (src), "atmp");
863 dest_info->descriptor = dest;
864 se->expr = dest;
866 /* Copy across the dtype field. */
867 gfc_add_modify (&se->pre,
868 gfc_conv_descriptor_dtype (dest),
869 gfc_conv_descriptor_dtype (src));
871 /* Copy the dimension information, renumbering dimension 1 to 0 and
872 0 to 1. */
873 for (n = 0; n < 2; n++)
875 dest_info->delta[n] = gfc_index_zero_node;
876 dest_info->start[n] = gfc_index_zero_node;
877 dest_info->end[n] = gfc_index_zero_node;
878 dest_info->stride[n] = gfc_index_one_node;
879 dest_info->dim[n] = n;
881 dest_index = gfc_rank_cst[n];
882 src_index = gfc_rank_cst[1 - n];
884 gfc_conv_descriptor_stride_set (&se->pre, dest, dest_index,
885 gfc_conv_descriptor_stride_get (src, src_index));
887 gfc_conv_descriptor_lbound_set (&se->pre, dest, dest_index,
888 gfc_conv_descriptor_lbound_get (src, src_index));
890 gfc_conv_descriptor_ubound_set (&se->pre, dest, dest_index,
891 gfc_conv_descriptor_ubound_get (src, src_index));
893 if (!loop->to[n])
895 gcc_assert (integer_zerop (loop->from[n]));
896 loop->to[n] =
897 fold_build2 (MINUS_EXPR, gfc_array_index_type,
898 gfc_conv_descriptor_ubound_get (dest, dest_index),
899 gfc_conv_descriptor_lbound_get (dest, dest_index));
903 /* Copy the data pointer. */
904 dest_info->data = gfc_conv_descriptor_data_get (src);
905 gfc_conv_descriptor_data_set (&se->pre, dest, dest_info->data);
907 /* Copy the offset. This is not changed by transposition; the top-left
908 element is still at the same offset as before, except where the loop
909 starts at zero. */
910 if (!integer_zerop (loop->from[0]))
911 dest_info->offset = gfc_conv_descriptor_offset_get (src);
912 else
913 dest_info->offset = gfc_index_zero_node;
915 gfc_conv_descriptor_offset_set (&se->pre, dest,
916 dest_info->offset);
918 if (dest_info->dimen > loop->temp_dim)
919 loop->temp_dim = dest_info->dimen;
923 /* Return the number of iterations in a loop that starts at START,
924 ends at END, and has step STEP. */
926 static tree
927 gfc_get_iteration_count (tree start, tree end, tree step)
929 tree tmp;
930 tree type;
932 type = TREE_TYPE (step);
933 tmp = fold_build2 (MINUS_EXPR, type, end, start);
934 tmp = fold_build2 (FLOOR_DIV_EXPR, type, tmp, step);
935 tmp = fold_build2 (PLUS_EXPR, type, tmp, build_int_cst (type, 1));
936 tmp = fold_build2 (MAX_EXPR, type, tmp, build_int_cst (type, 0));
937 return fold_convert (gfc_array_index_type, tmp);
941 /* Extend the data in array DESC by EXTRA elements. */
943 static void
944 gfc_grow_array (stmtblock_t * pblock, tree desc, tree extra)
946 tree arg0, arg1;
947 tree tmp;
948 tree size;
949 tree ubound;
951 if (integer_zerop (extra))
952 return;
954 ubound = gfc_conv_descriptor_ubound_get (desc, gfc_rank_cst[0]);
956 /* Add EXTRA to the upper bound. */
957 tmp = fold_build2 (PLUS_EXPR, gfc_array_index_type, ubound, extra);
958 gfc_conv_descriptor_ubound_set (pblock, desc, gfc_rank_cst[0], tmp);
960 /* Get the value of the current data pointer. */
961 arg0 = gfc_conv_descriptor_data_get (desc);
963 /* Calculate the new array size. */
964 size = TYPE_SIZE_UNIT (gfc_get_element_type (TREE_TYPE (desc)));
965 tmp = fold_build2 (PLUS_EXPR, gfc_array_index_type,
966 ubound, gfc_index_one_node);
967 arg1 = fold_build2 (MULT_EXPR, size_type_node,
968 fold_convert (size_type_node, tmp),
969 fold_convert (size_type_node, size));
971 /* Call the realloc() function. */
972 tmp = gfc_call_realloc (pblock, arg0, arg1);
973 gfc_conv_descriptor_data_set (pblock, desc, tmp);
977 /* Return true if the bounds of iterator I can only be determined
978 at run time. */
980 static inline bool
981 gfc_iterator_has_dynamic_bounds (gfc_iterator * i)
983 return (i->start->expr_type != EXPR_CONSTANT
984 || i->end->expr_type != EXPR_CONSTANT
985 || i->step->expr_type != EXPR_CONSTANT);
989 /* Split the size of constructor element EXPR into the sum of two terms,
990 one of which can be determined at compile time and one of which must
991 be calculated at run time. Set *SIZE to the former and return true
992 if the latter might be nonzero. */
994 static bool
995 gfc_get_array_constructor_element_size (mpz_t * size, gfc_expr * expr)
997 if (expr->expr_type == EXPR_ARRAY)
998 return gfc_get_array_constructor_size (size, expr->value.constructor);
999 else if (expr->rank > 0)
1001 /* Calculate everything at run time. */
1002 mpz_set_ui (*size, 0);
1003 return true;
1005 else
1007 /* A single element. */
1008 mpz_set_ui (*size, 1);
1009 return false;
1014 /* Like gfc_get_array_constructor_element_size, but applied to the whole
1015 of array constructor C. */
1017 static bool
1018 gfc_get_array_constructor_size (mpz_t * size, gfc_constructor_base base)
1020 gfc_constructor *c;
1021 gfc_iterator *i;
1022 mpz_t val;
1023 mpz_t len;
1024 bool dynamic;
1026 mpz_set_ui (*size, 0);
1027 mpz_init (len);
1028 mpz_init (val);
1030 dynamic = false;
1031 for (c = gfc_constructor_first (base); c; c = gfc_constructor_next (c))
1033 i = c->iterator;
1034 if (i && gfc_iterator_has_dynamic_bounds (i))
1035 dynamic = true;
1036 else
1038 dynamic |= gfc_get_array_constructor_element_size (&len, c->expr);
1039 if (i)
1041 /* Multiply the static part of the element size by the
1042 number of iterations. */
1043 mpz_sub (val, i->end->value.integer, i->start->value.integer);
1044 mpz_fdiv_q (val, val, i->step->value.integer);
1045 mpz_add_ui (val, val, 1);
1046 if (mpz_sgn (val) > 0)
1047 mpz_mul (len, len, val);
1048 else
1049 mpz_set_ui (len, 0);
1051 mpz_add (*size, *size, len);
1054 mpz_clear (len);
1055 mpz_clear (val);
1056 return dynamic;
1060 /* Make sure offset is a variable. */
1062 static void
1063 gfc_put_offset_into_var (stmtblock_t * pblock, tree * poffset,
1064 tree * offsetvar)
1066 /* We should have already created the offset variable. We cannot
1067 create it here because we may be in an inner scope. */
1068 gcc_assert (*offsetvar != NULL_TREE);
1069 gfc_add_modify (pblock, *offsetvar, *poffset);
1070 *poffset = *offsetvar;
1071 TREE_USED (*offsetvar) = 1;
1075 /* Variables needed for bounds-checking. */
1076 static bool first_len;
1077 static tree first_len_val;
1078 static bool typespec_chararray_ctor;
1080 static void
1081 gfc_trans_array_ctor_element (stmtblock_t * pblock, tree desc,
1082 tree offset, gfc_se * se, gfc_expr * expr)
1084 tree tmp;
1086 gfc_conv_expr (se, expr);
1088 /* Store the value. */
1089 tmp = build_fold_indirect_ref_loc (input_location,
1090 gfc_conv_descriptor_data_get (desc));
1091 tmp = gfc_build_array_ref (tmp, offset, NULL);
1093 if (expr->ts.type == BT_CHARACTER)
1095 int i = gfc_validate_kind (BT_CHARACTER, expr->ts.kind, false);
1096 tree esize;
1098 esize = size_in_bytes (gfc_get_element_type (TREE_TYPE (desc)));
1099 esize = fold_convert (gfc_charlen_type_node, esize);
1100 esize = fold_build2 (TRUNC_DIV_EXPR, gfc_charlen_type_node, esize,
1101 build_int_cst (gfc_charlen_type_node,
1102 gfc_character_kinds[i].bit_size / 8));
1104 gfc_conv_string_parameter (se);
1105 if (POINTER_TYPE_P (TREE_TYPE (tmp)))
1107 /* The temporary is an array of pointers. */
1108 se->expr = fold_convert (TREE_TYPE (tmp), se->expr);
1109 gfc_add_modify (&se->pre, tmp, se->expr);
1111 else
1113 /* The temporary is an array of string values. */
1114 tmp = gfc_build_addr_expr (gfc_get_pchar_type (expr->ts.kind), tmp);
1115 /* We know the temporary and the value will be the same length,
1116 so can use memcpy. */
1117 gfc_trans_string_copy (&se->pre, esize, tmp, expr->ts.kind,
1118 se->string_length, se->expr, expr->ts.kind);
1120 if ((gfc_option.rtcheck & GFC_RTCHECK_BOUNDS) && !typespec_chararray_ctor)
1122 if (first_len)
1124 gfc_add_modify (&se->pre, first_len_val,
1125 se->string_length);
1126 first_len = false;
1128 else
1130 /* Verify that all constructor elements are of the same
1131 length. */
1132 tree cond = fold_build2 (NE_EXPR, boolean_type_node,
1133 first_len_val, se->string_length);
1134 gfc_trans_runtime_check
1135 (true, false, cond, &se->pre, &expr->where,
1136 "Different CHARACTER lengths (%ld/%ld) in array constructor",
1137 fold_convert (long_integer_type_node, first_len_val),
1138 fold_convert (long_integer_type_node, se->string_length));
1142 else
1144 /* TODO: Should the frontend already have done this conversion? */
1145 se->expr = fold_convert (TREE_TYPE (tmp), se->expr);
1146 gfc_add_modify (&se->pre, tmp, se->expr);
1149 gfc_add_block_to_block (pblock, &se->pre);
1150 gfc_add_block_to_block (pblock, &se->post);
1154 /* Add the contents of an array to the constructor. DYNAMIC is as for
1155 gfc_trans_array_constructor_value. */
1157 static void
1158 gfc_trans_array_constructor_subarray (stmtblock_t * pblock,
1159 tree type ATTRIBUTE_UNUSED,
1160 tree desc, gfc_expr * expr,
1161 tree * poffset, tree * offsetvar,
1162 bool dynamic)
1164 gfc_se se;
1165 gfc_ss *ss;
1166 gfc_loopinfo loop;
1167 stmtblock_t body;
1168 tree tmp;
1169 tree size;
1170 int n;
1172 /* We need this to be a variable so we can increment it. */
1173 gfc_put_offset_into_var (pblock, poffset, offsetvar);
1175 gfc_init_se (&se, NULL);
1177 /* Walk the array expression. */
1178 ss = gfc_walk_expr (expr);
1179 gcc_assert (ss != gfc_ss_terminator);
1181 /* Initialize the scalarizer. */
1182 gfc_init_loopinfo (&loop);
1183 gfc_add_ss_to_loop (&loop, ss);
1185 /* Initialize the loop. */
1186 gfc_conv_ss_startstride (&loop);
1187 gfc_conv_loop_setup (&loop, &expr->where);
1189 /* Make sure the constructed array has room for the new data. */
1190 if (dynamic)
1192 /* Set SIZE to the total number of elements in the subarray. */
1193 size = gfc_index_one_node;
1194 for (n = 0; n < loop.dimen; n++)
1196 tmp = gfc_get_iteration_count (loop.from[n], loop.to[n],
1197 gfc_index_one_node);
1198 size = fold_build2 (MULT_EXPR, gfc_array_index_type, size, tmp);
1201 /* Grow the constructed array by SIZE elements. */
1202 gfc_grow_array (&loop.pre, desc, size);
1205 /* Make the loop body. */
1206 gfc_mark_ss_chain_used (ss, 1);
1207 gfc_start_scalarized_body (&loop, &body);
1208 gfc_copy_loopinfo_to_se (&se, &loop);
1209 se.ss = ss;
1211 gfc_trans_array_ctor_element (&body, desc, *poffset, &se, expr);
1212 gcc_assert (se.ss == gfc_ss_terminator);
1214 /* Increment the offset. */
1215 tmp = fold_build2 (PLUS_EXPR, gfc_array_index_type,
1216 *poffset, gfc_index_one_node);
1217 gfc_add_modify (&body, *poffset, tmp);
1219 /* Finish the loop. */
1220 gfc_trans_scalarizing_loops (&loop, &body);
1221 gfc_add_block_to_block (&loop.pre, &loop.post);
1222 tmp = gfc_finish_block (&loop.pre);
1223 gfc_add_expr_to_block (pblock, tmp);
1225 gfc_cleanup_loop (&loop);
1229 /* Assign the values to the elements of an array constructor. DYNAMIC
1230 is true if descriptor DESC only contains enough data for the static
1231 size calculated by gfc_get_array_constructor_size. When true, memory
1232 for the dynamic parts must be allocated using realloc. */
1234 static void
1235 gfc_trans_array_constructor_value (stmtblock_t * pblock, tree type,
1236 tree desc, gfc_constructor_base base,
1237 tree * poffset, tree * offsetvar,
1238 bool dynamic)
1240 tree tmp;
1241 stmtblock_t body;
1242 gfc_se se;
1243 mpz_t size;
1244 gfc_constructor *c;
1246 tree shadow_loopvar = NULL_TREE;
1247 gfc_saved_var saved_loopvar;
1249 mpz_init (size);
1250 for (c = gfc_constructor_first (base); c; c = gfc_constructor_next (c))
1252 /* If this is an iterator or an array, the offset must be a variable. */
1253 if ((c->iterator || c->expr->rank > 0) && INTEGER_CST_P (*poffset))
1254 gfc_put_offset_into_var (pblock, poffset, offsetvar);
1256 /* Shadowing the iterator avoids changing its value and saves us from
1257 keeping track of it. Further, it makes sure that there's always a
1258 backend-decl for the symbol, even if there wasn't one before,
1259 e.g. in the case of an iterator that appears in a specification
1260 expression in an interface mapping. */
1261 if (c->iterator)
1263 gfc_symbol *sym = c->iterator->var->symtree->n.sym;
1264 tree type = gfc_typenode_for_spec (&sym->ts);
1266 shadow_loopvar = gfc_create_var (type, "shadow_loopvar");
1267 gfc_shadow_sym (sym, shadow_loopvar, &saved_loopvar);
1270 gfc_start_block (&body);
1272 if (c->expr->expr_type == EXPR_ARRAY)
1274 /* Array constructors can be nested. */
1275 gfc_trans_array_constructor_value (&body, type, desc,
1276 c->expr->value.constructor,
1277 poffset, offsetvar, dynamic);
1279 else if (c->expr->rank > 0)
1281 gfc_trans_array_constructor_subarray (&body, type, desc, c->expr,
1282 poffset, offsetvar, dynamic);
1284 else
1286 /* This code really upsets the gimplifier so don't bother for now. */
1287 gfc_constructor *p;
1288 HOST_WIDE_INT n;
1289 HOST_WIDE_INT size;
1291 p = c;
1292 n = 0;
1293 while (p && !(p->iterator || p->expr->expr_type != EXPR_CONSTANT))
1295 p = gfc_constructor_next (p);
1296 n++;
1298 if (n < 4)
1300 /* Scalar values. */
1301 gfc_init_se (&se, NULL);
1302 gfc_trans_array_ctor_element (&body, desc, *poffset,
1303 &se, c->expr);
1305 *poffset = fold_build2 (PLUS_EXPR, gfc_array_index_type,
1306 *poffset, gfc_index_one_node);
1308 else
1310 /* Collect multiple scalar constants into a constructor. */
1311 tree list;
1312 tree init;
1313 tree bound;
1314 tree tmptype;
1315 HOST_WIDE_INT idx = 0;
1317 p = c;
1318 list = NULL_TREE;
1319 /* Count the number of consecutive scalar constants. */
1320 while (p && !(p->iterator
1321 || p->expr->expr_type != EXPR_CONSTANT))
1323 gfc_init_se (&se, NULL);
1324 gfc_conv_constant (&se, p->expr);
1326 if (c->expr->ts.type != BT_CHARACTER)
1327 se.expr = fold_convert (type, se.expr);
1328 /* For constant character array constructors we build
1329 an array of pointers. */
1330 else if (POINTER_TYPE_P (type))
1331 se.expr = gfc_build_addr_expr
1332 (gfc_get_pchar_type (p->expr->ts.kind),
1333 se.expr);
1335 list = tree_cons (build_int_cst (gfc_array_index_type,
1336 idx++), se.expr, list);
1337 c = p;
1338 p = gfc_constructor_next (p);
1341 bound = build_int_cst (NULL_TREE, n - 1);
1342 /* Create an array type to hold them. */
1343 tmptype = build_range_type (gfc_array_index_type,
1344 gfc_index_zero_node, bound);
1345 tmptype = build_array_type (type, tmptype);
1347 init = build_constructor_from_list (tmptype, nreverse (list));
1348 TREE_CONSTANT (init) = 1;
1349 TREE_STATIC (init) = 1;
1350 /* Create a static variable to hold the data. */
1351 tmp = gfc_create_var (tmptype, "data");
1352 TREE_STATIC (tmp) = 1;
1353 TREE_CONSTANT (tmp) = 1;
1354 TREE_READONLY (tmp) = 1;
1355 DECL_INITIAL (tmp) = init;
1356 init = tmp;
1358 /* Use BUILTIN_MEMCPY to assign the values. */
1359 tmp = gfc_conv_descriptor_data_get (desc);
1360 tmp = build_fold_indirect_ref_loc (input_location,
1361 tmp);
1362 tmp = gfc_build_array_ref (tmp, *poffset, NULL);
1363 tmp = gfc_build_addr_expr (NULL_TREE, tmp);
1364 init = gfc_build_addr_expr (NULL_TREE, init);
1366 size = TREE_INT_CST_LOW (TYPE_SIZE_UNIT (type));
1367 bound = build_int_cst (NULL_TREE, n * size);
1368 tmp = build_call_expr_loc (input_location,
1369 built_in_decls[BUILT_IN_MEMCPY], 3,
1370 tmp, init, bound);
1371 gfc_add_expr_to_block (&body, tmp);
1373 *poffset = fold_build2 (PLUS_EXPR, gfc_array_index_type,
1374 *poffset,
1375 build_int_cst (gfc_array_index_type, n));
1377 if (!INTEGER_CST_P (*poffset))
1379 gfc_add_modify (&body, *offsetvar, *poffset);
1380 *poffset = *offsetvar;
1384 /* The frontend should already have done any expansions
1385 at compile-time. */
1386 if (!c->iterator)
1388 /* Pass the code as is. */
1389 tmp = gfc_finish_block (&body);
1390 gfc_add_expr_to_block (pblock, tmp);
1392 else
1394 /* Build the implied do-loop. */
1395 stmtblock_t implied_do_block;
1396 tree cond;
1397 tree end;
1398 tree step;
1399 tree exit_label;
1400 tree loopbody;
1401 tree tmp2;
1403 loopbody = gfc_finish_block (&body);
1405 /* Create a new block that holds the implied-do loop. A temporary
1406 loop-variable is used. */
1407 gfc_start_block(&implied_do_block);
1409 /* Initialize the loop. */
1410 gfc_init_se (&se, NULL);
1411 gfc_conv_expr_val (&se, c->iterator->start);
1412 gfc_add_block_to_block (&implied_do_block, &se.pre);
1413 gfc_add_modify (&implied_do_block, shadow_loopvar, se.expr);
1415 gfc_init_se (&se, NULL);
1416 gfc_conv_expr_val (&se, c->iterator->end);
1417 gfc_add_block_to_block (&implied_do_block, &se.pre);
1418 end = gfc_evaluate_now (se.expr, &implied_do_block);
1420 gfc_init_se (&se, NULL);
1421 gfc_conv_expr_val (&se, c->iterator->step);
1422 gfc_add_block_to_block (&implied_do_block, &se.pre);
1423 step = gfc_evaluate_now (se.expr, &implied_do_block);
1425 /* If this array expands dynamically, and the number of iterations
1426 is not constant, we won't have allocated space for the static
1427 part of C->EXPR's size. Do that now. */
1428 if (dynamic && gfc_iterator_has_dynamic_bounds (c->iterator))
1430 /* Get the number of iterations. */
1431 tmp = gfc_get_iteration_count (shadow_loopvar, end, step);
1433 /* Get the static part of C->EXPR's size. */
1434 gfc_get_array_constructor_element_size (&size, c->expr);
1435 tmp2 = gfc_conv_mpz_to_tree (size, gfc_index_integer_kind);
1437 /* Grow the array by TMP * TMP2 elements. */
1438 tmp = fold_build2 (MULT_EXPR, gfc_array_index_type, tmp, tmp2);
1439 gfc_grow_array (&implied_do_block, desc, tmp);
1442 /* Generate the loop body. */
1443 exit_label = gfc_build_label_decl (NULL_TREE);
1444 gfc_start_block (&body);
1446 /* Generate the exit condition. Depending on the sign of
1447 the step variable we have to generate the correct
1448 comparison. */
1449 tmp = fold_build2 (GT_EXPR, boolean_type_node, step,
1450 build_int_cst (TREE_TYPE (step), 0));
1451 cond = fold_build3 (COND_EXPR, boolean_type_node, tmp,
1452 fold_build2 (GT_EXPR, boolean_type_node,
1453 shadow_loopvar, end),
1454 fold_build2 (LT_EXPR, boolean_type_node,
1455 shadow_loopvar, end));
1456 tmp = build1_v (GOTO_EXPR, exit_label);
1457 TREE_USED (exit_label) = 1;
1458 tmp = build3_v (COND_EXPR, cond, tmp,
1459 build_empty_stmt (input_location));
1460 gfc_add_expr_to_block (&body, tmp);
1462 /* The main loop body. */
1463 gfc_add_expr_to_block (&body, loopbody);
1465 /* Increase loop variable by step. */
1466 tmp = fold_build2 (PLUS_EXPR, TREE_TYPE (shadow_loopvar), shadow_loopvar, step);
1467 gfc_add_modify (&body, shadow_loopvar, tmp);
1469 /* Finish the loop. */
1470 tmp = gfc_finish_block (&body);
1471 tmp = build1_v (LOOP_EXPR, tmp);
1472 gfc_add_expr_to_block (&implied_do_block, tmp);
1474 /* Add the exit label. */
1475 tmp = build1_v (LABEL_EXPR, exit_label);
1476 gfc_add_expr_to_block (&implied_do_block, tmp);
1478 /* Finishe the implied-do loop. */
1479 tmp = gfc_finish_block(&implied_do_block);
1480 gfc_add_expr_to_block(pblock, tmp);
1482 gfc_restore_sym (c->iterator->var->symtree->n.sym, &saved_loopvar);
1485 mpz_clear (size);
1489 /* Figure out the string length of a variable reference expression.
1490 Used by get_array_ctor_strlen. */
1492 static void
1493 get_array_ctor_var_strlen (gfc_expr * expr, tree * len)
1495 gfc_ref *ref;
1496 gfc_typespec *ts;
1497 mpz_t char_len;
1499 /* Don't bother if we already know the length is a constant. */
1500 if (*len && INTEGER_CST_P (*len))
1501 return;
1503 ts = &expr->symtree->n.sym->ts;
1504 for (ref = expr->ref; ref; ref = ref->next)
1506 switch (ref->type)
1508 case REF_ARRAY:
1509 /* Array references don't change the string length. */
1510 break;
1512 case REF_COMPONENT:
1513 /* Use the length of the component. */
1514 ts = &ref->u.c.component->ts;
1515 break;
1517 case REF_SUBSTRING:
1518 if (ref->u.ss.start->expr_type != EXPR_CONSTANT
1519 || ref->u.ss.end->expr_type != EXPR_CONSTANT)
1520 break;
1521 mpz_init_set_ui (char_len, 1);
1522 mpz_add (char_len, char_len, ref->u.ss.end->value.integer);
1523 mpz_sub (char_len, char_len, ref->u.ss.start->value.integer);
1524 *len = gfc_conv_mpz_to_tree (char_len, gfc_default_integer_kind);
1525 *len = convert (gfc_charlen_type_node, *len);
1526 mpz_clear (char_len);
1527 return;
1529 default:
1530 /* TODO: Substrings are tricky because we can't evaluate the
1531 expression more than once. For now we just give up, and hope
1532 we can figure it out elsewhere. */
1533 return;
1537 *len = ts->u.cl->backend_decl;
1541 /* A catch-all to obtain the string length for anything that is not a
1542 constant, array or variable. */
1543 static void
1544 get_array_ctor_all_strlen (stmtblock_t *block, gfc_expr *e, tree *len)
1546 gfc_se se;
1547 gfc_ss *ss;
1549 /* Don't bother if we already know the length is a constant. */
1550 if (*len && INTEGER_CST_P (*len))
1551 return;
1553 if (!e->ref && e->ts.u.cl && e->ts.u.cl->length
1554 && e->ts.u.cl->length->expr_type == EXPR_CONSTANT)
1556 /* This is easy. */
1557 gfc_conv_const_charlen (e->ts.u.cl);
1558 *len = e->ts.u.cl->backend_decl;
1560 else
1562 /* Otherwise, be brutal even if inefficient. */
1563 ss = gfc_walk_expr (e);
1564 gfc_init_se (&se, NULL);
1566 /* No function call, in case of side effects. */
1567 se.no_function_call = 1;
1568 if (ss == gfc_ss_terminator)
1569 gfc_conv_expr (&se, e);
1570 else
1571 gfc_conv_expr_descriptor (&se, e, ss);
1573 /* Fix the value. */
1574 *len = gfc_evaluate_now (se.string_length, &se.pre);
1576 gfc_add_block_to_block (block, &se.pre);
1577 gfc_add_block_to_block (block, &se.post);
1579 e->ts.u.cl->backend_decl = *len;
1584 /* Figure out the string length of a character array constructor.
1585 If len is NULL, don't calculate the length; this happens for recursive calls
1586 when a sub-array-constructor is an element but not at the first position,
1587 so when we're not interested in the length.
1588 Returns TRUE if all elements are character constants. */
1590 bool
1591 get_array_ctor_strlen (stmtblock_t *block, gfc_constructor_base base, tree * len)
1593 gfc_constructor *c;
1594 bool is_const;
1596 is_const = TRUE;
1598 if (gfc_constructor_first (base) == NULL)
1600 if (len)
1601 *len = build_int_cstu (gfc_charlen_type_node, 0);
1602 return is_const;
1605 /* Loop over all constructor elements to find out is_const, but in len we
1606 want to store the length of the first, not the last, element. We can
1607 of course exit the loop as soon as is_const is found to be false. */
1608 for (c = gfc_constructor_first (base);
1609 c && is_const; c = gfc_constructor_next (c))
1611 switch (c->expr->expr_type)
1613 case EXPR_CONSTANT:
1614 if (len && !(*len && INTEGER_CST_P (*len)))
1615 *len = build_int_cstu (gfc_charlen_type_node,
1616 c->expr->value.character.length);
1617 break;
1619 case EXPR_ARRAY:
1620 if (!get_array_ctor_strlen (block, c->expr->value.constructor, len))
1621 is_const = false;
1622 break;
1624 case EXPR_VARIABLE:
1625 is_const = false;
1626 if (len)
1627 get_array_ctor_var_strlen (c->expr, len);
1628 break;
1630 default:
1631 is_const = false;
1632 if (len)
1633 get_array_ctor_all_strlen (block, c->expr, len);
1634 break;
1637 /* After the first iteration, we don't want the length modified. */
1638 len = NULL;
1641 return is_const;
1644 /* Check whether the array constructor C consists entirely of constant
1645 elements, and if so returns the number of those elements, otherwise
1646 return zero. Note, an empty or NULL array constructor returns zero. */
1648 unsigned HOST_WIDE_INT
1649 gfc_constant_array_constructor_p (gfc_constructor_base base)
1651 unsigned HOST_WIDE_INT nelem = 0;
1653 gfc_constructor *c = gfc_constructor_first (base);
1654 while (c)
1656 if (c->iterator
1657 || c->expr->rank > 0
1658 || c->expr->expr_type != EXPR_CONSTANT)
1659 return 0;
1660 c = gfc_constructor_next (c);
1661 nelem++;
1663 return nelem;
1667 /* Given EXPR, the constant array constructor specified by an EXPR_ARRAY,
1668 and the tree type of it's elements, TYPE, return a static constant
1669 variable that is compile-time initialized. */
1671 tree
1672 gfc_build_constant_array_constructor (gfc_expr * expr, tree type)
1674 tree tmptype, list, init, tmp;
1675 HOST_WIDE_INT nelem;
1676 gfc_constructor *c;
1677 gfc_array_spec as;
1678 gfc_se se;
1679 int i;
1681 /* First traverse the constructor list, converting the constants
1682 to tree to build an initializer. */
1683 nelem = 0;
1684 list = NULL_TREE;
1685 c = gfc_constructor_first (expr->value.constructor);
1686 while (c)
1688 gfc_init_se (&se, NULL);
1689 gfc_conv_constant (&se, c->expr);
1690 if (c->expr->ts.type != BT_CHARACTER)
1691 se.expr = fold_convert (type, se.expr);
1692 else if (POINTER_TYPE_P (type))
1693 se.expr = gfc_build_addr_expr (gfc_get_pchar_type (c->expr->ts.kind),
1694 se.expr);
1695 list = tree_cons (build_int_cst (gfc_array_index_type, nelem),
1696 se.expr, list);
1697 c = gfc_constructor_next (c);
1698 nelem++;
1701 /* Next determine the tree type for the array. We use the gfortran
1702 front-end's gfc_get_nodesc_array_type in order to create a suitable
1703 GFC_ARRAY_TYPE_P that may be used by the scalarizer. */
1705 memset (&as, 0, sizeof (gfc_array_spec));
1707 as.rank = expr->rank;
1708 as.type = AS_EXPLICIT;
1709 if (!expr->shape)
1711 as.lower[0] = gfc_get_int_expr (gfc_default_integer_kind, NULL, 0);
1712 as.upper[0] = gfc_get_int_expr (gfc_default_integer_kind,
1713 NULL, nelem - 1);
1715 else
1716 for (i = 0; i < expr->rank; i++)
1718 int tmp = (int) mpz_get_si (expr->shape[i]);
1719 as.lower[i] = gfc_get_int_expr (gfc_default_integer_kind, NULL, 0);
1720 as.upper[i] = gfc_get_int_expr (gfc_default_integer_kind,
1721 NULL, tmp - 1);
1724 tmptype = gfc_get_nodesc_array_type (type, &as, PACKED_STATIC, true);
1726 init = build_constructor_from_list (tmptype, nreverse (list));
1728 TREE_CONSTANT (init) = 1;
1729 TREE_STATIC (init) = 1;
1731 tmp = gfc_create_var (tmptype, "A");
1732 TREE_STATIC (tmp) = 1;
1733 TREE_CONSTANT (tmp) = 1;
1734 TREE_READONLY (tmp) = 1;
1735 DECL_INITIAL (tmp) = init;
1737 return tmp;
1741 /* Translate a constant EXPR_ARRAY array constructor for the scalarizer.
1742 This mostly initializes the scalarizer state info structure with the
1743 appropriate values to directly use the array created by the function
1744 gfc_build_constant_array_constructor. */
1746 static void
1747 gfc_trans_constant_array_constructor (gfc_loopinfo * loop,
1748 gfc_ss * ss, tree type)
1750 gfc_ss_info *info;
1751 tree tmp;
1752 int i;
1754 tmp = gfc_build_constant_array_constructor (ss->expr, type);
1756 info = &ss->data.info;
1758 info->descriptor = tmp;
1759 info->data = gfc_build_addr_expr (NULL_TREE, tmp);
1760 info->offset = gfc_index_zero_node;
1762 for (i = 0; i < info->dimen; i++)
1764 info->delta[i] = gfc_index_zero_node;
1765 info->start[i] = gfc_index_zero_node;
1766 info->end[i] = gfc_index_zero_node;
1767 info->stride[i] = gfc_index_one_node;
1768 info->dim[i] = i;
1771 if (info->dimen > loop->temp_dim)
1772 loop->temp_dim = info->dimen;
1775 /* Helper routine of gfc_trans_array_constructor to determine if the
1776 bounds of the loop specified by LOOP are constant and simple enough
1777 to use with gfc_trans_constant_array_constructor. Returns the
1778 iteration count of the loop if suitable, and NULL_TREE otherwise. */
1780 static tree
1781 constant_array_constructor_loop_size (gfc_loopinfo * loop)
1783 tree size = gfc_index_one_node;
1784 tree tmp;
1785 int i;
1787 for (i = 0; i < loop->dimen; i++)
1789 /* If the bounds aren't constant, return NULL_TREE. */
1790 if (!INTEGER_CST_P (loop->from[i]) || !INTEGER_CST_P (loop->to[i]))
1791 return NULL_TREE;
1792 if (!integer_zerop (loop->from[i]))
1794 /* Only allow nonzero "from" in one-dimensional arrays. */
1795 if (loop->dimen != 1)
1796 return NULL_TREE;
1797 tmp = fold_build2 (MINUS_EXPR, gfc_array_index_type,
1798 loop->to[i], loop->from[i]);
1800 else
1801 tmp = loop->to[i];
1802 tmp = fold_build2 (PLUS_EXPR, gfc_array_index_type,
1803 tmp, gfc_index_one_node);
1804 size = fold_build2 (MULT_EXPR, gfc_array_index_type, size, tmp);
1807 return size;
1811 /* Array constructors are handled by constructing a temporary, then using that
1812 within the scalarization loop. This is not optimal, but seems by far the
1813 simplest method. */
1815 static void
1816 gfc_trans_array_constructor (gfc_loopinfo * loop, gfc_ss * ss, locus * where)
1818 gfc_constructor_base c;
1819 tree offset;
1820 tree offsetvar;
1821 tree desc;
1822 tree type;
1823 bool dynamic;
1824 bool old_first_len, old_typespec_chararray_ctor;
1825 tree old_first_len_val;
1827 /* Save the old values for nested checking. */
1828 old_first_len = first_len;
1829 old_first_len_val = first_len_val;
1830 old_typespec_chararray_ctor = typespec_chararray_ctor;
1832 /* Do bounds-checking here and in gfc_trans_array_ctor_element only if no
1833 typespec was given for the array constructor. */
1834 typespec_chararray_ctor = (ss->expr->ts.u.cl
1835 && ss->expr->ts.u.cl->length_from_typespec);
1837 if ((gfc_option.rtcheck & GFC_RTCHECK_BOUNDS)
1838 && ss->expr->ts.type == BT_CHARACTER && !typespec_chararray_ctor)
1840 first_len_val = gfc_create_var (gfc_charlen_type_node, "len");
1841 first_len = true;
1844 ss->data.info.dimen = loop->dimen;
1846 c = ss->expr->value.constructor;
1847 if (ss->expr->ts.type == BT_CHARACTER)
1849 bool const_string;
1851 /* get_array_ctor_strlen walks the elements of the constructor, if a
1852 typespec was given, we already know the string length and want the one
1853 specified there. */
1854 if (typespec_chararray_ctor && ss->expr->ts.u.cl->length
1855 && ss->expr->ts.u.cl->length->expr_type != EXPR_CONSTANT)
1857 gfc_se length_se;
1859 const_string = false;
1860 gfc_init_se (&length_se, NULL);
1861 gfc_conv_expr_type (&length_se, ss->expr->ts.u.cl->length,
1862 gfc_charlen_type_node);
1863 ss->string_length = length_se.expr;
1864 gfc_add_block_to_block (&loop->pre, &length_se.pre);
1865 gfc_add_block_to_block (&loop->post, &length_se.post);
1867 else
1868 const_string = get_array_ctor_strlen (&loop->pre, c,
1869 &ss->string_length);
1871 /* Complex character array constructors should have been taken care of
1872 and not end up here. */
1873 gcc_assert (ss->string_length);
1875 ss->expr->ts.u.cl->backend_decl = ss->string_length;
1877 type = gfc_get_character_type_len (ss->expr->ts.kind, ss->string_length);
1878 if (const_string)
1879 type = build_pointer_type (type);
1881 else
1882 type = gfc_typenode_for_spec (&ss->expr->ts);
1884 /* See if the constructor determines the loop bounds. */
1885 dynamic = false;
1887 if (ss->expr->shape && loop->dimen > 1 && loop->to[0] == NULL_TREE)
1889 /* We have a multidimensional parameter. */
1890 int n;
1891 for (n = 0; n < ss->expr->rank; n++)
1893 loop->from[n] = gfc_index_zero_node;
1894 loop->to[n] = gfc_conv_mpz_to_tree (ss->expr->shape [n],
1895 gfc_index_integer_kind);
1896 loop->to[n] = fold_build2 (MINUS_EXPR, gfc_array_index_type,
1897 loop->to[n], gfc_index_one_node);
1901 if (loop->to[0] == NULL_TREE)
1903 mpz_t size;
1905 /* We should have a 1-dimensional, zero-based loop. */
1906 gcc_assert (loop->dimen == 1);
1907 gcc_assert (integer_zerop (loop->from[0]));
1909 /* Split the constructor size into a static part and a dynamic part.
1910 Allocate the static size up-front and record whether the dynamic
1911 size might be nonzero. */
1912 mpz_init (size);
1913 dynamic = gfc_get_array_constructor_size (&size, c);
1914 mpz_sub_ui (size, size, 1);
1915 loop->to[0] = gfc_conv_mpz_to_tree (size, gfc_index_integer_kind);
1916 mpz_clear (size);
1919 /* Special case constant array constructors. */
1920 if (!dynamic)
1922 unsigned HOST_WIDE_INT nelem = gfc_constant_array_constructor_p (c);
1923 if (nelem > 0)
1925 tree size = constant_array_constructor_loop_size (loop);
1926 if (size && compare_tree_int (size, nelem) == 0)
1928 gfc_trans_constant_array_constructor (loop, ss, type);
1929 goto finish;
1934 gfc_trans_create_temp_array (&loop->pre, &loop->post, loop, &ss->data.info,
1935 type, NULL_TREE, dynamic, true, false, where);
1937 desc = ss->data.info.descriptor;
1938 offset = gfc_index_zero_node;
1939 offsetvar = gfc_create_var_np (gfc_array_index_type, "offset");
1940 TREE_NO_WARNING (offsetvar) = 1;
1941 TREE_USED (offsetvar) = 0;
1942 gfc_trans_array_constructor_value (&loop->pre, type, desc, c,
1943 &offset, &offsetvar, dynamic);
1945 /* If the array grows dynamically, the upper bound of the loop variable
1946 is determined by the array's final upper bound. */
1947 if (dynamic)
1948 loop->to[0] = gfc_conv_descriptor_ubound_get (desc, gfc_rank_cst[0]);
1950 if (TREE_USED (offsetvar))
1951 pushdecl (offsetvar);
1952 else
1953 gcc_assert (INTEGER_CST_P (offset));
1954 #if 0
1955 /* Disable bound checking for now because it's probably broken. */
1956 if (gfc_option.rtcheck & GFC_RTCHECK_BOUNDS)
1958 gcc_unreachable ();
1960 #endif
1962 finish:
1963 /* Restore old values of globals. */
1964 first_len = old_first_len;
1965 first_len_val = old_first_len_val;
1966 typespec_chararray_ctor = old_typespec_chararray_ctor;
1970 /* INFO describes a GFC_SS_SECTION in loop LOOP, and this function is
1971 called after evaluating all of INFO's vector dimensions. Go through
1972 each such vector dimension and see if we can now fill in any missing
1973 loop bounds. */
1975 static void
1976 gfc_set_vector_loop_bounds (gfc_loopinfo * loop, gfc_ss_info * info)
1978 gfc_se se;
1979 tree tmp;
1980 tree desc;
1981 tree zero;
1982 int n;
1983 int dim;
1985 for (n = 0; n < loop->dimen; n++)
1987 dim = info->dim[n];
1988 if (info->ref->u.ar.dimen_type[dim] == DIMEN_VECTOR
1989 && loop->to[n] == NULL)
1991 /* Loop variable N indexes vector dimension DIM, and we don't
1992 yet know the upper bound of loop variable N. Set it to the
1993 difference between the vector's upper and lower bounds. */
1994 gcc_assert (loop->from[n] == gfc_index_zero_node);
1995 gcc_assert (info->subscript[dim]
1996 && info->subscript[dim]->type == GFC_SS_VECTOR);
1998 gfc_init_se (&se, NULL);
1999 desc = info->subscript[dim]->data.info.descriptor;
2000 zero = gfc_rank_cst[0];
2001 tmp = fold_build2 (MINUS_EXPR, gfc_array_index_type,
2002 gfc_conv_descriptor_ubound_get (desc, zero),
2003 gfc_conv_descriptor_lbound_get (desc, zero));
2004 tmp = gfc_evaluate_now (tmp, &loop->pre);
2005 loop->to[n] = tmp;
2011 /* Add the pre and post chains for all the scalar expressions in a SS chain
2012 to loop. This is called after the loop parameters have been calculated,
2013 but before the actual scalarizing loops. */
2015 static void
2016 gfc_add_loop_ss_code (gfc_loopinfo * loop, gfc_ss * ss, bool subscript,
2017 locus * where)
2019 gfc_se se;
2020 int n;
2022 /* TODO: This can generate bad code if there are ordering dependencies,
2023 e.g., a callee allocated function and an unknown size constructor. */
2024 gcc_assert (ss != NULL);
2026 for (; ss != gfc_ss_terminator; ss = ss->loop_chain)
2028 gcc_assert (ss);
2030 switch (ss->type)
2032 case GFC_SS_SCALAR:
2033 /* Scalar expression. Evaluate this now. This includes elemental
2034 dimension indices, but not array section bounds. */
2035 gfc_init_se (&se, NULL);
2036 gfc_conv_expr (&se, ss->expr);
2037 gfc_add_block_to_block (&loop->pre, &se.pre);
2039 if (ss->expr->ts.type != BT_CHARACTER)
2041 /* Move the evaluation of scalar expressions outside the
2042 scalarization loop, except for WHERE assignments. */
2043 if (subscript)
2044 se.expr = convert(gfc_array_index_type, se.expr);
2045 if (!ss->where)
2046 se.expr = gfc_evaluate_now (se.expr, &loop->pre);
2047 gfc_add_block_to_block (&loop->pre, &se.post);
2049 else
2050 gfc_add_block_to_block (&loop->post, &se.post);
2052 ss->data.scalar.expr = se.expr;
2053 ss->string_length = se.string_length;
2054 break;
2056 case GFC_SS_REFERENCE:
2057 /* Scalar reference. Evaluate this now. */
2058 gfc_init_se (&se, NULL);
2059 gfc_conv_expr_reference (&se, ss->expr);
2060 gfc_add_block_to_block (&loop->pre, &se.pre);
2061 gfc_add_block_to_block (&loop->post, &se.post);
2063 ss->data.scalar.expr = gfc_evaluate_now (se.expr, &loop->pre);
2064 ss->string_length = se.string_length;
2065 break;
2067 case GFC_SS_SECTION:
2068 /* Add the expressions for scalar and vector subscripts. */
2069 for (n = 0; n < GFC_MAX_DIMENSIONS; n++)
2070 if (ss->data.info.subscript[n])
2071 gfc_add_loop_ss_code (loop, ss->data.info.subscript[n], true,
2072 where);
2074 gfc_set_vector_loop_bounds (loop, &ss->data.info);
2075 break;
2077 case GFC_SS_VECTOR:
2078 /* Get the vector's descriptor and store it in SS. */
2079 gfc_init_se (&se, NULL);
2080 gfc_conv_expr_descriptor (&se, ss->expr, gfc_walk_expr (ss->expr));
2081 gfc_add_block_to_block (&loop->pre, &se.pre);
2082 gfc_add_block_to_block (&loop->post, &se.post);
2083 ss->data.info.descriptor = se.expr;
2084 break;
2086 case GFC_SS_INTRINSIC:
2087 gfc_add_intrinsic_ss_code (loop, ss);
2088 break;
2090 case GFC_SS_FUNCTION:
2091 /* Array function return value. We call the function and save its
2092 result in a temporary for use inside the loop. */
2093 gfc_init_se (&se, NULL);
2094 se.loop = loop;
2095 se.ss = ss;
2096 gfc_conv_expr (&se, ss->expr);
2097 gfc_add_block_to_block (&loop->pre, &se.pre);
2098 gfc_add_block_to_block (&loop->post, &se.post);
2099 ss->string_length = se.string_length;
2100 break;
2102 case GFC_SS_CONSTRUCTOR:
2103 if (ss->expr->ts.type == BT_CHARACTER
2104 && ss->string_length == NULL
2105 && ss->expr->ts.u.cl
2106 && ss->expr->ts.u.cl->length)
2108 gfc_init_se (&se, NULL);
2109 gfc_conv_expr_type (&se, ss->expr->ts.u.cl->length,
2110 gfc_charlen_type_node);
2111 ss->string_length = se.expr;
2112 gfc_add_block_to_block (&loop->pre, &se.pre);
2113 gfc_add_block_to_block (&loop->post, &se.post);
2115 gfc_trans_array_constructor (loop, ss, where);
2116 break;
2118 case GFC_SS_TEMP:
2119 case GFC_SS_COMPONENT:
2120 /* Do nothing. These are handled elsewhere. */
2121 break;
2123 default:
2124 gcc_unreachable ();
2130 /* Translate expressions for the descriptor and data pointer of a SS. */
2131 /*GCC ARRAYS*/
2133 static void
2134 gfc_conv_ss_descriptor (stmtblock_t * block, gfc_ss * ss, int base)
2136 gfc_se se;
2137 tree tmp;
2139 /* Get the descriptor for the array to be scalarized. */
2140 gcc_assert (ss->expr->expr_type == EXPR_VARIABLE);
2141 gfc_init_se (&se, NULL);
2142 se.descriptor_only = 1;
2143 gfc_conv_expr_lhs (&se, ss->expr);
2144 gfc_add_block_to_block (block, &se.pre);
2145 ss->data.info.descriptor = se.expr;
2146 ss->string_length = se.string_length;
2148 if (base)
2150 /* Also the data pointer. */
2151 tmp = gfc_conv_array_data (se.expr);
2152 /* If this is a variable or address of a variable we use it directly.
2153 Otherwise we must evaluate it now to avoid breaking dependency
2154 analysis by pulling the expressions for elemental array indices
2155 inside the loop. */
2156 if (!(DECL_P (tmp)
2157 || (TREE_CODE (tmp) == ADDR_EXPR
2158 && DECL_P (TREE_OPERAND (tmp, 0)))))
2159 tmp = gfc_evaluate_now (tmp, block);
2160 ss->data.info.data = tmp;
2162 tmp = gfc_conv_array_offset (se.expr);
2163 ss->data.info.offset = gfc_evaluate_now (tmp, block);
2168 /* Initialize a gfc_loopinfo structure. */
2170 void
2171 gfc_init_loopinfo (gfc_loopinfo * loop)
2173 int n;
2175 memset (loop, 0, sizeof (gfc_loopinfo));
2176 gfc_init_block (&loop->pre);
2177 gfc_init_block (&loop->post);
2179 /* Initially scalarize in order. */
2180 for (n = 0; n < GFC_MAX_DIMENSIONS; n++)
2181 loop->order[n] = n;
2183 loop->ss = gfc_ss_terminator;
2187 /* Copies the loop variable info to a gfc_se structure. Does not copy the SS
2188 chain. */
2190 void
2191 gfc_copy_loopinfo_to_se (gfc_se * se, gfc_loopinfo * loop)
2193 se->loop = loop;
2197 /* Return an expression for the data pointer of an array. */
2199 tree
2200 gfc_conv_array_data (tree descriptor)
2202 tree type;
2204 type = TREE_TYPE (descriptor);
2205 if (GFC_ARRAY_TYPE_P (type))
2207 if (TREE_CODE (type) == POINTER_TYPE)
2208 return descriptor;
2209 else
2211 /* Descriptorless arrays. */
2212 return gfc_build_addr_expr (NULL_TREE, descriptor);
2215 else
2216 return gfc_conv_descriptor_data_get (descriptor);
2220 /* Return an expression for the base offset of an array. */
2222 tree
2223 gfc_conv_array_offset (tree descriptor)
2225 tree type;
2227 type = TREE_TYPE (descriptor);
2228 if (GFC_ARRAY_TYPE_P (type))
2229 return GFC_TYPE_ARRAY_OFFSET (type);
2230 else
2231 return gfc_conv_descriptor_offset_get (descriptor);
2235 /* Get an expression for the array stride. */
2237 tree
2238 gfc_conv_array_stride (tree descriptor, int dim)
2240 tree tmp;
2241 tree type;
2243 type = TREE_TYPE (descriptor);
2245 /* For descriptorless arrays use the array size. */
2246 tmp = GFC_TYPE_ARRAY_STRIDE (type, dim);
2247 if (tmp != NULL_TREE)
2248 return tmp;
2250 tmp = gfc_conv_descriptor_stride_get (descriptor, gfc_rank_cst[dim]);
2251 return tmp;
2255 /* Like gfc_conv_array_stride, but for the lower bound. */
2257 tree
2258 gfc_conv_array_lbound (tree descriptor, int dim)
2260 tree tmp;
2261 tree type;
2263 type = TREE_TYPE (descriptor);
2265 tmp = GFC_TYPE_ARRAY_LBOUND (type, dim);
2266 if (tmp != NULL_TREE)
2267 return tmp;
2269 tmp = gfc_conv_descriptor_lbound_get (descriptor, gfc_rank_cst[dim]);
2270 return tmp;
2274 /* Like gfc_conv_array_stride, but for the upper bound. */
2276 tree
2277 gfc_conv_array_ubound (tree descriptor, int dim)
2279 tree tmp;
2280 tree type;
2282 type = TREE_TYPE (descriptor);
2284 tmp = GFC_TYPE_ARRAY_UBOUND (type, dim);
2285 if (tmp != NULL_TREE)
2286 return tmp;
2288 /* This should only ever happen when passing an assumed shape array
2289 as an actual parameter. The value will never be used. */
2290 if (GFC_ARRAY_TYPE_P (TREE_TYPE (descriptor)))
2291 return gfc_index_zero_node;
2293 tmp = gfc_conv_descriptor_ubound_get (descriptor, gfc_rank_cst[dim]);
2294 return tmp;
2298 /* Generate code to perform an array index bound check. */
2300 static tree
2301 gfc_trans_array_bound_check (gfc_se * se, tree descriptor, tree index, int n,
2302 locus * where, bool check_upper)
2304 tree fault;
2305 tree tmp_lo, tmp_up;
2306 char *msg;
2307 const char * name = NULL;
2309 if (!(gfc_option.rtcheck & GFC_RTCHECK_BOUNDS))
2310 return index;
2312 index = gfc_evaluate_now (index, &se->pre);
2314 /* We find a name for the error message. */
2315 if (se->ss)
2316 name = se->ss->expr->symtree->name;
2318 if (!name && se->loop && se->loop->ss && se->loop->ss->expr
2319 && se->loop->ss->expr->symtree)
2320 name = se->loop->ss->expr->symtree->name;
2322 if (!name && se->loop && se->loop->ss && se->loop->ss->loop_chain
2323 && se->loop->ss->loop_chain->expr
2324 && se->loop->ss->loop_chain->expr->symtree)
2325 name = se->loop->ss->loop_chain->expr->symtree->name;
2327 if (!name && se->loop && se->loop->ss && se->loop->ss->expr)
2329 if (se->loop->ss->expr->expr_type == EXPR_FUNCTION
2330 && se->loop->ss->expr->value.function.name)
2331 name = se->loop->ss->expr->value.function.name;
2332 else
2333 if (se->loop->ss->type == GFC_SS_CONSTRUCTOR
2334 || se->loop->ss->type == GFC_SS_SCALAR)
2335 name = "unnamed constant";
2338 if (descriptor->base.code != COMPONENT_REF)
2339 name = IDENTIFIER_POINTER (DECL_NAME (descriptor));
2341 /* If upper bound is present, include both bounds in the error message. */
2342 if (check_upper)
2344 tmp_lo = gfc_conv_array_lbound (descriptor, n);
2345 tmp_up = gfc_conv_array_ubound (descriptor, n);
2347 if (name)
2348 asprintf (&msg, "Index '%%ld' of dimension %d of array '%s' "
2349 "outside of expected range (%%ld:%%ld)", n+1, name);
2350 else
2351 asprintf (&msg, "Index '%%ld' of dimension %d "
2352 "outside of expected range (%%ld:%%ld)", n+1);
2354 fault = fold_build2 (LT_EXPR, boolean_type_node, index, tmp_lo);
2355 gfc_trans_runtime_check (true, false, fault, &se->pre, where, msg,
2356 fold_convert (long_integer_type_node, index),
2357 fold_convert (long_integer_type_node, tmp_lo),
2358 fold_convert (long_integer_type_node, tmp_up));
2359 fault = fold_build2 (GT_EXPR, boolean_type_node, index, tmp_up);
2360 gfc_trans_runtime_check (true, false, fault, &se->pre, where, msg,
2361 fold_convert (long_integer_type_node, index),
2362 fold_convert (long_integer_type_node, tmp_lo),
2363 fold_convert (long_integer_type_node, tmp_up));
2364 gfc_free (msg);
2366 else
2368 tmp_lo = gfc_conv_array_lbound (descriptor, n);
2370 if (name)
2371 asprintf (&msg, "Index '%%ld' of dimension %d of array '%s' "
2372 "below lower bound of %%ld", n+1, name);
2373 else
2374 asprintf (&msg, "Index '%%ld' of dimension %d "
2375 "below lower bound of %%ld", n+1);
2377 fault = fold_build2 (LT_EXPR, boolean_type_node, index, tmp_lo);
2378 gfc_trans_runtime_check (true, false, fault, &se->pre, where, msg,
2379 fold_convert (long_integer_type_node, index),
2380 fold_convert (long_integer_type_node, tmp_lo));
2381 gfc_free (msg);
2384 return index;
2388 /* Return the offset for an index. Performs bound checking for elemental
2389 dimensions. Single element references are processed separately. */
2391 static tree
2392 gfc_conv_array_index_offset (gfc_se * se, gfc_ss_info * info, int dim, int i,
2393 gfc_array_ref * ar, tree stride)
2395 tree index;
2396 tree desc;
2397 tree data;
2399 /* Get the index into the array for this dimension. */
2400 if (ar)
2402 gcc_assert (ar->type != AR_ELEMENT);
2403 switch (ar->dimen_type[dim])
2405 case DIMEN_ELEMENT:
2406 /* Elemental dimension. */
2407 gcc_assert (info->subscript[dim]
2408 && info->subscript[dim]->type == GFC_SS_SCALAR);
2409 /* We've already translated this value outside the loop. */
2410 index = info->subscript[dim]->data.scalar.expr;
2412 index = gfc_trans_array_bound_check (se, info->descriptor,
2413 index, dim, &ar->where,
2414 ar->as->type != AS_ASSUMED_SIZE
2415 || dim < ar->dimen - 1);
2416 break;
2418 case DIMEN_VECTOR:
2419 gcc_assert (info && se->loop);
2420 gcc_assert (info->subscript[dim]
2421 && info->subscript[dim]->type == GFC_SS_VECTOR);
2422 desc = info->subscript[dim]->data.info.descriptor;
2424 /* Get a zero-based index into the vector. */
2425 index = fold_build2 (MINUS_EXPR, gfc_array_index_type,
2426 se->loop->loopvar[i], se->loop->from[i]);
2428 /* Multiply the index by the stride. */
2429 index = fold_build2 (MULT_EXPR, gfc_array_index_type,
2430 index, gfc_conv_array_stride (desc, 0));
2432 /* Read the vector to get an index into info->descriptor. */
2433 data = build_fold_indirect_ref_loc (input_location,
2434 gfc_conv_array_data (desc));
2435 index = gfc_build_array_ref (data, index, NULL);
2436 index = gfc_evaluate_now (index, &se->pre);
2438 /* Do any bounds checking on the final info->descriptor index. */
2439 index = gfc_trans_array_bound_check (se, info->descriptor,
2440 index, dim, &ar->where,
2441 ar->as->type != AS_ASSUMED_SIZE
2442 || dim < ar->dimen - 1);
2443 break;
2445 case DIMEN_RANGE:
2446 /* Scalarized dimension. */
2447 gcc_assert (info && se->loop);
2449 /* Multiply the loop variable by the stride and delta. */
2450 index = se->loop->loopvar[i];
2451 if (!integer_onep (info->stride[i]))
2452 index = fold_build2 (MULT_EXPR, gfc_array_index_type, index,
2453 info->stride[i]);
2454 if (!integer_zerop (info->delta[i]))
2455 index = fold_build2 (PLUS_EXPR, gfc_array_index_type, index,
2456 info->delta[i]);
2457 break;
2459 default:
2460 gcc_unreachable ();
2463 else
2465 /* Temporary array or derived type component. */
2466 gcc_assert (se->loop);
2467 index = se->loop->loopvar[se->loop->order[i]];
2468 if (!integer_zerop (info->delta[i]))
2469 index = fold_build2 (PLUS_EXPR, gfc_array_index_type,
2470 index, info->delta[i]);
2473 /* Multiply by the stride. */
2474 if (!integer_onep (stride))
2475 index = fold_build2 (MULT_EXPR, gfc_array_index_type, index, stride);
2477 return index;
2481 /* Build a scalarized reference to an array. */
2483 static void
2484 gfc_conv_scalarized_array_ref (gfc_se * se, gfc_array_ref * ar)
2486 gfc_ss_info *info;
2487 tree decl = NULL_TREE;
2488 tree index;
2489 tree tmp;
2490 int n;
2492 info = &se->ss->data.info;
2493 if (ar)
2494 n = se->loop->order[0];
2495 else
2496 n = 0;
2498 index = gfc_conv_array_index_offset (se, info, info->dim[n], n, ar,
2499 info->stride0);
2500 /* Add the offset for this dimension to the stored offset for all other
2501 dimensions. */
2502 if (!integer_zerop (info->offset))
2503 index = fold_build2 (PLUS_EXPR, gfc_array_index_type, index, info->offset);
2505 if (se->ss->expr && is_subref_array (se->ss->expr))
2506 decl = se->ss->expr->symtree->n.sym->backend_decl;
2508 tmp = build_fold_indirect_ref_loc (input_location,
2509 info->data);
2510 se->expr = gfc_build_array_ref (tmp, index, decl);
2514 /* Translate access of temporary array. */
2516 void
2517 gfc_conv_tmp_array_ref (gfc_se * se)
2519 se->string_length = se->ss->string_length;
2520 gfc_conv_scalarized_array_ref (se, NULL);
2524 /* Build an array reference. se->expr already holds the array descriptor.
2525 This should be either a variable, indirect variable reference or component
2526 reference. For arrays which do not have a descriptor, se->expr will be
2527 the data pointer.
2528 a(i, j, k) = base[offset + i * stride[0] + j * stride[1] + k * stride[2]]*/
2530 void
2531 gfc_conv_array_ref (gfc_se * se, gfc_array_ref * ar, gfc_symbol * sym,
2532 locus * where)
2534 int n;
2535 tree index;
2536 tree tmp;
2537 tree stride;
2538 gfc_se indexse;
2539 gfc_se tmpse;
2541 if (ar->dimen == 0)
2542 return;
2544 /* Handle scalarized references separately. */
2545 if (ar->type != AR_ELEMENT)
2547 gfc_conv_scalarized_array_ref (se, ar);
2548 gfc_advance_se_ss_chain (se);
2549 return;
2552 index = gfc_index_zero_node;
2554 /* Calculate the offsets from all the dimensions. */
2555 for (n = 0; n < ar->dimen; n++)
2557 /* Calculate the index for this dimension. */
2558 gfc_init_se (&indexse, se);
2559 gfc_conv_expr_type (&indexse, ar->start[n], gfc_array_index_type);
2560 gfc_add_block_to_block (&se->pre, &indexse.pre);
2562 if (gfc_option.rtcheck & GFC_RTCHECK_BOUNDS)
2564 /* Check array bounds. */
2565 tree cond;
2566 char *msg;
2568 /* Evaluate the indexse.expr only once. */
2569 indexse.expr = save_expr (indexse.expr);
2571 /* Lower bound. */
2572 tmp = gfc_conv_array_lbound (se->expr, n);
2573 if (sym->attr.temporary)
2575 gfc_init_se (&tmpse, se);
2576 gfc_conv_expr_type (&tmpse, ar->as->lower[n],
2577 gfc_array_index_type);
2578 gfc_add_block_to_block (&se->pre, &tmpse.pre);
2579 tmp = tmpse.expr;
2582 cond = fold_build2 (LT_EXPR, boolean_type_node,
2583 indexse.expr, tmp);
2584 asprintf (&msg, "Index '%%ld' of dimension %d of array '%s' "
2585 "below lower bound of %%ld", n+1, sym->name);
2586 gfc_trans_runtime_check (true, false, cond, &se->pre, where, msg,
2587 fold_convert (long_integer_type_node,
2588 indexse.expr),
2589 fold_convert (long_integer_type_node, tmp));
2590 gfc_free (msg);
2592 /* Upper bound, but not for the last dimension of assumed-size
2593 arrays. */
2594 if (n < ar->dimen - 1 || ar->as->type != AS_ASSUMED_SIZE)
2596 tmp = gfc_conv_array_ubound (se->expr, n);
2597 if (sym->attr.temporary)
2599 gfc_init_se (&tmpse, se);
2600 gfc_conv_expr_type (&tmpse, ar->as->upper[n],
2601 gfc_array_index_type);
2602 gfc_add_block_to_block (&se->pre, &tmpse.pre);
2603 tmp = tmpse.expr;
2606 cond = fold_build2 (GT_EXPR, boolean_type_node,
2607 indexse.expr, tmp);
2608 asprintf (&msg, "Index '%%ld' of dimension %d of array '%s' "
2609 "above upper bound of %%ld", n+1, sym->name);
2610 gfc_trans_runtime_check (true, false, cond, &se->pre, where, msg,
2611 fold_convert (long_integer_type_node,
2612 indexse.expr),
2613 fold_convert (long_integer_type_node, tmp));
2614 gfc_free (msg);
2618 /* Multiply the index by the stride. */
2619 stride = gfc_conv_array_stride (se->expr, n);
2620 tmp = fold_build2 (MULT_EXPR, gfc_array_index_type, indexse.expr,
2621 stride);
2623 /* And add it to the total. */
2624 index = fold_build2 (PLUS_EXPR, gfc_array_index_type, index, tmp);
2627 tmp = gfc_conv_array_offset (se->expr);
2628 if (!integer_zerop (tmp))
2629 index = fold_build2 (PLUS_EXPR, gfc_array_index_type, index, tmp);
2631 /* Access the calculated element. */
2632 tmp = gfc_conv_array_data (se->expr);
2633 tmp = build_fold_indirect_ref (tmp);
2634 se->expr = gfc_build_array_ref (tmp, index, sym->backend_decl);
2638 /* Generate the code to be executed immediately before entering a
2639 scalarization loop. */
2641 static void
2642 gfc_trans_preloop_setup (gfc_loopinfo * loop, int dim, int flag,
2643 stmtblock_t * pblock)
2645 tree index;
2646 tree stride;
2647 gfc_ss_info *info;
2648 gfc_ss *ss;
2649 gfc_se se;
2650 int i;
2652 /* This code will be executed before entering the scalarization loop
2653 for this dimension. */
2654 for (ss = loop->ss; ss != gfc_ss_terminator; ss = ss->loop_chain)
2656 if ((ss->useflags & flag) == 0)
2657 continue;
2659 if (ss->type != GFC_SS_SECTION
2660 && ss->type != GFC_SS_FUNCTION && ss->type != GFC_SS_CONSTRUCTOR
2661 && ss->type != GFC_SS_COMPONENT)
2662 continue;
2664 info = &ss->data.info;
2666 if (dim >= info->dimen)
2667 continue;
2669 if (dim == info->dimen - 1)
2671 /* For the outermost loop calculate the offset due to any
2672 elemental dimensions. It will have been initialized with the
2673 base offset of the array. */
2674 if (info->ref)
2676 for (i = 0; i < info->ref->u.ar.dimen; i++)
2678 if (info->ref->u.ar.dimen_type[i] != DIMEN_ELEMENT)
2679 continue;
2681 gfc_init_se (&se, NULL);
2682 se.loop = loop;
2683 se.expr = info->descriptor;
2684 stride = gfc_conv_array_stride (info->descriptor, i);
2685 index = gfc_conv_array_index_offset (&se, info, i, -1,
2686 &info->ref->u.ar,
2687 stride);
2688 gfc_add_block_to_block (pblock, &se.pre);
2690 info->offset = fold_build2 (PLUS_EXPR, gfc_array_index_type,
2691 info->offset, index);
2692 info->offset = gfc_evaluate_now (info->offset, pblock);
2695 i = loop->order[0];
2696 stride = gfc_conv_array_stride (info->descriptor, info->dim[i]);
2698 else
2699 stride = gfc_conv_array_stride (info->descriptor, 0);
2701 /* Calculate the stride of the innermost loop. Hopefully this will
2702 allow the backend optimizers to do their stuff more effectively.
2704 info->stride0 = gfc_evaluate_now (stride, pblock);
2706 else
2708 /* Add the offset for the previous loop dimension. */
2709 gfc_array_ref *ar;
2711 if (info->ref)
2713 ar = &info->ref->u.ar;
2714 i = loop->order[dim + 1];
2716 else
2718 ar = NULL;
2719 i = dim + 1;
2722 gfc_init_se (&se, NULL);
2723 se.loop = loop;
2724 se.expr = info->descriptor;
2725 stride = gfc_conv_array_stride (info->descriptor, info->dim[i]);
2726 index = gfc_conv_array_index_offset (&se, info, info->dim[i], i,
2727 ar, stride);
2728 gfc_add_block_to_block (pblock, &se.pre);
2729 info->offset = fold_build2 (PLUS_EXPR, gfc_array_index_type,
2730 info->offset, index);
2731 info->offset = gfc_evaluate_now (info->offset, pblock);
2734 /* Remember this offset for the second loop. */
2735 if (dim == loop->temp_dim - 1)
2736 info->saved_offset = info->offset;
2741 /* Start a scalarized expression. Creates a scope and declares loop
2742 variables. */
2744 void
2745 gfc_start_scalarized_body (gfc_loopinfo * loop, stmtblock_t * pbody)
2747 int dim;
2748 int n;
2749 int flags;
2751 gcc_assert (!loop->array_parameter);
2753 for (dim = loop->dimen - 1; dim >= 0; dim--)
2755 n = loop->order[dim];
2757 gfc_start_block (&loop->code[n]);
2759 /* Create the loop variable. */
2760 loop->loopvar[n] = gfc_create_var (gfc_array_index_type, "S");
2762 if (dim < loop->temp_dim)
2763 flags = 3;
2764 else
2765 flags = 1;
2766 /* Calculate values that will be constant within this loop. */
2767 gfc_trans_preloop_setup (loop, dim, flags, &loop->code[n]);
2769 gfc_start_block (pbody);
2773 /* Generates the actual loop code for a scalarization loop. */
2775 void
2776 gfc_trans_scalarized_loop_end (gfc_loopinfo * loop, int n,
2777 stmtblock_t * pbody)
2779 stmtblock_t block;
2780 tree cond;
2781 tree tmp;
2782 tree loopbody;
2783 tree exit_label;
2784 tree stmt;
2785 tree init;
2786 tree incr;
2788 if ((ompws_flags & (OMPWS_WORKSHARE_FLAG | OMPWS_SCALARIZER_WS))
2789 == (OMPWS_WORKSHARE_FLAG | OMPWS_SCALARIZER_WS)
2790 && n == loop->dimen - 1)
2792 /* We create an OMP_FOR construct for the outermost scalarized loop. */
2793 init = make_tree_vec (1);
2794 cond = make_tree_vec (1);
2795 incr = make_tree_vec (1);
2797 /* Cycle statement is implemented with a goto. Exit statement must not
2798 be present for this loop. */
2799 exit_label = gfc_build_label_decl (NULL_TREE);
2800 TREE_USED (exit_label) = 1;
2802 /* Label for cycle statements (if needed). */
2803 tmp = build1_v (LABEL_EXPR, exit_label);
2804 gfc_add_expr_to_block (pbody, tmp);
2806 stmt = make_node (OMP_FOR);
2808 TREE_TYPE (stmt) = void_type_node;
2809 OMP_FOR_BODY (stmt) = loopbody = gfc_finish_block (pbody);
2811 OMP_FOR_CLAUSES (stmt) = build_omp_clause (input_location,
2812 OMP_CLAUSE_SCHEDULE);
2813 OMP_CLAUSE_SCHEDULE_KIND (OMP_FOR_CLAUSES (stmt))
2814 = OMP_CLAUSE_SCHEDULE_STATIC;
2815 if (ompws_flags & OMPWS_NOWAIT)
2816 OMP_CLAUSE_CHAIN (OMP_FOR_CLAUSES (stmt))
2817 = build_omp_clause (input_location, OMP_CLAUSE_NOWAIT);
2819 /* Initialize the loopvar. */
2820 TREE_VEC_ELT (init, 0) = build2_v (MODIFY_EXPR, loop->loopvar[n],
2821 loop->from[n]);
2822 OMP_FOR_INIT (stmt) = init;
2823 /* The exit condition. */
2824 TREE_VEC_ELT (cond, 0) = build2 (LE_EXPR, boolean_type_node,
2825 loop->loopvar[n], loop->to[n]);
2826 OMP_FOR_COND (stmt) = cond;
2827 /* Increment the loopvar. */
2828 tmp = build2 (PLUS_EXPR, gfc_array_index_type,
2829 loop->loopvar[n], gfc_index_one_node);
2830 TREE_VEC_ELT (incr, 0) = fold_build2 (MODIFY_EXPR,
2831 void_type_node, loop->loopvar[n], tmp);
2832 OMP_FOR_INCR (stmt) = incr;
2834 ompws_flags &= ~OMPWS_CURR_SINGLEUNIT;
2835 gfc_add_expr_to_block (&loop->code[n], stmt);
2837 else
2839 loopbody = gfc_finish_block (pbody);
2841 /* Initialize the loopvar. */
2842 if (loop->loopvar[n] != loop->from[n])
2843 gfc_add_modify (&loop->code[n], loop->loopvar[n], loop->from[n]);
2845 exit_label = gfc_build_label_decl (NULL_TREE);
2847 /* Generate the loop body. */
2848 gfc_init_block (&block);
2850 /* The exit condition. */
2851 cond = fold_build2 (GT_EXPR, boolean_type_node,
2852 loop->loopvar[n], loop->to[n]);
2853 tmp = build1_v (GOTO_EXPR, exit_label);
2854 TREE_USED (exit_label) = 1;
2855 tmp = build3_v (COND_EXPR, cond, tmp, build_empty_stmt (input_location));
2856 gfc_add_expr_to_block (&block, tmp);
2858 /* The main body. */
2859 gfc_add_expr_to_block (&block, loopbody);
2861 /* Increment the loopvar. */
2862 tmp = fold_build2 (PLUS_EXPR, gfc_array_index_type,
2863 loop->loopvar[n], gfc_index_one_node);
2864 gfc_add_modify (&block, loop->loopvar[n], tmp);
2866 /* Build the loop. */
2867 tmp = gfc_finish_block (&block);
2868 tmp = build1_v (LOOP_EXPR, tmp);
2869 gfc_add_expr_to_block (&loop->code[n], tmp);
2871 /* Add the exit label. */
2872 tmp = build1_v (LABEL_EXPR, exit_label);
2873 gfc_add_expr_to_block (&loop->code[n], tmp);
2879 /* Finishes and generates the loops for a scalarized expression. */
2881 void
2882 gfc_trans_scalarizing_loops (gfc_loopinfo * loop, stmtblock_t * body)
2884 int dim;
2885 int n;
2886 gfc_ss *ss;
2887 stmtblock_t *pblock;
2888 tree tmp;
2890 pblock = body;
2891 /* Generate the loops. */
2892 for (dim = 0; dim < loop->dimen; dim++)
2894 n = loop->order[dim];
2895 gfc_trans_scalarized_loop_end (loop, n, pblock);
2896 loop->loopvar[n] = NULL_TREE;
2897 pblock = &loop->code[n];
2900 tmp = gfc_finish_block (pblock);
2901 gfc_add_expr_to_block (&loop->pre, tmp);
2903 /* Clear all the used flags. */
2904 for (ss = loop->ss; ss; ss = ss->loop_chain)
2905 ss->useflags = 0;
2909 /* Finish the main body of a scalarized expression, and start the secondary
2910 copying body. */
2912 void
2913 gfc_trans_scalarized_loop_boundary (gfc_loopinfo * loop, stmtblock_t * body)
2915 int dim;
2916 int n;
2917 stmtblock_t *pblock;
2918 gfc_ss *ss;
2920 pblock = body;
2921 /* We finish as many loops as are used by the temporary. */
2922 for (dim = 0; dim < loop->temp_dim - 1; dim++)
2924 n = loop->order[dim];
2925 gfc_trans_scalarized_loop_end (loop, n, pblock);
2926 loop->loopvar[n] = NULL_TREE;
2927 pblock = &loop->code[n];
2930 /* We don't want to finish the outermost loop entirely. */
2931 n = loop->order[loop->temp_dim - 1];
2932 gfc_trans_scalarized_loop_end (loop, n, pblock);
2934 /* Restore the initial offsets. */
2935 for (ss = loop->ss; ss != gfc_ss_terminator; ss = ss->loop_chain)
2937 if ((ss->useflags & 2) == 0)
2938 continue;
2940 if (ss->type != GFC_SS_SECTION
2941 && ss->type != GFC_SS_FUNCTION && ss->type != GFC_SS_CONSTRUCTOR
2942 && ss->type != GFC_SS_COMPONENT)
2943 continue;
2945 ss->data.info.offset = ss->data.info.saved_offset;
2948 /* Restart all the inner loops we just finished. */
2949 for (dim = loop->temp_dim - 2; dim >= 0; dim--)
2951 n = loop->order[dim];
2953 gfc_start_block (&loop->code[n]);
2955 loop->loopvar[n] = gfc_create_var (gfc_array_index_type, "Q");
2957 gfc_trans_preloop_setup (loop, dim, 2, &loop->code[n]);
2960 /* Start a block for the secondary copying code. */
2961 gfc_start_block (body);
2965 /* Calculate the upper bound of an array section. */
2967 static tree
2968 gfc_conv_section_upper_bound (gfc_ss * ss, int n, stmtblock_t * pblock)
2970 int dim;
2971 gfc_expr *end;
2972 tree desc;
2973 tree bound;
2974 gfc_se se;
2975 gfc_ss_info *info;
2977 gcc_assert (ss->type == GFC_SS_SECTION);
2979 info = &ss->data.info;
2980 dim = info->dim[n];
2982 if (info->ref->u.ar.dimen_type[dim] == DIMEN_VECTOR)
2983 /* We'll calculate the upper bound once we have access to the
2984 vector's descriptor. */
2985 return NULL;
2987 gcc_assert (info->ref->u.ar.dimen_type[dim] == DIMEN_RANGE);
2988 desc = info->descriptor;
2989 end = info->ref->u.ar.end[dim];
2991 if (end)
2993 /* The upper bound was specified. */
2994 gfc_init_se (&se, NULL);
2995 gfc_conv_expr_type (&se, end, gfc_array_index_type);
2996 gfc_add_block_to_block (pblock, &se.pre);
2997 bound = se.expr;
2999 else
3001 /* No upper bound was specified, so use the bound of the array. */
3002 bound = gfc_conv_array_ubound (desc, dim);
3005 return bound;
3009 /* Calculate the lower bound of an array section. */
3011 static void
3012 gfc_conv_section_startstride (gfc_loopinfo * loop, gfc_ss * ss, int n)
3014 gfc_expr *start;
3015 gfc_expr *end;
3016 gfc_expr *stride;
3017 tree desc;
3018 gfc_se se;
3019 gfc_ss_info *info;
3020 int dim;
3022 gcc_assert (ss->type == GFC_SS_SECTION);
3024 info = &ss->data.info;
3025 dim = info->dim[n];
3027 if (info->ref->u.ar.dimen_type[dim] == DIMEN_VECTOR)
3029 /* We use a zero-based index to access the vector. */
3030 info->start[n] = gfc_index_zero_node;
3031 info->end[n] = gfc_index_zero_node;
3032 info->stride[n] = gfc_index_one_node;
3033 return;
3036 gcc_assert (info->ref->u.ar.dimen_type[dim] == DIMEN_RANGE);
3037 desc = info->descriptor;
3038 start = info->ref->u.ar.start[dim];
3039 end = info->ref->u.ar.end[dim];
3040 stride = info->ref->u.ar.stride[dim];
3042 /* Calculate the start of the range. For vector subscripts this will
3043 be the range of the vector. */
3044 if (start)
3046 /* Specified section start. */
3047 gfc_init_se (&se, NULL);
3048 gfc_conv_expr_type (&se, start, gfc_array_index_type);
3049 gfc_add_block_to_block (&loop->pre, &se.pre);
3050 info->start[n] = se.expr;
3052 else
3054 /* No lower bound specified so use the bound of the array. */
3055 info->start[n] = gfc_conv_array_lbound (desc, dim);
3057 info->start[n] = gfc_evaluate_now (info->start[n], &loop->pre);
3059 /* Similarly calculate the end. Although this is not used in the
3060 scalarizer, it is needed when checking bounds and where the end
3061 is an expression with side-effects. */
3062 if (end)
3064 /* Specified section start. */
3065 gfc_init_se (&se, NULL);
3066 gfc_conv_expr_type (&se, end, gfc_array_index_type);
3067 gfc_add_block_to_block (&loop->pre, &se.pre);
3068 info->end[n] = se.expr;
3070 else
3072 /* No upper bound specified so use the bound of the array. */
3073 info->end[n] = gfc_conv_array_ubound (desc, dim);
3075 info->end[n] = gfc_evaluate_now (info->end[n], &loop->pre);
3077 /* Calculate the stride. */
3078 if (stride == NULL)
3079 info->stride[n] = gfc_index_one_node;
3080 else
3082 gfc_init_se (&se, NULL);
3083 gfc_conv_expr_type (&se, stride, gfc_array_index_type);
3084 gfc_add_block_to_block (&loop->pre, &se.pre);
3085 info->stride[n] = gfc_evaluate_now (se.expr, &loop->pre);
3090 /* Calculates the range start and stride for a SS chain. Also gets the
3091 descriptor and data pointer. The range of vector subscripts is the size
3092 of the vector. Array bounds are also checked. */
3094 void
3095 gfc_conv_ss_startstride (gfc_loopinfo * loop)
3097 int n;
3098 tree tmp;
3099 gfc_ss *ss;
3100 tree desc;
3102 loop->dimen = 0;
3103 /* Determine the rank of the loop. */
3104 for (ss = loop->ss;
3105 ss != gfc_ss_terminator && loop->dimen == 0; ss = ss->loop_chain)
3107 switch (ss->type)
3109 case GFC_SS_SECTION:
3110 case GFC_SS_CONSTRUCTOR:
3111 case GFC_SS_FUNCTION:
3112 case GFC_SS_COMPONENT:
3113 loop->dimen = ss->data.info.dimen;
3114 break;
3116 /* As usual, lbound and ubound are exceptions!. */
3117 case GFC_SS_INTRINSIC:
3118 switch (ss->expr->value.function.isym->id)
3120 case GFC_ISYM_LBOUND:
3121 case GFC_ISYM_UBOUND:
3122 loop->dimen = ss->data.info.dimen;
3124 default:
3125 break;
3128 default:
3129 break;
3133 /* We should have determined the rank of the expression by now. If
3134 not, that's bad news. */
3135 gcc_assert (loop->dimen != 0);
3137 /* Loop over all the SS in the chain. */
3138 for (ss = loop->ss; ss != gfc_ss_terminator; ss = ss->loop_chain)
3140 if (ss->expr && ss->expr->shape && !ss->shape)
3141 ss->shape = ss->expr->shape;
3143 switch (ss->type)
3145 case GFC_SS_SECTION:
3146 /* Get the descriptor for the array. */
3147 gfc_conv_ss_descriptor (&loop->pre, ss, !loop->array_parameter);
3149 for (n = 0; n < ss->data.info.dimen; n++)
3150 gfc_conv_section_startstride (loop, ss, n);
3151 break;
3153 case GFC_SS_INTRINSIC:
3154 switch (ss->expr->value.function.isym->id)
3156 /* Fall through to supply start and stride. */
3157 case GFC_ISYM_LBOUND:
3158 case GFC_ISYM_UBOUND:
3159 break;
3160 default:
3161 continue;
3164 case GFC_SS_CONSTRUCTOR:
3165 case GFC_SS_FUNCTION:
3166 for (n = 0; n < ss->data.info.dimen; n++)
3168 ss->data.info.start[n] = gfc_index_zero_node;
3169 ss->data.info.end[n] = gfc_index_zero_node;
3170 ss->data.info.stride[n] = gfc_index_one_node;
3172 break;
3174 default:
3175 break;
3179 /* The rest is just runtime bound checking. */
3180 if (gfc_option.rtcheck & GFC_RTCHECK_BOUNDS)
3182 stmtblock_t block;
3183 tree lbound, ubound;
3184 tree end;
3185 tree size[GFC_MAX_DIMENSIONS];
3186 tree stride_pos, stride_neg, non_zerosized, tmp2, tmp3;
3187 gfc_ss_info *info;
3188 char *msg;
3189 int dim;
3191 gfc_start_block (&block);
3193 for (n = 0; n < loop->dimen; n++)
3194 size[n] = NULL_TREE;
3196 for (ss = loop->ss; ss != gfc_ss_terminator; ss = ss->loop_chain)
3198 stmtblock_t inner;
3200 if (ss->type != GFC_SS_SECTION)
3201 continue;
3203 gfc_start_block (&inner);
3205 /* TODO: range checking for mapped dimensions. */
3206 info = &ss->data.info;
3208 /* This code only checks ranges. Elemental and vector
3209 dimensions are checked later. */
3210 for (n = 0; n < loop->dimen; n++)
3212 bool check_upper;
3214 dim = info->dim[n];
3215 if (info->ref->u.ar.dimen_type[dim] != DIMEN_RANGE)
3216 continue;
3218 if (dim == info->ref->u.ar.dimen - 1
3219 && info->ref->u.ar.as->type == AS_ASSUMED_SIZE)
3220 check_upper = false;
3221 else
3222 check_upper = true;
3224 /* Zero stride is not allowed. */
3225 tmp = fold_build2 (EQ_EXPR, boolean_type_node, info->stride[n],
3226 gfc_index_zero_node);
3227 asprintf (&msg, "Zero stride is not allowed, for dimension %d "
3228 "of array '%s'", info->dim[n]+1,
3229 ss->expr->symtree->name);
3230 gfc_trans_runtime_check (true, false, tmp, &inner,
3231 &ss->expr->where, msg);
3232 gfc_free (msg);
3234 desc = ss->data.info.descriptor;
3236 /* This is the run-time equivalent of resolve.c's
3237 check_dimension(). The logical is more readable there
3238 than it is here, with all the trees. */
3239 lbound = gfc_conv_array_lbound (desc, dim);
3240 end = info->end[n];
3241 if (check_upper)
3242 ubound = gfc_conv_array_ubound (desc, dim);
3243 else
3244 ubound = NULL;
3246 /* non_zerosized is true when the selected range is not
3247 empty. */
3248 stride_pos = fold_build2 (GT_EXPR, boolean_type_node,
3249 info->stride[n], gfc_index_zero_node);
3250 tmp = fold_build2 (LE_EXPR, boolean_type_node, info->start[n],
3251 end);
3252 stride_pos = fold_build2 (TRUTH_AND_EXPR, boolean_type_node,
3253 stride_pos, tmp);
3255 stride_neg = fold_build2 (LT_EXPR, boolean_type_node,
3256 info->stride[n], gfc_index_zero_node);
3257 tmp = fold_build2 (GE_EXPR, boolean_type_node, info->start[n],
3258 end);
3259 stride_neg = fold_build2 (TRUTH_AND_EXPR, boolean_type_node,
3260 stride_neg, tmp);
3261 non_zerosized = fold_build2 (TRUTH_OR_EXPR, boolean_type_node,
3262 stride_pos, stride_neg);
3264 /* Check the start of the range against the lower and upper
3265 bounds of the array, if the range is not empty.
3266 If upper bound is present, include both bounds in the
3267 error message. */
3268 if (check_upper)
3270 tmp = fold_build2 (LT_EXPR, boolean_type_node,
3271 info->start[n], lbound);
3272 tmp = fold_build2 (TRUTH_AND_EXPR, boolean_type_node,
3273 non_zerosized, tmp);
3274 tmp2 = fold_build2 (GT_EXPR, boolean_type_node,
3275 info->start[n], ubound);
3276 tmp2 = fold_build2 (TRUTH_AND_EXPR, boolean_type_node,
3277 non_zerosized, tmp2);
3278 asprintf (&msg, "Index '%%ld' of dimension %d of array '%s' "
3279 "outside of expected range (%%ld:%%ld)",
3280 info->dim[n]+1, ss->expr->symtree->name);
3281 gfc_trans_runtime_check (true, false, tmp, &inner,
3282 &ss->expr->where, msg,
3283 fold_convert (long_integer_type_node, info->start[n]),
3284 fold_convert (long_integer_type_node, lbound),
3285 fold_convert (long_integer_type_node, ubound));
3286 gfc_trans_runtime_check (true, false, tmp2, &inner,
3287 &ss->expr->where, msg,
3288 fold_convert (long_integer_type_node, info->start[n]),
3289 fold_convert (long_integer_type_node, lbound),
3290 fold_convert (long_integer_type_node, ubound));
3291 gfc_free (msg);
3293 else
3295 tmp = fold_build2 (LT_EXPR, boolean_type_node,
3296 info->start[n], lbound);
3297 tmp = fold_build2 (TRUTH_AND_EXPR, boolean_type_node,
3298 non_zerosized, tmp);
3299 asprintf (&msg, "Index '%%ld' of dimension %d of array '%s' "
3300 "below lower bound of %%ld",
3301 info->dim[n]+1, ss->expr->symtree->name);
3302 gfc_trans_runtime_check (true, false, tmp, &inner,
3303 &ss->expr->where, msg,
3304 fold_convert (long_integer_type_node, info->start[n]),
3305 fold_convert (long_integer_type_node, lbound));
3306 gfc_free (msg);
3309 /* Compute the last element of the range, which is not
3310 necessarily "end" (think 0:5:3, which doesn't contain 5)
3311 and check it against both lower and upper bounds. */
3313 tmp = fold_build2 (MINUS_EXPR, gfc_array_index_type, end,
3314 info->start[n]);
3315 tmp = fold_build2 (TRUNC_MOD_EXPR, gfc_array_index_type, tmp,
3316 info->stride[n]);
3317 tmp = fold_build2 (MINUS_EXPR, gfc_array_index_type, end,
3318 tmp);
3319 tmp2 = fold_build2 (LT_EXPR, boolean_type_node, tmp, lbound);
3320 tmp2 = fold_build2 (TRUTH_AND_EXPR, boolean_type_node,
3321 non_zerosized, tmp2);
3322 if (check_upper)
3324 tmp3 = fold_build2 (GT_EXPR, boolean_type_node, tmp, ubound);
3325 tmp3 = fold_build2 (TRUTH_AND_EXPR, boolean_type_node,
3326 non_zerosized, tmp3);
3327 asprintf (&msg, "Index '%%ld' of dimension %d of array '%s' "
3328 "outside of expected range (%%ld:%%ld)",
3329 info->dim[n]+1, ss->expr->symtree->name);
3330 gfc_trans_runtime_check (true, false, tmp2, &inner,
3331 &ss->expr->where, msg,
3332 fold_convert (long_integer_type_node, tmp),
3333 fold_convert (long_integer_type_node, ubound),
3334 fold_convert (long_integer_type_node, lbound));
3335 gfc_trans_runtime_check (true, false, tmp3, &inner,
3336 &ss->expr->where, msg,
3337 fold_convert (long_integer_type_node, tmp),
3338 fold_convert (long_integer_type_node, ubound),
3339 fold_convert (long_integer_type_node, lbound));
3340 gfc_free (msg);
3342 else
3344 asprintf (&msg, "Index '%%ld' of dimension %d of array '%s' "
3345 "below lower bound of %%ld",
3346 info->dim[n]+1, ss->expr->symtree->name);
3347 gfc_trans_runtime_check (true, false, tmp2, &inner,
3348 &ss->expr->where, msg,
3349 fold_convert (long_integer_type_node, tmp),
3350 fold_convert (long_integer_type_node, lbound));
3351 gfc_free (msg);
3354 /* Check the section sizes match. */
3355 tmp = fold_build2 (MINUS_EXPR, gfc_array_index_type, end,
3356 info->start[n]);
3357 tmp = fold_build2 (FLOOR_DIV_EXPR, gfc_array_index_type, tmp,
3358 info->stride[n]);
3359 tmp = fold_build2 (PLUS_EXPR, gfc_array_index_type,
3360 gfc_index_one_node, tmp);
3361 tmp = fold_build2 (MAX_EXPR, gfc_array_index_type, tmp,
3362 build_int_cst (gfc_array_index_type, 0));
3363 /* We remember the size of the first section, and check all the
3364 others against this. */
3365 if (size[n])
3367 tmp3 = fold_build2 (NE_EXPR, boolean_type_node, tmp, size[n]);
3368 asprintf (&msg, "Array bound mismatch for dimension %d "
3369 "of array '%s' (%%ld/%%ld)",
3370 info->dim[n]+1, ss->expr->symtree->name);
3372 gfc_trans_runtime_check (true, false, tmp3, &inner,
3373 &ss->expr->where, msg,
3374 fold_convert (long_integer_type_node, tmp),
3375 fold_convert (long_integer_type_node, size[n]));
3377 gfc_free (msg);
3379 else
3380 size[n] = gfc_evaluate_now (tmp, &inner);
3383 tmp = gfc_finish_block (&inner);
3385 /* For optional arguments, only check bounds if the argument is
3386 present. */
3387 if (ss->expr->symtree->n.sym->attr.optional
3388 || ss->expr->symtree->n.sym->attr.not_always_present)
3389 tmp = build3_v (COND_EXPR,
3390 gfc_conv_expr_present (ss->expr->symtree->n.sym),
3391 tmp, build_empty_stmt (input_location));
3393 gfc_add_expr_to_block (&block, tmp);
3397 tmp = gfc_finish_block (&block);
3398 gfc_add_expr_to_block (&loop->pre, tmp);
3403 /* Return true if the two SS could be aliased, i.e. both point to the same data
3404 object. */
3405 /* TODO: resolve aliases based on frontend expressions. */
3407 static int
3408 gfc_could_be_alias (gfc_ss * lss, gfc_ss * rss)
3410 gfc_ref *lref;
3411 gfc_ref *rref;
3412 gfc_symbol *lsym;
3413 gfc_symbol *rsym;
3415 lsym = lss->expr->symtree->n.sym;
3416 rsym = rss->expr->symtree->n.sym;
3417 if (gfc_symbols_could_alias (lsym, rsym))
3418 return 1;
3420 if (rsym->ts.type != BT_DERIVED
3421 && lsym->ts.type != BT_DERIVED)
3422 return 0;
3424 /* For derived types we must check all the component types. We can ignore
3425 array references as these will have the same base type as the previous
3426 component ref. */
3427 for (lref = lss->expr->ref; lref != lss->data.info.ref; lref = lref->next)
3429 if (lref->type != REF_COMPONENT)
3430 continue;
3432 if (gfc_symbols_could_alias (lref->u.c.sym, rsym))
3433 return 1;
3435 for (rref = rss->expr->ref; rref != rss->data.info.ref;
3436 rref = rref->next)
3438 if (rref->type != REF_COMPONENT)
3439 continue;
3441 if (gfc_symbols_could_alias (lref->u.c.sym, rref->u.c.sym))
3442 return 1;
3446 for (rref = rss->expr->ref; rref != rss->data.info.ref; rref = rref->next)
3448 if (rref->type != REF_COMPONENT)
3449 break;
3451 if (gfc_symbols_could_alias (rref->u.c.sym, lsym))
3452 return 1;
3455 return 0;
3459 /* Resolve array data dependencies. Creates a temporary if required. */
3460 /* TODO: Calc dependencies with gfc_expr rather than gfc_ss, and move to
3461 dependency.c. */
3463 void
3464 gfc_conv_resolve_dependencies (gfc_loopinfo * loop, gfc_ss * dest,
3465 gfc_ss * rss)
3467 gfc_ss *ss;
3468 gfc_ref *lref;
3469 gfc_ref *rref;
3470 int nDepend = 0;
3472 loop->temp_ss = NULL;
3474 for (ss = rss; ss != gfc_ss_terminator; ss = ss->next)
3476 if (ss->type != GFC_SS_SECTION)
3477 continue;
3479 if (dest->expr->symtree->n.sym != ss->expr->symtree->n.sym)
3481 if (gfc_could_be_alias (dest, ss)
3482 || gfc_are_equivalenced_arrays (dest->expr, ss->expr))
3484 nDepend = 1;
3485 break;
3488 else
3490 lref = dest->expr->ref;
3491 rref = ss->expr->ref;
3493 nDepend = gfc_dep_resolver (lref, rref);
3494 if (nDepend == 1)
3495 break;
3496 #if 0
3497 /* TODO : loop shifting. */
3498 if (nDepend == 1)
3500 /* Mark the dimensions for LOOP SHIFTING */
3501 for (n = 0; n < loop->dimen; n++)
3503 int dim = dest->data.info.dim[n];
3505 if (lref->u.ar.dimen_type[dim] == DIMEN_VECTOR)
3506 depends[n] = 2;
3507 else if (! gfc_is_same_range (&lref->u.ar,
3508 &rref->u.ar, dim, 0))
3509 depends[n] = 1;
3512 /* Put all the dimensions with dependencies in the
3513 innermost loops. */
3514 dim = 0;
3515 for (n = 0; n < loop->dimen; n++)
3517 gcc_assert (loop->order[n] == n);
3518 if (depends[n])
3519 loop->order[dim++] = n;
3521 for (n = 0; n < loop->dimen; n++)
3523 if (! depends[n])
3524 loop->order[dim++] = n;
3527 gcc_assert (dim == loop->dimen);
3528 break;
3530 #endif
3534 if (nDepend == 1)
3536 tree base_type = gfc_typenode_for_spec (&dest->expr->ts);
3537 if (GFC_ARRAY_TYPE_P (base_type)
3538 || GFC_DESCRIPTOR_TYPE_P (base_type))
3539 base_type = gfc_get_element_type (base_type);
3540 loop->temp_ss = gfc_get_ss ();
3541 loop->temp_ss->type = GFC_SS_TEMP;
3542 loop->temp_ss->data.temp.type = base_type;
3543 loop->temp_ss->string_length = dest->string_length;
3544 loop->temp_ss->data.temp.dimen = loop->dimen;
3545 loop->temp_ss->next = gfc_ss_terminator;
3546 gfc_add_ss_to_loop (loop, loop->temp_ss);
3548 else
3549 loop->temp_ss = NULL;
3553 /* Initialize the scalarization loop. Creates the loop variables. Determines
3554 the range of the loop variables. Creates a temporary if required.
3555 Calculates how to transform from loop variables to array indices for each
3556 expression. Also generates code for scalar expressions which have been
3557 moved outside the loop. */
3559 void
3560 gfc_conv_loop_setup (gfc_loopinfo * loop, locus * where)
3562 int n;
3563 gfc_ss_info *info;
3564 gfc_ss_info *specinfo;
3565 gfc_ss *ss;
3566 tree tmp;
3567 gfc_ss *loopspec[GFC_MAX_DIMENSIONS];
3568 bool dynamic[GFC_MAX_DIMENSIONS];
3569 mpz_t *cshape;
3570 mpz_t i;
3572 mpz_init (i);
3573 for (n = 0; n < loop->dimen; n++)
3575 loopspec[n] = NULL;
3576 dynamic[n] = false;
3577 /* We use one SS term, and use that to determine the bounds of the
3578 loop for this dimension. We try to pick the simplest term. */
3579 for (ss = loop->ss; ss != gfc_ss_terminator; ss = ss->loop_chain)
3581 if (ss->shape)
3583 /* The frontend has worked out the size for us. */
3584 if (!loopspec[n] || !loopspec[n]->shape
3585 || !integer_zerop (loopspec[n]->data.info.start[n]))
3586 /* Prefer zero-based descriptors if possible. */
3587 loopspec[n] = ss;
3588 continue;
3591 if (ss->type == GFC_SS_CONSTRUCTOR)
3593 gfc_constructor_base base;
3594 /* An unknown size constructor will always be rank one.
3595 Higher rank constructors will either have known shape,
3596 or still be wrapped in a call to reshape. */
3597 gcc_assert (loop->dimen == 1);
3599 /* Always prefer to use the constructor bounds if the size
3600 can be determined at compile time. Prefer not to otherwise,
3601 since the general case involves realloc, and it's better to
3602 avoid that overhead if possible. */
3603 base = ss->expr->value.constructor;
3604 dynamic[n] = gfc_get_array_constructor_size (&i, base);
3605 if (!dynamic[n] || !loopspec[n])
3606 loopspec[n] = ss;
3607 continue;
3610 /* TODO: Pick the best bound if we have a choice between a
3611 function and something else. */
3612 if (ss->type == GFC_SS_FUNCTION)
3614 loopspec[n] = ss;
3615 continue;
3618 if (ss->type != GFC_SS_SECTION)
3619 continue;
3621 if (loopspec[n])
3622 specinfo = &loopspec[n]->data.info;
3623 else
3624 specinfo = NULL;
3625 info = &ss->data.info;
3627 if (!specinfo)
3628 loopspec[n] = ss;
3629 /* Criteria for choosing a loop specifier (most important first):
3630 doesn't need realloc
3631 stride of one
3632 known stride
3633 known lower bound
3634 known upper bound
3636 else if (loopspec[n]->type == GFC_SS_CONSTRUCTOR && dynamic[n])
3637 loopspec[n] = ss;
3638 else if (integer_onep (info->stride[n])
3639 && !integer_onep (specinfo->stride[n]))
3640 loopspec[n] = ss;
3641 else if (INTEGER_CST_P (info->stride[n])
3642 && !INTEGER_CST_P (specinfo->stride[n]))
3643 loopspec[n] = ss;
3644 else if (INTEGER_CST_P (info->start[n])
3645 && !INTEGER_CST_P (specinfo->start[n]))
3646 loopspec[n] = ss;
3647 /* We don't work out the upper bound.
3648 else if (INTEGER_CST_P (info->finish[n])
3649 && ! INTEGER_CST_P (specinfo->finish[n]))
3650 loopspec[n] = ss; */
3653 /* We should have found the scalarization loop specifier. If not,
3654 that's bad news. */
3655 gcc_assert (loopspec[n]);
3657 info = &loopspec[n]->data.info;
3659 /* Set the extents of this range. */
3660 cshape = loopspec[n]->shape;
3661 if (cshape && INTEGER_CST_P (info->start[n])
3662 && INTEGER_CST_P (info->stride[n]))
3664 loop->from[n] = info->start[n];
3665 mpz_set (i, cshape[n]);
3666 mpz_sub_ui (i, i, 1);
3667 /* To = from + (size - 1) * stride. */
3668 tmp = gfc_conv_mpz_to_tree (i, gfc_index_integer_kind);
3669 if (!integer_onep (info->stride[n]))
3670 tmp = fold_build2 (MULT_EXPR, gfc_array_index_type,
3671 tmp, info->stride[n]);
3672 loop->to[n] = fold_build2 (PLUS_EXPR, gfc_array_index_type,
3673 loop->from[n], tmp);
3675 else
3677 loop->from[n] = info->start[n];
3678 switch (loopspec[n]->type)
3680 case GFC_SS_CONSTRUCTOR:
3681 /* The upper bound is calculated when we expand the
3682 constructor. */
3683 gcc_assert (loop->to[n] == NULL_TREE);
3684 break;
3686 case GFC_SS_SECTION:
3687 /* Use the end expression if it exists and is not constant,
3688 so that it is only evaluated once. */
3689 if (info->end[n] && !INTEGER_CST_P (info->end[n]))
3690 loop->to[n] = info->end[n];
3691 else
3692 loop->to[n] = gfc_conv_section_upper_bound (loopspec[n], n,
3693 &loop->pre);
3694 break;
3696 case GFC_SS_FUNCTION:
3697 /* The loop bound will be set when we generate the call. */
3698 gcc_assert (loop->to[n] == NULL_TREE);
3699 break;
3701 default:
3702 gcc_unreachable ();
3706 /* Transform everything so we have a simple incrementing variable. */
3707 if (integer_onep (info->stride[n]))
3708 info->delta[n] = gfc_index_zero_node;
3709 else
3711 /* Set the delta for this section. */
3712 info->delta[n] = gfc_evaluate_now (loop->from[n], &loop->pre);
3713 /* Number of iterations is (end - start + step) / step.
3714 with start = 0, this simplifies to
3715 last = end / step;
3716 for (i = 0; i<=last; i++){...}; */
3717 tmp = fold_build2 (MINUS_EXPR, gfc_array_index_type,
3718 loop->to[n], loop->from[n]);
3719 tmp = fold_build2 (FLOOR_DIV_EXPR, gfc_array_index_type,
3720 tmp, info->stride[n]);
3721 tmp = fold_build2 (MAX_EXPR, gfc_array_index_type, tmp,
3722 build_int_cst (gfc_array_index_type, -1));
3723 loop->to[n] = gfc_evaluate_now (tmp, &loop->pre);
3724 /* Make the loop variable start at 0. */
3725 loop->from[n] = gfc_index_zero_node;
3729 /* Add all the scalar code that can be taken out of the loops.
3730 This may include calculating the loop bounds, so do it before
3731 allocating the temporary. */
3732 gfc_add_loop_ss_code (loop, loop->ss, false, where);
3734 /* If we want a temporary then create it. */
3735 if (loop->temp_ss != NULL)
3737 gcc_assert (loop->temp_ss->type == GFC_SS_TEMP);
3739 /* Make absolutely sure that this is a complete type. */
3740 if (loop->temp_ss->string_length)
3741 loop->temp_ss->data.temp.type
3742 = gfc_get_character_type_len_for_eltype
3743 (TREE_TYPE (loop->temp_ss->data.temp.type),
3744 loop->temp_ss->string_length);
3746 tmp = loop->temp_ss->data.temp.type;
3747 n = loop->temp_ss->data.temp.dimen;
3748 memset (&loop->temp_ss->data.info, 0, sizeof (gfc_ss_info));
3749 loop->temp_ss->type = GFC_SS_SECTION;
3750 loop->temp_ss->data.info.dimen = n;
3751 gfc_trans_create_temp_array (&loop->pre, &loop->post, loop,
3752 &loop->temp_ss->data.info, tmp, NULL_TREE,
3753 false, true, false, where);
3756 for (n = 0; n < loop->temp_dim; n++)
3757 loopspec[loop->order[n]] = NULL;
3759 mpz_clear (i);
3761 /* For array parameters we don't have loop variables, so don't calculate the
3762 translations. */
3763 if (loop->array_parameter)
3764 return;
3766 /* Calculate the translation from loop variables to array indices. */
3767 for (ss = loop->ss; ss != gfc_ss_terminator; ss = ss->loop_chain)
3769 if (ss->type != GFC_SS_SECTION && ss->type != GFC_SS_COMPONENT
3770 && ss->type != GFC_SS_CONSTRUCTOR)
3772 continue;
3774 info = &ss->data.info;
3776 for (n = 0; n < info->dimen; n++)
3778 /* If we are specifying the range the delta is already set. */
3779 if (loopspec[n] != ss)
3781 /* Calculate the offset relative to the loop variable.
3782 First multiply by the stride. */
3783 tmp = loop->from[n];
3784 if (!integer_onep (info->stride[n]))
3785 tmp = fold_build2 (MULT_EXPR, gfc_array_index_type,
3786 tmp, info->stride[n]);
3788 /* Then subtract this from our starting value. */
3789 tmp = fold_build2 (MINUS_EXPR, gfc_array_index_type,
3790 info->start[n], tmp);
3792 info->delta[n] = gfc_evaluate_now (tmp, &loop->pre);
3799 /* Fills in an array descriptor, and returns the size of the array. The size
3800 will be a simple_val, ie a variable or a constant. Also calculates the
3801 offset of the base. Returns the size of the array.
3803 stride = 1;
3804 offset = 0;
3805 for (n = 0; n < rank; n++)
3807 a.lbound[n] = specified_lower_bound;
3808 offset = offset + a.lbond[n] * stride;
3809 size = 1 - lbound;
3810 a.ubound[n] = specified_upper_bound;
3811 a.stride[n] = stride;
3812 size = siz >= 0 ? ubound + size : 0; //size = ubound + 1 - lbound
3813 stride = stride * size;
3815 return (stride);
3816 } */
3817 /*GCC ARRAYS*/
3819 static tree
3820 gfc_array_init_size (tree descriptor, int rank, tree * poffset,
3821 gfc_expr ** lower, gfc_expr ** upper,
3822 stmtblock_t * pblock)
3824 tree type;
3825 tree tmp;
3826 tree size;
3827 tree offset;
3828 tree stride;
3829 tree cond;
3830 tree or_expr;
3831 tree thencase;
3832 tree elsecase;
3833 tree var;
3834 stmtblock_t thenblock;
3835 stmtblock_t elseblock;
3836 gfc_expr *ubound;
3837 gfc_se se;
3838 int n;
3840 type = TREE_TYPE (descriptor);
3842 stride = gfc_index_one_node;
3843 offset = gfc_index_zero_node;
3845 /* Set the dtype. */
3846 tmp = gfc_conv_descriptor_dtype (descriptor);
3847 gfc_add_modify (pblock, tmp, gfc_get_dtype (TREE_TYPE (descriptor)));
3849 or_expr = NULL_TREE;
3851 for (n = 0; n < rank; n++)
3853 /* We have 3 possibilities for determining the size of the array:
3854 lower == NULL => lbound = 1, ubound = upper[n]
3855 upper[n] = NULL => lbound = 1, ubound = lower[n]
3856 upper[n] != NULL => lbound = lower[n], ubound = upper[n] */
3857 ubound = upper[n];
3859 /* Set lower bound. */
3860 gfc_init_se (&se, NULL);
3861 if (lower == NULL)
3862 se.expr = gfc_index_one_node;
3863 else
3865 gcc_assert (lower[n]);
3866 if (ubound)
3868 gfc_conv_expr_type (&se, lower[n], gfc_array_index_type);
3869 gfc_add_block_to_block (pblock, &se.pre);
3871 else
3873 se.expr = gfc_index_one_node;
3874 ubound = lower[n];
3877 gfc_conv_descriptor_lbound_set (pblock, descriptor, gfc_rank_cst[n],
3878 se.expr);
3880 /* Work out the offset for this component. */
3881 tmp = fold_build2 (MULT_EXPR, gfc_array_index_type, se.expr, stride);
3882 offset = fold_build2 (MINUS_EXPR, gfc_array_index_type, offset, tmp);
3884 /* Start the calculation for the size of this dimension. */
3885 size = fold_build2 (MINUS_EXPR, gfc_array_index_type,
3886 gfc_index_one_node, se.expr);
3888 /* Set upper bound. */
3889 gfc_init_se (&se, NULL);
3890 gcc_assert (ubound);
3891 gfc_conv_expr_type (&se, ubound, gfc_array_index_type);
3892 gfc_add_block_to_block (pblock, &se.pre);
3894 gfc_conv_descriptor_ubound_set (pblock, descriptor, gfc_rank_cst[n], se.expr);
3896 /* Store the stride. */
3897 gfc_conv_descriptor_stride_set (pblock, descriptor, gfc_rank_cst[n], stride);
3899 /* Calculate the size of this dimension. */
3900 size = fold_build2 (PLUS_EXPR, gfc_array_index_type, se.expr, size);
3902 /* Check whether the size for this dimension is negative. */
3903 cond = fold_build2 (LE_EXPR, boolean_type_node, size,
3904 gfc_index_zero_node);
3905 if (n == 0)
3906 or_expr = cond;
3907 else
3908 or_expr = fold_build2 (TRUTH_OR_EXPR, boolean_type_node, or_expr, cond);
3910 size = fold_build3 (COND_EXPR, gfc_array_index_type, cond,
3911 gfc_index_zero_node, size);
3913 /* Multiply the stride by the number of elements in this dimension. */
3914 stride = fold_build2 (MULT_EXPR, gfc_array_index_type, stride, size);
3915 stride = gfc_evaluate_now (stride, pblock);
3918 /* The stride is the number of elements in the array, so multiply by the
3919 size of an element to get the total size. */
3920 tmp = TYPE_SIZE_UNIT (gfc_get_element_type (type));
3921 size = fold_build2 (MULT_EXPR, gfc_array_index_type, stride,
3922 fold_convert (gfc_array_index_type, tmp));
3924 if (poffset != NULL)
3926 offset = gfc_evaluate_now (offset, pblock);
3927 *poffset = offset;
3930 if (integer_zerop (or_expr))
3931 return size;
3932 if (integer_onep (or_expr))
3933 return gfc_index_zero_node;
3935 var = gfc_create_var (TREE_TYPE (size), "size");
3936 gfc_start_block (&thenblock);
3937 gfc_add_modify (&thenblock, var, gfc_index_zero_node);
3938 thencase = gfc_finish_block (&thenblock);
3940 gfc_start_block (&elseblock);
3941 gfc_add_modify (&elseblock, var, size);
3942 elsecase = gfc_finish_block (&elseblock);
3944 tmp = gfc_evaluate_now (or_expr, pblock);
3945 tmp = build3_v (COND_EXPR, tmp, thencase, elsecase);
3946 gfc_add_expr_to_block (pblock, tmp);
3948 return var;
3952 /* Initializes the descriptor and generates a call to _gfor_allocate. Does
3953 the work for an ALLOCATE statement. */
3954 /*GCC ARRAYS*/
3956 bool
3957 gfc_array_allocate (gfc_se * se, gfc_expr * expr, tree pstat)
3959 tree tmp;
3960 tree pointer;
3961 tree offset;
3962 tree size;
3963 gfc_expr **lower;
3964 gfc_expr **upper;
3965 gfc_ref *ref, *prev_ref = NULL;
3966 bool allocatable_array;
3968 ref = expr->ref;
3970 /* Find the last reference in the chain. */
3971 while (ref && ref->next != NULL)
3973 gcc_assert (ref->type != REF_ARRAY || ref->u.ar.type == AR_ELEMENT
3974 || (ref->u.ar.dimen == 0 && ref->u.ar.codimen > 0));
3975 prev_ref = ref;
3976 ref = ref->next;
3979 if (ref == NULL || ref->type != REF_ARRAY)
3980 return false;
3982 /* Return if this is a scalar coarray. */
3983 if (!prev_ref && !expr->symtree->n.sym->attr.dimension)
3985 gcc_assert (expr->symtree->n.sym->attr.codimension);
3986 return false;
3988 else if (prev_ref && !prev_ref->u.c.component->attr.dimension)
3990 gcc_assert (prev_ref->u.c.component->attr.codimension);
3991 return false;
3994 if (!prev_ref)
3995 allocatable_array = expr->symtree->n.sym->attr.allocatable;
3996 else
3997 allocatable_array = prev_ref->u.c.component->attr.allocatable;
3999 /* Figure out the size of the array. */
4000 switch (ref->u.ar.type)
4002 case AR_ELEMENT:
4003 lower = NULL;
4004 upper = ref->u.ar.start;
4005 break;
4007 case AR_FULL:
4008 gcc_assert (ref->u.ar.as->type == AS_EXPLICIT);
4010 lower = ref->u.ar.as->lower;
4011 upper = ref->u.ar.as->upper;
4012 break;
4014 case AR_SECTION:
4015 lower = ref->u.ar.start;
4016 upper = ref->u.ar.end;
4017 break;
4019 default:
4020 gcc_unreachable ();
4021 break;
4024 size = gfc_array_init_size (se->expr, ref->u.ar.as->rank, &offset,
4025 lower, upper, &se->pre);
4027 /* Allocate memory to store the data. */
4028 pointer = gfc_conv_descriptor_data_get (se->expr);
4029 STRIP_NOPS (pointer);
4031 /* The allocate_array variants take the old pointer as first argument. */
4032 if (allocatable_array)
4033 tmp = gfc_allocate_array_with_status (&se->pre, pointer, size, pstat, expr);
4034 else
4035 tmp = gfc_allocate_with_status (&se->pre, size, pstat);
4036 tmp = fold_build2 (MODIFY_EXPR, void_type_node, pointer, tmp);
4037 gfc_add_expr_to_block (&se->pre, tmp);
4039 gfc_conv_descriptor_offset_set (&se->pre, se->expr, offset);
4041 if (expr->ts.type == BT_DERIVED
4042 && expr->ts.u.derived->attr.alloc_comp)
4044 tmp = gfc_nullify_alloc_comp (expr->ts.u.derived, se->expr,
4045 ref->u.ar.as->rank);
4046 gfc_add_expr_to_block (&se->pre, tmp);
4049 return true;
4053 /* Deallocate an array variable. Also used when an allocated variable goes
4054 out of scope. */
4055 /*GCC ARRAYS*/
4057 tree
4058 gfc_array_deallocate (tree descriptor, tree pstat, gfc_expr* expr)
4060 tree var;
4061 tree tmp;
4062 stmtblock_t block;
4064 gfc_start_block (&block);
4065 /* Get a pointer to the data. */
4066 var = gfc_conv_descriptor_data_get (descriptor);
4067 STRIP_NOPS (var);
4069 /* Parameter is the address of the data component. */
4070 tmp = gfc_deallocate_with_status (var, pstat, false, expr);
4071 gfc_add_expr_to_block (&block, tmp);
4073 /* Zero the data pointer. */
4074 tmp = fold_build2 (MODIFY_EXPR, void_type_node,
4075 var, build_int_cst (TREE_TYPE (var), 0));
4076 gfc_add_expr_to_block (&block, tmp);
4078 return gfc_finish_block (&block);
4082 /* Create an array constructor from an initialization expression.
4083 We assume the frontend already did any expansions and conversions. */
4085 tree
4086 gfc_conv_array_initializer (tree type, gfc_expr * expr)
4088 gfc_constructor *c;
4089 tree tmp;
4090 mpz_t maxval;
4091 gfc_se se;
4092 HOST_WIDE_INT hi;
4093 unsigned HOST_WIDE_INT lo;
4094 tree index, range;
4095 VEC(constructor_elt,gc) *v = NULL;
4097 switch (expr->expr_type)
4099 case EXPR_CONSTANT:
4100 case EXPR_STRUCTURE:
4101 /* A single scalar or derived type value. Create an array with all
4102 elements equal to that value. */
4103 gfc_init_se (&se, NULL);
4105 if (expr->expr_type == EXPR_CONSTANT)
4106 gfc_conv_constant (&se, expr);
4107 else
4108 gfc_conv_structure (&se, expr, 1);
4110 tmp = TYPE_MAX_VALUE (TYPE_DOMAIN (type));
4111 gcc_assert (tmp && INTEGER_CST_P (tmp));
4112 hi = TREE_INT_CST_HIGH (tmp);
4113 lo = TREE_INT_CST_LOW (tmp);
4114 lo++;
4115 if (lo == 0)
4116 hi++;
4117 /* This will probably eat buckets of memory for large arrays. */
4118 while (hi != 0 || lo != 0)
4120 CONSTRUCTOR_APPEND_ELT (v, NULL_TREE, se.expr);
4121 if (lo == 0)
4122 hi--;
4123 lo--;
4125 break;
4127 case EXPR_ARRAY:
4128 /* Create a vector of all the elements. */
4129 for (c = gfc_constructor_first (expr->value.constructor);
4130 c; c = gfc_constructor_next (c))
4132 if (c->iterator)
4134 /* Problems occur when we get something like
4135 integer :: a(lots) = (/(i, i=1, lots)/) */
4136 gfc_fatal_error ("The number of elements in the array constructor "
4137 "at %L requires an increase of the allowed %d "
4138 "upper limit. See -fmax-array-constructor "
4139 "option", &expr->where,
4140 gfc_option.flag_max_array_constructor);
4141 return NULL_TREE;
4143 if (mpz_cmp_si (c->offset, 0) != 0)
4144 index = gfc_conv_mpz_to_tree (c->offset, gfc_index_integer_kind);
4145 else
4146 index = NULL_TREE;
4147 mpz_init (maxval);
4148 if (mpz_cmp_si (c->repeat, 0) != 0)
4150 tree tmp1, tmp2;
4152 mpz_set (maxval, c->repeat);
4153 mpz_add (maxval, c->offset, maxval);
4154 mpz_sub_ui (maxval, maxval, 1);
4155 tmp2 = gfc_conv_mpz_to_tree (maxval, gfc_index_integer_kind);
4156 if (mpz_cmp_si (c->offset, 0) != 0)
4158 mpz_add_ui (maxval, c->offset, 1);
4159 tmp1 = gfc_conv_mpz_to_tree (maxval, gfc_index_integer_kind);
4161 else
4162 tmp1 = gfc_conv_mpz_to_tree (c->offset, gfc_index_integer_kind);
4164 range = fold_build2 (RANGE_EXPR, integer_type_node, tmp1, tmp2);
4166 else
4167 range = NULL;
4168 mpz_clear (maxval);
4170 gfc_init_se (&se, NULL);
4171 switch (c->expr->expr_type)
4173 case EXPR_CONSTANT:
4174 gfc_conv_constant (&se, c->expr);
4175 if (range == NULL_TREE)
4176 CONSTRUCTOR_APPEND_ELT (v, index, se.expr);
4177 else
4179 if (index != NULL_TREE)
4180 CONSTRUCTOR_APPEND_ELT (v, index, se.expr);
4181 CONSTRUCTOR_APPEND_ELT (v, range, se.expr);
4183 break;
4185 case EXPR_STRUCTURE:
4186 gfc_conv_structure (&se, c->expr, 1);
4187 CONSTRUCTOR_APPEND_ELT (v, index, se.expr);
4188 break;
4191 default:
4192 /* Catch those occasional beasts that do not simplify
4193 for one reason or another, assuming that if they are
4194 standard defying the frontend will catch them. */
4195 gfc_conv_expr (&se, c->expr);
4196 if (range == NULL_TREE)
4197 CONSTRUCTOR_APPEND_ELT (v, index, se.expr);
4198 else
4200 if (index != NULL_TREE)
4201 CONSTRUCTOR_APPEND_ELT (v, index, se.expr);
4202 CONSTRUCTOR_APPEND_ELT (v, range, se.expr);
4204 break;
4207 break;
4209 case EXPR_NULL:
4210 return gfc_build_null_descriptor (type);
4212 default:
4213 gcc_unreachable ();
4216 /* Create a constructor from the list of elements. */
4217 tmp = build_constructor (type, v);
4218 TREE_CONSTANT (tmp) = 1;
4219 return tmp;
4223 /* Generate code to evaluate non-constant array bounds. Sets *poffset and
4224 returns the size (in elements) of the array. */
4226 static tree
4227 gfc_trans_array_bounds (tree type, gfc_symbol * sym, tree * poffset,
4228 stmtblock_t * pblock)
4230 gfc_array_spec *as;
4231 tree size;
4232 tree stride;
4233 tree offset;
4234 tree ubound;
4235 tree lbound;
4236 tree tmp;
4237 gfc_se se;
4239 int dim;
4241 as = sym->as;
4243 size = gfc_index_one_node;
4244 offset = gfc_index_zero_node;
4245 for (dim = 0; dim < as->rank; dim++)
4247 /* Evaluate non-constant array bound expressions. */
4248 lbound = GFC_TYPE_ARRAY_LBOUND (type, dim);
4249 if (as->lower[dim] && !INTEGER_CST_P (lbound))
4251 gfc_init_se (&se, NULL);
4252 gfc_conv_expr_type (&se, as->lower[dim], gfc_array_index_type);
4253 gfc_add_block_to_block (pblock, &se.pre);
4254 gfc_add_modify (pblock, lbound, se.expr);
4256 ubound = GFC_TYPE_ARRAY_UBOUND (type, dim);
4257 if (as->upper[dim] && !INTEGER_CST_P (ubound))
4259 gfc_init_se (&se, NULL);
4260 gfc_conv_expr_type (&se, as->upper[dim], gfc_array_index_type);
4261 gfc_add_block_to_block (pblock, &se.pre);
4262 gfc_add_modify (pblock, ubound, se.expr);
4264 /* The offset of this dimension. offset = offset - lbound * stride. */
4265 tmp = fold_build2 (MULT_EXPR, gfc_array_index_type, lbound, size);
4266 offset = fold_build2 (MINUS_EXPR, gfc_array_index_type, offset, tmp);
4268 /* The size of this dimension, and the stride of the next. */
4269 if (dim + 1 < as->rank)
4270 stride = GFC_TYPE_ARRAY_STRIDE (type, dim + 1);
4271 else
4272 stride = GFC_TYPE_ARRAY_SIZE (type);
4274 if (ubound != NULL_TREE && !(stride && INTEGER_CST_P (stride)))
4276 /* Calculate stride = size * (ubound + 1 - lbound). */
4277 tmp = fold_build2 (MINUS_EXPR, gfc_array_index_type,
4278 gfc_index_one_node, lbound);
4279 tmp = fold_build2 (PLUS_EXPR, gfc_array_index_type, ubound, tmp);
4280 tmp = fold_build2 (MULT_EXPR, gfc_array_index_type, size, tmp);
4281 if (stride)
4282 gfc_add_modify (pblock, stride, tmp);
4283 else
4284 stride = gfc_evaluate_now (tmp, pblock);
4286 /* Make sure that negative size arrays are translated
4287 to being zero size. */
4288 tmp = fold_build2 (GE_EXPR, boolean_type_node,
4289 stride, gfc_index_zero_node);
4290 tmp = fold_build3 (COND_EXPR, gfc_array_index_type, tmp,
4291 stride, gfc_index_zero_node);
4292 gfc_add_modify (pblock, stride, tmp);
4295 size = stride;
4298 gfc_trans_vla_type_sizes (sym, pblock);
4300 *poffset = offset;
4301 return size;
4305 /* Generate code to initialize/allocate an array variable. */
4307 tree
4308 gfc_trans_auto_array_allocation (tree decl, gfc_symbol * sym, tree fnbody)
4310 stmtblock_t block;
4311 tree type;
4312 tree tmp;
4313 tree size;
4314 tree offset;
4315 bool onstack;
4317 gcc_assert (!(sym->attr.pointer || sym->attr.allocatable));
4319 /* Do nothing for USEd variables. */
4320 if (sym->attr.use_assoc)
4321 return fnbody;
4323 type = TREE_TYPE (decl);
4324 gcc_assert (GFC_ARRAY_TYPE_P (type));
4325 onstack = TREE_CODE (type) != POINTER_TYPE;
4327 gfc_start_block (&block);
4329 /* Evaluate character string length. */
4330 if (sym->ts.type == BT_CHARACTER
4331 && onstack && !INTEGER_CST_P (sym->ts.u.cl->backend_decl))
4333 gfc_conv_string_length (sym->ts.u.cl, NULL, &block);
4335 gfc_trans_vla_type_sizes (sym, &block);
4337 /* Emit a DECL_EXPR for this variable, which will cause the
4338 gimplifier to allocate storage, and all that good stuff. */
4339 tmp = fold_build1 (DECL_EXPR, TREE_TYPE (decl), decl);
4340 gfc_add_expr_to_block (&block, tmp);
4343 if (onstack)
4345 gfc_add_expr_to_block (&block, fnbody);
4346 return gfc_finish_block (&block);
4349 type = TREE_TYPE (type);
4351 gcc_assert (!sym->attr.use_assoc);
4352 gcc_assert (!TREE_STATIC (decl));
4353 gcc_assert (!sym->module);
4355 if (sym->ts.type == BT_CHARACTER
4356 && !INTEGER_CST_P (sym->ts.u.cl->backend_decl))
4357 gfc_conv_string_length (sym->ts.u.cl, NULL, &block);
4359 size = gfc_trans_array_bounds (type, sym, &offset, &block);
4361 /* Don't actually allocate space for Cray Pointees. */
4362 if (sym->attr.cray_pointee)
4364 if (TREE_CODE (GFC_TYPE_ARRAY_OFFSET (type)) == VAR_DECL)
4365 gfc_add_modify (&block, GFC_TYPE_ARRAY_OFFSET (type), offset);
4366 gfc_add_expr_to_block (&block, fnbody);
4367 return gfc_finish_block (&block);
4370 /* The size is the number of elements in the array, so multiply by the
4371 size of an element to get the total size. */
4372 tmp = TYPE_SIZE_UNIT (gfc_get_element_type (type));
4373 size = fold_build2 (MULT_EXPR, gfc_array_index_type, size,
4374 fold_convert (gfc_array_index_type, tmp));
4376 /* Allocate memory to hold the data. */
4377 tmp = gfc_call_malloc (&block, TREE_TYPE (decl), size);
4378 gfc_add_modify (&block, decl, tmp);
4380 /* Set offset of the array. */
4381 if (TREE_CODE (GFC_TYPE_ARRAY_OFFSET (type)) == VAR_DECL)
4382 gfc_add_modify (&block, GFC_TYPE_ARRAY_OFFSET (type), offset);
4385 /* Automatic arrays should not have initializers. */
4386 gcc_assert (!sym->value);
4388 gfc_add_expr_to_block (&block, fnbody);
4390 /* Free the temporary. */
4391 tmp = gfc_call_free (convert (pvoid_type_node, decl));
4392 gfc_add_expr_to_block (&block, tmp);
4394 return gfc_finish_block (&block);
4398 /* Generate entry and exit code for g77 calling convention arrays. */
4400 tree
4401 gfc_trans_g77_array (gfc_symbol * sym, tree body)
4403 tree parm;
4404 tree type;
4405 locus loc;
4406 tree offset;
4407 tree tmp;
4408 tree stmt;
4409 stmtblock_t block;
4411 gfc_get_backend_locus (&loc);
4412 gfc_set_backend_locus (&sym->declared_at);
4414 /* Descriptor type. */
4415 parm = sym->backend_decl;
4416 type = TREE_TYPE (parm);
4417 gcc_assert (GFC_ARRAY_TYPE_P (type));
4419 gfc_start_block (&block);
4421 if (sym->ts.type == BT_CHARACTER
4422 && TREE_CODE (sym->ts.u.cl->backend_decl) == VAR_DECL)
4423 gfc_conv_string_length (sym->ts.u.cl, NULL, &block);
4425 /* Evaluate the bounds of the array. */
4426 gfc_trans_array_bounds (type, sym, &offset, &block);
4428 /* Set the offset. */
4429 if (TREE_CODE (GFC_TYPE_ARRAY_OFFSET (type)) == VAR_DECL)
4430 gfc_add_modify (&block, GFC_TYPE_ARRAY_OFFSET (type), offset);
4432 /* Set the pointer itself if we aren't using the parameter directly. */
4433 if (TREE_CODE (parm) != PARM_DECL)
4435 tmp = convert (TREE_TYPE (parm), GFC_DECL_SAVED_DESCRIPTOR (parm));
4436 gfc_add_modify (&block, parm, tmp);
4438 stmt = gfc_finish_block (&block);
4440 gfc_set_backend_locus (&loc);
4442 gfc_start_block (&block);
4444 /* Add the initialization code to the start of the function. */
4446 if (sym->attr.optional || sym->attr.not_always_present)
4448 tmp = gfc_conv_expr_present (sym);
4449 stmt = build3_v (COND_EXPR, tmp, stmt, build_empty_stmt (input_location));
4452 gfc_add_expr_to_block (&block, stmt);
4453 gfc_add_expr_to_block (&block, body);
4455 return gfc_finish_block (&block);
4459 /* Modify the descriptor of an array parameter so that it has the
4460 correct lower bound. Also move the upper bound accordingly.
4461 If the array is not packed, it will be copied into a temporary.
4462 For each dimension we set the new lower and upper bounds. Then we copy the
4463 stride and calculate the offset for this dimension. We also work out
4464 what the stride of a packed array would be, and see it the two match.
4465 If the array need repacking, we set the stride to the values we just
4466 calculated, recalculate the offset and copy the array data.
4467 Code is also added to copy the data back at the end of the function.
4470 tree
4471 gfc_trans_dummy_array_bias (gfc_symbol * sym, tree tmpdesc, tree body)
4473 tree size;
4474 tree type;
4475 tree offset;
4476 locus loc;
4477 stmtblock_t block;
4478 stmtblock_t cleanup;
4479 tree lbound;
4480 tree ubound;
4481 tree dubound;
4482 tree dlbound;
4483 tree dumdesc;
4484 tree tmp;
4485 tree stmt;
4486 tree stride, stride2;
4487 tree stmt_packed;
4488 tree stmt_unpacked;
4489 tree partial;
4490 gfc_se se;
4491 int n;
4492 int checkparm;
4493 int no_repack;
4494 bool optional_arg;
4496 /* Do nothing for pointer and allocatable arrays. */
4497 if (sym->attr.pointer || sym->attr.allocatable)
4498 return body;
4500 if (sym->attr.dummy && gfc_is_nodesc_array (sym))
4501 return gfc_trans_g77_array (sym, body);
4503 gfc_get_backend_locus (&loc);
4504 gfc_set_backend_locus (&sym->declared_at);
4506 /* Descriptor type. */
4507 type = TREE_TYPE (tmpdesc);
4508 gcc_assert (GFC_ARRAY_TYPE_P (type));
4509 dumdesc = GFC_DECL_SAVED_DESCRIPTOR (tmpdesc);
4510 dumdesc = build_fold_indirect_ref_loc (input_location,
4511 dumdesc);
4512 gfc_start_block (&block);
4514 if (sym->ts.type == BT_CHARACTER
4515 && TREE_CODE (sym->ts.u.cl->backend_decl) == VAR_DECL)
4516 gfc_conv_string_length (sym->ts.u.cl, NULL, &block);
4518 checkparm = (sym->as->type == AS_EXPLICIT
4519 && (gfc_option.rtcheck & GFC_RTCHECK_BOUNDS));
4521 no_repack = !(GFC_DECL_PACKED_ARRAY (tmpdesc)
4522 || GFC_DECL_PARTIAL_PACKED_ARRAY (tmpdesc));
4524 if (GFC_DECL_PARTIAL_PACKED_ARRAY (tmpdesc))
4526 /* For non-constant shape arrays we only check if the first dimension
4527 is contiguous. Repacking higher dimensions wouldn't gain us
4528 anything as we still don't know the array stride. */
4529 partial = gfc_create_var (boolean_type_node, "partial");
4530 TREE_USED (partial) = 1;
4531 tmp = gfc_conv_descriptor_stride_get (dumdesc, gfc_rank_cst[0]);
4532 tmp = fold_build2 (EQ_EXPR, boolean_type_node, tmp, gfc_index_one_node);
4533 gfc_add_modify (&block, partial, tmp);
4535 else
4537 partial = NULL_TREE;
4540 /* The naming of stmt_unpacked and stmt_packed may be counter-intuitive
4541 here, however I think it does the right thing. */
4542 if (no_repack)
4544 /* Set the first stride. */
4545 stride = gfc_conv_descriptor_stride_get (dumdesc, gfc_rank_cst[0]);
4546 stride = gfc_evaluate_now (stride, &block);
4548 tmp = fold_build2 (EQ_EXPR, boolean_type_node,
4549 stride, gfc_index_zero_node);
4550 tmp = fold_build3 (COND_EXPR, gfc_array_index_type, tmp,
4551 gfc_index_one_node, stride);
4552 stride = GFC_TYPE_ARRAY_STRIDE (type, 0);
4553 gfc_add_modify (&block, stride, tmp);
4555 /* Allow the user to disable array repacking. */
4556 stmt_unpacked = NULL_TREE;
4558 else
4560 gcc_assert (integer_onep (GFC_TYPE_ARRAY_STRIDE (type, 0)));
4561 /* A library call to repack the array if necessary. */
4562 tmp = GFC_DECL_SAVED_DESCRIPTOR (tmpdesc);
4563 stmt_unpacked = build_call_expr_loc (input_location,
4564 gfor_fndecl_in_pack, 1, tmp);
4566 stride = gfc_index_one_node;
4568 if (gfc_option.warn_array_temp)
4569 gfc_warning ("Creating array temporary at %L", &loc);
4572 /* This is for the case where the array data is used directly without
4573 calling the repack function. */
4574 if (no_repack || partial != NULL_TREE)
4575 stmt_packed = gfc_conv_descriptor_data_get (dumdesc);
4576 else
4577 stmt_packed = NULL_TREE;
4579 /* Assign the data pointer. */
4580 if (stmt_packed != NULL_TREE && stmt_unpacked != NULL_TREE)
4582 /* Don't repack unknown shape arrays when the first stride is 1. */
4583 tmp = fold_build3 (COND_EXPR, TREE_TYPE (stmt_packed),
4584 partial, stmt_packed, stmt_unpacked);
4586 else
4587 tmp = stmt_packed != NULL_TREE ? stmt_packed : stmt_unpacked;
4588 gfc_add_modify (&block, tmpdesc, fold_convert (type, tmp));
4590 offset = gfc_index_zero_node;
4591 size = gfc_index_one_node;
4593 /* Evaluate the bounds of the array. */
4594 for (n = 0; n < sym->as->rank; n++)
4596 if (checkparm || !sym->as->upper[n])
4598 /* Get the bounds of the actual parameter. */
4599 dubound = gfc_conv_descriptor_ubound_get (dumdesc, gfc_rank_cst[n]);
4600 dlbound = gfc_conv_descriptor_lbound_get (dumdesc, gfc_rank_cst[n]);
4602 else
4604 dubound = NULL_TREE;
4605 dlbound = NULL_TREE;
4608 lbound = GFC_TYPE_ARRAY_LBOUND (type, n);
4609 if (!INTEGER_CST_P (lbound))
4611 gfc_init_se (&se, NULL);
4612 gfc_conv_expr_type (&se, sym->as->lower[n],
4613 gfc_array_index_type);
4614 gfc_add_block_to_block (&block, &se.pre);
4615 gfc_add_modify (&block, lbound, se.expr);
4618 ubound = GFC_TYPE_ARRAY_UBOUND (type, n);
4619 /* Set the desired upper bound. */
4620 if (sym->as->upper[n])
4622 /* We know what we want the upper bound to be. */
4623 if (!INTEGER_CST_P (ubound))
4625 gfc_init_se (&se, NULL);
4626 gfc_conv_expr_type (&se, sym->as->upper[n],
4627 gfc_array_index_type);
4628 gfc_add_block_to_block (&block, &se.pre);
4629 gfc_add_modify (&block, ubound, se.expr);
4632 /* Check the sizes match. */
4633 if (checkparm)
4635 /* Check (ubound(a) - lbound(a) == ubound(b) - lbound(b)). */
4636 char * msg;
4637 tree temp;
4639 temp = fold_build2 (MINUS_EXPR, gfc_array_index_type,
4640 ubound, lbound);
4641 temp = fold_build2 (PLUS_EXPR, gfc_array_index_type,
4642 gfc_index_one_node, temp);
4644 stride2 = fold_build2 (MINUS_EXPR, gfc_array_index_type,
4645 dubound, dlbound);
4646 stride2 = fold_build2 (PLUS_EXPR, gfc_array_index_type,
4647 gfc_index_one_node, stride2);
4649 tmp = fold_build2 (NE_EXPR, gfc_array_index_type, temp, stride2);
4650 asprintf (&msg, "Dimension %d of array '%s' has extent "
4651 "%%ld instead of %%ld", n+1, sym->name);
4653 gfc_trans_runtime_check (true, false, tmp, &block, &loc, msg,
4654 fold_convert (long_integer_type_node, temp),
4655 fold_convert (long_integer_type_node, stride2));
4657 gfc_free (msg);
4660 else
4662 /* For assumed shape arrays move the upper bound by the same amount
4663 as the lower bound. */
4664 tmp = fold_build2 (MINUS_EXPR, gfc_array_index_type,
4665 dubound, dlbound);
4666 tmp = fold_build2 (PLUS_EXPR, gfc_array_index_type, tmp, lbound);
4667 gfc_add_modify (&block, ubound, tmp);
4669 /* The offset of this dimension. offset = offset - lbound * stride. */
4670 tmp = fold_build2 (MULT_EXPR, gfc_array_index_type, lbound, stride);
4671 offset = fold_build2 (MINUS_EXPR, gfc_array_index_type, offset, tmp);
4673 /* The size of this dimension, and the stride of the next. */
4674 if (n + 1 < sym->as->rank)
4676 stride = GFC_TYPE_ARRAY_STRIDE (type, n + 1);
4678 if (no_repack || partial != NULL_TREE)
4680 stmt_unpacked =
4681 gfc_conv_descriptor_stride_get (dumdesc, gfc_rank_cst[n+1]);
4684 /* Figure out the stride if not a known constant. */
4685 if (!INTEGER_CST_P (stride))
4687 if (no_repack)
4688 stmt_packed = NULL_TREE;
4689 else
4691 /* Calculate stride = size * (ubound + 1 - lbound). */
4692 tmp = fold_build2 (MINUS_EXPR, gfc_array_index_type,
4693 gfc_index_one_node, lbound);
4694 tmp = fold_build2 (PLUS_EXPR, gfc_array_index_type,
4695 ubound, tmp);
4696 size = fold_build2 (MULT_EXPR, gfc_array_index_type,
4697 size, tmp);
4698 stmt_packed = size;
4701 /* Assign the stride. */
4702 if (stmt_packed != NULL_TREE && stmt_unpacked != NULL_TREE)
4703 tmp = fold_build3 (COND_EXPR, gfc_array_index_type, partial,
4704 stmt_unpacked, stmt_packed);
4705 else
4706 tmp = (stmt_packed != NULL_TREE) ? stmt_packed : stmt_unpacked;
4707 gfc_add_modify (&block, stride, tmp);
4710 else
4712 stride = GFC_TYPE_ARRAY_SIZE (type);
4714 if (stride && !INTEGER_CST_P (stride))
4716 /* Calculate size = stride * (ubound + 1 - lbound). */
4717 tmp = fold_build2 (MINUS_EXPR, gfc_array_index_type,
4718 gfc_index_one_node, lbound);
4719 tmp = fold_build2 (PLUS_EXPR, gfc_array_index_type,
4720 ubound, tmp);
4721 tmp = fold_build2 (MULT_EXPR, gfc_array_index_type,
4722 GFC_TYPE_ARRAY_STRIDE (type, n), tmp);
4723 gfc_add_modify (&block, stride, tmp);
4728 /* Set the offset. */
4729 if (TREE_CODE (GFC_TYPE_ARRAY_OFFSET (type)) == VAR_DECL)
4730 gfc_add_modify (&block, GFC_TYPE_ARRAY_OFFSET (type), offset);
4732 gfc_trans_vla_type_sizes (sym, &block);
4734 stmt = gfc_finish_block (&block);
4736 gfc_start_block (&block);
4738 /* Only do the entry/initialization code if the arg is present. */
4739 dumdesc = GFC_DECL_SAVED_DESCRIPTOR (tmpdesc);
4740 optional_arg = (sym->attr.optional
4741 || (sym->ns->proc_name->attr.entry_master
4742 && sym->attr.dummy));
4743 if (optional_arg)
4745 tmp = gfc_conv_expr_present (sym);
4746 stmt = build3_v (COND_EXPR, tmp, stmt, build_empty_stmt (input_location));
4748 gfc_add_expr_to_block (&block, stmt);
4750 /* Add the main function body. */
4751 gfc_add_expr_to_block (&block, body);
4753 /* Cleanup code. */
4754 if (!no_repack)
4756 gfc_start_block (&cleanup);
4758 if (sym->attr.intent != INTENT_IN)
4760 /* Copy the data back. */
4761 tmp = build_call_expr_loc (input_location,
4762 gfor_fndecl_in_unpack, 2, dumdesc, tmpdesc);
4763 gfc_add_expr_to_block (&cleanup, tmp);
4766 /* Free the temporary. */
4767 tmp = gfc_call_free (tmpdesc);
4768 gfc_add_expr_to_block (&cleanup, tmp);
4770 stmt = gfc_finish_block (&cleanup);
4772 /* Only do the cleanup if the array was repacked. */
4773 tmp = build_fold_indirect_ref_loc (input_location,
4774 dumdesc);
4775 tmp = gfc_conv_descriptor_data_get (tmp);
4776 tmp = fold_build2 (NE_EXPR, boolean_type_node, tmp, tmpdesc);
4777 stmt = build3_v (COND_EXPR, tmp, stmt, build_empty_stmt (input_location));
4779 if (optional_arg)
4781 tmp = gfc_conv_expr_present (sym);
4782 stmt = build3_v (COND_EXPR, tmp, stmt,
4783 build_empty_stmt (input_location));
4785 gfc_add_expr_to_block (&block, stmt);
4787 /* We don't need to free any memory allocated by internal_pack as it will
4788 be freed at the end of the function by pop_context. */
4789 return gfc_finish_block (&block);
4793 /* Calculate the overall offset, including subreferences. */
4794 static void
4795 gfc_get_dataptr_offset (stmtblock_t *block, tree parm, tree desc, tree offset,
4796 bool subref, gfc_expr *expr)
4798 tree tmp;
4799 tree field;
4800 tree stride;
4801 tree index;
4802 gfc_ref *ref;
4803 gfc_se start;
4804 int n;
4806 /* If offset is NULL and this is not a subreferenced array, there is
4807 nothing to do. */
4808 if (offset == NULL_TREE)
4810 if (subref)
4811 offset = gfc_index_zero_node;
4812 else
4813 return;
4816 tmp = gfc_conv_array_data (desc);
4817 tmp = build_fold_indirect_ref_loc (input_location,
4818 tmp);
4819 tmp = gfc_build_array_ref (tmp, offset, NULL);
4821 /* Offset the data pointer for pointer assignments from arrays with
4822 subreferences; e.g. my_integer => my_type(:)%integer_component. */
4823 if (subref)
4825 /* Go past the array reference. */
4826 for (ref = expr->ref; ref; ref = ref->next)
4827 if (ref->type == REF_ARRAY &&
4828 ref->u.ar.type != AR_ELEMENT)
4830 ref = ref->next;
4831 break;
4834 /* Calculate the offset for each subsequent subreference. */
4835 for (; ref; ref = ref->next)
4837 switch (ref->type)
4839 case REF_COMPONENT:
4840 field = ref->u.c.component->backend_decl;
4841 gcc_assert (field && TREE_CODE (field) == FIELD_DECL);
4842 tmp = fold_build3 (COMPONENT_REF, TREE_TYPE (field),
4843 tmp, field, NULL_TREE);
4844 break;
4846 case REF_SUBSTRING:
4847 gcc_assert (TREE_CODE (TREE_TYPE (tmp)) == ARRAY_TYPE);
4848 gfc_init_se (&start, NULL);
4849 gfc_conv_expr_type (&start, ref->u.ss.start, gfc_charlen_type_node);
4850 gfc_add_block_to_block (block, &start.pre);
4851 tmp = gfc_build_array_ref (tmp, start.expr, NULL);
4852 break;
4854 case REF_ARRAY:
4855 gcc_assert (TREE_CODE (TREE_TYPE (tmp)) == ARRAY_TYPE
4856 && ref->u.ar.type == AR_ELEMENT);
4858 /* TODO - Add bounds checking. */
4859 stride = gfc_index_one_node;
4860 index = gfc_index_zero_node;
4861 for (n = 0; n < ref->u.ar.dimen; n++)
4863 tree itmp;
4864 tree jtmp;
4866 /* Update the index. */
4867 gfc_init_se (&start, NULL);
4868 gfc_conv_expr_type (&start, ref->u.ar.start[n], gfc_array_index_type);
4869 itmp = gfc_evaluate_now (start.expr, block);
4870 gfc_init_se (&start, NULL);
4871 gfc_conv_expr_type (&start, ref->u.ar.as->lower[n], gfc_array_index_type);
4872 jtmp = gfc_evaluate_now (start.expr, block);
4873 itmp = fold_build2 (MINUS_EXPR, gfc_array_index_type, itmp, jtmp);
4874 itmp = fold_build2 (MULT_EXPR, gfc_array_index_type, itmp, stride);
4875 index = fold_build2 (PLUS_EXPR, gfc_array_index_type, itmp, index);
4876 index = gfc_evaluate_now (index, block);
4878 /* Update the stride. */
4879 gfc_init_se (&start, NULL);
4880 gfc_conv_expr_type (&start, ref->u.ar.as->upper[n], gfc_array_index_type);
4881 itmp = fold_build2 (MINUS_EXPR, gfc_array_index_type, start.expr, jtmp);
4882 itmp = fold_build2 (PLUS_EXPR, gfc_array_index_type,
4883 gfc_index_one_node, itmp);
4884 stride = fold_build2 (MULT_EXPR, gfc_array_index_type, stride, itmp);
4885 stride = gfc_evaluate_now (stride, block);
4888 /* Apply the index to obtain the array element. */
4889 tmp = gfc_build_array_ref (tmp, index, NULL);
4890 break;
4892 default:
4893 gcc_unreachable ();
4894 break;
4899 /* Set the target data pointer. */
4900 offset = gfc_build_addr_expr (gfc_array_dataptr_type (desc), tmp);
4901 gfc_conv_descriptor_data_set (block, parm, offset);
4905 /* gfc_conv_expr_descriptor needs the string length an expression
4906 so that the size of the temporary can be obtained. This is done
4907 by adding up the string lengths of all the elements in the
4908 expression. Function with non-constant expressions have their
4909 string lengths mapped onto the actual arguments using the
4910 interface mapping machinery in trans-expr.c. */
4911 static void
4912 get_array_charlen (gfc_expr *expr, gfc_se *se)
4914 gfc_interface_mapping mapping;
4915 gfc_formal_arglist *formal;
4916 gfc_actual_arglist *arg;
4917 gfc_se tse;
4919 if (expr->ts.u.cl->length
4920 && gfc_is_constant_expr (expr->ts.u.cl->length))
4922 if (!expr->ts.u.cl->backend_decl)
4923 gfc_conv_string_length (expr->ts.u.cl, expr, &se->pre);
4924 return;
4927 switch (expr->expr_type)
4929 case EXPR_OP:
4930 get_array_charlen (expr->value.op.op1, se);
4932 /* For parentheses the expression ts.u.cl is identical. */
4933 if (expr->value.op.op == INTRINSIC_PARENTHESES)
4934 return;
4936 expr->ts.u.cl->backend_decl =
4937 gfc_create_var (gfc_charlen_type_node, "sln");
4939 if (expr->value.op.op2)
4941 get_array_charlen (expr->value.op.op2, se);
4943 gcc_assert (expr->value.op.op == INTRINSIC_CONCAT);
4945 /* Add the string lengths and assign them to the expression
4946 string length backend declaration. */
4947 gfc_add_modify (&se->pre, expr->ts.u.cl->backend_decl,
4948 fold_build2 (PLUS_EXPR, gfc_charlen_type_node,
4949 expr->value.op.op1->ts.u.cl->backend_decl,
4950 expr->value.op.op2->ts.u.cl->backend_decl));
4952 else
4953 gfc_add_modify (&se->pre, expr->ts.u.cl->backend_decl,
4954 expr->value.op.op1->ts.u.cl->backend_decl);
4955 break;
4957 case EXPR_FUNCTION:
4958 if (expr->value.function.esym == NULL
4959 || expr->ts.u.cl->length->expr_type == EXPR_CONSTANT)
4961 gfc_conv_string_length (expr->ts.u.cl, expr, &se->pre);
4962 break;
4965 /* Map expressions involving the dummy arguments onto the actual
4966 argument expressions. */
4967 gfc_init_interface_mapping (&mapping);
4968 formal = expr->symtree->n.sym->formal;
4969 arg = expr->value.function.actual;
4971 /* Set se = NULL in the calls to the interface mapping, to suppress any
4972 backend stuff. */
4973 for (; arg != NULL; arg = arg->next, formal = formal ? formal->next : NULL)
4975 if (!arg->expr)
4976 continue;
4977 if (formal->sym)
4978 gfc_add_interface_mapping (&mapping, formal->sym, NULL, arg->expr);
4981 gfc_init_se (&tse, NULL);
4983 /* Build the expression for the character length and convert it. */
4984 gfc_apply_interface_mapping (&mapping, &tse, expr->ts.u.cl->length);
4986 gfc_add_block_to_block (&se->pre, &tse.pre);
4987 gfc_add_block_to_block (&se->post, &tse.post);
4988 tse.expr = fold_convert (gfc_charlen_type_node, tse.expr);
4989 tse.expr = fold_build2 (MAX_EXPR, gfc_charlen_type_node, tse.expr,
4990 build_int_cst (gfc_charlen_type_node, 0));
4991 expr->ts.u.cl->backend_decl = tse.expr;
4992 gfc_free_interface_mapping (&mapping);
4993 break;
4995 default:
4996 gfc_conv_string_length (expr->ts.u.cl, expr, &se->pre);
4997 break;
5003 /* Convert an array for passing as an actual argument. Expressions and
5004 vector subscripts are evaluated and stored in a temporary, which is then
5005 passed. For whole arrays the descriptor is passed. For array sections
5006 a modified copy of the descriptor is passed, but using the original data.
5008 This function is also used for array pointer assignments, and there
5009 are three cases:
5011 - se->want_pointer && !se->direct_byref
5012 EXPR is an actual argument. On exit, se->expr contains a
5013 pointer to the array descriptor.
5015 - !se->want_pointer && !se->direct_byref
5016 EXPR is an actual argument to an intrinsic function or the
5017 left-hand side of a pointer assignment. On exit, se->expr
5018 contains the descriptor for EXPR.
5020 - !se->want_pointer && se->direct_byref
5021 EXPR is the right-hand side of a pointer assignment and
5022 se->expr is the descriptor for the previously-evaluated
5023 left-hand side. The function creates an assignment from
5024 EXPR to se->expr. */
5026 void
5027 gfc_conv_expr_descriptor (gfc_se * se, gfc_expr * expr, gfc_ss * ss)
5029 gfc_loopinfo loop;
5030 gfc_ss *secss;
5031 gfc_ss_info *info;
5032 int need_tmp;
5033 int n;
5034 tree tmp;
5035 tree desc;
5036 stmtblock_t block;
5037 tree start;
5038 tree offset;
5039 int full;
5040 bool subref_array_target = false;
5042 gcc_assert (ss != gfc_ss_terminator);
5044 /* Special case things we know we can pass easily. */
5045 switch (expr->expr_type)
5047 case EXPR_VARIABLE:
5048 /* If we have a linear array section, we can pass it directly.
5049 Otherwise we need to copy it into a temporary. */
5051 /* Find the SS for the array section. */
5052 secss = ss;
5053 while (secss != gfc_ss_terminator && secss->type != GFC_SS_SECTION)
5054 secss = secss->next;
5056 gcc_assert (secss != gfc_ss_terminator);
5057 info = &secss->data.info;
5059 /* Get the descriptor for the array. */
5060 gfc_conv_ss_descriptor (&se->pre, secss, 0);
5061 desc = info->descriptor;
5063 subref_array_target = se->direct_byref && is_subref_array (expr);
5064 need_tmp = gfc_ref_needs_temporary_p (expr->ref)
5065 && !subref_array_target;
5067 if (need_tmp)
5068 full = 0;
5069 else if (GFC_ARRAY_TYPE_P (TREE_TYPE (desc)))
5071 /* Create a new descriptor if the array doesn't have one. */
5072 full = 0;
5074 else if (info->ref->u.ar.type == AR_FULL)
5075 full = 1;
5076 else if (se->direct_byref)
5077 full = 0;
5078 else
5079 full = gfc_full_array_ref_p (info->ref, NULL);
5081 if (full)
5083 if (se->direct_byref)
5085 /* Copy the descriptor for pointer assignments. */
5086 gfc_add_modify (&se->pre, se->expr, desc);
5088 /* Add any offsets from subreferences. */
5089 gfc_get_dataptr_offset (&se->pre, se->expr, desc, NULL_TREE,
5090 subref_array_target, expr);
5092 else if (se->want_pointer)
5094 /* We pass full arrays directly. This means that pointers and
5095 allocatable arrays should also work. */
5096 se->expr = gfc_build_addr_expr (NULL_TREE, desc);
5098 else
5100 se->expr = desc;
5103 if (expr->ts.type == BT_CHARACTER)
5104 se->string_length = gfc_get_expr_charlen (expr);
5106 return;
5108 break;
5110 case EXPR_FUNCTION:
5111 /* A transformational function return value will be a temporary
5112 array descriptor. We still need to go through the scalarizer
5113 to create the descriptor. Elemental functions ar handled as
5114 arbitrary expressions, i.e. copy to a temporary. */
5115 secss = ss;
5116 /* Look for the SS for this function. */
5117 while (secss != gfc_ss_terminator
5118 && (secss->type != GFC_SS_FUNCTION || secss->expr != expr))
5119 secss = secss->next;
5121 if (se->direct_byref)
5123 gcc_assert (secss != gfc_ss_terminator);
5125 /* For pointer assignments pass the descriptor directly. */
5126 se->ss = secss;
5127 se->expr = gfc_build_addr_expr (NULL_TREE, se->expr);
5128 gfc_conv_expr (se, expr);
5129 return;
5132 if (secss == gfc_ss_terminator)
5134 /* Elemental function. */
5135 need_tmp = 1;
5136 if (expr->ts.type == BT_CHARACTER
5137 && expr->ts.u.cl->length->expr_type != EXPR_CONSTANT)
5138 get_array_charlen (expr, se);
5140 info = NULL;
5142 else
5144 /* Transformational function. */
5145 info = &secss->data.info;
5146 need_tmp = 0;
5148 break;
5150 case EXPR_ARRAY:
5151 /* Constant array constructors don't need a temporary. */
5152 if (ss->type == GFC_SS_CONSTRUCTOR
5153 && expr->ts.type != BT_CHARACTER
5154 && gfc_constant_array_constructor_p (expr->value.constructor))
5156 need_tmp = 0;
5157 info = &ss->data.info;
5158 secss = ss;
5160 else
5162 need_tmp = 1;
5163 secss = NULL;
5164 info = NULL;
5166 break;
5168 default:
5169 /* Something complicated. Copy it into a temporary. */
5170 need_tmp = 1;
5171 secss = NULL;
5172 info = NULL;
5173 break;
5176 gfc_init_loopinfo (&loop);
5178 /* Associate the SS with the loop. */
5179 gfc_add_ss_to_loop (&loop, ss);
5181 /* Tell the scalarizer not to bother creating loop variables, etc. */
5182 if (!need_tmp)
5183 loop.array_parameter = 1;
5184 else
5185 /* The right-hand side of a pointer assignment mustn't use a temporary. */
5186 gcc_assert (!se->direct_byref);
5188 /* Setup the scalarizing loops and bounds. */
5189 gfc_conv_ss_startstride (&loop);
5191 if (need_tmp)
5193 /* Tell the scalarizer to make a temporary. */
5194 loop.temp_ss = gfc_get_ss ();
5195 loop.temp_ss->type = GFC_SS_TEMP;
5196 loop.temp_ss->next = gfc_ss_terminator;
5198 if (expr->ts.type == BT_CHARACTER
5199 && !expr->ts.u.cl->backend_decl)
5200 get_array_charlen (expr, se);
5202 loop.temp_ss->data.temp.type = gfc_typenode_for_spec (&expr->ts);
5204 if (expr->ts.type == BT_CHARACTER)
5205 loop.temp_ss->string_length = expr->ts.u.cl->backend_decl;
5206 else
5207 loop.temp_ss->string_length = NULL;
5209 se->string_length = loop.temp_ss->string_length;
5210 loop.temp_ss->data.temp.dimen = loop.dimen;
5211 gfc_add_ss_to_loop (&loop, loop.temp_ss);
5214 gfc_conv_loop_setup (&loop, & expr->where);
5216 if (need_tmp)
5218 /* Copy into a temporary and pass that. We don't need to copy the data
5219 back because expressions and vector subscripts must be INTENT_IN. */
5220 /* TODO: Optimize passing function return values. */
5221 gfc_se lse;
5222 gfc_se rse;
5224 /* Start the copying loops. */
5225 gfc_mark_ss_chain_used (loop.temp_ss, 1);
5226 gfc_mark_ss_chain_used (ss, 1);
5227 gfc_start_scalarized_body (&loop, &block);
5229 /* Copy each data element. */
5230 gfc_init_se (&lse, NULL);
5231 gfc_copy_loopinfo_to_se (&lse, &loop);
5232 gfc_init_se (&rse, NULL);
5233 gfc_copy_loopinfo_to_se (&rse, &loop);
5235 lse.ss = loop.temp_ss;
5236 rse.ss = ss;
5238 gfc_conv_scalarized_array_ref (&lse, NULL);
5239 if (expr->ts.type == BT_CHARACTER)
5241 gfc_conv_expr (&rse, expr);
5242 if (POINTER_TYPE_P (TREE_TYPE (rse.expr)))
5243 rse.expr = build_fold_indirect_ref_loc (input_location,
5244 rse.expr);
5246 else
5247 gfc_conv_expr_val (&rse, expr);
5249 gfc_add_block_to_block (&block, &rse.pre);
5250 gfc_add_block_to_block (&block, &lse.pre);
5252 lse.string_length = rse.string_length;
5253 tmp = gfc_trans_scalar_assign (&lse, &rse, expr->ts, true,
5254 expr->expr_type == EXPR_VARIABLE, true);
5255 gfc_add_expr_to_block (&block, tmp);
5257 /* Finish the copying loops. */
5258 gfc_trans_scalarizing_loops (&loop, &block);
5260 desc = loop.temp_ss->data.info.descriptor;
5262 gcc_assert (is_gimple_lvalue (desc));
5264 else if (expr->expr_type == EXPR_FUNCTION)
5266 desc = info->descriptor;
5267 se->string_length = ss->string_length;
5269 else
5271 /* We pass sections without copying to a temporary. Make a new
5272 descriptor and point it at the section we want. The loop variable
5273 limits will be the limits of the section.
5274 A function may decide to repack the array to speed up access, but
5275 we're not bothered about that here. */
5276 int dim, ndim;
5277 tree parm;
5278 tree parmtype;
5279 tree stride;
5280 tree from;
5281 tree to;
5282 tree base;
5284 /* Set the string_length for a character array. */
5285 if (expr->ts.type == BT_CHARACTER)
5286 se->string_length = gfc_get_expr_charlen (expr);
5288 desc = info->descriptor;
5289 gcc_assert (secss && secss != gfc_ss_terminator);
5290 if (se->direct_byref)
5292 /* For pointer assignments we fill in the destination. */
5293 parm = se->expr;
5294 parmtype = TREE_TYPE (parm);
5296 else
5298 /* Otherwise make a new one. */
5299 parmtype = gfc_get_element_type (TREE_TYPE (desc));
5300 parmtype = gfc_get_array_type_bounds (parmtype, loop.dimen,
5301 loop.from, loop.to, 0,
5302 GFC_ARRAY_UNKNOWN, false);
5303 parm = gfc_create_var (parmtype, "parm");
5306 offset = gfc_index_zero_node;
5307 dim = 0;
5309 /* The following can be somewhat confusing. We have two
5310 descriptors, a new one and the original array.
5311 {parm, parmtype, dim} refer to the new one.
5312 {desc, type, n, secss, loop} refer to the original, which maybe
5313 a descriptorless array.
5314 The bounds of the scalarization are the bounds of the section.
5315 We don't have to worry about numeric overflows when calculating
5316 the offsets because all elements are within the array data. */
5318 /* Set the dtype. */
5319 tmp = gfc_conv_descriptor_dtype (parm);
5320 gfc_add_modify (&loop.pre, tmp, gfc_get_dtype (parmtype));
5322 /* Set offset for assignments to pointer only to zero if it is not
5323 the full array. */
5324 if (se->direct_byref
5325 && info->ref && info->ref->u.ar.type != AR_FULL)
5326 base = gfc_index_zero_node;
5327 else if (GFC_ARRAY_TYPE_P (TREE_TYPE (desc)))
5328 base = gfc_evaluate_now (gfc_conv_array_offset (desc), &loop.pre);
5329 else
5330 base = NULL_TREE;
5332 ndim = info->ref ? info->ref->u.ar.dimen : info->dimen;
5333 for (n = 0; n < ndim; n++)
5335 stride = gfc_conv_array_stride (desc, n);
5337 /* Work out the offset. */
5338 if (info->ref
5339 && info->ref->u.ar.dimen_type[n] == DIMEN_ELEMENT)
5341 gcc_assert (info->subscript[n]
5342 && info->subscript[n]->type == GFC_SS_SCALAR);
5343 start = info->subscript[n]->data.scalar.expr;
5345 else
5347 /* Check we haven't somehow got out of sync. */
5348 gcc_assert (info->dim[dim] == n);
5350 /* Evaluate and remember the start of the section. */
5351 start = info->start[dim];
5352 stride = gfc_evaluate_now (stride, &loop.pre);
5355 tmp = gfc_conv_array_lbound (desc, n);
5356 tmp = fold_build2 (MINUS_EXPR, TREE_TYPE (tmp), start, tmp);
5358 tmp = fold_build2 (MULT_EXPR, TREE_TYPE (tmp), tmp, stride);
5359 offset = fold_build2 (PLUS_EXPR, TREE_TYPE (tmp), offset, tmp);
5361 if (info->ref
5362 && info->ref->u.ar.dimen_type[n] == DIMEN_ELEMENT)
5364 /* For elemental dimensions, we only need the offset. */
5365 continue;
5368 /* Vector subscripts need copying and are handled elsewhere. */
5369 if (info->ref)
5370 gcc_assert (info->ref->u.ar.dimen_type[n] == DIMEN_RANGE);
5372 /* Set the new lower bound. */
5373 from = loop.from[dim];
5374 to = loop.to[dim];
5376 /* If we have an array section or are assigning make sure that
5377 the lower bound is 1. References to the full
5378 array should otherwise keep the original bounds. */
5379 if ((!info->ref
5380 || info->ref->u.ar.type != AR_FULL)
5381 && !integer_onep (from))
5383 tmp = fold_build2 (MINUS_EXPR, gfc_array_index_type,
5384 gfc_index_one_node, from);
5385 to = fold_build2 (PLUS_EXPR, gfc_array_index_type, to, tmp);
5386 from = gfc_index_one_node;
5388 gfc_conv_descriptor_lbound_set (&loop.pre, parm,
5389 gfc_rank_cst[dim], from);
5391 /* Set the new upper bound. */
5392 gfc_conv_descriptor_ubound_set (&loop.pre, parm,
5393 gfc_rank_cst[dim], to);
5395 /* Multiply the stride by the section stride to get the
5396 total stride. */
5397 stride = fold_build2 (MULT_EXPR, gfc_array_index_type,
5398 stride, info->stride[dim]);
5400 if (se->direct_byref
5401 && info->ref
5402 && info->ref->u.ar.type != AR_FULL)
5404 base = fold_build2 (MINUS_EXPR, TREE_TYPE (base),
5405 base, stride);
5407 else if (GFC_ARRAY_TYPE_P (TREE_TYPE (desc)))
5409 tmp = gfc_conv_array_lbound (desc, n);
5410 tmp = fold_build2 (MINUS_EXPR, TREE_TYPE (base),
5411 tmp, loop.from[dim]);
5412 tmp = fold_build2 (MULT_EXPR, TREE_TYPE (base),
5413 tmp, gfc_conv_array_stride (desc, n));
5414 base = fold_build2 (PLUS_EXPR, TREE_TYPE (base),
5415 tmp, base);
5418 /* Store the new stride. */
5419 gfc_conv_descriptor_stride_set (&loop.pre, parm,
5420 gfc_rank_cst[dim], stride);
5422 dim++;
5425 if (se->data_not_needed)
5426 gfc_conv_descriptor_data_set (&loop.pre, parm,
5427 gfc_index_zero_node);
5428 else
5429 /* Point the data pointer at the 1st element in the section. */
5430 gfc_get_dataptr_offset (&loop.pre, parm, desc, offset,
5431 subref_array_target, expr);
5433 if ((se->direct_byref || GFC_ARRAY_TYPE_P (TREE_TYPE (desc)))
5434 && !se->data_not_needed)
5436 /* Set the offset. */
5437 gfc_conv_descriptor_offset_set (&loop.pre, parm, base);
5439 else
5441 /* Only the callee knows what the correct offset it, so just set
5442 it to zero here. */
5443 gfc_conv_descriptor_offset_set (&loop.pre, parm, gfc_index_zero_node);
5445 desc = parm;
5448 if (!se->direct_byref)
5450 /* Get a pointer to the new descriptor. */
5451 if (se->want_pointer)
5452 se->expr = gfc_build_addr_expr (NULL_TREE, desc);
5453 else
5454 se->expr = desc;
5457 gfc_add_block_to_block (&se->pre, &loop.pre);
5458 gfc_add_block_to_block (&se->post, &loop.post);
5460 /* Cleanup the scalarizer. */
5461 gfc_cleanup_loop (&loop);
5464 /* Helper function for gfc_conv_array_parameter if array size needs to be
5465 computed. */
5467 static void
5468 array_parameter_size (tree desc, gfc_expr *expr, tree *size)
5470 tree elem;
5471 if (GFC_ARRAY_TYPE_P (TREE_TYPE (desc)))
5472 *size = GFC_TYPE_ARRAY_SIZE (TREE_TYPE (desc));
5473 else if (expr->rank > 1)
5474 *size = build_call_expr_loc (input_location,
5475 gfor_fndecl_size0, 1,
5476 gfc_build_addr_expr (NULL, desc));
5477 else
5479 tree ubound = gfc_conv_descriptor_ubound_get (desc, gfc_index_zero_node);
5480 tree lbound = gfc_conv_descriptor_lbound_get (desc, gfc_index_zero_node);
5482 *size = fold_build2 (MINUS_EXPR, gfc_array_index_type, ubound, lbound);
5483 *size = fold_build2 (PLUS_EXPR, gfc_array_index_type, *size,
5484 gfc_index_one_node);
5485 *size = fold_build2 (MAX_EXPR, gfc_array_index_type, *size,
5486 gfc_index_zero_node);
5488 elem = TYPE_SIZE_UNIT (gfc_get_element_type (TREE_TYPE (desc)));
5489 *size = fold_build2 (MULT_EXPR, gfc_array_index_type, *size,
5490 fold_convert (gfc_array_index_type, elem));
5493 /* Convert an array for passing as an actual parameter. */
5494 /* TODO: Optimize passing g77 arrays. */
5496 void
5497 gfc_conv_array_parameter (gfc_se * se, gfc_expr * expr, gfc_ss * ss, bool g77,
5498 const gfc_symbol *fsym, const char *proc_name,
5499 tree *size)
5501 tree ptr;
5502 tree desc;
5503 tree tmp = NULL_TREE;
5504 tree stmt;
5505 tree parent = DECL_CONTEXT (current_function_decl);
5506 bool full_array_var;
5507 bool this_array_result;
5508 bool contiguous;
5509 bool no_pack;
5510 bool array_constructor;
5511 bool good_allocatable;
5512 bool ultimate_ptr_comp;
5513 bool ultimate_alloc_comp;
5514 gfc_symbol *sym;
5515 stmtblock_t block;
5516 gfc_ref *ref;
5518 ultimate_ptr_comp = false;
5519 ultimate_alloc_comp = false;
5520 for (ref = expr->ref; ref; ref = ref->next)
5522 if (ref->next == NULL)
5523 break;
5525 if (ref->type == REF_COMPONENT)
5527 ultimate_ptr_comp = ref->u.c.component->attr.pointer;
5528 ultimate_alloc_comp = ref->u.c.component->attr.allocatable;
5532 full_array_var = false;
5533 contiguous = false;
5535 if (expr->expr_type == EXPR_VARIABLE && ref && !ultimate_ptr_comp)
5536 full_array_var = gfc_full_array_ref_p (ref, &contiguous);
5538 sym = full_array_var ? expr->symtree->n.sym : NULL;
5540 /* The symbol should have an array specification. */
5541 gcc_assert (!sym || sym->as || ref->u.ar.as);
5543 if (expr->expr_type == EXPR_ARRAY && expr->ts.type == BT_CHARACTER)
5545 get_array_ctor_strlen (&se->pre, expr->value.constructor, &tmp);
5546 expr->ts.u.cl->backend_decl = tmp;
5547 se->string_length = tmp;
5550 /* Is this the result of the enclosing procedure? */
5551 this_array_result = (full_array_var && sym->attr.flavor == FL_PROCEDURE);
5552 if (this_array_result
5553 && (sym->backend_decl != current_function_decl)
5554 && (sym->backend_decl != parent))
5555 this_array_result = false;
5557 /* Passing address of the array if it is not pointer or assumed-shape. */
5558 if (full_array_var && g77 && !this_array_result)
5560 tmp = gfc_get_symbol_decl (sym);
5562 if (sym->ts.type == BT_CHARACTER)
5563 se->string_length = sym->ts.u.cl->backend_decl;
5565 if (sym->ts.type == BT_DERIVED)
5567 gfc_conv_expr_descriptor (se, expr, ss);
5568 se->expr = gfc_conv_array_data (se->expr);
5569 return;
5572 if (!sym->attr.pointer
5573 && sym->as
5574 && sym->as->type != AS_ASSUMED_SHAPE
5575 && !sym->attr.allocatable)
5577 /* Some variables are declared directly, others are declared as
5578 pointers and allocated on the heap. */
5579 if (sym->attr.dummy || POINTER_TYPE_P (TREE_TYPE (tmp)))
5580 se->expr = tmp;
5581 else
5582 se->expr = gfc_build_addr_expr (NULL_TREE, tmp);
5583 if (size)
5584 array_parameter_size (tmp, expr, size);
5585 return;
5588 if (sym->attr.allocatable)
5590 if (sym->attr.dummy || sym->attr.result)
5592 gfc_conv_expr_descriptor (se, expr, ss);
5593 tmp = se->expr;
5595 if (size)
5596 array_parameter_size (tmp, expr, size);
5597 se->expr = gfc_conv_array_data (tmp);
5598 return;
5602 /* A convenient reduction in scope. */
5603 contiguous = g77 && !this_array_result && contiguous;
5605 /* There is no need to pack and unpack the array, if it is contiguous
5606 and not deferred or assumed shape. */
5607 no_pack = ((sym && sym->as
5608 && !sym->attr.pointer
5609 && sym->as->type != AS_DEFERRED
5610 && sym->as->type != AS_ASSUMED_SHAPE)
5612 (ref && ref->u.ar.as
5613 && ref->u.ar.as->type != AS_DEFERRED
5614 && ref->u.ar.as->type != AS_ASSUMED_SHAPE));
5616 no_pack = contiguous && no_pack;
5618 /* Array constructors are always contiguous and do not need packing. */
5619 array_constructor = g77 && !this_array_result && expr->expr_type == EXPR_ARRAY;
5621 /* Same is true of contiguous sections from allocatable variables. */
5622 good_allocatable = contiguous
5623 && expr->symtree
5624 && expr->symtree->n.sym->attr.allocatable;
5626 /* Or ultimate allocatable components. */
5627 ultimate_alloc_comp = contiguous && ultimate_alloc_comp;
5629 if (no_pack || array_constructor || good_allocatable || ultimate_alloc_comp)
5631 gfc_conv_expr_descriptor (se, expr, ss);
5632 if (expr->ts.type == BT_CHARACTER)
5633 se->string_length = expr->ts.u.cl->backend_decl;
5634 if (size)
5635 array_parameter_size (se->expr, expr, size);
5636 se->expr = gfc_conv_array_data (se->expr);
5637 return;
5640 if (this_array_result)
5642 /* Result of the enclosing function. */
5643 gfc_conv_expr_descriptor (se, expr, ss);
5644 if (size)
5645 array_parameter_size (se->expr, expr, size);
5646 se->expr = gfc_build_addr_expr (NULL_TREE, se->expr);
5648 if (g77 && TREE_TYPE (TREE_TYPE (se->expr)) != NULL_TREE
5649 && GFC_DESCRIPTOR_TYPE_P (TREE_TYPE (TREE_TYPE (se->expr))))
5650 se->expr = gfc_conv_array_data (build_fold_indirect_ref_loc (input_location,
5651 se->expr));
5653 return;
5655 else
5657 /* Every other type of array. */
5658 se->want_pointer = 1;
5659 gfc_conv_expr_descriptor (se, expr, ss);
5660 if (size)
5661 array_parameter_size (build_fold_indirect_ref_loc (input_location,
5662 se->expr),
5663 expr, size);
5666 /* Deallocate the allocatable components of structures that are
5667 not variable. */
5668 if (expr->ts.type == BT_DERIVED
5669 && expr->ts.u.derived->attr.alloc_comp
5670 && expr->expr_type != EXPR_VARIABLE)
5672 tmp = build_fold_indirect_ref_loc (input_location,
5673 se->expr);
5674 tmp = gfc_deallocate_alloc_comp (expr->ts.u.derived, tmp, expr->rank);
5675 gfc_add_expr_to_block (&se->post, tmp);
5678 if (g77)
5680 desc = se->expr;
5681 /* Repack the array. */
5682 if (gfc_option.warn_array_temp)
5684 if (fsym)
5685 gfc_warning ("Creating array temporary at %L for argument '%s'",
5686 &expr->where, fsym->name);
5687 else
5688 gfc_warning ("Creating array temporary at %L", &expr->where);
5691 ptr = build_call_expr_loc (input_location,
5692 gfor_fndecl_in_pack, 1, desc);
5694 if (fsym && fsym->attr.optional && sym && sym->attr.optional)
5696 tmp = gfc_conv_expr_present (sym);
5697 ptr = build3 (COND_EXPR, TREE_TYPE (se->expr), tmp,
5698 fold_convert (TREE_TYPE (se->expr), ptr),
5699 fold_convert (TREE_TYPE (se->expr), null_pointer_node));
5702 ptr = gfc_evaluate_now (ptr, &se->pre);
5704 se->expr = ptr;
5706 if (gfc_option.rtcheck & GFC_RTCHECK_ARRAY_TEMPS)
5708 char * msg;
5710 if (fsym && proc_name)
5711 asprintf (&msg, "An array temporary was created for argument "
5712 "'%s' of procedure '%s'", fsym->name, proc_name);
5713 else
5714 asprintf (&msg, "An array temporary was created");
5716 tmp = build_fold_indirect_ref_loc (input_location,
5717 desc);
5718 tmp = gfc_conv_array_data (tmp);
5719 tmp = fold_build2 (NE_EXPR, boolean_type_node,
5720 fold_convert (TREE_TYPE (tmp), ptr), tmp);
5722 if (fsym && fsym->attr.optional && sym && sym->attr.optional)
5723 tmp = fold_build2 (TRUTH_AND_EXPR, boolean_type_node,
5724 gfc_conv_expr_present (sym), tmp);
5726 gfc_trans_runtime_check (false, true, tmp, &se->pre,
5727 &expr->where, msg);
5728 gfc_free (msg);
5731 gfc_start_block (&block);
5733 /* Copy the data back. */
5734 if (fsym == NULL || fsym->attr.intent != INTENT_IN)
5736 tmp = build_call_expr_loc (input_location,
5737 gfor_fndecl_in_unpack, 2, desc, ptr);
5738 gfc_add_expr_to_block (&block, tmp);
5741 /* Free the temporary. */
5742 tmp = gfc_call_free (convert (pvoid_type_node, ptr));
5743 gfc_add_expr_to_block (&block, tmp);
5745 stmt = gfc_finish_block (&block);
5747 gfc_init_block (&block);
5748 /* Only if it was repacked. This code needs to be executed before the
5749 loop cleanup code. */
5750 tmp = build_fold_indirect_ref_loc (input_location,
5751 desc);
5752 tmp = gfc_conv_array_data (tmp);
5753 tmp = fold_build2 (NE_EXPR, boolean_type_node,
5754 fold_convert (TREE_TYPE (tmp), ptr), tmp);
5756 if (fsym && fsym->attr.optional && sym && sym->attr.optional)
5757 tmp = fold_build2 (TRUTH_AND_EXPR, boolean_type_node,
5758 gfc_conv_expr_present (sym), tmp);
5760 tmp = build3_v (COND_EXPR, tmp, stmt, build_empty_stmt (input_location));
5762 gfc_add_expr_to_block (&block, tmp);
5763 gfc_add_block_to_block (&block, &se->post);
5765 gfc_init_block (&se->post);
5766 gfc_add_block_to_block (&se->post, &block);
5771 /* Generate code to deallocate an array, if it is allocated. */
5773 tree
5774 gfc_trans_dealloc_allocated (tree descriptor)
5776 tree tmp;
5777 tree var;
5778 stmtblock_t block;
5780 gfc_start_block (&block);
5782 var = gfc_conv_descriptor_data_get (descriptor);
5783 STRIP_NOPS (var);
5785 /* Call array_deallocate with an int * present in the second argument.
5786 Although it is ignored here, it's presence ensures that arrays that
5787 are already deallocated are ignored. */
5788 tmp = gfc_deallocate_with_status (var, NULL_TREE, true, NULL);
5789 gfc_add_expr_to_block (&block, tmp);
5791 /* Zero the data pointer. */
5792 tmp = fold_build2 (MODIFY_EXPR, void_type_node,
5793 var, build_int_cst (TREE_TYPE (var), 0));
5794 gfc_add_expr_to_block (&block, tmp);
5796 return gfc_finish_block (&block);
5800 /* This helper function calculates the size in words of a full array. */
5802 static tree
5803 get_full_array_size (stmtblock_t *block, tree decl, int rank)
5805 tree idx;
5806 tree nelems;
5807 tree tmp;
5808 idx = gfc_rank_cst[rank - 1];
5809 nelems = gfc_conv_descriptor_ubound_get (decl, idx);
5810 tmp = gfc_conv_descriptor_lbound_get (decl, idx);
5811 tmp = fold_build2 (MINUS_EXPR, gfc_array_index_type, nelems, tmp);
5812 tmp = fold_build2 (PLUS_EXPR, gfc_array_index_type,
5813 tmp, gfc_index_one_node);
5814 tmp = gfc_evaluate_now (tmp, block);
5816 nelems = gfc_conv_descriptor_stride_get (decl, idx);
5817 tmp = fold_build2 (MULT_EXPR, gfc_array_index_type, nelems, tmp);
5818 return gfc_evaluate_now (tmp, block);
5822 /* Allocate dest to the same size as src, and copy src -> dest.
5823 If no_malloc is set, only the copy is done. */
5825 static tree
5826 duplicate_allocatable(tree dest, tree src, tree type, int rank,
5827 bool no_malloc)
5829 tree tmp;
5830 tree size;
5831 tree nelems;
5832 tree null_cond;
5833 tree null_data;
5834 stmtblock_t block;
5836 /* If the source is null, set the destination to null. Then,
5837 allocate memory to the destination. */
5838 gfc_init_block (&block);
5840 if (rank == 0)
5842 tmp = null_pointer_node;
5843 tmp = fold_build2 (MODIFY_EXPR, type, dest, tmp);
5844 gfc_add_expr_to_block (&block, tmp);
5845 null_data = gfc_finish_block (&block);
5847 gfc_init_block (&block);
5848 size = TYPE_SIZE_UNIT (type);
5849 if (!no_malloc)
5851 tmp = gfc_call_malloc (&block, type, size);
5852 tmp = fold_build2 (MODIFY_EXPR, void_type_node, dest,
5853 fold_convert (type, tmp));
5854 gfc_add_expr_to_block (&block, tmp);
5857 tmp = built_in_decls[BUILT_IN_MEMCPY];
5858 tmp = build_call_expr_loc (input_location, tmp, 3,
5859 dest, src, size);
5861 else
5863 gfc_conv_descriptor_data_set (&block, dest, null_pointer_node);
5864 null_data = gfc_finish_block (&block);
5866 gfc_init_block (&block);
5867 nelems = get_full_array_size (&block, src, rank);
5868 tmp = fold_convert (gfc_array_index_type,
5869 TYPE_SIZE_UNIT (gfc_get_element_type (type)));
5870 size = fold_build2 (MULT_EXPR, gfc_array_index_type, nelems, tmp);
5871 if (!no_malloc)
5873 tmp = TREE_TYPE (gfc_conv_descriptor_data_get (src));
5874 tmp = gfc_call_malloc (&block, tmp, size);
5875 gfc_conv_descriptor_data_set (&block, dest, tmp);
5878 /* We know the temporary and the value will be the same length,
5879 so can use memcpy. */
5880 tmp = built_in_decls[BUILT_IN_MEMCPY];
5881 tmp = build_call_expr_loc (input_location,
5882 tmp, 3, gfc_conv_descriptor_data_get (dest),
5883 gfc_conv_descriptor_data_get (src), size);
5886 gfc_add_expr_to_block (&block, tmp);
5887 tmp = gfc_finish_block (&block);
5889 /* Null the destination if the source is null; otherwise do
5890 the allocate and copy. */
5891 if (rank == 0)
5892 null_cond = src;
5893 else
5894 null_cond = gfc_conv_descriptor_data_get (src);
5896 null_cond = convert (pvoid_type_node, null_cond);
5897 null_cond = fold_build2 (NE_EXPR, boolean_type_node,
5898 null_cond, null_pointer_node);
5899 return build3_v (COND_EXPR, null_cond, tmp, null_data);
5903 /* Allocate dest to the same size as src, and copy data src -> dest. */
5905 tree
5906 gfc_duplicate_allocatable (tree dest, tree src, tree type, int rank)
5908 return duplicate_allocatable(dest, src, type, rank, false);
5912 /* Copy data src -> dest. */
5914 tree
5915 gfc_copy_allocatable_data (tree dest, tree src, tree type, int rank)
5917 return duplicate_allocatable(dest, src, type, rank, true);
5921 /* Recursively traverse an object of derived type, generating code to
5922 deallocate, nullify or copy allocatable components. This is the work horse
5923 function for the functions named in this enum. */
5925 enum {DEALLOCATE_ALLOC_COMP = 1, NULLIFY_ALLOC_COMP, COPY_ALLOC_COMP,
5926 COPY_ONLY_ALLOC_COMP};
5928 static tree
5929 structure_alloc_comps (gfc_symbol * der_type, tree decl,
5930 tree dest, int rank, int purpose)
5932 gfc_component *c;
5933 gfc_loopinfo loop;
5934 stmtblock_t fnblock;
5935 stmtblock_t loopbody;
5936 tree tmp;
5937 tree comp;
5938 tree dcmp;
5939 tree nelems;
5940 tree index;
5941 tree var;
5942 tree cdecl;
5943 tree ctype;
5944 tree vref, dref;
5945 tree null_cond = NULL_TREE;
5947 gfc_init_block (&fnblock);
5949 if (POINTER_TYPE_P (TREE_TYPE (decl)) && rank != 0)
5950 decl = build_fold_indirect_ref_loc (input_location,
5951 decl);
5953 /* If this an array of derived types with allocatable components
5954 build a loop and recursively call this function. */
5955 if (TREE_CODE (TREE_TYPE (decl)) == ARRAY_TYPE
5956 || GFC_DESCRIPTOR_TYPE_P (TREE_TYPE (decl)))
5958 tmp = gfc_conv_array_data (decl);
5959 var = build_fold_indirect_ref_loc (input_location,
5960 tmp);
5962 /* Get the number of elements - 1 and set the counter. */
5963 if (GFC_DESCRIPTOR_TYPE_P (TREE_TYPE (decl)))
5965 /* Use the descriptor for an allocatable array. Since this
5966 is a full array reference, we only need the descriptor
5967 information from dimension = rank. */
5968 tmp = get_full_array_size (&fnblock, decl, rank);
5969 tmp = fold_build2 (MINUS_EXPR, gfc_array_index_type,
5970 tmp, gfc_index_one_node);
5972 null_cond = gfc_conv_descriptor_data_get (decl);
5973 null_cond = fold_build2 (NE_EXPR, boolean_type_node, null_cond,
5974 build_int_cst (TREE_TYPE (null_cond), 0));
5976 else
5978 /* Otherwise use the TYPE_DOMAIN information. */
5979 tmp = array_type_nelts (TREE_TYPE (decl));
5980 tmp = fold_convert (gfc_array_index_type, tmp);
5983 /* Remember that this is, in fact, the no. of elements - 1. */
5984 nelems = gfc_evaluate_now (tmp, &fnblock);
5985 index = gfc_create_var (gfc_array_index_type, "S");
5987 /* Build the body of the loop. */
5988 gfc_init_block (&loopbody);
5990 vref = gfc_build_array_ref (var, index, NULL);
5992 if (purpose == COPY_ALLOC_COMP)
5994 if (GFC_DESCRIPTOR_TYPE_P (TREE_TYPE (dest)))
5996 tmp = gfc_duplicate_allocatable (dest, decl, TREE_TYPE(decl), rank);
5997 gfc_add_expr_to_block (&fnblock, tmp);
5999 tmp = build_fold_indirect_ref_loc (input_location,
6000 gfc_conv_array_data (dest));
6001 dref = gfc_build_array_ref (tmp, index, NULL);
6002 tmp = structure_alloc_comps (der_type, vref, dref, rank, purpose);
6004 else if (purpose == COPY_ONLY_ALLOC_COMP)
6006 tmp = build_fold_indirect_ref_loc (input_location,
6007 gfc_conv_array_data (dest));
6008 dref = gfc_build_array_ref (tmp, index, NULL);
6009 tmp = structure_alloc_comps (der_type, vref, dref, rank,
6010 COPY_ALLOC_COMP);
6012 else
6013 tmp = structure_alloc_comps (der_type, vref, NULL_TREE, rank, purpose);
6015 gfc_add_expr_to_block (&loopbody, tmp);
6017 /* Build the loop and return. */
6018 gfc_init_loopinfo (&loop);
6019 loop.dimen = 1;
6020 loop.from[0] = gfc_index_zero_node;
6021 loop.loopvar[0] = index;
6022 loop.to[0] = nelems;
6023 gfc_trans_scalarizing_loops (&loop, &loopbody);
6024 gfc_add_block_to_block (&fnblock, &loop.pre);
6026 tmp = gfc_finish_block (&fnblock);
6027 if (null_cond != NULL_TREE)
6028 tmp = build3_v (COND_EXPR, null_cond, tmp,
6029 build_empty_stmt (input_location));
6031 return tmp;
6034 /* Otherwise, act on the components or recursively call self to
6035 act on a chain of components. */
6036 for (c = der_type->components; c; c = c->next)
6038 bool cmp_has_alloc_comps = (c->ts.type == BT_DERIVED)
6039 && c->ts.u.derived->attr.alloc_comp;
6040 cdecl = c->backend_decl;
6041 ctype = TREE_TYPE (cdecl);
6043 switch (purpose)
6045 case DEALLOCATE_ALLOC_COMP:
6046 /* Do not deallocate the components of ultimate pointer
6047 components. */
6048 if (cmp_has_alloc_comps && !c->attr.pointer)
6050 comp = fold_build3 (COMPONENT_REF, ctype,
6051 decl, cdecl, NULL_TREE);
6052 rank = c->as ? c->as->rank : 0;
6053 tmp = structure_alloc_comps (c->ts.u.derived, comp, NULL_TREE,
6054 rank, purpose);
6055 gfc_add_expr_to_block (&fnblock, tmp);
6058 if (c->attr.allocatable && c->attr.dimension)
6060 comp = fold_build3 (COMPONENT_REF, ctype,
6061 decl, cdecl, NULL_TREE);
6062 tmp = gfc_trans_dealloc_allocated (comp);
6063 gfc_add_expr_to_block (&fnblock, tmp);
6065 else if (c->attr.allocatable)
6067 /* Allocatable scalar components. */
6068 comp = fold_build3 (COMPONENT_REF, ctype, decl, cdecl, NULL_TREE);
6070 tmp = gfc_deallocate_with_status (comp, NULL_TREE, true, NULL);
6071 gfc_add_expr_to_block (&fnblock, tmp);
6073 tmp = fold_build2 (MODIFY_EXPR, void_type_node, comp,
6074 build_int_cst (TREE_TYPE (comp), 0));
6075 gfc_add_expr_to_block (&fnblock, tmp);
6077 else if (c->ts.type == BT_CLASS
6078 && c->ts.u.derived->components->attr.allocatable)
6080 /* Allocatable scalar CLASS components. */
6081 comp = fold_build3 (COMPONENT_REF, ctype, decl, cdecl, NULL_TREE);
6083 /* Add reference to '$data' component. */
6084 tmp = c->ts.u.derived->components->backend_decl;
6085 comp = fold_build3 (COMPONENT_REF, TREE_TYPE (tmp),
6086 comp, tmp, NULL_TREE);
6088 tmp = gfc_deallocate_with_status (comp, NULL_TREE, true, NULL);
6089 gfc_add_expr_to_block (&fnblock, tmp);
6091 tmp = fold_build2 (MODIFY_EXPR, void_type_node, comp,
6092 build_int_cst (TREE_TYPE (comp), 0));
6093 gfc_add_expr_to_block (&fnblock, tmp);
6095 break;
6097 case NULLIFY_ALLOC_COMP:
6098 if (c->attr.pointer)
6099 continue;
6100 else if (c->attr.allocatable && c->attr.dimension)
6102 comp = fold_build3 (COMPONENT_REF, ctype,
6103 decl, cdecl, NULL_TREE);
6104 gfc_conv_descriptor_data_set (&fnblock, comp, null_pointer_node);
6106 else if (c->attr.allocatable)
6108 /* Allocatable scalar components. */
6109 comp = fold_build3 (COMPONENT_REF, ctype, decl, cdecl, NULL_TREE);
6110 tmp = fold_build2 (MODIFY_EXPR, void_type_node, comp,
6111 build_int_cst (TREE_TYPE (comp), 0));
6112 gfc_add_expr_to_block (&fnblock, tmp);
6114 else if (c->ts.type == BT_CLASS
6115 && c->ts.u.derived->components->attr.allocatable)
6117 /* Allocatable scalar CLASS components. */
6118 comp = fold_build3 (COMPONENT_REF, ctype, decl, cdecl, NULL_TREE);
6119 /* Add reference to '$data' component. */
6120 tmp = c->ts.u.derived->components->backend_decl;
6121 comp = fold_build3 (COMPONENT_REF, TREE_TYPE (tmp),
6122 comp, tmp, NULL_TREE);
6123 tmp = fold_build2 (MODIFY_EXPR, void_type_node, comp,
6124 build_int_cst (TREE_TYPE (comp), 0));
6125 gfc_add_expr_to_block (&fnblock, tmp);
6127 else if (cmp_has_alloc_comps)
6129 comp = fold_build3 (COMPONENT_REF, ctype,
6130 decl, cdecl, NULL_TREE);
6131 rank = c->as ? c->as->rank : 0;
6132 tmp = structure_alloc_comps (c->ts.u.derived, comp, NULL_TREE,
6133 rank, purpose);
6134 gfc_add_expr_to_block (&fnblock, tmp);
6136 break;
6138 case COPY_ALLOC_COMP:
6139 if (c->attr.pointer)
6140 continue;
6142 /* We need source and destination components. */
6143 comp = fold_build3 (COMPONENT_REF, ctype, decl, cdecl, NULL_TREE);
6144 dcmp = fold_build3 (COMPONENT_REF, ctype, dest, cdecl, NULL_TREE);
6145 dcmp = fold_convert (TREE_TYPE (comp), dcmp);
6147 if (c->attr.allocatable && !cmp_has_alloc_comps)
6149 rank = c->as ? c->as->rank : 0;
6150 tmp = gfc_duplicate_allocatable(dcmp, comp, ctype, rank);
6151 gfc_add_expr_to_block (&fnblock, tmp);
6154 if (cmp_has_alloc_comps)
6156 rank = c->as ? c->as->rank : 0;
6157 tmp = fold_convert (TREE_TYPE (dcmp), comp);
6158 gfc_add_modify (&fnblock, dcmp, tmp);
6159 tmp = structure_alloc_comps (c->ts.u.derived, comp, dcmp,
6160 rank, purpose);
6161 gfc_add_expr_to_block (&fnblock, tmp);
6163 break;
6165 default:
6166 gcc_unreachable ();
6167 break;
6171 return gfc_finish_block (&fnblock);
6174 /* Recursively traverse an object of derived type, generating code to
6175 nullify allocatable components. */
6177 tree
6178 gfc_nullify_alloc_comp (gfc_symbol * der_type, tree decl, int rank)
6180 return structure_alloc_comps (der_type, decl, NULL_TREE, rank,
6181 NULLIFY_ALLOC_COMP);
6185 /* Recursively traverse an object of derived type, generating code to
6186 deallocate allocatable components. */
6188 tree
6189 gfc_deallocate_alloc_comp (gfc_symbol * der_type, tree decl, int rank)
6191 return structure_alloc_comps (der_type, decl, NULL_TREE, rank,
6192 DEALLOCATE_ALLOC_COMP);
6196 /* Recursively traverse an object of derived type, generating code to
6197 copy it and its allocatable components. */
6199 tree
6200 gfc_copy_alloc_comp (gfc_symbol * der_type, tree decl, tree dest, int rank)
6202 return structure_alloc_comps (der_type, decl, dest, rank, COPY_ALLOC_COMP);
6206 /* Recursively traverse an object of derived type, generating code to
6207 copy only its allocatable components. */
6209 tree
6210 gfc_copy_only_alloc_comp (gfc_symbol * der_type, tree decl, tree dest, int rank)
6212 return structure_alloc_comps (der_type, decl, dest, rank, COPY_ONLY_ALLOC_COMP);
6216 /* Check for default initializer; sym->value is not enough as it is also
6217 set for EXPR_NULL of allocatables. */
6219 static bool
6220 has_default_initializer (gfc_symbol *der)
6222 gfc_component *c;
6224 gcc_assert (der->attr.flavor == FL_DERIVED);
6225 for (c = der->components; c; c = c->next)
6226 if ((c->ts.type != BT_DERIVED && c->initializer)
6227 || (c->ts.type == BT_DERIVED
6228 && (!c->attr.pointer && has_default_initializer (c->ts.u.derived))))
6229 break;
6231 return c != NULL;
6235 /* NULLIFY an allocatable/pointer array on function entry, free it on exit.
6236 Do likewise, recursively if necessary, with the allocatable components of
6237 derived types. */
6239 tree
6240 gfc_trans_deferred_array (gfc_symbol * sym, tree body)
6242 tree type;
6243 tree tmp;
6244 tree descriptor;
6245 stmtblock_t fnblock;
6246 locus loc;
6247 int rank;
6248 bool sym_has_alloc_comp;
6250 sym_has_alloc_comp = (sym->ts.type == BT_DERIVED)
6251 && sym->ts.u.derived->attr.alloc_comp;
6253 /* Make sure the frontend gets these right. */
6254 if (!(sym->attr.pointer || sym->attr.allocatable || sym_has_alloc_comp))
6255 fatal_error ("Possible frontend bug: Deferred array size without pointer, "
6256 "allocatable attribute or derived type without allocatable "
6257 "components.");
6259 gfc_init_block (&fnblock);
6261 gcc_assert (TREE_CODE (sym->backend_decl) == VAR_DECL
6262 || TREE_CODE (sym->backend_decl) == PARM_DECL);
6264 if (sym->ts.type == BT_CHARACTER
6265 && !INTEGER_CST_P (sym->ts.u.cl->backend_decl))
6267 gfc_conv_string_length (sym->ts.u.cl, NULL, &fnblock);
6268 gfc_trans_vla_type_sizes (sym, &fnblock);
6271 /* Dummy, use associated and result variables don't need anything special. */
6272 if (sym->attr.dummy || sym->attr.use_assoc || sym->attr.result)
6274 gfc_add_expr_to_block (&fnblock, body);
6276 return gfc_finish_block (&fnblock);
6279 gfc_get_backend_locus (&loc);
6280 gfc_set_backend_locus (&sym->declared_at);
6281 descriptor = sym->backend_decl;
6283 /* Although static, derived types with default initializers and
6284 allocatable components must not be nulled wholesale; instead they
6285 are treated component by component. */
6286 if (TREE_STATIC (descriptor) && !sym_has_alloc_comp)
6288 /* SAVEd variables are not freed on exit. */
6289 gfc_trans_static_array_pointer (sym);
6290 return body;
6293 /* Get the descriptor type. */
6294 type = TREE_TYPE (sym->backend_decl);
6296 if (sym_has_alloc_comp && !(sym->attr.pointer || sym->attr.allocatable))
6298 if (!sym->attr.save
6299 && !(TREE_STATIC (sym->backend_decl) && sym->attr.is_main_program))
6301 if (sym->value == NULL || !has_default_initializer (sym->ts.u.derived))
6303 rank = sym->as ? sym->as->rank : 0;
6304 tmp = gfc_nullify_alloc_comp (sym->ts.u.derived, descriptor, rank);
6305 gfc_add_expr_to_block (&fnblock, tmp);
6307 else
6309 tmp = gfc_init_default_dt (sym, NULL, false);
6310 gfc_add_expr_to_block (&fnblock, tmp);
6314 else if (!GFC_DESCRIPTOR_TYPE_P (type))
6316 /* If the backend_decl is not a descriptor, we must have a pointer
6317 to one. */
6318 descriptor = build_fold_indirect_ref_loc (input_location,
6319 sym->backend_decl);
6320 type = TREE_TYPE (descriptor);
6323 /* NULLIFY the data pointer. */
6324 if (GFC_DESCRIPTOR_TYPE_P (type) && !sym->attr.save)
6325 gfc_conv_descriptor_data_set (&fnblock, descriptor, null_pointer_node);
6327 gfc_add_expr_to_block (&fnblock, body);
6329 gfc_set_backend_locus (&loc);
6331 /* Allocatable arrays need to be freed when they go out of scope.
6332 The allocatable components of pointers must not be touched. */
6333 if (sym_has_alloc_comp && !(sym->attr.function || sym->attr.result)
6334 && !sym->attr.pointer && !sym->attr.save)
6336 int rank;
6337 rank = sym->as ? sym->as->rank : 0;
6338 tmp = gfc_deallocate_alloc_comp (sym->ts.u.derived, descriptor, rank);
6339 gfc_add_expr_to_block (&fnblock, tmp);
6342 if (sym->attr.allocatable && sym->attr.dimension
6343 && !sym->attr.save && !sym->attr.result)
6345 tmp = gfc_trans_dealloc_allocated (sym->backend_decl);
6346 gfc_add_expr_to_block (&fnblock, tmp);
6349 return gfc_finish_block (&fnblock);
6352 /************ Expression Walking Functions ******************/
6354 /* Walk a variable reference.
6356 Possible extension - multiple component subscripts.
6357 x(:,:) = foo%a(:)%b(:)
6358 Transforms to
6359 forall (i=..., j=...)
6360 x(i,j) = foo%a(j)%b(i)
6361 end forall
6362 This adds a fair amount of complexity because you need to deal with more
6363 than one ref. Maybe handle in a similar manner to vector subscripts.
6364 Maybe not worth the effort. */
6367 static gfc_ss *
6368 gfc_walk_variable_expr (gfc_ss * ss, gfc_expr * expr)
6370 gfc_ref *ref;
6371 gfc_array_ref *ar;
6372 gfc_ss *newss;
6373 int n;
6375 for (ref = expr->ref; ref; ref = ref->next)
6376 if (ref->type == REF_ARRAY && ref->u.ar.type != AR_ELEMENT)
6377 break;
6379 for (; ref; ref = ref->next)
6381 if (ref->type == REF_SUBSTRING)
6383 newss = gfc_get_ss ();
6384 newss->type = GFC_SS_SCALAR;
6385 newss->expr = ref->u.ss.start;
6386 newss->next = ss;
6387 ss = newss;
6389 newss = gfc_get_ss ();
6390 newss->type = GFC_SS_SCALAR;
6391 newss->expr = ref->u.ss.end;
6392 newss->next = ss;
6393 ss = newss;
6396 /* We're only interested in array sections from now on. */
6397 if (ref->type != REF_ARRAY)
6398 continue;
6400 ar = &ref->u.ar;
6402 if (ar->as->rank == 0)
6404 /* Scalar coarray. */
6405 continue;
6408 switch (ar->type)
6410 case AR_ELEMENT:
6411 for (n = 0; n < ar->dimen; n++)
6413 newss = gfc_get_ss ();
6414 newss->type = GFC_SS_SCALAR;
6415 newss->expr = ar->start[n];
6416 newss->next = ss;
6417 ss = newss;
6419 break;
6421 case AR_FULL:
6422 newss = gfc_get_ss ();
6423 newss->type = GFC_SS_SECTION;
6424 newss->expr = expr;
6425 newss->next = ss;
6426 newss->data.info.dimen = ar->as->rank;
6427 newss->data.info.ref = ref;
6429 /* Make sure array is the same as array(:,:), this way
6430 we don't need to special case all the time. */
6431 ar->dimen = ar->as->rank;
6432 for (n = 0; n < ar->dimen; n++)
6434 newss->data.info.dim[n] = n;
6435 ar->dimen_type[n] = DIMEN_RANGE;
6437 gcc_assert (ar->start[n] == NULL);
6438 gcc_assert (ar->end[n] == NULL);
6439 gcc_assert (ar->stride[n] == NULL);
6441 ss = newss;
6442 break;
6444 case AR_SECTION:
6445 newss = gfc_get_ss ();
6446 newss->type = GFC_SS_SECTION;
6447 newss->expr = expr;
6448 newss->next = ss;
6449 newss->data.info.dimen = 0;
6450 newss->data.info.ref = ref;
6452 /* We add SS chains for all the subscripts in the section. */
6453 for (n = 0; n < ar->dimen; n++)
6455 gfc_ss *indexss;
6457 switch (ar->dimen_type[n])
6459 case DIMEN_ELEMENT:
6460 /* Add SS for elemental (scalar) subscripts. */
6461 gcc_assert (ar->start[n]);
6462 indexss = gfc_get_ss ();
6463 indexss->type = GFC_SS_SCALAR;
6464 indexss->expr = ar->start[n];
6465 indexss->next = gfc_ss_terminator;
6466 indexss->loop_chain = gfc_ss_terminator;
6467 newss->data.info.subscript[n] = indexss;
6468 break;
6470 case DIMEN_RANGE:
6471 /* We don't add anything for sections, just remember this
6472 dimension for later. */
6473 newss->data.info.dim[newss->data.info.dimen] = n;
6474 newss->data.info.dimen++;
6475 break;
6477 case DIMEN_VECTOR:
6478 /* Create a GFC_SS_VECTOR index in which we can store
6479 the vector's descriptor. */
6480 indexss = gfc_get_ss ();
6481 indexss->type = GFC_SS_VECTOR;
6482 indexss->expr = ar->start[n];
6483 indexss->next = gfc_ss_terminator;
6484 indexss->loop_chain = gfc_ss_terminator;
6485 newss->data.info.subscript[n] = indexss;
6486 newss->data.info.dim[newss->data.info.dimen] = n;
6487 newss->data.info.dimen++;
6488 break;
6490 default:
6491 /* We should know what sort of section it is by now. */
6492 gcc_unreachable ();
6495 /* We should have at least one non-elemental dimension. */
6496 gcc_assert (newss->data.info.dimen > 0);
6497 ss = newss;
6498 break;
6500 default:
6501 /* We should know what sort of section it is by now. */
6502 gcc_unreachable ();
6506 return ss;
6510 /* Walk an expression operator. If only one operand of a binary expression is
6511 scalar, we must also add the scalar term to the SS chain. */
6513 static gfc_ss *
6514 gfc_walk_op_expr (gfc_ss * ss, gfc_expr * expr)
6516 gfc_ss *head;
6517 gfc_ss *head2;
6518 gfc_ss *newss;
6520 head = gfc_walk_subexpr (ss, expr->value.op.op1);
6521 if (expr->value.op.op2 == NULL)
6522 head2 = head;
6523 else
6524 head2 = gfc_walk_subexpr (head, expr->value.op.op2);
6526 /* All operands are scalar. Pass back and let the caller deal with it. */
6527 if (head2 == ss)
6528 return head2;
6530 /* All operands require scalarization. */
6531 if (head != ss && (expr->value.op.op2 == NULL || head2 != head))
6532 return head2;
6534 /* One of the operands needs scalarization, the other is scalar.
6535 Create a gfc_ss for the scalar expression. */
6536 newss = gfc_get_ss ();
6537 newss->type = GFC_SS_SCALAR;
6538 if (head == ss)
6540 /* First operand is scalar. We build the chain in reverse order, so
6541 add the scalar SS after the second operand. */
6542 head = head2;
6543 while (head && head->next != ss)
6544 head = head->next;
6545 /* Check we haven't somehow broken the chain. */
6546 gcc_assert (head);
6547 newss->next = ss;
6548 head->next = newss;
6549 newss->expr = expr->value.op.op1;
6551 else /* head2 == head */
6553 gcc_assert (head2 == head);
6554 /* Second operand is scalar. */
6555 newss->next = head2;
6556 head2 = newss;
6557 newss->expr = expr->value.op.op2;
6560 return head2;
6564 /* Reverse a SS chain. */
6566 gfc_ss *
6567 gfc_reverse_ss (gfc_ss * ss)
6569 gfc_ss *next;
6570 gfc_ss *head;
6572 gcc_assert (ss != NULL);
6574 head = gfc_ss_terminator;
6575 while (ss != gfc_ss_terminator)
6577 next = ss->next;
6578 /* Check we didn't somehow break the chain. */
6579 gcc_assert (next != NULL);
6580 ss->next = head;
6581 head = ss;
6582 ss = next;
6585 return (head);
6589 /* Walk the arguments of an elemental function. */
6591 gfc_ss *
6592 gfc_walk_elemental_function_args (gfc_ss * ss, gfc_actual_arglist *arg,
6593 gfc_ss_type type)
6595 int scalar;
6596 gfc_ss *head;
6597 gfc_ss *tail;
6598 gfc_ss *newss;
6600 head = gfc_ss_terminator;
6601 tail = NULL;
6602 scalar = 1;
6603 for (; arg; arg = arg->next)
6605 if (!arg->expr)
6606 continue;
6608 newss = gfc_walk_subexpr (head, arg->expr);
6609 if (newss == head)
6611 /* Scalar argument. */
6612 newss = gfc_get_ss ();
6613 newss->type = type;
6614 newss->expr = arg->expr;
6615 newss->next = head;
6617 else
6618 scalar = 0;
6620 head = newss;
6621 if (!tail)
6623 tail = head;
6624 while (tail->next != gfc_ss_terminator)
6625 tail = tail->next;
6629 if (scalar)
6631 /* If all the arguments are scalar we don't need the argument SS. */
6632 gfc_free_ss_chain (head);
6633 /* Pass it back. */
6634 return ss;
6637 /* Add it onto the existing chain. */
6638 tail->next = ss;
6639 return head;
6643 /* Walk a function call. Scalar functions are passed back, and taken out of
6644 scalarization loops. For elemental functions we walk their arguments.
6645 The result of functions returning arrays is stored in a temporary outside
6646 the loop, so that the function is only called once. Hence we do not need
6647 to walk their arguments. */
6649 static gfc_ss *
6650 gfc_walk_function_expr (gfc_ss * ss, gfc_expr * expr)
6652 gfc_ss *newss;
6653 gfc_intrinsic_sym *isym;
6654 gfc_symbol *sym;
6655 gfc_component *comp = NULL;
6657 isym = expr->value.function.isym;
6659 /* Handle intrinsic functions separately. */
6660 if (isym)
6661 return gfc_walk_intrinsic_function (ss, expr, isym);
6663 sym = expr->value.function.esym;
6664 if (!sym)
6665 sym = expr->symtree->n.sym;
6667 /* A function that returns arrays. */
6668 gfc_is_proc_ptr_comp (expr, &comp);
6669 if ((!comp && gfc_return_by_reference (sym) && sym->result->attr.dimension)
6670 || (comp && comp->attr.dimension))
6672 newss = gfc_get_ss ();
6673 newss->type = GFC_SS_FUNCTION;
6674 newss->expr = expr;
6675 newss->next = ss;
6676 newss->data.info.dimen = expr->rank;
6677 return newss;
6680 /* Walk the parameters of an elemental function. For now we always pass
6681 by reference. */
6682 if (sym->attr.elemental)
6683 return gfc_walk_elemental_function_args (ss, expr->value.function.actual,
6684 GFC_SS_REFERENCE);
6686 /* Scalar functions are OK as these are evaluated outside the scalarization
6687 loop. Pass back and let the caller deal with it. */
6688 return ss;
6692 /* An array temporary is constructed for array constructors. */
6694 static gfc_ss *
6695 gfc_walk_array_constructor (gfc_ss * ss, gfc_expr * expr)
6697 gfc_ss *newss;
6698 int n;
6700 newss = gfc_get_ss ();
6701 newss->type = GFC_SS_CONSTRUCTOR;
6702 newss->expr = expr;
6703 newss->next = ss;
6704 newss->data.info.dimen = expr->rank;
6705 for (n = 0; n < expr->rank; n++)
6706 newss->data.info.dim[n] = n;
6708 return newss;
6712 /* Walk an expression. Add walked expressions to the head of the SS chain.
6713 A wholly scalar expression will not be added. */
6715 static gfc_ss *
6716 gfc_walk_subexpr (gfc_ss * ss, gfc_expr * expr)
6718 gfc_ss *head;
6720 switch (expr->expr_type)
6722 case EXPR_VARIABLE:
6723 head = gfc_walk_variable_expr (ss, expr);
6724 return head;
6726 case EXPR_OP:
6727 head = gfc_walk_op_expr (ss, expr);
6728 return head;
6730 case EXPR_FUNCTION:
6731 head = gfc_walk_function_expr (ss, expr);
6732 return head;
6734 case EXPR_CONSTANT:
6735 case EXPR_NULL:
6736 case EXPR_STRUCTURE:
6737 /* Pass back and let the caller deal with it. */
6738 break;
6740 case EXPR_ARRAY:
6741 head = gfc_walk_array_constructor (ss, expr);
6742 return head;
6744 case EXPR_SUBSTRING:
6745 /* Pass back and let the caller deal with it. */
6746 break;
6748 default:
6749 internal_error ("bad expression type during walk (%d)",
6750 expr->expr_type);
6752 return ss;
6756 /* Entry point for expression walking.
6757 A return value equal to the passed chain means this is
6758 a scalar expression. It is up to the caller to take whatever action is
6759 necessary to translate these. */
6761 gfc_ss *
6762 gfc_walk_expr (gfc_expr * expr)
6764 gfc_ss *res;
6766 res = gfc_walk_subexpr (gfc_ss_terminator, expr);
6767 return gfc_reverse_ss (res);