85 lines
3.2 KiB
Rust
85 lines
3.2 KiB
Rust
|
use windows::Win32::Graphics::Direct3D12::{D3D12_CPU_PAGE_PROPERTY_UNKNOWN, D3D12_FEATURE_DATA_FORMAT_SUPPORT, D3D12_FORMAT_SUPPORT1_MIP, D3D12_FORMAT_SUPPORT1_RENDER_TARGET, D3D12_FORMAT_SUPPORT1_SHADER_SAMPLE, D3D12_FORMAT_SUPPORT1_TEXTURE2D, D3D12_FORMAT_SUPPORT2_UAV_TYPED_STORE, D3D12_HEAP_FLAG_NONE, D3D12_HEAP_PROPERTIES, D3D12_HEAP_TYPE_DEFAULT, D3D12_MEMORY_POOL_UNKNOWN, D3D12_RESOURCE_DESC, D3D12_RESOURCE_DIMENSION_TEXTURE2D, D3D12_RESOURCE_FLAG_ALLOW_UNORDERED_ACCESS, D3D12_RESOURCE_STATE_PIXEL_SHADER_RESOURCE, ID3D12Device, ID3D12Resource};
|
||
|
|
use windows::Win32::Graphics::Dxgi::Common::{DXGI_FORMAT, DXGI_SAMPLE_DESC};
|
||
|
|
use librashader_common::{ImageFormat, Size};
|
||
|
|
use librashader_runtime::scaling::MipmapSize;
|
||
|
|
use crate::error;
|
||
|
|
use crate::error::assume_d3d12_init;
|
||
|
|
use crate::util::d3d12_get_closest_format;
|
||
|
|
|
||
|
|
#[derive(Debug, Clone)]
|
||
|
|
pub(crate) struct OwnedImage {
|
||
|
|
render: ID3D12Resource,
|
||
|
|
pub(crate) size: Size<u32>,
|
||
|
|
format: DXGI_FORMAT,
|
||
|
|
device: ID3D12Device,
|
||
|
|
max_mipmap: u16,
|
||
|
|
}
|
||
|
|
|
||
|
|
pub fn new(
|
||
|
|
device: &ID3D12Device,
|
||
|
|
size: Size<u32>,
|
||
|
|
format: ImageFormat,
|
||
|
|
mipmap: bool,
|
||
|
|
) -> error::Result<OwnedImage> {
|
||
|
|
unsafe {
|
||
|
|
let miplevels = source.size.calculate_miplevels() as u16;
|
||
|
|
let mut desc = D3D12_RESOURCE_DESC {
|
||
|
|
Dimension: D3D12_RESOURCE_DIMENSION_TEXTURE2D,
|
||
|
|
Alignment: 0,
|
||
|
|
Width: size.width as u64,
|
||
|
|
Height: size.height,
|
||
|
|
DepthOrArraySize: 1,
|
||
|
|
MipLevels: if mipmap { miplevels } else { 1 },
|
||
|
|
Format: format.into(),
|
||
|
|
SampleDesc: DXGI_SAMPLE_DESC {
|
||
|
|
Count: 1,
|
||
|
|
Quality: 0,
|
||
|
|
},
|
||
|
|
Layout: Default::default(),
|
||
|
|
Flags: Default::default(),
|
||
|
|
};
|
||
|
|
|
||
|
|
let mut format_support = D3D12_FEATURE_DATA_FORMAT_SUPPORT {
|
||
|
|
Format: desc.Format,
|
||
|
|
Support1: D3D12_FORMAT_SUPPORT1_TEXTURE2D
|
||
|
|
| D3D12_FORMAT_SUPPORT1_SHADER_SAMPLE
|
||
|
|
| D3D12_FORMAT_SUPPORT1_RENDER_TARGET,
|
||
|
|
..Default::default()
|
||
|
|
};
|
||
|
|
|
||
|
|
if mipmap {
|
||
|
|
desc.Flags |= D3D12_RESOURCE_FLAG_ALLOW_UNORDERED_ACCESS;
|
||
|
|
format_support.Support1 |= D3D12_FORMAT_SUPPORT1_MIP;
|
||
|
|
format_support.Support2 |= D3D12_FORMAT_SUPPORT2_UAV_TYPED_STORE;
|
||
|
|
}
|
||
|
|
|
||
|
|
desc.Format = d3d12_get_closest_format(device, desc.Format, format_support);
|
||
|
|
|
||
|
|
let mut resource: Option<ID3D12Resource> = None;
|
||
|
|
unsafe {
|
||
|
|
device.CreateCommittedResource(
|
||
|
|
&D3D12_HEAP_PROPERTIES {
|
||
|
|
Type: D3D12_HEAP_TYPE_DEFAULT,
|
||
|
|
CPUPageProperty: D3D12_CPU_PAGE_PROPERTY_UNKNOWN,
|
||
|
|
MemoryPoolPreference: D3D12_MEMORY_POOL_UNKNOWN,
|
||
|
|
CreationNodeMask: 1,
|
||
|
|
VisibleNodeMask: 1,
|
||
|
|
},
|
||
|
|
D3D12_HEAP_FLAG_NONE,
|
||
|
|
&desc,
|
||
|
|
D3D12_RESOURCE_STATE_PIXEL_SHADER_RESOURCE,
|
||
|
|
None,
|
||
|
|
&mut resource,
|
||
|
|
)?;
|
||
|
|
}
|
||
|
|
assume_d3d12_init!(resource, "CreateCommittedResource");
|
||
|
|
|
||
|
|
Ok(OwnedImage {
|
||
|
|
render: resource,
|
||
|
|
size,
|
||
|
|
format: desc.Format,
|
||
|
|
device: device.clone(),
|
||
|
|
max_mipmap: miplevels,
|
||
|
|
})
|
||
|
|
}
|
||
|
|
}
|