STDMETHODIMP CSubtitleInputPin::NewSegment(REFERENCE_TIME tStart, REFERENCE_TIME tStop, double dRate)
{
CAutoLock cAutoLock(&m_csReceive);
if (m_mt.majortype == MEDIATYPE_Text
|| m_mt.majortype == MEDIATYPE_Subtitle
&& (m_mt.subtype == MEDIASUBTYPE_UTF8
/*|| m_mt.subtype == MEDIASUBTYPE_USF*/
|| m_mt.subtype == MEDIASUBTYPE_SSA
|| m_mt.subtype == MEDIASUBTYPE_ASS
|| m_mt.subtype == MEDIASUBTYPE_ASS2)) {
CAutoLock cAutoLock2(m_pSubLock);
CRenderedTextSubtitle* pRTS = (CRenderedTextSubtitle*)(ISubStream*)m_pSubStream;
pRTS->RemoveAll();
pRTS->CreateSegments();
} else if (m_mt.majortype == MEDIATYPE_Subtitle && (m_mt.subtype == MEDIASUBTYPE_VOBSUB)) {
CAutoLock cAutoLock2(m_pSubLock);
CVobSubStream* pVSS = (CVobSubStream*)(ISubStream*)m_pSubStream;
pVSS->RemoveAll();
} else if (IsHdmvSub(&m_mt)) {
CAutoLock cAutoLock2(m_pSubLock);
CRenderedHdmvSubtitle* pHdmvSubtitle = (CRenderedHdmvSubtitle*)(ISubStream*)m_pSubStream;
pHdmvSubtitle->NewSegment(tStart, tStop, dRate);
}
TRACE(_T("NewSegment: InvalidateSubtitle(%I64d, ...)\n"), tStart);
// IMPORTANT: m_pSubLock must not be locked when calling this
InvalidateSubtitle(tStart, m_pSubStream);
return __super::NewSegment(tStart, tStop, dRate);
}
void CSubtitleInputPin::DecodeSamples()
{
SetThreadName(DWORD(-1), "Subtitle Input Pin Thread");
for (; !m_bExitDecodingThread;) {
std::unique_lock<std::mutex> lock(m_mutexQueue);
auto needStopProcessing = [this]() {
return m_bStopDecoding || m_bExitDecodingThread;
};
auto isQueueReady = [&]() {
return !m_sampleQueue.IsEmpty() || needStopProcessing();
};
m_condQueueReady.wait(lock, isQueueReady);
lock.unlock(); // Release this lock until we can acquire the other one
REFERENCE_TIME rtInvalidate = -1;
if (!needStopProcessing()) {
CAutoLock cAutoLock(m_pSubLock);
lock.lock(); // Reacquire the lock
while (!m_sampleQueue.IsEmpty() && !needStopProcessing()) {
auto pSample = m_sampleQueue.RemoveHead();
if (pSample) {
REFERENCE_TIME rtSampleInvalidate = DecodeSample(pSample);
if (rtSampleInvalidate >= 0 && (rtSampleInvalidate < rtInvalidate || rtInvalidate < 0)) {
rtInvalidate = rtSampleInvalidate;
}
} else { // marker for end of stream
if (IsRLECodedSub(&m_mt)) {
CRLECodedSubtitle* pRLECodedSubtitle = (CRLECodedSubtitle*)(ISubStream*)m_pSubStream;
pRLECodedSubtitle->EndOfStream();
}
}
}
}
if (rtInvalidate >= 0) {
TRACE(_T("NewSegment: InvalidateSubtitle(%I64d, ...)\n"), rtInvalidate);
// IMPORTANT: m_pSubLock must not be locked when calling this
InvalidateSubtitle(rtInvalidate, m_pSubStream);
}
}
}
STDMETHODIMP CSubtitleInputPin::Receive(IMediaSample* pSample)
{
TRACE_SAMPLE_TIMING(__FUNCTIONW__);
HRESULT hr;
REFERENCE_TIME tStart, tStop;
hr = pSample->GetTime(&tStart, &tStop);
ASSERT(SUCCEEDED(hr));
tStart += m_tStart;
hr = __super::Receive(pSample);
CAutoLock cAutoLock(m_pSubLock);
if (m_helper)
{
hr = m_helper->Receive(pSample);
TRACE_SAMPLE("InvalidateSubtitle :"<<ReftimeToCString(tStart));
InvalidateSubtitle(tStart, m_helper->GetSubStream());
}
hr = S_OK;
return hr;
}
STDMETHODIMP CSubtitleInputPin::Receive(IMediaSample* pSample)
{
HRESULT hr;
hr = __super::Receive(pSample);
if(FAILED(hr)) return hr;
CAutoLock cAutoLock(&m_csReceive);
REFERENCE_TIME tStart, tStop;
pSample->GetTime(&tStart, &tStop);
tStart += m_tStart;
tStop += m_tStart;
BYTE* pData = NULL;
hr = pSample->GetPointer(&pData);
if(FAILED(hr) || pData == NULL) return hr;
int len = pSample->GetActualDataLength();
bool fInvalidate = false;
if(m_mt.majortype == MEDIATYPE_Text)
{
CAutoLock cAutoLock(m_pSubLock);
CRenderedTextSubtitle* pRTS = (CRenderedTextSubtitle*)(ISubStream*)m_pSubStream;
if(!strncmp((char*)pData, __GAB1__, strlen(__GAB1__)))
{
char* ptr = (char*)&pData[strlen(__GAB1__)+1];
char* end = (char*)&pData[len];
while(ptr < end)
{
WORD tag = *((WORD*)(ptr));
ptr += 2;
WORD size = *((WORD*)(ptr));
ptr += 2;
if(tag == __GAB1_LANGUAGE__)
{
pRTS->m_name = CString(ptr);
}
else if(tag == __GAB1_ENTRY__)
{
pRTS->Add(AToW(&ptr[8]), false, *(int*)ptr, *(int*)(ptr+4));
fInvalidate = true;
}
else if(tag == __GAB1_LANGUAGE_UNICODE__)
{
pRTS->m_name = (WCHAR*)ptr;
}
else if(tag == __GAB1_ENTRY_UNICODE__)
{
pRTS->Add((WCHAR*)(ptr+8), true, *(int*)ptr, *(int*)(ptr+4));
fInvalidate = true;
}
ptr += size;
}
}
else if(!strncmp((char*)pData, __GAB2__, strlen(__GAB2__)))
{
char* ptr = (char*)&pData[strlen(__GAB2__)+1];
char* end = (char*)&pData[len];
while(ptr < end)
{
WORD tag = *((WORD*)(ptr));
ptr += 2;
DWORD size = *((DWORD*)(ptr));
ptr += 4;
if(tag == __GAB1_LANGUAGE_UNICODE__)
{
pRTS->m_name = (WCHAR*)ptr;
}
else if(tag == __GAB1_RAWTEXTSUBTITLE__)
{
pRTS->Open((BYTE*)ptr, size, DEFAULT_CHARSET, pRTS->m_name);
fInvalidate = true;
}
ptr += size;
}
}
else if(pData != 0 && len > 1 && *pData != 0)
{
CStringA str((char*)pData, len);
str.Replace("\r\n", "\n");
str.Trim();
if(!str.IsEmpty())
{
pRTS->Add(AToW(str), false, (int)(tStart / 10000), (int)(tStop / 10000));
fInvalidate = true;
}
}
}
else if(m_mt.majortype == MEDIATYPE_Subtitle)
{
CAutoLock cAutoLock(m_pSubLock);
if(m_mt.subtype == MEDIASUBTYPE_UTF8)
{
CRenderedTextSubtitle* pRTS = (CRenderedTextSubtitle*)(ISubStream*)m_pSubStream;
CStringW str = UTF8To16(CStringA((LPCSTR)pData, len)).Trim();
if(!str.IsEmpty())
{
pRTS->Add(str, true, (int)(tStart / 10000), (int)(tStop / 10000));
fInvalidate = true;
}
}
else if(m_mt.subtype == MEDIASUBTYPE_SSA || m_mt.subtype == MEDIASUBTYPE_ASS || m_mt.subtype == MEDIASUBTYPE_ASS2)
{
CRenderedTextSubtitle* pRTS = (CRenderedTextSubtitle*)(ISubStream*)m_pSubStream;
CStringW str = UTF8To16(CStringA((LPCSTR)pData, len)).Trim();
if(!str.IsEmpty())
{
STSEntry stse;
int fields = m_mt.subtype == MEDIASUBTYPE_ASS2 ? 10 : 9;
CAtlList<CStringW> sl;
Explode(str, sl, ',', fields);
if(sl.GetCount() == (size_t)fields)
{
stse.readorder = wcstol(sl.RemoveHead(), NULL, 10);
stse.layer = wcstol(sl.RemoveHead(), NULL, 10);
stse.style = sl.RemoveHead();
stse.actor = sl.RemoveHead();
stse.marginRect.left = wcstol(sl.RemoveHead(), NULL, 10);
stse.marginRect.right = wcstol(sl.RemoveHead(), NULL, 10);
stse.marginRect.top = stse.marginRect.bottom = wcstol(sl.RemoveHead(), NULL, 10);
if(fields == 10) stse.marginRect.bottom = wcstol(sl.RemoveHead(), NULL, 10);
stse.effect = sl.RemoveHead();
stse.str = sl.RemoveHead();
}
if(!stse.str.IsEmpty())
{
pRTS->Add(stse.str, true, (int)(tStart / 10000), (int)(tStop / 10000),
stse.style, stse.actor, stse.effect, stse.marginRect, stse.layer, stse.readorder);
fInvalidate = true;
}
}
}
else if(m_mt.subtype == MEDIASUBTYPE_SSF)
{
ssf::CRenderer* pSSF = (ssf::CRenderer*)(ISubStream*)m_pSubStream;
CStringW str = UTF8To16(CStringA((LPCSTR)pData, len)).Trim();
if(!str.IsEmpty())
{
pSSF->Append(tStart, tStop, str);
fInvalidate = true;
}
}
else if(m_mt.subtype == MEDIASUBTYPE_VOBSUB)
{
CVobSubStream* pVSS = (CVobSubStream*)(ISubStream*)m_pSubStream;
pVSS->Add(tStart, tStop, pData, len);
}
else if (IsHdmvSub(&m_mt))
{
CAutoLock cAutoLock(m_pSubLock);
CRenderedHdmvSubtitle* pHdmvSubtitle = (CRenderedHdmvSubtitle*)(ISubStream*)m_pSubStream;
pHdmvSubtitle->ParseSample (pSample);
}
}
if(fInvalidate)
{
TRACE(_T("InvalidateSubtitle(%I64d, ..)\n"), tStart);
// IMPORTANT: m_pSubLock must not be locked when calling this
InvalidateSubtitle(tStart, m_pSubStream);
}
hr = S_OK;
return hr;
}
