分享
 
 
 

DShow中实现抓图的几种方法

王朝other·作者佚名  2008-06-01
窄屏简体版  字體: |||超大  

1.加入Sample Grabber Filter

当我们加入Sample Grabber Filter的时候,我们可以直接调用其接口(interface)ISampleGrabber。该接口可以获取经过该Filter的单独的Media Samples。详情请参见DXSDK。

1.1 派生出自己的Sample Grabber

从ISampleGrabberCB中派生出自己的类,然后实现其虚函数,详情请参见SDK中的示例程序(DXSDK ROOT\Samples\C++\DirectShow\Editing\GrabBitmaps)。

1.2 直接调用Sample Grabber Filter的接口

假如我们在播放的过程中动态的加入Filter的话,操作和效率都不乐观。所以我采用下面的方法:

该方法传递的是时间,不是在播放的时候动态加入Filter然后截图,而是另外打开源文件进行操作。

A)申明以下接口:

#001 IGraphBuilder *pGraph = NULL; //for graph builder

#002 IMediaControl *pControl = NULL; //media control

#003 IMediaSeeking *pSeeking = NULL; //media seeking

#004 IMediaEventEx *pEvent = NULL; //media envent

#005 IBaseFilter *pNullFilter =NULL;//for holding the Sample grabber Filter

B)初始化接口:

#001 JIF(CoCreateInstance(CLSID_FilterGraph, NULL, CLSCTX_INPROC,

#002 IID_IGraphBuilder, (void **)&pGraph));

#003

#004 JIF(CoCreateInstance(CLSID_NullRenderer, NULL, CLSCTX_INPROC,

#005 IID_IBaseFilter, (void **)&pNullFilter));

#006 JIF(pGraph->QueryInterface(IID_IMediaControl,(void **)&pControl));

#007 JIF(pGraph->QueryInterface(IID_IMediaSeeking, (void **)&pSeeking));

#008 JIF(pGraph->QueryInterface(IID_IMediaEvent, (void **)&pEvent));

C)创建Sample Grabber

#001 // Create the Sample Grabber.

#002 IBaseFilter *pGrabberF = NULL;

#003 JIF(CoCreateInstance(CLSID_SampleGrabber,NULL, CLSCTX_INPROC_SERVER,

#004 IID_IBaseFilter, (void**)&pGrabberF));

#005

#006 JIF(pGraph->AddFilter(pGrabberF, L"Sample Grabber"));

#007 JIF(pGraph->AddFilter(pNullFilter, L"Null Render Filter"));

#008

#009 ISampleGrabber *pGrabber;

#010 JIF(pGrabberF->QueryInterface(IID_ISampleGrabber, (void**)&pGrabber));

设置Sample Grabber的媒体格式:调用SetMediaType,该函数接受一个AM_MEDIA_TYPE的结构,主要是设置该结构中的majortype,和suBType域。

D)添加Source Filter:

#001 IBaseFilter *pSrc;

#002 JIF(pGraph->AddSourceFilter(T2W(m_szFile), L"Source", &pSrc));

E)连接Grabber 和 NullRender两个Filter:

#001 IPin *pOutPin;

#002 hr = GetPin(pGrabberF, PINDIR_OUTPUT, &pOutPin);

#003

#004 IPin *pInPin;

#005 hr = GetPin(pNullFilter, PINDIR_INPUT, &pInPin);

#006

#007 pGraph->Connect(pOutPin, pInPin);

F)取得当前所连接媒体的类型

#001 AM_MEDIA_TYPE mt;

#002 hr = pGrabber->GetConnectedMediaType(&mt);

#003 // Examine the format block.

#004 VIDEOINFOHEADER *pVih;

#005 if ((mt.formattype == FORMAT_VideoInfo) &&

#006 (mt.cbFormat >= sizeof(VIDEOINFOHEADER)) &&

#007 (mt.pbFormat != NULL) )

#008 {

#009 pVih = (VIDEOINFOHEADER*)mt.pbFormat;

#010 }

#011 else

