diff --git a/.gitignore b/.gitignore index 441e471..d1d2522 100644 --- a/.gitignore +++ b/.gitignore @@ -3,3 +3,6 @@ build/ *.dylib *.log .DS_Store + +# 构建产物：由 scripts/gen-icons.js 从 src/assets/*.svg 生成 +src/generated/ diff --git a/binding.gyp b/binding.gyp index efcab16..146d579 100644 --- a/binding.gyp +++ b/binding.gyp @@ -27,7 +27,7 @@ [ "OS=='win'", { - "sources": ["src/binding_windows.cpp"], + "sources": ["src/binding_windows.cpp", "src/screenshot_windows.cpp"], "libraries": [ "user32.lib", "kernel32.lib", @@ -38,7 +38,8 @@ "uiautomationcore.lib", "gdiplus.lib", "dwmapi.lib", - "gdi32.lib" + "gdi32.lib", + "imm32.lib" ], "msvs_settings": { "VCCLCompilerTool": { diff --git a/scripts/build.js b/scripts/build.js index 87c7818..5cda9fd 100644 --- a/scripts/build.js +++ b/scripts/build.js @@ -7,11 +7,15 @@ const platform = os.platform(); console.log(`🔨 Building for ${platform}...\n`); try { - // Step 1: 编译 C++ 原生模块 (跨平台) + // Step 1: 把 SVG 资源生成成 C 头文件（截图工具栏图标） + console.log('🎨 Generating icon header from SVG...'); + execSync('node scripts/gen-icons.js', { stdio: 'inherit' }); + + // Step 2: 编译 C++ 原生模块 (跨平台) console.log('📦 Running node-gyp rebuild...'); execSync('npx node-gyp rebuild', { stdio: 'inherit' }); - // Step 2: macOS 需要额外编译 Swift + // Step 3: macOS 需要额外编译 Swift if (platform === 'darwin') { console.log('\n🍎 Building Swift library for macOS...'); execSync('npm run build:swift', { stdio: 'inherit' }); diff --git a/scripts/gen-icons.js b/scripts/gen-icons.js new file mode 100644 index 0000000..e95b1e1 --- /dev/null +++ b/scripts/gen-icons.js @@ -0,0 +1,90 @@ +#!/usr/bin/env node +/** + * 把 src/assets/*.svg 生成成 C 头文件 src/generated/icon_svgs.h。 + * + * 输出形如： + * static const char* kIconSvg_Rect = "..."; + * + * SVG 原文原样保留（含 currentColor），运行时再把 currentColor 替换成 + * 目标颜色（normal/hover/active 三态），从而一套文本支持多色。 + * + * 自动生成，请勿手动编辑。 + */ +const fs = require('fs'); +const path = require('path'); + +// 工具栏按钮 -> SVG 文件名映射（顺序与 src/screenshot_windows.cpp 的 ToolButton 枚举一致） +// 分隔线（Separator）对应 null，不生成图标。 +const ICON_MAP = { + Rect: 'square', + Circle: 'circle', + Arrow: 'arrow-up-right', + Brush: 'pencil', + Mosaic: 'mosaic', + Text: 'text', + Translate: 'translate', + Undo: 'arrow-back-up', + Redo: 'arrow-forward-up', + Save: 'download', + Cancel: 'x', + Confirm: 'check', +}; + +const ASSETS_DIR = path.join(__dirname, '..', 'src', 'assets'); +const OUT_DIR = path.join(__dirname, '..', 'src', 'generated'); +const OUT_FILE = path.join(OUT_DIR, 'icon_svgs.h'); + +// C 字符串字面量转义：处理 " \ 和换行 +function toCLiteral(str) { + // 把字符串按行拆开，每行用独立字面量拼接，保证可读性且避免超长行 + return str + .replace(/\\/g, '\\\\') + .replace(/"/g, '\\"') + .split(/\r?\n/) + .map((line) => ` "${line}"`) + .join('\n'); +} + +function main() { + if (!fs.existsSync(ASSETS_DIR)) { + console.error(`✖ assets dir not found: ${ASSETS_DIR}`); + process.exit(1); + } + fs.mkdirSync(OUT_DIR, { recursive: true }); + + const lines = []; + lines.push('#pragma once'); + lines.push('// AUTO-GENERATED by scripts/gen-icons.js — DO NOT EDIT.'); + lines.push('// SVG 原文保留 currentColor，运行时替换为目标颜色。'); + lines.push(''); + + for (const [key, fileBase] of Object.entries(ICON_MAP)) { + const svgPath = path.join(ASSETS_DIR, `${fileBase}.svg`); + if (!fs.existsSync(svgPath)) { + console.error(`✖ missing svg: ${svgPath}`); + process.exit(1); + } + const raw = fs.readFileSync(svgPath, 'utf8').trim(); + // 折叠成单行，去掉换行，再按 C 字面量拼接（更紧凑） + const oneLine = raw.replace(/\r?\n/g, '').replace(/\s+/g, ' '); + lines.push(`static const char* kIconSvg_${key} =`); + lines.push(` "${oneLine.replace(/\\/g, '\\\\').replace(/"/g, '\\"')}";`); + lines.push(''); + } + + // 所有图标文本的数组（按 ToolButton 枚举顺序，分隔线为 nullptr） + lines.push('// 按 ToolButton 枚举顺序的 SVG 文本数组，分隔线为 nullptr。'); + lines.push('static const char* const kIconSvgs[] = {'); + // 顺序：Rect Circle Arrow Brush Mosaic Text Translate SEP1 Undo Redo SEP2 Save Cancel Confirm + lines.push(' kIconSvg_Rect, kIconSvg_Circle, kIconSvg_Arrow, kIconSvg_Brush,'); + lines.push(' kIconSvg_Mosaic, kIconSvg_Text, kIconSvg_Translate, nullptr,'); + lines.push(' kIconSvg_Undo, kIconSvg_Redo, nullptr,'); + lines.push(' kIconSvg_Save, kIconSvg_Cancel, kIconSvg_Confirm'); + lines.push('};'); + lines.push(''); + + fs.writeFileSync(OUT_FILE, lines.join('\n'), 'utf8'); + console.log(`✓ generated ${path.relative(path.join(__dirname, '..'), OUT_FILE)} (${Object.keys(ICON_MAP).length} icons)`); +} + +main(); diff --git a/src/assets/arrow-back-up.svg b/src/assets/arrow-back-up.svg new file mode 100644 index 0000000..150ab62 --- /dev/null +++ b/src/assets/arrow-back-up.svg @@ -0,0 +1 @@ + \ No newline at end of file diff --git a/src/assets/arrow-forward-up.svg b/src/assets/arrow-forward-up.svg new file mode 100644 index 0000000..e73f85f --- /dev/null +++ b/src/assets/arrow-forward-up.svg @@ -0,0 +1 @@ + \ No newline at end of file diff --git a/src/assets/arrow-up-right.svg b/src/assets/arrow-up-right.svg new file mode 100644 index 0000000..7d37443 --- /dev/null +++ b/src/assets/arrow-up-right.svg @@ -0,0 +1 @@ + \ No newline at end of file diff --git a/src/assets/check.svg b/src/assets/check.svg new file mode 100644 index 0000000..a002cfa --- /dev/null +++ b/src/assets/check.svg @@ -0,0 +1 @@ + \ No newline at end of file diff --git a/src/assets/circle.svg b/src/assets/circle.svg new file mode 100644 index 0000000..344bca9 --- /dev/null +++ b/src/assets/circle.svg @@ -0,0 +1 @@ + \ No newline at end of file diff --git a/src/assets/download.svg b/src/assets/download.svg new file mode 100644 index 0000000..7d9aaef --- /dev/null +++ b/src/assets/download.svg @@ -0,0 +1,5 @@ + + + + \ No newline at end of file diff --git a/src/assets/mosaic.svg b/src/assets/mosaic.svg new file mode 100644 index 0000000..36e2be4 --- /dev/null +++ b/src/assets/mosaic.svg @@ -0,0 +1,8 @@ + + + + \ No newline at end of file diff --git a/src/assets/pencil.svg b/src/assets/pencil.svg new file mode 100644 index 0000000..1004c5d --- /dev/null +++ b/src/assets/pencil.svg @@ -0,0 +1 @@ + \ No newline at end of file diff --git a/src/assets/square.svg b/src/assets/square.svg new file mode 100644 index 0000000..1439270 --- /dev/null +++ b/src/assets/square.svg @@ -0,0 +1 @@ + \ No newline at end of file diff --git a/src/assets/text.svg b/src/assets/text.svg new file mode 100644 index 0000000..709ea3e --- /dev/null +++ b/src/assets/text.svg @@ -0,0 +1 @@ + \ No newline at end of file diff --git a/src/assets/translate.svg b/src/assets/translate.svg new file mode 100644 index 0000000..ef066d4 --- /dev/null +++ b/src/assets/translate.svg @@ -0,0 +1 @@ + \ No newline at end of file diff --git a/src/assets/x.svg b/src/assets/x.svg new file mode 100644 index 0000000..9f797b7 --- /dev/null +++ b/src/assets/x.svg @@ -0,0 +1 @@ + \ No newline at end of file diff --git a/src/binding_windows.cpp b/src/binding_windows.cpp index c04af2a..ebb107f 100644 --- a/src/binding_windows.cpp +++ b/src/binding_windows.cpp @@ -26,6 +26,8 @@ #pragma comment(lib, "dwmapi.lib") #pragma comment(lib, "uiautomationcore.lib") +#include "screenshot_windows.h" + // DWMWA_CLOAKED 在较新的 Windows SDK 中定义，为了兼容性手动定义 #ifndef DWMWA_CLOAKED #define DWMWA_CLOAKED 14 @@ -67,12 +69,6 @@ static std::thread g_windowMessageThread; static HWND g_lastMonitoredWindow = NULL; static std::string g_lastMonitoredTitle; -// 全局变量 - 区域截图 -static HWND g_screenshotOverlayWindow = NULL; -static std::atomic g_isCapturing(false); -static napi_threadsafe_function g_screenshotTsfn = nullptr; -static std::thread g_screenshotThread; - // 全局变量 - 鼠标监控 static HHOOK g_mouseHook = NULL; static std::atomic g_isMouseMonitoring(false); @@ -859,1160 +855,6 @@ Napi::Value ActivateWindow(const Napi::CallbackInfo& info) { return Napi::Boolean::New(env, newForeground == hwnd); } -// ==================== 区域截图功能（预截屏 + 双缓冲架构） ==================== - -// 截图常量 -static const int SC_PANEL_WIDTH = 140; -static const int SC_PANEL_HEIGHT = 140; -static const int SC_MAGNIFIER_HEIGHT = 74; -static const int SC_PANEL_MARGIN = 15; -static const int SC_PANEL_CORNER_RADIUS = 8; -static const int SC_ZOOM_FACTOR = 4; - -// 截图状态枚举 -enum CaptureState { CS_Idle, CS_Selecting, CS_Done, CS_Cancelled }; - -// 窗口信息 -struct SCWindowInfo { - HWND hwnd; - RECT rect; - std::wstring title; -}; - -// 截图结果结构 -struct ScreenshotResult { - bool success; - int x; - int y; - int x2; - int y2; - int width; - int height; - std::string base64; -}; - -// GDI 资源缓存 -struct SCGdiResources { - HBRUSH bgBrush; - HPEN borderPen; - HPEN crosshairPen; - HPEN selectionPen; - HPEN highlightPen; - HFONT smallFont; - - void Init() { - bgBrush = CreateSolidBrush(RGB(52, 52, 53)); - borderPen = CreatePen(PS_SOLID, 0, RGB(102, 102, 102)); - crosshairPen = CreatePen(PS_SOLID, 1, RGB(0, 136, 255)); - selectionPen = CreatePen(PS_SOLID, 1, RGB(0, 136, 255)); - highlightPen = CreatePen(PS_SOLID, 3, RGB(0, 136, 255)); - // 创建字体 - LOGFONTW lf = {}; - lf.lfHeight = -12; - lf.lfCharSet = DEFAULT_CHARSET; - wcscpy_s(lf.lfFaceName, L"微软雅黑"); - smallFont = CreateFontIndirectW(&lf); - } - - void Cleanup() { - if (bgBrush) { DeleteObject(bgBrush); bgBrush = NULL; } - if (borderPen) { DeleteObject(borderPen); borderPen = NULL; } - if (crosshairPen) { DeleteObject(crosshairPen); crosshairPen = NULL; } - if (selectionPen) { DeleteObject(selectionPen); selectionPen = NULL; } - if (highlightPen) { DeleteObject(highlightPen); highlightPen = NULL; } - if (smallFont) { DeleteObject(smallFont); smallFont = NULL; } - } -}; - -// 截图上下文 -struct CaptureContext { - CaptureState state; - int virtualX, virtualY, virtualW, virtualH; - int startX, startY, endX, endY; - int mouseX, mouseY; - COLORREF currentColor; - std::vector windows; - int hoveredWindow; // -1 = none - // 预截屏 - HBITMAP screenBitmap; - HDC memDC; - // 双缓冲 - HDC backDC; - HBITMAP backBitmap; - // 脏区域追踪 - RECT lastPanelRect; - RECT lastSelectionRect; - RECT lastLabelRect; - RECT lastHighlightRect; - bool needFullRedraw; - // DPI - double dpiScale; - // GDI 资源 - SCGdiResources gdi; -}; - -// 截图上下文指针（窗口过程使用） -static CaptureContext* g_captureCtx = nullptr; - -// 前向声明（定义在文件后面的应用图标提取部分） -static int GetPngEncoderClsid(CLSID* pClsid); - -// Base64 编码表 -static const char base64_chars[] = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/"; - -// Base64 编码 -static std::string Base64Encode(const BYTE* data, size_t len) { - std::string result; - result.reserve(((len + 2) / 3) * 4); - for (size_t i = 0; i < len; i += 3) { - unsigned int b = (data[i] << 16) | ((i + 1 < len ? data[i + 1] : 0) << 8) | (i + 2 < len ? data[i + 2] : 0); - result.push_back(base64_chars[(b >> 18) & 0x3F]); - result.push_back(base64_chars[(b >> 12) & 0x3F]); - result.push_back(i + 1 < len ? base64_chars[(b >> 6) & 0x3F] : '='); - result.push_back(i + 2 < len ? base64_chars[b & 0x3F] : '='); - } - return result; -} - -// ---- 工具函数 ---- - -// 获取 DPI 缩放因子 -static double GetDpiScaleFactor() { - typedef UINT (WINAPI *GetDpiForSystemProc)(); - HMODULE user32 = GetModuleHandleW(L"user32.dll"); - if (user32) { - auto proc = (GetDpiForSystemProc)GetProcAddress(user32, "GetDpiForSystem"); - if (proc) { - UINT dpi = proc(); - double scale = dpi / 96.0; - if (scale < 0.5) scale = 0.5; - if (scale > 4.0) scale = 4.0; - return scale; - } - } - return 1.0; -} - -// 显示器枚举回调数据 -struct MonitorEnumData { - LONG minLeft, minTop, maxRight, maxBottom; - double totalDpiScale; - int monitorCount; -}; - -// 显示器枚举回调 -static BOOL CALLBACK MonitorEnumProc(HMONITOR hMonitor, HDC hdcMonitor, LPRECT lprcMonitor, LPARAM dwData) { - MonitorEnumData* data = reinterpret_cast(dwData); - - // 获取显示器的物理尺寸 - MONITORINFOEXW mi; - mi.cbSize = sizeof(MONITORINFOEXW); - if (GetMonitorInfoW(hMonitor, &mi)) { - // 在 DPI 感知模式下，rcMonitor 已经是物理像素坐标 - data->minLeft = (std::min)(data->minLeft, mi.rcMonitor.left); - data->minTop = (std::min)(data->minTop, mi.rcMonitor.top); - data->maxRight = (std::max)(data->maxRight, mi.rcMonitor.right); - data->maxBottom = (std::max)(data->maxBottom, mi.rcMonitor.bottom); - data->monitorCount++; - - // 获取显示器 DPI - typedef HRESULT(WINAPI* GetDpiForMonitorProc)(HMONITOR, int, UINT*, UINT*); - HMODULE shcore = LoadLibraryW(L"shcore.dll"); - if (shcore) { - auto getDpiForMonitor = (GetDpiForMonitorProc)GetProcAddress(shcore, "GetDpiForMonitor"); - if (getDpiForMonitor) { - UINT dpiX, dpiY; - if (SUCCEEDED(getDpiForMonitor(hMonitor, 0/*MDT_EFFECTIVE_DPI*/, &dpiX, &dpiY))) { - data->totalDpiScale = (std::max)(data->totalDpiScale, dpiX / 96.0); - } - } - FreeLibrary(shcore); - } - } - return TRUE; -} - -// 截取整个虚拟屏幕到物理尺寸位图 -static bool CaptureVirtualScreen(HDC& outMemDC, HBITMAP& outBitmap, - int& vx, int& vy, int& vw, int& vh, double& dpiScale) { - // 获取逻辑坐标的虚拟屏幕尺寸 - vx = GetSystemMetrics(SM_XVIRTUALSCREEN); - vy = GetSystemMetrics(SM_YVIRTUALSCREEN); - vw = GetSystemMetrics(SM_CXVIRTUALSCREEN); - vh = GetSystemMetrics(SM_CYVIRTUALSCREEN); - - // 枚举所有显示器获取物理像素边界 - MonitorEnumData enumData = { INT_MAX, INT_MAX, INT_MIN, INT_MIN, 1.0, 0 }; - EnumDisplayMonitors(NULL, NULL, MonitorEnumProc, reinterpret_cast(&enumData)); - - // 计算物理尺寸（使用枚举得到的实际物理像素边界） - int physVx = enumData.minLeft; - int physVy = enumData.minTop; - int physVw = enumData.maxRight - enumData.minLeft; - int physVh = enumData.maxBottom - enumData.minTop; - - // 如果枚举失败，回退到 DPI 缩放计算 - if (physVw <= 0 || physVh <= 0 || enumData.monitorCount == 0) { - physVx = (int)(vx * dpiScale); - physVy = (int)(vy * dpiScale); - physVw = (int)(vw * dpiScale + 0.5); - physVh = (int)(vh * dpiScale + 0.5); - } - - HDC screenDC = GetDC(NULL); - if (!screenDC) return false; - - outMemDC = CreateCompatibleDC(screenDC); - if (!outMemDC) { ReleaseDC(NULL, screenDC); return false; } - - outBitmap = CreateCompatibleBitmap(screenDC, physVw, physVh); - if (!outBitmap) { DeleteDC(outMemDC); ReleaseDC(NULL, screenDC); return false; } - - SelectObject(outMemDC, outBitmap); - - // 设置高质量拉伸模式 - SetStretchBltMode(outMemDC, HALFTONE); - SetBrushOrgEx(outMemDC, 0, 0, NULL); - - // 直接 BitBlt 物理像素（在 DPI 感知模式下，屏幕 DC 和坐标都是物理像素级别） - BitBlt(outMemDC, 0, 0, physVw, physVh, screenDC, physVx, physVy, SRCCOPY); - - // 更新返回的 dpiScale 为实际的物理/逻辑比例 - // 这样后续的坐标转换才能正确 - if (vw > 0 && vh > 0) { - dpiScale = (double)physVw / vw; - } - - ReleaseDC(NULL, screenDC); - return true; -} - -// 创建双缓冲 -static bool CreateBackBuffer(HDC& outDC, HBITMAP& outBmp, int w, int h) { - HDC screenDC = GetDC(NULL); - if (!screenDC) return false; - outDC = CreateCompatibleDC(screenDC); - if (!outDC) { ReleaseDC(NULL, screenDC); return false; } - outBmp = CreateCompatibleBitmap(screenDC, w, h); - if (!outBmp) { DeleteDC(outDC); ReleaseDC(NULL, screenDC); return false; } - SelectObject(outDC, outBmp); - ReleaseDC(NULL, screenDC); - return true; -} - -// 从预截屏位图读取像素颜色（逻辑坐标） -static COLORREF GetPixelColorFromBitmap(HDC memDC, int x, int y, int vx, int vy, double dpiScale) { - int lx = x - vx; - int ly = y - vy; - int px = (int)(lx * dpiScale + 0.5); - int py = (int)(ly * dpiScale + 0.5); - return GetPixel(memDC, px, py); -} - -// COLORREF 转 HEX/RGB 字符串 -static void ColorrefToStrings(COLORREF color, char* hexBuf, char* rgbBuf) { - int r = color & 0xFF; - int g = (color >> 8) & 0xFF; - int b = (color >> 16) & 0xFF; - sprintf_s(hexBuf, 32, "#%02X%02X%02X", r, g, b); - sprintf_s(rgbBuf, 32, "%d, %d, %d", r, g, b); -} - -// 枚举窗口回调 -static BOOL CALLBACK SCEnumWindowsProc(HWND hwnd, LPARAM lParam) { - auto* windows = reinterpret_cast*>(lParam); - - if (!IsWindowVisible(hwnd)) return TRUE; - - LONG_PTR exStyle = GetWindowLongPtrW(hwnd, GWL_EXSTYLE); - if (exStyle & WS_EX_TOOLWINDOW) return TRUE; - - LONG_PTR style = GetWindowLongPtrW(hwnd, GWL_STYLE); - if (style == 0) return TRUE; - - // 检查是否为幽灵窗口（cloaked window） - // 幽灵窗口虽然 IsWindowVisible 返回 true，但实际上不可见 - BOOL isCloaked = FALSE; - HRESULT hrCloaked = DwmGetWindowAttribute(hwnd, DWMWA_CLOAKED, &isCloaked, sizeof(isCloaked)); - if (SUCCEEDED(hrCloaked) && isCloaked) { - return TRUE; // 跳过幽灵窗口 - } - - // 获取窗口类名以进行额外过滤 - const int MAX_CLASS_NAME = 256; - WCHAR className[MAX_CLASS_NAME] = {0}; - int classNameLen = GetClassNameW(hwnd, className, MAX_CLASS_NAME); - - // 过滤某些特殊的系统窗口类 - if (classNameLen > 0) { - // Windows 输入法相关窗口（如 Microsoft Text Input Application） - if (wcscmp(className, L"Windows.UI.Core.CoreWindow") == 0) { - // 对于 CoreWindow，再次确认是否真的可见（通过检查是否有有效的可视化区域） - RECT clientRect; - if (!GetClientRect(hwnd, &clientRect)) return TRUE; - - // 如果客户区太小，很可能是输入法等后台窗口 - int clientW = clientRect.right - clientRect.left; - int clientH = clientRect.bottom - clientRect.top; - if (clientW < 100 || clientH < 100) return TRUE; - } - - // 过滤 ApplicationFrameWindow 的空壳窗口 - // UWP 应用在未激活时可能留下空的 ApplicationFrameWindow - if (wcscmp(className, L"ApplicationFrameWindow") == 0) { - // 检查窗口是否被最小化或隐藏 - if (IsIconic(hwnd)) return TRUE; - - // 检查是否真的有内容（通过检查窗口透明度或其他属性） - BYTE opacity = 255; - DWORD cloakedReason = 0; - DwmGetWindowAttribute(hwnd, DWMWA_CLOAKED, &cloakedReason, sizeof(cloakedReason)); - if (cloakedReason != 0) return TRUE; - } - } - - int titleLen = GetWindowTextLengthW(hwnd); - if (titleLen == 0) return TRUE; - - std::wstring title(titleLen + 1, L'\0'); - GetWindowTextW(hwnd, &title[0], titleLen + 1); - title.resize(titleLen); - - if (hwnd == GetDesktopWindow()) return TRUE; - - // 使用 DWM 获取精确边界 - RECT rect = {}; - HRESULT hr = DwmGetWindowAttribute(hwnd, DWMWA_EXTENDED_FRAME_BOUNDS, &rect, sizeof(rect)); - if (FAILED(hr)) { - if (!GetWindowRect(hwnd, &rect)) return TRUE; - } - - int w = rect.right - rect.left; - int h = rect.bottom - rect.top; - if (w < 50 || h < 50) return TRUE; - - SCWindowInfo info; - info.hwnd = hwnd; - info.rect = rect; - info.title = title; - windows->push_back(info); - return TRUE; -} - -// 枚举窗口 -static std::vector EnumWindowsForCapture() { - std::vector windows; - EnumWindows(SCEnumWindowsProc, reinterpret_cast(&windows)); - return windows; -} - -// 查找鼠标下方的窗口 -static int FindWindowAtPoint(const std::vector& windows, int x, int y) { - for (size_t i = 0; i < windows.size(); i++) { - const RECT& r = windows[i].rect; - if (x >= r.left && x < r.right && y >= r.top && y < r.bottom) - return (int)i; - } - return -1; -} - -// 计算浮窗位置（优先右下，超出则翻转） -static void CalcPanelPosition(int mx, int my, int vx, int vy, int vw, int vh, int& px, int& py) { - int sr = vx + vw; - int sb = vy + vh; - px = mx + SC_PANEL_MARGIN; - py = my + SC_PANEL_MARGIN; - if (px + SC_PANEL_WIDTH > sr) px = mx - SC_PANEL_WIDTH - SC_PANEL_MARGIN; - if (py + SC_PANEL_HEIGHT > sb) py = my - SC_PANEL_HEIGHT - SC_PANEL_MARGIN; - if (px < vx) px = vx + SC_PANEL_MARGIN; - if (py < vy) py = vy + SC_PANEL_MARGIN; -} - -// 从预截屏位图恢复脏区域到后台缓冲 -static void RestoreDirtyRegion(HDC backDC, HDC memDC, const RECT& dirty, double dpiScale) { - int w = dirty.right - dirty.left; - int h = dirty.bottom - dirty.top; - if (w <= 0 || h <= 0) return; - int x = (std::max)((int)dirty.left, 0); - int y = (std::max)((int)dirty.top, 0); - w = dirty.right - x; - h = dirty.bottom - y; - if (dpiScale > 1.01 || dpiScale < 0.99) { - int px = (int)(x * dpiScale + 0.5); - int py = (int)(y * dpiScale + 0.5); - int pw = (int)(w * dpiScale + 0.5); - int ph = (int)(h * dpiScale + 0.5); - StretchBlt(backDC, x, y, w, h, memDC, px, py, pw, ph, SRCCOPY); - } else { - BitBlt(backDC, x, y, w, h, memDC, x, y, SRCCOPY); - } -} - -// 扩展矩形 -static RECT InflateRectBy(const RECT& r, int margin) { - return { r.left - margin, r.top - margin, r.right + margin, r.bottom + margin }; -} - -// ---- 绘制函数 ---- - -// 绘制放大镜 + 鼠标信息面板 -static void DrawInfoPanel(HDC hdc, int panelX, int panelY, COLORREF color, - HDC memDC, int vx, int vy, int mx, int my, double dpiScale, const SCGdiResources& gdi) { - HGDIOBJ oldBrush = SelectObject(hdc, gdi.bgBrush); - HGDIOBJ oldPen = SelectObject(hdc, gdi.borderPen); - - // 圆角矩形背景 - RoundRect(hdc, panelX, panelY, panelX + SC_PANEL_WIDTH, panelY + SC_PANEL_HEIGHT, - SC_PANEL_CORNER_RADIUS, SC_PANEL_CORNER_RADIUS); - - // 放大镜：从物理尺寸位图取像素 - int srcW = SC_PANEL_WIDTH / SC_ZOOM_FACTOR; - int srcH = SC_MAGNIFIER_HEIGHT / SC_ZOOM_FACTOR; - int mxLogical = mx - vx; - int myLogical = my - vy; - int mxPhysical = (int)(mxLogical * dpiScale + 0.5); - int myPhysical = (int)(myLogical * dpiScale + 0.5); - int srcWPhysical = (int)(srcW * dpiScale + 0.5); - int srcHPhysical = (int)(srcH * dpiScale + 0.5); - int srcXPhysical = mxPhysical - srcWPhysical / 2; - int srcYPhysical = myPhysical - srcHPhysical / 2; - - int magX = panelX + 2; - int magY = panelY + 2; - int magW = SC_PANEL_WIDTH - 4; - int magH = SC_MAGNIFIER_HEIGHT - 2; - - StretchBlt(hdc, magX, magY, magW, magH, memDC, - (std::max)(srcXPhysical, 0), (std::max)(srcYPhysical, 0), - srcWPhysical, srcHPhysical, SRCCOPY); - - // 十字准星 - SelectObject(hdc, gdi.crosshairPen); - int cx = magX + magW / 2; - int cy = magY + magH / 2; - MoveToEx(hdc, magX, cy, NULL); LineTo(hdc, magX + magW, cy); - MoveToEx(hdc, cx, magY, NULL); LineTo(hdc, cx, magY + magH); - - // 文字信息 - SetBkMode(hdc, TRANSPARENT); - SetTextColor(hdc, RGB(255, 255, 255)); - HGDIOBJ oldFont = SelectObject(hdc, gdi.smallFont); - - char hexBuf[32], rgbBuf[32]; - ColorrefToStrings(color, hexBuf, rgbBuf); - char posBuf[64]; - sprintf_s(posBuf, "%d, %d", mx, my); - - const int LABEL_PAD = 6; - int labelX = panelX + LABEL_PAD; - int valueRightX = panelX + SC_PANEL_WIDTH - LABEL_PAD; - - // 获取文字高度 - SIZE textSize; - GetTextExtentPoint32W(hdc, L"测试", 2, &textSize); - int lineH = textSize.cy; - int infoY = panelY + SC_PANEL_HEIGHT - LABEL_PAD - lineH * 3; - - // 辅助：右对齐绘制 - auto drawRightAligned = [&](const wchar_t* text, int len, int rx, int ry) { - SIZE sz; - GetTextExtentPoint32W(hdc, text, len, &sz); - TextOutW(hdc, rx - sz.cx, ry, text, len); - }; - - // 坐标 - TextOutW(hdc, labelX, infoY, L"坐标", 2); - std::wstring posW(posBuf, posBuf + strlen(posBuf)); - drawRightAligned(posW.c_str(), (int)posW.size(), valueRightX, infoY); - - // HEX - TextOutW(hdc, labelX, infoY + lineH, L"HEX", 3); - std::wstring hexW(hexBuf, hexBuf + strlen(hexBuf)); - drawRightAligned(hexW.c_str(), (int)hexW.size(), valueRightX, infoY + lineH); - - // RGB - TextOutW(hdc, labelX, infoY + lineH * 2, L"RGB", 3); - std::wstring rgbW(rgbBuf, rgbBuf + strlen(rgbBuf)); - drawRightAligned(rgbW.c_str(), (int)rgbW.size(), valueRightX, infoY + lineH * 2); - - SelectObject(hdc, oldFont); - SelectObject(hdc, oldBrush); - SelectObject(hdc, oldPen); -} - -// 绘制尺寸标签，返回标签矩形 -static RECT DrawSizeLabel(HDC hdc, int width, int height, - int refLeft, int refTop, int refRight, int refBottom, - int virtualW, int virtualH, const SCGdiResources& gdi) { - RECT empty = {0, 0, 0, 0}; - if (width < 0 || height < 0) return empty; - - wchar_t sizeBuf[64]; - swprintf_s(sizeBuf, L"%d × %d", width, height); - int sizeLen = (int)wcslen(sizeBuf); - - HGDIOBJ oldFont = SelectObject(hdc, gdi.smallFont); - SIZE textSize; - GetTextExtentPoint32W(hdc, sizeBuf, sizeLen, &textSize); - - const int LP = 12, LS = 5; - int labelW = textSize.cx + LP * 2; - int labelH = textSize.cy + 4; - - int lx = refLeft; - int ly = refTop - labelH - LS; - if (ly < 0) { - lx = refLeft + LS; - ly = refTop + LS; - if (lx + labelW > virtualW) lx = virtualW - labelW - LS; - if (ly + labelH > virtualH) ly = virtualH - labelH - LS; - if (lx + labelW > refRight) lx = refRight - labelW - LS; - if (ly + labelH > refBottom) ly = refBottom - labelH - LS; - } - if (lx < 0) lx = 0; - if (ly < 0) ly = 0; - if (lx + labelW > virtualW) lx = virtualW - labelW; - if (ly + labelH > virtualH) ly = virtualH - labelH; - - HGDIOBJ oldBrush = SelectObject(hdc, gdi.bgBrush); - HGDIOBJ oldPen = SelectObject(hdc, gdi.borderPen); - RoundRect(hdc, lx, ly, lx + labelW, ly + labelH, SC_PANEL_CORNER_RADIUS, SC_PANEL_CORNER_RADIUS); - - SetBkMode(hdc, TRANSPARENT); - SetTextColor(hdc, RGB(255, 255, 255)); - TextOutW(hdc, lx + LP, ly + 2, sizeBuf, sizeLen); - - SelectObject(hdc, oldFont); - SelectObject(hdc, oldBrush); - SelectObject(hdc, oldPen); - - RECT result = { lx, ly, lx + labelW, ly + labelH }; - return result; -} - -// 绘制选区矩形边框 + 尺寸标签 -static RECT DrawSelection(HDC hdc, int x1, int y1, int x2, int y2, - int vx, int vy, int vw, int vh, const SCGdiResources& gdi) { - int left = (std::min)(x1, x2) - vx; - int top = (std::min)(y1, y2) - vy; - int right = (std::max)(x1, x2) - vx; - int bottom = (std::max)(y1, y2) - vy; - - HGDIOBJ oldPen = SelectObject(hdc, gdi.selectionPen); - HGDIOBJ oldBrush = SelectObject(hdc, GetStockObject(NULL_BRUSH)); - Rectangle(hdc, left, top, right, bottom); - - int sizeW = right - left; - int sizeH = bottom - top; - RECT labelRect = DrawSizeLabel(hdc, sizeW, sizeH, left, top, right, bottom, vw, vh, gdi); - - SelectObject(hdc, oldPen); - SelectObject(hdc, oldBrush); - return labelRect; -} - -// 绘制窗口高亮边框 -static void DrawWindowHighlight(HDC hdc, const RECT& rect, int vx, int vy, const SCGdiResources& gdi) { - int left = rect.left - vx; - int top = rect.top - vy; - int right = rect.right - vx; - int bottom = rect.bottom - vy; - - HGDIOBJ oldPen = SelectObject(hdc, gdi.highlightPen); - HGDIOBJ oldBrush = SelectObject(hdc, GetStockObject(NULL_BRUSH)); - Rectangle(hdc, left, top, right, bottom); - SelectObject(hdc, oldPen); - SelectObject(hdc, oldBrush); -} - -// 将 HBITMAP 转换为 PNG base64 字符串 -static std::string BitmapToBase64Png(HBITMAP hBitmap) { - Gdiplus::GdiplusStartupInput gdiplusStartupInput; - ULONG_PTR gdiplusToken; - Gdiplus::GdiplusStartup(&gdiplusToken, &gdiplusStartupInput, NULL); - - std::string result; - { - Gdiplus::Bitmap* bmp = Gdiplus::Bitmap::FromHBITMAP(hBitmap, NULL); - if (bmp) { - CLSID pngClsid; - if (GetPngEncoderClsid(&pngClsid) >= 0) { - IStream* stream = NULL; - CreateStreamOnHGlobal(NULL, TRUE, &stream); - if (stream && bmp->Save(stream, &pngClsid, NULL) == Gdiplus::Ok) { - HGLOBAL hMem = NULL; - GetHGlobalFromStream(stream, &hMem); - size_t len = GlobalSize(hMem); - BYTE* ptr = (BYTE*)GlobalLock(hMem); - if (ptr && len > 0) { - result = "data:image/png;base64," + Base64Encode(ptr, len); - } - GlobalUnlock(hMem); - } - if (stream) stream->Release(); - } - delete bmp; - } - } - Gdiplus::GdiplusShutdown(gdiplusToken); - return result; -} - -// 保存位图到剪贴板 -static bool SaveBitmapToClipboard(HBITMAP hBitmap) { - if (!OpenClipboard(NULL)) return false; - EmptyClipboard(); - BITMAP bm; - GetObject(hBitmap, sizeof(BITMAP), &bm); - HBITMAP hCopy = (HBITMAP)CopyImage(hBitmap, IMAGE_BITMAP, bm.bmWidth, bm.bmHeight, LR_COPYRETURNORG); - SetClipboardData(CF_BITMAP, hCopy); - CloseClipboard(); - return true; -} - -// 从预截屏位图提取区域，生成 base64 并复制到剪贴板 -static ScreenshotResult* ExtractRegionResult(HDC memDC, const RECT& rect, - int vx, int vy, double dpiScale) { - ScreenshotResult* result = new ScreenshotResult(); - result->success = false; - int width = rect.right - rect.left; - int height = rect.bottom - rect.top; - result->x = rect.left; - result->y = rect.top; - result->x2 = rect.right; - result->y2 = rect.bottom; - result->width = width; - result->height = height; - - if (width <= 0 || height <= 0) return result; - - // 从物理尺寸位图提取区域 - int lx = rect.left - vx; - int ly = rect.top - vy; - int px = (int)(lx * dpiScale + 0.5); - int py = (int)(ly * dpiScale + 0.5); - int pw = (int)(width * dpiScale + 0.5); - int ph = (int)(height * dpiScale + 0.5); - - HDC screenDC = GetDC(NULL); - HDC regionDC = CreateCompatibleDC(screenDC); - HBITMAP regionBmp = CreateCompatibleBitmap(screenDC, pw, ph); - SelectObject(regionDC, regionBmp); - BitBlt(regionDC, 0, 0, pw, ph, memDC, px, py, SRCCOPY); - - // 如果有 DPI 缩放，缩放回逻辑尺寸 - HBITMAP finalBmp = regionBmp; - HDC finalDC = regionDC; - if (dpiScale > 1.01 || dpiScale < 0.99) { - HDC scaledDC = CreateCompatibleDC(screenDC); - HBITMAP scaledBmp = CreateCompatibleBitmap(screenDC, width, height); - SelectObject(scaledDC, scaledBmp); - SetStretchBltMode(scaledDC, HALFTONE); - SetBrushOrgEx(scaledDC, 0, 0, NULL); - StretchBlt(scaledDC, 0, 0, width, height, regionDC, 0, 0, pw, ph, SRCCOPY); - DeleteDC(regionDC); - DeleteObject(regionBmp); - finalBmp = scaledBmp; - finalDC = scaledDC; - } - - // 生成 base64 - result->base64 = BitmapToBase64Png(finalBmp); - // 复制到剪贴板 - result->success = SaveBitmapToClipboard(finalBmp); - - DeleteDC(finalDC); - DeleteObject(finalBmp); - ReleaseDC(NULL, screenDC); - - return result; -} - -// ---- 窗口过程和线程 ---- - -// 在主线程调用 JS 回调（截图完成） -static void CallScreenshotJs(napi_env env, napi_value js_callback, void* context, void* data) { - if (env != nullptr && js_callback != nullptr && data != nullptr) { - ScreenshotResult* result = static_cast(data); - - napi_value resultObj; - napi_create_object(env, &resultObj); - - napi_value success; - napi_get_boolean(env, result->success, &success); - napi_set_named_property(env, resultObj, "success", success); - - if (result->success) { - napi_value x, y, x2, y2, width, height, base64; - napi_create_int32(env, result->x, &x); - napi_set_named_property(env, resultObj, "x", x); - napi_create_int32(env, result->y, &y); - napi_set_named_property(env, resultObj, "y", y); - napi_create_int32(env, result->x2, &x2); - napi_set_named_property(env, resultObj, "x2", x2); - napi_create_int32(env, result->y2, &y2); - napi_set_named_property(env, resultObj, "y2", y2); - napi_create_int32(env, result->width, &width); - napi_set_named_property(env, resultObj, "width", width); - napi_create_int32(env, result->height, &height); - napi_set_named_property(env, resultObj, "height", height); - napi_create_string_utf8(env, result->base64.c_str(), result->base64.size(), &base64); - napi_set_named_property(env, resultObj, "base64", base64); - } - - napi_value global; - napi_get_global(env, &global); - napi_call_function(env, global, js_callback, 1, &resultObj, nullptr); - delete result; - } -} - -// 截图覆盖层窗口过程（双缓冲渲染） -static LRESULT CALLBACK ScreenshotOverlayWndProc(HWND hwnd, UINT msg, WPARAM wParam, LPARAM lParam) { - CaptureContext* ctx = g_captureCtx; - if (!ctx) return DefWindowProc(hwnd, msg, wParam, lParam); - - switch (msg) { - case WM_PAINT: { - PAINTSTRUCT ps; - HDC hdc = BeginPaint(hwnd, &ps); - HDC backDC = ctx->backDC; - - // 计算浮窗位置 - int panelX, panelY; - CalcPanelPosition(ctx->mouseX, ctx->mouseY, - ctx->virtualX, ctx->virtualY, ctx->virtualW, ctx->virtualH, panelX, panelY); - // 转为相对坐标 - int panelXRel = panelX - ctx->virtualX; - int panelYRel = panelY - ctx->virtualY; - - RECT curPanelRect = { panelXRel, panelYRel, - panelXRel + SC_PANEL_WIDTH, panelYRel + SC_PANEL_HEIGHT }; - - // 当前选区矩形 - RECT curSelRect = {0,0,0,0}; - if (ctx->state == CS_Selecting) { - curSelRect.left = (std::min)(ctx->startX, ctx->endX) - ctx->virtualX; - curSelRect.top = (std::min)(ctx->startY, ctx->endY) - ctx->virtualY; - curSelRect.right = (std::max)(ctx->startX, ctx->endX) - ctx->virtualX; - curSelRect.bottom = (std::max)(ctx->startY, ctx->endY) - ctx->virtualY; - } - - // 当前高亮窗口矩形 - RECT curHlRect = {0,0,0,0}; - if (ctx->state == CS_Idle && ctx->hoveredWindow >= 0 && ctx->hoveredWindow < (int)ctx->windows.size()) { - const RECT& wr = ctx->windows[ctx->hoveredWindow].rect; - curHlRect = { wr.left - ctx->virtualX, wr.top - ctx->virtualY, - wr.right - ctx->virtualX, wr.bottom - ctx->virtualY }; - } - - double ds = ctx->dpiScale; - int physW = (int)(ctx->virtualW * ds + 0.5); - int physH = (int)(ctx->virtualH * ds + 0.5); - - // 恢复背景 - if (ctx->needFullRedraw) { - if (ds > 1.01 || ds < 0.99) { - StretchBlt(backDC, 0, 0, ctx->virtualW, ctx->virtualH, - ctx->memDC, 0, 0, physW, physH, SRCCOPY); - } else { - BitBlt(backDC, 0, 0, ctx->virtualW, ctx->virtualH, - ctx->memDC, 0, 0, SRCCOPY); - } - ctx->needFullRedraw = false; - } else { - // 脏区域恢复 - RestoreDirtyRegion(backDC, ctx->memDC, InflateRectBy(ctx->lastPanelRect, 2), ds); - if (ctx->lastSelectionRect.right > ctx->lastSelectionRect.left) - RestoreDirtyRegion(backDC, ctx->memDC, InflateRectBy(ctx->lastSelectionRect, 5), ds); - if (ctx->lastLabelRect.right > ctx->lastLabelRect.left) - RestoreDirtyRegion(backDC, ctx->memDC, InflateRectBy(ctx->lastLabelRect, 2), ds); - if (ctx->lastHighlightRect.right > ctx->lastHighlightRect.left) - RestoreDirtyRegion(backDC, ctx->memDC, InflateRectBy(ctx->lastHighlightRect, 5), ds); - } - - // 绘制窗口高亮（Idle 状态） - if (ctx->state == CS_Idle) { - if (ctx->hoveredWindow >= 0 && ctx->hoveredWindow < (int)ctx->windows.size()) { - // 高亮悬停的窗口 - DrawWindowHighlight(backDC, ctx->windows[ctx->hoveredWindow].rect, - ctx->virtualX, ctx->virtualY, ctx->gdi); - } else { - // 没有匹配到窗口时，高亮鼠标所在的屏幕 - POINT pt = { ctx->mouseX, ctx->mouseY }; - HMONITOR hMonitor = MonitorFromPoint(pt, MONITOR_DEFAULTTONEAREST); - if (hMonitor) { - MONITORINFO monitorInfo; - monitorInfo.cbSize = sizeof(MONITORINFO); - if (GetMonitorInfo(hMonitor, &monitorInfo)) { - DrawWindowHighlight(backDC, monitorInfo.rcMonitor, - ctx->virtualX, ctx->virtualY, ctx->gdi); - } - } - } - } - - // 绘制选区或窗口尺寸标签 - RECT curLabelRect = {0,0,0,0}; - if (ctx->state == CS_Selecting) { - curLabelRect = DrawSelection(backDC, ctx->startX, ctx->startY, ctx->endX, ctx->endY, - ctx->virtualX, ctx->virtualY, ctx->virtualW, ctx->virtualH, ctx->gdi); - } else if (ctx->state == CS_Idle) { - RECT screenRect; - int ww, wh; - - if (ctx->hoveredWindow >= 0 && ctx->hoveredWindow < (int)ctx->windows.size()) { - // 显示悬停窗口的尺寸 - const RECT& wr = ctx->windows[ctx->hoveredWindow].rect; - ww = wr.right - wr.left; - wh = wr.bottom - wr.top; - screenRect = wr; - } else { - // 没有匹配到窗口时，显示当前屏幕的尺寸 - POINT pt = { ctx->mouseX, ctx->mouseY }; - HMONITOR hMonitor = MonitorFromPoint(pt, MONITOR_DEFAULTTONEAREST); - if (hMonitor) { - MONITORINFO monitorInfo; - monitorInfo.cbSize = sizeof(MONITORINFO); - if (GetMonitorInfo(hMonitor, &monitorInfo)) { - screenRect = monitorInfo.rcMonitor; - ww = screenRect.right - screenRect.left; - wh = screenRect.bottom - screenRect.top; - } - } - } - - curLabelRect = DrawSizeLabel(backDC, ww, wh, - screenRect.left - ctx->virtualX, screenRect.top - ctx->virtualY, - screenRect.right - ctx->virtualX, screenRect.bottom - ctx->virtualY, - ctx->virtualW, ctx->virtualH, ctx->gdi); - } - - // 绘制放大镜信息面板 - DrawInfoPanel(backDC, panelXRel, panelYRel, ctx->currentColor, - ctx->memDC, ctx->virtualX, ctx->virtualY, - ctx->mouseX, ctx->mouseY, ctx->dpiScale, ctx->gdi); - - // 更新脏区域追踪 - ctx->lastPanelRect = curPanelRect; - ctx->lastSelectionRect = curSelRect; - ctx->lastLabelRect = curLabelRect; - ctx->lastHighlightRect = curHlRect; - - // 后台缓冲 -> 窗口 - BitBlt(hdc, 0, 0, ctx->virtualW, ctx->virtualH, backDC, 0, 0, SRCCOPY); - EndPaint(hwnd, &ps); - return 0; - } - - case WM_LBUTTONDOWN: { - if (ctx->state == CS_Idle) { - ctx->startX = ctx->mouseX; - ctx->startY = ctx->mouseY; - ctx->endX = ctx->mouseX; - ctx->endY = ctx->mouseY; - ctx->state = CS_Selecting; - ctx->needFullRedraw = true; - } - return 0; - } - - case WM_MOUSEMOVE: { - POINT pt; - GetCursorPos(&pt); - if (pt.x != ctx->mouseX || pt.y != ctx->mouseY) { - ctx->mouseX = pt.x; - ctx->mouseY = pt.y; - ctx->currentColor = GetPixelColorFromBitmap(ctx->memDC, - ctx->mouseX, ctx->mouseY, ctx->virtualX, ctx->virtualY, ctx->dpiScale); - - if (ctx->state == CS_Selecting) { - ctx->endX = ctx->mouseX; - ctx->endY = ctx->mouseY; - } else if (ctx->state == CS_Idle) { - int newHovered = FindWindowAtPoint(ctx->windows, ctx->mouseX, ctx->mouseY); - if (newHovered != ctx->hoveredWindow) { - ctx->hoveredWindow = newHovered; - } - } - InvalidateRect(hwnd, NULL, FALSE); - } - return 0; - } - - case WM_LBUTTONUP: { - if (ctx->state == CS_Selecting) { - int w = abs(ctx->endX - ctx->startX); - int h = abs(ctx->endY - ctx->startY); - - RECT finalRect; - if (w <= 1 && h <= 1) { - // 点击 -> 使用悬停窗口矩形 - int idx = FindWindowAtPoint(ctx->windows, ctx->mouseX, ctx->mouseY); - if (idx >= 0) { - finalRect = ctx->windows[idx].rect; - } else { - // 匹配不到窗口时，默认选区为鼠标所在的屏幕 - POINT pt = { ctx->mouseX, ctx->mouseY }; - HMONITOR hMonitor = MonitorFromPoint(pt, MONITOR_DEFAULTTONEAREST); - if (hMonitor) { - MONITORINFO monitorInfo; - monitorInfo.cbSize = sizeof(MONITORINFO); - if (GetMonitorInfo(hMonitor, &monitorInfo)) { - finalRect = monitorInfo.rcMonitor; - } else { - // 获取失败，降级为虚拟屏幕 - finalRect = { ctx->virtualX, ctx->virtualY, - ctx->virtualX + ctx->virtualW, ctx->virtualY + ctx->virtualH }; - } - } else { - // 获取显示器失败，降级为虚拟屏幕 - finalRect = { ctx->virtualX, ctx->virtualY, - ctx->virtualX + ctx->virtualW, ctx->virtualY + ctx->virtualH }; - } - } - } else { - finalRect.left = (std::min)(ctx->startX, ctx->endX); - finalRect.top = (std::min)(ctx->startY, ctx->endY); - finalRect.right = (std::max)(ctx->startX, ctx->endX); - finalRect.bottom = (std::max)(ctx->startY, ctx->endY); - } - - // 提取结果 - ScreenshotResult* result = ExtractRegionResult(ctx->memDC, finalRect, - ctx->virtualX, ctx->virtualY, ctx->dpiScale); - - if (g_screenshotTsfn != nullptr) { - napi_call_threadsafe_function(g_screenshotTsfn, result, napi_tsfn_nonblocking); - } - ctx->state = CS_Done; - DestroyWindow(hwnd); - } - return 0; - } - - case WM_RBUTTONDOWN: { - ctx->state = CS_Cancelled; - // 回调失败结果 - if (g_screenshotTsfn != nullptr) { - ScreenshotResult* result = new ScreenshotResult(); - result->success = false; - result->x = 0; result->y = 0; result->x2 = 0; result->y2 = 0; - result->width = 0; result->height = 0; - napi_call_threadsafe_function(g_screenshotTsfn, result, napi_tsfn_nonblocking); - } - DestroyWindow(hwnd); - return 0; - } - - case WM_KEYDOWN: { - if (wParam == VK_ESCAPE) { - ctx->state = CS_Cancelled; - if (g_screenshotTsfn != nullptr) { - ScreenshotResult* result = new ScreenshotResult(); - result->success = false; - result->x = 0; result->y = 0; result->x2 = 0; result->y2 = 0; - result->width = 0; result->height = 0; - napi_call_threadsafe_function(g_screenshotTsfn, result, napi_tsfn_nonblocking); - } - DestroyWindow(hwnd); - } - return 0; - } - - case WM_DESTROY: { - g_screenshotOverlayWindow = NULL; - PostQuitMessage(0); - return 0; - } - } - - return DefWindowProc(hwnd, msg, wParam, lParam); -} - -// 截图线程（预截屏 + 双缓冲架构） -static void ScreenshotCaptureThread() { - // 设置 DPI 感知 - typedef DPI_AWARENESS_CONTEXT (WINAPI *SetThreadDpiAwarenessContextProc)(DPI_AWARENESS_CONTEXT); - HMODULE user32 = GetModuleHandleW(L"user32.dll"); - if (user32) { - auto setDpiProc = (SetThreadDpiAwarenessContextProc)GetProcAddress(user32, "SetThreadDpiAwarenessContext"); - if (setDpiProc) { - setDpiProc(DPI_AWARENESS_CONTEXT_PER_MONITOR_AWARE_V2); - } - } - - double dpiScale = GetDpiScaleFactor(); - - // 预截屏整个虚拟屏幕 - HDC memDC = NULL; - HBITMAP screenBitmap = NULL; - int vx, vy, vw, vh; - if (!CaptureVirtualScreen(memDC, screenBitmap, vx, vy, vw, vh, dpiScale)) { - g_isCapturing = false; - return; - } - - // 创建双缓冲 - HDC backDC = NULL; - HBITMAP backBmp = NULL; - if (!CreateBackBuffer(backDC, backBmp, vw, vh)) { - DeleteDC(memDC); - DeleteObject(screenBitmap); - g_isCapturing = false; - return; - } - - // 枚举窗口 - std::vector windows = EnumWindowsForCapture(); - - // 初始化 GDI 资源 - SCGdiResources gdi; - gdi.Init(); - - // 初始化上下文 - CaptureContext ctx = {}; - ctx.state = CS_Idle; - ctx.virtualX = vx; ctx.virtualY = vy; - ctx.virtualW = vw; ctx.virtualH = vh; - ctx.startX = 0; ctx.startY = 0; - ctx.endX = 0; ctx.endY = 0; - ctx.hoveredWindow = -1; - ctx.screenBitmap = screenBitmap; - ctx.memDC = memDC; - ctx.backDC = backDC; - ctx.backBitmap = backBmp; - ctx.lastPanelRect = {0,0,0,0}; - ctx.lastSelectionRect = {0,0,0,0}; - ctx.lastLabelRect = {0,0,0,0}; - ctx.lastHighlightRect = {0,0,0,0}; - ctx.needFullRedraw = true; - ctx.dpiScale = dpiScale; - ctx.gdi = gdi; - ctx.windows = std::move(windows); - - // 获取初始鼠标位置和颜色 - POINT pt; - GetCursorPos(&pt); - ctx.mouseX = pt.x; - ctx.mouseY = pt.y; - ctx.currentColor = GetPixelColorFromBitmap(memDC, pt.x, pt.y, vx, vy, dpiScale); - - g_captureCtx = &ctx; - - // 注册窗口类 - WNDCLASSEXW wc = {0}; - wc.cbSize = sizeof(WNDCLASSEXW); - wc.lpfnWndProc = ScreenshotOverlayWndProc; - wc.hInstance = GetModuleHandle(NULL); - wc.hCursor = LoadCursor(NULL, IDC_ARROW); // 默认鼠标样式 - wc.lpszClassName = L"ZToolsScreenshotOverlay"; - - if (!RegisterClassExW(&wc)) { - gdi.Cleanup(); - DeleteDC(backDC); DeleteObject(backBmp); - DeleteDC(memDC); DeleteObject(screenBitmap); - g_captureCtx = nullptr; - g_isCapturing = false; - return; - } - - // 创建普通 WS_POPUP 窗口（非分层窗口） - g_screenshotOverlayWindow = CreateWindowExW( - WS_EX_TOPMOST | WS_EX_TOOLWINDOW, - L"ZToolsScreenshotOverlay", - L"Screenshot Overlay", - WS_POPUP, - vx, vy, vw, vh, - NULL, NULL, GetModuleHandle(NULL), NULL - ); - - if (g_screenshotOverlayWindow == NULL) { - UnregisterClassW(L"ZToolsScreenshotOverlay", GetModuleHandle(NULL)); - gdi.Cleanup(); - DeleteDC(backDC); DeleteObject(backBmp); - DeleteDC(memDC); DeleteObject(screenBitmap); - g_captureCtx = nullptr; - g_isCapturing = false; - return; - } - - ShowWindow(g_screenshotOverlayWindow, SW_SHOW); - SetForegroundWindow(g_screenshotOverlayWindow); - - // 消息循环 - MSG msg; - while (true) { - if (ctx.state == CS_Done || ctx.state == CS_Cancelled) break; - - if (PeekMessage(&msg, NULL, 0, 0, PM_REMOVE)) { - if (msg.message == WM_QUIT) break; - TranslateMessage(&msg); - DispatchMessage(&msg); - } else { - // 检查 ESC 键（窗口可能没有焦点） - if (GetAsyncKeyState(VK_ESCAPE) & 0x8000) { - if (ctx.state != CS_Done && ctx.state != CS_Cancelled) { - ctx.state = CS_Cancelled; - if (g_screenshotTsfn != nullptr) { - ScreenshotResult* result = new ScreenshotResult(); - result->success = false; - result->x = 0; result->y = 0; result->x2 = 0; result->y2 = 0; - result->width = 0; result->height = 0; - napi_call_threadsafe_function(g_screenshotTsfn, result, napi_tsfn_nonblocking); - } - DestroyWindow(g_screenshotOverlayWindow); - break; - } - } - Sleep(1); - } - } - - // 清理 - g_captureCtx = nullptr; - gdi.Cleanup(); - DeleteDC(backDC); DeleteObject(backBmp); - DeleteDC(memDC); DeleteObject(screenBitmap); - UnregisterClassW(L"ZToolsScreenshotOverlay", GetModuleHandle(NULL)); - g_isCapturing = false; -} - -// 启动区域截图 -Napi::Value StartRegionCapture(const Napi::CallbackInfo& info) { - Napi::Env env = info.Env(); - - if (g_isCapturing) { - Napi::Error::New(env, "Screenshot already in progress").ThrowAsJavaScriptException(); - return env.Undefined(); - } - - // 可选的回调函数 - if (info.Length() > 0 && info[0].IsFunction()) { - Napi::Function callback = info[0].As(); - napi_value resource_name; - napi_create_string_utf8(env, "ScreenshotCallback", NAPI_AUTO_LENGTH, &resource_name); - - napi_status status = napi_create_threadsafe_function( - env, callback, nullptr, resource_name, - 0, 1, nullptr, nullptr, nullptr, - CallScreenshotJs, &g_screenshotTsfn - ); - - if (status != napi_ok) { - Napi::Error::New(env, "Failed to create threadsafe function").ThrowAsJavaScriptException(); - return env.Undefined(); - } - } - - g_isCapturing = true; - - g_screenshotThread = std::thread(ScreenshotCaptureThread); - g_screenshotThread.detach(); - - return env.Undefined(); -} - // ==================== 剪贴板文件功能 ==================== // 获取剪贴板中的文件列表 @@ -4029,7 +2871,7 @@ static std::unique_ptr CreateBitmapFromIcon( } // 获取 PNG 编码器 CLSID -static int GetPngEncoderClsid(CLSID* pClsid) { +int GetPngEncoderClsid(CLSID* pClsid) { UINT num = 0u; UINT size = 0u; Gdiplus::GetImageEncodersSize(std::addressof(num), std::addressof(size)); diff --git a/src/screenshot_windows.cpp b/src/screenshot_windows.cpp new file mode 100644 index 0000000..5c8ead2 --- /dev/null +++ b/src/screenshot_windows.cpp @@ -0,0 +1,4468 @@ +#include +#include +#include +#include // For GET_X_LPARAM, GET_Y_LPARAM +#include +#include // For IME (Input Method Editor) support +#include // For GetSaveFileNameW（保存对话框） +#include // For SHGetKnownFolderPath（已知文件夹路径，如图片库） +#include +#include +#include // For std::min, std::max +#include +#include +#include // For std::sqrt, std::fabs + +// DWMWA_CLOAKED 在较新的 Windows SDK 中定义，为了兼容性手动定义 +#ifndef DWMWA_CLOAKED +#define DWMWA_CLOAKED 14 +#endif + +// GDI+ 需要 min/max +namespace Gdiplus { + using std::min; + using std::max; +} +#include + +// AlphaBlend 需要 msimg32 +#pragma comment(lib, "msimg32.lib") + +#include "screenshot_windows.h" + +// ---- nanosvg：SVG 光栅化（单文件库，宏实例化）---- +// 两个 .h 必须在同一编译单元用宏实例化一次；这里在 screenshot_windows.cpp 内实例化。 +#define NANOSVG_IMPLEMENTATION +#include "third_party/nanosvg.h" +#define NANOSVGRAST_IMPLEMENTATION +#include "third_party/nanosvgrast.h" + +// ---- 截图工具栏图标 SVG 文本（构建期由 scripts/gen-icons.js 从 src/assets 生成）---- +#include "generated/icon_svgs.h" + +// 全局变量 - 区域截图 +static HWND g_screenshotOverlayWindow = NULL; +static std::atomic g_isCapturing(false); +static napi_threadsafe_function g_screenshotTsfn = nullptr; +static std::thread g_screenshotThread; + +// ==================== 区域截图功能（预截屏 + 双缓冲架构） ==================== + +// 截图常量 +static const int SC_PANEL_WIDTH = 140; +static const int SC_PANEL_HEIGHT = 140; +static const int SC_MAGNIFIER_HEIGHT = 74; +static const int SC_PANEL_MARGIN = 15; +static const int SC_PANEL_CORNER_RADIUS = 8; +static const int SC_ZOOM_FACTOR = 4; + +// 选区外遮罩：微信风格，选区内部保持清晰，外部覆盖半透明黑色 +// 取值 0~255，数值越大越暗（0 = 无遮罩，255 = 全黑） +static const BYTE SC_MASK_ALPHA = 120; + +// 截图状态枚举 +enum CaptureState { + CS_Idle, // 等待选择（hover 窗口/拖拽开始） + CS_Selecting, // 正在拖拽框选 + CS_Confirmed, // 已确认选区，可调整/拖动/打开工具栏 + CS_Resizing, // 正在拖拽手柄调整选区 + CS_Moving, // 正在整体拖动选区 + CS_Drawing, // 正在绘制标注（矩形/圆/箭头/画笔） + CS_TextEditing, // 正在输入文字 + CS_Done, + CS_Cancelled +}; + +// 选区调整手柄（8 个方向） +enum ResizeHandle { + RH_None = -1, + RH_Left = 0, + RH_Right = 1, + RH_Top = 2, + RH_Bottom = 3, + RH_TopLeft = 4, + RH_TopRight = 5, + RH_BottomLeft = 6, + RH_BottomRight = 7 +}; + +// 工具栏按钮 +enum ToolButton { + TB_Rect = 0, // 矩形 + TB_Circle, // 圆形（含椭圆） + TB_Arrow, // 箭头 + TB_Brush, // 画笔 + TB_Mosaic, // 马赛克 + TB_Text, // 文字 + TB_Translate, // 翻译 + TB_Separator1, // 分隔线 + TB_Undo, // 撤销 + TB_Redo, // 重做 + TB_Separator2, // 分隔线 + TB_Save, // 保存到本地 + TB_Cancel, // 取消 + TB_Confirm, // 确定 + TB_Count +}; + +// ==================== 标注绘制（矩形/圆/箭头/画笔） ==================== +// 所有标注统一用「选区相对逻辑坐标」存储（相对 selection.left/top 的偏移）： +// - 实时渲染时：backDC 局部坐标 = curSelRect.left + relX +// - 合成进 PNG 时：finalDC 原点正好是选区原点，标注直接画在 (relX, relY) +// - 移动选区时：relX/relY 不变，标注自动跟随，无需额外换算 +enum AnnotationType { + AT_Rect, + AT_Circle, + AT_Arrow, + AT_Brush, + AT_Text, + AT_Mosaic // 马赛克（框选区域 或 鼠标涂抹） +}; + +struct Annotation { + AnnotationType type; + COLORREF color; + int thickness; // 逻辑像素（矢量=线宽；文字=字号） + // 绝对虚拟屏幕坐标（与 ctx->mouseX/selection 同坐标系）。 + // 用绝对坐标而非选区相对，保证选区缩放/移动时标注位置固定不动。 + int x1, y1, x2, y2; // Rect / Circle / Arrow 的起止（绝对坐标）；AT_Text 的 x1/y1 为文字锚点； + // AT_Mosaic 框选模式的矩形起止（绝对坐标） + std::vector pts; // Brush 自由路径（绝对坐标）；AT_Mosaic 涂抹模式的路径（绝对坐标） + std::wstring text; // AT_Text 的文字内容 + // ---- AT_Mosaic 专用 ---- + bool mosaicRect; // true=框选区域马赛克；false=鼠标涂抹马赛克 + int mosaicSize; // 马赛克块大小（逻辑像素） + int brushRadius; // 涂抹半径（逻辑像素，仅涂抹模式有效） +}; + +// 粗细预设（逻辑像素，实际绘制粗细，渲染时乘 dpiScale） +static const int SC_THICK_PRESETS[] = { 1, 2, 4 }; +static const int SC_THICK_COUNT = sizeof(SC_THICK_PRESETS) / sizeof(SC_THICK_PRESETS[0]); +static const int SC_DEFAULT_THICK_IDX = 1; // 默认中粗 +// 子菜单圆点预览直径（逻辑像素，仅用于界面显示，与实际绘制粗细解耦） +static const int SC_THICK_DOT_SIZES[] = { 5, 10, 16 }; +static const int SC_THICK_DOT_COUNT = sizeof(SC_THICK_DOT_SIZES) / sizeof(SC_THICK_DOT_SIZES[0]); + +// 文字字号预设（逻辑像素），文字工具激活时子菜单第一组显示 +static const int SC_FONT_SIZES[] = { 16, 24, 36 }; +static const int SC_FONT_COUNT = sizeof(SC_FONT_SIZES) / sizeof(SC_FONT_SIZES[0]); +static const int SC_DEFAULT_FONT_IDX = 1; // 默认中号 +static const wchar_t* SC_FONT_FACE = L"微软雅黑"; + +// 马赛克块大小预设（逻辑像素），马赛克工具子菜单显示 +static const int SC_MOSAIC_SIZES[] = { 6, 10, 16 }; +static const int SC_MOSAIC_COUNT = sizeof(SC_MOSAIC_SIZES) / sizeof(SC_MOSAIC_SIZES[0]); +static const int SC_DEFAULT_MOSAIC_IDX = 1; // 默认中等块 +// 涂抹半径预设（逻辑像素），马赛克涂抹模式使用，与画笔粗细预设共用同一组子菜单第二组无效， +// 这里单独定义便于扩展。半径越大涂抹范围越宽。 +static const int SC_MOSAIC_RADIUS[] = { 12, 22, 36 }; +static const int SC_MOSAIC_RADIUS_COUNT = sizeof(SC_MOSAIC_RADIUS) / sizeof(SC_MOSAIC_RADIUS[0]); +static const int SC_DEFAULT_MOSAIC_RADIUS_IDX = 1; // 默认中等半径 + +// 颜色预设 +static const COLORREF SC_COLOR_PRESETS[] = { + RGB(0xE5, 0x39, 0x35), // 红 + RGB(0xFB, 0x8C, 0x00), // 橙 + RGB(0xFD, 0xD8, 0x35), // 黄 + RGB(0x43, 0xA0, 0x47), // 绿 + RGB(0x00, 0xAC, 0xC1), // 青 + RGB(0x1E, 0x88, 0xE5), // 蓝 + RGB(0xFF, 0xFF, 0xFF), // 白 + RGB(0x33, 0x33, 0x33), // 黑 +}; +static const int SC_COLOR_COUNT = sizeof(SC_COLOR_PRESETS) / sizeof(SC_COLOR_PRESETS[0]); +static const int SC_DEFAULT_COLOR_IDX = 0; // 默认红 + +// 判断某工具按钮是否为可绘制矢量工具 +static bool IsVectorTool(int btn) { + return btn == TB_Rect || btn == TB_Circle || btn == TB_Arrow || btn == TB_Brush; +} + +// ToolButton -> AnnotationType +static AnnotationType ToolToAnnotationType(int btn) { + switch (btn) { + case TB_Rect: return AT_Rect; + case TB_Circle: return AT_Circle; + case TB_Arrow: return AT_Arrow; + case TB_Brush: return AT_Brush; + default: return AT_Rect; + } +} + +// 手柄/工具栏几何常量 +static const int SC_HANDLE_SIZE = 8; // 调整手柄边长 +static const int SC_TOOLBAR_BTN = 32; // 按钮尺寸（正方形） +static const int SC_TOOLBAR_PAD = 6; // 按钮↔工具栏边缘内边距（四边一致） +static const int SC_TOOLBAR_H = SC_TOOLBAR_BTN + SC_TOOLBAR_PAD * 2; // 工具栏高度 = 按钮 + 上下内边距 +static const int SC_TOOLBAR_GAP = 1; // 按钮间距 +static const int SC_TOOLBAR_RADIUS = 8; // 工具栏圆角 +static const int SC_TOOLBAR_MARGIN = 6; // 选区到工具栏间距 +static const int SC_TOOLBAR_BORDER = 1; // 工具栏边框 +static const int SC_MIN_SELECTION = 10; // 最小选区尺寸 + +// 子菜单几何常量（100% DPI 基准值，运行时按 dpiScale 缩放） +// 单行布局：[粗细圆点×3] | [分隔线] | [颜色圆点×8]，无文案。 +// 单元格（点击区 + 选中背景区）大小与工具栏按钮一致，便于视觉对齐。 +static const int SC_POPUP_CELL = SC_TOOLBAR_BTN; // 单元格尺寸（= 工具栏按钮大小） +static const int SC_POPUP_PAD = 4; // 内边距 +static const int SC_POPUP_RADIUS = 8; // 圆角 +static const int SC_POPUP_COLOR_DOT = 18; // 颜色圆点直径（图标本身） +static const int SC_POPUP_SEP_GAP = 6; // 分隔线两侧间距 +static const int SC_POPUP_SEP_H = 20; // 分隔线高度 +static const int SC_POPUP_BORDER = 1; // 边框 +static const int SC_POPUP_MARGIN = 4; // 工具栏与子菜单间距 + +// 子菜单几何（DPI 缩放后） +struct SCPopupMetrics { + int pad; + int radius; + int cell; // 单元格尺寸（点击区 + 选中背景区，= 工具栏按钮大小） + int colorDot; // 颜色圆点直径（图标本身） + int sepGap; // 分隔线两侧间距 + int sepH; // 分隔线高度 + int border; + int margin; +}; + +// 窗口信息 +struct SCWindowInfo { + HWND hwnd; + RECT rect; + std::wstring title; +}; + +// 截图结果结构 +struct ScreenshotResult { + bool success; + int x; + int y; + int x2; + int y2; + int width; + int height; + std::string base64; +}; + +// GDI 资源缓存 +struct SCGdiResources { + HBRUSH bgBrush; + HPEN borderPen; + HPEN crosshairPen; + HPEN selectionPen; + HPEN highlightPen; + HFONT smallFont; + // 选区外遮罩缓冲（虚拟屏幕大小，纯黑 + 常量 alpha），用于 AlphaBlend + HDC maskDC; + HBITMAP maskBitmap; + + void Init() { + bgBrush = CreateSolidBrush(RGB(52, 52, 53)); + borderPen = CreatePen(PS_SOLID, 0, RGB(102, 102, 102)); + crosshairPen = CreatePen(PS_SOLID, 1, RGB(0, 136, 255)); + selectionPen = CreatePen(PS_SOLID, 1, RGB(0, 136, 255)); + highlightPen = CreatePen(PS_SOLID, 3, RGB(0, 136, 255)); + // 创建字体 + LOGFONTW lf = {}; + lf.lfHeight = -12; + lf.lfCharSet = DEFAULT_CHARSET; + wcscpy_s(lf.lfFaceName, L"微软雅黑"); + smallFont = CreateFontIndirectW(&lf); + maskDC = NULL; + maskBitmap = NULL; + } + + // 创建遮罩缓冲（纯黑位图，配合常量 alpha 实现 40%+ 半透明遮罩） + // 须在 CaptureContext 虚拟屏幕尺寸确定后调用 + void InitMask(int virtualW, int virtualH) { + HDC screenDC = GetDC(NULL); + if (!screenDC) return; + maskDC = CreateCompatibleDC(screenDC); + if (maskDC) { + maskBitmap = CreateCompatibleBitmap(screenDC, virtualW, virtualH); + if (maskBitmap) { + SelectObject(maskDC, maskBitmap); + // 填充纯黑（AlphaBlend 用常量 alpha，源颜色为黑） + RECT rc = { 0, 0, virtualW, virtualH }; + HBRUSH black = CreateSolidBrush(RGB(0, 0, 0)); + FillRect(maskDC, &rc, black); + DeleteObject(black); + } else { + DeleteDC(maskDC); + maskDC = NULL; + } + } + ReleaseDC(NULL, screenDC); + } + + void Cleanup() { + if (bgBrush) { DeleteObject(bgBrush); bgBrush = NULL; } + if (borderPen) { DeleteObject(borderPen); borderPen = NULL; } + if (crosshairPen) { DeleteObject(crosshairPen); crosshairPen = NULL; } + if (selectionPen) { DeleteObject(selectionPen); selectionPen = NULL; } + if (highlightPen) { DeleteObject(highlightPen); highlightPen = NULL; } + if (smallFont) { DeleteObject(smallFont); smallFont = NULL; } + if (maskBitmap) { DeleteObject(maskBitmap); maskBitmap = NULL; } + if (maskDC) { DeleteDC(maskDC); maskDC = NULL; } + } +}; + +// ---- 工具栏 DPI 缩放几何 ---- +// 基础逻辑尺寸（100% DPI）按 dpiScale 放大，保证 1080p → 4K 下工具栏尺寸与图标同步。 +// 基础值与原 SC_TOOLBAR_* 常量保持一致，便于回归。 +struct SCToolbarMetrics { + int btn; // 按钮边长 + int h; // 工具栏高度 + int gap; // 按钮间距 + int pad; // 按钮↔工具栏边缘内边距（四边一致） + int radius; // 圆角半径 + int margin; // 选区到工具栏间距 + int border; // 工具栏边框宽度 + int iconSize; // 图标光栅化尺寸（物理像素） +}; + +static SCToolbarMetrics CalcToolbarMetrics(double dpiScale) { + auto scale = [&](int v) { return (int)(v * dpiScale + 0.5); }; + SCToolbarMetrics m; + m.btn = scale(SC_TOOLBAR_BTN); + m.h = scale(SC_TOOLBAR_H); + m.gap = scale(SC_TOOLBAR_GAP); + m.pad = scale(SC_TOOLBAR_PAD); + m.radius = scale(SC_TOOLBAR_RADIUS); + m.margin = scale(SC_TOOLBAR_MARGIN); + m.border = scale(SC_TOOLBAR_BORDER); + // 图标视觉内容约占按钮 ~72%，留出内边距；额外 +2px 余量提升抗锯齿质量 + m.iconSize = scale(SC_TOOLBAR_BTN - 8) + 2; + return m; +} + +// ---- SVG 图标光栅化 + 缓存 ---- +// 将单个 SVG 文本光栅化为 32bpp 预乘 ARGB HBITMAP，尺寸 px×px。 +// 把 SVG 中的 currentColor 替换为目标 color（normal/active 两色复用同一文本）。 +static HBITMAP RenderSvgToBitmap(const char* svgText, COLORREF color, int px) { + if (!svgText || px <= 0) return NULL; + + // 1. 替换 currentColor -> #RRGGBB（nanosvg 解析会改写 buffer，需可写副本） + char colorHex[8]; + sprintf_s(colorHex, sizeof(colorHex), "#%02X%02X%02X", + color & 0xFF, (color >> 8) & 0xFF, (color >> 16) & 0xFF); + std::string svg(svgText); + const std::string token = "currentColor"; + size_t pos = 0; + while ((pos = svg.find(token, pos)) != std::string::npos) { + svg.replace(pos, token.size(), colorHex); + pos += 6; + } + + // 2. 解析（nsvgParse 会就地修改传入字符串） + NSVGimage* image = nsvgParse(&svg[0], "px", 96.0f); + if (!image) return NULL; + + // 3. 光栅化到 RGBA 缓冲（非预乘） + std::vector rgba(px * px * 4, 0); + NSVGrasterizer* rast = nsvgCreateRasterizer(); + if (!rast) { nsvgDelete(image); return NULL; } + // 内容缩放到 72% 并居中，四周各留 14% 内边距，避免图标顶满按钮。 + // nanosvg 解析后 shape 已在 image->width 坐标系（viewBox 已折算）， + // 故 scale = px * 0.72 / image->width，偏移 tx = ty = px * 0.14。 + const float contentScale = 0.72f; + const float pad = (1.0f - contentScale) * 0.5f; + float refSize = (image->width > 0) ? image->width + : (image->height > 0) ? image->height : 24.0f; + float scale = (float)px * contentScale / refSize; + float tx = px * pad; + float ty = px * pad; + nsvgRasterize(rast, image, tx, ty, scale, rgba.data(), px, px, px * 4); + nsvgDeleteRasterizer(rast); + nsvgDelete(image); + + // 4. RGBA -> 预乘 BGRA（用于 AlphaBlend，避免黑边） + // nanosvg 输出 RGBA 字节序 [R,G,B,A]；而 32bpp BI_RGB DIB 内存布局为 + // BGRA（像素值 0xAARRGGBB 在小端内存里是 B,G,R,A）。故预乘时需把 R、B + // 对调写入，否则 memcpy 后通道会反，蓝色被画成黄色（灰度色看不出来）。 + for (int i = 0; i < px * px; i++) { + unsigned char r = rgba[i * 4 + 0]; + unsigned char g = rgba[i * 4 + 1]; + unsigned char b = rgba[i * 4 + 2]; + unsigned char a = rgba[i * 4 + 3]; + // 预乘后按 DIB 的 BGRA 字节序写入 + rgba[i * 4 + 0] = (unsigned char)((b * a + 127) / 255); // B + rgba[i * 4 + 1] = (unsigned char)((g * a + 127) / 255); // G + rgba[i * 4 + 2] = (unsigned char)((r * a + 127) / 255); // R + rgba[i * 4 + 3] = a; // A + } + + // 5. 创建 32bpp ARGB HBITMAP + HDC screenDC = GetDC(NULL); + if (!screenDC) return NULL; + HDC memDC = CreateCompatibleDC(screenDC); + if (!memDC) { ReleaseDC(NULL, screenDC); return NULL; } + + BITMAPINFO bmi = {}; + bmi.bmiHeader.biSize = sizeof(BITMAPINFOHEADER); + bmi.bmiHeader.biWidth = px; + bmi.bmiHeader.biHeight = -px; // 自上而下，避免垂直翻转 + bmi.bmiHeader.biPlanes = 1; + bmi.bmiHeader.biBitCount = 32; + bmi.bmiHeader.biCompression = BI_RGB; + + void* bits = nullptr; + HBITMAP bmp = CreateDIBSection(memDC, &bmi, DIB_RGB_COLORS, &bits, NULL, 0); + if (bmp && bits) { + memcpy(bits, rgba.data(), px * px * 4); + } + DeleteDC(memDC); + ReleaseDC(NULL, screenDC); + return bmp; +} + +// 工具栏图标位图缓存：按当前 DPI 渲染一次，dark/white 两色版本。 +// dark = normal/hover 图标色，white = active（蓝底）图标色。 +struct SCIconCache { + bool inited; + int iconSize; + HBITMAP dark[TB_Count]; // 普通态：深灰图标 + HBITMAP active[TB_Count]; // 选中态：主题蓝图标（搭配浅蓝高亮底） + + SCIconCache() : inited(false), iconSize(0) { + for (int i = 0; i < TB_Count; i++) { dark[i] = NULL; active[i] = NULL; } + } + + void Init(int physicalIconSize) { + if (inited) Cleanup(); + iconSize = physicalIconSize; + COLORREF darkColor = RGB(60, 60, 60); + // 选中态图标用主题蓝 #3B8BF2，与浅蓝高亮底搭配 + COLORREF activeColor = RGB(0x3B, 0x8B, 0xF2); + for (int i = 0; i < TB_Count; i++) { + // 分隔线（TB_Separator1/2）及未映射的项：kIconSvgs[i] 为 nullptr，跳过 + if (kIconSvgs[i]) { + dark[i] = RenderSvgToBitmap(kIconSvgs[i], darkColor, iconSize); + active[i] = RenderSvgToBitmap(kIconSvgs[i], activeColor, iconSize); + } + } + inited = true; + } + + // 取按钮位图：isActive 时用主题蓝版本，其余用深灰 + HBITMAP Get(int btn, bool isActive) const { + if (btn < 0 || btn >= TB_Count) return NULL; + return isActive ? active[btn] : dark[btn]; + } + + void Cleanup() { + for (int i = 0; i < TB_Count; i++) { + if (dark[i]) { DeleteObject(dark[i]); dark[i] = NULL; } + if (active[i]) { DeleteObject(active[i]); active[i] = NULL; } + } + inited = false; + } +}; + +// 计算工具栏宽度（所有按钮 + 间距 + 左右内边距 + 边框），按 metrics 缩放 +static int CalcToolbarWidth(const SCToolbarMetrics& m) { + return TB_Count * (m.btn + m.gap) - m.gap + m.pad * 2 + m.border * 2; +} + +// 前向声明：PointInRect / AddRoundedRect 定义在本文件靠后位置，子菜单命中/绘制需要提前使用。 +static bool PointInRect(int x, int y, const RECT& r); +static void AddRoundedRect(Gdiplus::GraphicsPath& outPath, int x, int y, int w, int h, int radius); + +// ==================== 粗细/颜色子菜单 ==================== + +// 按 DPI 计算子菜单几何（单行布局）。 +// 布局：[粗细圆点×3] sepGap | 分隔线 | sepGap [颜色圆点×8] +// 单元格尺寸与工具栏按钮一致（= SC_TOOLBAR_BTN），单元格间距 = SC_TOOLBAR_GAP，视觉对齐。 +// 粗细圆点本身大小随预设值变化，颜色圆点固定直径，均居中在单元格内。 +static SCPopupMetrics CalcPopupMetrics(double dpiScale) { + auto scale = [&](int v) { return (int)(v * dpiScale + 0.5); }; + SCPopupMetrics m; + m.pad = scale(SC_POPUP_PAD); + m.radius = scale(SC_POPUP_RADIUS); + m.cell = scale(SC_POPUP_CELL); + m.colorDot = scale(SC_POPUP_COLOR_DOT); + m.sepGap = scale(SC_POPUP_SEP_GAP); + m.sepH = scale(SC_POPUP_SEP_H); + m.border = scale(SC_POPUP_BORDER); + m.margin = scale(SC_POPUP_MARGIN); + return m; +} + +// 子菜单总宽/高（单行）。单元格间距沿用工具栏按钮间距 SC_TOOLBAR_GAP。 +static void CalcPopupSize(const SCPopupMetrics& m, int& outW, int& outH) { + int cellGap = (int)(SC_TOOLBAR_GAP * (m.cell / (double)SC_POPUP_CELL) + 0.5); + // 粗细组宽 + int thickW = SC_THICK_COUNT * m.cell + (SC_THICK_COUNT - 1) * cellGap; + // 颜色组宽 + int colorW = SC_COLOR_COUNT * m.cell + (SC_COLOR_COUNT - 1) * cellGap; + int contentW = thickW + m.sepGap * 2 + 1 + colorW; // 1 = 分隔线宽度 + outW = contentW + m.pad * 2 + m.border * 2; + outH = m.cell + m.pad * 2 + m.border * 2; +} + +// 计算子菜单位置（通用）：贴工具栏下方，放不下则上方，左右钳制到虚拟屏幕内。 +// toolbarRect / out 均为相对虚拟屏幕坐标；pw/ph 为子菜单尺寸。 +static void CalcPopupPlacement(const RECT& toolbarRect, + int virtualW, int virtualH, + const SCPopupMetrics& m, int pw, int ph, RECT& out) { + // 水平：与工具栏左对齐 + int x = toolbarRect.left; + // 垂直：优先工具栏下方 + int y = toolbarRect.bottom + m.margin; + if (y + ph > virtualH) { + y = toolbarRect.top - m.margin - ph; + } + // 左右钳制 + if (x + pw > virtualW) x = virtualW - pw - m.margin; + if (x < 0) x = m.margin; + + out.left = x; + out.top = y; + out.right = x + pw; + out.bottom = y + ph; +} + +// 计算子菜单位置：贴工具栏下方，放不下则上方，左右钳制到虚拟屏幕内。 +// toolbarRect / out 均为相对虚拟屏幕坐标（与工具栏一致）。 +static void CalcPopupPosition(const RECT& toolbarRect, + int virtualW, int virtualH, + const SCPopupMetrics& m, RECT& out) { + int pw, ph; + CalcPopupSize(m, pw, ph); + CalcPopupPlacement(toolbarRect, virtualW, virtualH, m, pw, ph, out); +} + +// 马赛克子菜单几何常量 +// 模式切换组：2 个单元格（涂抹 / 框选）；块大小组：3 个单元格。 +static const int SC_MOSAIC_MODE_COUNT = 2; // 涂抹、框选 +// 涂抹模式用半径圆点表示，块大小用马赛克方块网格表示（绘制时按预设值缩放） + +// 计算马赛克子菜单总宽/高（单行）。 +// 布局：[模式×2] sepGap | 分隔线 | sepGap [块大小×3] sepGap | 分隔线 | sepGap [涂抹半径×3] +static void CalcMosaicPopupSize(const SCPopupMetrics& m, int& outW, int& outH) { + int cellGap = (int)(SC_TOOLBAR_GAP * (m.cell / (double)SC_POPUP_CELL) + 0.5); + int modeW = SC_MOSAIC_MODE_COUNT * m.cell + (SC_MOSAIC_MODE_COUNT - 1) * cellGap; + int sizeW = SC_MOSAIC_COUNT * m.cell + (SC_MOSAIC_COUNT - 1) * cellGap; + int radiusW = SC_MOSAIC_RADIUS_COUNT * m.cell + (SC_MOSAIC_RADIUS_COUNT - 1) * cellGap; + // 两组分隔线，每组分隔线宽 = sepGap*2 + 1 + int contentW = modeW + (m.sepGap * 2 + 1) + sizeW + (m.sepGap * 2 + 1) + radiusW; + outW = contentW + m.pad * 2 + m.border * 2; + outH = m.cell + m.pad * 2 + m.border * 2; +} + +// 命中测试马赛克子菜单，返回值约定： +// +1 = 涂抹模式；+2 = 框选模式 +// +101.. = 第 N 个块大小（100 + sizeIdx + 1） +// +201.. = 第 N 个涂抹半径（200 + radiusIdx + 1） +// 0 = 未命中 +static int HitTestMosaicPopup(int x, int y, const RECT& popupRect, const SCPopupMetrics& m) { + if (!PointInRect(x, y, popupRect)) return 0; + int contentLeft = popupRect.left + m.border + m.pad; + int contentTop = popupRect.top + m.border + m.pad; + int cellGap = (int)(SC_TOOLBAR_GAP * (m.cell / (double)SC_POPUP_CELL) + 0.5); + if (y < contentTop || y >= contentTop + m.cell) return 0; + + // 模式组 + for (int i = 0; i < SC_MOSAIC_MODE_COUNT; i++) { + int ix = contentLeft + i * (m.cell + cellGap); + if (x >= ix && x < ix + m.cell) return i + 1; // +1=涂抹 +2=框选 + } + int modeEndX = contentLeft + SC_MOSAIC_MODE_COUNT * m.cell + + (SC_MOSAIC_MODE_COUNT - 1) * cellGap; + int sizeStartX = modeEndX + m.sepGap * 2 + 1; + if (x < sizeStartX) return 0; // 第一条分隔线区域 + // 块大小组 + int sizeEndX = sizeStartX + SC_MOSAIC_COUNT * m.cell + + (SC_MOSAIC_COUNT - 1) * cellGap; + for (int i = 0; i < SC_MOSAIC_COUNT; i++) { + int ix = sizeStartX + i * (m.cell + cellGap); + if (x >= ix && x < ix + m.cell) return 100 + i + 1; + } + // 涂抹半径组 + int radiusStartX = sizeEndX + m.sepGap * 2 + 1; + if (x < radiusStartX) return 0; // 第二条分隔线区域 + for (int i = 0; i < SC_MOSAIC_RADIUS_COUNT; i++) { + int ix = radiusStartX + i * (m.cell + cellGap); + if (x >= ix && x < ix + m.cell) return 200 + i + 1; + } + return 0; +} + +// 绘制马赛克子菜单（单行）。 +// modeIdx：当前模式（0=涂抹 1=框选）；sizeIdx：当前块大小索引；radiusIdx：涂抹半径索引。 +static void DrawMosaicPopup(HDC hdc, const RECT& popupRect, + int modeIdx, int sizeIdx, int radiusIdx, + const SCPopupMetrics& m) { + Gdiplus::GdiplusStartupInput startupInput; + ULONG_PTR gdipToken; + Gdiplus::GdiplusStartup(&gdipToken, &startupInput, NULL); + { + Gdiplus::Graphics graphics(hdc); + graphics.SetSmoothingMode(Gdiplus::SmoothingModeAntiAlias); + + int pw = popupRect.right - popupRect.left; + int ph = popupRect.bottom - popupRect.top; + + // 白色圆角背景 + 浅灰边框 + Gdiplus::SolidBrush whiteBrush(Gdiplus::Color(255, 255, 255, 255)); + Gdiplus::GraphicsPath bgPath; + AddRoundedRect(bgPath, popupRect.left, popupRect.top, pw, ph, m.radius); + graphics.FillPath(&whiteBrush, &bgPath); + Gdiplus::Pen borderPen(Gdiplus::Color(255, 210, 210, 210), (Gdiplus::REAL)m.border); + graphics.DrawPath(&borderPen, &bgPath); + + int contentLeft = popupRect.left + m.border + m.pad; + int contentTop = popupRect.top + m.border + m.pad; + int midY = contentTop + m.cell / 2; + int cellGap = (int)(SC_TOOLBAR_GAP * (m.cell / (double)SC_POPUP_CELL) + 0.5); + + auto drawCellBg = [&](int cellLeft) { + Gdiplus::GraphicsPath p; + AddRoundedRect(p, cellLeft, contentTop, m.cell, m.cell, m.cell / 4); + Gdiplus::SolidBrush bgBrush(Gdiplus::Color(255, 225, 237, 253)); + graphics.FillPath(&bgBrush, &p); + Gdiplus::Pen edgePen(Gdiplus::Color(90, 160, 160, 160), 1.0f); + graphics.DrawPath(&edgePen, &p); + }; + auto cellColor = [&](bool sel) -> Gdiplus::Color { + return sel ? Gdiplus::Color(255, 0x3B, 0x8B, 0xF2) + : Gdiplus::Color(255, 0x33, 0x33, 0x33); + }; + + // 模式组 + for (int i = 0; i < SC_MOSAIC_MODE_COUNT; i++) { + int cellLeft = contentLeft + i * (m.cell + cellGap); + bool sel = (i == modeIdx); + if (sel) drawCellBg(cellLeft); + int cx = cellLeft + m.cell / 2; + Gdiplus::Color c = cellColor(sel); + if (i == 0) { + // 涂抹模式：画一个画笔/毛刷图标（一条波浪线 + 圆头） + Gdiplus::Pen pen(c, (Gdiplus::REAL)2.0f); + pen.SetLineJoin(Gdiplus::LineJoinRound); + pen.SetStartCap(Gdiplus::LineCapRound); + pen.SetEndCap(Gdiplus::LineCapRound); + int r = (int)(m.cell * 0.22); + // 自由曲线（模拟涂抹轨迹） + Gdiplus::PointF curve[] = { + Gdiplus::PointF((float)(cx - m.cell * 0.28), (float)(midY + m.cell * 0.18)), + Gdiplus::PointF((float)(cx - m.cell * 0.10), (float)(midY - m.cell * 0.18)), + Gdiplus::PointF((float)(cx + m.cell * 0.10), (float)(midY + m.cell * 0.18)), + Gdiplus::PointF((float)(cx + m.cell * 0.28), (float)(midY - m.cell * 0.18)), + }; + graphics.DrawLines(&pen, curve, 4); + } else { + // 框选模式：画一个虚线矩形 + Gdiplus::Pen pen(c, (Gdiplus::REAL)2.0f); + int r = (int)(m.cell * 0.24); + graphics.DrawRectangle(&pen, cx - r, midY - r, r * 2, r * 2); + } + } + + // 分隔线 + int modeEndX = contentLeft + SC_MOSAIC_MODE_COUNT * m.cell + + (SC_MOSAIC_MODE_COUNT - 1) * cellGap; + int sepX = modeEndX + m.sepGap; + Gdiplus::Pen sepPen(Gdiplus::Color(255, 220, 220, 220), 1.0f); + graphics.DrawLine(&sepPen, sepX, midY - m.sepH / 2, sepX, midY + m.sepH / 2); + + // 块大小组：每个单元格画一个 N×N 的马赛克方块网格，块越大网格越粗 + int sizeStartX = sepX + m.sepGap + 1; + for (int i = 0; i < SC_MOSAIC_COUNT; i++) { + int cellLeft = sizeStartX + i * (m.cell + cellGap); + bool sel = (i == sizeIdx); + if (sel) drawCellBg(cellLeft); + Gdiplus::Color c = cellColor(sel); + int cx = cellLeft + m.cell / 2; + // 网格区域边长（占单元格约 0.6） + int gridHalf = (int)(m.cell * 0.26); + int gridSize = gridHalf * 2; + int gx = cx - gridHalf; + int gy = midY - gridHalf; + // 块数随预设递增：i=0 -> 2x2, i=1 -> 3x3, i=2 -> 4x4 + int n = 2 + i; + int cellSz = gridSize / n; + if (cellSz < 1) cellSz = 1; + Gdiplus::SolidBrush b(c); + // 交错填充模拟马赛克质感（棋盘格） + for (int ry = 0; ry < n; ry++) { + for (int rx = 0; rx < n; rx++) { + if (((rx + ry) & 1) == 0) { + graphics.FillRectangle(&b, gx + rx * cellSz, gy + ry * cellSz, + cellSz, cellSz); + } + } + } + // 网格描边（未填充格用半透明） + Gdiplus::Pen gridPen(Gdiplus::Color(sel ? 200 : 120, + c.GetRed(), c.GetGreen(), c.GetBlue()), + 1.0f); + for (int k = 0; k <= n; k++) { + graphics.DrawLine(&gridPen, gx + k * cellSz, gy, + gx + k * cellSz, gy + n * cellSz); + graphics.DrawLine(&gridPen, gx, gy + k * cellSz, + gx + n * cellSz, gy + k * cellSz); + } + } + + // 第二条分隔线 + int sizeEndX = sizeStartX + SC_MOSAIC_COUNT * m.cell + + (SC_MOSAIC_COUNT - 1) * cellGap; + int sep2X = sizeEndX + m.sepGap; + graphics.DrawLine(&sepPen, sep2X, midY - m.sepH / 2, sep2X, midY + m.sepH / 2); + + // 涂抹半径组：用不同直径的圆点表示半径大小（类似画笔粗细）。 + // 仅涂抹模式下有意义；框选模式下置灰但仍可点击（切换后立即生效）。 + int radiusStartX = sep2X + m.sepGap + 1; + bool radiusEnabled = (modeIdx == 0); + for (int i = 0; i < SC_MOSAIC_RADIUS_COUNT; i++) { + int cellLeft = radiusStartX + i * (m.cell + cellGap); + bool sel = (i == radiusIdx) && radiusEnabled; + if (sel) drawCellBg(cellLeft); + Gdiplus::Color c = cellColor(sel); + int cx = cellLeft + m.cell / 2; + // 圆点直径随预设递增：小/中/大 + int dotD = (int)(SC_MOSAIC_RADIUS[i] * 0.5 * (m.cell / (double)SC_POPUP_CELL) + 0.5); + if (dotD < 5) dotD = 5; + if (dotD > m.cell - 4) dotD = m.cell - 4; + int r = dotD / 2; + Gdiplus::Color drawC = radiusEnabled ? c + : Gdiplus::Color(160, c.GetRed(), c.GetGreen(), c.GetBlue()); + Gdiplus::SolidBrush brush(drawC); + graphics.FillEllipse(&brush, cx - r, midY - r, r * 2, r * 2); + } + } + Gdiplus::GdiplusShutdown(gdipToken); +} + +// 命中测试子菜单，返回值约定： +// +1..+SC_THICK_COUNT = 第 N 个粗细 +// -1..-SC_COLOR_COUNT = 第 N 个颜色（取负为索引+1） +// 0 = 未命中（包括点在分隔线上） +static int HitTestPopup(int x, int y, const RECT& popupRect, const SCPopupMetrics& m) { + if (!PointInRect(x, y, popupRect)) return 0; + + // 内容左边界（绝对） + int contentLeft = popupRect.left + m.border + m.pad; + int contentTop = popupRect.top + m.border + m.pad; + int cellGap = (int)(SC_TOOLBAR_GAP * (m.cell / (double)SC_POPUP_CELL) + 0.5); + // y 必须在单元格高度内 + if (y < contentTop || y >= contentTop + m.cell) return 0; + + // 粗细组：contentLeft 起 + int thickX0 = contentLeft; + for (int i = 0; i < SC_THICK_COUNT; i++) { + int ix = thickX0 + i * (m.cell + cellGap); + if (x >= ix && x < ix + m.cell) return i + 1; + } + int thickEndX = thickX0 + SC_THICK_COUNT * m.cell + + (SC_THICK_COUNT - 1) * cellGap; + // 分隔线区域（不命中） + int colorStartX = thickEndX + m.sepGap * 2 + 1; + if (x < colorStartX) return 0; + + // 颜色组 + for (int i = 0; i < SC_COLOR_COUNT; i++) { + int ix = colorStartX + i * (m.cell + cellGap); + if (x >= ix && x < ix + m.cell) return -(i + 1); + } + return 0; +} + +// 绘制子菜单（GDI+ 抗锯齿白底圆角，单行布局）。 +// 第一组：isTextTool 时显示字号（不同大小 'A'），否则显示粗细（圆点直径区分）。 +// 第二组：颜色（固定直径圆点）。中间一条竖直分隔线。 +static void DrawPopup(HDC hdc, const RECT& popupRect, + int colorIdx, int firstIdx, bool isTextTool, + const SCPopupMetrics& m) { + Gdiplus::GdiplusStartupInput startupInput; + ULONG_PTR gdipToken; + Gdiplus::GdiplusStartup(&gdipToken, &startupInput, NULL); + { + Gdiplus::Graphics graphics(hdc); + graphics.SetSmoothingMode(Gdiplus::SmoothingModeAntiAlias); + + int pw = popupRect.right - popupRect.left; + int ph = popupRect.bottom - popupRect.top; + + // 白色圆角背景 + Gdiplus::SolidBrush whiteBrush(Gdiplus::Color(255, 255, 255, 255)); + Gdiplus::GraphicsPath bgPath; + AddRoundedRect(bgPath, popupRect.left, popupRect.top, pw, ph, m.radius); + graphics.FillPath(&whiteBrush, &bgPath); + // 浅灰边框 + Gdiplus::Pen borderPen(Gdiplus::Color(255, 210, 210, 210), (Gdiplus::REAL)m.border); + graphics.DrawPath(&borderPen, &bgPath); + + int contentLeft = popupRect.left + m.border + m.pad; + int contentTop = popupRect.top + m.border + m.pad; + int midY = contentTop + m.cell / 2; + int cellGap = (int)(SC_TOOLBAR_GAP * (m.cell / (double)SC_POPUP_CELL) + 0.5); + + // 选中态背景：单元格大小的圆角矩形。 + // 粗细沿用主工具栏选中高亮底（不透明浅蓝 RGB(225,237,253)）； + // 颜色用带透明（alpha 80）的选中色。统一加一条带透明灰色描边。 + auto drawCellBg = [&](int cellLeft, const Gdiplus::Color& bg) { + Gdiplus::GraphicsPath p; + AddRoundedRect(p, cellLeft, contentTop, m.cell, m.cell, m.cell / 4); + Gdiplus::SolidBrush bgBrush(bg); + graphics.FillPath(&bgBrush, &p); + // 背景边缘描边（带透明灰，alpha 90） + Gdiplus::Pen edgePen(Gdiplus::Color(90, 160, 160, 160), 1.0f); + graphics.DrawPath(&edgePen, &p); + }; + + // 第一组：文字工具显示字号（不同大小 'A'），矢量工具显示粗细（圆点直径区分） + int firstCount = isTextTool ? SC_FONT_COUNT : SC_THICK_COUNT; + int firstX0 = contentLeft; + for (int i = 0; i < firstCount; i++) { + int cellLeft = firstX0 + i * (m.cell + cellGap); + if (i == firstIdx) { + // 与主工具栏选中态一致的不透明浅蓝底 + drawCellBg(cellLeft, Gdiplus::Color(255, 225, 237, 253)); + } + Gdiplus::Color iconC = (i == firstIdx) + ? Gdiplus::Color(255, 0x3B, 0x8B, 0xF2) + : Gdiplus::Color(255, 0x33, 0x33, 0x33); + int cx = cellLeft + m.cell / 2; + if (isTextTool) { + // 字号：用不同大小的字母 'A' 表示，居中绘制。 + // 缩放系数：最大字号（36）额外缩小到 0.62，前两个字号（16/24）保持 0.72， + // 避免最大字号的 'A' 仍偏大溢出。 + double sizeScale = (i == SC_FONT_COUNT - 1) ? 0.62 : 0.72; + int fontPx = (int)(SC_FONT_SIZES[i] * (m.cell / (double)SC_POPUP_CELL) * sizeScale + 0.5); + if (fontPx < 6) fontPx = 6; + Gdiplus::FontFamily fontFamily(SC_FONT_FACE); + Gdiplus::FontStyle fs = Gdiplus::FontStyleRegular; + // 'A' 像素高 = fontPx，按高度反推 emSize（GDI+ 用 em） + Gdiplus::Font font(&fontFamily, (Gdiplus::REAL)fontPx, fs, Gdiplus::UnitPixel); + Gdiplus::SolidBrush b(iconC); + Gdiplus::StringFormat sf; + sf.SetAlignment(Gdiplus::StringAlignmentCenter); + sf.SetLineAlignment(Gdiplus::StringAlignmentCenter); + Gdiplus::RectF cellRect((Gdiplus::REAL)cellLeft, (Gdiplus::REAL)contentTop, + (Gdiplus::REAL)m.cell, (Gdiplus::REAL)m.cell); + graphics.DrawString(L"A", 1, &font, cellRect, &sf, &b); + } else { + // 粗细：圆点直径取自 SC_THICK_DOT_SIZES + int dotD = (int)(SC_THICK_DOT_SIZES[i] * (m.cell / (double)SC_POPUP_CELL) + 0.5); + if (dotD < 4) dotD = 4; + if (dotD > m.cell) dotD = m.cell; + int r = dotD / 2; + Gdiplus::SolidBrush brush(iconC); + graphics.FillEllipse(&brush, cx - r, midY - r, r * 2, r * 2); + } + } + + // 分隔线 + int firstEndX = firstX0 + firstCount * m.cell + + (firstCount - 1) * cellGap; + int sepX = firstEndX + m.sepGap; + Gdiplus::Pen sepPen(Gdiplus::Color(255, 220, 220, 220), 1.0f); + graphics.DrawLine(&sepPen, sepX, midY - m.sepH / 2, sepX, midY + m.sepH / 2); + + // 颜色组：选中时背景变为带透明的选中色 + int colorStartX = sepX + m.sepGap + 1; + for (int i = 0; i < SC_COLOR_COUNT; i++) { + COLORREF c = SC_COLOR_PRESETS[i]; + int cellLeft = colorStartX + i * (m.cell + cellGap); + int cx = cellLeft + m.cell / 2; + int r = m.colorDot / 2; + if (i == colorIdx) { + drawCellBg(cellLeft, + Gdiplus::Color(80, GetRValue(c), GetGValue(c), GetBValue(c))); + } + Gdiplus::Color gc(GetRValue(c), GetGValue(c), GetBValue(c)); + Gdiplus::SolidBrush brush(gc); + graphics.FillEllipse(&brush, cx - r, midY - r, r * 2, r * 2); + // 圆点本身始终保留极浅描边（白色块可见性），选中也保留 + Gdiplus::Pen outline(Gdiplus::Color(255, 220, 220, 220), 1.0f); + graphics.DrawEllipse(&outline, cx - r, midY - r, r * 2, r * 2); + } + } + Gdiplus::GdiplusShutdown(gdipToken); +} + +// 截图上下文 +struct CaptureContext { + CaptureState state; + int virtualX, virtualY, virtualW, virtualH; + int startX, startY, endX, endY; + int mouseX, mouseY; + COLORREF currentColor; + std::vector windows; + int hoveredWindow; // -1 = none + // 预截屏 + HBITMAP screenBitmap; + HDC memDC; + // 双缓冲 + HDC backDC; + HBITMAP backBitmap; + // 脏区域追踪 + RECT lastPanelRect; + RECT lastSelectionRect; + RECT lastLabelRect; + RECT lastHighlightRect; + RECT lastToolbarRect; + RECT lastPopupRect; + bool needFullRedraw; + // DPI + double dpiScale; + // GDI 资源 + SCGdiResources gdi; + + // ---- 确认态：可调整选区 ---- + // 已确认的选区（绝对屏幕坐标） + RECT selection; + // 当前正在拖拽的手柄（CS_Resizing 时有效），CS_Confirmed 下表示 hover 手柄 + int resizeHandle; + // 整体拖动/调整起点（绝对屏幕坐标） + int dragStartX, dragStartY; + RECT dragStartSelection; + + // ---- 悬浮工具栏 ---- + // 工具栏矩形（相对虚拟屏幕坐标，绘制用） + RECT toolbarRect; + // 工具栏 hover 按钮，-1 = none + int hoverToolbarBtn; + // 当前激活的绘制工具（高亮显示，仅界面） + int activeTool; + // 工具栏图标位图缓存（按 DPI 预渲染，dark/white 双色） + SCIconCache iconCache; + // 当前 DPI 下的工具栏几何（缓存，避免每次绘制重算） + SCToolbarMetrics toolbarMetrics; + + // ---- 标注绘制 ---- + std::vector annotations; // 已提交标注 + std::vector redoStack; // 撤销暂存 + Annotation curDrawing; // CS_Drawing 中正在绘制的标注 + bool hasCurDrawing; // curDrawing 是否有效 + int drawColorIdx; // 当前选中颜色索引 + int drawThickIdx; // 当前选中粗细索引（矢量工具） + int fontSizeIdx; // 当前选中字号索引（文字工具） + // 马赛克工具属性 + int mosaicSizeIdx; // 当前选中马赛克块大小索引 + int mosaicRadiusIdx; // 当前选中涂抹半径索引 + bool mosaicRectMode; // true=框选区域模式；false=涂抹模式 + // 涂抹模式光标：用系统光标机制（SetCursor）显示半径圆，由 OS 跟随鼠标， + // 无 WM_PAINT 重绘延迟（之前的 overlay 圆走 MOUSEMOVE→InvalidateRect→WM_PAINT 链路， + // 全屏重绘开销大导致不跟手）。按半径预设预生成彩色光标并缓存。 + HCURSOR mosaicBrushCursors[3]; // 对应 SC_MOSAIC_RADIUS_COUNT 个半径预设的光标 + bool mosaicBrushCursorsInited; + // ---- 马赛克渲染（reveal-mask 模型，消除不连续感）---- + // 预先把整张截图按当前块大小马赛克化得到 mosaicBase（逻辑像素，与 backDC 同尺寸）。 + // 马赛克标注只是「蒙版」：涂抹=路径圆形区域、框选=矩形区域，揭示其背后的 mosaicBase。 + // 这样任意区域、任意顺序叠加都连续无缝；切换块大小时只需重建 base，已揭示区域自动更新。 + // mosaicBase 覆盖整虚拟屏幕（绝对坐标），与选区无关，resize/move 无需重建。 + HDC mosaicBaseDC; + HBITMAP mosaicBaseBitmap; + int mosaicBaseW, mosaicBaseH; // base 尺寸（= 虚拟屏幕逻辑尺寸） + int mosaicBaseBlockPx; // 生成 base 时的块大小（检测变更触发重建） + // 涂抹模式增量绘制：记录上一帧最后绘制的路径点索引（reveal 模型下未使用，保留扩展）。 + int mosaicDrawLastIdx; + // 粗细/颜色子菜单 + bool popupOpen; + RECT popupRect; + SCPopupMetrics popupMetrics; + + // ---- 文字输入（CS_TextEditing）---- + std::wstring textBuf; // 正在输入的文字缓冲 + int textAnchorX, textAnchorY; // 文字锚点（绝对虚拟屏幕坐标） + int textCaretPos; // 插入符在 textBuf 中的 wchar 位置 + bool textCaretVisible; // 光标是否可见（闪烁控制） + DWORD textCaretLastBlink; // 上次光标闪烁时间（毫秒） + int textSelStart; // 文字选择起始位置（-1 表示无选择） + int textSelEnd; // 文字选择结束位置 + bool textDraggingSelection; // 是否正在拖动选择文字 + int hoveredTextAnnotation; // 悬浮命中的文字标注索引（-1 表示无，仅用于光标/即时反馈） + int selectedTextAnnotation; // 已选中的文字标注索引（-1 表示无，持久保持直到点空白） + int draggingTextAnnotation; // 正在拖动的文字标注索引（-1 表示无） + int textDragStartX, textDragStartY; // 文字拖动起始位置 +}; + +// 截图上下文指针（窗口过程使用） +static CaptureContext* g_captureCtx = nullptr; + +// Base64 编码表 +static const char base64_chars[] = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/"; + +// Base64 编码 +static std::string Base64Encode(const BYTE* data, size_t len) { + std::string result; + result.reserve(((len + 2) / 3) * 4); + for (size_t i = 0; i < len; i += 3) { + unsigned int b = (data[i] << 16) | ((i + 1 < len ? data[i + 1] : 0) << 8) | (i + 2 < len ? data[i + 2] : 0); + result.push_back(base64_chars[(b >> 18) & 0x3F]); + result.push_back(base64_chars[(b >> 12) & 0x3F]); + result.push_back(i + 1 < len ? base64_chars[(b >> 6) & 0x3F] : '='); + result.push_back(i + 2 < len ? base64_chars[b & 0x3F] : '='); + } + return result; +} + +// ---- 工具函数 ---- + +// 获取 DPI 缩放因子 +static double GetDpiScaleFactor() { + typedef UINT (WINAPI *GetDpiForSystemProc)(); + HMODULE user32 = GetModuleHandleW(L"user32.dll"); + if (user32) { + auto proc = (GetDpiForSystemProc)GetProcAddress(user32, "GetDpiForSystem"); + if (proc) { + UINT dpi = proc(); + double scale = dpi / 96.0; + if (scale < 0.5) scale = 0.5; + if (scale > 4.0) scale = 4.0; + return scale; + } + } + return 1.0; +} + +// 显示器枚举回调数据 +struct MonitorEnumData { + LONG minLeft, minTop, maxRight, maxBottom; + double totalDpiScale; + int monitorCount; +}; + +// 显示器枚举回调 +static BOOL CALLBACK MonitorEnumProc(HMONITOR hMonitor, HDC hdcMonitor, LPRECT lprcMonitor, LPARAM dwData) { + MonitorEnumData* data = reinterpret_cast(dwData); + + // 获取显示器的物理尺寸 + MONITORINFOEXW mi; + mi.cbSize = sizeof(MONITORINFOEXW); + if (GetMonitorInfoW(hMonitor, &mi)) { + // 在 DPI 感知模式下，rcMonitor 已经是物理像素坐标 + data->minLeft = (std::min)(data->minLeft, mi.rcMonitor.left); + data->minTop = (std::min)(data->minTop, mi.rcMonitor.top); + data->maxRight = (std::max)(data->maxRight, mi.rcMonitor.right); + data->maxBottom = (std::max)(data->maxBottom, mi.rcMonitor.bottom); + data->monitorCount++; + + // 获取显示器 DPI + typedef HRESULT(WINAPI* GetDpiForMonitorProc)(HMONITOR, int, UINT*, UINT*); + HMODULE shcore = LoadLibraryW(L"shcore.dll"); + if (shcore) { + auto getDpiForMonitor = (GetDpiForMonitorProc)GetProcAddress(shcore, "GetDpiForMonitor"); + if (getDpiForMonitor) { + UINT dpiX, dpiY; + if (SUCCEEDED(getDpiForMonitor(hMonitor, 0/*MDT_EFFECTIVE_DPI*/, &dpiX, &dpiY))) { + data->totalDpiScale = (std::max)(data->totalDpiScale, dpiX / 96.0); + } + } + FreeLibrary(shcore); + } + } + return TRUE; +} + +// 截取整个虚拟屏幕到物理尺寸位图 +static bool CaptureVirtualScreen(HDC& outMemDC, HBITMAP& outBitmap, + int& vx, int& vy, int& vw, int& vh, double& dpiScale) { + // 获取逻辑坐标的虚拟屏幕尺寸 + vx = GetSystemMetrics(SM_XVIRTUALSCREEN); + vy = GetSystemMetrics(SM_YVIRTUALSCREEN); + vw = GetSystemMetrics(SM_CXVIRTUALSCREEN); + vh = GetSystemMetrics(SM_CYVIRTUALSCREEN); + + // 枚举所有显示器获取物理像素边界 + MonitorEnumData enumData = { INT_MAX, INT_MAX, INT_MIN, INT_MIN, 1.0, 0 }; + EnumDisplayMonitors(NULL, NULL, MonitorEnumProc, reinterpret_cast(&enumData)); + + // 计算物理尺寸（使用枚举得到的实际物理像素边界） + int physVx = enumData.minLeft; + int physVy = enumData.minTop; + int physVw = enumData.maxRight - enumData.minLeft; + int physVh = enumData.maxBottom - enumData.minTop; + + // 如果枚举失败，回退到 DPI 缩放计算 + if (physVw <= 0 || physVh <= 0 || enumData.monitorCount == 0) { + physVx = (int)(vx * dpiScale); + physVy = (int)(vy * dpiScale); + physVw = (int)(vw * dpiScale + 0.5); + physVh = (int)(vh * dpiScale + 0.5); + } + + HDC screenDC = GetDC(NULL); + if (!screenDC) return false; + + outMemDC = CreateCompatibleDC(screenDC); + if (!outMemDC) { ReleaseDC(NULL, screenDC); return false; } + + outBitmap = CreateCompatibleBitmap(screenDC, physVw, physVh); + if (!outBitmap) { DeleteDC(outMemDC); ReleaseDC(NULL, screenDC); return false; } + + SelectObject(outMemDC, outBitmap); + + // 设置高质量拉伸模式 + SetStretchBltMode(outMemDC, HALFTONE); + SetBrushOrgEx(outMemDC, 0, 0, NULL); + + // 直接 BitBlt 物理像素（在 DPI 感知模式下，屏幕 DC 和坐标都是物理像素级别） + BitBlt(outMemDC, 0, 0, physVw, physVh, screenDC, physVx, physVy, SRCCOPY); + + // 更新返回的 dpiScale 为实际的物理/逻辑比例 + // 这样后续的坐标转换才能正确 + if (vw > 0 && vh > 0) { + dpiScale = (double)physVw / vw; + } + + ReleaseDC(NULL, screenDC); + return true; +} + +// 创建双缓冲 +static bool CreateBackBuffer(HDC& outDC, HBITMAP& outBmp, int w, int h) { + HDC screenDC = GetDC(NULL); + if (!screenDC) return false; + outDC = CreateCompatibleDC(screenDC); + if (!outDC) { ReleaseDC(NULL, screenDC); return false; } + outBmp = CreateCompatibleBitmap(screenDC, w, h); + if (!outBmp) { DeleteDC(outDC); ReleaseDC(NULL, screenDC); return false; } + SelectObject(outDC, outBmp); + ReleaseDC(NULL, screenDC); + return true; +} + +// 从预截屏位图读取像素颜色（逻辑坐标） +static COLORREF GetPixelColorFromBitmap(HDC memDC, int x, int y, int vx, int vy, double dpiScale) { + int lx = x - vx; + int ly = y - vy; + int px = (int)(lx * dpiScale + 0.5); + int py = (int)(ly * dpiScale + 0.5); + return GetPixel(memDC, px, py); +} + +// COLORREF 转 HEX/RGB 字符串 +static void ColorrefToStrings(COLORREF color, char* hexBuf, char* rgbBuf) { + int r = color & 0xFF; + int g = (color >> 8) & 0xFF; + int b = (color >> 16) & 0xFF; + sprintf_s(hexBuf, 32, "#%02X%02X%02X", r, g, b); + sprintf_s(rgbBuf, 32, "%d, %d, %d", r, g, b); +} + +// 枚举窗口回调 +static BOOL CALLBACK SCEnumWindowsProc(HWND hwnd, LPARAM lParam) { + auto* windows = reinterpret_cast*>(lParam); + + if (!IsWindowVisible(hwnd)) return TRUE; + + LONG_PTR exStyle = GetWindowLongPtrW(hwnd, GWL_EXSTYLE); + if (exStyle & WS_EX_TOOLWINDOW) return TRUE; + + LONG_PTR style = GetWindowLongPtrW(hwnd, GWL_STYLE); + if (style == 0) return TRUE; + + // 检查是否为幽灵窗口（cloaked window） + // 幽灵窗口虽然 IsWindowVisible 返回 true，但实际上不可见 + BOOL isCloaked = FALSE; + HRESULT hrCloaked = DwmGetWindowAttribute(hwnd, DWMWA_CLOAKED, &isCloaked, sizeof(isCloaked)); + if (SUCCEEDED(hrCloaked) && isCloaked) { + return TRUE; // 跳过幽灵窗口 + } + + // 获取窗口类名以进行额外过滤 + const int MAX_CLASS_NAME = 256; + WCHAR className[MAX_CLASS_NAME] = {0}; + int classNameLen = GetClassNameW(hwnd, className, MAX_CLASS_NAME); + + // 过滤某些特殊的系统窗口类 + if (classNameLen > 0) { + // Windows 输入法相关窗口（如 Microsoft Text Input Application） + if (wcscmp(className, L"Windows.UI.Core.CoreWindow") == 0) { + // 对于 CoreWindow，再次确认是否真的可见（通过检查是否有有效的可视化区域） + RECT clientRect; + if (!GetClientRect(hwnd, &clientRect)) return TRUE; + + // 如果客户区太小，很可能是输入法等后台窗口 + int clientW = clientRect.right - clientRect.left; + int clientH = clientRect.bottom - clientRect.top; + if (clientW < 100 || clientH < 100) return TRUE; + } + + // 过滤 ApplicationFrameWindow 的空壳窗口 + // UWP 应用在未激活时可能留下空的 ApplicationFrameWindow + if (wcscmp(className, L"ApplicationFrameWindow") == 0) { + // 检查窗口是否被最小化或隐藏 + if (IsIconic(hwnd)) return TRUE; + + // 检查是否真的有内容（通过检查窗口透明度或其他属性） + BYTE opacity = 255; + DWORD cloakedReason = 0; + DwmGetWindowAttribute(hwnd, DWMWA_CLOAKED, &cloakedReason, sizeof(cloakedReason)); + if (cloakedReason != 0) return TRUE; + } + } + + int titleLen = GetWindowTextLengthW(hwnd); + if (titleLen == 0) return TRUE; + + std::wstring title(titleLen + 1, L'\0'); + GetWindowTextW(hwnd, &title[0], titleLen + 1); + title.resize(titleLen); + + if (hwnd == GetDesktopWindow()) return TRUE; + + // 使用 DWM 获取精确边界 + RECT rect = {}; + HRESULT hr = DwmGetWindowAttribute(hwnd, DWMWA_EXTENDED_FRAME_BOUNDS, &rect, sizeof(rect)); + if (FAILED(hr)) { + if (!GetWindowRect(hwnd, &rect)) return TRUE; + } + + int w = rect.right - rect.left; + int h = rect.bottom - rect.top; + if (w < 50 || h < 50) return TRUE; + + SCWindowInfo info; + info.hwnd = hwnd; + info.rect = rect; + info.title = title; + windows->push_back(info); + return TRUE; +} + +// 枚举窗口 +static std::vector EnumWindowsForCapture() { + std::vector windows; + EnumWindows(SCEnumWindowsProc, reinterpret_cast(&windows)); + return windows; +} + +// 查找鼠标下方的窗口 +static int FindWindowAtPoint(const std::vector& windows, int x, int y) { + for (size_t i = 0; i < windows.size(); i++) { + const RECT& r = windows[i].rect; + if (x >= r.left && x < r.right && y >= r.top && y < r.bottom) + return (int)i; + } + return -1; +} + +// 计算浮窗位置（优先右下，超出则翻转） +static void CalcPanelPosition(int mx, int my, int vx, int vy, int vw, int vh, int& px, int& py) { + int sr = vx + vw; + int sb = vy + vh; + px = mx + SC_PANEL_MARGIN; + py = my + SC_PANEL_MARGIN; + if (px + SC_PANEL_WIDTH > sr) px = mx - SC_PANEL_WIDTH - SC_PANEL_MARGIN; + if (py + SC_PANEL_HEIGHT > sb) py = my - SC_PANEL_HEIGHT - SC_PANEL_MARGIN; + if (px < vx) px = vx + SC_PANEL_MARGIN; + if (py < vy) py = vy + SC_PANEL_MARGIN; +} + +// 从预截屏位图恢复脏区域到后台缓冲 +static void RestoreDirtyRegion(HDC backDC, HDC memDC, const RECT& dirty, double dpiScale) { + int w = dirty.right - dirty.left; + int h = dirty.bottom - dirty.top; + if (w <= 0 || h <= 0) return; + int x = (std::max)((int)dirty.left, 0); + int y = (std::max)((int)dirty.top, 0); + w = dirty.right - x; + h = dirty.bottom - y; + if (dpiScale > 1.01 || dpiScale < 0.99) { + int px = (int)(x * dpiScale + 0.5); + int py = (int)(y * dpiScale + 0.5); + int pw = (int)(w * dpiScale + 0.5); + int ph = (int)(h * dpiScale + 0.5); + StretchBlt(backDC, x, y, w, h, memDC, px, py, pw, ph, SRCCOPY); + } else { + BitBlt(backDC, x, y, w, h, memDC, x, y, SRCCOPY); + } +} + +// 扩展矩形 +static RECT InflateRectBy(const RECT& r, int margin) { + return { r.left - margin, r.top - margin, r.right + margin, r.bottom + margin }; +} + +// ---- 绘制函数 ---- + +// 绘制放大镜 + 鼠标信息面板 +static void DrawInfoPanel(HDC hdc, int panelX, int panelY, COLORREF color, + HDC memDC, int vx, int vy, int mx, int my, double dpiScale, const SCGdiResources& gdi) { + HGDIOBJ oldBrush = SelectObject(hdc, gdi.bgBrush); + HGDIOBJ oldPen = SelectObject(hdc, gdi.borderPen); + + // 圆角矩形背景 + RoundRect(hdc, panelX, panelY, panelX + SC_PANEL_WIDTH, panelY + SC_PANEL_HEIGHT, + SC_PANEL_CORNER_RADIUS, SC_PANEL_CORNER_RADIUS); + + // 放大镜：从物理尺寸位图取像素 + int srcW = SC_PANEL_WIDTH / SC_ZOOM_FACTOR; + int srcH = SC_MAGNIFIER_HEIGHT / SC_ZOOM_FACTOR; + int mxLogical = mx - vx; + int myLogical = my - vy; + int mxPhysical = (int)(mxLogical * dpiScale + 0.5); + int myPhysical = (int)(myLogical * dpiScale + 0.5); + int srcWPhysical = (int)(srcW * dpiScale + 0.5); + int srcHPhysical = (int)(srcH * dpiScale + 0.5); + int srcXPhysical = mxPhysical - srcWPhysical / 2; + int srcYPhysical = myPhysical - srcHPhysical / 2; + + int magX = panelX + 2; + int magY = panelY + 2; + int magW = SC_PANEL_WIDTH - 4; + int magH = SC_MAGNIFIER_HEIGHT - 2; + + StretchBlt(hdc, magX, magY, magW, magH, memDC, + (std::max)(srcXPhysical, 0), (std::max)(srcYPhysical, 0), + srcWPhysical, srcHPhysical, SRCCOPY); + + // 十字准星 + SelectObject(hdc, gdi.crosshairPen); + int cx = magX + magW / 2; + int cy = magY + magH / 2; + MoveToEx(hdc, magX, cy, NULL); LineTo(hdc, magX + magW, cy); + MoveToEx(hdc, cx, magY, NULL); LineTo(hdc, cx, magY + magH); + + // 文字信息 + SetBkMode(hdc, TRANSPARENT); + SetTextColor(hdc, RGB(255, 255, 255)); + HGDIOBJ oldFont = SelectObject(hdc, gdi.smallFont); + + char hexBuf[32], rgbBuf[32]; + ColorrefToStrings(color, hexBuf, rgbBuf); + char posBuf[64]; + sprintf_s(posBuf, "%d, %d", mx, my); + + const int LABEL_PAD = 6; + int labelX = panelX + LABEL_PAD; + int valueRightX = panelX + SC_PANEL_WIDTH - LABEL_PAD; + + // 获取文字高度 + SIZE textSize; + GetTextExtentPoint32W(hdc, L"测试", 2, &textSize); + int lineH = textSize.cy; + int infoY = panelY + SC_PANEL_HEIGHT - LABEL_PAD - lineH * 3; + + // 辅助：右对齐绘制 + auto drawRightAligned = [&](const wchar_t* text, int len, int rx, int ry) { + SIZE sz; + GetTextExtentPoint32W(hdc, text, len, &sz); + TextOutW(hdc, rx - sz.cx, ry, text, len); + }; + + // 坐标 + TextOutW(hdc, labelX, infoY, L"坐标", 2); + std::wstring posW(posBuf, posBuf + strlen(posBuf)); + drawRightAligned(posW.c_str(), (int)posW.size(), valueRightX, infoY); + + // HEX + TextOutW(hdc, labelX, infoY + lineH, L"HEX", 3); + std::wstring hexW(hexBuf, hexBuf + strlen(hexBuf)); + drawRightAligned(hexW.c_str(), (int)hexW.size(), valueRightX, infoY + lineH); + + // RGB + TextOutW(hdc, labelX, infoY + lineH * 2, L"RGB", 3); + std::wstring rgbW(rgbBuf, rgbBuf + strlen(rgbBuf)); + drawRightAligned(rgbW.c_str(), (int)rgbW.size(), valueRightX, infoY + lineH * 2); + + SelectObject(hdc, oldFont); + SelectObject(hdc, oldBrush); + SelectObject(hdc, oldPen); +} + +// 绘制尺寸标签，返回标签矩形 +static RECT DrawSizeLabel(HDC hdc, int width, int height, + int refLeft, int refTop, int refRight, int refBottom, + int virtualW, int virtualH, const SCGdiResources& gdi) { + RECT empty = {0, 0, 0, 0}; + if (width < 0 || height < 0) return empty; + + wchar_t sizeBuf[64]; + swprintf_s(sizeBuf, L"%d × %d", width, height); + int sizeLen = (int)wcslen(sizeBuf); + + HGDIOBJ oldFont = SelectObject(hdc, gdi.smallFont); + SIZE textSize; + GetTextExtentPoint32W(hdc, sizeBuf, sizeLen, &textSize); + + const int LP = 12, LS = 5; + int labelW = textSize.cx + LP * 2; + int labelH = textSize.cy + 4; + + int lx = refLeft; + int ly = refTop - labelH - LS; + if (ly < 0) { + lx = refLeft + LS; + ly = refTop + LS; + if (lx + labelW > virtualW) lx = virtualW - labelW - LS; + if (ly + labelH > virtualH) ly = virtualH - labelH - LS; + if (lx + labelW > refRight) lx = refRight - labelW - LS; + if (ly + labelH > refBottom) ly = refBottom - labelH - LS; + } + if (lx < 0) lx = 0; + if (ly < 0) ly = 0; + if (lx + labelW > virtualW) lx = virtualW - labelW; + if (ly + labelH > virtualH) ly = virtualH - labelH; + + HGDIOBJ oldBrush = SelectObject(hdc, gdi.bgBrush); + HGDIOBJ oldPen = SelectObject(hdc, gdi.borderPen); + RoundRect(hdc, lx, ly, lx + labelW, ly + labelH, SC_PANEL_CORNER_RADIUS, SC_PANEL_CORNER_RADIUS); + + SetBkMode(hdc, TRANSPARENT); + SetTextColor(hdc, RGB(255, 255, 255)); + TextOutW(hdc, lx + LP, ly + 2, sizeBuf, sizeLen); + + SelectObject(hdc, oldFont); + SelectObject(hdc, oldBrush); + SelectObject(hdc, oldPen); + + RECT result = { lx, ly, lx + labelW, ly + labelH }; + return result; +} + +// 绘制选区矩形边框 + 尺寸标签 +static RECT DrawSelection(HDC hdc, int x1, int y1, int x2, int y2, + int vx, int vy, int vw, int vh, const SCGdiResources& gdi) { + int left = (std::min)(x1, x2) - vx; + int top = (std::min)(y1, y2) - vy; + int right = (std::max)(x1, x2) - vx; + int bottom = (std::max)(y1, y2) - vy; + + HGDIOBJ oldPen = SelectObject(hdc, gdi.selectionPen); + HGDIOBJ oldBrush = SelectObject(hdc, GetStockObject(NULL_BRUSH)); + Rectangle(hdc, left, top, right, bottom); + + int sizeW = right - left; + int sizeH = bottom - top; + RECT labelRect = DrawSizeLabel(hdc, sizeW, sizeH, left, top, right, bottom, vw, vh, gdi); + + SelectObject(hdc, oldPen); + SelectObject(hdc, oldBrush); + return labelRect; +} + +// 绘制窗口高亮边框 +static void DrawWindowHighlight(HDC hdc, const RECT& rect, int vx, int vy, const SCGdiResources& gdi) { + int left = rect.left - vx; + int top = rect.top - vy; + int right = rect.right - vx; + int bottom = rect.bottom - vy; + + HGDIOBJ oldPen = SelectObject(hdc, gdi.highlightPen); + HGDIOBJ oldBrush = SelectObject(hdc, GetStockObject(NULL_BRUSH)); + Rectangle(hdc, left, top, right, bottom); + SelectObject(hdc, oldPen); + SelectObject(hdc, oldBrush); +} + +// 绘制选区外遮罩（微信风格） +// 在 backDC 上对"选区外部"区域 AlphaBlend 一层半透明黑色， +// 选区内部不绘制，保持原始截图清晰。 +// rect 为相对虚拟屏幕的逻辑坐标（已减去 virtualX/virtualY）。 +static void DrawDimMask(HDC backDC, const SCGdiResources& gdi, + int selLeft, int selTop, int selRight, int selBottom, + int virtualW, int virtualH) { + if (!gdi.maskDC || !gdi.maskBitmap) return; + + BLENDFUNCTION blend; + blend.BlendOp = AC_SRC_OVER; + blend.BlendFlags = 0; + blend.SourceConstantAlpha = SC_MASK_ALPHA; + blend.AlphaFormat = 0; // 不使用 per-pixel alpha，仅用常量 alpha + + // 将虚拟屏幕按选区划分为四周四块，逐块 AlphaBlend 遮罩 + // 上：x ∈ [0, vw], y ∈ [0, selTop] + if (selTop > 0) { + AlphaBlend(backDC, 0, 0, virtualW, selTop, + gdi.maskDC, 0, 0, virtualW, selTop, blend); + } + // 下：x ∈ [0, vw], y ∈ [selBottom, vh] + if (selBottom < virtualH) { + AlphaBlend(backDC, 0, selBottom, virtualW, virtualH - selBottom, + gdi.maskDC, 0, selBottom, virtualW, virtualH - selBottom, blend); + } + // 左：x ∈ [0, selLeft], y ∈ [selTop, selBottom] + if (selLeft > 0 && selBottom > selTop) { + AlphaBlend(backDC, 0, selTop, selLeft, selBottom - selTop, + gdi.maskDC, 0, selTop, selLeft, selBottom - selTop, blend); + } + // 右：x ∈ [selRight, vw], y ∈ [selTop, selBottom] + if (selRight < virtualW && selBottom > selTop) { + AlphaBlend(backDC, selRight, selTop, virtualW - selRight, selBottom - selTop, + gdi.maskDC, selRight, selTop, virtualW - selRight, selBottom - selTop, blend); + } +} + +// ---- 确认态辅助函数 ---- + +// 规范化矩形（保证 left= r.left && x < r.right && y >= r.top && y < r.bottom; +} + +// 命中测试调整手柄，返回 ResizeHandle（绝对坐标） +static int HitTestHandle(int x, int y, const RECT& sel) { + int hs = SC_HANDLE_SIZE; + int cx = (sel.left + sel.right) / 2; + int cy = (sel.top + sel.bottom) / 2; + // 8 个手柄的判定矩形（顺序与 ResizeHandle 一致） + struct { int hx, hy; int handle; } tests[] = { + { sel.left, cy, RH_Left }, + { sel.right, cy, RH_Right }, + { cx, sel.top, RH_Top }, + { cx, sel.bottom, RH_Bottom }, + { sel.left, sel.top, RH_TopLeft }, + { sel.right, sel.top, RH_TopRight }, + { sel.left, sel.bottom, RH_BottomLeft }, + { sel.right, sel.bottom, RH_BottomRight }, + }; + for (auto& t : tests) { + RECT box = { t.hx - hs, t.hy - hs, t.hx + hs, t.hy + hs }; + if (PointInRect(x, y, box)) return t.handle; + } + return RH_None; +} + +// 根据手柄返回对应的系统鼠标光标 +static LPCWSTR HandleCursor(int handle) { + switch (handle) { + case RH_Left: + case RH_Right: + return (LPCWSTR)IDC_SIZEWE; + case RH_Top: + case RH_Bottom: + return (LPCWSTR)IDC_SIZENS; + case RH_TopLeft: + case RH_BottomRight: + return (LPCWSTR)IDC_SIZENWSE; + case RH_TopRight: + case RH_BottomLeft: + return (LPCWSTR)IDC_SIZENESW; + default: + return (LPCWSTR)IDC_ARROW; + } +} + +// 计算工具栏位置：优先选区下方，放不下则上方，再放不下则选区内底部 +static void CalcToolbarPosition(const RECT& selRel, int virtualW, int virtualH, + const SCToolbarMetrics& m, RECT& out) { + int tw = CalcToolbarWidth(m); + int th = m.h; + int margin = m.margin; + + // 默认水平居中于选区，下方 + int x = selRel.left + ((selRel.right - selRel.left) - tw) / 2; + int y = selRel.bottom + margin; + + // 下方放不下 -> 上方 + if (y + th > virtualH) { + y = selRel.top - margin - th; + } + // 上方也放不下（选区很高），放选区内底部 + if (y < 0) { + y = selRel.bottom - margin - th; + if (y < selRel.top) y = selRel.top + margin; + } + // 水平边界约束 + if (x + tw > virtualW) x = virtualW - tw - margin; + if (x < 0) x = margin; + + out.left = x; + out.top = y; + out.right = x + tw; + out.bottom = y + th; +} + +// 命中测试工具栏按钮，返回 ToolButton 序号（相对虚拟屏幕坐标），-1 = none +static int HitTestToolbar(int x, int y, const RECT& toolbarRect, const SCToolbarMetrics& m) { + if (!PointInRect(x, y, toolbarRect)) return -1; + int idx = (x - toolbarRect.left - m.border - m.pad) / (m.btn + m.gap); + if (idx < 0 || idx >= TB_Count) return -1; + return idx; +} + +// 绘制单个工具图标：从 SCIconCache 取预渲染位图，AlphaBlend 居中绘制。 +// cx,cy 为按钮中心；iconSize 为缓存位图边长（物理像素）。 +static void DrawToolbarIcon(HDC hdc, int cx, int cy, int btn, bool active, + const SCIconCache& icons) { + HBITMAP bmp = icons.Get(btn, active); + if (!bmp) return; + + int sz = icons.iconSize; + int x = cx - sz / 2; + int y = cy - sz / 2; + + HDC srcDC = CreateCompatibleDC(hdc); + if (!srcDC) return; + HGDIOBJ oldBmp = SelectObject(srcDC, bmp); + + BLENDFUNCTION blend; + blend.BlendOp = AC_SRC_OVER; + blend.BlendFlags = 0; + blend.SourceConstantAlpha = 255; + blend.AlphaFormat = AC_SRC_ALPHA; // 使用 per-pixel alpha（位图已预乘） + AlphaBlend(hdc, x, y, sz, sz, srcDC, 0, 0, sz, sz, blend); + + SelectObject(srcDC, oldBmp); + DeleteDC(srcDC); +} + +// 绘制选区调整手柄（8 个），传入相对坐标矩形 +static void DrawResizeHandles(HDC hdc, const RECT& selRel, const SCGdiResources& gdi) { + int hs = SC_HANDLE_SIZE; + int half = hs / 2; + int cx = (selRel.left + selRel.right) / 2; + int cy = (selRel.top + selRel.bottom) / 2; + int pts[][2] = { + { selRel.left, cy }, + { selRel.right, cy }, + { cx, selRel.top }, + { cx, selRel.bottom }, + { selRel.left, selRel.top }, + { selRel.right, selRel.top }, + { selRel.left, selRel.bottom }, + { selRel.right, selRel.bottom }, + }; + HGDIOBJ oldPen = SelectObject(hdc, CreatePen(PS_SOLID, 1, RGB(255, 255, 255))); + HGDIOBJ oldBrush = SelectObject(hdc, CreateSolidBrush(RGB(0, 136, 255))); + for (auto& p : pts) { + Rectangle(hdc, p[0] - half, p[1] - half, p[0] + half, p[1] + half); + } + HGDIOBJ pen = SelectObject(hdc, oldPen); + DeleteObject(pen); + HGDIOBJ brush = SelectObject(hdc, oldBrush); + DeleteObject(brush); +} + +// 绘制确认态选区边框（细蓝框） +static void DrawConfirmedBorder(HDC hdc, const RECT& selRel, const SCGdiResources& gdi) { + HGDIOBJ oldPen = SelectObject(hdc, gdi.selectionPen); + HGDIOBJ oldBrush = SelectObject(hdc, GetStockObject(NULL_BRUSH)); + Rectangle(hdc, selRel.left, selRel.top, selRel.right, selRel.bottom); + SelectObject(hdc, oldPen); + SelectObject(hdc, oldBrush); +} + +// 用 GDI+ 圆角矩形路径填充 outPath（抗锯齿绘制的基础）。x,y,w,h 为整数像素矩形， +// radius 为圆角半径（自动钳制为不超过短边一半，避免重叠畸变）。 +// GDI+ 的 GraphicsPath 不可拷贝，故用 out 参数而非返回值。 +static void AddRoundedRect(Gdiplus::GraphicsPath& outPath, int x, int y, int w, int h, int radius) { + int r = (std::min)(radius, (std::min)(w, h) / 2); + if (r < 1) r = 1; + Gdiplus::Rect rect(x, y, w, h); + outPath.AddArc(rect.X, rect.Y, r * 2, r * 2, 180, 90); + outPath.AddArc(rect.GetRight() - r * 2, rect.Y, r * 2, r * 2, 270, 90); + outPath.AddArc(rect.GetRight() - r * 2, rect.GetBottom() - r * 2, r * 2, r * 2, 0, 90); + outPath.AddArc(rect.X, rect.GetBottom() - r * 2, r * 2, r * 2, 90, 90); + outPath.CloseFigure(); +} + +// 绘制悬浮工具栏（白底圆角 + 按钮图标 + 分组分隔线） +// 几何与图标尺寸均按 metrics（DPI 缩放）计算；图标从 icons 缓存取。 +static void DrawToolbar(HDC hdc, const RECT& toolbarRect, + int hoverBtn, int activeTool, const SCGdiResources& gdi, + const SCToolbarMetrics& m, const SCIconCache& icons) { + // 按钮在工具栏高度内的垂直留白 + int btnPad = (m.h - m.btn) / 2; + + // ---- 第一遍：GDI+ 抗锯齿绘制工具栏圆角背景 + 按钮圆角高亮 ---- + // GDI 的 RoundRect/FillRect 不支持抗锯齿，圆角边缘有锯齿，故改用 GDI+。 + Gdiplus::GdiplusStartupInput startupInput; + ULONG_PTR gdipToken; + Gdiplus::GdiplusStartup(&gdipToken, &startupInput, NULL); + { + Gdiplus::Graphics graphics(hdc); + graphics.SetSmoothingMode(Gdiplus::SmoothingModeAntiAlias); + + int tw = toolbarRect.right - toolbarRect.left; + int th = toolbarRect.bottom - toolbarRect.top; + + // 白色圆角背景填充（圆角外保持透明，露出后方截图） + Gdiplus::SolidBrush whiteBrush(Gdiplus::Color(255, 255, 255, 255)); + Gdiplus::GraphicsPath bgPath; + AddRoundedRect(bgPath, toolbarRect.left, toolbarRect.top, tw, th, m.radius); + graphics.FillPath(&whiteBrush, &bgPath); + + // 1px 浅灰边框 + Gdiplus::Pen borderPen(Gdiplus::Color(255, 210, 210, 210), (Gdiplus::REAL)m.border); + graphics.DrawPath(&borderPen, &bgPath); + + // 各按钮圆角高亮（hover/active） + // 按钮高亮圆角半径：约为按钮边长的 1/8，视觉柔和 + int hlRadius = m.btn / 8; + int hlInset = 2; + int hlSize = m.btn - hlInset * 2; + for (int i = 0; i < TB_Count; i++) { + if (i == TB_Separator1 || i == TB_Separator2) continue; + bool hover = (i == hoverBtn); + bool active = (i == activeTool); + if (!hover && !active) continue; + + int bx = toolbarRect.left + m.border + m.pad + i * (m.btn + m.gap); + int by = toolbarRect.top + btnPad; + // 选中态：主题蓝 #3B8BF2 叠白底 ~15% 的预混合浅蓝（225,237,253） + // hover 态：极浅蓝（235,243,255） + Gdiplus::Color hlColor = active + ? Gdiplus::Color(255, 225, 237, 253) + : Gdiplus::Color(255, 235, 243, 255); + Gdiplus::SolidBrush hlBrush(hlColor); + Gdiplus::GraphicsPath hlPath; + AddRoundedRect(hlPath, bx + hlInset, by + hlInset, hlSize, hlSize, hlRadius); + graphics.FillPath(&hlBrush, &hlPath); + } + } + Gdiplus::GdiplusShutdown(gdipToken); + + // ---- 第二遍：GDI 绘制分隔线 + 图标（位图直接 AlphaBlend，无需 AA）---- + for (int i = 0; i < TB_Count; i++) { + int bx = toolbarRect.left + m.border + m.pad + i * (m.btn + m.gap); + int by = toolbarRect.top + btnPad; + RECT btnRect = { bx, by, bx + m.btn, by + m.btn }; + + bool isSep = (i == TB_Separator1 || i == TB_Separator2); + if (isSep) { + // 分隔线 + int sx = bx + m.btn / 2; + int sepInset = m.btn / 8 + 2; + HGDIOBJ sepPen = CreatePen(PS_SOLID, 1, RGB(230, 230, 230)); + HGDIOBJ op = SelectObject(hdc, sepPen); + MoveToEx(hdc, sx, btnRect.top + sepInset, NULL); + LineTo(hdc, sx, btnRect.bottom - sepInset); + SelectObject(hdc, op); + DeleteObject(sepPen); + continue; + } + + bool active = (i == activeTool); + int cx = bx + m.btn / 2; + int cy = by + m.btn / 2; + DrawToolbarIcon(hdc, cx, cy, i, active, icons); + } +} + +// ==================== 标注绘制（GDI+ 抗锯齿） ==================== + +// 箭头头：在 (ex,ey) 方向（从 sx,sy 指来）画一个填充三角，大小随 thickness。 +// ox/oy 为整体偏移（覆盖层渲染时加 curSelRect.left/top；合成时为 0）。 +static void DrawArrowHead(Gdiplus::Graphics& graphics, Gdiplus::Brush& brush, + float sx, float sy, float ex, float ey, + float thickness, float ox, float oy) { + float dx = ex - sx; + float dy = ey - sy; + float len = std::sqrt(dx * dx + dy * dy); + if (len < 1.0f) return; + // 单位向量（箭头朝向） + float ux = dx / len; + float uy = dy / len; + // 箭头长度/半宽 ∝ thickness + float headLen = thickness * 3.0f + 6.0f; + float headHalfW = thickness * 1.8f + 4.0f; + if (headLen > len) headLen = len * 0.6f; + + // 箭尖 + float tipX = ex + ox, tipY = ey + oy; + // 底部中心（沿箭头方向后退 headLen） + float baseX = ex - ux * headLen + ox; + float baseY = ey - uy * headLen + oy; + // 法线方向 + float nx = -uy, ny = ux; + // 两翼 + float w1X = baseX + nx * headHalfW, w1Y = baseY + ny * headHalfW; + float w2X = baseX - nx * headHalfW, w2Y = baseY - ny * headHalfW; + + Gdiplus::PointF pts[3] = { + Gdiplus::PointF(tipX, tipY), + Gdiplus::PointF(w1X, w1Y), + Gdiplus::PointF(w2X, w2Y), + }; + graphics.FillPolygon(&brush, pts, 3); +} + +// 绘制单条标注（不含 clip）。 +// 标注坐标为绝对虚拟屏幕坐标；ox/oy 为绘制偏移，把绝对坐标换算到目标 DC 的局部坐标： +// - 覆盖层（backDC）：ox/oy = -virtualX/-virtualY（backDC 原点 = 虚拟屏幕左上角）。 +// - 合成（finalDC）：ox/oy = -rect.left/-rect.top（finalDC 原点 = 选区左上角）。 +static void DrawOneAnnotation(Gdiplus::Graphics& graphics, const Annotation& a, + float ox, float oy) { + Gdiplus::Color c(GetRValue(a.color), GetGValue(a.color), GetBValue(a.color)); + float thick = (float)a.thickness; + if (thick < 1.0f) thick = 1.0f; + Gdiplus::Pen pen(c, thick); + pen.SetLineJoin(Gdiplus::LineJoinRound); + pen.SetStartCap(Gdiplus::LineCapRound); + pen.SetEndCap(Gdiplus::LineCapRound); + Gdiplus::SolidBrush brush(c); + + switch (a.type) { + case AT_Rect: { + float x = a.x1 + ox; + float y = a.y1 + oy; + float w = (float)(a.x2 - a.x1); + float h = (float)(a.y2 - a.y1); + graphics.DrawRectangle(&pen, (std::min)(x, x + w), (std::min)(y, y + h), + std::fabs(w), std::fabs(h)); + break; + } + case AT_Circle: { + float x = a.x1 + ox; + float y = a.y1 + oy; + float w = (float)(a.x2 - a.x1); + float h = (float)(a.y2 - a.y1); + graphics.DrawEllipse(&pen, (std::min)(x, x + w), (std::min)(y, y + h), + std::fabs(w), std::fabs(h)); + break; + } + case AT_Arrow: { + float sx = a.x1 + ox; + float sy = a.y1 + oy; + float ex = a.x2 + ox; + float ey = a.y2 + oy; + // 线段终点回退一段，避免与箭头头重叠 + float dx = ex - sx, dy = ey - sy; + float len = std::sqrt(dx * dx + dy * dy); + float back = thick * 3.0f + 6.0f; + if (len > back + 2) { + float ex2 = ex - dx / len * back; + float ey2 = ey - dy / len * back; + graphics.DrawLine(&pen, sx, sy, ex2, ey2); + } + DrawArrowHead(graphics, brush, sx, sy, ex, ey, thick, 0, 0); + break; + } + case AT_Brush: { + if (a.pts.size() < 2) break; + std::vector pts; + pts.reserve(a.pts.size()); + for (const POINT& p : a.pts) { + pts.push_back(Gdiplus::PointF((float)(p.x) + ox, (float)(p.y) + oy)); + } + graphics.DrawLines(&pen, pts.data(), (INT)pts.size()); + break; + } + case AT_Text: { + if (a.text.empty()) break; + // 字号 = thickness；用 GDI+ 字体绘制，顶部对齐到锚点 y1 + int fontPx = a.thickness; + if (fontPx < 8) fontPx = 8; + Gdiplus::FontFamily fontFamily(SC_FONT_FACE); + Gdiplus::Font font(&fontFamily, (Gdiplus::REAL)fontPx, Gdiplus::FontStyleRegular, + Gdiplus::UnitPixel); + Gdiplus::SolidBrush textBrush(c); + Gdiplus::StringFormat sf; + sf.SetAlignment(Gdiplus::StringAlignmentNear); + sf.SetLineAlignment(Gdiplus::StringAlignmentNear); + // 用 MeasureString 得到合适布局矩形（避免被裁） + Gdiplus::RectF layoutRect((Gdiplus::REAL)(a.x1 + ox), (Gdiplus::REAL)(a.y1 + oy), + 10000.0f, (Gdiplus::REAL)(fontPx * 2)); + graphics.DrawString(a.text.c_str(), -1, &font, layoutRect, &sf, &textBrush); + break; + } + } +} + +// ==================== 马赛克渲染 ==================== +// 马赛克原理：从原始屏幕位图（srcDC = memDC，物理像素）取目标区域，按 mosaicSize 分块， +// 每块用「缩小到 1 像素再放大」得到平均色块（StretchBlt 降采样），形成马赛克效果。 +// 目标 DC（targetDC）为逻辑像素（backDC / finalDC），源 DC（srcDC）为物理像素（memDC）， +// 二者通过 dpiScale 换算：srcX = (absX - virtualX) * dpiScale。 + +// 对单个矩形区域做马赛克化并绘制到 targetDC。 +// dstX0/dstY0/dstW/dstH：目标逻辑像素矩形（已应用偏移到 targetDC 局部坐标）； +// srcAbsX0/srcAbsY0：该矩形左上角的绝对虚拟屏幕坐标（用于在 memDC 取源像素）； +// blockPx：马赛克块大小（逻辑像素）；srcDC/virtualX/virtualY/dpiScale：源参数。 +static void MosaicBlitRect(HDC targetDC, HDC srcDC, + int dstX0, int dstY0, int dstW, int dstH, + int srcAbsX0, int srcAbsY0, + int blockPx, int virtualX, int virtualY, double dpiScale) { + if (dstW <= 0 || dstH <= 0 || blockPx < 1) return; + + // 临时 1x1 DC：用作缩小缓冲（取块平均色） + HDC screenDC = GetDC(NULL); + if (!screenDC) return; + HDC onePxDC = CreateCompatibleDC(screenDC); + HBITMAP onePxBmp = CreateCompatibleBitmap(screenDC, 1, 1); + HGDIOBJ oldOne = SelectObject(onePxDC, onePxBmp); + ReleaseDC(NULL, screenDC); + + SetStretchBltMode(onePxDC, HALFTONE); + SetBrushOrgEx(onePxDC, 0, 0, NULL); + int oldTargetMode = GetStretchBltMode(targetDC); + SetStretchBltMode(targetDC, COLORONCOLOR); // 放大时取最近邻，保持块色纯净 + + // 按块遍历（逻辑像素步进） + for (int by = 0; by < dstH; by += blockPx) { + for (int bx = 0; bx < dstW; bx += blockPx) { + int dstBlockX = dstX0 + bx; + int dstBlockY = dstY0 + by; + int bw = (std::min)(blockPx, dstW - bx); + int bh = (std::min)(blockPx, dstH - by); + + // 源像素坐标（物理） + int srcAbsX = srcAbsX0 + bx; + int srcAbsY = srcAbsY0 + by; + int sx = (int)((srcAbsX - virtualX) * dpiScale + 0.5); + int sy = (int)((srcAbsY - virtualY) * dpiScale + 0.5); + int sw = (int)(bw * dpiScale + 0.5); + int sh = (int)(bh * dpiScale + 0.5); + if (sw < 1) sw = 1; + if (sh < 1) sh = 1; + + // 缩小到 1px（取整块平均色） + if (StretchBlt(onePxDC, 0, 0, 1, 1, srcDC, sx, sy, sw, sh, SRCCOPY)) { + // 放大回目标块 + StretchBlt(targetDC, dstBlockX, dstBlockY, bw, bh, + onePxDC, 0, 0, 1, 1, SRCCOPY); + } + } + } + SetStretchBltMode(targetDC, oldTargetMode); + + SelectObject(onePxDC, oldOne); + DeleteObject(onePxBmp); + DeleteDC(onePxDC); +} + + +// ==================== 马赛克渲染（reveal-mask 模型） ==================== +// 核心：预先把整张选区按当前块大小做一次完整马赛克化得到 mosaicBase（逻辑像素）。 +// 马赛克标注只是「蒙版」——涂抹=路径圆形区域、框选=矩形区域——揭示其背后的 mosaicBase。 +// 任意区域、任意顺序叠加都连续无缝；切换块大小只需重建 base，已揭示区域自动更新。 +// 不再对每个标注单独像素化，故无「松开后再处理一遍」的不连续感。 + +// 释放马赛克 base 资源 +static void FreeMosaicBase(CaptureContext* ctx) { + if (ctx->mosaicBaseDC) { DeleteDC(ctx->mosaicBaseDC); ctx->mosaicBaseDC = NULL; } + if (ctx->mosaicBaseBitmap) { DeleteObject(ctx->mosaicBaseBitmap); ctx->mosaicBaseBitmap = NULL; } + ctx->mosaicBaseW = 0; + ctx->mosaicBaseH = 0; + ctx->mosaicBaseBlockPx = 0; +} + +// 用 GDI+ 把单色位图转为带透明通道的 32bpp HBITMAP（用于光标）。 +static HBITMAP ColorBitmapFromBitmap(Gdiplus::Bitmap& bmp) { + HBITMAP hBmp = NULL; + Gdiplus::Color bg(0, 0, 0, 0); // 透明背景 + bmp.GetHBITMAP(bg, &hBmp); + return hBmp; +} + +// 生成单个涂抹光标：半径圆（白底 + 深色描边 + 中心十字）。 +// size 为光标位图边长（逻辑像素）；hotspot 在中心。 +static HCURSOR CreateMosaicBrushCursor(int radius) { + int pad = 3; + int size = (radius + pad) * 2; + if (size < 16) size = 16; + + Gdiplus::GdiplusStartupInput si; + ULONG_PTR token = 0; + HCURSOR result = NULL; + if (Gdiplus::GdiplusStartup(&token, &si, NULL) != Gdiplus::Ok) return NULL; + { + Gdiplus::Bitmap bmp(size, size, PixelFormat32bppARGB); + { + Gdiplus::Graphics g(&bmp); + g.SetSmoothingMode(Gdiplus::SmoothingModeAntiAlias); + int cx = size / 2; + int cy = size / 2; + // 外圈：白色描边底（保证暗背景可见） + Gdiplus::Pen whitePen(Gdiplus::Color(255, 255, 255, 255), 3.0f); + g.DrawEllipse(&whitePen, cx - radius, cy - radius, radius * 2, radius * 2); + // 内圈：深色虚线描边 + Gdiplus::Pen darkPen(Gdiplus::Color(255, 30, 30, 30), 1.5f); + darkPen.SetDashStyle(Gdiplus::DashStyleDash); + g.DrawEllipse(&darkPen, cx - radius, cy - radius, radius * 2, radius * 2); + // 中心十字（准星） + Gdiplus::Pen crossPen(Gdiplus::Color(255, 30, 30, 30), 1.0f); + int cl = (std::min)(6, radius); + g.DrawLine(&crossPen, cx - cl, cy, cx + cl, cy); + g.DrawLine(&crossPen, cx, cy - cl, cx, cy + cl); + } + HBITMAP hColor = ColorBitmapFromBitmap(bmp); + + // 掩码位图（全黑，使用彩色光标时掩码可忽略，但 CreateIcon 要求非空） + HDC screenDC = GetDC(NULL); + HDC maskDC = CreateCompatibleDC(screenDC); + HBITMAP hMask = CreateCompatibleBitmap(screenDC, size, size); + HGDIOBJ oldMask = SelectObject(maskDC, hMask); + PatBlt(maskDC, 0, 0, size, size, BLACKNESS); + SelectObject(maskDC, oldMask); + DeleteDC(maskDC); + ReleaseDC(NULL, screenDC); + + ICONINFO ii = {0}; + ii.fIcon = FALSE; + ii.xHotspot = size / 2; + ii.yHotspot = size / 2; + ii.hbmMask = hMask; + ii.hbmColor = hColor; + result = CreateIconIndirect(&ii); + DeleteObject(hMask); + DeleteObject(hColor); + } + Gdiplus::GdiplusShutdown(token); + return result; +} + +// 初始化涂抹光标缓存（按 DPI 缩放半径）。 +static void InitMosaicBrushCursors(CaptureContext* ctx) { + if (ctx->mosaicBrushCursorsInited) return; + for (int i = 0; i < SC_MOSAIC_RADIUS_COUNT; i++) { + ctx->mosaicBrushCursors[i] = CreateMosaicBrushCursor(SC_MOSAIC_RADIUS[i]); + } + ctx->mosaicBrushCursorsInited = true; +} + +static void FreeMosaicBrushCursors(CaptureContext* ctx) { + for (int i = 0; i < SC_MOSAIC_RADIUS_COUNT; i++) { + if (ctx->mosaicBrushCursors[i]) { DestroyIcon(ctx->mosaicBrushCursors[i]); ctx->mosaicBrushCursors[i] = NULL; } + } + ctx->mosaicBrushCursorsInited = false; +} + +// 生成马赛克 base：把整张虚拟屏幕原始底图按 blockPx 马赛克化，存为离屏位图。 +// base 用绝对（虚拟屏幕）坐标、尺寸 = virtualW×virtualH（与 backDC 一致），与选区无关。 +// 这样选区 resize/move 时 base 无需重建（标注蒙版用绝对坐标，任意选区下都正确对位）， +// 仅在块大小变化或初次生成时重建。代价是占一份全屏位图内存（与 backDC/memDC 同级）。 +static void RebuildMosaicBase(CaptureContext* ctx) { + int w = ctx->virtualW; + int h = ctx->virtualH; + if (w <= 0 || h <= 0) { FreeMosaicBase(ctx); return; } + + int blockPx = SC_MOSAIC_SIZES[ctx->mosaicSizeIdx]; + if (blockPx < 2) blockPx = 2; + + // 尺寸/块大小变化才重建位图 + if (w != ctx->mosaicBaseW || h != ctx->mosaicBaseH + || blockPx != ctx->mosaicBaseBlockPx || !ctx->mosaicBaseDC) { + FreeMosaicBase(ctx); + HDC screenDC = GetDC(NULL); + if (!screenDC) return; + ctx->mosaicBaseDC = CreateCompatibleDC(screenDC); + ctx->mosaicBaseBitmap = CreateCompatibleBitmap(screenDC, w, h); + ReleaseDC(NULL, screenDC); + if (!ctx->mosaicBaseDC || !ctx->mosaicBaseBitmap) { FreeMosaicBase(ctx); return; } + SelectObject(ctx->mosaicBaseDC, ctx->mosaicBaseBitmap); + ctx->mosaicBaseW = w; + ctx->mosaicBaseH = h; + ctx->mosaicBaseBlockPx = blockPx; + } + + // 整虚拟屏幕按 blockPx 马赛克化：base 原点 = 虚拟屏幕左上角，源绝对坐标 = virtualX/virtualY。 + MosaicBlitRect(ctx->mosaicBaseDC, ctx->memDC, 0, 0, w, h, + ctx->virtualX, ctx->virtualY, blockPx, + ctx->virtualX, ctx->virtualY, ctx->dpiScale); +} + +// 检查 base 是否需要重建（仅块大小变化 / 未生成）。 +// base 覆盖整屏且用绝对坐标，选区变化不影响 base，故 resize/move 无需重建。 +static bool MosaicBaseNeedsRebuild(const CaptureContext* ctx) { + int blockPx = SC_MOSAIC_SIZES[ctx->mosaicSizeIdx]; + if (blockPx < 2) blockPx = 2; + bool blockChanged = (blockPx != ctx->mosaicBaseBlockPx); + bool sizeChanged = (ctx->mosaicBaseW != ctx->virtualW || ctx->mosaicBaseH != ctx->virtualH); + return blockChanged || sizeChanged || !ctx->mosaicBaseDC; +} + +// 把单条马赛克标注对应的「蒙版区域」构建为 HRGN（选区相对坐标）。 +// ox/oy：绝对坐标 → 目标局部坐标的偏移（= -sel.left/-sel.top，因为 base/overlay 都是选区相对）。 +static HRGN BuildMosaicMaskRegion(const Annotation& a, float ox, float oy) { + if (a.mosaicRect) { + int absL = (std::min)(a.x1, a.x2); + int absT = (std::min)(a.y1, a.y2); + int absR = (std::max)(a.x1, a.x2); + int absB = (std::max)(a.y1, a.y2); + return CreateRectRgn((int)(absL + ox + 0.5f), (int)(absT + oy + 0.5f), + (int)(absR + ox + 0.5f), (int)(absB + oy + 0.5f)); + } else { + // 涂抹：把整条路径变为连续的「胶囊」区域（保证快速移动时不留空隙）。 + // 做法：沿相邻点之间的线段以不超过 radius/2 的步长插值取点，每个点画一个圆并并入区域， + // 相邻圆重叠从而形成无缝的粗笔触轨迹。 + int radius = a.brushRadius; + if (radius < 1) radius = 1; + HRGN rgn = CreateRectRgn(0, 0, 0, 0); + if (a.pts.empty()) return rgn; + // 步长：半径的一半，保证相邻圆重叠 ≥50%，无视觉缝隙 + double step = (std::max)(1.0, radius * 0.5); + + auto addCircle = [&](double cx, double cy) { + int ix = (int)(cx + 0.5); + int iy = (int)(cy + 0.5); + HRGN circle = CreateEllipticRgn(ix - radius, iy - radius, + ix + radius, iy + radius); + CombineRgn(rgn, rgn, circle, RGN_OR); + DeleteObject(circle); + }; + + // 第一个点 + addCircle(a.pts[0].x + ox, a.pts[0].y + oy); + for (size_t i = 1; i < a.pts.size(); i++) { + double x0 = a.pts[i - 1].x + ox; + double y0 = a.pts[i - 1].y + oy; + double x1 = a.pts[i].x + ox; + double y1 = a.pts[i].y + oy; + double dx = x1 - x0, dy = y1 - y0; + double segLen = std::sqrt(dx * dx + dy * dy); + if (segLen < 0.5) { + addCircle(x1, y1); + continue; + } + int n = (int)(segLen / step + 0.5); + if (n < 1) n = 1; + for (int k = 1; k <= n; k++) { + double t = (double)k / n; + addCircle(x0 + dx * t, y0 + dy * t); + } + } + return rgn; + } +} + +// 揭示马赛克：把 mosaicBase 中由 masks（已提交标注）+ curDrawing（正在绘制）覆盖的区域 +// BitBlt 到 targetDC（覆盖层 backDC）。 +// 全屏 base 用虚拟屏幕绝对坐标（原点=虚拟左上角，与 backDC 同坐标系）， +// 故 base 与 targetDC 1:1 对应，蒙版用绝对坐标（ox/oy=0）直接作为裁剪区，BitBlt 同位置拷贝。 +// ox/oy：标注坐标 → 目标局部坐标偏移（覆盖层=0；导出 finalDC 时=-rect.left/-rect.top）。 +static void RevealMosaicToTarget(HDC targetDC, HDC mosaicBase, + const std::vector& annotations, + const Annotation* curDrawing, + float ox, float oy) { + // 合并所有马赛克标注的蒙版区域（目标局部坐标） + HRGN mask = CreateRectRgn(0, 0, 0, 0); + bool any = false; + for (const Annotation& a : annotations) { + if (a.type != AT_Mosaic) continue; + HRGN r = BuildMosaicMaskRegion(a, ox, oy); + CombineRgn(mask, mask, r, RGN_OR); + DeleteObject(r); + any = true; + } + if (curDrawing && curDrawing->type == AT_Mosaic) { + HRGN r = BuildMosaicMaskRegion(*curDrawing, ox, oy); + CombineRgn(mask, mask, r, RGN_OR); + DeleteObject(r); + any = true; + } + if (any) { + SelectClipRgn(targetDC, mask); + // base 与 targetDC 同坐标系，1:1 拷贝 + BitBlt(targetDC, 0, 0, 0x7FFF, 0x7FFF, mosaicBase, 0, 0, SRCCOPY); + SelectClipRgn(targetDC, NULL); + } + DeleteObject(mask); +} + +// 覆盖层渲染矢量/文字标注（不含马赛克，马赛克由 reveal-mask 单独处理）。 +// selRel：选区在 backDC 局部坐标的矩形；标注为绝对虚拟屏幕坐标，偏移 = -virtualX/-virtualY。 +// 用 SetClip 限制到选区内部，避免画到遮罩区。 +static void DrawAnnotations(HDC hdc, const RECT& selRel, int virtualX, int virtualY, + const std::vector& annotations, + const Annotation* curDrawing) { + Gdiplus::GdiplusStartupInput startupInput; + ULONG_PTR gdipToken; + Gdiplus::GdiplusStartup(&gdipToken, &startupInput, NULL); + { + Gdiplus::Graphics graphics(hdc); + graphics.SetSmoothingMode(Gdiplus::SmoothingModeAntiAlias); + // 限制绘制范围在选区内 + Gdiplus::Rect clipRect(selRel.left, selRel.top, + selRel.right - selRel.left, + selRel.bottom - selRel.top); + graphics.SetClip(clipRect); + + float ox = (float)-virtualX; + float oy = (float)-virtualY; + + for (const Annotation& a : annotations) { + if (a.type == AT_Mosaic) continue; // 马赛克单独渲染 + DrawOneAnnotation(graphics, a, ox, oy); + } + if (curDrawing && curDrawing->type != AT_Mosaic) { + DrawOneAnnotation(graphics, *curDrawing, ox, oy); + } + } + Gdiplus::GdiplusShutdown(gdipToken); +} + + +// 合成标注进最终 PNG：finalDC 原点 = 选区左上角，故偏移 = -rect.left/-rect.top。 +// srcDC = memDC（原始屏幕位图，物理像素），用于马赛克像素化取源。 +// mosaicBlockPx：马赛克块大小（与编辑器当前全局块大小保持一致，保证导出与所见一致）。 +static void CompositeAnnotations(HDC finalDC, HDC srcDC, + const std::vector& annotations, + const RECT& rect, int virtualX, int virtualY, + double dpiScale, int mosaicBlockPx) { + if (annotations.empty()) return; + Gdiplus::GdiplusStartupInput startupInput; + ULONG_PTR gdipToken; + Gdiplus::GdiplusStartup(&gdipToken, &startupInput, NULL); + { + Gdiplus::Graphics graphics(finalDC); + graphics.SetSmoothingMode(Gdiplus::SmoothingModeAntiAlias); + float ox = (float)-rect.left; + float oy = (float)-rect.top; + for (const Annotation& a : annotations) { + if (a.type == AT_Mosaic) continue; // 马赛克单独渲染 + DrawOneAnnotation(graphics, a, ox, oy); + } + } + Gdiplus::GdiplusShutdown(gdipToken); + + // 马赛克标注单独渲染（reveal-mask 模型：生成整选区 base 再揭示蒙版区域） + bool hasMosaic = false; + for (const Annotation& a : annotations) { + if (a.type == AT_Mosaic) { hasMosaic = true; break; } + } + if (hasMosaic) { + int blockPx = mosaicBlockPx; + if (blockPx < 2) blockPx = 2; + int w = rect.right - rect.left; + int h = rect.bottom - rect.top; + HDC screenDC = GetDC(NULL); + if (screenDC) { + HDC baseDC = CreateCompatibleDC(screenDC); + HBITMAP baseBmp = CreateCompatibleBitmap(screenDC, w, h); + HGDIOBJ oldBase = SelectObject(baseDC, baseBmp); + MosaicBlitRect(baseDC, srcDC, 0, 0, w, h, + rect.left, rect.top, blockPx, virtualX, virtualY, dpiScale); + // 揭示蒙版区域（finalDC 原点 = 选区左上角，base 同为选区相对，ox=-rect.left） + float ox = (float)-rect.left; + float oy = (float)-rect.top; + RevealMosaicToTarget(finalDC, baseDC, annotations, nullptr, ox, oy); + SelectObject(baseDC, oldBase); + DeleteObject(baseBmp); + DeleteDC(baseDC); + ReleaseDC(NULL, screenDC); + } + } +} + +// 将 HBITMAP 转换为 PNG base64 字符串 +static std::string BitmapToBase64Png(HBITMAP hBitmap) { + Gdiplus::GdiplusStartupInput gdiplusStartupInput; + ULONG_PTR gdiplusToken; + Gdiplus::GdiplusStartup(&gdiplusToken, &gdiplusStartupInput, NULL); + + std::string result; + { + Gdiplus::Bitmap* bmp = Gdiplus::Bitmap::FromHBITMAP(hBitmap, NULL); + if (bmp) { + CLSID pngClsid; + if (GetPngEncoderClsid(&pngClsid) >= 0) { + IStream* stream = NULL; + CreateStreamOnHGlobal(NULL, TRUE, &stream); + if (stream && bmp->Save(stream, &pngClsid, NULL) == Gdiplus::Ok) { + HGLOBAL hMem = NULL; + GetHGlobalFromStream(stream, &hMem); + size_t len = GlobalSize(hMem); + BYTE* ptr = (BYTE*)GlobalLock(hMem); + if (ptr && len > 0) { + result = "data:image/png;base64," + Base64Encode(ptr, len); + } + GlobalUnlock(hMem); + } + if (stream) stream->Release(); + } + delete bmp; + } + } + Gdiplus::GdiplusShutdown(gdiplusToken); + return result; +} + +// 保存位图到剪贴板 +static bool SaveBitmapToClipboard(HBITMAP hBitmap) { + if (!OpenClipboard(NULL)) return false; + EmptyClipboard(); + BITMAP bm; + GetObject(hBitmap, sizeof(BITMAP), &bm); + HBITMAP hCopy = (HBITMAP)CopyImage(hBitmap, IMAGE_BITMAP, bm.bmWidth, bm.bmHeight, LR_COPYRETURNORG); + SetClipboardData(CF_BITMAP, hCopy); + CloseClipboard(); + return true; +} + +// 从预截屏位图提取区域，生成 base64 并复制到剪贴板。 +// anns：可选的标注列表，会合成进最终 PNG（选区相对坐标，finalDC 原点 = 选区原点）。 +static ScreenshotResult* ExtractRegionResult(HDC memDC, const RECT& rect, + int vx, int vy, double dpiScale, const std::vector& anns) { + ScreenshotResult* result = new ScreenshotResult(); + result->success = false; + int width = rect.right - rect.left; + int height = rect.bottom - rect.top; + result->x = rect.left; + result->y = rect.top; + result->x2 = rect.right; + result->y2 = rect.bottom; + result->width = width; + result->height = height; + + if (width <= 0 || height <= 0) return result; + + // 从物理尺寸位图提取区域 + int lx = rect.left - vx; + int ly = rect.top - vy; + int px = (int)(lx * dpiScale + 0.5); + int py = (int)(ly * dpiScale + 0.5); + int pw = (int)(width * dpiScale + 0.5); + int ph = (int)(height * dpiScale + 0.5); + + HDC screenDC = GetDC(NULL); + HDC regionDC = CreateCompatibleDC(screenDC); + HBITMAP regionBmp = CreateCompatibleBitmap(screenDC, pw, ph); + SelectObject(regionDC, regionBmp); + BitBlt(regionDC, 0, 0, pw, ph, memDC, px, py, SRCCOPY); + + // 如果有 DPI 缩放，缩放回逻辑尺寸 + HBITMAP finalBmp = regionBmp; + HDC finalDC = regionDC; + if (dpiScale > 1.01 || dpiScale < 0.99) { + HDC scaledDC = CreateCompatibleDC(screenDC); + HBITMAP scaledBmp = CreateCompatibleBitmap(screenDC, width, height); + SelectObject(scaledDC, scaledBmp); + SetStretchBltMode(scaledDC, HALFTONE); + SetBrushOrgEx(scaledDC, 0, 0, NULL); + StretchBlt(scaledDC, 0, 0, width, height, regionDC, 0, 0, pw, ph, SRCCOPY); + DeleteDC(regionDC); + DeleteObject(regionBmp); + finalBmp = scaledBmp; + finalDC = scaledDC; + } + + // 合成标注进最终图像（finalDC 原点 = 选区原点，标注为绝对坐标，偏移 = -rect.left/top） + CompositeAnnotations(finalDC, memDC, anns, rect, vx, vy, dpiScale, + SC_MOSAIC_SIZES[g_captureCtx ? g_captureCtx->mosaicSizeIdx : SC_DEFAULT_MOSAIC_IDX]); + + // 生成 base64 + result->base64 = BitmapToBase64Png(finalBmp); + // 复制到剪贴板 + result->success = SaveBitmapToClipboard(finalBmp); + + DeleteDC(finalDC); + DeleteObject(finalBmp); + ReleaseDC(NULL, screenDC); + + return result; +} + +// ---- 窗口过程和线程 ---- + +// 生成默认保存文件名：Screenshot_YYYYMMDD_HHMMSS.png +static std::wstring MakeDefaultScreenshotName() { + time_t now = time(NULL); + struct tm lt; + localtime_s(<, &now); + wchar_t buf[64]; + wsprintfW(buf, L"Screenshot_%04d%02d%02d_%02d%02d%02d.png", + lt.tm_year + 1900, lt.tm_mon + 1, lt.tm_mday, + lt.tm_hour, lt.tm_min, lt.tm_sec); + return std::wstring(buf); +} + +// 弹出系统保存对话框，返回用户选择的文件完整路径（含 .png 后缀）； +// 用户取消或失败时返回空字符串。 +// hwndOwner：父窗口句柄（截图覆盖层），用于模态居中。 +// 注意：覆盖层是 WS_EX_TOPMOST 全屏窗口，通用对话框可能被遮挡。 +// 弹出前临时移除其 TOPMOST（让对话框自然置顶），关闭后恢复，保证对话框可见可交互。 +static std::wstring PromptSaveFilePath(HWND hwndOwner) { + // 默认目录：图片库（FOLDERID_Pictures），获取失败则退化为桌面 + wchar_t* defaultDir = nullptr; + HRESULT hr = SHGetKnownFolderPath(FOLDERID_Pictures, 0, NULL, &defaultDir); + std::wstring initDir; + if (SUCCEEDED(hr) && defaultDir) { + initDir = defaultDir; + CoTaskMemFree(defaultDir); + } else { + wchar_t* desktop = nullptr; + if (SUCCEEDED(SHGetKnownFolderPath(FOLDERID_Desktop, 0, NULL, &desktop)) && desktop) { + initDir = desktop; + CoTaskMemFree(desktop); + } + } + + std::wstring defaultName = MakeDefaultScreenshotName(); + + wchar_t fileBuf[MAX_PATH] = {0}; + wcsncpy_s(fileBuf, MAX_PATH, defaultName.c_str(), _TRUNCATE); + + // 临时取消覆盖层 TOPMOST，确保保存对话框显示在最上层 + if (hwndOwner) { + SetWindowPos(hwndOwner, HWND_NOTOPMOST, 0, 0, 0, 0, + SWP_NOMOVE | SWP_NOSIZE | SWP_NOACTIVATE); + } + + OPENFILENAMEW ofn = {0}; + ofn.lStructSize = sizeof(OPENFILENAMEW); + ofn.hwndOwner = hwndOwner; + ofn.lpstrFile = fileBuf; + ofn.nMaxFile = MAX_PATH; + ofn.lpstrFilter = L"PNG 图像 (*.png)\0*.png\0"; + ofn.nFilterIndex = 1; + ofn.lpstrDefExt = L"png"; // 用户未输扩展名时自动补 .png + if (!initDir.empty()) { + ofn.lpstrInitialDir = initDir.c_str(); + } + ofn.Flags = OFN_OVERWRITEPROMPT | OFN_PATHMUSTEXIST | OFN_HIDEREADONLY + | OFN_NOCHANGEDIR; + + BOOL ok = GetSaveFileNameW(&ofn); + + // 恢复覆盖层 TOPMOST（用户取消保存时需要回到置顶全屏状态） + if (hwndOwner) { + SetWindowPos(hwndOwner, HWND_TOPMOST, 0, 0, 0, 0, + SWP_NOMOVE | SWP_NOSIZE | SWP_NOACTIVATE); + } + + if (ok) { + return std::wstring(fileBuf); + } + return std::wstring(); +} + +// 将已合成标注的选区 HBITMAP 保存为 PNG 文件。 +// 返回 true 表示保存成功。 +static bool SaveRegionToPngFile(HDC memDC, const RECT& rect, int vx, int vy, + double dpiScale, const std::vector& anns, + const std::wstring& filePath) { + int width = rect.right - rect.left; + int height = rect.bottom - rect.top; + if (width <= 0 || height <= 0 || filePath.empty()) return false; + + // 复用 ExtractRegionResult 的合成逻辑生成 finalBmp（含标注、按逻辑尺寸） + int lx = rect.left - vx; + int ly = rect.top - vy; + int px = (int)(lx * dpiScale + 0.5); + int py = (int)(ly * dpiScale + 0.5); + int pw = (int)(width * dpiScale + 0.5); + int ph = (int)(height * dpiScale + 0.5); + + HDC screenDC = GetDC(NULL); + HDC regionDC = CreateCompatibleDC(screenDC); + HBITMAP regionBmp = CreateCompatibleBitmap(screenDC, pw, ph); + SelectObject(regionDC, regionBmp); + BitBlt(regionDC, 0, 0, pw, ph, memDC, px, py, SRCCOPY); + + HBITMAP finalBmp = regionBmp; + HDC finalDC = regionDC; + if (dpiScale > 1.01 || dpiScale < 0.99) { + HDC scaledDC = CreateCompatibleDC(screenDC); + HBITMAP scaledBmp = CreateCompatibleBitmap(screenDC, width, height); + SelectObject(scaledDC, scaledBmp); + SetStretchBltMode(scaledDC, HALFTONE); + SetBrushOrgEx(scaledDC, 0, 0, NULL); + StretchBlt(scaledDC, 0, 0, width, height, regionDC, 0, 0, pw, ph, SRCCOPY); + DeleteDC(regionDC); + DeleteObject(regionBmp); + finalBmp = scaledBmp; + finalDC = scaledDC; + } + CompositeAnnotations(finalDC, memDC, anns, rect, vx, vy, dpiScale, + SC_MOSAIC_SIZES[g_captureCtx ? g_captureCtx->mosaicSizeIdx : SC_DEFAULT_MOSAIC_IDX]); + + // 用 GDI+ 保存为 PNG 文件 + bool ok = false; + Gdiplus::GdiplusStartupInput startupInput; + ULONG_PTR token = 0; + if (Gdiplus::GdiplusStartup(&token, &startupInput, NULL) == Gdiplus::Ok) { + { + Gdiplus::Bitmap* bmp = Gdiplus::Bitmap::FromHBITMAP(finalBmp, NULL); + if (bmp) { + CLSID pngClsid; + if (GetPngEncoderClsid(&pngClsid) >= 0) { + ok = (bmp->Save(filePath.c_str(), &pngClsid, NULL) == Gdiplus::Ok); + } + delete bmp; + } + } + Gdiplus::GdiplusShutdown(token); + } + + DeleteDC(finalDC); + DeleteObject(finalBmp); + ReleaseDC(NULL, screenDC); + return ok; +} + +// 在主线程调用 JS 回调（截图完成） +static void CallScreenshotJs(napi_env env, napi_value js_callback, void* context, void* data) { + if (env != nullptr && js_callback != nullptr && data != nullptr) { + ScreenshotResult* result = static_cast(data); + + napi_value resultObj; + napi_create_object(env, &resultObj); + + napi_value success; + napi_get_boolean(env, result->success, &success); + napi_set_named_property(env, resultObj, "success", success); + + if (result->success) { + napi_value x, y, x2, y2, width, height, base64; + napi_create_int32(env, result->x, &x); + napi_set_named_property(env, resultObj, "x", x); + napi_create_int32(env, result->y, &y); + napi_set_named_property(env, resultObj, "y", y); + napi_create_int32(env, result->x2, &x2); + napi_set_named_property(env, resultObj, "x2", x2); + napi_create_int32(env, result->y2, &y2); + napi_set_named_property(env, resultObj, "y2", y2); + napi_create_int32(env, result->width, &width); + napi_set_named_property(env, resultObj, "width", width); + napi_create_int32(env, result->height, &height); + napi_set_named_property(env, resultObj, "height", height); + napi_create_string_utf8(env, result->base64.c_str(), result->base64.size(), &base64); + napi_set_named_property(env, resultObj, "base64", base64); + } + + napi_value global; + napi_get_global(env, &global); + napi_call_function(env, global, js_callback, 1, &resultObj, nullptr); + delete result; + } +} + +// 进入确认态：规范化选区并切换状态 +static void EnterConfirmed(CaptureContext* ctx, const RECT& sel) { + RECT n = NormalizeRect(sel); + // 约束到虚拟屏幕内 + if (n.left < ctx->virtualX) { n.right += ctx->virtualX - n.left; n.left = ctx->virtualX; } + if (n.top < ctx->virtualY) { n.bottom += ctx->virtualY - n.top; n.top = ctx->virtualY; } + if (n.right > ctx->virtualX + ctx->virtualW) n.right = ctx->virtualX + ctx->virtualW; + if (n.bottom > ctx->virtualY + ctx->virtualH) n.bottom = ctx->virtualY + ctx->virtualH; + // 最小尺寸保护 + if (n.right - n.left < SC_MIN_SELECTION) n.right = n.left + SC_MIN_SELECTION; + if (n.bottom - n.top < SC_MIN_SELECTION) n.bottom = n.top + SC_MIN_SELECTION; + ctx->selection = n; + ctx->resizeHandle = RH_None; + ctx->state = CS_Confirmed; + ctx->needFullRedraw = true; +} + +// 计算所有标注内容的包围盒（选区相对逻辑坐标）。 +// 用于限制选区缩放：选区不可缩小到裁掉已添加内容。 +// 返回 false 表示无标注（无约束）。 +// 计算所有标注内容的包围盒（绝对虚拟屏幕坐标）。 +// 用于限制选区缩放：选区不可缩小到裁掉已添加内容。 +// 返回 false 表示无标注（无约束）。hdc 用于测量文字宽高。 +// 前置声明：文字包围盒复用 MeasureTextAnnotation（定义在下方）。 +static RECT MeasureTextAnnotation(HDC hdc, const Annotation& a); +static bool CalcAnnotationsBounds(const std::vector& anns, RECT& out, HDC hdc) { + if (anns.empty()) return false; + int minL = INT_MAX, minT = INT_MAX, maxR = INT_MIN, maxB = INT_MIN; + auto expand = [&](int x, int y) { + if (x < minL) minL = x; + if (y < minT) minT = y; + if (x > maxR) maxR = x; + if (y > maxB) maxB = y; + }; + for (const Annotation& a : anns) { + if (a.type == AT_Brush) { + for (const POINT& p : a.pts) expand(p.x, p.y); + } else if (a.type == AT_Mosaic) { + if (a.mosaicRect) { + // 框选模式：矩形角点 + expand(a.x1, a.y1); + expand(a.x2, a.y2); + } else { + // 涂抹模式：路径包围盒 + 半径 + int r = a.brushRadius; + for (const POINT& p : a.pts) { + expand(p.x - r, p.y - r); + expand(p.x + r, p.y + r); + } + } + } else if (a.type == AT_Text) { + // 复用 MeasureTextAnnotation：与选中时的外边框（含 padding）完全一致， + // 保证 resize 选区时文字包围盒约束 = 视觉选中边框，不会被裁掉。 + RECT r = MeasureTextAnnotation(hdc, a); + expand(r.left, r.top); + expand(r.right, r.bottom); + } else { + expand(a.x1, a.y1); + expand(a.x2, a.y2); + } + } + if (minL == INT_MAX) return false; + out.left = minL; + out.top = minT; + out.right = maxR; + out.bottom = maxB; + return true; +} + +// 测量文字标注的包围盒（绝对虚拟屏幕坐标） +// 用 GDI+ MeasureString 测量，与提交态 DrawString 渲染同源： +// - 字形左上角相对锚点 (x1,y1) 有内部偏移 (offX, offY) +// - 宽高为紧凑字形范围（不含 GDI GetTextExtentPoint32W 的尾部 overhang） +// - 边框左右 padding 对称，完整包住文字 +// 注意：GDI+ 对象须在 GdiplusShutdown 之前析构（内层 {} 保证）。 +static RECT MeasureTextAnnotation(HDC hdc, const Annotation& a) { + RECT rect = { a.x1, a.y1, a.x1, a.y1 }; + if (a.type != AT_Text || a.text.empty()) return rect; + + int fontPx = a.thickness; + if (fontPx < 8) fontPx = 8; + + float offX = 0, offY = 0, w = 0, h = 0; + Gdiplus::GdiplusStartupInput startupInput; + ULONG_PTR token = 0; + if (Gdiplus::GdiplusStartup(&token, &startupInput, NULL) == Gdiplus::Ok) { + { + Gdiplus::Graphics graphics(hdc); + graphics.SetTextRenderingHint(Gdiplus::TextRenderingHintAntiAlias); + Gdiplus::FontFamily fontFamily(SC_FONT_FACE); + Gdiplus::Font font(&fontFamily, (Gdiplus::REAL)fontPx, Gdiplus::FontStyleRegular, + Gdiplus::UnitPixel); + Gdiplus::StringFormat sf; + sf.SetAlignment(Gdiplus::StringAlignmentNear); + sf.SetLineAlignment(Gdiplus::StringAlignmentNear); + Gdiplus::RectF origin(0, 0, 0, 0); + Gdiplus::RectF bounds; + graphics.MeasureString(a.text.c_str(), (INT)a.text.size(), &font, origin, &sf, &bounds); + offX = bounds.X; + offY = bounds.Y; + w = bounds.Width; + h = bounds.Height; + } + Gdiplus::GdiplusShutdown(token); + } + + const int padding = 4; + // 字形左上角（绝对坐标）= 锚点 + GDI+ 内部偏移 + int glyphLeft = a.x1 + (int)floorf(offX); + int glyphTop = a.y1 + (int)floorf(offY); + int glyphRight = a.x1 + (int)ceilf(offX + w); + int glyphBottom = a.y1 + (int)ceilf(offY + h); + rect.left = glyphLeft - padding; + rect.top = glyphTop - padding; + rect.right = glyphRight + padding; + rect.bottom = glyphBottom + padding; + return rect; +} + +// 命中测试文字标注，返回标注索引（-1 表示未命中） +static int HitTestTextAnnotations(const std::vector& anns, int x, int y, HDC hdc) { + for (int i = (int)anns.size() - 1; i >= 0; i--) { + if (anns[i].type == AT_Text) { + RECT rect = MeasureTextAnnotation(hdc, anns[i]); + if (PointInRect(x, y, rect)) { + return i; + } + } + } + return -1; +} + +// ==================== GDI+ 文字测量（与提交态渲染同源） ==================== +// 提交态文字用 GDI+ DrawString 渲染（StringAlignmentNear 顶部左对齐）。 +// 为保证输入态边框/光标/命中与提交态视觉一致，输入态也必须用 GDI+ 测量。 +// 旧实现用 GDI GetTextExtentPoint32W，其宽度含尾部 overhang（右侧多出留白）， +// 且垂直基线与 GDI+ 不同（导致文字偏靠下），故统一改用 GDI+。 + +// 测量文字的紧凑包围盒：返回 DrawString(StringAlignmentNear/Near) 下字形相对锚点的偏移与尺寸。 +// outOffsetX/outOffsetY：字形左上角相对锚点 (0,0) 的偏移（通常 X≈0，Y 为字体内部顶部 leading）。 +// outW/outH：字形紧凑宽高。 +// 这样边框 = 锚点 + offset ± padding，可左右对称、紧贴字形。 +static void MeasureTextGdip(HDC hdc, const std::wstring& text, int fontPx, + float& outOffsetX, float& outOffsetY, float& outW, float& outH) { + outOffsetX = 0; outOffsetY = 0; outW = 0; outH = 0; + if (text.empty()) { + outH = (float)fontPx; + return; + } + Gdiplus::GdiplusStartupInput startupInput; + ULONG_PTR token = 0; + if (Gdiplus::GdiplusStartup(&token, &startupInput, NULL) != Gdiplus::Ok) return; + { + Gdiplus::Graphics graphics(hdc); + graphics.SetTextRenderingHint(Gdiplus::TextRenderingHintAntiAlias); + Gdiplus::FontFamily fontFamily(SC_FONT_FACE); + Gdiplus::Font font(&fontFamily, (Gdiplus::REAL)fontPx, Gdiplus::FontStyleRegular, + Gdiplus::UnitPixel); + Gdiplus::StringFormat sf; + sf.SetAlignment(Gdiplus::StringAlignmentNear); + sf.SetLineAlignment(Gdiplus::StringAlignmentNear); + // MeasureString 返回与 DrawString 一致的布局包围盒（含 GDI+ 的字体内部间距）。 + Gdiplus::RectF origin(0, 0, 0, 0); + Gdiplus::RectF bounds; + graphics.MeasureString(text.c_str(), (INT)text.size(), &font, origin, &sf, &bounds); + outOffsetX = bounds.X; + outOffsetY = bounds.Y; + outW = bounds.Width; + outH = bounds.Height; + } + Gdiplus::GdiplusShutdown(token); +} + +// 测量逐字符累计宽度（用于光标定位/选中高亮）。 +// widths[i] 表示前 i 个字符（text[0..i-1]）的累计紧凑宽度，widths[0]=0。 +// 与 DrawString 渲染进度一致，光标 x = textX + widths[i]。 +static void MeasureCharWidthsGdip(HDC hdc, const std::wstring& text, int fontPx, + std::vector& widths) { + widths.assign(text.size() + 1, 0.0f); + if (text.empty()) return; + Gdiplus::GdiplusStartupInput startupInput; + ULONG_PTR token = 0; + if (Gdiplus::GdiplusStartup(&token, &startupInput, NULL) != Gdiplus::Ok) return; + { + Gdiplus::Graphics graphics(hdc); + graphics.SetTextRenderingHint(Gdiplus::TextRenderingHintAntiAlias); + Gdiplus::FontFamily fontFamily(SC_FONT_FACE); + Gdiplus::Font font(&fontFamily, (Gdiplus::REAL)fontPx, Gdiplus::FontStyleRegular, + Gdiplus::UnitPixel); + Gdiplus::StringFormat sf; + sf.SetAlignment(Gdiplus::StringAlignmentNear); + sf.SetLineAlignment(Gdiplus::StringAlignmentNear); + Gdiplus::RectF origin(0, 0, 0, 0); + Gdiplus::RectF bounds; + std::wstring sub; + sub.reserve(text.size()); + for (size_t i = 1; i <= text.size(); i++) { + sub.assign(text, 0, i); + graphics.MeasureString(sub.c_str(), (INT)sub.size(), &font, origin, &sf, &bounds); + widths[i] = bounds.X + bounds.Width; + } + } + Gdiplus::GdiplusShutdown(token); +} + +// 根据鼠标位置计算光标在文字中的位置（字符索引） +// 用 GDI+ 逐字符测量，与 DrawString 渲染进度一致。 +static int CalcCaretPosFromMouse(HDC hdc, const std::wstring& text, int fontPx, int textX, int mouseX) { + if (text.empty()) return 0; + + std::vector widths; + MeasureCharWidthsGdip(hdc, text, fontPx, widths); + + int bestPos = 0; + float bestDist = FLT_MAX; + for (size_t i = 0; i <= text.size(); i++) { + float x = (float)textX + widths[i]; + float dist = fabsf(x - (float)mouseX); + if (dist < bestDist) { + bestDist = dist; + bestPos = (int)i; + } + } + return bestPos; +} + +// 截图覆盖层窗口过程（双缓冲渲染） +static LRESULT CALLBACK ScreenshotOverlayWndProc(HWND hwnd, UINT msg, WPARAM wParam, LPARAM lParam) { + CaptureContext* ctx = g_captureCtx; + if (!ctx) return DefWindowProc(hwnd, msg, wParam, lParam); + + switch (msg) { + case WM_PAINT: { + PAINTSTRUCT ps; + HDC hdc = BeginPaint(hwnd, &ps); + HDC backDC = ctx->backDC; + + // 计算浮窗位置 + int panelX, panelY; + CalcPanelPosition(ctx->mouseX, ctx->mouseY, + ctx->virtualX, ctx->virtualY, ctx->virtualW, ctx->virtualH, panelX, panelY); + // 转为相对坐标 + int panelXRel = panelX - ctx->virtualX; + int panelYRel = panelY - ctx->virtualY; + + RECT curPanelRect = { panelXRel, panelYRel, + panelXRel + SC_PANEL_WIDTH, panelYRel + SC_PANEL_HEIGHT }; + + // 当前选区矩形 + RECT curSelRect = {0,0,0,0}; + if (ctx->state == CS_Selecting) { + curSelRect.left = (std::min)(ctx->startX, ctx->endX) - ctx->virtualX; + curSelRect.top = (std::min)(ctx->startY, ctx->endY) - ctx->virtualY; + curSelRect.right = (std::max)(ctx->startX, ctx->endX) - ctx->virtualX; + curSelRect.bottom = (std::max)(ctx->startY, ctx->endY) - ctx->virtualY; + } else if (ctx->state == CS_Confirmed || ctx->state == CS_Resizing + || ctx->state == CS_Moving || ctx->state == CS_Drawing + || ctx->state == CS_TextEditing) { + curSelRect.left = ctx->selection.left - ctx->virtualX; + curSelRect.top = ctx->selection.top - ctx->virtualY; + curSelRect.right = ctx->selection.right - ctx->virtualX; + curSelRect.bottom = ctx->selection.bottom - ctx->virtualY; + } + + // 当前高亮窗口矩形 + RECT curHlRect = {0,0,0,0}; + if (ctx->state == CS_Idle && ctx->hoveredWindow >= 0 && ctx->hoveredWindow < (int)ctx->windows.size()) { + const RECT& wr = ctx->windows[ctx->hoveredWindow].rect; + curHlRect = { wr.left - ctx->virtualX, wr.top - ctx->virtualY, + wr.right - ctx->virtualX, wr.bottom - ctx->virtualY }; + } + + double ds = ctx->dpiScale; + int physW = (int)(ctx->virtualW * ds + 0.5); + int physH = (int)(ctx->virtualH * ds + 0.5); + + // 选区外遮罩覆盖整个虚拟屏幕，选区每次移动都会改变外部遮罩边界， + // 此时局部脏区域优化失效，必须整屏恢复背景 + 重绘遮罩。 + if (ctx->state == CS_Selecting || ctx->state == CS_Confirmed || + ctx->state == CS_Resizing || ctx->state == CS_Moving || + ctx->state == CS_Drawing || ctx->state == CS_TextEditing) { + ctx->needFullRedraw = true; + } + + // 恢复背景 + if (ctx->needFullRedraw) { + if (ds > 1.01 || ds < 0.99) { + StretchBlt(backDC, 0, 0, ctx->virtualW, ctx->virtualH, + ctx->memDC, 0, 0, physW, physH, SRCCOPY); + } else { + BitBlt(backDC, 0, 0, ctx->virtualW, ctx->virtualH, + ctx->memDC, 0, 0, SRCCOPY); + } + ctx->needFullRedraw = false; + } else { + // 脏区域恢复 + RestoreDirtyRegion(backDC, ctx->memDC, InflateRectBy(ctx->lastPanelRect, 2), ds); + if (ctx->lastSelectionRect.right > ctx->lastSelectionRect.left) + RestoreDirtyRegion(backDC, ctx->memDC, InflateRectBy(ctx->lastSelectionRect, 5), ds); + if (ctx->lastLabelRect.right > ctx->lastLabelRect.left) + RestoreDirtyRegion(backDC, ctx->memDC, InflateRectBy(ctx->lastLabelRect, 2), ds); + if (ctx->lastHighlightRect.right > ctx->lastHighlightRect.left) + RestoreDirtyRegion(backDC, ctx->memDC, InflateRectBy(ctx->lastHighlightRect, 5), ds); + if (ctx->lastToolbarRect.right > ctx->lastToolbarRect.left) + RestoreDirtyRegion(backDC, ctx->memDC, InflateRectBy(ctx->lastToolbarRect, 2), ds); + if (ctx->lastPopupRect.right > ctx->lastPopupRect.left) + RestoreDirtyRegion(backDC, ctx->memDC, InflateRectBy(ctx->lastPopupRect, 2), ds); + } + + // 绘制窗口高亮（Idle 状态） + if (ctx->state == CS_Idle) { + if (ctx->hoveredWindow >= 0 && ctx->hoveredWindow < (int)ctx->windows.size()) { + // 高亮悬停的窗口 + DrawWindowHighlight(backDC, ctx->windows[ctx->hoveredWindow].rect, + ctx->virtualX, ctx->virtualY, ctx->gdi); + } else { + // 没有匹配到窗口时，高亮鼠标所在的屏幕 + POINT pt = { ctx->mouseX, ctx->mouseY }; + HMONITOR hMonitor = MonitorFromPoint(pt, MONITOR_DEFAULTTONEAREST); + if (hMonitor) { + MONITORINFO monitorInfo; + monitorInfo.cbSize = sizeof(MONITORINFO); + if (GetMonitorInfo(hMonitor, &monitorInfo)) { + DrawWindowHighlight(backDC, monitorInfo.rcMonitor, + ctx->virtualX, ctx->virtualY, ctx->gdi); + } + } + } + } + + // ---- 确认/调整/移动/绘制态：遮罩 + 边框 + 手柄 + 标注 + 工具栏 ---- + RECT curLabelRect = {0,0,0,0}; + RECT curToolbarRect = {0,0,0,0}; + RECT curPopupRect = {0,0,0,0}; + bool confirmedMode = (ctx->state == CS_Confirmed || ctx->state == CS_Resizing + || ctx->state == CS_Moving || ctx->state == CS_Drawing + || ctx->state == CS_TextEditing); + if (confirmedMode) { + // 选区外遮罩（选区内部保持清晰） + DrawDimMask(backDC, ctx->gdi, + curSelRect.left, curSelRect.top, curSelRect.right, curSelRect.bottom, + ctx->virtualW, ctx->virtualH); + // 确认态边框 + DrawConfirmedBorder(backDC, curSelRect, ctx->gdi); + // 调整手柄（拖拽选区/调整选区/文字编辑时不绘制，避免遮挡；确认态/绘制标注时绘制） + if (ctx->state == CS_Confirmed || ctx->state == CS_Drawing) { + DrawResizeHandles(backDC, curSelRect, ctx->gdi); + } + // 已提交标注 + 正在绘制的标注（绘制范围 clip 在选区内） + // 调整选区时也保持显示，便于看清内容是否会被裁掉。 + if (ctx->state == CS_Confirmed || ctx->state == CS_Drawing || ctx->state == CS_Resizing) { + const Annotation* cur = ctx->hasCurDrawing ? &ctx->curDrawing : nullptr; + DrawAnnotations(backDC, curSelRect, ctx->virtualX, ctx->virtualY, ctx->annotations, cur); + // 马赛克（reveal-mask 模型）：先确保 base（整选区马赛克）已生成， + // 再把所有马赛克标注（含正在绘制的）的蒙版区域从 base 揭示到 backDC。 + // base 预计算后每帧只做带区域裁剪的 BitBlt，无逐标注像素化，连续无闪烁。 + if (MosaicBaseNeedsRebuild(ctx)) RebuildMosaicBase(ctx); + if (ctx->mosaicBaseDC) { + // base 与 backDC 同为虚拟屏幕绝对坐标，蒙版 ox/oy = 0 + RevealMosaicToTarget(backDC, ctx->mosaicBaseDC, + ctx->annotations, cur, 0.0f, 0.0f); + } + // 正在拖拽的矩形马赛克：叠加虚线边框提示当前框选范围（backDC 绝对坐标）。 + // 涂抹模式不画矩形边框（其范围由预览圆体现）。 + if (ctx->state == CS_Drawing && ctx->hasCurDrawing + && ctx->curDrawing.type == AT_Mosaic && ctx->curDrawing.mosaicRect) { + int rx1 = (std::min)(ctx->curDrawing.x1, ctx->curDrawing.x2); + int ry1 = (std::min)(ctx->curDrawing.y1, ctx->curDrawing.y2); + int rx2 = (std::max)(ctx->curDrawing.x1, ctx->curDrawing.x2); + int ry2 = (std::max)(ctx->curDrawing.y1, ctx->curDrawing.y2); + Gdiplus::GdiplusStartupInput startupInput; + ULONG_PTR bToken = 0; + if (Gdiplus::GdiplusStartup(&bToken, &startupInput, NULL) == Gdiplus::Ok) { + { + Gdiplus::Graphics graphics(backDC); + graphics.SetSmoothingMode(Gdiplus::SmoothingModeAntiAlias); + // 虚线笔：白色底 + 深色虚线，保证在任意背景上可见 + Gdiplus::Pen whitePen(Gdiplus::Color(255, 255, 255, 255), 3.0f); + graphics.DrawRectangle(&whitePen, (float)rx1, (float)ry1, + (float)(rx2 - rx1), (float)(ry2 - ry1)); + Gdiplus::Pen dashPen(Gdiplus::Color(255, 0x1E, 0x88, 0xE5), 1.5f); + dashPen.SetDashStyle(Gdiplus::DashStyleDash); + graphics.DrawRectangle(&dashPen, (float)rx1, (float)ry1, + (float)(rx2 - rx1), (float)(ry2 - ry1)); + } + Gdiplus::GdiplusShutdown(bToken); + } + } + } + // 文字编辑态：绘制输入光标和选中文字标注的边框 + if (ctx->state == CS_TextEditing) { + // 绘制已提交的标注 + DrawAnnotations(backDC, curSelRect, ctx->virtualX, ctx->virtualY, ctx->annotations, nullptr); + if (MosaicBaseNeedsRebuild(ctx)) RebuildMosaicBase(ctx); + if (ctx->mosaicBaseDC) { + RevealMosaicToTarget(backDC, ctx->mosaicBaseDC, + ctx->annotations, nullptr, 0.0f, 0.0f); + } + + // 绘制当前输入的文字和光标（统一用 GDI+，与提交态 DrawString 完全一致， + // 避免旧 GDI TextOutW 导致的文字偏靠下、右侧间距偏大的问题） + // 注意：测量与绘制共用同一个 GDI+ Graphics 作用域，避免在 backDC 上反复 + // Startup/Shutdown 及创建多个 Graphics 对象引发的状态混乱/崩溃。 + int fontPx = SC_FONT_SIZES[ctx->fontSizeIdx]; + COLORREF textColor = SC_COLOR_PRESETS[ctx->drawColorIdx]; + + // 文字锚点转换为相对坐标 + int textX = ctx->textAnchorX - ctx->virtualX; + int textY = ctx->textAnchorY - ctx->virtualY; + + // 单次 GDI+ 启动：完成测量（整体包围盒 + 逐字符宽度）与文字绘制 + // 关键：所有 GDI+ 对象（Graphics/Font/Brush 等）必须在 GdiplusShutdown 之前析构， + // 否则对象析构时访问已关闭的 GDI+ 内部状态会崩溃。故用内层 {} 包住对象生命周期， + // Shutdown 放在 } 之后（与 DrawAnnotations 的正确模式一致）。 + float offX = 0, offY = 0, textW = 0, textH = 0; + std::vector charWidths; + Gdiplus::GdiplusStartupInput startupInput; + ULONG_PTR gdipToken = 0; + if (Gdiplus::GdiplusStartup(&gdipToken, &startupInput, NULL) == Gdiplus::Ok) { + { + Gdiplus::Graphics graphics(backDC); + graphics.SetTextRenderingHint(Gdiplus::TextRenderingHintAntiAlias); + Gdiplus::FontFamily fontFamily(SC_FONT_FACE); + Gdiplus::Font font(&fontFamily, (Gdiplus::REAL)fontPx, + Gdiplus::FontStyleRegular, Gdiplus::UnitPixel); + Gdiplus::StringFormat sf; + sf.SetAlignment(Gdiplus::StringAlignmentNear); + sf.SetLineAlignment(Gdiplus::StringAlignmentNear); + + // 整体紧凑包围盒 + Gdiplus::RectF origin(0, 0, 0, 0); + Gdiplus::RectF bounds; + if (!ctx->textBuf.empty()) { + graphics.MeasureString(ctx->textBuf.c_str(), (INT)ctx->textBuf.size(), + &font, origin, &sf, &bounds); + offX = bounds.X; + offY = bounds.Y; + textW = bounds.Width; + textH = bounds.Height; + } else { + textH = (float)fontPx; + } + // 逐字符累计宽度（与 DrawString 渲染进度一致） + charWidths.assign(ctx->textBuf.size() + 1, 0.0f); + if (!ctx->textBuf.empty()) { + std::wstring sub; + sub.reserve(ctx->textBuf.size()); + for (size_t i = 1; i <= ctx->textBuf.size(); i++) { + sub.assign(ctx->textBuf, 0, i); + graphics.MeasureString(sub.c_str(), (INT)sub.size(), + &font, origin, &sf, &bounds); + charWidths[i] = bounds.X + bounds.Width; + } + } + + // 绘制文字（与提交态一致：顶部左对齐到锚点） + if (!ctx->textBuf.empty()) { + Gdiplus::SolidBrush textBrush(Gdiplus::Color(GetRValue(textColor), + GetGValue(textColor), + GetBValue(textColor))); + Gdiplus::RectF layoutRect((float)textX, (float)textY, 10000.0f, + (float)(fontPx * 2)); + graphics.DrawString(ctx->textBuf.c_str(), -1, &font, layoutRect, + &sf, &textBrush); + } + } + Gdiplus::GdiplusShutdown(gdipToken); + } + + // 字形可见区域的左上角（含 GDI+ 内部偏移） + float glyphLeft = (float)textX + offX; + float glyphTop = (float)textY + offY; + // 文字可见宽度（最小给一个占位宽度，空输入框也有合理大小） + float glyphW = (textW > 20.0f || !ctx->textBuf.empty()) ? textW : 20.0f; + float glyphH = (textH > 0 ? textH : (float)fontPx); + + // 绘制边框：左右对称 padding，上下基于字形紧凑高度。 + // 用字形可见区域 + padding，保证左侧与右侧间距一致，并完整包住文字。 + const float padding = 4.0f; + int boxLeft = (int)floorf(glyphLeft - padding); + int boxTop = (int)floorf(glyphTop - padding); + int boxRight = (int)ceilf(glyphLeft + glyphW + padding); + int boxBottom = (int)ceilf(glyphTop + glyphH + padding); + HPEN boxPen = CreatePen(PS_SOLID, 1, textColor); + HGDIOBJ oldPen2 = SelectObject(backDC, boxPen); + HGDIOBJ oldBrush2 = SelectObject(backDC, GetStockObject(NULL_BRUSH)); + Rectangle(backDC, boxLeft, boxTop, boxRight, boxBottom); + SelectObject(backDC, oldBrush2); + SelectObject(backDC, oldPen2); + DeleteObject(boxPen); + + // 绘制文字选择高亮（基于 GDI+ 字符宽度） + if (ctx->textSelStart >= 0 && ctx->textSelEnd >= 0 && ctx->textSelStart != ctx->textSelEnd + && ctx->textSelStart < (int)charWidths.size() + && ctx->textSelEnd < (int)charWidths.size()) { + int selStart = (std::min)(ctx->textSelStart, ctx->textSelEnd); + int selEnd = (std::max)(ctx->textSelStart, ctx->textSelEnd); + + float selLeftF = (float)textX + charWidths[selStart]; + float selRightF = (float)textX + charWidths[selEnd]; + int selLeft = (int)floorf(selLeftF); + int selRight = (int)ceilf(selRightF); + int selW = selRight - selLeft; + int selY = (int)floorf(glyphTop); + int selH = (int)ceilf(glyphH); + if (selW > 0 && selH > 0) { + HBRUSH selBrush = CreateSolidBrush(RGB(51, 153, 255)); + BLENDFUNCTION blend = { AC_SRC_OVER, 0, 100, 0 }; + HDC tempDC = CreateCompatibleDC(backDC); + HBITMAP tempBmp = CreateCompatibleBitmap(backDC, selW, selH); + SelectObject(tempDC, tempBmp); + RECT tempRect = {0, 0, selW, selH}; + FillRect(tempDC, &tempRect, selBrush); + AlphaBlend(backDC, selLeft, selY, selW, selH, tempDC, 0, 0, selW, selH, blend); + DeleteObject(selBrush); + DeleteDC(tempDC); + DeleteObject(tempBmp); + } + } + + // 绘制光标（闪烁效果，基于 GDI+ 字符宽度，高度对齐字形） + if (ctx->textCaretVisible && ctx->textCaretPos >= 0 + && ctx->textCaretPos < (int)charWidths.size()) { + float caretXF = (float)textX + charWidths[ctx->textCaretPos]; + int caretX = (int)floorf(caretXF); + int caretY2 = (int)floorf(glyphTop); + int caretH = (int)ceilf(glyphH); + HPEN caretPen = CreatePen(PS_SOLID, 2, textColor); + HGDIOBJ oldPenC = SelectObject(backDC, caretPen); + MoveToEx(backDC, caretX, caretY2, NULL); + LineTo(backDC, caretX, caretY2 + caretH); + SelectObject(backDC, oldPenC); + DeleteObject(caretPen); + } + } + // 确认态和文字编辑态：文字标注的选中/悬停边框 + // - selectedTextAnnotation：已选中的标注（点击后持久保持），实线高亮边框 + // - hoveredTextAnnotation：仅鼠标悬浮反馈，虚线边框（仅在与选中不同时绘制） + if (ctx->state == CS_Confirmed || ctx->state == CS_TextEditing) { + // 已选中：实线蓝色边框（醒目） + if (ctx->selectedTextAnnotation >= 0 + && ctx->selectedTextAnnotation < (int)ctx->annotations.size()) { + RECT annRect = MeasureTextAnnotation(backDC, ctx->annotations[ctx->selectedTextAnnotation]); + annRect.left -= ctx->virtualX; + annRect.top -= ctx->virtualY; + annRect.right -= ctx->virtualX; + annRect.bottom -= ctx->virtualY; + HPEN selPen = CreatePen(PS_SOLID, 2, RGB(0, 136, 255)); + HGDIOBJ oldPenS = SelectObject(backDC, selPen); + HGDIOBJ oldBrushS = SelectObject(backDC, GetStockObject(NULL_BRUSH)); + Rectangle(backDC, annRect.left, annRect.top, annRect.right, annRect.bottom); + SelectObject(backDC, oldBrushS); + SelectObject(backDC, oldPenS); + DeleteObject(selPen); + } + // 悬停：虚线边框（仅当悬浮的不是已选中项时绘制） + if (ctx->hoveredTextAnnotation >= 0 + && ctx->hoveredTextAnnotation < (int)ctx->annotations.size() + && ctx->hoveredTextAnnotation != ctx->selectedTextAnnotation) { + RECT annRect = MeasureTextAnnotation(backDC, ctx->annotations[ctx->hoveredTextAnnotation]); + annRect.left -= ctx->virtualX; + annRect.top -= ctx->virtualY; + annRect.right -= ctx->virtualX; + annRect.bottom -= ctx->virtualY; + HPEN selPen = CreatePen(PS_DASH, 1, RGB(0, 136, 255)); + HGDIOBJ oldPenS = SelectObject(backDC, selPen); + HGDIOBJ oldBrushS = SelectObject(backDC, GetStockObject(NULL_BRUSH)); + Rectangle(backDC, annRect.left, annRect.top, annRect.right, annRect.bottom); + SelectObject(backDC, oldBrushS); + SelectObject(backDC, oldPenS); + DeleteObject(selPen); + } + } + // 悬浮工具栏 + 粗细/颜色子菜单 + // 拖拽选区(CS_Moving)/调整选区(CS_Resizing)时隐藏（避免跟随抖动）； + // 绘制标注(CS_Drawing)/文字编辑(CS_TextEditing)时保持显示，便于随时查看/切换工具与样式。 + if (ctx->state == CS_Confirmed || ctx->state == CS_Drawing || ctx->state == CS_TextEditing) { + CalcToolbarPosition(curSelRect, ctx->virtualW, ctx->virtualH, + ctx->toolbarMetrics, curToolbarRect); + ctx->toolbarRect = curToolbarRect; + // hover 命中（相对坐标），确认态和文字编辑态均可更新 hover + if (ctx->state == CS_Confirmed || ctx->state == CS_TextEditing) { + int mxRel = ctx->mouseX - ctx->virtualX; + int myRel = ctx->mouseY - ctx->virtualY; + ctx->hoverToolbarBtn = HitTestToolbar(mxRel, myRel, curToolbarRect, ctx->toolbarMetrics); + } else { + ctx->hoverToolbarBtn = -1; + } + DrawToolbar(backDC, curToolbarRect, ctx->hoverToolbarBtn, ctx->activeTool, + ctx->gdi, ctx->toolbarMetrics, ctx->iconCache); + // 马赛克子菜单：模式切换 + 块大小 + if (ctx->popupOpen && ctx->activeTool == TB_Mosaic) { + int mpw, mph; + CalcMosaicPopupSize(ctx->popupMetrics, mpw, mph); + CalcPopupPlacement(curToolbarRect, ctx->virtualW, ctx->virtualH, + ctx->popupMetrics, mpw, mph, curPopupRect); + ctx->popupRect = curPopupRect; + int modeIdx = ctx->mosaicRectMode ? 1 : 0; + DrawMosaicPopup(backDC, curPopupRect, modeIdx, ctx->mosaicSizeIdx, + ctx->mosaicRadiusIdx, ctx->popupMetrics); + } + // 粗细/颜色子菜单：文字工具激活时始终显示（含文字编辑态） + else if (ctx->popupOpen && (IsVectorTool(ctx->activeTool) || ctx->activeTool == TB_Text)) { + CalcPopupPosition(curToolbarRect, ctx->virtualW, ctx->virtualH, + ctx->popupMetrics, curPopupRect); + ctx->popupRect = curPopupRect; + bool isText = (ctx->activeTool == TB_Text); + int firstIdx = isText ? ctx->fontSizeIdx : ctx->drawThickIdx; + DrawPopup(backDC, curPopupRect, ctx->drawColorIdx, firstIdx, isText, + ctx->popupMetrics); + } + } else { + ctx->hoverToolbarBtn = -1; + } + // 确认态不绘制放大镜/尺寸标签 + } else { + // ---- Idle/Selecting 态：原有逻辑 ---- + // 绘制选区外遮罩（微信风格，仅 Selecting 状态），选区内部保持清晰 + if (ctx->state == CS_Selecting) { + DrawDimMask(backDC, ctx->gdi, + curSelRect.left, curSelRect.top, curSelRect.right, curSelRect.bottom, + ctx->virtualW, ctx->virtualH); + } + + // 绘制选区或窗口尺寸标签 + if (ctx->state == CS_Selecting) { + curLabelRect = DrawSelection(backDC, ctx->startX, ctx->startY, ctx->endX, ctx->endY, + ctx->virtualX, ctx->virtualY, ctx->virtualW, ctx->virtualH, ctx->gdi); + } else if (ctx->state == CS_Idle) { + RECT screenRect; + int ww, wh; + + if (ctx->hoveredWindow >= 0 && ctx->hoveredWindow < (int)ctx->windows.size()) { + // 显示悬停窗口的尺寸 + const RECT& wr = ctx->windows[ctx->hoveredWindow].rect; + ww = wr.right - wr.left; + wh = wr.bottom - wr.top; + screenRect = wr; + } else { + // 没有匹配到窗口时，显示当前屏幕的尺寸 + POINT pt = { ctx->mouseX, ctx->mouseY }; + HMONITOR hMonitor = MonitorFromPoint(pt, MONITOR_DEFAULTTONEAREST); + if (hMonitor) { + MONITORINFO monitorInfo; + monitorInfo.cbSize = sizeof(MONITORINFO); + if (GetMonitorInfo(hMonitor, &monitorInfo)) { + screenRect = monitorInfo.rcMonitor; + ww = screenRect.right - screenRect.left; + wh = screenRect.bottom - screenRect.top; + } + } + } + + curLabelRect = DrawSizeLabel(backDC, ww, wh, + screenRect.left - ctx->virtualX, screenRect.top - ctx->virtualY, + screenRect.right - ctx->virtualX, screenRect.bottom - ctx->virtualY, + ctx->virtualW, ctx->virtualH, ctx->gdi); + } + + // 绘制放大镜信息面板 + DrawInfoPanel(backDC, panelXRel, panelYRel, ctx->currentColor, + ctx->memDC, ctx->virtualX, ctx->virtualY, + ctx->mouseX, ctx->mouseY, ctx->dpiScale, ctx->gdi); + } + + // 更新脏区域追踪 + ctx->lastPanelRect = curPanelRect; + ctx->lastSelectionRect = curSelRect; + ctx->lastLabelRect = curLabelRect; + ctx->lastHighlightRect = curHlRect; + ctx->lastToolbarRect = curToolbarRect; + ctx->lastPopupRect = curPopupRect; + + // 后台缓冲 -> 窗口 + BitBlt(hdc, 0, 0, ctx->virtualW, ctx->virtualH, backDC, 0, 0, SRCCOPY); + EndPaint(hwnd, &ps); + return 0; + } + + case WM_LBUTTONDOWN: { + if (ctx->state == CS_Idle) { + // 开始新的框选 + ctx->startX = ctx->mouseX; + ctx->startY = ctx->mouseY; + ctx->endX = ctx->mouseX; + ctx->endY = ctx->mouseY; + ctx->state = CS_Selecting; + ctx->needFullRedraw = true; + } else if (ctx->state == CS_TextEditing) { + // 文字编辑态：允许点击工具栏/子菜单操作，或点击当前输入框内选择文字 + int mxRel = ctx->mouseX - ctx->virtualX; + int myRel = ctx->mouseY - ctx->virtualY; + + // 命中粗细/颜色子菜单：切换属性，提交文字，回到确认态 + if (ctx->popupOpen) { + int hit = HitTestPopup(mxRel, myRel, ctx->popupRect, ctx->popupMetrics); + if (hit > 0) { + // 字号切换 + ctx->fontSizeIdx = hit - 1; + // 提交当前文字（如果有），应用新属性给选中的文字标注 + if (!ctx->textBuf.empty()) { + Annotation textAnnotation = {}; + textAnnotation.type = AT_Text; + textAnnotation.color = SC_COLOR_PRESETS[ctx->drawColorIdx]; + textAnnotation.thickness = SC_FONT_SIZES[ctx->fontSizeIdx]; + textAnnotation.x1 = ctx->textAnchorX; + textAnnotation.y1 = ctx->textAnchorY; + textAnnotation.text = ctx->textBuf; + ctx->annotations.push_back(textAnnotation); + ctx->redoStack.clear(); + } + ctx->textBuf.clear(); + ctx->textCaretPos = 0; + ctx->textSelStart = -1; + ctx->textSelEnd = -1; + ctx->state = CS_Confirmed; + ctx->needFullRedraw = true; + InvalidateRect(hwnd, NULL, FALSE); + return 0; + } + if (hit < 0) { + // 颜色切换 + ctx->drawColorIdx = -hit - 1; + // 提交当前文字 + if (!ctx->textBuf.empty()) { + Annotation textAnnotation = {}; + textAnnotation.type = AT_Text; + textAnnotation.color = SC_COLOR_PRESETS[ctx->drawColorIdx]; + textAnnotation.thickness = SC_FONT_SIZES[ctx->fontSizeIdx]; + textAnnotation.x1 = ctx->textAnchorX; + textAnnotation.y1 = ctx->textAnchorY; + textAnnotation.text = ctx->textBuf; + ctx->annotations.push_back(textAnnotation); + ctx->redoStack.clear(); + } + ctx->textBuf.clear(); + ctx->textCaretPos = 0; + ctx->textSelStart = -1; + ctx->textSelEnd = -1; + ctx->state = CS_Confirmed; + ctx->needFullRedraw = true; + InvalidateRect(hwnd, NULL, FALSE); + return 0; + } + } + + // 点击工具栏按钮：先提交当前文字，再回到确认态 + int toolbarBtn = HitTestToolbar(mxRel, myRel, ctx->toolbarRect, ctx->toolbarMetrics); + if (toolbarBtn >= 0 && toolbarBtn != TB_Separator1 && toolbarBtn != TB_Separator2) { + // 先提交当前文字（如果有） + if (!ctx->textBuf.empty()) { + Annotation textAnnotation = {}; + textAnnotation.type = AT_Text; + textAnnotation.color = SC_COLOR_PRESETS[ctx->drawColorIdx]; + textAnnotation.thickness = SC_FONT_SIZES[ctx->fontSizeIdx]; + textAnnotation.x1 = ctx->textAnchorX; + textAnnotation.y1 = ctx->textAnchorY; + textAnnotation.text = ctx->textBuf; + ctx->annotations.push_back(textAnnotation); + ctx->redoStack.clear(); + } + ctx->textBuf.clear(); + ctx->textCaretPos = 0; + ctx->textSelStart = -1; + ctx->textSelEnd = -1; + ctx->state = CS_Confirmed; + ctx->needFullRedraw = true; + InvalidateRect(hwnd, NULL, FALSE); + return 0; + } + + // 点击当前输入框内：开始文字选择或移动光标 + // 命中区与绘制边框完全一致（用 GDI+ 紧凑度量），否则点边框附近算"框外"会提交文字。 + int textX = ctx->textAnchorX - ctx->virtualX; + int textY = ctx->textAnchorY - ctx->virtualY; + int fontPx = SC_FONT_SIZES[ctx->fontSizeIdx]; + + float offX = 0, offY = 0, textW = 0, textH = 0; + MeasureTextGdip(ctx->backDC, ctx->textBuf, fontPx, offX, offY, textW, textH); + float glyphLeft = (float)textX + offX; + float glyphTop = (float)textY + offY; + float glyphW = (textW > 20.0f || !ctx->textBuf.empty()) ? textW : 20.0f; + float glyphH = (textH > 0 ? textH : (float)fontPx); + + const float padding = 4.0f; + RECT inputBox = { + (int)floorf(glyphLeft - padding), + (int)floorf(glyphTop - padding), + (int)ceilf(glyphLeft + glyphW + padding), + (int)ceilf(glyphTop + glyphH + padding) + }; + + if (PointInRect(mxRel, myRel, inputBox)) { + // 点击输入框内：计算光标位置并开始选择 + int caretPos = CalcCaretPosFromMouse(ctx->backDC, ctx->textBuf, fontPx, textX, mxRel); + + ctx->textCaretPos = caretPos; + ctx->textSelStart = caretPos; + ctx->textSelEnd = caretPos; + ctx->textDraggingSelection = true; + InvalidateRect(hwnd, NULL, FALSE); + return 0; + } + + // 点击选区内其他位置：提交当前文字，开始新的输入 + if (PointInRect(ctx->mouseX, ctx->mouseY, ctx->selection)) { + if (!ctx->textBuf.empty()) { + Annotation textAnnotation = {}; + textAnnotation.type = AT_Text; + textAnnotation.color = SC_COLOR_PRESETS[ctx->drawColorIdx]; + textAnnotation.thickness = SC_FONT_SIZES[ctx->fontSizeIdx]; + textAnnotation.x1 = ctx->textAnchorX; + textAnnotation.y1 = ctx->textAnchorY; + textAnnotation.text = ctx->textBuf; + ctx->annotations.push_back(textAnnotation); + ctx->redoStack.clear(); + } + ctx->textBuf.clear(); + ctx->textAnchorX = ctx->mouseX; + ctx->textAnchorY = ctx->mouseY; + ctx->textCaretPos = 0; + ctx->textSelStart = -1; + ctx->textSelEnd = -1; + ctx->needFullRedraw = true; + InvalidateRect(hwnd, NULL, FALSE); + return 0; + } + + // 点击选区外：提交文字并退出文字编辑态 + if (!ctx->textBuf.empty()) { + Annotation textAnnotation = {}; + textAnnotation.type = AT_Text; + textAnnotation.color = SC_COLOR_PRESETS[ctx->drawColorIdx]; + textAnnotation.thickness = SC_FONT_SIZES[ctx->fontSizeIdx]; + textAnnotation.x1 = ctx->textAnchorX; + textAnnotation.y1 = ctx->textAnchorY; + textAnnotation.text = ctx->textBuf; + ctx->annotations.push_back(textAnnotation); + ctx->redoStack.clear(); + } + ctx->textBuf.clear(); + ctx->textCaretPos = 0; + ctx->textSelStart = -1; + ctx->textSelEnd = -1; + ctx->state = CS_Confirmed; + ctx->needFullRedraw = true; + InvalidateRect(hwnd, NULL, FALSE); + return 0; + } else if (ctx->state == CS_Confirmed) { + // 实时命中测试（不依赖 hover 缓存值）。 + // 远程桌面（RDP）下 WM_MOUSEMOVE 常被节流/合并，hoverToolbarBtn/hoveredTextAnnotation + // 可能滞后于实际鼠标位置。若继续用缓存判断，点击文字标注时可能误命中残留的工具栏 + // 索引并提前 return，导致文字选中/拖拽分支永远执行不到。 + int mxRel = ctx->mouseX - ctx->virtualX; + int myRel = ctx->mouseY - ctx->virtualY; + int b = HitTestToolbar(mxRel, myRel, ctx->toolbarRect, ctx->toolbarMetrics); + + // 点击工具栏按钮（实时命中） + if (b >= 0 && b != TB_Separator1 && b != TB_Separator2) { + // 确定：提取选区并完成截图 + if (b == TB_Confirm) { + ScreenshotResult* result = ExtractRegionResult(ctx->memDC, ctx->selection, + ctx->virtualX, ctx->virtualY, ctx->dpiScale, ctx->annotations); + if (g_screenshotTsfn != nullptr) { + napi_call_threadsafe_function(g_screenshotTsfn, result, napi_tsfn_nonblocking); + } + ctx->state = CS_Done; + DestroyWindow(hwnd); + return 0; + } + // 取消：回调失败并关闭 + if (b == TB_Cancel) { + if (g_screenshotTsfn != nullptr) { + ScreenshotResult* result = new ScreenshotResult(); + result->success = false; + napi_call_threadsafe_function(g_screenshotTsfn, result, napi_tsfn_nonblocking); + } + ctx->state = CS_Cancelled; + DestroyWindow(hwnd); + return 0; + } + // 矢量工具：切换激活态 + 打开/关闭粗细颜色子菜单 + if (b == TB_Rect || b == TB_Circle || b == TB_Arrow || b == TB_Brush) { + if (ctx->activeTool == b) { + // 再次点同一工具：关闭工具与子菜单 + ctx->activeTool = -1; + ctx->popupOpen = false; + } else { + ctx->activeTool = b; + ctx->popupOpen = true; + } + InvalidateRect(hwnd, NULL, FALSE); + return 0; + } + // 文字工具：切换激活态 + 打开/关闭子菜单（字号+颜色） + if (b == TB_Text) { + if (ctx->activeTool == b) { + // 再次点同一工具：关闭工具与子菜单 + ctx->activeTool = -1; + ctx->popupOpen = false; + } else { + ctx->activeTool = b; + ctx->popupOpen = true; + } + InvalidateRect(hwnd, NULL, FALSE); + return 0; + } + // 马赛克工具：切换激活态 + 打开/关闭子菜单（模式+块大小） + if (b == TB_Mosaic) { + if (ctx->activeTool == b) { + // 再次点同一工具：关闭工具与子菜单 + ctx->activeTool = -1; + ctx->popupOpen = false; + } else { + ctx->activeTool = b; + ctx->popupOpen = true; + } + InvalidateRect(hwnd, NULL, FALSE); + return 0; + } + // 翻译工具：仅切换激活态占位，关闭子菜单 + if (b == TB_Translate) { + ctx->activeTool = (ctx->activeTool == b) ? -1 : b; + ctx->popupOpen = false; + InvalidateRect(hwnd, NULL, FALSE); + return 0; + } + // 撤销：弹出最后一条标注到 redo 栈 + if (b == TB_Undo) { + if (!ctx->annotations.empty()) { + ctx->redoStack.push_back(ctx->annotations.back()); + ctx->annotations.pop_back(); + // 修正选中索引：撤销后选中项可能已失效 + if (ctx->selectedTextAnnotation >= (int)ctx->annotations.size()) { + ctx->selectedTextAnnotation = -1; + } + ctx->hoveredTextAnnotation = -1; + InvalidateRect(hwnd, NULL, FALSE); + } + return 0; + } + // 重做：从 redo 栈回填 + if (b == TB_Redo) { + if (!ctx->redoStack.empty()) { + ctx->annotations.push_back(ctx->redoStack.back()); + ctx->redoStack.pop_back(); + InvalidateRect(hwnd, NULL, FALSE); + } + return 0; + } + // 保存到本地：弹出系统保存对话框，保存为 PNG 后关闭截图 + if (b == TB_Save) { + std::wstring filePath = PromptSaveFilePath(hwnd); + if (!filePath.empty()) { + bool saved = SaveRegionToPngFile(ctx->memDC, ctx->selection, + ctx->virtualX, ctx->virtualY, ctx->dpiScale, + ctx->annotations, filePath); + // 无论保存成功与否，均关闭截图窗口（用户已选择保存路径） + // 通过回调告知 JS 结果（成功/失败），不回传路径 + ScreenshotResult* result = new ScreenshotResult(); + result->success = saved; + if (saved) { + result->x = ctx->selection.left; + result->y = ctx->selection.top; + result->x2 = ctx->selection.right; + result->y2 = ctx->selection.bottom; + result->width = ctx->selection.right - ctx->selection.left; + result->height = ctx->selection.bottom - ctx->selection.top; + } + if (g_screenshotTsfn != nullptr) { + napi_call_threadsafe_function(g_screenshotTsfn, result, napi_tsfn_nonblocking); + } + ctx->state = CS_Done; + DestroyWindow(hwnd); + } + // 用户取消保存对话框：不关闭，留在编辑态 + return 0; + } + return 0; + } + + // 命中马赛克子菜单（模式切换 + 块大小 + 涂抹半径） + if (ctx->popupOpen && ctx->activeTool == TB_Mosaic) { + int hit = HitTestMosaicPopup(mxRel, myRel, ctx->popupRect, ctx->popupMetrics); + if (hit == 1) { + ctx->mosaicRectMode = false; // 涂抹模式 + InvalidateRect(hwnd, NULL, FALSE); + return 0; + } + if (hit == 2) { + ctx->mosaicRectMode = true; // 框选模式 + InvalidateRect(hwnd, NULL, FALSE); + return 0; + } + if (hit >= 101 && hit < 200) { + ctx->mosaicSizeIdx = hit - 101; + InvalidateRect(hwnd, NULL, FALSE); + return 0; + } + if (hit >= 201) { + ctx->mosaicRadiusIdx = hit - 201; + InvalidateRect(hwnd, NULL, FALSE); + return 0; + } + } + + // 命中粗细/颜色子菜单 + if (ctx->popupOpen) { + int hit = HitTestPopup(mxRel, myRel, ctx->popupRect, ctx->popupMetrics); + if (hit > 0) { + // 第一组：文字工具时为字号索引，矢量工具时为粗细索引 + if (ctx->activeTool == TB_Text) { + ctx->fontSizeIdx = hit - 1; + // 如果选中了文字标注，修改其字号 + if (ctx->selectedTextAnnotation >= 0 && ctx->selectedTextAnnotation < (int)ctx->annotations.size()) { + ctx->annotations[ctx->selectedTextAnnotation].thickness = SC_FONT_SIZES[ctx->fontSizeIdx]; + } + } else { + ctx->drawThickIdx = hit - 1; + } + InvalidateRect(hwnd, NULL, FALSE); + return 0; + } + if (hit < 0) { + ctx->drawColorIdx = -hit - 1; + // 如果选中了文字标注，修改其颜色 + if (ctx->activeTool == TB_Text && ctx->selectedTextAnnotation >= 0 + && ctx->selectedTextAnnotation < (int)ctx->annotations.size()) { + ctx->annotations[ctx->selectedTextAnnotation].color = SC_COLOR_PRESETS[ctx->drawColorIdx]; + } + InvalidateRect(hwnd, NULL, FALSE); + return 0; + } + } + + // 矢量工具激活时，点击选区内部/工具栏外 -> 开始绘制 + // 子菜单保持打开，绘制中可继续看到当前粗细/颜色。 + if (IsVectorTool(ctx->activeTool) && PointInRect(ctx->mouseX, ctx->mouseY, ctx->selection)) { + ctx->hasCurDrawing = true; + ctx->curDrawing = {}; + ctx->curDrawing.type = ToolToAnnotationType(ctx->activeTool); + ctx->curDrawing.color = SC_COLOR_PRESETS[ctx->drawColorIdx]; + ctx->curDrawing.thickness = SC_THICK_PRESETS[ctx->drawThickIdx]; + // 起点用绝对虚拟屏幕坐标 + ctx->curDrawing.x1 = ctx->mouseX; + ctx->curDrawing.y1 = ctx->mouseY; + ctx->curDrawing.x2 = ctx->curDrawing.x1; + ctx->curDrawing.y2 = ctx->curDrawing.y1; + if (ctx->curDrawing.type == AT_Brush) { + POINT p = { ctx->curDrawing.x1, ctx->curDrawing.y1 }; + ctx->curDrawing.pts.push_back(p); + } + ctx->state = CS_Drawing; + ctx->needFullRedraw = true; + return 0; + } + + // 马赛克工具激活时，点击选区内部 -> 开始马赛克绘制 + // 子菜单保持打开，绘制中可继续看到当前模式/块大小。 + if (ctx->activeTool == TB_Mosaic && PointInRect(ctx->mouseX, ctx->mouseY, ctx->selection)) { + ctx->hasCurDrawing = true; + ctx->curDrawing = {}; + ctx->curDrawing.type = AT_Mosaic; + ctx->curDrawing.color = 0; // 马赛克无颜色 + ctx->curDrawing.mosaicRect = ctx->mosaicRectMode; + ctx->curDrawing.mosaicSize = SC_MOSAIC_SIZES[ctx->mosaicSizeIdx]; + ctx->curDrawing.brushRadius = SC_MOSAIC_RADIUS[ctx->mosaicRadiusIdx]; + ctx->curDrawing.x1 = ctx->mouseX; + ctx->curDrawing.y1 = ctx->mouseY; + ctx->curDrawing.x2 = ctx->curDrawing.x1; + ctx->curDrawing.y2 = ctx->curDrawing.y1; + if (!ctx->mosaicRectMode) { + // 涂抹模式：记录路径起点（揭示由 WM_PAINT 统一处理） + POINT p = { ctx->curDrawing.x1, ctx->curDrawing.y1 }; + ctx->curDrawing.pts.push_back(p); + } + ctx->state = CS_Drawing; + ctx->needFullRedraw = true; + return 0; + } + + // 文字工具激活时：点击已确认文字 -> 选中并可拖动，点击空白 -> 进入输入态 + if (ctx->activeTool == TB_Text) { + int hitText = HitTestTextAnnotations(ctx->annotations, ctx->mouseX, ctx->mouseY, ctx->backDC); + if (hitText >= 0) { + // 选中文字标注，保持确认态。 + // selectedTextAnnotation 持久保持选中态（与 hover 解耦），鼠标移开仍高亮， + // 直到点击空白或进入其他操作才清除。 + ctx->selectedTextAnnotation = hitText; + ctx->hoveredTextAnnotation = hitText; + // 同步当前字号/颜色索引为该标注的值，子菜单高亮一致 + int curSize = ctx->annotations[hitText].thickness; + for (int i = 0; i < SC_FONT_COUNT; i++) { + if (SC_FONT_SIZES[i] == curSize) { ctx->fontSizeIdx = i; break; } + } + COLORREF curColor = ctx->annotations[hitText].color; + for (int i = 0; i < SC_COLOR_COUNT; i++) { + if (SC_COLOR_PRESETS[i] == curColor) { ctx->drawColorIdx = i; break; } + } + // 选中文字时确保字号/颜色子菜单打开，便于直接修改 + ctx->popupOpen = true; + // 同时进入拖动模式：按下即可拖拽移动文字（鼠标移动则在 WM_MOUSEMOVE + // 里更新标注坐标；若未移动，WM_LBUTTONUP 仅结束拖动，保持选中态）。 + ctx->draggingTextAnnotation = hitText; + ctx->textDragStartX = ctx->mouseX; + ctx->textDragStartY = ctx->mouseY; + ctx->dragStartX = ctx->annotations[hitText].x1; + ctx->dragStartY = ctx->annotations[hitText].y1; + InvalidateRect(hwnd, NULL, FALSE); + return 0; + } + // 点击选区内空白 -> 进入文字编辑态（清除已选中） + if (PointInRect(ctx->mouseX, ctx->mouseY, ctx->selection)) { + ctx->textBuf.clear(); + ctx->textAnchorX = ctx->mouseX; + ctx->textAnchorY = ctx->mouseY; + ctx->textCaretPos = 0; + ctx->textSelStart = -1; + ctx->textSelEnd = -1; + ctx->state = CS_TextEditing; + // 进入输入态时清除文字标注选中，避免残留选中边框 + ctx->hoveredTextAnnotation = -1; + ctx->selectedTextAnnotation = -1; + // 保持子菜单打开，清空绘制标志 + // ctx->popupOpen 保持不变 + ctx->hasCurDrawing = false; + // 文字工具保持激活，避免工具栏视觉状态丢失 + // ctx->activeTool 保持为 TB_Text + ctx->needFullRedraw = true; + InvalidateRect(hwnd, NULL, FALSE); + return 0; + } + } + + // 无工具激活时：点击已确认文字 -> 拖动文字位置 + int hitText = HitTestTextAnnotations(ctx->annotations, ctx->mouseX, ctx->mouseY, ctx->backDC); + if (hitText >= 0 && ctx->activeTool == -1) { + ctx->draggingTextAnnotation = hitText; + // 同步 hovered/selected，拖动期间选中边框即时显示 + ctx->hoveredTextAnnotation = hitText; + ctx->selectedTextAnnotation = hitText; + ctx->textDragStartX = ctx->mouseX; + ctx->textDragStartY = ctx->mouseY; + ctx->dragStartX = ctx->annotations[hitText].x1; + ctx->dragStartY = ctx->annotations[hitText].y1; + ctx->needFullRedraw = true; + return 0; + } + + // 命中调整手柄 -> 进入 Resizing + int h = HitTestHandle(ctx->mouseX, ctx->mouseY, ctx->selection); + if (h != RH_None) { + ctx->selectedTextAnnotation = -1; // 进入手柄调整，清除文字选中 + ctx->resizeHandle = h; + ctx->dragStartX = ctx->mouseX; + ctx->dragStartY = ctx->mouseY; + ctx->dragStartSelection = ctx->selection; + ctx->state = CS_Resizing; + ctx->needFullRedraw = true; + return 0; + } + // 点击选区内部 -> 整体拖动（已有标注内容时禁止，避免标注与背景错位） + if (PointInRect(ctx->mouseX, ctx->mouseY, ctx->selection)) { + if (!ctx->annotations.empty()) { + // 已有预处理内容：只能通过手柄改范围，不允许整体拖动 + return 0; + } + ctx->selectedTextAnnotation = -1; // 进入选区拖动，清除文字选中 + ctx->dragStartX = ctx->mouseX; + ctx->dragStartY = ctx->mouseY; + ctx->dragStartSelection = ctx->selection; + ctx->state = CS_Moving; + ctx->needFullRedraw = true; + return 0; + } + // 点击选区外空白：清除文字选中，进入确认态后不可重新框选，忽略点击 + ctx->selectedTextAnnotation = -1; + return 0; + } + return 0; + } + + case WM_MOUSEMOVE: { + POINT pt; + GetCursorPos(&pt); + bool moved = (pt.x != ctx->mouseX || pt.y != ctx->mouseY); + ctx->mouseX = pt.x; + ctx->mouseY = pt.y; + ctx->currentColor = GetPixelColorFromBitmap(ctx->memDC, + ctx->mouseX, ctx->mouseY, ctx->virtualX, ctx->virtualY, ctx->dpiScale); + + if (ctx->state == CS_Selecting) { + ctx->endX = ctx->mouseX; + ctx->endY = ctx->mouseY; + InvalidateRect(hwnd, NULL, FALSE); + } else if (ctx->state == CS_Idle) { + int newHovered = FindWindowAtPoint(ctx->windows, ctx->mouseX, ctx->mouseY); + ctx->hoveredWindow = newHovered; + // 像素信息浮窗跟随鼠标，每次移动都需重绘（与 CS_Selecting 行为一致） + InvalidateRect(hwnd, NULL, FALSE); + } else if (ctx->state == CS_Resizing) { + // 根据手柄调整选区边 + RECT& s = ctx->selection; + const RECT& o = ctx->dragStartSelection; + int dx = pt.x - ctx->dragStartX; + int dy = pt.y - ctx->dragStartY; + switch (ctx->resizeHandle) { + case RH_Left: s.left = o.left + dx; break; + case RH_Right: s.right = o.right + dx; break; + case RH_Top: s.top = o.top + dy; break; + case RH_Bottom: s.bottom = o.bottom + dy; break; + case RH_TopLeft: s.left = o.left + dx; s.top = o.top + dy; break; + case RH_TopRight: s.right = o.right + dx; s.top = o.top + dy; break; + case RH_BottomLeft: s.left = o.left + dx; s.bottom = o.bottom + dy; break; + case RH_BottomRight: s.right = o.right + dx; s.bottom = o.bottom + dy; break; + } + ctx->selection = NormalizeRect(s); + // 已有标注内容时，选区不可缩小到裁掉内容（标注为绝对坐标，四边各自钳制）： + // 左/上边不得越过内容包围盒的左/上；右/下边不得小于包围盒的右/下。 + RECT contentBounds; + if (CalcAnnotationsBounds(ctx->annotations, contentBounds, ctx->backDC)) { + if (s.left > contentBounds.left) { + if (ctx->resizeHandle == RH_Left || ctx->resizeHandle == RH_TopLeft + || ctx->resizeHandle == RH_BottomLeft) { + s.left = contentBounds.left; + } + } + if (s.top > contentBounds.top) { + if (ctx->resizeHandle == RH_Top || ctx->resizeHandle == RH_TopLeft + || ctx->resizeHandle == RH_TopRight) { + s.top = contentBounds.top; + } + } + if (s.right < contentBounds.right) { + if (ctx->resizeHandle == RH_Right || ctx->resizeHandle == RH_TopRight + || ctx->resizeHandle == RH_BottomRight) { + s.right = contentBounds.right; + } + } + if (s.bottom < contentBounds.bottom) { + if (ctx->resizeHandle == RH_Bottom || ctx->resizeHandle == RH_BottomLeft + || ctx->resizeHandle == RH_BottomRight) { + s.bottom = contentBounds.bottom; + } + } + } + InvalidateRect(hwnd, NULL, FALSE); + } else if (ctx->state == CS_Moving) { + // 整体平移 + int dx = pt.x - ctx->dragStartX; + int dy = pt.y - ctx->dragStartY; + int sw = ctx->dragStartSelection.right - ctx->dragStartSelection.left; + int sh = ctx->dragStartSelection.bottom - ctx->dragStartSelection.top; + int nl = ctx->dragStartSelection.left + dx; + int nt = ctx->dragStartSelection.top + dy; + // 约束到虚拟屏幕 + if (nl < ctx->virtualX) nl = ctx->virtualX; + if (nt < ctx->virtualY) nt = ctx->virtualY; + if (nl + sw > ctx->virtualX + ctx->virtualW) nl = ctx->virtualX + ctx->virtualW - sw; + if (nt + sh > ctx->virtualY + ctx->virtualH) nt = ctx->virtualY + ctx->virtualH - sh; + ctx->selection.left = nl; + ctx->selection.top = nt; + ctx->selection.right = nl + sw; + ctx->selection.bottom = nt + sh; + InvalidateRect(hwnd, NULL, FALSE); + } else if (ctx->state == CS_Drawing) { + // 更新正在绘制的标注终点/路径（选区相对坐标） + if (ctx->hasCurDrawing) { + // 终点用绝对坐标，钳制到选区内 + int selL = ctx->selection.left, selR = ctx->selection.right; + int selT = ctx->selection.top, selB = ctx->selection.bottom; + int ax = (std::max)(selL, (std::min)(ctx->mouseX, selR)); + int ay = (std::max)(selT, (std::min)(ctx->mouseY, selB)); + if (ctx->curDrawing.type == AT_Brush) { + POINT p = { ax, ay }; + ctx->curDrawing.pts.push_back(p); + } else if (ctx->curDrawing.type == AT_Mosaic && !ctx->curDrawing.mosaicRect) { + // 马赛克涂抹模式：记录路径点。揭示由 WM_PAINT 统一处理（reveal-mask 模型， + // 每帧从预计算的 base 揭示蒙版区域，无需增量绘制）。 + POINT p = { ax, ay }; + ctx->curDrawing.pts.push_back(p); + } else { + ctx->curDrawing.x2 = ax; + ctx->curDrawing.y2 = ay; + } + InvalidateRect(hwnd, NULL, FALSE); + } + } else if (ctx->state == CS_TextEditing) { + // 文字编辑态：拖动选择文字 + if (ctx->textDraggingSelection) { + int mxRel = ctx->mouseX - ctx->virtualX; + int textX = ctx->textAnchorX - ctx->virtualX; + int fontPx = SC_FONT_SIZES[ctx->fontSizeIdx]; + int caretPos = CalcCaretPosFromMouse(ctx->backDC, ctx->textBuf, fontPx, textX, mxRel); + + ctx->textSelEnd = caretPos; + ctx->textCaretPos = caretPos; + InvalidateRect(hwnd, NULL, FALSE); + } + } else if (ctx->draggingTextAnnotation >= 0) { + // 拖动文字标注位置 + int dx = ctx->mouseX - ctx->textDragStartX; + int dy = ctx->mouseY - ctx->textDragStartY; + ctx->annotations[ctx->draggingTextAnnotation].x1 = ctx->dragStartX + dx; + ctx->annotations[ctx->draggingTextAnnotation].y1 = ctx->dragStartY + dy; + InvalidateRect(hwnd, NULL, FALSE); + } else if (ctx->state == CS_Confirmed) { + // hover 手柄/工具栏/文字标注变化需重绘以更新光标提示与高亮 + int h = HitTestHandle(ctx->mouseX, ctx->mouseY, ctx->selection); + int mxRel = ctx->mouseX - ctx->virtualX; + int myRel = ctx->mouseY - ctx->virtualY; + int tb = HitTestToolbar(mxRel, myRel, ctx->toolbarRect, ctx->toolbarMetrics); + int ht = HitTestTextAnnotations(ctx->annotations, ctx->mouseX, ctx->mouseY, ctx->backDC); + if (moved || h != ctx->resizeHandle || tb != ctx->hoverToolbarBtn + || ht != ctx->hoveredTextAnnotation) { + ctx->resizeHandle = h; + ctx->hoverToolbarBtn = tb; + ctx->hoveredTextAnnotation = ht; + InvalidateRect(hwnd, NULL, FALSE); + } + } + return 0; + } + + case WM_LBUTTONUP: { + if (ctx->state == CS_Selecting) { + int w = abs(ctx->endX - ctx->startX); + int h = abs(ctx->endY - ctx->startY); + + RECT finalRect; + if (w <= 1 && h <= 1) { + // 点击 -> 使用悬停窗口矩形 + int idx = FindWindowAtPoint(ctx->windows, ctx->mouseX, ctx->mouseY); + if (idx >= 0) { + finalRect = ctx->windows[idx].rect; + } else { + // 匹配不到窗口时，默认选区为鼠标所在的屏幕 + POINT pt = { ctx->mouseX, ctx->mouseY }; + HMONITOR hMonitor = MonitorFromPoint(pt, MONITOR_DEFAULTTONEAREST); + if (hMonitor) { + MONITORINFO monitorInfo; + monitorInfo.cbSize = sizeof(MONITORINFO); + if (GetMonitorInfo(hMonitor, &monitorInfo)) { + finalRect = monitorInfo.rcMonitor; + } else { + // 获取失败，降级为虚拟屏幕 + finalRect = { ctx->virtualX, ctx->virtualY, + ctx->virtualX + ctx->virtualW, ctx->virtualY + ctx->virtualH }; + } + } else { + // 获取显示器失败，降级为虚拟屏幕 + finalRect = { ctx->virtualX, ctx->virtualY, + ctx->virtualX + ctx->virtualW, ctx->virtualY + ctx->virtualH }; + } + } + } else { + finalRect.left = (std::min)(ctx->startX, ctx->endX); + finalRect.top = (std::min)(ctx->startY, ctx->endY); + finalRect.right = (std::max)(ctx->startX, ctx->endX); + finalRect.bottom = (std::max)(ctx->startY, ctx->endY); + } + + // 进入确认态（可调整/拖动/工具栏），而非直接完成 + EnterConfirmed(ctx, finalRect); + InvalidateRect(hwnd, NULL, FALSE); + } else if (ctx->state == CS_Resizing || ctx->state == CS_Moving) { + // 调整/拖动结束 -> 回到确认态 + EnterConfirmed(ctx, ctx->selection); + InvalidateRect(hwnd, NULL, FALSE); + } else if (ctx->state == CS_Drawing) { + // 绘制结束 -> 提交标注（仅在有有效尺寸时） + bool valid = false; + if (ctx->hasCurDrawing) { + if (ctx->curDrawing.type == AT_Brush) { + valid = ctx->curDrawing.pts.size() >= 2; + } else if (ctx->curDrawing.type == AT_Mosaic) { + if (ctx->curDrawing.mosaicRect) { + // 框选模式：需要有效矩形尺寸 + valid = (abs(ctx->curDrawing.x2 - ctx->curDrawing.x1) >= 2 + || abs(ctx->curDrawing.y2 - ctx->curDrawing.y1) >= 2); + } else { + // 涂抹模式：至少 1 个点（单击也能产生一个马赛克圆） + valid = ctx->curDrawing.pts.size() >= 1; + } + } else { + valid = (abs(ctx->curDrawing.x2 - ctx->curDrawing.x1) >= 2 + || abs(ctx->curDrawing.y2 - ctx->curDrawing.y1) >= 2); + } + } + if (valid) { + ctx->annotations.push_back(ctx->curDrawing); + ctx->redoStack.clear(); // 新标注清空重做栈 + // reveal-mask 模型：base 与标注无关，下一帧 WM_PAINT 自动把新蒙版揭示出来。 + } + ctx->hasCurDrawing = false; + ctx->curDrawing = {}; + ctx->mosaicDrawLastIdx = 0; + ctx->state = CS_Confirmed; + ctx->needFullRedraw = true; + } else if (ctx->state == CS_TextEditing) { + // 文字选择结束 + if (ctx->textDraggingSelection) { + ctx->textDraggingSelection = false; + // 如果选择范围相同，清除选择 + if (ctx->textSelStart == ctx->textSelEnd) { + ctx->textSelStart = -1; + ctx->textSelEnd = -1; + } + } + } else if (ctx->draggingTextAnnotation >= 0) { + // 文字拖动结束 + ctx->draggingTextAnnotation = -1; + ctx->needFullRedraw = true; + } + return 0; + } + + case WM_LBUTTONDBLCLK: { + // 确认态下双击选区内部 -> 确认截图 + if ((ctx->state == CS_Confirmed) && PointInRect(ctx->mouseX, ctx->mouseY, ctx->selection)) { + ScreenshotResult* result = ExtractRegionResult(ctx->memDC, ctx->selection, + ctx->virtualX, ctx->virtualY, ctx->dpiScale, ctx->annotations); + if (g_screenshotTsfn != nullptr) { + napi_call_threadsafe_function(g_screenshotTsfn, result, napi_tsfn_nonblocking); + } + ctx->state = CS_Done; + DestroyWindow(hwnd); + } + return 0; + } + + case WM_RBUTTONDOWN: { + ctx->state = CS_Cancelled; + // 回调失败结果 + if (g_screenshotTsfn != nullptr) { + ScreenshotResult* result = new ScreenshotResult(); + result->success = false; + result->x = 0; result->y = 0; result->x2 = 0; result->y2 = 0; + result->width = 0; result->height = 0; + napi_call_threadsafe_function(g_screenshotTsfn, result, napi_tsfn_nonblocking); + } + DestroyWindow(hwnd); + return 0; + } + + case WM_KEYDOWN: { + if (wParam == VK_ESCAPE) { + // 文字编辑态：ESC 取消输入，回到确认态 + if (ctx->state == CS_TextEditing) { + ctx->textBuf.clear(); + ctx->textCaretPos = 0; + ctx->state = CS_Confirmed; + ctx->needFullRedraw = true; + InvalidateRect(hwnd, NULL, FALSE); + return 0; + } + // 其它状态：ESC 取消截图 + ctx->state = CS_Cancelled; + if (g_screenshotTsfn != nullptr) { + ScreenshotResult* result = new ScreenshotResult(); + result->success = false; + result->x = 0; result->y = 0; result->x2 = 0; result->y2 = 0; + result->width = 0; result->height = 0; + napi_call_threadsafe_function(g_screenshotTsfn, result, napi_tsfn_nonblocking); + } + DestroyWindow(hwnd); + } else if (wParam == VK_RETURN) { + // 文字编辑态：Enter 提交文字 + if (ctx->state == CS_TextEditing) { + if (!ctx->textBuf.empty()) { + Annotation textAnnotation = {}; + textAnnotation.type = AT_Text; + textAnnotation.color = SC_COLOR_PRESETS[ctx->drawColorIdx]; + textAnnotation.thickness = SC_FONT_SIZES[ctx->fontSizeIdx]; + textAnnotation.x1 = ctx->textAnchorX; + textAnnotation.y1 = ctx->textAnchorY; + textAnnotation.text = ctx->textBuf; + ctx->annotations.push_back(textAnnotation); + ctx->redoStack.clear(); + } + ctx->textBuf.clear(); + ctx->textCaretPos = 0; + ctx->state = CS_Confirmed; + ctx->needFullRedraw = true; + InvalidateRect(hwnd, NULL, FALSE); + return 0; + } + // 确认态：Enter 确认截图 + if (ctx->state == CS_Confirmed) { + ScreenshotResult* result = ExtractRegionResult(ctx->memDC, ctx->selection, + ctx->virtualX, ctx->virtualY, ctx->dpiScale, ctx->annotations); + if (g_screenshotTsfn != nullptr) { + napi_call_threadsafe_function(g_screenshotTsfn, result, napi_tsfn_nonblocking); + } + ctx->state = CS_Done; + DestroyWindow(hwnd); + } + } else if (wParam == VK_BACK) { + // 文字编辑态：退格删除 + if (ctx->state == CS_TextEditing && ctx->textCaretPos > 0) { + ctx->textBuf.erase(ctx->textCaretPos - 1, 1); + ctx->textCaretPos--; + InvalidateRect(hwnd, NULL, FALSE); + return 0; + } + } else if (wParam == VK_DELETE) { + // 文字编辑态：Delete 删除 + if (ctx->state == CS_TextEditing && ctx->textCaretPos < (int)ctx->textBuf.size()) { + ctx->textBuf.erase(ctx->textCaretPos, 1); + InvalidateRect(hwnd, NULL, FALSE); + return 0; + } + } else if (wParam == VK_LEFT) { + // 文字编辑态：左箭头移动光标 + if (ctx->state == CS_TextEditing && ctx->textCaretPos > 0) { + ctx->textCaretPos--; + InvalidateRect(hwnd, NULL, FALSE); + return 0; + } + } else if (wParam == VK_RIGHT) { + // 文字编辑态：右箭头移动光标 + if (ctx->state == CS_TextEditing && ctx->textCaretPos < (int)ctx->textBuf.size()) { + ctx->textCaretPos++; + InvalidateRect(hwnd, NULL, FALSE); + return 0; + } + } else if (wParam == VK_HOME) { + // 文字编辑态：Home 移动到行首 + if (ctx->state == CS_TextEditing) { + ctx->textCaretPos = 0; + InvalidateRect(hwnd, NULL, FALSE); + return 0; + } + } else if (wParam == VK_END) { + // 文字编辑态：End 移动到行尾 + if (ctx->state == CS_TextEditing) { + ctx->textCaretPos = (int)ctx->textBuf.size(); + InvalidateRect(hwnd, NULL, FALSE); + return 0; + } + } + return 0; + } + + case WM_IME_COMPOSITION: { + // 处理中文输入法（IME）输入 + if (ctx->state == CS_TextEditing) { + if (lParam & GCS_RESULTSTR) { + HIMC hIMC = ImmGetContext(hwnd); + if (hIMC) { + // 获取输入法完成的字符串长度 + LONG len = ImmGetCompositionStringW(hIMC, GCS_RESULTSTR, NULL, 0); + if (len > 0) { + // 分配缓冲区并获取字符串 + std::wstring result(len / sizeof(wchar_t), L'\0'); + ImmGetCompositionStringW(hIMC, GCS_RESULTSTR, &result[0], len); + result.resize(len / sizeof(wchar_t)); + + // 插入到当前光标位置 + ctx->textBuf.insert(ctx->textCaretPos, result); + ctx->textCaretPos += (int)result.size(); + InvalidateRect(hwnd, NULL, FALSE); + } + ImmReleaseContext(hwnd, hIMC); + } + return 0; + } + } + return DefWindowProc(hwnd, msg, wParam, lParam); + } + + case WM_CHAR: { + // 文字编辑态：接收字符输入（ASCII 和直接输入） + if (ctx->state == CS_TextEditing) { + wchar_t ch = (wchar_t)wParam; + // 过滤控制字符（除了可打印字符） + // 忽略 IME 相关的控制字符 + if (ch >= 32 && ch != 127) { + ctx->textBuf.insert(ctx->textCaretPos, 1, ch); + ctx->textCaretPos++; + InvalidateRect(hwnd, NULL, FALSE); + } + return 0; + } + return 0; + } + + case WM_SETCURSOR: { + // 马赛克涂抹模式（确认态/绘制态）：用预生成的半径圆光标（OS 跟随，无重绘延迟） + if (ctx->activeTool == TB_Mosaic && !ctx->mosaicRectMode + && (ctx->state == CS_Confirmed || ctx->state == CS_Drawing) + && ctx->mosaicBrushCursorsInited + && ctx->mosaicRadiusIdx >= 0 && ctx->mosaicRadiusIdx < SC_MOSAIC_RADIUS_COUNT + && ctx->mosaicBrushCursors[ctx->mosaicRadiusIdx]) { + // 工具栏/子菜单上仍用箭头/手型 + int mxRel = ctx->mouseX - ctx->virtualX; + int myRel = ctx->mouseY - ctx->virtualY; + if (PointInRect(mxRel, myRel, ctx->toolbarRect)) { + SetCursor(LoadCursorW(NULL, (LPCWSTR)IDC_ARROW)); + return TRUE; + } + if (ctx->popupOpen && PointInRect(mxRel, myRel, ctx->popupRect)) { + SetCursor(LoadCursorW(NULL, (LPCWSTR)IDC_HAND)); + return TRUE; + } + if (PointInRect(ctx->mouseX, ctx->mouseY, ctx->selection)) { + SetCursor(ctx->mosaicBrushCursors[ctx->mosaicRadiusIdx]); + return TRUE; + } + } + // 文字编辑中：I-beam 光标 + if (ctx->state == CS_TextEditing) { + SetCursor(LoadCursorW(NULL, (LPCWSTR)IDC_IBEAM)); + return TRUE; + } + // 绘制中：十字光标 + if (ctx->state == CS_Drawing) { + SetCursor(LoadCursorW(NULL, (LPCWSTR)IDC_CROSS)); + return TRUE; + } + // 拖动文字标注中：四向箭头光标 + if (ctx->draggingTextAnnotation >= 0) { + SetCursor(LoadCursorW(NULL, (LPCWSTR)IDC_SIZEALL)); + return TRUE; + } + // 确认态：根据 hover 位置切换 resize/move/箭头/十字光标 + if (ctx->state == CS_Confirmed) { + // 工具栏或子菜单 -> 箭头 + int mxRel = ctx->mouseX - ctx->virtualX; + int myRel = ctx->mouseY - ctx->virtualY; + if (PointInRect(mxRel, myRel, ctx->toolbarRect)) { + SetCursor(LoadCursorW(NULL, (LPCWSTR)IDC_ARROW)); + return TRUE; + } + if (ctx->popupOpen && PointInRect(mxRel, myRel, ctx->popupRect)) { + SetCursor(LoadCursorW(NULL, (LPCWSTR)IDC_HAND)); + return TRUE; + } + // 手柄 -> 对应 resize 光标 + int h = HitTestHandle(ctx->mouseX, ctx->mouseY, ctx->selection); + if (h != RH_None) { + SetCursor(LoadCursorW(NULL, HandleCursor(h))); + return TRUE; + } + // 文字标注悬停 -> 四向箭头（可拖动/选中） + // 注意：此处必须独立做命中测试，不能依赖 ctx->hoveredTextAnnotation。 + // 在远程桌面（RDP）下鼠标移动事件常被节流/合并，WM_MOUSEMOVE 更新 + // hoveredTextAnnotation 存在滞后，导致 WM_SETCURSOR 看到过期值。 + // 文字工具未激活时，悬停已确认文字 -> 拖动光标； + // 文字工具激活时，悬停已确认文字 -> 仍为拖动光标（可选中改属性）。 + if (HitTestTextAnnotations(ctx->annotations, ctx->mouseX, ctx->mouseY, ctx->backDC) >= 0) { + SetCursor(LoadCursorW(NULL, (LPCWSTR)IDC_SIZEALL)); + return TRUE; + } + // 选区内部： + // 矢量/文字/马赛克工具激活 -> 十字； + // 已有标注内容 -> 箭头（禁止整体拖动）； + // 否则 -> 移动光标 + if (PointInRect(ctx->mouseX, ctx->mouseY, ctx->selection)) { + if (IsVectorTool(ctx->activeTool) || ctx->activeTool == TB_Text + || ctx->activeTool == TB_Mosaic) { + SetCursor(LoadCursorW(NULL, (LPCWSTR)IDC_CROSS)); + } else if (!ctx->annotations.empty()) { + SetCursor(LoadCursorW(NULL, (LPCWSTR)IDC_ARROW)); + } else { + SetCursor(LoadCursorW(NULL, (LPCWSTR)IDC_SIZEALL)); + } + return TRUE; + } + // 选区外 -> 箭头 + SetCursor(LoadCursorW(NULL, (LPCWSTR)IDC_ARROW)); + return TRUE; + } + if (ctx->state == CS_Resizing) { + SetCursor(LoadCursorW(NULL, HandleCursor(ctx->resizeHandle))); + return TRUE; + } + if (ctx->state == CS_Moving) { + SetCursor(LoadCursorW(NULL, (LPCWSTR)IDC_SIZEALL)); + return TRUE; + } + return DefWindowProc(hwnd, msg, wParam, lParam); + } + + case WM_DESTROY: { + g_screenshotOverlayWindow = NULL; + PostQuitMessage(0); + return 0; + } + } + + return DefWindowProc(hwnd, msg, wParam, lParam); +} + +// 截图线程（预截屏 + 双缓冲架构） +static void ScreenshotCaptureThread() { + // 设置 DPI 感知 + typedef DPI_AWARENESS_CONTEXT (WINAPI *SetThreadDpiAwarenessContextProc)(DPI_AWARENESS_CONTEXT); + HMODULE user32 = GetModuleHandleW(L"user32.dll"); + if (user32) { + auto setDpiProc = (SetThreadDpiAwarenessContextProc)GetProcAddress(user32, "SetThreadDpiAwarenessContext"); + if (setDpiProc) { + setDpiProc(DPI_AWARENESS_CONTEXT_PER_MONITOR_AWARE_V2); + } + } + + double dpiScale = GetDpiScaleFactor(); + + // 预截屏整个虚拟屏幕 + HDC memDC = NULL; + HBITMAP screenBitmap = NULL; + int vx, vy, vw, vh; + if (!CaptureVirtualScreen(memDC, screenBitmap, vx, vy, vw, vh, dpiScale)) { + g_isCapturing = false; + return; + } + + // 创建双缓冲 + HDC backDC = NULL; + HBITMAP backBmp = NULL; + if (!CreateBackBuffer(backDC, backBmp, vw, vh)) { + DeleteDC(memDC); + DeleteObject(screenBitmap); + g_isCapturing = false; + return; + } + + // 枚举窗口 + std::vector windows = EnumWindowsForCapture(); + + // 初始化 GDI 资源 + SCGdiResources gdi; + gdi.Init(); + // 创建选区外遮罩缓冲（需虚拟屏幕尺寸） + gdi.InitMask(vw, vh); + + // 初始化上下文 + CaptureContext ctx = {}; + ctx.state = CS_Idle; + ctx.virtualX = vx; ctx.virtualY = vy; + ctx.virtualW = vw; ctx.virtualH = vh; + ctx.startX = 0; ctx.startY = 0; + ctx.endX = 0; ctx.endY = 0; + ctx.hoveredWindow = -1; + ctx.screenBitmap = screenBitmap; + ctx.memDC = memDC; + ctx.backDC = backDC; + ctx.backBitmap = backBmp; + ctx.lastPanelRect = {0,0,0,0}; + ctx.lastSelectionRect = {0,0,0,0}; + ctx.lastLabelRect = {0,0,0,0}; + ctx.lastHighlightRect = {0,0,0,0}; + ctx.lastToolbarRect = {0,0,0,0}; + ctx.lastPopupRect = {0,0,0,0}; + ctx.selection = {0,0,0,0}; + ctx.resizeHandle = RH_None; + ctx.dragStartX = 0; ctx.dragStartY = 0; + ctx.dragStartSelection = {0,0,0,0}; + ctx.toolbarRect = {0,0,0,0}; + ctx.hoverToolbarBtn = -1; + ctx.activeTool = -1; + ctx.needFullRedraw = true; + ctx.dpiScale = dpiScale; + ctx.gdi = gdi; + ctx.windows = std::move(windows); + + // 工具栏几何（按 DPI 缩放）+ 图标位图缓存（按 DPI 预渲染） + ctx.toolbarMetrics = CalcToolbarMetrics(dpiScale); + ctx.iconCache.Init(ctx.toolbarMetrics.iconSize); + // 涂抹光标缓存（按半径预生成，DPI 缩放半径） + for (int i = 0; i < SC_MOSAIC_RADIUS_COUNT; i++) ctx.mosaicBrushCursors[i] = NULL; + ctx.mosaicBrushCursorsInited = false; + InitMosaicBrushCursors(&ctx); + // 子菜单几何（按 DPI 缩放）+ 标注绘制默认值 + ctx.popupMetrics = CalcPopupMetrics(dpiScale); + ctx.popupOpen = false; + ctx.popupRect = {0,0,0,0}; + ctx.drawColorIdx = SC_DEFAULT_COLOR_IDX; + ctx.drawThickIdx = SC_DEFAULT_THICK_IDX; + ctx.fontSizeIdx = SC_DEFAULT_FONT_IDX; + ctx.mosaicSizeIdx = SC_DEFAULT_MOSAIC_IDX; + ctx.mosaicRadiusIdx = SC_DEFAULT_MOSAIC_RADIUS_IDX; + ctx.mosaicRectMode = false; // 默认涂抹模式 + ctx.mosaicBaseDC = NULL; + ctx.mosaicBaseBitmap = NULL; + ctx.mosaicBaseW = 0; + ctx.mosaicBaseH = 0; + ctx.mosaicBaseBlockPx = 0; + ctx.mosaicDrawLastIdx = 0; + ctx.hasCurDrawing = false; + // 文字编辑初始化 + ctx.textBuf.clear(); + ctx.textAnchorX = 0; + ctx.textAnchorY = 0; + ctx.textCaretPos = 0; + ctx.textCaretVisible = true; + ctx.textCaretLastBlink = GetTickCount(); + ctx.textSelStart = -1; + ctx.textSelEnd = -1; + ctx.textDraggingSelection = false; + ctx.hoveredTextAnnotation = -1; + ctx.selectedTextAnnotation = -1; + ctx.draggingTextAnnotation = -1; + ctx.textDragStartX = 0; + ctx.textDragStartY = 0; + + // 获取初始鼠标位置和颜色 + POINT pt; + GetCursorPos(&pt); + ctx.mouseX = pt.x; + ctx.mouseY = pt.y; + ctx.currentColor = GetPixelColorFromBitmap(memDC, pt.x, pt.y, vx, vy, dpiScale); + + g_captureCtx = &ctx; + + // 注册窗口类 + WNDCLASSEXW wc = {0}; + wc.cbSize = sizeof(WNDCLASSEXW); + wc.lpfnWndProc = ScreenshotOverlayWndProc; + wc.hInstance = GetModuleHandle(NULL); + wc.hCursor = LoadCursor(NULL, IDC_ARROW); // 默认鼠标样式 + wc.lpszClassName = L"ZToolsScreenshotOverlay"; + + if (!RegisterClassExW(&wc)) { + gdi.Cleanup(); + ctx.iconCache.Cleanup(); + DeleteDC(backDC); DeleteObject(backBmp); + DeleteDC(memDC); DeleteObject(screenBitmap); + g_captureCtx = nullptr; + g_isCapturing = false; + return; + } + + // 创建普通 WS_POPUP 窗口（非分层窗口） + g_screenshotOverlayWindow = CreateWindowExW( + WS_EX_TOPMOST | WS_EX_TOOLWINDOW, + L"ZToolsScreenshotOverlay", + L"Screenshot Overlay", + WS_POPUP, + vx, vy, vw, vh, + NULL, NULL, GetModuleHandle(NULL), NULL + ); + + if (g_screenshotOverlayWindow == NULL) { + UnregisterClassW(L"ZToolsScreenshotOverlay", GetModuleHandle(NULL)); + gdi.Cleanup(); + ctx.iconCache.Cleanup(); + DeleteDC(backDC); DeleteObject(backBmp); + DeleteDC(memDC); DeleteObject(screenBitmap); + g_captureCtx = nullptr; + g_isCapturing = false; + return; + } + + ShowWindow(g_screenshotOverlayWindow, SW_SHOW); + SetForegroundWindow(g_screenshotOverlayWindow); + + // 消息循环 + MSG msg; + while (true) { + if (ctx.state == CS_Done || ctx.state == CS_Cancelled) break; + + if (PeekMessage(&msg, NULL, 0, 0, PM_REMOVE)) { + if (msg.message == WM_QUIT) break; + TranslateMessage(&msg); + DispatchMessage(&msg); + } else { + // 文字编辑态：光标闪烁（每 500ms 切换） + if (ctx.state == CS_TextEditing) { + DWORD now = GetTickCount(); + if (now - ctx.textCaretLastBlink >= 500) { + ctx.textCaretVisible = !ctx.textCaretVisible; + ctx.textCaretLastBlink = now; + InvalidateRect(g_screenshotOverlayWindow, NULL, FALSE); + } + } + + // 检查 ESC 键（窗口可能没有焦点） + if (GetAsyncKeyState(VK_ESCAPE) & 0x8000) { + if (ctx.state != CS_Done && ctx.state != CS_Cancelled) { + ctx.state = CS_Cancelled; + if (g_screenshotTsfn != nullptr) { + ScreenshotResult* result = new ScreenshotResult(); + result->success = false; + result->x = 0; result->y = 0; result->x2 = 0; result->y2 = 0; + result->width = 0; result->height = 0; + napi_call_threadsafe_function(g_screenshotTsfn, result, napi_tsfn_nonblocking); + } + DestroyWindow(g_screenshotOverlayWindow); + break; + } + } + Sleep(1); + } + } + + // 清理 + g_captureCtx = nullptr; + gdi.Cleanup(); + ctx.iconCache.Cleanup(); + FreeMosaicBase(&ctx); + FreeMosaicBrushCursors(&ctx); + DeleteDC(backDC); DeleteObject(backBmp); + DeleteDC(memDC); DeleteObject(screenBitmap); + UnregisterClassW(L"ZToolsScreenshotOverlay", GetModuleHandle(NULL)); + g_isCapturing = false; +} + +// 启动区域截图 +Napi::Value StartRegionCapture(const Napi::CallbackInfo& info) { + Napi::Env env = info.Env(); + + if (g_isCapturing) { + Napi::Error::New(env, "Screenshot already in progress").ThrowAsJavaScriptException(); + return env.Undefined(); + } + + // 可选的回调函数 + if (info.Length() > 0 && info[0].IsFunction()) { + Napi::Function callback = info[0].As(); + napi_value resource_name; + napi_create_string_utf8(env, "ScreenshotCallback", NAPI_AUTO_LENGTH, &resource_name); + + napi_status status = napi_create_threadsafe_function( + env, callback, nullptr, resource_name, + 0, 1, nullptr, nullptr, nullptr, + CallScreenshotJs, &g_screenshotTsfn + ); + + if (status != napi_ok) { + Napi::Error::New(env, "Failed to create threadsafe function").ThrowAsJavaScriptException(); + return env.Undefined(); + } + } + + g_isCapturing = true; + + g_screenshotThread = std::thread(ScreenshotCaptureThread); + g_screenshotThread.detach(); + + return env.Undefined(); +} diff --git a/src/screenshot_windows.h b/src/screenshot_windows.h new file mode 100644 index 0000000..a56f46b --- /dev/null +++ b/src/screenshot_windows.h @@ -0,0 +1,11 @@ +#pragma once + +#include +#include + +// 区域截图入口（在 Init 中注册为 "startRegionCapture" 导出） +Napi::Value StartRegionCapture(const Napi::CallbackInfo& info); + +// 获取 PNG 编码器 CLSID +// 实际定义在 binding_windows.cpp（应用图标提取模块也在使用），截图模块复用 +int GetPngEncoderClsid(CLSID* pClsid); diff --git a/src/third_party/nanosvg.h b/src/third_party/nanosvg.h new file mode 100644 index 0000000..76bcc38 --- /dev/null +++ b/src/third_party/nanosvg.h @@ -0,0 +1,3277 @@ +/* + * Copyright (c) 2013-14 Mikko Mononen memon@inside.org + * + * This software is provided 'as-is', without any express or implied + * warranty. In no event will the authors be held liable for any damages + * arising from the use of this software. + * + * Permission is granted to anyone to use this software for any purpose, + * including commercial applications, and to alter it and redistribute it + * freely, subject to the following restrictions: + * + * 1. The origin of this software must not be misrepresented; you must not + * claim that you wrote the original software. If you use this software + * in a product, an acknowledgment in the product documentation would be + * appreciated but is not required. + * 2. Altered source versions must be plainly marked as such, and must not be + * misrepresented as being the original software. + * 3. This notice may not be removed or altered from any source distribution. + * + * The SVG parser is based on Anti-Grain Geometry 2.4 SVG example + * Copyright (C) 2002-2004 Maxim Shemanarev (McSeem) (http://www.antigrain.com/) + * + * Arc calculation code based on canvg (https://code.google.com/p/canvg/) + * + * Bounding box calculation based on http://blog.hackers-cafe.net/2009/06/how-to-calculate-bezier-curves-bounding.html + * + */ + +#ifndef NANOSVG_H +#define NANOSVG_H + +#ifndef NANOSVG_CPLUSPLUS +#ifdef __cplusplus +extern "C" { +#endif +#endif + +// NanoSVG is a simple stupid single-header-file SVG parse. The output of the parser is a list of cubic bezier shapes. +// +// The library suits well for anything from rendering scalable icons in your editor application to prototyping a game. +// +// NanoSVG supports a wide range of SVG features, but something may be missing, feel free to create a pull request! +// +// The shapes in the SVG images are transformed by the viewBox and converted to specified units. +// That is, you should get the same looking data as your designed in your favorite app. +// +// NanoSVG can return the paths in few different units. For example if you want to render an image, you may choose +// to get the paths in pixels, or if you are feeding the data into a CNC-cutter, you may want to use millimeters. +// +// The units passed to NanoSVG should be one of: 'px', 'pt', 'pc' 'mm', 'cm', or 'in'. +// DPI (dots-per-inch) controls how the unit conversion is done. +// +// If you don't know or care about the units stuff, "px" and 96 should get you going. + + +/* Example Usage: + // Load SVG + NSVGimage* image; + image = nsvgParseFromFile("test.svg", "px", 96); + printf("size: %f x %f\n", image->width, image->height); + // Use... + for (NSVGshape *shape = image->shapes; shape != NULL; shape = shape->next) { + for (NSVGpath *path = shape->paths; path != NULL; path = path->next) { + for (int i = 0; i < path->npts-1; i += 3) { + float* p = &path->pts[i*2]; + drawCubicBez(p[0],p[1], p[2],p[3], p[4],p[5], p[6],p[7]); + } + } + } + // Delete + nsvgDelete(image); +*/ + +enum NSVGpaintType { + NSVG_PAINT_UNDEF = -1, + NSVG_PAINT_NONE = 0, + NSVG_PAINT_COLOR = 1, + NSVG_PAINT_LINEAR_GRADIENT = 2, + NSVG_PAINT_RADIAL_GRADIENT = 3 +}; + +enum NSVGspreadType { + NSVG_SPREAD_PAD = 0, + NSVG_SPREAD_REFLECT = 1, + NSVG_SPREAD_REPEAT = 2 +}; + +enum NSVGlineJoin { + NSVG_JOIN_MITER = 0, + NSVG_JOIN_ROUND = 1, + NSVG_JOIN_BEVEL = 2 +}; + +enum NSVGlineCap { + NSVG_CAP_BUTT = 0, + NSVG_CAP_ROUND = 1, + NSVG_CAP_SQUARE = 2 +}; + +enum NSVGfillRule { + NSVG_FILLRULE_NONZERO = 0, + NSVG_FILLRULE_EVENODD = 1 +}; + +enum NSVGflags { + NSVG_FLAGS_VISIBLE = 0x01 +}; + +enum NSVGpaintOrder { + NSVG_PAINT_FILL = 0x00, + NSVG_PAINT_MARKERS = 0x01, + NSVG_PAINT_STROKE = 0x02, +}; + +typedef struct NSVGgradientStop { + unsigned int color; + float offset; +} NSVGgradientStop; + +typedef struct NSVGgradient { + float xform[6]; + char spread; + float fx, fy; + int nstops; + NSVGgradientStop stops[1]; +} NSVGgradient; + +typedef struct NSVGpaint { + signed char type; + union { + unsigned int color; + NSVGgradient* gradient; + }; +} NSVGpaint; + +typedef struct NSVGpath +{ + float* pts; // Cubic bezier points: x0,y0, [cpx1,cpx1,cpx2,cpy2,x1,y1], ... + int npts; // Total number of bezier points. + char closed; // Flag indicating if shapes should be treated as closed. + float bounds[4]; // Tight bounding box of the shape [minx,miny,maxx,maxy]. + struct NSVGpath* next; // Pointer to next path, or NULL if last element. +} NSVGpath; + +typedef struct NSVGshape +{ + char id[64]; // Optional 'id' attr of the shape or its group + NSVGpaint fill; // Fill paint + NSVGpaint stroke; // Stroke paint + float opacity; // Opacity of the shape. + float strokeWidth; // Stroke width (scaled). + float strokeDashOffset; // Stroke dash offset (scaled). + float strokeDashArray[8]; // Stroke dash array (scaled). + char strokeDashCount; // Number of dash values in dash array. + char strokeLineJoin; // Stroke join type. + char strokeLineCap; // Stroke cap type. + float miterLimit; // Miter limit + char fillRule; // Fill rule, see NSVGfillRule. + unsigned char paintOrder; // Encoded paint order (3×2-bit fields) see NSVGpaintOrder + unsigned char flags; // Logical or of NSVG_FLAGS_* flags + float bounds[4]; // Tight bounding box of the shape [minx,miny,maxx,maxy]. + char fillGradient[64]; // Optional 'id' of fill gradient + char strokeGradient[64]; // Optional 'id' of stroke gradient + float xform[6]; // Root transformation for fill/stroke gradient + NSVGpath* paths; // Linked list of paths in the image. + struct NSVGshape* next; // Pointer to next shape, or NULL if last element. +} NSVGshape; + +typedef struct NSVGimage +{ + float width; // Width of the image. + float height; // Height of the image. + NSVGshape* shapes; // Linked list of shapes in the image. +} NSVGimage; + +// Parses SVG file from a file, returns SVG image as paths. +NSVGimage* nsvgParseFromFile(const char* filename, const char* units, float dpi); + +// Parses SVG file from a null terminated string, returns SVG image as paths. +// Important note: changes the string. +NSVGimage* nsvgParse(char* input, const char* units, float dpi); + +// Duplicates a path. +NSVGpath* nsvgDuplicatePath(NSVGpath* p); + +// Deletes an image. +void nsvgDelete(NSVGimage* image); + +#ifndef NANOSVG_CPLUSPLUS +#ifdef __cplusplus +} +#endif +#endif + +#ifdef NANOSVG_IMPLEMENTATION + +#include +#include +#include +#include + +#define NSVG_PI (3.14159265358979323846264338327f) +#define NSVG_KAPPA90 (0.5522847493f) // Length proportional to radius of a cubic bezier handle for 90deg arcs. + +#define NSVG_ALIGN_MIN 0 +#define NSVG_ALIGN_MID 1 +#define NSVG_ALIGN_MAX 2 +#define NSVG_ALIGN_NONE 0 +#define NSVG_ALIGN_MEET 1 +#define NSVG_ALIGN_SLICE 2 + +#define NSVG_NOTUSED(v) do { (void)(1 ? (void)0 : ( (void)(v) ) ); } while(0) +#define NSVG_RGB(r, g, b) (((unsigned int)r) | ((unsigned int)g << 8) | ((unsigned int)b << 16)) + +#ifdef _MSC_VER + #pragma warning (disable: 4996) // Switch off security warnings + #pragma warning (disable: 4100) // Switch off unreferenced formal parameter warnings + #ifdef __cplusplus + #define NSVG_INLINE inline + #else + #define NSVG_INLINE + #endif +#else + #define NSVG_INLINE inline +#endif + + +static int nsvg__isspace(char c) +{ + return strchr(" \t\n\v\f\r", c) != 0; +} + +static int nsvg__isdigit(char c) +{ + return c >= '0' && c <= '9'; +} + +static NSVG_INLINE float nsvg__minf(float a, float b) { return a < b ? a : b; } +static NSVG_INLINE float nsvg__maxf(float a, float b) { return a > b ? a : b; } + + +// Simple XML parser + +#define NSVG_XML_TAG 1 +#define NSVG_XML_CONTENT 2 +#define NSVG_XML_MAX_ATTRIBS 256 + +static void nsvg__parseContent(char* s, + void (*contentCb)(void* ud, const char* s), + void* ud) +{ + // Trim start white spaces + while (*s && nsvg__isspace(*s)) s++; + if (!*s) return; + + if (contentCb) + (*contentCb)(ud, s); +} + +static void nsvg__parseElement(char* s, + void (*startelCb)(void* ud, const char* el, const char** attr), + void (*endelCb)(void* ud, const char* el), + void* ud) +{ + const char* attr[NSVG_XML_MAX_ATTRIBS]; + int nattr = 0; + char* name; + int start = 0; + int end = 0; + char quote; + + // Skip white space after the '<' + while (*s && nsvg__isspace(*s)) s++; + + // Check if the tag is end tag + if (*s == '/') { + s++; + end = 1; + } else { + start = 1; + } + + // Skip comments, data and preprocessor stuff. + if (!*s || *s == '?' || *s == '!') + return; + + // Get tag name + name = s; + while (*s && !nsvg__isspace(*s)) s++; + if (*s) { *s++ = '\0'; } + + // Get attribs + while (!end && *s && nattr < NSVG_XML_MAX_ATTRIBS-3) { + char* name = NULL; + char* value = NULL; + + // Skip white space before the attrib name + while (*s && nsvg__isspace(*s)) s++; + if (!*s) break; + if (*s == '/') { + end = 1; + break; + } + name = s; + // Find end of the attrib name. + while (*s && !nsvg__isspace(*s) && *s != '=') s++; + if (*s) { *s++ = '\0'; } + // Skip until the beginning of the value. + while (*s && *s != '\"' && *s != '\'') s++; + if (!*s) break; + quote = *s; + s++; + // Store value and find the end of it. + value = s; + while (*s && *s != quote) s++; + if (*s) { *s++ = '\0'; } + + // Store only well formed attributes + if (name && value) { + attr[nattr++] = name; + attr[nattr++] = value; + } + } + + // List terminator + attr[nattr++] = 0; + attr[nattr++] = 0; + + // Call callbacks. + if (start && startelCb) + (*startelCb)(ud, name, attr); + if (end && endelCb) + (*endelCb)(ud, name); +} + +int nsvg__parseXML(char* input, + void (*startelCb)(void* ud, const char* el, const char** attr), + void (*endelCb)(void* ud, const char* el), + void (*contentCb)(void* ud, const char* s), + void* ud) +{ + char* s = input; + char* mark = s; + int state = NSVG_XML_CONTENT; + while (*s) { + if (*s == '<' && state == NSVG_XML_CONTENT) { + // Start of a tag + *s++ = '\0'; + nsvg__parseContent(mark, contentCb, ud); + mark = s; + state = NSVG_XML_TAG; + } else if (*s == '>' && state == NSVG_XML_TAG) { + // Start of a content or new tag. + *s++ = '\0'; + nsvg__parseElement(mark, startelCb, endelCb, ud); + mark = s; + state = NSVG_XML_CONTENT; + } else { + s++; + } + } + + return 1; +} + + +/* Simple SVG parser. */ + +#define NSVG_MAX_ATTR 128 + +enum NSVGgradientUnits { + NSVG_USER_SPACE = 0, + NSVG_OBJECT_SPACE = 1 +}; + +#define NSVG_MAX_DASHES 8 +#define NSVG_MAX_CLASSES 32 + +enum NSVGunits { + NSVG_UNITS_USER, + NSVG_UNITS_PX, + NSVG_UNITS_PT, + NSVG_UNITS_PC, + NSVG_UNITS_MM, + NSVG_UNITS_CM, + NSVG_UNITS_IN, + NSVG_UNITS_PERCENT, + NSVG_UNITS_EM, + NSVG_UNITS_EX +}; + +typedef struct NSVGcoordinate { + float value; + int units; +} NSVGcoordinate; + +typedef struct NSVGlinearData { + NSVGcoordinate x1, y1, x2, y2; +} NSVGlinearData; + +typedef struct NSVGradialData { + NSVGcoordinate cx, cy, r, fx, fy; +} NSVGradialData; + +typedef struct NSVGgradientData +{ + char id[64]; + char ref[64]; + signed char type; + union { + NSVGlinearData linear; + NSVGradialData radial; + }; + char spread; + char units; + float xform[6]; + int nstops; + NSVGgradientStop* stops; + struct NSVGgradientData* next; +} NSVGgradientData; + +typedef struct NSVGattrib +{ + char id[64]; + float xform[6]; + unsigned int fillColor; + unsigned int strokeColor; + float opacity; + float fillOpacity; + float strokeOpacity; + char fillGradient[64]; + char strokeGradient[64]; + float strokeWidth; + float strokeDashOffset; + float strokeDashArray[NSVG_MAX_DASHES]; + int strokeDashCount; + char strokeLineJoin; + char strokeLineCap; + float miterLimit; + char fillRule; + float fontSize; + unsigned int stopColor; + float stopOpacity; + float stopOffset; + char hasFill; + char hasStroke; + char visible; + unsigned char paintOrder; +} NSVGattrib; + +typedef struct NSVGstyleDeclaration +{ + char* className; + char* propertiesText; + struct NSVGstyleDeclaration* next; +} NSVGstyleDeclaration; + +typedef struct NSVGparser +{ + NSVGattrib attr[NSVG_MAX_ATTR]; + int attrHead; + float* pts; + int npts; + int cpts; + NSVGpath* plist; + NSVGimage* image; + NSVGstyleDeclaration* styles; + NSVGgradientData* gradients; + NSVGshape* shapesTail; + float viewMinx, viewMiny, viewWidth, viewHeight; + int alignX, alignY, alignType; + float dpi; + char pathFlag; + char defsFlag; + char styleFlag; +} NSVGparser; + +static void nsvg__xformIdentity(float* t) +{ + t[0] = 1.0f; t[1] = 0.0f; + t[2] = 0.0f; t[3] = 1.0f; + t[4] = 0.0f; t[5] = 0.0f; +} + +static void nsvg__xformSetTranslation(float* t, float tx, float ty) +{ + t[0] = 1.0f; t[1] = 0.0f; + t[2] = 0.0f; t[3] = 1.0f; + t[4] = tx; t[5] = ty; +} + +static void nsvg__xformSetScale(float* t, float sx, float sy) +{ + t[0] = sx; t[1] = 0.0f; + t[2] = 0.0f; t[3] = sy; + t[4] = 0.0f; t[5] = 0.0f; +} + +static void nsvg__xformSetSkewX(float* t, float a) +{ + t[0] = 1.0f; t[1] = 0.0f; + t[2] = tanf(a); t[3] = 1.0f; + t[4] = 0.0f; t[5] = 0.0f; +} + +static void nsvg__xformSetSkewY(float* t, float a) +{ + t[0] = 1.0f; t[1] = tanf(a); + t[2] = 0.0f; t[3] = 1.0f; + t[4] = 0.0f; t[5] = 0.0f; +} + +static void nsvg__xformSetRotation(float* t, float a) +{ + float cs = cosf(a), sn = sinf(a); + t[0] = cs; t[1] = sn; + t[2] = -sn; t[3] = cs; + t[4] = 0.0f; t[5] = 0.0f; +} + +static void nsvg__xformMultiply(float* t, float* s) +{ + float t0 = t[0] * s[0] + t[1] * s[2]; + float t2 = t[2] * s[0] + t[3] * s[2]; + float t4 = t[4] * s[0] + t[5] * s[2] + s[4]; + t[1] = t[0] * s[1] + t[1] * s[3]; + t[3] = t[2] * s[1] + t[3] * s[3]; + t[5] = t[4] * s[1] + t[5] * s[3] + s[5]; + t[0] = t0; + t[2] = t2; + t[4] = t4; +} + +static void nsvg__xformInverse(float* inv, float* t) +{ + double invdet, det = (double)t[0] * t[3] - (double)t[2] * t[1]; + if (det > -1e-6 && det < 1e-6) { + nsvg__xformIdentity(t); + return; + } + invdet = 1.0 / det; + inv[0] = (float)(t[3] * invdet); + inv[2] = (float)(-t[2] * invdet); + inv[4] = (float)(((double)t[2] * t[5] - (double)t[3] * t[4]) * invdet); + inv[1] = (float)(-t[1] * invdet); + inv[3] = (float)(t[0] * invdet); + inv[5] = (float)(((double)t[1] * t[4] - (double)t[0] * t[5]) * invdet); +} + +static void nsvg__xformPremultiply(float* t, float* s) +{ + float s2[6]; + memcpy(s2, s, sizeof(float)*6); + nsvg__xformMultiply(s2, t); + memcpy(t, s2, sizeof(float)*6); +} + +static void nsvg__xformPoint(float* dx, float* dy, float x, float y, float* t) +{ + *dx = x*t[0] + y*t[2] + t[4]; + *dy = x*t[1] + y*t[3] + t[5]; +} + +static void nsvg__xformVec(float* dx, float* dy, float x, float y, float* t) +{ + *dx = x*t[0] + y*t[2]; + *dy = x*t[1] + y*t[3]; +} + +#define NSVG_EPSILON (1e-12) + +static int nsvg__ptInBounds(float* pt, float* bounds) +{ + return pt[0] >= bounds[0] && pt[0] <= bounds[2] && pt[1] >= bounds[1] && pt[1] <= bounds[3]; +} + + +static double nsvg__evalBezier(double t, double p0, double p1, double p2, double p3) +{ + double it = 1.0-t; + return it*it*it*p0 + 3.0*it*it*t*p1 + 3.0*it*t*t*p2 + t*t*t*p3; +} + +static void nsvg__curveBounds(float* bounds, float* curve) +{ + int i, j, count; + double roots[2], a, b, c, b2ac, t, v; + float* v0 = &curve[0]; + float* v1 = &curve[2]; + float* v2 = &curve[4]; + float* v3 = &curve[6]; + + // Start the bounding box by end points + bounds[0] = nsvg__minf(v0[0], v3[0]); + bounds[1] = nsvg__minf(v0[1], v3[1]); + bounds[2] = nsvg__maxf(v0[0], v3[0]); + bounds[3] = nsvg__maxf(v0[1], v3[1]); + + // Bezier curve fits inside the convex hull of it's control points. + // If control points are inside the bounds, we're done. + if (nsvg__ptInBounds(v1, bounds) && nsvg__ptInBounds(v2, bounds)) + return; + + // Add bezier curve inflection points in X and Y. + for (i = 0; i < 2; i++) { + a = -3.0 * v0[i] + 9.0 * v1[i] - 9.0 * v2[i] + 3.0 * v3[i]; + b = 6.0 * v0[i] - 12.0 * v1[i] + 6.0 * v2[i]; + c = 3.0 * v1[i] - 3.0 * v0[i]; + count = 0; + if (fabs(a) < NSVG_EPSILON) { + if (fabs(b) > NSVG_EPSILON) { + t = -c / b; + if (t > NSVG_EPSILON && t < 1.0-NSVG_EPSILON) + roots[count++] = t; + } + } else { + b2ac = b*b - 4.0*c*a; + if (b2ac > NSVG_EPSILON) { + t = (-b + sqrt(b2ac)) / (2.0 * a); + if (t > NSVG_EPSILON && t < 1.0-NSVG_EPSILON) + roots[count++] = t; + t = (-b - sqrt(b2ac)) / (2.0 * a); + if (t > NSVG_EPSILON && t < 1.0-NSVG_EPSILON) + roots[count++] = t; + } + } + for (j = 0; j < count; j++) { + v = nsvg__evalBezier(roots[j], v0[i], v1[i], v2[i], v3[i]); + bounds[0+i] = nsvg__minf(bounds[0+i], (float)v); + bounds[2+i] = nsvg__maxf(bounds[2+i], (float)v); + } + } +} + +static unsigned char nsvg__encodePaintOrder(enum NSVGpaintOrder a, enum NSVGpaintOrder b, enum NSVGpaintOrder c) { + return (a & 0x03) | ((b & 0x03) << 2) | ((c & 0x03) << 4); +} + +static NSVGparser* nsvg__createParser(void) +{ + NSVGparser* p; + p = (NSVGparser*)malloc(sizeof(NSVGparser)); + if (p == NULL) goto error; + memset(p, 0, sizeof(NSVGparser)); + + p->image = (NSVGimage*)malloc(sizeof(NSVGimage)); + if (p->image == NULL) goto error; + memset(p->image, 0, sizeof(NSVGimage)); + + // Init style + nsvg__xformIdentity(p->attr[0].xform); + memset(p->attr[0].id, 0, sizeof p->attr[0].id); + p->attr[0].fillColor = NSVG_RGB(0,0,0); + p->attr[0].strokeColor = NSVG_RGB(0,0,0); + p->attr[0].opacity = 1; + p->attr[0].fillOpacity = 1; + p->attr[0].strokeOpacity = 1; + p->attr[0].stopOpacity = 1; + p->attr[0].strokeWidth = 1; + p->attr[0].strokeLineJoin = NSVG_JOIN_MITER; + p->attr[0].strokeLineCap = NSVG_CAP_BUTT; + p->attr[0].miterLimit = 4; + p->attr[0].fillRule = NSVG_FILLRULE_NONZERO; + p->attr[0].hasFill = 1; + p->attr[0].visible = 1; + p->attr[0].paintOrder = nsvg__encodePaintOrder(NSVG_PAINT_FILL, NSVG_PAINT_STROKE, NSVG_PAINT_MARKERS); + + return p; + +error: + if (p) { + if (p->image) free(p->image); + free(p); + } + return NULL; +} +static void nsvg__deleteStyles(NSVGstyleDeclaration* style) { + while (style) { + NSVGstyleDeclaration* next = style->next; + free(style->className); + free(style->propertiesText); + free(style); + style = next; + } +} + +static void nsvg__deletePaths(NSVGpath* path) +{ + while (path) { + NSVGpath *next = path->next; + if (path->pts != NULL) + free(path->pts); + free(path); + path = next; + } +} + +static void nsvg__deletePaint(NSVGpaint* paint) +{ + if (paint->type == NSVG_PAINT_LINEAR_GRADIENT || paint->type == NSVG_PAINT_RADIAL_GRADIENT) + free(paint->gradient); +} + +static void nsvg__deleteGradientData(NSVGgradientData* grad) +{ + NSVGgradientData* next; + while (grad != NULL) { + next = grad->next; + free(grad->stops); + free(grad); + grad = next; + } +} + +static void nsvg__deleteParser(NSVGparser* p) +{ + if (p != NULL) { + nsvg__deleteStyles(p->styles); + nsvg__deletePaths(p->plist); + nsvg__deleteGradientData(p->gradients); + nsvgDelete(p->image); + free(p->pts); + free(p); + } +} + +static void nsvg__resetPath(NSVGparser* p) +{ + p->npts = 0; +} + +static void nsvg__addPoint(NSVGparser* p, float x, float y) +{ + if (p->npts+1 > p->cpts) { + p->cpts = p->cpts ? p->cpts*2 : 8; + p->pts = (float*)realloc(p->pts, p->cpts*2*sizeof(float)); + if (!p->pts) return; + } + p->pts[p->npts*2+0] = x; + p->pts[p->npts*2+1] = y; + p->npts++; +} + +static void nsvg__moveTo(NSVGparser* p, float x, float y) +{ + if (p->npts > 0) { + p->pts[(p->npts-1)*2+0] = x; + p->pts[(p->npts-1)*2+1] = y; + } else { + nsvg__addPoint(p, x, y); + } +} + +static void nsvg__lineTo(NSVGparser* p, float x, float y) +{ + float px,py, dx,dy; + if (p->npts > 0) { + px = p->pts[(p->npts-1)*2+0]; + py = p->pts[(p->npts-1)*2+1]; + dx = x - px; + dy = y - py; + nsvg__addPoint(p, px + dx/3.0f, py + dy/3.0f); + nsvg__addPoint(p, x - dx/3.0f, y - dy/3.0f); + nsvg__addPoint(p, x, y); + } +} + +static void nsvg__cubicBezTo(NSVGparser* p, float cpx1, float cpy1, float cpx2, float cpy2, float x, float y) +{ + if (p->npts > 0) { + nsvg__addPoint(p, cpx1, cpy1); + nsvg__addPoint(p, cpx2, cpy2); + nsvg__addPoint(p, x, y); + } +} + +static NSVGattrib* nsvg__getAttr(NSVGparser* p) +{ + return &p->attr[p->attrHead]; +} + +static void nsvg__pushAttr(NSVGparser* p) +{ + if (p->attrHead < NSVG_MAX_ATTR-1) { + p->attrHead++; + memcpy(&p->attr[p->attrHead], &p->attr[p->attrHead-1], sizeof(NSVGattrib)); + } +} + +static void nsvg__popAttr(NSVGparser* p) +{ + if (p->attrHead > 0) + p->attrHead--; +} + +static float nsvg__actualOrigX(NSVGparser* p) +{ + return p->viewMinx; +} + +static float nsvg__actualOrigY(NSVGparser* p) +{ + return p->viewMiny; +} + +static float nsvg__actualWidth(NSVGparser* p) +{ + return p->viewWidth; +} + +static float nsvg__actualHeight(NSVGparser* p) +{ + return p->viewHeight; +} + +static float nsvg__actualLength(NSVGparser* p) +{ + float w = nsvg__actualWidth(p), h = nsvg__actualHeight(p); + return sqrtf(w*w + h*h) / sqrtf(2.0f); +} + +static float nsvg__convertToPixels(NSVGparser* p, NSVGcoordinate c, float orig, float length) +{ + NSVGattrib* attr = nsvg__getAttr(p); + switch (c.units) { + case NSVG_UNITS_USER: return c.value; + case NSVG_UNITS_PX: return c.value; + case NSVG_UNITS_PT: return c.value / 72.0f * p->dpi; + case NSVG_UNITS_PC: return c.value / 6.0f * p->dpi; + case NSVG_UNITS_MM: return c.value / 25.4f * p->dpi; + case NSVG_UNITS_CM: return c.value / 2.54f * p->dpi; + case NSVG_UNITS_IN: return c.value * p->dpi; + case NSVG_UNITS_EM: return c.value * attr->fontSize; + case NSVG_UNITS_EX: return c.value * attr->fontSize * 0.52f; // x-height of Helvetica. + case NSVG_UNITS_PERCENT: return orig + c.value / 100.0f * length; + default: return c.value; + } + return c.value; +} + +static NSVGgradientData* nsvg__findGradientData(NSVGparser* p, const char* id) +{ + NSVGgradientData* grad = p->gradients; + if (id == NULL || *id == '\0') + return NULL; + while (grad != NULL) { + if (strcmp(grad->id, id) == 0) + return grad; + grad = grad->next; + } + return NULL; +} + +static NSVGgradient* nsvg__createGradient(NSVGparser* p, const char* id, const float* localBounds, float *xform, signed char* paintType) +{ + NSVGgradientData* data = NULL; + NSVGgradientData* ref = NULL; + NSVGgradientStop* stops = NULL; + NSVGgradient* grad; + float ox, oy, sw, sh, sl; + int nstops = 0; + int refIter; + + data = nsvg__findGradientData(p, id); + if (data == NULL) return NULL; + + // TODO: use ref to fill in all unset values too. + ref = data; + refIter = 0; + while (ref != NULL) { + NSVGgradientData* nextRef = NULL; + if (stops == NULL && ref->stops != NULL) { + stops = ref->stops; + nstops = ref->nstops; + break; + } + nextRef = nsvg__findGradientData(p, ref->ref); + if (nextRef == ref) break; // prevent infite loops on malformed data + ref = nextRef; + refIter++; + if (refIter > 32) break; // prevent infite loops on malformed data + } + if (stops == NULL) return NULL; + + grad = (NSVGgradient*)malloc(sizeof(NSVGgradient) + sizeof(NSVGgradientStop)*(nstops-1)); + if (grad == NULL) return NULL; + + // The shape width and height. + if (data->units == NSVG_OBJECT_SPACE) { + ox = localBounds[0]; + oy = localBounds[1]; + sw = localBounds[2] - localBounds[0]; + sh = localBounds[3] - localBounds[1]; + } else { + ox = nsvg__actualOrigX(p); + oy = nsvg__actualOrigY(p); + sw = nsvg__actualWidth(p); + sh = nsvg__actualHeight(p); + } + sl = sqrtf(sw*sw + sh*sh) / sqrtf(2.0f); + + if (data->type == NSVG_PAINT_LINEAR_GRADIENT) { + float x1, y1, x2, y2, dx, dy; + x1 = nsvg__convertToPixels(p, data->linear.x1, ox, sw); + y1 = nsvg__convertToPixels(p, data->linear.y1, oy, sh); + x2 = nsvg__convertToPixels(p, data->linear.x2, ox, sw); + y2 = nsvg__convertToPixels(p, data->linear.y2, oy, sh); + // Calculate transform aligned to the line + dx = x2 - x1; + dy = y2 - y1; + grad->xform[0] = dy; grad->xform[1] = -dx; + grad->xform[2] = dx; grad->xform[3] = dy; + grad->xform[4] = x1; grad->xform[5] = y1; + } else { + float cx, cy, fx, fy, r; + cx = nsvg__convertToPixels(p, data->radial.cx, ox, sw); + cy = nsvg__convertToPixels(p, data->radial.cy, oy, sh); + fx = nsvg__convertToPixels(p, data->radial.fx, ox, sw); + fy = nsvg__convertToPixels(p, data->radial.fy, oy, sh); + r = nsvg__convertToPixels(p, data->radial.r, 0, sl); + // Calculate transform aligned to the circle + grad->xform[0] = r; grad->xform[1] = 0; + grad->xform[2] = 0; grad->xform[3] = r; + grad->xform[4] = cx; grad->xform[5] = cy; + grad->fx = fx / r; + grad->fy = fy / r; + } + + nsvg__xformMultiply(grad->xform, data->xform); + nsvg__xformMultiply(grad->xform, xform); + + grad->spread = data->spread; + memcpy(grad->stops, stops, nstops*sizeof(NSVGgradientStop)); + grad->nstops = nstops; + + *paintType = data->type; + + return grad; +} + +static float nsvg__getAverageScale(float* t) +{ + float sx = sqrtf(t[0]*t[0] + t[2]*t[2]); + float sy = sqrtf(t[1]*t[1] + t[3]*t[3]); + return (sx + sy) * 0.5f; +} + +static void nsvg__getLocalBounds(float* bounds, NSVGshape *shape, float* xform) +{ + NSVGpath* path; + float curve[4*2], curveBounds[4]; + int i, first = 1; + for (path = shape->paths; path != NULL; path = path->next) { + nsvg__xformPoint(&curve[0], &curve[1], path->pts[0], path->pts[1], xform); + for (i = 0; i < path->npts-1; i += 3) { + nsvg__xformPoint(&curve[2], &curve[3], path->pts[(i+1)*2], path->pts[(i+1)*2+1], xform); + nsvg__xformPoint(&curve[4], &curve[5], path->pts[(i+2)*2], path->pts[(i+2)*2+1], xform); + nsvg__xformPoint(&curve[6], &curve[7], path->pts[(i+3)*2], path->pts[(i+3)*2+1], xform); + nsvg__curveBounds(curveBounds, curve); + if (first) { + bounds[0] = curveBounds[0]; + bounds[1] = curveBounds[1]; + bounds[2] = curveBounds[2]; + bounds[3] = curveBounds[3]; + first = 0; + } else { + bounds[0] = nsvg__minf(bounds[0], curveBounds[0]); + bounds[1] = nsvg__minf(bounds[1], curveBounds[1]); + bounds[2] = nsvg__maxf(bounds[2], curveBounds[2]); + bounds[3] = nsvg__maxf(bounds[3], curveBounds[3]); + } + curve[0] = curve[6]; + curve[1] = curve[7]; + } + } +} + +static void nsvg__addShape(NSVGparser* p) +{ + NSVGattrib* attr = nsvg__getAttr(p); + float scale = 1.0f; + NSVGshape* shape; + NSVGpath* path; + int i; + + if (p->plist == NULL) + return; + + shape = (NSVGshape*)malloc(sizeof(NSVGshape)); + if (shape == NULL) goto error; + memset(shape, 0, sizeof(NSVGshape)); + + memcpy(shape->id, attr->id, sizeof shape->id); + memcpy(shape->fillGradient, attr->fillGradient, sizeof shape->fillGradient); + memcpy(shape->strokeGradient, attr->strokeGradient, sizeof shape->strokeGradient); + memcpy(shape->xform, attr->xform, sizeof shape->xform); + scale = nsvg__getAverageScale(attr->xform); + shape->strokeWidth = attr->strokeWidth * scale; + shape->strokeDashOffset = attr->strokeDashOffset * scale; + shape->strokeDashCount = (char)attr->strokeDashCount; + for (i = 0; i < attr->strokeDashCount; i++) + shape->strokeDashArray[i] = attr->strokeDashArray[i] * scale; + shape->strokeLineJoin = attr->strokeLineJoin; + shape->strokeLineCap = attr->strokeLineCap; + shape->miterLimit = attr->miterLimit; + shape->fillRule = attr->fillRule; + shape->opacity = attr->opacity; + shape->paintOrder = attr->paintOrder; + + shape->paths = p->plist; + p->plist = NULL; + + // Calculate shape bounds + shape->bounds[0] = shape->paths->bounds[0]; + shape->bounds[1] = shape->paths->bounds[1]; + shape->bounds[2] = shape->paths->bounds[2]; + shape->bounds[3] = shape->paths->bounds[3]; + for (path = shape->paths->next; path != NULL; path = path->next) { + shape->bounds[0] = nsvg__minf(shape->bounds[0], path->bounds[0]); + shape->bounds[1] = nsvg__minf(shape->bounds[1], path->bounds[1]); + shape->bounds[2] = nsvg__maxf(shape->bounds[2], path->bounds[2]); + shape->bounds[3] = nsvg__maxf(shape->bounds[3], path->bounds[3]); + } + + // Set fill + if (attr->hasFill == 0) { + shape->fill.type = NSVG_PAINT_NONE; + } else if (attr->hasFill == 1) { + shape->fill.type = NSVG_PAINT_COLOR; + shape->fill.color = attr->fillColor; + shape->fill.color |= (unsigned int)(attr->fillOpacity*255) << 24; + } else if (attr->hasFill == 2) { + shape->fill.type = NSVG_PAINT_UNDEF; + } + + // Set stroke + if (attr->hasStroke == 0) { + shape->stroke.type = NSVG_PAINT_NONE; + } else if (attr->hasStroke == 1) { + shape->stroke.type = NSVG_PAINT_COLOR; + shape->stroke.color = attr->strokeColor; + shape->stroke.color |= (unsigned int)(attr->strokeOpacity*255) << 24; + } else if (attr->hasStroke == 2) { + shape->stroke.type = NSVG_PAINT_UNDEF; + } + + // Set flags + shape->flags = (attr->visible ? NSVG_FLAGS_VISIBLE : 0x00); + + // Add to tail + if (p->image->shapes == NULL) + p->image->shapes = shape; + else + p->shapesTail->next = shape; + p->shapesTail = shape; + + return; + +error: + if (shape) free(shape); +} + +static void nsvg__addPath(NSVGparser* p, char closed) +{ + NSVGattrib* attr = nsvg__getAttr(p); + NSVGpath* path = NULL; + float bounds[4]; + float* curve; + int i; + + if (p->npts < 4) + return; + + if (closed) + nsvg__lineTo(p, p->pts[0], p->pts[1]); + + // Expect 1 + N*3 points (N = number of cubic bezier segments). + if ((p->npts % 3) != 1) + return; + + path = (NSVGpath*)malloc(sizeof(NSVGpath)); + if (path == NULL) goto error; + memset(path, 0, sizeof(NSVGpath)); + + path->pts = (float*)malloc(p->npts*2*sizeof(float)); + if (path->pts == NULL) goto error; + path->closed = closed; + path->npts = p->npts; + + // Transform path. + for (i = 0; i < p->npts; ++i) + nsvg__xformPoint(&path->pts[i*2], &path->pts[i*2+1], p->pts[i*2], p->pts[i*2+1], attr->xform); + + // Find bounds + for (i = 0; i < path->npts-1; i += 3) { + curve = &path->pts[i*2]; + nsvg__curveBounds(bounds, curve); + if (i == 0) { + path->bounds[0] = bounds[0]; + path->bounds[1] = bounds[1]; + path->bounds[2] = bounds[2]; + path->bounds[3] = bounds[3]; + } else { + path->bounds[0] = nsvg__minf(path->bounds[0], bounds[0]); + path->bounds[1] = nsvg__minf(path->bounds[1], bounds[1]); + path->bounds[2] = nsvg__maxf(path->bounds[2], bounds[2]); + path->bounds[3] = nsvg__maxf(path->bounds[3], bounds[3]); + } + } + + path->next = p->plist; + p->plist = path; + + return; + +error: + if (path != NULL) { + if (path->pts != NULL) free(path->pts); + free(path); + } +} + +// We roll our own string to float because the std library one uses locale and messes things up. +static double nsvg__atof(const char* s) +{ + char* cur = (char*)s; + char* end = NULL; + double res = 0.0, sign = 1.0; + long long intPart = 0, fracPart = 0; + char hasIntPart = 0, hasFracPart = 0; + + // Parse optional sign + if (*cur == '+') { + cur++; + } else if (*cur == '-') { + sign = -1; + cur++; + } + + // Parse integer part + if (nsvg__isdigit(*cur)) { + // Parse digit sequence + intPart = strtoll(cur, &end, 10); + if (cur != end) { + res = (double)intPart; + hasIntPart = 1; + cur = end; + } + } + + // Parse fractional part. + if (*cur == '.') { + cur++; // Skip '.' + if (nsvg__isdigit(*cur)) { + // Parse digit sequence + fracPart = strtoll(cur, &end, 10); + if (cur != end) { + res += (double)fracPart / pow(10.0, (double)(end - cur)); + hasFracPart = 1; + cur = end; + } + } + } + + // A valid number should have integer or fractional part. + if (!hasIntPart && !hasFracPart) + return 0.0; + + // Parse optional exponent + if (*cur == 'e' || *cur == 'E') { + long expPart = 0; + cur++; // skip 'E' + expPart = strtol(cur, &end, 10); // Parse digit sequence with sign + if (cur != end) { + res *= pow(10.0, (double)expPart); + } + } + + return res * sign; +} + + +static const char* nsvg__parseNumber(const char* s, char* it, const int size) +{ + const int last = size-1; + int i = 0; + + // sign + if (*s == '-' || *s == '+') { + if (i < last) it[i++] = *s; + s++; + } + // integer part + while (*s && nsvg__isdigit(*s)) { + if (i < last) it[i++] = *s; + s++; + } + if (*s == '.') { + // decimal point + if (i < last) it[i++] = *s; + s++; + // fraction part + while (*s && nsvg__isdigit(*s)) { + if (i < last) it[i++] = *s; + s++; + } + } + // exponent + if ((*s == 'e' || *s == 'E') && (s[1] != 'm' && s[1] != 'x')) { + if (i < last) it[i++] = *s; + s++; + if (*s == '-' || *s == '+') { + if (i < last) it[i++] = *s; + s++; + } + while (*s && nsvg__isdigit(*s)) { + if (i < last) it[i++] = *s; + s++; + } + } + it[i] = '\0'; + + return s; +} + +static const char* nsvg__getNextPathItemWhenArcFlag(const char* s, char* it) +{ + it[0] = '\0'; + while (*s && (nsvg__isspace(*s) || *s == ',')) s++; + if (!*s) return s; + if (*s == '0' || *s == '1') { + it[0] = *s++; + it[1] = '\0'; + return s; + } + return s; +} + +static const char* nsvg__getNextPathItem(const char* s, char* it) +{ + it[0] = '\0'; + // Skip white spaces and commas + while (*s && (nsvg__isspace(*s) || *s == ',')) s++; + if (!*s) return s; + if (*s == '-' || *s == '+' || *s == '.' || nsvg__isdigit(*s)) { + s = nsvg__parseNumber(s, it, 64); + } else { + // Parse command + it[0] = *s++; + it[1] = '\0'; + return s; + } + + return s; +} + +static unsigned int nsvg__parseColorHex(const char* str) +{ + unsigned int r=0, g=0, b=0; + if (sscanf(str, "#%2x%2x%2x", &r, &g, &b) == 3 ) // 2 digit hex + return NSVG_RGB(r, g, b); + if (sscanf(str, "#%1x%1x%1x", &r, &g, &b) == 3 ) // 1 digit hex, e.g. #abc -> 0xccbbaa + return NSVG_RGB(r*17, g*17, b*17); // same effect as (r<<4|r), (g<<4|g), .. + return NSVG_RGB(128, 128, 128); +} + +// Parse rgb color. The pointer 'str' must point at "rgb(" (4+ characters). +// This function returns gray (rgb(128, 128, 128) == '#808080') on parse errors +// for backwards compatibility. Note: other image viewers return black instead. + +static unsigned int nsvg__parseColorRGB(const char* str) +{ + int i; + unsigned int rgbi[3]; + float rgbf[3]; + // try decimal integers first + if (sscanf(str, "rgb(%u, %u, %u)", &rgbi[0], &rgbi[1], &rgbi[2]) != 3) { + // integers failed, try percent values (float, locale independent) + const char delimiter[3] = {',', ',', ')'}; + str += 4; // skip "rgb(" + for (i = 0; i < 3; i++) { + while (*str && (nsvg__isspace(*str))) str++; // skip leading spaces + if (*str == '+') str++; // skip '+' (don't allow '-') + if (!*str) break; + rgbf[i] = nsvg__atof(str); + + // Note 1: it would be great if nsvg__atof() returned how many + // bytes it consumed but it doesn't. We need to skip the number, + // the '%' character, spaces, and the delimiter ',' or ')'. + + // Note 2: The following code does not allow values like "33.%", + // i.e. a decimal point w/o fractional part, but this is consistent + // with other image viewers, e.g. firefox, chrome, eog, gimp. + + while (*str && nsvg__isdigit(*str)) str++; // skip integer part + if (*str == '.') { + str++; + if (!nsvg__isdigit(*str)) break; // error: no digit after '.' + while (*str && nsvg__isdigit(*str)) str++; // skip fractional part + } + if (*str == '%') str++; else break; + while (*str && nsvg__isspace(*str)) str++; + if (*str == delimiter[i]) str++; + else break; + } + if (i == 3) { + rgbi[0] = roundf(rgbf[0] * 2.55f); + rgbi[1] = roundf(rgbf[1] * 2.55f); + rgbi[2] = roundf(rgbf[2] * 2.55f); + } else { + rgbi[0] = rgbi[1] = rgbi[2] = 128; + } + } + // clip values as the CSS spec requires + for (i = 0; i < 3; i++) { + if (rgbi[i] > 255) rgbi[i] = 255; + } + return NSVG_RGB(rgbi[0], rgbi[1], rgbi[2]); +} + +typedef struct NSVGNamedColor { + const char* name; + unsigned int color; +} NSVGNamedColor; + +NSVGNamedColor nsvg__colors[] = { + + { "red", NSVG_RGB(255, 0, 0) }, + { "green", NSVG_RGB( 0, 128, 0) }, + { "blue", NSVG_RGB( 0, 0, 255) }, + { "yellow", NSVG_RGB(255, 255, 0) }, + { "cyan", NSVG_RGB( 0, 255, 255) }, + { "magenta", NSVG_RGB(255, 0, 255) }, + { "black", NSVG_RGB( 0, 0, 0) }, + { "grey", NSVG_RGB(128, 128, 128) }, + { "gray", NSVG_RGB(128, 128, 128) }, + { "white", NSVG_RGB(255, 255, 255) }, + +#ifdef NANOSVG_ALL_COLOR_KEYWORDS + { "aliceblue", NSVG_RGB(240, 248, 255) }, + { "antiquewhite", NSVG_RGB(250, 235, 215) }, + { "aqua", NSVG_RGB( 0, 255, 255) }, + { "aquamarine", NSVG_RGB(127, 255, 212) }, + { "azure", NSVG_RGB(240, 255, 255) }, + { "beige", NSVG_RGB(245, 245, 220) }, + { "bisque", NSVG_RGB(255, 228, 196) }, + { "blanchedalmond", NSVG_RGB(255, 235, 205) }, + { "blueviolet", NSVG_RGB(138, 43, 226) }, + { "brown", NSVG_RGB(165, 42, 42) }, + { "burlywood", NSVG_RGB(222, 184, 135) }, + { "cadetblue", NSVG_RGB( 95, 158, 160) }, + { "chartreuse", NSVG_RGB(127, 255, 0) }, + { "chocolate", NSVG_RGB(210, 105, 30) }, + { "coral", NSVG_RGB(255, 127, 80) }, + { "cornflowerblue", NSVG_RGB(100, 149, 237) }, + { "cornsilk", NSVG_RGB(255, 248, 220) }, + { "crimson", NSVG_RGB(220, 20, 60) }, + { "darkblue", NSVG_RGB( 0, 0, 139) }, + { "darkcyan", NSVG_RGB( 0, 139, 139) }, + { "darkgoldenrod", NSVG_RGB(184, 134, 11) }, + { "darkgray", NSVG_RGB(169, 169, 169) }, + { "darkgreen", NSVG_RGB( 0, 100, 0) }, + { "darkgrey", NSVG_RGB(169, 169, 169) }, + { "darkkhaki", NSVG_RGB(189, 183, 107) }, + { "darkmagenta", NSVG_RGB(139, 0, 139) }, + { "darkolivegreen", NSVG_RGB( 85, 107, 47) }, + { "darkorange", NSVG_RGB(255, 140, 0) }, + { "darkorchid", NSVG_RGB(153, 50, 204) }, + { "darkred", NSVG_RGB(139, 0, 0) }, + { "darksalmon", NSVG_RGB(233, 150, 122) }, + { "darkseagreen", NSVG_RGB(143, 188, 143) }, + { "darkslateblue", NSVG_RGB( 72, 61, 139) }, + { "darkslategray", NSVG_RGB( 47, 79, 79) }, + { "darkslategrey", NSVG_RGB( 47, 79, 79) }, + { "darkturquoise", NSVG_RGB( 0, 206, 209) }, + { "darkviolet", NSVG_RGB(148, 0, 211) }, + { "deeppink", NSVG_RGB(255, 20, 147) }, + { "deepskyblue", NSVG_RGB( 0, 191, 255) }, + { "dimgray", NSVG_RGB(105, 105, 105) }, + { "dimgrey", NSVG_RGB(105, 105, 105) }, + { "dodgerblue", NSVG_RGB( 30, 144, 255) }, + { "firebrick", NSVG_RGB(178, 34, 34) }, + { "floralwhite", NSVG_RGB(255, 250, 240) }, + { "forestgreen", NSVG_RGB( 34, 139, 34) }, + { "fuchsia", NSVG_RGB(255, 0, 255) }, + { "gainsboro", NSVG_RGB(220, 220, 220) }, + { "ghostwhite", NSVG_RGB(248, 248, 255) }, + { "gold", NSVG_RGB(255, 215, 0) }, + { "goldenrod", NSVG_RGB(218, 165, 32) }, + { "greenyellow", NSVG_RGB(173, 255, 47) }, + { "honeydew", NSVG_RGB(240, 255, 240) }, + { "hotpink", NSVG_RGB(255, 105, 180) }, + { "indianred", NSVG_RGB(205, 92, 92) }, + { "indigo", NSVG_RGB( 75, 0, 130) }, + { "ivory", NSVG_RGB(255, 255, 240) }, + { "khaki", NSVG_RGB(240, 230, 140) }, + { "lavender", NSVG_RGB(230, 230, 250) }, + { "lavenderblush", NSVG_RGB(255, 240, 245) }, + { "lawngreen", NSVG_RGB(124, 252, 0) }, + { "lemonchiffon", NSVG_RGB(255, 250, 205) }, + { "lightblue", NSVG_RGB(173, 216, 230) }, + { "lightcoral", NSVG_RGB(240, 128, 128) }, + { "lightcyan", NSVG_RGB(224, 255, 255) }, + { "lightgoldenrodyellow", NSVG_RGB(250, 250, 210) }, + { "lightgray", NSVG_RGB(211, 211, 211) }, + { "lightgreen", NSVG_RGB(144, 238, 144) }, + { "lightgrey", NSVG_RGB(211, 211, 211) }, + { "lightpink", NSVG_RGB(255, 182, 193) }, + { "lightsalmon", NSVG_RGB(255, 160, 122) }, + { "lightseagreen", NSVG_RGB( 32, 178, 170) }, + { "lightskyblue", NSVG_RGB(135, 206, 250) }, + { "lightslategray", NSVG_RGB(119, 136, 153) }, + { "lightslategrey", NSVG_RGB(119, 136, 153) }, + { "lightsteelblue", NSVG_RGB(176, 196, 222) }, + { "lightyellow", NSVG_RGB(255, 255, 224) }, + { "lime", NSVG_RGB( 0, 255, 0) }, + { "limegreen", NSVG_RGB( 50, 205, 50) }, + { "linen", NSVG_RGB(250, 240, 230) }, + { "maroon", NSVG_RGB(128, 0, 0) }, + { "mediumaquamarine", NSVG_RGB(102, 205, 170) }, + { "mediumblue", NSVG_RGB( 0, 0, 205) }, + { "mediumorchid", NSVG_RGB(186, 85, 211) }, + { "mediumpurple", NSVG_RGB(147, 112, 219) }, + { "mediumseagreen", NSVG_RGB( 60, 179, 113) }, + { "mediumslateblue", NSVG_RGB(123, 104, 238) }, + { "mediumspringgreen", NSVG_RGB( 0, 250, 154) }, + { "mediumturquoise", NSVG_RGB( 72, 209, 204) }, + { "mediumvioletred", NSVG_RGB(199, 21, 133) }, + { "midnightblue", NSVG_RGB( 25, 25, 112) }, + { "mintcream", NSVG_RGB(245, 255, 250) }, + { "mistyrose", NSVG_RGB(255, 228, 225) }, + { "moccasin", NSVG_RGB(255, 228, 181) }, + { "navajowhite", NSVG_RGB(255, 222, 173) }, + { "navy", NSVG_RGB( 0, 0, 128) }, + { "oldlace", NSVG_RGB(253, 245, 230) }, + { "olive", NSVG_RGB(128, 128, 0) }, + { "olivedrab", NSVG_RGB(107, 142, 35) }, + { "orange", NSVG_RGB(255, 165, 0) }, + { "orangered", NSVG_RGB(255, 69, 0) }, + { "orchid", NSVG_RGB(218, 112, 214) }, + { "palegoldenrod", NSVG_RGB(238, 232, 170) }, + { "palegreen", NSVG_RGB(152, 251, 152) }, + { "paleturquoise", NSVG_RGB(175, 238, 238) }, + { "palevioletred", NSVG_RGB(219, 112, 147) }, + { "papayawhip", NSVG_RGB(255, 239, 213) }, + { "peachpuff", NSVG_RGB(255, 218, 185) }, + { "peru", NSVG_RGB(205, 133, 63) }, + { "pink", NSVG_RGB(255, 192, 203) }, + { "plum", NSVG_RGB(221, 160, 221) }, + { "powderblue", NSVG_RGB(176, 224, 230) }, + { "purple", NSVG_RGB(128, 0, 128) }, + { "rosybrown", NSVG_RGB(188, 143, 143) }, + { "royalblue", NSVG_RGB( 65, 105, 225) }, + { "saddlebrown", NSVG_RGB(139, 69, 19) }, + { "salmon", NSVG_RGB(250, 128, 114) }, + { "sandybrown", NSVG_RGB(244, 164, 96) }, + { "seagreen", NSVG_RGB( 46, 139, 87) }, + { "seashell", NSVG_RGB(255, 245, 238) }, + { "sienna", NSVG_RGB(160, 82, 45) }, + { "silver", NSVG_RGB(192, 192, 192) }, + { "skyblue", NSVG_RGB(135, 206, 235) }, + { "slateblue", NSVG_RGB(106, 90, 205) }, + { "slategray", NSVG_RGB(112, 128, 144) }, + { "slategrey", NSVG_RGB(112, 128, 144) }, + { "snow", NSVG_RGB(255, 250, 250) }, + { "springgreen", NSVG_RGB( 0, 255, 127) }, + { "steelblue", NSVG_RGB( 70, 130, 180) }, + { "tan", NSVG_RGB(210, 180, 140) }, + { "teal", NSVG_RGB( 0, 128, 128) }, + { "thistle", NSVG_RGB(216, 191, 216) }, + { "tomato", NSVG_RGB(255, 99, 71) }, + { "turquoise", NSVG_RGB( 64, 224, 208) }, + { "violet", NSVG_RGB(238, 130, 238) }, + { "wheat", NSVG_RGB(245, 222, 179) }, + { "whitesmoke", NSVG_RGB(245, 245, 245) }, + { "yellowgreen", NSVG_RGB(154, 205, 50) }, +#endif +}; + +static unsigned int nsvg__parseColorName(const char* str) +{ + int i, ncolors = sizeof(nsvg__colors) / sizeof(NSVGNamedColor); + + for (i = 0; i < ncolors; i++) { + if (strcmp(nsvg__colors[i].name, str) == 0) { + return nsvg__colors[i].color; + } + } + + return NSVG_RGB(128, 128, 128); +} + +static unsigned int nsvg__parseColor(const char* str) +{ + size_t len = 0; + while(*str == ' ') ++str; + len = strlen(str); + if (len >= 1 && *str == '#') + return nsvg__parseColorHex(str); + else if (len >= 4 && str[0] == 'r' && str[1] == 'g' && str[2] == 'b' && str[3] == '(') + return nsvg__parseColorRGB(str); + return nsvg__parseColorName(str); +} + +static float nsvg__parseOpacity(const char* str) +{ + float val = nsvg__atof(str); + if (val < 0.0f) val = 0.0f; + if (val > 1.0f) val = 1.0f; + return val; +} + +static float nsvg__parseMiterLimit(const char* str) +{ + float val = nsvg__atof(str); + if (val < 0.0f) val = 0.0f; + return val; +} + +static int nsvg__parseUnits(const char* units) +{ + if (units[0] == 'p' && units[1] == 'x') + return NSVG_UNITS_PX; + else if (units[0] == 'p' && units[1] == 't') + return NSVG_UNITS_PT; + else if (units[0] == 'p' && units[1] == 'c') + return NSVG_UNITS_PC; + else if (units[0] == 'm' && units[1] == 'm') + return NSVG_UNITS_MM; + else if (units[0] == 'c' && units[1] == 'm') + return NSVG_UNITS_CM; + else if (units[0] == 'i' && units[1] == 'n') + return NSVG_UNITS_IN; + else if (units[0] == '%') + return NSVG_UNITS_PERCENT; + else if (units[0] == 'e' && units[1] == 'm') + return NSVG_UNITS_EM; + else if (units[0] == 'e' && units[1] == 'x') + return NSVG_UNITS_EX; + return NSVG_UNITS_USER; +} + +static int nsvg__isCoordinate(const char* s) +{ + // optional sign + if (*s == '-' || *s == '+') + s++; + // must have at least one digit, or start by a dot + return (nsvg__isdigit(*s) || *s == '.'); +} + +static NSVGcoordinate nsvg__parseCoordinateRaw(const char* str) +{ + NSVGcoordinate coord = {0, NSVG_UNITS_USER}; + char buf[64]; + coord.units = nsvg__parseUnits(nsvg__parseNumber(str, buf, 64)); + coord.value = nsvg__atof(buf); + return coord; +} + +static NSVGcoordinate nsvg__coord(float v, int units) +{ + NSVGcoordinate coord = {v, units}; + return coord; +} + +static float nsvg__parseCoordinate(NSVGparser* p, const char* str, float orig, float length) +{ + NSVGcoordinate coord = nsvg__parseCoordinateRaw(str); + return nsvg__convertToPixels(p, coord, orig, length); +} + +static int nsvg__parseTransformArgs(const char* str, float* args, int maxNa, int* na) +{ + const char* end; + const char* ptr; + char it[64]; + + *na = 0; + ptr = str; + while (*ptr && *ptr != '(') ++ptr; + if (*ptr == 0) + return 1; + end = ptr; + while (*end && *end != ')') ++end; + if (*end == 0) + return 1; + + while (ptr < end) { + if (*ptr == '-' || *ptr == '+' || *ptr == '.' || nsvg__isdigit(*ptr)) { + if (*na >= maxNa) return 0; + ptr = nsvg__parseNumber(ptr, it, 64); + args[(*na)++] = (float)nsvg__atof(it); + } else { + ++ptr; + } + } + return (int)(end - str); +} + + +static int nsvg__parseMatrix(float* xform, const char* str) +{ + float t[6]; + int na = 0; + int len = nsvg__parseTransformArgs(str, t, 6, &na); + if (na != 6) return len; + memcpy(xform, t, sizeof(float)*6); + return len; +} + +static int nsvg__parseTranslate(float* xform, const char* str) +{ + float args[2]; + float t[6]; + int na = 0; + int len = nsvg__parseTransformArgs(str, args, 2, &na); + if (na == 1) args[1] = 0.0; + + nsvg__xformSetTranslation(t, args[0], args[1]); + memcpy(xform, t, sizeof(float)*6); + return len; +} + +static int nsvg__parseScale(float* xform, const char* str) +{ + float args[2]; + int na = 0; + float t[6]; + int len = nsvg__parseTransformArgs(str, args, 2, &na); + if (na == 1) args[1] = args[0]; + nsvg__xformSetScale(t, args[0], args[1]); + memcpy(xform, t, sizeof(float)*6); + return len; +} + +static int nsvg__parseSkewX(float* xform, const char* str) +{ + float args[1]; + int na = 0; + float t[6]; + int len = nsvg__parseTransformArgs(str, args, 1, &na); + nsvg__xformSetSkewX(t, args[0]/180.0f*NSVG_PI); + memcpy(xform, t, sizeof(float)*6); + return len; +} + +static int nsvg__parseSkewY(float* xform, const char* str) +{ + float args[1]; + int na = 0; + float t[6]; + int len = nsvg__parseTransformArgs(str, args, 1, &na); + nsvg__xformSetSkewY(t, args[0]/180.0f*NSVG_PI); + memcpy(xform, t, sizeof(float)*6); + return len; +} + +static int nsvg__parseRotate(float* xform, const char* str) +{ + float args[3]; + int na = 0; + float m[6]; + float t[6]; + int len = nsvg__parseTransformArgs(str, args, 3, &na); + if (na == 1) + args[1] = args[2] = 0.0f; + nsvg__xformIdentity(m); + + if (na > 1) { + nsvg__xformSetTranslation(t, -args[1], -args[2]); + nsvg__xformMultiply(m, t); + } + + nsvg__xformSetRotation(t, args[0]/180.0f*NSVG_PI); + nsvg__xformMultiply(m, t); + + if (na > 1) { + nsvg__xformSetTranslation(t, args[1], args[2]); + nsvg__xformMultiply(m, t); + } + + memcpy(xform, m, sizeof(float)*6); + + return len; +} + +static void nsvg__parseTransform(float* xform, const char* str) +{ + float t[6]; + int len; + nsvg__xformIdentity(xform); + while (*str) + { + if (strncmp(str, "matrix", 6) == 0) + len = nsvg__parseMatrix(t, str); + else if (strncmp(str, "translate", 9) == 0) + len = nsvg__parseTranslate(t, str); + else if (strncmp(str, "scale", 5) == 0) + len = nsvg__parseScale(t, str); + else if (strncmp(str, "rotate", 6) == 0) + len = nsvg__parseRotate(t, str); + else if (strncmp(str, "skewX", 5) == 0) + len = nsvg__parseSkewX(t, str); + else if (strncmp(str, "skewY", 5) == 0) + len = nsvg__parseSkewY(t, str); + else{ + ++str; + continue; + } + if (len != 0) { + str += len; + } else { + ++str; + continue; + } + + nsvg__xformPremultiply(xform, t); + } +} + +static void nsvg__parseUrl(char* id, const char* str) +{ + int i = 0; + str += 4; // "url("; + if (*str && *str == '#') + str++; + while (i < 63 && *str && *str != ')') { + id[i] = *str++; + i++; + } + id[i] = '\0'; +} + +static char nsvg__parseLineCap(const char* str) +{ + if (strcmp(str, "butt") == 0) + return NSVG_CAP_BUTT; + else if (strcmp(str, "round") == 0) + return NSVG_CAP_ROUND; + else if (strcmp(str, "square") == 0) + return NSVG_CAP_SQUARE; + // TODO: handle inherit. + return NSVG_CAP_BUTT; +} + +static char nsvg__parseLineJoin(const char* str) +{ + if (strcmp(str, "miter") == 0) + return NSVG_JOIN_MITER; + else if (strcmp(str, "round") == 0) + return NSVG_JOIN_ROUND; + else if (strcmp(str, "bevel") == 0) + return NSVG_JOIN_BEVEL; + // TODO: handle inherit. + return NSVG_JOIN_MITER; +} + +static char nsvg__parseFillRule(const char* str) +{ + if (strcmp(str, "nonzero") == 0) + return NSVG_FILLRULE_NONZERO; + else if (strcmp(str, "evenodd") == 0) + return NSVG_FILLRULE_EVENODD; + // TODO: handle inherit. + return NSVG_FILLRULE_NONZERO; +} + +static unsigned char nsvg__parsePaintOrder(const char* str) +{ + if (strcmp(str, "normal") == 0 || strcmp(str, "fill stroke markers") == 0) + return nsvg__encodePaintOrder(NSVG_PAINT_FILL, NSVG_PAINT_STROKE, NSVG_PAINT_MARKERS); + else if (strcmp(str, "fill markers stroke") == 0) + return nsvg__encodePaintOrder(NSVG_PAINT_FILL, NSVG_PAINT_MARKERS, NSVG_PAINT_STROKE); + else if (strcmp(str, "markers fill stroke") == 0) + return nsvg__encodePaintOrder(NSVG_PAINT_MARKERS, NSVG_PAINT_FILL, NSVG_PAINT_STROKE); + else if (strcmp(str, "markers stroke fill") == 0) + return nsvg__encodePaintOrder(NSVG_PAINT_MARKERS, NSVG_PAINT_STROKE, NSVG_PAINT_FILL); + else if (strcmp(str, "stroke fill markers") == 0) + return nsvg__encodePaintOrder(NSVG_PAINT_STROKE, NSVG_PAINT_FILL, NSVG_PAINT_MARKERS); + else if (strcmp(str, "stroke markers fill") == 0) + return nsvg__encodePaintOrder(NSVG_PAINT_STROKE, NSVG_PAINT_MARKERS, NSVG_PAINT_FILL); + // TODO: handle inherit. + return nsvg__encodePaintOrder(NSVG_PAINT_FILL, NSVG_PAINT_STROKE, NSVG_PAINT_MARKERS); +} + +static const char* nsvg__getNextDashItem(const char* s, char* it) +{ + int n = 0; + it[0] = '\0'; + // Skip white spaces and commas + while (*s && (nsvg__isspace(*s) || *s == ',')) s++; + // Advance until whitespace, comma or end. + while (*s && (!nsvg__isspace(*s) && *s != ',')) { + if (n < 63) + it[n++] = *s; + s++; + } + it[n++] = '\0'; + return s; +} + +static int nsvg__parseStrokeDashArray(NSVGparser* p, const char* str, float* strokeDashArray) +{ + char item[64]; + int count = 0, i; + float sum = 0.0f; + + // Handle "none" + if (str[0] == 'n') + return 0; + + // Parse dashes + while (*str) { + str = nsvg__getNextDashItem(str, item); + if (!*item) break; + if (count < NSVG_MAX_DASHES) + strokeDashArray[count++] = fabsf(nsvg__parseCoordinate(p, item, 0.0f, nsvg__actualLength(p))); + } + + for (i = 0; i < count; i++) + sum += strokeDashArray[i]; + if (sum <= 1e-6f) + count = 0; + + return count; +} + +static void nsvg__parseStyle(NSVGparser* p, const char* str); + +// Apply any matching class styles for a "class" attribute value. We support only simple class +// selectors. The class attribute may contain multiple space-separated class names. +static void nsvg__applyClassStyles(NSVGparser* p, const char* value) +{ + const char* cur = value; + while (*cur) { + const char* classStart; + size_t classLen; + NSVGstyleDeclaration* style; + + while (*cur && nsvg__isspace(*cur)) + cur++; + if (!*cur) + break; + + classStart = cur; + while (*cur && !nsvg__isspace(*cur)) + cur++; + classLen = (size_t)(cur - classStart); + + for (style = p->styles; style != NULL; style = style->next) { + if (strncmp(style->className, classStart, classLen) == 0 && style->className[classLen] == '\0') { + nsvg__parseStyle(p, style->propertiesText); + } + } + } +} + +static int nsvg__parseAttr(NSVGparser* p, const char* name, const char* value) +{ + float xform[6]; + NSVGattrib* attr = nsvg__getAttr(p); + if (!attr) return 0; + + if (strcmp(name, "style") == 0) { + nsvg__parseStyle(p, value); + } else if (strcmp(name, "display") == 0) { + if (strcmp(value, "none") == 0) + attr->visible = 0; + // Don't reset ->visible on display:inline, one display:none hides the whole subtree + + } else if (strcmp(name, "fill") == 0) { + if (strcmp(value, "none") == 0) { + attr->hasFill = 0; + } else if (strncmp(value, "url(", 4) == 0) { + attr->hasFill = 2; + nsvg__parseUrl(attr->fillGradient, value); + } else { + attr->hasFill = 1; + attr->fillColor = nsvg__parseColor(value); + } + } else if (strcmp(name, "opacity") == 0) { + attr->opacity = nsvg__parseOpacity(value); + } else if (strcmp(name, "fill-opacity") == 0) { + attr->fillOpacity = nsvg__parseOpacity(value); + } else if (strcmp(name, "stroke") == 0) { + if (strcmp(value, "none") == 0) { + attr->hasStroke = 0; + } else if (strncmp(value, "url(", 4) == 0) { + attr->hasStroke = 2; + nsvg__parseUrl(attr->strokeGradient, value); + } else { + attr->hasStroke = 1; + attr->strokeColor = nsvg__parseColor(value); + } + } else if (strcmp(name, "stroke-width") == 0) { + attr->strokeWidth = nsvg__parseCoordinate(p, value, 0.0f, nsvg__actualLength(p)); + } else if (strcmp(name, "stroke-dasharray") == 0) { + attr->strokeDashCount = nsvg__parseStrokeDashArray(p, value, attr->strokeDashArray); + } else if (strcmp(name, "stroke-dashoffset") == 0) { + attr->strokeDashOffset = nsvg__parseCoordinate(p, value, 0.0f, nsvg__actualLength(p)); + } else if (strcmp(name, "stroke-opacity") == 0) { + attr->strokeOpacity = nsvg__parseOpacity(value); + } else if (strcmp(name, "stroke-linecap") == 0) { + attr->strokeLineCap = nsvg__parseLineCap(value); + } else if (strcmp(name, "stroke-linejoin") == 0) { + attr->strokeLineJoin = nsvg__parseLineJoin(value); + } else if (strcmp(name, "stroke-miterlimit") == 0) { + attr->miterLimit = nsvg__parseMiterLimit(value); + } else if (strcmp(name, "fill-rule") == 0) { + attr->fillRule = nsvg__parseFillRule(value); + } else if (strcmp(name, "font-size") == 0) { + attr->fontSize = nsvg__parseCoordinate(p, value, 0.0f, nsvg__actualLength(p)); + } else if (strcmp(name, "transform") == 0) { + nsvg__parseTransform(xform, value); + nsvg__xformPremultiply(attr->xform, xform); + } else if (strcmp(name, "stop-color") == 0) { + attr->stopColor = nsvg__parseColor(value); + } else if (strcmp(name, "stop-opacity") == 0) { + attr->stopOpacity = nsvg__parseOpacity(value); + } else if (strcmp(name, "offset") == 0) { + attr->stopOffset = nsvg__parseCoordinate(p, value, 0.0f, 1.0f); + } else if (strcmp(name, "paint-order") == 0) { + attr->paintOrder = nsvg__parsePaintOrder(value); + } else if (strcmp(name, "id") == 0) { + strncpy(attr->id, value, 63); + attr->id[63] = '\0'; + } else if (strcmp(name, "class") == 0) { + nsvg__applyClassStyles(p, value); + } + else { + return 0; + } + return 1; +} + +static int nsvg__parseNameValue(NSVGparser* p, const char* start, const char* end) +{ + const char* str; + const char* val; + char name[512]; + char value[512]; + int n; + + str = start; + while (str < end && *str != ':') ++str; + + val = str; + + // Right Trim + while (str > start && (*str == ':' || nsvg__isspace(*str))) --str; + ++str; + + n = (int)(str - start); + if (n > 511) n = 511; + if (n) memcpy(name, start, n); + name[n] = 0; + + while (val < end && (*val == ':' || nsvg__isspace(*val))) ++val; + + n = (int)(end - val); + if (n > 511) n = 511; + if (n) memcpy(value, val, n); + value[n] = 0; + + return nsvg__parseAttr(p, name, value); +} + +static void nsvg__parseStyle(NSVGparser* p, const char* str) +{ + const char* start; + const char* end; + + while (*str) { + // Left Trim + while(*str && nsvg__isspace(*str)) ++str; + start = str; + while(*str && *str != ';') ++str; + end = str; + + // Right Trim + while (end > start && (*end == ';' || nsvg__isspace(*end))) --end; + ++end; + + nsvg__parseNameValue(p, start, end); + if (*str) ++str; + } +} + +static void nsvg__parseAttribs(NSVGparser* p, const char** attr) +{ + int i; + for (i = 0; attr[i]; i += 2) + { + if (strcmp(attr[i], "style") == 0) + nsvg__parseStyle(p, attr[i + 1]); + else + nsvg__parseAttr(p, attr[i], attr[i + 1]); + } +} + +static int nsvg__getArgsPerElement(char cmd) +{ + switch (cmd) { + case 'v': + case 'V': + case 'h': + case 'H': + return 1; + case 'm': + case 'M': + case 'l': + case 'L': + case 't': + case 'T': + return 2; + case 'q': + case 'Q': + case 's': + case 'S': + return 4; + case 'c': + case 'C': + return 6; + case 'a': + case 'A': + return 7; + case 'z': + case 'Z': + return 0; + } + return -1; +} + +static void nsvg__pathMoveTo(NSVGparser* p, float* cpx, float* cpy, float* args, int rel) +{ + if (rel) { + *cpx += args[0]; + *cpy += args[1]; + } else { + *cpx = args[0]; + *cpy = args[1]; + } + nsvg__moveTo(p, *cpx, *cpy); +} + +static void nsvg__pathLineTo(NSVGparser* p, float* cpx, float* cpy, float* args, int rel) +{ + if (rel) { + *cpx += args[0]; + *cpy += args[1]; + } else { + *cpx = args[0]; + *cpy = args[1]; + } + nsvg__lineTo(p, *cpx, *cpy); +} + +static void nsvg__pathHLineTo(NSVGparser* p, float* cpx, float* cpy, float* args, int rel) +{ + if (rel) + *cpx += args[0]; + else + *cpx = args[0]; + nsvg__lineTo(p, *cpx, *cpy); +} + +static void nsvg__pathVLineTo(NSVGparser* p, float* cpx, float* cpy, float* args, int rel) +{ + if (rel) + *cpy += args[0]; + else + *cpy = args[0]; + nsvg__lineTo(p, *cpx, *cpy); +} + +static void nsvg__pathCubicBezTo(NSVGparser* p, float* cpx, float* cpy, + float* cpx2, float* cpy2, float* args, int rel) +{ + float x2, y2, cx1, cy1, cx2, cy2; + + if (rel) { + cx1 = *cpx + args[0]; + cy1 = *cpy + args[1]; + cx2 = *cpx + args[2]; + cy2 = *cpy + args[3]; + x2 = *cpx + args[4]; + y2 = *cpy + args[5]; + } else { + cx1 = args[0]; + cy1 = args[1]; + cx2 = args[2]; + cy2 = args[3]; + x2 = args[4]; + y2 = args[5]; + } + + nsvg__cubicBezTo(p, cx1,cy1, cx2,cy2, x2,y2); + + *cpx2 = cx2; + *cpy2 = cy2; + *cpx = x2; + *cpy = y2; +} + +static void nsvg__pathCubicBezShortTo(NSVGparser* p, float* cpx, float* cpy, + float* cpx2, float* cpy2, float* args, int rel) +{ + float x1, y1, x2, y2, cx1, cy1, cx2, cy2; + + x1 = *cpx; + y1 = *cpy; + if (rel) { + cx2 = *cpx + args[0]; + cy2 = *cpy + args[1]; + x2 = *cpx + args[2]; + y2 = *cpy + args[3]; + } else { + cx2 = args[0]; + cy2 = args[1]; + x2 = args[2]; + y2 = args[3]; + } + + cx1 = 2*x1 - *cpx2; + cy1 = 2*y1 - *cpy2; + + nsvg__cubicBezTo(p, cx1,cy1, cx2,cy2, x2,y2); + + *cpx2 = cx2; + *cpy2 = cy2; + *cpx = x2; + *cpy = y2; +} + +static void nsvg__pathQuadBezTo(NSVGparser* p, float* cpx, float* cpy, + float* cpx2, float* cpy2, float* args, int rel) +{ + float x1, y1, x2, y2, cx, cy; + float cx1, cy1, cx2, cy2; + + x1 = *cpx; + y1 = *cpy; + if (rel) { + cx = *cpx + args[0]; + cy = *cpy + args[1]; + x2 = *cpx + args[2]; + y2 = *cpy + args[3]; + } else { + cx = args[0]; + cy = args[1]; + x2 = args[2]; + y2 = args[3]; + } + + // Convert to cubic bezier + cx1 = x1 + 2.0f/3.0f*(cx - x1); + cy1 = y1 + 2.0f/3.0f*(cy - y1); + cx2 = x2 + 2.0f/3.0f*(cx - x2); + cy2 = y2 + 2.0f/3.0f*(cy - y2); + + nsvg__cubicBezTo(p, cx1,cy1, cx2,cy2, x2,y2); + + *cpx2 = cx; + *cpy2 = cy; + *cpx = x2; + *cpy = y2; +} + +static void nsvg__pathQuadBezShortTo(NSVGparser* p, float* cpx, float* cpy, + float* cpx2, float* cpy2, float* args, int rel) +{ + float x1, y1, x2, y2, cx, cy; + float cx1, cy1, cx2, cy2; + + x1 = *cpx; + y1 = *cpy; + if (rel) { + x2 = *cpx + args[0]; + y2 = *cpy + args[1]; + } else { + x2 = args[0]; + y2 = args[1]; + } + + cx = 2*x1 - *cpx2; + cy = 2*y1 - *cpy2; + + // Convert to cubix bezier + cx1 = x1 + 2.0f/3.0f*(cx - x1); + cy1 = y1 + 2.0f/3.0f*(cy - y1); + cx2 = x2 + 2.0f/3.0f*(cx - x2); + cy2 = y2 + 2.0f/3.0f*(cy - y2); + + nsvg__cubicBezTo(p, cx1,cy1, cx2,cy2, x2,y2); + + *cpx2 = cx; + *cpy2 = cy; + *cpx = x2; + *cpy = y2; +} + +static float nsvg__sqr(float x) { return x*x; } +static float nsvg__vmag(float x, float y) { return sqrtf(x*x + y*y); } + +static float nsvg__vecrat(float ux, float uy, float vx, float vy) +{ + return (ux*vx + uy*vy) / (nsvg__vmag(ux,uy) * nsvg__vmag(vx,vy)); +} + +static float nsvg__vecang(float ux, float uy, float vx, float vy) +{ + float r = nsvg__vecrat(ux,uy, vx,vy); + if (r < -1.0f) r = -1.0f; + if (r > 1.0f) r = 1.0f; + return ((ux*vy < uy*vx) ? -1.0f : 1.0f) * acosf(r); +} + +static void nsvg__pathArcTo(NSVGparser* p, float* cpx, float* cpy, float* args, int rel) +{ + // Ported from canvg (https://code.google.com/p/canvg/) + float rx, ry, rotx; + float x1, y1, x2, y2, cx, cy, dx, dy, d; + float x1p, y1p, cxp, cyp, s, sa, sb; + float ux, uy, vx, vy, a1, da; + float x, y, tanx, tany, a, px = 0, py = 0, ptanx = 0, ptany = 0, t[6]; + float sinrx, cosrx; + int fa, fs; + int i, ndivs; + float hda, kappa; + + rx = fabsf(args[0]); // y radius + ry = fabsf(args[1]); // x radius + rotx = args[2] / 180.0f * NSVG_PI; // x rotation angle + fa = fabsf(args[3]) > 1e-6 ? 1 : 0; // Large arc + fs = fabsf(args[4]) > 1e-6 ? 1 : 0; // Sweep direction + x1 = *cpx; // start point + y1 = *cpy; + if (rel) { // end point + x2 = *cpx + args[5]; + y2 = *cpy + args[6]; + } else { + x2 = args[5]; + y2 = args[6]; + } + + dx = x1 - x2; + dy = y1 - y2; + d = sqrtf(dx*dx + dy*dy); + if (d < 1e-6f || rx < 1e-6f || ry < 1e-6f) { + // The arc degenerates to a line + nsvg__lineTo(p, x2, y2); + *cpx = x2; + *cpy = y2; + return; + } + + sinrx = sinf(rotx); + cosrx = cosf(rotx); + + // Convert to center point parameterization. + // http://www.w3.org/TR/SVG11/implnote.html#ArcImplementationNotes + // 1) Compute x1', y1' + x1p = cosrx * dx / 2.0f + sinrx * dy / 2.0f; + y1p = -sinrx * dx / 2.0f + cosrx * dy / 2.0f; + d = nsvg__sqr(x1p)/nsvg__sqr(rx) + nsvg__sqr(y1p)/nsvg__sqr(ry); + if (d > 1) { + d = sqrtf(d); + rx *= d; + ry *= d; + } + // 2) Compute cx', cy' + s = 0.0f; + sa = nsvg__sqr(rx)*nsvg__sqr(ry) - nsvg__sqr(rx)*nsvg__sqr(y1p) - nsvg__sqr(ry)*nsvg__sqr(x1p); + sb = nsvg__sqr(rx)*nsvg__sqr(y1p) + nsvg__sqr(ry)*nsvg__sqr(x1p); + if (sa < 0.0f) sa = 0.0f; + if (sb > 0.0f) + s = sqrtf(sa / sb); + if (fa == fs) + s = -s; + cxp = s * rx * y1p / ry; + cyp = s * -ry * x1p / rx; + + // 3) Compute cx,cy from cx',cy' + cx = (x1 + x2)/2.0f + cosrx*cxp - sinrx*cyp; + cy = (y1 + y2)/2.0f + sinrx*cxp + cosrx*cyp; + + // 4) Calculate theta1, and delta theta. + ux = (x1p - cxp) / rx; + uy = (y1p - cyp) / ry; + vx = (-x1p - cxp) / rx; + vy = (-y1p - cyp) / ry; + a1 = nsvg__vecang(1.0f,0.0f, ux,uy); // Initial angle + da = nsvg__vecang(ux,uy, vx,vy); // Delta angle + +// if (vecrat(ux,uy,vx,vy) <= -1.0f) da = NSVG_PI; +// if (vecrat(ux,uy,vx,vy) >= 1.0f) da = 0; + + if (fs == 0 && da > 0) + da -= 2 * NSVG_PI; + else if (fs == 1 && da < 0) + da += 2 * NSVG_PI; + + // Approximate the arc using cubic spline segments. + t[0] = cosrx; t[1] = sinrx; + t[2] = -sinrx; t[3] = cosrx; + t[4] = cx; t[5] = cy; + + // Split arc into max 90 degree segments. + // The loop assumes an iteration per end point (including start and end), this +1. + ndivs = (int)(fabsf(da) / (NSVG_PI*0.5f) + 1.0f); + hda = (da / (float)ndivs) / 2.0f; + // Fix for ticket #179: division by 0: avoid cotangens around 0 (infinite) + if ((hda < 1e-3f) && (hda > -1e-3f)) + hda *= 0.5f; + else + hda = (1.0f - cosf(hda)) / sinf(hda); + kappa = fabsf(4.0f / 3.0f * hda); + if (da < 0.0f) + kappa = -kappa; + + for (i = 0; i <= ndivs; i++) { + a = a1 + da * ((float)i/(float)ndivs); + dx = cosf(a); + dy = sinf(a); + nsvg__xformPoint(&x, &y, dx*rx, dy*ry, t); // position + nsvg__xformVec(&tanx, &tany, -dy*rx * kappa, dx*ry * kappa, t); // tangent + if (i > 0) + nsvg__cubicBezTo(p, px+ptanx,py+ptany, x-tanx, y-tany, x, y); + px = x; + py = y; + ptanx = tanx; + ptany = tany; + } + + *cpx = x2; + *cpy = y2; +} + +static void nsvg__parsePath(NSVGparser* p, const char** attr) +{ + const char* s = NULL; + char cmd = '\0'; + float args[10]; + int nargs; + int rargs = 0; + char initPoint; + float cpx, cpy, cpx2, cpy2; + const char* tmp[4]; + char closedFlag; + int i; + char item[64]; + + for (i = 0; attr[i]; i += 2) { + if (strcmp(attr[i], "d") == 0) { + s = attr[i + 1]; + } else { + tmp[0] = attr[i]; + tmp[1] = attr[i + 1]; + tmp[2] = 0; + tmp[3] = 0; + nsvg__parseAttribs(p, tmp); + } + } + + if (s) { + nsvg__resetPath(p); + cpx = 0; cpy = 0; + cpx2 = 0; cpy2 = 0; + initPoint = 0; + closedFlag = 0; + nargs = 0; + + while (*s) { + item[0] = '\0'; + if ((cmd == 'A' || cmd == 'a') && (nargs == 3 || nargs == 4)) + s = nsvg__getNextPathItemWhenArcFlag(s, item); + if (!*item) + s = nsvg__getNextPathItem(s, item); + if (!*item) break; + if (cmd != '\0' && nsvg__isCoordinate(item)) { + if (nargs < 10) + args[nargs++] = (float)nsvg__atof(item); + if (nargs >= rargs) { + switch (cmd) { + case 'm': + case 'M': + nsvg__pathMoveTo(p, &cpx, &cpy, args, cmd == 'm' ? 1 : 0); + // Moveto can be followed by multiple coordinate pairs, + // which should be treated as linetos. + cmd = (cmd == 'm') ? 'l' : 'L'; + rargs = nsvg__getArgsPerElement(cmd); + cpx2 = cpx; cpy2 = cpy; + initPoint = 1; + break; + case 'l': + case 'L': + nsvg__pathLineTo(p, &cpx, &cpy, args, cmd == 'l' ? 1 : 0); + cpx2 = cpx; cpy2 = cpy; + break; + case 'H': + case 'h': + nsvg__pathHLineTo(p, &cpx, &cpy, args, cmd == 'h' ? 1 : 0); + cpx2 = cpx; cpy2 = cpy; + break; + case 'V': + case 'v': + nsvg__pathVLineTo(p, &cpx, &cpy, args, cmd == 'v' ? 1 : 0); + cpx2 = cpx; cpy2 = cpy; + break; + case 'C': + case 'c': + nsvg__pathCubicBezTo(p, &cpx, &cpy, &cpx2, &cpy2, args, cmd == 'c' ? 1 : 0); + break; + case 'S': + case 's': + nsvg__pathCubicBezShortTo(p, &cpx, &cpy, &cpx2, &cpy2, args, cmd == 's' ? 1 : 0); + break; + case 'Q': + case 'q': + nsvg__pathQuadBezTo(p, &cpx, &cpy, &cpx2, &cpy2, args, cmd == 'q' ? 1 : 0); + break; + case 'T': + case 't': + nsvg__pathQuadBezShortTo(p, &cpx, &cpy, &cpx2, &cpy2, args, cmd == 't' ? 1 : 0); + break; + case 'A': + case 'a': + nsvg__pathArcTo(p, &cpx, &cpy, args, cmd == 'a' ? 1 : 0); + cpx2 = cpx; cpy2 = cpy; + break; + default: + if (nargs >= 2) { + cpx = args[nargs-2]; + cpy = args[nargs-1]; + cpx2 = cpx; cpy2 = cpy; + } + break; + } + nargs = 0; + } + } else { + cmd = item[0]; + if (cmd == 'M' || cmd == 'm') { + // Commit path. + if (p->npts > 0) + nsvg__addPath(p, closedFlag); + // Start new subpath. + nsvg__resetPath(p); + closedFlag = 0; + nargs = 0; + } else if (initPoint == 0) { + // Do not allow other commands until initial point has been set (moveTo called once). + cmd = '\0'; + } + if (cmd == 'Z' || cmd == 'z') { + closedFlag = 1; + // Commit path. + if (p->npts > 0) { + // Move current point to first point + cpx = p->pts[0]; + cpy = p->pts[1]; + cpx2 = cpx; cpy2 = cpy; + nsvg__addPath(p, closedFlag); + } + // Start new subpath. + nsvg__resetPath(p); + nsvg__moveTo(p, cpx, cpy); + closedFlag = 0; + nargs = 0; + } + rargs = nsvg__getArgsPerElement(cmd); + if (rargs == -1) { + // Command not recognized + cmd = '\0'; + rargs = 0; + } + } + } + // Commit path. + if (p->npts) + nsvg__addPath(p, closedFlag); + } + + nsvg__addShape(p); +} + +static void nsvg__parseRect(NSVGparser* p, const char** attr) +{ + float x = 0.0f; + float y = 0.0f; + float w = 0.0f; + float h = 0.0f; + float rx = -1.0f; // marks not set + float ry = -1.0f; + int i; + + for (i = 0; attr[i]; i += 2) { + if (!nsvg__parseAttr(p, attr[i], attr[i + 1])) { + if (strcmp(attr[i], "x") == 0) x = nsvg__parseCoordinate(p, attr[i+1], nsvg__actualOrigX(p), nsvg__actualWidth(p)); + if (strcmp(attr[i], "y") == 0) y = nsvg__parseCoordinate(p, attr[i+1], nsvg__actualOrigY(p), nsvg__actualHeight(p)); + if (strcmp(attr[i], "width") == 0) w = nsvg__parseCoordinate(p, attr[i+1], 0.0f, nsvg__actualWidth(p)); + if (strcmp(attr[i], "height") == 0) h = nsvg__parseCoordinate(p, attr[i+1], 0.0f, nsvg__actualHeight(p)); + if (strcmp(attr[i], "rx") == 0) rx = fabsf(nsvg__parseCoordinate(p, attr[i+1], 0.0f, nsvg__actualWidth(p))); + if (strcmp(attr[i], "ry") == 0) ry = fabsf(nsvg__parseCoordinate(p, attr[i+1], 0.0f, nsvg__actualHeight(p))); + } + } + + if (rx < 0.0f && ry > 0.0f) rx = ry; + if (ry < 0.0f && rx > 0.0f) ry = rx; + if (rx < 0.0f) rx = 0.0f; + if (ry < 0.0f) ry = 0.0f; + if (rx > w/2.0f) rx = w/2.0f; + if (ry > h/2.0f) ry = h/2.0f; + + if (w != 0.0f && h != 0.0f) { + nsvg__resetPath(p); + + if (rx < 0.00001f || ry < 0.0001f) { + nsvg__moveTo(p, x, y); + nsvg__lineTo(p, x+w, y); + nsvg__lineTo(p, x+w, y+h); + nsvg__lineTo(p, x, y+h); + } else { + // Rounded rectangle + nsvg__moveTo(p, x+rx, y); + nsvg__lineTo(p, x+w-rx, y); + nsvg__cubicBezTo(p, x+w-rx*(1-NSVG_KAPPA90), y, x+w, y+ry*(1-NSVG_KAPPA90), x+w, y+ry); + nsvg__lineTo(p, x+w, y+h-ry); + nsvg__cubicBezTo(p, x+w, y+h-ry*(1-NSVG_KAPPA90), x+w-rx*(1-NSVG_KAPPA90), y+h, x+w-rx, y+h); + nsvg__lineTo(p, x+rx, y+h); + nsvg__cubicBezTo(p, x+rx*(1-NSVG_KAPPA90), y+h, x, y+h-ry*(1-NSVG_KAPPA90), x, y+h-ry); + nsvg__lineTo(p, x, y+ry); + nsvg__cubicBezTo(p, x, y+ry*(1-NSVG_KAPPA90), x+rx*(1-NSVG_KAPPA90), y, x+rx, y); + } + + nsvg__addPath(p, 1); + + nsvg__addShape(p); + } +} + +static void nsvg__parseCircle(NSVGparser* p, const char** attr) +{ + float cx = 0.0f; + float cy = 0.0f; + float r = 0.0f; + int i; + + for (i = 0; attr[i]; i += 2) { + if (!nsvg__parseAttr(p, attr[i], attr[i + 1])) { + if (strcmp(attr[i], "cx") == 0) cx = nsvg__parseCoordinate(p, attr[i+1], nsvg__actualOrigX(p), nsvg__actualWidth(p)); + if (strcmp(attr[i], "cy") == 0) cy = nsvg__parseCoordinate(p, attr[i+1], nsvg__actualOrigY(p), nsvg__actualHeight(p)); + if (strcmp(attr[i], "r") == 0) r = fabsf(nsvg__parseCoordinate(p, attr[i+1], 0.0f, nsvg__actualLength(p))); + } + } + + if (r > 0.0f) { + nsvg__resetPath(p); + + nsvg__moveTo(p, cx+r, cy); + nsvg__cubicBezTo(p, cx+r, cy+r*NSVG_KAPPA90, cx+r*NSVG_KAPPA90, cy+r, cx, cy+r); + nsvg__cubicBezTo(p, cx-r*NSVG_KAPPA90, cy+r, cx-r, cy+r*NSVG_KAPPA90, cx-r, cy); + nsvg__cubicBezTo(p, cx-r, cy-r*NSVG_KAPPA90, cx-r*NSVG_KAPPA90, cy-r, cx, cy-r); + nsvg__cubicBezTo(p, cx+r*NSVG_KAPPA90, cy-r, cx+r, cy-r*NSVG_KAPPA90, cx+r, cy); + + nsvg__addPath(p, 1); + + nsvg__addShape(p); + } +} + +static void nsvg__parseEllipse(NSVGparser* p, const char** attr) +{ + float cx = 0.0f; + float cy = 0.0f; + float rx = 0.0f; + float ry = 0.0f; + int i; + + for (i = 0; attr[i]; i += 2) { + if (!nsvg__parseAttr(p, attr[i], attr[i + 1])) { + if (strcmp(attr[i], "cx") == 0) cx = nsvg__parseCoordinate(p, attr[i+1], nsvg__actualOrigX(p), nsvg__actualWidth(p)); + if (strcmp(attr[i], "cy") == 0) cy = nsvg__parseCoordinate(p, attr[i+1], nsvg__actualOrigY(p), nsvg__actualHeight(p)); + if (strcmp(attr[i], "rx") == 0) rx = fabsf(nsvg__parseCoordinate(p, attr[i+1], 0.0f, nsvg__actualWidth(p))); + if (strcmp(attr[i], "ry") == 0) ry = fabsf(nsvg__parseCoordinate(p, attr[i+1], 0.0f, nsvg__actualHeight(p))); + } + } + + if (rx > 0.0f && ry > 0.0f) { + + nsvg__resetPath(p); + + nsvg__moveTo(p, cx+rx, cy); + nsvg__cubicBezTo(p, cx+rx, cy+ry*NSVG_KAPPA90, cx+rx*NSVG_KAPPA90, cy+ry, cx, cy+ry); + nsvg__cubicBezTo(p, cx-rx*NSVG_KAPPA90, cy+ry, cx-rx, cy+ry*NSVG_KAPPA90, cx-rx, cy); + nsvg__cubicBezTo(p, cx-rx, cy-ry*NSVG_KAPPA90, cx-rx*NSVG_KAPPA90, cy-ry, cx, cy-ry); + nsvg__cubicBezTo(p, cx+rx*NSVG_KAPPA90, cy-ry, cx+rx, cy-ry*NSVG_KAPPA90, cx+rx, cy); + + nsvg__addPath(p, 1); + + nsvg__addShape(p); + } +} + +static void nsvg__parseLine(NSVGparser* p, const char** attr) +{ + float x1 = 0.0; + float y1 = 0.0; + float x2 = 0.0; + float y2 = 0.0; + int i; + + for (i = 0; attr[i]; i += 2) { + if (!nsvg__parseAttr(p, attr[i], attr[i + 1])) { + if (strcmp(attr[i], "x1") == 0) x1 = nsvg__parseCoordinate(p, attr[i + 1], nsvg__actualOrigX(p), nsvg__actualWidth(p)); + if (strcmp(attr[i], "y1") == 0) y1 = nsvg__parseCoordinate(p, attr[i + 1], nsvg__actualOrigY(p), nsvg__actualHeight(p)); + if (strcmp(attr[i], "x2") == 0) x2 = nsvg__parseCoordinate(p, attr[i + 1], nsvg__actualOrigX(p), nsvg__actualWidth(p)); + if (strcmp(attr[i], "y2") == 0) y2 = nsvg__parseCoordinate(p, attr[i + 1], nsvg__actualOrigY(p), nsvg__actualHeight(p)); + } + } + + nsvg__resetPath(p); + + nsvg__moveTo(p, x1, y1); + nsvg__lineTo(p, x2, y2); + + nsvg__addPath(p, 0); + + nsvg__addShape(p); +} + +static void nsvg__parsePoly(NSVGparser* p, const char** attr, int closeFlag) +{ + int i; + const char* s; + float args[2]; + int nargs, npts = 0; + char item[64]; + + nsvg__resetPath(p); + + for (i = 0; attr[i]; i += 2) { + if (!nsvg__parseAttr(p, attr[i], attr[i + 1])) { + if (strcmp(attr[i], "points") == 0) { + s = attr[i + 1]; + nargs = 0; + while (*s) { + s = nsvg__getNextPathItem(s, item); + args[nargs++] = (float)nsvg__atof(item); + if (nargs >= 2) { + if (npts == 0) + nsvg__moveTo(p, args[0], args[1]); + else + nsvg__lineTo(p, args[0], args[1]); + nargs = 0; + npts++; + } + } + } + } + } + + nsvg__addPath(p, (char)closeFlag); + + nsvg__addShape(p); +} + +static void nsvg__parseSVG(NSVGparser* p, const char** attr) +{ + int i; + for (i = 0; attr[i]; i += 2) { + if (!nsvg__parseAttr(p, attr[i], attr[i + 1])) { + if (strcmp(attr[i], "width") == 0) { + p->image->width = nsvg__parseCoordinate(p, attr[i + 1], 0.0f, 0.0f); + } else if (strcmp(attr[i], "height") == 0) { + p->image->height = nsvg__parseCoordinate(p, attr[i + 1], 0.0f, 0.0f); + } else if (strcmp(attr[i], "viewBox") == 0) { + const char *s = attr[i + 1]; + char buf[64]; + s = nsvg__parseNumber(s, buf, 64); + p->viewMinx = nsvg__atof(buf); + while (*s && (nsvg__isspace(*s) || *s == '%' || *s == ',')) s++; + if (!*s) return; + s = nsvg__parseNumber(s, buf, 64); + p->viewMiny = nsvg__atof(buf); + while (*s && (nsvg__isspace(*s) || *s == '%' || *s == ',')) s++; + if (!*s) return; + s = nsvg__parseNumber(s, buf, 64); + p->viewWidth = nsvg__atof(buf); + while (*s && (nsvg__isspace(*s) || *s == '%' || *s == ',')) s++; + if (!*s) return; + s = nsvg__parseNumber(s, buf, 64); + p->viewHeight = nsvg__atof(buf); + } else if (strcmp(attr[i], "preserveAspectRatio") == 0) { + if (strstr(attr[i + 1], "none") != 0) { + // No uniform scaling + p->alignType = NSVG_ALIGN_NONE; + } else { + // Parse X align + if (strstr(attr[i + 1], "xMin") != 0) + p->alignX = NSVG_ALIGN_MIN; + else if (strstr(attr[i + 1], "xMid") != 0) + p->alignX = NSVG_ALIGN_MID; + else if (strstr(attr[i + 1], "xMax") != 0) + p->alignX = NSVG_ALIGN_MAX; + // Parse X align + if (strstr(attr[i + 1], "yMin") != 0) + p->alignY = NSVG_ALIGN_MIN; + else if (strstr(attr[i + 1], "yMid") != 0) + p->alignY = NSVG_ALIGN_MID; + else if (strstr(attr[i + 1], "yMax") != 0) + p->alignY = NSVG_ALIGN_MAX; + // Parse meet/slice + p->alignType = NSVG_ALIGN_MEET; + if (strstr(attr[i + 1], "slice") != 0) + p->alignType = NSVG_ALIGN_SLICE; + } + } + } + } +} + +static void nsvg__parseGradient(NSVGparser* p, const char** attr, signed char type) +{ + int i; + NSVGgradientData* grad = (NSVGgradientData*)malloc(sizeof(NSVGgradientData)); + if (grad == NULL) return; + memset(grad, 0, sizeof(NSVGgradientData)); + grad->units = NSVG_OBJECT_SPACE; + grad->type = type; + if (grad->type == NSVG_PAINT_LINEAR_GRADIENT) { + grad->linear.x1 = nsvg__coord(0.0f, NSVG_UNITS_PERCENT); + grad->linear.y1 = nsvg__coord(0.0f, NSVG_UNITS_PERCENT); + grad->linear.x2 = nsvg__coord(100.0f, NSVG_UNITS_PERCENT); + grad->linear.y2 = nsvg__coord(0.0f, NSVG_UNITS_PERCENT); + } else if (grad->type == NSVG_PAINT_RADIAL_GRADIENT) { + grad->radial.cx = nsvg__coord(50.0f, NSVG_UNITS_PERCENT); + grad->radial.cy = nsvg__coord(50.0f, NSVG_UNITS_PERCENT); + grad->radial.r = nsvg__coord(50.0f, NSVG_UNITS_PERCENT); + } + + nsvg__xformIdentity(grad->xform); + + for (i = 0; attr[i]; i += 2) { + if (strcmp(attr[i], "id") == 0) { + strncpy(grad->id, attr[i+1], 63); + grad->id[63] = '\0'; + } else if (!nsvg__parseAttr(p, attr[i], attr[i + 1])) { + if (strcmp(attr[i], "gradientUnits") == 0) { + if (strcmp(attr[i+1], "objectBoundingBox") == 0) + grad->units = NSVG_OBJECT_SPACE; + else + grad->units = NSVG_USER_SPACE; + } else if (strcmp(attr[i], "gradientTransform") == 0) { + nsvg__parseTransform(grad->xform, attr[i + 1]); + } else if (strcmp(attr[i], "cx") == 0) { + grad->radial.cx = nsvg__parseCoordinateRaw(attr[i + 1]); + } else if (strcmp(attr[i], "cy") == 0) { + grad->radial.cy = nsvg__parseCoordinateRaw(attr[i + 1]); + } else if (strcmp(attr[i], "r") == 0) { + grad->radial.r = nsvg__parseCoordinateRaw(attr[i + 1]); + } else if (strcmp(attr[i], "fx") == 0) { + grad->radial.fx = nsvg__parseCoordinateRaw(attr[i + 1]); + } else if (strcmp(attr[i], "fy") == 0) { + grad->radial.fy = nsvg__parseCoordinateRaw(attr[i + 1]); + } else if (strcmp(attr[i], "x1") == 0) { + grad->linear.x1 = nsvg__parseCoordinateRaw(attr[i + 1]); + } else if (strcmp(attr[i], "y1") == 0) { + grad->linear.y1 = nsvg__parseCoordinateRaw(attr[i + 1]); + } else if (strcmp(attr[i], "x2") == 0) { + grad->linear.x2 = nsvg__parseCoordinateRaw(attr[i + 1]); + } else if (strcmp(attr[i], "y2") == 0) { + grad->linear.y2 = nsvg__parseCoordinateRaw(attr[i + 1]); + } else if (strcmp(attr[i], "spreadMethod") == 0) { + if (strcmp(attr[i+1], "pad") == 0) + grad->spread = NSVG_SPREAD_PAD; + else if (strcmp(attr[i+1], "reflect") == 0) + grad->spread = NSVG_SPREAD_REFLECT; + else if (strcmp(attr[i+1], "repeat") == 0) + grad->spread = NSVG_SPREAD_REPEAT; + } else if (strcmp(attr[i], "xlink:href") == 0) { + const char *href = attr[i+1]; + strncpy(grad->ref, href+1, 62); + grad->ref[62] = '\0'; + } + } + } + + grad->next = p->gradients; + p->gradients = grad; +} + +static void nsvg__parseGradientStop(NSVGparser* p, const char** attr) +{ + NSVGattrib* curAttr = nsvg__getAttr(p); + NSVGgradientData* grad; + NSVGgradientStop* stop; + int i, idx; + + curAttr->stopOffset = 0; + curAttr->stopColor = 0; + curAttr->stopOpacity = 1.0f; + + for (i = 0; attr[i]; i += 2) { + nsvg__parseAttr(p, attr[i], attr[i + 1]); + } + + // Add stop to the last gradient. + grad = p->gradients; + if (grad == NULL) return; + + grad->nstops++; + grad->stops = (NSVGgradientStop*)realloc(grad->stops, sizeof(NSVGgradientStop)*grad->nstops); + if (grad->stops == NULL) return; + + // Insert + idx = grad->nstops-1; + for (i = 0; i < grad->nstops-1; i++) { + if (curAttr->stopOffset < grad->stops[i].offset) { + idx = i; + break; + } + } + if (idx != grad->nstops-1) { + for (i = grad->nstops-1; i > idx; i--) + grad->stops[i] = grad->stops[i-1]; + } + + stop = &grad->stops[idx]; + stop->color = curAttr->stopColor; + stop->color |= (unsigned int)(curAttr->stopOpacity*255) << 24; + stop->offset = curAttr->stopOffset; +} + +static void nsvg__startElement(void* ud, const char* el, const char** attr) +{ + NSVGparser* p = (NSVGparser*)ud; + + if (p->defsFlag) { + // Skip everything but gradients and styles in defs + if (strcmp(el, "linearGradient") == 0) { + nsvg__parseGradient(p, attr, NSVG_PAINT_LINEAR_GRADIENT); + } else if (strcmp(el, "radialGradient") == 0) { + nsvg__parseGradient(p, attr, NSVG_PAINT_RADIAL_GRADIENT); + } else if (strcmp(el, "stop") == 0) { + nsvg__parseGradientStop(p, attr); + } else if (strcmp(el, "style") == 0) { + p->styleFlag = 1; + } + return; + } + + if (strcmp(el, "g") == 0) { + nsvg__pushAttr(p); + nsvg__parseAttribs(p, attr); + } else if (strcmp(el, "path") == 0) { + if (p->pathFlag) // Do not allow nested paths. + return; + nsvg__pushAttr(p); + nsvg__parsePath(p, attr); + nsvg__popAttr(p); + } else if (strcmp(el, "rect") == 0) { + nsvg__pushAttr(p); + nsvg__parseRect(p, attr); + nsvg__popAttr(p); + } else if (strcmp(el, "circle") == 0) { + nsvg__pushAttr(p); + nsvg__parseCircle(p, attr); + nsvg__popAttr(p); + } else if (strcmp(el, "ellipse") == 0) { + nsvg__pushAttr(p); + nsvg__parseEllipse(p, attr); + nsvg__popAttr(p); + } else if (strcmp(el, "line") == 0) { + nsvg__pushAttr(p); + nsvg__parseLine(p, attr); + nsvg__popAttr(p); + } else if (strcmp(el, "polyline") == 0) { + nsvg__pushAttr(p); + nsvg__parsePoly(p, attr, 0); + nsvg__popAttr(p); + } else if (strcmp(el, "polygon") == 0) { + nsvg__pushAttr(p); + nsvg__parsePoly(p, attr, 1); + nsvg__popAttr(p); + } else if (strcmp(el, "linearGradient") == 0) { + nsvg__parseGradient(p, attr, NSVG_PAINT_LINEAR_GRADIENT); + } else if (strcmp(el, "radialGradient") == 0) { + nsvg__parseGradient(p, attr, NSVG_PAINT_RADIAL_GRADIENT); + } else if (strcmp(el, "stop") == 0) { + nsvg__parseGradientStop(p, attr); + } else if (strcmp(el, "defs") == 0) { + p->defsFlag = 1; + } else if (strcmp(el, "svg") == 0) { + nsvg__parseSVG(p, attr); + } else if (strcmp(el, "style") == 0) { + p->styleFlag = 1; + } +} + +static void nsvg__endElement(void* ud, const char* el) +{ + NSVGparser* p = (NSVGparser*)ud; + + if (strcmp(el, "g") == 0) { + nsvg__popAttr(p); + } else if (strcmp(el, "path") == 0) { + p->pathFlag = 0; + } else if (strcmp(el, "defs") == 0) { + p->defsFlag = 0; + } else if (strcmp(el, "style") == 0) { + p->styleFlag = 0; + } +} + +static char *nsvg__strndup(const char *s, size_t n) +{ + char *result = (char *)malloc(n + 1); + if (result == NULL) + return NULL; + + memcpy(result, s, n); + result[n] = '\0'; + return result; +} + +static void nsvg__content(void* ud, const char* s) +{ + NSVGparser* p = (NSVGparser*)ud; + if (!p->styleFlag) + return; + + // Parse all the styles inside the style block. Each style's content will be later processed using nsvg__parseStyle(). + // Note: We only support selector lists of simple class selectors (e.g. ".foo, .bar { ... }"). + while (*s) { + NSVGstyleDeclaration* styles[NSVG_MAX_CLASSES]; + int nstyles = 0; + const char* propsStart; + const char* propsEnd; + int i; + + // 1) Parse the selector list up to '{'. For each simple class selector ('.name'), + // allocate a new NSVGstyleDeclaration into the local staging array. Styles are + // only committed to p->styles in step 3 below, once their propertiesText has + // also been allocated successfully. + while (*s && *s != '{') { + const char* selStart; + const char* selEnd; + NSVGstyleDeclaration* style; + + while (*s && (nsvg__isspace(*s) || *s == ',')) + s++; + if (!*s || *s == '{') + break; + + selStart = s; + while (*s && !nsvg__isspace(*s) && *s != ',' && *s != '{') + s++; + selEnd = s; + + if (*selStart != '.' || nstyles >= NSVG_MAX_CLASSES) + continue; // unsupported selector, or staging array full + selStart++; // strip leading '.' + + style = (NSVGstyleDeclaration*)malloc(sizeof(NSVGstyleDeclaration)); + if (style == NULL) + continue; + style->className = nsvg__strndup(selStart, (size_t)(selEnd - selStart)); + if (style->className == NULL) { + free(style); + continue; + } + style->propertiesText = NULL; + style->next = NULL; + styles[nstyles++] = style; + } + if (!*s) { + // No '{' found - discard pending styles and stop. + for (i = 0; i < nstyles; i++) { + free(styles[i]->className); + free(styles[i]); + } + break; + } + s++; // advance past '{' + + // 2) Find the end of the properties block (up to '}'). + propsStart = s; + while (*s && *s != '}') + s++; + propsEnd = s; + + // 3) Allocate propertiesText for each pending style. Commit successful ones to + // p->styles (head-insertion); free any whose allocation fails. + for (i = 0; i < nstyles; i++) { + styles[i]->propertiesText = nsvg__strndup(propsStart, (size_t)(propsEnd - propsStart)); + if (styles[i]->propertiesText == NULL) { + free(styles[i]->className); + free(styles[i]); + } else { + styles[i]->next = p->styles; + p->styles = styles[i]; + } + } + + if (*s) + s++; // advance past '}' + } +} + +static void nsvg__imageBounds(NSVGparser* p, float* bounds) +{ + NSVGshape* shape; + shape = p->image->shapes; + if (shape == NULL) { + bounds[0] = bounds[1] = bounds[2] = bounds[3] = 0.0; + return; + } + bounds[0] = shape->bounds[0]; + bounds[1] = shape->bounds[1]; + bounds[2] = shape->bounds[2]; + bounds[3] = shape->bounds[3]; + for (shape = shape->next; shape != NULL; shape = shape->next) { + bounds[0] = nsvg__minf(bounds[0], shape->bounds[0]); + bounds[1] = nsvg__minf(bounds[1], shape->bounds[1]); + bounds[2] = nsvg__maxf(bounds[2], shape->bounds[2]); + bounds[3] = nsvg__maxf(bounds[3], shape->bounds[3]); + } +} + +static float nsvg__viewAlign(float content, float container, int type) +{ + if (type == NSVG_ALIGN_MIN) + return 0; + else if (type == NSVG_ALIGN_MAX) + return container - content; + // mid + return (container - content) * 0.5f; +} + +static void nsvg__scaleGradient(NSVGgradient* grad, float tx, float ty, float sx, float sy) +{ + float t[6]; + nsvg__xformSetTranslation(t, tx, ty); + nsvg__xformMultiply (grad->xform, t); + + nsvg__xformSetScale(t, sx, sy); + nsvg__xformMultiply (grad->xform, t); +} + +static void nsvg__scaleToViewbox(NSVGparser* p, const char* units) +{ + NSVGshape* shape; + NSVGpath* path; + float tx, ty, sx, sy, us, bounds[4], t[6], avgs; + int i; + float* pt; + + // Guess image size if not set completely. + nsvg__imageBounds(p, bounds); + + if (p->viewWidth == 0) { + if (p->image->width > 0) { + p->viewWidth = p->image->width; + } else { + p->viewMinx = bounds[0]; + p->viewWidth = bounds[2] - bounds[0]; + } + } + if (p->viewHeight == 0) { + if (p->image->height > 0) { + p->viewHeight = p->image->height; + } else { + p->viewMiny = bounds[1]; + p->viewHeight = bounds[3] - bounds[1]; + } + } + if (p->image->width == 0) + p->image->width = p->viewWidth; + if (p->image->height == 0) + p->image->height = p->viewHeight; + + tx = -p->viewMinx; + ty = -p->viewMiny; + sx = p->viewWidth > 0 ? p->image->width / p->viewWidth : 0; + sy = p->viewHeight > 0 ? p->image->height / p->viewHeight : 0; + // Unit scaling + us = 1.0f / nsvg__convertToPixels(p, nsvg__coord(1.0f, nsvg__parseUnits(units)), 0.0f, 1.0f); + + // Fix aspect ratio + if (p->alignType == NSVG_ALIGN_MEET) { + // fit whole image into viewbox + sx = sy = nsvg__minf(sx, sy); + tx += nsvg__viewAlign(p->viewWidth*sx, p->image->width, p->alignX) / sx; + ty += nsvg__viewAlign(p->viewHeight*sy, p->image->height, p->alignY) / sy; + } else if (p->alignType == NSVG_ALIGN_SLICE) { + // fill whole viewbox with image + sx = sy = nsvg__maxf(sx, sy); + tx += nsvg__viewAlign(p->viewWidth*sx, p->image->width, p->alignX) / sx; + ty += nsvg__viewAlign(p->viewHeight*sy, p->image->height, p->alignY) / sy; + } + + // Transform + sx *= us; + sy *= us; + avgs = (sx+sy) / 2.0f; + for (shape = p->image->shapes; shape != NULL; shape = shape->next) { + shape->bounds[0] = (shape->bounds[0] + tx) * sx; + shape->bounds[1] = (shape->bounds[1] + ty) * sy; + shape->bounds[2] = (shape->bounds[2] + tx) * sx; + shape->bounds[3] = (shape->bounds[3] + ty) * sy; + for (path = shape->paths; path != NULL; path = path->next) { + path->bounds[0] = (path->bounds[0] + tx) * sx; + path->bounds[1] = (path->bounds[1] + ty) * sy; + path->bounds[2] = (path->bounds[2] + tx) * sx; + path->bounds[3] = (path->bounds[3] + ty) * sy; + for (i =0; i < path->npts; i++) { + pt = &path->pts[i*2]; + pt[0] = (pt[0] + tx) * sx; + pt[1] = (pt[1] + ty) * sy; + } + } + + if (shape->fill.type == NSVG_PAINT_LINEAR_GRADIENT || shape->fill.type == NSVG_PAINT_RADIAL_GRADIENT) { + nsvg__scaleGradient(shape->fill.gradient, tx,ty, sx,sy); + memcpy(t, shape->fill.gradient->xform, sizeof(float)*6); + nsvg__xformInverse(shape->fill.gradient->xform, t); + } + if (shape->stroke.type == NSVG_PAINT_LINEAR_GRADIENT || shape->stroke.type == NSVG_PAINT_RADIAL_GRADIENT) { + nsvg__scaleGradient(shape->stroke.gradient, tx,ty, sx,sy); + memcpy(t, shape->stroke.gradient->xform, sizeof(float)*6); + nsvg__xformInverse(shape->stroke.gradient->xform, t); + } + + shape->strokeWidth *= avgs; + shape->strokeDashOffset *= avgs; + for (i = 0; i < shape->strokeDashCount; i++) + shape->strokeDashArray[i] *= avgs; + } +} + +static void nsvg__createGradients(NSVGparser* p) +{ + NSVGshape* shape; + + for (shape = p->image->shapes; shape != NULL; shape = shape->next) { + if (shape->fill.type == NSVG_PAINT_UNDEF) { + if (shape->fillGradient[0] != '\0') { + float inv[6], localBounds[4]; + nsvg__xformInverse(inv, shape->xform); + nsvg__getLocalBounds(localBounds, shape, inv); + shape->fill.gradient = nsvg__createGradient(p, shape->fillGradient, localBounds, shape->xform, &shape->fill.type); + } + if (shape->fill.type == NSVG_PAINT_UNDEF) { + shape->fill.type = NSVG_PAINT_NONE; + } + } + if (shape->stroke.type == NSVG_PAINT_UNDEF) { + if (shape->strokeGradient[0] != '\0') { + float inv[6], localBounds[4]; + nsvg__xformInverse(inv, shape->xform); + nsvg__getLocalBounds(localBounds, shape, inv); + shape->stroke.gradient = nsvg__createGradient(p, shape->strokeGradient, localBounds, shape->xform, &shape->stroke.type); + } + if (shape->stroke.type == NSVG_PAINT_UNDEF) { + shape->stroke.type = NSVG_PAINT_NONE; + } + } + } +} + +NSVGimage* nsvgParse(char* input, const char* units, float dpi) +{ + NSVGparser* p; + NSVGimage* ret = 0; + + p = nsvg__createParser(); + if (p == NULL) { + return NULL; + } + p->dpi = dpi; + + nsvg__parseXML(input, nsvg__startElement, nsvg__endElement, nsvg__content, p); + + // Create gradients after all definitions have been parsed + nsvg__createGradients(p); + + // Scale to viewBox + nsvg__scaleToViewbox(p, units); + + ret = p->image; + p->image = NULL; + + nsvg__deleteParser(p); + + return ret; +} + +NSVGimage* nsvgParseFromFile(const char* filename, const char* units, float dpi) +{ + FILE* fp = NULL; + size_t size; + char* data = NULL; + NSVGimage* image = NULL; + + fp = fopen(filename, "rb"); + if (!fp) goto error; + fseek(fp, 0, SEEK_END); + size = ftell(fp); + fseek(fp, 0, SEEK_SET); + data = (char*)malloc(size+1); + if (data == NULL) goto error; + if (fread(data, 1, size, fp) != size) goto error; + data[size] = '\0'; // Must be null terminated. + fclose(fp); + image = nsvgParse(data, units, dpi); + free(data); + + return image; + +error: + if (fp) fclose(fp); + if (data) free(data); + if (image) nsvgDelete(image); + return NULL; +} + +NSVGpath* nsvgDuplicatePath(NSVGpath* p) +{ + NSVGpath* res = NULL; + + if (p == NULL) + return NULL; + + res = (NSVGpath*)malloc(sizeof(NSVGpath)); + if (res == NULL) goto error; + memset(res, 0, sizeof(NSVGpath)); + + res->pts = (float*)malloc(p->npts*2*sizeof(float)); + if (res->pts == NULL) goto error; + memcpy(res->pts, p->pts, p->npts * sizeof(float) * 2); + res->npts = p->npts; + + memcpy(res->bounds, p->bounds, sizeof(p->bounds)); + + res->closed = p->closed; + + return res; + +error: + if (res != NULL) { + free(res->pts); + free(res); + } + return NULL; +} + +void nsvgDelete(NSVGimage* image) +{ + NSVGshape *snext, *shape; + if (image == NULL) return; + shape = image->shapes; + while (shape != NULL) { + snext = shape->next; + nsvg__deletePaths(shape->paths); + nsvg__deletePaint(&shape->fill); + nsvg__deletePaint(&shape->stroke); + free(shape); + shape = snext; + } + free(image); +} + +#endif // NANOSVG_IMPLEMENTATION + +#endif // NANOSVG_H diff --git a/src/third_party/nanosvgrast.h b/src/third_party/nanosvgrast.h new file mode 100644 index 0000000..ba512c7 --- /dev/null +++ b/src/third_party/nanosvgrast.h @@ -0,0 +1,1472 @@ +/* + * Copyright (c) 2013-14 Mikko Mononen memon@inside.org + * + * This software is provided 'as-is', without any express or implied + * warranty. In no event will the authors be held liable for any damages + * arising from the use of this software. + * + * Permission is granted to anyone to use this software for any purpose, + * including commercial applications, and to alter it and redistribute it + * freely, subject to the following restrictions: + * + * 1. The origin of this software must not be misrepresented; you must not + * claim that you wrote the original software. If you use this software + * in a product, an acknowledgment in the product documentation would be + * appreciated but is not required. + * 2. Altered source versions must be plainly marked as such, and must not be + * misrepresented as being the original software. + * 3. This notice may not be removed or altered from any source distribution. + * + * The polygon rasterization is heavily based on stb_truetype rasterizer + * by Sean Barrett - http://nothings.org/ + * + */ + +#ifndef NANOSVGRAST_H +#define NANOSVGRAST_H + +#include "nanosvg.h" + +#ifndef NANOSVGRAST_CPLUSPLUS +#ifdef __cplusplus +extern "C" { +#endif +#endif + +typedef struct NSVGrasterizer NSVGrasterizer; + +/* Example Usage: + // Load SVG + NSVGimage* image; + image = nsvgParseFromFile("test.svg", "px", 96); + + // Create rasterizer (can be used to render multiple images). + struct NSVGrasterizer* rast = nsvgCreateRasterizer(); + // Allocate memory for image + unsigned char* img = malloc(w*h*4); + // Rasterize + nsvgRasterize(rast, image, 0,0,1, img, w, h, w*4); +*/ + +// Allocated rasterizer context. +NSVGrasterizer* nsvgCreateRasterizer(void); + +// Rasterizes SVG image, returns RGBA image (non-premultiplied alpha) +// r - pointer to rasterizer context +// image - pointer to image to rasterize +// tx,ty - image offset (applied after scaling) +// scale - image scale +// dst - pointer to destination image data, 4 bytes per pixel (RGBA) +// w - width of the image to render +// h - height of the image to render +// stride - number of bytes per scaleline in the destination buffer +void nsvgRasterize(NSVGrasterizer* r, + NSVGimage* image, float tx, float ty, float scale, + unsigned char* dst, int w, int h, int stride); + +// Deletes rasterizer context. +void nsvgDeleteRasterizer(NSVGrasterizer*); + + +#ifndef NANOSVGRAST_CPLUSPLUS +#ifdef __cplusplus +} +#endif +#endif + +#ifdef NANOSVGRAST_IMPLEMENTATION + +#include +#include +#include + +#define NSVG__SUBSAMPLES 5 +#define NSVG__FIXSHIFT 10 +#define NSVG__FIX (1 << NSVG__FIXSHIFT) +#define NSVG__FIXMASK (NSVG__FIX-1) +#define NSVG__MEMPAGE_SIZE 1024 + +typedef struct NSVGedge { + float x0,y0, x1,y1; + int dir; + struct NSVGedge* next; +} NSVGedge; + +typedef struct NSVGpoint { + float x, y; + float dx, dy; + float len; + float dmx, dmy; + unsigned char flags; +} NSVGpoint; + +typedef struct NSVGactiveEdge { + int x,dx; + float ey; + int dir; + struct NSVGactiveEdge *next; +} NSVGactiveEdge; + +typedef struct NSVGmemPage { + unsigned char mem[NSVG__MEMPAGE_SIZE]; + int size; + struct NSVGmemPage* next; +} NSVGmemPage; + +typedef struct NSVGcachedPaint { + signed char type; + char spread; + float xform[6]; + unsigned int colors[256]; +} NSVGcachedPaint; + +struct NSVGrasterizer +{ + float px, py; + + float tessTol; + float distTol; + + NSVGedge* edges; + int nedges; + int cedges; + + NSVGpoint* points; + int npoints; + int cpoints; + + NSVGpoint* points2; + int npoints2; + int cpoints2; + + NSVGactiveEdge* freelist; + NSVGmemPage* pages; + NSVGmemPage* curpage; + + unsigned char* scanline; + int cscanline; + + unsigned char* bitmap; + int width, height, stride; +}; + +NSVGrasterizer* nsvgCreateRasterizer(void) +{ + NSVGrasterizer* r = (NSVGrasterizer*)malloc(sizeof(NSVGrasterizer)); + if (r == NULL) goto error; + memset(r, 0, sizeof(NSVGrasterizer)); + + r->tessTol = 0.25f; + r->distTol = 0.01f; + + return r; + +error: + nsvgDeleteRasterizer(r); + return NULL; +} + +void nsvgDeleteRasterizer(NSVGrasterizer* r) +{ + NSVGmemPage* p; + + if (r == NULL) return; + + p = r->pages; + while (p != NULL) { + NSVGmemPage* next = p->next; + free(p); + p = next; + } + + if (r->edges) free(r->edges); + if (r->points) free(r->points); + if (r->points2) free(r->points2); + if (r->scanline) free(r->scanline); + + free(r); +} + +static NSVGmemPage* nsvg__nextPage(NSVGrasterizer* r, NSVGmemPage* cur) +{ + NSVGmemPage *newp; + + // If using existing chain, return the next page in chain + if (cur != NULL && cur->next != NULL) { + return cur->next; + } + + // Alloc new page + newp = (NSVGmemPage*)malloc(sizeof(NSVGmemPage)); + if (newp == NULL) return NULL; + memset(newp, 0, sizeof(NSVGmemPage)); + + // Add to linked list + if (cur != NULL) + cur->next = newp; + else + r->pages = newp; + + return newp; +} + +static void nsvg__resetPool(NSVGrasterizer* r) +{ + NSVGmemPage* p = r->pages; + while (p != NULL) { + p->size = 0; + p = p->next; + } + r->curpage = r->pages; +} + +static unsigned char* nsvg__alloc(NSVGrasterizer* r, int size) +{ + unsigned char* buf; + if (size > NSVG__MEMPAGE_SIZE) return NULL; + if (r->curpage == NULL || r->curpage->size+size > NSVG__MEMPAGE_SIZE) { + r->curpage = nsvg__nextPage(r, r->curpage); + } + buf = &r->curpage->mem[r->curpage->size]; + r->curpage->size += size; + return buf; +} + +static int nsvg__ptEquals(float x1, float y1, float x2, float y2, float tol) +{ + float dx = x2 - x1; + float dy = y2 - y1; + return dx*dx + dy*dy < tol*tol; +} + +static void nsvg__addPathPoint(NSVGrasterizer* r, float x, float y, int flags) +{ + NSVGpoint* pt; + + if (r->npoints > 0) { + pt = &r->points[r->npoints-1]; + if (nsvg__ptEquals(pt->x,pt->y, x,y, r->distTol)) { + pt->flags = (unsigned char)(pt->flags | flags); + return; + } + } + + if (r->npoints+1 > r->cpoints) { + r->cpoints = r->cpoints > 0 ? r->cpoints * 2 : 64; + r->points = (NSVGpoint*)realloc(r->points, sizeof(NSVGpoint) * r->cpoints); + if (r->points == NULL) return; + } + + pt = &r->points[r->npoints]; + pt->x = x; + pt->y = y; + pt->flags = (unsigned char)flags; + r->npoints++; +} + +static void nsvg__appendPathPoint(NSVGrasterizer* r, NSVGpoint pt) +{ + if (r->npoints+1 > r->cpoints) { + r->cpoints = r->cpoints > 0 ? r->cpoints * 2 : 64; + r->points = (NSVGpoint*)realloc(r->points, sizeof(NSVGpoint) * r->cpoints); + if (r->points == NULL) return; + } + r->points[r->npoints] = pt; + r->npoints++; +} + +static void nsvg__duplicatePoints(NSVGrasterizer* r) +{ + if (r->npoints > r->cpoints2) { + r->cpoints2 = r->npoints; + r->points2 = (NSVGpoint*)realloc(r->points2, sizeof(NSVGpoint) * r->cpoints2); + if (r->points2 == NULL) return; + } + + memcpy(r->points2, r->points, sizeof(NSVGpoint) * r->npoints); + r->npoints2 = r->npoints; +} + +static void nsvg__addEdge(NSVGrasterizer* r, float x0, float y0, float x1, float y1) +{ + NSVGedge* e; + + // Skip horizontal edges + if (y0 == y1) + return; + + if (r->nedges+1 > r->cedges) { + r->cedges = r->cedges > 0 ? r->cedges * 2 : 64; + r->edges = (NSVGedge*)realloc(r->edges, sizeof(NSVGedge) * r->cedges); + if (r->edges == NULL) return; + } + + e = &r->edges[r->nedges]; + r->nedges++; + + if (y0 < y1) { + e->x0 = x0; + e->y0 = y0; + e->x1 = x1; + e->y1 = y1; + e->dir = 1; + } else { + e->x0 = x1; + e->y0 = y1; + e->x1 = x0; + e->y1 = y0; + e->dir = -1; + } +} + +static float nsvg__normalize(float *x, float* y) +{ + float d = sqrtf((*x)*(*x) + (*y)*(*y)); + if (d > 1e-6f) { + float id = 1.0f / d; + *x *= id; + *y *= id; + } + return d; +} + +static float nsvg__absf(float x) { return x < 0 ? -x : x; } +static float nsvg__roundf(float x) { return (x >= 0) ? floorf(x + 0.5) : ceilf(x - 0.5); } + +static void nsvg__flattenCubicBez(NSVGrasterizer* r, + float x1, float y1, float x2, float y2, + float x3, float y3, float x4, float y4, + int level, int type) +{ + float x12,y12,x23,y23,x34,y34,x123,y123,x234,y234,x1234,y1234; + float dx,dy,d2,d3; + + if (level > 10) return; + + x12 = (x1+x2)*0.5f; + y12 = (y1+y2)*0.5f; + x23 = (x2+x3)*0.5f; + y23 = (y2+y3)*0.5f; + x34 = (x3+x4)*0.5f; + y34 = (y3+y4)*0.5f; + x123 = (x12+x23)*0.5f; + y123 = (y12+y23)*0.5f; + + dx = x4 - x1; + dy = y4 - y1; + d2 = nsvg__absf((x2 - x4) * dy - (y2 - y4) * dx); + d3 = nsvg__absf((x3 - x4) * dy - (y3 - y4) * dx); + + if ((d2 + d3)*(d2 + d3) < r->tessTol * (dx*dx + dy*dy)) { + nsvg__addPathPoint(r, x4, y4, type); + return; + } + + x234 = (x23+x34)*0.5f; + y234 = (y23+y34)*0.5f; + x1234 = (x123+x234)*0.5f; + y1234 = (y123+y234)*0.5f; + + nsvg__flattenCubicBez(r, x1,y1, x12,y12, x123,y123, x1234,y1234, level+1, 0); + nsvg__flattenCubicBez(r, x1234,y1234, x234,y234, x34,y34, x4,y4, level+1, type); +} + +static void nsvg__flattenShape(NSVGrasterizer* r, NSVGshape* shape, float scale) +{ + int i, j; + NSVGpath* path; + + for (path = shape->paths; path != NULL; path = path->next) { + r->npoints = 0; + // Flatten path + nsvg__addPathPoint(r, path->pts[0]*scale, path->pts[1]*scale, 0); + for (i = 0; i < path->npts-1; i += 3) { + float* p = &path->pts[i*2]; + nsvg__flattenCubicBez(r, p[0]*scale,p[1]*scale, p[2]*scale,p[3]*scale, p[4]*scale,p[5]*scale, p[6]*scale,p[7]*scale, 0, 0); + } + // Close path + nsvg__addPathPoint(r, path->pts[0]*scale, path->pts[1]*scale, 0); + // Build edges + for (i = 0, j = r->npoints-1; i < r->npoints; j = i++) + nsvg__addEdge(r, r->points[j].x, r->points[j].y, r->points[i].x, r->points[i].y); + } +} + +enum NSVGpointFlags +{ + NSVG_PT_CORNER = 0x01, + NSVG_PT_BEVEL = 0x02, + NSVG_PT_LEFT = 0x04 +}; + +static void nsvg__initClosed(NSVGpoint* left, NSVGpoint* right, NSVGpoint* p0, NSVGpoint* p1, float lineWidth) +{ + float w = lineWidth * 0.5f; + float dx = p1->x - p0->x; + float dy = p1->y - p0->y; + float len = nsvg__normalize(&dx, &dy); + float px = p0->x + dx*len*0.5f, py = p0->y + dy*len*0.5f; + float dlx = dy, dly = -dx; + float lx = px - dlx*w, ly = py - dly*w; + float rx = px + dlx*w, ry = py + dly*w; + left->x = lx; left->y = ly; + right->x = rx; right->y = ry; +} + +static void nsvg__buttCap(NSVGrasterizer* r, NSVGpoint* left, NSVGpoint* right, NSVGpoint* p, float dx, float dy, float lineWidth, int connect) +{ + float w = lineWidth * 0.5f; + float px = p->x, py = p->y; + float dlx = dy, dly = -dx; + float lx = px - dlx*w, ly = py - dly*w; + float rx = px + dlx*w, ry = py + dly*w; + + nsvg__addEdge(r, lx, ly, rx, ry); + + if (connect) { + nsvg__addEdge(r, left->x, left->y, lx, ly); + nsvg__addEdge(r, rx, ry, right->x, right->y); + } + left->x = lx; left->y = ly; + right->x = rx; right->y = ry; +} + +static void nsvg__squareCap(NSVGrasterizer* r, NSVGpoint* left, NSVGpoint* right, NSVGpoint* p, float dx, float dy, float lineWidth, int connect) +{ + float w = lineWidth * 0.5f; + float px = p->x - dx*w, py = p->y - dy*w; + float dlx = dy, dly = -dx; + float lx = px - dlx*w, ly = py - dly*w; + float rx = px + dlx*w, ry = py + dly*w; + + nsvg__addEdge(r, lx, ly, rx, ry); + + if (connect) { + nsvg__addEdge(r, left->x, left->y, lx, ly); + nsvg__addEdge(r, rx, ry, right->x, right->y); + } + left->x = lx; left->y = ly; + right->x = rx; right->y = ry; +} + +#ifndef NSVG_PI +#define NSVG_PI (3.14159265358979323846264338327f) +#endif + +static void nsvg__roundCap(NSVGrasterizer* r, NSVGpoint* left, NSVGpoint* right, NSVGpoint* p, float dx, float dy, float lineWidth, int ncap, int connect) +{ + int i; + float w = lineWidth * 0.5f; + float px = p->x, py = p->y; + float dlx = dy, dly = -dx; + float lx = 0, ly = 0, rx = 0, ry = 0, prevx = 0, prevy = 0; + + for (i = 0; i < ncap; i++) { + float a = (float)i/(float)(ncap-1)*NSVG_PI; + float ax = cosf(a) * w, ay = sinf(a) * w; + float x = px - dlx*ax - dx*ay; + float y = py - dly*ax - dy*ay; + + if (i > 0) + nsvg__addEdge(r, prevx, prevy, x, y); + + prevx = x; + prevy = y; + + if (i == 0) { + lx = x; ly = y; + } else if (i == ncap-1) { + rx = x; ry = y; + } + } + + if (connect) { + nsvg__addEdge(r, left->x, left->y, lx, ly); + nsvg__addEdge(r, rx, ry, right->x, right->y); + } + + left->x = lx; left->y = ly; + right->x = rx; right->y = ry; +} + +static void nsvg__bevelJoin(NSVGrasterizer* r, NSVGpoint* left, NSVGpoint* right, NSVGpoint* p0, NSVGpoint* p1, float lineWidth) +{ + float w = lineWidth * 0.5f; + float dlx0 = p0->dy, dly0 = -p0->dx; + float dlx1 = p1->dy, dly1 = -p1->dx; + float lx0 = p1->x - (dlx0 * w), ly0 = p1->y - (dly0 * w); + float rx0 = p1->x + (dlx0 * w), ry0 = p1->y + (dly0 * w); + float lx1 = p1->x - (dlx1 * w), ly1 = p1->y - (dly1 * w); + float rx1 = p1->x + (dlx1 * w), ry1 = p1->y + (dly1 * w); + + nsvg__addEdge(r, lx0, ly0, left->x, left->y); + nsvg__addEdge(r, lx1, ly1, lx0, ly0); + + nsvg__addEdge(r, right->x, right->y, rx0, ry0); + nsvg__addEdge(r, rx0, ry0, rx1, ry1); + + left->x = lx1; left->y = ly1; + right->x = rx1; right->y = ry1; +} + +static void nsvg__miterJoin(NSVGrasterizer* r, NSVGpoint* left, NSVGpoint* right, NSVGpoint* p0, NSVGpoint* p1, float lineWidth) +{ + float w = lineWidth * 0.5f; + float dlx0 = p0->dy, dly0 = -p0->dx; + float dlx1 = p1->dy, dly1 = -p1->dx; + float lx0, rx0, lx1, rx1; + float ly0, ry0, ly1, ry1; + + if (p1->flags & NSVG_PT_LEFT) { + lx0 = lx1 = p1->x - p1->dmx * w; + ly0 = ly1 = p1->y - p1->dmy * w; + nsvg__addEdge(r, lx1, ly1, left->x, left->y); + + rx0 = p1->x + (dlx0 * w); + ry0 = p1->y + (dly0 * w); + rx1 = p1->x + (dlx1 * w); + ry1 = p1->y + (dly1 * w); + nsvg__addEdge(r, right->x, right->y, rx0, ry0); + nsvg__addEdge(r, rx0, ry0, rx1, ry1); + } else { + lx0 = p1->x - (dlx0 * w); + ly0 = p1->y - (dly0 * w); + lx1 = p1->x - (dlx1 * w); + ly1 = p1->y - (dly1 * w); + nsvg__addEdge(r, lx0, ly0, left->x, left->y); + nsvg__addEdge(r, lx1, ly1, lx0, ly0); + + rx0 = rx1 = p1->x + p1->dmx * w; + ry0 = ry1 = p1->y + p1->dmy * w; + nsvg__addEdge(r, right->x, right->y, rx1, ry1); + } + + left->x = lx1; left->y = ly1; + right->x = rx1; right->y = ry1; +} + +static void nsvg__roundJoin(NSVGrasterizer* r, NSVGpoint* left, NSVGpoint* right, NSVGpoint* p0, NSVGpoint* p1, float lineWidth, int ncap) +{ + int i, n; + float w = lineWidth * 0.5f; + float dlx0 = p0->dy, dly0 = -p0->dx; + float dlx1 = p1->dy, dly1 = -p1->dx; + float a0 = atan2f(dly0, dlx0); + float a1 = atan2f(dly1, dlx1); + float da = a1 - a0; + float lx, ly, rx, ry; + + if (da < NSVG_PI) da += NSVG_PI*2; + if (da > NSVG_PI) da -= NSVG_PI*2; + + n = (int)ceilf((nsvg__absf(da) / NSVG_PI) * (float)ncap); + if (n < 2) n = 2; + if (n > ncap) n = ncap; + + lx = left->x; + ly = left->y; + rx = right->x; + ry = right->y; + + for (i = 0; i < n; i++) { + float u = (float)i/(float)(n-1); + float a = a0 + u*da; + float ax = cosf(a) * w, ay = sinf(a) * w; + float lx1 = p1->x - ax, ly1 = p1->y - ay; + float rx1 = p1->x + ax, ry1 = p1->y + ay; + + nsvg__addEdge(r, lx1, ly1, lx, ly); + nsvg__addEdge(r, rx, ry, rx1, ry1); + + lx = lx1; ly = ly1; + rx = rx1; ry = ry1; + } + + left->x = lx; left->y = ly; + right->x = rx; right->y = ry; +} + +static void nsvg__straightJoin(NSVGrasterizer* r, NSVGpoint* left, NSVGpoint* right, NSVGpoint* p1, float lineWidth) +{ + float w = lineWidth * 0.5f; + float lx = p1->x - (p1->dmx * w), ly = p1->y - (p1->dmy * w); + float rx = p1->x + (p1->dmx * w), ry = p1->y + (p1->dmy * w); + + nsvg__addEdge(r, lx, ly, left->x, left->y); + nsvg__addEdge(r, right->x, right->y, rx, ry); + + left->x = lx; left->y = ly; + right->x = rx; right->y = ry; +} + +static int nsvg__curveDivs(float r, float arc, float tol) +{ + float da = acosf(r / (r + tol)) * 2.0f; + int divs = (int)ceilf(arc / da); + if (divs < 2) divs = 2; + return divs; +} + +static void nsvg__expandStroke(NSVGrasterizer* r, NSVGpoint* points, int npoints, int closed, int lineJoin, int lineCap, float lineWidth) +{ + int ncap = nsvg__curveDivs(lineWidth*0.5f, NSVG_PI, r->tessTol); // Calculate divisions per half circle. + NSVGpoint left = {0,0,0,0,0,0,0,0}, right = {0,0,0,0,0,0,0,0}, firstLeft = {0,0,0,0,0,0,0,0}, firstRight = {0,0,0,0,0,0,0,0}; + NSVGpoint* p0, *p1; + int j, s, e; + + // Build stroke edges + if (closed) { + // Looping + p0 = &points[npoints-1]; + p1 = &points[0]; + s = 0; + e = npoints; + } else { + // Add cap + p0 = &points[0]; + p1 = &points[1]; + s = 1; + e = npoints-1; + } + + if (closed) { + nsvg__initClosed(&left, &right, p0, p1, lineWidth); + firstLeft = left; + firstRight = right; + } else { + // Add cap + float dx = p1->x - p0->x; + float dy = p1->y - p0->y; + nsvg__normalize(&dx, &dy); + if (lineCap == NSVG_CAP_BUTT) + nsvg__buttCap(r, &left, &right, p0, dx, dy, lineWidth, 0); + else if (lineCap == NSVG_CAP_SQUARE) + nsvg__squareCap(r, &left, &right, p0, dx, dy, lineWidth, 0); + else if (lineCap == NSVG_CAP_ROUND) + nsvg__roundCap(r, &left, &right, p0, dx, dy, lineWidth, ncap, 0); + } + + for (j = s; j < e; ++j) { + if (p1->flags & NSVG_PT_CORNER) { + if (lineJoin == NSVG_JOIN_ROUND) + nsvg__roundJoin(r, &left, &right, p0, p1, lineWidth, ncap); + else if (lineJoin == NSVG_JOIN_BEVEL || (p1->flags & NSVG_PT_BEVEL)) + nsvg__bevelJoin(r, &left, &right, p0, p1, lineWidth); + else + nsvg__miterJoin(r, &left, &right, p0, p1, lineWidth); + } else { + nsvg__straightJoin(r, &left, &right, p1, lineWidth); + } + p0 = p1++; + } + + if (closed) { + // Loop it + nsvg__addEdge(r, firstLeft.x, firstLeft.y, left.x, left.y); + nsvg__addEdge(r, right.x, right.y, firstRight.x, firstRight.y); + } else { + // Add cap + float dx = p1->x - p0->x; + float dy = p1->y - p0->y; + nsvg__normalize(&dx, &dy); + if (lineCap == NSVG_CAP_BUTT) + nsvg__buttCap(r, &right, &left, p1, -dx, -dy, lineWidth, 1); + else if (lineCap == NSVG_CAP_SQUARE) + nsvg__squareCap(r, &right, &left, p1, -dx, -dy, lineWidth, 1); + else if (lineCap == NSVG_CAP_ROUND) + nsvg__roundCap(r, &right, &left, p1, -dx, -dy, lineWidth, ncap, 1); + } +} + +static void nsvg__prepareStroke(NSVGrasterizer* r, float miterLimit, int lineJoin) +{ + int i, j; + NSVGpoint* p0, *p1; + + p0 = &r->points[r->npoints-1]; + p1 = &r->points[0]; + for (i = 0; i < r->npoints; i++) { + // Calculate segment direction and length + p0->dx = p1->x - p0->x; + p0->dy = p1->y - p0->y; + p0->len = nsvg__normalize(&p0->dx, &p0->dy); + // Advance + p0 = p1++; + } + + // calculate joins + p0 = &r->points[r->npoints-1]; + p1 = &r->points[0]; + for (j = 0; j < r->npoints; j++) { + float dlx0, dly0, dlx1, dly1, dmr2, cross; + dlx0 = p0->dy; + dly0 = -p0->dx; + dlx1 = p1->dy; + dly1 = -p1->dx; + // Calculate extrusions + p1->dmx = (dlx0 + dlx1) * 0.5f; + p1->dmy = (dly0 + dly1) * 0.5f; + dmr2 = p1->dmx*p1->dmx + p1->dmy*p1->dmy; + if (dmr2 > 0.000001f) { + float s2 = 1.0f / dmr2; + if (s2 > 600.0f) { + s2 = 600.0f; + } + p1->dmx *= s2; + p1->dmy *= s2; + } + + // Clear flags, but keep the corner. + p1->flags = (p1->flags & NSVG_PT_CORNER) ? NSVG_PT_CORNER : 0; + + // Keep track of left turns. + cross = p1->dx * p0->dy - p0->dx * p1->dy; + if (cross > 0.0f) + p1->flags |= NSVG_PT_LEFT; + + // Check to see if the corner needs to be beveled. + if (p1->flags & NSVG_PT_CORNER) { + if ((dmr2 * miterLimit*miterLimit) < 1.0f || lineJoin == NSVG_JOIN_BEVEL || lineJoin == NSVG_JOIN_ROUND) { + p1->flags |= NSVG_PT_BEVEL; + } + } + + p0 = p1++; + } +} + +static void nsvg__flattenShapeStroke(NSVGrasterizer* r, NSVGshape* shape, float scale) +{ + int i, j, closed; + NSVGpath* path; + NSVGpoint* p0, *p1; + float miterLimit = shape->miterLimit; + int lineJoin = shape->strokeLineJoin; + int lineCap = shape->strokeLineCap; + float lineWidth = shape->strokeWidth * scale; + + for (path = shape->paths; path != NULL; path = path->next) { + // Flatten path + r->npoints = 0; + nsvg__addPathPoint(r, path->pts[0]*scale, path->pts[1]*scale, NSVG_PT_CORNER); + for (i = 0; i < path->npts-1; i += 3) { + float* p = &path->pts[i*2]; + nsvg__flattenCubicBez(r, p[0]*scale,p[1]*scale, p[2]*scale,p[3]*scale, p[4]*scale,p[5]*scale, p[6]*scale,p[7]*scale, 0, NSVG_PT_CORNER); + } + if (r->npoints < 2) + continue; + + closed = path->closed; + + // If the first and last points are the same, remove the last, mark as closed path. + p0 = &r->points[r->npoints-1]; + p1 = &r->points[0]; + if (nsvg__ptEquals(p0->x,p0->y, p1->x,p1->y, r->distTol)) { + r->npoints--; + p0 = &r->points[r->npoints-1]; + closed = 1; + } + + if (shape->strokeDashCount > 0) { + int idash = 0, dashState = 1; + float totalDist = 0, dashLen, allDashLen, dashOffset; + NSVGpoint cur; + + if (closed) + nsvg__appendPathPoint(r, r->points[0]); + + // Duplicate points -> points2. + nsvg__duplicatePoints(r); + + r->npoints = 0; + cur = r->points2[0]; + nsvg__appendPathPoint(r, cur); + + // Figure out dash offset. + allDashLen = 0; + for (j = 0; j < shape->strokeDashCount; j++) + allDashLen += shape->strokeDashArray[j]; + if (shape->strokeDashCount & 1) + allDashLen *= 2.0f; + // Find location inside pattern + dashOffset = fmodf(shape->strokeDashOffset, allDashLen); + if (dashOffset < 0.0f) + dashOffset += allDashLen; + + while (dashOffset > shape->strokeDashArray[idash]) { + dashOffset -= shape->strokeDashArray[idash]; + idash = (idash + 1) % shape->strokeDashCount; + } + dashLen = (shape->strokeDashArray[idash] - dashOffset) * scale; + + for (j = 1; j < r->npoints2; ) { + float dx = r->points2[j].x - cur.x; + float dy = r->points2[j].y - cur.y; + float dist = sqrtf(dx*dx + dy*dy); + + if ((totalDist + dist) > dashLen) { + // Calculate intermediate point + float d = (dashLen - totalDist) / dist; + float x = cur.x + dx * d; + float y = cur.y + dy * d; + nsvg__addPathPoint(r, x, y, NSVG_PT_CORNER); + + // Stroke + if (r->npoints > 1 && dashState) { + nsvg__prepareStroke(r, miterLimit, lineJoin); + nsvg__expandStroke(r, r->points, r->npoints, 0, lineJoin, lineCap, lineWidth); + } + // Advance dash pattern + dashState = !dashState; + idash = (idash+1) % shape->strokeDashCount; + dashLen = shape->strokeDashArray[idash] * scale; + // Restart + cur.x = x; + cur.y = y; + cur.flags = NSVG_PT_CORNER; + totalDist = 0.0f; + r->npoints = 0; + nsvg__appendPathPoint(r, cur); + } else { + totalDist += dist; + cur = r->points2[j]; + nsvg__appendPathPoint(r, cur); + j++; + } + } + // Stroke any leftover path + if (r->npoints > 1 && dashState) { + nsvg__prepareStroke(r, miterLimit, lineJoin); + nsvg__expandStroke(r, r->points, r->npoints, 0, lineJoin, lineCap, lineWidth); + } + } else { + nsvg__prepareStroke(r, miterLimit, lineJoin); + nsvg__expandStroke(r, r->points, r->npoints, closed, lineJoin, lineCap, lineWidth); + } + } +} + +static int nsvg__cmpEdge(const void *p, const void *q) +{ + const NSVGedge* a = (const NSVGedge*)p; + const NSVGedge* b = (const NSVGedge*)q; + + if (a->y0 < b->y0) return -1; + if (a->y0 > b->y0) return 1; + return 0; +} + + +static NSVGactiveEdge* nsvg__addActive(NSVGrasterizer* r, NSVGedge* e, float startPoint) +{ + NSVGactiveEdge* z; + + if (r->freelist != NULL) { + // Restore from freelist. + z = r->freelist; + r->freelist = z->next; + } else { + // Alloc new edge. + z = (NSVGactiveEdge*)nsvg__alloc(r, sizeof(NSVGactiveEdge)); + if (z == NULL) return NULL; + } + + float dxdy = (e->x1 - e->x0) / (e->y1 - e->y0); +// STBTT_assert(e->y0 <= start_point); + // round dx down to avoid going too far + if (dxdy < 0) + z->dx = (int)(-nsvg__roundf(NSVG__FIX * -dxdy)); + else + z->dx = (int)nsvg__roundf(NSVG__FIX * dxdy); + z->x = (int)nsvg__roundf(NSVG__FIX * (e->x0 + dxdy * (startPoint - e->y0))); +// z->x -= off_x * FIX; + z->ey = e->y1; + z->next = 0; + z->dir = e->dir; + + return z; +} + +static void nsvg__freeActive(NSVGrasterizer* r, NSVGactiveEdge* z) +{ + z->next = r->freelist; + r->freelist = z; +} + +static void nsvg__fillScanline(unsigned char* scanline, int len, int x0, int x1, int maxWeight, int* xmin, int* xmax) +{ + int i = x0 >> NSVG__FIXSHIFT; + int j = x1 >> NSVG__FIXSHIFT; + if (i < *xmin) *xmin = i; + if (j > *xmax) *xmax = j; + if (i < len && j >= 0) { + if (i == j) { + // x0,x1 are the same pixel, so compute combined coverage + scanline[i] = (unsigned char)(scanline[i] + ((x1 - x0) * maxWeight >> NSVG__FIXSHIFT)); + } else { + if (i >= 0) // add antialiasing for x0 + scanline[i] = (unsigned char)(scanline[i] + (((NSVG__FIX - (x0 & NSVG__FIXMASK)) * maxWeight) >> NSVG__FIXSHIFT)); + else + i = -1; // clip + + if (j < len) // add antialiasing for x1 + scanline[j] = (unsigned char)(scanline[j] + (((x1 & NSVG__FIXMASK) * maxWeight) >> NSVG__FIXSHIFT)); + else + j = len; // clip + + for (++i; i < j; ++i) // fill pixels between x0 and x1 + scanline[i] = (unsigned char)(scanline[i] + maxWeight); + } + } +} + +// note: this routine clips fills that extend off the edges... ideally this +// wouldn't happen, but it could happen if the truetype glyph bounding boxes +// are wrong, or if the user supplies a too-small bitmap +static void nsvg__fillActiveEdges(unsigned char* scanline, int len, NSVGactiveEdge* e, int maxWeight, int* xmin, int* xmax, char fillRule) +{ + // non-zero winding fill + int x0 = 0, w = 0; + + if (fillRule == NSVG_FILLRULE_NONZERO) { + // Non-zero + while (e != NULL) { + if (w == 0) { + // if we're currently at zero, we need to record the edge start point + x0 = e->x; w += e->dir; + } else { + int x1 = e->x; w += e->dir; + // if we went to zero, we need to draw + if (w == 0) + nsvg__fillScanline(scanline, len, x0, x1, maxWeight, xmin, xmax); + } + e = e->next; + } + } else if (fillRule == NSVG_FILLRULE_EVENODD) { + // Even-odd + while (e != NULL) { + if (w == 0) { + // if we're currently at zero, we need to record the edge start point + x0 = e->x; w = 1; + } else { + int x1 = e->x; w = 0; + nsvg__fillScanline(scanline, len, x0, x1, maxWeight, xmin, xmax); + } + e = e->next; + } + } +} + +static float nsvg__clampf(float a, float mn, float mx) { + if (isnan(a)) + return mn; + return a < mn ? mn : (a > mx ? mx : a); +} + +static unsigned int nsvg__RGBA(unsigned char r, unsigned char g, unsigned char b, unsigned char a) +{ + return ((unsigned int)r) | ((unsigned int)g << 8) | ((unsigned int)b << 16) | ((unsigned int)a << 24); +} + +static unsigned int nsvg__lerpRGBA(unsigned int c0, unsigned int c1, float u) +{ + int iu = (int)(nsvg__clampf(u, 0.0f, 1.0f) * 256.0f); + int r = (((c0) & 0xff)*(256-iu) + (((c1) & 0xff)*iu)) >> 8; + int g = (((c0>>8) & 0xff)*(256-iu) + (((c1>>8) & 0xff)*iu)) >> 8; + int b = (((c0>>16) & 0xff)*(256-iu) + (((c1>>16) & 0xff)*iu)) >> 8; + int a = (((c0>>24) & 0xff)*(256-iu) + (((c1>>24) & 0xff)*iu)) >> 8; + return nsvg__RGBA((unsigned char)r, (unsigned char)g, (unsigned char)b, (unsigned char)a); +} + +static unsigned int nsvg__applyOpacity(unsigned int c, float u) +{ + int iu = (int)(nsvg__clampf(u, 0.0f, 1.0f) * 256.0f); + int r = (c) & 0xff; + int g = (c>>8) & 0xff; + int b = (c>>16) & 0xff; + int a = (((c>>24) & 0xff)*iu) >> 8; + return nsvg__RGBA((unsigned char)r, (unsigned char)g, (unsigned char)b, (unsigned char)a); +} + +static inline int nsvg__div255(int x) +{ + return ((x+1) * 257) >> 16; +} + +static void nsvg__scanlineSolid(unsigned char* dst, int count, unsigned char* cover, int x, int y, + float tx, float ty, float scale, NSVGcachedPaint* cache) +{ + + if (cache->type == NSVG_PAINT_COLOR) { + int i, cr, cg, cb, ca; + cr = cache->colors[0] & 0xff; + cg = (cache->colors[0] >> 8) & 0xff; + cb = (cache->colors[0] >> 16) & 0xff; + ca = (cache->colors[0] >> 24) & 0xff; + + for (i = 0; i < count; i++) { + int r,g,b; + int a = nsvg__div255((int)cover[0] * ca); + int ia = 255 - a; + // Premultiply + r = nsvg__div255(cr * a); + g = nsvg__div255(cg * a); + b = nsvg__div255(cb * a); + + // Blend over + r += nsvg__div255(ia * (int)dst[0]); + g += nsvg__div255(ia * (int)dst[1]); + b += nsvg__div255(ia * (int)dst[2]); + a += nsvg__div255(ia * (int)dst[3]); + + dst[0] = (unsigned char)r; + dst[1] = (unsigned char)g; + dst[2] = (unsigned char)b; + dst[3] = (unsigned char)a; + + cover++; + dst += 4; + } + } else if (cache->type == NSVG_PAINT_LINEAR_GRADIENT) { + // TODO: spread modes. + // TODO: plenty of opportunities to optimize. + float fx, fy, dx, gy; + float* t = cache->xform; + int i, cr, cg, cb, ca; + unsigned int c; + + fx = ((float)x - tx) / scale; + fy = ((float)y - ty) / scale; + dx = 1.0f / scale; + + for (i = 0; i < count; i++) { + int r,g,b,a,ia; + gy = fx*t[1] + fy*t[3] + t[5]; + c = cache->colors[(int)nsvg__clampf(gy*255.0f, 0, 255.0f)]; + cr = (c) & 0xff; + cg = (c >> 8) & 0xff; + cb = (c >> 16) & 0xff; + ca = (c >> 24) & 0xff; + + a = nsvg__div255((int)cover[0] * ca); + ia = 255 - a; + + // Premultiply + r = nsvg__div255(cr * a); + g = nsvg__div255(cg * a); + b = nsvg__div255(cb * a); + + // Blend over + r += nsvg__div255(ia * (int)dst[0]); + g += nsvg__div255(ia * (int)dst[1]); + b += nsvg__div255(ia * (int)dst[2]); + a += nsvg__div255(ia * (int)dst[3]); + + dst[0] = (unsigned char)r; + dst[1] = (unsigned char)g; + dst[2] = (unsigned char)b; + dst[3] = (unsigned char)a; + + cover++; + dst += 4; + fx += dx; + } + } else if (cache->type == NSVG_PAINT_RADIAL_GRADIENT) { + // TODO: spread modes. + // TODO: plenty of opportunities to optimize. + // TODO: focus (fx,fy) + float fx, fy, dx, gx, gy, gd; + float* t = cache->xform; + int i, cr, cg, cb, ca; + unsigned int c; + + fx = ((float)x - tx) / scale; + fy = ((float)y - ty) / scale; + dx = 1.0f / scale; + + for (i = 0; i < count; i++) { + int r,g,b,a,ia; + gx = fx*t[0] + fy*t[2] + t[4]; + gy = fx*t[1] + fy*t[3] + t[5]; + gd = sqrtf(gx*gx + gy*gy); + c = cache->colors[(int)nsvg__clampf(gd*255.0f, 0, 255.0f)]; + cr = (c) & 0xff; + cg = (c >> 8) & 0xff; + cb = (c >> 16) & 0xff; + ca = (c >> 24) & 0xff; + + a = nsvg__div255((int)cover[0] * ca); + ia = 255 - a; + + // Premultiply + r = nsvg__div255(cr * a); + g = nsvg__div255(cg * a); + b = nsvg__div255(cb * a); + + // Blend over + r += nsvg__div255(ia * (int)dst[0]); + g += nsvg__div255(ia * (int)dst[1]); + b += nsvg__div255(ia * (int)dst[2]); + a += nsvg__div255(ia * (int)dst[3]); + + dst[0] = (unsigned char)r; + dst[1] = (unsigned char)g; + dst[2] = (unsigned char)b; + dst[3] = (unsigned char)a; + + cover++; + dst += 4; + fx += dx; + } + } +} + +static void nsvg__rasterizeSortedEdges(NSVGrasterizer *r, float tx, float ty, float scale, NSVGcachedPaint* cache, char fillRule) +{ + NSVGactiveEdge *active = NULL; + int y, s; + int e = 0; + int maxWeight = (255 / NSVG__SUBSAMPLES); // weight per vertical scanline + int xmin, xmax; + + for (y = 0; y < r->height; y++) { + memset(r->scanline, 0, r->width); + xmin = r->width; + xmax = 0; + for (s = 0; s < NSVG__SUBSAMPLES; ++s) { + // find center of pixel for this scanline + float scany = (float)(y*NSVG__SUBSAMPLES + s) + 0.5f; + NSVGactiveEdge **step = &active; + + // update all active edges; + // remove all active edges that terminate before the center of this scanline + while (*step) { + NSVGactiveEdge *z = *step; + if (z->ey <= scany) { + *step = z->next; // delete from list +// NSVG__assert(z->valid); + nsvg__freeActive(r, z); + } else { + z->x += z->dx; // advance to position for current scanline + step = &((*step)->next); // advance through list + } + } + + // resort the list if needed + for (;;) { + int changed = 0; + step = &active; + while (*step && (*step)->next) { + if ((*step)->x > (*step)->next->x) { + NSVGactiveEdge* t = *step; + NSVGactiveEdge* q = t->next; + t->next = q->next; + q->next = t; + *step = q; + changed = 1; + } + step = &(*step)->next; + } + if (!changed) break; + } + + // insert all edges that start before the center of this scanline -- omit ones that also end on this scanline + while (e < r->nedges && r->edges[e].y0 <= scany) { + if (r->edges[e].y1 > scany) { + NSVGactiveEdge* z = nsvg__addActive(r, &r->edges[e], scany); + if (z == NULL) break; + // find insertion point + if (active == NULL) { + active = z; + } else if (z->x < active->x) { + // insert at front + z->next = active; + active = z; + } else { + // find thing to insert AFTER + NSVGactiveEdge* p = active; + while (p->next && p->next->x < z->x) + p = p->next; + // at this point, p->next->x is NOT < z->x + z->next = p->next; + p->next = z; + } + } + e++; + } + + // now process all active edges in non-zero fashion + if (active != NULL) + nsvg__fillActiveEdges(r->scanline, r->width, active, maxWeight, &xmin, &xmax, fillRule); + } + // Blit + if (xmin < 0) xmin = 0; + if (xmax > r->width-1) xmax = r->width-1; + if (xmin <= xmax) { + nsvg__scanlineSolid(&r->bitmap[y * r->stride] + xmin*4, xmax-xmin+1, &r->scanline[xmin], xmin, y, tx,ty, scale, cache); + } + } + +} + +static void nsvg__unpremultiplyAlpha(unsigned char* image, int w, int h, int stride) +{ + int x,y; + + // Unpremultiply + for (y = 0; y < h; y++) { + unsigned char *row = &image[y*stride]; + for (x = 0; x < w; x++) { + int r = row[0], g = row[1], b = row[2], a = row[3]; + if (a != 0) { + row[0] = (unsigned char)(r*255/a); + row[1] = (unsigned char)(g*255/a); + row[2] = (unsigned char)(b*255/a); + } + row += 4; + } + } + + // Defringe + for (y = 0; y < h; y++) { + unsigned char *row = &image[y*stride]; + for (x = 0; x < w; x++) { + int r = 0, g = 0, b = 0, a = row[3], n = 0; + if (a == 0) { + if (x-1 > 0 && row[-1] != 0) { + r += row[-4]; + g += row[-3]; + b += row[-2]; + n++; + } + if (x+1 < w && row[7] != 0) { + r += row[4]; + g += row[5]; + b += row[6]; + n++; + } + if (y-1 > 0 && row[-stride+3] != 0) { + r += row[-stride]; + g += row[-stride+1]; + b += row[-stride+2]; + n++; + } + if (y+1 < h && row[stride+3] != 0) { + r += row[stride]; + g += row[stride+1]; + b += row[stride+2]; + n++; + } + if (n > 0) { + row[0] = (unsigned char)(r/n); + row[1] = (unsigned char)(g/n); + row[2] = (unsigned char)(b/n); + } + } + row += 4; + } + } +} + + +static void nsvg__initPaint(NSVGcachedPaint* cache, NSVGpaint* paint, float opacity) +{ + int i, j; + NSVGgradient* grad; + + cache->type = paint->type; + + if (paint->type == NSVG_PAINT_COLOR) { + cache->colors[0] = nsvg__applyOpacity(paint->color, opacity); + return; + } + + grad = paint->gradient; + + cache->spread = grad->spread; + memcpy(cache->xform, grad->xform, sizeof(float)*6); + + if (grad->nstops == 0) { + for (i = 0; i < 256; i++) + cache->colors[i] = 0; + } else if (grad->nstops == 1) { + unsigned int color = nsvg__applyOpacity(grad->stops[0].color, opacity); + for (i = 0; i < 256; i++) + cache->colors[i] = color; + } else { + unsigned int ca, cb = 0; + float ua, ub, du, u; + int ia, ib, count; + + ca = nsvg__applyOpacity(grad->stops[0].color, opacity); + ua = nsvg__clampf(grad->stops[0].offset, 0, 1); + ub = nsvg__clampf(grad->stops[grad->nstops-1].offset, ua, 1); + ia = (int)(ua * 255.0f); + ib = (int)(ub * 255.0f); + for (i = 0; i < ia; i++) { + cache->colors[i] = ca; + } + + for (i = 0; i < grad->nstops-1; i++) { + ca = nsvg__applyOpacity(grad->stops[i].color, opacity); + cb = nsvg__applyOpacity(grad->stops[i+1].color, opacity); + ua = nsvg__clampf(grad->stops[i].offset, 0, 1); + ub = nsvg__clampf(grad->stops[i+1].offset, 0, 1); + ia = (int)(ua * 255.0f); + ib = (int)(ub * 255.0f); + count = ib - ia; + if (count <= 0) continue; + u = 0; + du = 1.0f / (float)count; + for (j = 0; j < count; j++) { + cache->colors[ia+j] = nsvg__lerpRGBA(ca,cb,u); + u += du; + } + } + + for (i = ib; i < 256; i++) + cache->colors[i] = cb; + } + +} + +/* +static void dumpEdges(NSVGrasterizer* r, const char* name) +{ + float xmin = 0, xmax = 0, ymin = 0, ymax = 0; + NSVGedge *e = NULL; + int i; + if (r->nedges == 0) return; + FILE* fp = fopen(name, "w"); + if (fp == NULL) return; + + xmin = xmax = r->edges[0].x0; + ymin = ymax = r->edges[0].y0; + for (i = 0; i < r->nedges; i++) { + e = &r->edges[i]; + xmin = nsvg__minf(xmin, e->x0); + xmin = nsvg__minf(xmin, e->x1); + xmax = nsvg__maxf(xmax, e->x0); + xmax = nsvg__maxf(xmax, e->x1); + ymin = nsvg__minf(ymin, e->y0); + ymin = nsvg__minf(ymin, e->y1); + ymax = nsvg__maxf(ymax, e->y0); + ymax = nsvg__maxf(ymax, e->y1); + } + + fprintf(fp, "", xmin, ymin, (xmax - xmin), (ymax - ymin)); + + for (i = 0; i < r->nedges; i++) { + e = &r->edges[i]; + fprintf(fp ,"", e->x0,e->y0, e->x1,e->y1); + } + + for (i = 0; i < r->npoints; i++) { + if (i+1 < r->npoints) + fprintf(fp ,"", r->points[i].x, r->points[i].y, r->points[i+1].x, r->points[i+1].y); + fprintf(fp ,"", r->points[i].x, r->points[i].y, r->points[i].flags == 0 ? "#f00" : "#0f0"); + } + + fprintf(fp, ""); + fclose(fp); +} +*/ + +void nsvgRasterize(NSVGrasterizer* r, + NSVGimage* image, float tx, float ty, float scale, + unsigned char* dst, int w, int h, int stride) +{ + NSVGshape *shape = NULL; + NSVGedge *e = NULL; + NSVGcachedPaint cache; + int i; + int j; + unsigned char paintOrder; + + r->bitmap = dst; + r->width = w; + r->height = h; + r->stride = stride; + + if (w > r->cscanline) { + r->cscanline = w; + r->scanline = (unsigned char*)realloc(r->scanline, w); + if (r->scanline == NULL) return; + } + + for (i = 0; i < h; i++) + memset(&dst[i*stride], 0, w*4); + + for (shape = image->shapes; shape != NULL; shape = shape->next) { + if (!(shape->flags & NSVG_FLAGS_VISIBLE)) + continue; + + for (j = 0; j < 3; j++) { + paintOrder = (shape->paintOrder >> (2 * j)) & 0x03; + + if (paintOrder == NSVG_PAINT_FILL && shape->fill.type != NSVG_PAINT_NONE) { + nsvg__resetPool(r); + r->freelist = NULL; + r->nedges = 0; + + nsvg__flattenShape(r, shape, scale); + + // Scale and translate edges + for (i = 0; i < r->nedges; i++) { + e = &r->edges[i]; + e->x0 = tx + e->x0; + e->y0 = (ty + e->y0) * NSVG__SUBSAMPLES; + e->x1 = tx + e->x1; + e->y1 = (ty + e->y1) * NSVG__SUBSAMPLES; + } + + // Rasterize edges + if (r->nedges != 0) + qsort(r->edges, r->nedges, sizeof(NSVGedge), nsvg__cmpEdge); + + // now, traverse the scanlines and find the intersections on each scanline, use non-zero rule + nsvg__initPaint(&cache, &shape->fill, shape->opacity); + + nsvg__rasterizeSortedEdges(r, tx,ty,scale, &cache, shape->fillRule); + } + if (paintOrder == NSVG_PAINT_STROKE && shape->stroke.type != NSVG_PAINT_NONE && (shape->strokeWidth * scale) > 0.01f) { + nsvg__resetPool(r); + r->freelist = NULL; + r->nedges = 0; + + nsvg__flattenShapeStroke(r, shape, scale); + + // dumpEdges(r, "edge.svg"); + + // Scale and translate edges + for (i = 0; i < r->nedges; i++) { + e = &r->edges[i]; + e->x0 = tx + e->x0; + e->y0 = (ty + e->y0) * NSVG__SUBSAMPLES; + e->x1 = tx + e->x1; + e->y1 = (ty + e->y1) * NSVG__SUBSAMPLES; + } + + // Rasterize edges + if (r->nedges != 0) + qsort(r->edges, r->nedges, sizeof(NSVGedge), nsvg__cmpEdge); + + // now, traverse the scanlines and find the intersections on each scanline, use non-zero rule + nsvg__initPaint(&cache, &shape->stroke, shape->opacity); + + nsvg__rasterizeSortedEdges(r, tx,ty,scale, &cache, NSVG_FILLRULE_NONZERO); + } + } + } + + nsvg__unpremultiplyAlpha(dst, w, h, stride); + + r->bitmap = NULL; + r->width = 0; + r->height = 0; + r->stride = 0; +} + +#endif // NANOSVGRAST_IMPLEMENTATION + +#endif // NANOSVGRAST_H