mirror of https://github.com/terhechte/postsack
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
59 lines
1.6 KiB
Rust
59 lines
1.6 KiB
Rust
use eframe::egui::Rect as EguiRect;
|
|
use treemap::{Rect, TreemapLayout};
|
|
|
|
use super::segment::Segment;
|
|
|
|
/// A small NewType so that we can keep all the `TreeMap` code in here and don't
|
|
/// have to do the layout calculation in a widget.
|
|
#[derive(Debug)]
|
|
pub struct Segmentation {
|
|
items: Vec<Segment>,
|
|
pub selected: Option<Segment>,
|
|
pub range: Option<std::ops::RangeInclusive<usize>>,
|
|
}
|
|
|
|
impl Segmentation {
|
|
pub fn new(items: Vec<Segment>) -> Self {
|
|
Self {
|
|
items,
|
|
selected: None,
|
|
range: None,
|
|
}
|
|
}
|
|
|
|
pub fn len(&self) -> usize {
|
|
self.items.len()
|
|
}
|
|
|
|
/// Update the layout information in the Segments
|
|
/// based on the current size
|
|
pub fn update_layout(&mut self, rect: EguiRect) {
|
|
let layout = TreemapLayout::new();
|
|
let bounds = Rect::from_points(
|
|
rect.left() as f64,
|
|
rect.top() as f64,
|
|
rect.width() as f64,
|
|
rect.height() as f64,
|
|
);
|
|
layout.layout_items(self.items(), bounds);
|
|
}
|
|
|
|
/// The total amount of items in all the `Segments`.
|
|
/// E.g. the sum of the count of the `Segments`
|
|
pub fn element_count(&self) -> usize {
|
|
self.items.iter().map(|e| e.count).sum::<usize>()
|
|
}
|
|
|
|
/// The items in this `Segmentation`, with range applied
|
|
pub fn items(&mut self) -> &mut [Segment] {
|
|
match &self.range {
|
|
Some(n) => {
|
|
// we reverse the range
|
|
let reversed_range = (self.len() - n.end())..=(self.len() - 1);
|
|
&mut self.items[reversed_range]
|
|
}
|
|
None => self.items.as_mut_slice(),
|
|
}
|
|
}
|
|
}
|