crp/OpenMC
1
0
Fork 0
mirror of https://github.com/openmc-dev/openmc.git synced 2026-07-28 22:26:08 -04:00
Code Issues Projects Releases Packages Wiki Activity Actions 5

Ability to start from binary source file (not in parallel).

Browse source
This commit is contained in:
Paul Romano 2012-03-22 15:59:01 -04:00
parent dffed0bfda
commit 17bc90a52e
3 changed files with 62 additions and 19 deletions
Download patch file Download diff file Expand all files Collapse all files

3
src/constants.F90
View file

@ -322,7 +322,8 @@ module constants
! Source types
integer, parameter :: &
SRC_BOX = 1, & ! Source in a rectangular prism
SRC_POINT = 2 ! Source at a single point
SRC_POINT = 2, & ! Source at a single point
SRC_FILE = 3 ! Source from a file
integer, parameter :: &
PROB_SOURCE = 1, & ! External source problem

26
src/input_xml.F90
View file

@ -137,29 +137,33 @@ contains
! Determine external source type
type = source_ % type
call lower_case(type)
select case (trim(type))
select case (type)
case ('box')
external_source % type = SRC_BOX
coeffs_reqd = 6
case ('point')
external_source % type = SRC_POINT
coeffs_reqd = 3
case ('file')
external_source % type = SRC_FILE
case default
message = "Invalid source type: " // trim(source_ % type)
call fatal_error()
end select
! Coefficients for external surface
n = size(source_ % coeffs)
if (n < coeffs_reqd) then
message = "Not enough coefficients specified for external source."
call fatal_error()
elseif (n > coeffs_reqd) then
message = "Too many coefficients specified for external source."
call fatal_error()
else
allocate(external_source % values(n))
external_source % values = source_ % coeffs
if (type /= 'file') then
n = size(source_ % coeffs)
if (n < coeffs_reqd) then
message = "Not enough coefficients specified for external source."
call fatal_error()
elseif (n > coeffs_reqd) then
message = "Too many coefficients specified for external source."
call fatal_error()
else
allocate(external_source % values(n))
external_source % values = source_ % coeffs
end if
end if
end if

52
src/source.F90
View file

@ -43,6 +43,13 @@ contains
! Determine maximum amount of particles to simulate on each processor
maxwork = ceiling(real(n_particles)/n_procs,8)
! ID's of first and last source particles
bank_first = rank*maxwork + 1
bank_last = min((rank+1)*maxwork, n_particles)
! number of particles for this processor
work = bank_last - bank_first + 1
! Allocate source bank
allocate(source_bank(maxwork), STAT=alloc_err)
if (alloc_err /= 0) then
@ -69,16 +76,16 @@ contains
call fatal_error()
end if
! Read the source from a binary file instead of sampling from some assumed
! source distribution
if (external_source % type == SRC_FILE) then
call read_source_binary()
return
end if
! Initialize first cycle source bank
do i = 0, n_procs - 1
if (rank == i) then
! ID's of first and last source particles
bank_first = rank*maxwork + 1
bank_last = min((rank+1)*maxwork, n_particles)
! number of particles for this processor
work = bank_last - bank_first + 1
do j = 1, work
id = bank_first + j - 1
source_bank(j) % id = id
@ -216,4 +223,35 @@ contains
end subroutine write_source_binary
!===============================================================================
! READ_SOURCE reads a source distribution from a source.binary file and
! initializes the source bank
!===============================================================================
subroutine read_source_binary()
integer(8) :: i ! index in source_bank
integer(8) :: n_sites ! number of sites in binary file
! Open binary source file for reading
open(UNIT=UNIT_SOURCE, FILE='source.binary', STATUS='old', &
ACCESS='stream')
! Read number of source sites in file
read(UNIT=UNIT_SOURCE) n_sites
do i = 1, maxwork
! Set ID for source site
source_bank(i) % id = i ! bank_first + i - 1
! Read position, angle, and energy
read(UNIT=UNIT_SOURCE) source_bank(i) % xyz, &
source_bank(i) % uvw, source_bank(i) % E
end do
! Close binary source file
close(UNIT=UNIT_SOURCE)
end subroutine read_source_binary
end module source