#012 {

#013 // Wrong format. Free the format block and return an error.

#014 FreeMediaType(mt);

#015 return VFW_E_INVALIDMEDIATYPE;

#016 }

#017

#018 // Do buffer the samples as they pass through

#019 //

#020 hr = pGrabber->SetBufferSamples(TRUE);

#021

#022 // Only grab one at a time, stop stream after

#023 // grabbing one sample

#024 //

#025 hr = pGrabber->SetOneShot( TRUE );

G)Seeking文件,使其到达要截图的时间帧

#001 pSeeking->SetPositions(pCurrentPos,

#002 AM_SEEKING_AbsolutePositioning,

#003 NULL, AM_SEEKING_NoPositioning );

#004

#005 pControl->Run();

#006

#007 long EvCode = 0;

#008

#009 hr = pEvent->WaitForCompletion( INFINITE, &EvCode );

H)取得当前的buffer数据

#001 // Find the required buffer size.

#002 long cbBuffer = 0;

#003 hr = pGrabber->GetCurrentBuffer(&cbBuffer, NULL);

#004 LONGLONG currentPos;

#005 pSeeking->GetCurrentPosition(&currentPos);

#006 BYTE *pBuffer = new BYTE[cbBuffer];

#007 if (!pBuffer)

#008 {

#009 // Out of memory. Return an error code.

#010 Msg("Out of Memory");

#011 }

#012 hr = pGrabber->GetCurrentBuffer(&cbBuffer, (long*)pBuffer);

I)写入文件

#001 // Create a file to hold the bitmap

#002 HANDLE hf = CreateFile(szFilename, GENERIC_WRITE, FILE_SHARE_READ,

#003 NULL, CREATE_ALWAYS, NULL, NULL );

#004

#005 if( hf == INVALID_HANDLE_VALUE )

#006 {

#007 // Failed to create file

#008 return 0;

#009 }

#010

#011 // Write out the file header

#012 //

#013 BITMAPFILEHEADER bfh;

#014 memset( &bfh, 0, sizeof( bfh ) );

#015 bfh.bfType = 'MB';

#016 bfh.bfSize = sizeof( bfh ) + cbBuffer + sizeof( BITMAPINFOHEADER );

#017 bfh.bfOffBits = sizeof(BITMAPINFOHEADER)+sizeof( BITMAPFILEHEADER );

#018

#019 DWord Written = 0;

#020 WriteFile( hf, &bfh, sizeof( bfh ), &Written, NULL );

#021

#022 // Write the bitmap format

#023 //

#024 BITMAPINFOHEADER bih;

#025 memset( &bih, 0, sizeof( bih ) );

#026 bih.biSize = sizeof( bih );

#027 bih.biWidth = pVih->bmiHeader.biWidth;

#028 bih.biHeight = pVih->bmiHeader.biHeight;

#029 bih.biPlanes = pVih->bmiHeader.biPlanes;

#030 bih.biBitCount = pVih->bmiHeader.biBitCount;

#031

#032 Written = 0;

#033

#034 WriteFile( hf, &bih, sizeof( bih ), &Written, NULL );

#035

#036 // Write the bitmap bits

#037 //

#038 Written = 0;

#039 WriteFile( hf, pBuffer, cbBuffer, &Written, NULL );

#040 FreeMediaType(mt);

#041 CloseHandle(hf);

J)释放资源

#001 pControl->Stop();

#002 SAFE_RELEASE(pControl);

#003 SAFE_RELEASE(pSeeking);

#004 SAFE_RELEASE(pEvent);

#005 SAFE_RELEASE(pSrc);

#006 SAFE_RELEASE(pNullFilter);

#007 SAFE_RELEASE(pGrabber);

#008 SAFE_RELEASE(pGrabberF);

#009 SAFE_RELEASE(pGraph);

K)其实我们可以不用NullRender,而是用IVideoWindow接口来实现。假如是那样的话,首先申明IVideoWindow *pVideo = NULL;将pVideo加入到Filter Graph中

