分享

delphi Directshow实现抓图,抠像

 独孤求财 2012-03-14

delphi Directshow实现抓图,抠像

时间:2011-5-30来源:yang 作者: peng点击: 68次

unit Unit1;

interface

uses
  Windows, Messages, SysUtils, Variants, Classes, Graphics, Controls, Forms,
  Dialogs, StdCtrls,shlobj, ExtCtrls,      {TFlatButtonUnit,TFlatPanelUnit,TFlatTitlebarUnit,}
    FlatBars, FlatBtns, FlatUtils, FlatPanel;

type

 //====================定义窗体类==========================//
  TForm1 = class(TForm)
  //界面控件
    open: TOpenDialog;    //打开对话框控件
    FlatPanel1: TFlatPanel; //FlatPanel控件
    Button1: TFlatButton;    //按钮控件
    Button2: TFlatButton;      //按钮控件
    FlatTitlebar1: TFlatTitlebar; //Titlebar 控件
    FlatButton1: TFlatButton;  //按钮控件
    FlatButton2: TFlatButton;    //按钮控件
    //窗口事件
    procedure FormClose(Sender: TObject; var Action: TCloseAction);     //窗口关闭事件
    procedure FormShow(Sender: TObject);  //窗口显示事件
    procedure Button1Click(Sender: TObject); //Button1的Click事件
    procedure Button2Click(Sender: TObject); //Button2的Click事件
    procedure FlatButton1Click(Sender: TObject); //FlatButton1的Click事件
    procedure FlatButton2Click(Sender: TObject);  //FlatButton2的Click事件
  private
    { Private declarations }

    mtv:widestring;     //视频文件名

  public
    { Public declarations }
     fillcolor:TColor;   //需要填充的颜色
    catchcolor:TColor;   //需要抠出的颜色
    detal:Byte;          //颜色误差范围
    procedure CreatDirectshow;    //创建Directshow环境资源
    Procedure FreeDirectshow;     //释放Directshow环境资源
    procedure catchpicture;       //捕捉视频图象,并做处理
  end;

var
  Form1: TForm1;  //定义窗体对象

implementation

{$R *.dfm}
uses
Directshow9,comobj,DSutil, Unit2; //包含附加的单元。Directshow9为微软DirectX SDK中Directshow的DELPHI单元文件
                                  //comobj为DELPHI的COM的单元
                                  //DSUtil为DIRECTSHOW的DELPHI工具单元
                                  //Unit2为 设置窗口 的单元

  //==================变量定义部分===============//
var
FilterGraph:IGraphBuilder=nil;  //Directshow的IGraphBuilder接口,用于Directshow的框架
MediaControl:IMediaControl=nil; //Directshow的IMediaControl接口,用于控制媒体播放,暂停,停止
 VideoRender:IBaseFilter=nil;  //Directshow的IBaseFilter接口,IBaseFiler是Directshow中所有Filter的基类.此处是保存的VMR Video Filter
 SampleGrabber:ISampleGrabber=nil;//Directshow的ISampleGrabber接口,此接口存在于SampleGrabber Filter上,用这接口可从 SampleGrabber Filter中获得视频图象颜色信息
 id:integer=0;//调试所用,用于在DIRECTSHOW SDK自带的 graphedt工具中申明个ID ,用于调试。

//===================函数实现部分===============//

