bite-gp-authoring 1.25.202

#[cfg(any(
    target_os = "linux",
    target_os = "macos",
    target_os = "ios",
    target_os = "windows"
))]
use crate::GpuCanvasContext;
use crate::{
    App, Bounds, Element, ElementId, GlobalElementId, IntoElement, LayoutId, ObjectFit, Pixels,
    Style, StyleRefinement, Styled, Window,
};
// The payload lives in the engine beside `PaintSurface`, the primitive it becomes: `surface()`
// takes one and `draw_surfaces` reads it, and neither the element nor a renderer invents a
// transport of its own.
#[cfg(target_os = "windows")]
use gpui_engine::DirectXSource;
use gpui_engine::SurfaceSource;
#[cfg(any(target_os = "macos", target_os = "ios"))]
use objc2_core_video::{CVPixelBufferGetHeight, CVPixelBufferGetWidth};
use refineable::Refineable;

/// A surface element: an external pixel source composited into the scene.
///
/// Built by [`surface`] and configured with [`Surface::object_fit`]. It holds the [`SurfaceSource`]
/// it was given until the window's renderer resolves that source into something it can sample; it
/// carries no pixel data itself. See [`surface`] for what a surface is and where a source comes
/// from.
pub struct Surface {
    source: SurfaceSource,
    object_fit: ObjectFit,
    style: StyleRefinement,
}

/// Composites pixels GPUI did not draw into the scene, as an ordinary element.
///
/// A surface is how an application shows content produced outside GPUI — a video decoder's frame, a
/// texture another graphics API rendered, or a resource that lives on another device — without
/// copying those pixels through the CPU. The producer fills a GPU resource and hands it here as a
/// source; the window's renderer samples that resource directly while it composites the frame.
///
/// The element lays out and stacks like any other child: it occupies space in the layout tree, and
/// it is painted in the frame's own pass in child order, so ordinary elements drawn beside or over
/// it compose with it rather than floating above it.
///
/// `source` is anything that converts into a [`SurfaceSource`]. On Windows that is a Direct3D
/// texture or shader-resource view, on macOS a CoreVideo image buffer, and on Linux a dma-buf
/// handle; [`SurfaceSource`] documents each platform's payloads and, where a platform offers more
/// than one, which to reach for. The Windows pair is the one that comes up most: handing over a
/// texture is the ergonomic default, since the renderer makes the view on the device it owns, while
/// handing over a view is the escape for a producer that is the authority on its own format and mip
/// interpretation.
///
/// A source is produced by someone else, before this element paints. A same-device resource is made
/// inside a [`gpu_canvas`](crate::gpu_canvas) callback, where the window renderer's device is
/// reachable through [`GpuCanvasContext::device`](crate::GpuCanvasContext::device); a source this
/// element takes was made off the window's device and shared with it — a dma-buf, a CoreVideo
/// buffer, a shared Direct3D view — the route a cross-process bridge takes.
///
/// ```rust
/// use gpui::{div, surface};
///
/// // `source` is a `SurfaceSource` for this platform; see the type's docs.
/// div().size_full().child(surface(source))
/// ```
///
/// For a working producer, run `cargo run -p gpui --example surface`: it shows this element and the
/// `gpu_canvas()` callback side by side, one payload at a time, on the host renderer.
pub fn surface(source: impl Into<SurfaceSource>) -> Surface {
    Surface {
        source: source.into(),
        object_fit: ObjectFit::Contain,
        style: Default::default(),
    }
}

impl Surface {
    /// Sets how the source's pixels are fitted into this element's bounds.
    ///
    /// The source carries its own pixel size; the element carries its layout size. The fit decides
    /// how the former maps onto the latter, using the same [`ObjectFit`] modes images use — for
    /// example [`ObjectFit::Contain`] (the default) letterboxes the source inside the bounds while
    /// preserving its aspect ratio, and [`ObjectFit::Fill`] stretches it to fill the bounds
    /// exactly. When the source's size cannot be determined, the pixels are drawn into the full
    /// bounds.
    pub fn object_fit(mut self, object_fit: ObjectFit) -> Self {
        self.object_fit = object_fit;
        self
    }
}

impl Element for Surface {
    type RequestLayoutState = ();
    type PrepaintState = ();

    fn id(&self) -> Option<ElementId> {
        None
    }

    fn source_location(&self) -> Option<&'static core::panic::Location<'static>> {
        None
    }

    fn request_layout(
        &mut self,
        _global_id: Option<&GlobalElementId>,
        window: &mut Window,
        cx: &mut App,
    ) -> (LayoutId, Self::RequestLayoutState) {
        let mut style = Style::default();
        style.refine(&self.style);
        let layout_id = window.request_layout(style, [], cx);
        (layout_id, ())
    }

    fn prepaint(
        &mut self,
        _global_id: Option<&GlobalElementId>,
        _bounds: Bounds<Pixels>,
        _request_layout: &mut Self::RequestLayoutState,
        _window: &mut Window,
        _cx: &mut App,
    ) -> Self::PrepaintState {
    }