#001 JIF(pGraph->QueryInterface(IID_IVideoWindow,(void**)&pVideo));

#002 hr = pGraph->Render(pOutPin);

#003 if (pVideo)

#004 {

#005 hr = pVideo->put_AutoShow(OAFALSE);

#006 }

通过IBasicVideo::GetCurrentImage接口

对于该接口,Video Renderer和Video Mixing Renderer(VMR)有不同的实现。

A)Video Renderer

假如该Renderer使用了DDraw加速的话,该调用会失败。在调用该接口的时候,必须首先暂停Renderer(可以通过IMediaControl::Pause()暂停,假如不能确信该操作是否成功,应该调用IMediaControl::GetState()判定状态)。

B)Video Mixing Renderer

对于VMR,该方法都会成功(不管是否运用了DDraw加速,也不管是否是暂停状态),此时对于它所有的状态(running, stopped, or paused)都适用。

函数Grabber代码如下(调用该函数的时候应该先将媒体文件暂停,原因上面已经说了):

#001 bool Grabber(IBasicVideo mBasicVideo, TCHAR *szFilename)

#002 {

#003 if (mBasicVideo)

#004 {

#005 long bitmapSize = 0;

#006 if(SUCCEEDED(mBasicVideo->GetCurrentImage(&bitmapSize, 0)))

#007 {

#008 //if语句里面的操作时取得buffer的size。

#009 //当我们在布确定image buffer的大小的情况下,我们给

#010 //GetCurrentImage的第二个参数传递0或者NULL,取得buffer的

#011 //大小供以后使用。

#012 bool pass = false;

#013 unsigned char * buffer = new unsigned char[bitmapSize];

#014 if(SUCCEEDED(mBasicVideo->GetCurrentImage(&bitmapSize,(long*)buffer)))

#015 {

#016 //此时已经用到刚才所取得的大小(分配空间)

#017 BITMAPFILEHEADER hdr; //Bitmap的头信息

#018 LPBITMAPINFOHEADER lpbi; // Bitmap的文件信息(包括数据)

#019

#020 lpbi = (LPBITMAPINFOHEADER)buffer;

#021

#022 int nColors = 1 << lpbi->biBitCount;

#023 if (nColors > 256)

#024 nColors = 0;

#025

#026 hdr.bfType = ((WORD) ('M' << 8) 'B'); //always is "BM"

#027 hdr.bfSize = bitmapSize + sizeof( hdr );

#028 hdr.bfReserved1 = 0;

#029 hdr.bfReserved2 = 0;

#030 hdr.bfOffBits = (DWORD) (sizeof(BITMAPFILEHEADER) + lpbi->biSize

#031 CFile bitmapFile(outFile, CFile::modeReadWrite CFile::modeCreate CFile::typeBinary);

#032 bitmapFile.Write(&hdr, sizeof(BITMAPFILEHEADER));

#033 bitmapFile.Write(buffer, bitmapSize);

#034 bitmapFile.Close();

#035 pass = true;

#036 }

#037 delete [] buffer; //数据用过之后记得要释放空间

#038 return true;

#039 }

#040 }

#041

#042 return false;

#043 }

IMediaDet接口

IMediaDet接口可以取得媒体文件的信息。SDK里面的示例代码(没有写入文件):

#001 long size;

#002 //取得图像帧的大小,给GetBitmapBits的第三个参数传递0 or NULL

#003 hr = pDet->GetBitmapBits(0, &size, 0, width, height);

#004 if(SUCCEEDED(hr))

#005 {

#006 char *pBuffer = new char[size];

#007 if(!pBuffer)

#008 {

#009 return E_OUTOFMEMORY;

#010 }

#011

#012 try

#013 {

#014 hr = pDet->GetbitmapsBits(0, 0, pBuffer, width, height);

#015 }

#016 catch(...)

#017 {

#018 delete [] pBuffer;

#019 throw;

#020 }

#021

#022 if(SUCCEEDED(hr))

#023 {

#024 BITMAPINFOHEADER *bmih = (BITMAPINFOHEADER*)pBuffer;

