如何修改Rust图像像素右移函数以支持左移操作？

问题描述 #

现有一段使用Rust image库实现像素沿X轴右移的代码，仅支持右移操作，希望修改使其同时支持左移：

原代码：

use image::{DynamicImage, GenericImageView, ImageBuffer};

fn move_pixels_right(img: &DynamicImage, shift: u32) -> DynamicImage {
    let (width, height) = img.dimensions();
    let mut new_img = ImageBuffer::new(width, height);

    for y in 0..height {
        for x in shift..width {
            let pixel = img.get_pixel(x - shift, y);
            new_img.put_pixel(x, y, pixel);
        }
    }

    DynamicImage::ImageRgba8(new_img)
}

fn main() {
    let img = image::open("center.png").unwrap();
    let moved_img = move_pixels_right(&img, 1);

    moved_img.save("moved.png").unwrap();
}

输入图像：中心带有非透明像素的图像
输出图像：非透明像素沿X轴右移1位后的图像

修改方案 #

可以将函数重命名为更通用的move_pixels_x，通过正负偏移量区分左右：正数表示右移，负数表示左移。同时增加边界检查避免越界访问，超出边界的位置填充透明像素。

修改后的完整代码：

use image::{DynamicImage, GenericImageView, ImageBuffer, Rgba};

fn move_pixels_x(img: &DynamicImage, shift: i32) -> DynamicImage {
    let (width, height) = img.dimensions();
    let width_i32 = width as i32;
    let mut new_img = ImageBuffer::new(width, height);

    for y in 0..height {
        for x in 0..width {
            let x_i32 = x as i32;
            // 计算原图像中对应像素的位置
            let src_x = x_i32 - shift;
            // 检查坐标是否在有效范围内
            if src_x >= 0 && src_x < width_i32 {
                let pixel = img.get_pixel(src_x as u32, y);
                new_img.put_pixel(x, y, pixel);
            } else {
                // 超出边界的位置填充透明像素
                new_img.put_pixel(x, y, Rgba([0, 0, 0, 0]));
            }
        }
    }

    DynamicImage::ImageRgba8(new_img)
}

fn main() {
    let img = image::open("center.png").unwrap();
    
    // 右移1位
    let moved_right = move_pixels_x(&img, 1);
    moved_right.save("moved_right.png").unwrap();
    
    // 左移1位
    let moved_left = move_pixels_x(&img, -1);
    moved_left.save("moved_left.png").unwrap();
}

关键修改点 #

• 偏移量参数改为i32类型：正数代表右移，负数代表左移
• 增加边界检查：确保原图像的像素坐标在有效范围内，防止越界访问
• 超出边界的位置填充透明像素，保持图像背景一致性
• 函数名改为move_pixels_x，明确功能为沿X轴平移

逻辑说明 #

• 右移时，shift为正：src_x = x - shift，从原图像的x-shift位置取像素放到新图像的x位置，与原右移逻辑一致
• 左移时，shift为负：src_x = x - (-n) = x + n，从原图像的x+n位置取像素放到新图像的x位置，实现左移效果

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