198 lines
6 KiB
C++
198 lines
6 KiB
C++
/*
|
|
|
|
Copyright (c) 1998-1999 Microsoft Corporation
|
|
|
|
*/
|
|
|
|
// Stream.h : Declaration of the CStream
|
|
|
|
#ifndef __STREAM_H_
|
|
#define __STREAM_H_
|
|
|
|
class CSample;
|
|
|
|
/////////////////////////////////////////////////////////////////////////////
|
|
// CStream
|
|
class ATL_NO_VTABLE CStream :
|
|
public CComObjectRootEx<CComMultiThreadModel>,
|
|
public IPin,
|
|
public IMemInputPin,
|
|
public IAMMediaStream,
|
|
public IMemAllocator
|
|
{
|
|
public:
|
|
typedef CComObjectRootEx<CComMultiThreadModel> _BaseClass;
|
|
DECLARE_GET_CONTROLLING_UNKNOWN()
|
|
//
|
|
// METHODS
|
|
//
|
|
CStream();
|
|
virtual ~CStream();
|
|
|
|
//
|
|
// IMediaStream
|
|
//
|
|
|
|
STDMETHODIMP GetMultiMediaStream(
|
|
/* [out] */ IMultiMediaStream **ppMultiMediaStream);
|
|
|
|
STDMETHODIMP GetInformation(
|
|
/* [optional][out] */ MSPID *pPurposeId,
|
|
/* [optional][out] */ STREAM_TYPE *pType);
|
|
|
|
STDMETHODIMP SendEndOfStream(DWORD dwFlags);
|
|
|
|
//
|
|
// IAMMediaStream
|
|
//
|
|
STDMETHODIMP Initialize(
|
|
IUnknown *pSourceObject,
|
|
DWORD dwFlags,
|
|
/* [in] */ REFMSPID PurposeId,
|
|
/* [in] */ const STREAM_TYPE StreamType);
|
|
|
|
STDMETHODIMP SetState(
|
|
/* [in] */ FILTER_STATE State) = 0; // derived class supplies pump class
|
|
|
|
STDMETHODIMP JoinAMMultiMediaStream(
|
|
/* [in] */ IAMMultiMediaStream *pAMMultiMediaStream);
|
|
|
|
STDMETHODIMP JoinFilter(
|
|
/* [in] */ IMediaStreamFilter *pMediaStreamFilter);
|
|
|
|
STDMETHODIMP JoinFilterGraph(
|
|
/* [in] */ IFilterGraph *pFilterGraph);
|
|
|
|
|
|
//
|
|
// IPin
|
|
//
|
|
STDMETHODIMP Disconnect();
|
|
STDMETHODIMP ConnectedTo(IPin **pPin);
|
|
STDMETHODIMP ConnectionMediaType(AM_MEDIA_TYPE *pmt);
|
|
STDMETHODIMP QueryPinInfo(PIN_INFO * pInfo);
|
|
STDMETHODIMP QueryDirection(PIN_DIRECTION * pPinDir);
|
|
STDMETHODIMP QueryId(LPWSTR * Id);
|
|
STDMETHODIMP QueryAccept(const AM_MEDIA_TYPE *pmt);
|
|
STDMETHODIMP QueryInternalConnections(IPin* *apPin, ULONG *nPin);
|
|
STDMETHODIMP EndOfStream(void);
|
|
STDMETHODIMP BeginFlush(void);
|
|
STDMETHODIMP EndFlush(void);
|
|
STDMETHODIMP NewSegment(REFERENCE_TIME tStart, REFERENCE_TIME tStop, double dRate);
|
|
|
|
//
|
|
// IMemInputPin
|
|
//
|
|
STDMETHODIMP GetAllocator(IMemAllocator ** ppAllocator);
|
|
STDMETHODIMP NotifyAllocator(IMemAllocator * pAllocator, BOOL bReadOnly);
|
|
STDMETHODIMP GetAllocatorRequirements(ALLOCATOR_PROPERTIES*pProps);
|
|
STDMETHODIMP ReceiveMultiple(IMediaSample **pSamples, long nSamples, long *nSamplesProcessed);
|
|
STDMETHODIMP ReceiveCanBlock();
|
|
STDMETHODIMP Connect(IPin * pReceivePin, const AM_MEDIA_TYPE *pmt);
|
|
STDMETHODIMP EnumMediaTypes(IEnumMediaTypes **ppEnum);
|
|
|
|
//
|
|
// IMemAllocator
|
|
//
|
|
STDMETHODIMP Commit() = 0; // derived class supplies pump class
|
|
STDMETHODIMP Decommit() = 0; // derived class supplies pump class
|
|
STDMETHODIMP ReleaseBuffer(IMediaSample *pBuffer);
|
|
|
|
// Note that NotifyAllocator calls this so override it
|
|
// if you care. Audio doesn't care becuase it's not
|
|
// really using this allocator at all.
|
|
STDMETHODIMP SetProperties(
|
|
ALLOCATOR_PROPERTIES* pRequest,
|
|
ALLOCATOR_PROPERTIES* pActual)
|
|
{
|
|
return S_OK;
|
|
}
|
|
STDMETHODIMP GetProperties(ALLOCATOR_PROPERTIES* pProps)
|
|
{
|
|
return E_UNEXPECTED;
|
|
}
|
|
|
|
//
|
|
// Special CStream methods
|
|
//
|
|
virtual HRESULT GetMediaType(ULONG Index, AM_MEDIA_TYPE **ppMediaType) = 0;
|
|
|
|
public:
|
|
//
|
|
// Private methods
|
|
//
|
|
void GetName(LPWSTR);
|
|
HRESULT AllocSampleFromPool(const REFERENCE_TIME * pStartTime, CSample **ppSample, DWORD dwFlags);
|
|
void AddSampleToFreePool(CSample *pSample);
|
|
bool StealSampleFromFreePool(CSample *pSample, BOOL bAbort);
|
|
HRESULT FinalConstruct(void);
|
|
HRESULT ConnectThisMediaType(IPin *pReceivePin, const AM_MEDIA_TYPE *pmt);
|
|
HRESULT CheckReceiveConnectionPin(IPin * pConnector);
|
|
#ifdef DEBUG
|
|
#define CHECKSAMPLELIST TM_ASSERT(CheckSampleList());
|
|
bool CheckSampleList();
|
|
#else
|
|
#define CHECKSAMPLELIST
|
|
#endif
|
|
|
|
BEGIN_COM_MAP(CStream)
|
|
COM_INTERFACE_ENTRY(IPin)
|
|
COM_INTERFACE_ENTRY(IMemInputPin)
|
|
COM_INTERFACE_ENTRY(IMemAllocator)
|
|
COM_INTERFACE_ENTRY2(IMediaStream, IAMMediaStream)
|
|
COM_INTERFACE_ENTRY(IAMMediaStream)
|
|
END_COM_MAP()
|
|
|
|
//
|
|
// MEMBER VARIABLES
|
|
//
|
|
public:
|
|
//
|
|
// These SHOULD NOT BE CCOMPTRS since we hold weak references to both of them
|
|
// (we never addref them).
|
|
//
|
|
IMediaStreamFilter *m_pFilter;
|
|
IBaseFilter *m_pBaseFilter;
|
|
IFilterGraph *m_pFilterGraph;
|
|
IAMMultiMediaStream *m_pMMStream;
|
|
|
|
// Allocator held during connection
|
|
CComPtr<IMemAllocator> m_pAllocator;
|
|
|
|
// Stream configuration
|
|
STREAM_TYPE m_StreamType;
|
|
PIN_DIRECTION m_Direction;
|
|
MSPID m_PurposeId;
|
|
REFERENCE_TIME m_rtSegmentStart;
|
|
|
|
// Allocator state information
|
|
bool m_bUsingMyAllocator;
|
|
bool m_bSamplesAreReadOnly;
|
|
bool m_bCommitted;
|
|
long m_lRequestedBufferCount;
|
|
|
|
// Sample list and semaphores
|
|
CSample *m_pFirstFree;
|
|
CSample *m_pLastFree;
|
|
long m_cAllocated;
|
|
long m_lWaiting;
|
|
HANDLE m_hWaitFreeSem;
|
|
REFERENCE_TIME m_rtWaiting;
|
|
|
|
// Filter state
|
|
FILTER_STATE m_FilterState;
|
|
|
|
// Pin state
|
|
CComPtr<IPin> m_pConnectedPin;
|
|
CComPtr<IQualityControl> m_pQC;
|
|
CComQIPtr<IMemInputPin, &IID_IMemInputPin> m_pConnectedMemInputPin;
|
|
AM_MEDIA_TYPE m_ConnectedMediaType;
|
|
AM_MEDIA_TYPE m_ActualMediaType;
|
|
bool m_bFlushing;
|
|
bool m_bEndOfStream;
|
|
bool m_bStopIfNoSamples;
|
|
};
|
|
|
|
|
|
#endif //__STREAM_H_
|