D3D11直接渲染YUV

#include "D3D11Render.h" #include "libyuv.h" #include <assert.h> #pragma comment(lib, "libyuv.lib") #define SAFE_RELEASE(p) { if((p)){ (p)->Release(); (p) = nullptr; }} #define USE_I420 const D3D11_INPUT_ELEMENT_DESC D3D11Render::VertexPosColor::inputLayout[2] = { {"POSITION", 0, DXGI_FORMAT_R32G32B32_FLOAT, 0, 0, D3D11_INPUT_PER_VERTEX_DATA, 0}, {"TEXCOORD", 0, DXGI_FORMAT_R32G32_FLOAT, 0, 12, D3D11_INPUT_PER_VERTEX_DATA, 0} }; void LogErrorText(HRESULT hr) { WCHAR strBufferError[300]; FormatMessageW(FORMAT_MESSAGE_FROM_SYSTEM | FORMAT_MESSAGE_IGNORE_INSERTS, nullptr, hr, MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT), strBufferError, 256, nullptr); std::cout << strBufferError << std::endl; } uint64_t get_timestamp1() { // struct timeval tm; // gettimeofday(&tm, NULL); // return (uint64_t)(tm.tv_sec*1000+tm.tv_usec/1000); LARGE_INTEGER counter; static LARGE_INTEGER freq = { 0 }; QueryPerformanceFrequency(&freq); QueryPerformanceCounter(&counter); return (counter.QuadPart * 1000) / (freq.QuadPart); } D3D11Render::D3D11Render() : m_width(0), m_height(0), m_orientation(0), m_fps(30), m_run(false), render_thread_(nullptr), m_renderBuffer(nullptr), m_Enable4xMsaa(false), m_hwnd(nullptr) { } D3D11Render::D3D11Render(int width, int height, char filepath[100]) : m_width(width), m_height(height), m_orientation(0), m_fps(30), m_run(false), render_thread_(nullptr), m_renderBuffer(nullptr), m_Enable4xMsaa(false), m_hwnd(nullptr) { memcpy(path, filepath, 100); } D3D11Render::~D3D11Render() { if (m_renderBuffer != nullptr) free(m_renderBuffer); if (m_pd3dImmediateContex) { m_pd3dImmediateContex->ClearState(); //恢复D3D设备上下文到默认状态 } } //int D3D11Render::InputMediaFrame(uint8_t* pFrame, int length, int width, int height, int orientation) //{ // if (pFrame == nullptr) { // std::cout << "pFrame is NULL." << std::endl; // return -1; // } // if (width <= 0 || height <= 0) { // std::cout << "width is:" << width << " height is:" << height << std::endl; // return -1; // } // uint8_t* frame = nullptr; // frame = new uint8_t[width * height * 3 / 2]; // std::unique_lock<std::mutex> sbguard1(m_Mutex); // if (frame == nullptr) { // return -1; // } // else { // if (width != m_width || height != m_height) { // int ret = OnResize(); // if (ret != 0) { // std::cout << "OnResize failed." << std::endl; // return -1; // } // while (!data.empty()) { // uint8_t* frame = data.front(); // data.pop_front(); // delete frame; // frame = nullptr; // } // m_width = width; // m_height = height; // m_orientation = orientation; // memcpy(frame, pFrame, width * height * 3 / 2); // data.push_back(frame); // return 0; // } // else { // memcpy(frame, pFrame, width * height * 3 / 2); // data.push_back(frame); // return 0; // } // } // return 0; //} int D3D11Render::InputMediaFrame1(uint8_t* pFrame, int length, int width, int height, int q_num) { if (pFrame == nullptr) { std::cout << "pFrame is NULL." << std::endl; return -1; } if (width <= 0 || height <= 0) { std::cout << "width is:" << width << " height is:" << height << std::endl; return -1; } std::unique_lock<std::mutex> sbguard(m_Mutex[0]); uint8_t* frame = new uint8_t[width * height * 3 / 2]; if (frame == nullptr) { return -1; } if (data[0].size() > 0) { while (data[0].size() > 15) { uint8_t* pd = data[0].front(); data[0].pop_front(); delete pd; pd = nullptr; } } memcpy(frame, pFrame, width * height * 3 / 2); data[0].push_back(frame); m_hasdata = true; return 0; } int D3D11Render::InputMediaFrame2(uint8_t* pFrame, int length, int width, int height, int q_num) { if (pFrame == nullptr) { std::cout << "pFrame is NULL." << std::endl; return -1; } if (width <= 0 || height <= 0) { std::cout << "width is:" << width << " height is:" << height << std::endl; return -1; } std::unique_lock<std::mutex> sbguard(m_Mutex[1]); uint8_t* frame = new uint8_t[width * height * 3 / 2]; if (frame == nullptr) { return -1; } if (data[1].size() > 0) { while (data[1].size() > 15) { uint8_t* pd = data[1].front(); data[1].pop_front(); delete pd; pd = nullptr; } } memcpy(frame, pFrame, width * height * 3 / 2); data[1].push_back(frame); m_hasdata = true; return 0; } int D3D11Render::InputMediaFrame3(uint8_t* pFrame, int length, int width, int height, int q_num) { if (pFrame == nullptr) { std::cout << "pFrame is NULL." << std::endl; return -1; } if (width <= 0 || height <= 0) { std::cout << "width is:" << width << " height is:" << height << std::endl; return -1; } std::unique_lock<std::mutex> sbguard(m_Mutex[2]); uint8_t* frame = new uint8_t[width * height * 3 / 2]; if (frame == nullptr) { return -1; } if (data[2].size() > 0) { while (data[2].size() > 15) { uint8_t* pd = data[2].front(); data[2].pop_front(); delete pd; pd = nullptr; } } memcpy(frame, pFrame, width * height * 3 / 2); data[2].push_back(frame); m_hasdata = true; return 0; } int D3D11Render::InputMediaFrame4(uint8_t* pFrame, int length, int width, int height, int q_num) { if (pFrame == nullptr) { std::cout << "pFrame is NULL." << std::endl; return -1; } if (width <= 0 || height <= 0) { std::cout << "width is:" << width << " height is:" << height << std::endl; return -1; } std::unique_lock<std::mutex> sbguard(m_Mutex[3]); uint8_t* frame = new uint8_t[width * height * 3 / 2]; if (frame == nullptr) { return -1; } if (data[3].size() > 0) { while (data[3].size() > 15) { uint8_t* pd = data[3].front(); data[3].pop_front(); delete pd; pd = nullptr; } } memcpy(frame, pFrame, width * height * 3 / 2); data[3].push_back(frame); m_hasdata = true; return 0; } int D3D11Render::InputMediaFrame5(uint8_t* pFrame, int length, int width, int height, int q_num) { if (pFrame == nullptr) { std::cout << "pFrame is NULL." << std::endl; return -1; } if (width <= 0 || height <= 0) { std::cout << "width is:" << width << " height is:" << height << std::endl; return -1; } std::unique_lock<std::mutex> sbguard(m_Mutex[4]); uint8_t* frame = new uint8_t[width * height * 3 / 2]; if (frame == nullptr) { return -1; } if (data[4].size() > 0) { while (data[4].size() > 15) { uint8_t* pd = data[4].front(); data[4].pop_front(); delete pd; pd = nullptr; } } memcpy(frame, pFrame, width * height * 3 / 2); data[4].push_back(frame); m_hasdata = true; return 0; } int D3D11Render::InputMediaFrame6(uint8_t* pFrame, int length, int width, int height, int q_num) { if (pFrame == nullptr) { std::cout << "pFrame is NULL." << std::endl; return -1; } if (width <= 0 || height <= 0) { std::cout << "width is:" << width << " height is:" << height << std::endl; return -1; } std::unique_lock<std::mutex> sbguard(m_Mutex[5]); uint8_t* frame = new uint8_t[width * height * 3 / 2]; if (frame == nullptr) { return -1; } if (data[5].size() > 0) { while (data[5].size() > 15) { uint8_t* pd = data[5].front(); data[5].pop_front(); delete pd; pd = nullptr; } } memcpy(frame, pFrame, width * height * 3 / 2); data[5].push_back(frame); m_hasdata = true; return 0; } int D3D11Render::InputMediaFrame7(uint8_t* pFrame, int length, int width, int height, int q_num) { if (pFrame == nullptr) { std::cout << "pFrame is NULL." << std::endl; return -1; } if (width <= 0 || height <= 0) { std::cout << "width is:" << width << " height is:" << height << std::endl; return -1; } std::unique_lock<std::mutex> sbguard(m_Mutex[6]); uint8_t* frame = new uint8_t[width * height * 3 / 2]; if (frame == nullptr) { return -1; } if (data[6].size() > 0) { while (data[6].size() > 15