move more to generic points

This commit is contained in:
Henry Schimke
2023-05-04 13:16:44 -05:00
parent 74a830f462
commit 592a4d694c
2 changed files with 137 additions and 62 deletions

View File

@@ -228,7 +228,7 @@ impl<'a> PatternView<'a> {
let mut size = size.unwrap_or(0); let mut size = size.unwrap_or(0);
if size == 0 { if size == 0 {
size = self.count - offset; size = self.count - offset;
} }
// else if size < 0 { // else if size < 0 {
// size += self.count - offset; // size += self.count - offset;
// } // }

View File

@@ -1,3 +1,4 @@
use std::process::Output;
use std::{fmt, iter::Sum}; use std::{fmt, iter::Sum};
use std::hash::Hash; use std::hash::Hash;
@@ -32,7 +33,7 @@ pub type Point = PointF;
impl Into<PointI> for Point { impl Into<PointI> for Point {
fn into(self) -> PointI { fn into(self) -> PointI {
PointI{ PointI {
x: self.x.floor() as u32, x: self.x.floor() as u32,
y: self.y.floor() as u32, y: self.y.floor() as u32,
} }
@@ -75,58 +76,82 @@ impl PartialEq for Point {
} }
impl Eq for Point {} impl Eq for Point {}
impl Point { impl<T> PointT<T>
pub const fn new(x: f32, y: f32) -> Self { where
Self { x, y } T: Copy,
{
pub const fn new(x: T, y: T) -> PointT<T> {
PointT { x: x, y: y }
} }
pub const fn with_single(x: T) -> Self {
pub const fn with_single(x: f32) -> Self {
Self { x, y: x } Self { x, y: x }
} }
} }
impl std::ops::AddAssign for Point { impl<T> std::ops::AddAssign for PointT<T>
where
T: std::ops::Add<Output = T> + Copy,
{
fn add_assign(&mut self, rhs: Self) { fn add_assign(&mut self, rhs: Self) {
self.x = self.x + rhs.x; self.x = self.x + rhs.x;
self.y = self.y + rhs.y; self.y = self.y + rhs.y;
} }
} }
impl std::ops::SubAssign for Point { impl<T> std::ops::SubAssign for PointT<T>
where
T: std::ops::Sub<Output = T> + Copy,
{
fn sub_assign(&mut self, rhs: Self) { fn sub_assign(&mut self, rhs: Self) {
self.x = self.x - rhs.x; self.x = self.x - rhs.x;
self.y = self.y - rhs.y; self.y = self.y - rhs.y;
} }
} }
impl<'a> Sum<&'a Point> for Point { impl<'a, T> Sum<&'a PointT<T>> for PointT<T>
where
T: std::ops::Add<Output = T> + 'a + Default,
PointT<T>: std::ops::Add<Output = PointT<T>> + Copy,
{
fn sum<I: Iterator<Item = &'a Self>>(iter: I) -> Self { fn sum<I: Iterator<Item = &'a Self>>(iter: I) -> Self {
iter.fold(Self::default(), |acc, &p| acc + p) iter.fold(Self::default(), |acc, &p| acc + p)
} }
} }
/** This impl is temporary and is there to ease refactoring. */ /** This impl is temporary and is there to ease refactoring. */
impl ResultPoint for Point { impl<T> ResultPoint for PointT<T>
where
T: Into<f32> + Copy,
{
fn getX(&self) -> f32 { fn getX(&self) -> f32 {
self.x self.x.into()
} }
fn getY(&self) -> f32 { fn getY(&self) -> f32 {
self.y self.y.into()
} }
fn to_rxing_result_point(&self) -> Self { fn to_rxing_result_point(&self) -> PointT<f32> {
*self PointT {
x: self.x.into(),
y: self.y.into(),
}
} }
} }
impl fmt::Display for Point { impl<T> fmt::Display for PointT<T>
where
T: std::fmt::Display,
{
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result { fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
write!(f, "({},{})", self.x, self.y) write!(f, "({},{})", self.x, self.y)
} }
} }
impl std::ops::Sub for Point { impl<T> std::ops::Sub for PointT<T>
where
T: std::ops::Sub<Output = T> + Copy,
{
type Output = Self; type Output = Self;
fn sub(self, rhs: Self) -> Self::Output { fn sub(self, rhs: Self) -> Self::Output {
@@ -134,7 +159,10 @@ impl std::ops::Sub for Point {
} }
} }
impl std::ops::Neg for Point { impl<T> std::ops::Neg for PointT<T>
where
T: std::ops::Neg<Output = T> + Copy,
{
type Output = Self; type Output = Self;
fn neg(self) -> Self::Output { fn neg(self) -> Self::Output {
@@ -142,7 +170,10 @@ impl std::ops::Neg for Point {
} }
} }
impl std::ops::Add for Point { impl<T> std::ops::Add for PointT<T>
where
T: std::ops::Add<Output = T> + Copy,
{
type Output = Self; type Output = Self;
fn add(self, rhs: Self) -> Self::Output { fn add(self, rhs: Self) -> Self::Output {
@@ -150,7 +181,10 @@ impl std::ops::Add for Point {
} }
} }
impl std::ops::Add<f32> for Point { impl<T> std::ops::Add<f32> for PointT<T>
where
T: Into<f32> + std::ops::Add<f32, Output = T> + Copy,
{
type Output = Self; type Output = Self;
fn add(self, rhs: f32) -> Self::Output { fn add(self, rhs: f32) -> Self::Output {
@@ -158,15 +192,21 @@ impl std::ops::Add<f32> for Point {
} }
} }
impl std::ops::Add<Point> for f32 { impl<T> std::ops::Add<PointT<T>> for f32
where
T: std::ops::Add<f32, Output = f32>,
{
type Output = Point; type Output = Point;
fn add(self, rhs: Point) -> Self::Output { fn add(self, rhs: PointT<T>) -> Self::Output {
Point::new(rhs.x + self, rhs.y + self) Point::new(rhs.x + self, rhs.y + self)
} }
} }
impl std::ops::Mul for Point { impl<T> std::ops::Mul for PointT<T>
where
T: std::ops::Mul<Output = T> + Copy,
{
type Output = Self; type Output = Self;
fn mul(self, rhs: Self) -> Self::Output { fn mul(self, rhs: Self) -> Self::Output {
@@ -230,36 +270,41 @@ impl std::ops::Mul<Point> for u32 {
} }
} }
impl Point { impl<T> PointT<T>
pub fn dot(self, p: Self) -> f32 { where
T: std::ops::Mul<Output = T> + std::ops::Sub<Output = T> + num::traits::real::Real,
PointT<T>: std::ops::Div<T, Output = PointT<T>>,
{
pub fn dot(self, p: Self) -> T {
self.x * p.x + self.y * p.y self.x * p.x + self.y * p.y
} }
pub fn cross(self, p: Self) -> f32 { pub fn cross(self, p: Self) -> T {
self.x * p.y - p.x * self.y self.x * p.y - p.x * self.y
} }
/// L1 norm /// L1 norm
pub fn sumAbsComponent(self) -> f32 { pub fn sumAbsComponent(self) -> T {
self.x.abs() + self.y.abs() self.x.abs() + self.y.abs()
} }
/// L2 norm /// L2 norm
pub fn length(self) -> f32 { pub fn length(self) -> T {
self.x.hypot(self.y) self.x.hypot(self.y)
} }
/// L-inf norm /// L-inf norm
pub fn maxAbsComponent(self) -> f32 { pub fn maxAbsComponent(self) -> T {
f32::max(self.x.abs(), self.y.abs()) self.x.abs().max(self.y.abs())
// f32::max(self.x.abs(), self.y.abs())
} }
pub fn squaredDistance(self, p: Self) -> f32 { pub fn squaredDistance(self, p: Self) -> T {
let diff = self - p; let diff = self - p;
diff.x * diff.x + diff.y * diff.y diff.x * diff.x + diff.y * diff.y
} }
pub fn distance(self, p: Self) -> f32 { pub fn distance(self, p: Self) -> T {
(self - p).length() (self - p).length()
} }
@@ -267,20 +312,15 @@ impl Point {
Self::new(self.x.abs(), self.y.abs()) Self::new(self.x.abs(), self.y.abs())
} }
pub fn fold<U, F: Fn(f32, f32) -> U>(self, f: F) -> U { pub fn fold<U, F: Fn(T, T) -> U>(self, f: F) -> U {
f(self.x, self.y) f(self.x, self.y)
} }
/// Calculate a floating point pixel coordinate representing the 'center' of the pixel. pub fn middle(self, p: Self) -> Self
/// This is sort of the inverse operation of the PointI(PointF) conversion constructor. where
/// See also the documentation of the GridSampler API. T: From<u8>,
#[inline(always)] {
pub fn centered(self) -> Self { (self + p) / 2.into()
Self::new(self.x.floor() + 0.5, self.y.floor() + 0.5)
}
pub fn middle(self, p: Self) -> Self {
(self + p) / 2.0
} }
pub fn normalized(self) -> Self { pub fn normalized(self) -> Self {
@@ -291,11 +331,14 @@ impl Point {
self / Self::maxAbsComponent(self) self / Self::maxAbsComponent(self)
} }
pub fn mainDirection(self) -> Self { pub fn mainDirection(self) -> Self
where
T: From<u8>,
{
if self.x.abs() > self.y.abs() { if self.x.abs() > self.y.abs() {
Self::new(self.x, 0.0) Self::new(self.x, 0.into())
} else { } else {
Self::new(0.0, self.y) Self::new(0.into(), self.y)
} }
} }
@@ -306,6 +349,17 @@ impl Point {
} }
} }
/// Calculate a floating point pixel coordinate representing the 'center' of the pixel.
/// This is sort of the inverse operation of the PointI(PointF) conversion constructor.
/// See also the documentation of the GridSampler API.
#[inline(always)]
pub fn centered(self) -> PointT<f32>
where
T: Into<f32>,
{
PointT::new(self.x.floor().into() + 0.5, self.y.floor().into() + 0.5)
}
pub fn floor(self) -> Self { pub fn floor(self) -> Self {
Self { Self {
x: self.x.floor(), x: self.x.floor(),
@@ -314,27 +368,48 @@ impl Point {
} }
} }
impl From<&(f32, f32)> for Point {
fn from(&(x, y): &(f32, f32)) -> Self {
Self::new(x, y)
}
}
impl From<(f32, f32)> for Point {
fn from((x, y): (f32, f32)) -> Self {
Self::new(x, y)
}
}
impl From<(i32, i32)> for Point { impl From<(i32, i32)> for Point {
fn from(value: (i32, i32)) -> Self { fn from((x, y): (i32, i32)) -> Self {
Self::new(value.0 as f32, value.1 as f32) Self::new(x as f32, y as f32)
} }
} }
impl From<(u32, u32)> for Point { impl From<(u32, u32)> for Point {
fn from(value: (u32, u32)) -> Self { fn from((x, y): (u32, u32)) -> Self {
Self::new(value.0 as f32, value.1 as f32) Self::new(x as f32, y as f32)
}
}
impl From<(f32, f32)> for PointI {
fn from((x, y): (f32, f32)) -> Self {
PointI {
x: x.floor() as u32,
y: y.floor() as u32,
}
}
}
impl<T> From<(T, T)> for PointT<T> {
fn from((x, y): (T, T)) -> PointT<T> {
PointT { x, y }
}
}
impl<T> From<&(T, T)> for PointT<T>
where
T: Copy,
{
fn from(&(x, y): &(T, T)) -> PointT<T> {
PointT { x, y }
}
}
impl<T> From<T> for PointT<T>
where
T: Copy,
{
fn from(value: T) -> Self {
Self::with_single(value)
} }
} }