d3d9query.h

#pragma once
#include "d3d9dev.h"
#include <d3d9.h>
#include <wrl/client.h>

interface hkIDirect3DQuery9 : public IDirect3DQuery9 {
  STDMETHOD(QueryInterface)(THIS_ REFIID riid, void** ppvObj);
  STDMETHOD_(ULONG, AddRef)(THIS);
  STDMETHOD_(ULONG, Release)(THIS);
  STDMETHOD(GetDevice)(THIS_ IDirect3DDevice9** ppDevice);
  STDMETHOD_(D3DQUERYTYPE, GetType)(THIS);
  STDMETHOD_(DWORD, GetDataSize)(THIS);
  STDMETHOD(Issue)(THIS_ DWORD dwIssueFlags);
  STDMETHOD(GetData)(THIS_ void* pData, DWORD dwSize, DWORD dwGetDataFlags);

  hkIDirect3DQuery9(IDirect3DQuery9* pIDirect3DQuery9, hkIDirect3DDevice9* device);
  virtual ~hkIDirect3DQuery9() = default;

private:
  Microsoft::WRL::ComPtr<IDirect3DQuery9> m_pD3Dquery;
  Microsoft::WRL::ComPtr<hkIDirect3DDevice9> m_device;
};