module fortplot_figure_plot_dispatch !! Plot rendering dispatch module !! !! Single Responsibility: Orchestrate rendering of all plots in a figure. !! Individual plot type renderers are in fortplot_figure_plot_renderers. use, intrinsic :: iso_fortran_env, only: wp => real64 use fortplot_context use fortplot_plot_data, only: plot_data_t, & PLOT_TYPE_LINE, & PLOT_TYPE_CONTOUR, & PLOT_TYPE_PCOLORMESH, & PLOT_TYPE_SCATTER, & PLOT_TYPE_FILL, & PLOT_TYPE_BOXPLOT, & PLOT_TYPE_ERRORBAR, & PLOT_TYPE_SURFACE, & PLOT_TYPE_PIE, & PLOT_TYPE_BAR, & PLOT_TYPE_HISTOGRAM, & PLOT_TYPE_REFLINE, & PLOT_TYPE_QUIVER, & PLOT_TYPE_POLAR, & AXIS_PRIMARY, & AXIS_TWINX, & AXIS_TWINY use fortplot_figure_initialization, only: figure_state_t use fortplot_figure_data_ranges, only: collect_figure_data_ranges use fortplot_rendering, only: render_line_plot, render_contour_plot, & render_pcolormesh_plot, render_fill_between_plot, & render_markers, render_boxplot_plot, & render_errorbar_plot, & render_pie_plot, render_bar_plot, & render_histogram_plot use fortplot_surface_rendering, only: render_surface_plot use fortplot_3d_data_rendering, only: render_3d_line_plot, render_3d_markers use fortplot_figure_plot_renderers, only: render_refline_plot, & render_quiver_plot, & render_streamplot_arrows, & render_polar_axes, & render_polar_plot_internal use fortplot_ascii, only: ascii_context use fortplot_ascii_drawing, only: fill_ascii_contour use fortplot_contour_level_calculation, only: compute_default_contour_levels implicit none private public :: render_all_plots contains subroutine render_all_plots(backend, plots, plot_count, & x_min_transformed, x_max_transformed, & y_min_transformed, y_max_transformed, & xscale, yscale, symlog_threshold, & width, height, margin_left, margin_right, & margin_bottom, margin_top, state) !! Render all plots in the figure class(plot_context), intent(inout) :: backend type(plot_data_t), intent(in) :: plots(:) integer, intent(in) :: plot_count real(wp), intent(in) :: x_min_transformed, x_max_transformed real(wp), intent(in) :: y_min_transformed, y_max_transformed character(len=*), intent(in) :: xscale, yscale real(wp), intent(in) :: symlog_threshold integer, intent(in) :: width, height real(wp), intent(in) :: margin_left, margin_right, margin_bottom, margin_top type(figure_state_t), intent(in), optional :: state integer :: i real(wp) :: x_min_curr, x_max_curr, y_min_curr, y_max_curr character(len=10) :: xscale_curr, yscale_curr real(wp) :: primary_x_min, primary_x_max, primary_y_min, primary_y_max real(wp) :: default_line_width, data_extent, raw_bounds(4) logical :: has_data logical :: restore_needed logical :: has_3d real(wp) :: z_min, z_max call resolve_primary_coordinates(state, primary_x_min, primary_x_max, & primary_y_min, primary_y_max, default_line_width) call detect_3d_z_extent(plots, plot_count, has_3d, z_min, z_max) if (.not. has_3d) then z_min = 0.0_wp z_max = 1.0_wp end if do i = 1, plot_count call resolve_plot_coordinates(plots(i), state, primary_x_min, primary_x_max, & primary_y_min, primary_y_max, & x_min_curr, x_max_curr, y_min_curr, y_max_curr, & xscale_curr, yscale_curr, restore_needed) if (restore_needed) then call backend%set_coordinates(x_min_curr, x_max_curr, y_min_curr, & y_max_curr) end if if (plots(i)%line_width > 0.0_wp) then call backend%set_line_width(plots(i)%line_width) else call backend%set_line_width(default_line_width) end if data_extent = 0.0_wp if (plots(i)%plot_type == PLOT_TYPE_QUIVER) then call collect_figure_data_ranges(plots, plot_count, & raw_bounds(1), raw_bounds(2), raw_bounds(3), raw_bounds(4), & has_data, axis_filter=plots(i)%axis) if (has_data) data_extent = maxval(abs(raw_bounds)) end if call dispatch_plot_render(backend, plots(i), & x_min_curr, x_max_curr, y_min_curr, y_max_curr, & xscale_curr, yscale_curr, symlog_threshold, & width, height, margin_left, margin_right, & margin_bottom, margin_top, default_line_width, & z_min, z_max, state, data_extent) if (present(state) .and. restore_needed) then call backend%set_coordinates(primary_x_min, primary_x_max, & primary_y_min, primary_y_max) end if end do if (present(state)) then call backend%set_coordinates(primary_x_min, primary_x_max, primary_y_min, & primary_y_max) end if end subroutine render_all_plots subroutine resolve_primary_coordinates(state, px_min, px_max, py_min, py_max, default_lw) !! Resolve primary coordinate ranges and default line width from figure state type(figure_state_t), intent(in), optional :: state real(wp), intent(out) :: px_min, px_max, py_min, py_max real(wp), intent(out) :: default_lw if (present(state)) then px_min = state%x_min_transformed px_max = state%x_max_transformed py_min = state%y_min_transformed py_max = state%y_max_transformed default_lw = state%current_line_width else px_min = 0.0_wp; px_max = 0.0_wp py_min = 0.0_wp; py_max = 0.0_wp default_lw = 1.5_wp end if end subroutine resolve_primary_coordinates subroutine resolve_plot_coordinates(plot, state, primary_x_min, primary_x_max, & primary_y_min, primary_y_max, & x_min, x_max, y_min, y_max, & xscale, yscale, restore_needed) !! Resolve coordinate ranges for a single plot, handling twin axes type(plot_data_t), intent(in) :: plot type(figure_state_t), intent(in), optional :: state real(wp), intent(in) :: primary_x_min, primary_x_max, primary_y_min, primary_y_max real(wp), intent(out) :: x_min, x_max, y_min, y_max character(len=10), intent(out) :: xscale, yscale logical, intent(out) :: restore_needed if (present(state)) then select case (plot%axis) case (AXIS_TWINX) x_min = state%x_min_transformed x_max = state%x_max_transformed y_min = state%twinx_y_min_transformed y_max = state%twinx_y_max_transformed xscale = state%xscale yscale = state%twinx_yscale restore_needed = .true. case (AXIS_TWINY) x_min = state%twiny_x_min_transformed x_max = state%twiny_x_max_transformed y_min = state%y_min_transformed y_max = state%y_max_transformed xscale = state%twiny_xscale yscale = state%yscale restore_needed = .true. case default x_min = primary_x_min x_max = primary_x_max y_min = primary_y_min y_max = primary_y_max xscale = state%xscale yscale = state%yscale restore_needed = .false. end select else x_min = 0.0_wp; x_max = 0.0_wp y_min = 0.0_wp; y_max = 0.0_wp xscale = ''; yscale = '' restore_needed = .false. end if end subroutine resolve_plot_coordinates subroutine dispatch_plot_render(backend, plot, x_min, x_max, y_min, y_max, & xscale, yscale, symlog_threshold, & width, height, margin_left, margin_right, & margin_bottom, margin_top, default_line_width, & z_min, z_max, state, data_extent) !! Dispatch rendering for a single plot type class(plot_context), intent(inout) :: backend type(plot_data_t), intent(in) :: plot real(wp), intent(in) :: x_min, x_max, y_min, y_max character(len=*), intent(in) :: xscale, yscale real(wp), intent(in) :: symlog_threshold integer, intent(in) :: width, height real(wp), intent(in) :: margin_left, margin_right, margin_bottom, margin_top real(wp), intent(in) :: default_line_width real(wp), intent(in) :: z_min, z_max, data_extent type(figure_state_t), intent(in), optional :: state logical :: has_z has_z = .false. if (allocated(plot%z)) has_z = size(plot%z) > 0 select case (plot%plot_type) case (PLOT_TYPE_LINE) if (has_z) then call render_3d_line_plot(backend, plot, x_min, x_max, y_min, & y_max, z_min, z_max) if (allocated(plot%marker)) then call render_3d_markers(backend, plot, x_min, x_max, y_min, & y_max, z_min, z_max) end if else call set_ascii_stream_mode(backend, plot%is_streamline) call render_line_plot(backend, plot, xscale, yscale, symlog_threshold) call set_ascii_stream_mode(backend, .false.) if (allocated(plot%marker)) then call render_markers(backend, plot, x_min, x_max, y_min, y_max, & xscale, yscale, symlog_threshold) end if end if case (PLOT_TYPE_SCATTER) if (allocated(plot%marker)) then if (has_z) then call render_3d_markers(backend, plot, x_min, x_max, y_min, & y_max, z_min, z_max) else call render_markers(backend, plot, x_min, x_max, y_min, y_max, & xscale, yscale, symlog_threshold) end if end if case (PLOT_TYPE_CONTOUR) ! ASCII filled contours use the ordered glyph ramp (issue #2077) and ! draw no contour lines, matching matplotlib contourf. if (.not. render_ascii_filled_contour(backend, plot)) then call render_contour_plot(backend, plot, x_min, x_max, & y_min, y_max, xscale, yscale, & symlog_threshold, width, height, & margin_left, margin_right, & margin_bottom, margin_top) end if case (PLOT_TYPE_PCOLORMESH) call render_pcolormesh_plot(backend, plot, x_min, x_max, y_min, y_max, & xscale, yscale, symlog_threshold, & width, height, margin_right) case (PLOT_TYPE_SURFACE) call render_surface_plot(backend, plot, x_min, x_max, y_min, y_max, & z_min, z_max, xscale, yscale, symlog_threshold) case (PLOT_TYPE_FILL) call render_fill_between_plot(backend, plot, xscale, yscale, symlog_threshold) case (PLOT_TYPE_BAR) call render_bar_plot(backend, plot, xscale, yscale, symlog_threshold) case (PLOT_TYPE_HISTOGRAM) call render_histogram_plot(backend, plot, xscale, yscale, symlog_threshold) case (PLOT_TYPE_PIE) call render_pie_plot(backend, plot, xscale, yscale, symlog_threshold) case (PLOT_TYPE_BOXPLOT) call render_boxplot_plot(backend, plot, xscale, yscale, symlog_threshold) case (PLOT_TYPE_ERRORBAR) ! matplotlib draws the connecting data line, then error bars, then ! markers (only when a marker is requested). if (allocated(plot%linestyle)) then if (trim(plot%linestyle) /= 'none' .and. & trim(plot%linestyle) /= 'None') then call render_line_plot(backend, plot, xscale, yscale, & symlog_threshold) end if end if call render_errorbar_plot(backend, plot, xscale, yscale, symlog_threshold, & default_line_width, width, height, & margin_left, margin_right, & margin_bottom, margin_top) call render_markers(backend, plot, x_min, x_max, y_min, y_max, & xscale, yscale, symlog_threshold) case (PLOT_TYPE_REFLINE) call render_refline_plot(backend, plot, x_min, x_max, y_min, y_max, & xscale, yscale, symlog_threshold) case (PLOT_TYPE_QUIVER) call render_quiver_plot(backend, plot, x_min, x_max, y_min, y_max, & xscale, yscale, symlog_threshold, data_extent) case (PLOT_TYPE_POLAR) call render_polar_plot_internal(backend, plot, x_min, x_max, y_min, y_max, state) end select end subroutine dispatch_plot_render subroutine set_ascii_stream_mode(backend, enable) !! Toggle thinned streamplot line rendering on the text backend so !! trajectory lines reduce to terminal-cell resolution (issue #2070). !! Non-text backends keep full-resolution trajectory geometry. class(plot_context), intent(inout) :: backend logical, intent(in) :: enable select type (bk => backend) class is (ascii_context) bk%stream_mode = enable end select end subroutine set_ascii_stream_mode subroutine detect_3d_z_extent(plots, plot_count, has_3d, z_min, z_max) type(plot_data_t), intent(in) :: plots(:) integer, intent(in) :: plot_count logical, intent(out) :: has_3d real(wp), intent(out) :: z_min, z_max integer :: i logical :: first has_3d = .false. first = .true. z_min = 0.0_wp z_max = 1.0_wp do i = 1, min(plot_count, size(plots)) if (.not. plots(i)%is_3d()) cycle has_3d = .true. if (allocated(plots(i)%z)) then call include_z_values(plots(i)%z, first, z_min, z_max) end if if (allocated(plots(i)%z_grid)) then call include_z_grid(plots(i)%z_grid, first, z_min, z_max) end if end do end subroutine detect_3d_z_extent subroutine include_z_values(values, first, z_min, z_max) real(wp), contiguous, intent(in) :: values(:) logical, intent(inout) :: first real(wp), intent(inout) :: z_min, z_max if (size(values) <= 0) return if (first) then z_min = minval(values) z_max = maxval(values) first = .false. else z_min = min(z_min, minval(values)) z_max = max(z_max, maxval(values)) end if end subroutine include_z_values logical function render_ascii_filled_contour(backend, plot) result(handled) !! Render a filled contour to the ASCII backend using the ordered glyph !! ramp (issue #2077). Returns .false. for other backends or line !! contours so the caller falls back to the generic renderer. class(plot_context), intent(inout) :: backend type(plot_data_t), intent(in) :: plot real(wp), allocatable :: levels(:) real(wp) :: y_lo, y_hi, z_min, z_max logical :: use_custom handled = .false. if (plot%plot_type /= PLOT_TYPE_CONTOUR) return if (.not. plot%fill_contours) return if (.not. allocated(plot%z_grid)) return if (.not. allocated(plot%x_grid)) return if (.not. allocated(plot%y_grid)) return if (size(plot%z_grid) <= 0) return select type (bk => backend) class is (ascii_context) use_custom = .false. if (allocated(plot%contour_levels)) then if (size(plot%contour_levels) >= 2) use_custom = .true. end if if (use_custom) then levels = plot%contour_levels call sort_levels_ascending(levels) else z_min = minval(plot%z_grid) z_max = maxval(plot%z_grid) call compute_default_contour_levels(z_min, z_max, levels) end if if (bk%has_stored_y_range) then y_lo = bk%stored_y_min y_hi = bk%stored_y_max else y_lo = bk%y_min y_hi = bk%y_max end if call fill_ascii_contour(bk%canvas, plot%x_grid, plot%y_grid, & plot%z_grid, levels, bk%x_min, bk%x_max, & y_lo, y_hi, bk%plot_width, bk%plot_height) handled = .true. class default handled = .false. end select end function render_ascii_filled_contour subroutine sort_levels_ascending(levels) !! Insertion sort; contour level arrays are small. real(wp), intent(inout) :: levels(:) integer :: i, j real(wp) :: key do i = 2, size(levels) key = levels(i) j = i - 1 do while (j >= 1) if (levels(j) <= key) exit levels(j + 1) = levels(j) j = j - 1 end do levels(j + 1) = key end do end subroutine sort_levels_ascending subroutine include_z_grid(values, first, z_min, z_max) real(wp), contiguous, intent(in) :: values(:, :) logical, intent(inout) :: first real(wp), intent(inout) :: z_min, z_max if (size(values) <= 0) return if (first) then z_min = minval(values) z_max = maxval(values) first = .false. else z_min = min(z_min, minval(values)) z_max = max(z_max, maxval(values)) end if end subroutine include_z_grid end module fortplot_figure_plot_dispatch