vista 0.0.2

A utility library for displaying and formatting arrays, matrices and tensors in the terminal.
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

//! ## `DisplayExt`
//!
//! This module provides the `DisplayExt` trait, which is used to add a `display` convenience method to `ndarray` arrays.

use ndarray::{ArrayBase, RawData};

use crate::{DisplayArray, display_method::DisplayMethod};

/// Common trait for displayable arrays.
pub trait DisplayExt<T: RawData, D> {
    /// Construct a `DisplayArray` type for the given data.
    fn display<M: DisplayMethod>(&self) -> DisplayArray<'_, T, D, M>;
}

impl<T: RawData, D> DisplayExt<T, D> for ArrayBase<T, D> {
    #[inline]
    fn display<M: DisplayMethod>(&self) -> DisplayArray<'_, T, D, M> {
        DisplayArray::new(vec![&self])
    }
}

impl<T: RawData, D, const N: usize> DisplayExt<T, D> for [&ArrayBase<T, D>; N] {
    #[inline]
    fn display<M: DisplayMethod>(&self) -> DisplayArray<'_, T, D, M> {
        DisplayArray::new(self.to_vec())
    }
}

impl<T: RawData, D> DisplayExt<T, D> for &[&ArrayBase<T, D>] {
    #[inline]
    fn display<M: DisplayMethod>(&self) -> DisplayArray<'_, T, D, M> {
        DisplayArray::new(self.to_vec())
    }
}