obs-studio源碼分析(一):窗口伶跷、顯示器(桌面)捕捉

windows下窗口辱揭、桌面捕捉有多種方式扁凛。如下:

1:使用GDI函數(shù)或者Windows Media API函數(shù)
2:使用DirectX技術(shù)
3:使用api hook技術(shù)(如D3D游戲捕捉)
4:使用圖形驅(qū)動(dòng)技術(shù)

obs的窗口、桌面捕捉在win-capture插件中裁赠。它使用到的是GDI技術(shù)殿漠。
相比于api hook或者圖形驅(qū)動(dòng)技術(shù),GDI在效率與性能方便確實(shí)略有差距佩捞。但確實(shí)不失為一種最簡(jiǎn)單的方法绞幌。
它的本質(zhì)就是抓取窗口或桌面快照(其實(shí)桌面也是一種特殊的窗口,也包含窗口句柄)
一忱。只要有句柄莲蜘,我們就可以得到了設(shè)備上下文(DC)谭确,就可以利用blit(復(fù)制)它的內(nèi)容到我們創(chuàng)建的DC中。

步驟如下:

1:上層枚舉窗口或桌面票渠,將選中的窗口名等屬性傳到此插件中逐哈。

static void update_settings(struct window_capture *wc, obs_data_t *s)

2:根據(jù)參數(shù)找到對(duì)應(yīng)的窗口或桌面。

HWND find_window(enum window_search_mode mode,
        enum window_priority priority,
        const char *class,
        const char *title,
        const char *exe)

3:libobs底層不斷調(diào)用static void wc_tick(void *data, float seconds)來獲取數(shù)據(jù)
问顷,而此函數(shù)調(diào)用了如下函數(shù)

//創(chuàng)建一個(gè)設(shè)備兼容的DC  
capture->hdc = CreateCompatibleDC(NULL);
//創(chuàng)建可以直接寫入的鞠眉、與設(shè)備無關(guān)的位圖(DIB)
capture->bmp = CreateDIBSection(capture->hdc, &bi, DIB_RGB_COLORS, (void**)&capture->bits, NULL, 0);
//將位圖選入到DC
capture->old_bmp = SelectObject(capture->hdc, capture->bmp);
//取得桌面窗口的DC
hdc_target = GetDC(window);
//將桌面窗口DC的圖象復(fù)制到兼容DC中   
BitBlt(hdc, 0, 0, capture->width, capture->height,hdc_target, capture->x, capture->y, SRCCOPY);
ReleaseDC(NULL, hdc_target);

5:釋放創(chuàng)建的對(duì)象

SelectObject(capture->hdc, capture->old_bmp);
DeleteDC(capture->hdc);
DeleteObject(capture->bmp);

6:當(dāng)然很多時(shí)候需要捕捉鼠標(biāo),需要在此位圖上畫一個(gè)icon

static void draw_cursor(struct dc_capture *capture, HDC hdc, HWND window)
保存圖像

有些特殊需求獲取捕捉的原始圖像择诈,數(shù)據(jù)保存在capture->bits中,是rgb數(shù)據(jù)出皇。當(dāng)然也可以直接保存為bmp圖片羞芍。如下(例子只保存5張,否則太多):

int nCount = 0;
void WriteBmp(void *data, int width, int height)
{
    if (nCount > 5)
        return;

    FILE *pFile = NULL;
    BITMAPFILEHEADER bmpheader;
    BITMAPINFO bmpinfo;

    char fileName[32];
    int bpp = 32;
    // open file
    sprintf(fileName, "frame%d.bmp", nCount);
    pFile = fopen(fileName, "wb");
    if (!pFile)
        return;

    bmpheader.bfType = ('M' << 8) | 'B';
    bmpheader.bfReserved1 = 0;
    bmpheader.bfReserved2 = 0;
    bmpheader.bfOffBits = sizeof(BITMAPFILEHEADER) + sizeof(BITMAPINFOHEADER);
    bmpheader.bfSize = bmpheader.bfOffBits + width * height * bpp / 8;

    bmpinfo.bmiHeader.biSize = sizeof(BITMAPINFOHEADER);
    bmpinfo.bmiHeader.biWidth = width;
    bmpinfo.bmiHeader.biHeight = -height; //reverse the image
    bmpinfo.bmiHeader.biPlanes = 1;
    bmpinfo.bmiHeader.biBitCount = bpp;
    bmpinfo.bmiHeader.biCompression = BI_RGB;
    bmpinfo.bmiHeader.biSizeImage = 0;
    bmpinfo.bmiHeader.biXPelsPerMeter = 100;
    bmpinfo.bmiHeader.biYPelsPerMeter = 100;
    bmpinfo.bmiHeader.biClrUsed = 0;
    bmpinfo.bmiHeader.biClrImportant = 0;

    fwrite(&bmpheader, sizeof(BITMAPFILEHEADER), 1, pFile);
    fwrite(&bmpinfo.bmiHeader, sizeof(BITMAPINFOHEADER), 1, pFile);
    uint8_t *buffer = (uint8_t *)data;

    fwrite(buffer, 1, width * height * bpp / 8, pFile);

    fclose(pFile);
    nCount++;
}
值得注意的是:

