[C++]DirectShow检测音视频输入设备及其采集参数

官方文档:https://docs.microsoft.com/zh-cn/windows/win32/directshow/directshow

创建CLR类库项目(CSharpDirectShow),编写托管的DirectShow类库,右键项目属性-->链接器--> 输入-->附加依赖项;添加静态库文件Strmiids.lib和Quartz.lib;

 

 

 定义头文件CSharpDirectShow.h,包含头文件dshow.h,定义如下方法:

#pragma once

#include <dshow.h>

using namespace System;
using namespace System::Runtime::InteropServices;

public ref class DirectShow
{
public:
    /// <summary>
    /// 在当前线程上初始化COM库,并将并发模型标识为单线程单元(STA)
    /// </summary>
    /// <returns></returns>
    static Boolean ComInit();
    /// <summary>
    /// 关闭当前线程上的COM库,卸载该线程加载的所有DLL,释放该线程维护的所有其他资源,并强制关闭该线程上的所有RPC连接
    /// </summary>
    static void ComUinit();
    /// <summary>
    /// 获取视频输入设备
    /// </summary>
    /// <param name="devices"></param>
    /// <returns></returns>
    static Int32 GetVideoInputDevices([Out]array<VideoInputDsDevice^>^% devices);
    /// <summary>
    /// 获取音频输入设备
    /// </summary>
    /// <param name="device"></param>
    /// <returns></returns>
    static Int32 GetAudioInputDevices([Out]array<AudioInputDsDevice^>^% device);
};

其中,初始化COM库只能在STA线程调用,控制台程序直接调用会返回失败,在Winform和WPF等应用程序应该在第一次使用时初始化COM库;
当前线程初始化COM库和关闭COM库:

Boolean DirectShow::ComInit()
{
    HRESULT hr = CoInitialize(NULL);
    return hr == S_OK || hr == S_FALSE;
}

void DirectShow::ComUinit()
{
    CoUninitialize();
}

一、枚举视频输入设备,参考:https://docs.microsoft.com/zh-cn/windows/win32/directshow/selecting-a-capture-device

1、创建视频输入设备的枚举器IEnumMoniker

HRESULT DirectShow::EnumerateDevices(REFGUID category, IEnumMoniker** ppEnum)
{
    // 创建系统设备枚举器
    ICreateDevEnum* pDevEnum;
    HRESULT hr = CoCreateInstance(CLSID_SystemDeviceEnum, NULL, CLSCTX_INPROC_SERVER, IID_PPV_ARGS(&pDevEnum));

    if (SUCCEEDED(hr))
    {
        // 为类别创建枚举数.
        hr = pDevEnum->CreateClassEnumerator(category, ppEnum, 0);

        if (hr == S_FALSE)
            hr = VFW_E_NOT_FOUND;  // 类别是空的,视为错误

        pDevEnum->Release();
    }
    return hr;
}

Int32 DirectShow::GetVideoInputDevices([Out]array<VideoInputDsDevice^>^% devices)
{
    devices = nullptr;

    IEnumMoniker* pEnum;
    HRESULT hr = EnumerateDevices(CLSID_VideoInputDeviceCategory, &pEnum);        // 创建视频输入设备枚举器

    if (FAILED(hr))
        return hr;

    hr = EnumerateVideoInputDevices(pEnum, devices);    // 枚举视频输入设备
    pEnum->Release();
    return hr;
}

2、调用IEnumMoniker::Next()方法枚举设备,调用IPropertyBag::Read方法读取设备属性,以获取设备的友好名称和设备标识字符串:

 

HRESULT DirectShow::EnumerateVideoInputDevices(IEnumMoniker* pEnum, [Out]array<VideoInputDsDevice^>^% devices)
{
    HRESULT hr = S_FALSE;
    devices = nullptr;
    List<VideoInputDsDevice^>^ list = gcnew List<VideoInputDsDevice^>();

    IMoniker* pMoniker;
    while (pEnum->Next(1, &pMoniker, NULL) == S_OK)
    {
        IPropertyBag* pPropBag;
        hr = pMoniker->BindToStorage(0, 0, IID_PPV_ARGS(&pPropBag));
        if (FAILED(hr))
        {
            pMoniker->Release();
            continue;
        }

        VideoInputDsDevice^ device = gcnew VideoInputDsDevice();

        VARIANT var;
        VariantInit(&var);

        // 获取设备友好名
        hr = pPropBag->Read(L"FriendlyName", &var, 0);

        if (FAILED(hr))
            hr = pPropBag->Read(L"Description", &var, 0);

        if (SUCCEEDED(hr))
        {
            device->FriendlyName = System::String(var.bstrVal).ToString();
            VariantClear(&var);
        }

        // 获取设备Moniker名
        LPOLESTR pOleDisplayName = reinterpret_cast<LPOLESTR>(CoTaskMemAlloc(MAX_MONIKER_NAME_LENGTH * 2));
        hr = pMoniker->GetDisplayName(NULL, NULL, &pOleDisplayName);

        if (SUCCEEDED(hr))
        {
            device->MonikerName = System::String(pOleDisplayName).ToString();
            array<VideoParams^>^ params;
            hr = EnumerateVideoParams(pMoniker, params);    // 枚举设备的采集参数
            if (SUCCEEDED(hr))
                device->Params = params;
        }

        CoTaskMemFree(pOleDisplayName);

        list->Add(device);
        pPropBag->Release();
        pMoniker->Release();
    }

    devices = list->ToArray();
    return S_OK;
}

 

