playwnddlg.cpp

这是我写的一个DSHOW用FFDSHOW解码 播放一个AVI文件小例子 欢迎跟初学者一起探讨 高手可以无视

// PlayWndDlg.cpp : 实现文件
//

#include "stdafx.h"
#include "PlayWnd.h"
#include "PlayWndDlg.h"


#ifdef _DEBUG
#define new DEBUG_NEW
#endif

#define WM_GRAPHNOTIFY WM_USER + 101

// 用于应用程序“关于”菜单项的 CAboutDlg 对话框

class CAboutDlg : public CDialog
{
public:
	CAboutDlg();

// 对话框数据
	enum { IDD = IDD_ABOUTBOX };

	protected:
	virtual void DoDataExchange(CDataExchange* pDX);    // DDX/DDV 支持

// 实现
protected:
	DECLARE_MESSAGE_MAP()
};

CAboutDlg::CAboutDlg() : CDialog(CAboutDlg::IDD)
{
}

void CAboutDlg::DoDataExchange(CDataExchange* pDX)
{
	CDialog::DoDataExchange(pDX);
}

BEGIN_MESSAGE_MAP(CAboutDlg, CDialog)
END_MESSAGE_MAP()


// CPlayWndDlg 对话框




CPlayWndDlg::CPlayWndDlg(CWnd* pParent /*=NULL*/)
	: CDialog(CPlayWndDlg::IDD, pParent)
{
	m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);

    m_pGraph = NULL;
    m_pMediaControl = NULL;
    m_pEvent = NULL;
    m_pMediaSeeking = NULL;
    m_strMediaFile = "";
    m_isPlaying = FALSE;
}

void CPlayWndDlg::DoDataExchange(CDataExchange* pDX)
{
	CDialog::DoDataExchange(pDX);
}

BEGIN_MESSAGE_MAP(CPlayWndDlg, CDialog)
	ON_WM_SYSCOMMAND()
	ON_WM_PAINT()
	ON_WM_QUERYDRAGICON()
	//}}AFX_MSG_MAP
    ON_BN_CLICKED(IDC_BROWSE_BUTTON, &CPlayWndDlg::OnBnClickedBrowseButton)
    ON_BN_CLICKED(IDC_PLAY_BUTTON, &CPlayWndDlg::OnBnClickedPlayButton)
    ON_BN_CLICKED(IDC_PAUSE_BUTTON, &CPlayWndDlg::OnBnClickedPauseButton)
    ON_BN_CLICKED(IDCANCEL, &CPlayWndDlg::OnBnClickedCancel)
    ON_MESSAGE(WM_GRAPHNOTIFY, OnGraphNotify)
END_MESSAGE_MAP()


// CPlayWndDlg 消息处理程序

BOOL CPlayWndDlg::OnInitDialog()
{
	CDialog::OnInitDialog();

	// 将“关于...”菜单项添加到系统菜单中。

	// IDM_ABOUTBOX 必须在系统命令范围内。
	ASSERT((IDM_ABOUTBOX & 0xFFF0) == IDM_ABOUTBOX);
	ASSERT(IDM_ABOUTBOX < 0xF000);

	CMenu* pSysMenu = GetSystemMenu(FALSE);
	if (pSysMenu != NULL)
	{
		CString strAboutMenu;
		strAboutMenu.LoadString(IDS_ABOUTBOX);
		if (!strAboutMenu.IsEmpty())
		{
			pSysMenu->AppendMenu(MF_SEPARATOR);
			pSysMenu->AppendMenu(MF_STRING, IDM_ABOUTBOX, strAboutMenu);
		}
	}

	// 设置此对话框的图标。当应用程序主窗口不是对话框时，框架将自动
	//  执行此操作
	SetIcon(m_hIcon, TRUE);			// 设置大图标
	SetIcon(m_hIcon, FALSE);		// 设置小图标

	// TODO: 在此添加额外的初始化代码
    ModifyStyle(0, WS_CLIPCHILDREN);
((CEdit*)GetDlgItem(IDC_MEDIAFILE_EDIT))->SetReadOnly(TRUE); 


	return TRUE;  // 除非将焦点设置到控件，否则返回 TRUE
}

void CPlayWndDlg::OnSysCommand(UINT nID, LPARAM lParam)
{
	if ((nID & 0xFFF0) == IDM_ABOUTBOX)
	{
		CAboutDlg dlgAbout;
		dlgAbout.DoModal();
	}
	else
	{
		CDialog::OnSysCommand(nID, lParam);
	}
}

// 如果向对话框添加最小化按钮，则需要下面的代码
//  来绘制该图标。对于使用文档/视图模型的 MFC 应用程序，
//  这将由框架自动完成。

