libsurfer/

hierarchy.rs

//! Functions for drawing the left hand panel showing scopes and variables.
use crate::SystemState;
use crate::data_container::{DataContainer, VariableType as VarType};
use crate::displayed_item_tree::VisibleItemIndex;
use crate::message::Message;
use crate::tooltips::{scope_tooltip_text, variable_tooltip_text};
use crate::transaction_container::StreamScopeRef;
use crate::transactions::{draw_transaction_root, draw_transaction_variable_list};
use crate::variable_direction::get_direction_string;
use crate::view::draw_true_name;
use crate::wave_container::{
    ScopeRef, ScopeRefExt, VariableMeta, VariableRef, VariableRefExt, WaveContainer,
};
use crate::wave_data::{ScopeType, WaveData};
use derive_more::{Display, FromStr};
use ecolor::Color32;
use egui::text::LayoutJob;
use egui::{CentralPanel, Frame, Layout, Panel, ScrollArea, TextStyle, Ui};
use egui_remixicon::icons;
use emath::Align;
use enum_iterator::Sequence;
use epaint::{
    Margin,
    text::{TextFormat, TextWrapMode},
};
use eyre::WrapErr as _;
use itertools::Itertools;
use num::BigUint;
use serde::{Deserialize, Serialize};
use std::ops::Range;
use std::rc::Rc;
use surfer_translation_types::translator::VariableNameInfo;
use tracing::warn;
#[derive(Clone, Copy, Debug, Deserialize, Display, FromStr, PartialEq, Eq, Serialize, Sequence)]
pub enum HierarchyStyle {
    Separate,
    Tree,
    Variables,
}

#[derive(Clone, Copy, Debug, Deserialize, Display, FromStr, PartialEq, Eq, Serialize, Sequence)]
pub enum ParameterDisplayLocation {
    Variables,
    Scopes,
    Tooltips,
    None,
}

#[derive(Clone, Debug, Deserialize, PartialEq, Eq, Serialize)]
pub enum ScopeExpandType {
    ExpandSpecific(ScopeRef),
    ExpandAll,
    CollapseAll,
}

use crate::variable_filter::VariableNameFilterType;
use crate::wave_source::WaveSource;

#[derive(Clone)]
pub(crate) struct VariableListRow {
    pub(crate) variable: VariableRef,
    pub(crate) meta: Option<VariableMeta>,
    pub(crate) name_info: Option<VariableNameInfo>,
}

/// Cache key for `draw_all_variables`. The cache is rebuilt whenever any field changes.
#[derive(PartialEq)]
pub(crate) struct AllVariableCacheKey {
    cache_generation: u64,
    translator_generation: u64,
    wave_source: WaveSource,
    filter_str: String,
    filter_type: VariableNameFilterType,
    case_insensitive: bool,
    include_inputs: bool,
    include_outputs: bool,
    include_inouts: bool,
    include_others: bool,
    group_by_direction: bool,
}

