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
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146

#[cfg(feature = "image_0-23")]
mod with_image_0_23 {
    use crate::image;
    use crate::opencv::{core as cv, prelude::*};
    use crate::{common::*, OpenCvElement, TryFromCv, TryIntoCv};
    use std::ops::Deref;

    impl<P, Container> TryFromCv<image::ImageBuffer<P, Container>> for cv::Mat
    where
        P: image::Pixel + 'static,
        P::Subpixel: 'static + OpenCvElement,
        Container: Deref<Target = [P::Subpixel]> + Clone,
    {
        type Error = Error;
        fn try_from_cv(from: image::ImageBuffer<P, Container>) -> Result<Self, Self::Error> {
            (&from).try_into_cv()
        }
    }

    impl<P, Container> TryFromCv<&image::ImageBuffer<P, Container>> for cv::Mat
    where
        P: image::Pixel + 'static,
        P::Subpixel: 'static + OpenCvElement,
        Container: Deref<Target = [P::Subpixel]> + Clone,
    {
        type Error = Error;
        fn try_from_cv(from: &image::ImageBuffer<P, Container>) -> Result<Self, Self::Error> {
            let (width, height) = from.dimensions();
            let cv_type = cv::CV_MAKETYPE(P::Subpixel::DEPTH, P::CHANNEL_COUNT as i32);
            let mat = unsafe {
                cv::Mat::new_rows_cols_with_data(
                    height as i32,
                    width as i32,
                    cv_type,
                    from.as_ptr() as *mut _,
                    cv::Mat_AUTO_STEP,
                )?
                .try_clone()?
            };
            Ok(mat)
        }
    }

    impl TryFromCv<&image::DynamicImage> for cv::Mat {
        type Error = Error;

        fn try_from_cv(from: &image::DynamicImage) -> Result<Self, Self::Error> {
            use image::DynamicImage as D;

            let mat = match from {
                D::ImageLuma8(image) => image.try_into_cv()?,
                D::ImageLumaA8(image) => image.try_into_cv()?,
                D::ImageBgr8(image) => image.try_into_cv()?,
                D::ImageBgra8(image) => image.try_into_cv()?,
                D::ImageRgb8(image) => image.try_into_cv()?,
                D::ImageRgba8(image) => image.try_into_cv()?,
                D::ImageLuma16(image) => image.try_into_cv()?,
                D::ImageLumaA16(image) => image.try_into_cv()?,
                D::ImageRgb16(image) => image.try_into_cv()?,
                D::ImageRgba16(image) => image.try_into_cv()?,
            };
            Ok(mat)
        }
    }

    impl TryFromCv<image::DynamicImage> for cv::Mat {
        type Error = Error;
        fn try_from_cv(from: image::DynamicImage) -> Result<Self, Self::Error> {
            (&from).try_into_cv()
        }
    }
}

#[cfg(feature = "image_0-24")]
mod with_image_0_24 {
    use crate::image;
    use crate::opencv::{core as cv, prelude::*};
    use crate::{common::*, OpenCvElement, TryFromCv, TryIntoCv};
    use std::ops::Deref;

    impl<P, Container> TryFromCv<image::ImageBuffer<P, Container>> for cv::Mat
    where
        P: image::Pixel + 'static,
        P::Subpixel: 'static + OpenCvElement,
        Container: Deref<Target = [P::Subpixel]> + Clone,
    {
        type Error = Error;
        fn try_from_cv(from: image::ImageBuffer<P, Container>) -> Result<Self, Self::Error> {
            (&from).try_into_cv()
        }
    }

    impl<P, Container> TryFromCv<&image::ImageBuffer<P, Container>> for cv::Mat
    where
        P: image::Pixel + 'static,
        P::Subpixel: 'static + OpenCvElement,
        Container: Deref<Target = [P::Subpixel]> + Clone,
    {
        type Error = Error;
        fn try_from_cv(from: &image::ImageBuffer<P, Container>) -> Result<Self, Self::Error> {
            let (width, height) = from.dimensions();
            let cv_type = cv::CV_MAKETYPE(P::Subpixel::DEPTH, P::CHANNEL_COUNT as i32);
            let mat = unsafe {
                cv::Mat::new_rows_cols_with_data(
                    height as i32,
                    width as i32,
                    cv_type,
                    from.as_ptr() as *mut _,
                    cv::Mat_AUTO_STEP,
                )?
                .try_clone()?
            };
            Ok(mat)
        }
    }

    impl TryFromCv<&image::DynamicImage> for cv::Mat {
        type Error = Error;

        fn try_from_cv(from: &image::DynamicImage) -> Result<Self, Self::Error> {
            use image::DynamicImage as D;

            let mat = match from {
                D::ImageLuma8(image) => image.try_into_cv()?,
                D::ImageLumaA8(image) => image.try_into_cv()?,
                D::ImageRgb8(image) => image.try_into_cv()?,
                D::ImageRgba8(image) => image.try_into_cv()?,
                D::ImageLuma16(image) => image.try_into_cv()?,
                D::ImageLumaA16(image) => image.try_into_cv()?,
                D::ImageRgb16(image) => image.try_into_cv()?,
                D::ImageRgba16(image) => image.try_into_cv()?,
                D::ImageRgb32F(image) => image.try_into_cv()?,
                D::ImageRgba32F(image) => image.try_into_cv()?,
                image => bail!("the color type {:?} is not supported", image.color()),
            };
            Ok(mat)
        }
    }

    impl TryFromCv<image::DynamicImage> for cv::Mat {
        type Error = Error;
        fn try_from_cv(from: image::DynamicImage) -> Result<Self, Self::Error> {
            (&from).try_into_cv()
        }
    }
}