fortplot_figure_core_ticks.f90 Source File

submodule (fortplot_figure_core) fortplot_figure_core_ticks
    !! Custom tick label operations for figure_t
    !!
    !! Single Responsibility: Handle custom tick positions and labels
    !! Extracted to maintain file size compliance

    use fortplot_logging, only: log_error, log_warning
    implicit none

contains

    subroutine set_ticks_impl(self, axis, positions, labels)
        !! Shared implementation for set_xticks and set_yticks
        class(figure_t), intent(inout) :: self
        integer, intent(in) :: axis
        real(wp), contiguous, intent(in) :: positions(:)
        character(len=*), intent(in), optional :: labels(:)

        integer :: n, i

        n = size(positions)
        if (n == 0) then
            select case (axis)
            case (1)
                call log_warning('set_xticks: empty positions array')
            case (2)
                call log_warning('set_yticks: empty positions array')
            end select
            return
        end if

        if (present(labels)) then
            if (size(labels) /= n) then
                select case (axis)
                case (1)
                    call log_error('set_xticks: labels array size must match positions')
                case (2)
                    call log_error('set_yticks: labels array size must match positions')
                end select
                return
            end if
        end if

        select case (axis)
        case (1)
            self%state%custom_xtick_positions = positions
            allocate(self%state%custom_xtick_labels(n))

            if (present(labels)) then
                do i = 1, n
                    self%state%custom_xtick_labels(i) = trim(labels(i))
                end do
            else
                call auto_gen_labels(self%state%custom_xtick_labels, positions, n)
            end if

            self%state%custom_xticks_set = .true.
        case (2)
            self%state%custom_ytick_positions = positions
            allocate(self%state%custom_ytick_labels(n))

            if (present(labels)) then
                do i = 1, n
                    self%state%custom_ytick_labels(i) = trim(labels(i))
                end do
            else
                call auto_gen_labels(self%state%custom_ytick_labels, positions, n)
            end if

            self%state%custom_yticks_set = .true.
        end select

        self%state%rendered = .false.
    end subroutine set_ticks_impl

    subroutine set_tick_labels_impl(self, axis, labels)
        !! Shared implementation for set_xtick_labels and set_ytick_labels
        class(figure_t), intent(inout) :: self
        integer, intent(in) :: axis
        character(len=*), intent(in) :: labels(:)

        integer :: n, i

        n = size(labels)
        if (n == 0) then
            select case (axis)
            case (1)
                call log_warning('set_xtick_labels: empty labels array')
            case (2)
                call log_warning('set_ytick_labels: empty labels array')
            end select
            return
        end if

        select case (axis)
        case (1)
            allocate(self%state%custom_xtick_labels(n))
            do i = 1, n
                self%state%custom_xtick_labels(i) = trim(labels(i))
            end do

            if (.not. allocated(self%state%custom_xtick_positions)) then
                allocate(self%state%custom_xtick_positions(n))
                do i = 1, n
                    self%state%custom_xtick_positions(i) = real(i, wp)
                end do
            end if

            self%state%custom_xticks_set = .true.
        case (2)
            allocate(self%state%custom_ytick_labels(n))
            do i = 1, n
                self%state%custom_ytick_labels(i) = trim(labels(i))
            end do

            if (.not. allocated(self%state%custom_ytick_positions)) then
                allocate(self%state%custom_ytick_positions(n))
                do i = 1, n
                    self%state%custom_ytick_positions(i) = real(i, wp)
                end do
            end if

            self%state%custom_yticks_set = .true.
        end select

        self%state%rendered = .false.
    end subroutine set_tick_labels_impl

    subroutine auto_gen_labels(labels_out, positions, n)
        !! Generate numeric labels from tick positions
        character(len=*), intent(out) :: labels_out(:)
        real(wp), contiguous, intent(in) :: positions(:)
        integer, intent(in) :: n

        integer :: i

        do i = 1, n
            write(labels_out(i), '(G12.5)') positions(i)
            labels_out(i) = adjustl(labels_out(i))
        end do
    end subroutine auto_gen_labels

    module subroutine set_xticks(self, positions, labels)
        !! Set custom x-axis tick positions and optionally labels
        !! If only positions provided, numeric labels are auto-generated
        !! If both provided, custom string labels are used
        class(figure_t), intent(inout) :: self
        real(wp), contiguous, intent(in) :: positions(:)
        character(len=*), intent(in), optional :: labels(:)

        call set_ticks_impl(self, 1, positions, labels)
    end subroutine set_xticks

    module subroutine set_yticks(self, positions, labels)
        !! Set custom y-axis tick positions and optionally labels
        class(figure_t), intent(inout) :: self
        real(wp), contiguous, intent(in) :: positions(:)
        character(len=*), intent(in), optional :: labels(:)

        call set_ticks_impl(self, 2, positions, labels)
    end subroutine set_yticks

    module subroutine set_xtick_labels(self, labels)
        !! Set custom x-axis tick labels for current tick positions
        !! Use after bar() or set_xticks() to provide categorical labels
        class(figure_t), intent(inout) :: self
        character(len=*), intent(in) :: labels(:)

        call set_tick_labels_impl(self, 1, labels)
    end subroutine set_xtick_labels

    module subroutine set_ytick_labels(self, labels)
        !! Set custom y-axis tick labels for current tick positions
        class(figure_t), intent(inout) :: self
        character(len=*), intent(in) :: labels(:)

        call set_tick_labels_impl(self, 2, labels)
    end subroutine set_ytick_labels

end submodule fortplot_figure_core_ticks