如何使用Rust的image=0.23.14库实现多张图片叠加及批量拼接

首先明确两种常见需求的差异：你要的是叠加（多张图在同一坐标区域重叠，上层图覆盖下层对应位置像素）还是拼接（多张图按水平/垂直方向依次排列不重叠）？两种场景的调整逻辑分别如下：

1. 需求为「多张图片叠加（图层堆叠）」的实现 #

首先确定底图画布的尺寸，一般取所有叠加图中最大的宽高作为画布尺寸，也可以自定义固定尺寸。调整后的代码如下：

use image::{
  GenericImage, 
  GenericImageView, 
  ImageBuffer, 
  Pixel, 
  Primitive
};
use anyhow::Result;

/// 多张图片叠加，后传入的图片会覆盖在前一张的相同坐标位置
fn overlay_images<I, P, S>(images: &[I], offset_x: u32, offset_y: u32) -> ImageBuffer<P, Vec<S>>
  where
   I: GenericImageView<Pixel = P>,
   P: Pixel<Subpixel = S> + 'static,
   S: Primitive + 'static {
    // 计算画布尺寸，取所有图片的最大宽高，可替换为自定义固定值
    let max_width = images.iter().map(|img| img.width()).max().unwrap_or(0);
    let max_height = images.iter().map(|img| img.height()).max().unwrap_or(0);
    let mut imgbuf = image::ImageBuffer::new(max_width, max_height);
   
    // 依次叠加所有图片，每张图的左上角坐标可通过offset自定义
    for img in images {
      // 如果要每张图设置不同偏移，可单独传入偏移量数组，此处简化为统一偏移
      imgbuf.copy_from(img, offset_x, offset_y).unwrap();
    }
    imgbuf
}

fn main() -> Result<()> {
  // 支持任意数量的图片叠加，直接在数组里新增路径即可
  let imgs: Vec<_> = [
    "images/img1.png",
    "images/img2.png",
    "images/img3.png",
    "images/img4.png"
  ].iter().map(|path| image::open(path).unwrap()).collect();
  
  // 所有图从(0,0)位置开始叠加，上层覆盖下层
  overlay_images(&imgs, 0, 0).save("overlay_result.png")?;
  
  Ok(())
}

如果需要支持半透明图的混合叠加（不是完全覆盖下层像素），把copy_from替换为逐像素计算alpha混合的逻辑即可。

2. 需求为「多张图片水平/垂直拼接」的实现 #

你贴的原始代码存在逻辑错误：固定了100x100的画布尺寸，也没有计算每张图的偏移位置，正确的多图拼接逻辑如下：

use image::{
  GenericImage, 
  GenericImageView, 
  ImageBuffer, 
  Pixel, 
  Primitive
};
use anyhow::Result;

/// 多张图片水平拼接，高度取所有图片的最大高度
fn h_concat<I, P, S>(images: &[I]) -> ImageBuffer<P, Vec<S>>
  where
   I: GenericImageView<Pixel = P>,
   P: Pixel<Subpixel = S> + 'static,
   S: Primitive + 'static {
    // 总宽度是所有图片宽度之和，高度取最大值
    let total_width: u32 = images.iter().map(|img| img.width()).sum();
    let max_height = images.iter().map(|img| img.height()).max().unwrap_or(0);
    let mut imgbuf = ImageBuffer::new(total_width, max_height);
   
    let mut current_x = 0;
    for img in images {
      imgbuf.copy_from(img, current_x, 0).unwrap();
      current_x += img.width();
    }
    imgbuf
}

/// 多张图片垂直拼接，宽度取所有图片的最大宽度
fn v_concat<I, P, S>(images: &[I]) -> ImageBuffer<P, Vec<S>>
  where
   I: GenericImageView<Pixel = P>,
   P: Pixel<Subpixel = S> + 'static,
   S: Primitive + 'static {
    let total_height: u32 = images.iter().map(|img| img.height()).sum();
    let max_width = images.iter().map(|img| img.width()).max().unwrap_or(0);
    let mut imgbuf = ImageBuffer::new(max_width, total_height);
   
    let mut current_y = 0;
    for img in images {
      imgbuf.copy_from(img, 0, current_y).unwrap();
      current_y += img.height();
    }
    imgbuf
}

fn main() -> Result<()> {
  let imgs: Vec<_> = [
    "images/img1.png",
    "images/img2.png",
    "images/img3.png"
  ].iter().map(|path| image::open(path).unwrap()).collect();
  
  h_concat(&imgs).save("h_concat_result.png")?;
  v_concat(&imgs).save("v_concat_result.png")?;
  
  Ok(())
}

两种场景都支持任意数量的图片传入，只需要把要处理的图片路径添加到数组即可。注意如果传入的图片像素格式不一致，需要先统一转换为相同格式再处理，避免编译时报类型不匹配错误。

内容的提问来源于stack exchange，提问作者IntFooBar