void CSubtitleDlDlg::OnOK()
{
SetStatus(ResStr(IDS_SUBDL_DLG_DOWNLOADING));
for (int i = 0; i < m_list.GetItemCount(); ++i) {
if (m_list.GetCheck(i)) {
m_selsubs.AddTail(*reinterpret_cast<isdb_subtitle*>(m_list.GetItemData(i)));
}
}
m_fReplaceSubs = IsDlgButtonChecked(IDC_CHECK1) == BST_CHECKED;
CMainFrame* pMF = static_cast<CMainFrame*>(GetParentFrame());
if (m_fReplaceSubs) {
pMF->m_pSubStreams.RemoveAll();
}
CComPtr<ISubStream> pSubStreamToSet;
POSITION pos = m_selsubs.GetHeadPosition();
while (pos) {
const isdb_subtitle& sub = m_selsubs.GetNext(pos);
AppSettings& s = AfxGetAppSettings();
CInternetSession is;
CStringA url = "http://" + s.strISDb + "/dl.php?";
CStringA args, ticket, str;
args.Format("id=%d&ticket=%s", sub.id, UrlEncode(ticket), true);
url.Append(args);
if (OpenUrl(is, CString(url), str)) {
CAutoPtr<CRenderedTextSubtitle> pRTS(DNew CRenderedTextSubtitle(&pMF->m_csSubLock, &s.subdefstyle, s.fUseDefaultSubtitlesStyle));
if (pRTS && pRTS->Open((BYTE*)(LPCSTR)str, str.GetLength(), DEFAULT_CHARSET, CString(sub.name)) && pRTS->GetStreamCount() > 0) {
CComPtr<ISubStream> pSubStream = pRTS.Detach();
pMF->m_pSubStreams.AddTail(pSubStream);
if (!pSubStreamToSet) {
pSubStreamToSet = pSubStream;
}
}
}
}
if (pSubStreamToSet) {
pMF->SetSubtitle(pSubStreamToSet);
}
__super::OnOK();
}
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);
}
void CSubtitleDlDlg::OnOK()
{
SetStatus(ResStr(IDS_SUBDL_DLG_DOWNLOADING));
for (int i = 0; i < m_list.GetItemCount(); ++i) {
if (m_list.GetCheck(i)) {
m_selsubs.AddTail(*reinterpret_cast<isdb_subtitle*>(m_list.GetItemData(i)));
}
}
m_fReplaceSubs = IsDlgButtonChecked(IDC_CHECK1) == BST_CHECKED;
CMainFrame* pMF = static_cast<CMainFrame*>(GetParentFrame());
if (m_fReplaceSubs) {
pMF->m_pSubStreams.RemoveAll();
}
CComPtr<ISubStream> pSubStreamToSet;
POSITION pos = m_selsubs.GetHeadPosition();
while (pos) {
const isdb_subtitle& sub = m_selsubs.GetNext(pos);
AppSettings& s = AfxGetAppSettings();
CInternetSession is;
CStringA url = "http://" + s.strISDb + "/dl.php?";
CStringA args, ticket, str;
args.Format("id=%d&ticket=%s", sub.id, UrlEncode(ticket));
url.Append(args);
if (OpenUrl(is, CString(url), str)) {
if (pMF->b_UseVSFilter) {
if (CComQIPtr<IDirectVobSub> pDVS = pMF->GetVSFilter()) {
TCHAR lpszTempPath[_MAX_PATH] = { 0 };
if (::GetTempPath(_MAX_PATH, lpszTempPath)) {
CString subFileName(lpszTempPath);
subFileName.Append(CString(sub.name));
if (::PathFileExists(subFileName)) {
::DeleteFile(subFileName);
}
CFile cf;
if (cf.Open(subFileName, CFile::modeCreate|CFile::modeWrite|CFile::shareDenyNone)) {
cf.Write(str.GetString(), str.GetLength());
cf.Close();
if (SUCCEEDED(pDVS->put_FileName((LPWSTR)(LPCWSTR)subFileName))) {
pDVS->put_SelectedLanguage(0);
pDVS->put_HideSubtitles(true);
pDVS->put_HideSubtitles(false);
}
::DeleteFile(subFileName);
}
}
__super::OnOK();
return;
}
}
CAutoPtr<CRenderedTextSubtitle> pRTS(DNew CRenderedTextSubtitle(&pMF->m_csSubLock, &s.subdefstyle, s.fUseDefaultSubtitlesStyle));
if (pRTS && pRTS->Open((BYTE*)(LPCSTR)str, str.GetLength(), DEFAULT_CHARSET, CString(sub.name)) && pRTS->GetStreamCount() > 0) {
CComPtr<ISubStream> pSubStream = pRTS.Detach();
pMF->m_pSubStreams.AddTail(pSubStream);
if (!pSubStreamToSet) {
pSubStreamToSet = pSubStream;
}
}
}
}
if (pSubStreamToSet) {
pMF->SetSubtitle(pSubStreamToSet);
AfxGetAppSettings().fEnableSubtitles = true;
}
__super::OnOK();
}
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;
}
