rendy-chain 0.5.1

use std::{
    fmt::Debug,
    ops::{BitOr, BitOrAssign},
};

/// Trait to abstract of specific access flags.
pub trait AccessFlags: Copy + Debug + BitOr<Output = Self> + BitOrAssign + 'static {
    /// Get flags value with no flags set.
    fn empty() -> Self;

    /// Check if this access must be exclusive.
    ///
    /// Basically this checks if all flags are known read flags.
    fn exclusive(&self) -> bool;
}

impl AccessFlags for rendy_core::hal::buffer::Access {
    #[inline]
    fn empty() -> Self {
        Self::empty()
    }

    #[inline]
    fn exclusive(&self) -> bool {
        self.intersects(
            Self::SHADER_WRITE | Self::TRANSFER_WRITE | Self::HOST_WRITE | Self::MEMORY_WRITE,
        )
    }
}

impl AccessFlags for rendy_core::hal::image::Access {
    #[inline]
    fn empty() -> Self {
        Self::empty()
    }

    #[inline]
    fn exclusive(&self) -> bool {
        self.intersects(
            Self::SHADER_WRITE
                | Self::COLOR_ATTACHMENT_WRITE
                | Self::DEPTH_STENCIL_ATTACHMENT_WRITE
                | Self::TRANSFER_WRITE
                | Self::HOST_WRITE
                | Self::MEMORY_WRITE,
        )
    }
}

/// Trait to abstract of specific usage flags.
pub trait UsageFlags: Copy + Debug + BitOr<Output = Self> + BitOrAssign + 'static {}

impl UsageFlags for rendy_core::hal::buffer::Usage {}
impl UsageFlags for rendy_core::hal::image::Usage {}

/// Abstracts resource types that uses different usage flags and layouts types.
pub trait Resource: 'static {
    /// Access flags for resource type.
    type Access: AccessFlags;

    /// Usage flags type for the resource.
    type Usage: UsageFlags;

    /// Layout type for the resource.
    type Layout: Copy + Debug + 'static;

    /// Empty usage.
    fn no_usage() -> Self::Usage;

    /// Layout suitable for specified accesses.
    fn layout_for(access: Self::Access) -> Self::Layout;
}

/// Buffer resource type.
#[derive(Clone, Copy, Debug, PartialEq, Eq, PartialOrd, Ord, Hash, Default)]
pub struct Buffer;

impl Resource for Buffer {
    type Access = rendy_core::hal::buffer::Access;
    type Usage = rendy_core::hal::buffer::Usage;
    type Layout = ();

    fn no_usage() -> Self::Usage {
        rendy_core::hal::buffer::Usage::empty()
    }

    fn layout_for(_access: rendy_core::hal::buffer::Access) {}
}

/// Image resource type.
#[derive(Clone, Copy, Debug, PartialEq, Eq, PartialOrd, Ord, Hash, Default)]
pub struct Image;

impl Resource for Image {
    type Access = rendy_core::hal::image::Access;

    type Usage = rendy_core::hal::image::Usage;

    type Layout = rendy_core::hal::image::Layout;

    fn no_usage() -> Self::Usage {
        rendy_core::hal::image::Usage::empty()
    }

    fn layout_for(access: rendy_core::hal::image::Access) -> rendy_core::hal::image::Layout {
        let mut acc = None;
        if access.contains(rendy_core::hal::image::Access::INPUT_ATTACHMENT_READ) {
            acc = Some(common_layout(
                acc,
                rendy_core::hal::image::Layout::ShaderReadOnlyOptimal,
            ));
        }
        if access.contains(rendy_core::hal::image::Access::COLOR_ATTACHMENT_READ) {
            acc = Some(common_layout(
                acc,
                rendy_core::hal::image::Layout::ColorAttachmentOptimal,
            ));
        }
        if access.contains(rendy_core::hal::image::Access::COLOR_ATTACHMENT_WRITE) {
            acc = Some(common_layout(
                acc,
                rendy_core::hal::image::Layout::ColorAttachmentOptimal,
            ));
        }
        if access.contains(rendy_core::hal::image::Access::DEPTH_STENCIL_ATTACHMENT_READ) {
            acc = Some(common_layout(
                acc,
                rendy_core::hal::image::Layout::DepthStencilReadOnlyOptimal,
            ));
        }
        if access.contains(rendy_core::hal::image::Access::DEPTH_STENCIL_ATTACHMENT_WRITE) {
            acc = Some(common_layout(
                acc,
                rendy_core::hal::image::Layout::DepthStencilAttachmentOptimal,
            ));
        }
        if access.contains(rendy_core::hal::image::Access::TRANSFER_READ) {
            acc = Some(common_layout(
                acc,
                rendy_core::hal::image::Layout::TransferSrcOptimal,
            ));
        }
        if access.contains(rendy_core::hal::image::Access::TRANSFER_WRITE) {
            acc = Some(common_layout(
                acc,
                rendy_core::hal::image::Layout::TransferDstOptimal,
            ));
        }
        acc.unwrap_or(rendy_core::hal::image::Layout::General)
    }
}

fn common_layout(
    acc: Option<rendy_core::hal::image::Layout>,
    layout: rendy_core::hal::image::Layout,
) -> rendy_core::hal::image::Layout {
    match (acc, layout) {
        (None, layout) => layout,
        (Some(left), right) if left == right => left,
        (
            Some(rendy_core::hal::image::Layout::DepthStencilReadOnlyOptimal),
            rendy_core::hal::image::Layout::DepthStencilAttachmentOptimal,
        ) => rendy_core::hal::image::Layout::DepthStencilAttachmentOptimal,
        (
            Some(rendy_core::hal::image::Layout::DepthStencilAttachmentOptimal),
            rendy_core::hal::image::Layout::DepthStencilReadOnlyOptimal,
        ) => rendy_core::hal::image::Layout::DepthStencilAttachmentOptimal,
        (Some(_), _) => rendy_core::hal::image::Layout::General,
    }
}