    fn paint(
        &mut self,
        _global_id: Option<&GlobalElementId>,
        #[cfg_attr(
            not(any(target_os = "macos", target_os = "ios", target_os = "windows")),
            allow(unused_variables)
        )]
        bounds: Bounds<Pixels>,
        _: &mut Self::RequestLayoutState,
        _: &mut Self::PrepaintState,
        #[cfg_attr(
            not(any(target_os = "macos", target_os = "ios", target_os = "windows")),
            allow(unused_variables)
        )]
        window: &mut Window,
        #[cfg_attr(
            not(any(target_os = "macos", target_os = "ios", target_os = "windows")),
            allow(unused_variables)
        )]
        cx: &mut App,
    ) {
        match &self.source {
            #[cfg(any(target_os = "macos", target_os = "ios"))]
            SurfaceSource::CoreVideo(image_buffer) => {
                let size = crate::size(
                    CVPixelBufferGetWidth(image_buffer).into(),
                    CVPixelBufferGetHeight(image_buffer).into(),
                );
                let new_bounds = self.object_fit.get_bounds(bounds, size);
                // TODO: Add support for corner_radii
                let mut ctx = GpuCanvasContext::new(window, cx, new_bounds);
                ctx.paint_surface(image_buffer.clone());
            }
            #[cfg(target_os = "windows")]
            SurfaceSource::DirectX(source) => {
                let new_bounds = match directx_source_size(source) {
                    Some(size) => self.object_fit.get_bounds(bounds, size),
                    None => bounds,
                };
                // TODO: Add support for corner_radii
                let mut ctx = GpuCanvasContext::new(window, cx, new_bounds);
                ctx.paint_surface(SurfaceSource::DirectX(source.clone()));
            }
            #[cfg(target_os = "linux")]
            SurfaceSource::DmaBuf(handle) => {
                let size = crate::size(
                    crate::DevicePixels::from(handle.width as i32),
                    crate::DevicePixels::from(handle.height as i32),
                );
                let new_bounds = self.object_fit.get_bounds(bounds, size);
                // TODO: Add support for corner_radii
                let mut ctx = GpuCanvasContext::new(window, cx, new_bounds);
                ctx.paint_surface(handle.clone());
            }
            #[allow(unreachable_patterns)]
            _ => {}
        }
    }
}

impl IntoElement for Surface {
    type Element = Self;

    fn into_element(self) -> Self::Element {
        self
    }
}

impl Styled for Surface {
    fn style(&mut self) -> &mut StyleRefinement {
        &mut self.style
    }
}

/// The pixel size of the texture behind a Direct3D surface source.
///
/// The Windows backend fits the surface to its bounds exactly as the macOS backend does, and neither
/// a texture nor a view carries a size of its own: the texture is the resource, and the view's size is
/// the resource behind it. A source whose resource is not a texture, or whose query fails, has no
/// size to fit, and the element falls back to its bounds.
#[cfg(target_os = "windows")]
fn directx_source_size(source: &DirectXSource) -> Option<crate::Size<crate::DevicePixels>> {
    use windows::Win32::Graphics::Direct3D11::{D3D11_TEXTURE2D_DESC, ID3D11Texture2D};
    use windows::core::Interface as _;

    unsafe {
        let mut desc = D3D11_TEXTURE2D_DESC::default();
        match source {
            DirectXSource::Texture(texture) => texture.GetDesc(&mut desc),
            DirectXSource::View(view) => {
                let resource = view.GetResource().ok()?;
                let texture: ID3D11Texture2D = resource.cast().ok()?;
                texture.GetDesc(&mut desc);
            }
            // A shared texture carries its size, because the renderer has not opened it yet.
            DirectXSource::Shared(shared) => {
                return Some(crate::size(
                    crate::DevicePixels::from(shared.width as i32),
                    crate::DevicePixels::from(shared.height as i32),
                ));
            }
        }
        Some(crate::size(
            crate::DevicePixels::from(desc.Width as i32),
            crate::DevicePixels::from(desc.Height as i32),
        ))
    }
}

#[cfg(all(test, target_os = "linux"))]
mod tests {
    use super::*;
    use crate::{Context, Render, TestAppContext, Window};
    use gpui_engine::{DmaBufHandle, DmaBufPlane, SurfaceFormatKind, SurfaceSource};
    use std::os::fd::OwnedFd;

    struct SurfaceView {
        handle: DmaBufHandle,
    }

    impl Render for SurfaceView {
        fn render(&mut self, _window: &mut Window, _cx: &mut Context<Self>) -> impl IntoElement {
            surface(self.handle.clone())
        }
    }

    /// A descriptor without a GPU: any open file is a valid one as far as the element cares.
    fn descriptor() -> OwnedFd {
        std::fs::File::open("/dev/null").expect("/dev/null").into()
    }

    /// The element's whole job is to put the handle into the scene as a surface: it names no
    /// renderer, so this needs no GPU.
    #[gpui::test]
    fn a_surface_handle_reaches_the_scene(cx: &mut TestAppContext) {
        let handle = DmaBufHandle::new(
            4,
            4,
            SurfaceFormatKind::rgba8(),
            DmaBufHandle::LINEAR,
            [DmaBufPlane::new(descriptor(), 0, 16)],
            None,
        );

        let expected = handle.clone();
        // `add_window_view` draws the frame the way the platform does; drawing by hand from inside
        // a `window.update` would re-borrow the view that closure already holds.
        let (_view, cx) = cx.add_window_view(|_, _| SurfaceView { handle });
        cx.run_until_parked();

        let surfaces = cx.update(|window, _cx| window.painted_surfaces());
        assert_eq!(
            surfaces.len(),
            1,
            "the surface should be in the scene exactly once"
        );
        assert_eq!(surfaces[0].source, SurfaceSource::DmaBuf(expected));
    }
}