簡體   English   中英

在泛型的 trait 實現中轉換為 f32

[英]Convert to f32 in trait implementation of generic

為了學習 rust,我嘗試實現一個通用圖像類/對象。

我對 Image::save(...) function 和 RgbaGeneric 的 Pixel::gray() 的實現有疑問。

編譯器說我必須使用 From 或 Into 特征。 但我不知道語法看起來如何。 所有讓它工作的嘗試都失敗了。

mod image {
    use std::fmt;

    use std::path::Path;
    use std::io::Write;
    use std::fs::File;

    #[derive(Clone, Debug)]
    pub struct RgbaGeneric<T> {
        r: T, g: T, b: T, a: T,
    }

    trait Pixel<T> {
        fn red(&self) -> T;
        fn green(&self) -> T;
        fn blue(&self) -> T;
        fn alpha(&self) -> T;
        fn gray(&self) -> T;
    }

    impl<T> Pixel<T> for RgbaGeneric<T> {
        fn red(&self) -> T { return self.r;}
        fn green(&self) -> T { return self.g; }
        fn blue(&self) -> T { return self.b; }
        fn alpha(&self) -> T { return self.a; }
        fn gray(&self) -> T {
           return (self.r as f32 * 0.21 +
                   self.g as f32 * 0.72 +
                   self.b  as f32 * 0.07) as T;
        }
    }

    impl<T> RgbaGeneric<T> {
        pub fn new(r: T, g: T, b: T, a: T) -> RgbaGeneric<T> {
            RgbaGeneric { r, g, b, a }
        }
    }

    pub type RgbaU8 = RgbaGeneric<u8>;

    pub struct Image<T> {
        width: usize, height: usize, data: Vec<T>,
    }

    pub type ImageRgbaU8 = Image<RgbaU8>;

    impl<T> Image<T>
    where
        T: Clone + Pixel<T>
    {
        pub fn new(width: usize, height: usize, val: T) -> Image<T> {
            Image {
                width,
                height,
                data: vec![val; width * height],
            }
        }

        pub fn width(&self) -> usize { return self.width; }
        pub fn height(&self) -> usize { return self.height; }
        pub fn pixel_count(&self) -> usize {
            return self.width * self.height;
        }

        pub fn pixels(&self) -> &[T] { return self.data.as_slice(); }

        pub fn pixel(&self, x: usize, y: usize) -> &T {
            return &self.data[self.index(x, y)];
        }

        pub fn set_pixel(&mut self, x: usize, y: usize, pixel: T) {
            let idx = self.index(x, y);
            self.data[idx] = pixel;
        }

        pub fn save(&self, filename: &str) -> std::io::Result<()> {
            let path = Path::new(filename);
            let mut file = try!(File::create(&path));
            let header = format!("P6 {} {} 255\n", self.width, self.height);
            try!(file.write(header.as_bytes()));
            for y in 0..self.height() {
                for x in 0..self.width() {
                    let pix = self.pixel(x,y);
                    try!(file.write(pix.red()));
                    try!(file.write(pix.green()));
                    try!(file.write(pix.blue()));
                }
            }
            return Ok(());
        }

        fn index(&self, x: usize, y: usize) -> usize {
            return y * &self.width + x;
        }
    }
}

gray方法的實現應該是這樣的:

impl<T> Pixel<T> for RgbaGeneric<T> where
        f32: std::convert::From<T>,
        T: std::convert::From<f32>{
    fn red(&self) -> T { return self.r;}
    fn green(&self) -> T { return self.g; }
    fn blue(&self) -> T { return self.b; }
    fn alpha(&self) -> T { return self.a; }
    fn gray(&self) -> T {
       return (Into::<f32>::into(self.r) * 0.21 +
               Into::<f32>::into(self.g) * 0.72 +
               Into::<f32>::into(self.b) * 0.07).into();
    }
}

在塊的 header 中,我們說我們的T可以轉換為f32類型,然后我們使用into()方法,因為as表達式只能用於在原始類型之間轉換(編譯器暗示T可以是任何類型)。

暫無
暫無

聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.

 
粵ICP備18138465號  © 2020-2024 STACKOOM.COM