llvm/flang/test/Lower/io-statement-3.f90

! Test lowering of IO read SIZE control-spec (12.6.2.15)
! RUN: bbc -emit-fir -hlfir=false -o - %s | FileCheck %s

! CHECK-LABEL: func @_QPtest_read_size(
! CHECK-SAME: %[[sizeVar:[^:]+]]: !fir.ref<i32>{{[^,]*}},
subroutine test_read_size(size, c1, c2, unit, stat)
  integer :: unit, size, stat
  character(*) :: c1, c2
  ! CHECK: %[[cookie:.*]] = fir.call @_FortranAioBeginExternalFormattedInput(
  ! CHECK: fir.call @_FortranAioEnableHandlers(
  ! CHECK: %[[ok1:.*]] = fir.call @_FortranAioSetAdvance(
  ! CHECK: fir.if %[[ok1]] {
  ! CHECK:   fir.if %[[ok1]] {
  ! CHECK:     %[[ok2:.*]] = fir.call @_FortranAioInputAscii(
  ! CHECK:     fir.if %[[ok2]] {
  ! CHECK:       fir.call @_FortranAioInputAscii(
  ! CHECK:     }
  ! CHECK:   }
  ! CHECK: }
  ! CHECK: %[[sizeValue:.*]] = fir.call @_FortranAioGetSize(%[[cookie]]) {{.*}}: (!fir.ref<i8>) -> i64
  ! CHECK: %[[sizeCast:.*]] = fir.convert %[[sizeValue]] : (i64) -> i32
  ! CHECK: fir.store %[[sizeCast]] to %[[sizeVar]] : !fir.ref<i32>
  ! CHECK: fir.call @_FortranAioEndIoStatement(%[[cookie]]) {{.*}}: (!fir.ref<i8>) -> i32
  READ(unit, '(A)', ADVANCE='NO', SIZE=size, IOSTAT=stat) c1, c2
end subroutine

  ! CHECK: %[[unit:.*]] = fir.alloca i32 {bindc_name = "unit", uniq_name = "_QFEunit"}
  integer :: unit
  character(7) :: c1
  character(4) :: c2
  integer :: size = 0
  integer :: stat = 0
  ! CHECK: %[[cookie:.*]] = fir.call @_FortranAioBeginOpenNewUnit(%{{.*}}, %{{.*}}) {{.*}}: (!fir.ref<i8>, i32) -> !fir.ref<i8>
  ! CHECK: fir.call @_FortranAioSetAccess(%[[cookie]], %{{.*}}, %{{.*}}) {{.*}}: (!fir.ref<i8>, !fir.ref<i8>, i64) -> i1
  ! CHECK: fir.call @_FortranAioSetAction(%[[cookie]], %{{.*}}, %{{.*}}) {{.*}}: (!fir.ref<i8>, !fir.ref<i8>, i64) -> i1
  ! CHECK: fir.call @_FortranAioSetForm(%[[cookie]], %{{.*}}, %{{.*}}) {{.*}}: (!fir.ref<i8>, !fir.ref<i8>, i64) -> i1
  ! CHECK: fir.call @_FortranAioSetStatus(%[[cookie]], %{{.*}}, %{{.*}}) {{.*}}: (!fir.ref<i8>, !fir.ref<i8>, i64) -> i1
  ! CHECK: %[[kind:.*]] = arith.constant 4 : i32
  ! CHECK: fir.call @_FortranAioGetNewUnit(%[[cookie]], %[[unit]], %[[kind]]) {{.*}}: (!fir.ref<i8>, !fir.ref<i32>, i32) -> i1
  ! CHECK: fir.call @_FortranAioEndIoStatement(%[[cookie]]) {{.*}}: (!fir.ref<i8>) -> i32
  OPEN(NEWUNIT=unit,ACCESS='SEQUENTIAL',ACTION='READWRITE',&
    FORM='FORMATTED',STATUS='SCRATCH')
  WRITE(unit, '(A)') "ABCDEF"
  WRITE(unit, '(A)') "GHIJKL"
  REWIND(unit)
  call test_read_size(size, c1, c2, unit, stat)
  print *, stat, size, c1
  CLOSE(unit)
end