impl SystemState {
    /// Scopes and variables in two separate lists
    pub fn separate(&mut self, ui: &mut Ui, msgs: &mut Vec<Message>) {
        ui.visuals_mut().override_text_color =
            Some(self.user.config.theme.primary_ui_color.foreground);

        let total_space = ui.available_height();
        Panel::top("scopes")
            .resizable(true)
            .default_size(total_space / 2.0)
            .max_size(total_space - 64.0)
            .frame(Frame::new().inner_margin(Margin::same(5)))
            .show_inside(ui, |ui| {
                ui.horizontal(|ui| {
                    ui.heading("Scopes")
                        .context_menu(|ui| self.hierarchy_menu(msgs, ui));
                    if self.user.waves.is_some() {
                        let default_padding = ui.spacing().button_padding;
                        ui.spacing_mut().button_padding = egui::vec2(0.0, default_padding.y);
                        ui.button(icons::MENU_UNFOLD_FILL)
                            .on_hover_text("Expand all scopes")
                            .clicked()
                            .then(|| msgs.push(Message::ExpandScope(ScopeExpandType::ExpandAll)));
                        ui.button(icons::MENU_FOLD_FILL)
                            .on_hover_text("Collapse all scopes")
                            .clicked()
                            .then(|| msgs.push(Message::ExpandScope(ScopeExpandType::CollapseAll)));
                        ui.spacing_mut().button_padding = default_padding;
                    }
                });
                ui.add_space(3.0);

                ScrollArea::both()
                    .id_salt("scopes")
                    .auto_shrink([false; 2])
                    .show(ui, |ui| {
                        ui.style_mut().wrap_mode = Some(TextWrapMode::Extend);
                        if let Some(waves) = &self.user.waves {
                            self.draw_all_scopes(msgs, waves, false, ui);
                        }
                    });
            });
        CentralPanel::default()
            .frame(Frame::new().inner_margin(Margin::same(5)))
            .show_inside(ui, |ui| {
                ui.with_layout(Layout::left_to_right(Align::TOP), |ui| {
                    ui.heading("Variables")
                        .context_menu(|ui| self.hierarchy_menu(msgs, ui));
                    ui.add_space(3.0);
                    self.draw_variable_filter_edit(ui, msgs, false);
                });
                ui.add_space(3.0);

                self.draw_variables(msgs, ui);
            });
        *self.scope_ref_to_expand.borrow_mut() = None;
    }

    fn draw_variable_list_header(&self, ui: &mut Ui) {
        let show_icons = self.show_hierarchy_icons();
        let show_direction = self.show_variable_direction();

        // Only show header if icons or direction are enabled
        if !show_icons && !show_direction {
            return;
        }

        ui.with_layout(
            Layout::top_down(Align::LEFT).with_cross_justify(true),
            |ui| {
                let monospace_font = ui
                    .style()
                    .text_styles
                    .get(&TextStyle::Monospace)
                    .cloned()
                    .unwrap();

                let text_format = TextFormat {
                    font_id: monospace_font,
                    color: self.user.config.theme.foreground,
                    ..Default::default()
                };

                let mut label = LayoutJob::default();
                // Type column - "T " to match "icon " (only if shown)
                if show_icons {
                    label.append("T ", 0.0, text_format.clone());
                }
                // Direction column - "D " to match "icon " (only if shown)
                if show_direction {
                    label.append("D ", 0.0, text_format.clone());
                }
                // Name column
                label.append("Name", 0.0, text_format);

                ui.add(egui::Button::selectable(false, label));
            },
        );
        ui.separator();
    }

    fn draw_variables(&mut self, msgs: &mut Vec<Message>, ui: &mut Ui) {
        if let Some(waves) = &self.user.waves {
            let empty_scope = if waves.inner.is_waves() {
                ScopeType::WaveScope(ScopeRef::empty())
            } else {
                ScopeType::StreamScope(StreamScopeRef::Empty(String::default()))
            };
            let active_scope = waves.active_scope.as_ref().unwrap_or(&empty_scope);
            match active_scope {
                ScopeType::WaveScope(scope) => {
                    let Some(wave_container) = waves.inner.as_waves() else {
                        return;
                    };
                    let variables = self.filtered_variables_unsorted(
                        &wave_container.variables_in_scope(scope),
                        false,
                    );
                    let variable_rows = self.build_variable_rows(wave_container, &variables);
                    // Draw header before scroll area
                    self.draw_variable_list_header(ui);
                    // Parameters shown in variable list
                    if self.parameter_display_location() == ParameterDisplayLocation::Variables {
                        let parameters = wave_container.parameters_in_scope(scope);
                        if !parameters.is_empty() {
                            ScrollArea::both()
                                .auto_shrink([false; 2])
                                .id_salt("variables")
                                .show(ui, |ui| {
                                    ui.style_mut().wrap_mode = Some(TextWrapMode::Extend);
                                    self.draw_parameters(msgs, wave_container, &parameters, ui);
                                    self.draw_variable_rows(
                                        msgs,
                                        wave_container,
                                        ui,
                                        &variable_rows,
                                        None,
                                        false,
                                    );
                                });
                            return; // Early exit
                        }
                    }
                    // Parameters not shown here or no parameters: use fast approach only drawing visible rows
                    let row_height = ui
                        .text_style_height(&TextStyle::Monospace)
                        .max(ui.text_style_height(&TextStyle::Body));
                    ScrollArea::both()
                        .auto_shrink([false; 2])
                        .id_salt("variables")
                        .show_rows(ui, row_height, variable_rows.len(), |ui, row_range| {
                            ui.style_mut().wrap_mode = Some(TextWrapMode::Extend);

                            self.draw_variable_rows(
                                msgs,
                                wave_container,
                                ui,
                                &variable_rows,
                                Some(&row_range),
                                false,
                            );
                        });
                }
                ScopeType::StreamScope(s) => {
                    ScrollArea::both()
                        .auto_shrink([false; 2])
                        .id_salt("variables")
                        .show(ui, |ui| {
                            ui.style_mut().wrap_mode = Some(TextWrapMode::Extend);

                            draw_transaction_variable_list(msgs, waves, ui, s);
                        });
                }
            }
        }
    }

