1#[cfg(all(feature = "plotters", not(target_arch = "wasm32")))]
2use plotters::prelude::IntoDrawingArea;
3
4use crate::aes::Aes;
5use crate::annotate::Annotation;
6use crate::build::PlotBuilder;
7use crate::coord::cartesian::CoordCartesian;
8use crate::coord::fixed::CoordFixed;
9use crate::coord::flip::CoordFlip;
10use crate::coord::polar::CoordPolar;
11use crate::coord::Coord;
12use crate::data::{DataFrame, GGData, Value};
13use crate::facet::{Facet, FacetLabeller, FacetScales, FacetSpace};
14use crate::geom::area::GeomArea;
15use crate::geom::bar::GeomBar;
16use crate::geom::bin2d::GeomBin2d;
17use crate::geom::blank::GeomBlank;
18use crate::geom::boxplot::GeomBoxplot;
19use crate::geom::col::GeomCol;
20use crate::geom::contour::GeomContour;
21use crate::geom::count::GeomCount;
22use crate::geom::crossbar::GeomCrossbar;
23use crate::geom::curve::GeomCurve;
24use crate::geom::density::GeomDensity;
25use crate::geom::density2d::GeomDensity2d;
26use crate::geom::dotplot::GeomDotplot;
27use crate::geom::errorbar::GeomErrorbar;
28use crate::geom::freqpoly::GeomFreqpoly;
29use crate::geom::hex::GeomHex;
30use crate::geom::histogram::GeomHistogram;
31use crate::geom::jitter::GeomJitter;
32use crate::geom::line::GeomLine;
33use crate::geom::linerange::GeomLinerange;
34use crate::geom::path::GeomPath;
35use crate::geom::point::GeomPoint;
36use crate::geom::pointrange::GeomPointrange;
37use crate::geom::polygon::GeomPolygon;
38use crate::geom::qq::{GeomQQ, GeomQQLine};
39use crate::geom::rect::GeomRect;
40use crate::geom::refline::{GeomAbline, GeomHline, GeomVline};
41use crate::geom::ribbon::GeomRibbon;
42use crate::geom::rug::GeomRug;
43use crate::geom::segment::GeomSegment;
44use crate::geom::smooth::GeomSmooth;
45use crate::geom::spoke::GeomSpoke;
46use crate::geom::step::GeomStep;
47use crate::geom::text::{GeomLabel, GeomText};
48use crate::geom::tile::GeomTile;
49use crate::geom::violin::GeomViolin;
50use crate::geom::{Geom, GeomParams};
51use crate::position::Position;
52use crate::render::layout::PlotLayout;
53#[cfg(all(feature = "plotters", not(target_arch = "wasm32")))]
54use crate::render::plotters_backend::PlottersAdapter;
55use crate::render::renderer::PlotRenderer;
56use crate::render::RenderError;
57use crate::scale::continuous::ScaleContinuous;
58use crate::scale::transform::ScaleTransform;
59use crate::scale::Scale;
60use crate::stat::Stat;
61use crate::theme::Theme;
62
63mod stat_layers;
66
67#[derive(Clone, Debug, Default)]
69pub struct Labels {
70 pub title: Option<String>,
71 pub subtitle: Option<String>,
72 pub x: Option<String>,
73 pub y: Option<String>,
74 pub caption: Option<String>,
75 pub tag: Option<String>,
77}
78
79pub struct Layer {
81 pub data: Option<DataFrame>,
82 pub mapping: Aes,
83 pub geom: Box<dyn Geom>,
84 pub stat: Box<dyn Stat>,
85 pub position: Box<dyn Position>,
86 pub params: GeomParams,
87 pub show_legend: Option<bool>,
88 pub explicit_style: bool,
91 pub tooltip_template: Option<String>,
94}
95
96pub struct GGPlot {
98 pub(crate) data: DataFrame,
99 pub(crate) mapping: Aes,
100 pub(crate) layers: Vec<Layer>,
101 pub(crate) scales: Vec<Box<dyn Scale>>,
102 pub(crate) coord: Box<dyn Coord>,
103 pub(crate) theme: Theme,
104 pub(crate) labels: Labels,
105 pub(crate) facet: Facet,
106 pub(crate) annotations: Vec<Annotation>,
107 pub(crate) guide_legend: crate::guide::config::GuideLegend,
108 pub(crate) warnings: Vec<String>,
111 pub(crate) default_aspect_ratio: Option<f64>,
114 pub(crate) locale: Option<crate::locale::Locale>,
116 pub(crate) svg_root_data: bool,
121 pub(crate) root_attrs: Vec<(String, String)>,
124 pub(crate) svg_warnings_attr: bool,
128 pub(crate) svg_formatted_values: bool,
132}
133
134impl GGPlot {
135 pub fn new(data: impl GGData) -> Self {
142 GGPlot {
143 data: data.into_dataframe(),
144 mapping: Aes::default(),
145 layers: Vec::new(),
146 scales: Vec::new(),
147 coord: Box::new(CoordCartesian::new()),
148 theme: Theme::default(),
149 labels: Labels::default(),
150 facet: Facet::default(),
151 annotations: Vec::new(),
152 guide_legend: crate::guide::config::GuideLegend::default(),
153 warnings: Vec::new(),
154 default_aspect_ratio: None,
155 locale: None,
156 svg_root_data: true,
157 root_attrs: Vec::new(),
158 svg_warnings_attr: false,
159 svg_formatted_values: false,
160 }
161 }
162
163 pub fn aes(mut self, mapping: Aes) -> Self {
165 self.warnings.extend(mapping.warnings.iter().cloned());
166 self.mapping = mapping;
167 self
168 }
169
170 pub fn geom_point(self) -> Self {
173 self.add_geom(GeomPoint::default())
174 }
175
176 pub fn geom_point_with(self, geom: GeomPoint) -> Self {
177 self.add_geom_with(geom)
178 }
179
180 pub fn geom_line(self) -> Self {
181 self.add_geom(GeomLine::default())
182 }
183
184 pub fn geom_line_with(self, geom: GeomLine) -> Self {
185 self.add_geom_with(geom)
186 }
187
188 pub fn geom_bar(self) -> Self {
189 self.add_geom(GeomBar::default())
190 }
191
192 pub fn geom_bar_with(self, geom: GeomBar) -> Self {
193 self.add_geom_with(geom)
194 }
195
196 pub fn geom_histogram(self) -> Self {
197 self.add_geom(GeomHistogram::default())
198 }
199
200 pub fn geom_histogram_with(self, geom: GeomHistogram) -> Self {
201 self.add_geom_with(geom)
202 }
203
204 pub fn geom_boxplot(self) -> Self {
205 self.add_geom(GeomBoxplot::default())
206 }
207
208 pub fn geom_boxplot_with(self, geom: GeomBoxplot) -> Self {
209 self.add_geom_with(geom)
210 }
211
212 pub fn geom_smooth(self) -> Self {
213 self.add_geom(GeomSmooth::default())
214 }
215
216 pub fn geom_smooth_with(self, geom: GeomSmooth) -> Self {
217 self.add_geom_with(geom)
218 }
219
220 pub fn geom_col(self) -> Self {
221 self.add_geom(GeomCol::default())
222 }
223
224 pub fn geom_col_with(self, geom: GeomCol) -> Self {
225 self.add_geom_with(geom)
226 }
227
228 pub fn geom_hline(self, yintercept: f64) -> Self {
232 self.add_refline(
233 GeomHline::new(yintercept),
234 false,
235 &[("yintercept", yintercept)],
236 )
237 }
238
239 pub fn geom_hline_with(self, geom: GeomHline) -> Self {
241 let y = geom.yintercept;
242 self.add_refline(geom, true, &[("yintercept", y)])
243 }
244
245 pub fn geom_hline_aes(self, mapping: Aes) -> Self {
269 self.add_geom(GeomHline::mapped()).layer_aes(mapping)
270 }
271
272 pub fn geom_hline_aes_with(self, geom: GeomHline, mapping: Aes) -> Self {
274 self.add_geom_with(geom).layer_aes(mapping)
275 }
276
277 pub fn geom_vline(self, xintercept: f64) -> Self {
280 self.add_refline(
281 GeomVline::new(xintercept),
282 false,
283 &[("xintercept", xintercept)],
284 )
285 }
286
287 pub fn geom_vline_with(self, geom: GeomVline) -> Self {
289 let x = geom.xintercept;
290 self.add_refline(geom, true, &[("xintercept", x)])
291 }
292
293 pub fn geom_vline_aes(self, mapping: Aes) -> Self {
296 self.add_geom(GeomVline::mapped()).layer_aes(mapping)
297 }
298
299 pub fn geom_vline_aes_with(self, geom: GeomVline, mapping: Aes) -> Self {
301 self.add_geom_with(geom).layer_aes(mapping)
302 }
303
304 pub fn geom_abline(self, slope: f64, intercept: f64) -> Self {
307 self.add_refline(
308 GeomAbline::new(slope, intercept),
309 false,
310 &[("slope", slope), ("intercept", intercept)],
311 )
312 }
313
314 pub fn geom_abline_with(self, geom: GeomAbline) -> Self {
316 let (b, a) = (geom.slope, geom.intercept);
317 self.add_refline(geom, true, &[("slope", b), ("intercept", a)])
318 }
319
320 pub fn geom_abline_aes(self, mapping: Aes) -> Self {
323 self.add_geom(GeomAbline::mapped()).layer_aes(mapping)
324 }
325
326 pub fn geom_abline_aes_with(self, geom: GeomAbline, mapping: Aes) -> Self {
328 self.add_geom_with(geom).layer_aes(mapping)
329 }
330
331 fn add_refline(
334 self,
335 geom: impl Geom + 'static,
336 explicit: bool,
337 values: &[(&str, f64)],
338 ) -> Self {
339 let mut data = DataFrame::new();
340 let mut mapping = Aes::new();
341 for (col, v) in values {
342 data.add_column(col.to_string(), vec![Value::Float(*v)]);
343 mapping = match *col {
344 "xintercept" => mapping.xintercept(col),
345 "yintercept" => mapping.yintercept(col),
346 "slope" => mapping.slope(col),
347 _ => mapping.intercept(col),
348 };
349 }
350 let plot = if explicit {
351 self.add_geom_with(geom)
352 } else {
353 self.add_geom(geom)
354 };
355 plot.layer_aes(mapping).layer_data(data)
356 }
357
358 pub fn geom_text(self) -> Self {
359 self.add_geom(GeomText::default())
360 }
361
362 pub fn geom_text_with(self, geom: GeomText) -> Self {
363 self.add_geom_with(geom)
364 }
365
366 #[cfg(feature = "ggpubr")]
371 pub fn stat_cor(self) -> Self {
372 self.geom_text().stat(crate::stat::cor::StatCor::default())
373 }
374
375 #[cfg(feature = "ggpubr")]
378 pub fn stat_cor_with(self, stat: crate::stat::cor::StatCor) -> Self {
379 self.geom_text().stat(stat)
380 }
381
382 #[cfg(feature = "ggpubr")]
387 pub fn stat_compare_means(self) -> Self {
388 self.geom_text()
389 .stat(crate::stat::compare_means::StatCompareMeans::default())
390 }
391
392 #[cfg(feature = "ggpubr")]
395 pub fn stat_compare_means_with(
396 self,
397 stat: crate::stat::compare_means::StatCompareMeans,
398 ) -> Self {
399 self.geom_text().stat(stat)
400 }
401
402 #[cfg(feature = "ggpubr")]
408 pub fn stat_compare_means_pairwise(self, comparisons: &[(&str, &str)]) -> Self {
409 self.stat_compare_means_pairwise_with(
410 crate::stat::compare_means::CompareMethod::Auto,
411 comparisons,
412 )
413 }
414
415 #[cfg(feature = "ggpubr")]
418 pub fn stat_compare_means_pairwise_with(
419 self,
420 method: crate::stat::compare_means::CompareMethod,
421 comparisons: &[(&str, &str)],
422 ) -> Self {
423 let col_for = |a: crate::aes::Aesthetic| {
425 self.mapping
426 .mappings
427 .iter()
428 .find(|m| m.aesthetic == a)
429 .map(|m| m.column.clone())
430 };
431 let (xcol, ycol) = match (
432 col_for(crate::aes::Aesthetic::X),
433 col_for(crate::aes::Aesthetic::Y),
434 ) {
435 (Some(x), Some(y)) => (x, y),
436 _ => return self,
437 };
438 let (xc, yc) = match (self.data.column(&xcol), self.data.column(&ycol)) {
439 (Some(x), Some(y)) => (x, y),
440 _ => return self,
441 };
442
443 let mut groups: Vec<(String, Vec<f64>)> = Vec::new();
445 let mut ymax = f64::NEG_INFINITY;
446 let mut ymin = f64::INFINITY;
447 for (xv, yv) in xc.iter().zip(yc.iter()) {
448 let y = match yv.as_f64() {
449 Some(y) if y.is_finite() => y,
450 _ => continue,
451 };
452 ymax = ymax.max(y);
453 ymin = ymin.min(y);
454 let key = xv.to_group_key();
455 if let Some(g) = groups.iter_mut().find(|(k, _)| *k == key) {
456 g.1.push(y);
457 } else {
458 groups.push((key, vec![y]));
459 }
460 }
461 if groups.len() < 2 {
462 return self;
463 }
464 let step = if ymax > ymin {
465 (ymax - ymin) * 0.12
466 } else {
467 1.0
468 };
469
470 let (mut xmin_v, mut xmax_v, mut y_v, mut label_v) =
472 (Vec::new(), Vec::new(), Vec::new(), Vec::new());
473 let mut placed = 0usize;
474 for (a, b) in comparisons {
475 let ga = groups.iter().find(|(k, _)| k == a).map(|g| &g.1);
476 let gb = groups.iter().find(|(k, _)| k == b).map(|g| &g.1);
477 let (ga, gb) = match (ga, gb) {
478 (Some(x), Some(y)) if x.len() >= 2 && y.len() >= 2 => (x, y),
479 _ => continue,
480 };
481 let p = match crate::stat::compare_means::pairwise_p(method, ga, gb) {
482 Some(p) => p,
483 None => continue,
484 };
485 xmin_v.push(Value::Str((*a).to_string()));
486 xmax_v.push(Value::Str((*b).to_string()));
487 y_v.push(Value::Float(ymax + step * (placed as f64 + 1.0)));
488 label_v.push(Value::Str(crate::stat::cor::format_p_value(p)));
489 placed += 1;
490 }
491 if xmin_v.is_empty() {
492 return self;
493 }
494
495 let data = vec![
496 ("xmin".to_string(), xmin_v),
497 ("xmax".to_string(), xmax_v),
498 ("y".to_string(), y_v),
499 ("label".to_string(), label_v),
500 ];
501 self.add_geom(crate::geom::bracket::GeomBracket::default())
502 .layer_data(data)
503 .layer_aes(Aes::new().xmin("xmin").xmax("xmax").y("y").label("label"))
504 }
505
506 pub fn geom_bracket(self, xmin: &str, xmax: &str, y: f64, label: &str) -> Self {
511 self.geom_bracket_many(
512 crate::geom::bracket::GeomBracket::default(),
513 &[(xmin, xmax, y, label)],
514 )
515 }
516
517 pub fn geom_bracket_many(
521 self,
522 geom: crate::geom::bracket::GeomBracket,
523 brackets: &[(&str, &str, f64, &str)],
524 ) -> Self {
525 let xmin = brackets
526 .iter()
527 .map(|b| Value::Str(b.0.to_string()))
528 .collect::<Vec<_>>();
529 let xmax = brackets
530 .iter()
531 .map(|b| Value::Str(b.1.to_string()))
532 .collect::<Vec<_>>();
533 let y = brackets
534 .iter()
535 .map(|b| Value::Float(b.2))
536 .collect::<Vec<_>>();
537 let label = brackets
538 .iter()
539 .map(|b| Value::Str(b.3.to_string()))
540 .collect::<Vec<_>>();
541 let data = vec![
542 ("xmin".to_string(), xmin),
543 ("xmax".to_string(), xmax),
544 ("y".to_string(), y),
545 ("label".to_string(), label),
546 ];
547 self.add_geom(geom)
548 .layer_data(data)
549 .layer_aes(Aes::new().xmin("xmin").xmax("xmax").y("y").label("label"))
550 }
551
552 pub fn geom_bracket_table(
561 mut self,
562 table: impl GGData,
563 spec: crate::geom::bracket::BracketTable,
564 ) -> Self {
565 let table = table.into_dataframe();
566 let col_for = |a: crate::aes::Aesthetic| {
567 self.mapping
568 .mappings
569 .iter()
570 .find(|m| m.aesthetic == a)
571 .map(|m| m.column.clone())
572 };
573 let x_levels: Vec<String> = col_for(crate::aes::Aesthetic::X)
576 .and_then(|c| self.data.column(&c))
577 .filter(|col| col.iter().any(|v| matches!(v, Value::Str(_))))
578 .map(|col| {
579 let mut seen = std::collections::HashSet::new();
580 col.iter()
581 .filter(|v| !v.is_na())
582 .map(|v| v.to_group_key())
583 .filter(|k| seen.insert(k.clone()))
584 .collect()
585 })
586 .unwrap_or_default();
587 let y_range = col_for(crate::aes::Aesthetic::Y)
588 .and_then(|c| self.data.column(&c))
589 .and_then(|col| {
590 let (lo, hi) = col
591 .iter()
592 .filter_map(|v| v.as_f64())
593 .filter(|v| v.is_finite())
594 .fold((f64::INFINITY, f64::NEG_INFINITY), |(lo, hi), v| {
595 (lo.min(v), hi.max(v))
596 });
597 (lo <= hi).then_some((lo, hi))
598 });
599 let mut warnings = Vec::new();
600 let resolved = spec.resolve(&table, &x_levels, y_range, &mut warnings);
601 self.warnings.extend(warnings);
602 match resolved {
603 Some(data) => self
604 .add_geom(crate::geom::bracket::GeomBracket::from(spec.geom))
605 .layer_data(data)
606 .layer_aes(Aes::new().xmin("xmin").xmax("xmax").y("y").label("label")),
607 None => self,
608 }
609 }
610
611 pub fn geom_text_repel(self) -> Self {
616 self.add_geom(crate::geom::repel::GeomTextRepel::default())
617 }
618
619 pub fn geom_text_repel_with(self, geom: crate::geom::repel::GeomTextRepel) -> Self {
622 self.add_geom_with(geom)
623 }
624
625 pub fn geom_label_repel(self) -> Self {
628 self.add_geom(crate::geom::repel::GeomLabelRepel::default())
629 }
630
631 pub fn geom_label_repel_with(self, geom: crate::geom::repel::GeomLabelRepel) -> Self {
633 self.add_geom_with(geom)
634 }
635
636 pub fn geom_label(self) -> Self {
637 self.add_geom(GeomLabel::default())
638 }
639
640 pub fn geom_label_with(self, geom: GeomLabel) -> Self {
641 self.add_geom_with(geom)
642 }
643
644 pub fn geom_area(self) -> Self {
645 self.add_geom(GeomArea::default())
646 }
647
648 pub fn geom_area_with(self, geom: GeomArea) -> Self {
649 self.add_geom_with(geom)
650 }
651
652 pub fn geom_ribbon(self) -> Self {
653 self.add_geom(GeomRibbon::default())
654 }
655
656 pub fn geom_ribbon_with(self, geom: GeomRibbon) -> Self {
657 self.add_geom_with(geom)
658 }
659
660 pub fn geom_errorbar(self) -> Self {
661 self.add_geom(GeomErrorbar::default())
662 }
663
664 pub fn geom_errorbar_with(self, geom: GeomErrorbar) -> Self {
665 self.add_geom_with(geom)
666 }
667
668 pub fn geom_segment(self) -> Self {
669 self.add_geom(GeomSegment::default())
670 }
671
672 pub fn geom_segment_with(self, geom: GeomSegment) -> Self {
673 self.add_geom_with(geom)
674 }
675
676 pub fn geom_density(self) -> Self {
677 self.add_geom(GeomDensity::default())
678 }
679
680 pub fn geom_density_with(self, geom: GeomDensity) -> Self {
681 self.add_geom_with(geom)
682 }
683
684 pub fn geom_rug(self) -> Self {
685 self.add_geom(GeomRug::default())
686 }
687
688 pub fn geom_rug_with(self, geom: GeomRug) -> Self {
689 self.add_geom_with(geom)
690 }
691
692 pub fn geom_jitter(self) -> Self {
693 self.add_geom(GeomJitter::default())
694 }
695
696 pub fn geom_jitter_with(self, geom: GeomJitter) -> Self {
697 self.add_geom_with(geom)
698 }
699
700 pub fn geom_path(self) -> Self {
701 self.add_geom(GeomPath::default())
702 }
703
704 pub fn geom_path_with(self, geom: GeomPath) -> Self {
705 self.add_geom_with(geom)
706 }
707
708 pub fn stat_ellipse(self) -> Self {
710 self.geom_path()
711 .stat(crate::stat::ellipse::StatEllipse::default())
712 }
713
714 pub fn stat_ellipse_level(self, level: f64) -> Self {
716 self.geom_path()
717 .stat(crate::stat::ellipse::StatEllipse::new(level))
718 }
719
720 #[cfg(feature = "regression")]
723 pub fn stat_quantile(mut self, taus: &[f64]) -> Self {
724 for &tau in taus {
725 self = self
726 .geom_path()
727 .stat(crate::stat::quantile::StatQuantile::new(tau));
728 }
729 self
730 }
731
732 #[cfg(feature = "regression")]
734 pub fn geom_quantile(self) -> Self {
735 self.stat_quantile(&[0.25, 0.5, 0.75])
736 }
737
738 pub fn geom_step(self) -> Self {
739 self.add_geom(GeomStep::default())
740 }
741
742 pub fn geom_step_with(self, geom: GeomStep) -> Self {
743 self.add_geom_with(geom)
744 }
745
746 pub fn geom_freqpoly(self) -> Self {
747 self.add_geom(GeomFreqpoly::default())
748 }
749
750 pub fn geom_freqpoly_with(self, geom: GeomFreqpoly) -> Self {
751 self.add_geom_with(geom)
752 }
753
754 pub fn geom_linerange(self) -> Self {
755 self.add_geom(GeomLinerange::default())
756 }
757
758 pub fn geom_linerange_with(self, geom: GeomLinerange) -> Self {
759 self.add_geom_with(geom)
760 }
761
762 pub fn geom_pointrange(self) -> Self {
763 self.add_geom(GeomPointrange::default())
764 }
765
766 pub fn geom_pointrange_with(self, geom: GeomPointrange) -> Self {
767 self.add_geom_with(geom)
768 }
769
770 pub fn geom_crossbar(self) -> Self {
771 self.add_geom(GeomCrossbar::default())
772 }
773
774 pub fn geom_crossbar_with(self, geom: GeomCrossbar) -> Self {
775 self.add_geom_with(geom)
776 }
777
778 pub fn geom_spoke(self) -> Self {
779 self.add_geom(GeomSpoke::default())
780 }
781
782 pub fn geom_spoke_with(self, geom: GeomSpoke) -> Self {
783 self.add_geom_with(geom)
784 }
785
786 pub fn geom_rect(self) -> Self {
787 self.add_geom(GeomRect::default())
788 }
789
790 pub fn geom_rect_with(self, geom: GeomRect) -> Self {
791 self.add_geom_with(geom)
792 }
793
794 pub fn geom_tile(self) -> Self {
795 self.add_geom(GeomTile::default())
796 }
797
798 pub fn geom_tile_with(self, geom: GeomTile) -> Self {
799 self.add_geom_with(geom)
800 }
801
802 pub fn geom_raster(self) -> Self {
804 self.add_geom(crate::geom::raster::GeomRaster::default())
805 }
806
807 pub fn geom_raster_with(self, geom: crate::geom::raster::GeomRaster) -> Self {
808 self.add_geom_with(geom)
809 }
810
811 pub fn geom_polygon(self) -> Self {
812 self.add_geom(GeomPolygon::default())
813 }
814
815 pub fn geom_polygon_with(self, geom: GeomPolygon) -> Self {
816 self.add_geom_with(geom)
817 }
818
819 #[cfg(feature = "sf")]
821 pub fn geom_sf(self) -> Self {
822 self.add_geom(crate::geom::sf::GeomSf::default())
823 }
824
825 #[cfg(feature = "sf")]
826 pub fn geom_sf_with(self, geom: crate::geom::sf::GeomSf) -> Self {
827 self.add_geom_with(geom)
828 }
829
830 pub fn geom_curve(self) -> Self {
831 self.add_geom(GeomCurve::default())
832 }
833
834 pub fn geom_curve_with(self, geom: GeomCurve) -> Self {
835 self.add_geom_with(geom)
836 }
837
838 pub fn geom_violin(self) -> Self {
839 self.add_geom(GeomViolin::default())
840 }
841
842 pub fn geom_violin_with(self, geom: GeomViolin) -> Self {
843 self.add_geom_with(geom)
844 }
845
846 pub fn geom_dotplot(self) -> Self {
847 self.add_geom(GeomDotplot::default())
848 }
849
850 pub fn geom_dotplot_with(self, geom: GeomDotplot) -> Self {
851 self.add_geom_with(geom)
852 }
853
854 pub fn geom_qq(self) -> Self {
855 self.add_geom(GeomQQ::default())
856 }
857
858 pub fn geom_qq_with(self, geom: GeomQQ) -> Self {
859 self.add_geom_with(geom)
860 }
861
862 pub fn geom_qq_line(self) -> Self {
863 self.add_geom(GeomQQLine::default())
864 }
865
866 pub fn geom_qq_line_with(self, geom: GeomQQLine) -> Self {
867 self.add_geom_with(geom)
868 }
869
870 pub fn geom_bin2d(self) -> Self {
871 self.add_geom(GeomBin2d::default())
872 }
873
874 pub fn geom_bin2d_with(self, geom: GeomBin2d) -> Self {
875 self.add_geom_with(geom)
876 }
877
878 pub fn geom_hex(self) -> Self {
879 self.add_geom(GeomHex::default())
880 }
881
882 pub fn geom_hex_with(self, geom: GeomHex) -> Self {
883 self.add_geom_with(geom)
884 }
885
886 pub fn geom_count(self) -> Self {
887 self.add_geom(GeomCount::default())
888 }
889
890 pub fn geom_count_with(self, geom: GeomCount) -> Self {
891 self.add_geom_with(geom)
892 }
893
894 pub fn geom_contour(self) -> Self {
895 self.add_geom(GeomContour::default())
896 }
897
898 pub fn geom_contour_with(self, geom: GeomContour) -> Self {
899 self.add_geom_with(geom)
900 }
901
902 pub fn geom_contour_filled(self) -> Self {
905 self.add_geom(GeomPolygon {
906 line_width: 0.0,
907 alpha: 1.0,
908 ..GeomPolygon::default()
909 })
910 .stat(crate::stat::contour_filled::StatContourFilled::default())
911 }
912
913 pub fn geom_density2d(self) -> Self {
914 self.add_geom(GeomDensity2d::default())
915 }
916
917 pub fn geom_density2d_with(self, geom: GeomDensity2d) -> Self {
918 self.add_geom_with(geom)
919 }
920
921 pub fn geom_candlestick(self) -> Self {
924 self.add_geom(crate::geom::candlestick::GeomCandlestick::default())
925 }
926
927 pub fn geom_candlestick_with(self, geom: crate::geom::candlestick::GeomCandlestick) -> Self {
928 self.add_geom_with(geom)
929 }
930
931 pub fn geom_ohlc(self) -> Self {
934 self.add_geom(crate::geom::candlestick::GeomOhlc::default())
935 }
936
937 pub fn geom_ohlc_with(self, geom: crate::geom::candlestick::GeomOhlc) -> Self {
938 self.add_geom_with(geom)
939 }
940
941 pub fn geom_calendar(self) -> Self {
952 self.geom_calendar_with(
953 GeomTile {
954 color: (255, 255, 255),
955 line_width: 1.5,
956 ..Default::default()
957 },
958 false,
959 )
960 }
961
962 pub fn geom_calendar_with(mut self, tile: GeomTile, monday_first: bool) -> Self {
965 use crate::stat::calendar::{
966 day_number, month_boundaries, month_breaks, month_breaks_localized, weekday_labels,
967 CalendarGrid, StatCalendar, MAX_CALENDAR_YEARS,
968 };
969 let grid = self
970 .mapping
971 .get_mapping(&crate::aes::Aesthetic::X)
972 .and_then(|c| self.data.column(c))
973 .and_then(|col| {
974 CalendarGrid::from_days(col.iter().filter_map(day_number), monday_first)
975 });
976 let Some((grid, clipped)) = grid else {
977 return self.add_geom_with(tile).stat(StatCalendar {
979 grid: None,
980 monday_first,
981 });
982 };
983 if clipped {
984 self.warnings.push(format!(
985 "geom_calendar: dates more than {MAX_CALENDAR_YEARS} years before the newest were dropped"
986 ));
987 }
988 let (xb, _) = month_breaks(&grid);
989 let yb = if monday_first {
991 vec![6.0, 4.0, 2.0]
992 } else {
993 vec![5.0, 3.0, 1.0]
994 };
995 let segs = month_boundaries(&grid);
996 let col = |f: fn(&(f64, f64, f64, f64)) -> f64| -> Vec<Value> {
997 segs.iter().map(|s| Value::Float(f(s))).collect()
998 };
999 let boundaries = vec![
1000 ("x".to_string(), col(|s| s.0)),
1001 ("y".to_string(), col(|s| s.1)),
1002 ("xend".to_string(), col(|s| s.2)),
1003 ("yend".to_string(), col(|s| s.3)),
1004 ];
1005 self.default_aspect_ratio = Some(7.0 / grid.n_weeks().max(1) as f64);
1006 let plot = self
1007 .add_geom_with(tile)
1008 .stat(StatCalendar {
1009 grid: Some(grid),
1010 monday_first,
1011 })
1012 .scale_x_continuous(
1013 ScaleContinuous::new()
1014 .with_breaks(xb)
1015 .with_localized_labels(move |l| month_breaks_localized(&grid, l).1)
1017 .with_expand(0.0, 0.0),
1018 )
1019 .scale_y_continuous(
1020 ScaleContinuous::new()
1021 .with_breaks(yb)
1022 .with_localized_labels(weekday_labels)
1023 .with_expand(0.0, 0.0),
1024 );
1025 if segs.is_empty() {
1026 return plot;
1027 }
1028 plot.add_geom_with(crate::geom::segment::GeomCalendarBorder)
1030 .layer_data(boundaries)
1031 .layer_aes(Aes::new().x("x").y("y").xend("xend").yend("yend"))
1032 .show_legend(false)
1033 }
1034
1035 pub fn geom_blank(self) -> Self {
1036 self.add_geom(GeomBlank)
1037 }
1038
1039 fn add_geom_with(self, geom: impl Geom + 'static) -> Self {
1042 let mut plot = self.add_geom(geom);
1043 if let Some(layer) = plot.layers.last_mut() {
1044 layer.explicit_style = true;
1045 }
1046 plot
1047 }
1048
1049 fn add_geom(mut self, geom: impl Geom + 'static) -> Self {
1050 let stat = geom.default_stat();
1051 let position = geom.default_position();
1052 let params = geom.default_params();
1053 self.layers.push(Layer {
1054 data: None,
1055 mapping: Aes::default(),
1056 geom: Box::new(geom),
1057 stat,
1058 position,
1059 params,
1060 show_legend: None,
1061 explicit_style: false,
1062 tooltip_template: None,
1063 });
1064 self
1065 }
1066
1067 pub fn stat(mut self, stat: impl Stat + 'static) -> Self {
1071 if let Some(layer) = self.layers.last_mut() {
1072 layer.stat = Box::new(stat);
1073 }
1074 self
1075 }
1076
1077 pub fn position(mut self, pos: impl Position + 'static) -> Self {
1079 if let Some(layer) = self.layers.last_mut() {
1080 layer.position = Box::new(pos);
1081 }
1082 self
1083 }
1084
1085 pub fn layer_data(mut self, data: impl GGData) -> Self {
1087 if let Some(layer) = self.layers.last_mut() {
1088 layer.data = Some(data.into_dataframe());
1089 }
1090 self
1091 }
1092
1093 pub fn layer_aes(mut self, mapping: Aes) -> Self {
1095 self.warnings.extend(mapping.warnings.iter().cloned());
1096 if let Some(layer) = self.layers.last_mut() {
1097 layer.mapping = mapping;
1098 }
1099 self
1100 }
1101
1102 pub fn tooltip(mut self, template: &str) -> Self {
1113 if let Some(layer) = self.layers.last_mut() {
1114 layer.tooltip_template = Some(template.to_string());
1115 }
1116 self
1117 }
1118
1119 pub fn show_legend(mut self, show: bool) -> Self {
1122 if let Some(layer) = self.layers.last_mut() {
1123 layer.show_legend = Some(show);
1124 }
1125 self
1126 }
1127
1128 pub fn scale_x_continuous(mut self, s: ScaleContinuous) -> Self {
1131 let s = s.for_aesthetic(crate::aes::Aesthetic::X);
1132 self.scales.push(Box::new(s));
1133 self
1134 }
1135
1136 pub fn scale_y_continuous(mut self, s: ScaleContinuous) -> Self {
1137 let s = s.for_aesthetic(crate::aes::Aesthetic::Y);
1138 self.scales.push(Box::new(s));
1139 self
1140 }
1141
1142 pub fn scale_x_discrete(mut self, s: crate::scale::discrete::ScaleDiscrete) -> Self {
1143 let s = s.for_aesthetic(crate::aes::Aesthetic::X);
1144 self.scales.push(Box::new(s));
1145 self
1146 }
1147
1148 pub fn scale_y_discrete(mut self, s: crate::scale::discrete::ScaleDiscrete) -> Self {
1149 let s = s.for_aesthetic(crate::aes::Aesthetic::Y);
1150 self.scales.push(Box::new(s));
1151 self
1152 }
1153
1154 pub fn scale_color(mut self, s: impl Scale + 'static) -> Self {
1155 self.scales.push(Box::new(s));
1156 self
1157 }
1158
1159 pub fn scale_fill(mut self, s: impl Scale + 'static) -> Self {
1160 self.scales.push(Box::new(s));
1161 self
1162 }
1163
1164 pub fn scale_color_manual(self, values: Vec<(&str, crate::scale::color::RGBAColor)>) -> Self {
1165 let s = crate::scale::manual::ScaleManual::new(crate::aes::Aesthetic::Color, values);
1166 self.scale_color(s)
1167 }
1168
1169 pub fn scale_fill_manual(self, values: Vec<(&str, crate::scale::color::RGBAColor)>) -> Self {
1170 let s = crate::scale::manual::ScaleManual::new(crate::aes::Aesthetic::Fill, values);
1171 self.scale_fill(s)
1172 }
1173
1174 pub fn scale_colour_identity(self) -> Self {
1177 let s = crate::scale::identity::ScaleColorIdentity::new(crate::aes::Aesthetic::Color);
1178 self.scale_color(s)
1179 }
1180
1181 pub fn scale_color_identity(self) -> Self {
1183 self.scale_colour_identity()
1184 }
1185
1186 pub fn scale_fill_identity(self) -> Self {
1189 let s = crate::scale::identity::ScaleColorIdentity::new(crate::aes::Aesthetic::Fill);
1190 self.scale_fill(s)
1191 }
1192
1193 pub fn scale_color_viridis(self) -> Self {
1194 use crate::scale::color::ScaleColorDiscrete;
1195 use crate::scale::palettes::PaletteName;
1196 let s = ScaleColorDiscrete::new(crate::aes::Aesthetic::Color)
1197 .with_named_palette(&PaletteName::Viridis);
1198 self.scale_color(s)
1199 }
1200
1201 pub fn scale_color_brewer(self, name: crate::scale::palettes::PaletteName) -> Self {
1202 use crate::scale::color::ScaleColorDiscrete;
1203 let s = ScaleColorDiscrete::new(crate::aes::Aesthetic::Color).with_named_palette(&name);
1204 self.scale_color(s)
1205 }
1206
1207 pub fn scale_color_gradient(
1208 self,
1209 low: crate::scale::color::RGBAColor,
1210 high: crate::scale::color::RGBAColor,
1211 ) -> Self {
1212 use crate::scale::color::ScaleColorContinuous;
1213 let s = ScaleColorContinuous::new(crate::aes::Aesthetic::Color).with_colors(low, high);
1214 self.scale_color(s)
1215 }
1216
1217 pub fn scale_fill_gradient(
1218 self,
1219 low: crate::scale::color::RGBAColor,
1220 high: crate::scale::color::RGBAColor,
1221 ) -> Self {
1222 use crate::scale::color::ScaleColorContinuous;
1223 let s = ScaleColorContinuous::new(crate::aes::Aesthetic::Fill).with_colors(low, high);
1224 self.scale_fill(s)
1225 }
1226
1227 pub fn scale_color_gradient2(
1228 self,
1229 low: crate::scale::color::RGBAColor,
1230 mid: crate::scale::color::RGBAColor,
1231 high: crate::scale::color::RGBAColor,
1232 ) -> Self {
1233 use crate::scale::gradient::ScaleColorGradient2;
1234 let s = ScaleColorGradient2::new(crate::aes::Aesthetic::Color).with_colors(low, mid, high);
1235 self.scale_color(s)
1236 }
1237
1238 pub fn scale_fill_gradient2(
1239 self,
1240 low: crate::scale::color::RGBAColor,
1241 mid: crate::scale::color::RGBAColor,
1242 high: crate::scale::color::RGBAColor,
1243 ) -> Self {
1244 use crate::scale::gradient::ScaleColorGradient2;
1245 let s = ScaleColorGradient2::new(crate::aes::Aesthetic::Fill).with_colors(low, mid, high);
1246 self.scale_fill(s)
1247 }
1248
1249 pub fn scale_fill_viridis(self) -> Self {
1250 use crate::scale::color::ScaleColorDiscrete;
1251 use crate::scale::palettes::PaletteName;
1252 let s = ScaleColorDiscrete::new(crate::aes::Aesthetic::Fill)
1253 .with_named_palette(&PaletteName::Viridis);
1254 self.scale_fill(s)
1255 }
1256
1257 pub fn scale_color_viridis_c(self) -> Self {
1259 use crate::scale::gradient_n::ScaleColorGradientN;
1260 let s = ScaleColorGradientN::viridis(crate::aes::Aesthetic::Color);
1261 self.scale_color(s)
1262 }
1263
1264 pub fn scale_fill_viridis_c(self) -> Self {
1266 use crate::scale::gradient_n::ScaleColorGradientN;
1267 let s = ScaleColorGradientN::viridis(crate::aes::Aesthetic::Fill);
1268 self.scale_fill(s)
1269 }
1270
1271 pub fn scale_color_gradientn(self, stops: Vec<(f64, crate::scale::color::RGBAColor)>) -> Self {
1273 use crate::scale::gradient_n::ScaleColorGradientN;
1274 let s = ScaleColorGradientN::new(crate::aes::Aesthetic::Color, stops);
1275 self.scale_color(s)
1276 }
1277
1278 pub fn scale_fill_gradientn(self, stops: Vec<(f64, crate::scale::color::RGBAColor)>) -> Self {
1280 use crate::scale::gradient_n::ScaleColorGradientN;
1281 let s = ScaleColorGradientN::new(crate::aes::Aesthetic::Fill, stops);
1282 self.scale_fill(s)
1283 }
1284
1285 pub fn scale_color_steps(
1288 self,
1289 low: crate::scale::color::RGBAColor,
1290 high: crate::scale::color::RGBAColor,
1291 n_bins: usize,
1292 ) -> Self {
1293 let s = crate::scale::steps::ScaleColorSteps::two(
1294 crate::aes::Aesthetic::Color,
1295 (low.r, low.g, low.b),
1296 (high.r, high.g, high.b),
1297 n_bins,
1298 );
1299 self.scale_color(s)
1300 }
1301
1302 pub fn scale_color_stepsn(
1304 self,
1305 stops: Vec<crate::scale::color::RGBAColor>,
1306 n_bins: usize,
1307 ) -> Self {
1308 let s = crate::scale::steps::ScaleColorSteps::new(
1309 crate::aes::Aesthetic::Color,
1310 stops.iter().map(|c| (c.r, c.g, c.b)).collect(),
1311 n_bins,
1312 );
1313 self.scale_color(s)
1314 }
1315
1316 pub fn scale_color_fermenter(
1318 self,
1319 name: crate::scale::palettes::PaletteName,
1320 n_bins: usize,
1321 ) -> Self {
1322 let stops = crate::scale::palettes::palette(&name)
1323 .iter()
1324 .map(|c| (c.r, c.g, c.b))
1325 .collect();
1326 let s =
1327 crate::scale::steps::ScaleColorSteps::new(crate::aes::Aesthetic::Color, stops, n_bins);
1328 self.scale_color(s)
1329 }
1330
1331 pub fn scale_fill_steps(
1333 self,
1334 low: crate::scale::color::RGBAColor,
1335 high: crate::scale::color::RGBAColor,
1336 n_bins: usize,
1337 ) -> Self {
1338 let s = crate::scale::steps::ScaleColorSteps::two(
1339 crate::aes::Aesthetic::Fill,
1340 (low.r, low.g, low.b),
1341 (high.r, high.g, high.b),
1342 n_bins,
1343 );
1344 self.scale_fill(s)
1345 }
1346
1347 pub fn scale_fill_fermenter(
1349 self,
1350 name: crate::scale::palettes::PaletteName,
1351 n_bins: usize,
1352 ) -> Self {
1353 let stops = crate::scale::palettes::palette(&name)
1354 .iter()
1355 .map(|c| (c.r, c.g, c.b))
1356 .collect();
1357 let s =
1358 crate::scale::steps::ScaleColorSteps::new(crate::aes::Aesthetic::Fill, stops, n_bins);
1359 self.scale_fill(s)
1360 }
1361
1362 pub fn scale_fill_discrete_sorted(self) -> Self {
1367 use crate::scale::color::ScaleColorDiscrete;
1368 self.scale_fill(ScaleColorDiscrete::new(crate::aes::Aesthetic::Fill).sorted())
1369 }
1370
1371 pub fn scale_color_discrete_sorted(self) -> Self {
1374 use crate::scale::color::ScaleColorDiscrete;
1375 self.scale_color(ScaleColorDiscrete::new(crate::aes::Aesthetic::Color).sorted())
1376 }
1377
1378 pub fn scale_fill_brewer(self, name: crate::scale::palettes::PaletteName) -> Self {
1379 use crate::scale::color::ScaleColorDiscrete;
1380 let s = ScaleColorDiscrete::new(crate::aes::Aesthetic::Fill).with_named_palette(&name);
1381 self.scale_fill(s)
1382 }
1383
1384 pub fn scale_linetype_manual(
1385 self,
1386 values: Vec<(&str, crate::render::backend::Linetype)>,
1387 ) -> Self {
1388 let s = crate::scale::linetype_manual::ScaleLinetypeManual::new(values);
1389 self.scale_color(s)
1390 }
1391
1392 pub fn scale_shape_manual(
1393 self,
1394 values: Vec<(&str, crate::render::backend::PointShape)>,
1395 ) -> Self {
1396 let s = crate::scale::shape_manual::ScaleShapeManual::new(values);
1397 self.scale_color(s)
1398 }
1399
1400 pub fn scale_color_grey(self) -> Self {
1401 let s = crate::scale::grey::ScaleColorGrey::new(crate::aes::Aesthetic::Color);
1402 self.scale_color(s)
1403 }
1404
1405 pub fn scale_fill_grey(self) -> Self {
1406 let s = crate::scale::grey::ScaleColorGrey::new(crate::aes::Aesthetic::Fill);
1407 self.scale_fill(s)
1408 }
1409
1410 pub fn scale_color_grey_with(self, s: crate::scale::grey::ScaleColorGrey) -> Self {
1411 self.scale_color(s)
1412 }
1413
1414 pub fn scale_fill_grey_with(self, s: crate::scale::grey::ScaleColorGrey) -> Self {
1415 self.scale_fill(s)
1416 }
1417
1418 pub fn scale_x_reverse(self) -> Self {
1419 self.scale_x_continuous(ScaleContinuous::new().with_transform(ScaleTransform::Reverse))
1420 }
1421
1422 pub fn scale_y_reverse(self) -> Self {
1423 self.scale_y_continuous(ScaleContinuous::new().with_transform(ScaleTransform::Reverse))
1424 }
1425
1426 pub fn scale_x_datetime(mut self, s: crate::scale::datetime::ScaleDateTime) -> Self {
1427 let s = s.for_aesthetic(crate::aes::Aesthetic::X);
1428 self.scales.push(Box::new(s));
1429 self
1430 }
1431
1432 pub fn scale_y_datetime(mut self, s: crate::scale::datetime::ScaleDateTime) -> Self {
1433 let s = s.for_aesthetic(crate::aes::Aesthetic::Y);
1434 self.scales.push(Box::new(s));
1435 self
1436 }
1437
1438 pub fn scale_size(mut self, s: crate::scale::size::ScaleSizeContinuous) -> Self {
1439 self.scales.push(Box::new(s));
1440 self
1441 }
1442
1443 pub fn scale_alpha(mut self, s: crate::scale::alpha::ScaleAlphaContinuous) -> Self {
1444 self.scales.push(Box::new(s));
1445 self
1446 }
1447
1448 pub fn scale_linewidth(mut self, s: crate::scale::linewidth::ScaleLinewidthContinuous) -> Self {
1452 self.scales.push(Box::new(s));
1453 self
1454 }
1455
1456 pub fn xlim(self, min: f64, max: f64) -> Self {
1457 self.scale_x_continuous(ScaleContinuous::new().with_limits(min, max))
1458 }
1459
1460 pub fn ylim(self, min: f64, max: f64) -> Self {
1461 self.scale_y_continuous(ScaleContinuous::new().with_limits(min, max))
1462 }
1463
1464 pub fn scale_x_log10(self) -> Self {
1465 self.scale_x_continuous(ScaleContinuous::new().with_transform(ScaleTransform::Log10))
1466 }
1467
1468 pub fn scale_y_log10(self) -> Self {
1469 self.scale_y_continuous(ScaleContinuous::new().with_transform(ScaleTransform::Log10))
1470 }
1471
1472 pub fn scale_x_sqrt(self) -> Self {
1473 self.scale_x_continuous(ScaleContinuous::new().with_transform(ScaleTransform::Sqrt))
1474 }
1475
1476 pub fn scale_y_sqrt(self) -> Self {
1477 self.scale_y_continuous(ScaleContinuous::new().with_transform(ScaleTransform::Sqrt))
1478 }
1479
1480 pub fn scale_x_log2(self) -> Self {
1481 self.scale_x_continuous(ScaleContinuous::new().with_transform(ScaleTransform::Log2))
1482 }
1483
1484 pub fn scale_y_log2(self) -> Self {
1485 self.scale_y_continuous(ScaleContinuous::new().with_transform(ScaleTransform::Log2))
1486 }
1487
1488 pub fn scale_x_ln(self) -> Self {
1489 self.scale_x_continuous(ScaleContinuous::new().with_transform(ScaleTransform::Ln))
1490 }
1491
1492 pub fn scale_y_ln(self) -> Self {
1493 self.scale_y_continuous(ScaleContinuous::new().with_transform(ScaleTransform::Ln))
1494 }
1495
1496 pub fn scale_x_logit(self) -> Self {
1498 self.scale_x_continuous(ScaleContinuous::new().with_transform(ScaleTransform::Logit))
1499 }
1500
1501 pub fn scale_y_logit(self) -> Self {
1502 self.scale_y_continuous(ScaleContinuous::new().with_transform(ScaleTransform::Logit))
1503 }
1504
1505 pub fn scale_x_probit(self) -> Self {
1507 self.scale_x_continuous(ScaleContinuous::new().with_transform(ScaleTransform::Probit))
1508 }
1509
1510 pub fn scale_y_probit(self) -> Self {
1511 self.scale_y_continuous(ScaleContinuous::new().with_transform(ScaleTransform::Probit))
1512 }
1513
1514 pub fn scale_x_pseudo_log(self) -> Self {
1516 self.scale_x_continuous(ScaleContinuous::new().with_transform(ScaleTransform::PseudoLog))
1517 }
1518
1519 pub fn scale_y_pseudo_log(self) -> Self {
1520 self.scale_y_continuous(ScaleContinuous::new().with_transform(ScaleTransform::PseudoLog))
1521 }
1522
1523 pub fn scale_x_reciprocal(self) -> Self {
1525 self.scale_x_continuous(ScaleContinuous::new().with_transform(ScaleTransform::Reciprocal))
1526 }
1527
1528 pub fn scale_y_reciprocal(self) -> Self {
1529 self.scale_y_continuous(ScaleContinuous::new().with_transform(ScaleTransform::Reciprocal))
1530 }
1531
1532 pub fn scale_x_exp(self) -> Self {
1534 self.scale_x_continuous(ScaleContinuous::new().with_transform(ScaleTransform::Exp))
1535 }
1536
1537 pub fn scale_y_exp(self) -> Self {
1538 self.scale_y_continuous(ScaleContinuous::new().with_transform(ScaleTransform::Exp))
1539 }
1540
1541 pub fn scale_x_boxcox(self, lambda: f64) -> Self {
1543 self.scale_x_continuous(
1544 ScaleContinuous::new().with_transform(ScaleTransform::BoxCox(lambda)),
1545 )
1546 }
1547
1548 pub fn scale_y_boxcox(self, lambda: f64) -> Self {
1549 self.scale_y_continuous(
1550 ScaleContinuous::new().with_transform(ScaleTransform::BoxCox(lambda)),
1551 )
1552 }
1553
1554 pub fn facet_wrap(mut self, var: &str, ncol: Option<usize>) -> Self {
1557 self.facet = Facet::Wrap {
1558 var: var.to_string(),
1559 ncol,
1560 scales: FacetScales::Fixed,
1561 labeller: FacetLabeller::default(),
1562 };
1563 self
1564 }
1565
1566 pub fn facet_wrap_free(mut self, var: &str, ncol: Option<usize>, scales: FacetScales) -> Self {
1567 self.facet = Facet::Wrap {
1568 var: var.to_string(),
1569 ncol,
1570 scales,
1571 labeller: FacetLabeller::default(),
1572 };
1573 self
1574 }
1575
1576 pub fn facet_wrap_labeller(
1577 mut self,
1578 var: &str,
1579 ncol: Option<usize>,
1580 labeller: FacetLabeller,
1581 ) -> Self {
1582 self.facet = Facet::Wrap {
1583 var: var.to_string(),
1584 ncol,
1585 scales: FacetScales::Fixed,
1586 labeller,
1587 };
1588 self
1589 }
1590
1591 pub fn facet_grid(mut self, row: Option<&str>, col: Option<&str>) -> Self {
1592 self.facet = Facet::Grid {
1593 row_var: row.map(String::from),
1594 col_var: col.map(String::from),
1595 scales: FacetScales::Fixed,
1596 labeller: FacetLabeller::default(),
1597 space: FacetSpace::Fixed,
1598 };
1599 self
1600 }
1601
1602 pub fn facet_grid_free(
1603 mut self,
1604 row: Option<&str>,
1605 col: Option<&str>,
1606 scales: FacetScales,
1607 ) -> Self {
1608 self.facet = Facet::Grid {
1609 row_var: row.map(String::from),
1610 col_var: col.map(String::from),
1611 scales,
1612 labeller: FacetLabeller::default(),
1613 space: FacetSpace::Fixed,
1614 };
1615 self
1616 }
1617
1618 pub fn facet_grid_multi(
1622 mut self,
1623 row: Option<&str>,
1624 cols: &[&str],
1625 scales: FacetScales,
1626 space: FacetSpace,
1627 ) -> Self {
1628 let col_var = if cols.len() > 1 {
1629 let n = self.data.nrows();
1631 let mut combined = Vec::with_capacity(n);
1632 for i in 0..n {
1633 let parts: Vec<String> = cols
1634 .iter()
1635 .map(|c| {
1636 self.data
1637 .column(c)
1638 .and_then(|col| col.get(i))
1639 .map(|v| v.to_group_key())
1640 .unwrap_or_default()
1641 })
1642 .collect();
1643 combined.push(crate::data::Value::Str(parts.join(" . ")));
1644 }
1645 let name = "__facet_cols__".to_string();
1646 self.data.add_column(name.clone(), combined);
1647 Some(name)
1648 } else {
1649 cols.first().map(|s| s.to_string())
1650 };
1651 self.facet = Facet::Grid {
1652 row_var: row.map(String::from),
1653 col_var,
1654 scales,
1655 labeller: FacetLabeller::default(),
1656 space,
1657 };
1658 self
1659 }
1660
1661 pub fn facet_grid_space(
1664 mut self,
1665 row: Option<&str>,
1666 col: Option<&str>,
1667 scales: FacetScales,
1668 space: FacetSpace,
1669 ) -> Self {
1670 self.facet = Facet::Grid {
1671 row_var: row.map(String::from),
1672 col_var: col.map(String::from),
1673 scales,
1674 labeller: FacetLabeller::default(),
1675 space,
1676 };
1677 self
1678 }
1679
1680 pub fn facet_grid_labeller(
1681 mut self,
1682 row: Option<&str>,
1683 col: Option<&str>,
1684 labeller: FacetLabeller,
1685 ) -> Self {
1686 self.facet = Facet::Grid {
1687 row_var: row.map(String::from),
1688 col_var: col.map(String::from),
1689 scales: FacetScales::Fixed,
1690 labeller,
1691 space: FacetSpace::Fixed,
1692 };
1693 self
1694 }
1695
1696 pub fn coord_flip(mut self) -> Self {
1699 self.coord = Box::new(CoordFlip);
1700 self
1701 }
1702
1703 pub fn coord_fixed(mut self, ratio: f64) -> Self {
1704 self.coord = Box::new(CoordFixed::new(ratio));
1705 self
1706 }
1707
1708 #[cfg(feature = "sf")]
1712 pub fn coord_sf(mut self) -> Self {
1713 self.coord = Box::new(crate::coord::sf::CoordSf::new());
1714 self
1715 }
1716
1717 #[cfg(feature = "sf")]
1722 pub fn coord_sf_with(mut self, coord: crate::coord::sf::CoordSf) -> Self {
1723 self.coord = Box::new(coord);
1724 self
1725 }
1726
1727 pub fn svg_root_data(mut self, on: bool) -> Self {
1734 self.svg_root_data = on;
1735 self
1736 }
1737
1738 pub fn svg_attr(mut self, name: impl Into<String>, value: impl Into<String>) -> Self {
1748 self.root_attrs.push((name.into(), value.into()));
1749 self
1750 }
1751
1752 pub fn svg_attrs(mut self, attrs: impl IntoIterator<Item = (String, String)>) -> Self {
1754 self.root_attrs.extend(attrs);
1755 self
1756 }
1757
1758 pub fn svg_warnings_attr(mut self, on: bool) -> Self {
1763 self.svg_warnings_attr = on;
1764 self
1765 }
1766
1767 pub fn svg_formatted_values(mut self, on: bool) -> Self {
1776 self.svg_formatted_values = on;
1777 self
1778 }
1779
1780 pub fn coord_trans(mut self, x: Option<ScaleTransform>, y: Option<ScaleTransform>) -> Self {
1785 self.coord = Box::new(crate::coord::trans::CoordTrans::new(x, y));
1786 self
1787 }
1788
1789 pub fn coord_trans_y(self, y: ScaleTransform) -> Self {
1791 self.coord_trans(None, Some(y))
1792 }
1793
1794 pub fn coord_trans_x(self, x: ScaleTransform) -> Self {
1796 self.coord_trans(Some(x), None)
1797 }
1798
1799 pub fn coord_cartesian_zoom(
1801 mut self,
1802 xlim: Option<(f64, f64)>,
1803 ylim: Option<(f64, f64)>,
1804 ) -> Self {
1805 let mut c = CoordCartesian::new();
1806 if let Some((min, max)) = xlim {
1807 c = c.xlim(min, max);
1808 }
1809 if let Some((min, max)) = ylim {
1810 c = c.ylim(min, max);
1811 }
1812 self.coord = Box::new(c);
1813 self
1814 }
1815
1816 pub fn coord_polar(mut self) -> Self {
1817 self.coord = Box::new(CoordPolar::new());
1818 self
1819 }
1820
1821 pub fn coord_radar(mut self) -> Self {
1825 self.coord = Box::new(crate::coord::radar::CoordRadar::new());
1826 self
1827 }
1828
1829 pub fn coord_radar_with(mut self, coord: crate::coord::radar::CoordRadar) -> Self {
1830 self.coord = Box::new(coord);
1831 self
1832 }
1833
1834 pub fn coord_polar_with(mut self, coord: CoordPolar) -> Self {
1835 self.coord = Box::new(coord);
1836 self
1837 }
1838
1839 pub fn theme(mut self, theme: Theme) -> Self {
1844 self.apply_preset(theme);
1845 self
1846 }
1847
1848 fn apply_preset(&mut self, theme: Theme) {
1849 let primary = self.theme.primary;
1850 self.theme = theme;
1851 if self.theme.primary.is_none() {
1852 self.theme.primary = primary;
1853 }
1854 }
1855
1856 pub fn axis_text_x_angle(mut self, degrees: f64) -> Self {
1860 self.theme.axis_text_x.angle = degrees;
1861 self
1862 }
1863
1864 pub fn axis_text_y_angle(mut self, degrees: f64) -> Self {
1866 self.theme.axis_text_y.angle = degrees;
1867 self
1868 }
1869
1870 pub fn axis_text_x_dodge(mut self, n: usize) -> Self {
1873 self.theme.axis_text_x_dodge = n.max(1);
1874 self
1875 }
1876
1877 pub fn aspect_ratio(mut self, ratio: f64) -> Self {
1880 self.theme.aspect_ratio = Some(ratio);
1881 self
1882 }
1883
1884 pub fn panel_ontop(mut self) -> Self {
1886 self.theme.panel_ontop = true;
1887 self
1888 }
1889
1890 pub fn axis_minor_ticks(mut self) -> Self {
1892 self.theme.axis_minor_ticks = true;
1893 self
1894 }
1895
1896 pub fn title_position(mut self, pos: crate::theme::TitlePosition) -> Self {
1899 self.theme.title_position = pos;
1900 self
1901 }
1902
1903 pub fn tag_position(mut self, pos: crate::theme::TagPosition) -> Self {
1905 self.theme.tag_position = pos;
1906 self
1907 }
1908
1909 pub fn legend_direction(mut self, dir: crate::theme::LegendDirection) -> Self {
1911 self.theme.legend_direction = Some(dir);
1912 self
1913 }
1914
1915 pub fn legend_position_inside(mut self, x: f64, y: f64) -> Self {
1922 self.theme.legend_position = crate::theme::LegendPosition::Inside(x, y);
1923 self
1924 }
1925
1926 pub fn legend_position(mut self, pos: crate::theme::LegendPosition) -> Self {
1928 self.theme.legend_position = pos;
1929 self
1930 }
1931
1932 pub fn primary_color(mut self, color: (u8, u8, u8)) -> Self {
1933 self.theme.primary = Some(color);
1934 self
1935 }
1936
1937 pub fn theme_minimal(mut self) -> Self {
1938 self.apply_preset(crate::theme::presets::theme_minimal());
1939 self
1940 }
1941
1942 pub fn theme_bw(mut self) -> Self {
1943 self.apply_preset(crate::theme::presets::theme_bw());
1944 self
1945 }
1946
1947 pub fn theme_gray(mut self) -> Self {
1948 self.apply_preset(crate::theme::presets::theme_gray());
1949 self
1950 }
1951
1952 pub fn theme_classic(mut self) -> Self {
1953 self.apply_preset(crate::theme::presets::theme_classic());
1954 self
1955 }
1956
1957 pub fn theme_pubr(mut self) -> Self {
1959 self.apply_preset(crate::theme::presets::theme_pubr());
1960 self
1961 }
1962
1963 pub fn theme_linedraw(mut self) -> Self {
1964 self.apply_preset(crate::theme::presets::theme_linedraw());
1965 self
1966 }
1967
1968 pub fn theme_light(mut self) -> Self {
1969 self.apply_preset(crate::theme::presets::theme_light());
1970 self
1971 }
1972
1973 pub fn theme_dark(mut self) -> Self {
1974 self.apply_preset(crate::theme::presets::theme_dark());
1975 self
1976 }
1977
1978 pub fn theme_void(mut self) -> Self {
1979 self.apply_preset(crate::theme::presets::theme_void());
1980 self
1981 }
1982
1983 pub fn theme_update(mut self, update: crate::theme::ThemeUpdate) -> Self {
1986 self.theme = self.theme.update(update);
1987 self
1988 }
1989
1990 pub fn guides(mut self, guide: crate::guide::config::GuideLegend) -> Self {
1994 self.guide_legend = guide;
1995 self
1996 }
1997
1998 pub fn labs(mut self, labels: Labels) -> Self {
2001 if labels.title.is_some() {
2002 self.labels.title = labels.title;
2003 }
2004 if labels.subtitle.is_some() {
2005 self.labels.subtitle = labels.subtitle;
2006 }
2007 if labels.x.is_some() {
2008 self.labels.x = labels.x;
2009 }
2010 if labels.y.is_some() {
2011 self.labels.y = labels.y;
2012 }
2013 if labels.caption.is_some() {
2014 self.labels.caption = labels.caption;
2015 }
2016 if labels.tag.is_some() {
2017 self.labels.tag = labels.tag;
2018 }
2019 self
2020 }
2021
2022 pub fn title(mut self, title: &str) -> Self {
2023 self.labels.title = Some(title.to_string());
2024 self
2025 }
2026
2027 pub fn tag(mut self, tag: &str) -> Self {
2030 self.labels.tag = Some(tag.to_string());
2031 self
2032 }
2033
2034 pub fn subtitle(mut self, subtitle: &str) -> Self {
2035 self.labels.subtitle = Some(subtitle.to_string());
2036 self
2037 }
2038
2039 pub fn xlab(mut self, label: &str) -> Self {
2040 self.labels.x = Some(label.to_string());
2041 self
2042 }
2043
2044 pub fn ylab(mut self, label: &str) -> Self {
2045 self.labels.y = Some(label.to_string());
2046 self
2047 }
2048
2049 pub fn caption(mut self, caption: &str) -> Self {
2050 self.labels.caption = Some(caption.to_string());
2051 self
2052 }
2053
2054 pub fn annotate(mut self, annotation: Annotation) -> Self {
2058 self.annotations.push(annotation);
2059 self
2060 }
2061
2062 pub fn annotate_text(self, label: &str, x: f64, y: f64) -> Self {
2064 self.annotate(Annotation::text(label, x, y))
2065 }
2066
2067 pub fn annotate_rect(self, xmin: f64, xmax: f64, ymin: f64, ymax: f64) -> Self {
2069 self.annotate(Annotation::rect(xmin, xmax, ymin, ymax))
2070 }
2071
2072 pub fn annotate_segment(self, x: f64, y: f64, xend: f64, yend: f64) -> Self {
2074 self.annotate(Annotation::segment(x, y, xend, yend))
2075 }
2076
2077 pub fn locale(mut self, locale: crate::locale::Locale) -> Self {
2100 self.locale = Some(locale);
2101 self
2102 }
2103
2104 pub fn try_build(self) -> Result<crate::build::BuiltPlot, GGError> {
2108 PlotBuilder::build(self)
2109 }
2110
2111 pub fn build(self) -> crate::build::BuiltPlot {
2115 self.try_build().expect("plot build failed")
2116 }
2117
2118 #[cfg(all(feature = "plotters", not(target_arch = "wasm32")))]
2122 pub fn save(self, path: &str) -> Result<(), GGError> {
2123 self.save_with_size(path, 800, 600)
2124 }
2125
2126 #[cfg(all(feature = "plotters", not(target_arch = "wasm32")))]
2128 pub fn save_with_size(self, path: &str, w: u32, h: u32) -> Result<(), GGError> {
2129 let (built, layout) = self.prepare(w, h)?;
2130
2131 let ext = path.rsplit('.').next().unwrap_or("svg").to_lowercase();
2133
2134 match ext.as_str() {
2135 "svg" => {
2136 let backend = plotters::prelude::SVGBackend::new(path, (w, h));
2137 Self::render_into(backend.into_drawing_area(), &built, &layout)?;
2138 }
2139 "png" | "bmp" | "gif" | "jpeg" | "jpg" | "tiff" => {
2140 let backend = plotters::prelude::BitMapBackend::new(path, (w, h));
2141 Self::render_into(backend.into_drawing_area(), &built, &layout)?;
2142 }
2143 _ => {
2144 return Err(GGError::UnsupportedFormat(ext));
2145 }
2146 }
2147
2148 Ok(())
2149 }
2150
2151 #[cfg(all(feature = "plotters", not(target_arch = "wasm32")))]
2157 pub fn render_svg(self) -> Result<String, GGError> {
2158 self.render_svg_with_size(800, 600)
2159 }
2160
2161 #[cfg(all(feature = "plotters", not(target_arch = "wasm32")))]
2165 pub fn render_svg_with_size(self, w: u32, h: u32) -> Result<String, GGError> {
2166 let (built, layout) = self.prepare(w, h)?;
2167 let mut buf = String::new();
2168 {
2169 let backend = plotters::prelude::SVGBackend::with_string(&mut buf, (w, h));
2170 Self::render_into(backend.into_drawing_area(), &built, &layout)?;
2171 }
2172 Ok(buf)
2173 }
2174
2175 pub fn render_svg_native(self) -> Result<String, GGError> {
2179 self.render_svg_native_with_size(800, 600)
2180 }
2181
2182 pub fn render_svg_native_with_size(self, w: u32, h: u32) -> Result<String, GGError> {
2190 Ok(self.render_native(w, h)?.0.finish())
2191 }
2192
2193 pub fn render_svg_native_with_warnings(
2198 self,
2199 w: u32,
2200 h: u32,
2201 ) -> Result<(String, Vec<String>), GGError> {
2202 let (backend, warnings, _) = self.render_native(w, h)?;
2203 Ok((backend.finish(), warnings))
2204 }
2205
2206 pub fn render_svg_native_at(self, x: f64, y: f64, w: u32, h: u32) -> Result<String, GGError> {
2218 Ok(self.render_native(w, h)?.0.finish_fragment(x, y))
2219 }
2220
2221 pub fn render_svg_area_with_size(self, w: u32, h: u32) -> Result<(String, [f64; 4]), GGError> {
2225 let (backend, _, pa) = self.render_native(w, h)?;
2226 Ok((backend.finish(), [pa.x, pa.y, pa.width, pa.height]))
2227 }
2228
2229 fn render_native(
2233 self,
2234 w: u32,
2235 h: u32,
2236 ) -> Result<
2237 (
2238 crate::render::svg_backend::SvgBackend,
2239 Vec<String>,
2240 crate::render::Rect,
2241 ),
2242 GGError,
2243 > {
2244 let svg_root_data = self.svg_root_data;
2245 let root_attrs = self.root_attrs.clone();
2246 let svg_warnings_attr = self.svg_warnings_attr;
2247 let svg_formatted_values = self.svg_formatted_values;
2248 let (built, layout) = self.prepare(w, h)?;
2249 let pa = layout.plot_area.clone();
2250 let mut backend = crate::render::svg_backend::SvgBackend::new(w, h, pa.clone());
2251 backend.set_formatted_values(svg_formatted_values);
2252 let existing_root_attrs = if svg_root_data {
2253 let attrs = crate::render::svg_backend::root_data_attrs(&built);
2254 backend.set_root_attrs(attrs.clone());
2255 let mut reserved = attrs;
2259 reserved.push(("data-plot".to_string(), String::new()));
2260 reserved
2261 } else {
2262 backend.without_plot_attr();
2263 Vec::new()
2264 };
2265 PlotRenderer::render(&built, &mut backend).map_err(GGError::Render)?;
2266 let mut warnings = built.warnings;
2267 warnings.extend(backend.take_warnings());
2268
2269 let mut reserved = existing_root_attrs;
2270 if svg_warnings_attr {
2271 reserved.push(("data-warnings".to_string(), String::new()));
2272 }
2273 let extra = crate::render::svg_backend::validate_root_attrs(
2274 &root_attrs,
2275 &reserved,
2276 "plot",
2277 &mut warnings,
2278 );
2279 backend.extend_root_attrs(extra);
2280 if svg_warnings_attr {
2281 backend.extend_root_attrs(vec![(
2282 "data-warnings".to_string(),
2283 crate::render::svg_backend::warnings_json(&warnings),
2284 )]);
2285 }
2286 Ok((backend, warnings, pa))
2287 }
2288
2289 #[cfg(feature = "canvas")]
2294 pub fn render_rgba_with_size(self, w: u32, h: u32) -> Result<(u32, u32, Vec<u8>), GGError> {
2295 let (built, layout) = self.prepare(w, h)?;
2296 let mut backend =
2297 crate::render::pixel_backend::PixelBackend::new(w, h, layout.plot_area.clone());
2298 PlotRenderer::render(&built, &mut backend).map_err(GGError::Render)?;
2299 let (ow, oh) = backend.dimensions();
2300 Ok((ow, oh, backend.into_rgba()))
2301 }
2302
2303 #[cfg(feature = "canvas")]
2307 pub fn render_rgba_area_with_size(
2308 self,
2309 w: u32,
2310 h: u32,
2311 ) -> Result<(Vec<u8>, [f64; 4]), GGError> {
2312 let (built, layout) = self.prepare(w, h)?;
2313 let pa = layout.plot_area.clone();
2314 let mut backend = crate::render::pixel_backend::PixelBackend::new(w, h, pa.clone());
2315 PlotRenderer::render(&built, &mut backend).map_err(GGError::Render)?;
2316 Ok((backend.into_rgba(), [pa.x, pa.y, pa.width, pa.height]))
2317 }
2318
2319 #[cfg(feature = "canvas")]
2323 pub fn render_png_raster_with_size(self, w: u32, h: u32) -> Result<Vec<u8>, GGError> {
2324 let (built, layout) = self.prepare(w, h)?;
2325 let mut backend =
2326 crate::render::pixel_backend::PixelBackend::new(w, h, layout.plot_area.clone());
2327 PlotRenderer::render(&built, &mut backend).map_err(GGError::Render)?;
2328 backend.into_png().map_err(GGError::Render)
2329 }
2330
2331 #[cfg(all(feature = "plotters", not(target_arch = "wasm32")))]
2338 pub fn render_png(self) -> Result<Vec<u8>, GGError> {
2339 self.render_png_with_size(800, 600)
2340 }
2341
2342 #[cfg(all(feature = "plotters", not(target_arch = "wasm32")))]
2345 pub fn render_png_with_size(self, w: u32, h: u32) -> Result<Vec<u8>, GGError> {
2346 let (built, layout) = self.prepare(w, h)?;
2347
2348 let mut rgb = vec![0u8; (w as usize) * (h as usize) * 3];
2351 {
2352 let backend = plotters::prelude::BitMapBackend::with_buffer(&mut rgb, (w, h));
2353 Self::render_into(backend.into_drawing_area(), &built, &layout)?;
2354 }
2355
2356 let img = image::RgbImage::from_raw(w, h, rgb).ok_or_else(|| {
2357 GGError::Render(RenderError::BackendError(
2358 "PNG buffer size mismatch".to_string(),
2359 ))
2360 })?;
2361 let mut out = std::io::Cursor::new(Vec::new());
2362 img.write_to(&mut out, image::ImageOutputFormat::Png)
2363 .map_err(|e| GGError::Render(RenderError::BackendError(format!("{:?}", e))))?;
2364 Ok(out.into_inner())
2365 }
2366
2367 pub(crate) fn prepare(
2369 self,
2370 w: u32,
2371 h: u32,
2372 ) -> Result<(crate::build::BuiltPlot, PlotLayout), GGError> {
2373 let (mut built, meta) = self.build_for_render()?;
2374 let layout = Self::layout_built(&mut built, &meta, w, h);
2375 Ok((built, layout))
2376 }
2377
2378 pub(crate) fn build_for_render(self) -> Result<(crate::build::BuiltPlot, RenderMeta), GGError> {
2381 let plot = self;
2382
2383 let has_title = plot.labels.title.is_some();
2384 let has_subtitle = plot.labels.subtitle.is_some();
2385 let has_caption = plot.labels.caption.is_some();
2386 let has_legend = plot.has_legend_mapping();
2387 let x_label = plot.labels.x.clone();
2388 let y_label = plot.labels.y.clone();
2389
2390 let mut built = PlotBuilder::build(plot)?;
2391
2392 built.theme.resolve_inheritance();
2394
2395 if let Some(ref label) = x_label {
2397 if let Some(s) = built.scales.get_mut(&crate::aes::Aesthetic::X) {
2398 s.set_name(label);
2399 }
2400 }
2401 if let Some(ref label) = y_label {
2402 if let Some(s) = built.scales.get_mut(&crate::aes::Aesthetic::Y) {
2403 s.set_name(label);
2404 }
2405 }
2406
2407 Ok((
2408 built,
2409 RenderMeta {
2410 has_title,
2411 has_subtitle,
2412 has_caption,
2413 has_legend,
2414 },
2415 ))
2416 }
2417
2418 pub(crate) fn layout_built(
2421 built: &mut crate::build::BuiltPlot,
2422 meta: &RenderMeta,
2423 w: u32,
2424 h: u32,
2425 ) -> PlotLayout {
2426 let RenderMeta {
2427 has_title,
2428 has_subtitle,
2429 has_caption,
2430 has_legend,
2431 } = *meta;
2432 let x_axis_top = built
2433 .scales
2434 .get(&crate::aes::Aesthetic::X)
2435 .map(|s| s.axis_position_opposite())
2436 .unwrap_or(false);
2437
2438 let flipped = built.coord.is_flipped();
2442 let h_aes = if flipped {
2443 crate::aes::Aesthetic::Y
2444 } else {
2445 crate::aes::Aesthetic::X
2446 };
2447 let v_aes = if flipped {
2448 crate::aes::Aesthetic::X
2449 } else {
2450 crate::aes::Aesthetic::Y
2451 };
2452
2453 let x_rot: Option<(f64, f64)> =
2457 if built.theme.axis_text_x.visible && built.theme.axis_text_x.angle.abs() < 1.0 {
2458 built.scales.get(&h_aes).and_then(|hs| {
2459 if !hs.is_discrete() {
2460 return None;
2461 }
2462 let breaks = hs.breaks();
2463 let n = breaks.len().max(1) as f64;
2464 let max_ch = breaks
2465 .iter()
2466 .map(|(_, l)| l.chars().count())
2467 .max()
2468 .unwrap_or(0) as f64;
2469 let font = built.theme.axis_text_x.size;
2470 let label_w = max_ch * font * 0.6; let spacing = 0.9 * w as f64 / n; if label_w <= spacing {
2473 return None; }
2475 let vertical = label_w * 0.707 > spacing;
2479 let angle = if vertical { 270.0 } else { 45.0 };
2480 let drop = if vertical { label_w } else { label_w * 0.707 };
2481 Some((angle, drop + font + 4.0))
2482 })
2483 } else {
2484 None
2485 };
2486 let x_label_height = x_rot.map(|(_, drop)| drop);
2487 if let Some((angle, _)) = x_rot {
2488 built.theme.axis_text_x.angle = angle;
2489 }
2490
2491 let y_label_width = if built.theme.axis_text_y.visible {
2495 built.scales.get(&v_aes).map(|vs| {
2496 let max_ch = vs
2497 .breaks()
2498 .iter()
2499 .map(|(_, l)| l.chars().count())
2500 .max()
2501 .unwrap_or(0);
2502 let measured = max_ch as f64 * built.theme.axis_text_y.size * 0.6 + 4.0;
2503 let default = built.theme.axis_text_y.size * 3.5 + 4.0;
2504 measured.max(default).min(0.35 * w as f64)
2505 })
2506 } else {
2507 None
2508 };
2509
2510 PlotLayout::compute_full(
2511 w as f64,
2512 h as f64,
2513 &built.theme,
2514 has_title,
2515 has_subtitle,
2516 has_caption,
2517 has_legend,
2518 x_axis_top,
2519 y_label_width,
2520 x_label_height,
2521 )
2522 }
2523
2524 #[cfg(all(feature = "plotters", not(target_arch = "wasm32")))]
2527 fn render_into<DB>(
2528 area: plotters::drawing::DrawingArea<DB, plotters::coord::Shift>,
2529 built: &crate::build::BuiltPlot,
2530 layout: &PlotLayout,
2531 ) -> Result<(), GGError>
2532 where
2533 DB: plotters::prelude::DrawingBackend,
2534 DB::ErrorType: 'static,
2535 {
2536 area.fill(&plotters::prelude::WHITE)
2537 .map_err(|e| GGError::Render(RenderError::BackendError(format!("{:?}", e))))?;
2538 let mut adapter = PlottersAdapter::new(&area, layout.plot_area.clone());
2539 PlotRenderer::render(built, &mut adapter).map_err(GGError::Render)?;
2540 area.present()
2541 .map_err(|e| GGError::Render(RenderError::BackendError(format!("{:?}", e))))?;
2542 Ok(())
2543 }
2544
2545 #[cfg(all(feature = "plotters", not(target_arch = "wasm32")))]
2548 pub fn ggsave(
2549 self,
2550 path: &str,
2551 width_inches: f64,
2552 height_inches: f64,
2553 dpi: f64,
2554 ) -> Result<(), GGError> {
2555 let w = (width_inches * dpi) as u32;
2556 let h = (height_inches * dpi) as u32;
2557 self.save_with_size(path, w, h)
2558 }
2559
2560 fn has_legend_mapping(&self) -> bool {
2561 self.mapping.mappings.iter().any(|m| {
2562 matches!(
2563 m.aesthetic,
2564 crate::aes::Aesthetic::Color
2565 | crate::aes::Aesthetic::Fill
2566 | crate::aes::Aesthetic::Shape
2567 | crate::aes::Aesthetic::Linetype
2568 | crate::aes::Aesthetic::Size
2569 | crate::aes::Aesthetic::Alpha
2570 )
2571 })
2572 }
2573}
2574
2575#[derive(Clone, Copy, Debug)]
2577pub(crate) struct RenderMeta {
2578 pub(crate) has_title: bool,
2579 pub(crate) has_subtitle: bool,
2580 pub(crate) has_caption: bool,
2581 pub(crate) has_legend: bool,
2582}
2583
2584#[derive(Debug)]
2586pub enum GGError {
2587 Render(RenderError),
2588 UnsupportedFormat(String),
2589 Io(std::io::Error),
2590 ValidationError(String),
2591}
2592
2593impl std::fmt::Display for GGError {
2594 fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
2595 match self {
2596 GGError::Render(e) => write!(f, "Render error: {e}"),
2597 GGError::UnsupportedFormat(ext) => write!(f, "Unsupported output format: {ext}"),
2598 GGError::Io(e) => write!(f, "IO error: {e}"),
2599 GGError::ValidationError(msg) => write!(f, "Validation error: {msg}"),
2600 }
2601 }
2602}
2603
2604impl std::error::Error for GGError {}