mirror of https://github.com/doitsujin/dxvk
131 lines
2.7 KiB
C++
131 lines
2.7 KiB
C++
#pragma once
|
|
|
|
#include "d3d11_include.h"
|
|
|
|
#include "../util/com/com_private_data.h"
|
|
|
|
namespace dxvk {
|
|
|
|
class D3D11Device;
|
|
|
|
template<typename Base>
|
|
class D3D11DeviceObject : public Base {
|
|
|
|
public:
|
|
|
|
D3D11DeviceObject(D3D11Device* pDevice)
|
|
: m_parent(pDevice) {
|
|
|
|
}
|
|
|
|
HRESULT STDMETHODCALLTYPE GetPrivateData(
|
|
REFGUID guid,
|
|
UINT *pDataSize,
|
|
void *pData) final {
|
|
return m_privateData.getData(
|
|
guid, pDataSize, pData);
|
|
}
|
|
|
|
HRESULT STDMETHODCALLTYPE SetPrivateData(
|
|
REFGUID guid,
|
|
UINT DataSize,
|
|
const void *pData) final {
|
|
return m_privateData.setData(
|
|
guid, DataSize, pData);
|
|
}
|
|
|
|
HRESULT STDMETHODCALLTYPE SetPrivateDataInterface(
|
|
REFGUID guid,
|
|
const IUnknown *pUnknown) final {
|
|
return m_privateData.setInterface(
|
|
guid, pUnknown);
|
|
}
|
|
|
|
void STDMETHODCALLTYPE GetDevice(
|
|
ID3D11Device** ppDevice) final {
|
|
*ppDevice = ref(GetParentInterface());
|
|
}
|
|
|
|
protected:
|
|
|
|
ID3D11Device* GetParentInterface() const {
|
|
// We don't know the definition of ID3D11Device
|
|
// here, because D3D11Device includes this file.
|
|
return reinterpret_cast<ID3D11Device*>(m_parent);
|
|
}
|
|
|
|
D3D11Device* const m_parent;
|
|
|
|
private:
|
|
|
|
ComPrivateData m_privateData;
|
|
|
|
};
|
|
|
|
|
|
template<typename Base>
|
|
class D3D11DeviceChild : public D3D11DeviceObject<ComObject<Base>> {
|
|
|
|
public:
|
|
|
|
D3D11DeviceChild(D3D11Device* pDevice)
|
|
: D3D11DeviceObject<ComObject<Base>>(pDevice) {
|
|
|
|
}
|
|
|
|
ULONG STDMETHODCALLTYPE AddRef() {
|
|
uint32_t refCount = this->m_refCount++;
|
|
if (unlikely(!refCount)) {
|
|
this->AddRefPrivate();
|
|
this->GetParentInterface()->AddRef();
|
|
}
|
|
|
|
return refCount + 1;
|
|
}
|
|
|
|
ULONG STDMETHODCALLTYPE Release() {
|
|
uint32_t refCount = --this->m_refCount;
|
|
if (unlikely(!refCount)) {
|
|
auto* parent = this->GetParentInterface();
|
|
this->ReleasePrivate();
|
|
parent->Release();
|
|
}
|
|
return refCount;
|
|
}
|
|
|
|
};
|
|
|
|
template<typename Base>
|
|
class D3D11StateObject : public D3D11DeviceObject<Base> {
|
|
|
|
public:
|
|
|
|
D3D11StateObject(D3D11Device* pDevice)
|
|
: D3D11DeviceObject<Base>(pDevice) {
|
|
|
|
}
|
|
|
|
ULONG STDMETHODCALLTYPE AddRef() {
|
|
uint32_t refCount = this->m_refCount++;
|
|
if (unlikely(!refCount))
|
|
this->GetParentInterface()->AddRef();
|
|
|
|
return refCount + 1;
|
|
}
|
|
|
|
ULONG STDMETHODCALLTYPE Release() {
|
|
uint32_t refCount = --this->m_refCount;
|
|
if (unlikely(!refCount))
|
|
this->GetParentInterface()->Release();
|
|
|
|
return refCount;
|
|
}
|
|
|
|
private:
|
|
|
|
std::atomic<uint32_t> m_refCount = { 0u };
|
|
|
|
};
|
|
|
|
}
|