std::string ProbeLangForLanguage(LPCSTR code)
{
if(strlen(code) == 3) {
return ISO6392ToLanguage(code);
} else if (strlen(code) >= 2) {
return ISO6391ToLanguage(code);
}
return std::string();
}
CString CSubtitleDlDlg::LangCodeToName(LPCSTR code)
{
// accept only three-letter language codes
size_t codeLen = strlen(code);
if (codeLen != 3) {
return _T("");
}
CString name = ISO6392ToLanguage(code);
if (!name.IsEmpty()) {
// workaround for ISO6392ToLanguage function behaivior
// for unknown language code it returns the code parameter back
if (code != name) {
return name;
}
}
// support abbreviations loosely based on first letters of language name
// this list is limited to upload-enabled languages
// retrieved with:
// wget -q -O- http://www.opensubtitles.org/addons/export_languages.php | \
// awk 'NR > 1 { if ($(NF-1) == "1") print ("\"" $(NF-2) "\",")}'
static LPCSTR ltable[] = {
"Albanian", "Arabic", "Armenian", "Basque", "Bengali", "Bosnian", "Breton", "Bulgarian",
"Burmese", "Catalan", "Chinese", "Czech", "Danish", "Dutch", "English", "Esperanto",
"Estonian", "Finnish", "French", "Georgian", "German", "Galician", "Greek", "Hebrew",
"Hindi", "Croatian", "Hungarian", "Icelandic", "Indonesian", "Italian", "Japanese", "Kazakh",
"Khmer", "Korean", "Latvian", "Lithuanian", "Luxembourgish", "Macedonian", "Malayalam", "Malay",
"Mongolian", "Norwegian", "Occitan", "Persian", "Polish", "Portuguese", "Russian", "Serbian",
"Sinhalese", "Slovak", "Slovenian", "Spanish", "Swahili", "Swedish", "Syriac", "Telugu",
"Tagalog", "Thai", "Turkish", "Ukrainian", "Urdu", "Vietnamese", "Romanian", "Brazilian",
};
for (size_t i = 0; i < _countof(ltable); ++i) {
CString name2 = ltable[i];
if (StrMatch(name2, CString(code)) == codeLen) {
return name2;
}
}
return _T("");
}
CString ISOLang::ISO639XToLanguage(LPCSTR code, bool bCheckForFullLangName /*= false*/)
{
CString lang;
switch (size_t nLen = strlen(code)) {
case 2:
lang = ISO6391ToLanguage(code);
break;
case 3:
lang = ISO6392ToLanguage(code);
if (lang == code) { // When it can't find a match, ISO6392ToLanguage returns the input string
lang.Empty();
}
break;
default:
if (bCheckForFullLangName && nLen > 3) {
if (IsISO639Language(code)) {
lang = code;
}
}
}
return lang;
}
HRESULT CSubtitleInputPin::CompleteConnect(IPin* pReceivePin)
{
if(m_mt.majortype == MEDIATYPE_Text)
{
if(!(m_pSubStream = DNew CRenderedTextSubtitle(m_pSubLock))) return E_FAIL;
CRenderedTextSubtitle* pRTS = (CRenderedTextSubtitle*)(ISubStream*)m_pSubStream;
pRTS->m_name = CString(GetPinName(pReceivePin)) + _T(" (embeded)");
pRTS->m_dstScreenSize = CSize(384, 288);
pRTS->CreateDefaultStyle(DEFAULT_CHARSET);
}
else if(m_mt.majortype == MEDIATYPE_Subtitle)
{
SUBTITLEINFO* psi = (SUBTITLEINFO*)m_mt.pbFormat;
DWORD dwOffset = 0;
CString name;
LCID lcid = 0;
if (psi != NULL)
{
dwOffset = psi->dwOffset;
name = ISO6392ToLanguage(psi->IsoLang);
lcid = ISO6392ToLcid(psi->IsoLang);
if(name.IsEmpty()) name = _T("Unknown");
if(wcslen(psi->TrackName) > 0) name += _T(" (") + CString(psi->TrackName) + _T(")");
}
if(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)
{
if(!(m_pSubStream = DNew CRenderedTextSubtitle(m_pSubLock))) return E_FAIL;
CRenderedTextSubtitle* pRTS = (CRenderedTextSubtitle*)(ISubStream*)m_pSubStream;
pRTS->m_name = name;
pRTS->m_lcid = lcid;
pRTS->m_dstScreenSize = CSize(384, 288);
pRTS->CreateDefaultStyle(DEFAULT_CHARSET);
if(dwOffset > 0 && m_mt.cbFormat - dwOffset > 0)
{
CMediaType mt = m_mt;
if(mt.pbFormat[dwOffset+0] != 0xef
&& mt.pbFormat[dwOffset+1] != 0xbb
&& mt.pbFormat[dwOffset+2] != 0xfb)
{
dwOffset -= 3;
mt.pbFormat[dwOffset+0] = 0xef;
mt.pbFormat[dwOffset+1] = 0xbb;
mt.pbFormat[dwOffset+2] = 0xbf;
}
pRTS->Open(mt.pbFormat + dwOffset, mt.cbFormat - dwOffset, DEFAULT_CHARSET, pRTS->m_name);
}
}
else if(m_mt.subtype == MEDIASUBTYPE_SSF)
{
if(!(m_pSubStream = DNew ssf::CRenderer(m_pSubLock))) return E_FAIL;
ssf::CRenderer* pSSF = (ssf::CRenderer*)(ISubStream*)m_pSubStream;
pSSF->Open(ssf::MemoryInputStream(m_mt.pbFormat + dwOffset, m_mt.cbFormat - dwOffset, false, false), name);
}
else if(m_mt.subtype == MEDIASUBTYPE_VOBSUB)
{
if(!(m_pSubStream = DNew CVobSubStream(m_pSubLock))) return E_FAIL;
CVobSubStream* pVSS = (CVobSubStream*)(ISubStream*)m_pSubStream;
pVSS->Open(name, m_mt.pbFormat + dwOffset, m_mt.cbFormat - dwOffset);
}
else if (IsHdmvSub(&m_mt))
{
if(!(m_pSubStream = DNew CRenderedHdmvSubtitle(m_pSubLock, (m_mt.subtype == MEDIASUBTYPE_DVB_SUBTITLES) ? ST_DVB : ST_HDMV))) return E_FAIL;
}
}
AddSubStream(m_pSubStream);
return __super::CompleteConnect(pReceivePin);
}
HRESULT CSubtitleInputPin::CompleteConnect(IPin* pReceivePin)
{
if (m_mt.majortype == MEDIATYPE_Text) {
if (!(m_pSubStream = DEBUG_NEW CRenderedTextSubtitle(m_pSubLock))) {
return E_FAIL;
}
CRenderedTextSubtitle* pRTS = (CRenderedTextSubtitle*)(ISubStream*)m_pSubStream;
pRTS->m_name = CString(GetPinName(pReceivePin)) + _T(" (embeded)");
pRTS->m_dstScreenSize = CSize(384, 288);
pRTS->CreateDefaultStyle(DEFAULT_CHARSET);
} else if (m_mt.majortype == MEDIATYPE_Subtitle) {
SUBTITLEINFO* psi = (SUBTITLEINFO*)m_mt.pbFormat;
DWORD dwOffset = 0;
CString name;
LCID lcid = 0;
if (psi != nullptr) {
dwOffset = psi->dwOffset;
name = ISO6392ToLanguage(psi->IsoLang);
lcid = ISO6392ToLcid(psi->IsoLang);
CString trackName(psi->TrackName);
trackName.Trim();
if (!trackName.IsEmpty()) {
if (!name.IsEmpty()) {
if (trackName[0] != _T('(') && trackName[0] != _T('[')) {
name += _T(",");
}
name += _T(" ");
}
name += trackName;
}
if (name.IsEmpty()) {
name = _T("Unknown");
}
}
name.Replace(_T(""), _T(""));
name.Replace(_T(""), _T(""));
if (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) {
if (!(m_pSubStream = DEBUG_NEW CRenderedTextSubtitle(m_pSubLock))) {
return E_FAIL;
}
CRenderedTextSubtitle* pRTS = (CRenderedTextSubtitle*)(ISubStream*)m_pSubStream;
pRTS->m_name = name;
pRTS->m_lcid = lcid;
pRTS->m_dstScreenSize = CSize(384, 288);
pRTS->CreateDefaultStyle(DEFAULT_CHARSET);
if (dwOffset > 0 && m_mt.cbFormat - dwOffset > 0) {
CMediaType mt = m_mt;
if (mt.pbFormat[dwOffset + 0] != 0xef
&& mt.pbFormat[dwOffset + 1] != 0xbb
&& mt.pbFormat[dwOffset + 2] != 0xfb) {
dwOffset -= 3;
mt.pbFormat[dwOffset + 0] = 0xef;
mt.pbFormat[dwOffset + 1] = 0xbb;
mt.pbFormat[dwOffset + 2] = 0xbf;
}
pRTS->Open(mt.pbFormat + dwOffset, mt.cbFormat - dwOffset, DEFAULT_CHARSET, pRTS->m_name);
}
} else if (m_mt.subtype == MEDIASUBTYPE_VOBSUB) {
if (!(m_pSubStream = DEBUG_NEW CVobSubStream(m_pSubLock))) {
return E_FAIL;
}
CVobSubStream* pVSS = (CVobSubStream*)(ISubStream*)m_pSubStream;
pVSS->Open(name, m_mt.pbFormat + dwOffset, m_mt.cbFormat - dwOffset);
} else if (IsHdmvSub(&m_mt)) {
if (!(m_pSubStream = DEBUG_NEW CRenderedHdmvSubtitle(m_pSubLock, (m_mt.subtype == MEDIASUBTYPE_DVB_SUBTITLES) ? ST_DVB : ST_HDMV, name, lcid))) {
return E_FAIL;
}
}
}
AddSubStream(m_pSubStream);
return __super::CompleteConnect(pReceivePin);
}
STDMETHODIMP_(CSubtitleInputPinHelper*) CSubtitleInputPin::CreateHelper( const CMediaType& mt, IPin* pReceivePin )
{
XY_LOG_INFO(XY_LOG_VAR_2_STR(pReceivePin));
CSubtitleInputPinHelper *ret = NULL;
if(mt.majortype == MEDIATYPE_Text)
{
XY_LOG_INFO("Create CTextSubtitleInputPinHepler");
CRenderedTextSubtitle* pRTS = DEBUG_NEW CRenderedTextSubtitle(m_pSubLock);
pRTS->m_name = CString(GetPinName(pReceivePin)) + _T(" (embeded)");
pRTS->m_dstScreenSize = CSize(384, 288);
ret = DEBUG_NEW CTextSubtitleInputPinHepler(pRTS, m_mt);
}
else if(mt.majortype == MEDIATYPE_Subtitle)
{
SUBTITLEINFO* psi = (SUBTITLEINFO*)mt.pbFormat;
DWORD dwOffset = 0;
CString name;
LCID lcid = 0;
if (psi != NULL) {
dwOffset = psi->dwOffset;
name = ISO6392ToLanguage(psi->IsoLang);
lcid = ISO6392ToLcid(psi->IsoLang);
CString trackName(psi->TrackName);
trackName.Trim();
if (!trackName.IsEmpty()) {
if (!name.IsEmpty()) {
if (trackName[0] != _T('(') && trackName[0] != _T('[')) {
name += _T(",");
}
name += _T(" ");
}
name += trackName;
}
if (name.IsEmpty()) {
name = _T("Unknown");
}
}
name.Replace(_T(""), _T(""));//CAUTION: VS may show name.Replace(_T(""),_T("")), however there is a character in the first _T("")
name.Replace(_T(""), _T(""));//CAUTION: VS may show name.Replace(_T(""),_T("")), however there is a character in the first _T("")
if(mt.subtype == MEDIASUBTYPE_UTF8
/*|| m_mt.subtype == MEDIASUBTYPE_USF*/
|| mt.subtype == MEDIASUBTYPE_SSA
|| mt.subtype == MEDIASUBTYPE_ASS
|| mt.subtype == MEDIASUBTYPE_ASS2)
{
XY_LOG_INFO("Create CTextSubtitleInputPinHepler");
CRenderedTextSubtitle* pRTS = DEBUG_NEW CRenderedTextSubtitle(m_pSubLock);
pRTS->m_name = name;
pRTS->m_lcid = lcid;
pRTS->m_dstScreenSize = CSize(384, 288);
if(dwOffset > 0 && mt.cbFormat - dwOffset > 0)
{
CMediaType mt1 = mt;
if(mt1.pbFormat[dwOffset+0] != 0xef
&& mt1.pbFormat[dwOffset+1] != 0xbb
&& mt1.pbFormat[dwOffset+2] != 0xfb)
{
dwOffset -= 3;
mt1.pbFormat[dwOffset+0] = 0xef;
mt1.pbFormat[dwOffset+1] = 0xbb;
mt1.pbFormat[dwOffset+2] = 0xbf;
}
pRTS->Open(mt1.pbFormat + dwOffset, mt1.cbFormat - dwOffset, DEFAULT_CHARSET, pRTS->m_name);
}
ret = DEBUG_NEW CTextSubtitleInputPinHepler(pRTS, m_mt);
}
else if(mt.subtype == MEDIASUBTYPE_SSF)
{
XY_LOG_INFO("Create CSSFInputPinHepler");
ssf::CRenderer* pSSF = DEBUG_NEW ssf::CRenderer(m_pSubLock);
pSSF->Open(ssf::MemoryInputStream(mt.pbFormat + dwOffset, mt.cbFormat - dwOffset, false, false), name);
ret = DEBUG_NEW CSSFInputPinHepler(pSSF, m_mt);
}
else if(mt.subtype == MEDIASUBTYPE_VOBSUB)
{
XY_LOG_INFO("Create CVobsubInputPinHepler");
CVobSubStream* pVSS = DEBUG_NEW CVobSubStream(m_pSubLock);
pVSS->Open(name, mt.pbFormat + dwOffset, mt.cbFormat - dwOffset);
ret = DEBUG_NEW CVobsubInputPinHepler(pVSS, m_mt);
}
else if (IsHdmvSub(&mt))
{
XY_LOG_INFO("Create CHdmvInputPinHepler");
CRenderedHdmvSubtitle *hdmv_sub = DEBUG_NEW CRenderedHdmvSubtitle(m_pSubLock,
(mt.subtype == MEDIASUBTYPE_DVB_SUBTITLES) ? ST_DVB : ST_HDMV, name, lcid);
ret = DEBUG_NEW CHdmvInputPinHepler(hdmv_sub, m_mt);
}
}
return ret;
}
HRESULT CDSMSplitterFilter::CreateOutputs(IAsyncReader* pAsyncReader)
{
CheckPointer(pAsyncReader, E_POINTER);
HRESULT hr = E_FAIL;
m_pFile.Free();
m_pFile.Attach(DNew CDSMSplitterFile(pAsyncReader, hr, *this, *this));
if (!m_pFile) {
return E_OUTOFMEMORY;
}
if (FAILED(hr)) {
m_pFile.Free();
return hr;
}
m_rtNewStart = m_rtCurrent = 0;
m_rtNewStop = m_rtStop = m_rtDuration = m_pFile->m_rtDuration;
CAtlArray<BYTE> ids;
POSITION pos = m_pFile->m_mts.GetStartPosition();
while (pos) {
BYTE id;
CMediaType mt;
m_pFile->m_mts.GetNextAssoc(pos, id, mt);
ids.Add(id);
}
qsort(ids.GetData(), ids.GetCount(), sizeof(BYTE), compare_id);
for (size_t i = 0; i < ids.GetCount(); i++) {
BYTE id = ids[i];
CMediaType& mt = m_pFile->m_mts[id];
CStringW name, lang;
name.Format(L"Output %02d", id);
CAtlArray<CMediaType> mts;
mts.Add(mt);
CAutoPtr<CBaseSplitterOutputPin> pPinOut(DNew CBaseSplitterOutputPin(mts, name, this, this, &hr));
name.Empty();
pos = m_pFile->m_sim[id].GetStartPosition();
while (pos) {
CStringA key;
CStringW value;
m_pFile->m_sim[id].GetNextAssoc(pos, key, value);
pPinOut->SetProperty(CStringW(key), value);
if (key == "NAME") {
name = value;
}
if (key == "LANG") if ((lang = ISO6392ToLanguage(CStringA(CString(value)))).IsEmpty()) {
lang = value;
}
}
if (!name.IsEmpty() || !lang.IsEmpty()) {
if (!name.IsEmpty()) {
if (!lang.IsEmpty()) {
name += L" (" + lang + L")";
}
} else if (!lang.IsEmpty()) {
name = lang;
}
pPinOut->SetName(name);
}
EXECUTE_ASSERT(SUCCEEDED(AddOutputPin(id, pPinOut)));
}
pos = m_pFile->m_fim.GetStartPosition();
while (pos) {
CStringA key;
CStringW value;
m_pFile->m_fim.GetNextAssoc(pos, key, value);
SetProperty(CStringW(key), value);
}
for (size_t i = 0; i < m_resources.GetCount(); i++) {
const CDSMResource& r = m_resources[i];
if (r.mime == "application/x-truetype-font" ||
r.mime == "application/x-font-ttf" ||
r.mime == "application/vnd.ms-opentype") {
//m_fontinst.InstallFont(r.data);
m_fontinst.InstallFontMemory(r.data.GetData(), r.data.GetCount());
}
}
return m_pOutputs.GetCount() > 0 ? S_OK : E_FAIL;
}
