65 lines
1.5 KiB
C
65 lines
1.5 KiB
C
|
#pragma once
|
||
|
|
|
||
|
|
#include <d3d11.h>
|
||
|
|
#include <wrl/client.h>
|
||
|
|
#include <windows.h>
|
||
|
|
#include <cstdint>
|
||
|
|
|
||
|
|
using Microsoft::WRL::ComPtr;
|
||
|
|
|
||
|
|
struct IddSharedHeader {
|
||
|
|
uint32_t width;
|
||
|
|
uint32_t height;
|
||
|
|
uint32_t format; // 0 = BGRA32
|
||
|
|
uint32_t stride; // bytes per row
|
||
|
|
uint64_t timestamp;
|
||
|
|
uint32_t frameId;
|
||
|
|
uint32_t dataSize;
|
||
|
|
};
|
||
|
|
|
||
|
|
class IddBridge {
|
||
|
|
public:
|
||
|
|
IddBridge();
|
||
|
|
~IddBridge();
|
||
|
|
|
||
|
|
bool Initialize();
|
||
|
|
bool CaptureFrame(ComPtr<ID3D11Texture2D>& texture);
|
||
|
|
void ReleaseFrame();
|
||
|
|
ID3D11Device* GetDevice() const { return device_.Get(); }
|
||
|
|
ID3D11DeviceContext* GetContext() const { return context_.Get(); }
|
||
|
|
|
||
|
|
private:
|
||
|
|
bool EnsureTexture(int width, int height);
|
||
|
|
|
||
|
|
ComPtr<ID3D11Device> device_;
|
||
|
|
ComPtr<ID3D11DeviceContext> context_;
|
||
|
|
ComPtr<ID3D11Texture2D> gpuTexture_;
|
||
|
|
bool frame_acquired_ = false;
|
||
|
|
|
||
|
|
HANDLE hMap_ = nullptr;
|
||
|
|
HANDLE hReadyEvent_ = nullptr;
|
||
|
|
HANDLE hConsumedEvent_ = nullptr;
|
||
|
|
uint8_t* shared_ = nullptr;
|
||
|
|
size_t shared_size_ = 0;
|
||
|
|
int tex_width_ = 0;
|
||
|
|
int tex_height_ = 0;
|
||
|
|
};
|
||
|
|
|
||
|
|
class IddProducer {
|
||
|
|
public:
|
||
|
|
IddProducer();
|
||
|
|
~IddProducer();
|
||
|
|
|
||
|
|
bool Initialize();
|
||
|
|
// Submit a frame. Data should be BGRA32.
|
||
|
|
bool SubmitFrame(const void* data, uint32_t width, uint32_t height, uint32_t stride);
|
||
|
|
|
||
|
|
private:
|
||
|
|
HANDLE hMap_ = nullptr;
|
||
|
|
HANDLE hReadyEvent_ = nullptr;
|
||
|
|
HANDLE hConsumedEvent_ = nullptr;
|
||
|
|
uint8_t* shared_ = nullptr;
|
||
|
|
size_t shared_size_ = 0;
|
||
|
|
uint32_t frameId_ = 0;
|
||
|
|
};
|