rust image找图

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

fn main() {
    let img_a = image::open("2.png").unwrap().to_rgb8();
    let img_b = image::open("1.png").unwrap().to_rgb8();

    if is_image_a_in_image_b(&img_a, &img_b) {
        println!("Image A is in Image B.");
    } else {
        println!("Image A is not in Image B.");
    }
}

fn is_image_a_in_image_b(img_a: &ImageBuffer<Rgb<u8>, Vec<u8>>, img_b: &ImageBuffer<Rgb<u8>, Vec<u8>>) -> bool {
    let (width_a, height_a) = img_a.dimensions();
    let (width_b, height_b) = img_b.dimensions();

    if width_a > width_b || height_a > height_b {
        // Image A is larger than Image B, cannot be contained in Image B.
        return false;
    }

    for y in 0..height_b - height_a {
        for x in 0..width_b - width_a {
            if is_image_a_at_position_in_image_b(img_a, img_b, x, y) {
                return true;
            }
        }
    }

    false
}

fn is_image_a_at_position_in_image_b(img_a: &ImageBuffer<Rgb<u8>, Vec<u8>>, img_b: &ImageBuffer<Rgb<u8>, Vec<u8>>, x: u32, y: u32) -> bool {
    let (width_a, height_a) = img_a.dimensions();

    for dy in 0..height_a {
        for dx in 0..width_a {
            let pixel_a = img_a.get_pixel(dx, dy);
            let pixel_b = img_b.get_pixel(x + dx, y + dy);

            if pixel_a != pixel_b {
                return false;
            }
        }
    }

    true
}