1627f7eb2Smrg /* Supporting functions for resolving DATA statement.
2*4c3eb207Smrg Copyright (C) 2002-2020 Free Software Foundation, Inc.
3627f7eb2Smrg Contributed by Lifang Zeng <zlf605@hotmail.com>
4627f7eb2Smrg
5627f7eb2Smrg This file is part of GCC.
6627f7eb2Smrg
7627f7eb2Smrg GCC is free software; you can redistribute it and/or modify it under
8627f7eb2Smrg the terms of the GNU General Public License as published by the Free
9627f7eb2Smrg Software Foundation; either version 3, or (at your option) any later
10627f7eb2Smrg version.
11627f7eb2Smrg
12627f7eb2Smrg GCC is distributed in the hope that it will be useful, but WITHOUT ANY
13627f7eb2Smrg WARRANTY; without even the implied warranty of MERCHANTABILITY or
14627f7eb2Smrg FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
15627f7eb2Smrg for more details.
16627f7eb2Smrg
17627f7eb2Smrg You should have received a copy of the GNU General Public License
18627f7eb2Smrg along with GCC; see the file COPYING3. If not see
19627f7eb2Smrg <http://www.gnu.org/licenses/>. */
20627f7eb2Smrg
21627f7eb2Smrg
22627f7eb2Smrg /* Notes for DATA statement implementation:
23627f7eb2Smrg
24627f7eb2Smrg We first assign initial value to each symbol by gfc_assign_data_value
25627f7eb2Smrg during resolving DATA statement. Refer to check_data_variable and
26627f7eb2Smrg traverse_data_list in resolve.c.
27627f7eb2Smrg
28627f7eb2Smrg The complexity exists in the handling of array section, implied do
29627f7eb2Smrg and array of struct appeared in DATA statement.
30627f7eb2Smrg
31627f7eb2Smrg We call gfc_conv_structure, gfc_con_array_array_initializer,
32627f7eb2Smrg etc., to convert the initial value. Refer to trans-expr.c and
33627f7eb2Smrg trans-array.c. */
34627f7eb2Smrg
35627f7eb2Smrg #include "config.h"
36627f7eb2Smrg #include "system.h"
37627f7eb2Smrg #include "coretypes.h"
38627f7eb2Smrg #include "gfortran.h"
39627f7eb2Smrg #include "data.h"
40627f7eb2Smrg #include "constructor.h"
41627f7eb2Smrg
42627f7eb2Smrg static void formalize_init_expr (gfc_expr *);
43627f7eb2Smrg
44627f7eb2Smrg /* Calculate the array element offset. */
45627f7eb2Smrg
46627f7eb2Smrg static void
get_array_index(gfc_array_ref * ar,mpz_t * offset)47627f7eb2Smrg get_array_index (gfc_array_ref *ar, mpz_t *offset)
48627f7eb2Smrg {
49627f7eb2Smrg gfc_expr *e;
50627f7eb2Smrg int i;
51627f7eb2Smrg mpz_t delta;
52627f7eb2Smrg mpz_t tmp;
53627f7eb2Smrg
54627f7eb2Smrg mpz_init (tmp);
55627f7eb2Smrg mpz_set_si (*offset, 0);
56627f7eb2Smrg mpz_init_set_si (delta, 1);
57627f7eb2Smrg for (i = 0; i < ar->dimen; i++)
58627f7eb2Smrg {
59627f7eb2Smrg e = gfc_copy_expr (ar->start[i]);
60627f7eb2Smrg gfc_simplify_expr (e, 1);
61627f7eb2Smrg
62627f7eb2Smrg if ((gfc_is_constant_expr (ar->as->lower[i]) == 0)
63627f7eb2Smrg || (gfc_is_constant_expr (ar->as->upper[i]) == 0)
64627f7eb2Smrg || (gfc_is_constant_expr (e) == 0))
65627f7eb2Smrg gfc_error ("non-constant array in DATA statement %L", &ar->where);
66627f7eb2Smrg
67627f7eb2Smrg mpz_set (tmp, e->value.integer);
68627f7eb2Smrg gfc_free_expr (e);
69627f7eb2Smrg mpz_sub (tmp, tmp, ar->as->lower[i]->value.integer);
70627f7eb2Smrg mpz_mul (tmp, tmp, delta);
71627f7eb2Smrg mpz_add (*offset, tmp, *offset);
72627f7eb2Smrg
73627f7eb2Smrg mpz_sub (tmp, ar->as->upper[i]->value.integer,
74627f7eb2Smrg ar->as->lower[i]->value.integer);
75627f7eb2Smrg mpz_add_ui (tmp, tmp, 1);
76627f7eb2Smrg mpz_mul (delta, tmp, delta);
77627f7eb2Smrg }
78627f7eb2Smrg mpz_clear (delta);
79627f7eb2Smrg mpz_clear (tmp);
80627f7eb2Smrg }
81627f7eb2Smrg
82627f7eb2Smrg /* Find if there is a constructor which component is equal to COM.
83627f7eb2Smrg TODO: remove this, use symbol.c(gfc_find_component) instead. */
84627f7eb2Smrg
85627f7eb2Smrg static gfc_constructor *
find_con_by_component(gfc_component * com,gfc_constructor_base base)86627f7eb2Smrg find_con_by_component (gfc_component *com, gfc_constructor_base base)
87627f7eb2Smrg {
88627f7eb2Smrg gfc_constructor *c;
89627f7eb2Smrg
90627f7eb2Smrg for (c = gfc_constructor_first (base); c; c = gfc_constructor_next (c))
91627f7eb2Smrg if (com == c->n.component)
92627f7eb2Smrg return c;
93627f7eb2Smrg
94627f7eb2Smrg return NULL;
95627f7eb2Smrg }
96627f7eb2Smrg
97627f7eb2Smrg
98627f7eb2Smrg /* Create a character type initialization expression from RVALUE.
99627f7eb2Smrg TS [and REF] describe [the substring of] the variable being initialized.
100627f7eb2Smrg INIT is the existing initializer, not NULL. Initialization is performed
101627f7eb2Smrg according to normal assignment rules. */
102627f7eb2Smrg
103627f7eb2Smrg static gfc_expr *
create_character_initializer(gfc_expr * init,gfc_typespec * ts,gfc_ref * ref,gfc_expr * rvalue)104627f7eb2Smrg create_character_initializer (gfc_expr *init, gfc_typespec *ts,
105627f7eb2Smrg gfc_ref *ref, gfc_expr *rvalue)
106627f7eb2Smrg {
107627f7eb2Smrg HOST_WIDE_INT len, start, end, tlen;
108627f7eb2Smrg gfc_char_t *dest;
109627f7eb2Smrg bool alloced_init = false;
110627f7eb2Smrg
111627f7eb2Smrg if (init && init->ts.type != BT_CHARACTER)
112627f7eb2Smrg return NULL;
113627f7eb2Smrg
114627f7eb2Smrg gfc_extract_hwi (ts->u.cl->length, &len);
115627f7eb2Smrg
116627f7eb2Smrg if (init == NULL)
117627f7eb2Smrg {
118627f7eb2Smrg /* Create a new initializer. */
119627f7eb2Smrg init = gfc_get_character_expr (ts->kind, NULL, NULL, len);
120627f7eb2Smrg init->ts = *ts;
121627f7eb2Smrg alloced_init = true;
122627f7eb2Smrg }
123627f7eb2Smrg
124627f7eb2Smrg dest = init->value.character.string;
125627f7eb2Smrg
126627f7eb2Smrg if (ref)
127627f7eb2Smrg {
128627f7eb2Smrg gfc_expr *start_expr, *end_expr;
129627f7eb2Smrg
130627f7eb2Smrg gcc_assert (ref->type == REF_SUBSTRING);
131627f7eb2Smrg
132627f7eb2Smrg /* Only set a substring of the destination. Fortran substring bounds
133627f7eb2Smrg are one-based [start, end], we want zero based [start, end). */
134627f7eb2Smrg start_expr = gfc_copy_expr (ref->u.ss.start);
135627f7eb2Smrg end_expr = gfc_copy_expr (ref->u.ss.end);
136627f7eb2Smrg
137627f7eb2Smrg if ((!gfc_simplify_expr(start_expr, 1))
138627f7eb2Smrg || !(gfc_simplify_expr(end_expr, 1)))
139627f7eb2Smrg {
140627f7eb2Smrg gfc_error ("failure to simplify substring reference in DATA "
141627f7eb2Smrg "statement at %L", &ref->u.ss.start->where);
142627f7eb2Smrg gfc_free_expr (start_expr);
143627f7eb2Smrg gfc_free_expr (end_expr);
144627f7eb2Smrg if (alloced_init)
145627f7eb2Smrg gfc_free_expr (init);
146627f7eb2Smrg return NULL;
147627f7eb2Smrg }
148627f7eb2Smrg
149627f7eb2Smrg gfc_extract_hwi (start_expr, &start);
150627f7eb2Smrg gfc_free_expr (start_expr);
151627f7eb2Smrg start--;
152627f7eb2Smrg gfc_extract_hwi (end_expr, &end);
153627f7eb2Smrg gfc_free_expr (end_expr);
154627f7eb2Smrg }
155627f7eb2Smrg else
156627f7eb2Smrg {
157627f7eb2Smrg /* Set the whole string. */
158627f7eb2Smrg start = 0;
159627f7eb2Smrg end = len;
160627f7eb2Smrg }
161627f7eb2Smrg
162627f7eb2Smrg /* Copy the initial value. */
163627f7eb2Smrg if (rvalue->ts.type == BT_HOLLERITH)
164627f7eb2Smrg len = rvalue->representation.length - rvalue->ts.u.pad;
165627f7eb2Smrg else
166627f7eb2Smrg len = rvalue->value.character.length;
167627f7eb2Smrg
168627f7eb2Smrg tlen = end - start;
169627f7eb2Smrg if (len > tlen)
170627f7eb2Smrg {
171627f7eb2Smrg if (tlen < 0)
172627f7eb2Smrg {
173627f7eb2Smrg gfc_warning_now (0, "Unused initialization string at %L because "
174627f7eb2Smrg "variable has zero length", &rvalue->where);
175627f7eb2Smrg len = 0;
176627f7eb2Smrg }
177627f7eb2Smrg else
178627f7eb2Smrg {
179627f7eb2Smrg gfc_warning_now (0, "Initialization string at %L was truncated to "
180627f7eb2Smrg "fit the variable (%ld/%ld)", &rvalue->where,
181627f7eb2Smrg (long) tlen, (long) len);
182627f7eb2Smrg len = tlen;
183627f7eb2Smrg }
184627f7eb2Smrg }
185627f7eb2Smrg
186627f7eb2Smrg if (rvalue->ts.type == BT_HOLLERITH)
187627f7eb2Smrg {
188627f7eb2Smrg for (size_t i = 0; i < (size_t) len; i++)
189627f7eb2Smrg dest[start+i] = rvalue->representation.string[i];
190627f7eb2Smrg }
191627f7eb2Smrg else
192627f7eb2Smrg memcpy (&dest[start], rvalue->value.character.string,
193627f7eb2Smrg len * sizeof (gfc_char_t));
194627f7eb2Smrg
195627f7eb2Smrg /* Pad with spaces. Substrings will already be blanked. */
196627f7eb2Smrg if (len < tlen && ref == NULL)
197627f7eb2Smrg gfc_wide_memset (&dest[start + len], ' ', end - (start + len));
198627f7eb2Smrg
199627f7eb2Smrg if (rvalue->ts.type == BT_HOLLERITH)
200627f7eb2Smrg {
201627f7eb2Smrg init->representation.length = init->value.character.length;
202627f7eb2Smrg init->representation.string
203627f7eb2Smrg = gfc_widechar_to_char (init->value.character.string,
204627f7eb2Smrg init->value.character.length);
205627f7eb2Smrg }
206627f7eb2Smrg
207627f7eb2Smrg return init;
208627f7eb2Smrg }
209627f7eb2Smrg
210627f7eb2Smrg
211627f7eb2Smrg /* Assign the initial value RVALUE to LVALUE's symbol->value. If the
212627f7eb2Smrg LVALUE already has an initialization, we extend this, otherwise we
213627f7eb2Smrg create a new one. If REPEAT is non-NULL, initialize *REPEAT
214627f7eb2Smrg consecutive values in LVALUE the same value in RVALUE. In that case,
215627f7eb2Smrg LVALUE must refer to a full array, not an array section. */
216627f7eb2Smrg
217627f7eb2Smrg bool
gfc_assign_data_value(gfc_expr * lvalue,gfc_expr * rvalue,mpz_t index,mpz_t * repeat)218627f7eb2Smrg gfc_assign_data_value (gfc_expr *lvalue, gfc_expr *rvalue, mpz_t index,
219627f7eb2Smrg mpz_t *repeat)
220627f7eb2Smrg {
221627f7eb2Smrg gfc_ref *ref;
222627f7eb2Smrg gfc_expr *init;
223627f7eb2Smrg gfc_expr *expr = NULL;
224*4c3eb207Smrg gfc_expr *rexpr;
225627f7eb2Smrg gfc_constructor *con;
226627f7eb2Smrg gfc_constructor *last_con;
227627f7eb2Smrg gfc_symbol *symbol;
228627f7eb2Smrg gfc_typespec *last_ts;
229627f7eb2Smrg mpz_t offset;
230*4c3eb207Smrg const char *msg = "F18(R841): data-implied-do object at %L is neither an "
231*4c3eb207Smrg "array-element nor a scalar-structure-component";
232627f7eb2Smrg
233627f7eb2Smrg symbol = lvalue->symtree->n.sym;
234627f7eb2Smrg init = symbol->value;
235627f7eb2Smrg last_ts = &symbol->ts;
236627f7eb2Smrg last_con = NULL;
237627f7eb2Smrg mpz_init_set_si (offset, 0);
238627f7eb2Smrg
239627f7eb2Smrg /* Find/create the parent expressions for subobject references. */
240627f7eb2Smrg for (ref = lvalue->ref; ref; ref = ref->next)
241627f7eb2Smrg {
242627f7eb2Smrg /* Break out of the loop if we find a substring. */
243627f7eb2Smrg if (ref->type == REF_SUBSTRING)
244627f7eb2Smrg {
245627f7eb2Smrg /* A substring should always be the last subobject reference. */
246627f7eb2Smrg gcc_assert (ref->next == NULL);
247627f7eb2Smrg break;
248627f7eb2Smrg }
249627f7eb2Smrg
250627f7eb2Smrg /* Use the existing initializer expression if it exists. Otherwise
251627f7eb2Smrg create a new one. */
252627f7eb2Smrg if (init == NULL)
253627f7eb2Smrg expr = gfc_get_expr ();
254627f7eb2Smrg else
255627f7eb2Smrg expr = init;
256627f7eb2Smrg
257627f7eb2Smrg /* Find or create this element. */
258627f7eb2Smrg switch (ref->type)
259627f7eb2Smrg {
260627f7eb2Smrg case REF_ARRAY:
261627f7eb2Smrg if (ref->u.ar.as->rank == 0)
262627f7eb2Smrg {
263627f7eb2Smrg gcc_assert (ref->u.ar.as->corank > 0);
264627f7eb2Smrg if (init == NULL)
265627f7eb2Smrg free (expr);
266627f7eb2Smrg continue;
267627f7eb2Smrg }
268627f7eb2Smrg
269627f7eb2Smrg if (init && expr->expr_type != EXPR_ARRAY)
270627f7eb2Smrg {
271627f7eb2Smrg gfc_error ("%qs at %L already is initialized at %L",
272627f7eb2Smrg lvalue->symtree->n.sym->name, &lvalue->where,
273627f7eb2Smrg &init->where);
274627f7eb2Smrg goto abort;
275627f7eb2Smrg }
276627f7eb2Smrg
277627f7eb2Smrg if (init == NULL)
278627f7eb2Smrg {
279627f7eb2Smrg /* The element typespec will be the same as the array
280627f7eb2Smrg typespec. */
281627f7eb2Smrg expr->ts = *last_ts;
282627f7eb2Smrg /* Setup the expression to hold the constructor. */
283627f7eb2Smrg expr->expr_type = EXPR_ARRAY;
284627f7eb2Smrg expr->rank = ref->u.ar.as->rank;
285627f7eb2Smrg }
286627f7eb2Smrg
287627f7eb2Smrg if (ref->u.ar.type == AR_ELEMENT)
288627f7eb2Smrg get_array_index (&ref->u.ar, &offset);
289627f7eb2Smrg else
290627f7eb2Smrg mpz_set (offset, index);
291627f7eb2Smrg
292627f7eb2Smrg /* Check the bounds. */
293627f7eb2Smrg if (mpz_cmp_si (offset, 0) < 0)
294627f7eb2Smrg {
295627f7eb2Smrg gfc_error ("Data element below array lower bound at %L",
296627f7eb2Smrg &lvalue->where);
297627f7eb2Smrg goto abort;
298627f7eb2Smrg }
299627f7eb2Smrg else if (repeat != NULL
300627f7eb2Smrg && ref->u.ar.type != AR_ELEMENT)
301627f7eb2Smrg {
302627f7eb2Smrg mpz_t size, end;
303627f7eb2Smrg gcc_assert (ref->u.ar.type == AR_FULL
304627f7eb2Smrg && ref->next == NULL);
305627f7eb2Smrg mpz_init_set (end, offset);
306627f7eb2Smrg mpz_add (end, end, *repeat);
307627f7eb2Smrg if (spec_size (ref->u.ar.as, &size))
308627f7eb2Smrg {
309627f7eb2Smrg if (mpz_cmp (end, size) > 0)
310627f7eb2Smrg {
311627f7eb2Smrg mpz_clear (size);
312627f7eb2Smrg gfc_error ("Data element above array upper bound at %L",
313627f7eb2Smrg &lvalue->where);
314627f7eb2Smrg goto abort;
315627f7eb2Smrg }
316627f7eb2Smrg mpz_clear (size);
317627f7eb2Smrg }
318627f7eb2Smrg
319627f7eb2Smrg con = gfc_constructor_lookup (expr->value.constructor,
320627f7eb2Smrg mpz_get_si (offset));
321627f7eb2Smrg if (!con)
322627f7eb2Smrg {
323627f7eb2Smrg con = gfc_constructor_lookup_next (expr->value.constructor,
324627f7eb2Smrg mpz_get_si (offset));
325627f7eb2Smrg if (con != NULL && mpz_cmp (con->offset, end) >= 0)
326627f7eb2Smrg con = NULL;
327627f7eb2Smrg }
328627f7eb2Smrg
329627f7eb2Smrg /* Overwriting an existing initializer is non-standard but
330627f7eb2Smrg usually only provokes a warning from other compilers. */
331627f7eb2Smrg if (con != NULL && con->expr != NULL)
332627f7eb2Smrg {
333627f7eb2Smrg /* Order in which the expressions arrive here depends on
334627f7eb2Smrg whether they are from data statements or F95 style
335627f7eb2Smrg declarations. Therefore, check which is the most
336627f7eb2Smrg recent. */
337627f7eb2Smrg gfc_expr *exprd;
338627f7eb2Smrg exprd = (LOCATION_LINE (con->expr->where.lb->location)
339627f7eb2Smrg > LOCATION_LINE (rvalue->where.lb->location))
340627f7eb2Smrg ? con->expr : rvalue;
341627f7eb2Smrg if (gfc_notify_std (GFC_STD_GNU,
342627f7eb2Smrg "re-initialization of %qs at %L",
343627f7eb2Smrg symbol->name, &exprd->where) == false)
344627f7eb2Smrg return false;
345627f7eb2Smrg }
346627f7eb2Smrg
347627f7eb2Smrg while (con != NULL)
348627f7eb2Smrg {
349627f7eb2Smrg gfc_constructor *next_con = gfc_constructor_next (con);
350627f7eb2Smrg
351627f7eb2Smrg if (mpz_cmp (con->offset, end) >= 0)
352627f7eb2Smrg break;
353627f7eb2Smrg if (mpz_cmp (con->offset, offset) < 0)
354627f7eb2Smrg {
355627f7eb2Smrg gcc_assert (mpz_cmp_si (con->repeat, 1) > 0);
356627f7eb2Smrg mpz_sub (con->repeat, offset, con->offset);
357627f7eb2Smrg }
358627f7eb2Smrg else if (mpz_cmp_si (con->repeat, 1) > 0
359627f7eb2Smrg && mpz_get_si (con->offset)
360627f7eb2Smrg + mpz_get_si (con->repeat) > mpz_get_si (end))
361627f7eb2Smrg {
362627f7eb2Smrg int endi;
363627f7eb2Smrg splay_tree_node node
364627f7eb2Smrg = splay_tree_lookup (con->base,
365627f7eb2Smrg mpz_get_si (con->offset));
366627f7eb2Smrg gcc_assert (node
367627f7eb2Smrg && con == (gfc_constructor *) node->value
368627f7eb2Smrg && node->key == (splay_tree_key)
369627f7eb2Smrg mpz_get_si (con->offset));
370627f7eb2Smrg endi = mpz_get_si (con->offset)
371627f7eb2Smrg + mpz_get_si (con->repeat);
372627f7eb2Smrg if (endi > mpz_get_si (end) + 1)
373627f7eb2Smrg mpz_set_si (con->repeat, endi - mpz_get_si (end));
374627f7eb2Smrg else
375627f7eb2Smrg mpz_set_si (con->repeat, 1);
376627f7eb2Smrg mpz_set (con->offset, end);
377627f7eb2Smrg node->key = (splay_tree_key) mpz_get_si (end);
378627f7eb2Smrg break;
379627f7eb2Smrg }
380627f7eb2Smrg else
381627f7eb2Smrg gfc_constructor_remove (con);
382627f7eb2Smrg con = next_con;
383627f7eb2Smrg }
384627f7eb2Smrg
385627f7eb2Smrg con = gfc_constructor_insert_expr (&expr->value.constructor,
386627f7eb2Smrg NULL, &rvalue->where,
387627f7eb2Smrg mpz_get_si (offset));
388627f7eb2Smrg mpz_set (con->repeat, *repeat);
389627f7eb2Smrg repeat = NULL;
390627f7eb2Smrg mpz_clear (end);
391627f7eb2Smrg break;
392627f7eb2Smrg }
393627f7eb2Smrg else
394627f7eb2Smrg {
395627f7eb2Smrg mpz_t size;
396627f7eb2Smrg if (spec_size (ref->u.ar.as, &size))
397627f7eb2Smrg {
398627f7eb2Smrg if (mpz_cmp (offset, size) >= 0)
399627f7eb2Smrg {
400627f7eb2Smrg mpz_clear (size);
401627f7eb2Smrg gfc_error ("Data element above array upper bound at %L",
402627f7eb2Smrg &lvalue->where);
403627f7eb2Smrg goto abort;
404627f7eb2Smrg }
405627f7eb2Smrg mpz_clear (size);
406627f7eb2Smrg }
407627f7eb2Smrg }
408627f7eb2Smrg
409627f7eb2Smrg con = gfc_constructor_lookup (expr->value.constructor,
410627f7eb2Smrg mpz_get_si (offset));
411627f7eb2Smrg if (!con)
412627f7eb2Smrg {
413627f7eb2Smrg con = gfc_constructor_insert_expr (&expr->value.constructor,
414627f7eb2Smrg NULL, &rvalue->where,
415627f7eb2Smrg mpz_get_si (offset));
416627f7eb2Smrg }
417627f7eb2Smrg else if (mpz_cmp_si (con->repeat, 1) > 0)
418627f7eb2Smrg {
419627f7eb2Smrg /* Need to split a range. */
420627f7eb2Smrg if (mpz_cmp (con->offset, offset) < 0)
421627f7eb2Smrg {
422627f7eb2Smrg gfc_constructor *pred_con = con;
423627f7eb2Smrg con = gfc_constructor_insert_expr (&expr->value.constructor,
424627f7eb2Smrg NULL, &con->where,
425627f7eb2Smrg mpz_get_si (offset));
426627f7eb2Smrg con->expr = gfc_copy_expr (pred_con->expr);
427627f7eb2Smrg mpz_add (con->repeat, pred_con->offset, pred_con->repeat);
428627f7eb2Smrg mpz_sub (con->repeat, con->repeat, offset);
429627f7eb2Smrg mpz_sub (pred_con->repeat, offset, pred_con->offset);
430627f7eb2Smrg }
431627f7eb2Smrg if (mpz_cmp_si (con->repeat, 1) > 0)
432627f7eb2Smrg {
433627f7eb2Smrg gfc_constructor *succ_con;
434627f7eb2Smrg succ_con
435627f7eb2Smrg = gfc_constructor_insert_expr (&expr->value.constructor,
436627f7eb2Smrg NULL, &con->where,
437627f7eb2Smrg mpz_get_si (offset) + 1);
438627f7eb2Smrg succ_con->expr = gfc_copy_expr (con->expr);
439627f7eb2Smrg mpz_sub_ui (succ_con->repeat, con->repeat, 1);
440627f7eb2Smrg mpz_set_si (con->repeat, 1);
441627f7eb2Smrg }
442627f7eb2Smrg }
443627f7eb2Smrg break;
444627f7eb2Smrg
445627f7eb2Smrg case REF_COMPONENT:
446627f7eb2Smrg if (init == NULL)
447627f7eb2Smrg {
448627f7eb2Smrg /* Setup the expression to hold the constructor. */
449627f7eb2Smrg expr->expr_type = EXPR_STRUCTURE;
450627f7eb2Smrg expr->ts.type = BT_DERIVED;
451627f7eb2Smrg expr->ts.u.derived = ref->u.c.sym;
452627f7eb2Smrg }
453627f7eb2Smrg else
454627f7eb2Smrg gcc_assert (expr->expr_type == EXPR_STRUCTURE);
455627f7eb2Smrg last_ts = &ref->u.c.component->ts;
456627f7eb2Smrg
457627f7eb2Smrg /* Find the same element in the existing constructor. */
458627f7eb2Smrg con = find_con_by_component (ref->u.c.component,
459627f7eb2Smrg expr->value.constructor);
460627f7eb2Smrg
461627f7eb2Smrg if (con == NULL)
462627f7eb2Smrg {
463627f7eb2Smrg /* Create a new constructor. */
464627f7eb2Smrg con = gfc_constructor_append_expr (&expr->value.constructor,
465627f7eb2Smrg NULL, NULL);
466627f7eb2Smrg con->n.component = ref->u.c.component;
467627f7eb2Smrg }
468627f7eb2Smrg break;
469627f7eb2Smrg
470*4c3eb207Smrg case REF_INQUIRY:
471*4c3eb207Smrg
472*4c3eb207Smrg /* After some discussion on clf it was determined that the following
473*4c3eb207Smrg violates F18(R841). If the error is removed, the expected result
474*4c3eb207Smrg is obtained. Leaving the code in place ensures a clean error
475*4c3eb207Smrg recovery. */
476*4c3eb207Smrg gfc_error (msg, &lvalue->where);
477*4c3eb207Smrg
478*4c3eb207Smrg /* This breaks with the other reference types in that the output
479*4c3eb207Smrg constructor has to be of type COMPLEX, whereas the lvalue is
480*4c3eb207Smrg of type REAL. The rvalue is copied to the real or imaginary
481*4c3eb207Smrg part as appropriate. In addition, for all except scalar
482*4c3eb207Smrg complex variables, a complex expression has to provided, where
483*4c3eb207Smrg the constructor does not have it, and the expression modified
484*4c3eb207Smrg with a new value for the real or imaginary part. */
485*4c3eb207Smrg gcc_assert (ref->next == NULL && last_ts->type == BT_COMPLEX);
486*4c3eb207Smrg rexpr = gfc_copy_expr (rvalue);
487*4c3eb207Smrg if (!gfc_compare_types (&lvalue->ts, &rexpr->ts))
488*4c3eb207Smrg gfc_convert_type (rexpr, &lvalue->ts, 0);
489*4c3eb207Smrg
490*4c3eb207Smrg /* This is the scalar, complex case, where an initializer exists. */
491*4c3eb207Smrg if (init && ref == lvalue->ref)
492*4c3eb207Smrg expr = symbol->value;
493*4c3eb207Smrg /* Then all cases, where a complex expression does not exist. */
494*4c3eb207Smrg else if (!last_con || !last_con->expr)
495*4c3eb207Smrg {
496*4c3eb207Smrg expr = gfc_get_constant_expr (BT_COMPLEX, lvalue->ts.kind,
497*4c3eb207Smrg &lvalue->where);
498*4c3eb207Smrg if (last_con)
499*4c3eb207Smrg last_con->expr = expr;
500*4c3eb207Smrg }
501*4c3eb207Smrg else
502*4c3eb207Smrg /* Finally, and existing constructor expression to be modified. */
503*4c3eb207Smrg expr = last_con->expr;
504*4c3eb207Smrg
505*4c3eb207Smrg /* Rejection of LEN and KIND inquiry references is handled
506*4c3eb207Smrg elsewhere. The error here is added as backup. The assertion
507*4c3eb207Smrg of F2008 for RE and IM is also done elsewhere. */
508*4c3eb207Smrg switch (ref->u.i)
509*4c3eb207Smrg {
510*4c3eb207Smrg case INQUIRY_LEN:
511*4c3eb207Smrg case INQUIRY_KIND:
512*4c3eb207Smrg gfc_error ("LEN or KIND inquiry ref in DATA statement at %L",
513*4c3eb207Smrg &lvalue->where);
514*4c3eb207Smrg goto abort;
515*4c3eb207Smrg case INQUIRY_RE:
516*4c3eb207Smrg mpfr_set (mpc_realref (expr->value.complex),
517*4c3eb207Smrg rexpr->value.real,
518*4c3eb207Smrg GFC_RND_MODE);
519*4c3eb207Smrg break;
520*4c3eb207Smrg case INQUIRY_IM:
521*4c3eb207Smrg mpfr_set (mpc_imagref (expr->value.complex),
522*4c3eb207Smrg rexpr->value.real,
523*4c3eb207Smrg GFC_RND_MODE);
524*4c3eb207Smrg break;
525*4c3eb207Smrg }
526*4c3eb207Smrg
527*4c3eb207Smrg /* Only the scalar, complex expression needs to be saved as the
528*4c3eb207Smrg symbol value since the last constructor expression is already
529*4c3eb207Smrg provided as the initializer in the code after the reference
530*4c3eb207Smrg cases. */
531*4c3eb207Smrg if (ref == lvalue->ref)
532*4c3eb207Smrg symbol->value = expr;
533*4c3eb207Smrg
534*4c3eb207Smrg gfc_free_expr (rexpr);
535*4c3eb207Smrg mpz_clear (offset);
536*4c3eb207Smrg return true;
537*4c3eb207Smrg
538627f7eb2Smrg default:
539627f7eb2Smrg gcc_unreachable ();
540627f7eb2Smrg }
541627f7eb2Smrg
542627f7eb2Smrg if (init == NULL)
543627f7eb2Smrg {
544627f7eb2Smrg /* Point the container at the new expression. */
545627f7eb2Smrg if (last_con == NULL)
546627f7eb2Smrg symbol->value = expr;
547627f7eb2Smrg else
548627f7eb2Smrg last_con->expr = expr;
549627f7eb2Smrg }
550627f7eb2Smrg init = con->expr;
551627f7eb2Smrg last_con = con;
552627f7eb2Smrg }
553627f7eb2Smrg
554627f7eb2Smrg mpz_clear (offset);
555627f7eb2Smrg gcc_assert (repeat == NULL);
556627f7eb2Smrg
557627f7eb2Smrg /* Overwriting an existing initializer is non-standard but usually only
558627f7eb2Smrg provokes a warning from other compilers. */
559627f7eb2Smrg if (init != NULL && init->where.lb && rvalue->where.lb)
560627f7eb2Smrg {
561627f7eb2Smrg /* Order in which the expressions arrive here depends on whether
562627f7eb2Smrg they are from data statements or F95 style declarations.
563627f7eb2Smrg Therefore, check which is the most recent. */
564627f7eb2Smrg expr = (LOCATION_LINE (init->where.lb->location)
565627f7eb2Smrg > LOCATION_LINE (rvalue->where.lb->location))
566627f7eb2Smrg ? init : rvalue;
567627f7eb2Smrg if (gfc_notify_std (GFC_STD_GNU, "re-initialization of %qs at %L",
568627f7eb2Smrg symbol->name, &expr->where) == false)
569627f7eb2Smrg return false;
570627f7eb2Smrg }
571627f7eb2Smrg
572*4c3eb207Smrg if (ref || (last_ts->type == BT_CHARACTER
573*4c3eb207Smrg && rvalue->expr_type == EXPR_CONSTANT))
574627f7eb2Smrg {
575627f7eb2Smrg /* An initializer has to be constant. */
576*4c3eb207Smrg if (lvalue->ts.u.cl->length == NULL && !(ref && ref->u.ss.length != NULL))
577*4c3eb207Smrg return false;
578*4c3eb207Smrg if (lvalue->ts.u.cl->length
579*4c3eb207Smrg && lvalue->ts.u.cl->length->expr_type != EXPR_CONSTANT)
580627f7eb2Smrg return false;
581627f7eb2Smrg expr = create_character_initializer (init, last_ts, ref, rvalue);
582627f7eb2Smrg }
583627f7eb2Smrg else
584627f7eb2Smrg {
585627f7eb2Smrg if (lvalue->ts.type == BT_DERIVED
586627f7eb2Smrg && gfc_has_default_initializer (lvalue->ts.u.derived))
587627f7eb2Smrg {
588627f7eb2Smrg gfc_error ("Nonpointer object %qs with default initialization "
589627f7eb2Smrg "shall not appear in a DATA statement at %L",
590627f7eb2Smrg symbol->name, &lvalue->where);
591627f7eb2Smrg return false;
592627f7eb2Smrg }
593627f7eb2Smrg
594627f7eb2Smrg expr = gfc_copy_expr (rvalue);
595627f7eb2Smrg if (!gfc_compare_types (&lvalue->ts, &expr->ts))
596627f7eb2Smrg gfc_convert_type (expr, &lvalue->ts, 0);
597627f7eb2Smrg }
598627f7eb2Smrg
599627f7eb2Smrg if (last_con == NULL)
600627f7eb2Smrg symbol->value = expr;
601627f7eb2Smrg else
602627f7eb2Smrg last_con->expr = expr;
603627f7eb2Smrg
604627f7eb2Smrg return true;
605627f7eb2Smrg
606627f7eb2Smrg abort:
607627f7eb2Smrg if (!init)
608627f7eb2Smrg gfc_free_expr (expr);
609627f7eb2Smrg mpz_clear (offset);
610627f7eb2Smrg return false;
611627f7eb2Smrg }
612627f7eb2Smrg
613627f7eb2Smrg
614627f7eb2Smrg /* Modify the index of array section and re-calculate the array offset. */
615627f7eb2Smrg
616627f7eb2Smrg void
gfc_advance_section(mpz_t * section_index,gfc_array_ref * ar,mpz_t * offset_ret)617627f7eb2Smrg gfc_advance_section (mpz_t *section_index, gfc_array_ref *ar,
618627f7eb2Smrg mpz_t *offset_ret)
619627f7eb2Smrg {
620627f7eb2Smrg int i;
621627f7eb2Smrg mpz_t delta;
622627f7eb2Smrg mpz_t tmp;
623627f7eb2Smrg bool forwards;
624627f7eb2Smrg int cmp;
625627f7eb2Smrg gfc_expr *start, *end, *stride;
626627f7eb2Smrg
627627f7eb2Smrg for (i = 0; i < ar->dimen; i++)
628627f7eb2Smrg {
629627f7eb2Smrg if (ar->dimen_type[i] != DIMEN_RANGE)
630627f7eb2Smrg continue;
631627f7eb2Smrg
632627f7eb2Smrg if (ar->stride[i])
633627f7eb2Smrg {
634627f7eb2Smrg stride = gfc_copy_expr(ar->stride[i]);
635627f7eb2Smrg if(!gfc_simplify_expr(stride, 1))
636627f7eb2Smrg gfc_internal_error("Simplification error");
637627f7eb2Smrg mpz_add (section_index[i], section_index[i],
638627f7eb2Smrg stride->value.integer);
639627f7eb2Smrg if (mpz_cmp_si (stride->value.integer, 0) >= 0)
640627f7eb2Smrg forwards = true;
641627f7eb2Smrg else
642627f7eb2Smrg forwards = false;
643627f7eb2Smrg gfc_free_expr(stride);
644627f7eb2Smrg }
645627f7eb2Smrg else
646627f7eb2Smrg {
647627f7eb2Smrg mpz_add_ui (section_index[i], section_index[i], 1);
648627f7eb2Smrg forwards = true;
649627f7eb2Smrg }
650627f7eb2Smrg
651627f7eb2Smrg if (ar->end[i])
652627f7eb2Smrg {
653627f7eb2Smrg end = gfc_copy_expr(ar->end[i]);
654627f7eb2Smrg if(!gfc_simplify_expr(end, 1))
655627f7eb2Smrg gfc_internal_error("Simplification error");
656627f7eb2Smrg cmp = mpz_cmp (section_index[i], end->value.integer);
657627f7eb2Smrg gfc_free_expr(end);
658627f7eb2Smrg }
659627f7eb2Smrg else
660627f7eb2Smrg cmp = mpz_cmp (section_index[i], ar->as->upper[i]->value.integer);
661627f7eb2Smrg
662627f7eb2Smrg if ((cmp > 0 && forwards) || (cmp < 0 && !forwards))
663627f7eb2Smrg {
664627f7eb2Smrg /* Reset index to start, then loop to advance the next index. */
665627f7eb2Smrg if (ar->start[i])
666627f7eb2Smrg {
667627f7eb2Smrg start = gfc_copy_expr(ar->start[i]);
668627f7eb2Smrg if(!gfc_simplify_expr(start, 1))
669627f7eb2Smrg gfc_internal_error("Simplification error");
670627f7eb2Smrg mpz_set (section_index[i], start->value.integer);
671627f7eb2Smrg gfc_free_expr(start);
672627f7eb2Smrg }
673627f7eb2Smrg else
674627f7eb2Smrg mpz_set (section_index[i], ar->as->lower[i]->value.integer);
675627f7eb2Smrg }
676627f7eb2Smrg else
677627f7eb2Smrg break;
678627f7eb2Smrg }
679627f7eb2Smrg
680627f7eb2Smrg mpz_set_si (*offset_ret, 0);
681627f7eb2Smrg mpz_init_set_si (delta, 1);
682627f7eb2Smrg mpz_init (tmp);
683627f7eb2Smrg for (i = 0; i < ar->dimen; i++)
684627f7eb2Smrg {
685627f7eb2Smrg mpz_sub (tmp, section_index[i], ar->as->lower[i]->value.integer);
686627f7eb2Smrg mpz_mul (tmp, tmp, delta);
687627f7eb2Smrg mpz_add (*offset_ret, tmp, *offset_ret);
688627f7eb2Smrg
689627f7eb2Smrg mpz_sub (tmp, ar->as->upper[i]->value.integer,
690627f7eb2Smrg ar->as->lower[i]->value.integer);
691627f7eb2Smrg mpz_add_ui (tmp, tmp, 1);
692627f7eb2Smrg mpz_mul (delta, tmp, delta);
693627f7eb2Smrg }
694627f7eb2Smrg mpz_clear (tmp);
695627f7eb2Smrg mpz_clear (delta);
696627f7eb2Smrg }
697627f7eb2Smrg
698627f7eb2Smrg
699627f7eb2Smrg /* Rearrange a structure constructor so the elements are in the specified
700627f7eb2Smrg order. Also insert NULL entries if necessary. */
701627f7eb2Smrg
702627f7eb2Smrg static void
formalize_structure_cons(gfc_expr * expr)703627f7eb2Smrg formalize_structure_cons (gfc_expr *expr)
704627f7eb2Smrg {
705627f7eb2Smrg gfc_constructor_base base = NULL;
706627f7eb2Smrg gfc_constructor *cur;
707627f7eb2Smrg gfc_component *order;
708627f7eb2Smrg
709627f7eb2Smrg /* Constructor is already formalized. */
710627f7eb2Smrg cur = gfc_constructor_first (expr->value.constructor);
711627f7eb2Smrg if (!cur || cur->n.component == NULL)
712627f7eb2Smrg return;
713627f7eb2Smrg
714627f7eb2Smrg for (order = expr->ts.u.derived->components; order; order = order->next)
715627f7eb2Smrg {
716627f7eb2Smrg cur = find_con_by_component (order, expr->value.constructor);
717627f7eb2Smrg if (cur)
718627f7eb2Smrg gfc_constructor_append_expr (&base, cur->expr, &cur->expr->where);
719627f7eb2Smrg else
720627f7eb2Smrg gfc_constructor_append_expr (&base, NULL, NULL);
721627f7eb2Smrg }
722627f7eb2Smrg
723627f7eb2Smrg /* For all what it's worth, one would expect
724627f7eb2Smrg gfc_constructor_free (expr->value.constructor);
725627f7eb2Smrg here. However, if the constructor is actually free'd,
726627f7eb2Smrg hell breaks loose in the testsuite?! */
727627f7eb2Smrg
728627f7eb2Smrg expr->value.constructor = base;
729627f7eb2Smrg }
730627f7eb2Smrg
731627f7eb2Smrg
732627f7eb2Smrg /* Make sure an initialization expression is in normalized form, i.e., all
733627f7eb2Smrg elements of the constructors are in the correct order. */
734627f7eb2Smrg
735627f7eb2Smrg static void
formalize_init_expr(gfc_expr * expr)736627f7eb2Smrg formalize_init_expr (gfc_expr *expr)
737627f7eb2Smrg {
738627f7eb2Smrg expr_t type;
739627f7eb2Smrg gfc_constructor *c;
740627f7eb2Smrg
741627f7eb2Smrg if (expr == NULL)
742627f7eb2Smrg return;
743627f7eb2Smrg
744627f7eb2Smrg type = expr->expr_type;
745627f7eb2Smrg switch (type)
746627f7eb2Smrg {
747627f7eb2Smrg case EXPR_ARRAY:
748627f7eb2Smrg for (c = gfc_constructor_first (expr->value.constructor);
749627f7eb2Smrg c; c = gfc_constructor_next (c))
750627f7eb2Smrg formalize_init_expr (c->expr);
751627f7eb2Smrg
752627f7eb2Smrg break;
753627f7eb2Smrg
754627f7eb2Smrg case EXPR_STRUCTURE:
755627f7eb2Smrg formalize_structure_cons (expr);
756627f7eb2Smrg break;
757627f7eb2Smrg
758627f7eb2Smrg default:
759627f7eb2Smrg break;
760627f7eb2Smrg }
761627f7eb2Smrg }
762627f7eb2Smrg
763627f7eb2Smrg
764627f7eb2Smrg /* Resolve symbol's initial value after all data statement. */
765627f7eb2Smrg
766627f7eb2Smrg void
gfc_formalize_init_value(gfc_symbol * sym)767627f7eb2Smrg gfc_formalize_init_value (gfc_symbol *sym)
768627f7eb2Smrg {
769627f7eb2Smrg formalize_init_expr (sym->value);
770627f7eb2Smrg }
771627f7eb2Smrg
772627f7eb2Smrg
773627f7eb2Smrg /* Get the integer value into RET_AS and SECTION from AS and AR, and return
774627f7eb2Smrg offset. */
775627f7eb2Smrg
776627f7eb2Smrg void
gfc_get_section_index(gfc_array_ref * ar,mpz_t * section_index,mpz_t * offset)777627f7eb2Smrg gfc_get_section_index (gfc_array_ref *ar, mpz_t *section_index, mpz_t *offset)
778627f7eb2Smrg {
779627f7eb2Smrg int i;
780627f7eb2Smrg mpz_t delta;
781627f7eb2Smrg mpz_t tmp;
782627f7eb2Smrg gfc_expr *start;
783627f7eb2Smrg
784627f7eb2Smrg mpz_set_si (*offset, 0);
785627f7eb2Smrg mpz_init (tmp);
786627f7eb2Smrg mpz_init_set_si (delta, 1);
787627f7eb2Smrg for (i = 0; i < ar->dimen; i++)
788627f7eb2Smrg {
789627f7eb2Smrg mpz_init (section_index[i]);
790627f7eb2Smrg switch (ar->dimen_type[i])
791627f7eb2Smrg {
792627f7eb2Smrg case DIMEN_ELEMENT:
793627f7eb2Smrg case DIMEN_RANGE:
794627f7eb2Smrg if (ar->start[i])
795627f7eb2Smrg {
796627f7eb2Smrg start = gfc_copy_expr(ar->start[i]);
797627f7eb2Smrg if(!gfc_simplify_expr(start, 1))
798627f7eb2Smrg gfc_internal_error("Simplification error");
799627f7eb2Smrg mpz_sub (tmp, start->value.integer,
800627f7eb2Smrg ar->as->lower[i]->value.integer);
801627f7eb2Smrg mpz_mul (tmp, tmp, delta);
802627f7eb2Smrg mpz_add (*offset, tmp, *offset);
803627f7eb2Smrg mpz_set (section_index[i], start->value.integer);
804627f7eb2Smrg gfc_free_expr(start);
805627f7eb2Smrg }
806627f7eb2Smrg else
807627f7eb2Smrg mpz_set (section_index[i], ar->as->lower[i]->value.integer);
808627f7eb2Smrg break;
809627f7eb2Smrg
810627f7eb2Smrg case DIMEN_VECTOR:
811627f7eb2Smrg gfc_internal_error ("TODO: Vector sections in data statements");
812627f7eb2Smrg
813627f7eb2Smrg default:
814627f7eb2Smrg gcc_unreachable ();
815627f7eb2Smrg }
816627f7eb2Smrg
817627f7eb2Smrg mpz_sub (tmp, ar->as->upper[i]->value.integer,
818627f7eb2Smrg ar->as->lower[i]->value.integer);
819627f7eb2Smrg mpz_add_ui (tmp, tmp, 1);
820627f7eb2Smrg mpz_mul (delta, tmp, delta);
821627f7eb2Smrg }
822627f7eb2Smrg
823627f7eb2Smrg mpz_clear (tmp);
824627f7eb2Smrg mpz_clear (delta);
825627f7eb2Smrg }
826627f7eb2Smrg
827