62 lines
2.1 KiB
JavaScript
62 lines
2.1 KiB
JavaScript
/**
|
||
* 反向 Alpha 混合模块
|
||
* 实现去除水印的核心算法
|
||
*/
|
||
|
||
// 常量定义
|
||
const ALPHA_THRESHOLD = 0.002; // 忽略极小的 alpha 值（噪声）
|
||
const MAX_ALPHA = 0.99; // 避免除以接近零的值
|
||
const LOGO_VALUE = 255; // 白色水印的颜色值
|
||
|
||
/**
|
||
* 使用反向 alpha 混合移除水印
|
||
*
|
||
* 原理：
|
||
* Gemini 添加水印: watermarked = α × logo + (1 - α) × original
|
||
* 反向求解: original = (watermarked - α × logo) / (1 - α)
|
||
*
|
||
* @param {ImageData} imageData - 要处理的图像数据（会被原地修改）
|
||
* @param {Float32Array} alphaMap - Alpha 通道数据
|
||
* @param {Object} position - 水印位置 {x, y, width, height}
|
||
*/
|
||
export function removeWatermark(imageData, alphaMap, position) {
|
||
const { x, y, width, height } = position;
|
||
|
||
// 遍历水印区域的每个像素
|
||
for (let row = 0; row < height; row++) {
|
||
for (let col = 0; col < width; col++) {
|
||
// 计算在原图中的索引（RGBA 格式，每个像素 4 个字节）
|
||
const imgIdx = ((y + row) * imageData.width + (x + col)) * 4;
|
||
|
||
// 计算在 alpha map 中的索引
|
||
const alphaIdx = row * width + col;
|
||
|
||
// 获取 alpha 值
|
||
let alpha = alphaMap[alphaIdx];
|
||
|
||
// 跳过极小的 alpha 值（噪声）
|
||
if (alpha < ALPHA_THRESHOLD) {
|
||
continue;
|
||
}
|
||
|
||
// 限制 alpha 值，避免除零
|
||
alpha = Math.min(alpha, MAX_ALPHA);
|
||
const oneMinusAlpha = 1.0 - alpha;
|
||
|
||
// 对 RGB 三个通道应用反向 alpha 混合公式
|
||
for (let c = 0; c < 3; c++) {
|
||
const watermarked = imageData.data[imgIdx + c];
|
||
|
||
// 反向 alpha 混合公式
|
||
const original = (watermarked - alpha * LOGO_VALUE) / oneMinusAlpha;
|
||
|
||
// 裁剪到 [0, 255] 范围
|
||
imageData.data[imgIdx + c] = Math.max(0, Math.min(255, Math.round(original)));
|
||
}
|
||
|
||
// Alpha 通道保持不变
|
||
// imageData.data[imgIdx + 3] 不需要修改
|
||
}
|
||
}
|
||
}
|