interface_query.hpp

/**
 * Copyright (C) 2015 Patrick Mours. All rights reserved.
 * License: https://github.com/crosire/d3d8to9#license
 */

#pragma once

#include <algorithm>
#include <unordered_map>

class Direct3D8;
class Direct3DDevice8;
class Direct3DSwapChain8;
class Direct3DTexture8;
class Direct3DCubeTexture8;
class Direct3DVolumeTexture8;
class Direct3DSurface8;
class Direct3DVolume8;
class Direct3DVertexBuffer8;
class Direct3DIndexBuffer8;

class AddressLookupTable
{
	template <typename T>
	struct AddressCacheIndex;


public:
	explicit AddressLookupTable(Direct3DDevice8 *Device) : Device(Device) {}
	~AddressLookupTable();

	template <typename T>
	T *FindAddress(void *pAddress9)
	{
		if (pAddress9 == nullptr)
			return nullptr;

		auto &cache = AddressCache[AddressCacheIndex<T>::CacheIndex];

		const auto it = cache.find(pAddress9);
		if (it != cache.end())
			return static_cast<T *>(it->second);

		return new T(Device, static_cast<typename AddressCacheIndex<T>::Type9 *>(pAddress9));
	}

	template <typename T>
	void SaveAddress(T *pAddress8, void *pAddress9)
	{
		if (pAddress8 == nullptr || pAddress9 == nullptr)
			return;

		AddressCache[AddressCacheIndex<T>::CacheIndex][pAddress9] = pAddress8;
	}

	template <typename T>
	void DeleteAddress(T *pAddress8)
	{
		if (pAddress8 == nullptr)
			return;

		auto &cache = AddressCache[AddressCacheIndex<T>::CacheIndex];

		const auto it = std::find_if(cache.begin(), cache.end(),
			[pAddress8](std::pair<void *, class AddressLookupTableObject *> Map) -> bool { return Map.second == pAddress8; });
		if (it != cache.end())
			cache.erase(it);
	}

private:
	Direct3DDevice8 *const Device;
	std::unordered_map<void *, class AddressLookupTableObject *> AddressCache[8];
};

template <>
struct AddressLookupTable::AddressCacheIndex<Direct3DSurface8>
{ static constexpr UINT CacheIndex = 0; using Type9 = IDirect3DSurface9; };
template <>
struct AddressLookupTable::AddressCacheIndex<Direct3DTexture8>
{ static constexpr UINT CacheIndex = 1; using Type9 = IDirect3DTexture9; };
template <>
struct AddressLookupTable::AddressCacheIndex<Direct3DVolumeTexture8>
{ static constexpr UINT CacheIndex = 2; using Type9 = IDirect3DVolumeTexture9; };
template <>
struct AddressLookupTable::AddressCacheIndex<Direct3DCubeTexture8>
{ static constexpr UINT CacheIndex = 3; using Type9 = IDirect3DCubeTexture9; };
template <>
struct AddressLookupTable::AddressCacheIndex<Direct3DVolume8>
{ static constexpr UINT CacheIndex = 4; using Type9 = IDirect3DVolume9; };
template <>
struct AddressLookupTable::AddressCacheIndex<Direct3DVertexBuffer8>
{ static constexpr UINT CacheIndex = 5; using Type9 = IDirect3DVertexBuffer9; };
template <>
struct AddressLookupTable::AddressCacheIndex<Direct3DIndexBuffer8>
{ static constexpr UINT CacheIndex = 6; using Type9 = IDirect3DIndexBuffer9; };
template <>
struct AddressLookupTable::AddressCacheIndex<Direct3DSwapChain8>
{ static constexpr UINT CacheIndex = 7; using Type9 = IDirect3DSwapChain9; };

class AddressLookupTableObject
{
public:
	virtual ~AddressLookupTableObject() {}

	void DeleteMe()
	{
		delete this;
	}
};

void GenericQueryInterface(REFIID riid, LPVOID *ppvObj, class Direct3DDevice8 *pDevice);

REFIID ConvertREFIID(REFIID riid);