! RUN: bbc -emit-hlfir %s -o - | FileCheck %s

! Test lowering of elemental calls with array arguments that use array
! elements as indices.
! As reported in issue #62981, wrong code was being generated in this case.

module test_ops
  implicit none
  interface
    integer elemental function elem_func_i(i)
      integer, intent(in) :: i
    end function
    real elemental function elem_func_r(r)
      real, intent(in) :: r
    end function
  end interface

  integer :: a(3), b(3), v(3), i, j, k, l
  real :: x(2), y(2), u

contains
! CHECK-LABEL: func @_QMtest_opsPcheck_array_elems_as_indices() {
subroutine check_array_elems_as_indices()
! CHECK: %[[A_ADDR:.*]] = fir.address_of(@_QMtest_opsEa) : !fir.ref<!fir.array<3xi32>>
! CHECK: %[[A_DECL:.*]]:2 = hlfir.declare %[[A_ADDR]]
! CHECK: %[[V_ADDR:.*]] = fir.address_of(@_QMtest_opsEv) : !fir.ref<!fir.array<3xi32>>
! CHECK: %[[V_DECL:.*]]:2 = hlfir.declare %[[V_ADDR]]
! CHECK: hlfir.forall lb {
! CHECK: } ub {
! CHECK: } (%[[I:.*]]: i32) {
! CHECK:   %[[I_IDX:.*]] = hlfir.forall_index "i" %[[I]]
! CHECK:   hlfir.region_assign {
! CHECK:     %[[SLICE:.*]] = hlfir.designate %[[A_DECL]]#0
! CHECK:     hlfir.elemental {{.*}} unordered : (!fir.shape<1>) -> !hlfir.expr<?xi32> {
! CHECK:     ^bb0(%[[ARG:.*]]: index):
! CHECK:       %[[ELEM_ADDR:.*]] = hlfir.designate %[[SLICE]] (%[[ARG]])
! CHECK:       %[[RES:.*]] = fir.call @_QPelem_func_i(%[[ELEM_ADDR]])
! CHECK:       hlfir.yield_element %[[RES]] : i32
! CHECK:     }
! CHECK:   }
! CHECK: }
  forall (i=1:3)
    b(i:i) = elem_func_i(a(v(i):v(i)))
  end forall
end subroutine

! CHECK-LABEL: func @_QMtest_opsPcheck_not_assert() {
subroutine check_not_assert()
  ! Implicit path.
! CHECK: hlfir.elemental
! CHECK: fir.call @_QPelem_func_i
  b = 10 + elem_func_i(a)

  ! Expression as argument, instead of variable.
  forall (i=1:3)
    b(i:i) = elem_func_i(a(i:i) + a(i:i))
  end forall

  ! Nested elemental function calls.
  y = elem_func_r(cos(x))
  y = elem_func_r(cos(x) + u)

  ! Array constructors as elemental function arguments.
  y = atan2( (/ (real(i, 4), i = 1, 2) /), &
             real( (/ (i, i = j, k, l) /), 4) )
end subroutine

end module