alex/librashader
1
0
Fork 0
Code Issues Pull requests Packages Projects Releases Wiki Activity

runtime: fix image swizzle for remainder

Browse source
This commit is contained in:
chyyran 2024-08-21 20:11:22 -04:00 committed by Ronny Chan
parent af05cc5bd8
commit b7fd3bc507
1 changed files with 8 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
librashader-runtime/src/image.rs
View file

@ -92,6 +92,8 @@ impl<P: PixelFormat> Image<P> {
} }
} }
// load-bearing #[inline(always)], without it llvm will not vectorize.
#[inline(always)]
fn swizzle_pixels(pixels: &mut Vec<u8>, swizzle: &'static [usize; 32]) { fn swizzle_pixels(pixels: &mut Vec<u8>, swizzle: &'static [usize; 32]) {
assert!(pixels.len() % 4 == 0); assert!(pixels.len() % 4 == 0);
let mut chunks = pixels.chunks_exact_mut(32); let mut chunks = pixels.chunks_exact_mut(32);
@ -104,7 +106,12 @@ fn swizzle_pixels(pixels: &mut Vec<u8>, swizzle: &'static [usize; 32]) {
let remainder = chunks.into_remainder(); let remainder = chunks.into_remainder();
for chunk in remainder.chunks_exact_mut(4) { for chunk in remainder.chunks_exact_mut(4) {
let argb = [chunk[3], chunk[0], chunk[1], chunk[2]]; let argb = [
chunk[swizzle[0]],
chunk[swizzle[1]],
chunk[swizzle[2]],
chunk[swizzle[3]],
];
chunk.copy_from_slice(&argb[..]) chunk.copy_from_slice(&argb[..])
} }
} }