summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
-rw-r--r--src/coords.rs113
-rw-r--r--src/lib.rs2
2 files changed, 115 insertions, 0 deletions
diff --git a/src/coords.rs b/src/coords.rs
new file mode 100644
index 0000000..c43b7e9
--- /dev/null
+++ b/src/coords.rs
@@ -0,0 +1,113 @@
+//! Coordinate spaces.
+
+use crate::distance::{Distance, Value};
+
+/// A coordinate space.
+pub trait Coordinates {
+ /// The type of individual coordinates.
+ type Value: Value;
+
+ /// Get the number of dims this point has.
+ fn dims(&self) -> usize;
+
+ /// Get the `i`th coordinate of this point.
+ fn coord(&self, i: usize) -> Self::Value;
+
+ /// Create a vector with this point's coordinates as values.
+ fn as_vec(&self) -> Vec<Self::Value> {
+ let len = self.dims();
+ let mut vec = Vec::with_capacity(len);
+ for i in 0..len {
+ vec.push(self.coord(i));
+ }
+ vec
+ }
+}
+
+/// [Coordinates] implementation for slices.
+impl<T: Value> Coordinates for [T] {
+ type Value = T;
+
+ fn dims(&self) -> usize {
+ self.len()
+ }
+
+ fn coord(&self, i: usize) -> T {
+ self[i]
+ }
+}
+
+/// [Coordinates] implementation for arrays.
+macro_rules! array_coordinates {
+ ($n:expr) => (
+ impl<T: Value> Coordinates for [T; $n] {
+ type Value = T;
+
+ fn dims(&self) -> usize {
+ $n
+ }
+
+ fn coord(&self, i: usize) -> T {
+ self[i]
+ }
+ }
+ );
+}
+
+array_coordinates!(1);
+array_coordinates!(2);
+array_coordinates!(3);
+array_coordinates!(4);
+array_coordinates!(5);
+array_coordinates!(6);
+array_coordinates!(7);
+array_coordinates!(8);
+
+/// [Coordinates] implemention for vectors.
+impl<T: Value> Coordinates for Vec<T> {
+ type Value = T;
+
+ fn dims(&self) -> usize {
+ self.len()
+ }
+
+ fn coord(&self, i: usize) -> T {
+ self[i]
+ }
+}
+
+/// Blanket [Coordinates] implementation for references.
+impl<T: ?Sized + Coordinates> Coordinates for &T {
+ type Value = T::Value;
+
+ fn dims(&self) -> usize {
+ (*self).dims()
+ }
+
+ fn coord(&self, i: usize) -> Self::Value {
+ (*self).coord(i)
+ }
+}
+
+/// Types that support computing distances to raw slices of coordinates.
+pub trait CoordinateProximity<T> {
+ type Distance: Distance;
+
+ /// Compute the distance to a point specified by its coordinates.
+ fn distance_to_coords(&self, coords: &[T]) -> Self::Distance;
+}
+
+/// Blanket [CoordinateProximity] implementation for references.
+impl<T: CoordinateProximity<U>, U> CoordinateProximity<U> for &T {
+ type Distance = T::Distance;
+
+ fn distance_to_coords(&self, coords: &[U]) -> Self::Distance {
+ (*self).distance_to_coords(coords)
+ }
+}
+
+/// Marker trait for coordinate proximities that are [metrics][crate::distance::Metric].
+pub trait CoordinateMetric<T>: CoordinateProximity<T> {}
+
+/// Blanket [CoordinateMetric] implementation for references.
+impl<T: CoordinateMetric<U>, U> CoordinateMetric<U> for &T {}
diff --git a/src/lib.rs b/src/lib.rs
index 8b0c32b..1532211 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -2,6 +2,8 @@
//!
//! [nearest neighbor search]: https://en.wikipedia.org/wiki/Nearest_neighbor_search
+pub mod coords;
pub mod distance;
+pub use coords::Coordinates;
pub use distance::{Distance, Metric, Proximity};