dear-imgui-wgpu 0.16.0

#[cfg(feature = "mv-log")]
use std::sync::{Mutex, OnceLock};

use super::{
    RendererRenderStateGuard, WgpuRenderer, draw::FramebufferExtent,
    map_renderer_render_state_error,
};
use crate::wgpu;
use crate::{GammaMode, RendererError, RendererResult, Uniforms};
use dear_imgui_rs::{
    ContextBinding,
    render::{DrawData, PendingFrame, ReconciledFrame},
    sys,
};
use wgpu::RenderPass;

#[allow(unused_macros)]
macro_rules! mvlog {
    ($($arg:tt)*) => {
        if cfg!(feature = "mv-log") { eprintln!($($arg)*); }
    }
}

fn with_bound_context<R>(
    binding: &ContextBinding,
    callback: impl FnOnce() -> RendererResult<R>,
) -> RendererResult<R> {
    binding
        .try_with_bound_context(callback)
        .map_err(|_| RendererError::ContextDropped)?
}

fn platform_io_for_current_context() -> RendererResult<*mut sys::ImGuiPlatformIO> {
    let context = unsafe { sys::igGetCurrentContext() };
    let platform_io = unsafe { sys::igGetPlatformIO_ContextPtr(context) };
    if platform_io.is_null() {
        Err(RendererError::InvalidRenderState(
            "bound Dear ImGui context has no PlatformIO".to_owned(),
        ))
    } else {
        Ok(platform_io)
    }
}

impl WgpuRenderer {
    fn ensure_frame_prepared(&self) -> RendererResult<()> {
        let native_frame_count = unsafe { sys::igGetFrameCount() };
        match self.backend_data.as_ref() {
            Some(backend_data)
                if backend_data
                    .frame_cursor
                    .is_native_frame(native_frame_count) =>
            {
                Ok(())
            }
            Some(_) => Err(RendererError::FrameNotPrepared),
            None => Err(RendererError::InvalidRenderState(
                "Renderer not initialized".to_owned(),
            )),
        }
    }

    fn prepare_frame_bound(&mut self, frame: &PendingFrame<'_>) -> RendererResult<()> {
        let epoch = frame.epoch();
        self.prepare_frame_epoch(epoch.sequence(), unsafe { sys::igGetFrameCount() })
    }

    /// Renders one Context-borrowed Dear ImGui frame into an explicit framebuffer extent.
    ///
    /// Managed texture requests are reconciled before draw commands resolve
    /// renderer texture IDs. WGPU render passes do not expose their attachment dimensions, so
    /// callers must pass the physical width and height of the target being rendered.
    pub fn render(
        &mut self,
        frame: PendingFrame<'_>,
        render_pass: &mut RenderPass<'_>,
        framebuffer_extent: FramebufferExtent,
    ) -> RendererResult<()> {
        let frame = self.reconcile_frame(frame)?;
        self.render_reconciled(frame, render_pass, framebuffer_extent)
    }

    /// Renders an already reconciled frame into an explicit framebuffer extent.
    ///
    /// Recording consumes the reconciled capability. Command submission and presentation remain
    /// the application's responsibility.
    pub fn render_reconciled(
        &mut self,
        frame: ReconciledFrame<'_>,
        render_pass: &mut RenderPass<'_>,
        framebuffer_extent: FramebufferExtent,
    ) -> RendererResult<()> {
        self.ensure_renderer_contract()?;
        self.ensure_reconciled_frame_matches(&frame)?;
        let binding = self.bound_context()?;
        with_bound_context(&binding, || {
            let platform_io = platform_io_for_current_context()?;
            self.render_read_only_draw_data_with_fb_size(
                frame.draw_data(),
                render_pass,
                framebuffer_extent.width(),
                framebuffer_extent.height(),
                true,
                platform_io,
            )
        })
    }

