开通博客赚积分
发布资源赚积分
分类

源码开发语言/平台
当前位置：首页 > 源码/资料 > 图形图象 > 查看源码
MainFrm.cpp
资源名称：Video_detection.rar　[点击查看]
上传用户：smdfuse
上传日期：2015-11-06
资源大小：98k
文件大小：15k
源码类别：

图形图象

开发平台：

Visual C++

MainFrm.cpp：源码内容
  1. // MainFrm.cpp : implementation of the CMainFrame class
  2. //
  4. #include "stdafx.h"
  5. #include "Video Demo.h"
  6. #include "Video demodoc.h"
  7. #include "MainFrm.h"
  8. #include "AVIHandler.h"   // AVI文件处理
  9. #include "colortrans.h"
  10. #include "StaticDetect.h"
  11. #include "ChafenMul.h"
  12. #include "POSDiag.h"
  13. #include "MotionDetectDiag.h"
  14. #include "GravityTrack.h"
  15. #include "GravityCenter.h"
  16. #ifdef _DEBUG
  17. #define new DEBUG_NEW
  18. #undef THIS_FILE
  19. static char THIS_FILE[] = __FILE__;
  20. #endif
  22. /////////////////////////////////////////////////////////////////////////////
  23. // CMainFrame
  25. IMPLEMENT_DYNCREATE(CMainFrame, CFrameWnd)
  27. BEGIN_MESSAGE_MAP(CMainFrame, CFrameWnd)
  28. //{{AFX_MSG_MAP(CMainFrame)
  29. ON_WM_CREATE()
  30. ON_COMMAND(ID_FILE_OPENVIDEO, OnFileOpenvideo)
  31. ON_COMMAND(ID_MOTION_SD, OnMotionSd)
  32. ON_COMMAND(ID_MOTION_DD, OnMotionDd)
  33. ON_COMMAND(ID_TRACEK_MEANSHIFT, OnTracekMeanshift)
  34. ON_COMMAND(ID_TRACEK_GAV, OnTracekGav)
  35. //}}AFX_MSG_MAP
  36. END_MESSAGE_MAP()
  38. static UINT indicators[] =
  39. {
  40. ID_SEPARATOR,           // status line indicator
  41. ID_INDICATOR_CAPS,
  42. ID_INDICATOR_NUM,
  43. ID_INDICATOR_SCRL,
  44. };
  46. /////////////////////////////////////////////////////////////////////////////
  47. // CMainFrame construction/destruction
  49. CMainFrame::CMainFrame()
  50. {
  51. // TODO: add member initialization code here
  53. }
  55. CMainFrame::~CMainFrame()
  56. {
  57. }
  59. int CMainFrame::OnCreate(LPCREATESTRUCT lpCreateStruct)
  60. {
  61. if (CFrameWnd::OnCreate(lpCreateStruct) == -1)
  62. return -1;
  64. //if (!m_wndToolBar.CreateEx(this, TBSTYLE_FLAT, WS_CHILD | WS_VISIBLE | CBRS_TOP
  65. // | CBRS_GRIPPER | CBRS_TOOLTIPS | CBRS_FLYBY | CBRS_SIZE_DYNAMIC) ||
  66. // !m_wndToolBar.LoadToolBar(IDR_MAINFRAME))
  67. // {
  68. // TRACE0("Failed to create toolbarn");
  69. // return -1;      // fail to create
  70. // }
  72. if (!m_wndStatusBar.Create(this) ||
  73. !m_wndStatusBar.SetIndicators(indicators,
  74.   sizeof(indicators)/sizeof(UINT)))
  75. {
  76. TRACE0("Failed to create status barn");
  77. return -1;      // fail to create
  78. }
  80. // TODO: Delete these three lines if you don't want the toolbar to
  81. //  be dockable
  82. //m_wndToolBar.EnableDocking(CBRS_ALIGN_ANY);
  83. // EnableDocking(CBRS_ALIGN_ANY);
  84. // DockControlBar(&m_wndToolBar);
  86. return 0;
  87. }
  89. BOOL CMainFrame::PreCreateWindow(CREATESTRUCT& cs)
  90. {
  91. if( !CFrameWnd::PreCreateWindow(cs) )
  92. return FALSE;
  93. // TODO: Modify the Window class or styles here by modifying
  94. //  the CREATESTRUCT cs
  96. return TRUE;
  97. }
  99. /************************************************************************/
  100. /* OnOpenOpenVideo                                                      */
  101. /* 打开AVI视频文件                                                      */
  102. /* 基于中央数据级控架构 */
  103. /************************************************************************/
  104. void CMainFrame::OnOpenOpenVideo()
  105. {
  106. AVIHandler *avi = NULL;
  108. //添加一路视频
  109. //考虑的问题是，如何将参数传递，到框架
  110. CString strFilter("Data Files (*.avi)|*.avi");
  111. CFileDialog diag(TRUE,NULL,NULL,OFN_HIDEREADONLY | OFN_OVERWRITEPROMPT,strFilter);
  112. //打开第一个文件
  113. int nID = diag.DoModal();
  114. if(nID == IDOK){
  116. // 新建一个子窗口
  117. POSITION pos=theApp.m_pDocManager->GetFirstDocTemplatePosition();
  118. // 取得一个文档模板，我取的是第一个，你可以去任何一个模板
  119. CDocTemplate* pDocTempl=theApp.m_pDocManager->GetNextDocTemplate(pos);
  120. // 通过打开文件的方式获得一个新窗口，并得到CDocument对象
  121. CVideoDemoDoc * doc = (CVideoDemoDoc *)pDocTempl->OpenDocumentFile(NULL);//theApp.m_pDocManager->OpenDocumentFile(NULL);
  123. // 建立视频数据源
  124. avi = new AVIHandler(diag.GetFileName());    // 打开AVI文件,AVI RGB
  125. theApp.curVideo = new VideoData();           // 当前视频数据源
  126. theApp.curVideo->doc = doc;
  127. theApp.curVideo->index = 0;
  128. theApp.curVideo->ms = (MediaSource*)avi;
  130. // 设置View的显示缓冲
  131. ImageData *id = new ImageData();   
  132. id->bytes = 3;
  133. id->coreDoc = doc;
  134. id->format = IMAGE_FORMAT_RGB24;
  135. id->imgWidth = avi->GetFrameWidth();
  136. id->imgHeight = avi->GetFrameHeight();
  137. id->data = avi->GetSingleFrame(0);                    // 获得第一帧图像
  138. doc->SetData(id);                                     // 设置第一帧图像
  140. // 设置文档和视图的视频源数据
  141. doc->SetVideoSource(theApp.curVideo);                 // 设置Document数据
  142. doc->RefreshView();               // 刷新显示View
  145. }
  146. }
  148. /////////////////////////////////////////////////////////////////////////////
  149. // CMainFrame diagnostics
  151. #ifdef _DEBUG
  152. void CMainFrame::AssertValid() const
  153. {
  154. CFrameWnd::AssertValid();
  155. }
  157. void CMainFrame::Dump(CDumpContext& dc) const
  158. {
  159. CFrameWnd::Dump(dc);
  160. }
  162. #endif //_DEBUG
  164. /////////////////////////////////////////////////////////////////////////////
  165. // CMainFrame message handlers
  167. /************************************************************************/
  168. /* 打开视频文件AVI                                                      */
  169. /************************************************************************/
  170. void CMainFrame::OnFileOpenvideo() 
  171. {
  172. AVIHandler *avi = NULL;
  174. //添加一路视频
  175. //考虑的问题是，如何将参数传递，到框架
  176. CString strFilter("Data Files (*.avi)|*.avi");
  177. CFileDialog diag(TRUE,NULL,NULL,OFN_HIDEREADONLY | OFN_OVERWRITEPROMPT,strFilter);
  178. //打开第一个文件
  179. int nID = diag.DoModal();
  180. if(nID == IDOK){
  182. // 新建一个子窗口
  183. POSITION pos=theApp.m_pDocManager->GetFirstDocTemplatePosition();
  184. // 取得一个文档模板，我取的是第一个，你可以去任何一个模板
  185. CDocTemplate* pDocTempl=theApp.m_pDocManager->GetNextDocTemplate(pos);
  186. // 通过打开文件的方式获得一个新窗口，并得到CDocument对象
  187. CVideoDemoDoc * doc = (CVideoDemoDoc *)pDocTempl->OpenDocumentFile(NULL);//theApp.m_pDocManager->OpenDocumentFile(NULL);
  189. // 建立视频数据源
  190. avi = new AVIHandler(diag.GetFileName());    // 打开AVI文件,AVI RGB
  192. if(!avi->isLoaded())
  193. {
  194. AfxMessageBox("打开文件错误!");
  195. return;
  196. }
  198. if(avi->GetLPBM()->biBitCount != 24)
  199. {
  200. delete avi;
  201. AfxMessageBox("AVI文件必须是24位非压缩彩色视频!");
  202. return;
  203. }
  205. theApp.curVideo = new VideoData();           // 当前视频数据源
  206. theApp.curVideo->doc = doc;
  207. theApp.curVideo->index = 0;
  208. theApp.curVideo->ms = (MediaSource*)avi;
  210. // 设置View的显示缓冲
  211. ImageData *id = new ImageData();   
  212. id->bytes = 3;
  213. id->coreDoc = doc;
  214. id->format = IMAGE_FORMAT_RGB24;
  215. id->imgWidth = avi->GetFrameWidth();
  216. id->imgHeight = avi->GetFrameHeight();
  217. id->data = avi->GetSingleFrame(0);                    // 获得第一帧图像
  218. doc->SetData(id);                                     // 设置第一帧图像
  220. // 设置文档和视图的视频源数据
  221. doc->SetVideoSource(theApp.curVideo);                 // 设置Document数据
  222. doc->RefreshView();               // 刷新显示View
  225. }
  226. }
  228. /************************************************************************/
  229. /* 静止背景目标检测                                                     */
  230. /************************************************************************/
  231. DWORD WINAPI StaticDetectT(LPVOID data){
  233. CVideoDemoDoc *doc = (CVideoDemoDoc*)data;
  235. int frameLength = theApp.curVideo->ms->GetFrameCount();
  236. MediaSource *avi = theApp.curVideo->ms;
  238. unsigned char *frame = avi->GetSingleFrame(0);
  239. int frameWidth = avi->GetFrameWidth();
  240. int frameHeight = avi->GetFrameHeight();
  242. // 色彩变换
  243. ColorTrans ct;
  245. CStatsticDetect sd(frameWidth,frameHeight);
  248. // 输出缓冲
  249. unsigned char * outBuffer = new unsigned char[frameHeight * frameWidth * 3];
  251. for(int i = 0; i < frameLength - 1; i++)
  252. {
  253. // 获得一帧图像
  254. frame = avi->GetSingleFrame(i);
  255. // 转换到灰度图像
  256. //ct.RGB2Gray8(frame,grayBuffer,frameWidth,frameHeight);
  258. sd.ReceiveFrame(i,frame,D_STATISTIC);
  259. // 添加一帧数据
  261. ct.gray2RGB2(sd.m_pDiffImage,outBuffer,frameWidth,frameHeight);
  262. doc->id->data = outBuffer;
  264. doc->RefreshView();
  265. }
  268. delete[] outBuffer;
  269. return 0;
  271. }
  273. /************************************************************************/
  274. /* 静止背景目标检测                                                     */
  275. /************************************************************************/
  276. void CMainFrame::OnMotionSd()
  277. {
  278. DWORD exitCode;
  280. if(GetExitCodeThread(th,&exitCode)){
  281.    TerminateThread(th,exitCode);
  282. }
  284. DWORD pID;
  285. //控制线程播放
  287. int frameLength = theApp.curVideo->ms->GetFrameCount();
  289. MediaSource *avi = theApp.curVideo->ms;
  291. if(!theApp.curVideo)
  292. {
  293. AfxMessageBox("没有打开视频文件 ！");
  294. return;
  295. }
  297. // 创建线程
  298. th = CreateThread(NULL,0,StaticDetectT,theApp.curVideo->doc,0,&pID);
  299. }
  301. /************************************************************************/
  302. /* 动止背景目标检测                                                     */
  303. /************************************************************************/
  304. DWORD WINAPI MotionDetectT(LPVOID data){
  306. CVideoDemoDoc *doc = (CVideoDemoDoc*)theApp.curVideo->doc;
  310. int frameLength = theApp.curVideo->ms->GetFrameCount();
  311. MediaSource *avi = theApp.curVideo->ms;
  313. unsigned char *frame = avi->GetSingleFrame(0);
  314. int frameWidth = avi->GetFrameWidth();
  315. int frameHeight = avi->GetFrameHeight();
  317. // 色彩变换
  318. ColorTrans ct;
  320. int *threshold = (int*)data;
  322. MotionDetectDiag diag;
  323. //if(diag.DoModal()){
  324. // threshold = diag.m_index;
  325. //}
  326. CStatsticDetect sd(frameWidth,frameHeight);
  327. //CStaticDetect sd(frameWidth,frameHeight);
  328. ChafenMul cm(frameWidth,frameHeight,*threshold);
  330. // 输出缓冲
  331. unsigned char * outBuffer = new unsigned char[frameHeight * frameWidth * 3];
  332. unsigned char * gray = new unsigned char[frameWidth * frameHeight * 3];
  333. for(int i = 0; i < frameLength - 1; i++)
  334. {
  335. // 获得一帧图像
  336. frame = avi->GetSingleFrame(i);
  338. ct.RGB2Gray8(frame,gray,frameWidth,frameHeight);
  340. // 准备数据
  341. cm.PrepareData(gray,frameWidth,frameHeight);
  344. if(i % 4 == 0 && i != 0)
  345. {
  346. cm.process();
  349. sd.DeNoise(cm.frame1,frameWidth,frameHeight);
  350. ct.gray2RGB2(cm.frame1,outBuffer,frameWidth,frameHeight);
  351. doc->id->data = outBuffer;
  353. doc->RefreshView();
  355. }
  356. }
  358. delete[] outBuffer;
  359. return 0;
  361. }
  363. /************************************************************************/
  364. /* 动背景目标检测                                                       */
  365. /************************************************************************/
  366. void CMainFrame::OnMotionDd() 
  367. {
  368. DWORD exitCode;
  370. if(GetExitCodeThread(th,&exitCode)){
  371.    TerminateThread(th,exitCode);
  372. }
  374. if(theApp.curVideo == NULL)
  375. AfxMessageBox("没有打开视频文件！");
  376. DWORD pID;
  377. //控制线程播放
  379. int frameLength = theApp.curVideo->ms->GetFrameCount();
  381. MediaSource *avi = theApp.curVideo->ms;
  384. int threshold = 100;
  385. MotionDetectDiag mdd;
  386. if(mdd.DoModal())
  387. {
  388. threshold = mdd.m_index;
  389. }
  391. int * thre = (int*)malloc(sizeof(int));
  392. *thre = threshold;
  393. // 创建线程
  394. th = CreateThread(NULL,0,MotionDetectT,thre,0,&pID);
  396. }
  399. /************************************************************************/
  400. /* 动止背景目标检测                                                     */
  401. /************************************************************************/
  402. DWORD WINAPI MSS(LPVOID data){
  404. CVideoDemoDoc *doc = (CVideoDemoDoc*)theApp.curVideo->doc;//(CVideoDemoDoc*)data;
  406. int frameLength = theApp.curVideo->ms->GetFrameCount();
  407. MediaSource *avi = theApp.curVideo->ms;
  409. unsigned char *frame = avi->GetSingleFrame(0);
  410. int frameWidth = avi->GetFrameWidth();
  411. int frameHeight = avi->GetFrameHeight();
  413. MeanShiftSegger* mms = (MeanShiftSegger*)data;
  415. // 输出缓冲
  416. unsigned char * outBuffer = new unsigned char[frameHeight * frameWidth * 3];
  418. for(int i = 0; i < frameLength - 1; i++)
  419. {
  420. // 获得一帧图像
  421. frame = avi->GetSingleFrame(i);
  423. mms->MeanShiftTrackProcess(frame,i);
  425. doc->id->data = frame;
  427. doc->RefreshView();
  428. }
  430. delete[] outBuffer;
  431. return 0;
  433. }
  436. /************************************************************************/
  437. /* MeanShift跟踪                                                        */
  438. /************************************************************************/
  439. void CMainFrame::OnTracekMeanshift() 
  440. {
  441. DWORD exitCode;
  443. if(GetExitCodeThread(th,&exitCode)){
  444.    TerminateThread(th,exitCode);
  445. }
  446. DWORD pID;
  447. if(theApp.curVideo == NULL){
  448. AfxMessageBox("没有打开视频！");
  449. return;
  450. }
  452. int x,y,width,height;
  453. POSDiag pd;
  454. if(pd.DoModal())
  455. {
  456. x= pd.m_x;
  457. y= pd.m_y;
  458. width = pd.m_width;
  459. height = pd.m_height;
  460. }
  461. else
  462. return;
  464. mss = new MeanShiftSegger();
  465. int frameLength = theApp.curVideo->ms->GetFrameCount();
  466. MediaSource *avi = theApp.curVideo->ms;
  468. unsigned char *frame = avi->GetSingleFrame(0);
  469. int frameWidth = avi->GetFrameWidth();
  470. int frameHeight = avi->GetFrameHeight();
  472. mss->InitMeanShiftTracker(frame,frameWidth,frameHeight,x,y,width,height);
  475. th = CreateThread(NULL,0,MSS,mss,0,&pID);
  479. }
  482. /************************************************************************/
  483. /* 重心 法 跟踪                                                         */
  484. /************************************************************************/
  485. DWORD WINAPI GAV(LPVOID data){
  487. int* threshold = (int*)data;
  489. CVideoDemoDoc *doc = (CVideoDemoDoc*)theApp.curVideo->doc;
  491. int frameLength = theApp.curVideo->ms->GetFrameCount();
  492. MediaSource *avi = theApp.curVideo->ms;
  494. unsigned char *frame = avi->GetSingleFrame(0);
  495. int frameWidth = avi->GetFrameWidth();
  496. int frameHeight = avi->GetFrameHeight();
  498. // 输出缓冲
  499. unsigned char * outBuffer = new unsigned char[frameHeight * frameWidth * 3];
  501. unsigned char * grayBuffer =new unsigned char[frameWidth * frameHeight];
  503. ColorTrans ct;
  504. CGravityCenter* m_gravitycenter;
  505. TARGETPARA m_target;
  506. for(int i = 0; i < frameLength - 1; i++)
  507. {
  508. // 获得一帧图像
  509. frame = avi->GetSingleFrame(i);
  511. ct.RGB2Gray8(frame,grayBuffer,frameWidth,frameHeight);
  513. if(i == 0)
  514. {
  515. m_gravitycenter = new CGravityCenter(frameWidth,frameHeight);
  516. m_gravitycenter->InitalObjectTracker(T_GRACENTER,*threshold);
  517. m_gravitycenter->GravityCenter(i, frame);
  519. }
  520. else 
  521. {
  522. m_gravitycenter->GravityCenter(i, frame);
  523. }
  525. m_target = m_gravitycenter->m_result;
  526. /*
  527. // 重心跟踪
  528. WININFO info = GravityTrack(grayBuffer,frameWidth,frameHeight,*threshold);
  529. if(!info.flag)
  530. {
  531. for(int j = max(info.pt.y - info.h / 2,0);j <= min(info.pt.y + info.h / 2,frameHeight); j ++)
  532. for(int i = max(info.pt.x - info.w / 2,0);i <= min(info.pt.x + info.w / 2,frameWidth); i ++)
  533. {
  534. frame[j * frameWidth * 3 + i * 3] = 255;
  535. }
  537. }*/
  539. doc->id->data = frame;
  541. doc->RefreshView();
  542. }
  544. delete[] grayBuffer;
  546. delete[] outBuffer;
  547. return 0;
  549. }
  550. /************************************************************************/
  551. /* 重心法跟踪                                                           */
  552. /************************************************************************/
  553. void CMainFrame::OnTracekGav() 
  554. {
  555. DWORD exitCode;
  557. if(GetExitCodeThread(th,&exitCode)){
  558.    TerminateThread(th,exitCode);
  559. }
  560. DWORD pID;
  561. if(theApp.curVideo == NULL){
  562. AfxMessageBox("没有打开视频！");
  563. return;
  564. }
  567. int threshold = 100;
  568. MotionDetectDiag mdd;
  569. if(mdd.DoModal())
  570. {
  571. threshold = mdd.m_index;
  572. }
  574. int * thre = (int*)malloc(sizeof(int));
  575. *thre = threshold;
  577. th = CreateThread(NULL,0,GAV,thre,0,&pID);
  578. }