use super::{Plot3D, Plot3DError, validate_lengths, validate_multiple, validate_nonempty};
use crate::{Plot3DUi, Triangle3DFlags};
pub struct Triangles3D<'a> {
pub label: &'a str,
pub xs_f32: Option<&'a [f32]>,
pub ys_f32: Option<&'a [f32]>,
pub zs_f32: Option<&'a [f32]>,
pub xs_f64: Option<&'a [f64]>,
pub ys_f64: Option<&'a [f64]>,
pub zs_f64: Option<&'a [f64]>,
pub flags: Triangle3DFlags,
pub offset: i32,
pub stride: i32,
pub points_f32: Option<&'a [[f32; 3]]>,
pub points_f64: Option<&'a [[f64; 3]]>,
}
impl<'a> Triangles3D<'a> {
pub fn f32(label: &'a str, xs: &'a [f32], ys: &'a [f32], zs: &'a [f32]) -> Self {
Self {
label,
xs_f32: Some(xs),
ys_f32: Some(ys),
zs_f32: Some(zs),
xs_f64: None,
ys_f64: None,
zs_f64: None,
flags: Triangle3DFlags::NONE,
offset: 0,
stride: 0,
points_f32: None,
points_f64: None,
}
}
pub fn f64(label: &'a str, xs: &'a [f64], ys: &'a [f64], zs: &'a [f64]) -> Self {
Self {
label,
xs_f32: None,
ys_f32: None,
zs_f32: None,
xs_f64: Some(xs),
ys_f64: Some(ys),
zs_f64: Some(zs),
flags: Triangle3DFlags::NONE,
offset: 0,
stride: 0,
points_f32: None,
points_f64: None,
}
}
pub fn points_f32(label: &'a str, pts: &'a [[f32; 3]]) -> Self {
Self {
label,
xs_f32: None,
ys_f32: None,
zs_f32: None,
xs_f64: None,
ys_f64: None,
zs_f64: None,
flags: Triangle3DFlags::NONE,
offset: 0,
stride: 0,
points_f32: Some(pts),
points_f64: None,
}
}
pub fn points_f64(label: &'a str, pts: &'a [[f64; 3]]) -> Self {
Self {
label,
xs_f32: None,
ys_f32: None,
zs_f32: None,
xs_f64: None,
ys_f64: None,
zs_f64: None,
flags: Triangle3DFlags::NONE,
offset: 0,
stride: 0,
points_f32: None,
points_f64: Some(pts),
}
}
pub fn flags(mut self, flags: Triangle3DFlags) -> Self {
self.flags = flags;
self
}
pub fn offset(mut self, o: i32) -> Self {
self.offset = o;
self
}
pub fn stride(mut self, s: i32) -> Self {
self.stride = s;
self
}
}
impl<'a> Plot3D for Triangles3D<'a> {
fn label(&self) -> &str {
self.label
}
fn try_plot(&self, ui: &Plot3DUi<'_>) -> Result<(), Plot3DError> {
if let Some(pts) = self.points_f32 {
validate_nonempty(pts)?;
validate_multiple(pts.len(), 3, "triangles(points)")?;
let mut xs = Vec::with_capacity(pts.len());
let mut ys = Vec::with_capacity(pts.len());
let mut zs = Vec::with_capacity(pts.len());
for p in pts {
xs.push(p[0]);
ys.push(p[1]);
zs.push(p[2]);
}
ui.plot_triangles_f32(self.label, &xs, &ys, &zs, self.flags);
return Ok(());
}
if let Some(pts) = self.points_f64 {
validate_nonempty(pts)?;
validate_multiple(pts.len(), 3, "triangles(points)")?;
let mut xs = Vec::with_capacity(pts.len());
let mut ys = Vec::with_capacity(pts.len());
let mut zs = Vec::with_capacity(pts.len());
for p in pts {
xs.push(p[0]);
ys.push(p[1]);
zs.push(p[2]);
}
ui.plot_triangles_f64(self.label, &xs, &ys, &zs, self.flags);
return Ok(());
}
if let (Some(x), Some(y), Some(z)) = (self.xs_f32, self.ys_f32, self.zs_f32) {
validate_nonempty(x)?;
validate_lengths(x, y, "x/y")?;
validate_lengths(y, z, "y/z")?;
validate_multiple(x.len(), 3, "triangles")?;
ui.plot_triangles_f32_raw(self.label, x, y, z, self.flags, self.offset, self.stride);
return Ok(());
}
if let (Some(x), Some(y), Some(z)) = (self.xs_f64, self.ys_f64, self.zs_f64) {
validate_nonempty(x)?;
validate_lengths(x, y, "x/y")?;
validate_lengths(y, z, "y/z")?;
validate_multiple(x.len(), 3, "triangles")?;
ui.plot_triangles_f64_raw(self.label, x, y, z, self.flags, self.offset, self.stride);
return Ok(());
}
Err(Plot3DError::EmptyData)
}
}