mirror of
https://gitlab.freedesktop.org/gstreamer/gstreamer.git
synced 2024-11-27 12:11:13 +00:00
c57fe82a93
This object will upload system memory to GPU and preprocess texture using compute shader or software converter if needed Part-of: <https://gitlab.freedesktop.org/gstreamer/gstreamer/-/merge_requests/5691>
110 lines
3 KiB
C++
110 lines
3 KiB
C++
/* GStreamer
|
|
* Copyright (C) 2023 Seungha Yang <seungha@centricular.com>
|
|
*
|
|
* This library is free software; you can redistribute it and/or
|
|
* modify it under the terms of the GNU Library General Public
|
|
* License as published by the Free Software Foundation; either
|
|
* version 2 of the License, or (at your option) any later version.
|
|
*
|
|
* This library is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
* Library General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU Library General Public
|
|
* License along with this library; if not, write to the
|
|
* Free Software Foundation, Inc., 51 Franklin St, Fifth Floor,
|
|
* Boston, MA 02110-1301, USA.
|
|
*/
|
|
|
|
#ifdef HAVE_CONFIG_H
|
|
#include "config.h"
|
|
#endif
|
|
|
|
#include "gstd3d11shadercache.h"
|
|
#include "gstd3d11compile.h"
|
|
#include <string.h>
|
|
#include <mutex>
|
|
#include <map>
|
|
|
|
/* *INDENT-OFF* */
|
|
static std::mutex cache_lock_;
|
|
static std::map <gint64, ID3DBlob *> ps_blob_;
|
|
static std::map <gint64, ID3DBlob *> vs_blob_;
|
|
static std::map <gint64, ID3DBlob *> cs_blob_;
|
|
/* *INDENT-ON* */
|
|
|
|
HRESULT
|
|
gst_d3d11_shader_cache_get_pixel_shader_blob (gint64 token,
|
|
const gchar * source, gsize source_size, const gchar * entry_point,
|
|
const D3D_SHADER_MACRO * defines, ID3DBlob ** blob)
|
|
{
|
|
std::lock_guard < std::mutex > lk (cache_lock_);
|
|
|
|
auto cached = ps_blob_.find (token);
|
|
if (cached != ps_blob_.end ()) {
|
|
*blob = cached->second;
|
|
cached->second->AddRef ();
|
|
return S_OK;
|
|
}
|
|
|
|
HRESULT hr = gst_d3d11_compile (source, source_size, nullptr, defines,
|
|
nullptr, entry_point, "ps_5_0", 0, 0, blob, nullptr);
|
|
if (FAILED (hr))
|
|
return hr;
|
|
|
|
(*blob)->AddRef ();
|
|
ps_blob_[token] = *blob;
|
|
|
|
return S_OK;
|
|
}
|
|
|
|
HRESULT
|
|
gst_d3d11_shader_cache_get_vertex_shader_blob (gint64 token,
|
|
const gchar * source, gsize source_size, const gchar * entry_point,
|
|
ID3DBlob ** blob)
|
|
{
|
|
std::lock_guard < std::mutex > lk (cache_lock_);
|
|
|
|
auto cached = vs_blob_.find (token);
|
|
if (cached != vs_blob_.end ()) {
|
|
*blob = cached->second;
|
|
cached->second->AddRef ();
|
|
return S_OK;
|
|
}
|
|
|
|
HRESULT hr = gst_d3d11_compile (source, source_size, nullptr, nullptr,
|
|
nullptr, entry_point, "vs_5_0", 0, 0, blob, nullptr);
|
|
if (FAILED (hr))
|
|
return hr;
|
|
|
|
(*blob)->AddRef ();
|
|
vs_blob_[token] = *blob;
|
|
|
|
return S_OK;
|
|
}
|
|
|
|
HRESULT
|
|
gst_d3d11_shader_cache_get_compute_shader_blob (gint64 token,
|
|
const gchar * source, gsize source_size, const gchar * entry_point,
|
|
const D3D_SHADER_MACRO * defines, ID3DBlob ** blob)
|
|
{
|
|
std::lock_guard < std::mutex > lk (cache_lock_);
|
|
|
|
auto cached = cs_blob_.find (token);
|
|
if (cached != cs_blob_.end ()) {
|
|
*blob = cached->second;
|
|
cached->second->AddRef ();
|
|
return S_OK;
|
|
}
|
|
|
|
HRESULT hr = gst_d3d11_compile (source, source_size, nullptr, defines,
|
|
nullptr, entry_point, "cs_5_0", 0, 0, blob, nullptr);
|
|
if (FAILED (hr))
|
|
return hr;
|
|
|
|
(*blob)->AddRef ();
|
|
cs_blob_[token] = *blob;
|
|
|
|
return S_OK;
|
|
}
|