BaseSource.h
上传用户:xjjlds
上传日期:2015-12-05
资源大小:22823k
文件大小:3k
源码类别:

多媒体编程

开发平台:

Visual C++

  1. /* 
  2.  * Copyright (C) 2003-2005 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. #pragma once
  22. #include "......DSUtilDSUtil.h"
  23. template<class TStream>
  24. class CBaseSource
  25. : public CSource
  26. , public IFileSourceFilter
  27. , public IAMFilterMiscFlags
  28. {
  29. protected:
  30. CStringW m_fn;
  31. public:
  32. CBaseSource(TCHAR* name, LPUNKNOWN lpunk, HRESULT* phr, const CLSID& clsid)
  33. : CSource(name, lpunk, clsid)
  34. {
  35. if(phr) *phr = S_OK;
  36. }
  37. DECLARE_IUNKNOWN;
  38.     STDMETHODIMP NonDelegatingQueryInterface(REFIID riid, void** ppv)
  39. {
  40. CheckPointer(ppv, E_POINTER);
  41. return 
  42. QI(IFileSourceFilter)
  43. QI(IAMFilterMiscFlags)
  44. __super::NonDelegatingQueryInterface(riid, ppv);
  45. }
  46. // IFileSourceFilter
  47. STDMETHODIMP Load(LPCOLESTR pszFileName, const AM_MEDIA_TYPE* pmt)
  48. {
  49. // TODO: destroy any already existing pins and create new, now we are just going die nicely instead of doing it :)
  50. if(GetPinCount() > 0)
  51. return VFW_E_ALREADY_CONNECTED;
  52. HRESULT hr = S_OK;
  53. if(!(new TStream(pszFileName, this, &hr)))
  54. return E_OUTOFMEMORY;
  55. if(FAILED(hr))
  56. return hr;
  57. m_fn = pszFileName;
  58. return S_OK;
  59. }
  60. STDMETHODIMP GetCurFile(LPOLESTR* ppszFileName, AM_MEDIA_TYPE* pmt)
  61. {
  62. if(!ppszFileName) return E_POINTER;
  63. if(!(*ppszFileName = (LPOLESTR)CoTaskMemAlloc((m_fn.GetLength()+1)*sizeof(WCHAR))))
  64. return E_OUTOFMEMORY;
  65. wcscpy(*ppszFileName, m_fn);
  66. return S_OK;
  67. }
  68. // IAMFilterMiscFlags
  69. STDMETHODIMP_(ULONG) GetMiscFlags()
  70. {
  71. return AM_FILTER_MISC_FLAGS_IS_SOURCE;
  72. }
  73. };
  74. class CBaseStream 
  75. : public CSourceStream
  76. , public CSourceSeeking
  77. {
  78. protected:
  79. CCritSec m_cSharedState;
  80. REFERENCE_TIME m_AvgTimePerFrame;
  81. REFERENCE_TIME m_rtSampleTime, m_rtPosition;
  82. BOOL m_bDiscontinuity, m_bFlushing;
  83. HRESULT OnThreadStartPlay();
  84. HRESULT OnThreadCreate();
  85. private:
  86. void UpdateFromSeek();
  87. STDMETHODIMP SetRate(double dRate);
  88. HRESULT ChangeStart();
  89.     HRESULT ChangeStop();
  90.     HRESULT ChangeRate() {return S_OK;}
  91. public:
  92.     CBaseStream(TCHAR* name, CSource* pParent, HRESULT* phr);
  93. virtual ~CBaseStream();
  94.     STDMETHODIMP NonDelegatingQueryInterface(REFIID riid, void** ppv);
  95.     HRESULT FillBuffer(IMediaSample* pSample);
  96.     virtual HRESULT FillBuffer(IMediaSample* pSample, int nFrame, BYTE* pOut, long& len /*in+out*/) = 0;
  97. STDMETHODIMP Notify(IBaseFilter* pSender, Quality q);
  98. };