    fn draw_parameters(
        &self,
        msgs: &mut Vec<Message>,
        wave_container: &WaveContainer,
        parameters: &[VariableRef],
        ui: &mut Ui,
    ) {
        egui::collapsing_header::CollapsingState::load_with_default_open(
            ui.ctx(),
            egui::Id::new(parameters),
            self.expand_parameter_section,
        )
        .show_header(ui, |ui| {
            ui.with_layout(
                Layout::top_down(Align::LEFT).with_cross_justify(true),
                |ui| {
                    ui.label("Parameters");
                },
            );
        })
        .body(|ui| {
            self.filter_and_draw_variable_list(msgs, wave_container, ui, parameters, None);
        });
    }

    /// Scopes and variables in a joint tree.
    pub fn tree(&mut self, ui: &mut Ui, msgs: &mut Vec<Message>) {
        ui.visuals_mut().override_text_color =
            Some(self.user.config.theme.primary_ui_color.foreground);

        ui.with_layout(
            Layout::top_down(Align::LEFT).with_cross_justify(true),
            |ui| {
                Frame::new().inner_margin(Margin::same(5)).show(ui, |ui| {
                    ui.with_layout(Layout::left_to_right(Align::TOP), |ui| {
                        ui.heading("Hierarchy")
                            .context_menu(|ui| self.hierarchy_menu(msgs, ui));
                        ui.add_space(3.0);
                        self.draw_variable_filter_edit(ui, msgs, false);
                    });
                    ui.add_space(3.0);

                    ScrollArea::both().id_salt("hierarchy").show(ui, |ui| {
                        ui.style_mut().wrap_mode = Some(TextWrapMode::Extend);
                        if let Some(waves) = &self.user.waves {
                            ui.style_mut().wrap_mode = Some(TextWrapMode::Extend);
                            self.draw_all_scopes(msgs, waves, true, ui);
                        }
                    });
                });
            },
        );
    }

    /// List with all variables.
    pub fn variable_list(&mut self, ui: &mut Ui, msgs: &mut Vec<Message>) {
        ui.visuals_mut().override_text_color =
            Some(self.user.config.theme.primary_ui_color.foreground);

        ui.with_layout(
            Layout::top_down(Align::LEFT).with_cross_justify(true),
            |ui| {
                Frame::new().inner_margin(Margin::same(5)).show(ui, |ui| {
                    ui.with_layout(Layout::left_to_right(Align::TOP), |ui| {
                        ui.heading("Variables")
                            .context_menu(|ui| self.hierarchy_menu(msgs, ui));
                        ui.add_space(3.0);
                        self.draw_variable_filter_edit(ui, msgs, true);
                    });
                    ui.add_space(3.0);
                    self.draw_all_variables(msgs, ui);
                });
            },
        );
    }

