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
|
module CoProcessor
implicit none
public initializecoprocessor, runcoprocessor, finalizecoprocessor
contains
subroutine initializecoprocessor()
use catalyst
use catalyst_python
implicit none
integer :: ilen, i
character(len=200) :: arg
call catalyst_initialize()
do i=1, iargc()
call getarg(i, arg)
call catalyst_add_python_script(arg)
enddo
end subroutine initializecoprocessor
subroutine runcoprocessor(dimensions, step, time, x)
use iso_c_binding
use catalyst
implicit none
integer, intent(in) :: dimensions(3), step
real(kind=8), dimension(:), intent(in) :: x
real(kind=8), intent(in) :: time
integer :: extent(6)
logical :: flag
real(kind=8), DIMENSION(:), allocatable :: xcp(:)
flag = catalyst_request_data_description(step, time)
if (flag) then
flag = catalyst_need_to_create_grid()
call getvtkextent(dimensions, extent)
if (flag) then
call createcpimagedata(dimensions, extent)
end if
! x is the array with global values, we need just this process's
! values for Catalyst which will be put in xcp
allocate(xcp((extent(2)-extent(1)+1)*(extent(4)-extent(3)+1)*(extent(6)-extent(5)+1)))
call getlocalfield(dimensions, extent, x, xcp)
! adding //char(0) appends the C++ terminating character
! to the Fortran array
call addfield(xcp,"solution"//char(0))
call catalyst_process()
deallocate(xcp)
end if
end subroutine runcoprocessor
subroutine finalizecoprocessor()
use catalyst
call catalyst_finalize()
end subroutine finalizecoprocessor
! helper methods
subroutine getvtkextent(dimensions, extent)
use Box
implicit none
include 'mpif.h'
integer, intent(in) :: dimensions(3)
integer, intent(out) :: extent(6)
integer :: numtasks, rank, ierr
call mpi_comm_size(MPI_COMM_WORLD, numtasks, ierr)
call mpi_comm_rank(MPI_COMM_WORLD, rank, ierr)
call getlocalbox(rank+1, numtasks, dimensions, extent)
end subroutine getvtkextent
subroutine getlocalfield(dimensions, extent, x, xcp)
implicit none
integer :: i, j, k, counter
integer, intent(in) :: dimensions(3), extent(6)
real(kind=8), dimension(:), intent(in) :: x
real(kind=8), dimension(:), intent(out) :: xcp
counter = 1
do k=extent(5), extent(6)
do j=extent(3), extent(4)
do i=extent(1), extent(2)
xcp(counter) = x(i+(j-1)*dimensions(1)+(k-1)*dimensions(1)*dimensions(2))
counter = counter + 1
enddo
enddo
enddo
end subroutine getlocalfield
end module CoProcessor
|