{DELPHI函数格式简介:

  1.procedure 过程名称 (参数) ;   //如果没有参数则可以省略"()"
    //局部变量申明

    var //变量申明系统关键字
    MediaRect:TRect;//其中MediaRect为变量名称,TRect为变量类型

    begin
     ...        //函数体部分
     ...
    end;    // begin.....end 结构类似于C++中的 {.......}

{  2.function 函数名称(参数):返回值类型; //如果没有参数则可以省略"()"
      //局部变量申明
     
    var //变量申明系统关键字
    MediaRect:TRect;//其中MediaRect为变量名称,TRect为变量类型

     begin
     ...        //函数体部分
     ...
     result:=...;//result为返回值变量,不需要自己声明,它代表函数的返回值变量,":="为DELPHI的赋值符号  
    end;    // begin.....end 结构类似于C++中的 {.......}
//}


procedure TForm1.CreatDirectshow;
//函数局部变量申明部分
var
MediaRect:TRect;     //区域局部变量,用于设置视频在窗口显示的区域
pType:_AMMEDIATYPE; //视频媒体类型变量,用于设置SampleGrabber Filter的接收视频的类型(如告诉系统:接受的为视频信息,视频内部图象格式为RGB32位)
VideoWindow:IVideoWindow;     //Directshow的IVideoWindow接口,用于对视频的显示窗口做设置(例如:设置其父窗体)
begin
  FilterGraph:=CreateComobject(CLSID_FilterGraph) as IGraphBuilder;  //创建Directshow中的FilterGraph COM对象并返回IGraphBuilder接口
  VideoRender:=createcomobject(CLSID_VideoMixingRenderer) as IBaseFilter; //创建Directshow中的VideoMixingRenderer Filter COM对象并返回IBaseFilter接口
  SampleGrabber:=createcomobject(CLSID_SampleGrabber) AS ISampleGrabber;  //创建Directshow中的SampleGrabber Filter COM对象并返回ISampleGrabber接口


  Filtergraph.AddFilter(VideoRender,‘VideoRender‘); //在Filtregraph框架中添加刚创建的VideoMixingRenderer Filter
  FilterGraph.AddFilter(SampleGrabber as IBaseFilter,‘SampleGrabber‘); //在Filtregraph框架中添加刚创建的SampleGrabber Filter


  fillchar(ptype,sizeof(_AMMEDIATYPE),0);//对ptype变量初始化
   ptype.majortype:=MEDIATYPE_Video;//设置主类型为视频格式
  ptype.subtype:=MEDIASUBTYPE_RGB32; //设置辅助类型为RGB32(就是接受视频流为RGB32的,FilterGraph会智能加入相应的转换Filter)
   ptype.formattype:=FORMAT_VideoInfo;  //设置媒体信息格式为FORMAT_VideoInfo格式

    SampleGrabber.SetMediaType(pType);//将SampleGrabber Filter的类型设置为pType变量所指定的类型
    SampleGrabber.SetBufferSamples(true);//将SampleGrabber Filter设置为缓冲模式




 FilterGraph.RenderFile(pwidechar(mtv),nil); //完成播放指定文件的所有Filter连接

 VideoRender.QueryInterface(IID_IVideoWindow, VideoWindow);//从VideoRender中获得IVideoWindow接口
  Videowindow.put_Owner(OAHWND(FlatPanel1.Handle));   //设置视频窗口的父窗体为FlatPanel控件
   Videowindow.put_windowstyle(WS_CHILD or WS_Clipsiblings);   //设置视频窗体为子窗体,剪裁类型

 Videowindow.put_Left(0);//设置视频窗体的LEFT
 Videowindow.put_Top(0);//设置视频窗体的TOP
 Videowindow.put_Width(FlatPanel1.Width); //设置视频窗体的Width
 Videowindow.put_Height(FlatPanel1.Height);//设置视频窗体的Height

 Videowindow.put_Visible(true); //让视频窗体可见

 FilterGraph.QueryInterface(IID_IMediaControl,MediaControl);//从FilterGraph中获得IMediaControl接口

 dsutil.AddGraphToRot(filterGraph,id); //注册GraphEdit所用的ID,可以方便在GraphEdit控件中调试

end;

procedure TForm1.FreeDirectshow;
begin
dsutil.RemoveGraphFromRot(id);   //清除在GraphEdit中注册的ID

   if VideoRender<>nil then VideoRender:=nil;   //释放VideoRender
   if FilterGraph<>nil then FilterGraph:=nil;   //释放FilterGraph
   IF SampleGrabber<>NIL THEN SampleGrabber:=NIL;//释放SampleGrabber
end;

procedure TForm1.FormClose(Sender: TObject; var Action: TCloseAction);
begin
  freedirectshow; //释放Directshow环境
end;

procedure TForm1.FormShow(Sender: TObject);
begin
open.DefaultExt:=extractfiledir(application.ExeName); //设置打开对话框的打开默认路径,为应用程序的当前路径
 if open.Execute=false then  close; //如果没有选择打开的文件,则关闭程序
 if trim(open.FileName)=‘‘ then close; //如果打开文件的文件名为空,则关闭程序
 if not fileexists(open.FileName) then close; //如果打开不存在的文件,则关闭程序
  mtv:=open.FileName; //将打开窗口中用户选择的文件的名称赋值给MTV变量
CreatDirectshow;  //创建Directshow环境

end;

//============抓图处理关键代码部分==============//
procedure TForm1.catchpicture;
var
ptype:_AMMEDIATYPE; //视频媒体类型变量,详细介绍见,CreateDirectshow函数部分
size:integer;
buf:pchar;         //用于存放获得的视频数据的缓冲区
bmphead:tagBitmapfileheader;//BMP图片的文件头结构体变量
BMPINFO:PBitmapinfoheader;//BMP图片的文件头结构体指针变量
bmpcore:tagBITMAPCoreheader; //BMP图片的Core头结构体变量
filestream:TFilestream;  //DELPHI的文件流类型变量

step:integer;
i:integer;
catch,current,fill:integer;
absvalue:integer;

test1,test2,test3,test4,test5,test6:byte;
begin
try
  MediaControl.Pause;  //播放暂停
  SampleGrabber.GetConnectedMediaType(ptype);   //获得SampleGrabber Filter上的输入PIN上的连接类型
  size:=0;
  buf:=nil;
  SampleGrabber.GetCurrentBuffer(size,nil);//获得SampleGrabber Filter上的缓冲区大小




  try
       buf:=allocmem(size);  //分配buf空间
       SampleGrabber.GetCurrentBuffer(size,buf);//获得视频的当前数据,保存到buf中

    if  (ptype.cbFormat=sizeof(VIDEOINFOHEADER))  and (ptype.pbFormat<>nil) and (IsEqualGUID(ptype.formattype,FORMAT_VideoInfo)) then //判断Sampler Grabber中的输入Pin上的类型是否满足我们的要求
        begin

            //=设置bmp Head部分=//
            with bmphead do
            begin
              bftype:=$4d42;
              bfsize:=sizeof(tagBitmapfileheader);
              bfreserved1:=0;
              bfreserved2:=0;
              bfOffBits:=bfsize+sizeof(tagBITMAPCoreheader);
            end;
             bmpinfo:=PBitmapinfoheader(@(PVideoinfoheader(ptype.pbFormat).bmiHeader));
             step:=bmpinfo.biBitCount div 8;
             bmpcore.bcSize:=sizeof(tagBITMAPCoreheader);
             bmpcore.bcWidth:=bmpinfo.biWidth;
             bmpcore.bcHeight:=bmpinfo.biHeight;
             bmpcore.bcPlanes:=1;
             bmpcore.bcBitCount:=bmpinfo.biBitCount;


           if catchcolor<>fillcolor then
             begin
              catch:=COLORTORGB(Catchcolor);
              fill:=COLORTORGB(fillcolor);
              fill:=RGB(GETBValue(fillcolor),GETGValue(fillcolor),GetRValue(fillcolor));

              current:=0;
              for i:=0 to bmpinfo.biWidth*bmpinfo.biHeight-1 do//循环查找视频图象颜色中的数据
                begin

                 Current:=PInteger(Pchar(buf)+i*step)^;
                 

                  if (abs(GetBValue(Current)-GetRValue(catch))<=detal) and (abs(GetGValue(Current)-GetGValue(catch))<=detal) and (abs(GetRValue(Current)-GetBValue(catch))<=detal) then //判断当前像素中的颜色是否符合抠像填充条件
                    begin
                        PInteger(pchar(buf)+i*step)^:=fill;//填充用户设置的填充颜色
                    end;
                end;
             end;

            try

                 filestream:=TFilestream.Create(Extractfiledir(Application.ExeName)+‘/抓下来的视频图.bmp‘,fmCreate  or fmOpenWrite);  //创建FileStream视频流
                 filestream.WriteBuffer(bmphead,bmphead.bfSize);//将bmphead变量中的数据写入流中

                filestream.WriteBuffer(bmpcore,sizeof(tagBITMAPCoreheader)); //将bmpcore变量中的数据写入流中
                 filestream.WriteBuffer(buf^,size); //将buf指针所指向的数据写入流中

                 windows.MessageBox(handle,‘抓图成功!‘,‘提示‘,MB_ICONINFORMATION);//提示用户,抓图成功
            finally
                filestream.Free;//释放Filestream视频流
            end;    

        end;
  finally
    freemem(buf);   //释放Buf所占用的空间
  end;


finally


MediaControl.Run;//重新播放
end;
end;

procedure TForm1.Button1Click(Sender: TObject);
begin

  MediaControl.Run;//播放视频
  button1.Enabled:=false;  //将Button1按钮设为不可用
  button2.Enabled:=true;   //将Button2按钮设置为可用
end;

procedure TForm1.Button2Click(Sender: TObject);
begin
catchpicture();    //调用自己编写的抓图处理函数
end;

procedure TForm1.FlatButton1Click(Sender: TObject);
begin
close;  //关闭窗体,也将关闭程序
end;

procedure TForm1.FlatButton2Click(Sender: TObject);
begin
 form2.ShowModal;//显示FORM2窗体(模态显示,也就是你必须关闭了FORM2,才能操作程序的其他窗体)
end;

end.

    本站是提供个人知识管理的网络存储空间,所有内容均由用户发布,不代表本站观点。请注意甄别内容中的联系方式、诱导购买等信息,谨防诈骗。如发现有害或侵权内容,请点击一键举报。
    转藏 分享 献花(0

    0条评论

    发表

    请遵守用户 评论公约

    类似文章 更多