    fn make_all_variable_cache_key(
        &self,
        cache_generation: u64,
        wave_source: &WaveSource,
    ) -> AllVariableCacheKey {
        let f = &self.user.variable_filter;
        AllVariableCacheKey {
            cache_generation,
            translator_generation: self.translator_generation,
            wave_source: wave_source.clone(),
            filter_str: f.name_filter_str.clone(),
            filter_type: f.name_filter_type.clone(),
            case_insensitive: f.name_filter_case_insensitive,
            include_inputs: f.include_inputs,
            include_outputs: f.include_outputs,
            include_inouts: f.include_inouts,
            include_others: f.include_others,
            group_by_direction: f.group_by_direction,
        }
    }

    fn draw_all_variables(&mut self, msgs: &mut Vec<Message>, ui: &mut Ui) {
        // Phase 1: Rebuild the row cache only when the key changes.
        // wave_container borrows self.user.waves; all_variable_rows_cache is a disjoint field.
        if let Some(waves) = &self.user.waves
            && let DataContainer::Waves(wave_container) = &waves.inner
        {
            let key = self.make_all_variable_cache_key(waves.cache_generation, &waves.source);
            let is_stale = self
                .all_variable_rows_cache
                .as_ref()
                .is_none_or(|(k, _)| k != &key);
            if is_stale {
                let variables = self.filtered_variables_unsorted(&wave_container.variables(), true);
                let rows = self.build_variable_rows(wave_container, &variables);
                self.all_variable_rows_cache = Some((key, Rc::new(rows)));
            }
        }

        // Phase 2: Draw using the cached rows.
        if let Some(waves) = &self.user.waves {
            match &waves.inner {
                DataContainer::Waves(wave_container) => {
                    // Clone the Rc (O(1)) to get an owned handle usable inside the closure.
                    let variable_rows = self
                        .all_variable_rows_cache
                        .as_ref()
                        .map_or_else(|| Rc::new(Vec::new()), |(_, rows)| Rc::clone(rows));
                    let row_height = ui
                        .text_style_height(&TextStyle::Monospace)
                        .max(ui.text_style_height(&TextStyle::Body));
                    // Draw header before scroll area
                    self.draw_variable_list_header(ui);
                    ScrollArea::both()
                        .auto_shrink([false; 2])
                        .id_salt("variables")
                        .show_rows(ui, row_height, variable_rows.len(), |ui, row_range| {
                            ui.style_mut().wrap_mode = Some(TextWrapMode::Extend);
                            self.draw_variable_rows(
                                msgs,
                                wave_container,
                                ui,
                                &variable_rows,
                                Some(&row_range),
                                true,
                            );
                        });
                }
                DataContainer::Transactions(_) => {
                    // No support for Streams yet
                    ui.with_layout(
                        Layout::top_down(Align::LEFT).with_cross_justify(true),
                        |ui| {
                            ui.label("Streams are not yet supported.");
                            ui.label("Select another view.");
                        },
                    );
                }
                DataContainer::Empty => {}
            }
        }
    }

    fn draw_all_scopes(
        &self,
        msgs: &mut Vec<Message>,
        wave: &WaveData,
        draw_variables: bool,
        ui: &mut Ui,
    ) {
        for scope in wave.inner.root_scopes() {
            match scope {
                ScopeType::WaveScope(scope) => {
                    self.draw_selectable_child_or_orphan_scope(
                        msgs,
                        wave,
                        &scope,
                        draw_variables,
                        ui,
                    );
                }
                ScopeType::StreamScope(_) => {
                    draw_transaction_root(msgs, wave, ui);
                }
            }
        }
        if draw_variables && let Some(wave_container) = wave.inner.as_waves() {
            let scope = ScopeRef::empty();
            let variables = wave_container.variables_in_scope(&scope);
            self.filter_and_draw_variable_list(msgs, wave_container, ui, &variables, None);
        }
    }

