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_size_area(mut self, max_size: f64) -> Self {
1447 self.scales.push(Box::new(
1448 crate::scale::size::ScaleSizeContinuous::new().area(max_size),
1449 ));
1450 self
1451 }
1452
1453 pub fn scale_alpha(mut self, s: crate::scale::alpha::ScaleAlphaContinuous) -> Self {
1454 self.scales.push(Box::new(s));
1455 self
1456 }
1457
1458 pub fn scale_linewidth(mut self, s: crate::scale::linewidth::ScaleLinewidthContinuous) -> Self {
1462 self.scales.push(Box::new(s));
1463 self
1464 }
1465
1466 pub fn xlim(self, min: f64, max: f64) -> Self {
1467 self.scale_x_continuous(ScaleContinuous::new().with_limits(min, max))
1468 }
1469
1470 pub fn ylim(self, min: f64, max: f64) -> Self {
1471 self.scale_y_continuous(ScaleContinuous::new().with_limits(min, max))
1472 }
1473
1474 pub fn scale_x_log10(self) -> Self {
1475 self.scale_x_continuous(ScaleContinuous::new().with_transform(ScaleTransform::Log10))
1476 }
1477
1478 pub fn scale_y_log10(self) -> Self {
1479 self.scale_y_continuous(ScaleContinuous::new().with_transform(ScaleTransform::Log10))
1480 }
1481
1482 pub fn scale_x_sqrt(self) -> Self {
1483 self.scale_x_continuous(ScaleContinuous::new().with_transform(ScaleTransform::Sqrt))
1484 }
1485
1486 pub fn scale_y_sqrt(self) -> Self {
1487 self.scale_y_continuous(ScaleContinuous::new().with_transform(ScaleTransform::Sqrt))
1488 }
1489
1490 pub fn scale_x_log2(self) -> Self {
1491 self.scale_x_continuous(ScaleContinuous::new().with_transform(ScaleTransform::Log2))
1492 }
1493
1494 pub fn scale_y_log2(self) -> Self {
1495 self.scale_y_continuous(ScaleContinuous::new().with_transform(ScaleTransform::Log2))
1496 }
1497
1498 pub fn scale_x_ln(self) -> Self {
1499 self.scale_x_continuous(ScaleContinuous::new().with_transform(ScaleTransform::Ln))
1500 }
1501
1502 pub fn scale_y_ln(self) -> Self {
1503 self.scale_y_continuous(ScaleContinuous::new().with_transform(ScaleTransform::Ln))
1504 }
1505
1506 pub fn scale_x_logit(self) -> Self {
1508 self.scale_x_continuous(ScaleContinuous::new().with_transform(ScaleTransform::Logit))
1509 }
1510
1511 pub fn scale_y_logit(self) -> Self {
1512 self.scale_y_continuous(ScaleContinuous::new().with_transform(ScaleTransform::Logit))
1513 }
1514
1515 pub fn scale_x_probit(self) -> Self {
1517 self.scale_x_continuous(ScaleContinuous::new().with_transform(ScaleTransform::Probit))
1518 }
1519
1520 pub fn scale_y_probit(self) -> Self {
1521 self.scale_y_continuous(ScaleContinuous::new().with_transform(ScaleTransform::Probit))
1522 }
1523
1524 pub fn scale_x_pseudo_log(self) -> Self {
1526 self.scale_x_continuous(ScaleContinuous::new().with_transform(ScaleTransform::PseudoLog))
1527 }
1528
1529 pub fn scale_y_pseudo_log(self) -> Self {
1530 self.scale_y_continuous(ScaleContinuous::new().with_transform(ScaleTransform::PseudoLog))
1531 }
1532
1533 pub fn scale_x_reciprocal(self) -> Self {
1535 self.scale_x_continuous(ScaleContinuous::new().with_transform(ScaleTransform::Reciprocal))
1536 }
1537
1538 pub fn scale_y_reciprocal(self) -> Self {
1539 self.scale_y_continuous(ScaleContinuous::new().with_transform(ScaleTransform::Reciprocal))
1540 }
1541
1542 pub fn scale_x_exp(self) -> Self {
1544 self.scale_x_continuous(ScaleContinuous::new().with_transform(ScaleTransform::Exp))
1545 }
1546
1547 pub fn scale_y_exp(self) -> Self {
1548 self.scale_y_continuous(ScaleContinuous::new().with_transform(ScaleTransform::Exp))
1549 }
1550
1551 pub fn scale_x_boxcox(self, lambda: f64) -> Self {
1553 self.scale_x_continuous(
1554 ScaleContinuous::new().with_transform(ScaleTransform::BoxCox(lambda)),
1555 )
1556 }
1557
1558 pub fn scale_y_boxcox(self, lambda: f64) -> Self {
1559 self.scale_y_continuous(
1560 ScaleContinuous::new().with_transform(ScaleTransform::BoxCox(lambda)),
1561 )
1562 }
1563
1564 pub fn facet_wrap(mut self, var: &str, ncol: Option<usize>) -> Self {
1567 self.facet = Facet::Wrap {
1568 var: var.to_string(),
1569 ncol,
1570 scales: FacetScales::Fixed,
1571 labeller: FacetLabeller::default(),
1572 };
1573 self
1574 }
1575
1576 pub fn facet_wrap_free(mut self, var: &str, ncol: Option<usize>, scales: FacetScales) -> Self {
1577 self.facet = Facet::Wrap {
1578 var: var.to_string(),
1579 ncol,
1580 scales,
1581 labeller: FacetLabeller::default(),
1582 };
1583 self
1584 }
1585
1586 pub fn facet_wrap_labeller(
1587 mut self,
1588 var: &str,
1589 ncol: Option<usize>,
1590 labeller: FacetLabeller,
1591 ) -> Self {
1592 self.facet = Facet::Wrap {
1593 var: var.to_string(),
1594 ncol,
1595 scales: FacetScales::Fixed,
1596 labeller,
1597 };
1598 self
1599 }
1600
1601 pub fn facet_grid(mut self, row: Option<&str>, col: Option<&str>) -> Self {
1602 self.facet = Facet::Grid {
1603 row_var: row.map(String::from),
1604 col_var: col.map(String::from),
1605 scales: FacetScales::Fixed,
1606 labeller: FacetLabeller::default(),
1607 space: FacetSpace::Fixed,
1608 };
1609 self
1610 }
1611
1612 pub fn facet_grid_free(
1613 mut self,
1614 row: Option<&str>,
1615 col: Option<&str>,
1616 scales: FacetScales,
1617 ) -> Self {
1618 self.facet = Facet::Grid {
1619 row_var: row.map(String::from),
1620 col_var: col.map(String::from),
1621 scales,
1622 labeller: FacetLabeller::default(),
1623 space: FacetSpace::Fixed,
1624 };
1625 self
1626 }
1627
1628 pub fn facet_grid_multi(
1632 mut self,
1633 row: Option<&str>,
1634 cols: &[&str],
1635 scales: FacetScales,
1636 space: FacetSpace,
1637 ) -> Self {
1638 let col_var = if cols.len() > 1 {
1639 let n = self.data.nrows();
1641 let mut combined = Vec::with_capacity(n);
1642 for i in 0..n {
1643 let parts: Vec<String> = cols
1644 .iter()
1645 .map(|c| {
1646 self.data
1647 .column(c)
1648 .and_then(|col| col.get(i))
1649 .map(|v| v.to_group_key())
1650 .unwrap_or_default()
1651 })
1652 .collect();
1653 combined.push(crate::data::Value::Str(parts.join(" . ")));
1654 }
1655 let name = "__facet_cols__".to_string();
1656 self.data.add_column(name.clone(), combined);
1657 Some(name)
1658 } else {
1659 cols.first().map(|s| s.to_string())
1660 };
1661 self.facet = Facet::Grid {
1662 row_var: row.map(String::from),
1663 col_var,
1664 scales,
1665 labeller: FacetLabeller::default(),
1666 space,
1667 };
1668 self
1669 }
1670
1671 pub fn facet_grid_space(
1674 mut self,
1675 row: Option<&str>,
1676 col: Option<&str>,
1677 scales: FacetScales,
1678 space: FacetSpace,
1679 ) -> Self {
1680 self.facet = Facet::Grid {
1681 row_var: row.map(String::from),
1682 col_var: col.map(String::from),
1683 scales,
1684 labeller: FacetLabeller::default(),
1685 space,
1686 };
1687 self
1688 }
1689
1690 pub fn facet_grid_labeller(
1691 mut self,
1692 row: Option<&str>,
1693 col: Option<&str>,
1694 labeller: FacetLabeller,
1695 ) -> Self {
1696 self.facet = Facet::Grid {
1697 row_var: row.map(String::from),
1698 col_var: col.map(String::from),
1699 scales: FacetScales::Fixed,
1700 labeller,
1701 space: FacetSpace::Fixed,
1702 };
1703 self
1704 }
1705
1706 pub fn coord_flip(mut self) -> Self {
1709 self.coord = Box::new(CoordFlip);
1710 self
1711 }
1712
1713 pub fn coord_fixed(mut self, ratio: f64) -> Self {
1714 self.coord = Box::new(CoordFixed::new(ratio));
1715 self
1716 }
1717
1718 #[cfg(feature = "sf")]
1722 pub fn coord_sf(mut self) -> Self {
1723 self.coord = Box::new(crate::coord::sf::CoordSf::new());
1724 self
1725 }
1726
1727 #[cfg(feature = "sf")]
1732 pub fn coord_sf_with(mut self, coord: crate::coord::sf::CoordSf) -> Self {
1733 self.coord = Box::new(coord);
1734 self
1735 }
1736
1737 pub fn svg_root_data(mut self, on: bool) -> Self {
1744 self.svg_root_data = on;
1745 self
1746 }
1747
1748 pub fn svg_attr(mut self, name: impl Into<String>, value: impl Into<String>) -> Self {
1758 self.root_attrs.push((name.into(), value.into()));
1759 self
1760 }
1761
1762 pub fn svg_attrs(mut self, attrs: impl IntoIterator<Item = (String, String)>) -> Self {
1764 self.root_attrs.extend(attrs);
1765 self
1766 }
1767
1768 pub fn svg_warnings_attr(mut self, on: bool) -> Self {
1773 self.svg_warnings_attr = on;
1774 self
1775 }
1776
1777 pub fn svg_formatted_values(mut self, on: bool) -> Self {
1786 self.svg_formatted_values = on;
1787 self
1788 }
1789
1790 pub fn coord_trans(mut self, x: Option<ScaleTransform>, y: Option<ScaleTransform>) -> Self {
1795 self.coord = Box::new(crate::coord::trans::CoordTrans::new(x, y));
1796 self
1797 }
1798
1799 pub fn coord_trans_y(self, y: ScaleTransform) -> Self {
1801 self.coord_trans(None, Some(y))
1802 }
1803
1804 pub fn coord_trans_x(self, x: ScaleTransform) -> Self {
1806 self.coord_trans(Some(x), None)
1807 }
1808
1809 pub fn coord_cartesian_zoom(
1811 mut self,
1812 xlim: Option<(f64, f64)>,
1813 ylim: Option<(f64, f64)>,
1814 ) -> Self {
1815 let mut c = CoordCartesian::new();
1816 if let Some((min, max)) = xlim {
1817 c = c.xlim(min, max);
1818 }
1819 if let Some((min, max)) = ylim {
1820 c = c.ylim(min, max);
1821 }
1822 self.coord = Box::new(c);
1823 self
1824 }
1825
1826 pub fn coord_polar(mut self) -> Self {
1827 self.coord = Box::new(CoordPolar::new());
1828 self
1829 }
1830
1831 pub fn coord_radar(mut self) -> Self {
1835 self.coord = Box::new(crate::coord::radar::CoordRadar::new());
1836 self
1837 }
1838
1839 pub fn coord_radar_with(mut self, coord: crate::coord::radar::CoordRadar) -> Self {
1840 self.coord = Box::new(coord);
1841 self
1842 }
1843
1844 pub fn coord_polar_with(mut self, coord: CoordPolar) -> Self {
1845 self.coord = Box::new(coord);
1846 self
1847 }
1848
1849 pub fn theme(mut self, theme: Theme) -> Self {
1854 self.apply_preset(theme);
1855 self
1856 }
1857
1858 fn apply_preset(&mut self, theme: Theme) {
1859 let primary = self.theme.primary;
1860 self.theme = theme;
1861 if self.theme.primary.is_none() {
1862 self.theme.primary = primary;
1863 }
1864 }
1865
1866 pub fn axis_text_x_angle(mut self, degrees: f64) -> Self {
1870 self.theme.axis_text_x.angle = degrees;
1871 self
1872 }
1873
1874 pub fn axis_text_y_angle(mut self, degrees: f64) -> Self {
1876 self.theme.axis_text_y.angle = degrees;
1877 self
1878 }
1879
1880 pub fn axis_text_x_dodge(mut self, n: usize) -> Self {
1883 self.theme.axis_text_x_dodge = n.max(1);
1884 self
1885 }
1886
1887 pub fn aspect_ratio(mut self, ratio: f64) -> Self {
1890 self.theme.aspect_ratio = Some(ratio);
1891 self
1892 }
1893
1894 pub fn panel_ontop(mut self) -> Self {
1896 self.theme.panel_ontop = true;
1897 self
1898 }
1899
1900 pub fn axis_minor_ticks(mut self) -> Self {
1902 self.theme.axis_minor_ticks = true;
1903 self
1904 }
1905
1906 pub fn axis_text_x_fit(mut self, fit: crate::theme::AxisLabelFit) -> Self {
1911 self.theme.axis_text_x_fit = fit;
1912 self
1913 }
1914
1915 pub fn axis_text_x_check_overlap(mut self, on: bool) -> Self {
1918 self.theme.axis_text_x_check_overlap = on;
1919 self
1920 }
1921
1922 pub fn title_position(mut self, pos: crate::theme::TitlePosition) -> Self {
1925 self.theme.title_position = pos;
1926 self
1927 }
1928
1929 pub fn tag_position(mut self, pos: crate::theme::TagPosition) -> Self {
1931 self.theme.tag_position = pos;
1932 self
1933 }
1934
1935 pub fn legend_direction(mut self, dir: crate::theme::LegendDirection) -> Self {
1937 self.theme.legend_direction = Some(dir);
1938 self
1939 }
1940
1941 pub fn legend_position_inside(mut self, x: f64, y: f64) -> Self {
1948 self.theme.legend_position = crate::theme::LegendPosition::Inside(x, y);
1949 self
1950 }
1951
1952 pub fn legend_position(mut self, pos: crate::theme::LegendPosition) -> Self {
1954 self.theme.legend_position = pos;
1955 self
1956 }
1957
1958 pub fn primary_color(mut self, color: (u8, u8, u8)) -> Self {
1959 self.theme.primary = Some(color);
1960 self
1961 }
1962
1963 pub fn theme_minimal(mut self) -> Self {
1964 self.apply_preset(crate::theme::presets::theme_minimal());
1965 self
1966 }
1967
1968 pub fn theme_bw(mut self) -> Self {
1969 self.apply_preset(crate::theme::presets::theme_bw());
1970 self
1971 }
1972
1973 pub fn theme_gray(mut self) -> Self {
1974 self.apply_preset(crate::theme::presets::theme_gray());
1975 self
1976 }
1977
1978 pub fn theme_classic(mut self) -> Self {
1979 self.apply_preset(crate::theme::presets::theme_classic());
1980 self
1981 }
1982
1983 pub fn theme_pubr(mut self) -> Self {
1985 self.apply_preset(crate::theme::presets::theme_pubr());
1986 self
1987 }
1988
1989 pub fn theme_linedraw(mut self) -> Self {
1990 self.apply_preset(crate::theme::presets::theme_linedraw());
1991 self
1992 }
1993
1994 pub fn theme_light(mut self) -> Self {
1995 self.apply_preset(crate::theme::presets::theme_light());
1996 self
1997 }
1998
1999 pub fn theme_dark(mut self) -> Self {
2000 self.apply_preset(crate::theme::presets::theme_dark());
2001 self
2002 }
2003
2004 pub fn theme_void(mut self) -> Self {
2005 self.apply_preset(crate::theme::presets::theme_void());
2006 self
2007 }
2008
2009 pub fn theme_update(mut self, update: crate::theme::ThemeUpdate) -> Self {
2012 self.theme = self.theme.update(update);
2013 self
2014 }
2015
2016 pub fn guides(mut self, guide: crate::guide::config::GuideLegend) -> Self {
2020 self.guide_legend = guide;
2021 self
2022 }
2023
2024 pub fn labs(mut self, labels: Labels) -> Self {
2027 if labels.title.is_some() {
2028 self.labels.title = labels.title;
2029 }
2030 if labels.subtitle.is_some() {
2031 self.labels.subtitle = labels.subtitle;
2032 }
2033 if labels.x.is_some() {
2034 self.labels.x = labels.x;
2035 }
2036 if labels.y.is_some() {
2037 self.labels.y = labels.y;
2038 }
2039 if labels.caption.is_some() {
2040 self.labels.caption = labels.caption;
2041 }
2042 if labels.tag.is_some() {
2043 self.labels.tag = labels.tag;
2044 }
2045 self
2046 }
2047
2048 pub fn title(mut self, title: &str) -> Self {
2049 self.labels.title = Some(title.to_string());
2050 self
2051 }
2052
2053 pub fn tag(mut self, tag: &str) -> Self {
2056 self.labels.tag = Some(tag.to_string());
2057 self
2058 }
2059
2060 pub fn subtitle(mut self, subtitle: &str) -> Self {
2061 self.labels.subtitle = Some(subtitle.to_string());
2062 self
2063 }
2064
2065 pub fn xlab(mut self, label: &str) -> Self {
2066 self.labels.x = Some(label.to_string());
2067 self
2068 }
2069
2070 pub fn ylab(mut self, label: &str) -> Self {
2071 self.labels.y = Some(label.to_string());
2072 self
2073 }
2074
2075 pub fn caption(mut self, caption: &str) -> Self {
2076 self.labels.caption = Some(caption.to_string());
2077 self
2078 }
2079
2080 pub fn annotate(mut self, annotation: Annotation) -> Self {
2084 self.annotations.push(annotation);
2085 self
2086 }
2087
2088 pub fn annotate_text(self, label: &str, x: f64, y: f64) -> Self {
2090 self.annotate(Annotation::text(label, x, y))
2091 }
2092
2093 pub fn annotate_rect(self, xmin: f64, xmax: f64, ymin: f64, ymax: f64) -> Self {
2095 self.annotate(Annotation::rect(xmin, xmax, ymin, ymax))
2096 }
2097
2098 pub fn annotate_segment(self, x: f64, y: f64, xend: f64, yend: f64) -> Self {
2100 self.annotate(Annotation::segment(x, y, xend, yend))
2101 }
2102
2103 pub fn locale(mut self, locale: crate::locale::Locale) -> Self {
2126 self.locale = Some(locale);
2127 self
2128 }
2129
2130 pub fn try_build(self) -> Result<crate::build::BuiltPlot, GGError> {
2134 PlotBuilder::build(self)
2135 }
2136
2137 pub fn build(self) -> crate::build::BuiltPlot {
2141 self.try_build().expect("plot build failed")
2142 }
2143
2144 #[cfg(all(feature = "plotters", not(target_arch = "wasm32")))]
2148 pub fn save(self, path: &str) -> Result<(), GGError> {
2149 self.save_with_size(path, 800, 600)
2150 }
2151
2152 #[cfg(all(feature = "plotters", not(target_arch = "wasm32")))]
2154 pub fn save_with_size(self, path: &str, w: u32, h: u32) -> Result<(), GGError> {
2155 let (built, layout) = self.prepare(w, h)?;
2156
2157 let ext = path.rsplit('.').next().unwrap_or("svg").to_lowercase();
2159
2160 match ext.as_str() {
2161 "svg" => {
2162 let backend = plotters::prelude::SVGBackend::new(path, (w, h));
2163 Self::render_into(backend.into_drawing_area(), &built, &layout)?;
2164 }
2165 "png" | "bmp" | "gif" | "jpeg" | "jpg" | "tiff" => {
2166 let backend = plotters::prelude::BitMapBackend::new(path, (w, h));
2167 Self::render_into(backend.into_drawing_area(), &built, &layout)?;
2168 }
2169 _ => {
2170 return Err(GGError::UnsupportedFormat(ext));
2171 }
2172 }
2173
2174 Ok(())
2175 }
2176
2177 #[cfg(all(feature = "plotters", not(target_arch = "wasm32")))]
2183 pub fn render_svg(self) -> Result<String, GGError> {
2184 self.render_svg_with_size(800, 600)
2185 }
2186
2187 #[cfg(all(feature = "plotters", not(target_arch = "wasm32")))]
2191 pub fn render_svg_with_size(self, w: u32, h: u32) -> Result<String, GGError> {
2192 let (built, layout) = self.prepare(w, h)?;
2193 let mut buf = String::new();
2194 {
2195 let backend = plotters::prelude::SVGBackend::with_string(&mut buf, (w, h));
2196 Self::render_into(backend.into_drawing_area(), &built, &layout)?;
2197 }
2198 Ok(buf)
2199 }
2200
2201 pub fn render_svg_native(self) -> Result<String, GGError> {
2205 self.render_svg_native_with_size(800, 600)
2206 }
2207
2208 pub fn render_svg_native_with_size(self, w: u32, h: u32) -> Result<String, GGError> {
2216 Ok(self.render_native(w, h)?.0.finish())
2217 }
2218
2219 pub fn render_svg_native_with_warnings(
2224 self,
2225 w: u32,
2226 h: u32,
2227 ) -> Result<(String, Vec<String>), GGError> {
2228 let (backend, warnings, _) = self.render_native(w, h)?;
2229 Ok((backend.finish(), warnings))
2230 }
2231
2232 pub fn render_svg_native_at(self, x: f64, y: f64, w: u32, h: u32) -> Result<String, GGError> {
2244 Ok(self.render_native(w, h)?.0.finish_fragment(x, y))
2245 }
2246
2247 pub fn render_svg_area_with_size(self, w: u32, h: u32) -> Result<(String, [f64; 4]), GGError> {
2251 let (backend, _, pa) = self.render_native(w, h)?;
2252 Ok((backend.finish(), [pa.x, pa.y, pa.width, pa.height]))
2253 }
2254
2255 fn render_native(
2259 self,
2260 w: u32,
2261 h: u32,
2262 ) -> Result<
2263 (
2264 crate::render::svg_backend::SvgBackend,
2265 Vec<String>,
2266 crate::render::Rect,
2267 ),
2268 GGError,
2269 > {
2270 let svg_root_data = self.svg_root_data;
2271 let root_attrs = self.root_attrs.clone();
2272 let svg_warnings_attr = self.svg_warnings_attr;
2273 let svg_formatted_values = self.svg_formatted_values;
2274 let (built, layout) = self.prepare(w, h)?;
2275 let pa = layout.plot_area.clone();
2276 let mut backend = crate::render::svg_backend::SvgBackend::new(w, h, pa.clone());
2277 backend.set_formatted_values(svg_formatted_values);
2278 let existing_root_attrs = if svg_root_data {
2279 let attrs = crate::render::svg_backend::root_data_attrs(&built);
2280 backend.set_root_attrs(attrs.clone());
2281 let mut reserved = attrs;
2285 reserved.push(("data-plot".to_string(), String::new()));
2286 reserved
2287 } else {
2288 backend.without_plot_attr();
2289 Vec::new()
2290 };
2291 PlotRenderer::render(&built, &mut backend).map_err(GGError::Render)?;
2292 let mut warnings = built.warnings;
2293 warnings.extend(backend.take_warnings());
2294
2295 let mut reserved = existing_root_attrs;
2296 if svg_warnings_attr {
2297 reserved.push(("data-warnings".to_string(), String::new()));
2298 }
2299 let extra = crate::render::svg_backend::validate_root_attrs(
2300 &root_attrs,
2301 &reserved,
2302 "plot",
2303 &mut warnings,
2304 );
2305 backend.extend_root_attrs(extra);
2306 if svg_warnings_attr {
2307 backend.extend_root_attrs(vec![(
2308 "data-warnings".to_string(),
2309 crate::render::svg_backend::warnings_json(&warnings),
2310 )]);
2311 }
2312 Ok((backend, warnings, pa))
2313 }
2314
2315 #[cfg(feature = "canvas")]
2320 pub fn render_rgba_with_size(self, w: u32, h: u32) -> Result<(u32, u32, Vec<u8>), GGError> {
2321 let (built, layout) = self.prepare(w, h)?;
2322 let mut backend =
2323 crate::render::pixel_backend::PixelBackend::new(w, h, layout.plot_area.clone());
2324 PlotRenderer::render(&built, &mut backend).map_err(GGError::Render)?;
2325 let (ow, oh) = backend.dimensions();
2326 Ok((ow, oh, backend.into_rgba()))
2327 }
2328
2329 #[cfg(feature = "canvas")]
2333 pub fn render_rgba_area_with_size(
2334 self,
2335 w: u32,
2336 h: u32,
2337 ) -> Result<(Vec<u8>, [f64; 4]), GGError> {
2338 let (built, layout) = self.prepare(w, h)?;
2339 let pa = layout.plot_area.clone();
2340 let mut backend = crate::render::pixel_backend::PixelBackend::new(w, h, pa.clone());
2341 PlotRenderer::render(&built, &mut backend).map_err(GGError::Render)?;
2342 Ok((backend.into_rgba(), [pa.x, pa.y, pa.width, pa.height]))
2343 }
2344
2345 #[cfg(feature = "canvas")]
2349 pub fn render_png_raster_with_size(self, w: u32, h: u32) -> Result<Vec<u8>, GGError> {
2350 let (built, layout) = self.prepare(w, h)?;
2351 let mut backend =
2352 crate::render::pixel_backend::PixelBackend::new(w, h, layout.plot_area.clone());
2353 PlotRenderer::render(&built, &mut backend).map_err(GGError::Render)?;
2354 backend.into_png().map_err(GGError::Render)
2355 }
2356
2357 #[cfg(all(feature = "plotters", not(target_arch = "wasm32")))]
2364 pub fn render_png(self) -> Result<Vec<u8>, GGError> {
2365 self.render_png_with_size(800, 600)
2366 }
2367
2368 #[cfg(all(feature = "plotters", not(target_arch = "wasm32")))]
2371 pub fn render_png_with_size(self, w: u32, h: u32) -> Result<Vec<u8>, GGError> {
2372 let (built, layout) = self.prepare(w, h)?;
2373
2374 let mut rgb = vec![0u8; (w as usize) * (h as usize) * 3];
2377 {
2378 let backend = plotters::prelude::BitMapBackend::with_buffer(&mut rgb, (w, h));
2379 Self::render_into(backend.into_drawing_area(), &built, &layout)?;
2380 }
2381
2382 let img = image::RgbImage::from_raw(w, h, rgb).ok_or_else(|| {
2383 GGError::Render(RenderError::BackendError(
2384 "PNG buffer size mismatch".to_string(),
2385 ))
2386 })?;
2387 let mut out = std::io::Cursor::new(Vec::new());
2388 img.write_to(&mut out, image::ImageOutputFormat::Png)
2389 .map_err(|e| GGError::Render(RenderError::BackendError(format!("{:?}", e))))?;
2390 Ok(out.into_inner())
2391 }
2392
2393 pub(crate) fn prepare(
2395 self,
2396 w: u32,
2397 h: u32,
2398 ) -> Result<(crate::build::BuiltPlot, PlotLayout), GGError> {
2399 let (mut built, meta) = self.build_for_render()?;
2400 let layout = Self::layout_built(&mut built, &meta, w, h);
2401 Ok((built, layout))
2402 }
2403
2404 pub(crate) fn build_for_render(self) -> Result<(crate::build::BuiltPlot, RenderMeta), GGError> {
2407 let plot = self;
2408
2409 let has_title = plot.labels.title.is_some();
2410 let has_subtitle = plot.labels.subtitle.is_some();
2411 let has_caption = plot.labels.caption.is_some();
2412 let has_legend = plot.has_legend_mapping();
2413 let x_label = plot.labels.x.clone();
2414 let y_label = plot.labels.y.clone();
2415
2416 let mut built = PlotBuilder::build(plot)?;
2417
2418 built.theme.resolve_inheritance();
2420
2421 if let Some(ref label) = x_label {
2423 if let Some(s) = built.scales.get_mut(&crate::aes::Aesthetic::X) {
2424 s.set_name(label);
2425 }
2426 }
2427 if let Some(ref label) = y_label {
2428 if let Some(s) = built.scales.get_mut(&crate::aes::Aesthetic::Y) {
2429 s.set_name(label);
2430 }
2431 }
2432
2433 Ok((
2434 built,
2435 RenderMeta {
2436 has_title,
2437 has_subtitle,
2438 has_caption,
2439 has_legend,
2440 },
2441 ))
2442 }
2443
2444 pub(crate) fn layout_built(
2447 built: &mut crate::build::BuiltPlot,
2448 meta: &RenderMeta,
2449 w: u32,
2450 h: u32,
2451 ) -> PlotLayout {
2452 let RenderMeta {
2453 has_title,
2454 has_subtitle,
2455 has_caption,
2456 has_legend,
2457 } = *meta;
2458 let x_axis_top = built
2459 .scales
2460 .get(&crate::aes::Aesthetic::X)
2461 .map(|s| s.axis_position_opposite())
2462 .unwrap_or(false);
2463
2464 let flipped = built.coord.is_flipped();
2468 let h_aes = if flipped {
2469 crate::aes::Aesthetic::Y
2470 } else {
2471 crate::aes::Aesthetic::X
2472 };
2473 let v_aes = if flipped {
2474 crate::aes::Aesthetic::X
2475 } else {
2476 crate::aes::Aesthetic::Y
2477 };
2478
2479 let x_rot: Option<(f64, f64)> =
2483 if built.theme.axis_text_x.visible && built.theme.axis_text_x.angle.abs() < 1.0 {
2484 built.scales.get(&h_aes).and_then(|hs| {
2485 if !hs.is_discrete() {
2486 return None;
2487 }
2488 let breaks = hs.breaks();
2489 let n = breaks.len().max(1) as f64;
2490 let max_ch = breaks
2491 .iter()
2492 .map(|(_, l)| l.chars().count())
2493 .max()
2494 .unwrap_or(0) as f64;
2495 let font = built.theme.axis_text_x.size;
2496 let label_w = max_ch * font * 0.6; let spacing = 0.9 * w as f64 / n; if label_w <= spacing {
2499 return None; }
2501 let vertical = label_w * 0.707 > spacing;
2505 let angle = if vertical { 270.0 } else { 45.0 };
2506 let drop = if vertical { label_w } else { label_w * 0.707 };
2507 Some((angle, drop + font + 4.0))
2508 })
2509 } else {
2510 None
2511 };
2512 let x_label_height = x_rot.map(|(_, drop)| drop);
2513 if let Some((angle, _)) = x_rot {
2514 built.theme.axis_text_x.angle = angle;
2515 }
2516
2517 let y_label_width = if built.theme.axis_text_y.visible {
2521 built.scales.get(&v_aes).map(|vs| {
2522 let max_ch = vs
2523 .breaks()
2524 .iter()
2525 .map(|(_, l)| l.chars().count())
2526 .max()
2527 .unwrap_or(0);
2528 let measured = max_ch as f64 * built.theme.axis_text_y.size * 0.6 + 4.0;
2529 let default = built.theme.axis_text_y.size * 3.5 + 4.0;
2530 measured.max(default).min(0.35 * w as f64)
2531 })
2532 } else {
2533 None
2534 };
2535
2536 let mut extras = crate::render::layout::LayoutExtras::default();
2547 let can_reserve = built.facet.is_none()
2548 && built.theme.axis_text_x.visible
2549 && built.theme.axis_text_x_fit == crate::theme::AxisLabelFit::Reserve
2550 && !built.coord.is_polar()
2551 && built.coord.supports_axis_reserve();
2552 if can_reserve {
2553 if let Some(hs) = built.scales.get(&h_aes) {
2554 let breaks = hs.breaks();
2555 if let (Some(first), Some(last)) = (breaks.first(), breaks.last()) {
2556 let angle = built.theme.axis_text_x.angle;
2557 let anchor = if angle.abs() > 10.0 { 1.0 } else { 0.5 };
2558 let size = built.theme.axis_text_x.size;
2559 const EDGE: f64 = 1.0;
2560 for _ in 0..2 {
2561 let layout = PlotLayout::compute_with_extras(
2562 w as f64,
2563 h as f64,
2564 &built.theme,
2565 has_title,
2566 has_subtitle,
2567 has_caption,
2568 has_legend,
2569 x_axis_top,
2570 y_label_width,
2571 x_label_height,
2572 extras,
2573 );
2574 let area = &layout.plot_area;
2575 let edge_y = area.y + area.height;
2576 let to_point = |pos: f64| if flipped { (0.0, pos) } else { (pos, 0.0) };
2577 let px_first = built.coord.transform(to_point(first.0), area).0;
2578 let px_last = built.coord.transform(to_point(last.0), area).0;
2579 let bbox_first = crate::render::text::label_bbox(
2580 &first.1, size, px_first, edge_y, angle, anchor,
2581 );
2582 let bbox_last = crate::render::text::label_bbox(
2583 &last.1, size, px_last, edge_y, angle, anchor,
2584 );
2585 let deficit_left = (EDGE - bbox_first[0]).max(0.0);
2586 let deficit_right = (bbox_last[2] - (w as f64 - EDGE)).max(0.0);
2587 if deficit_left <= 0.0 && deficit_right <= 0.0 {
2588 break;
2589 }
2590 let (p_first, p_last) = (first.0, last.0);
2599 if deficit_left > 0.0 && deficit_right > 0.0 {
2600 let (a, b, c, d) = (1.0 - p_first, p_first, 1.0 - p_last, p_last);
2605 let det = a * d - b * c;
2606 if det.abs() > 0.01 {
2607 extras.left +=
2608 ((deficit_left * d + deficit_right * b) / det).max(0.0);
2609 extras.right +=
2610 ((deficit_right * a + deficit_left * c) / det).max(0.0);
2611 } else {
2612 extras.left += deficit_left / (1.0 - p_first).max(0.05);
2616 extras.right += deficit_right / p_last.max(0.05);
2617 }
2618 } else if deficit_left > 0.0 {
2619 extras.left += deficit_left / (1.0 - p_first).max(0.05);
2620 } else if deficit_right > 0.0 {
2621 extras.right += deficit_right / p_last.max(0.05);
2622 }
2623 const MIN_PANEL_WIDTH: f64 = 50.0;
2638 let cap = (w as f64 - MIN_PANEL_WIDTH).max(0.0);
2639 if extras.left + extras.right > cap {
2640 let scale = if extras.left + extras.right > 0.0 {
2641 cap / (extras.left + extras.right)
2642 } else {
2643 0.0
2644 };
2645 extras.left *= scale;
2646 extras.right *= scale;
2647 }
2648 }
2649 }
2650 }
2651 }
2652
2653 if has_legend
2659 && !matches!(
2660 built.theme.legend_position,
2661 crate::theme::LegendPosition::None
2662 )
2663 {
2664 let legend_aes =
2665 crate::guide::legend::legend_scales_for(&built.scales, &built.suppressed_aes);
2666 if legend_aes.len() == 1 {
2667 if let Some(scale) = built.scales.get(&legend_aes[0]) {
2668 if scale.is_discrete() {
2669 let breaks = scale.breaks();
2670 let layout0 = PlotLayout::compute_with_extras(
2671 w as f64,
2672 h as f64,
2673 &built.theme,
2674 has_title,
2675 has_subtitle,
2676 has_caption,
2677 has_legend,
2678 x_axis_top,
2679 y_label_width,
2680 x_label_height,
2681 extras,
2682 );
2683 let (legend_x0, _, mut is_horizontal) =
2684 crate::guide::legend::legend_position(&built.theme, &layout0.plot_area);
2685 if let Some(dir) = built.theme.legend_direction {
2686 is_horizontal =
2687 matches!(dir, crate::theme::LegendDirection::Horizontal);
2688 }
2689 let avail_width =
2690 (layout0.plot_area.x + layout0.plot_area.width - legend_x0).max(0.0);
2691 let mut legend_warnings = Vec::new();
2692 let validated_guide = built
2693 .guide_legend
2694 .validated(breaks.len(), &mut legend_warnings);
2695 let plan = crate::guide::legend::plan_legend(
2696 &breaks,
2697 &validated_guide,
2698 &built.theme,
2699 avail_width,
2700 layout0.plot_area.height,
2701 is_horizontal,
2702 );
2703 built.warnings.extend(legend_warnings);
2704 if let Some(plan) = &plan {
2705 extras.legend_extra_rows = plan.rows.saturating_sub(1);
2706 extras.legend_extra_width =
2713 plan.cols.saturating_sub(1) as f64 * plan.col_pitch;
2714 if !is_horizontal && plan.cols > 1 {
2728 let final_layout = PlotLayout::compute_with_extras(
2729 w as f64,
2730 h as f64,
2731 &built.theme,
2732 has_title,
2733 has_subtitle,
2734 has_caption,
2735 has_legend,
2736 x_axis_top,
2737 y_label_width,
2738 x_label_height,
2739 extras,
2740 );
2741 let (legend_x_final, _, _) = crate::guide::legend::legend_position(
2742 &built.theme,
2743 &final_layout.plot_area,
2744 );
2745 let swatch_and_gap =
2746 built.theme.legend_key_width + built.theme.legend_spacing;
2747 let text_size = built.theme.legend_text.size;
2748 let legend_right_edge = plan
2749 .entries
2750 .iter()
2751 .map(|e| {
2752 let label = &breaks[e.break_idx].1;
2753 legend_x_final
2754 + e.x_offset
2755 + swatch_and_gap
2756 + crate::render::text::text_width(label, text_size)
2757 })
2758 .fold(0.0_f64, f64::max);
2759 if legend_right_edge > w as f64 {
2760 built.warnings.push(format!(
2761 "legend: the {}-column grid still doesn't fit the {w}px chart (needs about {} px)",
2762 plan.cols,
2763 legend_right_edge.ceil() as i64
2764 ));
2765 }
2766 }
2767 if let Some(ov) = &plan.overflow {
2768 let n = breaks.len();
2769 let shown = n - ov.dropped;
2770 built.warnings.push(format!(
2771 "legend: {} of {n} entries don't fit the chart; the legend shows {shown} and \"+{} more\"",
2772 ov.dropped, ov.dropped
2773 ));
2774 }
2775 }
2776 built.legend_plan = plan;
2782 }
2783 }
2784 }
2785 }
2786
2787 PlotLayout::compute_with_extras(
2788 w as f64,
2789 h as f64,
2790 &built.theme,
2791 has_title,
2792 has_subtitle,
2793 has_caption,
2794 has_legend,
2795 x_axis_top,
2796 y_label_width,
2797 x_label_height,
2798 extras,
2799 )
2800 }
2801
2802 #[cfg(all(feature = "plotters", not(target_arch = "wasm32")))]
2805 fn render_into<DB>(
2806 area: plotters::drawing::DrawingArea<DB, plotters::coord::Shift>,
2807 built: &crate::build::BuiltPlot,
2808 layout: &PlotLayout,
2809 ) -> Result<(), GGError>
2810 where
2811 DB: plotters::prelude::DrawingBackend,
2812 DB::ErrorType: 'static,
2813 {
2814 area.fill(&plotters::prelude::WHITE)
2815 .map_err(|e| GGError::Render(RenderError::BackendError(format!("{:?}", e))))?;
2816 let mut adapter = PlottersAdapter::new(&area, layout.plot_area.clone());
2817 PlotRenderer::render(built, &mut adapter).map_err(GGError::Render)?;
2818 area.present()
2819 .map_err(|e| GGError::Render(RenderError::BackendError(format!("{:?}", e))))?;
2820 Ok(())
2821 }
2822
2823 #[cfg(all(feature = "plotters", not(target_arch = "wasm32")))]
2826 pub fn ggsave(
2827 self,
2828 path: &str,
2829 width_inches: f64,
2830 height_inches: f64,
2831 dpi: f64,
2832 ) -> Result<(), GGError> {
2833 let w = (width_inches * dpi) as u32;
2834 let h = (height_inches * dpi) as u32;
2835 self.save_with_size(path, w, h)
2836 }
2837
2838 fn has_legend_mapping(&self) -> bool {
2839 self.mapping.mappings.iter().any(|m| {
2840 matches!(
2841 m.aesthetic,
2842 crate::aes::Aesthetic::Color
2843 | crate::aes::Aesthetic::Fill
2844 | crate::aes::Aesthetic::Shape
2845 | crate::aes::Aesthetic::Linetype
2846 | crate::aes::Aesthetic::Size
2847 | crate::aes::Aesthetic::Alpha
2848 )
2849 })
2850 }
2851}
2852
2853#[derive(Clone, Copy, Debug)]
2855pub(crate) struct RenderMeta {
2856 pub(crate) has_title: bool,
2857 pub(crate) has_subtitle: bool,
2858 pub(crate) has_caption: bool,
2859 pub(crate) has_legend: bool,
2860}
2861
2862#[derive(Debug)]
2864pub enum GGError {
2865 Render(RenderError),
2866 UnsupportedFormat(String),
2867 Io(std::io::Error),
2868 ValidationError(String),
2869}
2870
2871impl std::fmt::Display for GGError {
2872 fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
2873 match self {
2874 GGError::Render(e) => write!(f, "Render error: {e}"),
2875 GGError::UnsupportedFormat(ext) => write!(f, "Unsupported output format: {ext}"),
2876 GGError::Io(e) => write!(f, "IO error: {e}"),
2877 GGError::ValidationError(msg) => write!(f, "Validation error: {msg}"),
2878 }
2879 }
2880}
2881
2882impl std::error::Error for GGError {}