OSDN Git Service

67fc796f86fefb0575d0d6f7ad3f9bcc1e2ffc4d
[pf3gnuchains/gcc-fork.git] / gcc / fortran / trans-io.c
1 /* IO Code translation/library interface
2    Copyright (C) 2002, 2003, 2004, 2005 Free Software Foundation, Inc.
3    Contributed by Paul Brook
4
5 This file is part of GCC.
6
7 GCC is free software; you can redistribute it and/or modify it under
8 the terms of the GNU General Public License as published by the Free
9 Software Foundation; either version 2, or (at your option) any later
10 version.
11
12 GCC is distributed in the hope that it will be useful, but WITHOUT ANY
13 WARRANTY; without even the implied warranty of MERCHANTABILITY or
14 FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
15 for more details.
16
17 You should have received a copy of the GNU General Public License
18 along with GCC; see the file COPYING.  If not, write to the Free
19 Software Foundation, 59 Temple Place - Suite 330, Boston, MA
20 02111-1307, USA.  */
21
22
23 #include "config.h"
24 #include "system.h"
25 #include "coretypes.h"
26 #include "tree.h"
27 #include "tree-gimple.h"
28 #include "ggc.h"
29 #include "toplev.h"
30 #include "real.h"
31 #include "gfortran.h"
32 #include "trans.h"
33 #include "trans-stmt.h"
34 #include "trans-array.h"
35 #include "trans-types.h"
36 #include "trans-const.h"
37
38
39 /* Members of the ioparm structure.  */
40
41 static GTY(()) tree ioparm_unit;
42 static GTY(()) tree ioparm_err;
43 static GTY(()) tree ioparm_end;
44 static GTY(()) tree ioparm_eor;
45 static GTY(()) tree ioparm_list_format;
46 static GTY(()) tree ioparm_library_return;
47 static GTY(()) tree ioparm_iostat;
48 static GTY(()) tree ioparm_exist;
49 static GTY(()) tree ioparm_opened;
50 static GTY(()) tree ioparm_number;
51 static GTY(()) tree ioparm_named;
52 static GTY(()) tree ioparm_rec;
53 static GTY(()) tree ioparm_nextrec;
54 static GTY(()) tree ioparm_size;
55 static GTY(()) tree ioparm_recl_in;
56 static GTY(()) tree ioparm_recl_out;
57 static GTY(()) tree ioparm_iolength;
58 static GTY(()) tree ioparm_file;
59 static GTY(()) tree ioparm_file_len;
60 static GTY(()) tree ioparm_status;
61 static GTY(()) tree ioparm_status_len;
62 static GTY(()) tree ioparm_access;
63 static GTY(()) tree ioparm_access_len;
64 static GTY(()) tree ioparm_form;
65 static GTY(()) tree ioparm_form_len;
66 static GTY(()) tree ioparm_blank;
67 static GTY(()) tree ioparm_blank_len;
68 static GTY(()) tree ioparm_position;
69 static GTY(()) tree ioparm_position_len;
70 static GTY(()) tree ioparm_action;
71 static GTY(()) tree ioparm_action_len;
72 static GTY(()) tree ioparm_delim;
73 static GTY(()) tree ioparm_delim_len;
74 static GTY(()) tree ioparm_pad;
75 static GTY(()) tree ioparm_pad_len;
76 static GTY(()) tree ioparm_format;
77 static GTY(()) tree ioparm_format_len;
78 static GTY(()) tree ioparm_advance;
79 static GTY(()) tree ioparm_advance_len;
80 static GTY(()) tree ioparm_name;
81 static GTY(()) tree ioparm_name_len;
82 static GTY(()) tree ioparm_internal_unit;
83 static GTY(()) tree ioparm_internal_unit_len;
84 static GTY(()) tree ioparm_sequential;
85 static GTY(()) tree ioparm_sequential_len;
86 static GTY(()) tree ioparm_direct;
87 static GTY(()) tree ioparm_direct_len;
88 static GTY(()) tree ioparm_formatted;
89 static GTY(()) tree ioparm_formatted_len;
90 static GTY(()) tree ioparm_unformatted;
91 static GTY(()) tree ioparm_unformatted_len;
92 static GTY(()) tree ioparm_read;
93 static GTY(()) tree ioparm_read_len;
94 static GTY(()) tree ioparm_write;
95 static GTY(()) tree ioparm_write_len;
96 static GTY(()) tree ioparm_readwrite;
97 static GTY(()) tree ioparm_readwrite_len;
98 static GTY(()) tree ioparm_namelist_name;
99 static GTY(()) tree ioparm_namelist_name_len;
100 static GTY(()) tree ioparm_namelist_read_mode;
101
102 /* The global I/O variables */
103
104 static GTY(()) tree ioparm_var;
105 static GTY(()) tree locus_file;
106 static GTY(()) tree locus_line;
107
108
109 /* Library I/O subroutines */
110
111 static GTY(()) tree iocall_read;
112 static GTY(()) tree iocall_read_done;
113 static GTY(()) tree iocall_write;
114 static GTY(()) tree iocall_write_done;
115 static GTY(()) tree iocall_x_integer;
116 static GTY(()) tree iocall_x_logical;
117 static GTY(()) tree iocall_x_character;
118 static GTY(()) tree iocall_x_real;
119 static GTY(()) tree iocall_x_complex;
120 static GTY(()) tree iocall_open;
121 static GTY(()) tree iocall_close;
122 static GTY(()) tree iocall_inquire;
123 static GTY(()) tree iocall_iolength;
124 static GTY(()) tree iocall_iolength_done;
125 static GTY(()) tree iocall_rewind;
126 static GTY(()) tree iocall_backspace;
127 static GTY(()) tree iocall_endfile;
128 static GTY(()) tree iocall_set_nml_val_int;
129 static GTY(()) tree iocall_set_nml_val_float;
130 static GTY(()) tree iocall_set_nml_val_char;
131 static GTY(()) tree iocall_set_nml_val_complex;
132 static GTY(()) tree iocall_set_nml_val_log;
133
134 /* Variable for keeping track of what the last data transfer statement
135    was.  Used for deciding which subroutine to call when the data
136    transfer is complete.  */
137 static enum { READ, WRITE, IOLENGTH } last_dt;
138
139 #define ADD_FIELD(name, type)                                           \
140   ioparm_ ## name = gfc_add_field_to_struct                             \
141         (&(TYPE_FIELDS (ioparm_type)), ioparm_type,                     \
142          get_identifier (stringize(name)), type)
143
144 #define ADD_STRING(name) \
145   ioparm_ ## name = gfc_add_field_to_struct                             \
146         (&(TYPE_FIELDS (ioparm_type)), ioparm_type,                     \
147          get_identifier (stringize(name)), pchar_type_node);            \
148   ioparm_ ## name ## _len = gfc_add_field_to_struct                     \
149         (&(TYPE_FIELDS (ioparm_type)), ioparm_type,                     \
150          get_identifier (stringize(name) "_len"), gfc_charlen_type_node)
151
152
153 /* Create function decls for IO library functions.  */
154
155 void
156 gfc_build_io_library_fndecls (void)
157 {
158   tree gfc_int4_type_node;
159   tree gfc_pint4_type_node;
160   tree ioparm_type;
161
162   gfc_int4_type_node = gfc_get_int_type (4);
163   gfc_pint4_type_node = build_pointer_type (gfc_int4_type_node);
164
165   /* Build the st_parameter structure.  Information associated with I/O
166      calls are transferred here.  This must match the one defined in the
167      library exactly.  */
168
169   ioparm_type = make_node (RECORD_TYPE);
170   TYPE_NAME (ioparm_type) = get_identifier ("_gfc_ioparm");
171
172   ADD_FIELD (unit, gfc_int4_type_node);
173   ADD_FIELD (err, gfc_int4_type_node);
174   ADD_FIELD (end, gfc_int4_type_node);
175   ADD_FIELD (eor, gfc_int4_type_node);
176   ADD_FIELD (list_format, gfc_int4_type_node);
177   ADD_FIELD (library_return, gfc_int4_type_node);
178
179   ADD_FIELD (iostat, gfc_pint4_type_node);
180   ADD_FIELD (exist, gfc_pint4_type_node);
181   ADD_FIELD (opened, gfc_pint4_type_node);
182   ADD_FIELD (number, gfc_pint4_type_node);
183   ADD_FIELD (named, gfc_pint4_type_node);
184   ADD_FIELD (rec, gfc_int4_type_node);
185   ADD_FIELD (nextrec, gfc_pint4_type_node);
186   ADD_FIELD (size, gfc_pint4_type_node);
187
188   ADD_FIELD (recl_in, gfc_int4_type_node);
189   ADD_FIELD (recl_out, gfc_pint4_type_node);
190
191   ADD_FIELD (iolength, gfc_pint4_type_node);
192
193   ADD_STRING (file);
194   ADD_STRING (status);
195
196   ADD_STRING (access);
197   ADD_STRING (form);
198   ADD_STRING (blank);
199   ADD_STRING (position);
200   ADD_STRING (action);
201   ADD_STRING (delim);
202   ADD_STRING (pad);
203   ADD_STRING (format);
204   ADD_STRING (advance);
205   ADD_STRING (name);
206   ADD_STRING (internal_unit);
207   ADD_STRING (sequential);
208
209   ADD_STRING (direct);
210   ADD_STRING (formatted);
211   ADD_STRING (unformatted);
212   ADD_STRING (read);
213   ADD_STRING (write);
214   ADD_STRING (readwrite);
215
216   ADD_STRING (namelist_name);
217   ADD_FIELD (namelist_read_mode, gfc_int4_type_node);
218
219   gfc_finish_type (ioparm_type);
220
221   ioparm_var = build_decl (VAR_DECL, get_identifier (PREFIX("ioparm")),
222                            ioparm_type);
223   DECL_EXTERNAL (ioparm_var) = 1;
224   TREE_PUBLIC (ioparm_var) = 1;
225
226   locus_line = build_decl (VAR_DECL, get_identifier (PREFIX("line")),
227                            gfc_int4_type_node);
228   DECL_EXTERNAL (locus_line) = 1;
229   TREE_PUBLIC (locus_line) = 1;
230
231   locus_file = build_decl (VAR_DECL, get_identifier (PREFIX("filename")),
232                            pchar_type_node);
233   DECL_EXTERNAL (locus_file) = 1;
234   TREE_PUBLIC (locus_file) = 1;
235
236   /* Define the transfer functions.  */
237
238   iocall_x_integer =
239     gfc_build_library_function_decl (get_identifier
240                                      (PREFIX("transfer_integer")),
241                                      void_type_node, 2, pvoid_type_node,
242                                      gfc_int4_type_node);
243
244   iocall_x_logical =
245     gfc_build_library_function_decl (get_identifier
246                                      (PREFIX("transfer_logical")),
247                                      void_type_node, 2, pvoid_type_node,
248                                      gfc_int4_type_node);
249
250   iocall_x_character =
251     gfc_build_library_function_decl (get_identifier
252                                      (PREFIX("transfer_character")),
253                                      void_type_node, 2, pvoid_type_node,
254                                      gfc_int4_type_node);
255
256   iocall_x_real =
257     gfc_build_library_function_decl (get_identifier (PREFIX("transfer_real")),
258                                      void_type_node, 2,
259                                      pvoid_type_node, gfc_int4_type_node);
260
261   iocall_x_complex =
262     gfc_build_library_function_decl (get_identifier
263                                      (PREFIX("transfer_complex")),
264                                      void_type_node, 2, pvoid_type_node,
265                                      gfc_int4_type_node);
266
267   /* Library entry points */
268
269   iocall_read =
270     gfc_build_library_function_decl (get_identifier (PREFIX("st_read")),
271                                      void_type_node, 0);
272
273   iocall_write =
274     gfc_build_library_function_decl (get_identifier (PREFIX("st_write")),
275                                      void_type_node, 0);
276   iocall_open =
277     gfc_build_library_function_decl (get_identifier (PREFIX("st_open")),
278                                      void_type_node, 0);
279
280   iocall_close =
281     gfc_build_library_function_decl (get_identifier (PREFIX("st_close")),
282                                      void_type_node, 0);
283
284   iocall_inquire =
285     gfc_build_library_function_decl (get_identifier (PREFIX("st_inquire")),
286                                      gfc_int4_type_node, 0);
287
288   iocall_iolength =
289     gfc_build_library_function_decl(get_identifier (PREFIX("st_iolength")),
290                                     void_type_node, 0);
291
292   iocall_rewind =
293     gfc_build_library_function_decl (get_identifier (PREFIX("st_rewind")),
294                                      gfc_int4_type_node, 0);
295
296   iocall_backspace =
297     gfc_build_library_function_decl (get_identifier (PREFIX("st_backspace")),
298                                      gfc_int4_type_node, 0);
299
300   iocall_endfile =
301     gfc_build_library_function_decl (get_identifier (PREFIX("st_endfile")),
302                                      gfc_int4_type_node, 0);
303   /* Library helpers */
304
305   iocall_read_done =
306     gfc_build_library_function_decl (get_identifier (PREFIX("st_read_done")),
307                                      gfc_int4_type_node, 0);
308
309   iocall_write_done =
310     gfc_build_library_function_decl (get_identifier (PREFIX("st_write_done")),
311                                      gfc_int4_type_node, 0);
312
313   iocall_iolength_done =
314     gfc_build_library_function_decl (get_identifier (PREFIX("st_iolength_done")),
315                                      gfc_int4_type_node, 0);
316
317   iocall_set_nml_val_int =
318     gfc_build_library_function_decl (get_identifier (PREFIX("st_set_nml_var_int")),
319                                      void_type_node, 4,
320                                      pvoid_type_node, pvoid_type_node,
321                                      gfc_int4_type_node,gfc_int4_type_node);
322
323   iocall_set_nml_val_float =
324     gfc_build_library_function_decl (get_identifier (PREFIX("st_set_nml_var_float")),
325                                      void_type_node, 4,
326                                      pvoid_type_node, pvoid_type_node,
327                                      gfc_int4_type_node,gfc_int4_type_node);
328   iocall_set_nml_val_char =
329     gfc_build_library_function_decl (get_identifier (PREFIX("st_set_nml_var_char")),
330                                      void_type_node, 5,
331                                      pvoid_type_node, pvoid_type_node,
332                                      gfc_int4_type_node, gfc_int4_type_node, 
333                                      gfc_charlen_type_node);
334   iocall_set_nml_val_complex =
335     gfc_build_library_function_decl (get_identifier (PREFIX("st_set_nml_var_complex")),
336                                      void_type_node, 4,
337                                      pvoid_type_node, pvoid_type_node,
338                                      gfc_int4_type_node,gfc_int4_type_node);
339   iocall_set_nml_val_log =
340     gfc_build_library_function_decl (get_identifier (PREFIX("st_set_nml_var_log")),
341                                      void_type_node, 4,
342                                      pvoid_type_node, pvoid_type_node,
343                                      gfc_int4_type_node,gfc_int4_type_node);
344
345 }
346
347
348 /* Generate code to store an non-string I/O parameter into the
349    ioparm structure.  This is a pass by value.  */
350
351 static void
352 set_parameter_value (stmtblock_t * block, tree var, gfc_expr * e)
353 {
354   gfc_se se;
355   tree tmp;
356
357   gfc_init_se (&se, NULL);
358   gfc_conv_expr_type (&se, e, TREE_TYPE (var));
359   gfc_add_block_to_block (block, &se.pre);
360
361   tmp = build3 (COMPONENT_REF, TREE_TYPE (var), ioparm_var, var, NULL_TREE);
362   gfc_add_modify_expr (block, tmp, se.expr);
363 }
364
365
366 /* Generate code to store an non-string I/O parameter into the
367    ioparm structure.  This is pass by reference.  */
368
369 static void
370 set_parameter_ref (stmtblock_t * block, tree var, gfc_expr * e)
371 {
372   gfc_se se;
373   tree tmp;
374
375   gfc_init_se (&se, NULL);
376   se.want_pointer = 1;
377
378   gfc_conv_expr_type (&se, e, TREE_TYPE (var));
379   gfc_add_block_to_block (block, &se.pre);
380
381   tmp = build3 (COMPONENT_REF, TREE_TYPE (var), ioparm_var, var, NULL_TREE);
382   gfc_add_modify_expr (block, tmp, se.expr);
383 }
384
385
386 /* Generate code to store a string and its length into the
387    ioparm structure.  */
388
389 static void
390 set_string (stmtblock_t * block, stmtblock_t * postblock, tree var,
391             tree var_len, gfc_expr * e)
392 {
393   gfc_se se;
394   tree tmp;
395   tree msg;
396   tree io;
397   tree len;
398
399   gfc_init_se (&se, NULL);
400   gfc_conv_expr (&se, e);
401
402   io = build3 (COMPONENT_REF, TREE_TYPE (var), ioparm_var, var, NULL_TREE);
403   len = build3 (COMPONENT_REF, TREE_TYPE (var_len), ioparm_var, var_len,
404                 NULL_TREE);
405
406   /* Integer variable assigned a format label.  */
407   if (e->ts.type == BT_INTEGER && e->symtree->n.sym->attr.assign == 1)
408     {
409       msg =
410         gfc_build_cstring_const ("Assigned label is not a format label");
411       tmp = GFC_DECL_STRING_LEN (se.expr);
412       tmp = build2 (LE_EXPR, boolean_type_node,
413                     tmp, convert (TREE_TYPE (tmp), integer_minus_one_node));
414       gfc_trans_runtime_check (tmp, msg, &se.pre);
415       gfc_add_modify_expr (&se.pre, io, GFC_DECL_ASSIGN_ADDR (se.expr));
416       gfc_add_modify_expr (&se.pre, len, GFC_DECL_STRING_LEN (se.expr));
417     }
418   else
419     {
420       gfc_conv_string_parameter (&se);
421       gfc_add_modify_expr (&se.pre, io, fold_convert (TREE_TYPE (io), se.expr));
422       gfc_add_modify_expr (&se.pre, len, se.string_length);
423     }
424
425   gfc_add_block_to_block (block, &se.pre);
426   gfc_add_block_to_block (postblock, &se.post);
427
428 }
429
430
431 /* Set a member of the ioparm structure to one.  */
432 static void
433 set_flag (stmtblock_t *block, tree var)
434 {
435   tree tmp, type = TREE_TYPE (var);
436
437   tmp = build3 (COMPONENT_REF, type, ioparm_var, var, NULL_TREE);
438   gfc_add_modify_expr (block, tmp, convert (type, integer_one_node));
439 }
440
441
442 /* Add a case to a IO-result switch.  */
443
444 static void
445 add_case (int label_value, gfc_st_label * label, stmtblock_t * body)
446 {
447   tree tmp, value;
448
449   if (label == NULL)
450     return;                     /* No label, no case */
451
452   value = build_int_cst (NULL_TREE, label_value);
453
454   /* Make a backend label for this case.  */
455   tmp = build_decl (LABEL_DECL, NULL_TREE, NULL_TREE);
456   DECL_CONTEXT (tmp) = current_function_decl;
457
458   /* And the case itself.  */
459   tmp = build3_v (CASE_LABEL_EXPR, value, NULL_TREE, tmp);
460   gfc_add_expr_to_block (body, tmp);
461
462   /* Jump to the label.  */
463   tmp = build1_v (GOTO_EXPR, gfc_get_label_decl (label));
464   gfc_add_expr_to_block (body, tmp);
465 }
466
467
468 /* Generate a switch statement that branches to the correct I/O
469    result label.  The last statement of an I/O call stores the
470    result into a variable because there is often cleanup that
471    must be done before the switch, so a temporary would have to
472    be created anyway.  */
473
474 static void
475 io_result (stmtblock_t * block, gfc_st_label * err_label,
476            gfc_st_label * end_label, gfc_st_label * eor_label)
477 {
478   stmtblock_t body;
479   tree tmp, rc;
480
481   /* If no labels are specified, ignore the result instead
482      of building an empty switch.  */
483   if (err_label == NULL
484       && end_label == NULL
485       && eor_label == NULL)
486     return;
487
488   /* Build a switch statement.  */
489   gfc_start_block (&body);
490
491   /* The label values here must be the same as the values
492      in the library_return enum in the runtime library */
493   add_case (1, err_label, &body);
494   add_case (2, end_label, &body);
495   add_case (3, eor_label, &body);
496
497   tmp = gfc_finish_block (&body);
498
499   rc = build3 (COMPONENT_REF, TREE_TYPE (ioparm_library_return), ioparm_var,
500                ioparm_library_return, NULL_TREE);
501
502   tmp = build3_v (SWITCH_EXPR, rc, tmp, NULL_TREE);
503
504   gfc_add_expr_to_block (block, tmp);
505 }
506
507
508 /* Store the current file and line number to variables so that if a
509    library call goes awry, we can tell the user where the problem is.  */
510
511 static void
512 set_error_locus (stmtblock_t * block, locus * where)
513 {
514   gfc_file *f;
515   tree tmp;
516   int line;
517
518   f = where->lb->file;
519   tmp = gfc_build_cstring_const (f->filename);
520
521   tmp = gfc_build_addr_expr (pchar_type_node, tmp);
522   gfc_add_modify_expr (block, locus_file, tmp);
523
524 #ifdef USE_MAPPED_LOCATION
525   line = LOCATION_LINE (where->lb->location);
526 #else
527   line = where->lb->linenum;
528 #endif
529   gfc_add_modify_expr (block, locus_line, build_int_cst (NULL_TREE, line));
530 }
531
532
533 /* Translate an OPEN statement.  */
534
535 tree
536 gfc_trans_open (gfc_code * code)
537 {
538   stmtblock_t block, post_block;
539   gfc_open *p;
540   tree tmp;
541
542   gfc_init_block (&block);
543   gfc_init_block (&post_block);
544
545   set_error_locus (&block, &code->loc);
546   p = code->ext.open;
547
548   if (p->unit)
549     set_parameter_value (&block, ioparm_unit, p->unit);
550
551   if (p->file)
552     set_string (&block, &post_block, ioparm_file, ioparm_file_len, p->file);
553
554   if (p->status)
555     set_string (&block, &post_block, ioparm_status,
556                 ioparm_status_len, p->status);
557
558   if (p->access)
559     set_string (&block, &post_block, ioparm_access,
560                 ioparm_access_len, p->access);
561
562   if (p->form)
563     set_string (&block, &post_block, ioparm_form, ioparm_form_len, p->form);
564
565   if (p->recl)
566     set_parameter_value (&block, ioparm_recl_in, p->recl);
567
568   if (p->blank)
569     set_string (&block, &post_block, ioparm_blank, ioparm_blank_len,
570                 p->blank);
571
572   if (p->position)
573     set_string (&block, &post_block, ioparm_position,
574                 ioparm_position_len, p->position);
575
576   if (p->action)
577     set_string (&block, &post_block, ioparm_action,
578                 ioparm_action_len, p->action);
579
580   if (p->delim)
581     set_string (&block, &post_block, ioparm_delim, ioparm_delim_len,
582                 p->delim);
583
584   if (p->pad)
585     set_string (&block, &post_block, ioparm_pad, ioparm_pad_len, p->pad);
586
587   if (p->iostat)
588     set_parameter_ref (&block, ioparm_iostat, p->iostat);
589
590   if (p->err)
591     set_flag (&block, ioparm_err);
592
593   tmp = gfc_build_function_call (iocall_open, NULL_TREE);
594   gfc_add_expr_to_block (&block, tmp);
595
596   gfc_add_block_to_block (&block, &post_block);
597
598   io_result (&block, p->err, NULL, NULL);
599
600   return gfc_finish_block (&block);
601 }
602
603
604 /* Translate a CLOSE statement.  */
605
606 tree
607 gfc_trans_close (gfc_code * code)
608 {
609   stmtblock_t block, post_block;
610   gfc_close *p;
611   tree tmp;
612
613   gfc_init_block (&block);
614   gfc_init_block (&post_block);
615
616   set_error_locus (&block, &code->loc);
617   p = code->ext.close;
618
619   if (p->unit)
620     set_parameter_value (&block, ioparm_unit, p->unit);
621
622   if (p->status)
623     set_string (&block, &post_block, ioparm_status,
624                 ioparm_status_len, p->status);
625
626   if (p->iostat)
627     set_parameter_ref (&block, ioparm_iostat, p->iostat);
628
629   if (p->err)
630     set_flag (&block, ioparm_err);
631
632   tmp = gfc_build_function_call (iocall_close, NULL_TREE);
633   gfc_add_expr_to_block (&block, tmp);
634
635   gfc_add_block_to_block (&block, &post_block);
636
637   io_result (&block, p->err, NULL, NULL);
638
639   return gfc_finish_block (&block);
640 }
641
642
643 /* Common subroutine for building a file positioning statement.  */
644
645 static tree
646 build_filepos (tree function, gfc_code * code)
647 {
648   stmtblock_t block;
649   gfc_filepos *p;
650   tree tmp;
651
652   p = code->ext.filepos;
653
654   gfc_init_block (&block);
655
656   set_error_locus (&block, &code->loc);
657
658   if (p->unit)
659     set_parameter_value (&block, ioparm_unit, p->unit);
660
661   if (p->iostat)
662     set_parameter_ref (&block, ioparm_iostat, p->iostat);
663
664   if (p->err)
665     set_flag (&block, ioparm_err);
666
667   tmp = gfc_build_function_call (function, NULL);
668   gfc_add_expr_to_block (&block, tmp);
669
670   io_result (&block, p->err, NULL, NULL);
671
672   return gfc_finish_block (&block);
673 }
674
675
676 /* Translate a BACKSPACE statement.  */
677
678 tree
679 gfc_trans_backspace (gfc_code * code)
680 {
681
682   return build_filepos (iocall_backspace, code);
683 }
684
685
686 /* Translate an ENDFILE statement.  */
687
688 tree
689 gfc_trans_endfile (gfc_code * code)
690 {
691
692   return build_filepos (iocall_endfile, code);
693 }
694
695
696 /* Translate a REWIND statement.  */
697
698 tree
699 gfc_trans_rewind (gfc_code * code)
700 {
701
702   return build_filepos (iocall_rewind, code);
703 }
704
705
706 /* Translate the non-IOLENGTH form of an INQUIRE statement.  */
707
708 tree
709 gfc_trans_inquire (gfc_code * code)
710 {
711   stmtblock_t block, post_block;
712   gfc_inquire *p;
713   tree tmp;
714
715   gfc_init_block (&block);
716   gfc_init_block (&post_block);
717
718   set_error_locus (&block, &code->loc);
719   p = code->ext.inquire;
720
721   if (p->unit)
722     set_parameter_value (&block, ioparm_unit, p->unit);
723
724   if (p->file)
725     set_string (&block, &post_block, ioparm_file, ioparm_file_len, p->file);
726
727   if (p->iostat)
728     set_parameter_ref (&block, ioparm_iostat, p->iostat);
729
730   if (p->exist)
731     set_parameter_ref (&block, ioparm_exist, p->exist);
732
733   if (p->opened)
734     set_parameter_ref (&block, ioparm_opened, p->opened);
735
736   if (p->number)
737     set_parameter_ref (&block, ioparm_number, p->number);
738
739   if (p->named)
740     set_parameter_ref (&block, ioparm_named, p->named);
741
742   if (p->name)
743     set_string (&block, &post_block, ioparm_name, ioparm_name_len, p->name);
744
745   if (p->access)
746     set_string (&block, &post_block, ioparm_access,
747                 ioparm_access_len, p->access);
748
749   if (p->sequential)
750     set_string (&block, &post_block, ioparm_sequential,
751                 ioparm_sequential_len, p->sequential);
752
753   if (p->direct)
754     set_string (&block, &post_block, ioparm_direct,
755                 ioparm_direct_len, p->direct);
756
757   if (p->form)
758     set_string (&block, &post_block, ioparm_form, ioparm_form_len, p->form);
759
760   if (p->formatted)
761     set_string (&block, &post_block, ioparm_formatted,
762                 ioparm_formatted_len, p->formatted);
763
764   if (p->unformatted)
765     set_string (&block, &post_block, ioparm_unformatted,
766                 ioparm_unformatted_len, p->unformatted);
767
768   if (p->recl)
769     set_parameter_ref (&block, ioparm_recl_out, p->recl);
770
771   if (p->nextrec)
772     set_parameter_ref (&block, ioparm_nextrec, p->nextrec);
773
774   if (p->blank)
775     set_string (&block, &post_block, ioparm_blank, ioparm_blank_len,
776                 p->blank);
777
778   if (p->position)
779     set_string (&block, &post_block, ioparm_position,
780                 ioparm_position_len, p->position);
781
782   if (p->action)
783     set_string (&block, &post_block, ioparm_action,
784                 ioparm_action_len, p->action);
785
786   if (p->read)
787     set_string (&block, &post_block, ioparm_read, ioparm_read_len, p->read);
788
789   if (p->write)
790     set_string (&block, &post_block, ioparm_write,
791                 ioparm_write_len, p->write);
792
793   if (p->readwrite)
794     set_string (&block, &post_block, ioparm_readwrite,
795                 ioparm_readwrite_len, p->readwrite);
796
797   if (p->delim)
798     set_string (&block, &post_block, ioparm_delim, ioparm_delim_len,
799                 p->delim);
800
801   if (p->pad)
802     set_string (&block, &post_block, ioparm_pad, ioparm_pad_len,
803                 p->pad); 
804
805   if (p->err)
806     set_flag (&block, ioparm_err);
807
808   tmp = gfc_build_function_call (iocall_inquire, NULL);
809   gfc_add_expr_to_block (&block, tmp);
810
811   gfc_add_block_to_block (&block, &post_block);
812
813   io_result (&block, p->err, NULL, NULL);
814
815   return gfc_finish_block (&block);
816 }
817
818
819 static gfc_expr *
820 gfc_new_nml_name_expr (char * name)
821 {
822    gfc_expr * nml_name;
823    nml_name = gfc_get_expr();
824    nml_name->ref = NULL;
825    nml_name->expr_type = EXPR_CONSTANT;
826    nml_name->ts.kind = gfc_default_character_kind;
827    nml_name->ts.type = BT_CHARACTER;
828    nml_name->value.character.length = strlen(name);
829    nml_name->value.character.string = name;
830
831    return nml_name;
832 }
833
834 static gfc_expr *
835 get_new_var_expr(gfc_symbol * sym)
836 {
837   gfc_expr * nml_var;
838
839   nml_var = gfc_get_expr();
840   nml_var->expr_type = EXPR_VARIABLE;
841   nml_var->ts = sym->ts;
842   if (sym->as)
843     nml_var->rank = sym->as->rank;
844   nml_var->symtree = (gfc_symtree *)gfc_getmem (sizeof (gfc_symtree));
845   nml_var->symtree->n.sym = sym;
846   nml_var->where = sym->declared_at;
847   sym->attr.referenced = 1;
848
849   return nml_var;
850 }
851
852 /* For a scalar variable STRING whose address is ADDR_EXPR, generate a
853    call to iocall_set_nml_val.  For derived type variable, recursively
854    generate calls to iocall_set_nml_val for each leaf field. The leafs
855    have no names -- their STRING field is null, and are interpreted by
856    the run-time library as having only the value, as in the example:
857
858    &foo bzz=1,2,3,4,5/
859
860    Note that the first output field appears after the name of the
861    variable, not of the field name.  This causes a little complication
862    documented below.  */
863
864 static void
865 transfer_namelist_element (stmtblock_t * block, gfc_typespec * ts, tree addr_expr, 
866                            tree string, tree string_length)
867 {
868   tree tmp, args, arg2;
869   tree expr;
870
871   gcc_assert (POINTER_TYPE_P (TREE_TYPE (addr_expr)));
872
873   if (ts->type == BT_DERIVED)
874     {
875       gfc_component *c;
876       expr = gfc_build_indirect_ref (addr_expr);
877
878       for (c = ts->derived->components; c; c = c->next)
879         {
880           tree field = c->backend_decl;
881           gcc_assert (field && TREE_CODE (field) == FIELD_DECL);
882           tmp = build3 (COMPONENT_REF, TREE_TYPE (field), 
883                         expr, field, NULL_TREE);
884
885           if (c->dimension)
886             gfc_todo_error ("NAMELIST IO of array in derived type");
887           if (!c->pointer)
888             tmp = gfc_build_addr_expr (NULL, tmp);
889           transfer_namelist_element (block, &c->ts, tmp, string, string_length);
890
891           /* The first output field bears the name of the topmost
892              derived type variable.  All other fields are anonymous
893              and appear with nulls in their string and string_length
894              fields.  After the first use, we set string and
895              string_length to null.  */
896           string = null_pointer_node;
897           string_length = integer_zero_node;
898         }
899
900       return;
901     }
902
903   args = gfc_chainon_list (NULL_TREE, addr_expr);
904   args = gfc_chainon_list (args, string);
905   args = gfc_chainon_list (args, string_length);
906   arg2 = build_int_cst (gfc_array_index_type, ts->kind);
907   args = gfc_chainon_list (args,arg2);
908
909   switch (ts->type)
910     {
911     case BT_INTEGER:
912       tmp = gfc_build_function_call (iocall_set_nml_val_int, args);
913       break;
914
915     case BT_CHARACTER:
916       expr = gfc_build_indirect_ref (addr_expr);
917       gcc_assert (TREE_CODE (TREE_TYPE (expr)) == ARRAY_TYPE);
918       args = gfc_chainon_list (args,
919                                TYPE_MAX_VALUE (TYPE_DOMAIN (TREE_TYPE (expr))));
920       tmp = gfc_build_function_call (iocall_set_nml_val_char, args);
921       break;
922
923     case BT_REAL:
924       tmp = gfc_build_function_call (iocall_set_nml_val_float, args);
925       break;
926
927     case BT_LOGICAL:
928       tmp = gfc_build_function_call (iocall_set_nml_val_log, args);
929       break;
930
931     case BT_COMPLEX:
932       tmp = gfc_build_function_call (iocall_set_nml_val_complex, args);
933       break;
934
935     default :
936       internal_error ("Bad namelist IO basetype (%d)", ts->type);
937     }
938
939   gfc_add_expr_to_block (block, tmp);
940 }
941
942 /* Create a data transfer statement.  Not all of the fields are valid
943    for both reading and writing, but improper use has been filtered
944    out by now.  */
945
946 static tree
947 build_dt (tree * function, gfc_code * code)
948 {
949   stmtblock_t block, post_block;
950   gfc_dt *dt;
951   tree tmp;
952   gfc_expr *nmlname, *nmlvar;
953   gfc_namelist *nml;
954   gfc_se se,se2;
955
956   gfc_init_block (&block);
957   gfc_init_block (&post_block);
958
959   set_error_locus (&block, &code->loc);
960   dt = code->ext.dt;
961
962   gcc_assert (dt != NULL);
963
964   if (dt->io_unit)
965     {
966       if (dt->io_unit->ts.type == BT_CHARACTER)
967         {
968           set_string (&block, &post_block, ioparm_internal_unit,
969                       ioparm_internal_unit_len, dt->io_unit);
970         }
971       else
972         set_parameter_value (&block, ioparm_unit, dt->io_unit);
973     }
974
975   if (dt->rec)
976     set_parameter_value (&block, ioparm_rec, dt->rec);
977
978   if (dt->advance)
979     set_string (&block, &post_block, ioparm_advance, ioparm_advance_len,
980                 dt->advance);
981
982   if (dt->format_expr)
983     set_string (&block, &post_block, ioparm_format, ioparm_format_len,
984                 dt->format_expr);
985
986   if (dt->format_label)
987     {
988       if (dt->format_label == &format_asterisk)
989         set_flag (&block, ioparm_list_format);
990       else
991         set_string (&block, &post_block, ioparm_format,
992                     ioparm_format_len, dt->format_label->format);
993     }
994
995   if (dt->iostat)
996     set_parameter_ref (&block, ioparm_iostat, dt->iostat);
997
998   if (dt->size)
999     set_parameter_ref (&block, ioparm_size, dt->size);
1000
1001   if (dt->err)
1002     set_flag (&block, ioparm_err);
1003
1004   if (dt->eor)
1005     set_flag(&block, ioparm_eor);
1006
1007   if (dt->end)
1008     set_flag(&block, ioparm_end);
1009
1010   if (dt->namelist)
1011     {
1012        if (dt->format_expr || dt->format_label)
1013           fatal_error("A format cannot be specified with a namelist");
1014
1015        nmlname = gfc_new_nml_name_expr(dt->namelist->name);
1016
1017        set_string (&block, &post_block, ioparm_namelist_name,
1018                 ioparm_namelist_name_len, nmlname);
1019
1020        if (last_dt == READ)
1021           set_flag (&block, ioparm_namelist_read_mode);
1022
1023         for (nml = dt->namelist->namelist; nml; nml = nml->next)
1024           {
1025             gfc_init_se (&se, NULL);
1026             gfc_init_se (&se2, NULL);
1027             nmlvar = get_new_var_expr (nml->sym);
1028             nmlname = gfc_new_nml_name_expr (nml->sym->name);
1029             gfc_conv_expr_reference (&se2, nmlname);
1030             gfc_conv_expr_reference (&se, nmlvar);
1031             gfc_evaluate_now (se.expr, &se.pre); 
1032
1033             transfer_namelist_element (&block, &nml->sym->ts, se.expr,
1034                                        se2.expr, se2.string_length);
1035           }
1036     }
1037
1038   tmp = gfc_build_function_call (*function, NULL_TREE);
1039   gfc_add_expr_to_block (&block, tmp);
1040
1041   gfc_add_block_to_block (&block, &post_block);
1042
1043   return gfc_finish_block (&block);
1044 }
1045
1046
1047 /* Translate the IOLENGTH form of an INQUIRE statement.  We treat
1048    this as a third sort of data transfer statement, except that
1049    lengths are summed instead of actually transferring any data.  */
1050
1051 tree
1052 gfc_trans_iolength (gfc_code * code)
1053 {
1054   stmtblock_t block;
1055   gfc_inquire *inq;
1056   tree dt;
1057
1058   gfc_init_block (&block);
1059
1060   set_error_locus (&block, &code->loc);
1061
1062   inq = code->ext.inquire;
1063
1064   /* First check that preconditions are met.  */
1065   gcc_assert (inq != NULL);
1066   gcc_assert (inq->iolength != NULL);
1067
1068   /* Connect to the iolength variable.  */
1069   if (inq->iolength)
1070     set_parameter_ref (&block, ioparm_iolength, inq->iolength);
1071
1072   /* Actual logic.  */
1073   last_dt = IOLENGTH;
1074   dt = build_dt(&iocall_iolength, code);
1075
1076   gfc_add_expr_to_block (&block, dt);
1077
1078   return gfc_finish_block (&block);
1079 }
1080
1081
1082 /* Translate a READ statement.  */
1083
1084 tree
1085 gfc_trans_read (gfc_code * code)
1086 {
1087
1088   last_dt = READ;
1089   return build_dt (&iocall_read, code);
1090 }
1091
1092
1093 /* Translate a WRITE statement */
1094
1095 tree
1096 gfc_trans_write (gfc_code * code)
1097 {
1098
1099   last_dt = WRITE;
1100   return build_dt (&iocall_write, code);
1101 }
1102
1103
1104 /* Finish a data transfer statement.  */
1105
1106 tree
1107 gfc_trans_dt_end (gfc_code * code)
1108 {
1109   tree function, tmp;
1110   stmtblock_t block;
1111
1112   gfc_init_block (&block);
1113
1114   switch (last_dt)
1115     {
1116     case READ:
1117       function = iocall_read_done;
1118       break;
1119
1120     case WRITE:
1121       function = iocall_write_done;
1122       break;
1123
1124     case IOLENGTH:
1125       function = iocall_iolength_done;
1126       break;
1127
1128     default:
1129       gcc_unreachable ();
1130     }
1131
1132   tmp = gfc_build_function_call (function, NULL);
1133   gfc_add_expr_to_block (&block, tmp);
1134
1135   if (last_dt != IOLENGTH)
1136     {
1137       gcc_assert (code->ext.dt != NULL);
1138       io_result (&block, code->ext.dt->err,
1139                  code->ext.dt->end, code->ext.dt->eor);
1140     }
1141
1142   return gfc_finish_block (&block);
1143 }
1144
1145 static void
1146 transfer_expr (gfc_se * se, gfc_typespec * ts, tree addr_expr);
1147
1148 /* Given an array field in a derived type variable, generate the code
1149    for the loop that iterates over array elements, and the code that
1150    accesses those array elements.  Use transfer_expr to generate code
1151    for transferring that element.  Because elements may also be
1152    derived types, transfer_expr and transfer_array_component are mutually
1153    recursive.  */
1154
1155 static tree
1156 transfer_array_component (tree expr, gfc_component * cm)
1157 {
1158   tree tmp;
1159   stmtblock_t body;
1160   stmtblock_t block;
1161   gfc_loopinfo loop;
1162   int n;
1163   gfc_ss *ss;
1164   gfc_se se;
1165
1166   gfc_start_block (&block);
1167   gfc_init_se (&se, NULL);
1168
1169   /* Create and initialize Scalarization Status.  Unlike in
1170      gfc_trans_transfer, we can't simply use gfc_walk_expr to take
1171      care of this task, because we don't have a gfc_expr at hand.
1172      Build one manually, as in gfc_trans_subarray_assign.  */
1173
1174   ss = gfc_get_ss ();
1175   ss->type = GFC_SS_COMPONENT;
1176   ss->expr = NULL;
1177   ss->shape = gfc_get_shape (cm->as->rank);
1178   ss->next = gfc_ss_terminator;
1179   ss->data.info.dimen = cm->as->rank;
1180   ss->data.info.descriptor = expr;
1181   ss->data.info.data = gfc_conv_array_data (expr);
1182   ss->data.info.offset = gfc_conv_array_offset (expr);
1183   for (n = 0; n < cm->as->rank; n++)
1184     {
1185       ss->data.info.dim[n] = n;
1186       ss->data.info.start[n] = gfc_conv_array_lbound (expr, n);
1187       ss->data.info.stride[n] = gfc_index_one_node;
1188
1189       mpz_init (ss->shape[n]);
1190       mpz_sub (ss->shape[n], cm->as->upper[n]->value.integer,
1191                cm->as->lower[n]->value.integer);
1192       mpz_add_ui (ss->shape[n], ss->shape[n], 1);
1193     }
1194
1195   /* Once we got ss, we use scalarizer to create the loop.  */
1196
1197   gfc_init_loopinfo (&loop);
1198   gfc_add_ss_to_loop (&loop, ss);
1199   gfc_conv_ss_startstride (&loop);
1200   gfc_conv_loop_setup (&loop);
1201   gfc_mark_ss_chain_used (ss, 1);
1202   gfc_start_scalarized_body (&loop, &body);
1203
1204   gfc_copy_loopinfo_to_se (&se, &loop);
1205   se.ss = ss;
1206
1207   /* gfc_conv_tmp_array_ref assumes that se.expr contains the array.  */
1208   se.expr = expr;
1209   gfc_conv_tmp_array_ref (&se);
1210
1211   /* Now se.expr contains an element of the array.  Take the address and pass
1212      it to the IO routines.  */
1213   tmp = gfc_build_addr_expr (NULL, se.expr);
1214   transfer_expr (&se, &cm->ts, tmp);
1215
1216   /* We are done now with the loop body.  Wrap up the scalarizer and
1217      return.  */
1218
1219   gfc_add_block_to_block (&body, &se.pre);
1220   gfc_add_block_to_block (&body, &se.post);
1221
1222   gfc_trans_scalarizing_loops (&loop, &body);
1223
1224   gfc_add_block_to_block (&block, &loop.pre);
1225   gfc_add_block_to_block (&block, &loop.post);
1226
1227   for (n = 0; n < cm->as->rank; n++)
1228     mpz_clear (ss->shape[n]);
1229   gfc_free (ss->shape);
1230
1231   gfc_cleanup_loop (&loop);
1232
1233   return gfc_finish_block (&block);
1234 }
1235
1236 /* Generate the call for a scalar transfer node.  */
1237
1238 static void
1239 transfer_expr (gfc_se * se, gfc_typespec * ts, tree addr_expr)
1240 {
1241   tree args, tmp, function, arg2, field, expr;
1242   gfc_component *c;
1243   int kind;
1244
1245   kind = ts->kind;
1246   function = NULL;
1247   arg2 = NULL;
1248
1249   switch (ts->type)
1250     {
1251     case BT_INTEGER:
1252       arg2 = build_int_cst (NULL_TREE, kind);
1253       function = iocall_x_integer;
1254       break;
1255
1256     case BT_REAL:
1257       arg2 = build_int_cst (NULL_TREE, kind);
1258       function = iocall_x_real;
1259       break;
1260
1261     case BT_COMPLEX:
1262       arg2 = build_int_cst (NULL_TREE, kind);
1263       function = iocall_x_complex;
1264       break;
1265
1266     case BT_LOGICAL:
1267       arg2 = build_int_cst (NULL_TREE, kind);
1268       function = iocall_x_logical;
1269       break;
1270
1271     case BT_CHARACTER:
1272       if (se->string_length)
1273         arg2 = se->string_length;
1274       else
1275         {
1276           tmp = gfc_build_indirect_ref (addr_expr);
1277           gcc_assert (TREE_CODE (TREE_TYPE (tmp)) == ARRAY_TYPE);
1278           arg2 = TYPE_MAX_VALUE (TYPE_DOMAIN (TREE_TYPE (tmp)));
1279         }
1280       function = iocall_x_character;
1281       break;
1282
1283     case BT_DERIVED:
1284       /* Recurse into the elements of the derived type.  */
1285       expr = gfc_evaluate_now (addr_expr, &se->pre);
1286       expr = gfc_build_indirect_ref (expr);
1287
1288       for (c = ts->derived->components; c; c = c->next)
1289         {
1290           field = c->backend_decl;
1291           gcc_assert (field && TREE_CODE (field) == FIELD_DECL);
1292
1293           tmp = build3 (COMPONENT_REF, TREE_TYPE (field), expr, field,
1294                         NULL_TREE);
1295
1296           if (c->dimension)
1297             {
1298               tmp = transfer_array_component (tmp, c);
1299               gfc_add_expr_to_block (&se->pre, tmp);
1300             }
1301           else
1302             {
1303               if (!c->pointer)
1304                 tmp = gfc_build_addr_expr (NULL, tmp);
1305               transfer_expr (se, &c->ts, tmp);
1306             }
1307         }
1308       return;
1309
1310     default:
1311       internal_error ("Bad IO basetype (%d)", ts->type);
1312     }
1313
1314   args = gfc_chainon_list (NULL_TREE, addr_expr);
1315   args = gfc_chainon_list (args, arg2);
1316
1317   tmp = gfc_build_function_call (function, args);
1318   gfc_add_expr_to_block (&se->pre, tmp);
1319   gfc_add_block_to_block (&se->pre, &se->post);
1320
1321 }
1322
1323
1324 /* gfc_trans_transfer()-- Translate a TRANSFER code node */
1325
1326 tree
1327 gfc_trans_transfer (gfc_code * code)
1328 {
1329   stmtblock_t block, body;
1330   gfc_loopinfo loop;
1331   gfc_expr *expr;
1332   gfc_ss *ss;
1333   gfc_se se;
1334   tree tmp;
1335
1336   gfc_start_block (&block);
1337
1338   expr = code->expr;
1339   ss = gfc_walk_expr (expr);
1340
1341   gfc_init_se (&se, NULL);
1342
1343   if (ss == gfc_ss_terminator)
1344     gfc_init_block (&body);
1345   else
1346     {
1347       /* Initialize the scalarizer.  */
1348       gfc_init_loopinfo (&loop);
1349       gfc_add_ss_to_loop (&loop, ss);
1350
1351       /* Initialize the loop.  */
1352       gfc_conv_ss_startstride (&loop);
1353       gfc_conv_loop_setup (&loop);
1354
1355       /* The main loop body.  */
1356       gfc_mark_ss_chain_used (ss, 1);
1357       gfc_start_scalarized_body (&loop, &body);
1358
1359       gfc_copy_loopinfo_to_se (&se, &loop);
1360       se.ss = ss;
1361     }
1362
1363   gfc_conv_expr_reference (&se, expr);
1364
1365   transfer_expr (&se, &expr->ts, se.expr);
1366
1367   gfc_add_block_to_block (&body, &se.pre);
1368   gfc_add_block_to_block (&body, &se.post);
1369
1370   if (se.ss == NULL)
1371     tmp = gfc_finish_block (&body);
1372   else
1373     {
1374       gcc_assert (se.ss == gfc_ss_terminator);
1375       gfc_trans_scalarizing_loops (&loop, &body);
1376
1377       gfc_add_block_to_block (&loop.pre, &loop.post);
1378       tmp = gfc_finish_block (&loop.pre);
1379       gfc_cleanup_loop (&loop);
1380     }
1381
1382   gfc_add_expr_to_block (&block, tmp);
1383
1384   return gfc_finish_block (&block);
1385 }
1386
1387 #include "gt-fortran-trans-io.h"
1388