    fn add_scope_selectable_label(
        &self,
        msgs: &mut Vec<Message>,
        wave: &WaveData,
        scope: &ScopeRef,
        ui: &mut Ui,
        scroll_to_label: bool,
    ) {
        let name = scope.name();
        let is_selected = wave.active_scope == Some(ScopeType::WaveScope(scope.clone()));
        let mut response = if self.show_hierarchy_icons() {
            let scope_type = wave
                .inner
                .as_waves()
                .and_then(|wc| wc.get_scope_type(scope));
            let (icon, icon_color) = self.user.config.theme.scope_icons.get_icon(scope_type);

            let body_font = ui
                .style()
                .text_styles
                .get(&TextStyle::Body)
                .cloned()
                .unwrap_or_default();
            let icon_format = TextFormat {
                font_id: body_font.clone(),
                color: icon_color,
                ..Default::default()
            };
            let name_format = TextFormat {
                font_id: body_font,
                color: self.user.config.theme.foreground,
                ..Default::default()
            };
            let mut label = LayoutJob::default();
            label.append(icon, 0.0, icon_format);
            label.append(" ", 0.0, name_format.clone());
            label.append(&name, 0.0, name_format);

            ui.add(egui::Button::selectable(is_selected, label))
        } else {
            ui.add(egui::Button::selectable(is_selected, name))
        };
        let _ = response.interact(egui::Sense::click_and_drag());
        response.drag_started().then(|| {
            msgs.push(Message::VariableDragStarted(VisibleItemIndex(
                wave.display_item_ref_counter,
            )));
        });

        if scroll_to_label {
            response.scroll_to_me(Some(Align::Center));
        }

        response.drag_stopped().then(|| {
            if ui.input(|i| i.pointer.hover_pos().unwrap_or_default().x)
                > self.user.sidepanel_width.unwrap_or_default()
            {
                let scope_t = ScopeType::WaveScope(scope.clone());
                let variables = wave
                    .inner
                    .variables_in_scope(&scope_t)
                    .iter()
                    .filter_map(|var| match var {
                        VarType::Variable(var) => Some(var.clone()),
                        VarType::Generator(_) => None,
                    })
                    .collect_vec();

                msgs.push(Message::AddDraggedVariables(
                    self.filtered_variables(variables.as_slice(), false),
                ));
            }
        });
        if self.show_scope_tooltip() {
            response = response.on_hover_ui(|ui| {
                ui.set_max_width(ui.spacing().tooltip_width);
                ui.add(egui::Label::new(scope_tooltip_text(
                    wave,
                    scope,
                    self.parameter_display_location() == ParameterDisplayLocation::Tooltips,
                )));
            });
        }
        response.context_menu(|ui| {
            if ui.button("Add scope").clicked() {
                msgs.push(Message::AddScope(scope.clone(), false));
            }
            if ui.button("Add scope recursively").clicked() {
                msgs.push(Message::AddScope(scope.clone(), true));
            }
            if ui.button("Add scope as group").clicked() {
                msgs.push(Message::AddScopeAsGroup(scope.clone(), false));
            }
            if ui.button("Add scope as group recursively").clicked() {
                msgs.push(Message::AddScopeAsGroup(scope.clone(), true));
            }
            if wave
                .inner
                .as_waves()
                .is_some_and(|wc| wc.scope_is_array(scope))
                && ui.button("Show frame buffer").clicked()
            {
                msgs.push(Message::SetFrameBufferArray(scope.clone()));
            }
        });
        response.clicked().then(|| {
            msgs.push(Message::SetActiveScope(if is_selected {
                None
            } else {
                Some(ScopeType::WaveScope(scope.clone()))
            }));
        });
    }