在桌面或窗口捕捉時(shí)郊艘,記得設(shè)置多適配器兼容模式荷科,否則你回經(jīng)常發(fā)現(xiàn)抓的數(shù)據(jù)為空,即為黑色纱注。

上層調(diào)用

既然知道了底層實(shí)現(xiàn)畏浆,那上層如何調(diào)用呢?

    ui->pCBoxWindows->clear();
    //創(chuàng)建窗口捕捉資源
    OBSSource source = obs_source_create("window_capture", "test", NULL, nullptr);
    if (source) {
        //添加到指定場(chǎng)景中
        OBSSceneItem sceneitem;
        sceneitem = obs_scene_add(scene, source);
        obs_sceneitem_set_visible(sceneitem, true);
    }
    //通過source獲取屬性
    obs_properties_t *pProperties = obs_source_properties(source);
    obs_property_t *property = obs_properties_first(pProperties);
    while (property) {
        //獲取屬性名字
        const char *name = obs_property_name(property);
        //屬性為窗口
        if (strcmp("window", name) == 0){
            obs_combo_format format = obs_property_list_format(property);
            size_t           count = obs_property_list_item_count(property);
            for (size_t i = 0; i < count; i++){
                //獲取窗口名字
                const char *name = obs_property_list_item_name(property, i);
                QVariant var;
                if (format == OBS_COMBO_FORMAT_STRING) {
                    var = obs_property_list_item_string(property, i);
                    ui->pCBoxWindows->addItem(QString::fromUtf8(name), var);
                }
            }
        }
        obs_property_next(&property);
    }
    OBSData settings = obs_source_get_settings(source);
    QString strWindow = ui->pCBoxWindows->currentData().toString();
    obs_data_set_string(settings, "window", strWindow.toLocal8Bit().data());
    obs_data_set_int(settings, "priority", 0);
    //設(shè)置多適配器兼容
    obs_data_set_bool(settings, "compatibility", true);
    //更新設(shè)置狞贱,只有更新刻获,底層才會(huì)生效
    obs_source_update(source, settings);
    obs_properties_destroy(pProperties);

以上就是obs中Windows窗口,桌面捕捉的實(shí)現(xiàn)瞎嬉。mac的原理類似蝎毡,只是使用了不同的系統(tǒng)API。