#025 HDC hdcDest = GetDC(0);

#026

#027 //Find the address of the start of the image data

#028 void *pData = pBuffer + sizeof(BITMAPINFOHEADER);

#029

#030 //Note: In general a BITMAPINFOHEADER can include extra color

#031 //information at the end, so calculating the offset to the image

#032 //data i snot generally correct. However, the IMediaDet interface

#033 //always returns an RGB-24 image with no extra color information

#034

#035 BITMAPINFO bmi;

#036 ZeroMemory(&bmi, sizeof(BITMAPINFO));

#037 CopyMemory(&(bmi.bmiHeader), bmih, sizeof(BITMAPINFOHEADER));

#038 HBITMAP hBitmap = CreateDIBitmap(hdcDect, bmih, CBM_INIT,

#039 pData, &bmi, DIB_RGB_COLORS);

#040 }

#041

#042 delete [] pBuffer;

#043 }

该方法并没有写入bitmap,具体的写入过程可以参加上面的几种方法。

 
 
 
免责声明:本文为网络用户发布,其观点仅代表作者个人观点,与本站无关,本站仅提供信息存储服务。文中陈述内容未经本站证实,其真实性、完整性、及时性本站不作任何保证或承诺,请读者仅作参考,并请自行核实相关内容。
2023年上半年GDP全球前十五强
 百态   2023-10-24
美众议院议长启动对拜登的弹劾调查
 百态   2023-09-13
上海、济南、武汉等多地出现不明坠落物
 探索   2023-09-06
印度或要将国名改为“巴拉特”
 百态   2023-09-06
男子为女友送行,买票不登机被捕
 百态   2023-08-20
手机地震预警功能怎么开?
 干货   2023-08-06
女子4年卖2套房花700多万做美容:不但没变美脸,面部还出现变形
 百态   2023-08-04
住户一楼被水淹 还冲来8头猪
 百态   2023-07-31
女子体内爬出大量瓜子状活虫
 百态   2023-07-25
地球连续35年收到神秘规律性信号,网友:不要回答!
 探索   2023-07-21
全球镓价格本周大涨27%
 探索   2023-07-09
钱都流向了那些不缺钱的人,苦都留给了能吃苦的人
 探索   2023-07-02
倩女手游刀客魅者强控制(强混乱强眩晕强睡眠)和对应控制抗性的关系
 百态   2020-08-20
美国5月9日最新疫情:美国确诊人数突破131万
 百态   2020-05-09
荷兰政府宣布将集体辞职
 干货   2020-04-30
倩女幽魂手游师徒任务情义春秋猜成语答案逍遥观:鹏程万里
 干货   2019-11-12
倩女幽魂手游师徒任务情义春秋猜成语答案神机营:射石饮羽
 干货   2019-11-12
倩女幽魂手游师徒任务情义春秋猜成语答案昆仑山:拔刀相助
 干货   2019-11-12
倩女幽魂手游师徒任务情义春秋猜成语答案天工阁:鬼斧神工
 干货   2019-11-12
倩女幽魂手游师徒任务情义春秋猜成语答案丝路古道:单枪匹马
 干货   2019-11-12
倩女幽魂手游师徒任务情义春秋猜成语答案镇郊荒野:与虎谋皮
 干货   2019-11-12
倩女幽魂手游师徒任务情义春秋猜成语答案镇郊荒野:李代桃僵
 干货   2019-11-12
倩女幽魂手游师徒任务情义春秋猜成语答案镇郊荒野:指鹿为马
 干货   2019-11-12
倩女幽魂手游师徒任务情义春秋猜成语答案金陵:小鸟依人
 干货   2019-11-12
倩女幽魂手游师徒任务情义春秋猜成语答案金陵:千金买邻
 干货   2019-11-12
 
推荐阅读
 
 
 
>>返回首頁<<
 
靜靜地坐在廢墟上,四周的荒凉一望無際,忽然覺得,淒涼也很美
© 2005- 王朝網路 版權所有