imgddcore 0.1.3

Performance-first perceptual hashing library; perfect for handling large datasets. Designed to quickly process nested folder structures, commonly found in image datasets
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96

#[cfg(test)]
mod tests {
    use anyhow::Result;
    use image::{DynamicImage, Rgba};
    use imgddcore::hashing::ImageHash;

    /// Creates a mock image with alternating pixel values for testing.
    fn create_mock_image(size: (u32, u32)) -> DynamicImage {
        let (width, height) = size;
        DynamicImage::ImageRgba8(image::ImageBuffer::from_fn(width, height, |x, _| {
            if x % 2 == 0 {
                Rgba([255, 0, 0, 255]) // Red pixel
            } else {
                Rgba([0, 0, 0, 255]) // Black pixel
            }
        }))
        .grayscale()
    }

    #[test]
    fn test_ahash() -> Result<()> {
        let test_image = create_mock_image((8, 8));
        let hash = ImageHash::ahash(&test_image)?;
        println!("aHash: {:064b}", hash.get_hash());
        let expected_hash = 0b1010101010101010101010101010101010101010101010101010101010101010;
        assert_eq!(
            hash.get_hash(),
            expected_hash,
            "aHash does not match expected value"
        );

        Ok(())
    }

    #[test]
    fn test_mhash() -> Result<()> {
        let test_image = create_mock_image((8, 8));
        let hash = ImageHash::mhash(&test_image)?;
        println!("mHash: {:064b}", hash.get_hash());
        let expected_hash = 0b1010101010101010101010101010101010101010101010101010101010101010;

        assert_eq!(
            hash.get_hash(),
            expected_hash,
            "mHash does not match expected value"
        );

        Ok(())
    }

    #[test]
    fn test_dhash() -> Result<()> {
        let test_image = create_mock_image((9, 8));
        let hash = ImageHash::dhash(&test_image)?;
        println!("dHash: {:064b}", hash.get_hash());
        let expected_hash = 0b0101010101010101010101010101010101010101010101010101010101010101;
        assert_eq!(
            hash.get_hash(),
            expected_hash,
            "dHash does not match expected value"
        );

        Ok(())
    }

    #[test]
    fn test_phash() -> Result<()> {
        let test_image = create_mock_image((32, 32));
        let hash = ImageHash::phash(&test_image)?;
        let expected_hash = 0b1101010100000000000000000000000000000000000000000000000000000000;
        println!("pHash: {:064b}", hash.get_hash());
        assert_eq!(
            hash.get_hash(),
            expected_hash,
            "pHash does not match expected value"
        );

        Ok(())
    }

    #[test]
    fn test_whash() -> Result<()> {
        let test_image = create_mock_image((8, 8));
        let hash = ImageHash::whash(&test_image)?;
        println!("wHash: {:064b}", hash.get_hash());
        let expected_hash = 0b1010101010101010101010101010101010101010101010101010101010101010;

        assert_eq!(
            hash.get_hash(),
            expected_hash,
            "wHash does not match expected value"
        );

        Ok(())
    }
}