rvlib/

main_loop.rs

#![deny(clippy::all)]
#![forbid(unsafe_code)]
use crate::autosave::{AUTOSAVE_INTERVAL_S, autosave};
use crate::control::{Control, Info};
use crate::drawme::ImageInfo;
use crate::events::{Events, KeyCode};
use crate::file_util::{DEFAULT_PRJ_PATH, get_prj_name};
use crate::history::{History, Record};
use crate::menu::{Menu, ToolSelectMenu, are_tools_active};
use crate::result::trace_ok_err;
use crate::tools::{
    ALWAYS_ACTIVE_ZOOM, BBOX_NAME, Manipulate, ToolState, ToolWrapper, ZOOM_NAME, make_tool_vec,
};
use crate::util::Visibility;
use crate::world::World;
use crate::{Annotation, UpdateView, apply_tool_method_mut, httpserver, image_util, measure_time};
use image::{DynamicImage, GenericImageView};
use rvimage_domain::{BbI, PtI, RvResult, ShapeF};
use std::fmt::Debug;
use std::mem;
use std::path::{Path, PathBuf};
use std::sync::mpsc::Receiver;
use std::time::Instant;
use tracing::{error, info, warn};

fn pos_2_string_gen<T>(im: &T, x: u32, y: u32) -> String
where
    T: GenericImageView,
    <T as GenericImageView>::Pixel: Debug,
{
    let p = format!("{:?}", im.get_pixel(x, y));
    format!("({x}, {y}) -> ({})", &p[6..p.len() - 2])
}

fn pos_2_string(im: &DynamicImage, x: u32, y: u32) -> String {
    if x < im.width() && y < im.height() {
        image_util::apply_to_matched_image(
            im,
            |im| pos_2_string_gen(im, x, y),
            |im| pos_2_string_gen(im, x, y),
            |im| pos_2_string_gen(im, x, y),
            |im| pos_2_string_gen(im, x, y),
        )
    } else {
        "".to_string()
    }
}

fn get_pixel_on_orig_str(world: &World, mouse_pos: &Option<PtI>) -> Option<String> {
    mouse_pos.map(|p| pos_2_string(world.data.im_background(), p.x, p.y))
}

fn apply_tools(
    tools: &mut [ToolState],
    mut world: World,
    mut history: History,
    input_event: &Events,
) -> (World, History) {
    let aaz = tools
        .iter_mut()
        .find(|t| t.name == ALWAYS_ACTIVE_ZOOM)
        .unwrap();
    (world, history) = apply_tool_method_mut!(aaz, events_tf, world, history, input_event);
    let aaz_hbu = apply_tool_method_mut!(aaz, has_been_used, input_event);
    let not_aaz = tools
        .iter_mut()
        .filter(|t| t.name != ALWAYS_ACTIVE_ZOOM && t.is_active());
    for t in not_aaz {
        (world, history) = apply_tool_method_mut!(t, events_tf, world, history, input_event);
        if aaz_hbu == Some(true) {
            (world, history) = apply_tool_method_mut!(t, on_always_active_zoom, world, history);
        }
    }
    (world, history)
}

macro_rules! activate_tool_event {
    ($key:ident, $name:expr, $input:expr, $rat:expr, $tools:expr) => {
        if $input.held_alt() && $input.pressed(KeyCode::$key) {
            $rat = Some(
                $tools
                    .iter()
                    .enumerate()
                    .find(|(_, t)| t.name == $name)
                    .unwrap()
                    .0,
            );
        }
    };
}

fn find_active_tool(tools: &[ToolState]) -> Option<&str> {
    tools
        .iter()
        .find(|t| t.is_active() && !t.is_always_active())
        .map(|t| t.name)
}

pub struct MainEventLoop {
    menu: Menu,
    tools_select_menu: ToolSelectMenu,
    world: World,
    ctrl: Control,
    history: History,
    tools: Vec<ToolState>,
    recently_clicked_tool_idx: Option<usize>,
    rx_from_http: Option<Receiver<RvResult<String>>>,
    http_addr: String,
    autosave_timer: Instant,
    next_image_held_timer: Instant,
}
impl Default for MainEventLoop {
    fn default() -> Self {
        let file_path = std::env::args().nth(1).map(PathBuf::from);
        Self::new(file_path)
    }
}

