topopt.f90

program topopt
  use simulation_m, only: simulation_t
  use design, only: design_t, design_factory
  use problem, only: problem_t
  use optimizer, only: optimizer_t, optimizer_factory
 
  use json_module, only: json_file
  use json_utils, only: json_extract_object
  use utils, only: neko_error
  use json_utils_ext, only: json_read_file
  use neko_top, only: neko_top_register_types
 
  use mpi_f08, only: mpi_init
  implicit none
 
  ! JSON related arguments
  integer :: argc
  character(len=256) :: parameter_file
  type(json_file) :: parameters, design_parameters
 
  ! MPI parameters
  integer :: ierr
 
  type(simulation_t) :: sim
  class(design_t), allocatable :: des
  type(problem_t) :: prob
  class(optimizer_t), allocatable :: opt
 
  ! -------------------------------------------------------------------------- !
  ! Initialize the MPI environment
 
  call mpi_init(ierr)
  call neko_top_register_types()
 
  ! -------------------------------------------------------------------------- !
  ! Read the parameters file as the first terminal argument
 
  argc = command_argument_count()
  if (argc .lt. 1) call neko_error('Missing parameter file')
  call get_command_argument(1, parameter_file)
 
  ! Read the parameters file
  parameters = json_read_file(trim(parameter_file))
  call json_extract_object(parameters, 'optimization.design', design_parameters)
 
  ! -------------------------------------------------------------------------- !
  ! Initialization of the components
 
  call sim%init(parameters)
  call design_factory(des, design_parameters, sim)
  call prob%init(parameters, des, sim)
  call optimizer_factory(opt, parameters, prob, des, sim)
 
  ! -------------------------------------------------------------------------- !
  ! Execute the optimization
 
  call opt%run(prob, des, sim)
 
  ! -------------------------------------------------------------------------- !
  ! Clean up the components
 
  call opt%free()
  call prob%free()
  call des%free()
  call sim%free()
 
  if (allocated(des)) deallocate(des)
  if (allocated(opt)) deallocate(opt)
 
end program topopt