void CPlayWndDlg::OnPaint()
{
	if (IsIconic())
	{
		CPaintDC dc(this); // 用于绘制的设备上下文

		SendMessage(WM_ICONERASEBKGND, reinterpret_cast<WPARAM>(dc.GetSafeHdc()), 0);

		// 使图标在工作矩形中居中
		int cxIcon = GetSystemMetrics(SM_CXICON);
		int cyIcon = GetSystemMetrics(SM_CYICON);
		CRect rect;
		GetClientRect(&rect);
		int x = (rect.Width() - cxIcon + 1) / 2;
		int y = (rect.Height() - cyIcon + 1) / 2;

		// 绘制图标
        dc.DrawIcon(x, y, m_hIcon);
	}
	else
	{
        if(m_isPlaying == FALSE)
        {
            CClientDC dc(GetDlgItem(IDC_VW_FRAME));
            dc.SetBkColor(RGB(0,0,0));
            CRect rc; 
            GetDlgItem(IDC_VW_FRAME)->GetClientRect(rc);

            //ClientToScreen(rc);

            dc.FillRect(rc, CBrush::FromHandle((HBRUSH)GetStockObject(BLACK_BRUSH)));
            GetDlgItem(IDC_VW_FRAME)->Invalidate();
        }

		CDialog::OnPaint();
	}
}

//当用户拖动最小化窗口时系统调用此函数取得光标显示。
//
HCURSOR CPlayWndDlg::OnQueryDragIcon()
{
	return static_cast<HCURSOR>(m_hIcon);
}

BOOL CPlayWndDlg::DestroyWindow()
{
    // TODO: 在此添加专用代码和/或调用基类

    if(m_pGraph)
    m_pGraph->Release();
    if(m_pMediaControl)
    m_pMediaControl->Release();
    if(m_pEvent)
    m_pEvent->Release();
    if(m_pMediaSeeking)
    m_pMediaSeeking->Release();

    m_pGraph = NULL;
    m_pMediaControl = NULL;
    m_pEvent = NULL;
    m_pMediaSeeking = NULL;
    return CDialog::DestroyWindow();
}


void CPlayWndDlg::OnBnClickedBrowseButton()
{
    m_strMediaFile = "D:\\test.avi";
    GetDlgItem(IDC_MEDIAFILE_EDIT)->SetWindowText(m_strMediaFile);

    HRESULT hr = CoCreateInstance(CLSID_FilterGraph, NULL, CLSCTX_INPROC, IID_IGraphBuilder, (void **)&m_pGraph);

    //hr = m_pGraph->RenderFile(LPCWSTR(m_strMediaFile), NULL);

    // Source Filter
    IBaseFilter* pFilter;
    hr = m_pGraph->AddSourceFilter(LPCWSTR(m_strMediaFile),LPCWSTR("test"),&pFilter);

    // AVI Splitter Filter
    IBaseFilter *paviSplitterF = NULL; 
    hr = CoCreateInstance(CLSID_AviSpliter, NULL, CLSCTX_INPROC, IID_IBaseFilter, (void **)&paviSplitterF);
         if(SUCCEEDED(hr))
     {
              hr = m_pGraph->AddFilter(paviSplitterF, LPCWSTR("aviSplitter"));
         if(SUCCEEDED(hr))
         {
                 hr = ConnectFilters(m_pGraph, pFilter,paviSplitterF);

             paviSplitterF->Release();
         }
         pFilter->Release();
     }


    // H264 Transform Filter
    IBaseFilter *pFFdshowH264 = NULL; 
    hr = CoCreateInstance(CLSID_FFdshowFilter, 0, CLSCTX_INPROC, IID_IBaseFilter, (void **)&pFFdshowH264);
     if(SUCCEEDED(hr))
     {
         hr = m_pGraph->AddFilter(pFFdshowH264, LPCWSTR("ffdshow"));
         if(SUCCEEDED(hr))
         {
             hr = ConnectFilters(m_pGraph, paviSplitterF,pFFdshowH264);
             pFFdshowH264->Release();
         }
         pFilter->Release();
     }

    IPin * pPin;   

   IEnumPins *pEnum = NULL;

   hr = pFFdshowH264->EnumPins(&pEnum);

   while(pEnum->Next(1,&pPin,NULL)==S_OK)
   {
          PIN_DIRECTION PinDir;
          hr = pPin->QueryDirection(&PinDir);

          if(PinDir == PINDIR_OUTPUT)
          {
                 IPin *pTemp;

                 if(SUCCEEDED(pPin->ConnectedTo(&pTemp)))
                 {
                        pTemp->Release();
                 }
                 else
                 {
                        hr = m_pGraph->Render(pPin); 
                 }
          }
   }



     // IBaseFilter *pVMRRender = NULL; 
     //hr = CoCreateInstance(CLSID_VMRRenderer,NULL,CLSCTX_INPROC_SERVER,IID_IBaseFilter,(void **)&pVMRRender); 
     //if(SUCCEEDED(hr))
     //{
     //      hr = m_pGraph->AddFilter(pVMRRender, LPCWSTR("VMRRender"));
     //    if(SUCCEEDED(hr))
     //    {
     //            hr = ConnectFilters(m_pGraph, pFFdshowH264,pVMRRender);

     //        pVMRRender->Release();
     //    }
     //    pFFdshowH264->Release();
     //}

    if(FAILED(hr))
    {
        char szMsg[200];
        AMGetErrorText(hr, LPWSTR(szMsg), sizeof(szMsg));
        AfxMessageBox(LPCTSTR(szMsg));
    }

    //指定父窗体
    IVideoWindow* pVidWin = NULL;
    m_pGraph->QueryInterface(IID_IVideoWindow, (void **)&pVidWin);
    pVidWin->put_Owner((OAHWND)m_hWnd);
    pVidWin->put_WindowStyle(WS_CHILD | WS_CLIPSIBLINGS | WS_CLIPCHILDREN);

    CRect rc;

    GetDlgItem(IDC_VW_FRAME)->GetWindowRect(rc);
    ScreenToClient(rc);
    pVidWin->SetWindowPosition(rc.left, rc.top, rc.Width(), rc.Height());

    // 注意此处Filter Graph Manager的事件以WM_GRAPHNOTIFY发出(用户定义的消息).
    m_pGraph->QueryInterface(IID_IMediaEventEx, (void **)&m_pEvent);
    m_pEvent->SetNotifyWindow((OAHWND)m_hWnd, WM_GRAPHNOTIFY, 0);

    // 设置Seeking
    m_pGraph->QueryInterface(IID_IMediaSeeking, (void **)&m_pMediaSeeking);

    //Play();
}

