diff --git a/Overlay/Overlay.vcxproj b/Overlay/Overlay.vcxproj index e06dac2..d581838 100644 --- a/Overlay/Overlay.vcxproj +++ b/Overlay/Overlay.vcxproj @@ -43,7 +43,7 @@ Application true v143 - Unicode + MultiByte Application @@ -73,6 +73,9 @@ true + + true + Level3 @@ -107,9 +110,11 @@ true _DEBUG;_CONSOLE;%(PreprocessorDefinitions) true + stdcpp20 + MultiThreadedDebug - Console + Windows true @@ -122,6 +127,7 @@ NDEBUG;_CONSOLE;%(PreprocessorDefinitions) true stdcpp20 + MultiThreaded Windows diff --git a/Overlay/functions/captureScreen.h b/Overlay/functions/captureScreen.h index d9865a5..6317ce0 100644 --- a/Overlay/functions/captureScreen.h +++ b/Overlay/functions/captureScreen.h @@ -1,8 +1,10 @@ #pragma once #include "addWatermark.h" -void captureScreen(std::string file_screenshot) +inline void captureScreen() { + std::string file_screenshot = "screenshot_" + std::to_string(time(NULL)) + ".bmp"; + HDC hdcScreen; HDC hdcDIB; HBITMAP hbmDIB; diff --git a/Overlay/functions/screenRecorder.h b/Overlay/functions/screenRecorder.h index de25395..9d1504b 100644 --- a/Overlay/functions/screenRecorder.h +++ b/Overlay/functions/screenRecorder.h @@ -1,8 +1,56 @@ #pragma once #include "../Include/Include.h" +#include -void screenRecorder(std::string file_video) +inline void screenRecorder(std::string ffilename) { - std::string command = "ffmpeg -f gdigrab -framerate 60 -i desktop -c:v libx264 -preset ultrafast -tune zerolatency -crf 25 -pix_fmt yuv420p " + ("C:\\Users\\" + (std::string)std::getenv("USERNAME") + "\\Documents\\vos.team\\video\\" + file_video); - system(command.c_str()); + std::string filename = "video.avi"; //temporary fix + + int screen_width = GetSystemMetrics(SM_CXSCREEN); + int screen_height = GetSystemMetrics(SM_CYSCREEN); + cv::Size screen_size(screen_width, screen_height); + + int fourcc = cv::VideoWriter::fourcc('M', 'J', 'P', 'G'); + + // Create video writer object + cv::VideoWriter writer(filename, fourcc, 60.0, screen_size); + + // Check if writer is ready + if (!writer.isOpened()) + { + std::cout << "Error opening video file" << std::endl; + return; + } + + // Capture screen and write frames to video file + while (true) + { + // Capture screen image + HDC hScreen = GetDC(NULL); + HDC hDC = CreateCompatibleDC(hScreen); + HBITMAP hBitmap = CreateCompatibleBitmap(hScreen, screen_width, screen_height); + HGDIOBJ hOld = SelectObject(hDC, hBitmap); + BitBlt(hDC, 0, 0, screen_width, screen_height, hScreen, 0, 0, SRCCOPY); + cv::Mat img = cv::Mat(screen_height, screen_width, CV_8UC4); + //cv::cuda::GpuMat img = cv::cuda::GpuMat(screen_height, screen_width, CV_8UC4); + GetBitmapBits(hBitmap, screen_width * screen_height * 4, img.data); + cv::cvtColor(img, img, cv::COLOR_RGBA2RGB); + + // Write frame to video file + writer.write(img); + + // Release resources + SelectObject(hDC, hOld); + DeleteDC(hDC); + DeleteObject(hBitmap); + ReleaseDC(NULL, hScreen); + + // Check for escape key to exit loop (doesn't work) + if (GetAsyncKeyState(VK_ESCAPE)) + break; + } + + // Release video writer and close windows + writer.release(); + cv::destroyAllWindows(); } \ No newline at end of file diff --git a/Overlay/main.cpp b/Overlay/main.cpp index e44d963..41cd13a 100644 --- a/Overlay/main.cpp +++ b/Overlay/main.cpp @@ -10,7 +10,6 @@ LRESULT CALLBACK WndProc(HWND hwnd, UINT msg, WPARAM wParam, LPARAM lParam) } std::string file_screenshot = "screenshot_" + std::to_string(time(NULL)) + ".bmp"; - std::string file_video = "video_" + std::to_string(time(NULL)) + ".mp4"; switch (msg) { @@ -26,11 +25,12 @@ LRESULT CALLBACK WndProc(HWND hwnd, UINT msg, WPARAM wParam, LPARAM lParam) } else if (wParam == 3) { - captureScreen(file_screenshot); + captureScreen(); } else if (wParam == 4) { - screenRecorder(file_video); + std::thread t(screenRecorder, file_screenshot); + t.join(); } break; case WM_PAINT: @@ -104,6 +104,5 @@ int WINAPI WinMain(HINSTANCE hInstance, HINSTANCE hPrevInstance, LPSTR lpCmdLine UnregisterHotKey(hwnd, 3); UnregisterHotKey(hwnd, 4); - exit(0); return Msg.wParam; } \ No newline at end of file