    /// Applies managed-texture requests without drawing or acquiring a surface.
    ///
    /// Callback capability is checked before texture reconciliation, so an
    /// unsupported callback-bearing frame is not consumed partially.
    pub fn reconcile_frame<'frame>(
        &mut self,
        frame: PendingFrame<'frame>,
    ) -> RendererResult<ReconciledFrame<'frame>> {
        self.ensure_renderer_contract()?;
        self.ensure_pending_frame_matches(&frame)?;
        let binding = self.bound_context()?;
        with_bound_context(&binding, || {
            Self::preflight_draw_callback_support(frame.draw_requirements())?;
            self.prepare_frame_bound(&frame)?;
            self.reconcile_frame_bound(frame)
        })
    }

    fn reconcile_frame_bound<'frame>(
        &mut self,
        frame: PendingFrame<'frame>,
    ) -> RendererResult<ReconciledFrame<'frame>> {
        let request_epoch = frame.epoch().sequence();
        let backend_data = self.backend_data.as_mut().ok_or_else(|| {
            RendererError::InvalidRenderState("Renderer not initialized".to_owned())
        })?;
        let feedback = self.texture_manager.handle_texture_requests(
            frame.texture_requests(),
            request_epoch,
            &backend_data.device,
            &backend_data.queue,
            &mut backend_data.render_resources,
        )?;
        let frame = frame.reconcile_texture_feedback(feedback)?;
        self.texture_manager
            .prune_destroyed_managed_textures(frame.completion_progress().watermark());
        Ok(frame)
    }

    /// Internal explicit-size variant used by renderer-owned secondary viewports.
    pub(super) fn render_read_only_draw_data_with_fb_size(
        &mut self,
        draw_data: &DrawData,
        render_pass: &mut RenderPass<'_>,
        fb_width: u32,
        fb_height: u32,
        main_viewport: bool,
        platform_io: *mut sys::ImGuiPlatformIO,
    ) -> RendererResult<()> {
        self.ensure_frame_prepared()?;
        self.log_framebuffer_mismatch(draw_data, fb_width, fb_height, main_viewport);
        let extent = FramebufferExtent::new(fb_width, fb_height);
        if extent.is_empty() {
            return Ok(());
        }
        self.render_draw_data_at_extent(draw_data, render_pass, extent, platform_io)
    }

    fn render_draw_data_at_extent(
        &mut self,
        draw_data: &DrawData,
        render_pass: &mut RenderPass<'_>,
        extent: FramebufferExtent,
        platform_io: *mut sys::ImGuiPlatformIO,
    ) -> RendererResult<()> {
        self.ensure_renderer_contract()?;
        if !draw_data.valid() {
            return Ok(());
        }
        Self::preflight_draw_callback_support(draw_data.requirements())?;
        unsafe {
            RendererRenderStateGuard::<crate::WgpuRenderStateStorage>::preflight(platform_io)
        }
        .map_err(map_renderer_render_state_error)?;

        let backend_data = self.backend_data.as_mut().ok_or_else(|| {
            RendererError::InvalidRenderState("Renderer not initialized".to_owned())
        })?;
        let prepared = Self::prepare_draw_data(
            &self.texture_manager,
            &self.default_texture,
            draw_data,
            extent,
            backend_data,
        )?;
        if prepared.is_empty() {
            return Ok(());
        }

        let gamma = match self.gamma_mode {
            GammaMode::Auto => Uniforms::gamma_for_format(backend_data.render_target_format),
            GammaMode::Linear => 1.0,
            GammaMode::Gamma22 => 2.2,
        };
        let state = Self::prepare_render_state_static(
            draw_data,
            backend_data,
            gamma,
            prepared.has_elements(),
        )?;
        let device = backend_data.device.clone();
        Self::execute_prepared_draw_data(
            prepared,
            &state,
            extent,
            render_pass,
            platform_io,
            &device,
        )
    }

    #[cfg(feature = "mv-log")]
    fn log_framebuffer_mismatch(
        &self,
        draw_data: &DrawData,
        fb_width: u32,
        fb_height: u32,
        main_viewport: bool,
    ) {
        static LAST_MISMATCH: OnceLock<Mutex<Option<(u32, u32, u32, u32, bool)>>> = OnceLock::new();
        let expected_width = (draw_data.display_size()[0] * draw_data.framebuffer_scale()[0])
            .round()
            .max(0.0) as u32;
        let expected_height = (draw_data.display_size()[1] * draw_data.framebuffer_scale()[1])
            .round()
            .max(0.0) as u32;
        if expected_width == fb_width && expected_height == fb_height {
            return;
        }
        let key = (
            expected_width,
            expected_height,
            fb_width,
            fb_height,
            main_viewport,
        );
        let mut previous = LAST_MISMATCH
            .get_or_init(|| Mutex::new(None))
            .lock()
            .unwrap();
        if *previous != Some(key) {
            mvlog!(
                "[wgpu-mv] fb mismatch expected=({}, {}) override=({}, {}) disp=({:.1},{:.1}) fb_scale=({:.2},{:.2}) main={}",
                expected_width,
                expected_height,
                fb_width,
                fb_height,
                draw_data.display_size()[0],
                draw_data.display_size()[1],
                draw_data.framebuffer_scale()[0],
                draw_data.framebuffer_scale()[1],
                main_viewport
            );
            *previous = Some(key);
        }
    }

    #[cfg(not(feature = "mv-log"))]
    fn log_framebuffer_mismatch(
        &self,
        _draw_data: &DrawData,
        _fb_width: u32,
        _fb_height: u32,
        _main_viewport: bool,
    ) {
    }
}