void CPlayWndDlg::OnBnClickedPlayButton()
{
    // TODO: 在此添加控件通知处理程序代码
    Play();
}

void CPlayWndDlg::OnBnClickedPauseButton()
{
    // TODO: 在此添加控件通知处理程序代码
    m_pMediaControl->Pause();
    m_isPlaying = TRUE;

    GetDlgItem(IDC_PLAY_BUTTON)->EnableWindow(TRUE);
    GetDlgItem(IDC_PAUSE_BUTTON)->EnableWindow(FALSE);

}

void CPlayWndDlg::OnBnClickedCancel()
{
    // TODO: 在此添加控件通知处理程序代码
    CleanUp();
    OnCancel();
}

// IDC_VW_FRAME控件Picture Control主要作用是控制Vedio Window的显示位置
void CPlayWndDlg::MoveVideoWindow(void)
{
    IVideoWindow* pVideoWinow = NULL;
    if(m_pGraph)
    {
        m_pGraph->QueryInterface(IID_IVideoWindow, (void **)&pVideoWinow);
        CRect rc;
        GetDlgItem(IDC_VW_FRAME)->GetWindowRect(rc);
        ScreenToClient(rc);
        pVideoWinow->SetWindowPosition(rc.left, rc.top, rc.Width(), rc.Height());
        pVideoWinow->Release();
        pVideoWinow = NULL;
    }
}

void CPlayWndDlg::CleanUp(void)
{
    long levCode;
    IVideoWindow *pVidWin = NULL;

    if(!m_pGraph)
    return;

    m_pGraph->QueryInterface(IID_IVideoWindow, (void **)&pVidWin);
    m_pEvent->WaitForCompletion(INFINITE, &levCode);

    pVidWin->put_Visible(OAFALSE);
    pVidWin->Release();
    m_pMediaSeeking->Release();
    m_pMediaControl->Release();
    m_pEvent->Release();
    m_pGraph->Release();
    m_pMediaSeeking = NULL;
    m_pMediaControl = NULL;
    m_pEvent = NULL;
    m_pGraph = NULL;

    UpdateData(FALSE);
    CClientDC dc(GetDlgItem(IDC_VW_FRAME));
    dc.SetBkColor(RGB(0,0,0));

    CRect rc;

    GetDlgItem(IDC_VW_FRAME)->GetClientRect(rc);
    ClientToScreen(rc);
    dc.FillRect(rc, CBrush::FromHandle((HBRUSH)GetStockObject(BLACK_BRUSH)));
    Invalidate();
}

