obs-studio源码分析(一):窗口、显示器(桌面)捕捉

windows下窗口、桌面捕捉有多种方式。如下:

1:使用GDI函数或者Windows Media API函数
2:使用DirectX技术
3:使用api hook技术(如D3D游戏捕捉)
4:使用图形驱动技术

obs的窗口、桌面捕捉在win-capture插件中。它使用到的是GDI技术。
相比于api hook或者图形驱动技术,GDI在效率与性能方便确实略有差距。但确实不失为一种最简单的方法。
它的本质就是抓取窗口或桌面快照(其实桌面也是一种特殊的窗口,也包含窗口句柄)
。只要有句柄,我们就可以得到了设备上下文(DC),就可以利用blit(复制)它的内容到我们创建的DC中。

步骤如下:

1:上层枚举窗口或桌面,将选中的窗口名等属性传到此插件中。

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

2:根据参数找到对应的窗口或桌面。

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

3:libobs底层不断调用static void wc_tick(void *data, float seconds)来获取数据
,而此函数调用了如下函数

//创建一个设备兼容的DC  
capture->hdc = CreateCompatibleDC(NULL);
//创建可以直接写入的、与设备无关的位图(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的图象复制到兼容DC中   
BitBlt(hdc, 0, 0, capture->width, capture->height,hdc_target, capture->x, capture->y, SRCCOPY);
ReleaseDC(NULL, hdc_target);

5:释放创建的对象

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

6:当然很多时候需要捕捉鼠标,需要在此位图上画一个icon

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

有些特殊需求获取捕捉的原始图像,数据保存在capture->bits中,是rgb数据。当然也可以直接保存为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++;
}
值得注意的是:

在桌面或窗口捕捉时,记得设置多适配器兼容模式,否则你回经常发现抓的数据为空,即为黑色。

上层调用

既然知道了底层实现,那上层如何调用呢?

    ui->pCBoxWindows->clear();
    //创建窗口捕捉资源
    OBSSource source = obs_source_create("window_capture", "test", NULL, nullptr);
    if (source) {
        //添加到指定场景中
        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);
    //设置多适配器兼容
    obs_data_set_bool(settings, "compatibility", true);
    //更新设置,只有更新,底层才会生效
    obs_source_update(source, settings);
    obs_properties_destroy(pProperties);

以上就是obs中Windows窗口,桌面捕捉的实现。mac的原理类似,只是使用了不同的系统API。

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

推荐阅读更多精彩内容