149 lines
4.5 KiB
Fortran
149 lines
4.5 KiB
Fortran
!================================================================================
|
|
! M_WORK - module that contains working arrays wrk1....wrk15.
|
|
!
|
|
! Time-stamp: <2009-08-18 13:47:59 (chumakov)>
|
|
!================================================================================
|
|
module m_work
|
|
|
|
! use m_openmpi
|
|
use m_parameters
|
|
use m_io
|
|
|
|
|
|
implicit none
|
|
|
|
! --- work arrays
|
|
real*8, allocatable :: tmp4(:,:,:)
|
|
|
|
real*8, allocatable :: wrk(:,:,:,:)
|
|
|
|
real*8, allocatable :: wrkp(:,:)
|
|
|
|
real*8, allocatable :: rhs_old(:,:,:,:)
|
|
|
|
contains
|
|
|
|
!================================================================================
|
|
!================================================================================
|
|
!================================================================================
|
|
!================================================================================
|
|
subroutine m_work_init
|
|
|
|
use m_parameters
|
|
implicit none
|
|
|
|
!!$ write(out,*) "Inside m_work_init: ",task
|
|
!!$ call flush(out)
|
|
|
|
! allocating work arrays
|
|
if (task.eq.'hydro') then
|
|
|
|
if (les_model.ge.4) then
|
|
! The dynamic structure LES model needs more wrk arrays than usual
|
|
call m_work_allocate(max(6,3+n_scalars+n_les+4))
|
|
else
|
|
call m_work_allocate(max(6,3+n_scalars+n_les+2))
|
|
end if
|
|
|
|
elseif (task.eq.'stats') then
|
|
|
|
call m_work_allocate(6)
|
|
|
|
elseif (task.eq.'parts') then
|
|
|
|
! required recources are different for the "parts" part of the code
|
|
! we need several layers of velocities for velocity interpolation
|
|
select case (particles_tracking_scheme)
|
|
case (0)
|
|
allocate(wrk(1:nx+2,1:ny,1:3,0:0),stat=ierr)
|
|
write(out,*) "Allocated wrk(1:nx+2,1:ny,1:3,0:0)", ierr
|
|
wrk = zip
|
|
case (1)
|
|
allocate(wrk(1:nx+2,1:ny,1:3,1:3),stat=ierr)
|
|
write(out,*) "Allocated wrk(1:nx+2,1:ny,1:3,1:3)", ierr
|
|
wrk = zip
|
|
case default
|
|
stop 'wrong particles_tracking_scheme'
|
|
end select
|
|
allocate(tmp4(nx,ny,nz), stat=ierr)
|
|
write(out,*) "Allocated tmp4."
|
|
call flush(out)
|
|
|
|
else
|
|
write(out,*) 'TASK variable is set in such a way that I dont know how to allocate work arrays'
|
|
write(out,*) 'task = ',task
|
|
call my_exit(-1)
|
|
end if
|
|
|
|
!!$ write(out,*) "Finished m_work_init"
|
|
!!$ call flush(out)
|
|
|
|
return
|
|
end subroutine m_work_init
|
|
|
|
!================================================================================
|
|
!================================================================================
|
|
! allocating and defining the prescribed number of arrays
|
|
!================================================================================
|
|
!================================================================================
|
|
|
|
subroutine m_work_allocate(number)
|
|
|
|
implicit none
|
|
integer :: number
|
|
integer :: i, ierr, ierr_total
|
|
|
|
ierr = 0
|
|
|
|
!!$ write(out,"('Allocating work: ',i3)") number
|
|
!!$ call flush(out)
|
|
|
|
! array that is needed for output (nx,ny,nz)
|
|
allocate(tmp4(nx,ny,nz),stat=i); ierr = ierr + i;
|
|
|
|
! main working array, needed for FFT etc, so (nx+2,ny,nz)
|
|
allocate(wrk(nx+2,ny,nz,0:number),stat=i); ierr = ierr + i
|
|
|
|
! array for the spare RHS for Adams-Bashforth time-stepping scheme methods
|
|
if (task.eq.'hydro') then
|
|
allocate(rhs_old(nx+2,ny,nz,3+n_scalars+n_les),stat=i); ierr = ierr + i
|
|
end if
|
|
|
|
if (ierr.ne.0) then
|
|
print *,'*** WORK_INIT: error in allocation, stopping. ',ierr
|
|
print *,'*** task = ',task
|
|
print *,'*** myid = ',myid
|
|
call my_exit(-1)
|
|
stop
|
|
end if
|
|
|
|
if (allocated(wrk)) write(out,"('allocated wrk(nx+2,ny,nz,0:',i3,')')") number
|
|
if (allocated(rhs_old)) write(out,"('allocated rhs_old(nx+2,ny,nz,1:',i3,')')") 3+n_scalars+n_les
|
|
call flush(out)
|
|
|
|
tmp4 = 0.0
|
|
wrk = zip
|
|
if (allocated(rhs_old)) rhs_old = 0.d0
|
|
|
|
|
|
return
|
|
end subroutine m_work_allocate
|
|
|
|
!================================================================================
|
|
!================================================================================
|
|
|
|
subroutine m_work_exit
|
|
implicit none
|
|
|
|
! write(out,*) 'deallocaing tmp4'; call flush(out)
|
|
if (allocated(tmp4)) deallocate(tmp4)
|
|
! write(out,*) 'deallocaing wrk'; call flush(out)
|
|
if (allocated(wrk)) deallocate(wrk)
|
|
! write(out,*) 'deallocaing wrkp'; call flush(out)
|
|
if (allocated(wrkp)) deallocate(wrkp)
|
|
write(out,*) 'deallocated wrk arrays'; call flush(out)
|
|
|
|
return
|
|
end subroutine m_work_exit
|
|
|
|
end module m_work
|