在DirectShow的视频图像上叠加线条和文字

在DirectShow的视频图像上叠加线条和文字

最近一直在从事工业测量方面的开发工作，难免会用到各种各样的相机，其中支持DX的USB相机开发起来比较方便，由于工作需要经常要在视频图像上叠加线条和文字，图1便是我最近一段时间写的一套工业检测系统，图像是从USB相机中实时获取的。看到网上有些帖子也在讨论这个问题，现在给出我的一个非常简单的思路并附上源代码(vc++6.0编译通过，需要连接USB相机，可用普通摄像头来代替。地址:http://xiaolang86.download.csdn.net/)。

 

 

图1

用过DirectShow的就知道，用DirectShow连接相机并绘制视频图像十分方便，只需要把当前窗口的句柄传给IVideoWindow过滤器，然后设置一下图像绘制的区域就可以实现了，但是这样一来要在图像窗口上叠加一些线条和文字就比较难了。我的思路是，只用DirectShow来连接相机和获取数据，绘制则自己来写，在DC里面实现。大家可以参考我上传的DDShowDemo的程序。具体步骤如下：

1.     设置环境变量：

DShow里面的静态库很多，如果只是需要连接相机，只有几个库就可以了。根据我个人的实际我只用到了Winmm.lib，strmiids.lib，quartz.lib三个库，第一个好像是windows的库。头文件的话我建议把DX里所有的头文件都拷过来，大家也可以把我的源代码目录下的Include/DX这个文件夹全部拷贝过去。具体在VC6.0里面的设置，大家可以去Project->Setting里面看一下。

2.     连接相机：

(1)首先介绍一下连接相机的类：

连接相机是利用CCaptureVideo类，这个类是根据DX中的Demo改编过来的，在网上都可以下到，我的程序中做了一些改动，大家可以直接将源程序中的CaptureView.cpp和CaptureView.h文件拷贝到自己的工程中。在CCaptureVideo类中其实还嵌套了一个CSampleGrabberCB的友元类，该友元类里的BufferCB函数是用来获取每一帧的图像的。

在CCaptureVideo的cpp文件中引入了两个头文件:

#include "DDShowDemoView.h"

#include "GlobalVar.h"

第一个是要绘制的视图的头文件，大家可以改成自己的；第二个是放一些全局变量的头文件，大家可以去看一下定义，需要介绍的是定义在GlobalVar.h里面的一个结构体，个人认为非常有用：

//struct

typedef struct tagCallBackInfo

{

    double dblSampleTime;

    long lBufferSize;

    BYTE *pBuffer;

    BITMAPINFOHEADER bih;

} CALLBACKINFO;

这个结构里面定义了采样时间，图像大小，图像数据，图像信息头。并定义了一个实例：CALLBACKINFO cb. 定义好之后，如果要对图像进行某些操作，我们关心的就只是cb.pBuffer就可以了，并且这是一个全局变量在程序的任何地方都可以引用，非常方便。

(2)具体代码

在App中定义一个相机实例CCaptureVideo m_cap;注意需要把theApp进行extern声明，这样可以在其它地方访问m_cap，当然也可以把m_cap放到上面说的全局定义文件中。

在View类的InitialUpdate函数中添加如下代码：黑体字所示

void CDDShowDemoView::OnInitialUpdate()

{

    CView::OnInitialUpdate();

    //Contact Camera

    //zhou's comment:if put this->GetSafeHwnd(),

    //DDShow will draw itself, but no vector can be drawed up the buffer,so //give it NULL

    HRESULT hr = theApp.m_cap.Init(0,NULL);

    //Start Invalidate

    SetTimer(0,20,NULL);

}

需要说明的是，如果在Init函数中把当前视图的句柄传进去，那么绘制的工作就可以由DShow来帮你是实现了，这个是我所不希望的，所以我传NULL进去。大家可以跟进Init函数里去看一看，其实真正将图像和当前视图联系起来是函数SetupVideoWindow()，在这个函数里面又需要把视图句柄传进去，如黑体字代码。这不是矛盾吗？大家可以试一下，如果在SetupVideoWindow函数里面也传NULL进去，效果是一样的，但是DX会认为你指定一个视图绘制数据，它会帮你开启一个ActiveWindow的窗口。

HRESULT CCaptureVideo::SetupVideoWindow()

{

    HRESULT hr;

    //can not be NULL, or will active activewindow

    hr = m_pVW->put_Owner((OAHWND)g_pView->m_hWnd);

//  hr = m_pVW->put_Owner((OAHWND)m_hWnd);

    if (FAILED(hr))return hr;

    hr = m_pVW->put_WindowStyle(WS_CHILD);

    if (FAILED(hr))return hr;

    ResizeVideoWindow();

    hr = m_pVW->put_Visible(OATRUE);

    return hr;

}

连接好相机之后，就需要通知视图有视频流数据过来，需要刷新。这里的刷新通过开启一个Timer进程来实现。这个是我最近才改动的，之前在做视频开发的时候，我都是将视图刷新放到CSampleGrabberCB的BufferCB函数中，效果也还不错，但是后来换了一种品牌的相机之后，程序也可以获取数据，但是整个程序占用CPU非常高，达到80%以上，才改成在视图里面开启一个时间进程，在视图内部刷新。大家可以看到在OnTimer函数里面就一句话:Invalidate(false):

void CDDShowDemoView::OnTimer(UINT nIDEvent)

{

    if( nIDEvent == 0 )

       Invalidate( false );

    CView::OnTimer(nIDEvent);

}

这样相机就连接成功，现在需要自己动手写一个绘制函数了。

3.     绘制数据

在视图类中添加一个DrawImgBuffer函数

void CDDShowDemoView::DrawImgBuffer(CDC *pMemDC, BYTE *pBuffer, BITMAPINFOHEADER *pbih, CRect rtInput, CRect rtOutput, HWND hwnd )

{

// If we haven't yet snapped a still, return

if (!pBuffer)

    return;// FALSE;

HDC hdcDraw = pMemDC->GetSafeHdc();//::GetDC( hwnd );

//select into dc

CBitmap *pOldbitmap = (CBitmap*)pMemDC->SelectObject( pBuffer );

PAINTSTRUCT ps;

::BeginPaint(hwnd, &ps);

::SetStretchBltMode(hdcDraw, COLORONCOLOR);

StretchDIBits(

    hdcDraw,

    rtOutput.left, rtOutput.top, rtOutput.Width(), rtOutput.Height(),

    rtInput.top, rtInput.left, rtInput.Width(), rtInput.Height(),

    pBuffer,

    (BITMAPINFO*) pbih,

    DIB_RGB_COLORS,

    SRCCOPY );

::EndPaint(hwnd, &ps);

::ReleaseDC( hwnd, hdcDraw ); 

}

接着在OnDraw函数里面加上这个函数就好了：

void CDDShowDemoView::OnDraw(CDC* pDC)

{

CDDShowDemoDoc* pDoc = GetDocument();

ASSERT_VALID(pDoc);

 

CMemDC *pMemDC = new CMemDC(pDC);

CRect rt(0,0,1280,1024);

if( cb.pBuffer )

{

    DrawImgBuffer( pMemDC, cb.pBuffer, &cb.bih, rt,rt, this->GetSafeHwnd() );

    //just for test:zhoulm's demo

    //you can put your line and text there:

    CFont font;

    font.CreatePointFont( 120, _T("Arial") );

    CFont *pOldFont = pMemDC->SelectObject( &font );

    pMemDC->SetBkMode(TRANSPARENT);

    pMemDC->SetTextColor(RGB(0,255,0));

    pMemDC->TextOut( 100, 100, "zhoulm's Demo! Welcome to:www.whudpcv.cn" );

    pMemDC->SelectObject( pOldFont );

}

if( pMemDC )

{

    delete pMemDC;

    pMemDC = NULL;

}

}

注意这里一定要用MemDC来绘制，而不能用微软自带的DC。MemDC定义在MemDC.h文件中，大家需要把这个文件也加入到工程中来。在这里绘制的时候，就可以看到CALLBACKINFO结构体是多么的方便了，直接访问cb.pBuffer就好了，信息头也直接使用cb.bih就好。

编译，完成！看看最后效果吧：