.

cpp/winsharedutils/screenshot.cpp

@@ -1,6 +1,5 @@
 
 
-
 void GetVirtualDesktopRect(RECT &rect)
 {
     // 获取虚拟桌面的尺寸和位置
@@ -31,72 +30,6 @@ HBITMAP GetBitmap(RECT &rect, HDC hDC)
 
     return hBitmap;
 }
-int SaveBitmapToFile(HBITMAP hBitmap, LPCWSTR lpFileName)
-{
-    WORD wBitCount; // 位图中每个像素所占字节数
-    // 定义调色板大小，位图中像素字节大小，位图文件大小，写入文件字节数
-    DWORD dwPaletteSize = 0, dwBmBitsSize, dwDIBSize, dwWritten;
-    BITMAP Bitmap;           // 位图属性结构
-    BITMAPFILEHEADER bmfHdr; // 位图文件头结构
-    BITMAPINFOHEADER bi;     // 位图信息头结构
-    HANDLE fh;               // 定义文件，分配内存句柄，调色板句柄
-    LPSTR lpbk, lpmem;
-
-    wBitCount = 32;
-    // 设置位图信息头结构
-    GetObject(hBitmap, sizeof(BITMAP), (LPSTR)&Bitmap);
-    bi.biSize = sizeof(BITMAPINFOHEADER);
-    bi.biWidth = Bitmap.bmWidth;
-    bi.biHeight = Bitmap.bmHeight; // 为负,正向的位图;为正,倒向的位图
-    bi.biPlanes = 1;
-    bi.biBitCount = wBitCount;
-    bi.biCompression = BI_RGB;
-    bi.biSizeImage = 0;
-    bi.biXPelsPerMeter = 0;
-    bi.biYPelsPerMeter = 0;
-    bi.biClrUsed = 0;
-    bi.biClrImportant = 0;
-    dwBmBitsSize = ((Bitmap.bmWidth * wBitCount + 31) / 32) * 4 * Bitmap.bmHeight;
-
-    // 创建位图文件
-    fh = CreateFile(lpFileName, GENERIC_WRITE, 0, NULL, CREATE_ALWAYS,
-                    FILE_ATTRIBUTE_NORMAL | FILE_FLAG_SEQUENTIAL_SCAN, NULL);
-    if (fh == INVALID_HANDLE_VALUE)
-        return FALSE;
-    // 设置位图文件头
-    bmfHdr.bfType = 0x4D42; //   "BM"
-    dwDIBSize = sizeof(BITMAPFILEHEADER) + sizeof(BITMAPINFOHEADER) + dwBmBitsSize;
-    bmfHdr.bfSize = dwDIBSize;
-    bmfHdr.bfReserved1 = 0;
-    bmfHdr.bfReserved2 = 0;
-    bmfHdr.bfOffBits = (DWORD)sizeof(BITMAPFILEHEADER) + (DWORD)sizeof(BITMAPINFOHEADER);
-
-    // 写入位图文件头
-    WriteFile(fh, (LPSTR)&bmfHdr, sizeof(BITMAPFILEHEADER), &dwWritten, NULL);
-    // 写入位图信息头
-    WriteFile(fh, (LPSTR)&bi, sizeof(BITMAPINFOHEADER), &dwWritten, NULL);
-
-    // 获取位图阵列
-    lpmem = new char[dwBmBitsSize];
-    lpbk = (LPSTR) new char[dwBmBitsSize];
-    GetBitmapBits(hBitmap, dwBmBitsSize, lpmem); // 正向的内存图象数据
-
-    // 转化为倒向数据(仅在bmHeight为正时需要)
-    for (int i = 0; i < Bitmap.bmHeight; i++)
-    {
-        memcpy(lpbk + Bitmap.bmWidth * i * 4, lpmem + Bitmap.bmWidth * (Bitmap.bmHeight - i - 1) * 4, Bitmap.bmWidth * 4);
-    }
-    // 写位图数据
-    WriteFile(fh, lpbk, dwBmBitsSize, &dwWritten, NULL);
-
-    // 清除
-    delete[] lpbk;
-    delete[] lpmem;
-
-    CloseHandle(fh);
-    return TRUE;
-}
-
 std::vector<byte> SaveBitmapToBuffer(HBITMAP hBitmap)
 {
     WORD wBitCount; // 位图中每个像素所占字节数
@@ -167,8 +100,6 @@ DECLARE_API void gdi_screenshot(HWND hwnd, RECT rect, void (*cb)(byte *, size_t)
     if (!hdc)
         return;
     auto bm = GetBitmap(rect, hdc);
-    // SaveBitmapToFile(bm, LR"(.\2.bmp)");
-    size_t size;
     auto bf = std::move(SaveBitmapToBuffer(bm));
     if (bf.size())
         cb(bf.data(), bf.size());

py/LunaTranslator/myutils/hwnd.py

@@ -23,11 +23,12 @@ def clipboard_set_image(p: QImage):
 
 @threader
 def grabwindow(app="PNG", callback_origin=None, tocliponly=False):
+    tmsp = time.strftime("%Y-%m-%d-%H-%M-%S", time.localtime())
     if tocliponly:
         fname = ""
         uid = None
     elif callback_origin or tocliponly:
-        fname = gobject.gettempdir(time.strftime("%Y-%m-%d-%H-%M-%S", time.localtime()))
+        fname = gobject.gettempdir(tmsp)
         uid = None
     else:
 
@@ -36,29 +37,26 @@ def grabwindow(app="PNG", callback_origin=None, tocliponly=False):
             hwndx = windows.GetForegroundWindow()
         hwndx = windows.GetAncestor(hwndx)
         gamepath = getpidexe(windows.GetWindowThreadProcessId(hwndx))
-        exename = os.path.basename(gamepath).replace(
-            "." + os.path.basename(gamepath).split(".")[-1], ""
-        )
+        exename = os.path.splitext(os.path.basename(gamepath))[0]
         uid = gobject.baseobject.gameuid
         screenshot_savepath: str = globalconfig.get("screenshot_savepath", "")
-        fnamef = lambda: gobject.getcachedir(
-            "screenshot/"
-            + exename
-            + "/"
-            + time.strftime("%Y-%m-%d-%H-%M-%S", time.localtime()),
-            abspath=False,
-        )
-        if screenshot_savepath:
-            try:
-                dirname = screenshot_savepath.format(exename=exename)
-                os.makedirs(dirname, exist_ok=True)
-                fname = os.path.join(
-                    dirname, time.strftime("%Y-%m-%d-%H-%M-%S", time.localtime())
-                )
-            except:
-                fname = fnamef()
-        else:
-            fname = fnamef()
+
+        try:
+            if not screenshot_savepath:
+                raise
+            dirname = screenshot_savepath.format(exename=exename)
+            os.makedirs(dirname, exist_ok=True)
+            fname = os.path.join(dirname, tmsp)
+        except:
+            fnamebase = "cache/screenshot"
+            if not os.path.exists(fnamebase):
+                fnamebase = windows.SHGetFolderPathW(windows.CSIDL_MYPICTURES)
+                if not (fnamebase and os.path.exists(fnamebase)):
+                    fnamebase = "cache/screenshot"
+                else:
+                    fnamebase = os.path.join(fnamebase, "LunaTranslator")
+            fname = os.path.join(fnamebase, exename, tmsp)
+            os.makedirs(os.path.join(fnamebase, exename), exist_ok=True)
 
     def callback_1(callback_origin, uid, tocliponly, p: QPixmap, fn):
         if p.isNull():
@@ -80,7 +78,6 @@ def grabwindow(app="PNG", callback_origin=None, tocliponly=False):
     hwnd = windows.GetAncestor(hwnd)
     _ = windows.GetClientRect(hwnd)
     p = gdi_screenshot(0, 0, _[2], _[3], hwnd)
-
     callback(p, fname + "_gdi." + app)
     isshit = (not callback_origin) and (not tocliponly)
     if p.isNull() or isshit:

py/LunaTranslator/windows.py

@@ -53,6 +53,7 @@ from ctypes.wintypes import (
     USHORT,
 )
 
+HRESULT = LONG
 HWINEVENTHOOK = HANDLE
 LRESULT = LPLONG
 WAIT_TIMEOUT = 258
@@ -962,3 +963,23 @@ SetWinEventHook.argtypes = DWORD, DWORD, HMODULE, WINEVENTPROC, DWORD, DWORD, DW
 PathFileExists = windll.Shlwapi.PathFileExistsW
 PathFileExists.argtypes = (LPCWSTR,)
 PathFileExists.restype = BOOL
+
+_SHGetFolderPathW = _shell32.SHGetFolderPathW
+_SHGetFolderPathW.argtypes = (
+    HWND,
+    c_int,
+    HANDLE,
+    DWORD,
+    LPWSTR,
+)
+_SHGetFolderPathW.restype = HRESULT
+CSIDL_MYPICTURES = 0x27
+SHGFP_TYPE_CURRENT = 0
+S_OK = 0
+
+
+def SHGetFolderPathW(csidl):
+    buff = create_unicode_buffer(MAX_PATH + 100)
+    if _SHGetFolderPathW(None, csidl, None, SHGFP_TYPE_CURRENT, buff) != S_OK:
+        return None
+    return buff.value