使用D3D渲染YUV_RGB

下面我们封装一个类,使用d3d实现yuv,rgb最简单的渲染功能。只需传窗口句柄,数据大小即可完成初始化,然后就可以按照自己的频率往里送数据。

//JYD3DRender.h
#include <stdio.h>
#include <tchar.h>
#include <d3d9.h>

class JYD3DRender{

public:
    JYD3DRender();
    ~JYD3DRender();

public:
    int Init(HWND hwnd, unsigned int nWidth, unsigned int nHeight, bool isYuv);
    void Cleanup();
    bool Render(char *buffer);

private:
    bool                    m_bIsYuv;
    int                     m_nWidth;
    int                     m_nHeight;
    RECT                    m_rtViewport;
    CRITICAL_SECTION        m_critial;
    IDirect3D9              *m_pDirect3D9;
    IDirect3DDevice9        *m_pDirect3DDevice;
    IDirect3DSurface9       *m_pDirect3DSurfaceRender;
};

//JYD3DRender.cpp
#include "JYD3DRender.h"

JYD3DRender::JYD3DRender()
{
    m_pDirect3D9 = NULL;
    m_pDirect3DDevice = NULL;
    m_pDirect3DSurfaceRender = NULL;
}
JYD3DRender::~JYD3DRender()
{
}


int JYD3DRender::Init(HWND hwnd, unsigned int nWidth, unsigned int nHeight, bool isYuv){

    HRESULT lRet;
    InitializeCriticalSection(&m_critial);
    Cleanup();

    m_pDirect3D9 = Direct3DCreate9(D3D_SDK_VERSION);
    if (m_pDirect3D9 == NULL)
        return -1;

    D3DPRESENT_PARAMETERS d3dpp;
    ZeroMemory(&d3dpp, sizeof(d3dpp));
    d3dpp.Windowed = TRUE;
    d3dpp.SwapEffect = D3DSWAPEFFECT_DISCARD;
    d3dpp.BackBufferFormat = D3DFMT_UNKNOWN;

    GetClientRect(hwnd, &m_rtViewport);

    lRet = m_pDirect3D9->CreateDevice(D3DADAPTER_DEFAULT, D3DDEVTYPE_HAL, hwnd, D3DCREATE_SOFTWARE_VERTEXPROCESSING, &d3dpp, &m_pDirect3DDevice);
    if (FAILED(lRet))
        return -1;

    if (isYuv){
        lRet = m_pDirect3DDevice->CreateOffscreenPlainSurface(nWidth, nHeight, (D3DFORMAT)MAKEFOURCC('Y', 'V', '1', '2'), D3DPOOL_DEFAULT, &m_pDirect3DSurfaceRender, NULL);
        if (FAILED(lRet))
            return -1;
    }
    else{
        lRet = m_pDirect3DDevice->CreateOffscreenPlainSurface(nWidth, nHeight, D3DFMT_X8R8G8B8, D3DPOOL_DEFAULT, &m_pDirect3DSurfaceRender, NULL);
        if (FAILED(lRet))
            return -1;
    }
    
    m_nWidth = nWidth;
    m_nHeight = nHeight;
    m_bIsYuv = isYuv;

    return 0;
}

void JYD3DRender::Cleanup()
{
    EnterCriticalSection(&m_critial);
    if (m_pDirect3DSurfaceRender)
        m_pDirect3DSurfaceRender->Release();
    if (m_pDirect3DDevice)
        m_pDirect3DDevice->Release();
    if (m_pDirect3D9)
        m_pDirect3D9->Release();
    LeaveCriticalSection(&m_critial);
}