impl MainEventLoop {
    pub fn new(prj_file_path: Option<PathBuf>) -> Self {
        let ctrl = Control::new();

        let mut world = World::empty();
        let mut tools = make_tool_vec();
        for t in &mut tools {
            if t.is_active() {
                (world, _) = t.activate(world, History::default());
            }
        }
        let http_addr = ctrl.http_address();
        // http server state
        let rx_from_http = if let Ok((_, rx)) = httpserver::launch(http_addr.clone()) {
            Some(rx)
        } else {
            None
        };
        let mut self_ = Self {
            world,
            ctrl,
            tools,
            http_addr,
            tools_select_menu: ToolSelectMenu::default(),
            menu: Menu::default(),
            history: History::default(),
            recently_clicked_tool_idx: None,
            rx_from_http,
            autosave_timer: Instant::now(),
            next_image_held_timer: Instant::now(),
        };

        trace_ok_err(self_.load_prj_during_startup(prj_file_path));
        self_
    }
    pub fn one_iteration(
        &mut self,
        e: &Events,
        ui_image_rect: Option<ShapeF>,
        tmp_anno_buffer: Option<Annotation>,
        request_file_label_to_load: Option<&str>,
        ui: &mut egui::Ui,
    ) -> RvResult<(UpdateView, bool, bool, &str)> {
        measure_time!("whole iteration", {
            measure_time!("part 1", {
                self.world.set_image_rect(ui_image_rect);
                self.world.update_view.tmp_anno_buffer = tmp_anno_buffer;
                let project_loaded_in_curr_iter = self.menu.ui(
                    ui,
                    &mut self.ctrl,
                    &mut self.world.data.tools_data_map,
                    find_active_tool(&self.tools),
                );
                let new_annos = self
                    .ctrl
                    .check_wand_many_output(&mut self.world.data.tools_data_map)?;
                if new_annos {
                    self.history
                        .push(Record::new(self.world.clone(), "wand many"));
                }
                if new_annos && let Some(active_tool_name) = find_active_tool(&self.tools) {
                    self.world
                        .request_redraw_annotations(active_tool_name, Visibility::All);
                }

                self.world.data.meta_data.ssh_cfg = Some(self.ctrl.cfg.ssh_cfg());
                if project_loaded_in_curr_iter {
                    for t in &mut self.tools {
                        self.world = t.deactivate(mem::take(&mut self.world));
                    }
                }
                if let Some(elf) = &self.ctrl.log_export_path {
                    trace_ok_err(self.ctrl.export_logs(elf));
                }
                if self.ctrl.log_export_path.is_some() {
                    self.ctrl.log_export_path = None;
                }
                if e.held_ctrl() && e.pressed(KeyCode::S) {
                    let prj_path = self.ctrl.cfg.current_prj_path().to_path_buf();
                    if let Err(e) = self
                        .ctrl
                        .save(prj_path, &self.world.data.tools_data_map, true)
                    {
                        self.menu
                            .show_info(Info::Error(format!("could not save project due to {e:?}")));
                    }
                }
            });

            egui::Panel::right("my_panel")
                .show(ui, |ui| {
                    ui.vertical(|ui| {
                        self.tools_select_menu.ui(
                            ui,
                            &mut self.tools,
                            &mut self.world.data.tools_data_map,
                        )
                    })
                    .inner
                })
                .inner?;

            // tool activation
            if self.recently_clicked_tool_idx.is_none() {
                self.recently_clicked_tool_idx = self.tools_select_menu.recently_clicked_tool();
            }
            if let (Some(idx_active), Some(_)) = (
                self.recently_clicked_tool_idx,
                &self.world.data.meta_data.file_path_absolute(),
            ) && !self.ctrl.flags().is_loading_screen_active
            {
                // first deactivate, then activate
                for (i, t) in self.tools.iter_mut().enumerate() {
                    if i != idx_active && t.is_active() && !t.is_always_active() {
                        let meta_data = self.ctrl.meta_data(
                            self.ctrl.file_selected_idx,
                            Some(self.ctrl.flags().is_loading_screen_active),
                        );
                        self.world.data.meta_data = meta_data;
                        self.world = t.deactivate(mem::take(&mut self.world));
                    }
                }
                for (i, t) in self.tools.iter_mut().enumerate() {
                    if i == idx_active {
                        (self.world, self.history) =
                            t.activate(mem::take(&mut self.world), mem::take(&mut self.history));
                    }
                }
                self.recently_clicked_tool_idx = None;
            }

            if e.held_alt() && e.pressed(KeyCode::Q) {
                info!("deactivate all tools");
                let was_any_tool_active = self
                    .tools
                    .iter()
                    .any(|t| t.is_active() && !t.is_always_active());
                for t in self.tools.iter_mut() {
                    if !t.is_always_active() && t.is_active() {
                        let meta_data = self.ctrl.meta_data(
                            self.ctrl.file_selected_idx,
                            Some(self.ctrl.flags().is_loading_screen_active),
                        );
                        self.world.data.meta_data = meta_data;
                        self.world = t.deactivate(mem::take(&mut self.world));
                    }
                }
                if was_any_tool_active {
                    self.history
                        .push(Record::new(self.world.clone(), "deactivation of all tools"));
                }
            }
            // tool activation keyboard shortcuts
            activate_tool_event!(B, BBOX_NAME, e, self.recently_clicked_tool_idx, self.tools);
            activate_tool_event!(Z, ZOOM_NAME, e, self.recently_clicked_tool_idx, self.tools);

            const DOUBLE_SKIP_TH_MS: u128 = 500;
            if e.held_ctrl() && e.pressed(KeyCode::M) {
                self.menu.toggle();
            } else if e.released(KeyCode::F5) {
                if let Err(e) = self.ctrl.reload(None) {
                    self.menu
                        .show_info(Info::Error(format!("could not reload due to {e:?}")));
                }
            } else if e.held(KeyCode::PageDown) || e.held(KeyCode::PageUp) {
                if self.world.data.meta_data.flags.is_loading_screen_active == Some(true) {
                    self.next_image_held_timer = Instant::now();
                } else {
                    let elapsed = self.next_image_held_timer.elapsed().as_millis();
                    let interval = self.ctrl.cfg.usr.image_change_delay_on_held_key_ms as u128;
                    if elapsed > interval {
                        if e.held(KeyCode::PageDown) {
                            self.ctrl.paths_navigator.next();
                        } else if e.held(KeyCode::PageUp) {
                            self.ctrl.paths_navigator.prev();
                        }
                        self.next_image_held_timer = Instant::now();
                    }
                }
            } else if e.released(KeyCode::PageDown)
                && self.next_image_held_timer.elapsed().as_millis() > DOUBLE_SKIP_TH_MS
            {
                self.ctrl.paths_navigator.next();
            } else if e.released(KeyCode::PageUp)
                && self.next_image_held_timer.elapsed().as_millis() > DOUBLE_SKIP_TH_MS
            {
                self.ctrl.paths_navigator.prev();
            } else if e.released(KeyCode::Escape) {
                self.world.set_zoom_box(None);
            }

            // check for file load request from image/thumbnail UI
            if let Some(file_label) = request_file_label_to_load {
                self.ctrl.paths_navigator.select_file_label(file_label);
                self.ctrl
                    .paths_navigator
                    .activate_scroll_to_selected_label();
            }
            // check for new image requests from http server
            let rx_match = &self.rx_from_http.as_ref().map(|rx| rx.try_iter().last());
            if let Some(Some(Ok(file_label))) = rx_match {
                self.ctrl.paths_navigator.select_file_label(file_label);
                self.ctrl
                    .paths_navigator
                    .activate_scroll_to_selected_label();
            } else if let Some(Some(Err(e))) = rx_match {
                // if the server thread sends an error we restart the server
                warn!("{e:?}");
                (self.http_addr, self.rx_from_http) =
                    match httpserver::restart_with_increased_port(&self.http_addr) {
                        Ok(x) => x,
                        Err(e) => {
                            error!("{e:?}");
                            (self.http_addr.to_string(), None)
                        }
                    };
            }

            let world_idx_pair = measure_time!("load image", {
                // Regular per-frame update that runs regardless of whether tools
                // are active. Lets tools flush pending menu state, e.g. attribute edits.
                for t in &mut self.tools {
                    if t.is_active() {
                        self.world = t.update(mem::take(&mut self.world));
                    }
                }
                // load new image if requested by a menu click or by the http server
                if e.held_ctrl() && e.pressed(KeyCode::Z) {
                    info!("undo");
                    self.ctrl.undo(&mut self.history)
                } else if e.held_ctrl() && e.pressed(KeyCode::Y) {
                    info!("redo");
                    self.ctrl.redo(&mut self.history)
                } else {
                    // let mut world = measure_time!("world clone", self.world.clone());
                    match measure_time!(
                        "load if",
                        self.ctrl
                            .load_new_image_if_triggered(&self.world, &mut self.history)
                    ) {
                        Ok(iip) => iip,
                        Err(e) => {
                            measure_time!(
                                "show info",
                                self.menu.show_info(Info::Error(format!("{e:?}")))
                            );
                            None
                        }
                    }
                }
            });

            if let Some((world, file_label_idx)) = world_idx_pair {
                self.world = world;
                if let Some(active_tool_name) = find_active_tool(&self.tools) {
                    self.world
                        .request_redraw_annotations(active_tool_name, Visibility::All);
                }
                if file_label_idx.is_some() {
                    self.ctrl.paths_navigator.select_label_idx(file_label_idx);
                    let meta_data = self.ctrl.meta_data(
                        self.ctrl.file_selected_idx,
                        Some(self.ctrl.flags().is_loading_screen_active),
                    );
                    self.world.data.meta_data = meta_data;
                    if !self.ctrl.flags().is_loading_screen_active {
                        for t in &mut self.tools {
                            if t.is_active() {
                                (self.world, self.history) = t.file_changed(
                                    mem::take(&mut self.world),
                                    mem::take(&mut self.history),
                                );
                            }
                        }
                    }
                }
            }

            if are_tools_active(&self.menu, &self.tools_select_menu) {
                let meta_data = self.ctrl.meta_data(
                    self.ctrl.file_selected_idx,
                    Some(self.ctrl.flags().is_loading_screen_active),
                );
                self.world.data.meta_data = meta_data;
                (self.world, self.history) = apply_tools(
                    &mut self.tools,
                    mem::take(&mut self.world),
                    mem::take(&mut self.history),
                    e,
                );
            }

            // show position and rgb value
            if let Some(idx) = self.ctrl.paths_navigator.file_label_selected_idx() {
                let pixel_pos = e.mouse_pos_on_orig.map(|mp| mp.into());
                let data_point = get_pixel_on_orig_str(&self.world, &pixel_pos);
                let shape = self.world.shape_orig();
                let file_label = self.ctrl.file_label(idx);
                let active_tool = self.tools.iter().find(|t| t.is_active());
                let tool_string = if let Some(t) = active_tool {
                    format!("{} tool is active", t.name)
                } else {
                    "".to_string()
                };
                let zoom_box_coords = self
                    .world
                    .zoom_box()
                    .map(|zb| {
                        let zb = BbI::from(zb);
                        format!("zoom x {}, y {}, w {}, h {}", zb.x, zb.y, zb.w, zb.h)
                    })
                    .unwrap_or("no zoom".into());
                let s = match data_point {
                    Some(s) => ImageInfo {
                        filename: file_label.to_string(),
                        shape_info: format!("{}x{}", shape.w, shape.h),
                        pixel_value: s,
                        tool_info: tool_string,
                        zoom_box_coords,
                    },
                    None => ImageInfo {
                        filename: file_label.to_string(),
                        shape_info: format!("{}x{}", shape.w, shape.h),
                        pixel_value: "(x, y) -> (r, g, b)".to_string(),
                        tool_info: tool_string,
                        zoom_box_coords,
                    },
                };
                self.world.update_view.image_info = Some(s);
            }
            if let Some(n_autosaves) = self.ctrl.cfg.usr.n_autosaves
                && self.autosave_timer.elapsed().as_secs() > AUTOSAVE_INTERVAL_S
            {
                self.autosave_timer = Instant::now();
                let homefolder = self.ctrl.cfg.home_folder().to_string();
                let current_prj_path = self.ctrl.cfg.current_prj_path().to_path_buf();
                let save_prj = |prj_path| {
                    self.ctrl
                        .save(prj_path, &self.world.data.tools_data_map, false)
                };
                trace_ok_err(autosave(
                    &current_prj_path,
                    homefolder,
                    n_autosaves,
                    save_prj,
                ));
            }

            Ok((
                mem::take(&mut self.world.update_view),
                self.ctrl.cfg.usr.show_main_image(),
                self.ctrl.cfg.usr.show_thumbs(),
                get_prj_name(self.ctrl.cfg.current_prj_path(), None),
            ))
        })
    }
    pub fn load_prj_during_startup(&mut self, file_path: Option<PathBuf>) -> RvResult<()> {
        if let Some(file_path) = file_path {
            info!("loaded project {file_path:?}");
            self.world.data.tools_data_map = self.ctrl.load(file_path)?;
        } else {
            let pp = self.ctrl.cfg.current_prj_path().to_path_buf();
            // load last project
            match self.ctrl.load(pp) {
                Ok(td) => {
                    info!(
                        "loaded last saved project {:?}",
                        self.ctrl.cfg.current_prj_path()
                    );
                    self.world.data.tools_data_map = td;
                }
                Err(e) => {
                    if DEFAULT_PRJ_PATH.as_os_str() != self.ctrl.cfg.current_prj_path().as_os_str()
                    {
                        info!(
                            "could not read last saved project {:?} due to {e:?} ",
                            self.ctrl.cfg.current_prj_path()
                        );
                    }
                }
            }
        }
        Ok(())
    }
    pub fn import_prj(&mut self, file_path: &Path) -> RvResult<()> {
        self.world.data.tools_data_map = self.ctrl.replace_with_save(file_path)?;
        Ok(())
    }
}