3、枚举视频输入设备的采集参数:

 

HRESULT DirectShow::EnumerateVideoParams(IMoniker* pMoniker, [Out]array<VideoParams^>^% params)
{
    params = nullptr;
    IBaseFilter* pFilter;
    HRESULT    hr = pMoniker->BindToObject(NULL, NULL, IID_IBaseFilter, (void**)&pFilter);

    if (FAILED(hr))
        return hr;

    IEnumPins* pinEnum;
    hr = pFilter->EnumPins(&pinEnum);

    if (FAILED(hr)) {
        pFilter->Release();
        return hr;
    }

    List<VideoParams^>^ list = gcnew List<VideoParams^>();
    IPin* pPins;
    while (pinEnum->Next(1, &pPins, NULL) == S_OK)
    {
        PIN_INFO pinInfo;
        hr = pPins->QueryPinInfo(&pinInfo);

        if (FAILED(hr) || pinInfo.dir != PINDIR_OUTPUT)
        {
            pPins->Release();
            continue;
        }

        IEnumMediaTypes* mtEnum;
        hr = pPins->EnumMediaTypes(&mtEnum);

        if (FAILED(hr))
        {
            pPins->Release();
            continue;
        }

        AM_MEDIA_TYPE* mt;
        while (mtEnum->Next(1, &mt, NULL) == S_OK)
        {
            VideoParams^ param = nullptr;
            if (mt->formattype == FORMAT_VideoInfo)
            {
                VIDEOINFOHEADER* pVih = reinterpret_cast<VIDEOINFOHEADER*>(mt->pbFormat);
                param = gcnew VideoParams();
                param->FrameWidth = pVih->bmiHeader.biWidth;
                param->FrameHeight = pVih->bmiHeader.biHeight;
                param->AverageFrameRate = pVih->AvgTimePerFrame == 0 ? 0 : 10000000 / pVih->AvgTimePerFrame;
            }
            else if (mt->formattype == FORMAT_VideoInfo2) {
                VIDEOINFOHEADER2* pVih = reinterpret_cast<VIDEOINFOHEADER2*>(mt->pbFormat);
                param = gcnew VideoParams();
                param->FrameWidth = pVih->bmiHeader.biWidth;
                param->FrameHeight = pVih->bmiHeader.biHeight;
                param->AverageFrameRate = pVih->AvgTimePerFrame == 0 ? 0 : 10000000 / pVih->AvgTimePerFrame;
            }

            if (param && param->AverageFrameRate > 1)
            {
                Boolean isExit = false;
                for each (VideoParams ^ item in list)
                {
                    if (item->FrameWidth == param->FrameWidth && item->FrameHeight == param->FrameHeight && item->AverageFrameRate == param->AverageFrameRate)
                    {
                        isExit = true;
                        break;
                    }
                }
                if (!isExit)
                    list->Add(param);
            }
        }
        pPins->Release();
    }

    pFilter->Release();

    params = list->ToArray();
    return S_OK;
}

 

调用代码及结果:

 

var ret = DirectShow.GetVideoInputDevices (out VideoInputDsDevice[] videoInputDevices);

if (ret == 0) {
    Console.WriteLine ("视频输入设备:");

    foreach (var videoInputDevice in videoInputDevices) {
        Console.WriteLine ($"{videoInputDevice.FriendlyName}\t{videoInputDevice.MonikerName}");

        if (videoInputDevice.Params.Length > 0) {
            Console.WriteLine ("像素宽度\t像素高度\t1秒平均帧数");

            foreach (var param in videoInputDevice.Params) {
                Console.WriteLine ($"{param.FrameWidth}\t{param.FrameHeight}\t{param.AverageFrameRate}");
            }
        }

        Console.WriteLine ();
    }
}

 

 

 

二、枚举音频输入设备,参考https://docs.microsoft.com/zh-cn/windows/win32/directshow/selecting-a-capture-device,过程和视频一样:

ret = DirectShow.GetAudioInputDevices (out AudioInputDsDevice[] audioInputDevices);

if (ret == 0) {
    Console.WriteLine ("音频输入设备:");

    foreach (var audioInputDevice in audioInputDevices) {
        Console.WriteLine ($"{audioInputDevice.FriendlyName}\t{audioInputDevice.MonikerName}");

        if (audioInputDevice.Params.Length > 0) {
            Console.WriteLine ("音频格式\t通道数\t采样速率\t块对齐\t位数");

            foreach (var param in audioInputDevice.Params) {
                Console.WriteLine ($"{param.Format}\t{param.Channels}\t{param.SampleRate}\t{param.BlockAlign}\t{param.BitsPerSample}");
            }
        }

        Console.WriteLine ();
    }
}

 

 代码已上传至Github:https://github.com/LowPlayer/CameraCapture

posted @ 2021-01-09 17:56  孤独成派  阅读(855)  评论(0编辑  收藏  举报