vkd3d: Implement vkd3d_shader_cache_get.

This commit is contained in:
Stefan Dösinger 2024-04-05 17:21:53 +03:00 committed by Alexandre Julliard
parent aad76f4ed2
commit 71f1955cd6
Notes: Alexandre Julliard 2024-04-25 00:16:00 +02:00
Approved-by: Henri Verbeet (@hverbeet)
Approved-by: Alexandre Julliard (@julliard)
Merge-Request: https://gitlab.winehq.org/wine/vkd3d/-/merge_requests/791
5 changed files with 77 additions and 2 deletions

View File

@ -55,6 +55,10 @@ enum vkd3d_result
VKD3D_ERROR_NOT_IMPLEMENTED = -5,
/** The object or entry already exists. \since 1.12 */
VKD3D_ERROR_KEY_ALREADY_EXISTS = -6,
/** The requested object was not found. \since 1.12 */
VKD3D_ERROR_NOT_FOUND = -7,
/** The output buffer is larger than the requested object \since 1.12. */
VKD3D_ERROR_MORE_DATA = -8,
VKD3D_FORCE_32_BIT_ENUM(VKD3D_RESULT),
};

View File

@ -37,6 +37,10 @@ HRESULT hresult_from_vkd3d_result(int vkd3d_result)
return E_NOTIMPL;
case VKD3D_ERROR_KEY_ALREADY_EXISTS:
return DXGI_ERROR_ALREADY_EXISTS;
case VKD3D_ERROR_NOT_FOUND:
return DXGI_ERROR_NOT_FOUND;
case VKD3D_ERROR_MORE_DATA:
return DXGI_ERROR_MORE_DATA;
default:
FIXME("Unhandled vkd3d result %d.\n", vkd3d_result);
return E_FAIL;

View File

@ -179,3 +179,54 @@ int vkd3d_shader_cache_put(struct vkd3d_shader_cache *cache,
done:
return ret;
}
int vkd3d_shader_cache_get(struct vkd3d_shader_cache *cache,
const void *key, size_t key_size, void *value, size_t *value_size)
{
struct shader_cache_entry *e;
struct shader_cache_key k;
struct rb_entry *entry;
enum vkd3d_result ret;
size_t size_in;
TRACE("%p, %p, %#zx, %p, %p.\n", cache, key, key_size, value, value_size);
size_in = *value_size;
k.hash = vkd3d_shader_cache_hash_key(key, key_size);
k.key = key;
k.key_size = key_size;
entry = rb_get(&cache->tree, &k);
if (!entry)
{
WARN("Entry not found.\n");
ret = VKD3D_ERROR_NOT_FOUND;
goto done;
}
e = RB_ENTRY_VALUE(entry, struct shader_cache_entry, entry);
*value_size = e->h.value_size;
if (!value)
{
TRACE("Found item %#"PRIx64", returning needed size %#"PRIx64".\n",
e->h.hash, e->h.value_size);
ret = VKD3D_OK;
goto done;
}
if (size_in < e->h.value_size)
{
WARN("Output buffer is too small for item %#"PRIx64", got %#zx want %#"PRIx64".\n",
e->h.hash, size_in, e->h.value_size);
ret = VKD3D_ERROR_MORE_DATA;
goto done;
}
memcpy(value, e->payload + e->h.key_size, e->h.value_size);
ret = VKD3D_OK;
TRACE("Returning cached item %#"PRIx64".\n", e->h.hash);
done:
return ret;
}

View File

@ -2685,10 +2685,24 @@ static HRESULT STDMETHODCALLTYPE d3d12_cache_session_GetDevice(ID3D12ShaderCache
static HRESULT STDMETHODCALLTYPE d3d12_cache_session_FindValue(ID3D12ShaderCacheSession *iface,
const void *key, UINT key_size, void *value, UINT *value_size)
{
FIXME("iface %p, key %p, key_size %#x, value %p, value_size %p stub!\n",
struct d3d12_cache_session *session = impl_from_ID3D12ShaderCacheSession(iface);
enum vkd3d_result ret;
size_t size;
TRACE("iface %p, key %p, key_size %#x, value %p, value_size %p.\n",
iface, key, key_size, value, value_size);
return DXGI_ERROR_NOT_FOUND;
if (!value_size)
{
WARN("value_size is NULL, returning E_INVALIDARG.\n");
return E_INVALIDARG;
}
size = *value_size;
ret = vkd3d_shader_cache_get(session->cache, key, key_size, value, &size);
*value_size = size;
return hresult_from_vkd3d_result(ret);
}
static HRESULT STDMETHODCALLTYPE d3d12_cache_session_StoreValue(ID3D12ShaderCacheSession *iface,

View File

@ -1778,5 +1778,7 @@ unsigned int vkd3d_shader_cache_incref(struct vkd3d_shader_cache *cache);
unsigned int vkd3d_shader_cache_decref(struct vkd3d_shader_cache *cache);
int vkd3d_shader_cache_put(struct vkd3d_shader_cache *cache,
const void *key, size_t key_size, const void *value, size_t value_size);
int vkd3d_shader_cache_get(struct vkd3d_shader_cache *cache,
const void *key, size_t key_size, void *value, size_t *value_size);
#endif /* __VKD3D_PRIVATE_H */