sam/write_field.f at master · tslund/sam · GitHub

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
      subroutine write_field( fname, u, work )

      include 'sam.h'
      include 'mpif.h'

      complex u(Nze,Ny_min,nxp,Lu), work(Nz_min,Ny_min)
      character(*) fname
      character(6) ext
      integer(kind=mpi_offset_kind) :: offset, irecl_out
      integer amode, status(mpi_status_size)

      n_words_out = Nz_min*Ny_min
      irecl_out   = n_words_out*16

      amode = ior( mpi_mode_create, mpi_mode_wronly )

      call mpi_file_open( mpi_comm_world, fname, amode,
     &                    mpi_info_null, fh(3), ierr )

      do n=1, Lu
         do i=1, nxp
            ii = ixs + i-1
            do j=1, Ny_min
               do k=1, Nz_min
                  work(k,j) = u(k,j,i,n)
               end do
            end do
            offset = int(((n-1)*Nx_min + (ii-1)),8)*irecl_out
            call mpi_file_write_at( fh(3), offset, work,
     &           n_words_out, mpi_double_complex, status, ierr )
         end do
      end do

      call mpi_file_close( fh(3), ierr )

      if( l_root .and. i_stat .ne. 0 ) then
         write(ext,'(i6.6)') nt
         open(unit=1,file='stat.'//ext,form='unformatted')
         write(1) Nx, Ny, Nz
         write(1) time, dt, i_cfl, dt0, cfl0, nrk_max
         write(1) vis, i_les
         write(1) t_stat0+(t-t_start)
         write(1) stat
         close(1)
      end if

      return
      end