    fn draw_selectable_child_or_orphan_scope(
        &self,
        msgs: &mut Vec<Message>,
        wave: &WaveData,
        scope: &ScopeRef,
        draw_variables: bool,
        ui: &mut Ui,
    ) {
        // Extract wave container once to avoid repeated as_waves().unwrap() calls
        let Some(wave_container) = wave.inner.as_waves() else {
            return;
        };

        let Some(child_scopes) = wave_container
            .child_scopes(scope)
            .context("Failed to get child scopes")
            .map_err(|e| warn!("{e:#?}"))
            .ok()
        else {
            return;
        };

        let no_variables_in_scope = wave_container.no_variables_in_scope(scope);
        if child_scopes.is_empty() && no_variables_in_scope && !self.show_empty_scopes() {
            return;
        }

        if child_scopes.is_empty() && (!draw_variables || no_variables_in_scope) {
            // Indent our label by both icon width and icon spacing to
            // match the other headers that actually have an icon.
            ui.horizontal(|ui| {
                ui.add_space(ui.spacing().icon_width + ui.spacing().icon_spacing);
                self.add_scope_selectable_label(msgs, wave, scope, ui, false);
            });
        } else {
            let should_open_header = self.should_open_header_and_scroll_to(scope);
            let mut collapsing_header =
                egui::collapsing_header::CollapsingState::load_with_default_open(
                    ui.ctx(),
                    egui::Id::new(scope),
                    false,
                );
            if let Some((header_state, _)) = should_open_header {
                collapsing_header.set_open(header_state);
            }
            collapsing_header
                .show_header(ui, |ui| {
                    ui.with_layout(
                        Layout::top_down(Align::LEFT).with_cross_justify(true),
                        |ui| {
                            self.add_scope_selectable_label(
                                msgs,
                                wave,
                                scope,
                                ui,
                                should_open_header.is_some_and(|(_, scroll)| scroll),
                            );
                        },
                    );
                })
                .body(|ui| {
                    if (draw_variables
                        && !(matches!(
                            self.parameter_display_location(),
                            ParameterDisplayLocation::Tooltips | ParameterDisplayLocation::None
                        )))
                        || self.parameter_display_location() == ParameterDisplayLocation::Scopes
                    {
                        let parameters = wave_container.parameters_in_scope(scope);
                        if !parameters.is_empty() {
                            self.draw_parameters(msgs, wave_container, &parameters, ui);
                        }
                    }
                    self.draw_root_scope_view(msgs, wave, scope, draw_variables, ui);
                    if draw_variables {
                        let variables = wave_container.variables_in_scope(scope);
                        self.filter_and_draw_variable_list(
                            msgs,
                            wave_container,
                            ui,
                            &variables,
                            None,
                        );
                    }
                });
        }
    }

    fn draw_root_scope_view(
        &self,
        msgs: &mut Vec<Message>,
        wave: &WaveData,
        root_scope: &ScopeRef,
        draw_variables: bool,
        ui: &mut Ui,
    ) {
        // Extract wave container once to avoid unwrap
        let Some(wave_container) = wave.inner.as_waves() else {
            return;
        };

        wave_container
            .child_scopes(root_scope)
            .context("Failed to get child scopes")
            .map_err(|e| warn!("{e:#?}"))
            .ok()
            .into_iter()
            .flatten()
            .sorted_by(|a, b| numeric_sort::cmp(&a.name(), &b.name()))
            .for_each(|child_scope| {
                self.draw_selectable_child_or_orphan_scope(
                    msgs,
                    wave,
                    &child_scope,
                    draw_variables,
                    ui,
                );
            });
    }

    fn filter_and_draw_variable_list(
        &self,
        msgs: &mut Vec<Message>,
        wave_container: &WaveContainer,
        ui: &mut Ui,
        variables: &[VariableRef],
        row_range: Option<&Range<usize>>,
    ) {
        let filtered_variables = self.filtered_variables_unsorted(variables, false);
        self.draw_variable_list(
            msgs,
            wave_container,
            ui,
            &filtered_variables,
            row_range,
            false,
        );
    }

