use num_traits::Zero;
use std::fmt;
use std::marker::PhantomData;
use std::ops::{Deref, DerefMut, Index, IndexMut, Range};
use std::path::Path;
use std::slice::{ChunksExact, ChunksExactMut};
use crate::color::{FromColor, Luma, LumaA, Rgb, Rgba, Bgr, Bgra};
use crate::flat::{FlatSamples, SampleLayout};
use crate::dynimage::{save_buffer, save_buffer_with_format};
use crate::error::ImageResult;
use crate::image::{GenericImage, GenericImageView, ImageFormat};
use crate::math::Rect;
use crate::traits::{EncodableLayout, Pixel};
use crate::utils::expand_packed;
pub struct Pixels<'a, P: Pixel + 'a>
where
P::Subpixel: 'a,
{
chunks: ChunksExact<'a, P::Subpixel>,
}
impl<'a, P: Pixel + 'a> Iterator for Pixels<'a, P>
where
P::Subpixel: 'a,
{
type Item = &'a P;
#[inline(always)]
fn next(&mut self) -> Option<&'a P> {
self.chunks.next().map(|v| <P as Pixel>::from_slice(v))
}
}
impl<'a, P: Pixel + 'a> ExactSizeIterator for Pixels<'a, P>
where
P::Subpixel: 'a,
{
fn len(&self) -> usize {
self.chunks.len()
}
}
impl<'a, P: Pixel + 'a> DoubleEndedIterator for Pixels<'a, P>
where
P::Subpixel: 'a,
{
#[inline(always)]
fn next_back(&mut self) -> Option<&'a P> {
self.chunks.next_back().map(|v| <P as Pixel>::from_slice(v))
}
}
impl<P: Pixel> Clone for Pixels<'_, P> {
fn clone(&self) -> Self {
Pixels { chunks: self.chunks.clone() }
}
}
impl<P: Pixel> fmt::Debug for Pixels<'_, P>
where
P::Subpixel: fmt::Debug,
{
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
f
.debug_struct("Pixels")
.field("chunks", &self.chunks)
.finish()
}
}
pub struct PixelsMut<'a, P: Pixel + 'a>
where
P::Subpixel: 'a,
{
chunks: ChunksExactMut<'a, P::Subpixel>,
}
impl<'a, P: Pixel + 'a> Iterator for PixelsMut<'a, P>
where
P::Subpixel: 'a,
{
type Item = &'a mut P;
#[inline(always)]
fn next(&mut self) -> Option<&'a mut P> {
self.chunks.next().map(|v| <P as Pixel>::from_slice_mut(v))
}
}
impl<'a, P: Pixel + 'a> ExactSizeIterator for PixelsMut<'a, P>
where
P::Subpixel: 'a,
{
fn len(&self) -> usize {
self.chunks.len()
}
}
impl<'a, P: Pixel + 'a> DoubleEndedIterator for PixelsMut<'a, P>
where
P::Subpixel: 'a,
{
#[inline(always)]
fn next_back(&mut self) -> Option<&'a mut P> {
self.chunks
.next_back()
.map(|v| <P as Pixel>::from_slice_mut(v))
}
}
impl<P: Pixel> fmt::Debug for PixelsMut<'_, P>
where
P::Subpixel: fmt::Debug,
{
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
f
.debug_struct("PixelsMut")
.field("chunks", &self.chunks)
.finish()
}
}
pub struct Rows<'a, P: Pixel + 'a>
where
<P as Pixel>::Subpixel: 'a,
{
pixels: ChunksExact<'a, P::Subpixel>,
}
impl<'a, P: Pixel + 'a> Rows<'a, P> {
fn with_image(pixels: &'a [P::Subpixel], width: u32, height: u32) -> Self {
let row_len = (width as usize) * usize::from(<P as Pixel>::CHANNEL_COUNT);
if row_len == 0 {
Rows {
pixels: [].chunks_exact(1),
}
} else {
let pixels = pixels.get(..row_len*height as usize)
.expect("Pixel buffer has too few subpixels");
Rows {
pixels: pixels.chunks_exact(row_len),
}
}
}
}
impl<'a, P: Pixel + 'a> Iterator for Rows<'a, P>
where
P::Subpixel: 'a,
{
type Item = Pixels<'a, P>;
#[inline(always)]
fn next(&mut self) -> Option<Pixels<'a, P>> {
let row = self.pixels.next()?;
Some(Pixels {
chunks: row.chunks_exact(<P as Pixel>::CHANNEL_COUNT as usize),
})
}
}
impl<'a, P: Pixel + 'a> ExactSizeIterator for Rows<'a, P>
where
P::Subpixel: 'a,
{
fn len(&self) -> usize {
self.pixels.len()
}
}
impl<'a, P: Pixel + 'a> DoubleEndedIterator for Rows<'a, P>
where
P::Subpixel: 'a,
{
#[inline(always)]
fn next_back(&mut self) -> Option<Pixels<'a, P>> {
let row = self.pixels.next_back()?;
Some(Pixels {
chunks: row.chunks_exact(<P as Pixel>::CHANNEL_COUNT as usize),
})
}
}
impl<P: Pixel> Clone for Rows<'_, P> {
fn clone(&self) -> Self {
Rows { pixels: self.pixels.clone() }
}
}
impl<P: Pixel> fmt::Debug for Rows<'_, P>
where
P::Subpixel: fmt::Debug,
{
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
f
.debug_struct("Rows")
.field("pixels", &self.pixels)
.finish()
}
}
pub struct RowsMut<'a, P: Pixel + 'a>
where
<P as Pixel>::Subpixel: 'a,
{
pixels: ChunksExactMut<'a, P::Subpixel>,
}
impl<'a, P: Pixel + 'a> RowsMut<'a, P> {
fn with_image(pixels: &'a mut [P::Subpixel], width: u32, height: u32) -> Self {
let row_len = (width as usize) * usize::from(<P as Pixel>::CHANNEL_COUNT);
if row_len == 0 {
RowsMut {
pixels: [].chunks_exact_mut(1),
}
} else {
let pixels = pixels.get_mut(..row_len*height as usize)
.expect("Pixel buffer has too few subpixels");
RowsMut {
pixels: pixels.chunks_exact_mut(row_len),
}
}
}
}
impl<'a, P: Pixel + 'a> Iterator for RowsMut<'a, P>
where
P::Subpixel: 'a,
{
type Item = PixelsMut<'a, P>;
#[inline(always)]
fn next(&mut self) -> Option<PixelsMut<'a, P>> {
let row = self.pixels.next()?;
Some(PixelsMut {
chunks: row.chunks_exact_mut(<P as Pixel>::CHANNEL_COUNT as usize),
})
}
}
impl<'a, P: Pixel + 'a> ExactSizeIterator for RowsMut<'a, P>
where
P::Subpixel: 'a,
{
fn len(&self) -> usize {
self.pixels.len()
}
}
impl<'a, P: Pixel + 'a> DoubleEndedIterator for RowsMut<'a, P>
where
P::Subpixel: 'a,
{
#[inline(always)]
fn next_back(&mut self) -> Option<PixelsMut<'a, P>> {
let row = self.pixels.next_back()?;
Some(PixelsMut {
chunks: row.chunks_exact_mut(<P as Pixel>::CHANNEL_COUNT as usize),
})
}
}
impl<P: Pixel> fmt::Debug for RowsMut<'_, P>
where
P::Subpixel: fmt::Debug,
{
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
f
.debug_struct("RowsMut")
.field("pixels", &self.pixels)
.finish()
}
}
pub struct EnumeratePixels<'a, P: Pixel + 'a>
where
<P as Pixel>::Subpixel: 'a,
{
pixels: Pixels<'a, P>,
x: u32,
y: u32,
width: u32,
}
impl<'a, P: Pixel + 'a> Iterator for EnumeratePixels<'a, P>
where
P::Subpixel: 'a,
{
type Item = (u32, u32, &'a P);
#[inline(always)]
fn next(&mut self) -> Option<(u32, u32, &'a P)> {
if self.x >= self.width {
self.x = 0;
self.y += 1;
}
let (x, y) = (self.x, self.y);
self.x += 1;
self.pixels.next().map(|p| (x, y, p))
}
}
impl<'a, P: Pixel + 'a> ExactSizeIterator for EnumeratePixels<'a, P>
where
P::Subpixel: 'a,
{
fn len(&self) -> usize {
self.pixels.len()
}
}
impl<P: Pixel> Clone for EnumeratePixels<'_, P> {
fn clone(&self) -> Self {
EnumeratePixels {
pixels: self.pixels.clone(),
..*self
}
}
}
impl<P: Pixel> fmt::Debug for EnumeratePixels<'_, P>
where
P::Subpixel: fmt::Debug,
{
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
f
.debug_struct("EnumeratePixels")
.field("pixels", &self.pixels)
.field("x", &self.x)
.field("y", &self.y)
.field("width", &self.width)
.finish()
}
}
pub struct EnumerateRows<'a, P: Pixel + 'a>
where
<P as Pixel>::Subpixel: 'a,
{
rows: Rows<'a, P>,
y: u32,
width: u32,
}
impl<'a, P: Pixel + 'a> Iterator for EnumerateRows<'a, P>
where
P::Subpixel: 'a,
{
type Item = (u32, EnumeratePixels<'a, P>);
#[inline(always)]
fn next(&mut self) -> Option<(u32, EnumeratePixels<'a, P>)> {
let y = self.y;
self.y += 1;
self.rows.next().map(|r| {
(
y,
EnumeratePixels {
x: 0,
y,
width: self.width,
pixels: r,
},
)
})
}
}
impl<'a, P: Pixel + 'a> ExactSizeIterator for EnumerateRows<'a, P>
where
P::Subpixel: 'a,
{
fn len(&self) -> usize {
self.rows.len()
}
}
impl<P: Pixel> Clone for EnumerateRows<'_, P> {
fn clone(&self) -> Self {
EnumerateRows {
rows: self.rows.clone(),
..*self
}
}
}
impl<P: Pixel> fmt::Debug for EnumerateRows<'_, P>
where
P::Subpixel: fmt::Debug,
{
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
f
.debug_struct("EnumerateRows")
.field("rows", &self.rows)
.field("y", &self.y)
.field("width", &self.width)
.finish()
}
}
pub struct EnumeratePixelsMut<'a, P: Pixel + 'a>
where
<P as Pixel>::Subpixel: 'a,
{
pixels: PixelsMut<'a, P>,
x: u32,
y: u32,
width: u32,
}
impl<'a, P: Pixel + 'a> Iterator for EnumeratePixelsMut<'a, P>
where
P::Subpixel: 'a,
{
type Item = (u32, u32, &'a mut P);
#[inline(always)]
fn next(&mut self) -> Option<(u32, u32, &'a mut P)> {
if self.x >= self.width {
self.x = 0;
self.y += 1;
}
let (x, y) = (self.x, self.y);
self.x += 1;
self.pixels.next().map(|p| (x, y, p))
}
}
impl<'a, P: Pixel + 'a> ExactSizeIterator for EnumeratePixelsMut<'a, P>
where
P::Subpixel: 'a,
{
fn len(&self) -> usize {
self.pixels.len()
}
}
impl<P: Pixel> fmt::Debug for EnumeratePixelsMut<'_, P>
where
P::Subpixel: fmt::Debug,
{
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
f
.debug_struct("EnumeratePixelsMut")
.field("pixels", &self.pixels)
.field("x", &self.x)
.field("y", &self.y)
.field("width", &self.width)
.finish()
}
}
pub struct EnumerateRowsMut<'a, P: Pixel + 'a>
where
<P as Pixel>::Subpixel: 'a,
{
rows: RowsMut<'a, P>,
y: u32,
width: u32,
}
impl<'a, P: Pixel + 'a> Iterator for EnumerateRowsMut<'a, P>
where
P::Subpixel: 'a,
{
type Item = (u32, EnumeratePixelsMut<'a, P>);
#[inline(always)]
fn next(&mut self) -> Option<(u32, EnumeratePixelsMut<'a, P>)> {
let y = self.y;
self.y += 1;
self.rows.next().map(|r| {
(
y,
EnumeratePixelsMut {
x: 0,
y,
width: self.width,
pixels: r,
},
)
})
}
}
impl<'a, P: Pixel + 'a> ExactSizeIterator for EnumerateRowsMut<'a, P>
where
P::Subpixel: 'a,
{
fn len(&self) -> usize {
self.rows.len()
}
}
impl<P: Pixel> fmt::Debug for EnumerateRowsMut<'_, P>
where
P::Subpixel: fmt::Debug,
{
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
f
.debug_struct("EnumerateRowsMut")
.field("rows", &self.rows)
.field("y", &self.y)
.field("width", &self.width)
.finish()
}
}
#[derive(Debug, Hash, PartialEq, Eq)]
pub struct ImageBuffer<P: Pixel, Container> {
width: u32,
height: u32,
_phantom: PhantomData<P>,
data: Container,
}
impl<P, Container> ImageBuffer<P, Container>
where
P: Pixel + 'static,
P::Subpixel: 'static,
Container: Deref<Target = [P::Subpixel]>,
{
pub fn from_raw(width: u32, height: u32, buf: Container) -> Option<ImageBuffer<P, Container>> {
if Self::check_image_fits(width, height, buf.len()) {
Some(ImageBuffer {
data: buf,
width,
height,
_phantom: PhantomData,
})
} else {
None
}
}
pub fn into_raw(self) -> Container {
self.data
}
pub fn as_raw(&self) -> &Container {
&self.data
}
pub fn dimensions(&self) -> (u32, u32) {
(self.width, self.height)
}
pub fn width(&self) -> u32 {
self.width
}
pub fn height(&self) -> u32 {
self.height
}
fn inner_pixels(&self) -> &[P::Subpixel] {
let len = Self::image_buffer_len(self.width, self.height).unwrap();
&self.data[..len]
}
pub fn pixels(&self) -> Pixels<P> {
Pixels {
chunks: self.inner_pixels().chunks_exact(<P as Pixel>::CHANNEL_COUNT as usize),
}
}
pub fn rows(&self) -> Rows<P> {
Rows::with_image(&self.data, self.width, self.height)
}
pub fn enumerate_pixels(&self) -> EnumeratePixels<P> {
EnumeratePixels {
pixels: self.pixels(),
x: 0,
y: 0,
width: self.width,
}
}
pub fn enumerate_rows(&self) -> EnumerateRows<P> {
EnumerateRows {
rows: self.rows(),
y: 0,
width: self.width,
}
}
pub fn get_pixel(&self, x: u32, y: u32) -> &P {
match self.pixel_indices(x, y) {
None => panic!("Image index {:?} out of bounds {:?}", (x, y), (self.width, self.height)),
Some(pixel_indices) => <P as Pixel>::from_slice(&self.data[pixel_indices]),
}
}
fn check_image_fits(width: u32, height: u32, len: usize) -> bool {
let checked_len = Self::image_buffer_len(width, height);
checked_len.map(|min_len| min_len <= len).unwrap_or(false)
}
fn image_buffer_len(width: u32, height: u32) -> Option<usize> {
Some(<P as Pixel>::CHANNEL_COUNT as usize)
.and_then(|size| size.checked_mul(width as usize))
.and_then(|size| size.checked_mul(height as usize))
}
#[inline(always)]
fn pixel_indices(&self, x: u32, y: u32) -> Option<Range<usize>> {
if x >= self.width || y >= self.height {
return None
}
Some(self.pixel_indices_unchecked(x, y))
}
#[inline(always)]
fn pixel_indices_unchecked(&self, x: u32, y: u32) -> Range<usize> {
let no_channels = <P as Pixel>::CHANNEL_COUNT as usize;
let min_index = (y as usize*self.width as usize + x as usize)*no_channels;
min_index..min_index+no_channels
}
pub fn sample_layout(&self) -> SampleLayout {
SampleLayout::row_major_packed(<P as Pixel>::CHANNEL_COUNT, self.width, self.height)
}
pub fn into_flat_samples(self) -> FlatSamples<Container>
where Container: AsRef<[P::Subpixel]>
{
let layout = self.sample_layout();
FlatSamples {
samples: self.data,
layout,
color_hint: Some(P::COLOR_TYPE),
}
}
pub fn as_flat_samples(&self) -> FlatSamples<&[P::Subpixel]>
where Container: AsRef<[P::Subpixel]>
{
let layout = self.sample_layout();
FlatSamples {
samples: self.data.as_ref(),
layout,
color_hint: Some(P::COLOR_TYPE),
}
}
pub fn as_flat_samples_mut(&mut self) -> FlatSamples<&mut [P::Subpixel]>
where Container: AsMut<[P::Subpixel]>
{
let layout = self.sample_layout();
FlatSamples {
samples: self.data.as_mut(),
layout,
color_hint: Some(P::COLOR_TYPE),
}
}
}
impl<P, Container> ImageBuffer<P, Container>
where
P: Pixel + 'static,
P::Subpixel: 'static,
Container: Deref<Target = [P::Subpixel]> + DerefMut,
{
fn inner_pixels_mut(&mut self) -> &mut [P::Subpixel] {
let len = Self::image_buffer_len(self.width, self.height).unwrap();
&mut self.data[..len]
}
pub fn pixels_mut(&mut self) -> PixelsMut<P> {
PixelsMut {
chunks: self.inner_pixels_mut().chunks_exact_mut(<P as Pixel>::CHANNEL_COUNT as usize),
}
}
pub fn rows_mut(&mut self) -> RowsMut<P> {
RowsMut::with_image(&mut self.data, self.width, self.height)
}
pub fn enumerate_pixels_mut(&mut self) -> EnumeratePixelsMut<P> {
let width = self.width;
EnumeratePixelsMut {
pixels: self.pixels_mut(),
x: 0,
y: 0,
width,
}
}
pub fn enumerate_rows_mut(&mut self) -> EnumerateRowsMut<P> {
let width = self.width;
EnumerateRowsMut {
rows: self.rows_mut(),
y: 0,
width,
}
}
pub fn get_pixel_mut(&mut self, x: u32, y: u32) -> &mut P {
match self.pixel_indices(x, y) {
None => panic!("Image index {:?} out of bounds {:?}", (x, y), (self.width, self.height)),
Some(pixel_indices) => <P as Pixel>::from_slice_mut(&mut self.data[pixel_indices]),
}
}
pub fn put_pixel(&mut self, x: u32, y: u32, pixel: P) {
*self.get_pixel_mut(x, y) = pixel
}
}
impl<P, Container> ImageBuffer<P, Container>
where
P: Pixel + 'static,
[P::Subpixel]: EncodableLayout,
Container: Deref<Target = [P::Subpixel]>,
{
pub fn save<Q>(&self, path: Q) -> ImageResult<()>
where
Q: AsRef<Path>,
{
save_buffer(
path,
self.as_bytes(),
self.width(),
self.height(),
<P as Pixel>::COLOR_TYPE,
)
}
}
impl<P, Container> ImageBuffer<P, Container>
where
P: Pixel + 'static,
[P::Subpixel]: EncodableLayout,
Container: Deref<Target = [P::Subpixel]>,
{
pub fn save_with_format<Q>(&self, path: Q, format: ImageFormat) -> ImageResult<()>
where
Q: AsRef<Path>,
{
save_buffer_with_format(
path,
self.as_bytes(),
self.width(),
self.height(),
<P as Pixel>::COLOR_TYPE,
format,
)
}
}
impl<P, Container> Default for ImageBuffer<P, Container>
where
P: Pixel,
Container: Default,
{
fn default() -> Self {
Self {
width: 0,
height: 0,
_phantom: PhantomData,
data: Default::default(),
}
}
}
impl<P, Container> Deref for ImageBuffer<P, Container>
where
P: Pixel + 'static,
P::Subpixel: 'static,
Container: Deref<Target = [P::Subpixel]>,
{
type Target = [P::Subpixel];
fn deref(&self) -> &<Self as Deref>::Target {
&*self.data
}
}
impl<P, Container> DerefMut for ImageBuffer<P, Container>
where
P: Pixel + 'static,
P::Subpixel: 'static,
Container: Deref<Target = [P::Subpixel]> + DerefMut,
{
fn deref_mut(&mut self) -> &mut <Self as Deref>::Target {
&mut *self.data
}
}
impl<P, Container> Index<(u32, u32)> for ImageBuffer<P, Container>
where
P: Pixel + 'static,
P::Subpixel: 'static,
Container: Deref<Target = [P::Subpixel]>,
{
type Output = P;
fn index(&self, (x, y): (u32, u32)) -> &P {
self.get_pixel(x, y)
}
}
impl<P, Container> IndexMut<(u32, u32)> for ImageBuffer<P, Container>
where
P: Pixel + 'static,
P::Subpixel: 'static,
Container: Deref<Target = [P::Subpixel]> + DerefMut,
{
fn index_mut(&mut self, (x, y): (u32, u32)) -> &mut P {
self.get_pixel_mut(x, y)
}
}
impl<P, Container> Clone for ImageBuffer<P, Container>
where
P: Pixel,
Container: Deref<Target = [P::Subpixel]> + Clone,
{
fn clone(&self) -> ImageBuffer<P, Container> {
ImageBuffer {
data: self.data.clone(),
width: self.width,
height: self.height,
_phantom: PhantomData,
}
}
}
impl<P, Container> GenericImageView for ImageBuffer<P, Container>
where
P: Pixel + 'static,
Container: Deref<Target = [P::Subpixel]> + Deref,
P::Subpixel: 'static,
{
type Pixel = P;
type InnerImageView = Self;
fn dimensions(&self) -> (u32, u32) {
self.dimensions()
}
fn bounds(&self) -> (u32, u32, u32, u32) {
(0, 0, self.width, self.height)
}
fn get_pixel(&self, x: u32, y: u32) -> P {
*self.get_pixel(x, y)
}
#[inline(always)]
unsafe fn unsafe_get_pixel(&self, x: u32, y: u32) -> P {
let indices = self.pixel_indices_unchecked(x, y);
*<P as Pixel>::from_slice(self.data.get_unchecked(indices))
}
fn inner(&self) -> &Self::InnerImageView {
self
}
}
impl<P, Container> GenericImage for ImageBuffer<P, Container>
where
P: Pixel + 'static,
Container: Deref<Target = [P::Subpixel]> + DerefMut,
P::Subpixel: 'static,
{
type InnerImage = Self;
fn get_pixel_mut(&mut self, x: u32, y: u32) -> &mut P {
self.get_pixel_mut(x, y)
}
fn put_pixel(&mut self, x: u32, y: u32, pixel: P) {
*self.get_pixel_mut(x, y) = pixel
}
#[inline(always)]
unsafe fn unsafe_put_pixel(&mut self, x: u32, y: u32, pixel: P) {
let indices = self.pixel_indices_unchecked(x, y);
let p = <P as Pixel>::from_slice_mut(self.data.get_unchecked_mut(indices));
*p = pixel
}
fn blend_pixel(&mut self, x: u32, y: u32, p: P) {
self.get_pixel_mut(x, y).blend(&p)
}
fn copy_within(&mut self, source: Rect, x: u32, y: u32) -> bool {
let Rect { x: sx, y: sy, width, height } = source;
let dx = x;
let dy = y;
assert!(sx < self.width() && dx < self.width());
assert!(sy < self.height() && dy < self.height());
if self.width() - dx.max(sx) < width || self.height() - dy.max(sy) < height {
return false;
}
if sy < dy {
for y in (0..height).rev() {
let sy = sy + y;
let dy = dy + y;
let Range { start, .. } = self.pixel_indices_unchecked(sx, sy);
let Range { end, .. } = self.pixel_indices_unchecked(sx + width - 1, sy);
let dst = self.pixel_indices_unchecked(dx, dy).start;
slice_copy_within(self, start..end, dst);
}
} else {
for y in 0..height {
let sy = sy + y;
let dy = dy + y;
let Range { start, .. } = self.pixel_indices_unchecked(sx, sy);
let Range { end, .. } = self.pixel_indices_unchecked(sx + width - 1, sy);
let dst = self.pixel_indices_unchecked(dx, dy).start;
slice_copy_within(self, start..end, dst);
}
}
true
}
fn inner_mut(&mut self) -> &mut Self::InnerImage {
self
}
}
#[inline(always)]
fn slice_copy_within<T: Copy>(slice: &mut [T], Range { start: src_start, end: src_end }: Range<usize>, dest: usize) {
assert!(src_start <= src_end, "src end is before src start");
assert!(src_end <= slice.len(), "src is out of bounds");
let count = src_end - src_start;
assert!(dest <= slice.len() - count, "dest is out of bounds");
unsafe {
std::ptr::copy(
slice.as_ptr().add(src_start),
slice.as_mut_ptr().add(dest),
count,
);
}
}
impl<P: Pixel + 'static> ImageBuffer<P, Vec<P::Subpixel>>
where
P::Subpixel: 'static,
{
pub fn new(width: u32, height: u32) -> ImageBuffer<P, Vec<P::Subpixel>> {
let size = Self::image_buffer_len(width, height)
.expect("Buffer length in `ImageBuffer::new` overflows usize");
ImageBuffer {
data: vec![Zero::zero(); size],
width,
height,
_phantom: PhantomData,
}
}
pub fn from_pixel(width: u32, height: u32, pixel: P) -> ImageBuffer<P, Vec<P::Subpixel>> {
let mut buf = ImageBuffer::new(width, height);
for p in buf.pixels_mut() {
*p = pixel
}
buf
}
pub fn from_fn<F>(width: u32, height: u32, mut f: F) -> ImageBuffer<P, Vec<P::Subpixel>>
where
F: FnMut(u32, u32) -> P,
{
let mut buf = ImageBuffer::new(width, height);
for (x, y, p) in buf.enumerate_pixels_mut() {
*p = f(x, y)
}
buf
}
pub fn from_vec(
width: u32,
height: u32,
buf: Vec<P::Subpixel>,
) -> Option<ImageBuffer<P, Vec<P::Subpixel>>> {
ImageBuffer::from_raw(width, height, buf)
}
pub fn into_vec(self) -> Vec<P::Subpixel> {
self.into_raw()
}
}
pub trait ConvertBuffer<T> {
fn convert(&self) -> T;
}
impl GrayImage {
pub fn expand_palette(
self,
palette: &[(u8, u8, u8)],
transparent_idx: Option<u8>,
) -> RgbaImage {
let (width, height) = self.dimensions();
let mut data = self.into_raw();
let entries = data.len();
data.resize(entries.checked_mul(4).unwrap(), 0);
let mut buffer = ImageBuffer::from_vec(width, height, data).unwrap();
expand_packed(&mut buffer, 4, 8, |idx, pixel| {
let (r, g, b) = palette[idx as usize];
let a = if let Some(t_idx) = transparent_idx {
if t_idx == idx {
0
} else {
255
}
} else {
255
};
pixel[0] = r;
pixel[1] = g;
pixel[2] = b;
pixel[3] = a;
});
buffer
}
}
impl<'a, 'b, Container, FromType: Pixel + 'static, ToType: Pixel + 'static>
ConvertBuffer<ImageBuffer<ToType, Vec<ToType::Subpixel>>> for ImageBuffer<FromType, Container>
where
Container: Deref<Target = [FromType::Subpixel]>,
ToType: FromColor<FromType>,
FromType::Subpixel: 'static,
ToType::Subpixel: 'static,
{
fn convert(&self) -> ImageBuffer<ToType, Vec<ToType::Subpixel>> {
let mut buffer: ImageBuffer<ToType, Vec<ToType::Subpixel>> =
ImageBuffer::new(self.width, self.height);
for (to, from) in buffer.pixels_mut().zip(self.pixels()) {
to.from_color(from)
}
buffer
}
}
pub type RgbImage = ImageBuffer<Rgb<u8>, Vec<u8>>;
pub type RgbaImage = ImageBuffer<Rgba<u8>, Vec<u8>>;
pub type GrayImage = ImageBuffer<Luma<u8>, Vec<u8>>;
pub type GrayAlphaImage = ImageBuffer<LumaA<u8>, Vec<u8>>;
pub(crate) type BgrImage = ImageBuffer<Bgr<u8>, Vec<u8>>;
pub(crate) type BgraImage = ImageBuffer<Bgra<u8>, Vec<u8>>;
pub(crate) type Rgb16Image = ImageBuffer<Rgb<u16>, Vec<u16>>;
pub(crate) type Rgba16Image = ImageBuffer<Rgba<u16>, Vec<u16>>;
pub(crate) type Gray16Image = ImageBuffer<Luma<u16>, Vec<u16>>;
pub(crate) type GrayAlpha16Image = ImageBuffer<LumaA<u16>, Vec<u16>>;
#[cfg(test)]
mod test {
use super::{ImageBuffer, RgbImage};
use crate::{color, Rgb};
#[test]
fn slice_buffer() {
let data = [0; 9];
let buf: ImageBuffer<color::Luma<u8>, _> = ImageBuffer::from_raw(3, 3, &data[..]).unwrap();
assert_eq!(&*buf, &data[..])
}
#[test]
fn get_pixel() {
let mut a: RgbImage = ImageBuffer::new(10, 10);
{
let b = a.get_mut(3 * 10).unwrap();
*b = 255;
}
assert_eq!(a.get_pixel(0, 1)[0], 255)
}
#[test]
fn mut_iter() {
let mut a: RgbImage = ImageBuffer::new(10, 10);
{
let val = a.pixels_mut().next().unwrap();
*val = color::Rgb([42, 0, 0]);
}
assert_eq!(a.data[0], 42)
}
#[test]
fn zero_width_zero_height() {
let mut image = RgbImage::new(0, 0);
assert_eq!(image.rows_mut().count(), 0);
assert_eq!(image.pixels_mut().count(), 0);
assert_eq!(image.rows().count(), 0);
assert_eq!(image.pixels().count(), 0);
}
#[test]
fn zero_width_nonzero_height() {
let mut image = RgbImage::new(0, 2);
assert_eq!(image.rows_mut().count(), 0);
assert_eq!(image.pixels_mut().count(), 0);
assert_eq!(image.rows().count(), 0);
assert_eq!(image.pixels().count(), 0);
}
#[test]
fn nonzero_width_zero_height() {
let mut image = RgbImage::new(2, 0);
assert_eq!(image.rows_mut().count(), 0);
assert_eq!(image.pixels_mut().count(), 0);
assert_eq!(image.rows().count(), 0);
assert_eq!(image.pixels().count(), 0);
}
#[test]
fn pixels_on_large_buffer() {
let mut image = RgbImage::from_raw(1, 1, vec![0; 6]).unwrap();
assert_eq!(image.pixels().count(), 1);
assert_eq!(image.enumerate_pixels().count(), 1);
assert_eq!(image.pixels_mut().count(), 1);
assert_eq!(image.enumerate_pixels_mut().count(), 1);
assert_eq!(image.rows().count(), 1);
assert_eq!(image.rows_mut().count(), 1);
}
#[test]
fn default() {
let image = ImageBuffer::<Rgb<u8>, Vec<u8>>::default();
assert_eq!(image.dimensions(), (0, 0));
}
}
#[cfg(test)]
#[cfg(feature = "benchmarks")]
mod benchmarks {
use super::{ConvertBuffer, GrayImage, ImageBuffer, Pixel, RgbImage};
use crate::GenericImage;
use crate::math::Rect;
use test;
#[bench]
fn conversion(b: &mut test::Bencher) {
let mut a: RgbImage = ImageBuffer::new(1000, 1000);
for p in a.pixels_mut() {
let rgb = p.channels_mut();
rgb[0] = 255;
rgb[1] = 23;
rgb[2] = 42;
}
assert!(a.data[0] != 0);
b.iter(|| {
let b: GrayImage = a.convert();
assert!(0 != b.data[0]);
assert!(a.data[0] != b.data[0]);
test::black_box(b);
});
b.bytes = 1000 * 1000 * 3
}
#[bench]
fn image_access_row_by_row(b: &mut test::Bencher) {
let mut a: RgbImage = ImageBuffer::new(1000, 1000);
for p in a.pixels_mut() {
let rgb = p.channels_mut();
rgb[0] = 255;
rgb[1] = 23;
rgb[2] = 42;
}
b.iter(move || {
let image: &RgbImage = test::black_box(&a);
let mut sum: usize = 0;
for y in 0..1000 {
for x in 0..1000 {
let pixel = image.get_pixel(x, y);
sum = sum.wrapping_add(pixel[0] as usize);
sum = sum.wrapping_add(pixel[1] as usize);
sum = sum.wrapping_add(pixel[2] as usize);
}
}
test::black_box(sum)
});
b.bytes = 1000 * 1000 * 3;
}
#[bench]
fn image_access_col_by_col(b: &mut test::Bencher) {
let mut a: RgbImage = ImageBuffer::new(1000, 1000);
for p in a.pixels_mut() {
let rgb = p.channels_mut();
rgb[0] = 255;
rgb[1] = 23;
rgb[2] = 42;
}
b.iter(move || {
let image: &RgbImage = test::black_box(&a);
let mut sum: usize = 0;
for x in 0..1000 {
for y in 0..1000 {
let pixel = image.get_pixel(x, y);
sum = sum.wrapping_add(pixel[0] as usize);
sum = sum.wrapping_add(pixel[1] as usize);
sum = sum.wrapping_add(pixel[2] as usize);
}
}
test::black_box(sum)
});
b.bytes = 1000 * 1000 * 3;
}
#[test]
fn test_image_buffer_copy_within_oob() {
let mut image: GrayImage = ImageBuffer::from_raw(4, 4, vec![0u8; 16]).unwrap();
assert!(!image.copy_within(Rect { x: 0, y: 0, width: 5, height: 4 }, 0, 0));
assert!(!image.copy_within(Rect { x: 0, y: 0, width: 4, height: 5 }, 0, 0));
assert!(!image.copy_within(Rect { x: 1, y: 0, width: 4, height: 4 }, 0, 0));
assert!(!image.copy_within(Rect { x: 0, y: 0, width: 4, height: 4 }, 1, 0));
assert!(!image.copy_within(Rect { x: 0, y: 1, width: 4, height: 4 }, 0, 0));
assert!(!image.copy_within(Rect { x: 0, y: 0, width: 4, height: 4 }, 0, 1));
assert!(!image.copy_within(Rect { x: 1, y: 1, width: 4, height: 4 }, 0, 0));
}
#[test]
fn test_image_buffer_copy_within_tl() {
let data = &[
00, 01, 02, 03,
04, 05, 06, 07,
08, 09, 10, 11,
12, 13, 14, 15
];
let expected = [
00, 01, 02, 03,
04, 00, 01, 02,
08, 04, 05, 06,
12, 08, 09, 10,
];
let mut image: GrayImage = ImageBuffer::from_raw(4, 4, Vec::from(&data[..])).unwrap();
assert!(image.copy_within(Rect { x: 0, y: 0, width: 3, height: 3 }, 1, 1));
assert_eq!(&image.into_raw(), &expected);
}
#[test]
fn test_image_buffer_copy_within_tr() {
let data = &[
00, 01, 02, 03,
04, 05, 06, 07,
08, 09, 10, 11,
12, 13, 14, 15
];
let expected = [
00, 01, 02, 03,
01, 02, 03, 07,
05, 06, 07, 11,
09, 10, 11, 15
];
let mut image: GrayImage = ImageBuffer::from_raw(4, 4, Vec::from(&data[..])).unwrap();
assert!(image.copy_within(Rect { x: 1, y: 0, width: 3, height: 3 }, 0, 1));
assert_eq!(&image.into_raw(), &expected);
}
#[test]
fn test_image_buffer_copy_within_bl() {
let data = &[
00, 01, 02, 03,
04, 05, 06, 07,
08, 09, 10, 11,
12, 13, 14, 15
];
let expected = [
00, 04, 05, 06,
04, 08, 09, 10,
08, 12, 13, 14,
12, 13, 14, 15
];
let mut image: GrayImage = ImageBuffer::from_raw(4, 4, Vec::from(&data[..])).unwrap();
assert!(image.copy_within(Rect { x: 0, y: 1, width: 3, height: 3 }, 1, 0));
assert_eq!(&image.into_raw(), &expected);
}
#[test]
fn test_image_buffer_copy_within_br() {
let data = &[
00, 01, 02, 03,
04, 05, 06, 07,
08, 09, 10, 11,
12, 13, 14, 15
];
let expected = [
05, 06, 07, 03,
09, 10, 11, 07,
13, 14, 15, 11,
12, 13, 14, 15
];
let mut image: GrayImage = ImageBuffer::from_raw(4, 4, Vec::from(&data[..])).unwrap();
assert!(image.copy_within(Rect { x: 1, y: 1, width: 3, height: 3 }, 0, 0));
assert_eq!(&image.into_raw(), &expected);
}
}