picto 0.4.4

Image handling library.
Documentation 
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

//            DO WHAT THE FUCK YOU WANT TO PUBLIC LICENSE
//                    Version 2, December 2004
//
// Copyleft (ↄ) meh. <meh@schizofreni.co> | http://meh.schizofreni.co
//
// Everyone is permitted to copy and distribute verbatim or modified
// copies of this license document, and changing it is allowed as long
// as the name is changed.
//
//            DO WHAT THE FUCK YOU WANT TO PUBLIC LICENSE
//   TERMS AND CONDITIONS FOR COPYING, DISTRIBUTION AND MODIFICATION
//
//  0. You just DO WHAT THE FUCK YOU WANT TO.

use crate::{
	buffer::Buffer,
	color::{ComponentWise, Limited, Rgba},
	pixel,
	processing::blur,
	view,
};

/// Trait for blurrable types.
pub trait Sharpen<P, C>
where
	P: From<Rgba> + Into<Rgba>,
	P: pixel::Read<C> + pixel::Write<C>,
	C: pixel::Channel,
{
	/// Sharpen by the given radius and threshold.
	///
	/// # Example
	///
	/// ```
	/// use picto::read;
	/// use picto::color::Rgb;
	/// use picto::processing::prelude::*;
	///
	/// let image = read::from_path::<Rgb, u8, _>("tests/boat.xyz").unwrap();
	///
	/// image.sharpen(4.0, 0.02);
	/// ```
	fn sharpen(self, sigma: f32, threshold: f32) -> Buffer<P, C, Vec<C>>;
}

impl<'i, P, C, I> Sharpen<P, C> for I
where
	P: From<Rgba> + Into<Rgba>,
	P: pixel::Read<C> + pixel::Write<C>,
	C: pixel::Channel,
	I: Into<view::Read<'i, P, C>>,
{
	#[inline]
	fn sharpen(self, sigma: f32, threshold: f32) -> Buffer<P, C, Vec<C>> {
		by::<_, P, C, P, C>(self, sigma, threshold)
	}
}

/// Sharpen by the given radius and threshold.
#[inline]
pub fn by<'i, I, PI, CI, PO, CO>(input: I, sigma: f32, threshold: f32) -> Buffer<PO, CO, Vec<CO>>
where
	I: Into<view::Read<'i, PI, CI>>,
	PO: From<Rgba> + Into<Rgba>,
	PO: pixel::Read<CO> + pixel::Write<CO>,
	CO: pixel::Channel,
	PI: Into<Rgba>,
	PI: pixel::Read<CI>,
	CI: pixel::Channel,
{
	let input = input.into();
	let mut output = blur::by::<_, PI, CI, PO, CO>(&input, sigma);

	for ((_, _, i), (_, _, mut o)) in input.pixels().zip(output.pixels_mut()) {
		let a = i.get().into();
		let b = o.get().into();

		o.set(
			&a.component_wise(&b, |a, b| {
				let diff = (a - b).abs();

				if diff > threshold {
					a + diff
				}
				else {
					a
				}
			})
			.clamp()
			.into(),
		)
	}

	output
}