94 lines
2.3 KiB
Rust
94 lines
2.3 KiB
Rust
use std::ops::{Deref, DerefMut, Range};
|
|
|
|
use evenio::component::Component;
|
|
use nalgebra::{Rotation3, Vector3};
|
|
use ndarray::{Array3, ArrayBase, Dim, SliceArg};
|
|
|
|
use crate::client::render::voxel::VoxelColor;
|
|
|
|
#[derive(Debug, Component, Default)]
|
|
pub struct Pos(pub Vector3<f32>);
|
|
#[derive(Debug, Component, Default)]
|
|
pub struct Orientation(pub Rotation3<f32>);
|
|
|
|
pub type VoxelGrid = TrackedGrid<VoxelColor>;
|
|
pub type GridRegion = (Range<usize>, Range<usize>, Range<usize>);
|
|
#[derive(Debug, Clone, Component)]
|
|
pub struct TrackedGrid<T> {
|
|
data: Array3<T>,
|
|
changes: Vec<GridRegion>,
|
|
}
|
|
|
|
impl<T> TrackedGrid<T> {
|
|
pub fn new(data: Array3<T>) -> Self {
|
|
Self {
|
|
data,
|
|
changes: Vec::new(),
|
|
}
|
|
}
|
|
pub fn view_slice_mut<I: SliceArg<Dim<[usize; 3]>>>(
|
|
&mut self,
|
|
slice: I,
|
|
) -> ArrayBase<ndarray::ViewRepr<&mut T>, <I as SliceArg<Dim<[usize; 3]>>>::OutDim> {
|
|
self.data.slice_mut(slice)
|
|
}
|
|
pub fn take_changes(&mut self) -> Vec<GridRegion> {
|
|
std::mem::take(&mut self.changes)
|
|
}
|
|
}
|
|
|
|
impl Pos {
|
|
pub fn new(x: f32, y: f32, z: f32) -> Self {
|
|
Self(Vector3::new(x, y, z))
|
|
}
|
|
}
|
|
impl Orientation {
|
|
pub fn from_axis_angle<SB: nalgebra::Storage<f32, nalgebra::Const<3>>>(
|
|
axis: &nalgebra::Unit<nalgebra::Matrix<f32, nalgebra::Const<3>, nalgebra::Const<1>, SB>>,
|
|
angle: f32,
|
|
) -> Self {
|
|
Self(Rotation3::from_axis_angle(axis, angle))
|
|
}
|
|
}
|
|
impl From<Vector3<f32>> for Pos {
|
|
fn from(val: Vector3<f32>) -> Self {
|
|
Pos(val)
|
|
}
|
|
}
|
|
impl From<Rotation3<f32>> for Orientation {
|
|
fn from(val: Rotation3<f32>) -> Self {
|
|
Orientation(val)
|
|
}
|
|
}
|
|
|
|
// reref boilerplate :pensive:
|
|
|
|
impl Deref for Pos {
|
|
type Target = Vector3<f32>;
|
|
fn deref(&self) -> &Self::Target {
|
|
&self.0
|
|
}
|
|
}
|
|
impl DerefMut for Pos {
|
|
fn deref_mut(&mut self) -> &mut Self::Target {
|
|
&mut self.0
|
|
}
|
|
}
|
|
impl Deref for Orientation {
|
|
type Target = Rotation3<f32>;
|
|
fn deref(&self) -> &Self::Target {
|
|
&self.0
|
|
}
|
|
}
|
|
impl DerefMut for Orientation {
|
|
fn deref_mut(&mut self) -> &mut Self::Target {
|
|
&mut self.0
|
|
}
|
|
}
|
|
impl<T> Deref for TrackedGrid<T> {
|
|
type Target = Array3<T>;
|
|
fn deref(&self) -> &Self::Target {
|
|
&self.data
|
|
}
|
|
}
|