Question

为了简单起见，我正在使用我机器上提供的H264编码器MFT软件编写H.264编码器（与我最终将使用的英特尔QSV硬件编码器相反）。按照一些教程，我将以下代码放在一起：

IMFTransform* encoder;
DWORD numInputs, numOutputs;
DWORD *inputIDs, *outputIDs;
MFRatio fps = { 24, 1 }, par = { 1, 1 };
HRESULT hr;

hr = CoInitializeEx(NULL, COINIT_APARTMENTTHREADED);
if (FAILED(hr)) {
    return hr;
}

// Create the MFT
hr = FindVideoEncoder(false, true, true, &encoder);
if (FAILED(hr))
{
    return hr;
}

// Obtain number of streams attached to MFT
encoder->GetStreamCount(&numInputs, &numOutputs);

// Allocate appropriate size arrays
inputIDs = new DWORD[numInputs];
outputIDs = new DWORD[numOutputs];

// Obtain stream IDs
encoder->GetStreamIDs(numInputs, inputIDs, numOutputs, outputIDs);

// Set input and output types
IMFMediaType *inputType, *outputType;

CreateVideoType(FRAME_WIDTH, FRAME_HEIGHT, MFVideoInterlace_Unknown, fps, par,
    &outputType, MFMediaType_Video, MFVideoFormat_H264);
hr = encoder->SetOutputType(outputIDs[0], outputType, 0);
if (FAILED(hr))
{
    return hr;
}

CreateVideoType(FRAME_WIDTH, FRAME_HEIGHT, MFVideoInterlace_Unknown, fps, par,
    &inputType, MFMediaType_Video, MFVideoFormat_Base);
hr = encoder->SetInputType(inputIDs[0], inputType, 0);
if (FAILED(hr))
{
    return hr;
}

代码在以下行失败，HRESULT值为“提供的流号无效”。

hr = encoder->SetOutputType(outputIDs[0], outputType, 0);

我试图在SetOutputType之前调用SetInputType但是它会产生相同的错误。我究竟做错了什么？提前感谢您的答案。

编辑：添加了CreateVideoType函数以显示如何创建outputType和inputType

HRESULT CreateVideoType(
    UINT32               width,
    UINT32               height,
    MFVideoInterlaceMode interlaceMode,
    const MFRatio&       frameRate,
    const MFRatio&       par,
    IMFMediaType         **ppType,
    GUID                 majorType,
    GUID                 subtype
    )
{
    if (ppType == NULL)
    {
        return E_POINTER;
    }

    LONG    lStride = 0;
    UINT    cbImage = 0;

    IMFMediaType *pType = NULL;

    // Set the subtype GUID from the FOURCC or D3DFORMAT value.
    //subtype.Data1 = D3DFMT_A8R8G8B8;

    if (FAILED(MFCreateMediaType(&pType)))
    {
        goto done;
    }

    if (FAILED(pType->SetGUID(MF_MT_MAJOR_TYPE, majorType)))
    {
        goto done;
    }

    if (FAILED(pType->SetGUID(MF_MT_SUBTYPE, subtype)))
    {
        goto done;
    }

    if (FAILED(pType->SetUINT32(MF_MT_INTERLACE_MODE, interlaceMode)))
    {
        goto done;
    }


    if (FAILED(MFSetAttributeSize(pType, MF_MT_FRAME_SIZE, width, height)))
    {
        goto done;
    }

    // Calculate the default stride value.
    if (FAILED(pType->SetUINT32(MF_MT_DEFAULT_STRIDE, UINT32(lStride))))
    {
        goto done;
    }

    // Calculate the image size in bytes.
    if (FAILED(MFCalculateImageSize(subtype, width, height, &cbImage)))
    {
        goto done;
    }

    if (FAILED(pType->SetUINT32(MF_MT_SAMPLE_SIZE, cbImage)))
    {
        goto done;
    }

    if (FAILED(pType->SetUINT32(MF_MT_FIXED_SIZE_SAMPLES, TRUE)))
    {
        goto done;
    }

    if (FAILED(pType->SetUINT32(MF_MT_ALL_SAMPLES_INDEPENDENT, TRUE)))
    {
        goto done;
    }

    // Frame rate
    if (FAILED(MFSetAttributeRatio(pType, MF_MT_FRAME_RATE, frameRate.Numerator, frameRate.Denominator)))
    {
        goto done;
    }

    // Pixel aspect ratio   
    if (FAILED(MFSetAttributeRatio(pType, MF_MT_PIXEL_ASPECT_RATIO, par.Numerator, par.Denominator)))
    {
        goto done;
    }

    // Return the pointer to the caller.
    *ppType = pType;
    (*ppType)->AddRef();

done:
    SafeRelease(&pType);
    return S_OK;
}

Answer 1

Microsoft H264编码器MFT未实现GetStreamIDs - 它返回E_NOTIMPL。您必须将输入和输出流ID设置为0。

Answer 2

正如Evgeny在讨论中指出的那样，在代码工作之前有几个问题需要解决。将inputID和outputID更改为0可以解决SetInputType和SetOutputType来电，但视频类型的创建存在一些问题。

首先输出类型创建：

CreateH264VideoType(FRAME_WIDTH, FRAME_HEIGHT, MFVideoInterlace_Unknown, fps, par,
        &outputType, MFMediaType_Video, MFVideoFormat_H264);

在我删除的函数CreateH264VideoType中

 // Calculate the image size in bytes.

if (FAILED(MFCalculateImageSize(subtype, width, height, &cbImage)))
{
    goto done;
}

我不得不添加

// Profile
if (FAILED(pType->SetUINT32(MF_MT_MPEG2_PROFILE, eAVEncH264VProfile_Base)))
{
    goto done;
}

// Bitrate
if (FAILED(pType->SetUINT32(MF_MT_AVG_BITRATE, 166723584)))
{
    goto done;
}

对于CreateUncompressedVideoType，我删除了MFCalculateImageSize并添加了比特率属性集调用。这解决了编码器初始化的错误。

Media Foundation SetInputType返回“提供的流编号无效”

2 个答案: