///|
/// 随机点阵水印:按 `DotConfig` 的归一化网格 + PRNG 布局,在全图散布深色点。
///
/// - 落点网格使用**归一化坐标**(单元边长 = 图宽 8%),点位置按比例跟随
/// 图像尺寸——因此缩放(resize)后提取仍能命中,鲁棒性矩阵实测通过。
/// - 每个单元由 `seed` 驱动的确定性 PRNG 决定是否落点(`density` 为落点
/// 单元概率)及点在单元内的偏移比例(叠加 `grid_offset_x / grid_offset_y`
/// 像素偏移,非零偏移在缩放场景会有轻微漂移,建议默认 0)。
/// - 点颜色固定为深色(`0x0000_00FF`),半透明时由 `opacity` 控制;
/// 同一配置(尤其同一 `seed`)嵌入结果完全一致,可被 `extract_dots` 复现。
pub fn RgbaImage::embed_dots(
self : RgbaImage,
cfg : DotConfig,
) -> Result[RgbaImage, String] {
let alpha = clamp_opacity(cfg.opacity)
if alpha == 0 {
return Ok(self)
}
for (px, py) in dot_points(cfg, self.width, self.height) {
for dy in 0..= 0 && y >= 0 && x < self.width && y < self.height {
let dst = self.get_pixel(x, y)
self.set_pixel(x, y, blend_pixel(dst, 0x0000_00FF, alpha))
}
}
}
}
Ok(self)
}
///|
/// 提取点阵水印:用同一 `cfg` 复现点位置,统计深色命中率(0.0~1.0)。
///
/// 命中判定:像素亮度(RGB 均值)低于 `brightness_threshold` 视为命中。
/// **注意**:这是绝对命中率,不含基线对照——暗色调原图本身可能天然
/// 高命中(假阳性)。需要严谨判别的场景请用 `verify_dots` / `dot_contrast`
/// (成对对照,自动扣除原图暗区干扰)。
pub fn extract_dots(
img : RgbaImage,
cfg : DotConfig,
brightness_threshold? : Int32,
) -> Double {
let thr = match brightness_threshold {
Some(v) => v
None => 128
}
let (hits, total) = count_hits(img, dot_points(cfg, img.width, img.height), cfg.dot_size, thr)
if total == 0 {
0.0
} else {
hits.to_double() / total.to_double()
}
}
///|
/// 成对对照判定:`目标点位命中率 − 对照点位命中率`。
///
/// - **目标点位**:`cfg.seed` 派生的落点(即嵌入者实际画点处);
/// - **对照点位**:由 `cfg.seed` 确定性派生的另一组同密度随机落点(未画点处)。
///
/// 两者在同一张图上用同一亮度阈值统计,原图的暗区/纹理等系统性偏差
/// 同时作用于两侧,差值即"水印带来的真实信号":
///
/// - 未嵌入图(含暗色原图):目标命中率 ≈ 对照命中率 → contrast ≈ 0;
/// - 已嵌入图:目标点位与真水印重合(全暗)→ contrast 显著 > 0。
///
/// 这解决了绝对命中率的两个缺陷:暗色原图假阳性(原图干扰)与
/// 邻近 seed 同高命中(候选歧义)。
pub fn dot_contrast(
img : RgbaImage,
cfg : DotConfig,
brightness_threshold? : Int32,
) -> Double {
let thr = match brightness_threshold {
Some(v) => v
None => 128
}
let (th, tt) = count_hits(img, dot_points(cfg, img.width, img.height), cfg.dot_size, thr)
let (rh, rt) = count_hits(
img,
dot_points_with_seed(cfg, img.width, img.height, ref_seed(cfg.seed)),
cfg.dot_size,
thr,
)
let target = if tt == 0 { 0.0 } else { th.to_double() / tt.to_double() }
let refr = if rt == 0 { 0.0 } else { rh.to_double() / rt.to_double() }
target - refr
}
///|
/// 统计一组点位的深色命中(embed 与 extract/contrast 共用,两侧公式一致)。
///
/// 邻域容差兜住 2 倍缩放错位(dot_size=3 时 4×4);与 embed 对齐:
/// embed 对完全越界的格不画点,提取时也跳过(避免分母虚增);
/// 部分越界的格 embed 已画界内部分,提取用邻域"任一像素在界内"即计入,两侧一致。
fn count_hits(
img : RgbaImage,
pts : Array[(Int32, Int32)],
dot_size : Int32,
thr : Int32,
) -> (Int32, Int32) {
let mut hits = 0
let mut total = 0
for (px, py) in pts {
let size = dot_size + 1
let mut any = false
let mut checked = false
for dy in 0..= 0 && y >= 0 && x < img.width && y < img.height {
checked = true
let p = img.get_pixel(x, y)
let r2 = (p >> 24) & 0xFF
let g2 = (p >> 16) & 0xFF
let b2 = (p >> 8) & 0xFF
if (r2 + g2 + b2) / 3 < thr {
any = true
}
}
}
}
if checked {
total += 1
if any {
hits += 1
}
}
}
(hits, total)
}
///|
/// 对照 seed:由目标 seed 确定性派生(黄金比例常数异或),保证同密度、
/// 同规模、位置与目标点位不相干;embed 不会使用此 seed,因此对照点位
/// 恒为"未画点处"。
fn ref_seed(seed : Int64) -> Int64 {
seed ^ 0x9E37_79B9_7F4A_7C15L
}
///|
/// 按 `cfg` 计算全部落点坐标(embed 与 extract 共用,保证两侧公式一致)。
///
/// - 归一化网格 + PRNG 序列在此统一推进;任何一侧改动落点公式,另一侧
/// 自动跟随,避免"改一处漏一处"导致的提取失配。
/// - 坐标已做偶数像素对齐(2 的幂次缩放下奇偶网格重合);`grid_offset`
/// 叠加在最终坐标上(可为负,越界点由调用侧按边界裁剪)。
fn dot_points(cfg : DotConfig, w : Int32, h : Int32) -> Array[(Int32, Int32)] {
dot_points_with_seed(cfg, w, h, cfg.seed)
}
///|
/// 按指定 `seed` 计算全部落点坐标(embed/extract 用 `cfg.seed`,
/// 对照判定用 `ref_seed(cfg.seed)`——两侧共用同一落点公式,保证点
/// 规模/密度一致,仅位置由 seed 决定)。
fn dot_points_with_seed(
cfg : DotConfig,
w : Int32,
h : Int32,
seed : Int64,
) -> Array[(Int32, Int32)] {
let cell_ratio = 0.08 // 单元边长占图宽的比例(≈ 原 8px @100px 的观感)
// density clamp 到 [0,1]:<0 无点、>1 全图点;embed/extract 共用本函数故两侧一致
let mut threshold = (cfg.density * 1000.0).to_int()
if threshold < 0 {
threshold = 0
} else if threshold > 1000 {
threshold = 1000
}
let cols = (1.0 / cell_ratio).ceil().to_int()
let mut st = seed_to_state(seed)
let pts = Array::make(0, (0, 0))
for cy in 0..