Merge pull request #28 from alceal:feature/export_to-json

feat: Convert plots into JSON with the `to_json`method

Cargo.toml

@@ -21,3 +21,4 @@ polars = { version = "0.44.2", features = [
     "strings",
 ] }
 serde = "1.0.214"
+serde_json = "1.0.132"

src/common/plot.rs

@@ -1,13 +1,16 @@
 use std::env;
 
 use plotly::{Layout, Plot as Plotly, Trace};
+use serde::Serialize;
 
 /// A trait representing a generic plot that can be displayed or rendered.
 pub trait Plot {
     fn plot(&self);
 
     fn write_html(&self, path: impl Into<String>);
 
+    fn to_json(&self) -> Result<String, serde_json::Error>;
+
     // fn write_image(
     //     &self,
     //     path: impl Into<String>,
@@ -33,7 +36,7 @@ pub(crate) trait PlotHelper {
 // Implement the public trait `Plot` for any type that implements `PlotHelper`.
 impl<T> Plot for T
 where
-    T: PlotHelper,
+    T: PlotHelper + Serialize + Clone,
 {
     fn plot(&self) {
         let mut plot = Plotly::new();
@@ -53,6 +56,10 @@ where
         plot.write_html(path.into());
     }
 
+    fn to_json(&self) -> Result<String, serde_json::Error> {
+        serde_json::to_string(self)
+    }
+
     // fn write_image(
     //     &self,
     //     path: impl Into<String>,

src/plots/barplot.rs

@@ -7,6 +7,7 @@ use plotly::{
 };
 
 use polars::frame::DataFrame;
+use serde::Serialize;
 
 use crate::{
     common::{Layout, Marker, PlotHelper, Polar},
@@ -105,6 +106,7 @@ use crate::{
 /// ```
 ///
 /// ![Example](https://imgur.com/2alZlO5.png)
+#[derive(Clone, Serialize)]
 pub struct BarPlot {
     traces: Vec<Box<dyn Trace + 'static>>,
     layout: LayoutPlotly,

src/plots/boxplot.rs

@@ -6,6 +6,7 @@ use plotly::{
 };
 
 use polars::frame::DataFrame;
+use serde::Serialize;
 
 use crate::{
     common::{Layout, Marker, PlotHelper, Polar},
@@ -106,6 +107,7 @@ use crate::{
 /// ```
 ///
 /// ![Example](https://imgur.com/uj1LY90.png)
+#[derive(Clone, Serialize)]
 pub struct BoxPlot {
     traces: Vec<Box<dyn Trace + 'static>>,
     layout: LayoutPlotly,

src/plots/heatmap.rs

@@ -70,6 +70,7 @@ use crate::{
 /// ```
 ///
 /// ![Example](https://imgur.com/5uFih4M.png)
+#[derive(Clone, Serialize)]
 pub struct HeatMap {
     pub traces: Vec<Box<dyn Trace + 'static>>,
     pub layout: LayoutPlotly,

src/plots/histogram.rs

@@ -6,6 +6,7 @@ use plotly::{
 };
 
 use polars::frame::DataFrame;
+use serde::Serialize;
 
 use crate::{
     common::{Layout, Marker, PlotHelper, Polar},
@@ -98,6 +99,7 @@ use crate::{
 /// ```
 ///
 /// ![Example](https://imgur.com/w2oiuIo.png)
+#[derive(Clone, Serialize)]
 pub struct Histogram {
     traces: Vec<Box<dyn Trace + 'static>>,
     layout: LayoutPlotly,

src/plots/lineplot.rs

@@ -9,6 +9,7 @@ use polars::{
     frame::DataFrame,
     prelude::{col, IntoLazy},
 };
+use serde::Serialize;
 
 use crate::{
     common::{Layout, Line, Marker, PlotHelper, Polar},
@@ -109,6 +110,7 @@ use crate::{
 /// ```
 ///
 /// ![Example](https://imgur.com/PaXG300.png)
+#[derive(Clone, Serialize)]
 pub struct LinePlot {
     traces: Vec<Box<dyn Trace + 'static>>,
     layout: LayoutPlotly,

src/plots/scatter3dplot.rs

@@ -6,6 +6,7 @@ use plotly::{
 };
 
 use polars::frame::DataFrame;
+use serde::Serialize;
 
 use crate::{
     common::{Layout, Marker, PlotHelper, Polar},
@@ -89,6 +90,7 @@ use crate::{
 /// ```
 ///
 /// ![Example](https://imgur.com/BXlxKfg.png)
+#[derive(Clone, Serialize)]
 pub struct Scatter3dPlot {
     traces: Vec<Box<dyn Trace + 'static>>,
     layout: LayoutPlotly,

src/plots/scatterplot.rs

@@ -6,6 +6,7 @@ use plotly::{
 };
 
 use polars::frame::DataFrame;
+use serde::Serialize;
 
 use crate::{
     common::{Layout, Marker, PlotHelper, Polar},
@@ -105,6 +106,7 @@ use crate::{
 /// ```
 ///
 /// ![Example](https://imgur.com/9jfO8RU.png)
+#[derive(Clone, Serialize)]
 pub struct ScatterPlot {
     traces: Vec<Box<dyn Trace + 'static>>,
     layout: LayoutPlotly,

src/plots/timeseriesplot.rs

@@ -9,6 +9,7 @@ use polars::{
     frame::DataFrame,
     prelude::{col, IntoLazy},
 };
+use serde::Serialize;
 
 use crate::{
     common::{Layout, Line, Marker, PlotHelper, Polar},
@@ -89,6 +90,7 @@ use crate::{
 /// ```
 ///
 /// ![Example](https://imgur.com/1GaGFbk.png)
+#[derive(Clone, Serialize)]
 pub struct TimeSeriesPlot {
     traces: Vec<Box<dyn Trace + 'static>>,
     layout: LayoutPlotly,