bool JYD3DRender::Render(char *buffer){

    if (!m_pDirect3DSurfaceRender || !buffer)
        return false;

    HRESULT lRet;
    D3DLOCKED_RECT d3d_rect;
    lRet = m_pDirect3DSurfaceRender->LockRect(&d3d_rect, NULL, D3DLOCK_DONOTWAIT);
    if (FAILED(lRet))
        return false;

    byte *pSrc = (byte *)buffer;
    byte * pDest = (BYTE *)d3d_rect.pBits;
    int stride = d3d_rect.Pitch;

    unsigned long i = 0;
    if (m_bIsYuv){  //如果送入的数据是YUV420P,需要注意它与YV12的区别(U,V的位置刚好相反)
        for (i = 0; i < m_nHeight; i++){
            memcpy(pDest + i * stride, pSrc + i * m_nWidth, m_nWidth);
        }
        for (i = 0; i < m_nHeight / 2; i++){
            memcpy(pDest + stride * m_nHeight + i * stride / 2, pSrc + m_nWidth * m_nHeight * 5 / 4 + i * m_nWidth / 2, m_nWidth / 2);
        }
        for (i = 0; i < m_nHeight / 2; i++){
            memcpy(pDest + stride * m_nHeight + stride * m_nHeight / 4 + i * stride / 2, pSrc + m_nWidth * m_nHeight + i * m_nWidth / 2, m_nWidth / 2);
        }
    }
    else{
        int pixel_w_size = m_nWidth * 4;
        for (i = 0; i< m_nHeight; i++){
            memcpy(pDest, pSrc, pixel_w_size);
            pDest += stride;
            pSrc += pixel_w_size;
        }
    }

    lRet = m_pDirect3DSurfaceRender->UnlockRect();
    if (FAILED(lRet))
        return false;

    if (m_pDirect3DDevice == NULL)
        return false;

    m_pDirect3DDevice->Clear(0, NULL, D3DCLEAR_TARGET, D3DCOLOR_XRGB(0, 0, 0), 1.0f, 0);
    m_pDirect3DDevice->BeginScene();
    IDirect3DSurface9 * pBackBuffer = NULL;

    m_pDirect3DDevice->GetBackBuffer(0, 0, D3DBACKBUFFER_TYPE_MONO, &pBackBuffer);
    m_pDirect3DDevice->StretchRect(m_pDirect3DSurfaceRender, NULL, pBackBuffer, &m_rtViewport, D3DTEXF_LINEAR);
    m_pDirect3DDevice->EndScene();
    m_pDirect3DDevice->Present(NULL, NULL, NULL, NULL);
    pBackBuffer->Release();

    return true;
}

下面是一个使用方法,在这里我用Qt窗口句柄

#include "JYD3DRender.h"
#include <windows.h>
#include <process.h>

#define WIDTH 1280
#define HEIGHT 720

static void video_frame_yuv(void *opaque)
{
    JYD3DRender pD3DRender;
    pD3DRender.Init((HWND)opaque, WIDTH, HEIGHT, true);

    FILE *fileYuv = fopen("1280x720.yuv", "rb");
    if (!fileYuv)
        return;
    int size = WIDTH * HEIGHT * 3 / 2;
    char *buffer = (char *)malloc(size * 3 / 2);

    while (1){
        if (fread(buffer, 1, size, fileYuv) != size){
            fseek(fileYuv, 0, 0);
            fread(buffer, 1, size, fileYuv);
        }
        pD3DRender.Render(buffer);
        Sleep(30);
    }
    fclose(fileYuv);
    fileYuv = NULL;
}

static void video_frame_rgb(void *opaque)
{
    JYD3DRender pD3DRender;
    pD3DRender.Init((HWND)opaque, WIDTH, HEIGHT, true);

    FILE *fileRGB = fopen("1280x720.rgb", "rb");
    if (!fileRGB)
        return;
    int size = WIDTH * HEIGHT * 4;
    char *buffer = (char *)malloc(size * 4);

    while (1){
        if (fread(buffer, 1, size, fileRGB) != size){
            fseek(fileRGB, 0, 0);
            fread(buffer, 1, size, fileRGB);
        }
        pD3DRender.Render(buffer);
        Sleep(30);
    }
    fclose(fileRGB);
    fileRGB = NULL;
}

int main(int argc, char **argv)
{
    QApplication a(argc, argv);

    QWidget *w = new QWidget();
    w->show();

    _beginthreadex(NULL, 0, (unsigned int(_stdcall*)(void*))video_frame_yuv, (void *)w->winId(), 0, NULL);

    return a.exec();
}

程序设置记得添加 D3D9.lib

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 204,445评论 6 478
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 85,889评论 2 381
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 151,047评论 0 337
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 54,760评论 1 276
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 63,745评论 5 367
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 48,638评论 1 281
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 38,011评论 3 398
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 36,669评论 0 258
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 40,923评论 1 299
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 35,655评论 2 321
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 37,740评论 1 330
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 33,406评论 4 320
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 38,995评论 3 307
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 29,961评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 31,197评论 1 260
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 45,023评论 2 350
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 42,483评论 2 342

推荐阅读更多精彩内容