#include <uuids.h>
#include <InitGuid.h>
#include <d3d9.h>
+#include <qdebug.h>
// This MFT sends all samples it processes to connected video probes.
// Sample is sent to probes in ProcessInput.
if (m_outputType)
m_outputType->Release();
+
+ for (int i = 0; i < m_mediaTypes.size(); ++i)
+ m_mediaTypes[i]->Release();
}
void MFTransform::addProbe(MFVideoProbeControl *probe)
m_videoProbes.removeOne(probe);
}
+void MFTransform::addSupportedMediaType(IMFMediaType *type)
+{
+ if (!type)
+ return;
+ QMutexLocker locker(&m_mutex);
+ m_mediaTypes.append(type);
+}
+
STDMETHODIMP MFTransform::QueryInterface(REFIID riid, void** ppv)
{
if (!ppv)
STDMETHODIMP MFTransform::GetInputStreamInfo(DWORD dwInputStreamID, MFT_INPUT_STREAM_INFO *pStreamInfo)
{
+ QMutexLocker locker(&m_mutex);
+
if (dwInputStreamID > 0)
return MF_E_INVALIDSTREAMNUMBER;
STDMETHODIMP MFTransform::GetOutputStreamInfo(DWORD dwOutputStreamID, MFT_OUTPUT_STREAM_INFO *pStreamInfo)
{
+ QMutexLocker locker(&m_mutex);
+
if (dwOutputStreamID > 0)
return MF_E_INVALIDSTREAMNUMBER;
QMutexLocker locker(&m_mutex);
+ if (m_sample)
+ return MF_E_TRANSFORM_CANNOT_CHANGE_MEDIATYPE_WHILE_PROCESSING;
+
+ if (!isMediaTypeSupported(pType))
+ return MF_E_INVALIDMEDIATYPE;
+
DWORD flags = 0;
- if (m_outputType && m_outputType->IsEqual(pType, &flags) != S_OK) {
+ if (pType && !m_inputType && m_outputType && m_outputType->IsEqual(pType, &flags) != S_OK)
return MF_E_INVALIDMEDIATYPE;
- }
if (dwFlags == MFT_SET_TYPE_TEST_ONLY)
return pType ? S_OK : E_POINTER;
- if (m_inputType)
+ if (m_inputType) {
m_inputType->Release();
+ // Input type has changed, discard output type (if it's set) so it's reset later on
+ if (m_outputType && m_outputType->IsEqual(pType, &flags) != S_OK) {
+ m_outputType->Release();
+ m_outputType = 0;
+ }
+ }
m_inputType = pType;
QMutexLocker locker(&m_mutex);
+ if (m_sample)
+ return MF_E_TRANSFORM_CANNOT_CHANGE_MEDIATYPE_WHILE_PROCESSING;
+
+ if (!isMediaTypeSupported(pType))
+ return MF_E_INVALIDMEDIATYPE;
+
DWORD flags = 0;
- if (m_inputType && m_inputType->IsEqual(pType, &flags) != S_OK) {
+ if (pType && !m_outputType && m_inputType && m_inputType->IsEqual(pType, &flags) != S_OK)
return MF_E_INVALIDMEDIATYPE;
- }
if (dwFlags == MFT_SET_TYPE_TEST_ONLY)
return pType ? S_OK : E_POINTER;
- if (m_outputType)
+ if (m_outputType) {
m_outputType->Release();
+ // Output type has changed, discard input type (if it's set) so it's reset later on
+ if (m_inputType && m_inputType->IsEqual(pType, &flags) != S_OK) {
+ m_inputType->Release();
+ m_inputType = 0;
+ }
+ }
m_outputType = pType;
return array;
}
+
+bool MFTransform::isMediaTypeSupported(IMFMediaType *type)
+{
+ // if the list is empty, it supports all formats
+ if (!type || m_mediaTypes.isEmpty())
+ return true;
+
+ for (int i = 0; i < m_mediaTypes.size(); ++i) {
+ DWORD flags = 0;
+ m_mediaTypes.at(i)->IsEqual(type, &flags);
+ if (flags & MF_MEDIATYPE_EQUAL_FORMAT_TYPES)
+ return true;
+ }
+
+ return false;
+}
m_request.rate = 1.0f;
m_audioSampleGrabber = new AudioSampleGrabberCallback;
- m_videoProbeMFT = new MFTransform;
}
void MFPlayerSession::close()
m_sourceResolver->Release();
m_sourceResolver = 0;
}
+ if (m_videoProbeMFT) {
+ m_videoProbeMFT->Release();
+ m_videoProbeMFT = 0;
+ }
+
if (m_session)
m_session->Release();
void MFPlayerSession::addProbe(MFVideoProbeControl* probe)
{
- m_videoProbeMFT->addProbe(probe);
+ if (m_videoProbes.contains(probe))
+ return;
+
+ m_videoProbes.append(probe);
+
+ if (m_videoProbeMFT)
+ m_videoProbeMFT->addProbe(probe);
}
void MFPlayerSession::removeProbe(MFVideoProbeControl* probe)
{
- m_videoProbeMFT->removeProbe(probe);
+ m_videoProbes.removeOne(probe);
+
+ if (m_videoProbeMFT)
+ m_videoProbeMFT->removeProbe(probe);
}
MFPlayerSession::~MFPlayerSession()
{
m_audioSampleGrabber->Release();
- m_videoProbeMFT->Release();
}
if (FAILED(topoLoader->Load(topology, &resolvedTopology, NULL)))
break;
-// FIXME!! VideoProbe disabled in Qt 5.0 because it is unstable.
-// Commented out the following code to skip inserting the transform node
-// getting the video frames.
-
// Get all output nodes and search for video output node.
-// if (FAILED(resolvedTopology->GetOutputNodeCollection(&outputNodes)))
-// break;
+ if (FAILED(resolvedTopology->GetOutputNodeCollection(&outputNodes)))
+ break;
-// DWORD elementCount = 0;
-// if (FAILED(outputNodes->GetElementCount(&elementCount)))
-// break;
+ DWORD elementCount = 0;
+ if (FAILED(outputNodes->GetElementCount(&elementCount)))
+ break;
-// for (DWORD n = 0; n < elementCount; n++) {
-// IUnknown *element = 0;
-// IMFTopologyNode *node = 0;
-// IMFTopologyNode *inputNode = 0;
-// IMFTopologyNode *mftNode = 0;
+ for (DWORD n = 0; n < elementCount; n++) {
+ IUnknown *element = 0;
+ IMFTopologyNode *node = 0;
+ IUnknown *outputObject = 0;
+ IMFMediaTypeHandler *videoSink = 0;
+ IMFTopologyNode *inputNode = 0;
+ IMFTopologyNode *mftNode = 0;
-// do {
-// if (FAILED(outputNodes->GetElement(n, &element)))
-// break;
+ do {
+ if (FAILED(outputNodes->GetElement(n, &element)))
+ break;
-// if (FAILED(element->QueryInterface(IID_IMFTopologyNode, (void**)&node)))
-// break;
+ if (FAILED(element->QueryInterface(IID_IMFTopologyNode, (void**)&node)))
+ break;
-// TOPOID id;
-// if (FAILED(node->GetTopoNodeID(&id)))
-// break;
+ TOPOID id;
+ if (FAILED(node->GetTopoNodeID(&id)))
+ break;
-// if (id != outputNodeId)
-// break;
+ if (id != outputNodeId)
+ break;
-// // Insert MFT between the output node and the node connected to it.
-// DWORD outputIndex = 0;
-// if (FAILED(node->GetInput(0, &inputNode, &outputIndex)))
-// break;
+ // Use output supported media types for the MFT
+ if (FAILED(node->GetObject(&outputObject)))
+ break;
-// if (FAILED(MFCreateTopologyNode(MF_TOPOLOGY_TRANSFORM_NODE, &mftNode)))
-// break;
+ if (FAILED(outputObject->QueryInterface(IID_IMFMediaTypeHandler, (void**)&videoSink)))
+ break;
-// if (FAILED(mftNode->SetObject(m_videoProbeMFT)))
-// break;
+ DWORD mtCount;
+ if (FAILED(videoSink->GetMediaTypeCount(&mtCount)))
+ break;
-// if (FAILED(resolvedTopology->AddNode(mftNode)))
-// break;
+ for (DWORD i = 0; i < mtCount; ++i) {
+ IMFMediaType *type = 0;
+ if (SUCCEEDED(videoSink->GetMediaTypeByIndex(i, &type)))
+ m_videoProbeMFT->addSupportedMediaType(type);
+ }
-// if (FAILED(inputNode->ConnectOutput(0, mftNode, 0)))
-// break;
+ // Insert MFT between the output node and the node connected to it.
+ DWORD outputIndex = 0;
+ if (FAILED(node->GetInput(0, &inputNode, &outputIndex)))
+ break;
-// if (FAILED(mftNode->ConnectOutput(0, node, 0)))
-// break;
+ if (FAILED(MFCreateTopologyNode(MF_TOPOLOGY_TRANSFORM_NODE, &mftNode)))
+ break;
-// isNewTopology = true;
-// } while (false);
+ if (FAILED(mftNode->SetObject(m_videoProbeMFT)))
+ break;
-// if (mftNode)
-// mftNode->Release();
-// if (inputNode)
-// inputNode->Release();
-// if (node)
-// node->Release();
-// if (element)
-// element->Release();
+ if (FAILED(resolvedTopology->AddNode(mftNode)))
+ break;
-// if (isNewTopology)
-// break;
-// }
+ if (FAILED(inputNode->ConnectOutput(0, mftNode, 0)))
+ break;
+
+ if (FAILED(mftNode->ConnectOutput(0, node, 0)))
+ break;
+
+ isNewTopology = true;
+ } while (false);
+
+ if (mftNode)
+ mftNode->Release();
+ if (inputNode)
+ inputNode->Release();
+ if (node)
+ node->Release();
+ if (element)
+ element->Release();
+ if (videoSink)
+ videoSink->Release();
+ if (outputObject)
+ outputObject->Release();
+
+ if (isNewTopology)
+ break;
+ }
} while (false);
if (outputNodes)
QObject::connect(m_sourceResolver, SIGNAL(mediaSourceReady()), this, SLOT(handleMediaSourceReady()));
QObject::connect(m_sourceResolver, SIGNAL(error(long)), this, SLOT(handleSourceError(long)));
+ m_videoProbeMFT = new MFTransform;
+ for (int i = 0; i < m_videoProbes.size(); ++i)
+ m_videoProbeMFT->addProbe(m_videoProbes.at(i));
+
Q_ASSERT(m_session == NULL);
HRESULT hr = MFCreateMediaSession(NULL, &m_session);
if (FAILED(hr)) {
break;
case MESourceUnknown:
changeStatus(QMediaPlayer::InvalidMedia);
+ break;
case MEError:
+ changeStatus(QMediaPlayer::UnknownMediaStatus);
qWarning() << "handleSessionEvent: serious error = " << hrStatus;
emit error(QMediaPlayer::ResourceError, tr("Media session serious error."), true);
break;