?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請(qǐng)聯(lián)系作者
  • 序言:七十年代末氧枣,一起剝皮案震驚了整個(gè)濱河市沐兵,隨后出現(xiàn)的幾起案子,更是在濱河造成了極大的恐慌便监,老刑警劉巖扎谎,帶你破解...
    沈念sama閱讀 222,627評(píng)論 6 517
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件,死亡現(xiàn)場(chǎng)離奇詭異烧董,居然都是意外死亡毁靶,警方通過查閱死者的電腦和手機(jī),發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 95,180評(píng)論 3 399
  • 文/潘曉璐 我一進(jìn)店門解藻,熙熙樓的掌柜王于貴愁眉苦臉地迎上來老充,“玉大人,你說我怎么就攤上這事螟左》茸牵” “怎么了觅够?”我有些...
    開封第一講書人閱讀 169,346評(píng)論 0 362
  • 文/不壞的土叔 我叫張陵,是天一觀的道長(zhǎng)巷嚣。 經(jīng)常有香客問我喘先,道長(zhǎng),這世上最難降的妖魔是什么廷粒? 我笑而不...
    開封第一講書人閱讀 60,097評(píng)論 1 300
  • 正文 為了忘掉前任窘拯,我火速辦了婚禮,結(jié)果婚禮上坝茎,老公的妹妹穿的比我還像新娘涤姊。我一直安慰自己,他們只是感情好嗤放,可當(dāng)我...
    茶點(diǎn)故事閱讀 69,100評(píng)論 6 398
  • 文/花漫 我一把揭開白布思喊。 她就那樣靜靜地躺著,像睡著了一般次酌。 火紅的嫁衣襯著肌膚如雪恨课。 梳的紋絲不亂的頭發(fā)上,一...
    開封第一講書人閱讀 52,696評(píng)論 1 312
  • 那天岳服,我揣著相機(jī)與錄音剂公,去河邊找鬼。 笑死吊宋,一個(gè)胖子當(dāng)著我的面吹牛纲辽,可吹牛的內(nèi)容都是我干的。 我是一名探鬼主播璃搜,決...
    沈念sama閱讀 41,165評(píng)論 3 422
  • 文/蒼蘭香墨 我猛地睜開眼文兑,長(zhǎng)吁一口氣:“原來是場(chǎng)噩夢(mèng)啊……” “哼!你這毒婦竟也來了腺劣?” 一聲冷哼從身側(cè)響起绿贞,我...
    開封第一講書人閱讀 40,108評(píng)論 0 277
  • 序言:老撾萬榮一對(duì)情侶失蹤,失蹤者是張志新(化名)和其女友劉穎橘原,沒想到半個(gè)月后籍铁,有當(dāng)?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體,經(jīng)...
    沈念sama閱讀 46,646評(píng)論 1 319
  • 正文 獨(dú)居荒郊野嶺守林人離奇死亡趾断,尸身上長(zhǎng)有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點(diǎn)故事閱讀 38,709評(píng)論 3 342
  • 正文 我和宋清朗相戀三年拒名,在試婚紗的時(shí)候發(fā)現(xiàn)自己被綠了。 大學(xué)時(shí)的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片芋酌。...
    茶點(diǎn)故事閱讀 40,861評(píng)論 1 353
  • 序言:一個(gè)原本活蹦亂跳的男人離奇死亡增显,死狀恐怖,靈堂內(nèi)的尸體忽然破棺而出脐帝,到底是詐尸還是另有隱情同云,我是刑警寧澤糖权,帶...
    沈念sama閱讀 36,527評(píng)論 5 351
  • 正文 年R本政府宣布,位于F島的核電站炸站,受9級(jí)特大地震影響星澳,放射性物質(zhì)發(fā)生泄漏。R本人自食惡果不足惜旱易,卻給世界環(huán)境...
    茶點(diǎn)故事閱讀 42,196評(píng)論 3 336
  • 文/蒙蒙 一禁偎、第九天 我趴在偏房一處隱蔽的房頂上張望。 院中可真熱鬧阀坏,春花似錦如暖、人聲如沸。這莊子的主人今日做“春日...
    開封第一講書人閱讀 32,698評(píng)論 0 25
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽。三九已至浸船,卻和暖如春,著一層夾襖步出監(jiān)牢的瞬間寝蹈,已是汗流浹背李命。 一陣腳步聲響...
    開封第一講書人閱讀 33,804評(píng)論 1 274
  • 我被黑心中介騙來泰國(guó)打工, 沒想到剛下飛機(jī)就差點(diǎn)兒被人妖公主榨干…… 1. 我叫王不留箫老,地道東北人封字。 一個(gè)月前我還...
    沈念sama閱讀 49,287評(píng)論 3 379
  • 正文 我出身青樓,卻偏偏與公主長(zhǎng)得像耍鬓,于是被迫代替她去往敵國(guó)和親阔籽。 傳聞我的和親對(duì)象是個(gè)殘疾皇子,可洞房花燭夜當(dāng)晚...
    茶點(diǎn)故事閱讀 45,860評(píng)論 2 361

推薦閱讀更多精彩內(nèi)容