    fn draw_variable_list(
        &self,
        msgs: &mut Vec<Message>,
        wave_container: &WaveContainer,
        ui: &mut Ui,
        variables: &[VariableRef],
        row_range: Option<&Range<usize>>,
        display_full_path: bool,
    ) {
        let variable_rows = self.build_variable_rows(wave_container, variables);
        self.draw_variable_rows(
            msgs,
            wave_container,
            ui,
            &variable_rows,
            row_range,
            display_full_path,
        );
    }

    fn build_variable_rows(
        &self,
        wave_container: &WaveContainer,
        variables: &[VariableRef],
    ) -> Vec<VariableListRow> {
        let mut rows = variables
            .iter()
            .map(|var| {
                let meta = wave_container.variable_meta(var).ok();
                let name_info = self.get_variable_name_info(var, meta.as_ref());
                VariableListRow {
                    variable: var.clone(),
                    meta,
                    name_info,
                }
            })
            .collect::<Vec<_>>();

        rows.sort_by(|a, b| {
            let a_priority = a
                .name_info
                .as_ref()
                .and_then(|info| info.priority)
                .unwrap_or_default();
            let b_priority = b
                .name_info
                .as_ref()
                .and_then(|info| info.priority)
                .unwrap_or_default();
            b_priority
                .cmp(&a_priority)
                .then_with(|| self.variable_cmp(&a.variable, &b.variable, Some(wave_container)))
        });
        rows
    }

