VideoCaptureView.cpp 34 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221
  1. // 这段 MFC 示例源代码演示如何使用 MFC Microsoft Office Fluent 用户界面
  2. // (“Fluent UI”)。该示例仅供参考,
  3. // 用以补充《Microsoft 基础类参考》和
  4. // MFC C++ 库软件随附的相关电子文档。
  5. // 复制、使用或分发 Fluent UI 的许可条款是单独提供的。
  6. // 若要了解有关 Fluent UI 许可计划的详细信息,请访问
  7. // https://go.microsoft.com/fwlink/?LinkId=238214.
  8. //
  9. // 版权所有(C) Microsoft Corporation
  10. // 保留所有权利。
  11. // VideoCaptureView.cpp: CVideoCaptureView 类的实现
  12. //
  13. #include "stdafx.h"
  14. #include "framework.h"
  15. // SHARED_HANDLERS 可以在实现预览、缩略图和搜索筛选器句柄的
  16. // ATL 项目中进行定义,并允许与该项目共享文档代码。
  17. #ifndef SHARED_HANDLERS
  18. #include "VideoCapture.h"
  19. #endif
  20. #include "VideoCaptureDoc.h"
  21. #include "VideoCaptureView.h"
  22. #include "MainFrm.h"
  23. #ifdef _DEBUG
  24. #define new DEBUG_NEW
  25. #endif
  26. CMemoryClient g_ub530;
  27. static CMainFrame* g_pMainFrame = NULL;
  28. VOID Dbg(CHAR* pszStr, ...)
  29. {
  30. char szData[MAX_PATH] = "[SC] ";
  31. va_list args;
  32. va_start(args, pszStr);
  33. _vsnprintf_s(szData + 5, MAX_PATH - 5, MAX_PATH - 6, pszStr, args);
  34. va_end(args);
  35. strcat_s(szData, "\n");
  36. OutputDebugStringA(szData);
  37. }
  38. // FORMAT CHANGED CALLBACK FUNCTION
  39. //
  40. QRETURN on_process_format_changed(PVOID pDevice, ULONG nVideoInput, ULONG nAudioInput, ULONG nVideoWidth, ULONG nVideoHeight, BOOL bVideoIsInterleaved, double dVideoFrameRate, ULONG nAudioChannels, ULONG nAudioBitsPerSample, ULONG nAudioSampleFrequency, PVOID pUserData)
  41. {
  42. //Dbg("format changed Detected \n");
  43. CVideoCaptureView* pView = (CVideoCaptureView*)pUserData;
  44. if (pView == NULL) { return QCAP_RT_OK; }
  45. if (pView->GetSafeHwnd() == NULL) { return QCAP_RT_OK; }
  46. pView->m_nVideoWidth = nVideoWidth;
  47. pView->m_nVideoHeight = nVideoHeight;
  48. pView->m_bVideoIsInterleaved = bVideoIsInterleaved;
  49. pView->m_dVideoFrameRate = dVideoFrameRate;
  50. pView->m_nAudioChannels = nAudioChannels;
  51. pView->m_nAudioBitsPerSample = nAudioBitsPerSample;
  52. pView->m_nAudioSampleFrequency = nAudioSampleFrequency;
  53. // OUTPUT FORMAT CHANGED MESSAGE
  54. //
  55. CHAR strVideoInput[MAX_PATH] = { 0 };
  56. CHAR strAudioInput[MAX_PATH] = { 0 };
  57. CHAR strFrameType[MAX_PATH] = { 0 };
  58. UINT nVH = 0;
  59. if (nVideoInput == 0) { sprintf_s(strVideoInput, "COMPOSITE"); }
  60. if (nVideoInput == 1) { sprintf_s(strVideoInput, "SVIDEO"); }
  61. if (nVideoInput == 2) { sprintf_s(strVideoInput, "HDMI"); }
  62. if (nVideoInput == 3) { sprintf_s(strVideoInput, "DVI_D"); }
  63. if (nVideoInput == 4) { sprintf_s(strVideoInput, "COMPONENTS (YCBCR)"); }
  64. if (nVideoInput == 5) { sprintf_s(strVideoInput, "DVI_A (RGB / VGA)"); }
  65. if (nVideoInput == 6) { sprintf_s(strVideoInput, "SDI"); }
  66. if (nVideoInput == 7) { sprintf_s(strVideoInput, "AUTO"); }
  67. if (nAudioInput == 0) { sprintf_s(strAudioInput, "EMBEDDED_AUDIO"); }
  68. if (nAudioInput == 1) { sprintf_s(strAudioInput, "LINE_IN"); }
  69. if (bVideoIsInterleaved == TRUE) { nVH = nVideoHeight / 2; }
  70. else { nVH = nVideoHeight; }
  71. if (bVideoIsInterleaved == TRUE) { sprintf_s(strFrameType, " I "); }
  72. else { sprintf_s(strFrameType, " P "); }
  73. pView->m_strFormatChangedOutput.Format(
  74. _T("INFO : %d x %d%s @%2.3f FPS , %d CH x %d BITS x %d HZ , VIDEO INPUT : %s , AUDIO INPUT : %s"),
  75. nVideoWidth,
  76. nVH,
  77. strFrameType,
  78. dVideoFrameRate,
  79. nAudioChannels,
  80. nAudioBitsPerSample,
  81. nAudioSampleFrequency,
  82. strVideoInput,
  83. strAudioInput
  84. );
  85. Global::WriteTextLog(_T("采集卡格式变更:%d x %d%s @%2.3f FPS , %d CH x %d BITS x %d HZ , 视频输入源 : %s , 音频输入源 : %s"),
  86. nVideoWidth,
  87. nVH,
  88. strFrameType,
  89. dVideoFrameRate,
  90. nAudioChannels,
  91. nAudioBitsPerSample,
  92. nAudioSampleFrequency,
  93. strVideoInput,
  94. strAudioInput);
  95. ::SendMessage(g_pMainFrame->m_hWnd, MSG_STATUS_BAR, (WPARAM)& pView->m_strFormatChangedOutput, ID_STATUSBAR_PANE1);
  96. // NO SOURCE
  97. //
  98. if (nVideoWidth == 0 &&
  99. nVideoHeight == 0 &&
  100. dVideoFrameRate == 0.0 &&
  101. nAudioChannels == 0 &&
  102. nAudioBitsPerSample == 0 &&
  103. nAudioSampleFrequency == 0)
  104. {
  105. pView->m_bNoSignal = TRUE;
  106. }
  107. else
  108. {
  109. pView->m_bNoSignal = FALSE;
  110. }
  111. pView->SetTimer(0x00000000, 1, NULL);
  112. return QCAP_RT_OK;
  113. }
  114. // NO SIGNAL DETEACTED CALLBACK FUNCTION
  115. //
  116. QRETURN on_process_no_signal_detected(PVOID pDevice, ULONG nVideoInput, ULONG nAudioInput, PVOID pUserData)
  117. {
  118. AFX_MANAGE_STATE(AfxGetStaticModuleState());
  119. CVideoCaptureView* pView = (CVideoCaptureView*)pUserData;
  120. //Dbg("No Signal Detected \n");
  121. if (pView == NULL) { return QCAP_RT_OK; }
  122. if (pView->GetSafeHwnd() == NULL) { return QCAP_RT_OK; }
  123. //pView->SetRibbonStatusBarText("当前采集卡无信号。", ID_STATUSBAR_PANE1);
  124. Global::WriteTextLog(_T("当前采集卡无信号"));
  125. ::SendMessage(g_pMainFrame->m_hWnd, MSG_STATUS_BAR, (WPARAM)& CString("当前采集卡无信号"), ID_STATUSBAR_PANE1);
  126. pView->m_bNoSignal = TRUE;
  127. pView->SetTimer(0x00000000, 1, NULL);
  128. return QCAP_RT_OK;
  129. }
  130. // SIGNAL REMOVED CALLBACK FUNCTION
  131. //
  132. QRETURN on_process_signal_removed(PVOID pDevice, ULONG nVideoInput, ULONG nAudioInput, PVOID pUserData)
  133. {
  134. AFX_MANAGE_STATE(AfxGetStaticModuleState());
  135. CVideoCaptureView* pView = (CVideoCaptureView*)pUserData;
  136. if (pView == NULL) { return QCAP_RT_OK; }
  137. if (pView->GetSafeHwnd() == NULL) { return QCAP_RT_OK; }
  138. //pView->SetRibbonStatusBarText("当前采集卡信号被移除。", ID_STATUSBAR_PANE1);
  139. Global::WriteTextLog(_T("当前采集卡信号被移除"));
  140. ::SendMessage(g_pMainFrame->m_hWnd, MSG_STATUS_BAR, (WPARAM)& CString("当前采集卡信号被移除"), ID_STATUSBAR_PANE1);
  141. pView->m_bNoSignal = TRUE;
  142. pView->SetTimer(0x00000000, 1, NULL);
  143. return QCAP_RT_OK;
  144. }
  145. // PREVIEW VIDEO CALLBACK FUNCTION
  146. //
  147. QRETURN on_process_preview_video_buffer(PVOID pDevice, double dSampleTime, BYTE* pFrameBuffer, ULONG nFrameBufferLen, PVOID pUserData)
  148. {
  149. CVideoCaptureView* pView = (CVideoCaptureView*)pUserData;
  150. if (pView)
  151. {
  152. // 尝试加锁;
  153. if (!pView->m_mut_cpature.try_lock())
  154. {
  155. #ifdef DEBUG
  156. OutputDebugStringA("!pView->m_mut_cpature.try_lock()\n");
  157. #endif
  158. return QCAP_RT_OK;
  159. }
  160. // 是否触发截图;
  161. if (pView->m_bCaptureImage)
  162. {
  163. // 加锁;
  164. std::lock_guard<std::mutex> lk(pView->m_mut_thread);
  165. // 复制数据;
  166. pView->m_dwBufferLen = nFrameBufferLen;
  167. pView->m_pBuffer = new BYTE[nFrameBufferLen];
  168. memcpy(pView->m_pBuffer, pFrameBuffer, nFrameBufferLen);
  169. // 通知线程保存图片;
  170. pView->m_thread_cond.notify_one();
  171. // 恢复截图标记,防止重复截图;
  172. pView->m_bCaptureImage = FALSE;
  173. #ifdef DEBUG
  174. OutputDebugStringA("preview_video_buffer:pView->m_bCaptureImage\n");
  175. #endif
  176. }
  177. // 解锁;
  178. pView->m_mut_cpature.unlock();
  179. }
  180. return QCAP_RT_OK;
  181. }
  182. // PREVIEW AUDIO CALLBACK FUNCTION
  183. //
  184. QRETURN on_process_preview_audio_buffer(PVOID pDevice, double dSampleTime, BYTE* pFrameBuffer, ULONG nFrameBufferLen, PVOID pUserData)
  185. {
  186. return QCAP_RT_OK;
  187. }
  188. // VIDEO HARDARE ENCODER CALLBACK FUNCTION
  189. //
  190. QRETURN on_process_hardware_encoder_video_buffer(PVOID pDevice, UINT iRecNum, double dSampleTime, BYTE* pFrameBuffer, ULONG nFrameBufferLen, BOOL bIsKeyFrame, PVOID pUserData)
  191. {
  192. return QCAP_RT_OK;
  193. }
  194. QRETURN on_process_snapshot_done(PVOID pDevice, CHAR* pszFilePathName, PVOID pUserData)
  195. {
  196. return QCAP_RT_OK;
  197. }
  198. QRETURN on_process_snapshot_stream(PVOID pDevice, CHAR* pszFilePathName, BYTE* pStreamBuffer, ULONG nStreamBufferLen, PVOID pUserData)
  199. {
  200. // 注意:在调用本回调函数前,已经保存了图片!
  201. CVideoCaptureView* pView = (CVideoCaptureView*)pUserData;
  202. if (pView->m_bHoriontal || pView->m_bVertically)
  203. Global::SaveImgByRotate(pszFilePathName, pStreamBuffer, nStreamBufferLen, pView->m_bHoriontal, pView->m_bVertically);
  204. return QCAP_RT_OK;
  205. }
  206. // CVideoCaptureView
  207. IMPLEMENT_DYNCREATE(CVideoCaptureView, CView)
  208. BEGIN_MESSAGE_MAP(CVideoCaptureView, CView)
  209. // 标准打印命令
  210. ON_COMMAND(ID_FILE_PRINT, &CView::OnFilePrint)
  211. ON_COMMAND(ID_FILE_PRINT_DIRECT, &CView::OnFilePrint)
  212. ON_COMMAND(ID_FILE_PRINT_PREVIEW, &CVideoCaptureView::OnFilePrintPreview)
  213. ON_WM_CONTEXTMENU()
  214. ON_WM_RBUTTONUP()
  215. ON_WM_CREATE()
  216. ON_WM_TIMER()
  217. ON_COMMAND(ID_CUT_BMP, &CVideoCaptureView::OnCutBmp)
  218. ON_COMMAND(ID_CUT_JPG, &CVideoCaptureView::OnCutJpg)
  219. ON_COMMAND(ID_START_RECORD, &CVideoCaptureView::OnStartRecord)
  220. ON_COMMAND(ID_STOP_RECORD, &CVideoCaptureView::OnStopRecord)
  221. ON_COMMAND(ID_CHECK_VERTICALLY, &CVideoCaptureView::OnCheckVertically)
  222. ON_COMMAND(ID_CHECK_HORIONTAL, &CVideoCaptureView::OnCheckHoriontal)
  223. ON_UPDATE_COMMAND_UI(ID_CHECK_VERTICALLY, &CVideoCaptureView::OnUpdateCheckVertically)
  224. ON_UPDATE_COMMAND_UI(ID_CHECK_HORIONTAL, &CVideoCaptureView::OnUpdateCheckHoriontal)
  225. ON_COMMAND(ID_TRAYMENU_RECONNECT, &CVideoCaptureView::OnTraymenuReconnect)
  226. ON_COMMAND(ID_CHECK_SUPORT_GPU, &CVideoCaptureView::OnCheckSuportGpu)
  227. ON_COMMAND(ID_CHECK_FORMAT_MP4, &CVideoCaptureView::OnCheckFormatMp4)
  228. ON_COMMAND(ID_CHECK_FORMAT_AVI, &CVideoCaptureView::OnCheckFormatAvi)
  229. ON_UPDATE_COMMAND_UI(ID_CHECK_SUPORT_GPU, &CVideoCaptureView::OnUpdateCheckSuportGpu)
  230. ON_UPDATE_COMMAND_UI(ID_CHECK_FORMAT_MP4, &CVideoCaptureView::OnUpdateCheckFormatMp4)
  231. ON_UPDATE_COMMAND_UI(ID_CHECK_FORMAT_AVI, &CVideoCaptureView::OnUpdateCheckFormatAvi)
  232. ON_UPDATE_COMMAND_UI(ID_START_RECORD, &CVideoCaptureView::OnUpdateStartRecord)
  233. ON_UPDATE_COMMAND_UI(ID_STOP_RECORD, &CVideoCaptureView::OnUpdateStopRecord)
  234. END_MESSAGE_MAP()
  235. // CVideoCaptureView 构造/析构
  236. CVideoCaptureView::CVideoCaptureView() noexcept
  237. {
  238. // TODO: 在此处添加构造代码
  239. m_hVideoDevice = NULL;
  240. m_bIsRecord = FALSE;
  241. m_bNoSignal = FALSE;
  242. m_bVertically = FALSE;
  243. m_bHoriontal = FALSE;
  244. m_bAppQuit = FALSE;
  245. m_pBuffer = NULL;
  246. m_bCaptureImage = FALSE;
  247. m_dwBufferLen = 0;
  248. m_bIsMp4 = TRUE;
  249. m_bSupportGPU = FALSE;
  250. }
  251. CVideoCaptureView::~CVideoCaptureView()
  252. {
  253. g_ub530.EndOfThread();
  254. m_bAppQuit = TRUE;
  255. HwUninitialize();
  256. }
  257. BOOL CVideoCaptureView::PreCreateWindow(CREATESTRUCT& cs)
  258. {
  259. // TODO: 在此处通过修改
  260. // CREATESTRUCT cs 来修改窗口类或样式
  261. return CView::PreCreateWindow(cs);
  262. }
  263. // CVideoCaptureView 绘图
  264. void CVideoCaptureView::OnDraw(CDC* pDC)
  265. {
  266. CVideoCaptureDoc* pDoc = GetDocument();
  267. ASSERT_VALID(pDoc);
  268. if (!pDoc)
  269. return;
  270. // TO FILL IN THE TEXT
  271. //
  272. if (m_hVideoDevice == 0 || m_bNoSignal) {
  273. CFont font;
  274. ULONG nFontSize = 300;
  275. font.CreatePointFont(nFontSize, TEXT("Arial"), NULL);
  276. HFONT hOldFont = (HFONT)pDC->SelectObject(&font);
  277. pDC->SetBkMode(TRANSPARENT);
  278. pDC->SetTextColor(RGB(255, 255, 255));
  279. CRect rect_client;
  280. GetClientRect(&rect_client);
  281. CString string = _T("");
  282. if (m_hVideoDevice == 0)
  283. {
  284. CBrush brush_fill_rect(RGB(0, 0, 0));
  285. pDC->FillRect(&rect_client, &brush_fill_rect);
  286. string = TEXT("No Device");
  287. //g_pMainFrame->SetRibbonStatusBarText("当前采集卡设备被移除。", ID_STATUSBAR_PANE1);
  288. }
  289. else if(m_bNoSignal)
  290. {
  291. CBrush brush_fill_rect(RGB(0, 0, 255));
  292. pDC->FillRect(&rect_client, &brush_fill_rect);
  293. string = TEXT("No Source");
  294. //g_pMainFrame->SetRibbonStatusBarText("当前采集卡信号被移除。", ID_STATUSBAR_PANE1);
  295. }
  296. pDC->DrawText(string, &rect_client, DT_SINGLELINE | DT_CENTER | DT_VCENTER);
  297. pDC->SelectObject(&hOldFont);
  298. font.DeleteObject();
  299. }
  300. // TODO: 在此处为本机数据添加绘制代码
  301. }
  302. // CVideoCaptureView 打印
  303. BOOL CVideoCaptureView::HwInitialize()
  304. {
  305. if (m_hVideoDevice != NULL)
  306. return TRUE;
  307. // CALLBACK FUNCTION
  308. PF_FORMAT_CHANGED_CALLBACK pFormatChangedCB = { 0 };
  309. PF_VIDEO_PREVIEW_CALLBACK pPreviewVideoCB = { 0 };
  310. PF_AUDIO_PREVIEW_CALLBACK pPreviewAudioCB = { 0 };
  311. PF_VIDEO_HARDWARE_ENCODER_CALLBACK pHardwareEncoderVideoCB = { 0 };
  312. PF_NO_SIGNAL_DETECTED_CALLBACK pNoSignalDetectedCB = { 0 };
  313. PF_SIGNAL_REMOVED_CALLBACK pSignalRemovedCB = { 0 };
  314. PF_SNAPSHOT_DONE_CALLBACK pSnapShotDoneCB = { 0 };
  315. PF_SNAPSHOT_STREAM_CALLBACK pSnapShotStreamCB = { 0 };
  316. // CREATE CAPTURE DEVICE
  317. //
  318. if (m_hVideoDevice == NULL)
  319. QCAP_CREATE("CY3014 USB", 0, m_hWnd, &m_hVideoDevice, 1);
  320. if (m_hVideoDevice == NULL) {
  321. return FALSE;
  322. }
  323. // REGISTER FORMAT CHANGED CALLBACK FUNCTION
  324. pFormatChangedCB = on_process_format_changed;
  325. QCAP_REGISTER_FORMAT_CHANGED_CALLBACK(m_hVideoDevice, pFormatChangedCB, this);
  326. // REGISTER PREVIEW VIDEO CALLBACK FUNCTION
  327. pPreviewVideoCB = on_process_preview_video_buffer;
  328. QCAP_REGISTER_VIDEO_PREVIEW_CALLBACK(m_hVideoDevice, pPreviewVideoCB, this);
  329. // REGISTER PREVIEW AUDIO CALLBACK FUNCTION
  330. //pPreviewAudioCB = on_process_preview_audio_buffer;
  331. //QCAP_REGISTER_AUDIO_PREVIEW_CALLBACK(m_hVideoDevice, pPreviewAudioCB, this);
  332. // REGISTER HARDWARE ENCODER VIDEO CALLBACK FUNCTION
  333. //pHardwareEncoderVideoCB = on_process_hardware_encoder_video_buffer;
  334. //QCAP_REGISTER_VIDEO_HARDWARE_ENCODER_CALLBACK(m_hVideoDevice, 0, pHardwareEncoderVideoCB, this);
  335. // REGISTER NO SIGNAL DETECTED CALLBACK FUNCTION
  336. pNoSignalDetectedCB = on_process_no_signal_detected;
  337. QCAP_REGISTER_NO_SIGNAL_DETECTED_CALLBACK(m_hVideoDevice, pNoSignalDetectedCB, this);
  338. // REGISTER SIGNAL REMOVED CALLBACK FUNCTION
  339. pSignalRemovedCB = on_process_signal_removed;
  340. QCAP_REGISTER_SIGNAL_REMOVED_CALLBACK(m_hVideoDevice, pSignalRemovedCB, this);
  341. #if 1
  342. //
  343. pSnapShotDoneCB = on_process_snapshot_done;
  344. QCAP_REGISTER_SNAPSHOT_DONE_CALLBACK(m_hVideoDevice, pSnapShotDoneCB, this);
  345. //
  346. pSnapShotStreamCB = on_process_snapshot_stream;
  347. QCAP_REGISTER_SNAPSHOT_STREAM_CALLBACK(m_hVideoDevice, pSnapShotStreamCB, this);
  348. #endif
  349. QCAP_SET_VIDEO_DEINTERLACE_TYPE(m_hVideoDevice, QCAP_SOFTWARE_DEINTERLACE_TYPE_BLENDING);
  350. QCAP_SET_VIDEO_DEINTERLACE(m_hVideoDevice, 0);
  351. QCAP_SET_AUDIO_VOLUME(m_hVideoDevice, 100);
  352. QCAP_SET_VIDEO_HARDWARE_ENCODER_PROPERTY(m_hVideoDevice, 0, QCAP_ENCODER_FORMAT_H264, QCAP_RECORD_MODE_CBR, 8000, 12 * 1024 * 1024, 30, 0, 0, QCAP_DOWNSCALE_MODE_OFF, 0, 0);
  353. QCAP_SET_VIDEO_INPUT(m_hVideoDevice, QCAP_INPUT_TYPE_AUTO);
  354. QCAP_RUN(m_hVideoDevice);
  355. // 刷新区域;
  356. Invalidate();
  357. // UPDATE USER INTERFACE RESOURCE
  358. if (m_hVideoDevice == 0) {
  359. /*m_oSetupDialog.m_btnVideoInput.EnableWindow(FALSE);
  360. m_oSetupDialog.m_btnVideoQuality.EnableWindow(FALSE);
  361. m_oSetupDialog.m_btnAudioInput.EnableWindow(FALSE);
  362. m_oSetupDialog.m_btnSnapshot_bmp.EnableWindow(FALSE);
  363. m_oSetupDialog.m_btnSnapshot_jpg.EnableWindow(FALSE);
  364. m_oSetupDialog.m_btnRecordStart_1_1.EnableWindow(FALSE);
  365. m_oSetupDialog.m_btnRecordStop_1_1.EnableWindow(FALSE);
  366. m_oSetupDialog.m_btnRecordStart_1_2.EnableWindow(FALSE);
  367. m_oSetupDialog.m_btnRecordStop_1_2.EnableWindow(FALSE);
  368. m_oSetupDialog.m_btnRecordStart_1_3.EnableWindow(FALSE);
  369. m_oSetupDialog.m_btnRecordStop_1_3.EnableWindow(FALSE);
  370. m_oSetupDialog.m_checkGPU_1_1.EnableWindow(FALSE);
  371. m_oSetupDialog.m_checkGPU_1_1.SetCheck(0);
  372. m_oSetupDialog.m_checkGPU_1_2.EnableWindow(FALSE);
  373. m_oSetupDialog.m_checkGPU_1_2.SetCheck(0);
  374. m_oSetupDialog.m_checkGPU_1_3.EnableWindow(FALSE);
  375. m_oSetupDialog.m_checkGPU_1_3.SetCheck(0);
  376. m_oSetupDialog.m_checkAutoDeinterlace.EnableWindow(FALSE);
  377. m_oSetupDialog.m_checkAutoDeinterlace.SetCheck(0);*/
  378. }
  379. else {
  380. /*m_oSetupDialog.m_btnSnapshot_bmp.EnableWindow(TRUE);
  381. m_oSetupDialog.m_btnSnapshot_jpg.EnableWindow(TRUE);
  382. m_oSetupDialog.m_btnRecordStart_1_1.EnableWindow(TRUE);
  383. m_oSetupDialog.m_btnRecordStop_1_1.EnableWindow(FALSE);
  384. m_oSetupDialog.m_btnRecordStart_1_2.EnableWindow(TRUE);
  385. m_oSetupDialog.m_btnRecordStop_1_2.EnableWindow(FALSE);
  386. m_oSetupDialog.m_btnRecordStart_1_3.EnableWindow(TRUE);
  387. m_oSetupDialog.m_btnRecordStop_1_3.EnableWindow(FALSE);
  388. m_oSetupDialog.m_checkGPU_1_1.EnableWindow(FALSE);
  389. m_oSetupDialog.m_checkGPU_1_1.SetCheck(0);
  390. m_oSetupDialog.m_checkGPU_1_2.EnableWindow(TRUE);
  391. m_oSetupDialog.m_checkGPU_1_2.SetCheck(0);
  392. m_oSetupDialog.m_checkGPU_1_3.EnableWindow(TRUE);
  393. m_oSetupDialog.m_checkGPU_1_3.SetCheck(0);
  394. m_oSetupDialog.m_checkAutoDeinterlace.EnableWindow(TRUE);
  395. m_oSetupDialog.m_checkAutoDeinterlace.SetCheck(0);*/
  396. }
  397. return TRUE;
  398. }
  399. BOOL CVideoCaptureView::HwUninitialize()
  400. {
  401. if (m_hVideoDevice != 0) {
  402. QCAP_STOP(m_hVideoDevice);
  403. QCAP_DESTROY(m_hVideoDevice);
  404. m_hVideoDevice = NULL;
  405. }
  406. return TRUE;
  407. }
  408. void CVideoCaptureView::CaptureSingleImage(LPTSTR lpszFileName, BOOL bIsJPG)
  409. {
  410. if (m_hVideoDevice != NULL)
  411. {
  412. if (bIsJPG)
  413. {
  414. QCAP_SNAPSHOT_JPG(m_hVideoDevice, lpszFileName, 100, FALSE, 5000); //参数4:是否异步;
  415. }
  416. else
  417. {
  418. QCAP_SNAPSHOT_BMP(m_hVideoDevice, lpszFileName, FALSE, 5000);
  419. }
  420. #ifdef _DEBUG
  421. Global::WriteTextLog("完成:CaptureSingleImage");
  422. #endif
  423. }
  424. }
  425. std::string CVideoCaptureView::CaptureSingleImageAutoName(LPCTSTR lpszDir, BOOL bIsJPG)
  426. {
  427. if (m_hVideoDevice != NULL)
  428. {
  429. TCHAR szPath[MAX_PATH] = { 0 };
  430. // 毫秒级;
  431. time_point<system_clock, ::chrono::milliseconds> tp = time_point_cast<std::chrono::milliseconds>(system_clock::now());
  432. auto tt = std::chrono::system_clock::to_time_t(tp);
  433. std::tm now = { 0 };
  434. localtime_s(&now, &tt); // 时间戳转成本地时间;
  435. int msc = tp.time_since_epoch().count() % 1000;
  436. if (bIsJPG)
  437. {
  438. _stprintf_s(szPath, _T("%s%04d%02d%02d%02d%02d%02d%03d.jpg"),
  439. lpszDir ? lpszDir : Global::g_szCurModuleDir,
  440. now.tm_year + 1990,
  441. now.tm_mon + 1,
  442. now.tm_mday,
  443. now.tm_hour,
  444. now.tm_min,
  445. now.tm_sec,
  446. msc);
  447. QCAP_SNAPSHOT_JPG(m_hVideoDevice, szPath, 100, FALSE, 5000);
  448. }
  449. else
  450. {
  451. _stprintf_s(szPath, _T("%s%04d%02d%02d%02d%02d%02d%03d.bmp"),
  452. lpszDir ? lpszDir : Global::g_szCurModuleDir,
  453. now.tm_year + 1990,
  454. now.tm_mon + 1,
  455. now.tm_mday,
  456. now.tm_hour,
  457. now.tm_min,
  458. now.tm_sec,
  459. msc);
  460. QCAP_SNAPSHOT_BMP(m_hVideoDevice, szPath, FALSE, 5000);
  461. }
  462. #ifdef _DEBUG
  463. Global::WriteTextLog("完成:CaptureSingleImageAutoName");
  464. #endif
  465. return std::string(szPath);
  466. }
  467. return std::string();
  468. }
  469. void CVideoCaptureView::CaptureMultiImage(LPCTSTR lpszDir, LPCTSTR lpszPrefix, BOOL bIsJPG, int nDurationTime)
  470. {
  471. std::thread t([&](CVideoCaptureView* p, LPCTSTR lpszDir, LPCTSTR lpszPrefix, BOOL bIsJPG, int nDurationTime) {
  472. // 连续截图;
  473. TCHAR szDir[MAX_PATH] = { 0 };
  474. TCHAR szPath[MAX_PATH] = { 0 };
  475. _stprintf_s(szDir, lpszPrefix ? _T("%s%s-") : _T("%s%s"),
  476. lpszDir ? lpszDir : Global::g_szCurModuleDir,
  477. lpszPrefix ? lpszPrefix : _T(""));
  478. auto start = system_clock::now();
  479. while (true)
  480. {
  481. // C++11获取当前时间;
  482. time_point<system_clock, ::chrono::milliseconds> tp = time_point_cast<std::chrono::milliseconds>(system_clock::now());
  483. if (bIsJPG)
  484. {
  485. _stprintf_s(szPath, _T("%s%lld.jpg"), szDir, tp.time_since_epoch().count());
  486. QCAP_SNAPSHOT_JPG(m_hVideoDevice, szPath, 100, TRUE);
  487. // 等待磁盘完成写入;
  488. //Sleep(20);
  489. }
  490. else
  491. {
  492. _stprintf_s(szPath, _T("%s%lld.bmp"), szDir, tp.time_since_epoch().count());
  493. QCAP_SNAPSHOT_BMP(m_hVideoDevice, szPath, TRUE);
  494. // 等待磁盘完成写入;
  495. //Sleep(530);
  496. }
  497. auto duration = duration_cast<milliseconds>(system_clock::now() - start);
  498. if (duration.count() >= nDurationTime)
  499. break;
  500. }
  501. }, this, lpszDir, lpszPrefix, bIsJPG, nDurationTime);
  502. t.detach();
  503. }
  504. void CVideoCaptureView::CaptureSingleImageEx(LPTSTR lpszFileName, BOOL bIsJPG)
  505. {
  506. {
  507. // 加锁;
  508. std::lock_guard<std::mutex> lk(m_mut_cpature);
  509. // 标记截图;
  510. m_bCaptureImage = TRUE;
  511. // 初始化数据;
  512. ZeroMemory(&m_CaptureInfo, sizeof(CaptureInfo));
  513. m_CaptureInfo.bIsJPG = bIsJPG;
  514. m_CaptureInfo.bSingle = TRUE;
  515. m_CaptureInfo.IsAutoName = FALSE;
  516. m_CaptureInfo.nCaputerCount = 0;
  517. m_CaptureInfo.nKeepTime = 0;
  518. memset(m_CaptureInfo.szPrefix, 0, 64);
  519. //_stprintf_s(m_CaptureInfo.szSaveDir, _T("%s"), lpszFileName);
  520. _stprintf_s(m_CaptureInfo.szSaveDir, _T("%s"), lpszFileName);
  521. //m_strCaptureName = lpszFileName;
  522. }
  523. // 等待截图完成;
  524. {
  525. // 加锁;
  526. std::unique_lock<std::mutex> lk(m_mut_cpature);
  527. m_capture_cond.wait(lk, [&]() {return !m_pBuffer; });
  528. // 解锁;
  529. lk.unlock();
  530. }
  531. }
  532. std::string CVideoCaptureView::CaptureSingleImageAutoNameEx(LPCTSTR lpszDir, BOOL bIsJPG)
  533. {
  534. {
  535. // 加锁;
  536. std::lock_guard<std::mutex> lk(m_mut_cpature);
  537. // 标记截图;
  538. m_bCaptureImage = TRUE;
  539. // 初始化数据;
  540. ZeroMemory(&m_CaptureInfo, sizeof(CaptureInfo));
  541. m_CaptureInfo.bIsJPG = bIsJPG;
  542. m_CaptureInfo.bSingle = TRUE;
  543. m_CaptureInfo.IsAutoName = FALSE;
  544. m_CaptureInfo.nCaputerCount = 0;
  545. m_CaptureInfo.nKeepTime = 0;
  546. memset(m_CaptureInfo.szPrefix, 0, 64);
  547. memset(m_CaptureInfo.szSaveDir, 0, MAX_PATH);
  548. // 毫秒级;
  549. time_point<system_clock, ::chrono::milliseconds> tp = time_point_cast<std::chrono::milliseconds>(system_clock::now());
  550. auto tt = std::chrono::system_clock::to_time_t(tp);
  551. std::tm now = { 0 };
  552. localtime_s(&now, &tt); // 时间戳转成本地时间;
  553. int msc = tp.time_since_epoch().count() % 1000;
  554. if (bIsJPG)
  555. {
  556. _stprintf_s(m_CaptureInfo.szSaveDir, _T("%s%04d%02d%02d%02d%02d%02d%03d.jpg"),
  557. lpszDir ? lpszDir : Global::g_szCurModuleDir,
  558. now.tm_year + 1990,
  559. now.tm_mon + 1,
  560. now.tm_mday,
  561. now.tm_hour,
  562. now.tm_min,
  563. now.tm_sec,
  564. msc);
  565. }
  566. else
  567. {
  568. _stprintf_s(m_CaptureInfo.szSaveDir, _T("%s%04d%02d%02d%02d%02d%02d%03d.bmp"),
  569. lpszDir ? lpszDir : Global::g_szCurModuleDir,
  570. now.tm_year + 1990,
  571. now.tm_mon + 1,
  572. now.tm_mday,
  573. now.tm_hour,
  574. now.tm_min,
  575. now.tm_sec,
  576. msc);
  577. }
  578. #ifdef DEBUG
  579. OutputDebugStringA(m_CaptureInfo.szSaveDir);
  580. OutputDebugStringA("\tCVideoCaptureView::CaptureSingleImageAutoNameEx\n");
  581. #endif // DEBUG
  582. }
  583. // 等待截图完成;
  584. {
  585. // 加锁;
  586. std::unique_lock<std::mutex> lk(m_mut_cpature);
  587. m_capture_cond.wait(lk, [&]() {return !m_pBuffer; });
  588. // 解锁;
  589. lk.unlock();
  590. }
  591. return std::string(m_CaptureInfo.szSaveDir);
  592. }
  593. void CVideoCaptureView::CaptureMultiImageEx(LPCTSTR lpszDir, LPCTSTR lpszPrefix, BOOL bIsJPG, int nDurationTime)
  594. {
  595. std::thread t([&](CVideoCaptureView* p, LPCTSTR lpszDir, LPCTSTR lpszPrefix, BOOL bIsJPG, int nDurationTime) {
  596. // 连续截图;
  597. TCHAR szDir[MAX_PATH] = { 0 };
  598. _stprintf_s(szDir, lpszPrefix ? _T("%s%s-") : _T("%s%s"),
  599. lpszDir ? lpszDir : Global::g_szCurModuleDir,
  600. lpszPrefix ? lpszPrefix : _T(""));
  601. auto start = system_clock::now();
  602. while (true)
  603. {
  604. // C++11获取当前时间;
  605. time_point<system_clock, ::chrono::milliseconds> tp = time_point_cast<std::chrono::milliseconds>(system_clock::now());
  606. {
  607. // 加锁;
  608. std::lock_guard<std::mutex> lk(m_mut_cpature);
  609. // 标记截图;
  610. m_bCaptureImage = TRUE;
  611. // 初始化数据;
  612. ZeroMemory(&m_CaptureInfo, sizeof(CaptureInfo));
  613. m_CaptureInfo.bIsJPG = bIsJPG;
  614. m_CaptureInfo.bSingle = FALSE;
  615. m_CaptureInfo.IsAutoName = TRUE;
  616. m_CaptureInfo.nCaputerCount = 0;
  617. m_CaptureInfo.nKeepTime = nDurationTime;
  618. _stprintf_s(m_CaptureInfo.szPrefix, _T("%s"), lpszPrefix);
  619. if (bIsJPG)
  620. {
  621. _stprintf_s(m_CaptureInfo.szSaveDir, _T("%s%lld.jpg"), szDir, tp.time_since_epoch().count());
  622. }
  623. else
  624. {
  625. _stprintf_s(m_CaptureInfo.szSaveDir, _T("%s%lld.bmp"), szDir, tp.time_since_epoch().count());
  626. }
  627. }
  628. // 等待截图完成;
  629. {
  630. // 加锁;
  631. std::unique_lock<std::mutex> lk(m_mut_cpature);
  632. m_capture_cond.wait(lk, [&]() {return !m_pBuffer; });
  633. // 解锁;
  634. lk.unlock();
  635. }
  636. auto duration = duration_cast<milliseconds>(system_clock::now() - start);
  637. if (duration.count() >= nDurationTime)
  638. break;
  639. }
  640. }, this, lpszDir, lpszPrefix, bIsJPG, nDurationTime);
  641. t.detach();
  642. }
  643. BOOL CVideoCaptureView::SaveImageByCaptureInfo(const CaptureInfo& capInfo)
  644. {
  645. if (m_pBuffer == NULL)
  646. {
  647. return FALSE;
  648. }
  649. BOOL bRet = FALSE;
  650. // 保存截图;
  651. HGLOBAL hMemery = GlobalAlloc(GMEM_MOVEABLE, m_nVideoWidth * m_nVideoHeight * 4);
  652. if (hMemery != NULL)
  653. {
  654. BYTE* pDstFrameBuffer = NULL;
  655. pDstFrameBuffer = (BYTE*)GlobalLock(hMemery);
  656. if (pDstFrameBuffer != NULL)
  657. {
  658. QRESULT QRet = QCAP_COLORSPACE_YUY2_TO_ABGR32(m_pBuffer,
  659. m_nVideoWidth,
  660. m_nVideoHeight,
  661. 0,
  662. pDstFrameBuffer,
  663. m_nVideoWidth,
  664. m_nVideoHeight,
  665. 0);
  666. Status stat = GenericError;
  667. Bitmap* pImg = ::new Bitmap(
  668. m_nVideoWidth,
  669. m_nVideoHeight,
  670. m_nVideoWidth * 4,
  671. PixelFormat32bppRGB,
  672. pDstFrameBuffer);
  673. if (m_bHoriontal && !m_bVertically)
  674. pImg->RotateFlip(RotateNoneFlipX);// 水平翻转;
  675. else if (m_bHoriontal && m_bVertically)
  676. pImg->RotateFlip(Rotate180FlipNone);// 270度;
  677. else if (!m_bHoriontal && m_bVertically)
  678. pImg->RotateFlip(Rotate180FlipX);// 垂直翻转;
  679. CLSID encoderClsid = { 0 };
  680. CString strFileName = m_CaptureInfo.szSaveDir;
  681. // 需要判断路径是否存在,不存在创建目录;
  682. int nIndex = strFileName.ReverseFind(_T('\\'));
  683. if (nIndex != -1)
  684. {
  685. if (!PathFileExists(strFileName.Left(nIndex)))
  686. {
  687. // 如果文件夹不存在,创建;
  688. SHCreateDirectoryEx(NULL, strFileName.Left(nIndex), NULL);
  689. }
  690. }
  691. BSTR newfile = strFileName.AllocSysString();
  692. if (!capInfo.bIsJPG)
  693. {
  694. Global::GetEncoderClsid(L"image/bmp", &encoderClsid);
  695. stat = pImg->Save(newfile, &encoderClsid, NULL);
  696. }
  697. else
  698. {
  699. Global::GetEncoderClsid(L"image/jpeg", &encoderClsid);
  700. EncoderParameters encoderParameters;
  701. encoderParameters.Count = 1;
  702. encoderParameters.Parameter[0].Guid = EncoderQuality;
  703. encoderParameters.Parameter[0].Type = EncoderParameterValueTypeLong;
  704. encoderParameters.Parameter[0].NumberOfValues = 1;
  705. // Save the image as a JPEG with quality level 100.
  706. ULONG uQuality = 100;
  707. encoderParameters.Parameter[0].Value = &uQuality;
  708. stat = pImg->Save(newfile, &encoderClsid, &encoderParameters);
  709. }
  710. if (pImg)
  711. ::delete pImg;
  712. pImg = NULL;
  713. SysFreeString(newfile);
  714. bRet = (stat == 0 ? TRUE : FALSE);
  715. #ifdef DEBUG
  716. if (bRet)
  717. {
  718. OutputDebugStringA(strFileName);
  719. OutputDebugStringA("\tSave Image::Ok\n");
  720. }
  721. #endif
  722. }
  723. #ifdef DEBUG
  724. else
  725. {
  726. OutputDebugStringA("pDstFrameBuffer == NULL\n");
  727. }
  728. #endif
  729. GlobalUnlock(hMemery);
  730. }
  731. #ifdef DEBUG
  732. else
  733. {
  734. DWORD dwError = GetLastError();
  735. CString strError = _T("");
  736. strError.Format(_T("hMemery == NULL分配内存出错:%ld\n"), dwError);
  737. OutputDebugStringA(strError);
  738. }
  739. #endif
  740. if (m_pBuffer)
  741. delete[]m_pBuffer;
  742. m_pBuffer = NULL;
  743. // 必须调用GlobalFree释放;
  744. GlobalFree(hMemery);
  745. return bRet;
  746. }
  747. void CVideoCaptureView::CaptureImageThread(CVideoCaptureView* pView)
  748. {
  749. CString strFileName = _T("");
  750. while (!pView->m_bAppQuit)
  751. {
  752. std::unique_lock<std::mutex> lk(pView->m_mut_thread); // 这里使用unique_lock是为了后面方便解锁
  753. #if 1 // 避免线程虚假唤醒;
  754. pView->m_thread_cond.wait(lk, [&]() {return pView->m_pBuffer; });
  755. #else
  756. while (!pView->m_pBuffer)
  757. pView->m_data_cond.wait(lk);
  758. #endif
  759. // 保存截图;
  760. pView->SaveImageByCaptureInfo(pView->m_CaptureInfo);
  761. // 通知截图完成;
  762. pView->m_capture_cond.notify_one();
  763. // 解锁;
  764. lk.unlock();
  765. // sleep 10ms;
  766. //this_thread::sleep_for(chrono::milliseconds(10));
  767. }
  768. }
  769. void CVideoCaptureView::StartRecord(DWORD dwDuration, LPCTSTR lpSavePath)
  770. {
  771. QRESULT QRet = QCAP_RS_SUCCESSFUL;
  772. if (!m_bIsMp4) // 音频格式;
  773. {
  774. QRet = QCAP_SET_AUDIO_RECORD_PROPERTY(m_hVideoDevice, 0, QCAP_ENCODER_TYPE_SOFTWARE, QCAP_ENCODER_FORMAT_PCM);
  775. //_tcscat_s(szPath, _T(".avi"));
  776. }
  777. else
  778. {
  779. QRet = QCAP_SET_AUDIO_RECORD_PROPERTY(m_hVideoDevice, 0, QCAP_ENCODER_TYPE_SOFTWARE, QCAP_ENCODER_FORMAT_AAC);
  780. //_tcscat_s(szPath, _T(".mp4"));
  781. }
  782. // 设置视频属性;
  783. QRet = QCAP_SET_VIDEO_RECORD_PROPERTY(
  784. m_hVideoDevice,
  785. 0,
  786. m_bSupportGPU ? QCAP_ENCODER_TYPE_INTEL_MEDIA_SDK : QCAP_ENCODER_TYPE_SOFTWARE,
  787. QCAP_ENCODER_FORMAT_H264,
  788. QCAP_RECORD_MODE_CBR,
  789. 8000,
  790. 16 * 1024 * 1024, // 16最高质量;
  791. 30, // 每秒多少帧
  792. 0,
  793. 0,
  794. QCAP_DOWNSCALE_MODE_OFF);
  795. // 开始录屏;
  796. QRet = QCAP_START_RECORD(m_hVideoDevice, 0, const_cast<CHAR*>(lpSavePath), QCAP_RECORD_FLAG_FULL, 0.0, 0.0, 0.0, 0, NULL);
  797. m_bIsRecord = TRUE;
  798. }
  799. void CVideoCaptureView::StopRecord()
  800. {
  801. QCAP_STOP_RECORD(m_hVideoDevice, 0);
  802. m_bIsRecord = FALSE;
  803. }
  804. void CVideoCaptureView::OnFilePrintPreview()
  805. {
  806. #ifndef SHARED_HANDLERS
  807. AFXPrintPreview(this);
  808. #endif
  809. }
  810. BOOL CVideoCaptureView::OnPreparePrinting(CPrintInfo* pInfo)
  811. {
  812. // 默认准备
  813. return DoPreparePrinting(pInfo);
  814. }
  815. void CVideoCaptureView::OnBeginPrinting(CDC* /*pDC*/, CPrintInfo* /*pInfo*/)
  816. {
  817. // TODO: 添加额外的打印前进行的初始化过程
  818. }
  819. void CVideoCaptureView::OnEndPrinting(CDC* /*pDC*/, CPrintInfo* /*pInfo*/)
  820. {
  821. // TODO: 添加打印后进行的清理过程
  822. }
  823. void CVideoCaptureView::OnRButtonUp(UINT /* nFlags */, CPoint point)
  824. {
  825. ClientToScreen(&point);
  826. OnContextMenu(this, point);
  827. }
  828. void CVideoCaptureView::OnContextMenu(CWnd* /* pWnd */, CPoint point)
  829. {
  830. #ifndef SHARED_HANDLERS
  831. theApp.GetContextMenuManager()->ShowPopupMenu(IDR_POPUP_EDIT, point.x, point.y, this, TRUE);
  832. #endif
  833. }
  834. // CVideoCaptureView 诊断
  835. #ifdef _DEBUG
  836. void CVideoCaptureView::AssertValid() const
  837. {
  838. CView::AssertValid();
  839. }
  840. void CVideoCaptureView::Dump(CDumpContext& dc) const
  841. {
  842. CView::Dump(dc);
  843. }
  844. CVideoCaptureDoc* CVideoCaptureView::GetDocument() const // 非调试版本是内联的
  845. {
  846. ASSERT(m_pDocument->IsKindOf(RUNTIME_CLASS(CVideoCaptureDoc)));
  847. return (CVideoCaptureDoc*)m_pDocument;
  848. }
  849. #endif //_DEBUG
  850. // CVideoCaptureView 消息处理程序
  851. void CVideoCaptureView::SetRibbonStatusBarText(CString strText, int uId)
  852. {
  853. if ( g_pMainFrame )
  854. {
  855. g_pMainFrame->SetRibbonStatusBarText(strText, uId);
  856. }
  857. }
  858. int CVideoCaptureView::OnCreate(LPCREATESTRUCT lpCreateStruct)
  859. {
  860. if (CView::OnCreate(lpCreateStruct) == -1)
  861. return -1;
  862. // 初始化共享内存;
  863. CMemoryClient::m_pView = this;
  864. g_ub530.InitMemery();
  865. g_ub530.StartThread();
  866. g_pMainFrame = (CMainFrame*)this->GetParent();
  867. // TODO: 在此添加您专用的创建代码
  868. // 创建视频流,并关联指定显示窗口;
  869. HwInitialize();
  870. // 创建截图线程;
  871. std::thread t(CaptureImageThread, this);
  872. t.detach();
  873. return 0;
  874. }
  875. void CVideoCaptureView::OnTimer(UINT_PTR nIDEvent)
  876. {
  877. // TODO: 在此添加消息处理程序代码和/或调用默认值
  878. if (nIDEvent == 0)
  879. {
  880. KillTimer(0);
  881. if (m_bNoSignal)
  882. {
  883. if (m_bIsRecord)
  884. {
  885. QCAP_STOP_RECORD(m_hVideoDevice, 0);
  886. m_bIsRecord = FALSE;
  887. }
  888. }
  889. else
  890. {
  891. }
  892. }
  893. CView::OnTimer(nIDEvent);
  894. }
  895. void CVideoCaptureView::OnCutBmp()
  896. {
  897. // TODO: 在此添加命令处理程序代码
  898. CaptureSingleImageAutoName(NULL, FALSE);
  899. }
  900. void CVideoCaptureView::OnCutJpg()
  901. {
  902. // TODO: 在此添加命令处理程序代码
  903. CaptureSingleImageAutoName(NULL, TRUE);
  904. }
  905. void CVideoCaptureView::OnStartRecord()
  906. {
  907. // TODO: 在此添加命令处理程序代码
  908. TCHAR szPath[MAX_PATH] = { 0 };
  909. // 毫秒级;
  910. time_point<system_clock, ::chrono::milliseconds> tp = time_point_cast<std::chrono::milliseconds>(system_clock::now());
  911. auto tt = std::chrono::system_clock::to_time_t(tp);
  912. std::tm now = { 0 };
  913. localtime_s(&now, &tt); // 时间戳转成本地时间;
  914. int msc = tp.time_since_epoch().count() % 1000;
  915. _stprintf_s(szPath, _T("%s%04d%02d%02d%02d%02d%02d%03d"),
  916. #ifdef _DEBUG
  917. _T("D:\\bin\\VideoCapture\\"),
  918. #else
  919. Global::g_szCurModuleDir,
  920. #endif
  921. now.tm_year + 1990,
  922. now.tm_mon + 1,
  923. now.tm_mday,
  924. now.tm_hour,
  925. now.tm_min,
  926. now.tm_sec,
  927. msc);
  928. QRESULT QRet = QCAP_RS_SUCCESSFUL;
  929. if (!m_bIsMp4) // 音频格式;
  930. {
  931. QRet = QCAP_SET_AUDIO_RECORD_PROPERTY(m_hVideoDevice, 0, QCAP_ENCODER_TYPE_SOFTWARE, QCAP_ENCODER_FORMAT_PCM);
  932. _tcscat_s(szPath, _T(".avi"));
  933. }
  934. else
  935. {
  936. QRet = QCAP_SET_AUDIO_RECORD_PROPERTY(m_hVideoDevice, 0, QCAP_ENCODER_TYPE_SOFTWARE, QCAP_ENCODER_FORMAT_AAC);
  937. _tcscat_s(szPath, _T(".mp4"));
  938. }
  939. // 设置视频属性;
  940. QRet = QCAP_SET_VIDEO_RECORD_PROPERTY(
  941. m_hVideoDevice,
  942. 0,
  943. m_bSupportGPU ? QCAP_ENCODER_TYPE_INTEL_MEDIA_SDK : QCAP_ENCODER_TYPE_SOFTWARE,
  944. QCAP_ENCODER_FORMAT_H264,
  945. QCAP_RECORD_MODE_CBR,
  946. 8000,
  947. 16*1024*1024, // 16最高质量;
  948. 30, // 每秒多少帧
  949. 0,
  950. 0,
  951. QCAP_DOWNSCALE_MODE_OFF);
  952. // 开始录屏;
  953. QRet = QCAP_START_RECORD(m_hVideoDevice, 0, szPath, QCAP_RECORD_FLAG_FULL, 0.0, 0.0, 0.0, 0, NULL);
  954. m_bIsRecord = TRUE;
  955. }
  956. void CVideoCaptureView::OnStopRecord()
  957. {
  958. // TODO: 在此添加命令处理程序代码
  959. QCAP_STOP_RECORD(m_hVideoDevice, 0);
  960. m_bIsRecord = FALSE;
  961. }
  962. void CVideoCaptureView::OnCheckVertically()
  963. {
  964. // TODO: 在此添加命令处理程序代码
  965. CMainFrame* pMain = (CMainFrame*)AfxGetMainWnd();
  966. CMFCRibbonButton* pCheckbox = pMain->GetRibbonButton(ID_CHECK_VERTICALLY);
  967. m_bVertically = !pCheckbox->IsChecked();
  968. if (m_hVideoDevice)
  969. {
  970. QCAP_SET_VIDEO_MIRROR(m_hVideoDevice, m_bHoriontal, m_bVertically);
  971. }
  972. }
  973. void CVideoCaptureView::OnCheckHoriontal()
  974. {
  975. // TODO: 在此添加命令处理程序代码
  976. CMainFrame* pMain = (CMainFrame*)AfxGetMainWnd();
  977. CMFCRibbonButton* pCheckbox = pMain->GetRibbonButton(ID_CHECK_HORIONTAL);
  978. m_bHoriontal = !pCheckbox->IsChecked();
  979. if (m_hVideoDevice)
  980. {
  981. QCAP_SET_VIDEO_MIRROR(m_hVideoDevice, m_bHoriontal, m_bVertically);
  982. }
  983. }
  984. void CVideoCaptureView::OnUpdateCheckVertically(CCmdUI* pCmdUI)
  985. {
  986. // TODO: 在此添加命令更新用户界面处理程序代码
  987. pCmdUI->SetCheck(m_bVertically);
  988. }
  989. void CVideoCaptureView::OnUpdateCheckHoriontal(CCmdUI* pCmdUI)
  990. {
  991. // TODO: 在此添加命令更新用户界面处理程序代码
  992. pCmdUI->SetCheck(m_bHoriontal);
  993. }
  994. void CVideoCaptureView::OnTraymenuReconnect()
  995. {
  996. // TODO: 在此添加命令处理程序代码
  997. HwUninitialize();// 断开;
  998. Invalidate();
  999. HwInitialize();// 重连;
  1000. }
  1001. void CVideoCaptureView::OnCheckSuportGpu()
  1002. {
  1003. // TODO: 在此添加命令处理程序代码
  1004. CMainFrame* pMain = (CMainFrame*)AfxGetMainWnd();
  1005. CMFCRibbonButton* pCheckbox = pMain->GetRibbonButton(ID_CHECK_SUPORT_GPU);
  1006. m_bSupportGPU = !pCheckbox->IsChecked();
  1007. }
  1008. void CVideoCaptureView::OnCheckFormatMp4()
  1009. {
  1010. CMainFrame* pMain = (CMainFrame*)AfxGetMainWnd();
  1011. CMFCRibbonButton* pCheckbox = pMain->GetRibbonButton(ID_CHECK_FORMAT_MP4);
  1012. m_bIsMp4 = !pCheckbox->IsChecked();
  1013. }
  1014. void CVideoCaptureView::OnCheckFormatAvi()
  1015. {
  1016. CMainFrame* pMain = (CMainFrame*)AfxGetMainWnd();
  1017. CMFCRibbonButton* pCheckbox = pMain->GetRibbonButton(ID_CHECK_FORMAT_AVI);
  1018. m_bIsMp4 = pCheckbox->IsChecked();
  1019. }
  1020. void CVideoCaptureView::OnUpdateCheckSuportGpu(CCmdUI* pCmdUI)
  1021. {
  1022. // TODO: 在此添加命令更新用户界面处理程序代码
  1023. pCmdUI->SetCheck(m_bSupportGPU);
  1024. }
  1025. void CVideoCaptureView::OnUpdateCheckFormatMp4(CCmdUI* pCmdUI)
  1026. {
  1027. // TODO: 在此添加命令更新用户界面处理程序代码
  1028. pCmdUI->SetCheck(m_bIsMp4);
  1029. }
  1030. void CVideoCaptureView::OnUpdateCheckFormatAvi(CCmdUI* pCmdUI)
  1031. {
  1032. // TODO: 在此添加命令更新用户界面处理程序代码
  1033. pCmdUI->SetCheck(!m_bIsMp4);
  1034. }
  1035. void CVideoCaptureView::OnUpdateStartRecord(CCmdUI* pCmdUI)
  1036. {
  1037. // TODO: 在此添加命令更新用户界面处理程序代码
  1038. pCmdUI->Enable(!m_bIsRecord);
  1039. }
  1040. void CVideoCaptureView::OnUpdateStopRecord(CCmdUI* pCmdUI)
  1041. {
  1042. // TODO: 在此添加命令更新用户界面处理程序代码
  1043. pCmdUI->Enable(m_bIsRecord);
  1044. }