KinectSDK2.0红外帧获取

#include "stdafx.h" #include "included.h" #define TITLE L"Title" #define WNDWIDTH 1024 #define WNDHEIGHT 768 #define lengthof(a) sizeof(a)/sizeof(*a) // ThisApp构造函数 ThisApp::ThisApp(){ } // ThisApp析构函数 ThisApp::~ThisApp(){ // 销毁事件 if (m_hInfraredFrameArrived && m_pInfraredFrameReader){ m_pInfraredFrameReader->UnsubscribeFrameArrived(m_hInfraredFrameArrived); m_hInfraredFrameArrived = 0; } // 释放InfraredFrameReader SafeRelease(m_pInfraredFrameReader); // 优雅地关闭Kinect if (m_pKinect){ m_pKinect->Close(); } SafeRelease(m_pKinect); } // 初始化 HRESULT ThisApp::Initialize(HINSTANCE hInstance, int nCmdShow){ HRESULT hr = E_FAIL; if (SUCCEEDED(static_cast<HRESULT>(m_ImagaRenderer))) { //register window class WNDCLASSEX wcex = { sizeof(WNDCLASSEX) }; wcex.style = CS_HREDRAW | CS_VREDRAW; wcex.lpfnWndProc = ThisApp::WndProc; wcex.cbClsExtra = 0; wcex.cbWndExtra = sizeof(LONG_PTR); wcex.hInstance = hInstance; wcex.hCursor = LoadCursorW(nullptr, IDC_ARROW); wcex.hbrBackground = nullptr; wcex.lpszMenuName = nullptr; wcex.lpszClassName = L"Direct2DTemplate"; wcex.hIcon = nullptr; // 注册窗口 RegisterClassEx(&wcex); // 创建窗口 RECT window_rect = { 0, 0, WNDWIDTH, WNDHEIGHT }; DWORD window_style = WS_CAPTION | WS_VISIBLE | WS_POPUP | WS_SYSMENU | WS_MINIMIZEBOX; AdjustWindowRect(&window_rect, window_style, FALSE); AdjustWindowRect(&window_rect, window_style, FALSE); window_rect.right -= window_rect.left; window_rect.bottom -= window_rect.top; window_rect.left = (GetSystemMetrics(SM_CXFULLSCREEN) - window_rect.right) / 2; window_rect.top = (GetSystemMetrics(SM_CYFULLSCREEN) - window_rect.bottom) / 2; m_hwnd = CreateWindowExW(0, wcex.lpszClassName, TITLE, window_style, window_rect.left, window_rect.top, window_rect.right, window_rect.bottom, 0, 0, hInstance, this); hr = m_hwnd ? S_OK : E_FAIL; if (SUCCEEDED(hr)) { // 设置窗口句柄 m_ImagaRenderer.SetHwnd(m_hwnd); // 显示窗口 ShowWindow(m_hwnd, nCmdShow); UpdateWindow(m_hwnd); } } return hr; } // 消息循环 void ThisApp::RunMessageLoop() { MSG msg; HANDLE events[] = { reinterpret_cast<HANDLE>(m_hInfraredFrameArrived) }; while (true){ // 消息处理 if (PeekMessageW(&msg, nullptr, 0, 0, PM_REMOVE)){ TranslateMessage(&msg); DispatchMessageW(&msg); } // 设置事件 // 事件0: 红外临帧事件 events[0] = reinterpret_cast<HANDLE>(m_hInfraredFrameArrived); // 检查事件 switch (MsgWaitForMultipleObjects(lengthof(events), events, FALSE, INFINITE, QS_ALLINPUT)) { // events[0] case WAIT_OBJECT_0 + 0: this->check_infrared_frame(); break; default: break; } // 退出 if (msg.message == WM_QUIT){ break; } } } // 窗口过程函数 LRESULT CALLBACK ThisApp::WndProc(HWND hwnd, UINT message, WPARAM wParam, LPARAM lParam) { LRESULT result = 0; if (message == WM_CREATE) { LPCREATESTRUCT pcs = (LPCREATESTRUCT)lParam; ThisApp *pOurApp = (ThisApp *)pcs->lpCreateParams; ::SetWindowLongPtrW( hwnd, GWLP_USERDATA, PtrToUlong(pOurApp) ); // 并初始化Kinect if (FAILED(pOurApp->init_kinect())){ ::MessageBoxW(hwnd, L"初始化Kinect v2失败", L"真的很遗憾", MB_ICONERROR); } result = 1; } else { ThisApp *pOurApp = reinterpret_cast<ThisApp *>(static_cast<LONG_PTR>( ::GetWindowLongPtrW( hwnd, GWLP_USERDATA ))); bool wasHandled = false; if (pOurApp) { switch (message) { case WM_DISPLAYCHANGE: InvalidateRect(hwnd, NULL, FALSE); result = 0; wasHandled = true; break; case WM_MOUSEWHEEL: pOurApp->m_ImagaRenderer.matrix._11 += 0.05f * static_cast<float>(static_cast<short>(HIWORD(wParam))) / static_cast<float>(WHEEL_DELTA); pOurApp->m_ImagaRenderer.matrix._22 = pOurApp->m_ImagaRenderer.matrix._11; pOurApp->m_ImagaRenderer.OnRender(); break; case WM_PAINT: pOurApp->m_ImagaRenderer.OnRender(); break; case WM_CLOSE: // 将收尾操作(如结束全部子线程)放在这里 DestroyWindow(hwnd); result = 1; wasHandled = true; break; case WM_DESTROY: PostQuitMessage(0); result = 1; wasHandled = true; break; } } if (!wasHandled) { result = DefWindowProc(hwnd, message, wParam, lParam); } } return result; } // 初始化Kinect HRESULT ThisApp::init_kinect(){ IInfraredFrameSource* pInfraredFrameSource = nullptr; // 查找当前默认Kinect HRESULT hr = ::GetDefaultKinectSensor(&m_pKinect); // 绅士地打开Kinect if (SUCCEEDED(hr)){ hr = m_pKinect->Open(); } // 获取红外帧源(InfraredFrameSource) if (SUCCEEDED(hr)){ hr = m_pKinect->get_InfraredFrameSource(&pInfraredFrameSource); } // 再获取红外帧读取器 if (SUCCEEDED(hr)){ #ifdef _DEBUG // 检查是否已存在 if (m_pInfraredFrameReader) ::MessageBoxW(m_hwnd, L"成员变量:m_pInfraredFrameReader 值已存在", L"<ThisApp::init_kinect>", MB_ICONERROR); #endif hr = pInfraredFrameSource->OpenReader(&m_pInfraredFrameReader); } // 注册临帧事件 if (SUCCEEDED(hr)){ #ifdef _DEBUG // 检查是否已存在 if (m_hInfraredFrameArrived) ::MessageBoxW(m_hwnd, L"成员变量:m_hInfraredFrameArrived 值已存在", L"<ThisApp::init_kinect>", MB_ICONERROR); #endif m_pInfraredFrameReader->SubscribeFrameArrived(&m_hInfraredFrameArrived); } SafeRelease(pInfraredFrameSource); return hr; } // 检查红外帧 void ThisApp::check_infrared_frame(){ if (!m_pInfraredFrameReader) return; #ifdef _DEBUG static int frame_num = 0; ++frame_num; _cwprintf(L"<ThisApp::check_infrared_frame>Frame@%8d ", frame_num); #endif // 红外临帧事件参数 IInfraredFrameArrivedEventArgs* pArgs = nullptr; // 红外帧引用 IInfraredFrameReference* pIRFrameRef = nullptr; // 红外帧 IInfraredFrame* pInfraredFrame = nullptr; // 帧描述 IFrameDescription* pFrameDescription = nullptr; // 红外帧宽度数据 int width = 0; // 红外帧高度数据 int height = 0; // 帧缓存大小 UINT nBufferSize = 0; // 帧缓存 UINT16 *pBuffer = nullptr; // 获取参数 HRESULT hr = m_pInfraredFrameReader->GetFrameArrivedEventData(m_hInfraredFrameArrived, &pArgs); // 获取引用 if (SUCCEEDED(hr)) { hr = pArgs->get_FrameReference(&pIRFrameRef); } // 获取红外帧 if (SUCCEEDED(hr)) { hr = pIRFrameRef->AcquireFrame(&pInfraredFrame); } // 获取帧描述 if (SUCCEEDED(hr)) { hr = pInfraredFrame->get_FrameDescription(&pFrameDescription); } // 获取帧宽度 if (SUCCEEDED(hr)) { hr = pFrameDescription->get_Width(&width); } // 获取帧高度 if (SUCCEEDED(hr)) { hr = pFrameDescription->get_Height(&height); } // 获取红外数据 if (SUCCEEDED(hr)) { hr = pInfraredFrame->AccessUnderlyingBuffer(&nBufferSize, &pBuffer); } // 处理并传输数据 if (SUCCEEDED(hr)){ auto pRGBXBuffer = m_ImagaRenderer.GetBuffer(); for (UINT i = 0U; i < nBufferSize; ++i){ pRGBXBuffer[i].rgbBlue = static_cast<BYTE>(pBuffer[i] >> 8); pRGBXBuffer[i].rgbGreen = pRGBXBuffer[i].rgbBlue; pRGBXBuffer[i].rgbRed = pRGBXBuffer[i].rgbBlue; pRGBXBuffer[i].rgbReserved = 0xFF; } m_ImagaRenderer.WriteBitmapData(width, height); } // 安全释放 SafeRelease(pFrame