bem_simulator.f90 Source File

!> 吸着(insulator)モデルのメインループを実行し、電荷堆積と統計更新を行う。
module bem_simulator
!$ use omp_lib
  use, intrinsic :: iso_fortran_env, only: output_unit
  use, intrinsic :: ieee_arithmetic, only: ieee_is_finite
  use bem_kinds, only: dp, i32, i64
  use bem_types, only: sim_stats, mesh_type, particles_soa, injection_state, sim_config, hit_info
  use bem_app_config, only: app_config, init_particle_batch_from_config
  use bem_physics_config_types, only: field_physics_config, panel_kernel_config, derive_field_panel_config
  use bem_config_helpers, only: resolve_particle_boundaries
  use bem_app_config_runtime, only: particle_source_plan_type, build_particle_source_plan
  use bem_electrostatic_snapshot, only: electrostatic_snapshot_type, electrostatic_diagnostics_type
  use bem_particle_stepper, only: build_particle_step_candidate, resolve_particle_boundary_candidate, &
                                  advance_particle_step_upper_panel_fourier, &
                                  particle_step_result, particle_step_invalid_boundary, &
                                  particle_step_multiple_box_events, particle_step_ambiguous_open_corner
  use bem_collision, only: collision_query_grid_stalled, collision_query_image_limit, &
                           collision_query_index_range, collision_query_invalid_segment, collision_query_ok, find_first_hit
  use bem_surface_models, only: apply_surface_model_charge_relaxation
  use bem_charge_ledger, only: charge_ledger_type, accumulate_charge_ledger
  use bem_string_utils, only: lower_ascii
  use bem_external_boundary_contract, only: external_boundary_contract_type, external_boundary_ok, &
                                            resolve_external_boundary_contract
  use bem_simulator_workspace, only: simulator_batch_workspace_type
  use bem_surface_closure_contract, only: surface_closure_contract_type
  use bem_surface_current_model, only: evaluate_surface_closure
  use bem_output_writer, only: write_matching_plane_history_snapshot
  use bem_mpi, only: mpi_context, mpi_is_root, mpi_allreduce_sum_real_dp_array, mpi_allreduce_sum_real_dp_scalar, &
                     mpi_allreduce_sum_i32_scalar, mpi_allreduce_sum_i64_array, &
                     mpi_allreduce_min_i32_scalar, mpi_allreduce_max_i32_scalar, mpi_select_lowest_rank_i32_values
  implicit none
  private

  public :: run_absorption_insulator

  interface
    module subroutine run_absorption_insulator( &
      mesh, app, stats, history_unit, history_stride, initial_stats, inject_state, mpi, mesh_potential_v, &
      potential_history_unit, top_reference_history_unit, charge_ledger, electrostatic_diagnostics, &
      matching_plane_history_unit &
      )
      type(mesh_type), intent(inout) :: mesh
      type(app_config), intent(in) :: app
      type(sim_stats), intent(out) :: stats
      integer, intent(in), optional :: history_unit
      integer(i32), intent(in), optional :: history_stride
      type(sim_stats), intent(in), optional :: initial_stats
      type(injection_state), intent(inout), optional :: inject_state
      type(mpi_context), intent(in), optional :: mpi
      real(dp), allocatable, intent(out), optional :: mesh_potential_v(:)
      integer, intent(in), optional :: potential_history_unit
      integer, intent(in), optional :: top_reference_history_unit
      type(charge_ledger_type), intent(inout), optional :: charge_ledger
      type(electrostatic_diagnostics_type), intent(out), optional :: electrostatic_diagnostics
      integer, intent(in), optional :: matching_plane_history_unit
    end subroutine run_absorption_insulator

    module subroutine prepare_batch_state( &
      mesh, app, source_plan, snapshot, stats, batch_idx, workspace, pcls_batch, mpi, inject_state, &
      collision_failure_status, collision_failure_species, collision_failure_ray, collision_failure_bounce &
      )
      type(mesh_type), intent(in) :: mesh
      type(app_config), intent(in) :: app
      type(particle_source_plan_type), intent(in) :: source_plan
      type(electrostatic_snapshot_type), intent(inout) :: snapshot
      type(sim_stats), intent(in) :: stats
      integer(i32), intent(out) :: batch_idx
      type(simulator_batch_workspace_type), intent(inout) :: workspace
      type(particles_soa), intent(out) :: pcls_batch
      type(mpi_context), intent(in) :: mpi
      type(injection_state), intent(inout), optional :: inject_state
      integer(i32), intent(out) :: collision_failure_status, collision_failure_species
      integer(i32), intent(out) :: collision_failure_ray, collision_failure_bounce
    end subroutine prepare_batch_state

    module subroutine process_particle_batch( &
      mesh, app, boundary_contract, current_model, snapshot, pcls_batch, dq_thread, escaped_boundary_flag, absorbed_flag, &
      absorbed_element, soft_discarded_boundary_flag, bfield, batch_idx, mpi_rank, &
      actual_team_size, &
      collision_failure_status, collision_failure_particle, collision_failure_step, &
      collision_failure_x, collision_failure_v, matching_plane_moments_thread, retry_counts &
      )
      type(mesh_type), intent(in) :: mesh
      type(app_config), intent(in) :: app
      type(external_boundary_contract_type), intent(in) :: boundary_contract
      type(surface_closure_contract_type), intent(in) :: current_model
      type(electrostatic_snapshot_type), intent(inout) :: snapshot
      type(particles_soa), intent(inout) :: pcls_batch
      real(dp), intent(inout) :: dq_thread(:, :)
      logical, intent(inout) :: escaped_boundary_flag(:), absorbed_flag(:)
      integer(i32), intent(inout) :: absorbed_element(:)
      logical, intent(inout) :: soft_discarded_boundary_flag(:)
      real(dp), intent(in) :: bfield(3)
      integer(i32), intent(in) :: batch_idx, mpi_rank
      integer(i32), intent(out) :: actual_team_size
      integer(i32), intent(out) :: collision_failure_status, collision_failure_particle, collision_failure_step
      real(dp), intent(out) :: collision_failure_x(3), collision_failure_v(3)
      real(dp), intent(inout) :: matching_plane_moments_thread(:, :, :)
      integer(i64), intent(out) :: retry_counts(2)
    end subroutine process_particle_batch

    module subroutine commit_batch_charge(mesh, q_floor, external_e, field_bc_mode, workspace, rel, mpi)
      type(mesh_type), intent(inout) :: mesh
      real(dp), intent(in) :: q_floor
      real(dp), intent(in) :: external_e(3)
      character(len=*), intent(in) :: field_bc_mode
      type(simulator_batch_workspace_type), intent(inout) :: workspace
      real(dp), intent(out) :: rel
      type(mpi_context), intent(in) :: mpi
    end subroutine commit_batch_charge

    module subroutine prepare_adaptive_charge_candidate(mesh, workspace, mpi)
      type(mesh_type), intent(in) :: mesh
      type(simulator_batch_workspace_type), intent(inout) :: workspace
      type(mpi_context), intent(in) :: mpi
    end subroutine prepare_adaptive_charge_candidate

    module subroutine apply_neutral_return_surface_closure(app, pcls_batch, fresh_particle_count, workspace, mpi)
      type(app_config), intent(in) :: app
      type(particles_soa), intent(in) :: pcls_batch
      integer(i32), intent(in) :: fresh_particle_count
      type(simulator_batch_workspace_type), intent(inout) :: workspace
      type(mpi_context), intent(in) :: mpi
    end subroutine apply_neutral_return_surface_closure

    module subroutine apply_fixed_surface_current_closure( &
      app, current_model, pcls_batch, fresh_particle_count, workspace, mpi &
      )
      type(app_config), intent(in) :: app
      type(surface_closure_contract_type), intent(in) :: current_model
      type(particles_soa), intent(in) :: pcls_batch
      integer(i32), intent(in) :: fresh_particle_count
      type(simulator_batch_workspace_type), intent(inout) :: workspace
      type(mpi_context), intent(in) :: mpi
    end subroutine apply_fixed_surface_current_closure

    module subroutine record_batch_initial_charge(app, pcls_batch, fresh_particle_count, ledger)
      type(app_config), intent(in) :: app
      type(particles_soa), intent(in) :: pcls_batch
      integer(i32), intent(in) :: fresh_particle_count
      type(charge_ledger_type), intent(inout) :: ledger
    end subroutine record_batch_initial_charge

    module subroutine record_batch_outcome_charge( &
      pcls_batch, escaped_boundary_flag, absorbed_flag, soft_discarded_boundary_flag, ledger &
      )
      type(particles_soa), intent(in) :: pcls_batch
      logical, intent(in) :: escaped_boundary_flag(:), absorbed_flag(:), soft_discarded_boundary_flag(:)
      type(charge_ledger_type), intent(inout) :: ledger
    end subroutine record_batch_outcome_charge

    module subroutine reduce_charge_ledger_fluxes(ledger, mpi, workspace)
      type(charge_ledger_type), intent(inout) :: ledger
      type(mpi_context), intent(in) :: mpi
      type(simulator_batch_workspace_type), intent(inout) :: workspace
    end subroutine reduce_charge_ledger_fluxes

    module subroutine count_batch_outcomes( &
      pcls_batch, escaped_boundary_flag, absorbed_flag, soft_discarded_boundary_flag, &
      batch_counts, soft_discarded_abs_charge &
      )
      type(particles_soa), intent(in) :: pcls_batch
      logical, intent(in) :: escaped_boundary_flag(:), absorbed_flag(:), soft_discarded_boundary_flag(:)
      integer(i64), intent(out) :: batch_counts(6)
      real(dp), intent(out) :: soft_discarded_abs_charge
    end subroutine count_batch_outcomes

    module subroutine accumulate_batch_stats(stats, batch_counts, soft_discarded_abs_charge, retry_counts, rel)
      type(sim_stats), intent(inout) :: stats
      integer(i64), intent(in) :: batch_counts(6)
      integer(i64), intent(in) :: retry_counts(2)
      real(dp), intent(in) :: soft_discarded_abs_charge, rel
    end subroutine accumulate_batch_stats

    module subroutine print_batch_progress(batch_idx, final_batch_idx, rel_change)
      integer(i32), intent(in) :: batch_idx, final_batch_idx
      real(dp), intent(in) :: rel_change
    end subroutine print_batch_progress

    module subroutine maybe_write_history_snapshot(history_enabled, hist_unit, hist_stride, stats, rel, q_elem)
      logical, intent(in) :: history_enabled
      integer, intent(in) :: hist_unit
      integer(i32), intent(in) :: hist_stride
      type(sim_stats), intent(in) :: stats
      real(dp), intent(in) :: rel, q_elem(:)
    end subroutine maybe_write_history_snapshot

    module subroutine write_history_snapshot(unit_id, batch_idx, processed_particles, rel_change, q_elem)
      integer, intent(in) :: unit_id
      integer(i32), intent(in) :: batch_idx
      integer(i64), intent(in) :: processed_particles
      real(dp), intent(in) :: rel_change, q_elem(:)
    end subroutine write_history_snapshot

    module subroutine maybe_write_potential_history_snapshot( &
      potential_history_enabled, pot_hist_unit, hist_stride, stats, snapshot, mesh, sim, potential_buf, &
      top_reference_history_enabled, top_reference_history_unit &
      )
      logical, intent(in) :: potential_history_enabled
      integer, intent(in) :: pot_hist_unit
      integer(i32), intent(in) :: hist_stride
      type(sim_stats), intent(in) :: stats
      type(electrostatic_snapshot_type), intent(inout) :: snapshot
      type(mesh_type), intent(inout) :: mesh
      type(sim_config), intent(in) :: sim
      real(dp), intent(inout) :: potential_buf(:)
      logical, intent(in) :: top_reference_history_enabled
      integer, intent(in) :: top_reference_history_unit
    end subroutine maybe_write_potential_history_snapshot

    module subroutine write_potential_history_snapshot(unit_id, batch_idx, potential_v)
      integer, intent(in) :: unit_id
      integer(i32), intent(in) :: batch_idx
      real(dp), intent(in) :: potential_v(:)
    end subroutine write_potential_history_snapshot
  end interface

end module bem_simulator