BOOL CPlayWndDlg::Stop(void)
{
    IVideoWindow *pVidWin = NULL;
    HRESULT hr;

    if(m_pMediaControl)
    {
    LONGLONG pos = 0;
    hr = m_pMediaControl->Stop();
    hr = m_pMediaSeeking->SetPositions(&pos, AM_SEEKING_AbsolutePositioning ,&pos, AM_SEEKING_NoPositioning);

    m_pGraph->QueryInterface(IID_IVideoWindow, (void **)&pVidWin);
    pVidWin->put_Visible(OAFALSE);

    m_isPlaying = FALSE;
    GetDlgItem(IDC_PLAY_BUTTON)->EnableWindow(TRUE);
    GetDlgItem(IDC_PAUSE_BUTTON)->EnableWindow(FALSE);
    pVidWin->Release();
    long levCode;
    m_pEvent->WaitForCompletion(INFINITE, &levCode);
    m_pMediaControl->Release();

    return TRUE;
    }
    return FALSE;
}

BOOL CPlayWndDlg::Play(void)
{
    // 运行
    IVideoWindow *pVidWin = NULL;

    if(m_pGraph)
    {
        m_pGraph->QueryInterface(IID_IVideoWindow, (void **)&pVidWin);
        pVidWin->put_Visible(OATRUE);
        m_pGraph->QueryInterface(IID_IMediaControl, (void **)&m_pMediaControl);
        m_pMediaControl->Run();
        m_isPlaying = TRUE;
        GetDlgItem(IDC_PLAY_BUTTON)->EnableWindow(FALSE);
        GetDlgItem(IDC_PAUSE_BUTTON)->EnableWindow(TRUE);

        return TRUE;
    }
    return FALSE;
}

HRESULT CPlayWndDlg::OnGraphNotify(WPARAM wParam,LPARAM lParam)
{
    long levCode, lparam1, lparam2;
    HRESULT hr;

    while (hr = m_pEvent->GetEvent(&levCode, &lparam1, &lparam2, 0), SUCCEEDED(hr))
    {
        hr = m_pEvent->FreeEventParams(levCode, lparam1, lparam2);
        if ((EC_COMPLETE == levCode) || (EC_USERABORT == levCode))
        {
            TRACE("End of the media file!!.\n");
            Stop();

            //CleanUp();

            break;
        }
    }
    return hr;
}

// 得到Filter上的未连接Pin
HRESULT CPlayWndDlg::GetUnconnectedPin(IBaseFilter *pFileter, PIN_DIRECTION PinDir, IPin **ppPin)
{
    *ppPin =0;
    IEnumPins *pEnum = 0;
    IPin *pPin =0;
    HRESULT hr = pFileter->EnumPins(&pEnum);
    if(FAILED(hr))
    {
        return hr;
    }
    while(pEnum->Next(1, &pPin, NULL) == S_OK)
    {
        PIN_DIRECTION ThisPinDir;
        pPin->QueryDirection(&ThisPinDir);
        if(ThisPinDir == PinDir)
        {
            IPin *pTmp= 0;
            hr =pPin->ConnectedTo(&pTmp);
            if(SUCCEEDED(hr))
            {
                pTmp->Release();
            }
            else
            {
                pEnum->Release();
                *ppPin =pPin;
                return S_OK;
            }
            pPin->Release();
        }
        pEnum->Release();
        return E_FAIL;
    }
}

HRESULT CPlayWndDlg::ConnectFilters( IGraphBuilder *pGraph, IPin *pOut, IBaseFilter *pDest)
{
    if((pGraph == NULL) || (pOut == NULL) || (pDest == NULL))
    {
        return E_POINTER;
    }
#ifdef debug
     PIN_DIRECTION PinDir;
     pOut->QueryDirection(&PinDir);
     _ASSERTE(PinDir == PINDIR_OUTPUT);
#endif
     //得到下一级Filter的输入Pin
     IPin *pIn = 0;
     HRESULT hr =GetUnconnectedPin(pDest, PINDIR_INPUT, &pIn);
     if((FAILED(hr)))
     {
         return hr;
     }
     //将输入Pin链接到输出Pin
     hr = pGraph->Connect(pOut,pIn);
     pIn->Release();
     return hr;
}
// 不同参数的ConnectFilters函数重载形式
HRESULT CPlayWndDlg::ConnectFilters( IGraphBuilder *pGraph,  IBaseFilter *pSrc, IBaseFilter *pDest)
{
     if((pGraph == NULL) || (pSrc == NULL) || (pDest == NULL))
    {
        return E_POINTER;
    }
     //Find an out pin on the fist fliter
     IPin *pOut =0;
     HRESULT hr = GetUnconnectedPin(pSrc, PINDIR_OUTPUT, &pOut);
     if((FAILED(hr)))
     {
         return hr;
     }
     hr = ConnectFilters(pGraph, pOut, pDest);
     pOut->Release();
     return hr;
}