HRESULT CStreamSwitcherInputPin::CompleteConnect(IPin* pReceivePin)
{
HRESULT hr = __super::CompleteConnect(pReceivePin);
if(FAILED(hr)) return hr;
((CStreamSwitcherFilter*)m_pFilter)->CompleteConnect(PINDIR_INPUT, this, pReceivePin);
m_fCanBlock = false;
bool fForkedSomewhere = false;
CStringW fileName;
CStringW pinName;
IPin* pPin = (IPin*)this;
IBaseFilter* pBF = (IBaseFilter*)m_pFilter;
while((pPin = GetUpStreamPin(pBF, pPin)) && (pBF = GetFilterFromPin(pPin)))
{
if(IsSplitter(pBF))
{
pinName = GetPinName(pPin);
}
CLSID clsid = GetCLSID(pBF);
if(clsid == CLSID_AviSplitter || clsid == CLSID_OggSplitter)
m_fCanBlock = true;
int nIn, nOut, nInC, nOutC;
CountPins(pBF, nIn, nOut, nInC, nOutC);
fForkedSomewhere = fForkedSomewhere || nIn > 1 || nOut > 1;
if(CComQIPtr<IFileSourceFilter> pFSF = pBF)
{
WCHAR* pszName = NULL;
AM_MEDIA_TYPE mt;
if(SUCCEEDED(pFSF->GetCurFile(&pszName, &mt)) && pszName)
{
fileName = pszName;
CoTaskMemFree(pszName);
fileName.Replace('\\', '/');
CStringW fn = fileName.Mid(fileName.ReverseFind('/')+1);
if(!fn.IsEmpty()) fileName = fn;
if(!pinName.IsEmpty()) fileName += L" / " + pinName;
WCHAR* pName = new WCHAR[fileName.GetLength()+1];
if(pName)
{
wcscpy(pName, fileName);
if(m_pName) delete [] m_pName;
m_pName = pName;
}
}
break;
}
pPin = GetFirstPin(pBF);
}
if(!fForkedSomewhere)
m_fCanBlock = true;
m_hNotifyEvent = NULL;
return S_OK;
}
HRESULT CStreamSwitcherInputPin::CompleteConnect(IPin* pReceivePin)
{
HRESULT hr = __super::CompleteConnect(pReceivePin);
if (FAILED(hr)) {
return hr;
}
(static_cast<CStreamSwitcherFilter*>(m_pFilter))->CompleteConnect(PINDIR_INPUT, this, pReceivePin);
m_fCanBlock = false;
bool fForkedSomewhere = false;
CStringW fileName;
CStringW pinName;
IPin* pPin = (IPin*)this;
IBaseFilter* pBF = (IBaseFilter*)m_pFilter;
pPin = GetUpStreamPin(pBF, pPin);
if (pPin) {
pBF = GetFilterFromPin(pPin);
}
while (pPin && pBF) {
if (IsSplitter(pBF)) {
pinName = GetPinName(pPin);
}
CLSID clsid = GetCLSID(pBF);
if (clsid == CLSID_AviSplitter || clsid == CLSID_OggSplitter) {
m_fCanBlock = true;
}
int nIn, nOut, nInC, nOutC;
CountPins(pBF, nIn, nOut, nInC, nOutC);
fForkedSomewhere = fForkedSomewhere || nIn > 1 || nOut > 1;
DWORD cStreams = 0;
if (CComQIPtr<IAMStreamSelect> pSSF = pBF) {
hr = pSSF->Count(&cStreams);
if (SUCCEEDED(hr)) {
for (int i = 0; i < (int)cStreams; i++) {
AM_MEDIA_TYPE* pmt = nullptr;
hr = pSSF->Info(i, &pmt, nullptr, nullptr, nullptr, nullptr, nullptr, nullptr);
if (SUCCEEDED(hr) && pmt && pmt->majortype == MEDIATYPE_Audio) {
m_pSSF = pSSF;
DeleteMediaType(pmt);
break;
}
DeleteMediaType(pmt);
}
}
}
if (CComQIPtr<IFileSourceFilter> pFSF = pBF) {
WCHAR* pszName = nullptr;
AM_MEDIA_TYPE mt;
if (SUCCEEDED(pFSF->GetCurFile(&pszName, &mt)) && pszName) {
fileName = pszName;
CoTaskMemFree(pszName);
fileName.Replace('\\', '/');
CStringW fn = fileName.Mid(fileName.ReverseFind('/') + 1);
if (!fn.IsEmpty()) {
fileName = fn;
}
// Haali & LAVFSplitter return only one "Audio" pin name, cause CMainFrame::OnInitMenuPopup lookup find the wrong popmenu,
// add space at the end to prevent this, internal filter never return "Audio" only.
if (!pinName.IsEmpty()) {
fileName = pinName + L" ";
}
WCHAR* pName = DEBUG_NEW WCHAR[fileName.GetLength() + 1];
if (pName) {
wcscpy_s(pName, fileName.GetLength() + 1, fileName);
if (m_pName) {
delete [] m_pName;
}
m_pName = pName;
if (cStreams == 1) { // Simple external track, no need to use the info from IAMStreamSelect
m_pSSF.Release();
}
}
}
break;
}
pPin = GetFirstPin(pBF);
pPin = GetUpStreamPin(pBF, pPin);
if (pPin) {
pBF = GetFilterFromPin(pPin);
}
}
if (!fForkedSomewhere) {
m_fCanBlock = true;
}
m_hNotifyEvent = nullptr;
return S_OK;
}
HRESULT CStreamSwitcherInputPin::CompleteConnect(IPin* pReceivePin)
{
HRESULT hr = __super::CompleteConnect(pReceivePin);
if (FAILED(hr)) {
return hr;
}
(static_cast<CStreamSwitcherFilter*>(m_pFilter))->CompleteConnect(PINDIR_INPUT, this, pReceivePin);
m_fCanBlock = false;
bool fForkedSomewhere = false;
CStringW fileName;
CStringW pinName;
IPin* pPin = (IPin*)this;
IBaseFilter* pBF = (IBaseFilter*)m_pFilter;
pPin = GetUpStreamPin(pBF, pPin);
if (pPin) {
pBF = GetFilterFromPin(pPin);
}
while (pPin && pBF) {
if (IsSplitter(pBF)) {
pinName = GetPinName(pPin);
}
CLSID clsid = GetCLSID(pBF);
if (clsid == CLSID_AviSplitter || clsid == CLSID_OggSplitter) {
m_fCanBlock = true;
}
int nIn, nOut, nInC, nOutC;
CountPins(pBF, nIn, nOut, nInC, nOutC);
fForkedSomewhere = fForkedSomewhere || nIn > 1 || nOut > 1;
if (CComQIPtr<IFileSourceFilter> pFSF = pBF) {
WCHAR* pszName = NULL;
AM_MEDIA_TYPE mt;
if (SUCCEEDED(pFSF->GetCurFile(&pszName, &mt)) && pszName) {
fileName = pszName;
CoTaskMemFree(pszName);
fileName.Replace('\\', '/');
CStringW fn = fileName.Mid(fileName.ReverseFind('/')+1);
if (!fn.IsEmpty()) {
fileName = fn;
}
// Haali & LAVFSplitter return only one "Audio" pin name, cause CMainFrame::OnInitMenuPopup lookup find the wrong popmenu,
// add space at the end to prevent this, internal filter never return "Audio" only.
if (!pinName.IsEmpty()) {
fileName = pinName + L" ";
}
WCHAR* pName = DNew WCHAR[fileName.GetLength()+1];
if (pName) {
wcscpy_s(pName, fileName.GetLength() + 1, fileName);
if (m_pName) {
delete [] m_pName;
}
m_pName = pName;
}
}
break;
}
pPin = GetFirstPin(pBF);
pPin = GetUpStreamPin(pBF, pPin);
if (pPin) {
pBF = GetFilterFromPin(pPin);
}
}
if (!fForkedSomewhere) {
m_fCanBlock = true;
}
m_hNotifyEvent = NULL;
return S_OK;
}