use crate::core::ElementPlacement;
use super::data::{Chart, ChartSeries};
use super::types::ChartType;
pub struct ChartBuilder {
title: String,
chart_type: ChartType,
categories: Vec<String>,
series: Vec<ChartSeries>,
placement: ElementPlacement,
}
impl ChartBuilder {
pub fn new(title: &str, chart_type: ChartType) -> Self {
ChartBuilder {
title: title.to_string(),
chart_type,
categories: Vec::new(),
series: Vec::new(),
placement: ElementPlacement::chart_defaults(),
}
}
pub fn position(mut self, x: u32, y: u32) -> Self {
self.placement.set_position(x, y);
self
}
pub fn size(mut self, width: u32, height: u32) -> Self {
self.placement.set_size(width, height);
self
}
pub fn categories(mut self, categories: Vec<&str>) -> Self {
self.categories = categories.into_iter().map(|c| c.to_string()).collect();
self
}
pub fn add_series(mut self, series: ChartSeries) -> Self {
self.series.push(series);
self
}
pub fn build(self) -> Chart {
Chart {
title: self.title,
chart_type: self.chart_type,
categories: self.categories,
series: self.series,
x: self.placement.x,
y: self.placement.y,
width: self.placement.width,
height: self.placement.height,
}
}
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn test_chart_builder() {
let chart = ChartBuilder::new("Revenue", ChartType::Bar)
.categories(vec!["Q1", "Q2", "Q3"])
.add_series(ChartSeries::new("2023", vec![100.0, 150.0, 200.0]))
.add_series(ChartSeries::new("2024", vec![120.0, 180.0, 220.0]))
.position(100000, 200000)
.size(4000000, 3000000)
.build();
assert_eq!(chart.title, "Revenue");
assert_eq!(chart.chart_type, ChartType::Bar);
assert_eq!(chart.category_count(), 3);
assert_eq!(chart.series_count(), 2);
assert_eq!(chart.x, 100000);
assert_eq!(chart.y, 200000);
}
}