pixmux 0.1.0

An interactive TUI for exploring CSV-backed image datasets.
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

use std::ptr;

use pixmux::Pane;
use ratatui::Frame;
use ratatui::layout::{Constraint, Layout, Rect};
use ratatui::style::Color;
use ratatui::widgets::Block;
use ratatui_image::{StatefulImage, picker};

use crate::app::App;

pub struct ImageTabUI;

impl ImageTabUI {
    fn render_pane(pane: &Pane, frame: &mut Frame, area: Rect, app: &App, picker: &picker::Picker) {
        match pane {
            Pane::Leaf { image_id } => {
                let block = Block::bordered();

                let current_pane = app.pane_tree.get_node_at(&app.current_pane_path).unwrap();
                if ptr::eq(current_pane, pane) {
                    frame.render_widget(block.clone().style(Color::LightYellow), area);
                } else {
                    frame.render_widget(block.clone(), area);
                }

                let imagedir = app.imagedir_paths.get(app.current_datarow_index).unwrap();
                let imagefile_basename = app.imagefile_basenames.get(*image_id).unwrap();
                let imagefile = imagedir.join(imagefile_basename);

                if imagefile.exists() {
                    let image_source = image::ImageReader::open(imagefile)
                        .unwrap()
                        .decode()
                        .unwrap();
                    let mut image = picker.new_resize_protocol(image_source);

                    frame.render_stateful_widget(
                        StatefulImage::default(),
                        block.inner(area),
                        &mut image,
                    );
                }
            }
            Pane::Split {
                direction,
                pct,
                first,
                second,
                ..
            } => {
                let constraints = vec![
                    Constraint::Percentage(*pct as u16),
                    Constraint::Percentage((100 - pct) as u16),
                ];
                let chunks = Layout::default()
                    .spacing(-1)
                    .direction(*direction)
                    .constraints(constraints)
                    .split(area);
                Self::render_pane(first, frame, chunks[0], app, picker);
                Self::render_pane(second, frame, chunks[1], app, picker);
            }
        }
    }

    pub fn render(frame: &mut Frame, area: Rect, app: &App) {
        let picker = picker::Picker::from_query_stdio().unwrap();
        Self::render_pane(&app.pane_tree, frame, area, app, &picker);
    }
}