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

use std::cell::UnsafeCell;

use rayon::prelude::*;

use crate::core::traits::{GenericImageView, Pixel, TryConvert, TryConvertSlice};
use crate::packed::generic::{ImageView, ImageViewMut, ImageBuffer};

// This is a private helper struct to share buffers between threads in a lock free manner where we
// would usually need a Mutex. Only use this when you can ensure that all usage of the wrapped
// value is safe and never interleaved!
struct UnsafeShared<T: ?Sized> {
    value: UnsafeCell<T>,
}

impl<T> UnsafeShared<T> {
    pub fn new(t: T) -> UnsafeShared<T> {
        UnsafeShared {
            value: UnsafeCell::new(t),
        }
    }

    #[allow(clippy::mut_from_ref)]
    pub fn get(&self) -> &mut T {
        unsafe { &mut *self.value.get() }
    }
}

unsafe impl<T: ?Sized + Send> Send for UnsafeShared<T> {}
unsafe impl<T: ?Sized + Send> Sync for UnsafeShared<T> {}

macro_rules! impl_TryConvert {
    () => {
        type Error = ();

        fn try_convert(&self, output: &mut ImageBuffer<DP>) -> Result<(), Self::Error> {
            if output.width() < self.width() || output.height() < self.height() {
                *output = ImageBuffer::new(self.width(), self.height());
            }

            // It is safe to use the shared, lock free wrapper here because each thread
            // accesses a distinct pixel row, so pixel access is never interleaved.
            let output = UnsafeShared::new(output);

            (0..self.height()).into_par_iter().for_each(|i| {
                let output = output.get();
                let row_in = &self[i as usize];
                let row_out = &mut output[i as usize];
                // TODO: marshal error
                SP::try_convert(row_in, row_out).unwrap();
            });

            Ok(())
        }
    };
}

macro_rules! impl_TryConvertFlat {
    () => {
        type Error = ();

        fn try_convert(&self, output: &mut ImageViewMut<'b, DP>) -> Result<(), Self::Error> {
            if output.width() < self.width() || output.height() < self.height() {
                return Err(());
            }

            // It is safe to use the shared, lock free wrapper here because each thread
            // accesses a distinct pixel row, so pixel access is never interleaved.
            let output = UnsafeShared::new(output);

            (0..self.height()).into_par_iter().for_each(|i| {
                let output = output.get();
                let row_in = &self[i as usize];
                let row_out = &mut output[i as usize];
                // TODO: marshal error
                SP::try_convert(row_in, row_out).unwrap();
            });

            Ok(())
        }
    };
}

impl<'a, SP, DP> TryConvert<ImageBuffer<DP>> for ImageView<'a, SP>
where
    SP: Pixel,
    DP: Pixel,
    SP: TryConvertSlice<DP>,
{
    impl_TryConvert!();
}

impl<'a, SP, DP> TryConvert<ImageBuffer<DP>> for ImageViewMut<'a, SP>
where
    SP: Pixel,
    DP: Pixel,
    SP: TryConvertSlice<DP>,
{
    impl_TryConvert!();
}

impl<SP, DP> TryConvert<ImageBuffer<DP>> for ImageBuffer<SP>
where
    SP: Pixel,
    DP: Pixel,
    SP: TryConvertSlice<DP>,
{
    impl_TryConvert!();
}

impl<'a, 'b, SP, DP> TryConvert<ImageViewMut<'b, DP>> for ImageView<'a, SP>
where
    SP: Pixel,
    DP: Pixel,
    SP: TryConvertSlice<DP>,
{
    impl_TryConvertFlat!();
}

impl<'a, 'b, SP, DP> TryConvert<ImageViewMut<'b, DP>> for ImageViewMut<'a, SP>
where
    SP: Pixel,
    DP: Pixel,
    SP: TryConvertSlice<DP>,
{
    impl_TryConvertFlat!();
}

impl<'b, SP, DP> TryConvert<ImageViewMut<'b, DP>> for ImageBuffer<SP>
where
    SP: Pixel,
    DP: Pixel,
    SP: TryConvertSlice<DP>,
{
    impl_TryConvertFlat!();
}