2023-02-06 08:17:23 +11:00
|
|
|
use crate::descriptor_heap::{CpuStagingHeap, D3D12DescriptorHeapSlot, RenderTargetHeap};
|
|
|
|
use librashader_common::{FilterMode, ImageFormat, Size, WrapMode};
|
2023-02-02 10:09:34 +11:00
|
|
|
use std::ops::Deref;
|
2023-02-06 10:03:38 +11:00
|
|
|
use windows::Win32::Graphics::Direct3D12::{ID3D12Resource, D3D12_CPU_DESCRIPTOR_HANDLE};
|
2023-02-01 17:25:39 +11:00
|
|
|
use windows::Win32::Graphics::Dxgi::Common::DXGI_FORMAT;
|
2023-02-01 09:50:47 +11:00
|
|
|
|
2023-02-02 10:09:34 +11:00
|
|
|
#[derive(Clone)]
|
2023-02-01 11:23:57 +11:00
|
|
|
pub(crate) enum InputDescriptor {
|
2023-02-01 09:50:47 +11:00
|
|
|
Owned(D3D12DescriptorHeapSlot<CpuStagingHeap>),
|
2023-02-06 08:17:23 +11:00
|
|
|
Raw(D3D12_CPU_DESCRIPTOR_HANDLE),
|
2023-02-01 09:50:47 +11:00
|
|
|
}
|
|
|
|
|
2023-02-02 10:09:34 +11:00
|
|
|
#[derive(Clone)]
|
2023-02-01 11:23:57 +11:00
|
|
|
pub(crate) enum OutputDescriptor {
|
2023-02-01 16:16:06 +11:00
|
|
|
Owned(D3D12DescriptorHeapSlot<RenderTargetHeap>),
|
2023-02-06 08:17:23 +11:00
|
|
|
Raw(D3D12_CPU_DESCRIPTOR_HANDLE),
|
2023-02-01 09:50:47 +11:00
|
|
|
}
|
|
|
|
|
|
|
|
impl AsRef<D3D12_CPU_DESCRIPTOR_HANDLE> for InputDescriptor {
|
|
|
|
fn as_ref(&self) -> &D3D12_CPU_DESCRIPTOR_HANDLE {
|
|
|
|
match self {
|
2023-02-02 10:09:34 +11:00
|
|
|
InputDescriptor::Owned(h) => h.deref().as_ref(),
|
2023-02-06 08:17:23 +11:00
|
|
|
InputDescriptor::Raw(h) => h,
|
2023-02-01 09:50:47 +11:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
impl AsRef<D3D12_CPU_DESCRIPTOR_HANDLE> for OutputDescriptor {
|
|
|
|
fn as_ref(&self) -> &D3D12_CPU_DESCRIPTOR_HANDLE {
|
|
|
|
match self {
|
2023-02-02 10:09:34 +11:00
|
|
|
OutputDescriptor::Owned(h) => h.deref().as_ref(),
|
2023-02-06 08:17:23 +11:00
|
|
|
OutputDescriptor::Raw(h) => h,
|
2023-02-01 09:50:47 +11:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2023-02-02 10:09:34 +11:00
|
|
|
#[derive(Clone)]
|
2023-02-01 09:50:47 +11:00
|
|
|
pub struct OutputTexture {
|
2023-02-01 16:16:06 +11:00
|
|
|
pub(crate) descriptor: OutputDescriptor,
|
|
|
|
pub(crate) size: Size<u32>,
|
2023-02-01 09:50:47 +11:00
|
|
|
}
|
|
|
|
|
|
|
|
impl OutputTexture {
|
2023-02-06 08:17:23 +11:00
|
|
|
pub fn new(
|
|
|
|
handle: D3D12DescriptorHeapSlot<RenderTargetHeap>,
|
|
|
|
size: Size<u32>,
|
|
|
|
) -> OutputTexture {
|
2023-02-01 09:50:47 +11:00
|
|
|
let descriptor = OutputDescriptor::Owned(handle);
|
2023-02-06 08:17:23 +11:00
|
|
|
OutputTexture { descriptor, size }
|
2023-02-01 09:50:47 +11:00
|
|
|
}
|
|
|
|
|
|
|
|
// unsafe since the lifetime of the handle has to survive
|
2023-02-06 08:17:23 +11:00
|
|
|
pub unsafe fn new_from_raw(
|
|
|
|
handle: D3D12_CPU_DESCRIPTOR_HANDLE,
|
|
|
|
size: Size<u32>,
|
2023-02-01 09:50:47 +11:00
|
|
|
) -> OutputTexture {
|
|
|
|
let descriptor = OutputDescriptor::Raw(handle);
|
2023-02-06 08:17:23 +11:00
|
|
|
OutputTexture { descriptor, size }
|
2023-02-01 09:50:47 +11:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2023-02-02 10:09:34 +11:00
|
|
|
#[derive(Clone)]
|
2023-02-01 09:50:47 +11:00
|
|
|
pub struct InputTexture {
|
2023-02-06 10:03:38 +11:00
|
|
|
pub(crate) resource: ID3D12Resource,
|
2023-02-01 11:23:57 +11:00
|
|
|
pub(crate) descriptor: InputDescriptor,
|
2023-02-01 10:59:49 +11:00
|
|
|
pub(crate) size: Size<u32>,
|
2023-02-06 10:03:38 +11:00
|
|
|
pub(crate) format: DXGI_FORMAT,
|
2023-02-01 11:23:57 +11:00
|
|
|
pub(crate) wrap_mode: WrapMode,
|
2023-02-06 08:17:23 +11:00
|
|
|
pub(crate) filter: FilterMode,
|
2023-02-01 09:50:47 +11:00
|
|
|
}
|
|
|
|
|
|
|
|
impl InputTexture {
|
2023-02-06 08:17:23 +11:00
|
|
|
pub fn new(
|
2023-02-06 10:03:38 +11:00
|
|
|
resource: ID3D12Resource,
|
2023-02-06 08:17:23 +11:00
|
|
|
handle: D3D12DescriptorHeapSlot<CpuStagingHeap>,
|
|
|
|
size: Size<u32>,
|
|
|
|
format: ImageFormat,
|
|
|
|
wrap_mode: WrapMode,
|
|
|
|
filter: FilterMode,
|
|
|
|
) -> InputTexture {
|
2023-02-01 09:50:47 +11:00
|
|
|
let srv = InputDescriptor::Owned(handle);
|
|
|
|
InputTexture {
|
2023-02-06 10:03:38 +11:00
|
|
|
resource,
|
2023-02-01 09:50:47 +11:00
|
|
|
descriptor: srv,
|
|
|
|
size,
|
2023-02-01 17:25:39 +11:00
|
|
|
format: format.into(),
|
2023-02-01 09:50:47 +11:00
|
|
|
wrap_mode,
|
2023-02-06 08:17:23 +11:00
|
|
|
filter,
|
2023-02-01 09:50:47 +11:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// unsafe since the lifetime of the handle has to survive
|
2023-02-06 08:17:23 +11:00
|
|
|
pub unsafe fn new_from_raw(
|
2023-02-06 10:03:38 +11:00
|
|
|
resource: ID3D12Resource,
|
2023-02-06 08:17:23 +11:00
|
|
|
handle: D3D12_CPU_DESCRIPTOR_HANDLE,
|
|
|
|
size: Size<u32>,
|
|
|
|
format: DXGI_FORMAT,
|
|
|
|
wrap_mode: WrapMode,
|
|
|
|
filter: FilterMode,
|
2023-02-01 09:50:47 +11:00
|
|
|
) -> InputTexture {
|
|
|
|
let srv = InputDescriptor::Raw(handle);
|
|
|
|
InputTexture {
|
2023-02-06 10:03:38 +11:00
|
|
|
resource,
|
2023-02-01 09:50:47 +11:00
|
|
|
descriptor: srv,
|
|
|
|
size,
|
|
|
|
format,
|
|
|
|
wrap_mode,
|
2023-02-06 08:17:23 +11:00
|
|
|
filter,
|
2023-02-01 09:50:47 +11:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
2023-02-01 11:23:57 +11:00
|
|
|
|
|
|
|
impl AsRef<InputTexture> for InputTexture {
|
|
|
|
fn as_ref(&self) -> &InputTexture {
|
|
|
|
self
|
|
|
|
}
|
2023-02-06 08:17:23 +11:00
|
|
|
}
|