src/apps/mplayerc/TextPassThruFilter.cpp

   1 /*
   2  *      Copyright (C) 2003-2006 Gabest
   3  *      http://www.gabest.org
   4  *
   5  *  This Program is free software; you can redistribute it and/or modify
   6  *  it under the terms of the GNU General Public License as published by
   7  *  the Free Software Foundation; either version 2, or (at your option)
   8  *  any later version.
   9  *
  10  *  This Program is distributed in the hope that it will be useful,
  11  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
  12  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  13  *  GNU General Public License for more details.
  14  *
  15  *  You should have received a copy of the GNU General Public License
  16  *  along with GNU Make; see the file COPYING.  If not, write to
  17  *  the Free Software Foundation, 675 Mass Ave, Cambridge, MA 02139, USA.
  18  *  http://www.gnu.org/copyleft/gpl.html
  19  *
  20  */
  21
  22 #include "stdafx.h"
  23 #include <windows.h>
  24 #include <commdlg.h>
  25 #include "mplayerc.h"
  26 #include "mainfrm.h"
  27 #include "TextPassThruFilter.h"
  28 #include "..\..\..\include\moreuuids.h"
  29 #include "..\..\DSUtil\DSUtil.h"
  30 #include "..\..\subtitles\SubtitleInputPin.h"
  31
  32 //
  33 // CTextPassThruInputPin
  34 //
  35
  36 class CTextPassThruInputPin : public CSubtitleInputPin
  37 {
  38         CTextPassThruFilter* m_pTPTFilter;
  39         CComPtr<ISubStream> m_pSubStreamOld;
  40
  41 protected:
  42         void AddSubStream(ISubStream* pSubStream)
  43         {
  44                 if(m_pSubStreamOld)
  45                 {
  46                         if(pSubStream) m_pTPTFilter->m_pMainFrame->ReplaceSubtitle(m_pSubStreamOld, pSubStream);
  47                         m_pSubStreamOld = NULL;
  48                 }
  49         }
  50
  51         void RemoveSubStream(ISubStream* pSubStream)
  52         {
  53                 m_pSubStreamOld = pSubStream;
  54         }
  55
  56         void InvalidateSubtitle(REFERENCE_TIME rtStart, ISubStream* pSubStream)
  57         {
  58                 m_pTPTFilter->m_pMainFrame->InvalidateSubtitle((DWORD_PTR)pSubStream, rtStart);
  59         }
  60
  61 public:
  62     CTextPassThruInputPin(CTextPassThruFilter* pTPTFilter, CCritSec* pLock, CCritSec* pSubLock, HRESULT* phr);
  63         STDMETHODIMP NewSegment(REFERENCE_TIME tStart, REFERENCE_TIME tStop, double dRate);
  64         STDMETHODIMP Receive(IMediaSample* pSample);
  65     STDMETHODIMP EndOfStream();
  66     STDMETHODIMP BeginFlush();
  67     STDMETHODIMP EndFlush();
  68 };
  69
  70 //
  71 // CTextPassThruOutputPin
  72 //
  73
  74 class CTextPassThruOutputPin : public CBaseOutputPin
  75 {
  76         CTextPassThruFilter* m_pTPTFilter;
  77
  78 public:
  79     CTextPassThruOutputPin(CTextPassThruFilter* pTPTFilter, CCritSec* pLock, HRESULT* phr);
  80
  81     HRESULT CheckMediaType(const CMediaType* mtOut);
  82         HRESULT DecideBufferSize(IMemAllocator* pAllocator, ALLOCATOR_PROPERTIES* pProperties);
  83         HRESULT GetMediaType(int iPosition, CMediaType* pmt);
  84         STDMETHODIMP Notify(IBaseFilter* pSender, Quality q) {return S_OK;}
  85 };
  86
  87 ///////////
  88
  89 CTextPassThruInputPin::CTextPassThruInputPin(CTextPassThruFilter* pTPTFilter, CCritSec* pLock, CCritSec* pSubLock, HRESULT* phr)
  90         : CSubtitleInputPin(pTPTFilter, pLock, pSubLock, phr)
  91         , m_pTPTFilter(pTPTFilter)
  92 {
  93 }
  94
  95 STDMETHODIMP CTextPassThruInputPin::NewSegment(REFERENCE_TIME tStart, REFERENCE_TIME tStop, double dRate)
  96 {
  97         HRESULT hr = __super::NewSegment(tStart, tStop, dRate);
  98         if(FAILED(hr)) return hr;
  99         return m_pTPTFilter->m_pOutput->DeliverNewSegment(tStart, tStop, dRate);
 100 }
 101
 102 STDMETHODIMP CTextPassThruInputPin::Receive(IMediaSample* pSample)
 103 {
 104         HRESULT hr = __super::Receive(pSample);
 105         if(FAILED(hr)) return hr;
 106         return m_pTPTFilter->m_pOutput->Deliver(pSample);
 107 }
 108
 109 STDMETHODIMP CTextPassThruInputPin::EndOfStream()
 110 {
 111         HRESULT hr = __super::EndOfStream();
 112         if(FAILED(hr)) return hr;
 113         return m_pTPTFilter->m_pOutput->DeliverEndOfStream();
 114 }
 115
 116 STDMETHODIMP CTextPassThruInputPin::BeginFlush()
 117 {
 118         HRESULT hr = __super::BeginFlush();
 119         if(FAILED(hr)) return hr;
 120         return m_pTPTFilter->m_pOutput->DeliverBeginFlush();
 121 }
 122
 123 STDMETHODIMP CTextPassThruInputPin::EndFlush()
 124 {
 125         HRESULT hr = __super::EndFlush();
 126         if(FAILED(hr)) return hr;
 127         return m_pTPTFilter->m_pOutput->DeliverEndFlush();
 128 }
 129
 130 //
 131
 132 CTextPassThruOutputPin::CTextPassThruOutputPin(CTextPassThruFilter* pTPTFilter, CCritSec* pLock, HRESULT* phr)
 133         : CBaseOutputPin(NAME(""), pTPTFilter, pLock, phr, L"Out")
 134         , m_pTPTFilter(pTPTFilter)
 135 {
 136 }
 137
 138 HRESULT CTextPassThruOutputPin::CheckMediaType(const CMediaType* mtOut)
 139 {
 140         CMediaType mt;
 141         return S_OK == m_pTPTFilter->m_pInput->ConnectionMediaType(&mt) && mt == *mtOut
 142                 ? S_OK
 143                 : E_FAIL;
 144 }
 145
 146 HRESULT CTextPassThruOutputPin::DecideBufferSize(IMemAllocator* pAllocator, ALLOCATOR_PROPERTIES* pProperties)
 147 {
 148         if(m_pTPTFilter->m_pInput->IsConnected() == FALSE)
 149                 return E_UNEXPECTED;
 150
 151         CComPtr<IMemAllocator> pAllocatorIn;
 152         m_pTPTFilter->m_pInput->GetAllocator(&pAllocatorIn);
 153         if(!pAllocatorIn) return E_UNEXPECTED;
 154
 155         pAllocatorIn->GetProperties(pProperties);
 156
 157         HRESULT hr;
 158         ALLOCATOR_PROPERTIES Actual;
 159         if(FAILED(hr = pAllocator->SetProperties(pProperties, &Actual)))
 160                 return hr;
 161
 162         return(pProperties->cBuffers > Actual.cBuffers || pProperties->cbBuffer > Actual.cbBuffer
 163                 ? E_FAIL
 164                 : NOERROR);
 165 }
 166
 167 HRESULT CTextPassThruOutputPin::GetMediaType(int iPosition, CMediaType* pmt)
 168 {
 169         if(m_pTPTFilter->m_pInput->IsConnected() == FALSE)
 170                 return E_UNEXPECTED;
 171
 172         if(iPosition < 0) return E_INVALIDARG;
 173         if(iPosition > 0) return VFW_S_NO_MORE_ITEMS;
 174
 175         m_pTPTFilter->m_pInput->ConnectionMediaType(pmt);
 176
 177         return S_OK;
 178 }
 179
 180 //
 181 // CTextPassThruFilter
 182 //
 183
 184 CTextPassThruFilter::CTextPassThruFilter(CMainFrame* pMainFrame)
 185         : CBaseFilter(NAME("CTextPassThruFilter"), NULL, this, __uuidof(this))
 186         , m_pMainFrame(pMainFrame)
 187 {
 188         HRESULT hr;
 189         m_pInput = new CTextPassThruInputPin(this, this, &m_pMainFrame->m_csSubLock, &hr);
 190         m_pOutput = new CTextPassThruOutputPin(this, this, &hr);
 191 }
 192
 193 CTextPassThruFilter::~CTextPassThruFilter()
 194 {
 195         delete m_pInput; m_pInput = NULL;
 196         delete m_pOutput; m_pOutput = NULL;
 197 }
 198
 199 STDMETHODIMP CTextPassThruFilter::NonDelegatingQueryInterface(REFIID riid, void** ppv)
 200 {
 201         if(m_pInput && riid == __uuidof(ISubStream))
 202         {
 203                 if(CComPtr<ISubStream> pSubStream = m_pInput->GetSubStream())
 204                 {
 205                         *ppv = pSubStream.Detach();
 206                         return S_OK;
 207                 }
 208         }
 209
 210         return __super::NonDelegatingQueryInterface(riid, ppv);
 211 }
 212
 213 int CTextPassThruFilter::GetPinCount()
 214 {
 215         return 2;
 216 }
 217
 218 CBasePin* CTextPassThruFilter::GetPin(int n)
 219 {
 220         if(n == 0) return m_pInput;
 221         else if(n == 1) return m_pOutput;
 222         return NULL;
 223 }
 224
 225 /*
 226 void CTextPassThruFilter::SetName()
 227 {
 228         CRenderedTextSubtitle* pRTS = (CRenderedTextSubtitle*)(ISubStream*)m_pRTS;
 229
 230         CAutoLock cAutoLock(&m_pMainFrame->m_csSubLock);
 231
 232         if(CComQIPtr<IPropertyBag> pPB = m_pTPTInput->GetConnected())
 233         {
 234                 CComVariant var;
 235                 if(SUCCEEDED(pPB->Read(L"LANGUAGE", &var, NULL)))
 236                 {
 237                         pRTS->m_name = CString(var.bstrVal) + _T(" (embeded)");
 238                 }
 239         }
 240
 241         if(pRTS->m_name.IsEmpty())
 242         {
 243                 CPinInfo pi;
 244                 m_pTPTInput->GetConnected()->QueryPinInfo(&pi);
 245                 pRTS->m_name = CString(CStringW(pi.achName)) + _T(" (embeded)");
 246
 247         }
 248 }
 249 */