image_energy 0.1.0

Library crate for calculating energy of an image
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

use image::DynamicImage;
use image::io::Reader as ImageReader;

use crate::energy_of_image::energy::get_energy_image;

pub struct ImageWrapper {
    original: Option<DynamicImage>,
}

impl ImageWrapper {
    pub fn from_file(path: &str) -> Self {
        let result = ImageReader::open(path);
        match result {
            Ok(reader) => {
                let image_result = reader.decode();
                match image_result {
                    Ok(image) => {
                        ImageWrapper {
                            original: Some(image)
                        }
                    }
                    Err(e) => {
                        panic!("{}", e)
                    }
                }
            }
            Err(e) => {
                panic!("{}", e)
            }
        }
    }

    pub fn get_energy(&self) -> DynamicImage {
        match &self.original {
            Some(image) => {
                get_energy_image(image)
            }
            None => {
                panic!("No image was loaded to calculate energy")
            }
        }
    }
}