bendn/fimg
1
0
Fork 0
mirror of https://github.com/bend-n/fimg.git synced 2024-12-22 10:28:21 -06:00
Code Issues Actions Packages Projects Releases Wiki Activity

generic

Browse source
This commit is contained in:
bendn 2023-11-02 09:15:05 +07:00
parent 997da104d2
commit 16fbab6ca2
No known key found for this signature in database
GPG key ID: 0D9D3A2A3B2A93D6
2 changed files with 16 additions and 34 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

40
src/overlay.rs
View file

@ -129,20 +129,16 @@ impl<T: AsMut<[u8]> + AsRef<[u8]>, U: AsRef<[u8]>> Overlay<Image<U, 4>> for Imag
} }
} }
impl BlendingOverlay<Image<&[u8], 4>> for Image<&mut [u8], 4> { impl<const A: usize, const B: usize, T: AsMut<[u8]> + AsRef<[u8]>, U: AsRef<[u8]>>
BlendingOverlay<Image<U, B>> for Image<T, A>
where
[u8; A]: Blend<B>,
{
#[inline] #[inline]
unsafe fn overlay_blended(&mut self, with: &Image<&[u8], 4>) -> &mut Self { unsafe fn overlay_blended(&mut self, with: &Image<U, B>) -> &mut Self {
debug_assert!(self.width() == with.width()); debug_assert!(self.width() == with.width());
debug_assert!(self.height() == with.height()); debug_assert!(self.height() == with.height());
for (i, other_pixels) in with.chunked().enumerate() { for (other_pixels, own_pixels) in with.chunked().zip(self.chunked_mut()) {
// SAFETY: caller assures us, all is well.
let own_pixels = unsafe {
&mut *(self
.buffer
.as_mut()
.get_unchecked_mut(i * 4..i * 4 + 4)
.as_mut_ptr() as *mut [u8; 4])
};
own_pixels.blend(*other_pixels); own_pixels.blend(*other_pixels);
} }
self self
@ -259,28 +255,6 @@ impl<T: AsMut<[u8]> + AsRef<[u8]>, U: AsRef<[u8]>> Overlay<Image<U, 4>> for Imag
} }
} }
impl<T: AsMut<[u8]> + AsRef<[u8]>, U: AsRef<[u8]>> BlendingOverlay<Image<U, 4>> for Image<T, 3> {
#[inline]
unsafe fn overlay_blended(&mut self, with: &Image<U, 4>) -> &mut Self {
debug_assert!(self.width() == with.width());
debug_assert!(self.height() == with.height());
for (i, other_pixels) in with.chunked().enumerate() {
// SAFETY: caller assures us, all is well.
let [r, g, b] = unsafe {
&mut *(self
.buffer
.as_mut()
.get_unchecked_mut(i * 3..i * 3 + 3)
.as_mut_ptr() as *mut [u8; 3])
};
let mut us = [*r, *g, *b, 255];
us.blend(*other_pixels);
(*r, *g, *b) = (us[0], us[1], us[2]);
}
self
}
}
impl ClonerOverlay<4, 3> for ImageCloner<'_, 3> { impl ClonerOverlay<4, 3> for ImageCloner<'_, 3> {
#[inline] #[inline]
#[must_use = "function does not modify the original image"] #[must_use = "function does not modify the original image"]

10
src/pixels/blending.rs
View file

@ -1,5 +1,5 @@
//! module for pixel blending ops //! module for pixel blending ops
use super::{unfloat, Floatify, PMap, Trunc, Unfloatify}; use super::{convert::PFrom, unfloat, Floatify, PMap, Trunc, Unfloatify};
use umath::FF32; use umath::FF32;
/// Trait for blending pixels together. /// Trait for blending pixels together.
@ -41,6 +41,14 @@ impl Blend<3> for [u8; 3] {
} }
} }
impl Blend<4> for [u8; 3] {
fn blend(&mut self, with: [u8; 4]) {
let mut us: [u8; 4] = PFrom::pfrom(*self);
us.blend(with);
*self = PFrom::pfrom(us);
}
}
impl Blend<2> for [u8; 2] { impl Blend<2> for [u8; 2] {
fn blend(&mut self, with: [u8; 2]) { fn blend(&mut self, with: [u8; 2]) {
let bg = self.float(); let bg = self.float();