// Finds and returns the h264 MFT if available...otherwise fails with style.</para>
HRESULT EncodeTransform::FindEncoder(const GUID& subtype)
{
HRESULT hr = S_OK;
UINT32 count = 0;
CLSID *ppCLSIDs = NULL;
MFT_REGISTER_TYPE_INFO info = { 0 };
info.guidMajorType = MFMediaType_Video;
info.guidSubtype = subtype;
hr = MFTEnum(
MFT_CATEGORY_VIDEO_ENCODER,
0,
NULL,
&info,
NULL,
&ppCLSIDs,
&count
);
if (SUCCEEDED(hr) && count == 0)
{
hr = MF_E_TOPO_CODEC_NOT_FOUND;
}
if (SUCCEEDED(hr))
{
hr = CoCreateInstance(ppCLSIDs[0], NULL, CLSCTX_INPROC_SERVER, IID_PPV_ARGS(&mpEncoder));
}
CoTaskMemFree(ppCLSIDs);
return hr;
}
static void test_register(void)
{
static WCHAR name[] = {'W','i','n','e',' ','t','e','s','t',0};
MFT_REGISTER_TYPE_INFO input[] =
{
{ DUMMY_CLSID, DUMMY_GUID1 }
};
MFT_REGISTER_TYPE_INFO output[] =
{
{ DUMMY_CLSID, DUMMY_GUID2 }
};
CLSID *clsids;
UINT32 count;
HRESULT ret;
ret = MFTRegister(DUMMY_CLSID, MFT_CATEGORY_OTHER, name, 0, 1, input, 1, output, NULL);
if (ret == E_ACCESSDENIED)
{
win_skip("Not enough permissions to register a filter\n");
return;
}
ok(ret == S_OK, "Failed to register dummy filter: %x\n", ret);
if(0)
{
/* NULL name crashes on windows */
ret = MFTRegister(DUMMY_CLSID, MFT_CATEGORY_OTHER, NULL, 0, 1, input, 1, output, NULL);
ok(ret == E_INVALIDARG, "got %x\n", ret);
}
ret = MFTRegister(DUMMY_CLSID, MFT_CATEGORY_OTHER, name, 0, 0, NULL, 0, NULL, NULL);
ok(ret == S_OK, "Failed to register dummy filter: %x\n", ret);
ret = MFTRegister(DUMMY_CLSID, MFT_CATEGORY_OTHER, name, 0, 1, NULL, 0, NULL, NULL);
ok(ret == S_OK, "Failed to register dummy filter: %x\n", ret);
ret = MFTRegister(DUMMY_CLSID, MFT_CATEGORY_OTHER, name, 0, 0, NULL, 1, NULL, NULL);
ok(ret == S_OK, "Failed to register dummy filter: %x\n", ret);
if(0)
{
/* NULL clsids/count crashes on windows (vista) */
count = 0;
ret = MFTEnum(MFT_CATEGORY_OTHER, 0, NULL, NULL, NULL, NULL, &count);
ok(ret == E_POINTER, "Failed to enumerate filters: %x\n", ret);
ok(count == 0, "Expected count > 0\n");
clsids = NULL;
ret = MFTEnum(MFT_CATEGORY_OTHER, 0, NULL, NULL, NULL, &clsids, NULL);
ok(ret == E_POINTER, "Failed to enumerate filters: %x\n", ret);
ok(count == 0, "Expected count > 0\n");
}
count = 0;
clsids = NULL;
ret = MFTEnum(MFT_CATEGORY_OTHER, 0, NULL, NULL, NULL, &clsids, &count);
ok(ret == S_OK, "Failed to enumerate filters: %x\n", ret);
ok(count > 0, "Expected count > 0\n");
ok(clsids != NULL, "Expected clsids != NULL\n");
ok(check_clsid(clsids, count), "Filter was not part of enumeration\n");
CoTaskMemFree(clsids);
count = 0;
clsids = NULL;
ret = MFTEnum(MFT_CATEGORY_OTHER, 0, input, NULL, NULL, &clsids, &count);
ok(ret == S_OK, "Failed to enumerate filters: %x\n", ret);
ok(count > 0, "Expected count > 0\n");
ok(clsids != NULL, "Expected clsids != NULL\n");
ok(check_clsid(clsids, count), "Filter was not part of enumeration\n");
CoTaskMemFree(clsids);
count = 0;
clsids = NULL;
ret = MFTEnum(MFT_CATEGORY_OTHER, 0, NULL, output, NULL, &clsids, &count);
ok(ret == S_OK, "Failed to enumerate filters: %x\n", ret);
ok(count > 0, "Expected count > 0\n");
ok(clsids != NULL, "Expected clsids != NULL\n");
ok(check_clsid(clsids, count), "Filter was not part of enumeration\n");
CoTaskMemFree(clsids);
count = 0;
clsids = NULL;
ret = MFTEnum(MFT_CATEGORY_OTHER, 0, input, output, NULL, &clsids, &count);
ok(ret == S_OK, "Failed to enumerate filters: %x\n", ret);
ok(count > 0, "Expected count > 0\n");
ok(clsids != NULL, "Expected clsids != NULL\n");
ok(check_clsid(clsids, count), "Filter was not part of enumeration\n");
CoTaskMemFree(clsids);
/* exchange input and output */
count = 0;
clsids = NULL;
ret = MFTEnum(MFT_CATEGORY_OTHER, 0, output, input, NULL, &clsids, &count);
ok(ret == S_OK, "Failed to enumerate filters: %x\n", ret);
ok(!count, "got %d\n", count);
ok(clsids == NULL, "Expected clsids == NULL\n");
ret = MFTUnregister(DUMMY_CLSID);
ok(ret == S_OK ||
/* w7pro64 */
broken(ret == HRESULT_FROM_WIN32(ERROR_FILE_NOT_FOUND)), "got %x\n", ret);
ret = MFTUnregister(DUMMY_CLSID);
ok(ret == S_OK || broken(ret == HRESULT_FROM_WIN32(ERROR_FILE_NOT_FOUND)), "got %x\n", ret);
}
HRESULT CASFManager::SetupStreamDecoder (WORD wStreamNumber,
GUID* pguidCurrentMediaType)
{
if (! m_pContentInfo)
{
return MF_E_NOT_INITIALIZED;
}
if (wStreamNumber == 0)
{
return E_INVALIDARG;
}
IMFASFProfile* pProfile = NULL;
IMFMediaType* pMediaType = NULL;
IMFASFStreamConfig *pStream = NULL;
GUID guidMajorType = GUID_NULL;
GUID guidSubType = GUID_NULL;
GUID guidDecoderCategory = GUID_NULL;
BOOL fIsCompressed = TRUE;
CLSID *pDecoderCLSIDs = NULL; // Pointer to an array of CLISDs.
UINT32 cDecoderCLSIDs = 0; // Size of the array.
HRESULT hr = S_OK;
//Get the profile object that stores stream information
CHECK_HR(hr = m_pContentInfo->GetProfile(&pProfile));
//Get stream configuration object from the profile
CHECK_HR(hr = pProfile->GetStreamByNumber(wStreamNumber, &pStream));
//Get the media type
CHECK_HR(hr = pStream->GetMediaType(&pMediaType));
//Get the major media type
CHECK_HR(hr = pMediaType->GetMajorType(&guidMajorType));
//Get the sub media type
CHECK_HR(hr = pMediaType->GetGUID(MF_MT_SUBTYPE, &guidSubType));
//find out if the media type is compressed
CHECK_HR(hr = pMediaType->IsCompressedFormat(&fIsCompressed));
if (fIsCompressed)
{
//get decoder category
if (guidMajorType == MFMediaType_Video)
{
guidDecoderCategory = MFT_CATEGORY_VIDEO_DECODER;
}
else if (guidMajorType == MFMediaType_Audio)
{
guidDecoderCategory = MFT_CATEGORY_AUDIO_DECODER;
}
else
{
CHECK_HR(hr = MF_E_INVALIDMEDIATYPE);
}
// Look for a decoder.
MFT_REGISTER_TYPE_INFO tinfo;
tinfo.guidMajorType = guidMajorType;
tinfo.guidSubtype = guidSubType;
CHECK_HR(hr = MFTEnum(
guidDecoderCategory,
0, // Reserved
&tinfo, // Input type to match. (Encoded type.)
NULL, // Output type to match. (Don't care.)
NULL, // Attributes to match. (None.)
&pDecoderCLSIDs, // Receives a pointer to an array of CLSIDs.
&cDecoderCLSIDs // Receives the size of the array.
));
// MFTEnum can return zero matches.
if (cDecoderCLSIDs == 0)
{
hr = MF_E_TOPO_CODEC_NOT_FOUND;
}
else
{
//if the CDecoder instance does not exist, create one.
if (!m_pDecoder)
{
CHECK_HR(hr = CDecoder::CreateInstance(&m_pDecoder));
}
//Load the first MFT in the array for the current media type
CHECK_HR(hr = m_pDecoder->Initialize(pDecoderCLSIDs[0], pMediaType));
}
*pguidCurrentMediaType = guidMajorType;
}
else
{
// Not compressed. Don't need a decoder.
CHECK_HR(hr = MF_E_INVALIDREQUEST);
}
TRACE((L"Stream decoder loaded.\n"));
done:
LOG_MSG_IF_FAILED(L"CASFManager::SetupStreamDecoder failed.\n", hr);
SAFE_RELEASE(pProfile);
SAFE_RELEASE(pMediaType);
SAFE_RELEASE(pStream);
CoTaskMemFree(pDecoderCLSIDs);
return hr;
}
