]> git.ipfire.org Git - thirdparty/gcc.git/blob - gcc/fortran/trans-array.c
63a1ea08fbcf866d19fbc93df797f72005297ec9
[thirdparty/gcc.git] / gcc / fortran / trans-array.c
1 /* Array translation routines
2 Copyright (C) 2002, 2003, 2004, 2005, 2006, 2007
3 Free Software Foundation, Inc.
4 Contributed by Paul Brook <paul@nowt.org>
5 and Steven Bosscher <s.bosscher@student.tudelft.nl>
6
7 This file is part of GCC.
8
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 2, or (at your option) any later
12 version.
13
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.
18
19 You should have received a copy of the GNU General Public License
20 along with GCC; see the file COPYING. If not, write to the Free
21 Software Foundation, 51 Franklin Street, Fifth Floor, Boston, MA
22 02110-1301, USA. */
23
24 /* trans-array.c-- Various array related code, including scalarization,
25 allocation, initialization and other support routines. */
26
27 /* How the scalarizer works.
28 In gfortran, array expressions use the same core routines as scalar
29 expressions.
30 First, a Scalarization State (SS) chain is built. This is done by walking
31 the expression tree, and building a linear list of the terms in the
32 expression. As the tree is walked, scalar subexpressions are translated.
33
34 The scalarization parameters are stored in a gfc_loopinfo structure.
35 First the start and stride of each term is calculated by
36 gfc_conv_ss_startstride. During this process the expressions for the array
37 descriptors and data pointers are also translated.
38
39 If the expression is an assignment, we must then resolve any dependencies.
40 In fortran all the rhs values of an assignment must be evaluated before
41 any assignments take place. This can require a temporary array to store the
42 values. We also require a temporary when we are passing array expressions
43 or vector subecripts as procedure parameters.
44
45 Array sections are passed without copying to a temporary. These use the
46 scalarizer to determine the shape of the section. The flag
47 loop->array_parameter tells the scalarizer that the actual values and loop
48 variables will not be required.
49
50 The function gfc_conv_loop_setup generates the scalarization setup code.
51 It determines the range of the scalarizing loop variables. If a temporary
52 is required, this is created and initialized. Code for scalar expressions
53 taken outside the loop is also generated at this time. Next the offset and
54 scaling required to translate from loop variables to array indices for each
55 term is calculated.
56
57 A call to gfc_start_scalarized_body marks the start of the scalarized
58 expression. This creates a scope and declares the loop variables. Before
59 calling this gfc_make_ss_chain_used must be used to indicate which terms
60 will be used inside this loop.
61
62 The scalar gfc_conv_* functions are then used to build the main body of the
63 scalarization loop. Scalarization loop variables and precalculated scalar
64 values are automatically substituted. Note that gfc_advance_se_ss_chain
65 must be used, rather than changing the se->ss directly.
66
67 For assignment expressions requiring a temporary two sub loops are
68 generated. The first stores the result of the expression in the temporary,
69 the second copies it to the result. A call to
70 gfc_trans_scalarized_loop_boundary marks the end of the main loop code and
71 the start of the copying loop. The temporary may be less than full rank.
72
73 Finally gfc_trans_scalarizing_loops is called to generate the implicit do
74 loops. The loops are added to the pre chain of the loopinfo. The post
75 chain may still contain cleanup code.
76
77 After the loop code has been added into its parent scope gfc_cleanup_loop
78 is called to free all the SS allocated by the scalarizer. */
79
80 #include "config.h"
81 #include "system.h"
82 #include "coretypes.h"
83 #include "tree.h"
84 #include "tree-gimple.h"
85 #include "ggc.h"
86 #include "toplev.h"
87 #include "real.h"
88 #include "flags.h"
89 #include "gfortran.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"
96
97 static gfc_ss *gfc_walk_subexpr (gfc_ss *, gfc_expr *);
98 static bool gfc_get_array_constructor_size (mpz_t *, gfc_constructor *);
99
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;
103
104
105 static tree
106 gfc_array_dataptr_type (tree desc)
107 {
108 return (GFC_TYPE_ARRAY_DATAPTR_TYPE (TREE_TYPE (desc)));
109 }
110
111
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.
117
118 To understand these magic numbers, look at the comments
119 before gfc_build_array_type() in trans-types.c.
120
121 The code within these defines should be the only code which knows the format
122 of an array descriptor.
123
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.
127
128 Don't forget to #undef these! */
129
130 #define DATA_FIELD 0
131 #define OFFSET_FIELD 1
132 #define DTYPE_FIELD 2
133 #define DIMENSION_FIELD 3
134
135 #define STRIDE_SUBFIELD 0
136 #define LBOUND_SUBFIELD 1
137 #define UBOUND_SUBFIELD 2
138
139 /* This provides READ-ONLY access to the data field. The field itself
140 doesn't have the proper type. */
141
142 tree
143 gfc_conv_descriptor_data_get (tree desc)
144 {
145 tree field, type, t;
146
147 type = TREE_TYPE (desc);
148 gcc_assert (GFC_DESCRIPTOR_TYPE_P (type));
149
150 field = TYPE_FIELDS (type);
151 gcc_assert (DATA_FIELD == 0);
152
153 t = build3 (COMPONENT_REF, TREE_TYPE (field), desc, field, NULL_TREE);
154 t = fold_convert (GFC_TYPE_ARRAY_DATAPTR_TYPE (type), t);
155
156 return t;
157 }
158
159 /* This provides WRITE access to the data field.
160
161 TUPLES_P is true if we are generating tuples.
162
163 This function gets called through the following macros:
164 gfc_conv_descriptor_data_set
165 gfc_conv_descriptor_data_set_tuples. */
166
167 void
168 gfc_conv_descriptor_data_set_internal (stmtblock_t *block,
169 tree desc, tree value,
170 bool tuples_p)
171 {
172 tree field, type, t;
173
174 type = TREE_TYPE (desc);
175 gcc_assert (GFC_DESCRIPTOR_TYPE_P (type));
176
177 field = TYPE_FIELDS (type);
178 gcc_assert (DATA_FIELD == 0);
179
180 t = build3 (COMPONENT_REF, TREE_TYPE (field), desc, field, NULL_TREE);
181 gfc_add_modify (block, t, fold_convert (TREE_TYPE (field), value), tuples_p);
182 }
183
184
185 /* This provides address access to the data field. This should only be
186 used by array allocation, passing this on to the runtime. */
187
188 tree
189 gfc_conv_descriptor_data_addr (tree desc)
190 {
191 tree field, type, t;
192
193 type = TREE_TYPE (desc);
194 gcc_assert (GFC_DESCRIPTOR_TYPE_P (type));
195
196 field = TYPE_FIELDS (type);
197 gcc_assert (DATA_FIELD == 0);
198
199 t = build3 (COMPONENT_REF, TREE_TYPE (field), desc, field, NULL_TREE);
200 return build_fold_addr_expr (t);
201 }
202
203 tree
204 gfc_conv_descriptor_offset (tree desc)
205 {
206 tree type;
207 tree field;
208
209 type = TREE_TYPE (desc);
210 gcc_assert (GFC_DESCRIPTOR_TYPE_P (type));
211
212 field = gfc_advance_chain (TYPE_FIELDS (type), OFFSET_FIELD);
213 gcc_assert (field != NULL_TREE && TREE_TYPE (field) == gfc_array_index_type);
214
215 return build3 (COMPONENT_REF, TREE_TYPE (field), desc, field, NULL_TREE);
216 }
217
218 tree
219 gfc_conv_descriptor_dtype (tree desc)
220 {
221 tree field;
222 tree type;
223
224 type = TREE_TYPE (desc);
225 gcc_assert (GFC_DESCRIPTOR_TYPE_P (type));
226
227 field = gfc_advance_chain (TYPE_FIELDS (type), DTYPE_FIELD);
228 gcc_assert (field != NULL_TREE && TREE_TYPE (field) == gfc_array_index_type);
229
230 return build3 (COMPONENT_REF, TREE_TYPE (field), desc, field, NULL_TREE);
231 }
232
233 static tree
234 gfc_conv_descriptor_dimension (tree desc, tree dim)
235 {
236 tree field;
237 tree type;
238 tree tmp;
239
240 type = TREE_TYPE (desc);
241 gcc_assert (GFC_DESCRIPTOR_TYPE_P (type));
242
243 field = gfc_advance_chain (TYPE_FIELDS (type), DIMENSION_FIELD);
244 gcc_assert (field != NULL_TREE
245 && TREE_CODE (TREE_TYPE (field)) == ARRAY_TYPE
246 && TREE_CODE (TREE_TYPE (TREE_TYPE (field))) == RECORD_TYPE);
247
248 tmp = build3 (COMPONENT_REF, TREE_TYPE (field), desc, field, NULL_TREE);
249 tmp = gfc_build_array_ref (tmp, dim);
250 return tmp;
251 }
252
253 tree
254 gfc_conv_descriptor_stride (tree desc, tree dim)
255 {
256 tree tmp;
257 tree field;
258
259 tmp = gfc_conv_descriptor_dimension (desc, dim);
260 field = TYPE_FIELDS (TREE_TYPE (tmp));
261 field = gfc_advance_chain (field, STRIDE_SUBFIELD);
262 gcc_assert (field != NULL_TREE && TREE_TYPE (field) == gfc_array_index_type);
263
264 tmp = build3 (COMPONENT_REF, TREE_TYPE (field), tmp, field, NULL_TREE);
265 return tmp;
266 }
267
268 tree
269 gfc_conv_descriptor_lbound (tree desc, tree dim)
270 {
271 tree tmp;
272 tree field;
273
274 tmp = gfc_conv_descriptor_dimension (desc, dim);
275 field = TYPE_FIELDS (TREE_TYPE (tmp));
276 field = gfc_advance_chain (field, LBOUND_SUBFIELD);
277 gcc_assert (field != NULL_TREE && TREE_TYPE (field) == gfc_array_index_type);
278
279 tmp = build3 (COMPONENT_REF, TREE_TYPE (field), tmp, field, NULL_TREE);
280 return tmp;
281 }
282
283 tree
284 gfc_conv_descriptor_ubound (tree desc, tree dim)
285 {
286 tree tmp;
287 tree field;
288
289 tmp = gfc_conv_descriptor_dimension (desc, dim);
290 field = TYPE_FIELDS (TREE_TYPE (tmp));
291 field = gfc_advance_chain (field, UBOUND_SUBFIELD);
292 gcc_assert (field != NULL_TREE && TREE_TYPE (field) == gfc_array_index_type);
293
294 tmp = build3 (COMPONENT_REF, TREE_TYPE (field), tmp, field, NULL_TREE);
295 return tmp;
296 }
297
298
299 /* Build a null array descriptor constructor. */
300
301 tree
302 gfc_build_null_descriptor (tree type)
303 {
304 tree field;
305 tree tmp;
306
307 gcc_assert (GFC_DESCRIPTOR_TYPE_P (type));
308 gcc_assert (DATA_FIELD == 0);
309 field = TYPE_FIELDS (type);
310
311 /* Set a NULL data pointer. */
312 tmp = build_constructor_single (type, field, null_pointer_node);
313 TREE_CONSTANT (tmp) = 1;
314 TREE_INVARIANT (tmp) = 1;
315 /* All other fields are ignored. */
316
317 return tmp;
318 }
319
320
321 /* Cleanup those #defines. */
322
323 #undef DATA_FIELD
324 #undef OFFSET_FIELD
325 #undef DTYPE_FIELD
326 #undef DIMENSION_FIELD
327 #undef STRIDE_SUBFIELD
328 #undef LBOUND_SUBFIELD
329 #undef UBOUND_SUBFIELD
330
331
332 /* Mark a SS chain as used. Flags specifies in which loops the SS is used.
333 flags & 1 = Main loop body.
334 flags & 2 = temp copy loop. */
335
336 void
337 gfc_mark_ss_chain_used (gfc_ss * ss, unsigned flags)
338 {
339 for (; ss != gfc_ss_terminator; ss = ss->next)
340 ss->useflags = flags;
341 }
342
343 static void gfc_free_ss (gfc_ss *);
344
345
346 /* Free a gfc_ss chain. */
347
348 static void
349 gfc_free_ss_chain (gfc_ss * ss)
350 {
351 gfc_ss *next;
352
353 while (ss != gfc_ss_terminator)
354 {
355 gcc_assert (ss != NULL);
356 next = ss->next;
357 gfc_free_ss (ss);
358 ss = next;
359 }
360 }
361
362
363 /* Free a SS. */
364
365 static void
366 gfc_free_ss (gfc_ss * ss)
367 {
368 int n;
369
370 switch (ss->type)
371 {
372 case GFC_SS_SECTION:
373 for (n = 0; n < GFC_MAX_DIMENSIONS; n++)
374 {
375 if (ss->data.info.subscript[n])
376 gfc_free_ss_chain (ss->data.info.subscript[n]);
377 }
378 break;
379
380 default:
381 break;
382 }
383
384 gfc_free (ss);
385 }
386
387
388 /* Free all the SS associated with a loop. */
389
390 void
391 gfc_cleanup_loop (gfc_loopinfo * loop)
392 {
393 gfc_ss *ss;
394 gfc_ss *next;
395
396 ss = loop->ss;
397 while (ss != gfc_ss_terminator)
398 {
399 gcc_assert (ss != NULL);
400 next = ss->loop_chain;
401 gfc_free_ss (ss);
402 ss = next;
403 }
404 }
405
406
407 /* Associate a SS chain with a loop. */
408
409 void
410 gfc_add_ss_to_loop (gfc_loopinfo * loop, gfc_ss * head)
411 {
412 gfc_ss *ss;
413
414 if (head == gfc_ss_terminator)
415 return;
416
417 ss = head;
418 for (; ss && ss != gfc_ss_terminator; ss = ss->next)
419 {
420 if (ss->next == gfc_ss_terminator)
421 ss->loop_chain = loop->ss;
422 else
423 ss->loop_chain = ss->next;
424 }
425 gcc_assert (ss == gfc_ss_terminator);
426 loop->ss = head;
427 }
428
429
430 /* Generate an initializer for a static pointer or allocatable array. */
431
432 void
433 gfc_trans_static_array_pointer (gfc_symbol * sym)
434 {
435 tree type;
436
437 gcc_assert (TREE_STATIC (sym->backend_decl));
438 /* Just zero the data member. */
439 type = TREE_TYPE (sym->backend_decl);
440 DECL_INITIAL (sym->backend_decl) = gfc_build_null_descriptor (type);
441 }
442
443
444 /* If the bounds of SE's loop have not yet been set, see if they can be
445 determined from array spec AS, which is the array spec of a called
446 function. MAPPING maps the callee's dummy arguments to the values
447 that the caller is passing. Add any initialization and finalization
448 code to SE. */
449
450 void
451 gfc_set_loop_bounds_from_array_spec (gfc_interface_mapping * mapping,
452 gfc_se * se, gfc_array_spec * as)
453 {
454 int n, dim;
455 gfc_se tmpse;
456 tree lower;
457 tree upper;
458 tree tmp;
459
460 if (as && as->type == AS_EXPLICIT)
461 for (dim = 0; dim < se->loop->dimen; dim++)
462 {
463 n = se->loop->order[dim];
464 if (se->loop->to[n] == NULL_TREE)
465 {
466 /* Evaluate the lower bound. */
467 gfc_init_se (&tmpse, NULL);
468 gfc_apply_interface_mapping (mapping, &tmpse, as->lower[dim]);
469 gfc_add_block_to_block (&se->pre, &tmpse.pre);
470 gfc_add_block_to_block (&se->post, &tmpse.post);
471 lower = tmpse.expr;
472
473 /* ...and the upper bound. */
474 gfc_init_se (&tmpse, NULL);
475 gfc_apply_interface_mapping (mapping, &tmpse, as->upper[dim]);
476 gfc_add_block_to_block (&se->pre, &tmpse.pre);
477 gfc_add_block_to_block (&se->post, &tmpse.post);
478 upper = tmpse.expr;
479
480 /* Set the upper bound of the loop to UPPER - LOWER. */
481 tmp = fold_build2 (MINUS_EXPR, gfc_array_index_type, upper, lower);
482 tmp = gfc_evaluate_now (tmp, &se->pre);
483 se->loop->to[n] = tmp;
484 }
485 }
486 }
487
488
489 /* Generate code to allocate an array temporary, or create a variable to
490 hold the data. If size is NULL, zero the descriptor so that the
491 callee will allocate the array. If DEALLOC is true, also generate code to
492 free the array afterwards.
493
494 Initialization code is added to PRE and finalization code to POST.
495 DYNAMIC is true if the caller may want to extend the array later
496 using realloc. This prevents us from putting the array on the stack. */
497
498 static void
499 gfc_trans_allocate_array_storage (stmtblock_t * pre, stmtblock_t * post,
500 gfc_ss_info * info, tree size, tree nelem,
501 bool dynamic, bool dealloc)
502 {
503 tree tmp;
504 tree desc;
505 bool onstack;
506
507 desc = info->descriptor;
508 info->offset = gfc_index_zero_node;
509 if (size == NULL_TREE || integer_zerop (size))
510 {
511 /* A callee allocated array. */
512 gfc_conv_descriptor_data_set (pre, desc, null_pointer_node);
513 onstack = FALSE;
514 }
515 else
516 {
517 /* Allocate the temporary. */
518 onstack = !dynamic && gfc_can_put_var_on_stack (size);
519
520 if (onstack)
521 {
522 /* Make a temporary variable to hold the data. */
523 tmp = fold_build2 (MINUS_EXPR, TREE_TYPE (nelem), nelem,
524 gfc_index_one_node);
525 tmp = build_range_type (gfc_array_index_type, gfc_index_zero_node,
526 tmp);
527 tmp = build_array_type (gfc_get_element_type (TREE_TYPE (desc)),
528 tmp);
529 tmp = gfc_create_var (tmp, "A");
530 tmp = build_fold_addr_expr (tmp);
531 gfc_conv_descriptor_data_set (pre, desc, tmp);
532 }
533 else
534 {
535 /* Allocate memory to hold the data. */
536 tmp = gfc_call_malloc (pre, NULL, size);
537 tmp = gfc_evaluate_now (tmp, pre);
538 gfc_conv_descriptor_data_set (pre, desc, tmp);
539 }
540 }
541 info->data = gfc_conv_descriptor_data_get (desc);
542
543 /* The offset is zero because we create temporaries with a zero
544 lower bound. */
545 tmp = gfc_conv_descriptor_offset (desc);
546 gfc_add_modify_expr (pre, tmp, gfc_index_zero_node);
547
548 if (dealloc && !onstack)
549 {
550 /* Free the temporary. */
551 tmp = gfc_conv_descriptor_data_get (desc);
552 tmp = gfc_call_free (fold_convert (pvoid_type_node, tmp));
553 gfc_add_expr_to_block (post, tmp);
554 }
555 }
556
557
558 /* Generate code to create and initialize the descriptor for a temporary
559 array. This is used for both temporaries needed by the scalarizer, and
560 functions returning arrays. Adjusts the loop variables to be
561 zero-based, and calculates the loop bounds for callee allocated arrays.
562 Allocate the array unless it's callee allocated (we have a callee
563 allocated array if 'callee_alloc' is true, or if loop->to[n] is
564 NULL_TREE for any n). Also fills in the descriptor, data and offset
565 fields of info if known. Returns the size of the array, or NULL for a
566 callee allocated array.
567
568 PRE, POST, DYNAMIC and DEALLOC are as for gfc_trans_allocate_array_storage.
569 */
570
571 tree
572 gfc_trans_create_temp_array (stmtblock_t * pre, stmtblock_t * post,
573 gfc_loopinfo * loop, gfc_ss_info * info,
574 tree eltype, bool dynamic, bool dealloc,
575 bool callee_alloc)
576 {
577 tree type;
578 tree desc;
579 tree tmp;
580 tree size;
581 tree nelem;
582 tree cond;
583 tree or_expr;
584 int n;
585 int dim;
586
587 gcc_assert (info->dimen > 0);
588 /* Set the lower bound to zero. */
589 for (dim = 0; dim < info->dimen; dim++)
590 {
591 n = loop->order[dim];
592 if (n < loop->temp_dim)
593 gcc_assert (integer_zerop (loop->from[n]));
594 else
595 {
596 /* Callee allocated arrays may not have a known bound yet. */
597 if (loop->to[n])
598 loop->to[n] = fold_build2 (MINUS_EXPR, gfc_array_index_type,
599 loop->to[n], loop->from[n]);
600 loop->from[n] = gfc_index_zero_node;
601 }
602
603 info->delta[dim] = gfc_index_zero_node;
604 info->start[dim] = gfc_index_zero_node;
605 info->end[dim] = gfc_index_zero_node;
606 info->stride[dim] = gfc_index_one_node;
607 info->dim[dim] = dim;
608 }
609
610 /* Initialize the descriptor. */
611 type =
612 gfc_get_array_type_bounds (eltype, info->dimen, loop->from, loop->to, 1);
613 desc = gfc_create_var (type, "atmp");
614 GFC_DECL_PACKED_ARRAY (desc) = 1;
615
616 info->descriptor = desc;
617 size = gfc_index_one_node;
618
619 /* Fill in the array dtype. */
620 tmp = gfc_conv_descriptor_dtype (desc);
621 gfc_add_modify_expr (pre, tmp, gfc_get_dtype (TREE_TYPE (desc)));
622
623 /*
624 Fill in the bounds and stride. This is a packed array, so:
625
626 size = 1;
627 for (n = 0; n < rank; n++)
628 {
629 stride[n] = size
630 delta = ubound[n] + 1 - lbound[n];
631 size = size * delta;
632 }
633 size = size * sizeof(element);
634 */
635
636 or_expr = NULL_TREE;
637
638 for (n = 0; n < info->dimen; n++)
639 {
640 if (loop->to[n] == NULL_TREE)
641 {
642 /* For a callee allocated array express the loop bounds in terms
643 of the descriptor fields. */
644 tmp = build2 (MINUS_EXPR, gfc_array_index_type,
645 gfc_conv_descriptor_ubound (desc, gfc_rank_cst[n]),
646 gfc_conv_descriptor_lbound (desc, gfc_rank_cst[n]));
647 loop->to[n] = tmp;
648 size = NULL_TREE;
649 continue;
650 }
651
652 /* Store the stride and bound components in the descriptor. */
653 tmp = gfc_conv_descriptor_stride (desc, gfc_rank_cst[n]);
654 gfc_add_modify_expr (pre, tmp, size);
655
656 tmp = gfc_conv_descriptor_lbound (desc, gfc_rank_cst[n]);
657 gfc_add_modify_expr (pre, tmp, gfc_index_zero_node);
658
659 tmp = gfc_conv_descriptor_ubound (desc, gfc_rank_cst[n]);
660 gfc_add_modify_expr (pre, tmp, loop->to[n]);
661
662 tmp = fold_build2 (PLUS_EXPR, gfc_array_index_type,
663 loop->to[n], gfc_index_one_node);
664
665 /* Check whether the size for this dimension is negative. */
666 cond = fold_build2 (LE_EXPR, boolean_type_node, tmp,
667 gfc_index_zero_node);
668 cond = gfc_evaluate_now (cond, pre);
669
670 if (n == 0)
671 or_expr = cond;
672 else
673 or_expr = fold_build2 (TRUTH_OR_EXPR, boolean_type_node, or_expr, cond);
674
675 size = fold_build2 (MULT_EXPR, gfc_array_index_type, size, tmp);
676 size = gfc_evaluate_now (size, pre);
677 }
678
679 /* Get the size of the array. */
680
681 if (size && !callee_alloc)
682 {
683 /* If or_expr is true, then the extent in at least one
684 dimension is zero and the size is set to zero. */
685 size = fold_build3 (COND_EXPR, gfc_array_index_type,
686 or_expr, gfc_index_zero_node, size);
687
688 nelem = size;
689 size = fold_build2 (MULT_EXPR, gfc_array_index_type, size,
690 fold_convert (gfc_array_index_type,
691 TYPE_SIZE_UNIT (gfc_get_element_type (type))));
692 }
693 else
694 {
695 nelem = size;
696 size = NULL_TREE;
697 }
698
699 gfc_trans_allocate_array_storage (pre, post, info, size, nelem, dynamic,
700 dealloc);
701
702 if (info->dimen > loop->temp_dim)
703 loop->temp_dim = info->dimen;
704
705 return size;
706 }
707
708
709 /* Generate code to transpose array EXPR by creating a new descriptor
710 in which the dimension specifications have been reversed. */
711
712 void
713 gfc_conv_array_transpose (gfc_se * se, gfc_expr * expr)
714 {
715 tree dest, src, dest_index, src_index;
716 gfc_loopinfo *loop;
717 gfc_ss_info *dest_info, *src_info;
718 gfc_ss *dest_ss, *src_ss;
719 gfc_se src_se;
720 int n;
721
722 loop = se->loop;
723
724 src_ss = gfc_walk_expr (expr);
725 dest_ss = se->ss;
726
727 src_info = &src_ss->data.info;
728 dest_info = &dest_ss->data.info;
729 gcc_assert (dest_info->dimen == 2);
730 gcc_assert (src_info->dimen == 2);
731
732 /* Get a descriptor for EXPR. */
733 gfc_init_se (&src_se, NULL);
734 gfc_conv_expr_descriptor (&src_se, expr, src_ss);
735 gfc_add_block_to_block (&se->pre, &src_se.pre);
736 gfc_add_block_to_block (&se->post, &src_se.post);
737 src = src_se.expr;
738
739 /* Allocate a new descriptor for the return value. */
740 dest = gfc_create_var (TREE_TYPE (src), "atmp");
741 dest_info->descriptor = dest;
742 se->expr = dest;
743
744 /* Copy across the dtype field. */
745 gfc_add_modify_expr (&se->pre,
746 gfc_conv_descriptor_dtype (dest),
747 gfc_conv_descriptor_dtype (src));
748
749 /* Copy the dimension information, renumbering dimension 1 to 0 and
750 0 to 1. */
751 for (n = 0; n < 2; n++)
752 {
753 dest_info->delta[n] = gfc_index_zero_node;
754 dest_info->start[n] = gfc_index_zero_node;
755 dest_info->end[n] = gfc_index_zero_node;
756 dest_info->stride[n] = gfc_index_one_node;
757 dest_info->dim[n] = n;
758
759 dest_index = gfc_rank_cst[n];
760 src_index = gfc_rank_cst[1 - n];
761
762 gfc_add_modify_expr (&se->pre,
763 gfc_conv_descriptor_stride (dest, dest_index),
764 gfc_conv_descriptor_stride (src, src_index));
765
766 gfc_add_modify_expr (&se->pre,
767 gfc_conv_descriptor_lbound (dest, dest_index),
768 gfc_conv_descriptor_lbound (src, src_index));
769
770 gfc_add_modify_expr (&se->pre,
771 gfc_conv_descriptor_ubound (dest, dest_index),
772 gfc_conv_descriptor_ubound (src, src_index));
773
774 if (!loop->to[n])
775 {
776 gcc_assert (integer_zerop (loop->from[n]));
777 loop->to[n] = build2 (MINUS_EXPR, gfc_array_index_type,
778 gfc_conv_descriptor_ubound (dest, dest_index),
779 gfc_conv_descriptor_lbound (dest, dest_index));
780 }
781 }
782
783 /* Copy the data pointer. */
784 dest_info->data = gfc_conv_descriptor_data_get (src);
785 gfc_conv_descriptor_data_set (&se->pre, dest, dest_info->data);
786
787 /* Copy the offset. This is not changed by transposition: the top-left
788 element is still at the same offset as before. */
789 dest_info->offset = gfc_conv_descriptor_offset (src);
790 gfc_add_modify_expr (&se->pre,
791 gfc_conv_descriptor_offset (dest),
792 dest_info->offset);
793
794 if (dest_info->dimen > loop->temp_dim)
795 loop->temp_dim = dest_info->dimen;
796 }
797
798
799 /* Return the number of iterations in a loop that starts at START,
800 ends at END, and has step STEP. */
801
802 static tree
803 gfc_get_iteration_count (tree start, tree end, tree step)
804 {
805 tree tmp;
806 tree type;
807
808 type = TREE_TYPE (step);
809 tmp = fold_build2 (MINUS_EXPR, type, end, start);
810 tmp = fold_build2 (FLOOR_DIV_EXPR, type, tmp, step);
811 tmp = fold_build2 (PLUS_EXPR, type, tmp, build_int_cst (type, 1));
812 tmp = fold_build2 (MAX_EXPR, type, tmp, build_int_cst (type, 0));
813 return fold_convert (gfc_array_index_type, tmp);
814 }
815
816
817 /* Extend the data in array DESC by EXTRA elements. */
818
819 static void
820 gfc_grow_array (stmtblock_t * pblock, tree desc, tree extra)
821 {
822 tree arg0, arg1;
823 tree tmp;
824 tree size;
825 tree ubound;
826
827 if (integer_zerop (extra))
828 return;
829
830 ubound = gfc_conv_descriptor_ubound (desc, gfc_rank_cst[0]);
831
832 /* Add EXTRA to the upper bound. */
833 tmp = build2 (PLUS_EXPR, gfc_array_index_type, ubound, extra);
834 gfc_add_modify_expr (pblock, ubound, tmp);
835
836 /* Get the value of the current data pointer. */
837 arg0 = gfc_conv_descriptor_data_get (desc);
838
839 /* Calculate the new array size. */
840 size = TYPE_SIZE_UNIT (gfc_get_element_type (TREE_TYPE (desc)));
841 tmp = build2 (PLUS_EXPR, gfc_array_index_type, ubound, gfc_index_one_node);
842 arg1 = build2 (MULT_EXPR, gfc_array_index_type, tmp,
843 fold_convert (gfc_array_index_type, size));
844
845 /* Pick the realloc function. */
846 if (gfc_index_integer_kind == 4 || gfc_index_integer_kind == 8)
847 tmp = gfor_fndecl_internal_realloc;
848 else
849 gcc_unreachable ();
850
851 /* Set the new data pointer. */
852 tmp = build_call_expr (tmp, 2, arg0, arg1);
853 gfc_conv_descriptor_data_set (pblock, desc, tmp);
854 }
855
856
857 /* Return true if the bounds of iterator I can only be determined
858 at run time. */
859
860 static inline bool
861 gfc_iterator_has_dynamic_bounds (gfc_iterator * i)
862 {
863 return (i->start->expr_type != EXPR_CONSTANT
864 || i->end->expr_type != EXPR_CONSTANT
865 || i->step->expr_type != EXPR_CONSTANT);
866 }
867
868
869 /* Split the size of constructor element EXPR into the sum of two terms,
870 one of which can be determined at compile time and one of which must
871 be calculated at run time. Set *SIZE to the former and return true
872 if the latter might be nonzero. */
873
874 static bool
875 gfc_get_array_constructor_element_size (mpz_t * size, gfc_expr * expr)
876 {
877 if (expr->expr_type == EXPR_ARRAY)
878 return gfc_get_array_constructor_size (size, expr->value.constructor);
879 else if (expr->rank > 0)
880 {
881 /* Calculate everything at run time. */
882 mpz_set_ui (*size, 0);
883 return true;
884 }
885 else
886 {
887 /* A single element. */
888 mpz_set_ui (*size, 1);
889 return false;
890 }
891 }
892
893
894 /* Like gfc_get_array_constructor_element_size, but applied to the whole
895 of array constructor C. */
896
897 static bool
898 gfc_get_array_constructor_size (mpz_t * size, gfc_constructor * c)
899 {
900 gfc_iterator *i;
901 mpz_t val;
902 mpz_t len;
903 bool dynamic;
904
905 mpz_set_ui (*size, 0);
906 mpz_init (len);
907 mpz_init (val);
908
909 dynamic = false;
910 for (; c; c = c->next)
911 {
912 i = c->iterator;
913 if (i && gfc_iterator_has_dynamic_bounds (i))
914 dynamic = true;
915 else
916 {
917 dynamic |= gfc_get_array_constructor_element_size (&len, c->expr);
918 if (i)
919 {
920 /* Multiply the static part of the element size by the
921 number of iterations. */
922 mpz_sub (val, i->end->value.integer, i->start->value.integer);
923 mpz_fdiv_q (val, val, i->step->value.integer);
924 mpz_add_ui (val, val, 1);
925 if (mpz_sgn (val) > 0)
926 mpz_mul (len, len, val);
927 else
928 mpz_set_ui (len, 0);
929 }
930 mpz_add (*size, *size, len);
931 }
932 }
933 mpz_clear (len);
934 mpz_clear (val);
935 return dynamic;
936 }
937
938
939 /* Make sure offset is a variable. */
940
941 static void
942 gfc_put_offset_into_var (stmtblock_t * pblock, tree * poffset,
943 tree * offsetvar)
944 {
945 /* We should have already created the offset variable. We cannot
946 create it here because we may be in an inner scope. */
947 gcc_assert (*offsetvar != NULL_TREE);
948 gfc_add_modify_expr (pblock, *offsetvar, *poffset);
949 *poffset = *offsetvar;
950 TREE_USED (*offsetvar) = 1;
951 }
952
953
954 /* Assign an element of an array constructor. */
955
956 static void
957 gfc_trans_array_ctor_element (stmtblock_t * pblock, tree desc,
958 tree offset, gfc_se * se, gfc_expr * expr)
959 {
960 tree tmp;
961
962 gfc_conv_expr (se, expr);
963
964 /* Store the value. */
965 tmp = build_fold_indirect_ref (gfc_conv_descriptor_data_get (desc));
966 tmp = gfc_build_array_ref (tmp, offset);
967 if (expr->ts.type == BT_CHARACTER)
968 {
969 gfc_conv_string_parameter (se);
970 if (POINTER_TYPE_P (TREE_TYPE (tmp)))
971 {
972 /* The temporary is an array of pointers. */
973 se->expr = fold_convert (TREE_TYPE (tmp), se->expr);
974 gfc_add_modify_expr (&se->pre, tmp, se->expr);
975 }
976 else
977 {
978 /* The temporary is an array of string values. */
979 tmp = gfc_build_addr_expr (pchar_type_node, tmp);
980 /* We know the temporary and the value will be the same length,
981 so can use memcpy. */
982 tmp = build_call_expr (built_in_decls[BUILT_IN_MEMCPY], 3,
983 tmp, se->expr, se->string_length);
984 gfc_add_expr_to_block (&se->pre, tmp);
985 }
986 }
987 else
988 {
989 /* TODO: Should the frontend already have done this conversion? */
990 se->expr = fold_convert (TREE_TYPE (tmp), se->expr);
991 gfc_add_modify_expr (&se->pre, tmp, se->expr);
992 }
993
994 gfc_add_block_to_block (pblock, &se->pre);
995 gfc_add_block_to_block (pblock, &se->post);
996 }
997
998
999 /* Add the contents of an array to the constructor. DYNAMIC is as for
1000 gfc_trans_array_constructor_value. */
1001
1002 static void
1003 gfc_trans_array_constructor_subarray (stmtblock_t * pblock,
1004 tree type ATTRIBUTE_UNUSED,
1005 tree desc, gfc_expr * expr,
1006 tree * poffset, tree * offsetvar,
1007 bool dynamic)
1008 {
1009 gfc_se se;
1010 gfc_ss *ss;
1011 gfc_loopinfo loop;
1012 stmtblock_t body;
1013 tree tmp;
1014 tree size;
1015 int n;
1016
1017 /* We need this to be a variable so we can increment it. */
1018 gfc_put_offset_into_var (pblock, poffset, offsetvar);
1019
1020 gfc_init_se (&se, NULL);
1021
1022 /* Walk the array expression. */
1023 ss = gfc_walk_expr (expr);
1024 gcc_assert (ss != gfc_ss_terminator);
1025
1026 /* Initialize the scalarizer. */
1027 gfc_init_loopinfo (&loop);
1028 gfc_add_ss_to_loop (&loop, ss);
1029
1030 /* Initialize the loop. */
1031 gfc_conv_ss_startstride (&loop);
1032 gfc_conv_loop_setup (&loop);
1033
1034 /* Make sure the constructed array has room for the new data. */
1035 if (dynamic)
1036 {
1037 /* Set SIZE to the total number of elements in the subarray. */
1038 size = gfc_index_one_node;
1039 for (n = 0; n < loop.dimen; n++)
1040 {
1041 tmp = gfc_get_iteration_count (loop.from[n], loop.to[n],
1042 gfc_index_one_node);
1043 size = fold_build2 (MULT_EXPR, gfc_array_index_type, size, tmp);
1044 }
1045
1046 /* Grow the constructed array by SIZE elements. */
1047 gfc_grow_array (&loop.pre, desc, size);
1048 }
1049
1050 /* Make the loop body. */
1051 gfc_mark_ss_chain_used (ss, 1);
1052 gfc_start_scalarized_body (&loop, &body);
1053 gfc_copy_loopinfo_to_se (&se, &loop);
1054 se.ss = ss;
1055
1056 gfc_trans_array_ctor_element (&body, desc, *poffset, &se, expr);
1057 gcc_assert (se.ss == gfc_ss_terminator);
1058
1059 /* Increment the offset. */
1060 tmp = build2 (PLUS_EXPR, gfc_array_index_type, *poffset, gfc_index_one_node);
1061 gfc_add_modify_expr (&body, *poffset, tmp);
1062
1063 /* Finish the loop. */
1064 gfc_trans_scalarizing_loops (&loop, &body);
1065 gfc_add_block_to_block (&loop.pre, &loop.post);
1066 tmp = gfc_finish_block (&loop.pre);
1067 gfc_add_expr_to_block (pblock, tmp);
1068
1069 gfc_cleanup_loop (&loop);
1070 }
1071
1072
1073 /* Assign the values to the elements of an array constructor. DYNAMIC
1074 is true if descriptor DESC only contains enough data for the static
1075 size calculated by gfc_get_array_constructor_size. When true, memory
1076 for the dynamic parts must be allocated using realloc. */
1077
1078 static void
1079 gfc_trans_array_constructor_value (stmtblock_t * pblock, tree type,
1080 tree desc, gfc_constructor * c,
1081 tree * poffset, tree * offsetvar,
1082 bool dynamic)
1083 {
1084 tree tmp;
1085 stmtblock_t body;
1086 gfc_se se;
1087 mpz_t size;
1088
1089 mpz_init (size);
1090 for (; c; c = c->next)
1091 {
1092 /* If this is an iterator or an array, the offset must be a variable. */
1093 if ((c->iterator || c->expr->rank > 0) && INTEGER_CST_P (*poffset))
1094 gfc_put_offset_into_var (pblock, poffset, offsetvar);
1095
1096 gfc_start_block (&body);
1097
1098 if (c->expr->expr_type == EXPR_ARRAY)
1099 {
1100 /* Array constructors can be nested. */
1101 gfc_trans_array_constructor_value (&body, type, desc,
1102 c->expr->value.constructor,
1103 poffset, offsetvar, dynamic);
1104 }
1105 else if (c->expr->rank > 0)
1106 {
1107 gfc_trans_array_constructor_subarray (&body, type, desc, c->expr,
1108 poffset, offsetvar, dynamic);
1109 }
1110 else
1111 {
1112 /* This code really upsets the gimplifier so don't bother for now. */
1113 gfc_constructor *p;
1114 HOST_WIDE_INT n;
1115 HOST_WIDE_INT size;
1116
1117 p = c;
1118 n = 0;
1119 while (p && !(p->iterator || p->expr->expr_type != EXPR_CONSTANT))
1120 {
1121 p = p->next;
1122 n++;
1123 }
1124 if (n < 4)
1125 {
1126 /* Scalar values. */
1127 gfc_init_se (&se, NULL);
1128 gfc_trans_array_ctor_element (&body, desc, *poffset,
1129 &se, c->expr);
1130
1131 *poffset = fold_build2 (PLUS_EXPR, gfc_array_index_type,
1132 *poffset, gfc_index_one_node);
1133 }
1134 else
1135 {
1136 /* Collect multiple scalar constants into a constructor. */
1137 tree list;
1138 tree init;
1139 tree bound;
1140 tree tmptype;
1141
1142 p = c;
1143 list = NULL_TREE;
1144 /* Count the number of consecutive scalar constants. */
1145 while (p && !(p->iterator
1146 || p->expr->expr_type != EXPR_CONSTANT))
1147 {
1148 gfc_init_se (&se, NULL);
1149 gfc_conv_constant (&se, p->expr);
1150 if (p->expr->ts.type == BT_CHARACTER
1151 && POINTER_TYPE_P (type))
1152 {
1153 /* For constant character array constructors we build
1154 an array of pointers. */
1155 se.expr = gfc_build_addr_expr (pchar_type_node,
1156 se.expr);
1157 }
1158
1159 list = tree_cons (NULL_TREE, se.expr, list);
1160 c = p;
1161 p = p->next;
1162 }
1163
1164 bound = build_int_cst (NULL_TREE, n - 1);
1165 /* Create an array type to hold them. */
1166 tmptype = build_range_type (gfc_array_index_type,
1167 gfc_index_zero_node, bound);
1168 tmptype = build_array_type (type, tmptype);
1169
1170 init = build_constructor_from_list (tmptype, nreverse (list));
1171 TREE_CONSTANT (init) = 1;
1172 TREE_INVARIANT (init) = 1;
1173 TREE_STATIC (init) = 1;
1174 /* Create a static variable to hold the data. */
1175 tmp = gfc_create_var (tmptype, "data");
1176 TREE_STATIC (tmp) = 1;
1177 TREE_CONSTANT (tmp) = 1;
1178 TREE_INVARIANT (tmp) = 1;
1179 TREE_READONLY (tmp) = 1;
1180 DECL_INITIAL (tmp) = init;
1181 init = tmp;
1182
1183 /* Use BUILTIN_MEMCPY to assign the values. */
1184 tmp = gfc_conv_descriptor_data_get (desc);
1185 tmp = build_fold_indirect_ref (tmp);
1186 tmp = gfc_build_array_ref (tmp, *poffset);
1187 tmp = build_fold_addr_expr (tmp);
1188 init = build_fold_addr_expr (init);
1189
1190 size = TREE_INT_CST_LOW (TYPE_SIZE_UNIT (type));
1191 bound = build_int_cst (NULL_TREE, n * size);
1192 tmp = build_call_expr (built_in_decls[BUILT_IN_MEMCPY], 3,
1193 tmp, init, bound);
1194 gfc_add_expr_to_block (&body, tmp);
1195
1196 *poffset = fold_build2 (PLUS_EXPR, gfc_array_index_type,
1197 *poffset,
1198 build_int_cst (gfc_array_index_type, n));
1199 }
1200 if (!INTEGER_CST_P (*poffset))
1201 {
1202 gfc_add_modify_expr (&body, *offsetvar, *poffset);
1203 *poffset = *offsetvar;
1204 }
1205 }
1206
1207 /* The frontend should already have done any expansions possible
1208 at compile-time. */
1209 if (!c->iterator)
1210 {
1211 /* Pass the code as is. */
1212 tmp = gfc_finish_block (&body);
1213 gfc_add_expr_to_block (pblock, tmp);
1214 }
1215 else
1216 {
1217 /* Build the implied do-loop. */
1218 tree cond;
1219 tree end;
1220 tree step;
1221 tree loopvar;
1222 tree exit_label;
1223 tree loopbody;
1224 tree tmp2;
1225 tree tmp_loopvar;
1226
1227 loopbody = gfc_finish_block (&body);
1228
1229 gfc_init_se (&se, NULL);
1230 gfc_conv_expr (&se, c->iterator->var);
1231 gfc_add_block_to_block (pblock, &se.pre);
1232 loopvar = se.expr;
1233
1234 /* Make a temporary, store the current value in that
1235 and return it, once the loop is done. */
1236 tmp_loopvar = gfc_create_var (TREE_TYPE (loopvar), "loopvar");
1237 gfc_add_modify_expr (pblock, tmp_loopvar, loopvar);
1238
1239 /* Initialize the loop. */
1240 gfc_init_se (&se, NULL);
1241 gfc_conv_expr_val (&se, c->iterator->start);
1242 gfc_add_block_to_block (pblock, &se.pre);
1243 gfc_add_modify_expr (pblock, loopvar, se.expr);
1244
1245 gfc_init_se (&se, NULL);
1246 gfc_conv_expr_val (&se, c->iterator->end);
1247 gfc_add_block_to_block (pblock, &se.pre);
1248 end = gfc_evaluate_now (se.expr, pblock);
1249
1250 gfc_init_se (&se, NULL);
1251 gfc_conv_expr_val (&se, c->iterator->step);
1252 gfc_add_block_to_block (pblock, &se.pre);
1253 step = gfc_evaluate_now (se.expr, pblock);
1254
1255 /* If this array expands dynamically, and the number of iterations
1256 is not constant, we won't have allocated space for the static
1257 part of C->EXPR's size. Do that now. */
1258 if (dynamic && gfc_iterator_has_dynamic_bounds (c->iterator))
1259 {
1260 /* Get the number of iterations. */
1261 tmp = gfc_get_iteration_count (loopvar, end, step);
1262
1263 /* Get the static part of C->EXPR's size. */
1264 gfc_get_array_constructor_element_size (&size, c->expr);
1265 tmp2 = gfc_conv_mpz_to_tree (size, gfc_index_integer_kind);
1266
1267 /* Grow the array by TMP * TMP2 elements. */
1268 tmp = fold_build2 (MULT_EXPR, gfc_array_index_type, tmp, tmp2);
1269 gfc_grow_array (pblock, desc, tmp);
1270 }
1271
1272 /* Generate the loop body. */
1273 exit_label = gfc_build_label_decl (NULL_TREE);
1274 gfc_start_block (&body);
1275
1276 /* Generate the exit condition. Depending on the sign of
1277 the step variable we have to generate the correct
1278 comparison. */
1279 tmp = fold_build2 (GT_EXPR, boolean_type_node, step,
1280 build_int_cst (TREE_TYPE (step), 0));
1281 cond = fold_build3 (COND_EXPR, boolean_type_node, tmp,
1282 build2 (GT_EXPR, boolean_type_node,
1283 loopvar, end),
1284 build2 (LT_EXPR, boolean_type_node,
1285 loopvar, end));
1286 tmp = build1_v (GOTO_EXPR, exit_label);
1287 TREE_USED (exit_label) = 1;
1288 tmp = build3_v (COND_EXPR, cond, tmp, build_empty_stmt ());
1289 gfc_add_expr_to_block (&body, tmp);
1290
1291 /* The main loop body. */
1292 gfc_add_expr_to_block (&body, loopbody);
1293
1294 /* Increase loop variable by step. */
1295 tmp = build2 (PLUS_EXPR, TREE_TYPE (loopvar), loopvar, step);
1296 gfc_add_modify_expr (&body, loopvar, tmp);
1297
1298 /* Finish the loop. */
1299 tmp = gfc_finish_block (&body);
1300 tmp = build1_v (LOOP_EXPR, tmp);
1301 gfc_add_expr_to_block (pblock, tmp);
1302
1303 /* Add the exit label. */
1304 tmp = build1_v (LABEL_EXPR, exit_label);
1305 gfc_add_expr_to_block (pblock, tmp);
1306
1307 /* Restore the original value of the loop counter. */
1308 gfc_add_modify_expr (pblock, loopvar, tmp_loopvar);
1309 }
1310 }
1311 mpz_clear (size);
1312 }
1313
1314
1315 /* Figure out the string length of a variable reference expression.
1316 Used by get_array_ctor_strlen. */
1317
1318 static void
1319 get_array_ctor_var_strlen (gfc_expr * expr, tree * len)
1320 {
1321 gfc_ref *ref;
1322 gfc_typespec *ts;
1323 mpz_t char_len;
1324
1325 /* Don't bother if we already know the length is a constant. */
1326 if (*len && INTEGER_CST_P (*len))
1327 return;
1328
1329 ts = &expr->symtree->n.sym->ts;
1330 for (ref = expr->ref; ref; ref = ref->next)
1331 {
1332 switch (ref->type)
1333 {
1334 case REF_ARRAY:
1335 /* Array references don't change the string length. */
1336 break;
1337
1338 case REF_COMPONENT:
1339 /* Use the length of the component. */
1340 ts = &ref->u.c.component->ts;
1341 break;
1342
1343 case REF_SUBSTRING:
1344 if (ref->u.ss.start->expr_type != EXPR_CONSTANT
1345 || ref->u.ss.start->expr_type != EXPR_CONSTANT)
1346 break;
1347 mpz_init_set_ui (char_len, 1);
1348 mpz_add (char_len, char_len, ref->u.ss.end->value.integer);
1349 mpz_sub (char_len, char_len, ref->u.ss.start->value.integer);
1350 *len = gfc_conv_mpz_to_tree (char_len,
1351 gfc_default_character_kind);
1352 *len = convert (gfc_charlen_type_node, *len);
1353 mpz_clear (char_len);
1354 return;
1355
1356 default:
1357 /* TODO: Substrings are tricky because we can't evaluate the
1358 expression more than once. For now we just give up, and hope
1359 we can figure it out elsewhere. */
1360 return;
1361 }
1362 }
1363
1364 *len = ts->cl->backend_decl;
1365 }
1366
1367
1368 /* A catch-all to obtain the string length for anything that is not a
1369 constant, array or variable. */
1370 static void
1371 get_array_ctor_all_strlen (stmtblock_t *block, gfc_expr *e, tree *len)
1372 {
1373 gfc_se se;
1374 gfc_ss *ss;
1375
1376 /* Don't bother if we already know the length is a constant. */
1377 if (*len && INTEGER_CST_P (*len))
1378 return;
1379
1380 if (!e->ref && e->ts.cl->length
1381 && e->ts.cl->length->expr_type == EXPR_CONSTANT)
1382 {
1383 /* This is easy. */
1384 gfc_conv_const_charlen (e->ts.cl);
1385 *len = e->ts.cl->backend_decl;
1386 }
1387 else
1388 {
1389 /* Otherwise, be brutal even if inefficient. */
1390 ss = gfc_walk_expr (e);
1391 gfc_init_se (&se, NULL);
1392
1393 /* No function call, in case of side effects. */
1394 se.no_function_call = 1;
1395 if (ss == gfc_ss_terminator)
1396 gfc_conv_expr (&se, e);
1397 else
1398 gfc_conv_expr_descriptor (&se, e, ss);
1399
1400 /* Fix the value. */
1401 *len = gfc_evaluate_now (se.string_length, &se.pre);
1402
1403 gfc_add_block_to_block (block, &se.pre);
1404 gfc_add_block_to_block (block, &se.post);
1405
1406 e->ts.cl->backend_decl = *len;
1407 }
1408 }
1409
1410
1411 /* Figure out the string length of a character array constructor.
1412 Returns TRUE if all elements are character constants. */
1413
1414 bool
1415 get_array_ctor_strlen (stmtblock_t *block, gfc_constructor * c, tree * len)
1416 {
1417 bool is_const;
1418
1419 is_const = TRUE;
1420 for (; c; c = c->next)
1421 {
1422 switch (c->expr->expr_type)
1423 {
1424 case EXPR_CONSTANT:
1425 if (!(*len && INTEGER_CST_P (*len)))
1426 *len = build_int_cstu (gfc_charlen_type_node,
1427 c->expr->value.character.length);
1428 break;
1429
1430 case EXPR_ARRAY:
1431 if (!get_array_ctor_strlen (block, c->expr->value.constructor, len))
1432 is_const = false;
1433 break;
1434
1435 case EXPR_VARIABLE:
1436 is_const = false;
1437 get_array_ctor_var_strlen (c->expr, len);
1438 break;
1439
1440 default:
1441 is_const = false;
1442 get_array_ctor_all_strlen (block, c->expr, len);
1443 break;
1444 }
1445 }
1446
1447 return is_const;
1448 }
1449
1450 /* Check whether the array constructor C consists entirely of constant
1451 elements, and if so returns the number of those elements, otherwise
1452 return zero. Note, an empty or NULL array constructor returns zero. */
1453
1454 unsigned HOST_WIDE_INT
1455 gfc_constant_array_constructor_p (gfc_constructor * c)
1456 {
1457 unsigned HOST_WIDE_INT nelem = 0;
1458
1459 while (c)
1460 {
1461 if (c->iterator
1462 || c->expr->rank > 0
1463 || c->expr->expr_type != EXPR_CONSTANT)
1464 return 0;
1465 c = c->next;
1466 nelem++;
1467 }
1468 return nelem;
1469 }
1470
1471
1472 /* Given EXPR, the constant array constructor specified by an EXPR_ARRAY,
1473 and the tree type of it's elements, TYPE, return a static constant
1474 variable that is compile-time initialized. */
1475
1476 tree
1477 gfc_build_constant_array_constructor (gfc_expr * expr, tree type)
1478 {
1479 tree tmptype, list, init, tmp;
1480 HOST_WIDE_INT nelem;
1481 gfc_constructor *c;
1482 gfc_array_spec as;
1483 gfc_se se;
1484 int i;
1485
1486 /* First traverse the constructor list, converting the constants
1487 to tree to build an initializer. */
1488 nelem = 0;
1489 list = NULL_TREE;
1490 c = expr->value.constructor;
1491 while (c)
1492 {
1493 gfc_init_se (&se, NULL);
1494 gfc_conv_constant (&se, c->expr);
1495 if (c->expr->ts.type == BT_CHARACTER
1496 && POINTER_TYPE_P (type))
1497 se.expr = gfc_build_addr_expr (pchar_type_node, se.expr);
1498 list = tree_cons (NULL_TREE, se.expr, list);
1499 c = c->next;
1500 nelem++;
1501 }
1502
1503 /* Next determine the tree type for the array. We use the gfortran
1504 front-end's gfc_get_nodesc_array_type in order to create a suitable
1505 GFC_ARRAY_TYPE_P that may be used by the scalarizer. */
1506
1507 memset (&as, 0, sizeof (gfc_array_spec));
1508
1509 as.rank = expr->rank;
1510 as.type = AS_EXPLICIT;
1511 if (!expr->shape)
1512 {
1513 as.lower[0] = gfc_int_expr (0);
1514 as.upper[0] = gfc_int_expr (nelem - 1);
1515 }
1516 else
1517 for (i = 0; i < expr->rank; i++)
1518 {
1519 int tmp = (int) mpz_get_si (expr->shape[i]);
1520 as.lower[i] = gfc_int_expr (0);
1521 as.upper[i] = gfc_int_expr (tmp - 1);
1522 }
1523
1524 tmptype = gfc_get_nodesc_array_type (type, &as, PACKED_STATIC);
1525
1526 init = build_constructor_from_list (tmptype, nreverse (list));
1527
1528 TREE_CONSTANT (init) = 1;
1529 TREE_INVARIANT (init) = 1;
1530 TREE_STATIC (init) = 1;
1531
1532 tmp = gfc_create_var (tmptype, "A");
1533 TREE_STATIC (tmp) = 1;
1534 TREE_CONSTANT (tmp) = 1;
1535 TREE_INVARIANT (tmp) = 1;
1536 TREE_READONLY (tmp) = 1;
1537 DECL_INITIAL (tmp) = init;
1538
1539 return tmp;
1540 }
1541
1542
1543 /* Translate a constant EXPR_ARRAY array constructor for the scalarizer.
1544 This mostly initializes the scalarizer state info structure with the
1545 appropriate values to directly use the array created by the function
1546 gfc_build_constant_array_constructor. */
1547
1548 static void
1549 gfc_trans_constant_array_constructor (gfc_loopinfo * loop,
1550 gfc_ss * ss, tree type)
1551 {
1552 gfc_ss_info *info;
1553 tree tmp;
1554 int i;
1555
1556 tmp = gfc_build_constant_array_constructor (ss->expr, type);
1557
1558 info = &ss->data.info;
1559
1560 info->descriptor = tmp;
1561 info->data = build_fold_addr_expr (tmp);
1562 info->offset = fold_build1 (NEGATE_EXPR, gfc_array_index_type,
1563 loop->from[0]);
1564
1565 for (i = 0; i < info->dimen; i++)
1566 {
1567 info->delta[i] = gfc_index_zero_node;
1568 info->start[i] = gfc_index_zero_node;
1569 info->end[i] = gfc_index_zero_node;
1570 info->stride[i] = gfc_index_one_node;
1571 info->dim[i] = i;
1572 }
1573
1574 if (info->dimen > loop->temp_dim)
1575 loop->temp_dim = info->dimen;
1576 }
1577
1578 /* Helper routine of gfc_trans_array_constructor to determine if the
1579 bounds of the loop specified by LOOP are constant and simple enough
1580 to use with gfc_trans_constant_array_constructor. Returns the
1581 the iteration count of the loop if suitable, and NULL_TREE otherwise. */
1582
1583 static tree
1584 constant_array_constructor_loop_size (gfc_loopinfo * loop)
1585 {
1586 tree size = gfc_index_one_node;
1587 tree tmp;
1588 int i;
1589
1590 for (i = 0; i < loop->dimen; i++)
1591 {
1592 /* If the bounds aren't constant, return NULL_TREE. */
1593 if (!INTEGER_CST_P (loop->from[i]) || !INTEGER_CST_P (loop->to[i]))
1594 return NULL_TREE;
1595 if (!integer_zerop (loop->from[i]))
1596 {
1597 /* Only allow nonzero "from" in one-dimensional arrays. */
1598 if (loop->dimen != 1)
1599 return NULL_TREE;
1600 tmp = fold_build2 (MINUS_EXPR, gfc_array_index_type,
1601 loop->to[i], loop->from[i]);
1602 }
1603 else
1604 tmp = loop->to[i];
1605 tmp = fold_build2 (PLUS_EXPR, gfc_array_index_type,
1606 tmp, gfc_index_one_node);
1607 size = fold_build2 (MULT_EXPR, gfc_array_index_type, size, tmp);
1608 }
1609
1610 return size;
1611 }
1612
1613
1614 /* Array constructors are handled by constructing a temporary, then using that
1615 within the scalarization loop. This is not optimal, but seems by far the
1616 simplest method. */
1617
1618 static void
1619 gfc_trans_array_constructor (gfc_loopinfo * loop, gfc_ss * ss)
1620 {
1621 gfc_constructor *c;
1622 tree offset;
1623 tree offsetvar;
1624 tree desc;
1625 tree type;
1626 bool dynamic;
1627
1628 ss->data.info.dimen = loop->dimen;
1629
1630 c = ss->expr->value.constructor;
1631 if (ss->expr->ts.type == BT_CHARACTER)
1632 {
1633 bool const_string = get_array_ctor_strlen (&loop->pre, c, &ss->string_length);
1634 if (!ss->string_length)
1635 gfc_todo_error ("complex character array constructors");
1636
1637 /* It is surprising but still possible to wind up with expressions that
1638 lack a character length.
1639 TODO Find the offending part of the front end and cure this properly.
1640 Concatenation involving arrays is the main culprit. */
1641 if (!ss->expr->ts.cl)
1642 {
1643 ss->expr->ts.cl = gfc_get_charlen ();
1644 ss->expr->ts.cl->next = gfc_current_ns->cl_list;
1645 gfc_current_ns->cl_list = ss->expr->ts.cl->next;
1646 }
1647
1648 ss->expr->ts.cl->backend_decl = ss->string_length;
1649
1650 type = gfc_get_character_type_len (ss->expr->ts.kind, ss->string_length);
1651 if (const_string)
1652 type = build_pointer_type (type);
1653 }
1654 else
1655 type = gfc_typenode_for_spec (&ss->expr->ts);
1656
1657 /* See if the constructor determines the loop bounds. */
1658 dynamic = false;
1659
1660 if (ss->expr->shape && loop->dimen > 1 && loop->to[0] == NULL_TREE)
1661 {
1662 /* We have a multidimensional parameter. */
1663 int n;
1664 for (n = 0; n < ss->expr->rank; n++)
1665 {
1666 loop->from[n] = gfc_index_zero_node;
1667 loop->to[n] = gfc_conv_mpz_to_tree (ss->expr->shape [n],
1668 gfc_index_integer_kind);
1669 loop->to[n] = fold_build2 (MINUS_EXPR, gfc_array_index_type,
1670 loop->to[n], gfc_index_one_node);
1671 }
1672 }
1673
1674 if (loop->to[0] == NULL_TREE)
1675 {
1676 mpz_t size;
1677
1678 /* We should have a 1-dimensional, zero-based loop. */
1679 gcc_assert (loop->dimen == 1);
1680 gcc_assert (integer_zerop (loop->from[0]));
1681
1682 /* Split the constructor size into a static part and a dynamic part.
1683 Allocate the static size up-front and record whether the dynamic
1684 size might be nonzero. */
1685 mpz_init (size);
1686 dynamic = gfc_get_array_constructor_size (&size, c);
1687 mpz_sub_ui (size, size, 1);
1688 loop->to[0] = gfc_conv_mpz_to_tree (size, gfc_index_integer_kind);
1689 mpz_clear (size);
1690 }
1691
1692 /* Special case constant array constructors. */
1693 if (!dynamic)
1694 {
1695 unsigned HOST_WIDE_INT nelem = gfc_constant_array_constructor_p (c);
1696 if (nelem > 0)
1697 {
1698 tree size = constant_array_constructor_loop_size (loop);
1699 if (size && compare_tree_int (size, nelem) == 0)
1700 {
1701 gfc_trans_constant_array_constructor (loop, ss, type);
1702 return;
1703 }
1704 }
1705 }
1706
1707 gfc_trans_create_temp_array (&loop->pre, &loop->post, loop, &ss->data.info,
1708 type, dynamic, true, false);
1709
1710 desc = ss->data.info.descriptor;
1711 offset = gfc_index_zero_node;
1712 offsetvar = gfc_create_var_np (gfc_array_index_type, "offset");
1713 TREE_NO_WARNING (offsetvar) = 1;
1714 TREE_USED (offsetvar) = 0;
1715 gfc_trans_array_constructor_value (&loop->pre, type, desc, c,
1716 &offset, &offsetvar, dynamic);
1717
1718 /* If the array grows dynamically, the upper bound of the loop variable
1719 is determined by the array's final upper bound. */
1720 if (dynamic)
1721 loop->to[0] = gfc_conv_descriptor_ubound (desc, gfc_rank_cst[0]);
1722
1723 if (TREE_USED (offsetvar))
1724 pushdecl (offsetvar);
1725 else
1726 gcc_assert (INTEGER_CST_P (offset));
1727 #if 0
1728 /* Disable bound checking for now because it's probably broken. */
1729 if (flag_bounds_check)
1730 {
1731 gcc_unreachable ();
1732 }
1733 #endif
1734 }
1735
1736
1737 /* INFO describes a GFC_SS_SECTION in loop LOOP, and this function is
1738 called after evaluating all of INFO's vector dimensions. Go through
1739 each such vector dimension and see if we can now fill in any missing
1740 loop bounds. */
1741
1742 static void
1743 gfc_set_vector_loop_bounds (gfc_loopinfo * loop, gfc_ss_info * info)
1744 {
1745 gfc_se se;
1746 tree tmp;
1747 tree desc;
1748 tree zero;
1749 int n;
1750 int dim;
1751
1752 for (n = 0; n < loop->dimen; n++)
1753 {
1754 dim = info->dim[n];
1755 if (info->ref->u.ar.dimen_type[dim] == DIMEN_VECTOR
1756 && loop->to[n] == NULL)
1757 {
1758 /* Loop variable N indexes vector dimension DIM, and we don't
1759 yet know the upper bound of loop variable N. Set it to the
1760 difference between the vector's upper and lower bounds. */
1761 gcc_assert (loop->from[n] == gfc_index_zero_node);
1762 gcc_assert (info->subscript[dim]
1763 && info->subscript[dim]->type == GFC_SS_VECTOR);
1764
1765 gfc_init_se (&se, NULL);
1766 desc = info->subscript[dim]->data.info.descriptor;
1767 zero = gfc_rank_cst[0];
1768 tmp = fold_build2 (MINUS_EXPR, gfc_array_index_type,
1769 gfc_conv_descriptor_ubound (desc, zero),
1770 gfc_conv_descriptor_lbound (desc, zero));
1771 tmp = gfc_evaluate_now (tmp, &loop->pre);
1772 loop->to[n] = tmp;
1773 }
1774 }
1775 }
1776
1777
1778 /* Add the pre and post chains for all the scalar expressions in a SS chain
1779 to loop. This is called after the loop parameters have been calculated,
1780 but before the actual scalarizing loops. */
1781
1782 static void
1783 gfc_add_loop_ss_code (gfc_loopinfo * loop, gfc_ss * ss, bool subscript)
1784 {
1785 gfc_se se;
1786 int n;
1787
1788 /* TODO: This can generate bad code if there are ordering dependencies.
1789 eg. a callee allocated function and an unknown size constructor. */
1790 gcc_assert (ss != NULL);
1791
1792 for (; ss != gfc_ss_terminator; ss = ss->loop_chain)
1793 {
1794 gcc_assert (ss);
1795
1796 switch (ss->type)
1797 {
1798 case GFC_SS_SCALAR:
1799 /* Scalar expression. Evaluate this now. This includes elemental
1800 dimension indices, but not array section bounds. */
1801 gfc_init_se (&se, NULL);
1802 gfc_conv_expr (&se, ss->expr);
1803 gfc_add_block_to_block (&loop->pre, &se.pre);
1804
1805 if (ss->expr->ts.type != BT_CHARACTER)
1806 {
1807 /* Move the evaluation of scalar expressions outside the
1808 scalarization loop. */
1809 if (subscript)
1810 se.expr = convert(gfc_array_index_type, se.expr);
1811 se.expr = gfc_evaluate_now (se.expr, &loop->pre);
1812 gfc_add_block_to_block (&loop->pre, &se.post);
1813 }
1814 else
1815 gfc_add_block_to_block (&loop->post, &se.post);
1816
1817 ss->data.scalar.expr = se.expr;
1818 ss->string_length = se.string_length;
1819 break;
1820
1821 case GFC_SS_REFERENCE:
1822 /* Scalar reference. Evaluate this now. */
1823 gfc_init_se (&se, NULL);
1824 gfc_conv_expr_reference (&se, ss->expr);
1825 gfc_add_block_to_block (&loop->pre, &se.pre);
1826 gfc_add_block_to_block (&loop->post, &se.post);
1827
1828 ss->data.scalar.expr = gfc_evaluate_now (se.expr, &loop->pre);
1829 ss->string_length = se.string_length;
1830 break;
1831
1832 case GFC_SS_SECTION:
1833 /* Add the expressions for scalar and vector subscripts. */
1834 for (n = 0; n < GFC_MAX_DIMENSIONS; n++)
1835 if (ss->data.info.subscript[n])
1836 gfc_add_loop_ss_code (loop, ss->data.info.subscript[n], true);
1837
1838 gfc_set_vector_loop_bounds (loop, &ss->data.info);
1839 break;
1840
1841 case GFC_SS_VECTOR:
1842 /* Get the vector's descriptor and store it in SS. */
1843 gfc_init_se (&se, NULL);
1844 gfc_conv_expr_descriptor (&se, ss->expr, gfc_walk_expr (ss->expr));
1845 gfc_add_block_to_block (&loop->pre, &se.pre);
1846 gfc_add_block_to_block (&loop->post, &se.post);
1847 ss->data.info.descriptor = se.expr;
1848 break;
1849
1850 case GFC_SS_INTRINSIC:
1851 gfc_add_intrinsic_ss_code (loop, ss);
1852 break;
1853
1854 case GFC_SS_FUNCTION:
1855 /* Array function return value. We call the function and save its
1856 result in a temporary for use inside the loop. */
1857 gfc_init_se (&se, NULL);
1858 se.loop = loop;
1859 se.ss = ss;
1860 gfc_conv_expr (&se, ss->expr);
1861 gfc_add_block_to_block (&loop->pre, &se.pre);
1862 gfc_add_block_to_block (&loop->post, &se.post);
1863 ss->string_length = se.string_length;
1864 break;
1865
1866 case GFC_SS_CONSTRUCTOR:
1867 gfc_trans_array_constructor (loop, ss);
1868 break;
1869
1870 case GFC_SS_TEMP:
1871 case GFC_SS_COMPONENT:
1872 /* Do nothing. These are handled elsewhere. */
1873 break;
1874
1875 default:
1876 gcc_unreachable ();
1877 }
1878 }
1879 }
1880
1881
1882 /* Translate expressions for the descriptor and data pointer of a SS. */
1883 /*GCC ARRAYS*/
1884
1885 static void
1886 gfc_conv_ss_descriptor (stmtblock_t * block, gfc_ss * ss, int base)
1887 {
1888 gfc_se se;
1889 tree tmp;
1890
1891 /* Get the descriptor for the array to be scalarized. */
1892 gcc_assert (ss->expr->expr_type == EXPR_VARIABLE);
1893 gfc_init_se (&se, NULL);
1894 se.descriptor_only = 1;
1895 gfc_conv_expr_lhs (&se, ss->expr);
1896 gfc_add_block_to_block (block, &se.pre);
1897 ss->data.info.descriptor = se.expr;
1898 ss->string_length = se.string_length;
1899
1900 if (base)
1901 {
1902 /* Also the data pointer. */
1903 tmp = gfc_conv_array_data (se.expr);
1904 /* If this is a variable or address of a variable we use it directly.
1905 Otherwise we must evaluate it now to avoid breaking dependency
1906 analysis by pulling the expressions for elemental array indices
1907 inside the loop. */
1908 if (!(DECL_P (tmp)
1909 || (TREE_CODE (tmp) == ADDR_EXPR
1910 && DECL_P (TREE_OPERAND (tmp, 0)))))
1911 tmp = gfc_evaluate_now (tmp, block);
1912 ss->data.info.data = tmp;
1913
1914 tmp = gfc_conv_array_offset (se.expr);
1915 ss->data.info.offset = gfc_evaluate_now (tmp, block);
1916 }
1917 }
1918
1919
1920 /* Initialize a gfc_loopinfo structure. */
1921
1922 void
1923 gfc_init_loopinfo (gfc_loopinfo * loop)
1924 {
1925 int n;
1926
1927 memset (loop, 0, sizeof (gfc_loopinfo));
1928 gfc_init_block (&loop->pre);
1929 gfc_init_block (&loop->post);
1930
1931 /* Initially scalarize in order. */
1932 for (n = 0; n < GFC_MAX_DIMENSIONS; n++)
1933 loop->order[n] = n;
1934
1935 loop->ss = gfc_ss_terminator;
1936 }
1937
1938
1939 /* Copies the loop variable info to a gfc_se structure. Does not copy the SS
1940 chain. */
1941
1942 void
1943 gfc_copy_loopinfo_to_se (gfc_se * se, gfc_loopinfo * loop)
1944 {
1945 se->loop = loop;
1946 }
1947
1948
1949 /* Return an expression for the data pointer of an array. */
1950
1951 tree
1952 gfc_conv_array_data (tree descriptor)
1953 {
1954 tree type;
1955
1956 type = TREE_TYPE (descriptor);
1957 if (GFC_ARRAY_TYPE_P (type))
1958 {
1959 if (TREE_CODE (type) == POINTER_TYPE)
1960 return descriptor;
1961 else
1962 {
1963 /* Descriptorless arrays. */
1964 return build_fold_addr_expr (descriptor);
1965 }
1966 }
1967 else
1968 return gfc_conv_descriptor_data_get (descriptor);
1969 }
1970
1971
1972 /* Return an expression for the base offset of an array. */
1973
1974 tree
1975 gfc_conv_array_offset (tree descriptor)
1976 {
1977 tree type;
1978
1979 type = TREE_TYPE (descriptor);
1980 if (GFC_ARRAY_TYPE_P (type))
1981 return GFC_TYPE_ARRAY_OFFSET (type);
1982 else
1983 return gfc_conv_descriptor_offset (descriptor);
1984 }
1985
1986
1987 /* Get an expression for the array stride. */
1988
1989 tree
1990 gfc_conv_array_stride (tree descriptor, int dim)
1991 {
1992 tree tmp;
1993 tree type;
1994
1995 type = TREE_TYPE (descriptor);
1996
1997 /* For descriptorless arrays use the array size. */
1998 tmp = GFC_TYPE_ARRAY_STRIDE (type, dim);
1999 if (tmp != NULL_TREE)
2000 return tmp;
2001
2002 tmp = gfc_conv_descriptor_stride (descriptor, gfc_rank_cst[dim]);
2003 return tmp;
2004 }
2005
2006
2007 /* Like gfc_conv_array_stride, but for the lower bound. */
2008
2009 tree
2010 gfc_conv_array_lbound (tree descriptor, int dim)
2011 {
2012 tree tmp;
2013 tree type;
2014
2015 type = TREE_TYPE (descriptor);
2016
2017 tmp = GFC_TYPE_ARRAY_LBOUND (type, dim);
2018 if (tmp != NULL_TREE)
2019 return tmp;
2020
2021 tmp = gfc_conv_descriptor_lbound (descriptor, gfc_rank_cst[dim]);
2022 return tmp;
2023 }
2024
2025
2026 /* Like gfc_conv_array_stride, but for the upper bound. */
2027
2028 tree
2029 gfc_conv_array_ubound (tree descriptor, int dim)
2030 {
2031 tree tmp;
2032 tree type;
2033
2034 type = TREE_TYPE (descriptor);
2035
2036 tmp = GFC_TYPE_ARRAY_UBOUND (type, dim);
2037 if (tmp != NULL_TREE)
2038 return tmp;
2039
2040 /* This should only ever happen when passing an assumed shape array
2041 as an actual parameter. The value will never be used. */
2042 if (GFC_ARRAY_TYPE_P (TREE_TYPE (descriptor)))
2043 return gfc_index_zero_node;
2044
2045 tmp = gfc_conv_descriptor_ubound (descriptor, gfc_rank_cst[dim]);
2046 return tmp;
2047 }
2048
2049
2050 /* Generate code to perform an array index bound check. */
2051
2052 static tree
2053 gfc_trans_array_bound_check (gfc_se * se, tree descriptor, tree index, int n,
2054 locus * where, bool check_upper)
2055 {
2056 tree fault;
2057 tree tmp;
2058 char *msg;
2059 const char * name = NULL;
2060
2061 if (!flag_bounds_check)
2062 return index;
2063
2064 index = gfc_evaluate_now (index, &se->pre);
2065
2066 /* We find a name for the error message. */
2067 if (se->ss)
2068 name = se->ss->expr->symtree->name;
2069
2070 if (!name && se->loop && se->loop->ss && se->loop->ss->expr
2071 && se->loop->ss->expr->symtree)
2072 name = se->loop->ss->expr->symtree->name;
2073
2074 if (!name && se->loop && se->loop->ss && se->loop->ss->loop_chain
2075 && se->loop->ss->loop_chain->expr
2076 && se->loop->ss->loop_chain->expr->symtree)
2077 name = se->loop->ss->loop_chain->expr->symtree->name;
2078
2079 if (!name && se->loop && se->loop->ss && se->loop->ss->loop_chain
2080 && se->loop->ss->loop_chain->expr->symtree)
2081 name = se->loop->ss->loop_chain->expr->symtree->name;
2082
2083 if (!name && se->loop && se->loop->ss && se->loop->ss->expr)
2084 {
2085 if (se->loop->ss->expr->expr_type == EXPR_FUNCTION
2086 && se->loop->ss->expr->value.function.name)
2087 name = se->loop->ss->expr->value.function.name;
2088 else
2089 if (se->loop->ss->type == GFC_SS_CONSTRUCTOR
2090 || se->loop->ss->type == GFC_SS_SCALAR)
2091 name = "unnamed constant";
2092 }
2093
2094 /* Check lower bound. */
2095 tmp = gfc_conv_array_lbound (descriptor, n);
2096 fault = fold_build2 (LT_EXPR, boolean_type_node, index, tmp);
2097 if (name)
2098 asprintf (&msg, "%s for array '%s', lower bound of dimension %d exceeded",
2099 gfc_msg_fault, name, n+1);
2100 else
2101 asprintf (&msg, "%s, lower bound of dimension %d exceeded",
2102 gfc_msg_fault, n+1);
2103 gfc_trans_runtime_check (fault, msg, &se->pre, where);
2104 gfc_free (msg);
2105
2106 /* Check upper bound. */
2107 if (check_upper)
2108 {
2109 tmp = gfc_conv_array_ubound (descriptor, n);
2110 fault = fold_build2 (GT_EXPR, boolean_type_node, index, tmp);
2111 if (name)
2112 asprintf (&msg, "%s for array '%s', upper bound of dimension %d "
2113 " exceeded", gfc_msg_fault, name, n+1);
2114 else
2115 asprintf (&msg, "%s, upper bound of dimension %d exceeded",
2116 gfc_msg_fault, n+1);
2117 gfc_trans_runtime_check (fault, msg, &se->pre, where);
2118 gfc_free (msg);
2119 }
2120
2121 return index;
2122 }
2123
2124
2125 /* Return the offset for an index. Performs bound checking for elemental
2126 dimensions. Single element references are processed separately. */
2127
2128 static tree
2129 gfc_conv_array_index_offset (gfc_se * se, gfc_ss_info * info, int dim, int i,
2130 gfc_array_ref * ar, tree stride)
2131 {
2132 tree index;
2133 tree desc;
2134 tree data;
2135
2136 /* Get the index into the array for this dimension. */
2137 if (ar)
2138 {
2139 gcc_assert (ar->type != AR_ELEMENT);
2140 switch (ar->dimen_type[dim])
2141 {
2142 case DIMEN_ELEMENT:
2143 gcc_assert (i == -1);
2144 /* Elemental dimension. */
2145 gcc_assert (info->subscript[dim]
2146 && info->subscript[dim]->type == GFC_SS_SCALAR);
2147 /* We've already translated this value outside the loop. */
2148 index = info->subscript[dim]->data.scalar.expr;
2149
2150 index = gfc_trans_array_bound_check (se, info->descriptor,
2151 index, dim, &ar->where,
2152 (ar->as->type != AS_ASSUMED_SIZE
2153 && !ar->as->cp_was_assumed) || dim < ar->dimen - 1);
2154 break;
2155
2156 case DIMEN_VECTOR:
2157 gcc_assert (info && se->loop);
2158 gcc_assert (info->subscript[dim]
2159 && info->subscript[dim]->type == GFC_SS_VECTOR);
2160 desc = info->subscript[dim]->data.info.descriptor;
2161
2162 /* Get a zero-based index into the vector. */
2163 index = fold_build2 (MINUS_EXPR, gfc_array_index_type,
2164 se->loop->loopvar[i], se->loop->from[i]);
2165
2166 /* Multiply the index by the stride. */
2167 index = fold_build2 (MULT_EXPR, gfc_array_index_type,
2168 index, gfc_conv_array_stride (desc, 0));
2169
2170 /* Read the vector to get an index into info->descriptor. */
2171 data = build_fold_indirect_ref (gfc_conv_array_data (desc));
2172 index = gfc_build_array_ref (data, index);
2173 index = gfc_evaluate_now (index, &se->pre);
2174
2175 /* Do any bounds checking on the final info->descriptor index. */
2176 index = gfc_trans_array_bound_check (se, info->descriptor,
2177 index, dim, &ar->where,
2178 (ar->as->type != AS_ASSUMED_SIZE
2179 && !ar->as->cp_was_assumed) || dim < ar->dimen - 1);
2180 break;
2181
2182 case DIMEN_RANGE:
2183 /* Scalarized dimension. */
2184 gcc_assert (info && se->loop);
2185
2186 /* Multiply the loop variable by the stride and delta. */
2187 index = se->loop->loopvar[i];
2188 if (!integer_onep (info->stride[i]))
2189 index = fold_build2 (MULT_EXPR, gfc_array_index_type, index,
2190 info->stride[i]);
2191 if (!integer_zerop (info->delta[i]))
2192 index = fold_build2 (PLUS_EXPR, gfc_array_index_type, index,
2193 info->delta[i]);
2194 break;
2195
2196 default:
2197 gcc_unreachable ();
2198 }
2199 }
2200 else
2201 {
2202 /* Temporary array or derived type component. */
2203 gcc_assert (se->loop);
2204 index = se->loop->loopvar[se->loop->order[i]];
2205 if (!integer_zerop (info->delta[i]))
2206 index = fold_build2 (PLUS_EXPR, gfc_array_index_type,
2207 index, info->delta[i]);
2208 }
2209
2210 /* Multiply by the stride. */
2211 if (!integer_onep (stride))
2212 index = fold_build2 (MULT_EXPR, gfc_array_index_type, index, stride);
2213
2214 return index;
2215 }
2216
2217
2218 /* Build a scalarized reference to an array. */
2219
2220 static void
2221 gfc_conv_scalarized_array_ref (gfc_se * se, gfc_array_ref * ar)
2222 {
2223 gfc_ss_info *info;
2224 tree index;
2225 tree tmp;
2226 int n;
2227
2228 info = &se->ss->data.info;
2229 if (ar)
2230 n = se->loop->order[0];
2231 else
2232 n = 0;
2233
2234 index = gfc_conv_array_index_offset (se, info, info->dim[n], n, ar,
2235 info->stride0);
2236 /* Add the offset for this dimension to the stored offset for all other
2237 dimensions. */
2238 if (!integer_zerop (info->offset))
2239 index = fold_build2 (PLUS_EXPR, gfc_array_index_type, index, info->offset);
2240
2241 tmp = build_fold_indirect_ref (info->data);
2242 se->expr = gfc_build_array_ref (tmp, index);
2243 }
2244
2245
2246 /* Translate access of temporary array. */
2247
2248 void
2249 gfc_conv_tmp_array_ref (gfc_se * se)
2250 {
2251 se->string_length = se->ss->string_length;
2252 gfc_conv_scalarized_array_ref (se, NULL);
2253 }
2254
2255
2256 /* Build an array reference. se->expr already holds the array descriptor.
2257 This should be either a variable, indirect variable reference or component
2258 reference. For arrays which do not have a descriptor, se->expr will be
2259 the data pointer.
2260 a(i, j, k) = base[offset + i * stride[0] + j * stride[1] + k * stride[2]]*/
2261
2262 void
2263 gfc_conv_array_ref (gfc_se * se, gfc_array_ref * ar, gfc_symbol * sym,
2264 locus * where)
2265 {
2266 int n;
2267 tree index;
2268 tree tmp;
2269 tree stride;
2270 gfc_se indexse;
2271
2272 /* Handle scalarized references separately. */
2273 if (ar->type != AR_ELEMENT)
2274 {
2275 gfc_conv_scalarized_array_ref (se, ar);
2276 gfc_advance_se_ss_chain (se);
2277 return;
2278 }
2279
2280 index = gfc_index_zero_node;
2281
2282 /* Calculate the offsets from all the dimensions. */
2283 for (n = 0; n < ar->dimen; n++)
2284 {
2285 /* Calculate the index for this dimension. */
2286 gfc_init_se (&indexse, se);
2287 gfc_conv_expr_type (&indexse, ar->start[n], gfc_array_index_type);
2288 gfc_add_block_to_block (&se->pre, &indexse.pre);
2289
2290 if (flag_bounds_check)
2291 {
2292 /* Check array bounds. */
2293 tree cond;
2294 char *msg;
2295
2296 /* Evaluate the indexse.expr only once. */
2297 indexse.expr = save_expr (indexse.expr);
2298
2299 /* Lower bound. */
2300 tmp = gfc_conv_array_lbound (se->expr, n);
2301 cond = fold_build2 (LT_EXPR, boolean_type_node,
2302 indexse.expr, tmp);
2303 asprintf (&msg, "%s for array '%s', "
2304 "lower bound of dimension %d exceeded", gfc_msg_fault,
2305 sym->name, n+1);
2306 gfc_trans_runtime_check (cond, msg, &se->pre, where);
2307 gfc_free (msg);
2308
2309 /* Upper bound, but not for the last dimension of assumed-size
2310 arrays. */
2311 if (n < ar->dimen - 1
2312 || (ar->as->type != AS_ASSUMED_SIZE && !ar->as->cp_was_assumed))
2313 {
2314 tmp = gfc_conv_array_ubound (se->expr, n);
2315 cond = fold_build2 (GT_EXPR, boolean_type_node,
2316 indexse.expr, tmp);
2317 asprintf (&msg, "%s for array '%s', "
2318 "upper bound of dimension %d exceeded", gfc_msg_fault,
2319 sym->name, n+1);
2320 gfc_trans_runtime_check (cond, msg, &se->pre, where);
2321 gfc_free (msg);
2322 }
2323 }
2324
2325 /* Multiply the index by the stride. */
2326 stride = gfc_conv_array_stride (se->expr, n);
2327 tmp = fold_build2 (MULT_EXPR, gfc_array_index_type, indexse.expr,
2328 stride);
2329
2330 /* And add it to the total. */
2331 index = fold_build2 (PLUS_EXPR, gfc_array_index_type, index, tmp);
2332 }
2333
2334 tmp = gfc_conv_array_offset (se->expr);
2335 if (!integer_zerop (tmp))
2336 index = fold_build2 (PLUS_EXPR, gfc_array_index_type, index, tmp);
2337
2338 /* Access the calculated element. */
2339 tmp = gfc_conv_array_data (se->expr);
2340 tmp = build_fold_indirect_ref (tmp);
2341 se->expr = gfc_build_array_ref (tmp, index);
2342 }
2343
2344
2345 /* Generate the code to be executed immediately before entering a
2346 scalarization loop. */
2347
2348 static void
2349 gfc_trans_preloop_setup (gfc_loopinfo * loop, int dim, int flag,
2350 stmtblock_t * pblock)
2351 {
2352 tree index;
2353 tree stride;
2354 gfc_ss_info *info;
2355 gfc_ss *ss;
2356 gfc_se se;
2357 int i;
2358
2359 /* This code will be executed before entering the scalarization loop
2360 for this dimension. */
2361 for (ss = loop->ss; ss != gfc_ss_terminator; ss = ss->loop_chain)
2362 {
2363 if ((ss->useflags & flag) == 0)
2364 continue;
2365
2366 if (ss->type != GFC_SS_SECTION
2367 && ss->type != GFC_SS_FUNCTION && ss->type != GFC_SS_CONSTRUCTOR
2368 && ss->type != GFC_SS_COMPONENT)
2369 continue;
2370
2371 info = &ss->data.info;
2372
2373 if (dim >= info->dimen)
2374 continue;
2375
2376 if (dim == info->dimen - 1)
2377 {
2378 /* For the outermost loop calculate the offset due to any
2379 elemental dimensions. It will have been initialized with the
2380 base offset of the array. */
2381 if (info->ref)
2382 {
2383 for (i = 0; i < info->ref->u.ar.dimen; i++)
2384 {
2385 if (info->ref->u.ar.dimen_type[i] != DIMEN_ELEMENT)
2386 continue;
2387
2388 gfc_init_se (&se, NULL);
2389 se.loop = loop;
2390 se.expr = info->descriptor;
2391 stride = gfc_conv_array_stride (info->descriptor, i);
2392 index = gfc_conv_array_index_offset (&se, info, i, -1,
2393 &info->ref->u.ar,
2394 stride);
2395 gfc_add_block_to_block (pblock, &se.pre);
2396
2397 info->offset = fold_build2 (PLUS_EXPR, gfc_array_index_type,
2398 info->offset, index);
2399 info->offset = gfc_evaluate_now (info->offset, pblock);
2400 }
2401
2402 i = loop->order[0];
2403 stride = gfc_conv_array_stride (info->descriptor, info->dim[i]);
2404 }
2405 else
2406 stride = gfc_conv_array_stride (info->descriptor, 0);
2407
2408 /* Calculate the stride of the innermost loop. Hopefully this will
2409 allow the backend optimizers to do their stuff more effectively.
2410 */
2411 info->stride0 = gfc_evaluate_now (stride, pblock);
2412 }
2413 else
2414 {
2415 /* Add the offset for the previous loop dimension. */
2416 gfc_array_ref *ar;
2417
2418 if (info->ref)
2419 {
2420 ar = &info->ref->u.ar;
2421 i = loop->order[dim + 1];
2422 }
2423 else
2424 {
2425 ar = NULL;
2426 i = dim + 1;
2427 }
2428
2429 gfc_init_se (&se, NULL);
2430 se.loop = loop;
2431 se.expr = info->descriptor;
2432 stride = gfc_conv_array_stride (info->descriptor, info->dim[i]);
2433 index = gfc_conv_array_index_offset (&se, info, info->dim[i], i,
2434 ar, stride);
2435 gfc_add_block_to_block (pblock, &se.pre);
2436 info->offset = fold_build2 (PLUS_EXPR, gfc_array_index_type,
2437 info->offset, index);
2438 info->offset = gfc_evaluate_now (info->offset, pblock);
2439 }
2440
2441 /* Remember this offset for the second loop. */
2442 if (dim == loop->temp_dim - 1)
2443 info->saved_offset = info->offset;
2444 }
2445 }
2446
2447
2448 /* Start a scalarized expression. Creates a scope and declares loop
2449 variables. */
2450
2451 void
2452 gfc_start_scalarized_body (gfc_loopinfo * loop, stmtblock_t * pbody)
2453 {
2454 int dim;
2455 int n;
2456 int flags;
2457
2458 gcc_assert (!loop->array_parameter);
2459
2460 for (dim = loop->dimen - 1; dim >= 0; dim--)
2461 {
2462 n = loop->order[dim];
2463
2464 gfc_start_block (&loop->code[n]);
2465
2466 /* Create the loop variable. */
2467 loop->loopvar[n] = gfc_create_var (gfc_array_index_type, "S");
2468
2469 if (dim < loop->temp_dim)
2470 flags = 3;
2471 else
2472 flags = 1;
2473 /* Calculate values that will be constant within this loop. */
2474 gfc_trans_preloop_setup (loop, dim, flags, &loop->code[n]);
2475 }
2476 gfc_start_block (pbody);
2477 }
2478
2479
2480 /* Generates the actual loop code for a scalarization loop. */
2481
2482 static void
2483 gfc_trans_scalarized_loop_end (gfc_loopinfo * loop, int n,
2484 stmtblock_t * pbody)
2485 {
2486 stmtblock_t block;
2487 tree cond;
2488 tree tmp;
2489 tree loopbody;
2490 tree exit_label;
2491
2492 loopbody = gfc_finish_block (pbody);
2493
2494 /* Initialize the loopvar. */
2495 gfc_add_modify_expr (&loop->code[n], loop->loopvar[n], loop->from[n]);
2496
2497 exit_label = gfc_build_label_decl (NULL_TREE);
2498
2499 /* Generate the loop body. */
2500 gfc_init_block (&block);
2501
2502 /* The exit condition. */
2503 cond = build2 (GT_EXPR, boolean_type_node, loop->loopvar[n], loop->to[n]);
2504 tmp = build1_v (GOTO_EXPR, exit_label);
2505 TREE_USED (exit_label) = 1;
2506 tmp = build3_v (COND_EXPR, cond, tmp, build_empty_stmt ());
2507 gfc_add_expr_to_block (&block, tmp);
2508
2509 /* The main body. */
2510 gfc_add_expr_to_block (&block, loopbody);
2511
2512 /* Increment the loopvar. */
2513 tmp = build2 (PLUS_EXPR, gfc_array_index_type,
2514 loop->loopvar[n], gfc_index_one_node);
2515 gfc_add_modify_expr (&block, loop->loopvar[n], tmp);
2516
2517 /* Build the loop. */
2518 tmp = gfc_finish_block (&block);
2519 tmp = build1_v (LOOP_EXPR, tmp);
2520 gfc_add_expr_to_block (&loop->code[n], tmp);
2521
2522 /* Add the exit label. */
2523 tmp = build1_v (LABEL_EXPR, exit_label);
2524 gfc_add_expr_to_block (&loop->code[n], tmp);
2525 }
2526
2527
2528 /* Finishes and generates the loops for a scalarized expression. */
2529
2530 void
2531 gfc_trans_scalarizing_loops (gfc_loopinfo * loop, stmtblock_t * body)
2532 {
2533 int dim;
2534 int n;
2535 gfc_ss *ss;
2536 stmtblock_t *pblock;
2537 tree tmp;
2538
2539 pblock = body;
2540 /* Generate the loops. */
2541 for (dim = 0; dim < loop->dimen; dim++)
2542 {
2543 n = loop->order[dim];
2544 gfc_trans_scalarized_loop_end (loop, n, pblock);
2545 loop->loopvar[n] = NULL_TREE;
2546 pblock = &loop->code[n];
2547 }
2548
2549 tmp = gfc_finish_block (pblock);
2550 gfc_add_expr_to_block (&loop->pre, tmp);
2551
2552 /* Clear all the used flags. */
2553 for (ss = loop->ss; ss; ss = ss->loop_chain)
2554 ss->useflags = 0;
2555 }
2556
2557
2558 /* Finish the main body of a scalarized expression, and start the secondary
2559 copying body. */
2560
2561 void
2562 gfc_trans_scalarized_loop_boundary (gfc_loopinfo * loop, stmtblock_t * body)
2563 {
2564 int dim;
2565 int n;
2566 stmtblock_t *pblock;
2567 gfc_ss *ss;
2568
2569 pblock = body;
2570 /* We finish as many loops as are used by the temporary. */
2571 for (dim = 0; dim < loop->temp_dim - 1; dim++)
2572 {
2573 n = loop->order[dim];
2574 gfc_trans_scalarized_loop_end (loop, n, pblock);
2575 loop->loopvar[n] = NULL_TREE;
2576 pblock = &loop->code[n];
2577 }
2578
2579 /* We don't want to finish the outermost loop entirely. */
2580 n = loop->order[loop->temp_dim - 1];
2581 gfc_trans_scalarized_loop_end (loop, n, pblock);
2582
2583 /* Restore the initial offsets. */
2584 for (ss = loop->ss; ss != gfc_ss_terminator; ss = ss->loop_chain)
2585 {
2586 if ((ss->useflags & 2) == 0)
2587 continue;
2588
2589 if (ss->type != GFC_SS_SECTION
2590 && ss->type != GFC_SS_FUNCTION && ss->type != GFC_SS_CONSTRUCTOR
2591 && ss->type != GFC_SS_COMPONENT)
2592 continue;
2593
2594 ss->data.info.offset = ss->data.info.saved_offset;
2595 }
2596
2597 /* Restart all the inner loops we just finished. */
2598 for (dim = loop->temp_dim - 2; dim >= 0; dim--)
2599 {
2600 n = loop->order[dim];
2601
2602 gfc_start_block (&loop->code[n]);
2603
2604 loop->loopvar[n] = gfc_create_var (gfc_array_index_type, "Q");
2605
2606 gfc_trans_preloop_setup (loop, dim, 2, &loop->code[n]);
2607 }
2608
2609 /* Start a block for the secondary copying code. */
2610 gfc_start_block (body);
2611 }
2612
2613
2614 /* Calculate the upper bound of an array section. */
2615
2616 static tree
2617 gfc_conv_section_upper_bound (gfc_ss * ss, int n, stmtblock_t * pblock)
2618 {
2619 int dim;
2620 gfc_expr *end;
2621 tree desc;
2622 tree bound;
2623 gfc_se se;
2624 gfc_ss_info *info;
2625
2626 gcc_assert (ss->type == GFC_SS_SECTION);
2627
2628 info = &ss->data.info;
2629 dim = info->dim[n];
2630
2631 if (info->ref->u.ar.dimen_type[dim] == DIMEN_VECTOR)
2632 /* We'll calculate the upper bound once we have access to the
2633 vector's descriptor. */
2634 return NULL;
2635
2636 gcc_assert (info->ref->u.ar.dimen_type[dim] == DIMEN_RANGE);
2637 desc = info->descriptor;
2638 end = info->ref->u.ar.end[dim];
2639
2640 if (end)
2641 {
2642 /* The upper bound was specified. */
2643 gfc_init_se (&se, NULL);
2644 gfc_conv_expr_type (&se, end, gfc_array_index_type);
2645 gfc_add_block_to_block (pblock, &se.pre);
2646 bound = se.expr;
2647 }
2648 else
2649 {
2650 /* No upper bound was specified, so use the bound of the array. */
2651 bound = gfc_conv_array_ubound (desc, dim);
2652 }
2653
2654 return bound;
2655 }
2656
2657
2658 /* Calculate the lower bound of an array section. */
2659
2660 static void
2661 gfc_conv_section_startstride (gfc_loopinfo * loop, gfc_ss * ss, int n)
2662 {
2663 gfc_expr *start;
2664 gfc_expr *end;
2665 gfc_expr *stride;
2666 tree desc;
2667 gfc_se se;
2668 gfc_ss_info *info;
2669 int dim;
2670
2671 gcc_assert (ss->type == GFC_SS_SECTION);
2672
2673 info = &ss->data.info;
2674 dim = info->dim[n];
2675
2676 if (info->ref->u.ar.dimen_type[dim] == DIMEN_VECTOR)
2677 {
2678 /* We use a zero-based index to access the vector. */
2679 info->start[n] = gfc_index_zero_node;
2680 info->end[n] = gfc_index_zero_node;
2681 info->stride[n] = gfc_index_one_node;
2682 return;
2683 }
2684
2685 gcc_assert (info->ref->u.ar.dimen_type[dim] == DIMEN_RANGE);
2686 desc = info->descriptor;
2687 start = info->ref->u.ar.start[dim];
2688 end = info->ref->u.ar.end[dim];
2689 stride = info->ref->u.ar.stride[dim];
2690
2691 /* Calculate the start of the range. For vector subscripts this will
2692 be the range of the vector. */
2693 if (start)
2694 {
2695 /* Specified section start. */
2696 gfc_init_se (&se, NULL);
2697 gfc_conv_expr_type (&se, start, gfc_array_index_type);
2698 gfc_add_block_to_block (&loop->pre, &se.pre);
2699 info->start[n] = se.expr;
2700 }
2701 else
2702 {
2703 /* No lower bound specified so use the bound of the array. */
2704 info->start[n] = gfc_conv_array_lbound (desc, dim);
2705 }
2706 info->start[n] = gfc_evaluate_now (info->start[n], &loop->pre);
2707
2708 /* Similarly calculate the end. Although this is not used in the
2709 scalarizer, it is needed when checking bounds and where the end
2710 is an expression with side-effects. */
2711 if (end)
2712 {
2713 /* Specified section start. */
2714 gfc_init_se (&se, NULL);
2715 gfc_conv_expr_type (&se, end, gfc_array_index_type);
2716 gfc_add_block_to_block (&loop->pre, &se.pre);
2717 info->end[n] = se.expr;
2718 }
2719 else
2720 {
2721 /* No upper bound specified so use the bound of the array. */
2722 info->end[n] = gfc_conv_array_ubound (desc, dim);
2723 }
2724 info->end[n] = gfc_evaluate_now (info->end[n], &loop->pre);
2725
2726 /* Calculate the stride. */
2727 if (stride == NULL)
2728 info->stride[n] = gfc_index_one_node;
2729 else
2730 {
2731 gfc_init_se (&se, NULL);
2732 gfc_conv_expr_type (&se, stride, gfc_array_index_type);
2733 gfc_add_block_to_block (&loop->pre, &se.pre);
2734 info->stride[n] = gfc_evaluate_now (se.expr, &loop->pre);
2735 }
2736 }
2737
2738
2739 /* Calculates the range start and stride for a SS chain. Also gets the
2740 descriptor and data pointer. The range of vector subscripts is the size
2741 of the vector. Array bounds are also checked. */
2742
2743 void
2744 gfc_conv_ss_startstride (gfc_loopinfo * loop)
2745 {
2746 int n;
2747 tree tmp;
2748 gfc_ss *ss;
2749 tree desc;
2750
2751 loop->dimen = 0;
2752 /* Determine the rank of the loop. */
2753 for (ss = loop->ss;
2754 ss != gfc_ss_terminator && loop->dimen == 0; ss = ss->loop_chain)
2755 {
2756 switch (ss->type)
2757 {
2758 case GFC_SS_SECTION:
2759 case GFC_SS_CONSTRUCTOR:
2760 case GFC_SS_FUNCTION:
2761 case GFC_SS_COMPONENT:
2762 loop->dimen = ss->data.info.dimen;
2763 break;
2764
2765 /* As usual, lbound and ubound are exceptions!. */
2766 case GFC_SS_INTRINSIC:
2767 switch (ss->expr->value.function.isym->id)
2768 {
2769 case GFC_ISYM_LBOUND:
2770 case GFC_ISYM_UBOUND:
2771 loop->dimen = ss->data.info.dimen;
2772
2773 default:
2774 break;
2775 }
2776
2777 default:
2778 break;
2779 }
2780 }
2781
2782 if (loop->dimen == 0)
2783 gfc_todo_error ("Unable to determine rank of expression");
2784
2785
2786 /* Loop over all the SS in the chain. */
2787 for (ss = loop->ss; ss != gfc_ss_terminator; ss = ss->loop_chain)
2788 {
2789 if (ss->expr && ss->expr->shape && !ss->shape)
2790 ss->shape = ss->expr->shape;
2791
2792 switch (ss->type)
2793 {
2794 case GFC_SS_SECTION:
2795 /* Get the descriptor for the array. */
2796 gfc_conv_ss_descriptor (&loop->pre, ss, !loop->array_parameter);
2797
2798 for (n = 0; n < ss->data.info.dimen; n++)
2799 gfc_conv_section_startstride (loop, ss, n);
2800 break;
2801
2802 case GFC_SS_INTRINSIC:
2803 switch (ss->expr->value.function.isym->id)
2804 {
2805 /* Fall through to supply start and stride. */
2806 case GFC_ISYM_LBOUND:
2807 case GFC_ISYM_UBOUND:
2808 break;
2809 default:
2810 continue;
2811 }
2812
2813 case GFC_SS_CONSTRUCTOR:
2814 case GFC_SS_FUNCTION:
2815 for (n = 0; n < ss->data.info.dimen; n++)
2816 {
2817 ss->data.info.start[n] = gfc_index_zero_node;
2818 ss->data.info.end[n] = gfc_index_zero_node;
2819 ss->data.info.stride[n] = gfc_index_one_node;
2820 }
2821 break;
2822
2823 default:
2824 break;
2825 }
2826 }
2827
2828 /* The rest is just runtime bound checking. */
2829 if (flag_bounds_check)
2830 {
2831 stmtblock_t block;
2832 tree lbound, ubound;
2833 tree end;
2834 tree size[GFC_MAX_DIMENSIONS];
2835 tree stride_pos, stride_neg, non_zerosized, tmp2;
2836 gfc_ss_info *info;
2837 char *msg;
2838 int dim;
2839
2840 gfc_start_block (&block);
2841
2842 for (n = 0; n < loop->dimen; n++)
2843 size[n] = NULL_TREE;
2844
2845 for (ss = loop->ss; ss != gfc_ss_terminator; ss = ss->loop_chain)
2846 {
2847 if (ss->type != GFC_SS_SECTION)
2848 continue;
2849
2850 /* TODO: range checking for mapped dimensions. */
2851 info = &ss->data.info;
2852
2853 /* This code only checks ranges. Elemental and vector
2854 dimensions are checked later. */
2855 for (n = 0; n < loop->dimen; n++)
2856 {
2857 bool check_upper;
2858
2859 dim = info->dim[n];
2860 if (info->ref->u.ar.dimen_type[dim] != DIMEN_RANGE)
2861 continue;
2862
2863 if (n == info->ref->u.ar.dimen - 1
2864 && (info->ref->u.ar.as->type == AS_ASSUMED_SIZE
2865 || info->ref->u.ar.as->cp_was_assumed))
2866 check_upper = false;
2867 else
2868 check_upper = true;
2869
2870 /* Zero stride is not allowed. */
2871 tmp = fold_build2 (EQ_EXPR, boolean_type_node, info->stride[n],
2872 gfc_index_zero_node);
2873 asprintf (&msg, "Zero stride is not allowed, for dimension %d "
2874 "of array '%s'", info->dim[n]+1,
2875 ss->expr->symtree->name);
2876 gfc_trans_runtime_check (tmp, msg, &block, &ss->expr->where);
2877 gfc_free (msg);
2878
2879 desc = ss->data.info.descriptor;
2880
2881 /* This is the run-time equivalent of resolve.c's
2882 check_dimension(). The logical is more readable there
2883 than it is here, with all the trees. */
2884 lbound = gfc_conv_array_lbound (desc, dim);
2885 end = info->end[n];
2886 if (check_upper)
2887 ubound = gfc_conv_array_ubound (desc, dim);
2888 else
2889 ubound = NULL;
2890
2891 /* non_zerosized is true when the selected range is not
2892 empty. */
2893 stride_pos = fold_build2 (GT_EXPR, boolean_type_node,
2894 info->stride[n], gfc_index_zero_node);
2895 tmp = fold_build2 (LE_EXPR, boolean_type_node, info->start[n],
2896 end);
2897 stride_pos = fold_build2 (TRUTH_AND_EXPR, boolean_type_node,
2898 stride_pos, tmp);
2899
2900 stride_neg = fold_build2 (LT_EXPR, boolean_type_node,
2901 info->stride[n], gfc_index_zero_node);
2902 tmp = fold_build2 (GE_EXPR, boolean_type_node, info->start[n],
2903 end);
2904 stride_neg = fold_build2 (TRUTH_AND_EXPR, boolean_type_node,
2905 stride_neg, tmp);
2906 non_zerosized = fold_build2 (TRUTH_OR_EXPR, boolean_type_node,
2907 stride_pos, stride_neg);
2908
2909 /* Check the start of the range against the lower and upper
2910 bounds of the array, if the range is not empty. */
2911 tmp = fold_build2 (LT_EXPR, boolean_type_node, info->start[n],
2912 lbound);
2913 tmp = fold_build2 (TRUTH_AND_EXPR, boolean_type_node,
2914 non_zerosized, tmp);
2915 asprintf (&msg, "%s, lower bound of dimension %d of array '%s'"
2916 " exceeded", gfc_msg_fault, info->dim[n]+1,
2917 ss->expr->symtree->name);
2918 gfc_trans_runtime_check (tmp, msg, &block, &ss->expr->where);
2919 gfc_free (msg);
2920
2921 if (check_upper)
2922 {
2923 tmp = fold_build2 (GT_EXPR, boolean_type_node,
2924 info->start[n], ubound);
2925 tmp = fold_build2 (TRUTH_AND_EXPR, boolean_type_node,
2926 non_zerosized, tmp);
2927 asprintf (&msg, "%s, upper bound of dimension %d of array "
2928 "'%s' exceeded", gfc_msg_fault, info->dim[n]+1,
2929 ss->expr->symtree->name);
2930 gfc_trans_runtime_check (tmp, msg, &block, &ss->expr->where);
2931 gfc_free (msg);
2932 }
2933
2934 /* Compute the last element of the range, which is not
2935 necessarily "end" (think 0:5:3, which doesn't contain 5)
2936 and check it against both lower and upper bounds. */
2937 tmp2 = fold_build2 (MINUS_EXPR, gfc_array_index_type, end,
2938 info->start[n]);
2939 tmp2 = fold_build2 (TRUNC_MOD_EXPR, gfc_array_index_type, tmp2,
2940 info->stride[n]);
2941 tmp2 = fold_build2 (MINUS_EXPR, gfc_array_index_type, end,
2942 tmp2);
2943
2944 tmp = fold_build2 (LT_EXPR, boolean_type_node, tmp2, lbound);
2945 tmp = fold_build2 (TRUTH_AND_EXPR, boolean_type_node,
2946 non_zerosized, tmp);
2947 asprintf (&msg, "%s, lower bound of dimension %d of array '%s'"
2948 " exceeded", gfc_msg_fault, info->dim[n]+1,
2949 ss->expr->symtree->name);
2950 gfc_trans_runtime_check (tmp, msg, &block, &ss->expr->where);
2951 gfc_free (msg);
2952
2953 if (check_upper)
2954 {
2955 tmp = fold_build2 (GT_EXPR, boolean_type_node, tmp2, ubound);
2956 tmp = fold_build2 (TRUTH_AND_EXPR, boolean_type_node,
2957 non_zerosized, tmp);
2958 asprintf (&msg, "%s, upper bound of dimension %d of array "
2959 "'%s' exceeded", gfc_msg_fault, info->dim[n]+1,
2960 ss->expr->symtree->name);
2961 gfc_trans_runtime_check (tmp, msg, &block, &ss->expr->where);
2962 gfc_free (msg);
2963 }
2964
2965 /* Check the section sizes match. */
2966 tmp = fold_build2 (MINUS_EXPR, gfc_array_index_type, end,
2967 info->start[n]);
2968 tmp = fold_build2 (FLOOR_DIV_EXPR, gfc_array_index_type, tmp,
2969 info->stride[n]);
2970 /* We remember the size of the first section, and check all the
2971 others against this. */
2972 if (size[n])
2973 {
2974 tmp =
2975 fold_build2 (NE_EXPR, boolean_type_node, tmp, size[n]);
2976 asprintf (&msg, "%s, size mismatch for dimension %d "
2977 "of array '%s'", gfc_msg_bounds, info->dim[n]+1,
2978 ss->expr->symtree->name);
2979 gfc_trans_runtime_check (tmp, msg, &block, &ss->expr->where);
2980 gfc_free (msg);
2981 }
2982 else
2983 size[n] = gfc_evaluate_now (tmp, &block);
2984 }
2985 }
2986
2987 tmp = gfc_finish_block (&block);
2988 gfc_add_expr_to_block (&loop->pre, tmp);
2989 }
2990 }
2991
2992
2993 /* Return true if the two SS could be aliased, i.e. both point to the same data
2994 object. */
2995 /* TODO: resolve aliases based on frontend expressions. */
2996
2997 static int
2998 gfc_could_be_alias (gfc_ss * lss, gfc_ss * rss)
2999 {
3000 gfc_ref *lref;
3001 gfc_ref *rref;
3002 gfc_symbol *lsym;
3003 gfc_symbol *rsym;
3004
3005 lsym = lss->expr->symtree->n.sym;
3006 rsym = rss->expr->symtree->n.sym;
3007 if (gfc_symbols_could_alias (lsym, rsym))
3008 return 1;
3009
3010 if (rsym->ts.type != BT_DERIVED
3011 && lsym->ts.type != BT_DERIVED)
3012 return 0;
3013
3014 /* For derived types we must check all the component types. We can ignore
3015 array references as these will have the same base type as the previous
3016 component ref. */
3017 for (lref = lss->expr->ref; lref != lss->data.info.ref; lref = lref->next)
3018 {
3019 if (lref->type != REF_COMPONENT)
3020 continue;
3021
3022 if (gfc_symbols_could_alias (lref->u.c.sym, rsym))
3023 return 1;
3024
3025 for (rref = rss->expr->ref; rref != rss->data.info.ref;
3026 rref = rref->next)
3027 {
3028 if (rref->type != REF_COMPONENT)
3029 continue;
3030
3031 if (gfc_symbols_could_alias (lref->u.c.sym, rref->u.c.sym))
3032 return 1;
3033 }
3034 }
3035
3036 for (rref = rss->expr->ref; rref != rss->data.info.ref; rref = rref->next)
3037 {
3038 if (rref->type != REF_COMPONENT)
3039 break;
3040
3041 if (gfc_symbols_could_alias (rref->u.c.sym, lsym))
3042 return 1;
3043 }
3044
3045 return 0;
3046 }
3047
3048
3049 /* Resolve array data dependencies. Creates a temporary if required. */
3050 /* TODO: Calc dependencies with gfc_expr rather than gfc_ss, and move to
3051 dependency.c. */
3052
3053 void
3054 gfc_conv_resolve_dependencies (gfc_loopinfo * loop, gfc_ss * dest,
3055 gfc_ss * rss)
3056 {
3057 gfc_ss *ss;
3058 gfc_ref *lref;
3059 gfc_ref *rref;
3060 gfc_ref *aref;
3061 int nDepend = 0;
3062 int temp_dim = 0;
3063
3064 loop->temp_ss = NULL;
3065 aref = dest->data.info.ref;
3066 temp_dim = 0;
3067
3068 for (ss = rss; ss != gfc_ss_terminator; ss = ss->next)
3069 {
3070 if (ss->type != GFC_SS_SECTION)
3071 continue;
3072
3073 if (gfc_could_be_alias (dest, ss)
3074 || gfc_are_equivalenced_arrays (dest->expr, ss->expr))
3075 {
3076 nDepend = 1;
3077 break;
3078 }
3079
3080 if (dest->expr->symtree->n.sym == ss->expr->symtree->n.sym)
3081 {
3082 lref = dest->expr->ref;
3083 rref = ss->expr->ref;
3084
3085 nDepend = gfc_dep_resolver (lref, rref);
3086 if (nDepend == 1)
3087 break;
3088 #if 0
3089 /* TODO : loop shifting. */
3090 if (nDepend == 1)
3091 {
3092 /* Mark the dimensions for LOOP SHIFTING */
3093 for (n = 0; n < loop->dimen; n++)
3094 {
3095 int dim = dest->data.info.dim[n];
3096
3097 if (lref->u.ar.dimen_type[dim] == DIMEN_VECTOR)
3098 depends[n] = 2;
3099 else if (! gfc_is_same_range (&lref->u.ar,
3100 &rref->u.ar, dim, 0))
3101 depends[n] = 1;
3102 }
3103
3104 /* Put all the dimensions with dependencies in the
3105 innermost loops. */
3106 dim = 0;
3107 for (n = 0; n < loop->dimen; n++)
3108 {
3109 gcc_assert (loop->order[n] == n);
3110 if (depends[n])
3111 loop->order[dim++] = n;
3112 }
3113 temp_dim = dim;
3114 for (n = 0; n < loop->dimen; n++)
3115 {
3116 if (! depends[n])
3117 loop->order[dim++] = n;
3118 }
3119
3120 gcc_assert (dim == loop->dimen);
3121 break;
3122 }
3123 #endif
3124 }
3125 }
3126
3127 if (nDepend == 1)
3128 {
3129 tree base_type = gfc_typenode_for_spec (&dest->expr->ts);
3130 if (GFC_ARRAY_TYPE_P (base_type)
3131 || GFC_DESCRIPTOR_TYPE_P (base_type))
3132 base_type = gfc_get_element_type (base_type);
3133 loop->temp_ss = gfc_get_ss ();
3134 loop->temp_ss->type = GFC_SS_TEMP;
3135 loop->temp_ss->data.temp.type = base_type;
3136 loop->temp_ss->string_length = dest->string_length;
3137 loop->temp_ss->data.temp.dimen = loop->dimen;
3138 loop->temp_ss->next = gfc_ss_terminator;
3139 gfc_add_ss_to_loop (loop, loop->temp_ss);
3140 }
3141 else
3142 loop->temp_ss = NULL;
3143 }
3144
3145
3146 /* Initialize the scalarization loop. Creates the loop variables. Determines
3147 the range of the loop variables. Creates a temporary if required.
3148 Calculates how to transform from loop variables to array indices for each
3149 expression. Also generates code for scalar expressions which have been
3150 moved outside the loop. */
3151
3152 void
3153 gfc_conv_loop_setup (gfc_loopinfo * loop)
3154 {
3155 int n;
3156 int dim;
3157 gfc_ss_info *info;
3158 gfc_ss_info *specinfo;
3159 gfc_ss *ss;
3160 tree tmp;
3161 tree len;
3162 gfc_ss *loopspec[GFC_MAX_DIMENSIONS];
3163 bool dynamic[GFC_MAX_DIMENSIONS];
3164 gfc_constructor *c;
3165 mpz_t *cshape;
3166 mpz_t i;
3167
3168 mpz_init (i);
3169 for (n = 0; n < loop->dimen; n++)
3170 {
3171 loopspec[n] = NULL;
3172 dynamic[n] = false;
3173 /* We use one SS term, and use that to determine the bounds of the
3174 loop for this dimension. We try to pick the simplest term. */
3175 for (ss = loop->ss; ss != gfc_ss_terminator; ss = ss->loop_chain)
3176 {
3177 if (ss->shape)
3178 {
3179 /* The frontend has worked out the size for us. */
3180 loopspec[n] = ss;
3181 continue;
3182 }
3183
3184 if (ss->type == GFC_SS_CONSTRUCTOR)
3185 {
3186 /* An unknown size constructor will always be rank one.
3187 Higher rank constructors will either have known shape,
3188 or still be wrapped in a call to reshape. */
3189 gcc_assert (loop->dimen == 1);
3190
3191 /* Always prefer to use the constructor bounds if the size
3192 can be determined at compile time. Prefer not to otherwise,
3193 since the general case involves realloc, and it's better to
3194 avoid that overhead if possible. */
3195 c = ss->expr->value.constructor;
3196 dynamic[n] = gfc_get_array_constructor_size (&i, c);
3197 if (!dynamic[n] || !loopspec[n])
3198 loopspec[n] = ss;
3199 continue;
3200 }
3201
3202 /* TODO: Pick the best bound if we have a choice between a
3203 function and something else. */
3204 if (ss->type == GFC_SS_FUNCTION)
3205 {
3206 loopspec[n] = ss;
3207 continue;
3208 }
3209
3210 if (ss->type != GFC_SS_SECTION)
3211 continue;
3212
3213 if (loopspec[n])
3214 specinfo = &loopspec[n]->data.info;
3215 else
3216 specinfo = NULL;
3217 info = &ss->data.info;
3218
3219 if (!specinfo)
3220 loopspec[n] = ss;
3221 /* Criteria for choosing a loop specifier (most important first):
3222 doesn't need realloc
3223 stride of one
3224 known stride
3225 known lower bound
3226 known upper bound
3227 */
3228 else if (loopspec[n]->type == GFC_SS_CONSTRUCTOR && dynamic[n])
3229 loopspec[n] = ss;
3230 else if (integer_onep (info->stride[n])
3231 && !integer_onep (specinfo->stride[n]))
3232 loopspec[n] = ss;
3233 else if (INTEGER_CST_P (info->stride[n])
3234 && !INTEGER_CST_P (specinfo->stride[n]))
3235 loopspec[n] = ss;
3236 else if (INTEGER_CST_P (info->start[n])
3237 && !INTEGER_CST_P (specinfo->start[n]))
3238 loopspec[n] = ss;
3239 /* We don't work out the upper bound.
3240 else if (INTEGER_CST_P (info->finish[n])
3241 && ! INTEGER_CST_P (specinfo->finish[n]))
3242 loopspec[n] = ss; */
3243 }
3244
3245 if (!loopspec[n])
3246 gfc_todo_error ("Unable to find scalarization loop specifier");
3247
3248 info = &loopspec[n]->data.info;
3249
3250 /* Set the extents of this range. */
3251 cshape = loopspec[n]->shape;
3252 if (cshape && INTEGER_CST_P (info->start[n])
3253 && INTEGER_CST_P (info->stride[n]))
3254 {
3255 loop->from[n] = info->start[n];
3256 mpz_set (i, cshape[n]);
3257 mpz_sub_ui (i, i, 1);
3258 /* To = from + (size - 1) * stride. */
3259 tmp = gfc_conv_mpz_to_tree (i, gfc_index_integer_kind);
3260 if (!integer_onep (info->stride[n]))
3261 tmp = fold_build2 (MULT_EXPR, gfc_array_index_type,
3262 tmp, info->stride[n]);
3263 loop->to[n] = fold_build2 (PLUS_EXPR, gfc_array_index_type,
3264 loop->from[n], tmp);
3265 }
3266 else
3267 {
3268 loop->from[n] = info->start[n];
3269 switch (loopspec[n]->type)
3270 {
3271 case GFC_SS_CONSTRUCTOR:
3272 /* The upper bound is calculated when we expand the
3273 constructor. */
3274 gcc_assert (loop->to[n] == NULL_TREE);
3275 break;
3276
3277 case GFC_SS_SECTION:
3278 loop->to[n] = gfc_conv_section_upper_bound (loopspec[n], n,
3279 &loop->pre);
3280 break;
3281
3282 case GFC_SS_FUNCTION:
3283 /* The loop bound will be set when we generate the call. */
3284 gcc_assert (loop->to[n] == NULL_TREE);
3285 break;
3286
3287 default:
3288 gcc_unreachable ();
3289 }
3290 }
3291
3292 /* Transform everything so we have a simple incrementing variable. */
3293 if (integer_onep (info->stride[n]))
3294 info->delta[n] = gfc_index_zero_node;
3295 else
3296 {
3297 /* Set the delta for this section. */
3298 info->delta[n] = gfc_evaluate_now (loop->from[n], &loop->pre);
3299 /* Number of iterations is (end - start + step) / step.
3300 with start = 0, this simplifies to
3301 last = end / step;
3302 for (i = 0; i<=last; i++){...}; */
3303 tmp = fold_build2 (MINUS_EXPR, gfc_array_index_type,
3304 loop->to[n], loop->from[n]);
3305 tmp = fold_build2 (TRUNC_DIV_EXPR, gfc_array_index_type,
3306 tmp, info->stride[n]);
3307 loop->to[n] = gfc_evaluate_now (tmp, &loop->pre);
3308 /* Make the loop variable start at 0. */
3309 loop->from[n] = gfc_index_zero_node;
3310 }
3311 }
3312
3313 /* Add all the scalar code that can be taken out of the loops.
3314 This may include calculating the loop bounds, so do it before
3315 allocating the temporary. */
3316 gfc_add_loop_ss_code (loop, loop->ss, false);
3317
3318 /* If we want a temporary then create it. */
3319 if (loop->temp_ss != NULL)
3320 {
3321 gcc_assert (loop->temp_ss->type == GFC_SS_TEMP);
3322 tmp = loop->temp_ss->data.temp.type;
3323 len = loop->temp_ss->string_length;
3324 n = loop->temp_ss->data.temp.dimen;
3325 memset (&loop->temp_ss->data.info, 0, sizeof (gfc_ss_info));
3326 loop->temp_ss->type = GFC_SS_SECTION;
3327 loop->temp_ss->data.info.dimen = n;
3328 gfc_trans_create_temp_array (&loop->pre, &loop->post, loop,
3329 &loop->temp_ss->data.info, tmp, false, true,
3330 false);
3331 }
3332
3333 for (n = 0; n < loop->temp_dim; n++)
3334 loopspec[loop->order[n]] = NULL;
3335
3336 mpz_clear (i);
3337
3338 /* For array parameters we don't have loop variables, so don't calculate the
3339 translations. */
3340 if (loop->array_parameter)
3341 return;
3342
3343 /* Calculate the translation from loop variables to array indices. */
3344 for (ss = loop->ss; ss != gfc_ss_terminator; ss = ss->loop_chain)
3345 {
3346 if (ss->type != GFC_SS_SECTION && ss->type != GFC_SS_COMPONENT)
3347 continue;
3348
3349 info = &ss->data.info;
3350
3351 for (n = 0; n < info->dimen; n++)
3352 {
3353 dim = info->dim[n];
3354
3355 /* If we are specifying the range the delta is already set. */
3356 if (loopspec[n] != ss)
3357 {
3358 /* Calculate the offset relative to the loop variable.
3359 First multiply by the stride. */
3360 tmp = loop->from[n];
3361 if (!integer_onep (info->stride[n]))
3362 tmp = fold_build2 (MULT_EXPR, gfc_array_index_type,
3363 tmp, info->stride[n]);
3364
3365 /* Then subtract this from our starting value. */
3366 tmp = fold_build2 (MINUS_EXPR, gfc_array_index_type,
3367 info->start[n], tmp);
3368
3369 info->delta[n] = gfc_evaluate_now (tmp, &loop->pre);
3370 }
3371 }
3372 }
3373 }
3374
3375
3376 /* Fills in an array descriptor, and returns the size of the array. The size
3377 will be a simple_val, ie a variable or a constant. Also calculates the
3378 offset of the base. Returns the size of the array.
3379 {
3380 stride = 1;
3381 offset = 0;
3382 for (n = 0; n < rank; n++)
3383 {
3384 a.lbound[n] = specified_lower_bound;
3385 offset = offset + a.lbond[n] * stride;
3386 size = 1 - lbound;
3387 a.ubound[n] = specified_upper_bound;
3388 a.stride[n] = stride;
3389 size = ubound + size; //size = ubound + 1 - lbound
3390 stride = stride * size;
3391 }
3392 return (stride);
3393 } */
3394 /*GCC ARRAYS*/
3395
3396 static tree
3397 gfc_array_init_size (tree descriptor, int rank, tree * poffset,
3398 gfc_expr ** lower, gfc_expr ** upper,
3399 stmtblock_t * pblock)
3400 {
3401 tree type;
3402 tree tmp;
3403 tree size;
3404 tree offset;
3405 tree stride;
3406 tree cond;
3407 tree or_expr;
3408 tree thencase;
3409 tree elsecase;
3410 tree var;
3411 stmtblock_t thenblock;
3412 stmtblock_t elseblock;
3413 gfc_expr *ubound;
3414 gfc_se se;
3415 int n;
3416
3417 type = TREE_TYPE (descriptor);
3418
3419 stride = gfc_index_one_node;
3420 offset = gfc_index_zero_node;
3421
3422 /* Set the dtype. */
3423 tmp = gfc_conv_descriptor_dtype (descriptor);
3424 gfc_add_modify_expr (pblock, tmp, gfc_get_dtype (TREE_TYPE (descriptor)));
3425
3426 or_expr = NULL_TREE;
3427
3428 for (n = 0; n < rank; n++)
3429 {
3430 /* We have 3 possibilities for determining the size of the array:
3431 lower == NULL => lbound = 1, ubound = upper[n]
3432 upper[n] = NULL => lbound = 1, ubound = lower[n]
3433 upper[n] != NULL => lbound = lower[n], ubound = upper[n] */
3434 ubound = upper[n];
3435
3436 /* Set lower bound. */
3437 gfc_init_se (&se, NULL);
3438 if (lower == NULL)
3439 se.expr = gfc_index_one_node;
3440 else
3441 {
3442 gcc_assert (lower[n]);
3443 if (ubound)
3444 {
3445 gfc_conv_expr_type (&se, lower[n], gfc_array_index_type);
3446 gfc_add_block_to_block (pblock, &se.pre);
3447 }
3448 else
3449 {
3450 se.expr = gfc_index_one_node;
3451 ubound = lower[n];
3452 }
3453 }
3454 tmp = gfc_conv_descriptor_lbound (descriptor, gfc_rank_cst[n]);
3455 gfc_add_modify_expr (pblock, tmp, se.expr);
3456
3457 /* Work out the offset for this component. */
3458 tmp = fold_build2 (MULT_EXPR, gfc_array_index_type, se.expr, stride);
3459 offset = fold_build2 (MINUS_EXPR, gfc_array_index_type, offset, tmp);
3460
3461 /* Start the calculation for the size of this dimension. */
3462 size = build2 (MINUS_EXPR, gfc_array_index_type,
3463 gfc_index_one_node, se.expr);
3464
3465 /* Set upper bound. */
3466 gfc_init_se (&se, NULL);
3467 gcc_assert (ubound);
3468 gfc_conv_expr_type (&se, ubound, gfc_array_index_type);
3469 gfc_add_block_to_block (pblock, &se.pre);
3470
3471 tmp = gfc_conv_descriptor_ubound (descriptor, gfc_rank_cst[n]);
3472 gfc_add_modify_expr (pblock, tmp, se.expr);
3473
3474 /* Store the stride. */
3475 tmp = gfc_conv_descriptor_stride (descriptor, gfc_rank_cst[n]);
3476 gfc_add_modify_expr (pblock, tmp, stride);
3477
3478 /* Calculate the size of this dimension. */
3479 size = fold_build2 (PLUS_EXPR, gfc_array_index_type, se.expr, size);
3480
3481 /* Check whether the size for this dimension is negative. */
3482 cond = fold_build2 (LE_EXPR, boolean_type_node, size,
3483 gfc_index_zero_node);
3484 if (n == 0)
3485 or_expr = cond;
3486 else
3487 or_expr = fold_build2 (TRUTH_OR_EXPR, boolean_type_node, or_expr, cond);
3488
3489 /* Multiply the stride by the number of elements in this dimension. */
3490 stride = fold_build2 (MULT_EXPR, gfc_array_index_type, stride, size);
3491 stride = gfc_evaluate_now (stride, pblock);
3492 }
3493
3494 /* The stride is the number of elements in the array, so multiply by the
3495 size of an element to get the total size. */
3496 tmp = TYPE_SIZE_UNIT (gfc_get_element_type (type));
3497 size = fold_build2 (MULT_EXPR, gfc_array_index_type, stride,
3498 fold_convert (gfc_array_index_type, tmp));
3499
3500 if (poffset != NULL)
3501 {
3502 offset = gfc_evaluate_now (offset, pblock);
3503 *poffset = offset;
3504 }
3505
3506 if (integer_zerop (or_expr))
3507 return size;
3508 if (integer_onep (or_expr))
3509 return gfc_index_zero_node;
3510
3511 var = gfc_create_var (TREE_TYPE (size), "size");
3512 gfc_start_block (&thenblock);
3513 gfc_add_modify_expr (&thenblock, var, gfc_index_zero_node);
3514 thencase = gfc_finish_block (&thenblock);
3515
3516 gfc_start_block (&elseblock);
3517 gfc_add_modify_expr (&elseblock, var, size);
3518 elsecase = gfc_finish_block (&elseblock);
3519
3520 tmp = gfc_evaluate_now (or_expr, pblock);
3521 tmp = build3_v (COND_EXPR, tmp, thencase, elsecase);
3522 gfc_add_expr_to_block (pblock, tmp);
3523
3524 return var;
3525 }
3526
3527
3528 /* Initializes the descriptor and generates a call to _gfor_allocate. Does
3529 the work for an ALLOCATE statement. */
3530 /*GCC ARRAYS*/
3531
3532 bool
3533 gfc_array_allocate (gfc_se * se, gfc_expr * expr, tree pstat)
3534 {
3535 tree tmp;
3536 tree pointer;
3537 tree allocate;
3538 tree offset;
3539 tree size;
3540 gfc_expr **lower;
3541 gfc_expr **upper;
3542 gfc_ref *ref, *prev_ref = NULL;
3543 bool allocatable_array;
3544
3545 ref = expr->ref;
3546
3547 /* Find the last reference in the chain. */
3548 while (ref && ref->next != NULL)
3549 {
3550 gcc_assert (ref->type != REF_ARRAY || ref->u.ar.type == AR_ELEMENT);
3551 prev_ref = ref;
3552 ref = ref->next;
3553 }
3554
3555 if (ref == NULL || ref->type != REF_ARRAY)
3556 return false;
3557
3558 if (!prev_ref)
3559 allocatable_array = expr->symtree->n.sym->attr.allocatable;
3560 else
3561 allocatable_array = prev_ref->u.c.component->allocatable;
3562
3563 /* Figure out the size of the array. */
3564 switch (ref->u.ar.type)
3565 {
3566 case AR_ELEMENT:
3567 lower = NULL;
3568 upper = ref->u.ar.start;
3569 break;
3570
3571 case AR_FULL:
3572 gcc_assert (ref->u.ar.as->type == AS_EXPLICIT);
3573
3574 lower = ref->u.ar.as->lower;
3575 upper = ref->u.ar.as->upper;
3576 break;
3577
3578 case AR_SECTION:
3579 lower = ref->u.ar.start;
3580 upper = ref->u.ar.end;
3581 break;
3582
3583 default:
3584 gcc_unreachable ();
3585 break;
3586 }
3587
3588 size = gfc_array_init_size (se->expr, ref->u.ar.as->rank, &offset,
3589 lower, upper, &se->pre);
3590
3591 /* Allocate memory to store the data. */
3592 pointer = gfc_conv_descriptor_data_get (se->expr);
3593 STRIP_NOPS (pointer);
3594
3595 if (TYPE_PRECISION (gfc_array_index_type) == 32 ||
3596 TYPE_PRECISION (gfc_array_index_type) == 64)
3597 {
3598 if (allocatable_array)
3599 allocate = gfor_fndecl_allocate_array;
3600 else
3601 allocate = gfor_fndecl_allocate;
3602 }
3603 else
3604 gcc_unreachable ();
3605
3606 /* The allocate_array variants take the old pointer as first argument. */
3607 if (allocatable_array)
3608 tmp = build_call_expr (allocate, 3, pointer, size, pstat);
3609 else
3610 tmp = build_call_expr (allocate, 2, size, pstat);
3611 tmp = build2 (MODIFY_EXPR, void_type_node, pointer, tmp);
3612 gfc_add_expr_to_block (&se->pre, tmp);
3613
3614 tmp = gfc_conv_descriptor_offset (se->expr);
3615 gfc_add_modify_expr (&se->pre, tmp, offset);
3616
3617 if (expr->ts.type == BT_DERIVED
3618 && expr->ts.derived->attr.alloc_comp)
3619 {
3620 tmp = gfc_nullify_alloc_comp (expr->ts.derived, se->expr,
3621 ref->u.ar.as->rank);
3622 gfc_add_expr_to_block (&se->pre, tmp);
3623 }
3624
3625 return true;
3626 }
3627
3628
3629 /* Deallocate an array variable. Also used when an allocated variable goes
3630 out of scope. */
3631 /*GCC ARRAYS*/
3632
3633 tree
3634 gfc_array_deallocate (tree descriptor, tree pstat)
3635 {
3636 tree var;
3637 tree tmp;
3638 stmtblock_t block;
3639
3640 gfc_start_block (&block);
3641 /* Get a pointer to the data. */
3642 var = gfc_conv_descriptor_data_get (descriptor);
3643 STRIP_NOPS (var);
3644
3645 /* Parameter is the address of the data component. */
3646 tmp = build_call_expr (gfor_fndecl_deallocate, 2, var, pstat);
3647 gfc_add_expr_to_block (&block, tmp);
3648
3649 /* Zero the data pointer. */
3650 tmp = build2 (MODIFY_EXPR, void_type_node,
3651 var, build_int_cst (TREE_TYPE (var), 0));
3652 gfc_add_expr_to_block (&block, tmp);
3653
3654 return gfc_finish_block (&block);
3655 }
3656
3657
3658 /* Create an array constructor from an initialization expression.
3659 We assume the frontend already did any expansions and conversions. */
3660
3661 tree
3662 gfc_conv_array_initializer (tree type, gfc_expr * expr)
3663 {
3664 gfc_constructor *c;
3665 tree tmp;
3666 mpz_t maxval;
3667 gfc_se se;
3668 HOST_WIDE_INT hi;
3669 unsigned HOST_WIDE_INT lo;
3670 tree index, range;
3671 VEC(constructor_elt,gc) *v = NULL;
3672
3673 switch (expr->expr_type)
3674 {
3675 case EXPR_CONSTANT:
3676 case EXPR_STRUCTURE:
3677 /* A single scalar or derived type value. Create an array with all
3678 elements equal to that value. */
3679 gfc_init_se (&se, NULL);
3680
3681 if (expr->expr_type == EXPR_CONSTANT)
3682 gfc_conv_constant (&se, expr);
3683 else
3684 gfc_conv_structure (&se, expr, 1);
3685
3686 tmp = TYPE_MAX_VALUE (TYPE_DOMAIN (type));
3687 gcc_assert (tmp && INTEGER_CST_P (tmp));
3688 hi = TREE_INT_CST_HIGH (tmp);
3689 lo = TREE_INT_CST_LOW (tmp);
3690 lo++;
3691 if (lo == 0)
3692 hi++;
3693 /* This will probably eat buckets of memory for large arrays. */
3694 while (hi != 0 || lo != 0)
3695 {
3696 CONSTRUCTOR_APPEND_ELT (v, NULL_TREE, se.expr);
3697 if (lo == 0)
3698 hi--;
3699 lo--;
3700 }
3701 break;
3702
3703 case EXPR_ARRAY:
3704 /* Create a vector of all the elements. */
3705 for (c = expr->value.constructor; c; c = c->next)
3706 {
3707 if (c->iterator)
3708 {
3709 /* Problems occur when we get something like
3710 integer :: a(lots) = (/(i, i=1,lots)/) */
3711 /* TODO: Unexpanded array initializers. */
3712 internal_error
3713 ("Possible frontend bug: array constructor not expanded");
3714 }
3715 if (mpz_cmp_si (c->n.offset, 0) != 0)
3716 index = gfc_conv_mpz_to_tree (c->n.offset, gfc_index_integer_kind);
3717 else
3718 index = NULL_TREE;
3719 mpz_init (maxval);
3720 if (mpz_cmp_si (c->repeat, 0) != 0)
3721 {
3722 tree tmp1, tmp2;
3723
3724 mpz_set (maxval, c->repeat);
3725 mpz_add (maxval, c->n.offset, maxval);
3726 mpz_sub_ui (maxval, maxval, 1);
3727 tmp2 = gfc_conv_mpz_to_tree (maxval, gfc_index_integer_kind);
3728 if (mpz_cmp_si (c->n.offset, 0) != 0)
3729 {
3730 mpz_add_ui (maxval, c->n.offset, 1);
3731 tmp1 = gfc_conv_mpz_to_tree (maxval, gfc_index_integer_kind);
3732 }
3733 else
3734 tmp1 = gfc_conv_mpz_to_tree (c->n.offset, gfc_index_integer_kind);
3735
3736 range = build2 (RANGE_EXPR, integer_type_node, tmp1, tmp2);
3737 }
3738 else
3739 range = NULL;
3740 mpz_clear (maxval);
3741
3742 gfc_init_se (&se, NULL);
3743 switch (c->expr->expr_type)
3744 {
3745 case EXPR_CONSTANT:
3746 gfc_conv_constant (&se, c->expr);
3747 if (range == NULL_TREE)
3748 CONSTRUCTOR_APPEND_ELT (v, index, se.expr);
3749 else
3750 {
3751 if (index != NULL_TREE)
3752 CONSTRUCTOR_APPEND_ELT (v, index, se.expr);
3753 CONSTRUCTOR_APPEND_ELT (v, range, se.expr);
3754 }
3755 break;
3756
3757 case EXPR_STRUCTURE:
3758 gfc_conv_structure (&se, c->expr, 1);
3759 CONSTRUCTOR_APPEND_ELT (v, index, se.expr);
3760 break;
3761
3762 default:
3763 gcc_unreachable ();
3764 }
3765 }
3766 break;
3767
3768 case EXPR_NULL:
3769 return gfc_build_null_descriptor (type);
3770
3771 default:
3772 gcc_unreachable ();
3773 }
3774
3775 /* Create a constructor from the list of elements. */
3776 tmp = build_constructor (type, v);
3777 TREE_CONSTANT (tmp) = 1;
3778 TREE_INVARIANT (tmp) = 1;
3779 return tmp;
3780 }
3781
3782
3783 /* Generate code to evaluate non-constant array bounds. Sets *poffset and
3784 returns the size (in elements) of the array. */
3785
3786 static tree
3787 gfc_trans_array_bounds (tree type, gfc_symbol * sym, tree * poffset,
3788 stmtblock_t * pblock)
3789 {
3790 gfc_array_spec *as;
3791 tree size;
3792 tree stride;
3793 tree offset;
3794 tree ubound;
3795 tree lbound;
3796 tree tmp;
3797 gfc_se se;
3798
3799 int dim;
3800
3801 as = sym->as;
3802
3803 size = gfc_index_one_node;
3804 offset = gfc_index_zero_node;
3805 for (dim = 0; dim < as->rank; dim++)
3806 {
3807 /* Evaluate non-constant array bound expressions. */
3808 lbound = GFC_TYPE_ARRAY_LBOUND (type, dim);
3809 if (as->lower[dim] && !INTEGER_CST_P (lbound))
3810 {
3811 gfc_init_se (&se, NULL);
3812 gfc_conv_expr_type (&se, as->lower[dim], gfc_array_index_type);
3813 gfc_add_block_to_block (pblock, &se.pre);
3814 gfc_add_modify_expr (pblock, lbound, se.expr);
3815 }
3816 ubound = GFC_TYPE_ARRAY_UBOUND (type, dim);
3817 if (as->upper[dim] && !INTEGER_CST_P (ubound))
3818 {
3819 gfc_init_se (&se, NULL);
3820 gfc_conv_expr_type (&se, as->upper[dim], gfc_array_index_type);
3821 gfc_add_block_to_block (pblock, &se.pre);
3822 gfc_add_modify_expr (pblock, ubound, se.expr);
3823 }
3824 /* The offset of this dimension. offset = offset - lbound * stride. */
3825 tmp = fold_build2 (MULT_EXPR, gfc_array_index_type, lbound, size);
3826 offset = fold_build2 (MINUS_EXPR, gfc_array_index_type, offset, tmp);
3827
3828 /* The size of this dimension, and the stride of the next. */
3829 if (dim + 1 < as->rank)
3830 stride = GFC_TYPE_ARRAY_STRIDE (type, dim + 1);
3831 else
3832 stride = GFC_TYPE_ARRAY_SIZE (type);
3833
3834 if (ubound != NULL_TREE && !(stride && INTEGER_CST_P (stride)))
3835 {
3836 /* Calculate stride = size * (ubound + 1 - lbound). */
3837 tmp = fold_build2 (MINUS_EXPR, gfc_array_index_type,
3838 gfc_index_one_node, lbound);
3839 tmp = fold_build2 (PLUS_EXPR, gfc_array_index_type, ubound, tmp);
3840 tmp = fold_build2 (MULT_EXPR, gfc_array_index_type, size, tmp);
3841 if (stride)
3842 gfc_add_modify_expr (pblock, stride, tmp);
3843 else
3844 stride = gfc_evaluate_now (tmp, pblock);
3845
3846 /* Make sure that negative size arrays are translated
3847 to being zero size. */
3848 tmp = build2 (GE_EXPR, boolean_type_node,
3849 stride, gfc_index_zero_node);
3850 tmp = build3 (COND_EXPR, gfc_array_index_type, tmp,
3851 stride, gfc_index_zero_node);
3852 gfc_add_modify_expr (pblock, stride, tmp);
3853 }
3854
3855 size = stride;
3856 }
3857
3858 gfc_trans_vla_type_sizes (sym, pblock);
3859
3860 *poffset = offset;
3861 return size;
3862 }
3863
3864
3865 /* Generate code to initialize/allocate an array variable. */
3866
3867 tree
3868 gfc_trans_auto_array_allocation (tree decl, gfc_symbol * sym, tree fnbody)
3869 {
3870 stmtblock_t block;
3871 tree type;
3872 tree tmp;
3873 tree size;
3874 tree offset;
3875 bool onstack;
3876
3877 gcc_assert (!(sym->attr.pointer || sym->attr.allocatable));
3878
3879 /* Do nothing for USEd variables. */
3880 if (sym->attr.use_assoc)
3881 return fnbody;
3882
3883 type = TREE_TYPE (decl);
3884 gcc_assert (GFC_ARRAY_TYPE_P (type));
3885 onstack = TREE_CODE (type) != POINTER_TYPE;
3886
3887 gfc_start_block (&block);
3888
3889 /* Evaluate character string length. */
3890 if (sym->ts.type == BT_CHARACTER
3891 && onstack && !INTEGER_CST_P (sym->ts.cl->backend_decl))
3892 {
3893 gfc_trans_init_string_length (sym->ts.cl, &block);
3894
3895 gfc_trans_vla_type_sizes (sym, &block);
3896
3897 /* Emit a DECL_EXPR for this variable, which will cause the
3898 gimplifier to allocate storage, and all that good stuff. */
3899 tmp = build1 (DECL_EXPR, TREE_TYPE (decl), decl);
3900 gfc_add_expr_to_block (&block, tmp);
3901 }
3902
3903 if (onstack)
3904 {
3905 gfc_add_expr_to_block (&block, fnbody);
3906 return gfc_finish_block (&block);
3907 }
3908
3909 type = TREE_TYPE (type);
3910
3911 gcc_assert (!sym->attr.use_assoc);
3912 gcc_assert (!TREE_STATIC (decl));
3913 gcc_assert (!sym->module);
3914
3915 if (sym->ts.type == BT_CHARACTER
3916 && !INTEGER_CST_P (sym->ts.cl->backend_decl))
3917 gfc_trans_init_string_length (sym->ts.cl, &block);
3918
3919 size = gfc_trans_array_bounds (type, sym, &offset, &block);
3920
3921 /* Don't actually allocate space for Cray Pointees. */
3922 if (sym->attr.cray_pointee)
3923 {
3924 if (TREE_CODE (GFC_TYPE_ARRAY_OFFSET (type)) == VAR_DECL)
3925 gfc_add_modify_expr (&block, GFC_TYPE_ARRAY_OFFSET (type), offset);
3926 gfc_add_expr_to_block (&block, fnbody);
3927 return gfc_finish_block (&block);
3928 }
3929
3930 /* The size is the number of elements in the array, so multiply by the
3931 size of an element to get the total size. */
3932 tmp = TYPE_SIZE_UNIT (gfc_get_element_type (type));
3933 size = fold_build2 (MULT_EXPR, gfc_array_index_type, size,
3934 fold_convert (gfc_array_index_type, tmp));
3935
3936 /* Allocate memory to hold the data. */
3937 tmp = gfc_call_malloc (&block, TREE_TYPE (decl), size);
3938 gfc_add_modify_expr (&block, decl, tmp);
3939
3940 /* Set offset of the array. */
3941 if (TREE_CODE (GFC_TYPE_ARRAY_OFFSET (type)) == VAR_DECL)
3942 gfc_add_modify_expr (&block, GFC_TYPE_ARRAY_OFFSET (type), offset);
3943
3944
3945 /* Automatic arrays should not have initializers. */
3946 gcc_assert (!sym->value);
3947
3948 gfc_add_expr_to_block (&block, fnbody);
3949
3950 /* Free the temporary. */
3951 tmp = gfc_call_free (convert (pvoid_type_node, decl));
3952 gfc_add_expr_to_block (&block, tmp);
3953
3954 return gfc_finish_block (&block);
3955 }
3956
3957
3958 /* Generate entry and exit code for g77 calling convention arrays. */
3959
3960 tree
3961 gfc_trans_g77_array (gfc_symbol * sym, tree body)
3962 {
3963 tree parm;
3964 tree type;
3965 locus loc;
3966 tree offset;
3967 tree tmp;
3968 tree stmt;
3969 stmtblock_t block;
3970
3971 gfc_get_backend_locus (&loc);
3972 gfc_set_backend_locus (&sym->declared_at);
3973
3974 /* Descriptor type. */
3975 parm = sym->backend_decl;
3976 type = TREE_TYPE (parm);
3977 gcc_assert (GFC_ARRAY_TYPE_P (type));
3978
3979 gfc_start_block (&block);
3980
3981 if (sym->ts.type == BT_CHARACTER
3982 && TREE_CODE (sym->ts.cl->backend_decl) == VAR_DECL)
3983 gfc_trans_init_string_length (sym->ts.cl, &block);
3984
3985 /* Evaluate the bounds of the array. */
3986 gfc_trans_array_bounds (type, sym, &offset, &block);
3987
3988 /* Set the offset. */
3989 if (TREE_CODE (GFC_TYPE_ARRAY_OFFSET (type)) == VAR_DECL)
3990 gfc_add_modify_expr (&block, GFC_TYPE_ARRAY_OFFSET (type), offset);
3991
3992 /* Set the pointer itself if we aren't using the parameter directly. */
3993 if (TREE_CODE (parm) != PARM_DECL)
3994 {
3995 tmp = convert (TREE_TYPE (parm), GFC_DECL_SAVED_DESCRIPTOR (parm));
3996 gfc_add_modify_expr (&block, parm, tmp);
3997 }
3998 stmt = gfc_finish_block (&block);
3999
4000 gfc_set_backend_locus (&loc);
4001
4002 gfc_start_block (&block);
4003
4004 /* Add the initialization code to the start of the function. */
4005
4006 if (sym->attr.optional || sym->attr.not_always_present)
4007 {
4008 tmp = gfc_conv_expr_present (sym);
4009 stmt = build3_v (COND_EXPR, tmp, stmt, build_empty_stmt ());
4010 }
4011
4012 gfc_add_expr_to_block (&block, stmt);
4013 gfc_add_expr_to_block (&block, body);
4014
4015 return gfc_finish_block (&block);
4016 }
4017
4018
4019 /* Modify the descriptor of an array parameter so that it has the
4020 correct lower bound. Also move the upper bound accordingly.
4021 If the array is not packed, it will be copied into a temporary.
4022 For each dimension we set the new lower and upper bounds. Then we copy the
4023 stride and calculate the offset for this dimension. We also work out
4024 what the stride of a packed array would be, and see it the two match.
4025 If the array need repacking, we set the stride to the values we just
4026 calculated, recalculate the offset and copy the array data.
4027 Code is also added to copy the data back at the end of the function.
4028 */
4029
4030 tree
4031 gfc_trans_dummy_array_bias (gfc_symbol * sym, tree tmpdesc, tree body)
4032 {
4033 tree size;
4034 tree type;
4035 tree offset;
4036 locus loc;
4037 stmtblock_t block;
4038 stmtblock_t cleanup;
4039 tree lbound;
4040 tree ubound;
4041 tree dubound;
4042 tree dlbound;
4043 tree dumdesc;
4044 tree tmp;
4045 tree stmt;
4046 tree stride, stride2;
4047 tree stmt_packed;
4048 tree stmt_unpacked;
4049 tree partial;
4050 gfc_se se;
4051 int n;
4052 int checkparm;
4053 int no_repack;
4054 bool optional_arg;
4055
4056 /* Do nothing for pointer and allocatable arrays. */
4057 if (sym->attr.pointer || sym->attr.allocatable)
4058 return body;
4059
4060 if (sym->attr.dummy && gfc_is_nodesc_array (sym))
4061 return gfc_trans_g77_array (sym, body);
4062
4063 gfc_get_backend_locus (&loc);
4064 gfc_set_backend_locus (&sym->declared_at);
4065
4066 /* Descriptor type. */
4067 type = TREE_TYPE (tmpdesc);
4068 gcc_assert (GFC_ARRAY_TYPE_P (type));
4069 dumdesc = GFC_DECL_SAVED_DESCRIPTOR (tmpdesc);
4070 dumdesc = build_fold_indirect_ref (dumdesc);
4071 gfc_start_block (&block);
4072
4073 if (sym->ts.type == BT_CHARACTER
4074 && TREE_CODE (sym->ts.cl->backend_decl) == VAR_DECL)
4075 gfc_trans_init_string_length (sym->ts.cl, &block);
4076
4077 checkparm = (sym->as->type == AS_EXPLICIT && flag_bounds_check);
4078
4079 no_repack = !(GFC_DECL_PACKED_ARRAY (tmpdesc)
4080 || GFC_DECL_PARTIAL_PACKED_ARRAY (tmpdesc));
4081
4082 if (GFC_DECL_PARTIAL_PACKED_ARRAY (tmpdesc))
4083 {
4084 /* For non-constant shape arrays we only check if the first dimension
4085 is contiguous. Repacking higher dimensions wouldn't gain us
4086 anything as we still don't know the array stride. */
4087 partial = gfc_create_var (boolean_type_node, "partial");
4088 TREE_USED (partial) = 1;
4089 tmp = gfc_conv_descriptor_stride (dumdesc, gfc_rank_cst[0]);
4090 tmp = fold_build2 (EQ_EXPR, boolean_type_node, tmp, gfc_index_one_node);
4091 gfc_add_modify_expr (&block, partial, tmp);
4092 }
4093 else
4094 {
4095 partial = NULL_TREE;
4096 }
4097
4098 /* The naming of stmt_unpacked and stmt_packed may be counter-intuitive
4099 here, however I think it does the right thing. */
4100 if (no_repack)
4101 {
4102 /* Set the first stride. */
4103 stride = gfc_conv_descriptor_stride (dumdesc, gfc_rank_cst[0]);
4104 stride = gfc_evaluate_now (stride, &block);
4105
4106 tmp = build2 (EQ_EXPR, boolean_type_node, stride, gfc_index_zero_node);
4107 tmp = build3 (COND_EXPR, gfc_array_index_type, tmp,
4108 gfc_index_one_node, stride);
4109 stride = GFC_TYPE_ARRAY_STRIDE (type, 0);
4110 gfc_add_modify_expr (&block, stride, tmp);
4111
4112 /* Allow the user to disable array repacking. */
4113 stmt_unpacked = NULL_TREE;
4114 }
4115 else
4116 {
4117 gcc_assert (integer_onep (GFC_TYPE_ARRAY_STRIDE (type, 0)));
4118 /* A library call to repack the array if necessary. */
4119 tmp = GFC_DECL_SAVED_DESCRIPTOR (tmpdesc);
4120 stmt_unpacked = build_call_expr (gfor_fndecl_in_pack, 1, tmp);
4121
4122 stride = gfc_index_one_node;
4123 }
4124
4125 /* This is for the case where the array data is used directly without
4126 calling the repack function. */
4127 if (no_repack || partial != NULL_TREE)
4128 stmt_packed = gfc_conv_descriptor_data_get (dumdesc);
4129 else
4130 stmt_packed = NULL_TREE;
4131
4132 /* Assign the data pointer. */
4133 if (stmt_packed != NULL_TREE && stmt_unpacked != NULL_TREE)
4134 {
4135 /* Don't repack unknown shape arrays when the first stride is 1. */
4136 tmp = build3 (COND_EXPR, TREE_TYPE (stmt_packed), partial,
4137 stmt_packed, stmt_unpacked);
4138 }
4139 else
4140 tmp = stmt_packed != NULL_TREE ? stmt_packed : stmt_unpacked;
4141 gfc_add_modify_expr (&block, tmpdesc, fold_convert (type, tmp));
4142
4143 offset = gfc_index_zero_node;
4144 size = gfc_index_one_node;
4145
4146 /* Evaluate the bounds of the array. */
4147 for (n = 0; n < sym->as->rank; n++)
4148 {
4149 if (checkparm || !sym->as->upper[n])
4150 {
4151 /* Get the bounds of the actual parameter. */
4152 dubound = gfc_conv_descriptor_ubound (dumdesc, gfc_rank_cst[n]);
4153 dlbound = gfc_conv_descriptor_lbound (dumdesc, gfc_rank_cst[n]);
4154 }
4155 else
4156 {
4157 dubound = NULL_TREE;
4158 dlbound = NULL_TREE;
4159 }
4160
4161 lbound = GFC_TYPE_ARRAY_LBOUND (type, n);
4162 if (!INTEGER_CST_P (lbound))
4163 {
4164 gfc_init_se (&se, NULL);
4165 gfc_conv_expr_type (&se, sym->as->lower[n],
4166 gfc_array_index_type);
4167 gfc_add_block_to_block (&block, &se.pre);
4168 gfc_add_modify_expr (&block, lbound, se.expr);
4169 }
4170
4171 ubound = GFC_TYPE_ARRAY_UBOUND (type, n);
4172 /* Set the desired upper bound. */
4173 if (sym->as->upper[n])
4174 {
4175 /* We know what we want the upper bound to be. */
4176 if (!INTEGER_CST_P (ubound))
4177 {
4178 gfc_init_se (&se, NULL);
4179 gfc_conv_expr_type (&se, sym->as->upper[n],
4180 gfc_array_index_type);
4181 gfc_add_block_to_block (&block, &se.pre);
4182 gfc_add_modify_expr (&block, ubound, se.expr);
4183 }
4184
4185 /* Check the sizes match. */
4186 if (checkparm)
4187 {
4188 /* Check (ubound(a) - lbound(a) == ubound(b) - lbound(b)). */
4189 char * msg;
4190
4191 tmp = fold_build2 (MINUS_EXPR, gfc_array_index_type,
4192 ubound, lbound);
4193 stride2 = build2 (MINUS_EXPR, gfc_array_index_type,
4194 dubound, dlbound);
4195 tmp = fold_build2 (NE_EXPR, gfc_array_index_type, tmp, stride2);
4196 asprintf (&msg, "%s for dimension %d of array '%s'",
4197 gfc_msg_bounds, n+1, sym->name);
4198 gfc_trans_runtime_check (tmp, msg, &block, &loc);
4199 gfc_free (msg);
4200 }
4201 }
4202 else
4203 {
4204 /* For assumed shape arrays move the upper bound by the same amount
4205 as the lower bound. */
4206 tmp = build2 (MINUS_EXPR, gfc_array_index_type, dubound, dlbound);
4207 tmp = fold_build2 (PLUS_EXPR, gfc_array_index_type, tmp, lbound);
4208 gfc_add_modify_expr (&block, ubound, tmp);
4209 }
4210 /* The offset of this dimension. offset = offset - lbound * stride. */
4211 tmp = fold_build2 (MULT_EXPR, gfc_array_index_type, lbound, stride);
4212 offset = fold_build2 (MINUS_EXPR, gfc_array_index_type, offset, tmp);
4213
4214 /* The size of this dimension, and the stride of the next. */
4215 if (n + 1 < sym->as->rank)
4216 {
4217 stride = GFC_TYPE_ARRAY_STRIDE (type, n + 1);
4218
4219 if (no_repack || partial != NULL_TREE)
4220 {
4221 stmt_unpacked =
4222 gfc_conv_descriptor_stride (dumdesc, gfc_rank_cst[n+1]);
4223 }
4224
4225 /* Figure out the stride if not a known constant. */
4226 if (!INTEGER_CST_P (stride))
4227 {
4228 if (no_repack)
4229 stmt_packed = NULL_TREE;
4230 else
4231 {
4232 /* Calculate stride = size * (ubound + 1 - lbound). */
4233 tmp = fold_build2 (MINUS_EXPR, gfc_array_index_type,
4234 gfc_index_one_node, lbound);
4235 tmp = fold_build2 (PLUS_EXPR, gfc_array_index_type,
4236 ubound, tmp);
4237 size = fold_build2 (MULT_EXPR, gfc_array_index_type,
4238 size, tmp);
4239 stmt_packed = size;
4240 }
4241
4242 /* Assign the stride. */
4243 if (stmt_packed != NULL_TREE && stmt_unpacked != NULL_TREE)
4244 tmp = build3 (COND_EXPR, gfc_array_index_type, partial,
4245 stmt_unpacked, stmt_packed);
4246 else
4247 tmp = (stmt_packed != NULL_TREE) ? stmt_packed : stmt_unpacked;
4248 gfc_add_modify_expr (&block, stride, tmp);
4249 }
4250 }
4251 else
4252 {
4253 stride = GFC_TYPE_ARRAY_SIZE (type);
4254
4255 if (stride && !INTEGER_CST_P (stride))
4256 {
4257 /* Calculate size = stride * (ubound + 1 - lbound). */
4258 tmp = fold_build2 (MINUS_EXPR, gfc_array_index_type,
4259 gfc_index_one_node, lbound);
4260 tmp = fold_build2 (PLUS_EXPR, gfc_array_index_type,
4261 ubound, tmp);
4262 tmp = fold_build2 (MULT_EXPR, gfc_array_index_type,
4263 GFC_TYPE_ARRAY_STRIDE (type, n), tmp);
4264 gfc_add_modify_expr (&block, stride, tmp);
4265 }
4266 }
4267 }
4268
4269 /* Set the offset. */
4270 if (TREE_CODE (GFC_TYPE_ARRAY_OFFSET (type)) == VAR_DECL)
4271 gfc_add_modify_expr (&block, GFC_TYPE_ARRAY_OFFSET (type), offset);
4272
4273 gfc_trans_vla_type_sizes (sym, &block);
4274
4275 stmt = gfc_finish_block (&block);
4276
4277 gfc_start_block (&block);
4278
4279 /* Only do the entry/initialization code if the arg is present. */
4280 dumdesc = GFC_DECL_SAVED_DESCRIPTOR (tmpdesc);
4281 optional_arg = (sym->attr.optional
4282 || (sym->ns->proc_name->attr.entry_master
4283 && sym->attr.dummy));
4284 if (optional_arg)
4285 {
4286 tmp = gfc_conv_expr_present (sym);
4287 stmt = build3_v (COND_EXPR, tmp, stmt, build_empty_stmt ());
4288 }
4289 gfc_add_expr_to_block (&block, stmt);
4290
4291 /* Add the main function body. */
4292 gfc_add_expr_to_block (&block, body);
4293
4294 /* Cleanup code. */
4295 if (!no_repack)
4296 {
4297 gfc_start_block (&cleanup);
4298
4299 if (sym->attr.intent != INTENT_IN)
4300 {
4301 /* Copy the data back. */
4302 tmp = build_call_expr (gfor_fndecl_in_unpack, 2, dumdesc, tmpdesc);
4303 gfc_add_expr_to_block (&cleanup, tmp);
4304 }
4305
4306 /* Free the temporary. */
4307 tmp = gfc_call_free (tmpdesc);
4308 gfc_add_expr_to_block (&cleanup, tmp);
4309
4310 stmt = gfc_finish_block (&cleanup);
4311
4312 /* Only do the cleanup if the array was repacked. */
4313 tmp = build_fold_indirect_ref (dumdesc);
4314 tmp = gfc_conv_descriptor_data_get (tmp);
4315 tmp = build2 (NE_EXPR, boolean_type_node, tmp, tmpdesc);
4316 stmt = build3_v (COND_EXPR, tmp, stmt, build_empty_stmt ());
4317
4318 if (optional_arg)
4319 {
4320 tmp = gfc_conv_expr_present (sym);
4321 stmt = build3_v (COND_EXPR, tmp, stmt, build_empty_stmt ());
4322 }
4323 gfc_add_expr_to_block (&block, stmt);
4324 }
4325 /* We don't need to free any memory allocated by internal_pack as it will
4326 be freed at the end of the function by pop_context. */
4327 return gfc_finish_block (&block);
4328 }
4329
4330
4331 /* Convert an array for passing as an actual argument. Expressions and
4332 vector subscripts are evaluated and stored in a temporary, which is then
4333 passed. For whole arrays the descriptor is passed. For array sections
4334 a modified copy of the descriptor is passed, but using the original data.
4335
4336 This function is also used for array pointer assignments, and there
4337 are three cases:
4338
4339 - se->want_pointer && !se->direct_byref
4340 EXPR is an actual argument. On exit, se->expr contains a
4341 pointer to the array descriptor.
4342
4343 - !se->want_pointer && !se->direct_byref
4344 EXPR is an actual argument to an intrinsic function or the
4345 left-hand side of a pointer assignment. On exit, se->expr
4346 contains the descriptor for EXPR.
4347
4348 - !se->want_pointer && se->direct_byref
4349 EXPR is the right-hand side of a pointer assignment and
4350 se->expr is the descriptor for the previously-evaluated
4351 left-hand side. The function creates an assignment from
4352 EXPR to se->expr. */
4353
4354 void
4355 gfc_conv_expr_descriptor (gfc_se * se, gfc_expr * expr, gfc_ss * ss)
4356 {
4357 gfc_loopinfo loop;
4358 gfc_ss *secss;
4359 gfc_ss_info *info;
4360 int need_tmp;
4361 int n;
4362 tree tmp;
4363 tree desc;
4364 stmtblock_t block;
4365 tree start;
4366 tree offset;
4367 int full;
4368
4369 gcc_assert (ss != gfc_ss_terminator);
4370
4371 /* Special case things we know we can pass easily. */
4372 switch (expr->expr_type)
4373 {
4374 case EXPR_VARIABLE:
4375 /* If we have a linear array section, we can pass it directly.
4376 Otherwise we need to copy it into a temporary. */
4377
4378 /* Find the SS for the array section. */
4379 secss = ss;
4380 while (secss != gfc_ss_terminator && secss->type != GFC_SS_SECTION)
4381 secss = secss->next;
4382
4383 gcc_assert (secss != gfc_ss_terminator);
4384 info = &secss->data.info;
4385
4386 /* Get the descriptor for the array. */
4387 gfc_conv_ss_descriptor (&se->pre, secss, 0);
4388 desc = info->descriptor;
4389
4390 need_tmp = gfc_ref_needs_temporary_p (expr->ref);
4391 if (need_tmp)
4392 full = 0;
4393 else if (GFC_ARRAY_TYPE_P (TREE_TYPE (desc)))
4394 {
4395 /* Create a new descriptor if the array doesn't have one. */
4396 full = 0;
4397 }
4398 else if (info->ref->u.ar.type == AR_FULL)
4399 full = 1;
4400 else if (se->direct_byref)
4401 full = 0;
4402 else
4403 full = gfc_full_array_ref_p (info->ref);
4404
4405 if (full)
4406 {
4407 if (se->direct_byref)
4408 {
4409 /* Copy the descriptor for pointer assignments. */
4410 gfc_add_modify_expr (&se->pre, se->expr, desc);
4411 }
4412 else if (se->want_pointer)
4413 {
4414 /* We pass full arrays directly. This means that pointers and
4415 allocatable arrays should also work. */
4416 se->expr = build_fold_addr_expr (desc);
4417 }
4418 else
4419 {
4420 se->expr = desc;
4421 }
4422
4423 if (expr->ts.type == BT_CHARACTER)
4424 se->string_length = gfc_get_expr_charlen (expr);
4425
4426 return;
4427 }
4428 break;
4429
4430 case EXPR_FUNCTION:
4431 /* A transformational function return value will be a temporary
4432 array descriptor. We still need to go through the scalarizer
4433 to create the descriptor. Elemental functions ar handled as
4434 arbitrary expressions, i.e. copy to a temporary. */
4435 secss = ss;
4436 /* Look for the SS for this function. */
4437 while (secss != gfc_ss_terminator
4438 && (secss->type != GFC_SS_FUNCTION || secss->expr != expr))
4439 secss = secss->next;
4440
4441 if (se->direct_byref)
4442 {
4443 gcc_assert (secss != gfc_ss_terminator);
4444
4445 /* For pointer assignments pass the descriptor directly. */
4446 se->ss = secss;
4447 se->expr = build_fold_addr_expr (se->expr);
4448 gfc_conv_expr (se, expr);
4449 return;
4450 }
4451
4452 if (secss == gfc_ss_terminator)
4453 {
4454 /* Elemental function. */
4455 need_tmp = 1;
4456 info = NULL;
4457 }
4458 else
4459 {
4460 /* Transformational function. */
4461 info = &secss->data.info;
4462 need_tmp = 0;
4463 }
4464 break;
4465
4466 case EXPR_ARRAY:
4467 /* Constant array constructors don't need a temporary. */
4468 if (ss->type == GFC_SS_CONSTRUCTOR
4469 && expr->ts.type != BT_CHARACTER
4470 && gfc_constant_array_constructor_p (expr->value.constructor))
4471 {
4472 need_tmp = 0;
4473 info = &ss->data.info;
4474 secss = ss;
4475 }
4476 else
4477 {
4478 need_tmp = 1;
4479 secss = NULL;
4480 info = NULL;
4481 }
4482 break;
4483
4484 default:
4485 /* Something complicated. Copy it into a temporary. */
4486 need_tmp = 1;
4487 secss = NULL;
4488 info = NULL;
4489 break;
4490 }
4491
4492
4493 gfc_init_loopinfo (&loop);
4494
4495 /* Associate the SS with the loop. */
4496 gfc_add_ss_to_loop (&loop, ss);
4497
4498 /* Tell the scalarizer not to bother creating loop variables, etc. */
4499 if (!need_tmp)
4500 loop.array_parameter = 1;
4501 else
4502 /* The right-hand side of a pointer assignment mustn't use a temporary. */
4503 gcc_assert (!se->direct_byref);
4504
4505 /* Setup the scalarizing loops and bounds. */
4506 gfc_conv_ss_startstride (&loop);
4507
4508 if (need_tmp)
4509 {
4510 /* Tell the scalarizer to make a temporary. */
4511 loop.temp_ss = gfc_get_ss ();
4512 loop.temp_ss->type = GFC_SS_TEMP;
4513 loop.temp_ss->next = gfc_ss_terminator;
4514 if (expr->ts.type == BT_CHARACTER)
4515 {
4516 if (expr->ts.cl == NULL)
4517 {
4518 /* This had better be a substring reference! */
4519 gfc_ref *char_ref = expr->ref;
4520 for (; char_ref; char_ref = char_ref->next)
4521 if (char_ref->type == REF_SUBSTRING)
4522 {
4523 mpz_t char_len;
4524 expr->ts.cl = gfc_get_charlen ();
4525 expr->ts.cl->next = char_ref->u.ss.length->next;
4526 char_ref->u.ss.length->next = expr->ts.cl;
4527
4528 mpz_init_set_ui (char_len, 1);
4529 mpz_add (char_len, char_len,
4530 char_ref->u.ss.end->value.integer);
4531 mpz_sub (char_len, char_len,
4532 char_ref->u.ss.start->value.integer);
4533 expr->ts.cl->backend_decl
4534 = gfc_conv_mpz_to_tree (char_len,
4535 gfc_default_character_kind);
4536 /* Cast is necessary for *-charlen refs. */
4537 expr->ts.cl->backend_decl
4538 = convert (gfc_charlen_type_node,
4539 expr->ts.cl->backend_decl);
4540 mpz_clear (char_len);
4541 break;
4542 }
4543 gcc_assert (char_ref != NULL);
4544 loop.temp_ss->data.temp.type
4545 = gfc_typenode_for_spec (&expr->ts);
4546 loop.temp_ss->string_length = expr->ts.cl->backend_decl;
4547 }
4548 else if (expr->ts.cl->length
4549 && expr->ts.cl->length->expr_type == EXPR_CONSTANT)
4550 {
4551 expr->ts.cl->backend_decl
4552 = gfc_conv_mpz_to_tree (expr->ts.cl->length->value.integer,
4553 expr->ts.cl->length->ts.kind);
4554 loop.temp_ss->data.temp.type
4555 = gfc_typenode_for_spec (&expr->ts);
4556 loop.temp_ss->string_length
4557 = TYPE_SIZE_UNIT (loop.temp_ss->data.temp.type);
4558 }
4559 else
4560 {
4561 loop.temp_ss->data.temp.type
4562 = gfc_typenode_for_spec (&expr->ts);
4563 loop.temp_ss->string_length = expr->ts.cl->backend_decl;
4564 }
4565 se->string_length = loop.temp_ss->string_length;
4566 }
4567 else
4568 {
4569 loop.temp_ss->data.temp.type
4570 = gfc_typenode_for_spec (&expr->ts);
4571 loop.temp_ss->string_length = NULL;
4572 }
4573 loop.temp_ss->data.temp.dimen = loop.dimen;
4574 gfc_add_ss_to_loop (&loop, loop.temp_ss);
4575 }
4576
4577 gfc_conv_loop_setup (&loop);
4578
4579 if (need_tmp)
4580 {
4581 /* Copy into a temporary and pass that. We don't need to copy the data
4582 back because expressions and vector subscripts must be INTENT_IN. */
4583 /* TODO: Optimize passing function return values. */
4584 gfc_se lse;
4585 gfc_se rse;
4586
4587 /* Start the copying loops. */
4588 gfc_mark_ss_chain_used (loop.temp_ss, 1);
4589 gfc_mark_ss_chain_used (ss, 1);
4590 gfc_start_scalarized_body (&loop, &block);
4591
4592 /* Copy each data element. */
4593 gfc_init_se (&lse, NULL);
4594 gfc_copy_loopinfo_to_se (&lse, &loop);
4595 gfc_init_se (&rse, NULL);
4596 gfc_copy_loopinfo_to_se (&rse, &loop);
4597
4598 lse.ss = loop.temp_ss;
4599 rse.ss = ss;
4600
4601 gfc_conv_scalarized_array_ref (&lse, NULL);
4602 if (expr->ts.type == BT_CHARACTER)
4603 {
4604 gfc_conv_expr (&rse, expr);
4605 if (POINTER_TYPE_P (TREE_TYPE (rse.expr)))
4606 rse.expr = build_fold_indirect_ref (rse.expr);
4607 }
4608 else
4609 gfc_conv_expr_val (&rse, expr);
4610
4611 gfc_add_block_to_block (&block, &rse.pre);
4612 gfc_add_block_to_block (&block, &lse.pre);
4613
4614 gfc_add_modify_expr (&block, lse.expr, rse.expr);
4615
4616 /* Finish the copying loops. */
4617 gfc_trans_scalarizing_loops (&loop, &block);
4618
4619 desc = loop.temp_ss->data.info.descriptor;
4620
4621 gcc_assert (is_gimple_lvalue (desc));
4622 }
4623 else if (expr->expr_type == EXPR_FUNCTION)
4624 {
4625 desc = info->descriptor;
4626 se->string_length = ss->string_length;
4627 }
4628 else
4629 {
4630 /* We pass sections without copying to a temporary. Make a new
4631 descriptor and point it at the section we want. The loop variable
4632 limits will be the limits of the section.
4633 A function may decide to repack the array to speed up access, but
4634 we're not bothered about that here. */
4635 int dim, ndim;
4636 tree parm;
4637 tree parmtype;
4638 tree stride;
4639 tree from;
4640 tree to;
4641 tree base;
4642
4643 /* Set the string_length for a character array. */
4644 if (expr->ts.type == BT_CHARACTER)
4645 se->string_length = gfc_get_expr_charlen (expr);
4646
4647 desc = info->descriptor;
4648 gcc_assert (secss && secss != gfc_ss_terminator);
4649 if (se->direct_byref)
4650 {
4651 /* For pointer assignments we fill in the destination. */
4652 parm = se->expr;
4653 parmtype = TREE_TYPE (parm);
4654 }
4655 else
4656 {
4657 /* Otherwise make a new one. */
4658 parmtype = gfc_get_element_type (TREE_TYPE (desc));
4659 parmtype = gfc_get_array_type_bounds (parmtype, loop.dimen,
4660 loop.from, loop.to, 0);
4661 parm = gfc_create_var (parmtype, "parm");
4662 }
4663
4664 offset = gfc_index_zero_node;
4665 dim = 0;
4666
4667 /* The following can be somewhat confusing. We have two
4668 descriptors, a new one and the original array.
4669 {parm, parmtype, dim} refer to the new one.
4670 {desc, type, n, secss, loop} refer to the original, which maybe
4671 a descriptorless array.
4672 The bounds of the scalarization are the bounds of the section.
4673 We don't have to worry about numeric overflows when calculating
4674 the offsets because all elements are within the array data. */
4675
4676 /* Set the dtype. */
4677 tmp = gfc_conv_descriptor_dtype (parm);
4678 gfc_add_modify_expr (&loop.pre, tmp, gfc_get_dtype (parmtype));
4679
4680 if (se->direct_byref)
4681 base = gfc_index_zero_node;
4682 else if (GFC_ARRAY_TYPE_P (TREE_TYPE (desc)))
4683 base = gfc_evaluate_now (gfc_conv_array_offset (desc), &loop.pre);
4684 else
4685 base = NULL_TREE;
4686
4687 ndim = info->ref ? info->ref->u.ar.dimen : info->dimen;
4688 for (n = 0; n < ndim; n++)
4689 {
4690 stride = gfc_conv_array_stride (desc, n);
4691
4692 /* Work out the offset. */
4693 if (info->ref
4694 && info->ref->u.ar.dimen_type[n] == DIMEN_ELEMENT)
4695 {
4696 gcc_assert (info->subscript[n]
4697 && info->subscript[n]->type == GFC_SS_SCALAR);
4698 start = info->subscript[n]->data.scalar.expr;
4699 }
4700 else
4701 {
4702 /* Check we haven't somehow got out of sync. */
4703 gcc_assert (info->dim[dim] == n);
4704
4705 /* Evaluate and remember the start of the section. */
4706 start = info->start[dim];
4707 stride = gfc_evaluate_now (stride, &loop.pre);
4708 }
4709
4710 tmp = gfc_conv_array_lbound (desc, n);
4711 tmp = fold_build2 (MINUS_EXPR, TREE_TYPE (tmp), start, tmp);
4712
4713 tmp = fold_build2 (MULT_EXPR, TREE_TYPE (tmp), tmp, stride);
4714 offset = fold_build2 (PLUS_EXPR, TREE_TYPE (tmp), offset, tmp);
4715
4716 if (info->ref
4717 && info->ref->u.ar.dimen_type[n] == DIMEN_ELEMENT)
4718 {
4719 /* For elemental dimensions, we only need the offset. */
4720 continue;
4721 }
4722
4723 /* Vector subscripts need copying and are handled elsewhere. */
4724 if (info->ref)
4725 gcc_assert (info->ref->u.ar.dimen_type[n] == DIMEN_RANGE);
4726
4727 /* Set the new lower bound. */
4728 from = loop.from[dim];
4729 to = loop.to[dim];
4730
4731 /* If we have an array section or are assigning to a pointer,
4732 make sure that the lower bound is 1. References to the full
4733 array should otherwise keep the original bounds. */
4734 if ((!info->ref
4735 || info->ref->u.ar.type != AR_FULL
4736 || se->direct_byref)
4737 && !integer_onep (from))
4738 {
4739 tmp = fold_build2 (MINUS_EXPR, gfc_array_index_type,
4740 gfc_index_one_node, from);
4741 to = fold_build2 (PLUS_EXPR, gfc_array_index_type, to, tmp);
4742 from = gfc_index_one_node;
4743 }
4744 tmp = gfc_conv_descriptor_lbound (parm, gfc_rank_cst[dim]);
4745 gfc_add_modify_expr (&loop.pre, tmp, from);
4746
4747 /* Set the new upper bound. */
4748 tmp = gfc_conv_descriptor_ubound (parm, gfc_rank_cst[dim]);
4749 gfc_add_modify_expr (&loop.pre, tmp, to);
4750
4751 /* Multiply the stride by the section stride to get the
4752 total stride. */
4753 stride = fold_build2 (MULT_EXPR, gfc_array_index_type,
4754 stride, info->stride[dim]);
4755
4756 if (se->direct_byref)
4757 {
4758 base = fold_build2 (MINUS_EXPR, TREE_TYPE (base),
4759 base, stride);
4760 }
4761 else if (GFC_ARRAY_TYPE_P (TREE_TYPE (desc)))
4762 {
4763 tmp = gfc_conv_array_lbound (desc, n);
4764 tmp = fold_build2 (MINUS_EXPR, TREE_TYPE (base),
4765 tmp, loop.from[dim]);
4766 tmp = fold_build2 (MULT_EXPR, TREE_TYPE (base),
4767 tmp, gfc_conv_array_stride (desc, n));
4768 base = fold_build2 (PLUS_EXPR, TREE_TYPE (base),
4769 tmp, base);
4770 }
4771
4772 /* Store the new stride. */
4773 tmp = gfc_conv_descriptor_stride (parm, gfc_rank_cst[dim]);
4774 gfc_add_modify_expr (&loop.pre, tmp, stride);
4775
4776 dim++;
4777 }
4778
4779 if (se->data_not_needed)
4780 gfc_conv_descriptor_data_set (&loop.pre, parm, gfc_index_zero_node);
4781 else
4782 {
4783 /* Point the data pointer at the first element in the section. */
4784 tmp = gfc_conv_array_data (desc);
4785 tmp = build_fold_indirect_ref (tmp);
4786 tmp = gfc_build_array_ref (tmp, offset);
4787 offset = gfc_build_addr_expr (gfc_array_dataptr_type (desc), tmp);
4788 gfc_conv_descriptor_data_set (&loop.pre, parm, offset);
4789 }
4790
4791 if ((se->direct_byref || GFC_ARRAY_TYPE_P (TREE_TYPE (desc)))
4792 && !se->data_not_needed)
4793 {
4794 /* Set the offset. */
4795 tmp = gfc_conv_descriptor_offset (parm);
4796 gfc_add_modify_expr (&loop.pre, tmp, base);
4797 }
4798 else
4799 {
4800 /* Only the callee knows what the correct offset it, so just set
4801 it to zero here. */
4802 tmp = gfc_conv_descriptor_offset (parm);
4803 gfc_add_modify_expr (&loop.pre, tmp, gfc_index_zero_node);
4804 }
4805 desc = parm;
4806 }
4807
4808 if (!se->direct_byref)
4809 {
4810 /* Get a pointer to the new descriptor. */
4811 if (se->want_pointer)
4812 se->expr = build_fold_addr_expr (desc);
4813 else
4814 se->expr = desc;
4815 }
4816
4817 gfc_add_block_to_block (&se->pre, &loop.pre);
4818 gfc_add_block_to_block (&se->post, &loop.post);
4819
4820 /* Cleanup the scalarizer. */
4821 gfc_cleanup_loop (&loop);
4822 }
4823
4824
4825 /* Convert an array for passing as an actual parameter. */
4826 /* TODO: Optimize passing g77 arrays. */
4827
4828 void
4829 gfc_conv_array_parameter (gfc_se * se, gfc_expr * expr, gfc_ss * ss, int g77)
4830 {
4831 tree ptr;
4832 tree desc;
4833 tree tmp = NULL_TREE;
4834 tree stmt;
4835 tree parent = DECL_CONTEXT (current_function_decl);
4836 bool full_array_var, this_array_result;
4837 gfc_symbol *sym;
4838 stmtblock_t block;
4839
4840 full_array_var = (expr->expr_type == EXPR_VARIABLE
4841 && expr->ref->u.ar.type == AR_FULL);
4842 sym = full_array_var ? expr->symtree->n.sym : NULL;
4843
4844 if (expr->expr_type == EXPR_ARRAY && expr->ts.type == BT_CHARACTER)
4845 {
4846 get_array_ctor_strlen (&se->pre, expr->value.constructor, &tmp);
4847 expr->ts.cl->backend_decl = gfc_evaluate_now (tmp, &se->pre);
4848 se->string_length = expr->ts.cl->backend_decl;
4849 }
4850
4851 /* Is this the result of the enclosing procedure? */
4852 this_array_result = (full_array_var && sym->attr.flavor == FL_PROCEDURE);
4853 if (this_array_result
4854 && (sym->backend_decl != current_function_decl)
4855 && (sym->backend_decl != parent))
4856 this_array_result = false;
4857
4858 /* Passing address of the array if it is not pointer or assumed-shape. */
4859 if (full_array_var && g77 && !this_array_result)
4860 {
4861 tmp = gfc_get_symbol_decl (sym);
4862
4863 if (sym->ts.type == BT_CHARACTER)
4864 se->string_length = sym->ts.cl->backend_decl;
4865 if (!sym->attr.pointer && sym->as->type != AS_ASSUMED_SHAPE
4866 && !sym->attr.allocatable)
4867 {
4868 /* Some variables are declared directly, others are declared as
4869 pointers and allocated on the heap. */
4870 if (sym->attr.dummy || POINTER_TYPE_P (TREE_TYPE (tmp)))
4871 se->expr = tmp;
4872 else
4873 se->expr = build_fold_addr_expr (tmp);
4874 return;
4875 }
4876 if (sym->attr.allocatable)
4877 {
4878 if (sym->attr.dummy)
4879 {
4880 gfc_conv_expr_descriptor (se, expr, ss);
4881 se->expr = gfc_conv_array_data (se->expr);
4882 }
4883 else
4884 se->expr = gfc_conv_array_data (tmp);
4885 return;
4886 }
4887 }
4888
4889 if (this_array_result)
4890 {
4891 /* Result of the enclosing function. */
4892 gfc_conv_expr_descriptor (se, expr, ss);
4893 se->expr = build_fold_addr_expr (se->expr);
4894
4895 if (g77 && TREE_TYPE (TREE_TYPE (se->expr)) != NULL_TREE
4896 && GFC_DESCRIPTOR_TYPE_P (TREE_TYPE (TREE_TYPE (se->expr))))
4897 se->expr = gfc_conv_array_data (build_fold_indirect_ref (se->expr));
4898
4899 return;
4900 }
4901 else
4902 {
4903 /* Every other type of array. */
4904 se->want_pointer = 1;
4905 gfc_conv_expr_descriptor (se, expr, ss);
4906 }
4907
4908
4909 /* Deallocate the allocatable components of structures that are
4910 not variable. */
4911 if (expr->ts.type == BT_DERIVED
4912 && expr->ts.derived->attr.alloc_comp
4913 && expr->expr_type != EXPR_VARIABLE)
4914 {
4915 tmp = build_fold_indirect_ref (se->expr);
4916 tmp = gfc_deallocate_alloc_comp (expr->ts.derived, tmp, expr->rank);
4917 gfc_add_expr_to_block (&se->post, tmp);
4918 }
4919
4920 if (g77)
4921 {
4922 desc = se->expr;
4923 /* Repack the array. */
4924 ptr = build_call_expr (gfor_fndecl_in_pack, 1, desc);
4925 ptr = gfc_evaluate_now (ptr, &se->pre);
4926 se->expr = ptr;
4927
4928 gfc_start_block (&block);
4929
4930 /* Copy the data back. */
4931 tmp = build_call_expr (gfor_fndecl_in_unpack, 2, desc, ptr);
4932 gfc_add_expr_to_block (&block, tmp);
4933
4934 /* Free the temporary. */
4935 tmp = gfc_call_free (convert (pvoid_type_node, ptr));
4936 gfc_add_expr_to_block (&block, tmp);
4937
4938 stmt = gfc_finish_block (&block);
4939
4940 gfc_init_block (&block);
4941 /* Only if it was repacked. This code needs to be executed before the
4942 loop cleanup code. */
4943 tmp = build_fold_indirect_ref (desc);
4944 tmp = gfc_conv_array_data (tmp);
4945 tmp = build2 (NE_EXPR, boolean_type_node,
4946 fold_convert (TREE_TYPE (tmp), ptr), tmp);
4947 tmp = build3_v (COND_EXPR, tmp, stmt, build_empty_stmt ());
4948
4949 gfc_add_expr_to_block (&block, tmp);
4950 gfc_add_block_to_block (&block, &se->post);
4951
4952 gfc_init_block (&se->post);
4953 gfc_add_block_to_block (&se->post, &block);
4954 }
4955 }
4956
4957
4958 /* Generate code to deallocate an array, if it is allocated. */
4959
4960 tree
4961 gfc_trans_dealloc_allocated (tree descriptor)
4962 {
4963 tree tmp;
4964 tree ptr;
4965 tree var;
4966 stmtblock_t block;
4967
4968 gfc_start_block (&block);
4969
4970 var = gfc_conv_descriptor_data_get (descriptor);
4971 STRIP_NOPS (var);
4972 tmp = gfc_create_var (gfc_array_index_type, NULL);
4973 ptr = build_fold_addr_expr (tmp);
4974
4975 /* Call array_deallocate with an int* present in the second argument.
4976 Although it is ignored here, it's presence ensures that arrays that
4977 are already deallocated are ignored. */
4978 tmp = build_call_expr (gfor_fndecl_deallocate, 2, var, ptr);
4979 gfc_add_expr_to_block (&block, tmp);
4980
4981 /* Zero the data pointer. */
4982 tmp = build2 (MODIFY_EXPR, void_type_node,
4983 var, build_int_cst (TREE_TYPE (var), 0));
4984 gfc_add_expr_to_block (&block, tmp);
4985
4986 return gfc_finish_block (&block);
4987 }
4988
4989
4990 /* This helper function calculates the size in words of a full array. */
4991
4992 static tree
4993 get_full_array_size (stmtblock_t *block, tree decl, int rank)
4994 {
4995 tree idx;
4996 tree nelems;
4997 tree tmp;
4998 idx = gfc_rank_cst[rank - 1];
4999 nelems = gfc_conv_descriptor_ubound (decl, idx);
5000 tmp = gfc_conv_descriptor_lbound (decl, idx);
5001 tmp = build2 (MINUS_EXPR, gfc_array_index_type, nelems, tmp);
5002 tmp = build2 (PLUS_EXPR, gfc_array_index_type,
5003 tmp, gfc_index_one_node);
5004 tmp = gfc_evaluate_now (tmp, block);
5005
5006 nelems = gfc_conv_descriptor_stride (decl, idx);
5007 tmp = build2 (MULT_EXPR, gfc_array_index_type, nelems, tmp);
5008 return gfc_evaluate_now (tmp, block);
5009 }
5010
5011
5012 /* Allocate dest to the same size as src, and copy src -> dest. */
5013
5014 tree
5015 gfc_duplicate_allocatable(tree dest, tree src, tree type, int rank)
5016 {
5017 tree tmp;
5018 tree size;
5019 tree nelems;
5020 tree null_cond;
5021 tree null_data;
5022 stmtblock_t block;
5023
5024 /* If the source is null, set the destination to null. */
5025 gfc_init_block (&block);
5026 gfc_conv_descriptor_data_set (&block, dest, null_pointer_node);
5027 null_data = gfc_finish_block (&block);
5028
5029 gfc_init_block (&block);
5030
5031 nelems = get_full_array_size (&block, src, rank);
5032 size = fold_build2 (MULT_EXPR, gfc_array_index_type, nelems,
5033 fold_convert (gfc_array_index_type,
5034 TYPE_SIZE_UNIT (gfc_get_element_type (type))));
5035
5036 /* Allocate memory to the destination. */
5037 tmp = gfc_call_malloc (&block, TREE_TYPE (gfc_conv_descriptor_data_get (src)),
5038 size);
5039 gfc_conv_descriptor_data_set (&block, dest, tmp);
5040
5041 /* We know the temporary and the value will be the same length,
5042 so can use memcpy. */
5043 tmp = built_in_decls[BUILT_IN_MEMCPY];
5044 tmp = build_call_expr (tmp, 3, gfc_conv_descriptor_data_get (dest),
5045 gfc_conv_descriptor_data_get (src), size);
5046 gfc_add_expr_to_block (&block, tmp);
5047 tmp = gfc_finish_block (&block);
5048
5049 /* Null the destination if the source is null; otherwise do
5050 the allocate and copy. */
5051 null_cond = gfc_conv_descriptor_data_get (src);
5052 null_cond = convert (pvoid_type_node, null_cond);
5053 null_cond = build2 (NE_EXPR, boolean_type_node, null_cond,
5054 null_pointer_node);
5055 return build3_v (COND_EXPR, null_cond, tmp, null_data);
5056 }
5057
5058
5059 /* Recursively traverse an object of derived type, generating code to
5060 deallocate, nullify or copy allocatable components. This is the work horse
5061 function for the functions named in this enum. */
5062
5063 enum {DEALLOCATE_ALLOC_COMP = 1, NULLIFY_ALLOC_COMP, COPY_ALLOC_COMP};
5064
5065 static tree
5066 structure_alloc_comps (gfc_symbol * der_type, tree decl,
5067 tree dest, int rank, int purpose)
5068 {
5069 gfc_component *c;
5070 gfc_loopinfo loop;
5071 stmtblock_t fnblock;
5072 stmtblock_t loopbody;
5073 tree tmp;
5074 tree comp;
5075 tree dcmp;
5076 tree nelems;
5077 tree index;
5078 tree var;
5079 tree cdecl;
5080 tree ctype;
5081 tree vref, dref;
5082 tree null_cond = NULL_TREE;
5083
5084 gfc_init_block (&fnblock);
5085
5086 if (POINTER_TYPE_P (TREE_TYPE (decl)))
5087 decl = build_fold_indirect_ref (decl);
5088
5089 /* If this an array of derived types with allocatable components
5090 build a loop and recursively call this function. */
5091 if (TREE_CODE (TREE_TYPE (decl)) == ARRAY_TYPE
5092 || GFC_DESCRIPTOR_TYPE_P (TREE_TYPE (decl)))
5093 {
5094 tmp = gfc_conv_array_data (decl);
5095 var = build_fold_indirect_ref (tmp);
5096
5097 /* Get the number of elements - 1 and set the counter. */
5098 if (GFC_DESCRIPTOR_TYPE_P (TREE_TYPE (decl)))
5099 {
5100 /* Use the descriptor for an allocatable array. Since this
5101 is a full array reference, we only need the descriptor
5102 information from dimension = rank. */
5103 tmp = get_full_array_size (&fnblock, decl, rank);
5104 tmp = build2 (MINUS_EXPR, gfc_array_index_type,
5105 tmp, gfc_index_one_node);
5106
5107 null_cond = gfc_conv_descriptor_data_get (decl);
5108 null_cond = build2 (NE_EXPR, boolean_type_node, null_cond,
5109 build_int_cst (TREE_TYPE (null_cond), 0));
5110 }
5111 else
5112 {
5113 /* Otherwise use the TYPE_DOMAIN information. */
5114 tmp = array_type_nelts (TREE_TYPE (decl));
5115 tmp = fold_convert (gfc_array_index_type, tmp);
5116 }
5117
5118 /* Remember that this is, in fact, the no. of elements - 1. */
5119 nelems = gfc_evaluate_now (tmp, &fnblock);
5120 index = gfc_create_var (gfc_array_index_type, "S");
5121
5122 /* Build the body of the loop. */
5123 gfc_init_block (&loopbody);
5124
5125 vref = gfc_build_array_ref (var, index);
5126
5127 if (purpose == COPY_ALLOC_COMP)
5128 {
5129 tmp = gfc_duplicate_allocatable (dest, decl, TREE_TYPE(decl), rank);
5130 gfc_add_expr_to_block (&fnblock, tmp);
5131
5132 tmp = build_fold_indirect_ref (gfc_conv_descriptor_data_get (dest));
5133 dref = gfc_build_array_ref (tmp, index);
5134 tmp = structure_alloc_comps (der_type, vref, dref, rank, purpose);
5135 }
5136 else
5137 tmp = structure_alloc_comps (der_type, vref, NULL_TREE, rank, purpose);
5138
5139 gfc_add_expr_to_block (&loopbody, tmp);
5140
5141 /* Build the loop and return. */
5142 gfc_init_loopinfo (&loop);
5143 loop.dimen = 1;
5144 loop.from[0] = gfc_index_zero_node;
5145 loop.loopvar[0] = index;
5146 loop.to[0] = nelems;
5147 gfc_trans_scalarizing_loops (&loop, &loopbody);
5148 gfc_add_block_to_block (&fnblock, &loop.pre);
5149
5150 tmp = gfc_finish_block (&fnblock);
5151 if (null_cond != NULL_TREE)
5152 tmp = build3_v (COND_EXPR, null_cond, tmp, build_empty_stmt ());
5153
5154 return tmp;
5155 }
5156
5157 /* Otherwise, act on the components or recursively call self to
5158 act on a chain of components. */
5159 for (c = der_type->components; c; c = c->next)
5160 {
5161 bool cmp_has_alloc_comps = (c->ts.type == BT_DERIVED)
5162 && c->ts.derived->attr.alloc_comp;
5163 cdecl = c->backend_decl;
5164 ctype = TREE_TYPE (cdecl);
5165
5166 switch (purpose)
5167 {
5168 case DEALLOCATE_ALLOC_COMP:
5169 /* Do not deallocate the components of ultimate pointer
5170 components. */
5171 if (cmp_has_alloc_comps && !c->pointer)
5172 {
5173 comp = build3 (COMPONENT_REF, ctype, decl, cdecl, NULL_TREE);
5174 rank = c->as ? c->as->rank : 0;
5175 tmp = structure_alloc_comps (c->ts.derived, comp, NULL_TREE,
5176 rank, purpose);
5177 gfc_add_expr_to_block (&fnblock, tmp);
5178 }
5179
5180 if (c->allocatable)
5181 {
5182 comp = build3 (COMPONENT_REF, ctype, decl, cdecl, NULL_TREE);
5183 tmp = gfc_trans_dealloc_allocated (comp);
5184 gfc_add_expr_to_block (&fnblock, tmp);
5185 }
5186 break;
5187
5188 case NULLIFY_ALLOC_COMP:
5189 if (c->pointer)
5190 continue;
5191 else if (c->allocatable)
5192 {
5193 comp = build3 (COMPONENT_REF, ctype, decl, cdecl, NULL_TREE);
5194 gfc_conv_descriptor_data_set (&fnblock, comp, null_pointer_node);
5195 }
5196 else if (cmp_has_alloc_comps)
5197 {
5198 comp = build3 (COMPONENT_REF, ctype, decl, cdecl, NULL_TREE);
5199 rank = c->as ? c->as->rank : 0;
5200 tmp = structure_alloc_comps (c->ts.derived, comp, NULL_TREE,
5201 rank, purpose);
5202 gfc_add_expr_to_block (&fnblock, tmp);
5203 }
5204 break;
5205
5206 case COPY_ALLOC_COMP:
5207 if (c->pointer)
5208 continue;
5209
5210 /* We need source and destination components. */
5211 comp = build3 (COMPONENT_REF, ctype, decl, cdecl, NULL_TREE);
5212 dcmp = build3 (COMPONENT_REF, ctype, dest, cdecl, NULL_TREE);
5213 dcmp = fold_convert (TREE_TYPE (comp), dcmp);
5214
5215 if (c->allocatable && !cmp_has_alloc_comps)
5216 {
5217 tmp = gfc_duplicate_allocatable(dcmp, comp, ctype, c->as->rank);
5218 gfc_add_expr_to_block (&fnblock, tmp);
5219 }
5220
5221 if (cmp_has_alloc_comps)
5222 {
5223 rank = c->as ? c->as->rank : 0;
5224 tmp = fold_convert (TREE_TYPE (dcmp), comp);
5225 gfc_add_modify_expr (&fnblock, dcmp, tmp);
5226 tmp = structure_alloc_comps (c->ts.derived, comp, dcmp,
5227 rank, purpose);
5228 gfc_add_expr_to_block (&fnblock, tmp);
5229 }
5230 break;
5231
5232 default:
5233 gcc_unreachable ();
5234 break;
5235 }
5236 }
5237
5238 return gfc_finish_block (&fnblock);
5239 }
5240
5241 /* Recursively traverse an object of derived type, generating code to
5242 nullify allocatable components. */
5243
5244 tree
5245 gfc_nullify_alloc_comp (gfc_symbol * der_type, tree decl, int rank)
5246 {
5247 return structure_alloc_comps (der_type, decl, NULL_TREE, rank,
5248 NULLIFY_ALLOC_COMP);
5249 }
5250
5251
5252 /* Recursively traverse an object of derived type, generating code to
5253 deallocate allocatable components. */
5254
5255 tree
5256 gfc_deallocate_alloc_comp (gfc_symbol * der_type, tree decl, int rank)
5257 {
5258 return structure_alloc_comps (der_type, decl, NULL_TREE, rank,
5259 DEALLOCATE_ALLOC_COMP);
5260 }
5261
5262
5263 /* Recursively traverse an object of derived type, generating code to
5264 copy its allocatable components. */
5265
5266 tree
5267 gfc_copy_alloc_comp (gfc_symbol * der_type, tree decl, tree dest, int rank)
5268 {
5269 return structure_alloc_comps (der_type, decl, dest, rank, COPY_ALLOC_COMP);
5270 }
5271
5272
5273 /* NULLIFY an allocatable/pointer array on function entry, free it on exit.
5274 Do likewise, recursively if necessary, with the allocatable components of
5275 derived types. */
5276
5277 tree
5278 gfc_trans_deferred_array (gfc_symbol * sym, tree body)
5279 {
5280 tree type;
5281 tree tmp;
5282 tree descriptor;
5283 stmtblock_t fnblock;
5284 locus loc;
5285 int rank;
5286 bool sym_has_alloc_comp;
5287
5288 sym_has_alloc_comp = (sym->ts.type == BT_DERIVED)
5289 && sym->ts.derived->attr.alloc_comp;
5290
5291 /* Make sure the frontend gets these right. */
5292 if (!(sym->attr.pointer || sym->attr.allocatable || sym_has_alloc_comp))
5293 fatal_error ("Possible frontend bug: Deferred array size without pointer, "
5294 "allocatable attribute or derived type without allocatable "
5295 "components.");
5296
5297 gfc_init_block (&fnblock);
5298
5299 gcc_assert (TREE_CODE (sym->backend_decl) == VAR_DECL
5300 || TREE_CODE (sym->backend_decl) == PARM_DECL);
5301
5302 if (sym->ts.type == BT_CHARACTER
5303 && !INTEGER_CST_P (sym->ts.cl->backend_decl))
5304 {
5305 gfc_trans_init_string_length (sym->ts.cl, &fnblock);
5306 gfc_trans_vla_type_sizes (sym, &fnblock);
5307 }
5308
5309 /* Dummy and use associated variables don't need anything special. */
5310 if (sym->attr.dummy || sym->attr.use_assoc)
5311 {
5312 gfc_add_expr_to_block (&fnblock, body);
5313
5314 return gfc_finish_block (&fnblock);
5315 }
5316
5317 gfc_get_backend_locus (&loc);
5318 gfc_set_backend_locus (&sym->declared_at);
5319 descriptor = sym->backend_decl;
5320
5321 /* Although static, derived types with default initializers and
5322 allocatable components must not be nulled wholesale; instead they
5323 are treated component by component. */
5324 if (TREE_STATIC (descriptor) && !sym_has_alloc_comp)
5325 {
5326 /* SAVEd variables are not freed on exit. */
5327 gfc_trans_static_array_pointer (sym);
5328 return body;
5329 }
5330
5331 /* Get the descriptor type. */
5332 type = TREE_TYPE (sym->backend_decl);
5333
5334 if (sym_has_alloc_comp && !(sym->attr.pointer || sym->attr.allocatable))
5335 {
5336 if (!sym->attr.save)
5337 {
5338 rank = sym->as ? sym->as->rank : 0;
5339 tmp = gfc_nullify_alloc_comp (sym->ts.derived, descriptor, rank);
5340 gfc_add_expr_to_block (&fnblock, tmp);
5341 }
5342 }
5343 else if (!GFC_DESCRIPTOR_TYPE_P (type))
5344 {
5345 /* If the backend_decl is not a descriptor, we must have a pointer
5346 to one. */
5347 descriptor = build_fold_indirect_ref (sym->backend_decl);
5348 type = TREE_TYPE (descriptor);
5349 }
5350
5351 /* NULLIFY the data pointer. */
5352 if (GFC_DESCRIPTOR_TYPE_P (type))
5353 gfc_conv_descriptor_data_set (&fnblock, descriptor, null_pointer_node);
5354
5355 gfc_add_expr_to_block (&fnblock, body);
5356
5357 gfc_set_backend_locus (&loc);
5358
5359 /* Allocatable arrays need to be freed when they go out of scope.
5360 The allocatable components of pointers must not be touched. */
5361 if (sym_has_alloc_comp && !(sym->attr.function || sym->attr.result)
5362 && !sym->attr.pointer && !sym->attr.save)
5363 {
5364 int rank;
5365 rank = sym->as ? sym->as->rank : 0;
5366 tmp = gfc_deallocate_alloc_comp (sym->ts.derived, descriptor, rank);
5367 gfc_add_expr_to_block (&fnblock, tmp);
5368 }
5369
5370 if (sym->attr.allocatable)
5371 {
5372 tmp = gfc_trans_dealloc_allocated (sym->backend_decl);
5373 gfc_add_expr_to_block (&fnblock, tmp);
5374 }
5375
5376 return gfc_finish_block (&fnblock);
5377 }
5378
5379 /************ Expression Walking Functions ******************/
5380
5381 /* Walk a variable reference.
5382
5383 Possible extension - multiple component subscripts.
5384 x(:,:) = foo%a(:)%b(:)
5385 Transforms to
5386 forall (i=..., j=...)
5387 x(i,j) = foo%a(j)%b(i)
5388 end forall
5389 This adds a fair amount of complexity because you need to deal with more
5390 than one ref. Maybe handle in a similar manner to vector subscripts.
5391 Maybe not worth the effort. */
5392
5393
5394 static gfc_ss *
5395 gfc_walk_variable_expr (gfc_ss * ss, gfc_expr * expr)
5396 {
5397 gfc_ref *ref;
5398 gfc_array_ref *ar;
5399 gfc_ss *newss;
5400 gfc_ss *head;
5401 int n;
5402
5403 for (ref = expr->ref; ref; ref = ref->next)
5404 if (ref->type == REF_ARRAY && ref->u.ar.type != AR_ELEMENT)
5405 break;
5406
5407 for (; ref; ref = ref->next)
5408 {
5409 if (ref->type == REF_SUBSTRING)
5410 {
5411 newss = gfc_get_ss ();
5412 newss->type = GFC_SS_SCALAR;
5413 newss->expr = ref->u.ss.start;
5414 newss->next = ss;
5415 ss = newss;
5416
5417 newss = gfc_get_ss ();
5418 newss->type = GFC_SS_SCALAR;
5419 newss->expr = ref->u.ss.end;
5420 newss->next = ss;
5421 ss = newss;
5422 }
5423
5424 /* We're only interested in array sections from now on. */
5425 if (ref->type != REF_ARRAY)
5426 continue;
5427
5428 ar = &ref->u.ar;
5429 switch (ar->type)
5430 {
5431 case AR_ELEMENT:
5432 for (n = 0; n < ar->dimen; n++)
5433 {
5434 newss = gfc_get_ss ();
5435 newss->type = GFC_SS_SCALAR;
5436 newss->expr = ar->start[n];
5437 newss->next = ss;
5438 ss = newss;
5439 }
5440 break;
5441
5442 case AR_FULL:
5443 newss = gfc_get_ss ();
5444 newss->type = GFC_SS_SECTION;
5445 newss->expr = expr;
5446 newss->next = ss;
5447 newss->data.info.dimen = ar->as->rank;
5448 newss->data.info.ref = ref;
5449
5450 /* Make sure array is the same as array(:,:), this way
5451 we don't need to special case all the time. */
5452 ar->dimen = ar->as->rank;
5453 for (n = 0; n < ar->dimen; n++)
5454 {
5455 newss->data.info.dim[n] = n;
5456 ar->dimen_type[n] = DIMEN_RANGE;
5457
5458 gcc_assert (ar->start[n] == NULL);
5459 gcc_assert (ar->end[n] == NULL);
5460 gcc_assert (ar->stride[n] == NULL);
5461 }
5462 ss = newss;
5463 break;
5464
5465 case AR_SECTION:
5466 newss = gfc_get_ss ();
5467 newss->type = GFC_SS_SECTION;
5468 newss->expr = expr;
5469 newss->next = ss;
5470 newss->data.info.dimen = 0;
5471 newss->data.info.ref = ref;
5472
5473 head = newss;
5474
5475 /* We add SS chains for all the subscripts in the section. */
5476 for (n = 0; n < ar->dimen; n++)
5477 {
5478 gfc_ss *indexss;
5479
5480 switch (ar->dimen_type[n])
5481 {
5482 case DIMEN_ELEMENT:
5483 /* Add SS for elemental (scalar) subscripts. */
5484 gcc_assert (ar->start[n]);
5485 indexss = gfc_get_ss ();
5486 indexss->type = GFC_SS_SCALAR;
5487 indexss->expr = ar->start[n];
5488 indexss->next = gfc_ss_terminator;
5489 indexss->loop_chain = gfc_ss_terminator;
5490 newss->data.info.subscript[n] = indexss;
5491 break;
5492
5493 case DIMEN_RANGE:
5494 /* We don't add anything for sections, just remember this
5495 dimension for later. */
5496 newss->data.info.dim[newss->data.info.dimen] = n;
5497 newss->data.info.dimen++;
5498 break;
5499
5500 case DIMEN_VECTOR:
5501 /* Create a GFC_SS_VECTOR index in which we can store
5502 the vector's descriptor. */
5503 indexss = gfc_get_ss ();
5504 indexss->type = GFC_SS_VECTOR;
5505 indexss->expr = ar->start[n];
5506 indexss->next = gfc_ss_terminator;
5507 indexss->loop_chain = gfc_ss_terminator;
5508 newss->data.info.subscript[n] = indexss;
5509 newss->data.info.dim[newss->data.info.dimen] = n;
5510 newss->data.info.dimen++;
5511 break;
5512
5513 default:
5514 /* We should know what sort of section it is by now. */
5515 gcc_unreachable ();
5516 }
5517 }
5518 /* We should have at least one non-elemental dimension. */
5519 gcc_assert (newss->data.info.dimen > 0);
5520 ss = newss;
5521 break;
5522
5523 default:
5524 /* We should know what sort of section it is by now. */
5525 gcc_unreachable ();
5526 }
5527
5528 }
5529 return ss;
5530 }
5531
5532
5533 /* Walk an expression operator. If only one operand of a binary expression is
5534 scalar, we must also add the scalar term to the SS chain. */
5535
5536 static gfc_ss *
5537 gfc_walk_op_expr (gfc_ss * ss, gfc_expr * expr)
5538 {
5539 gfc_ss *head;
5540 gfc_ss *head2;
5541 gfc_ss *newss;
5542
5543 head = gfc_walk_subexpr (ss, expr->value.op.op1);
5544 if (expr->value.op.op2 == NULL)
5545 head2 = head;
5546 else
5547 head2 = gfc_walk_subexpr (head, expr->value.op.op2);
5548
5549 /* All operands are scalar. Pass back and let the caller deal with it. */
5550 if (head2 == ss)
5551 return head2;
5552
5553 /* All operands require scalarization. */
5554 if (head != ss && (expr->value.op.op2 == NULL || head2 != head))
5555 return head2;
5556
5557 /* One of the operands needs scalarization, the other is scalar.
5558 Create a gfc_ss for the scalar expression. */
5559 newss = gfc_get_ss ();
5560 newss->type = GFC_SS_SCALAR;
5561 if (head == ss)
5562 {
5563 /* First operand is scalar. We build the chain in reverse order, so
5564 add the scarar SS after the second operand. */
5565 head = head2;
5566 while (head && head->next != ss)
5567 head = head->next;
5568 /* Check we haven't somehow broken the chain. */
5569 gcc_assert (head);
5570 newss->next = ss;
5571 head->next = newss;
5572 newss->expr = expr->value.op.op1;
5573 }
5574 else /* head2 == head */
5575 {
5576 gcc_assert (head2 == head);
5577 /* Second operand is scalar. */
5578 newss->next = head2;
5579 head2 = newss;
5580 newss->expr = expr->value.op.op2;
5581 }
5582
5583 return head2;
5584 }
5585
5586
5587 /* Reverse a SS chain. */
5588
5589 gfc_ss *
5590 gfc_reverse_ss (gfc_ss * ss)
5591 {
5592 gfc_ss *next;
5593 gfc_ss *head;
5594
5595 gcc_assert (ss != NULL);
5596
5597 head = gfc_ss_terminator;
5598 while (ss != gfc_ss_terminator)
5599 {
5600 next = ss->next;
5601 /* Check we didn't somehow break the chain. */
5602 gcc_assert (next != NULL);
5603 ss->next = head;
5604 head = ss;
5605 ss = next;
5606 }
5607
5608 return (head);
5609 }
5610
5611
5612 /* Walk the arguments of an elemental function. */
5613
5614 gfc_ss *
5615 gfc_walk_elemental_function_args (gfc_ss * ss, gfc_actual_arglist *arg,
5616 gfc_ss_type type)
5617 {
5618 int scalar;
5619 gfc_ss *head;
5620 gfc_ss *tail;
5621 gfc_ss *newss;
5622
5623 head = gfc_ss_terminator;
5624 tail = NULL;
5625 scalar = 1;
5626 for (; arg; arg = arg->next)
5627 {
5628 if (!arg->expr)
5629 continue;
5630
5631 newss = gfc_walk_subexpr (head, arg->expr);
5632 if (newss == head)
5633 {
5634 /* Scalar argument. */
5635 newss = gfc_get_ss ();
5636 newss->type = type;
5637 newss->expr = arg->expr;
5638 newss->next = head;
5639 }
5640 else
5641 scalar = 0;
5642
5643 head = newss;
5644 if (!tail)
5645 {
5646 tail = head;
5647 while (tail->next != gfc_ss_terminator)
5648 tail = tail->next;
5649 }
5650 }
5651
5652 if (scalar)
5653 {
5654 /* If all the arguments are scalar we don't need the argument SS. */
5655 gfc_free_ss_chain (head);
5656 /* Pass it back. */
5657 return ss;
5658 }
5659
5660 /* Add it onto the existing chain. */
5661 tail->next = ss;
5662 return head;
5663 }
5664
5665
5666 /* Walk a function call. Scalar functions are passed back, and taken out of
5667 scalarization loops. For elemental functions we walk their arguments.
5668 The result of functions returning arrays is stored in a temporary outside
5669 the loop, so that the function is only called once. Hence we do not need
5670 to walk their arguments. */
5671
5672 static gfc_ss *
5673 gfc_walk_function_expr (gfc_ss * ss, gfc_expr * expr)
5674 {
5675 gfc_ss *newss;
5676 gfc_intrinsic_sym *isym;
5677 gfc_symbol *sym;
5678
5679 isym = expr->value.function.isym;
5680
5681 /* Handle intrinsic functions separately. */
5682 if (isym)
5683 return gfc_walk_intrinsic_function (ss, expr, isym);
5684
5685 sym = expr->value.function.esym;
5686 if (!sym)
5687 sym = expr->symtree->n.sym;
5688
5689 /* A function that returns arrays. */
5690 if (gfc_return_by_reference (sym) && sym->result->attr.dimension)
5691 {
5692 newss = gfc_get_ss ();
5693 newss->type = GFC_SS_FUNCTION;
5694 newss->expr = expr;
5695 newss->next = ss;
5696 newss->data.info.dimen = expr->rank;
5697 return newss;
5698 }
5699
5700 /* Walk the parameters of an elemental function. For now we always pass
5701 by reference. */
5702 if (sym->attr.elemental)
5703 return gfc_walk_elemental_function_args (ss, expr->value.function.actual,
5704 GFC_SS_REFERENCE);
5705
5706 /* Scalar functions are OK as these are evaluated outside the scalarization
5707 loop. Pass back and let the caller deal with it. */
5708 return ss;
5709 }
5710
5711
5712 /* An array temporary is constructed for array constructors. */
5713
5714 static gfc_ss *
5715 gfc_walk_array_constructor (gfc_ss * ss, gfc_expr * expr)
5716 {
5717 gfc_ss *newss;
5718 int n;
5719
5720 newss = gfc_get_ss ();
5721 newss->type = GFC_SS_CONSTRUCTOR;
5722 newss->expr = expr;
5723 newss->next = ss;
5724 newss->data.info.dimen = expr->rank;
5725 for (n = 0; n < expr->rank; n++)
5726 newss->data.info.dim[n] = n;
5727
5728 return newss;
5729 }
5730
5731
5732 /* Walk an expression. Add walked expressions to the head of the SS chain.
5733 A wholly scalar expression will not be added. */
5734
5735 static gfc_ss *
5736 gfc_walk_subexpr (gfc_ss * ss, gfc_expr * expr)
5737 {
5738 gfc_ss *head;
5739
5740 switch (expr->expr_type)
5741 {
5742 case EXPR_VARIABLE:
5743 head = gfc_walk_variable_expr (ss, expr);
5744 return head;
5745
5746 case EXPR_OP:
5747 head = gfc_walk_op_expr (ss, expr);
5748 return head;
5749
5750 case EXPR_FUNCTION:
5751 head = gfc_walk_function_expr (ss, expr);
5752 return head;
5753
5754 case EXPR_CONSTANT:
5755 case EXPR_NULL:
5756 case EXPR_STRUCTURE:
5757 /* Pass back and let the caller deal with it. */
5758 break;
5759
5760 case EXPR_ARRAY:
5761 head = gfc_walk_array_constructor (ss, expr);
5762 return head;
5763
5764 case EXPR_SUBSTRING:
5765 /* Pass back and let the caller deal with it. */
5766 break;
5767
5768 default:
5769 internal_error ("bad expression type during walk (%d)",
5770 expr->expr_type);
5771 }
5772 return ss;
5773 }
5774
5775
5776 /* Entry point for expression walking.
5777 A return value equal to the passed chain means this is
5778 a scalar expression. It is up to the caller to take whatever action is
5779 necessary to translate these. */
5780
5781 gfc_ss *
5782 gfc_walk_expr (gfc_expr * expr)
5783 {
5784 gfc_ss *res;
5785
5786 res = gfc_walk_subexpr (gfc_ss_terminator, expr);
5787 return gfc_reverse_ss (res);
5788 }