d3d12converter: Add support for sampler filter update

Creates new root signature and pipeline state object
if sampler filter method is updated

Part-of: <https://gitlab.freedesktop.org/gstreamer/gstreamer/-/merge_requests/7550>
This commit is contained in:
Seungha Yang 2024-09-18 20:21:22 +09:00 committed by GStreamer Marge Bot
parent c75e6d01c5
commit a1029d5b87

View file

@ -206,6 +206,7 @@ enum
PROP_FILL_BORDER, PROP_FILL_BORDER,
PROP_BORDER_COLOR, PROP_BORDER_COLOR,
PROP_VIDEO_DIRECTION, PROP_VIDEO_DIRECTION,
PROP_SAMPLER_FILTER,
}; };
/* *INDENT-OFF* */ /* *INDENT-OFF* */
@ -261,6 +262,7 @@ struct _GstD3D12ConverterPrivate
FLOAT blend_factor[4]; FLOAT blend_factor[4];
DXGI_SAMPLE_DESC sample_desc; DXGI_SAMPLE_DESC sample_desc;
gboolean update_pso = FALSE; gboolean update_pso = FALSE;
gboolean update_sampler = FALSE;
ConverterRootSignaturePtr crs; ConverterRootSignaturePtr crs;
ComPtr<ID3D12RootSignature> rs; ComPtr<ID3D12RootSignature> rs;
@ -319,6 +321,7 @@ struct _GstD3D12ConverterPrivate
GST_D3D12_CONVERTER_ALPHA_MODE_UNSPECIFIED; GST_D3D12_CONVERTER_ALPHA_MODE_UNSPECIFIED;
GstD3D12ConverterAlphaMode dst_alpha_mode = GstD3D12ConverterAlphaMode dst_alpha_mode =
GST_D3D12_CONVERTER_ALPHA_MODE_UNSPECIFIED; GST_D3D12_CONVERTER_ALPHA_MODE_UNSPECIFIED;
D3D12_FILTER sampler_filter = D3D12_FILTER_MIN_MAG_LINEAR_MIP_POINT;
}; };
/* *INDENT-ON* */ /* *INDENT-ON* */
@ -386,6 +389,10 @@ gst_d3d12_converter_class_init (GstD3D12ConverterClass * klass)
g_param_spec_enum ("video-direction", "Video Direction", g_param_spec_enum ("video-direction", "Video Direction",
"Video direction", GST_TYPE_VIDEO_ORIENTATION_METHOD, "Video direction", GST_TYPE_VIDEO_ORIENTATION_METHOD,
GST_VIDEO_ORIENTATION_IDENTITY, param_flags)); GST_VIDEO_ORIENTATION_IDENTITY, param_flags));
g_object_class_install_property (object_class, PROP_SAMPLER_FILTER,
g_param_spec_enum ("sampler-filter", "Sampler Filter",
"Sampler Filter", GST_TYPE_D3D12_CONVERTER_SAMPLER_FILTER,
D3D12_FILTER_MIN_MAG_LINEAR_MIP_POINT, param_flags));
GST_DEBUG_CATEGORY_INIT (gst_d3d12_converter_debug, GST_DEBUG_CATEGORY_INIT (gst_d3d12_converter_debug,
"d3d12converter", 0, "d3d12converter"); "d3d12converter", 0, "d3d12converter");
@ -472,7 +479,8 @@ gst_d3d12_converter_set_property (GObject * object, guint prop_id,
case PROP_ALPHA: case PROP_ALPHA:
priv->alpha = g_value_get_double (value); priv->alpha = g_value_get_double (value);
break; break;
case PROP_FILL_BORDER:{ case PROP_FILL_BORDER:
{
gboolean fill_border = g_value_get_boolean (value); gboolean fill_border = g_value_get_boolean (value);
if (fill_border != priv->fill_border) { if (fill_border != priv->fill_border) {
@ -481,7 +489,8 @@ gst_d3d12_converter_set_property (GObject * object, guint prop_id,
} }
break; break;
} }
case PROP_BORDER_COLOR:{ case PROP_BORDER_COLOR:
{
guint64 border_color = g_value_get_uint64 (value); guint64 border_color = g_value_get_uint64 (value);
if (border_color != priv->border_color) { if (border_color != priv->border_color) {
@ -490,7 +499,8 @@ gst_d3d12_converter_set_property (GObject * object, guint prop_id,
} }
break; break;
} }
case PROP_VIDEO_DIRECTION:{ case PROP_VIDEO_DIRECTION:
{
GstVideoOrientationMethod video_direction = GstVideoOrientationMethod video_direction =
(GstVideoOrientationMethod) g_value_get_enum (value); (GstVideoOrientationMethod) g_value_get_enum (value);
if (video_direction != priv->video_direction) { if (video_direction != priv->video_direction) {
@ -499,6 +509,15 @@ gst_d3d12_converter_set_property (GObject * object, guint prop_id,
} }
break; break;
} }
case PROP_SAMPLER_FILTER:
{
auto filter = (D3D12_FILTER) g_value_get_enum (value);
if (filter != priv->sampler_filter) {
priv->sampler_filter = filter;
priv->update_sampler = TRUE;
}
break;
}
default: default:
G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec); G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
break; break;
@ -550,6 +569,9 @@ gst_d3d12_converter_get_property (GObject * object, guint prop_id,
case PROP_VIDEO_DIRECTION: case PROP_VIDEO_DIRECTION:
g_value_set_enum (value, priv->video_direction); g_value_set_enum (value, priv->video_direction);
break; break;
case PROP_SAMPLER_FILTER:
g_value_set_enum (value, priv->sampler_filter);
break;
default: default:
G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec); G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
break; break;
@ -656,6 +678,62 @@ reorder_rtv_index (GstVideoFormat output_format, guint index)
return index; return index;
} }
static gboolean
gst_d3d12_converter_create_sampler (GstD3D12Converter * self,
D3D12_FILTER filter, ID3D12DescriptorHeap ** heap)
{
auto priv = self->priv;
ComPtr < ID3D12DescriptorHeap > sampler_heap;
auto hr = gst_d3d12_device_get_sampler_state (self->device, filter,
&sampler_heap);
if (!gst_d3d12_result (hr, self->device))
return FALSE;
if (priv->crs->HaveLut ()) {
D3D12_DESCRIPTOR_HEAP_DESC heap_desc = { };
heap_desc.NumDescriptors = 1;
heap_desc.Type = D3D12_DESCRIPTOR_HEAP_TYPE_SAMPLER;
heap_desc.Flags = D3D12_DESCRIPTOR_HEAP_FLAG_SHADER_VISIBLE;
ComPtr < ID3D12DescriptorHeap > new_heap;
auto device = gst_d3d12_device_get_device_handle (self->device);
hr = device->CreateDescriptorHeap (&heap_desc, IID_PPV_ARGS (&new_heap));
if (!gst_d3d12_result (hr, self->device)) {
GST_ERROR_OBJECT (self, "Couldn't create sampler heap");
return FALSE;
}
auto dst_handle = CD3DX12_CPU_DESCRIPTOR_HANDLE
(GetCPUDescriptorHandleForHeapStart (new_heap));
device->CopyDescriptorsSimple (1, dst_handle,
GetCPUDescriptorHandleForHeapStart (sampler_heap),
D3D12_DESCRIPTOR_HEAP_TYPE_SAMPLER);
if (filter != D3D12_FILTER_MIN_MAG_LINEAR_MIP_POINT) {
hr = gst_d3d12_device_get_sampler_state (self->device,
D3D12_FILTER_MIN_MAG_LINEAR_MIP_POINT,
sampler_heap.ReleaseAndGetAddressOf ());
if (!gst_d3d12_result (hr, self->device)) {
GST_ERROR_OBJECT (self, "Couldn't create sampler heap");
return FALSE;
}
}
dst_handle.Offset (priv->sampler_inc_size);
device->CopyDescriptorsSimple (1, dst_handle,
GetCPUDescriptorHandleForHeapStart (sampler_heap),
D3D12_DESCRIPTOR_HEAP_TYPE_SAMPLER);
*heap = new_heap.Detach ();
} else {
*heap = sampler_heap.Detach ();
}
return TRUE;
}
static gboolean static gboolean
gst_d3d12_converter_setup_resource (GstD3D12Converter * self, gst_d3d12_converter_setup_resource (GstD3D12Converter * self,
const GstVideoInfo * in_info, const GstVideoInfo * out_info, const GstVideoInfo * in_info, const GstVideoInfo * out_info,
@ -695,59 +773,20 @@ gst_d3d12_converter_setup_resource (GstD3D12Converter * self,
return FALSE; return FALSE;
} }
ComPtr < ID3D12DescriptorHeap > sampler_heap; if (!gst_d3d12_converter_create_sampler (self, sampler_filter,
hr = gst_d3d12_device_get_sampler_state (self->device, sampler_filter, &priv->sampler_heap)) {
&sampler_heap);
if (FAILED (hr) && sampler_filter != D3D12_FILTER_MIN_MAG_LINEAR_MIP_POINT) {
sampler_filter = D3D12_FILTER_MIN_MAG_LINEAR_MIP_POINT;
GST_WARNING_OBJECT (self,
"Couldn't create requested sampler, trying linear sampler");
hr = gst_d3d12_device_get_sampler_state (self->device,
sampler_filter, &sampler_heap);
}
if (!gst_d3d12_result (hr, self->device)) {
GST_ERROR_OBJECT (self, "Couldn't create sampler");
return FALSE;
}
if (priv->crs->HaveLut ()) {
D3D12_DESCRIPTOR_HEAP_DESC heap_desc = { };
heap_desc.NumDescriptors = 1;
heap_desc.Type = D3D12_DESCRIPTOR_HEAP_TYPE_SAMPLER;
heap_desc.Flags = D3D12_DESCRIPTOR_HEAP_FLAG_SHADER_VISIBLE;
hr = device->CreateDescriptorHeap (&heap_desc,
IID_PPV_ARGS (&priv->sampler_heap));
if (!gst_d3d12_result (hr, self->device)) {
GST_ERROR_OBJECT (self, "Couldn't create sampler heap");
return FALSE;
}
auto dst_handle = CD3DX12_CPU_DESCRIPTOR_HANDLE
(GetCPUDescriptorHandleForHeapStart (priv->sampler_heap));
device->CopyDescriptorsSimple (1, dst_handle,
GetCPUDescriptorHandleForHeapStart (sampler_heap),
D3D12_DESCRIPTOR_HEAP_TYPE_SAMPLER);
if (sampler_filter != D3D12_FILTER_MIN_MAG_LINEAR_MIP_POINT) { if (sampler_filter != D3D12_FILTER_MIN_MAG_LINEAR_MIP_POINT) {
hr = gst_d3d12_device_get_sampler_state (self->device, sampler_filter = D3D12_FILTER_MIN_MAG_LINEAR_MIP_POINT;
D3D12_FILTER_MIN_MAG_LINEAR_MIP_POINT, if (!gst_d3d12_converter_create_sampler (self, sampler_filter,
sampler_heap.ReleaseAndGetAddressOf ()); &priv->sampler_heap)) {
return FALSE;
if (!gst_d3d12_result (hr, self->device)) { }
GST_ERROR_OBJECT (self, "Couldn't create sampler heap"); } else {
return FALSE; return FALSE;
} }
} }
dst_handle.Offset (priv->sampler_inc_size); priv->sampler_filter = sampler_filter;
device->CopyDescriptorsSimple (1, dst_handle,
GetCPUDescriptorHandleForHeapStart (sampler_heap),
D3D12_DESCRIPTOR_HEAP_TYPE_SAMPLER);
} else {
priv->sampler_heap = sampler_heap;
}
auto psblob_list = auto psblob_list =
gst_d3d12_get_converter_pixel_shader_blob (GST_VIDEO_INFO_FORMAT gst_d3d12_get_converter_pixel_shader_blob (GST_VIDEO_INFO_FORMAT
@ -2080,6 +2119,21 @@ gst_d3d12_converter_update_pso (GstD3D12Converter * self)
return TRUE; return TRUE;
} }
static void
gst_d3d12_converter_update_sampler (GstD3D12Converter * self)
{
auto priv = self->priv;
if (!priv->update_sampler)
return;
priv->update_sampler = FALSE;
ComPtr < ID3D12DescriptorHeap > sampler_heap;
if (gst_d3d12_converter_create_sampler (self, priv->sampler_filter,
&sampler_heap)) {
priv->sampler_heap = sampler_heap;
}
}
static void static void
reorder_rtv_handles (GstVideoFormat output_format, reorder_rtv_handles (GstVideoFormat output_format,
D3D12_CPU_DESCRIPTOR_HANDLE * src, D3D12_CPU_DESCRIPTOR_HANDLE * dst) D3D12_CPU_DESCRIPTOR_HANDLE * src, D3D12_CPU_DESCRIPTOR_HANDLE * dst)
@ -2137,6 +2191,8 @@ gst_d3d12_converter_execute (GstD3D12Converter * self, GstD3D12Frame * in_frame,
return FALSE; return FALSE;
} }
gst_d3d12_converter_update_sampler (self);
if (priv->vertex_upload) { if (priv->vertex_upload) {
auto barrier = auto barrier =
CD3DX12_RESOURCE_BARRIER::Transition (priv->shader_buf.Get (), CD3DX12_RESOURCE_BARRIER::Transition (priv->shader_buf.Get (),
@ -2244,6 +2300,10 @@ gst_d3d12_converter_execute (GstD3D12Converter * self, GstD3D12Frame * in_frame,
FENCE_NOTIFY_COM (priv->vertex_upload.Detach ())); FENCE_NOTIFY_COM (priv->vertex_upload.Detach ()));
} }
auto sampler = priv->sampler_heap.Get ();
sampler->AddRef ();
gst_d3d12_fence_data_push (fence_data, FENCE_NOTIFY_COM (sampler));
return TRUE; return TRUE;
} }