Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add Table trace. #165

Closed
wants to merge 8 commits into from
Closed
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
15 changes: 15 additions & 0 deletions examples/basic_charts/src/main.rs
Original file line number Diff line number Diff line change
Expand Up @@ -10,6 +10,7 @@ use plotly::{
layout::{Axis, BarMode, Layout, Legend, TicksDirection, TraceOrder},
sankey::{Line as SankeyLine, Link, Node},
Bar, Plot, Sankey, Scatter, ScatterPolar,
Table,
};
use rand_distr::{Distribution, Normal, Uniform};

Expand Down Expand Up @@ -604,6 +605,19 @@ fn basic_sankey_diagram() {
plot.show();
}

fn table_chart() {
let trace = Table::new(
vec![String::from("col1"), String::from("col2")],
vec![
vec![1, 2],
vec![2, 3],
]
);
let mut plot = Plot::new();
plot.add_trace(table_trace);
Copy link
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

@baiguoname There is a typo here. This should be trace

plot.show();
}

fn main() {
// Uncomment any of these lines to display the example.

Expand All @@ -629,6 +643,7 @@ fn main() {
// basic_bar_chart();
// grouped_bar_chart();
// stacked_bar_chart();
// table_chart();

// Sankey Diagrams
// basic_sankey_diagram();
Expand Down
1 change: 1 addition & 0 deletions plotly/src/common/mod.rs
Original file line number Diff line number Diff line change
Expand Up @@ -201,6 +201,7 @@ pub enum PlotType {
Ohlc,
Sankey,
Surface,
Table,
}

#[derive(Serialize, Clone, Debug)]
Expand Down
2 changes: 1 addition & 1 deletion plotly/src/lib.rs
Original file line number Diff line number Diff line change
Expand Up @@ -34,7 +34,7 @@ pub use traces::{box_plot, contour, histogram, image, mesh3d, sankey, scatter_ma
// Bring the different trace types into the top-level scope
pub use traces::{
Bar, BoxPlot, Candlestick, Contour, HeatMap, Histogram, Image, Mesh3D, Ohlc, Sankey, Scatter,
Scatter3D, ScatterMapbox, ScatterPolar, Surface,
Scatter3D, ScatterMapbox, ScatterPolar, Surface, Table,
};

pub trait Restyle: serde::Serialize {}
Expand Down
2 changes: 2 additions & 0 deletions plotly/src/traces/mod.rs
Original file line number Diff line number Diff line change
Expand Up @@ -15,6 +15,7 @@ mod scatter3d;
pub mod scatter_mapbox;
mod scatter_polar;
pub mod surface;
pub mod table;

pub use bar::Bar;
pub use box_plot::BoxPlot;
Expand All @@ -30,5 +31,6 @@ pub use scatter3d::Scatter3D;
pub use scatter_mapbox::ScatterMapbox;
pub use scatter_polar::ScatterPolar;
pub use surface::Surface;
pub use table::Table;

pub use self::image::Image;
172 changes: 172 additions & 0 deletions plotly/src/traces/table.rs
Original file line number Diff line number Diff line change
@@ -0,0 +1,172 @@
//! Table trace

use plotly_derive::FieldSetter;
use serde::Serialize;

use crate::{
color::Color,
common::{Font, Line, PlotType, Visible},
Trace,
};

#[serde_with::skip_serializing_none]
#[derive(Serialize, Clone, Debug, FieldSetter)]
#[field_setter(box_self, kind = "trace")]
pub struct Table<T, N>
where
T: Serialize + Clone + 'static,
N: Serialize + Clone + 'static,
{
#[field_setter(default = "PlotType::Table")]
r#type: PlotType,
/// Sets the trace name. The trace name appear as the legend item and on
/// hover.
name: Option<String>,
#[serde(rename = "columnorder")]
/// Determines whether or not this trace is visible. If
/// `Visible::LegendOnly`, the trace is not drawn, but can appear as a
/// legend item (provided that the legend itself is visible).
visible: Option<Visible>,
///Specifies the rendered order of the data columns; for example, a value `2` at position `0`,
///means that column index `0` in the data will be rendered as the,
///third column, as columns have an index base of zero.
column_order: Option<Vec<usize>>,
#[serde(rename = "columnwidth")]
///The width of columns expressed as a ratio. Columns fill the available width,
///in proportion of their specified column widths.
column_width: Option<f64>,
///Header cell values. `values[m][n]` represents the value of the `n`th point in column `m`,,
///therefore the `values[m]` vector length for all columns must be the same (longer vectors,
///will be truncated). Each value must be a finite number or a string.
header: Option<Header<T>>,
///Cell values. `values[m][n]` represents the value of the `n`th point in column `m`,,
///therefore the `values[m]` vector length for all columns must be the same (longer vectors,
///will be truncated). Each value must be a finite number or a string.
cells: Option<Cells<N>>,
}

impl<T, N> Table<T, N>
where
T: Serialize + Clone + Default + 'static,
N: Serialize + Clone + Default + 'static,
{
pub fn new(header: Header<T>, cells: Cells<N>) -> Box<Self> {
Box::new(Table {
header: Some(header),
cells: Some(cells),
..Default::default()
})
}
}

impl<T, N> Trace for Table<T, N>
where
T: Serialize + Clone + 'static,
N: Serialize + Clone + 'static,
{
fn to_json(&self) -> String {
serde_json::to_string(self).unwrap()
}
}

#[serde_with::skip_serializing_none]
#[derive(Serialize, Clone, Debug, FieldSetter)]
pub struct Cells<N> {
///Cell values. `values[m][n]` represents the value of the `n`th point in column `m`,
///therefore the `values[m]` vector length for all columns must be the same (longer vectors,
///will be truncated). Each value must be a finite number or a string
values: Option<Vec<Vec<N>>>,
///Prefix for cell values.
prefix: Option<String>,
///Suffix for cell values.
suffix: Option<String>,
height: Option<f64>,
align: Option<String>,
line: Option<Line>,
///Sets the cell fill color. It accepts either a specific color,
///or an array of colors or a 2D array of colors
fill: Option<Fill>,
font: Option<Font>,
}

impl<N> Cells<N>
where
N: Serialize + Clone + Default + 'static,
{
pub fn new(values: Vec<Vec<N>>) -> Self {
Cells {
values: Some(values),
..Default::default()
}
}
}

#[serde_with::skip_serializing_none]
#[derive(Serialize, Clone, Debug, FieldSetter)]
pub struct Header<T> {
///Header cell values. `values[m][n]` represents the value of the `n`th point in column `m`,
///therefore the `values[m]` vector length for all columns must be the same (longer vectors,
///will be truncated). Each value must be a finite number or a string.
values: Option<Vec<T>>,
///Prefix for cell values.
prefix: Option<String>,
///Suffix for cell values.
suffix: Option<String>,
height: Option<f64>,
align: Option<String>,
line: Option<Line>,
///Sets the cell fill color. It accepts either a specific color,
///or an array of colors or a 2D array of colors
fill: Option<Fill>,
font: Option<Font>,
}

impl<T> Header<T>
where
T: Serialize + Clone + Default + 'static,
{
pub fn new(values: Vec<T>) -> Self {
Header {
values: Some(values),
..Default::default()
}
}
}

#[serde_with::skip_serializing_none]
#[derive(Serialize, Clone, Debug, FieldSetter)]
pub struct Fill {
color: Option<Box<dyn Color>>,
}

impl Fill {
pub fn new() -> Self {
Default::default()
}
}

#[cfg(test)]
mod tests {
use serde_json::{json, to_value};

use super::*;

#[test]
fn test_serialize_table() {
let columns = Header::new(vec![String::from("col1"), String::from("col2")]);
let values = Cells::new(vec![vec![1, 2], vec![2, 3]]);
let trace = Table::new(columns, values);

let expected = json!({
"type": "table",
"cells": {
"values": [[1, 2], [2, 3]],
},
"header": {
"values": ["col1", "col2"],
},
});

assert_eq!(to_value(trace).unwrap(), expected);
}
}