    fn draw_variable_rows(
        &self,
        msgs: &mut Vec<Message>,
        wave_container: &WaveContainer,
        ui: &mut Ui,
        variable_rows: &[VariableListRow],
        row_range: Option<&Range<usize>>,
        display_full_path: bool,
    ) {
        let variable_rows = if let Some(range) = row_range {
            let start = range.start.min(variable_rows.len());
            let end = range.end.min(variable_rows.len());
            &variable_rows[start..end]
        } else {
            variable_rows
        };

        // Precompute common font metrics once per frame to avoid expensive per-row work.
        // NOTE: Safe unwrap, we know that egui has its own built-in font.
        // Use precomputed font and char width where available to reduce work.
        let monospace_font = ui
            .style()
            .text_styles
            .get(&TextStyle::Monospace)
            .cloned()
            .unwrap();
        let body_font = ui
            .style()
            .text_styles
            .get(&TextStyle::Body)
            .cloned()
            .unwrap();
        let char_width_mono = ui.fonts_mut(|fonts| {
            fonts
                .layout_no_wrap(" ".to_string(), monospace_font.clone(), Color32::BLACK)
                .size()
                .x
        });
        // The button padding is added by egui on selectable labels
        let available_space = ui.available_width() - ui.spacing().button_padding.x * 2.;

        // Draw variables
        for row in variable_rows {
            let variable = &row.variable;
            let meta = row.meta.as_ref();
            let name_info = row.name_info.clone();

            // Get index string
            let index = meta
                .and_then(|meta| meta.index)
                .map(|index| {
                    if self.show_variable_indices() {
                        format!(" {index}")
                    } else {
                        String::new()
                    }
                })
                .unwrap_or_default();

            // Get type icon with color
            let (type_icon, icon_color) = if self.show_hierarchy_icons() {
                let (icon, color) = self.user.config.theme.variable_icons.get_icon(meta);
                (format!("{icon} "), color)
            } else {
                (String::new(), self.user.config.theme.foreground)
            };

            // Get direction icon
            let direction = self
                .show_variable_direction()
                .then(|| get_direction_string(meta, name_info.as_ref()))
                .flatten()
                .unwrap_or_default();
            // Get value in case of parameter
            let value = if meta.is_some_and(surfer_translation_types::VariableMeta::is_parameter) {
                let res = wave_container.query_variable(variable, &BigUint::ZERO).ok();
                res.and_then(|o| o.and_then(|q| q.current.map(|v| format!(": {}", v.1))))
                    .unwrap_or_else(|| ": Undefined".to_string())
            } else {
                String::new()
            };

            ui.with_layout(
                Layout::top_down(Align::LEFT).with_cross_justify(true),
                |ui| {
                    let mut label = LayoutJob::default();
                    let true_name = name_info.and_then(|info| info.true_name);

                    let font = if true_name.is_some() {
                        monospace_font.clone()
                    } else {
                        body_font.clone()
                    };
                    let icon_format = TextFormat {
                        font_id: font.clone(),
                        color: icon_color,
                        ..Default::default()
                    };
                    let text_format = TextFormat {
                        font_id: font,
                        color: self.user.config.theme.foreground,
                        ..Default::default()
                    };

                    if let Some(name) = true_name {
                        let type_icon_size = type_icon.chars().count();
                        let direction_size = direction.chars().count();
                        let index_size = index.chars().count();
                        let value_size = value.chars().count();
                        let used_space = (type_icon_size + direction_size + index_size + value_size)
                            as f32
                            * char_width_mono;
                        let space_for_name = available_space - used_space;

                        label.append(&type_icon, 0.0, icon_format);
                        label.append(&direction, 0.0, text_format.clone());

                        draw_true_name(
                            &name,
                            &mut label,
                            monospace_font.clone(),
                            self.user.config.theme.foreground,
                            char_width_mono,
                            space_for_name,
                            self.user.config.layout.waveforms_line_height,
                        );

                        label.append(&index, 0.0, text_format.clone());
                        label.append(&value, 0.0, text_format);
                    } else {
                        let name = if display_full_path {
                            variable.full_path_string()
                        } else {
                            variable.name.clone()
                        };
                        label.append(&type_icon, 0.0, icon_format);
                        label.append(&direction, 0.0, text_format.clone());
                        label.append(&name, 0.0, text_format.clone());
                        label.append(&index, 0.0, text_format.clone());
                        label.append(&value, 0.0, text_format);
                    }

                    let mut response = ui.add(egui::Button::selectable(false, label));

                    let _ = response.interact(egui::Sense::click_and_drag());

                    if self.show_tooltip() {
                        // Reuse the already-obtained `meta` and pass a clone of the variable
                        // reference into the closure so we don't call `variable_meta` again.
                        let tooltip_meta = meta;
                        let tooltip_var = variable.clone();
                        response = response.on_hover_ui(move |ui| {
                            ui.set_max_width(ui.spacing().tooltip_width);
                            ui.add(egui::Label::new(variable_tooltip_text(
                                tooltip_meta,
                                &tooltip_var,
                            )));
                        });
                    }
                    response.drag_started().then(|| {
                        msgs.push(Message::VariableDragStarted(VisibleItemIndex(
                            self.user.waves.as_ref().unwrap().display_item_ref_counter,
                        )));
                    });
                    response.drag_stopped().then(|| {
                        if ui.input(|i| i.pointer.hover_pos().unwrap_or_default().x)
                            > self.user.sidepanel_width.unwrap_or_default()
                        {
                            msgs.push(Message::AddDraggedVariables(vec![variable.clone()]));
                        }
                    });
                    response
                        .clicked()
                        .then(|| msgs.push(Message::AddVariables(vec![variable.clone()])));
                },
            );
        }
    }

    fn should_open_header_and_scroll_to(&self, scope: &ScopeRef) -> Option<(bool, bool)> {
        let mut scope_ref_cell = self.scope_ref_to_expand.borrow_mut();
        if let Some(state) = scope_ref_cell.as_mut() {
            match state {
                ScopeExpandType::ExpandAll => return Some((true, false)),
                ScopeExpandType::CollapseAll => return Some((false, false)),
                ScopeExpandType::ExpandSpecific(state) => {
                    if state.strs.starts_with(&scope.strs) {
                        if (state.strs.len() - 1) == scope.strs.len() {
                            // need to compare vs. parent of signal
                            *scope_ref_cell = None;
                        }
                        return Some((true, true));
                    }
                }
            }
        }
        None
    }
}