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
|
module template_matrix_01_m
implicit none
private
public :: matrix_t
requirement elemental_op(t, op)
type, deferred :: t
pure elemental function op(l, r) result(rs)
type(t), intent(in) :: l, r
type(t) :: rs
end function
end requirement
template matrix_t(t, plus, times, n)
require :: elemental_op(t, plus), elemental_op(t, times)
integer :: n
private
public :: add_matrix
type :: matrix
type(t) :: elements(n, n)
end type
contains
pure function add_matrix(a, b) result(r)
type(matrix), intent(in) :: a, b
type(matrix) :: r
r%elements = plus(a%elements, b%elements)
end function
pure function mul_matrix(a, b) result(r)
type(matrix), intent(in) :: a, b
type(matrix) :: r
type(t) :: dot(n)
type(t) :: temp
integer :: i, j, k
do i = 1, n
do j = 1, n
dot = times(a%elements(i,:), b%elements(:,j))
temp = dot(1)
do k = 2, n
temp = plus(temp, dot(i))
end do
r%elements(i,j) = temp
end do
end do
end function
end template
end module
program template_matrix_01
use template_matrix_01_m
integer, parameter :: n = 2
integer :: i, j
instantiate matrix_t(integer, operator(+), operator(*), n), &
only: int_matrix => matrix, &
int_add_matrix => add_matrix, &
int_mul_matrix => mul_matrix
type(int_matrix) :: am, bm, cm, dm
do i = 1, n
do j = 1, n
am%elements(i,j) = i
bm%elements(i,j) = i
end do
end do
cm = int_add_matrix(am, bm)
dm = int_mul_matrix(am, bm)
print *, cm%elements(1,1), cm%elements(1,2)
print *, cm%elements(2,1), cm%elements(2,2), achar(10)
print *, dm%elements(1,1), dm%elements(1,2)
print *, dm%elements(2,1), dm%elements(2,2)
end program
|
