xref: /llvm-project/flang/test/Lower/io-statement-3.f90 (revision f35f863a88f83332bef9605ef4cfe4f05c066efb)
1! Test lowering of IO read SIZE control-spec (12.6.2.15)
2! RUN: bbc -emit-fir -hlfir=false -o - %s | FileCheck %s
3
4! CHECK-LABEL: func @_QPtest_read_size(
5! CHECK-SAME: %[[sizeVar:[^:]+]]: !fir.ref<i32>{{[^,]*}},
6subroutine test_read_size(size, c1, c2, unit, stat)
7  integer :: unit, size, stat
8  character(*) :: c1, c2
9  ! CHECK: %[[cookie:.*]] = fir.call @_FortranAioBeginExternalFormattedInput(
10  ! CHECK: fir.call @_FortranAioEnableHandlers(
11  ! CHECK: %[[ok1:.*]] = fir.call @_FortranAioSetAdvance(
12  ! CHECK: fir.if %[[ok1]] {
13  ! CHECK:   fir.if %[[ok1]] {
14  ! CHECK:     %[[ok2:.*]] = fir.call @_FortranAioInputAscii(
15  ! CHECK:     fir.if %[[ok2]] {
16  ! CHECK:       fir.call @_FortranAioInputAscii(
17  ! CHECK:     }
18  ! CHECK:   }
19  ! CHECK: }
20  ! CHECK: %[[sizeValue:.*]] = fir.call @_FortranAioGetSize(%[[cookie]]) {{.*}}: (!fir.ref<i8>) -> i64
21  ! CHECK: %[[sizeCast:.*]] = fir.convert %[[sizeValue]] : (i64) -> i32
22  ! CHECK: fir.store %[[sizeCast]] to %[[sizeVar]] : !fir.ref<i32>
23  ! CHECK: fir.call @_FortranAioEndIoStatement(%[[cookie]]) {{.*}}: (!fir.ref<i8>) -> i32
24  READ(unit, '(A)', ADVANCE='NO', SIZE=size, IOSTAT=stat) c1, c2
25end subroutine
26
27  ! CHECK: %[[unit:.*]] = fir.alloca i32 {bindc_name = "unit", uniq_name = "_QFEunit"}
28  integer :: unit
29  character(7) :: c1
30  character(4) :: c2
31  integer :: size = 0
32  integer :: stat = 0
33  ! CHECK: %[[cookie:.*]] = fir.call @_FortranAioBeginOpenNewUnit(%{{.*}}, %{{.*}}) {{.*}}: (!fir.ref<i8>, i32) -> !fir.ref<i8>
34  ! CHECK: fir.call @_FortranAioSetAccess(%[[cookie]], %{{.*}}, %{{.*}}) {{.*}}: (!fir.ref<i8>, !fir.ref<i8>, i64) -> i1
35  ! CHECK: fir.call @_FortranAioSetAction(%[[cookie]], %{{.*}}, %{{.*}}) {{.*}}: (!fir.ref<i8>, !fir.ref<i8>, i64) -> i1
36  ! CHECK: fir.call @_FortranAioSetForm(%[[cookie]], %{{.*}}, %{{.*}}) {{.*}}: (!fir.ref<i8>, !fir.ref<i8>, i64) -> i1
37  ! CHECK: fir.call @_FortranAioSetStatus(%[[cookie]], %{{.*}}, %{{.*}}) {{.*}}: (!fir.ref<i8>, !fir.ref<i8>, i64) -> i1
38  ! CHECK: %[[kind:.*]] = arith.constant 4 : i32
39  ! CHECK: fir.call @_FortranAioGetNewUnit(%[[cookie]], %[[unit]], %[[kind]]) {{.*}}: (!fir.ref<i8>, !fir.ref<i32>, i32) -> i1
40  ! CHECK: fir.call @_FortranAioEndIoStatement(%[[cookie]]) {{.*}}: (!fir.ref<i8>) -> i32
41  OPEN(NEWUNIT=unit,ACCESS='SEQUENTIAL',ACTION='READWRITE',&
42    FORM='FORMATTED',STATUS='SCRATCH')
43  WRITE(unit, '(A)') "ABCDEF"
44  WRITE(unit, '(A)') "GHIJKL"
45  REWIND(unit)
46  call test_read_size(size, c1, c2, unit, stat)
47  print *, stat, size, c1
48  CLOSE(unit)
49end
50