1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
|
/*
* Copyright (C) 2003-2006 Gabest
* http://www.gabest.org
*
* This Program is free software; you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation; either version 2, or (at your option)
* any later version.
*
* This Program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with GNU Make; see the file COPYING. If not, write to
* the Free Software Foundation, 675 Mass Ave, Cambridge, MA 02139, USA.
* http://www.gnu.org/copyleft/gpl.html
*
*/
#pragma once
#include "..\..\..\DSUtil\DSUtil.h"
template<class TStream>
class CBaseSource
: public CSource
, public IFileSourceFilter
, public IAMFilterMiscFlags
{
protected:
CStringW m_fn;
public:
CBaseSource(TCHAR* name, LPUNKNOWN lpunk, HRESULT* phr, const CLSID& clsid)
: CSource(name, lpunk, clsid)
{
if(phr) *phr = S_OK;
}
DECLARE_IUNKNOWN;
STDMETHODIMP NonDelegatingQueryInterface(REFIID riid, void** ppv)
{
CheckPointer(ppv, E_POINTER);
return
QI(IFileSourceFilter)
QI(IAMFilterMiscFlags)
__super::NonDelegatingQueryInterface(riid, ppv);
}
// IFileSourceFilter
STDMETHODIMP Load(LPCOLESTR pszFileName, const AM_MEDIA_TYPE* pmt)
{
// TODO: destroy any already existing pins and create new, now we are just going die nicely instead of doing it :)
if(GetPinCount() > 0)
return VFW_E_ALREADY_CONNECTED;
HRESULT hr = S_OK;
if(!(DNew TStream(pszFileName, this, &hr)))
return E_OUTOFMEMORY;
if(FAILED(hr))
return hr;
m_fn = pszFileName;
return S_OK;
}
STDMETHODIMP GetCurFile(LPOLESTR* ppszFileName, AM_MEDIA_TYPE* pmt)
{
size_t nCount;
if(!ppszFileName) return E_POINTER;
nCount = m_fn.GetLength()+1;
if(!(*ppszFileName = (LPOLESTR)CoTaskMemAlloc(nCount*sizeof(WCHAR))))
return E_OUTOFMEMORY;
wcscpy_s(*ppszFileName, nCount, m_fn);
return S_OK;
}
// IAMFilterMiscFlags
STDMETHODIMP_(ULONG) GetMiscFlags()
{
return AM_FILTER_MISC_FLAGS_IS_SOURCE;
}
};
class CBaseStream
: public CSourceStream
, public CSourceSeeking
{
protected:
CCritSec m_cSharedState;
REFERENCE_TIME m_AvgTimePerFrame;
REFERENCE_TIME m_rtSampleTime, m_rtPosition;
BOOL m_bDiscontinuity, m_bFlushing;
HRESULT OnThreadStartPlay();
HRESULT OnThreadCreate();
private:
void UpdateFromSeek();
STDMETHODIMP SetRate(double dRate);
HRESULT ChangeStart();
HRESULT ChangeStop();
HRESULT ChangeRate() {return S_OK;}
public:
CBaseStream(TCHAR* name, CSource* pParent, HRESULT* phr);
virtual ~CBaseStream();
STDMETHODIMP NonDelegatingQueryInterface(REFIID riid, void** ppv);
HRESULT FillBuffer(IMediaSample* pSample);
virtual HRESULT FillBuffer(IMediaSample* pSample, int nFrame, BYTE* pOut, long& len /*in+out*/) = 0;
STDMETHODIMP Notify(IBaseFilter* pSender, Quality q);
};
|