summaryrefslogtreecommitdiffstats
path: root/third_party/rust/d3d12/src/pso.rs
diff options
context:
space:
mode:
authorDaniel Baumann <daniel.baumann@progress-linux.org>2024-04-07 19:33:14 +0000
committerDaniel Baumann <daniel.baumann@progress-linux.org>2024-04-07 19:33:14 +0000
commit36d22d82aa202bb199967e9512281e9a53db42c9 (patch)
tree105e8c98ddea1c1e4784a60a5a6410fa416be2de /third_party/rust/d3d12/src/pso.rs
parentInitial commit. (diff)
downloadfirefox-esr-36d22d82aa202bb199967e9512281e9a53db42c9.tar.xz
firefox-esr-36d22d82aa202bb199967e9512281e9a53db42c9.zip
Adding upstream version 115.7.0esr.upstream/115.7.0esrupstream
Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
Diffstat (limited to 'third_party/rust/d3d12/src/pso.rs')
-rw-r--r--third_party/rust/d3d12/src/pso.rs169
1 files changed, 169 insertions, 0 deletions
diff --git a/third_party/rust/d3d12/src/pso.rs b/third_party/rust/d3d12/src/pso.rs
new file mode 100644
index 0000000000..e7dbc0a34b
--- /dev/null
+++ b/third_party/rust/d3d12/src/pso.rs
@@ -0,0 +1,169 @@
+//! Pipeline state
+
+use crate::{com::WeakPtr, Blob, D3DResult, Error};
+use std::{ffi, ops::Deref, ptr};
+use winapi::um::{d3d12, d3dcompiler};
+
+bitflags! {
+ pub struct PipelineStateFlags: u32 {
+ const TOOL_DEBUG = d3d12::D3D12_PIPELINE_STATE_FLAG_TOOL_DEBUG;
+ }
+}
+
+bitflags! {
+ pub struct ShaderCompileFlags: u32 {
+ const DEBUG = d3dcompiler::D3DCOMPILE_DEBUG;
+ const SKIP_VALIDATION = d3dcompiler::D3DCOMPILE_SKIP_VALIDATION;
+ const SKIP_OPTIMIZATION = d3dcompiler::D3DCOMPILE_SKIP_OPTIMIZATION;
+ const PACK_MATRIX_ROW_MAJOR = d3dcompiler::D3DCOMPILE_PACK_MATRIX_ROW_MAJOR;
+ const PACK_MATRIX_COLUMN_MAJOR = d3dcompiler::D3DCOMPILE_PACK_MATRIX_COLUMN_MAJOR;
+ const PARTIAL_PRECISION = d3dcompiler::D3DCOMPILE_PARTIAL_PRECISION;
+ // TODO: add missing flags
+ }
+}
+
+#[derive(Copy, Clone)]
+pub struct Shader(d3d12::D3D12_SHADER_BYTECODE);
+impl Shader {
+ pub fn null() -> Self {
+ Shader(d3d12::D3D12_SHADER_BYTECODE {
+ BytecodeLength: 0,
+ pShaderBytecode: ptr::null(),
+ })
+ }
+
+ pub fn from_raw(data: &[u8]) -> Self {
+ Shader(d3d12::D3D12_SHADER_BYTECODE {
+ BytecodeLength: data.len() as _,
+ pShaderBytecode: data.as_ptr() as _,
+ })
+ }
+
+ // `blob` may not be null.
+ pub fn from_blob(blob: Blob) -> Self {
+ Shader(d3d12::D3D12_SHADER_BYTECODE {
+ BytecodeLength: unsafe { blob.GetBufferSize() },
+ pShaderBytecode: unsafe { blob.GetBufferPointer() },
+ })
+ }
+
+ /// Compile a shader from raw HLSL.
+ ///
+ /// * `target`: example format: `ps_5_1`.
+ pub fn compile(
+ code: &[u8],
+ target: &ffi::CStr,
+ entry: &ffi::CStr,
+ flags: ShaderCompileFlags,
+ ) -> D3DResult<(Blob, Error)> {
+ let mut shader = Blob::null();
+ let mut error = Error::null();
+
+ let hr = unsafe {
+ d3dcompiler::D3DCompile(
+ code.as_ptr() as *const _,
+ code.len(),
+ ptr::null(), // defines
+ ptr::null(), // include
+ ptr::null_mut(),
+ entry.as_ptr() as *const _,
+ target.as_ptr() as *const _,
+ flags.bits(),
+ 0,
+ shader.mut_void() as *mut *mut _,
+ error.mut_void() as *mut *mut _,
+ )
+ };
+
+ ((shader, error), hr)
+ }
+}
+
+impl Deref for Shader {
+ type Target = d3d12::D3D12_SHADER_BYTECODE;
+ fn deref(&self) -> &Self::Target {
+ &self.0
+ }
+}
+
+impl From<Option<Blob>> for Shader {
+ fn from(blob: Option<Blob>) -> Self {
+ match blob {
+ Some(b) => Shader::from_blob(b),
+ None => Shader::null(),
+ }
+ }
+}
+
+#[derive(Copy, Clone)]
+pub struct CachedPSO(d3d12::D3D12_CACHED_PIPELINE_STATE);
+impl CachedPSO {
+ pub fn null() -> Self {
+ CachedPSO(d3d12::D3D12_CACHED_PIPELINE_STATE {
+ CachedBlobSizeInBytes: 0,
+ pCachedBlob: ptr::null(),
+ })
+ }
+
+ // `blob` may not be null.
+ pub fn from_blob(blob: Blob) -> Self {
+ CachedPSO(d3d12::D3D12_CACHED_PIPELINE_STATE {
+ CachedBlobSizeInBytes: unsafe { blob.GetBufferSize() },
+ pCachedBlob: unsafe { blob.GetBufferPointer() },
+ })
+ }
+}
+
+impl Deref for CachedPSO {
+ type Target = d3d12::D3D12_CACHED_PIPELINE_STATE;
+ fn deref(&self) -> &Self::Target {
+ &self.0
+ }
+}
+
+pub type PipelineState = WeakPtr<d3d12::ID3D12PipelineState>;
+
+#[repr(u32)]
+pub enum Subobject {
+ RootSignature = d3d12::D3D12_PIPELINE_STATE_SUBOBJECT_TYPE_ROOT_SIGNATURE,
+ VS = d3d12::D3D12_PIPELINE_STATE_SUBOBJECT_TYPE_VS,
+ PS = d3d12::D3D12_PIPELINE_STATE_SUBOBJECT_TYPE_PS,
+ DS = d3d12::D3D12_PIPELINE_STATE_SUBOBJECT_TYPE_DS,
+ HS = d3d12::D3D12_PIPELINE_STATE_SUBOBJECT_TYPE_HS,
+ GS = d3d12::D3D12_PIPELINE_STATE_SUBOBJECT_TYPE_GS,
+ CS = d3d12::D3D12_PIPELINE_STATE_SUBOBJECT_TYPE_CS,
+ StreamOutput = d3d12::D3D12_PIPELINE_STATE_SUBOBJECT_TYPE_STREAM_OUTPUT,
+ Blend = d3d12::D3D12_PIPELINE_STATE_SUBOBJECT_TYPE_BLEND,
+ SampleMask = d3d12::D3D12_PIPELINE_STATE_SUBOBJECT_TYPE_SAMPLE_MASK,
+ Rasterizer = d3d12::D3D12_PIPELINE_STATE_SUBOBJECT_TYPE_RASTERIZER,
+ DepthStencil = d3d12::D3D12_PIPELINE_STATE_SUBOBJECT_TYPE_DEPTH_STENCIL,
+ InputLayout = d3d12::D3D12_PIPELINE_STATE_SUBOBJECT_TYPE_INPUT_LAYOUT,
+ IBStripCut = d3d12::D3D12_PIPELINE_STATE_SUBOBJECT_TYPE_IB_STRIP_CUT_VALUE,
+ PrimitiveTopology = d3d12::D3D12_PIPELINE_STATE_SUBOBJECT_TYPE_PRIMITIVE_TOPOLOGY,
+ RTFormats = d3d12::D3D12_PIPELINE_STATE_SUBOBJECT_TYPE_RENDER_TARGET_FORMATS,
+ DSFormat = d3d12::D3D12_PIPELINE_STATE_SUBOBJECT_TYPE_DEPTH_STENCIL_FORMAT,
+ SampleDesc = d3d12::D3D12_PIPELINE_STATE_SUBOBJECT_TYPE_SAMPLE_DESC,
+ NodeMask = d3d12::D3D12_PIPELINE_STATE_SUBOBJECT_TYPE_NODE_MASK,
+ CachedPSO = d3d12::D3D12_PIPELINE_STATE_SUBOBJECT_TYPE_CACHED_PSO,
+ Flags = d3d12::D3D12_PIPELINE_STATE_SUBOBJECT_TYPE_FLAGS,
+ DepthStencil1 = d3d12::D3D12_PIPELINE_STATE_SUBOBJECT_TYPE_DEPTH_STENCIL1,
+ // ViewInstancing = d3d12::D3D12_PIPELINE_STATE_SUBOBJECT_TYPE_VIEW_INSTANCING,
+}
+
+/// Subobject of a pipeline stream description
+#[repr(C)]
+pub struct PipelineStateSubobject<T> {
+ subobject_align: [usize; 0], // Subobjects must have the same alignment as pointers.
+ subobject_type: d3d12::D3D12_PIPELINE_STATE_SUBOBJECT_TYPE,
+ subobject: T,
+}
+
+impl<T> PipelineStateSubobject<T> {
+ pub fn new(subobject_type: Subobject, subobject: T) -> Self {
+ PipelineStateSubobject {
+ subobject_align: [],
+ subobject_type: subobject_type as _,
+ subobject,
+ }
+ }
+}