control.F90 Source File

module module_control
    use, intrinsic :: iso_fortran_env
    use :: module_core
    use :: control_time, only:type_time
    use :: control_iteration, only:type_iteration
    use :: control_openmp, only:initialize_openmp
    use :: module_input, only:type_input
    implicit none
    private

    public :: type_time
    public :: type_iteration
    public :: type_controls

    public :: calc_thermal, calc_hydraulic, calc_mechanical
    integer(int32), parameter :: calc_thermal = 1
    integer(int32), parameter :: calc_hydraulic = 2
    integer(int32), parameter :: calc_mechanical = 3

    type :: type_controls
        logical :: calculate_thermal
        logical :: calculate_hydraulic
        logical :: calculate_mechanical
        character(:), allocatable :: coupling_mode
        ! --- マテリアルごとのフラグ ---
        logical, allocatable :: thermal(:)
        logical, allocatable :: hydraulic(:)
        logical, allocatable :: mechanical(:)

        type(type_iteration) :: iteration
        type(type_time) :: time
    contains
        procedure :: initialize => initialize_type_controls
        procedure :: is_target => should_calculate_target
    end type type_controls

contains
    subroutine initialize_type_controls(self, input)
        implicit none
        class(type_controls), intent(inout) :: self
        class(type_input), intent(in) :: input

        integer(int32), allocatable :: unique_material_ids(:)
        integer(int32) :: ierr
        integer(int32) :: i, num_unique_regions, max_region_id
        integer(int32) :: current_material_id

        ierr = 0
        ! アクティブなマテリアル領域の情報を取得
        call input%geometry%vtk%get_active_region_info(unique_material_ids, ierr)
        if (ierr /= 0) return
        if (.not. allocated(unique_material_ids) .or. size(unique_material_ids) == 0) then
            ierr = -1 ! エラーコード
            print *, "Error: No active material regions found."
            stop 1
        end if

        num_unique_regions = size(unique_material_ids)
        max_region_id = maxval(unique_material_ids)

        ! 全体的な計算フラグを設定し、配列を割り当てて初期化
        if (input%basic%analysis_controls%calculate_thermal) then
            self%calculate_thermal = .true.
            allocate (self%thermal(max_region_id))
            self%thermal = .false. ! 配列全体を .false. で初期化
        else
            self%calculate_thermal = .false.
        end if

        if (input%basic%analysis_controls%calculate_hydraulic) then
            self%calculate_hydraulic = .true.
            allocate (self%hydraulic(max_region_id))
            self%hydraulic = .false.
        else
            self%calculate_hydraulic = .false.
        end if

        if (input%basic%analysis_controls%calculate_mechanical) then
            self%calculate_mechanical = .true.
            allocate (self%mechanical(max_region_id))
            self%mechanical = .false.
        else
            self%calculate_mechanical = .false.
        end if

        ! アクティブな各マテリアル領域に対してフラグを立てる
        do i = 1, num_unique_regions
            current_material_id = unique_material_ids(i)
            if (self%calculate_thermal) then
                ! materials配列の添え字として実際の材料IDを使用
                self%thermal(current_material_id) = input%basic%materials(i)%calculate_thermal
            end if

            if (self%calculate_hydraulic) then
                ! materials配列の添え字として実際の材料IDを使用
                self%hydraulic(current_material_id) = input%basic%materials(i)%calculate_hydraulic
            end if

            if (self%calculate_mechanical) then
                ! materials配列の添え字として実際の材料IDを使用
                self%mechanical(current_material_id) = input%basic%materials(i)%calculate_mechanical
            end if
        end do

        ! print *,

        ! coupling_modeの設定
        self%coupling_mode = input%basic%analysis_controls%coupling_mode

        ! call self%time%initialize(input)
        ! call self%iteration%initialize(input)

        call initialize_openmp(input)

        call deallocate_array(unique_material_ids)

    end subroutine initialize_type_controls

    ! -----------------------------------------------------------------
    ! <<< ここからが追加した関数 >>>
    ! 指定された物理現象と材料IDが計算対象かどうかを判定する
    ! -----------------------------------------------------------------
    pure function should_calculate_target(self, target_id, i_material) result(is_active)
        implicit none
        class(type_controls), intent(in) :: self
        integer, intent(in) :: target_id
        integer(int32), intent(in) :: i_material
        logical :: is_active

        is_active = .false.

        ! 高速な整数比較
        select case (target_id)
        case (calc_thermal)
#ifdef USE_DEBUG
            if (allocated(self%thermal)) then
                if (i_material <= ubound(self%thermal, 1)) then
#endif
                    is_active = self%thermal(i_material)
#ifdef USE_DEBUG
                end if
            end if
#endif

        case (calc_hydraulic)
#ifdef USE_DEBUG
            if (allocated(self%hydraulic)) then
                if (i_material <= ubound(self%hydraulic, 1)) then
#endif
                    is_active = self%hydraulic(i_material)
#ifdef USE_DEBUG
                end if
            end if
#endif

        case (calc_mechanical)
#ifdef USE_DEBUG
            if (allocated(self%mechanical)) then
                if (i_material <= ubound(self%mechanical, 1)) then
#endif
                    is_active = self%mechanical(i_material)
#ifdef USE_DEBUG
                end if
            end if
#endif
        end select
    end function should_calculate_target

end module module_control