monolith

use crate::image::{ImageInner, color_range};

use ash::vk;

/// A full barrier between all earlier and later work on the image, optionally

/// transferring ownership between queue families. Coarse, but correct for

/// every transition the renderer makes.

pub(crate) unsafe fn barrier(

    d: &ash::Device,

    cmd: vk::CommandBuffer,

    image: vk::Image,

    old: vk::ImageLayout,

    new: vk::ImageLayout,

    src_family: u32,

    dst_family: u32,

) {

    let b = [vk::ImageMemoryBarrier2::default()

        .src_stage_mask(vk::PipelineStageFlags2::ALL_COMMANDS)

        .src_access_mask(vk::AccessFlags2::MEMORY_WRITE)

        .dst_stage_mask(vk::PipelineStageFlags2::ALL_COMMANDS)

        .dst_access_mask(vk::AccessFlags2::MEMORY_READ | vk::AccessFlags2::MEMORY_WRITE)

        .old_layout(old)

        .new_layout(new)

        .src_queue_family_index(src_family)

        .dst_queue_family_index(dst_family)

        .image(image)

        .subresource_range(color_range())];

    unsafe {

        d.cmd_pipeline_barrier2(

            cmd,

            &vk::DependencyInfo::default().image_memory_barriers(&b),

        )

    };

}

/// Moves an image into `layout` for use on this queue. Dmabuf images come

/// from the foreign queue family in the general layout, others continue from

/// their tracked layout.

pub(crate) unsafe fn acquire(

    d: &ash::Device,

    cmd: vk::CommandBuffer,

    family: u32,

    image: &ImageInner,

    layout: vk::ImageLayout,

) {

    if image.external {

        unsafe {

            barrier(

                d,

                cmd,

                image.image,

                vk::ImageLayout::GENERAL,

                layout,

                vk::QUEUE_FAMILY_FOREIGN_EXT,

                family,

            )

        };

    } else {

        let old = image.layout();

        if old != layout {

            unsafe {

                barrier(

                    d,

                    cmd,

                    image.image,

                    old,

                    layout,

                    vk::QUEUE_FAMILY_IGNORED,

                    vk::QUEUE_FAMILY_IGNORED,

                )

            };

            image.set_layout(layout);

        }

    }

}

/// Hands a dmabuf image back to the foreign queue family after use.

pub(crate) unsafe fn release(

    d: &ash::Device,

    cmd: vk::CommandBuffer,

    family: u32,

    image: &ImageInner,

    layout: vk::ImageLayout,

) {

    if image.external {

        unsafe {

            barrier(

                d,

                cmd,

                image.image,

                layout,

                vk::ImageLayout::GENERAL,

                family,

                vk::QUEUE_FAMILY_FOREIGN_EXT,

            )

        };

    }

}