chart_js_rs/
traits.rs

1use crate::objects::*;
2use serde::Serialize;
3use serde_json::Value;
4use std::{collections::*, fmt::Display};
5
6pub(crate) trait ChartJsRsObject {}
7impl<T: Display> ChartJsRsObject for T {}
8
9pub trait DatasetTrait: Serialize + Default + Clone {
10    fn labels(self) -> Vec<NumberOrDateString>;
11}
12pub trait DatasetDataExt {
13    fn presorted_to_dataset_data(self) -> DatasetData;
14    fn unsorted_to_dataset_data(self) -> DatasetData;
15}
16
17impl<I> DatasetDataExt for I
18where
19    I: Iterator<Item = (NumberOrDateString, NumberString, Option<Value>)>,
20{
21    fn presorted_to_dataset_data(self) -> DatasetData {
22        DatasetData(serde_json::to_value(self.map(XYPoint::from).collect::<Vec<_>>()).unwrap())
23    }
24    fn unsorted_to_dataset_data(self) -> DatasetData {
25        DatasetData(serde_json::to_value(self.map(XYPoint::from).collect::<BTreeSet<_>>()).unwrap())
26    }
27}
28
29pub trait DatasetIterExt: Iterator {
30    fn into_data_iter<X, Y>(
31        self,
32    ) -> impl Iterator<Item = (NumberOrDateString, NumberString, Option<Value>)>
33    where
34        Self: Iterator<Item = (X, Y)> + Sized,
35        X: Into<NumberOrDateString>,
36        Y: Into<NumberString>,
37    {
38        self.map(|(x, y)| (x.into(), y.into(), None))
39    }
40    fn into_data_iter_with_description<X, Y, D>(
41        self,
42    ) -> impl Iterator<Item = (NumberOrDateString, NumberString, Option<Value>)>
43    where
44        Self: Iterator<Item = (X, Y, D)> + Sized,
45        X: Into<NumberOrDateString>,
46        Y: Into<NumberString>,
47        D: Serialize,
48    {
49        self.map(|(x, y, d)| (x.into(), y.into(), Some(serde_json::to_value(d).unwrap())))
50    }
51}
52impl<T> DatasetIterExt for T where T: Iterator + ?Sized {}
53
54pub trait Annotation: Serialize + Default {}