netwrite.h

wmvnetwrite.介绍将把微软的媒体格式流化到网络端口的代码

//*****************************************************************************
//
// Microsoft Windows Media
// Copyright ( C) Microsoft Corporation. All rights reserved.
//
// FileName:            NetWrite.cpp
//
// Abstract:            CNetWrite class declaration
//                      
//
//*****************************************************************************

#pragma once

#include "wmsdk.h"

#ifndef SAFE_RELEASE

#define SAFE_RELEASE( x )  \
    if( NULL != x )       \
    {                      \
        x->Release( );     \
        x = NULL;          \
    }

#endif

#ifndef SAFE_DELETE
#define SAFE_DELETE( x )	\
    if( NULL != x )		\
    {						\
        delete x;			\
        x = NULL;			\
    }
#endif

#ifndef SAFE_ARRAYDELETE
#define SAFE_ARRAYDELETE( x )	\
    if( NULL != x )		\
    {						\
        delete[] x;			\
        x = NULL;			\
    }
#endif

#define NETWRITE_ASYNC_EVENT	_T( "6d12fe9b-d029-4d08-b2eb-92c8cab323c7" )

class CNetWrite :public IWMReaderCallback, public IWMReaderCallbackAdvanced
{
public:
    CNetWrite();
    ~CNetWrite();
    HRESULT Configure( DWORD dwPortNum, const WCHAR *pwszFile, UINT nMaxClient, const WCHAR *pwszServerURL );
    HRESULT WritetoNet();
    HRESULT Init();
    
	//
	//Methods of IWMReaderCallback
	//
	HRESULT STDMETHODCALLTYPE OnSample( /* [in] */ DWORD dwOutputNum,
		/* [in] */ QWORD cnsSampleTime,
		/* [in] */ QWORD cnsSampleDuration,
		/* [in] */ DWORD dwFlags,
		/* [in] */ INSSBuffer __RPC_FAR *pSample,
		/* [in] */ void __RPC_FAR *pvContext);

	HRESULT STDMETHODCALLTYPE OnStatus( /* [in] */ WMT_STATUS Status,
		/* [in] */ HRESULT hr,
		/* [in] */ WMT_ATTR_DATATYPE dwType,
		/* [in] */ BYTE __RPC_FAR *pValue,
		/* [in] */ void __RPC_FAR *pvContext);


	//
	//Methods of IWMReaderCallbackAdvanced
	//
	HRESULT STDMETHODCALLTYPE OnStreamSample( /* [in] */ WORD wStreamNum,
		/* [in] */ QWORD cnsSampleTime,
		/* [in] */ QWORD cnsSampleDuration,
		/* [in] */ DWORD dwFlags,
		/* [in] */ INSSBuffer __RPC_FAR *pSample,
		/* [in] */ void __RPC_FAR *pvContext);

	HRESULT STDMETHODCALLTYPE OnTime( /* [in] */ QWORD cnsCurrentTime,
		/* [in] */ void __RPC_FAR *pvContext);











	HRESULT STDMETHODCALLTYPE OnStreamSelection( /* [in] */ WORD wStreamCount,
		/* [in] */ WORD __RPC_FAR *pStreamNumbers,
		/* [in] */ WMT_STREAM_SELECTION __RPC_FAR *pSelections,
		/* [in] */ void __RPC_FAR *pvContext)
	{
		return S_OK;
	}

	HRESULT STDMETHODCALLTYPE OnOutputPropsChanged( /* [in] */ DWORD dwOutputNum,
		/* [in] */ WM_MEDIA_TYPE __RPC_FAR *pMediaType,
		/* [in] */ void __RPC_FAR *pvContext)
	{
		return S_OK;
	}

	HRESULT STDMETHODCALLTYPE AllocateForStream( /* [in] */ WORD wStreamNum,
		/* [in] */ DWORD cbBuffer,
		/* [out] */ INSSBuffer __RPC_FAR *__RPC_FAR *ppBuffer,
		/* [in] */ void __RPC_FAR *pvContext)
	{
		return E_NOTIMPL;
	}

	HRESULT STDMETHODCALLTYPE AllocateForOutput( /* [in] */ DWORD dwOutputNum,
		/* [in] */ DWORD cbBuffer,
		/* [out] */ INSSBuffer __RPC_FAR *__RPC_FAR *ppBuffer,
		/* [in] */ void __RPC_FAR *pvContext)
	{
		return E_NOTIMPL;
	}
    
    //
    //Methods of IUnknown
    //
    HRESULT STDMETHODCALLTYPE QueryInterface( REFIID riid,
        void __RPC_FAR *__RPC_FAR *ppvObject) 
    {
        if( riid == IID_IWMReaderCallback )
        {
            *ppvObject = ( IWMReaderCallback* )this;
        }
        else if( riid == IID_IWMReaderCallbackAdvanced )
        {
            *ppvObject = ( IWMReaderCallbackAdvanced* )this;
        }
        else if( riid == IID_IWMStatusCallback )
        {
            *ppvObject = ( IWMStatusCallback* )this;
        }
        else
        {
            return E_NOINTERFACE;
        }
        return S_OK;
    }
    
    ULONG STDMETHODCALLTYPE AddRef( void ) { return 1; }
    
    ULONG STDMETHODCALLTYPE Release( void ) { return 1; }
    
private:
    HRESULT WriteHeader( const WCHAR * pwszName );
    HRESULT WriteScript();
private:		
    HANDLE						m_hEvent;    // Event for handling asynchronous calls
    HRESULT						m_hrAsync;   // Receives error or success codes for asynchronous operations.
    QWORD						m_qwTime;    // Specifies the time interval for the next batch of samples.

    // Miscellaneous interfaces that we need.
    IWMWriterAdvanced*			m_pWriterAdvanced;
    IWMReaderAdvanced*			m_pReaderAdvanced;
    IWMReader*					m_pReader;
    IWMWriter*					m_pWriter;
    IWMWriterNetworkSink*		m_pNetSink;
    IWMWriterPushSink*          m_pPushSink;
    IWMRegisterCallback*        m_pPushSinkCallbackCtrl;
    bool						m_bEOF;
    IWMHeaderInfo*				m_pReaderHeaderInfo;
    IWMHeaderInfo*				m_pWriterHeaderInfo;
};