Add label to scatter.rs

yuankunzhang · Oct 11, 2024 · 196fb3f · 196fb3f
2 parents 371ac0e + 77e1130
commit 196fb3f
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/charming/src/series/scatter.rs b/charming/src/series/scatter.rs
@@ -3,7 +3,7 @@ use serde::Serialize;
 use crate::{
     datatype::{DataFrame, DataPoint},
     element::{
-        ColorBy, CoordinateSystem, DimensionEncode, Emphasis, ItemStyle, MarkArea, MarkLine,
+        ColorBy, CoordinateSystem, DimensionEncode, Emphasis, ItemStyle, Label, MarkArea, MarkLine,
         Symbol, SymbolSize,
     },
 };
@@ -23,6 +23,9 @@ pub struct Scatter {
     #[serde(skip_serializing_if = "Option::is_none")]
     color_by: Option<ColorBy>,
 
+    #[serde(skip_serializing_if = "Option::is_none")]
+    label: Option<Label>,
+
     #[serde(skip_serializing_if = "Option::is_none")]
     dataset_index: Option<f64>,
 
@@ -73,6 +76,7 @@ impl Scatter {
             id: None,
             name: None,
             color_by: None,
+            label: None,
             dataset_index: None,
             coordinate_system: None,
             x_axis_index: None,
@@ -103,6 +107,11 @@ impl Scatter {
         self
     }
 
+    pub fn label<L: Into<Label>>(mut self, label: L) -> Self {
+        self.label = Some(label.into());
+        self
+    }
+
     pub fn dataset_index<F: Into<f64>>(mut self, dataset_index: F) -> Self {
         self.dataset_index = Some(dataset_index.into());
         self