File: template_interface_01.f90

package info (click to toggle)
lfortran 0.45.0-1
  • links: PTS, VCS
  • area: main
  • in suites: sid, trixie
  • size: 46,332 kB
  • sloc: cpp: 137,068; f90: 51,260; python: 6,444; ansic: 4,277; yacc: 2,285; fortran: 806; sh: 524; makefile: 30; javascript: 15
file content (105 lines) | stat: -rw-r--r-- 2,857 bytes parent folder | download
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
102
103
104
105
module template_interface_01_m
    implicit none
    public :: test_template

    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

    template sum_t(T, add, cast)
        require :: operator_r(T, T, T, add), cast_r(T, cast)
        private
        public :: generic_sum

        interface operator(+)
            procedure add
        end interface
    contains
        pure function generic_sum(arr) result(res)
            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 = res + arr(i)
                end do
            end if       
        end function
    end template

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 function simple_generic_sum {T, add, cast} (arr) result(res)
        require :: operator_r(T, T, T, add), cast_r(T, cast)
        interface operator(+)
            procedure add
        end interface
        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 = res + arr(i)
            end do
        end if    
    end function

    subroutine test_template()
        instantiate sum_t(integer, operator(+), cast_integer), only: generic_sum_integer => generic_sum
        instantiate sum_t(real, operator(+), cast_real), only: generic_sum_real => generic_sum
        integer :: ai(10), i, ri
        real :: ar(10), rr
        do i = 1, 10
            ai(i) = i
            ar(i) = i
        end do
        ri = generic_sum_integer(ai)
        rr = generic_sum_real(ar)
        print *, ri
        print *, rr
        print *, simple_generic_sum{integer, operator(+), cast_integer}(ai)
        print *, simple_generic_sum{real, operator(+), cast_real}(ar)
    end subroutine

end module

program template_interface_01
use template_interface_01_m
implicit none

call test_template()

end program