wombat 1.1.0

Hex file viewer
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121

//! Histogram

use bladvak::eframe::egui;
use bladvak::eframe::egui::ahash::{HashMap, HashMapExt};
use bladvak::egui_plot::{Bar, BarChart, Legend, Plot};
use bladvak::errors::ErrorManager;

/// Histogram data
#[derive(serde::Deserialize, serde::Serialize, Debug)]
pub(crate) struct Histogram {
    /// is open
    pub(crate) is_open: bool,

    #[serde(skip)]
    /// histogram values
    data: Option<HashMap<u8, usize>>,
    /// vertical or horizontal
    vertical: bool,
    /// bar width
    bar_width: f64,
}

impl Histogram {
    /// Create empty histogram
    pub(crate) fn new() -> Self {
        Self {
            is_open: false,
            data: None,
            vertical: false,
            bar_width: 1.0,
        }
    }

    /// reset data
    pub(crate) fn reset(&mut self) {
        self.data = None;
    }

    /// Calculate the histogram
    fn calculate_histogram(binary_data: &[u8]) -> HashMap<u8, usize> {
        let mut map = binary_data.iter().fold(HashMap::new(), |mut map, &byte| {
            *map.entry(byte).or_insert(0) += 1;
            map
        });
        for idx in 0u8..=255 {
            map.entry(idx).or_insert(0);
        }
        map
    }

    /// Show the histogram ui
    pub(crate) fn ui(
        &mut self,
        binary_data: &[u8],
        ui: &mut egui::Ui,
        _error_manager: &mut ErrorManager,
    ) {
        if self.is_open {
            if self.data.is_some() {
                let mut is_open = self.is_open;
                egui::Window::new("Histogram")
                    .open(&mut is_open)
                    .vscroll(true)
                    .show(ui.ctx(), |ui| {
                        ui.horizontal(|ui| {
                            ui.label("Orientation:");
                            ui.selectable_value(&mut self.vertical, true, "Vertical");
                            ui.selectable_value(&mut self.vertical, false, "Horizontal");
                        });
                        ui.horizontal(|ui| {
                            ui.label("Bar width");
                            ui.add(egui::Slider::new(&mut self.bar_width, 0.001..=2.0));
                        });
                        self.show_plot(ui);
                    });
                self.is_open = is_open;
            } else {
                self.data = Some(Self::calculate_histogram(binary_data));
            }
        }
    }

    /// Show the plot
    pub fn show_plot(&self, ui: &mut egui::Ui) {
        let Some(data) = &self.data else {
            return;
        };
        let mut min = f64::INFINITY;
        let mut max = f64::NEG_INFINITY;
        let mut chart = BarChart::new(
            "Histogram",
            data.iter()
                .map(|(byte, value)| {
                    #[allow(clippy::cast_precision_loss)]
                    let value = *value as f64;
                    min = value.min(min);
                    max = value.max(max);
                    Bar::new(f64::from(*byte), value)
                })
                .collect(),
        )
        .width(self.bar_width)
        .color(egui::Color32::LIGHT_BLUE);

        if !self.vertical {
            chart = chart.horizontal();
        }

        let x_label = "bytes (0 to 255)";
        let y_label = format!("count (min:{min} max:{max})");
        Plot::new("Distribution")
            .legend(Legend::default())
            .x_axis_label(if self.vertical { x_label } else { &y_label })
            .y_axis_label(if self.vertical { &y_label } else { x_label })
            .clamp_grid(true)
            .allow_zoom(egui::Vec2b::new(true, true))
            // .allow_drag(egui::Vec2b::new(true, true))
            // .allow_scroll(egui::Vec2b::new(true, true))
            .show(ui, |plot_ui| plot_ui.bar_chart(chart));
    }
}