1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101
|
module template_simple_03_m
implicit none
private
public :: generic_sum
requirement operator_r(T, U, V, binary_func)
type, deferred :: T
type, deferred :: U
type, deferred :: V
pure elemental function binary_func(lhs, rhs) result(res)
type(T), intent(in) :: lhs
type(U), intent(in) :: rhs
type(V) :: res
end function
end requirement
requirement cast_r(T, cast)
type, deferred :: T
pure elemental function cast(arg) result(res)
integer, intent(in) :: arg
type(T) :: res
end function
end requirement
contains
pure elemental function cast_integer(arg) result(res)
integer, intent(in) :: arg
integer :: res
res = 0
end function
pure elemental function cast_real(arg) result(res)
integer, intent(in) :: arg
real :: res
res = 0.0
end function
pure elemental function div_integer(lhs, rhs) result(res)
integer, intent(in) :: lhs, rhs
integer :: res
res = lhs / rhs
end function
pure elemental function div_real(lhs, rhs) result(res)
real, intent(in) :: lhs
integer, intent(in) :: rhs
real :: res
res = lhs / rhs
end function
pure function generic_sum {T, add, cast} (arr) result(res)
require :: operator_r(T, T, T, add), cast_r(T, cast)
type(T), intent(in) :: arr(:)
type(T) :: res
integer :: n, i
n = size(arr)
res = cast(0)
if (n > 0) then
res = arr(1)
do i=2,n
res = add(res, arr(i))
end do
end if
end function
pure function generic_avg {T, add, cast, div} (arr) result(res)
require :: operator_r(T, T, T, add), cast_r(T, cast)
require :: operator_r(T, integer, T, div)
type(T), intent(in) :: arr(:)
type(T) :: res
integer :: n, i
n = size(arr)
if (n > 0) then
res = div(generic_sum{T, add, cast}(arr), n)
else
res = cast(0)
end if
end function
subroutine test_template()
integer :: a_i(10), i, s_i
real :: a_r(10), s_r
do i = 1, size(a_i)
a_i(i) = i
a_r(i) = i
end do
s_i = generic_avg{integer, operator(+), cast_integer, div_integer}(a_i)
s_r = generic_avg{real, operator(+), cast_real, div_real}(a_r)
print *, s_i
end subroutine
end module
program template_simple_03
use template_simple_03_m
implicit none
call test_template()
end
|