turbo_trace_server/

viewer.rs

use std::cmp::{Reverse, max};

use either::Either;
use itertools::Itertools;
use rayon::iter::{IntoParallelRefIterator, ParallelIterator};
use rustc_hash::{FxHashMap, FxHashSet};
use serde::Serialize;
use turbo_rcstr::rcstr;

use crate::{
    server::ViewRect,
    span_bottom_up_ref::SpanBottomUpRef,
    span_graph_ref::{SpanGraphEventRef, SpanGraphRef},
    span_ref::{SpanEventRef, SpanRef},
    store::{SpanId, Store},
    timestamp::Timestamp,
    u64_empty_string,
};

const EXTRA_WIDTH_PERCENTAGE: u64 = 50;
const EXTRA_HEIGHT: u64 = 5;

#[derive(Default)]
pub struct Viewer {
    span_options: FxHashMap<SpanId, SpanOptions>,
}

#[derive(Clone, Copy, Debug)]
pub enum ValueMode {
    Duration,
    Cpu,
    Allocations,
    Deallocations,
    PersistentAllocations,
    AllocationCount,
    Count,
    AllocationsPerTime,
    PersistentAllocationsPerTime,
    AllocationCountPerTime,
}

impl ValueMode {
    fn secondary(&self) -> ValueMode {
        match self {
            ValueMode::Duration => ValueMode::Cpu,
            ValueMode::Cpu => ValueMode::Duration,
            ValueMode::Allocations => ValueMode::PersistentAllocations,
            ValueMode::Deallocations => ValueMode::PersistentAllocations,
            ValueMode::PersistentAllocations => ValueMode::Allocations,
            ValueMode::AllocationCount => ValueMode::Allocations,
            ValueMode::Count => ValueMode::Count,
            ValueMode::AllocationsPerTime => ValueMode::PersistentAllocationsPerTime,
            ValueMode::PersistentAllocationsPerTime => ValueMode::AllocationsPerTime,
            ValueMode::AllocationCountPerTime => ValueMode::AllocationsPerTime,
        }
    }

    fn value_from_span(&self, span: &SpanRef<'_>) -> u64 {
        match self {
            ValueMode::Duration => *span.corrected_total_time(),
            ValueMode::Cpu => *span.total_time(),
            ValueMode::Allocations => span.total_allocations(),
            ValueMode::Deallocations => span.total_deallocations(),
            ValueMode::PersistentAllocations => span.total_persistent_allocations(),
            ValueMode::AllocationCount => span.total_allocation_count(),
            ValueMode::Count => span.total_span_count(),
            ValueMode::AllocationsPerTime => {
                value_over_time(span.total_allocations(), span.corrected_total_time())
            }
            ValueMode::AllocationCountPerTime => {
                value_over_time(span.total_allocation_count(), span.corrected_total_time())
            }
            ValueMode::PersistentAllocationsPerTime => value_over_time(
                span.total_persistent_allocations(),
                span.corrected_total_time(),
            ),
        }
    }

    fn value_from_event(&self, event: &SpanEventRef<'_>) -> u64 {
        match self {
            ValueMode::Duration => *event.corrected_self_time(),
            ValueMode::Cpu => *event.total_time(),
            _ => match event {
                SpanEventRef::Child { span } => self.value_from_span(span),
                SpanEventRef::SelfTime { .. } => 0,
            },
        }
    }

    fn value_from_graph(&self, graph: &SpanGraphRef<'_>) -> u64 {
        match self {
            ValueMode::Duration => *graph.corrected_total_time(),
            ValueMode::Cpu => *graph.total_time(),
            ValueMode::Allocations => graph.total_allocations(),
            ValueMode::Deallocations => graph.total_deallocations(),
            ValueMode::PersistentAllocations => graph.total_persistent_allocations(),
            ValueMode::AllocationCount => graph.total_allocation_count(),
            ValueMode::Count => graph.total_span_count(),
            ValueMode::AllocationsPerTime => {
                value_over_time(graph.total_allocations(), graph.corrected_total_time())
            }
            ValueMode::AllocationCountPerTime => {
                value_over_time(graph.total_allocation_count(), graph.corrected_total_time())
            }
            ValueMode::PersistentAllocationsPerTime => value_over_time(
                graph.total_persistent_allocations(),
                graph.corrected_total_time(),
            ),
        }
    }

    fn value_from_graph_event(&self, event: &SpanGraphEventRef<'_>) -> u64 {
        match self {
            ValueMode::Duration => *event.corrected_total_time(),
            ValueMode::Cpu => *event.total_time(),
            ValueMode::Allocations => event.total_allocations(),
            ValueMode::Deallocations => event.total_deallocations(),
            ValueMode::PersistentAllocations => event.total_persistent_allocations(),
            ValueMode::AllocationCount => event.total_allocation_count(),
            ValueMode::Count => event.total_span_count(),
            ValueMode::AllocationsPerTime => {
                value_over_time(event.total_allocations(), event.corrected_total_time())
            }
            ValueMode::AllocationCountPerTime => {
                value_over_time(event.total_allocation_count(), event.corrected_total_time())
            }
            ValueMode::PersistentAllocationsPerTime => value_over_time(
                event.total_persistent_allocations(),
                event.corrected_total_time(),
            ),
        }
    }

    fn value_from_bottom_up(&self, bottom_up: &SpanBottomUpRef<'_>) -> u64 {
        match self {
            ValueMode::Duration => *bottom_up.corrected_self_time(),
            ValueMode::Cpu => *bottom_up.self_time(),
            ValueMode::Allocations => bottom_up.self_allocations(),
            ValueMode::Deallocations => bottom_up.self_deallocations(),
            ValueMode::PersistentAllocations => bottom_up.self_persistent_allocations(),
            ValueMode::AllocationCount => bottom_up.self_allocation_count(),
            ValueMode::Count => bottom_up.self_span_count(),
            ValueMode::AllocationsPerTime => value_over_time(
                bottom_up.self_allocations(),
                bottom_up.corrected_self_time(),
            ),
            ValueMode::AllocationCountPerTime => value_over_time(
                bottom_up.self_allocation_count(),
                bottom_up.corrected_self_time(),
            ),
            ValueMode::PersistentAllocationsPerTime => value_over_time(
                bottom_up.self_persistent_allocations(),
                bottom_up.corrected_self_time(),
            ),
        }
    }

    fn value_from_bottom_up_span(&self, bottom_up_span: &SpanRef<'_>) -> u64 {
        match self {
            ValueMode::Duration => *bottom_up_span.corrected_self_time(),
            ValueMode::Cpu => *bottom_up_span.self_time(),
            ValueMode::Allocations => bottom_up_span.self_allocations(),
            ValueMode::Deallocations => bottom_up_span.self_deallocations(),
            ValueMode::PersistentAllocations => bottom_up_span.self_persistent_allocations(),
            ValueMode::AllocationCount => bottom_up_span.self_allocation_count(),
            ValueMode::Count => bottom_up_span.self_span_count(),
            ValueMode::AllocationsPerTime => value_over_time(
                bottom_up_span.self_allocations(),
                bottom_up_span.corrected_self_time(),
            ),
            ValueMode::AllocationCountPerTime => value_over_time(
                bottom_up_span.self_allocation_count(),
                bottom_up_span.corrected_self_time(),
            ),
            ValueMode::PersistentAllocationsPerTime => value_over_time(
                bottom_up_span.self_persistent_allocations(),
                bottom_up_span.corrected_self_time(),
            ),
        }
    }
}

/// this is unfortunately int division but itll have to do.
///
/// cases where count per time is very low is probably not important
fn value_over_time(value: u64, time: Timestamp) -> u64 {
    value.checked_div(*time).unwrap_or(0)
}

#[derive(Clone, Copy, Debug, PartialEq, Eq)]
pub enum SortMode {
    ExecutionOrder,
    Value,
    Name,
}

#[derive(Clone, Copy, Debug)]
pub enum ViewMode {
    RawSpans { sort_mode: SortMode },
    Aggregated { sort_mode: SortMode },
    BottomUp { sort_mode: SortMode },
    AggregatedBottomUp { sort_mode: SortMode },
}

impl ViewMode {
    fn as_spans(self) -> Self {
        match self {
            ViewMode::RawSpans { sort_mode } => ViewMode::RawSpans { sort_mode },
            ViewMode::Aggregated { sort_mode } => ViewMode::RawSpans { sort_mode },
            ViewMode::BottomUp { sort_mode } => ViewMode::BottomUp { sort_mode },
            ViewMode::AggregatedBottomUp { sort_mode } => ViewMode::BottomUp { sort_mode },
        }
    }

    fn as_bottom_up(self) -> Self {
        match self {
            ViewMode::RawSpans { sort_mode } => ViewMode::BottomUp { sort_mode },
            ViewMode::Aggregated { sort_mode } => ViewMode::AggregatedBottomUp { sort_mode },
            ViewMode::BottomUp { sort_mode } => ViewMode::BottomUp { sort_mode },
            ViewMode::AggregatedBottomUp { sort_mode } => {
                ViewMode::AggregatedBottomUp { sort_mode }
            }
        }
    }

    fn aggregate_children(&self) -> bool {
        match self {
            ViewMode::RawSpans { .. } => false,
            ViewMode::Aggregated { .. } => true,
            ViewMode::BottomUp { .. } => false,
            ViewMode::AggregatedBottomUp { .. } => true,
        }
    }

    fn bottom_up(&self) -> bool {
        match self {
            ViewMode::RawSpans { .. } => false,
            ViewMode::Aggregated { .. } => false,
            ViewMode::BottomUp { .. } => true,
            ViewMode::AggregatedBottomUp { .. } => true,
        }
    }

    fn sort_children(&self) -> SortMode {
        match self {
            ViewMode::RawSpans { sort_mode } => *sort_mode,
            ViewMode::Aggregated { sort_mode } => *sort_mode,
            ViewMode::BottomUp { sort_mode } => *sort_mode,
            ViewMode::AggregatedBottomUp { sort_mode } => *sort_mode,
        }
    }
}

#[derive(Default)]
struct SpanOptions {
    view_mode: Option<(ViewMode, bool)>,
}

pub struct Update {
    pub lines: Vec<ViewLineUpdate>,
    pub max: u64,
}

#[derive(Serialize, Debug)]
#[serde(rename_all = "camelCase")]
pub struct ViewLineUpdate {
    y: u64,
    spans: Vec<ViewSpan>,
}

#[derive(Serialize, Debug)]
#[serde(rename_all = "camelCase")]
pub struct ViewSpan {
    #[serde(with = "u64_empty_string")]
    id: u64,
    #[serde(rename = "x")]
    start: u64,
    #[serde(rename = "w")]
    width: u64,
    #[serde(rename = "cat")]
    category: String,
    #[serde(rename = "t")]
    text: String,
    #[serde(rename = "c")]
    count: u64,
    #[serde(rename = "k")]
    kind: u8,
    #[serde(rename = "s")]
    start_in_parent: u32,
    #[serde(rename = "e")]
    end_in_parent: u32,
    #[serde(rename = "v")]
    secondary: u64,
}

#[derive(Debug)]
enum QueueItem<'a> {
    Span(SpanRef<'a>),
    SpanGraph(SpanGraphRef<'a>),
    SpanBottomUp(SpanBottomUpRef<'a>),
    SpanBottomUpSpan(SpanRef<'a>),
}

impl QueueItem<'_> {
    fn value(&self, value_mode: ValueMode) -> u64 {
        match self {
            QueueItem::Span(span) => value_mode.value_from_span(span),
            QueueItem::SpanGraph(span_graph) => value_mode.value_from_graph(span_graph),
            QueueItem::SpanBottomUp(span_bottom_up) => {
                value_mode.value_from_bottom_up(span_bottom_up)
            }
            QueueItem::SpanBottomUpSpan(span) => value_mode.value_from_bottom_up_span(span),
        }
    }

    fn max_depth(&self) -> u32 {
        match self {
            QueueItem::Span(span) => span.max_depth(),
            QueueItem::SpanGraph(span_graph) => span_graph.max_depth(),
            QueueItem::SpanBottomUp(span_bottom_up) => span_bottom_up.max_depth(),
            QueueItem::SpanBottomUpSpan(span) => span.max_depth(),
        }
    }
}

#[derive(Debug, PartialEq, Eq)]
enum FilterMode {
    SelectedItem,
    Parent,
    Child,
}

#[derive(Debug)]
struct QueueItemWithState<'a> {
    item: QueueItem<'a>,
    line_index: usize,
    start: u64,
    placeholder: bool,
    view_mode: ViewMode,
    filtered: Option<FilterMode>,
}

struct ChildItem<'a> {
    item: QueueItemWithState<'a>,
    depth: u32,
    pixel_range: (u64, u64),
}

impl Viewer {
    pub fn new() -> Self {
        Self::default()
    }

    pub fn set_view_mode(&mut self, id: SpanId, view_mode: Option<(ViewMode, bool)>) {
        self.span_options.entry(id).or_default().view_mode = view_mode;
    }

    pub fn compute_update(&mut self, store: &Store, view_rect: &ViewRect) -> Update {
        let mut highlighted_spans: FxHashSet<SpanId> = FxHashSet::default();
        let mut highlighted_span_parents: FxHashSet<SpanId> = FxHashSet::default();
        let search_mode = !view_rect.query.is_empty();
        let (query, focus_mode) = if let Some(query) = view_rect.query.strip_suffix('!') {
            (query, true)
        } else {
            (view_rect.query.as_str(), false)
        };

        let default_view_mode = view_rect.view_mode.as_str();
        let (default_view_mode, default_sort_mode) =
            if let Some(s) = default_view_mode.strip_suffix("-sorted-by-name") {
                (s, SortMode::Name)
            } else if let Some(s) = default_view_mode.strip_suffix("-sorted-by-value") {
                (s, SortMode::Value)
            } else if let Some(s) = default_view_mode.strip_suffix("-sorted") {
                (s, SortMode::Value)
            } else {
                (default_view_mode, SortMode::ExecutionOrder)
            };
        let (default_view_mode, with_root) = match default_view_mode {
            "aggregated" => (
                ViewMode::Aggregated {
                    sort_mode: default_sort_mode,
                },
                false,
            ),
            "root-aggregated" => (
                ViewMode::Aggregated {
                    sort_mode: default_sort_mode,
                },
                true,
            ),
            "raw-spans" => (
                ViewMode::RawSpans {
                    sort_mode: default_sort_mode,
                },
                false,
            ),
            "bottom-up" => (
                ViewMode::BottomUp {
                    sort_mode: default_sort_mode,
                },
                false,
            ),
            "aggregated-bottom-up" => (
                ViewMode::AggregatedBottomUp {
                    sort_mode: default_sort_mode,
                },
                false,
            ),
            "root-aggregated-bottom-up" => (
                ViewMode::AggregatedBottomUp {
                    sort_mode: default_sort_mode,
                },
                true,
            ),
            _ => (
                ViewMode::Aggregated {
                    sort_mode: default_sort_mode,
                },
                false,
            ),
        };

        let value_mode = match view_rect.value_mode.as_str() {
            "duration" => ValueMode::Duration,
            "cpu" => ValueMode::Cpu,
            "allocations" => ValueMode::Allocations,
            "deallocations" => ValueMode::Deallocations,
            "persistent-deallocations" => ValueMode::PersistentAllocations,
            "allocation-count" => ValueMode::AllocationCount,
            "allocations-per-time" => ValueMode::AllocationsPerTime,
            "allocation-count-per-time" => ValueMode::AllocationCountPerTime,
            "persistent-allocations-per-time" => ValueMode::PersistentAllocationsPerTime,
            "count" => ValueMode::Count,
            _ => ValueMode::Duration,
        };

        if !store.has_time_info() && matches!(value_mode, ValueMode::Duration) {
            return Update {
                lines: vec![ViewLineUpdate {
                    spans: vec![ViewSpan {
                        id: 0,
                        start: 0,
                        width: 1,
                        category: "info".to_string(),
                        text: "No time info in trace".to_string(),
                        count: 1,
                        kind: 0,
                        start_in_parent: 0,
                        end_in_parent: 0,
                        secondary: 0,
                    }],
                    y: 0,
                }],
                max: 1,
            };
        }

        let mut queue = Vec::new();

        let root_spans = if with_root {
            vec![store.root_span()]
        } else {
            let mut root_spans = store.root_spans().collect::<Vec<_>>();
            root_spans.sort_by_key(|span| span.start());
            root_spans
        };

        let mut children = Vec::new();
        let mut current = 0;
        let offset = root_spans
            .iter()
            .min_by_key(|span| span.start())
            .map_or(Timestamp::ZERO, |span| span.start());
        root_spans.par_iter().for_each(|span| {
            span.max_depth();
            QueueItem::Span(*span).value(value_mode);
        });
        for span in root_spans {
            if matches!(value_mode, ValueMode::Duration) {
                // Move current to start if needed.
                current = max(current, *(span.start() - offset));
            }
            if add_child_item(
                &mut children,
                &mut current,
                view_rect,
                0,
                default_view_mode,
                value_mode,
                QueueItem::Span(span),
                Some(if search_mode {
                    FilterMode::Parent
                } else {
                    FilterMode::SelectedItem
                }),
            ) && search_mode
            {
                let mut has_results = false;
                for mut result in span.search(query) {
                    has_results = true;
                    highlighted_spans.insert(result.id());
                    while let Some(parent) = result.parent() {
                        result = parent;
                        if !highlighted_span_parents.insert(result.id()) {
                            break;
                        }
                    }
                }
                if has_results {
                    highlighted_spans.insert(span.id());
                } else {
                    children.last_mut().unwrap().item.filtered = None;
                }
            }
        }
        enqueue_children(children, &mut queue);
        queue.par_iter().for_each(|item| {
            let QueueItem::Span(span) = item.item else {
                return;
            };
            let view_mode = if span.is_complete() {
                item.view_mode
            } else {
                item.view_mode.as_spans()
            };

            match (view_mode.bottom_up(), view_mode.aggregate_children()) {
                (false, false) => {}
                (false, true) => {
                    span.graph()
                        .collect::<Vec<_>>()
                        .par_iter()
                        .for_each(|event| {
                            value_mode.value_from_graph_event(event);
                        });
                }
                (true, false) => {
                    span.bottom_up()
                        .collect::<Vec<_>>()
                        .par_iter()
                        .for_each(|bu| {
                            bu.spans().collect::<Vec<_>>().par_iter().for_each(|span| {
                                value_mode.value_from_bottom_up_span(span);
                            });
                        });
                }
                (true, true) => {
                    span.bottom_up()
                        .collect::<Vec<_>>()
                        .par_iter()
                        .for_each(|bu| {
                            value_mode.value_from_bottom_up(bu);
                        });
                }
            }
        });

        let mut lines: Vec<Vec<LineEntry<'_>>> = vec![];

        while let Some(QueueItemWithState {
            item: span,
            line_index,
            start,
            placeholder,
            view_mode,
            mut filtered,
        }) = queue.pop()
        {
            let line = get_line(&mut lines, line_index);
            let width = span.value(value_mode);
            let secondary = span.value(value_mode.secondary());

            let skipped_by_focus =
                focus_mode && matches!(filtered, Some(FilterMode::Parent) | None);

            let get_filter_mode = |span: SpanId| {
                if focus_mode
                    && matches!(filtered, Some(FilterMode::SelectedItem | FilterMode::Child))
                {
                    Some(FilterMode::Child)
                } else if search_mode {
                    if highlighted_spans.contains(&span) {
                        Some(FilterMode::SelectedItem)
                    } else if highlighted_span_parents.contains(&span) {
                        Some(FilterMode::Parent)
                    } else {
                        None
                    }
                } else {
                    Some(FilterMode::SelectedItem)
                }
            };

            // compute children
            let mut children = Vec::new();
            let mut current = start;
            let child_line_index = if skipped_by_focus {
                line_index
            } else {
                line_index + 1
            };
            match &span {
                QueueItem::Span(span) => {
                    let (selected_view_mode, inherit) = (!span.is_root())
                        .then(|| self.span_options.get(&span.id()).and_then(|o| o.view_mode))
                        .flatten()
                        .unwrap_or_else(|| {
                            (
                                if span.is_complete() {
                                    view_mode
                                } else {
                                    view_mode.as_spans()
                                },
                                false,
                            )
                        });

                    let view_mode = if inherit {
                        selected_view_mode
                    } else {
                        view_mode
                    };

                    let selected_view_mode =
                        if search_mode && highlighted_span_parents.contains(&span.id()) {
                            selected_view_mode.as_spans()
                        } else {
                            selected_view_mode
                        };

                    if selected_view_mode.bottom_up() {
                        let bottom_up = span.bottom_up();
                        if selected_view_mode.aggregate_children() {
                            let bottom_up = match selected_view_mode.sort_children() {
                                SortMode::Value => {
                                    Either::Left(bottom_up.sorted_by_cached_key(|child| {
                                        Reverse(value_mode.value_from_bottom_up(child))
                                    }))
                                }
                                SortMode::Name => {
                                    Either::Left(bottom_up.sorted_by_cached_key(|child| {
                                        let (cat, title) = child.nice_name();
                                        (title.to_string(), cat.to_string())
                                    }))
                                }
                                SortMode::ExecutionOrder => Either::Right(bottom_up),
                            };
                            for child in bottom_up {
                                // TODO search
                                add_child_item(
                                    &mut children,
                                    &mut current,
                                    view_rect,
                                    child_line_index,
                                    view_mode,
                                    value_mode,
                                    QueueItem::SpanBottomUp(child),
                                    Some(FilterMode::SelectedItem),
                                );
                            }
                        } else {
                            let bottom_up = bottom_up
                                .flat_map(|bottom_up| bottom_up.spans().collect::<Vec<_>>());
                            let bottom_up = match selected_view_mode.sort_children() {
                                SortMode::Value => {
                                    Either::Left(bottom_up.sorted_by_cached_key(|child| {
                                        Reverse(value_mode.value_from_bottom_up_span(child))
                                    }))
                                }
                                SortMode::Name => {
                                    Either::Left(bottom_up.sorted_by_cached_key(|child| {
                                        let (cat, title) = child.nice_name();
                                        (title.to_string(), cat.to_string())
                                    }))
                                }
                                SortMode::ExecutionOrder => Either::Right(bottom_up),
                            };
                            for child in bottom_up {
                                let filtered = get_filter_mode(child.id());
                                add_child_item(
                                    &mut children,
                                    &mut current,
                                    view_rect,
                                    child_line_index,
                                    view_mode,
                                    value_mode,
                                    QueueItem::SpanBottomUpSpan(child),
                                    filtered,
                                );
                            }
                        }
                    } else if !selected_view_mode.aggregate_children() {
                        let spans = match selected_view_mode.sort_children() {
                            SortMode::Value => {
                                Either::Left(span.events().sorted_by_cached_key(|child| {
                                    Reverse(value_mode.value_from_event(child))
                                }))
                            }
                            SortMode::Name => {
                                Either::Left(span.events().sorted_by_cached_key(|child| {
                                    let (cat, title) = match child {
                                        SpanEventRef::Child { span } => span.nice_name(),
                                        SpanEventRef::SelfTime { .. } => (&rcstr!(""), &rcstr!("")),
                                    };
                                    (title.to_string(), cat.to_string())
                                }))
                            }
                            SortMode::ExecutionOrder => Either::Right(span.events()),
                        };
                        for child in spans {
                            match child {
                                SpanEventRef::SelfTime { .. } => {
                                    current += value_mode.value_from_event(&child);
                                }
                                SpanEventRef::Child { span: child } => {
                                    let filtered = get_filter_mode(child.id());
                                    add_child_item(
                                        &mut children,
                                        &mut current,
                                        view_rect,
                                        child_line_index,
                                        view_mode,
                                        value_mode,
                                        QueueItem::Span(child),
                                        filtered,
                                    );
                                }
                            }
                        }
                    } else {
                        let events = match selected_view_mode.sort_children() {
                            SortMode::Value => {
                                Either::Left(span.graph().sorted_by_cached_key(|child| {
                                    Reverse(value_mode.value_from_graph_event(child))
                                }))
                            }
                            SortMode::Name => {
                                Either::Left(span.graph().sorted_by_cached_key(|child| {
                                    let (cat, title) = match child {
                                        SpanGraphEventRef::Child { graph } => graph.nice_name(),
                                        SpanGraphEventRef::SelfTime { .. } => {
                                            (&rcstr!(""), &rcstr!(""))
                                        }
                                    };
                                    (title.to_string(), cat.to_string())
                                }))
                            }
                            SortMode::ExecutionOrder => Either::Right(span.graph()),
                        };
                        for event in events {
                            let filtered = if search_mode {
                                None
                            } else {
                                Some(FilterMode::SelectedItem)
                            };
                            match event {
                                SpanGraphEventRef::SelfTime { duration: _ } => {}
                                SpanGraphEventRef::Child { graph } => {
                                    add_child_item(
                                        &mut children,
                                        &mut current,
                                        view_rect,
                                        child_line_index,
                                        view_mode,
                                        value_mode,
                                        QueueItem::SpanGraph(graph),
                                        filtered,
                                    );
                                }
                            }
                        }
                    }
                }
                QueueItem::SpanGraph(span_graph) => {
                    let (selected_view_mode, inherit) = self
                        .span_options
                        .get(&span_graph.id())
                        .and_then(|o| o.view_mode)
                        .unwrap_or((view_mode, false));

                    let view_mode = if inherit {
                        selected_view_mode
                    } else {
                        view_mode
                    };
                    if selected_view_mode.bottom_up() {
                        let bottom_up = span_graph.bottom_up();
                        if selected_view_mode.aggregate_children() {
                            let bottom_up = match selected_view_mode.sort_children() {
                                SortMode::Value => {
                                    Either::Left(bottom_up.sorted_by_cached_key(|child| {
                                        Reverse(value_mode.value_from_bottom_up(child))
                                    }))
                                }
                                SortMode::Name => {
                                    Either::Left(bottom_up.sorted_by_cached_key(|child| {
                                        let (cat, title) = child.nice_name();
                                        (title.to_string(), cat.to_string())
                                    }))
                                }
                                SortMode::ExecutionOrder => Either::Right(bottom_up),
                            };
                            for child in bottom_up {
                                // TODO search
                                add_child_item(
                                    &mut children,
                                    &mut current,
                                    view_rect,
                                    child_line_index,
                                    view_mode,
                                    value_mode,
                                    QueueItem::SpanBottomUp(child),
                                    Some(FilterMode::SelectedItem),
                                );
                            }
                        } else {
                            let bottom_up = bottom_up
                                .flat_map(|bottom_up| bottom_up.spans().collect::<Vec<_>>());
                            let bottom_up = match selected_view_mode.sort_children() {
                                SortMode::Value => {
                                    Either::Left(bottom_up.sorted_by_cached_key(|child| {
                                        Reverse(value_mode.value_from_bottom_up_span(child))
                                    }))
                                }
                                SortMode::Name => {
                                    Either::Left(bottom_up.sorted_by_cached_key(|child| {
                                        let (cat, title) = child.nice_name();
                                        (title.to_string(), cat.to_string())
                                    }))
                                }
                                SortMode::ExecutionOrder => {
                                    Either::Right(bottom_up.sorted_by_key(|child| child.start()))
                                }
                            };
                            for child in bottom_up {
                                let filtered = get_filter_mode(child.id());
                                add_child_item(
                                    &mut children,
                                    &mut current,
                                    view_rect,
                                    child_line_index,
                                    view_mode,
                                    value_mode,
                                    QueueItem::SpanBottomUpSpan(child),
                                    filtered,
                                );
                            }
                        }
                    } else if !selected_view_mode.aggregate_children() && span_graph.count() > 1 {
                        let spans = match selected_view_mode.sort_children() {
                            SortMode::Value => {
                                Either::Left(span_graph.root_spans().sorted_by_cached_key(
                                    |child| Reverse(value_mode.value_from_span(child)),
                                ))
                            }
                            SortMode::Name => Either::Left(
                                span_graph.root_spans().sorted_by_cached_key(|child| {
                                    let (cat, title) = child.nice_name();
                                    (title.to_string(), cat.to_string())
                                }),
                            ),
                            SortMode::ExecutionOrder => Either::Right(
                                span_graph.root_spans().sorted_by_key(|child| child.start()),
                            ),
                        };
                        for child in spans {
                            let filtered = get_filter_mode(child.id());
                            add_child_item(
                                &mut children,
                                &mut current,
                                view_rect,
                                child_line_index,
                                view_mode,
                                value_mode,
                                QueueItem::Span(child),
                                filtered,
                            );
                        }
                    } else {
                        let events = match selected_view_mode.sort_children() {
                            SortMode::Value => {
                                Either::Left(span_graph.events().sorted_by_cached_key(|child| {
                                    Reverse(value_mode.value_from_graph_event(child))
                                }))
                            }
                            SortMode::Name => {
                                Either::Left(span_graph.events().sorted_by_cached_key(|child| {
                                    let (cat, title) = match child {
                                        SpanGraphEventRef::Child { graph } => graph.nice_name(),
                                        SpanGraphEventRef::SelfTime { .. } => {
                                            (&rcstr!(""), &rcstr!(""))
                                        }
                                    };
                                    (title.to_string(), cat.to_string())
                                }))
                            }
                            SortMode::ExecutionOrder => Either::Right(span_graph.events()),
                        };
                        for child in events {
                            if let SpanGraphEventRef::Child { graph } = child {
                                let filtered = if search_mode {
                                    None
                                } else {
                                    Some(FilterMode::SelectedItem)
                                };
                                add_child_item(
                                    &mut children,
                                    &mut current,
                                    view_rect,
                                    child_line_index,
                                    view_mode,
                                    value_mode,
                                    QueueItem::SpanGraph(graph),
                                    filtered,
                                );
                            }
                        }
                    }
                }
                QueueItem::SpanBottomUp(bottom_up) => {
                    let view_mode = self
                        .span_options
                        .get(&bottom_up.id())
                        .and_then(|o| o.view_mode)
                        .map(|(v, _)| v.as_bottom_up())
                        .unwrap_or(view_mode);

                    if view_mode.aggregate_children() {
                        let bottom_up = match view_mode.sort_children() {
                            SortMode::Value => {
                                Either::Left(bottom_up.children().sorted_by_cached_key(|child| {
                                    Reverse(value_mode.value_from_bottom_up(child))
                                }))
                            }
                            SortMode::Name => {
                                Either::Left(bottom_up.children().sorted_by_cached_key(|child| {
                                    let (cat, title) = child.nice_name();
                                    (title.to_string(), cat.to_string())
                                }))
                            }
                            SortMode::ExecutionOrder => Either::Right(bottom_up.children()),
                        };
                        for child in bottom_up {
                            // TODO search
                            add_child_item(
                                &mut children,
                                &mut current,
                                view_rect,
                                child_line_index,
                                view_mode,
                                value_mode,
                                QueueItem::SpanBottomUp(child),
                                Some(FilterMode::SelectedItem),
                            );
                        }
                    } else {
                        let spans = match view_mode.sort_children() {
                            SortMode::Value => {
                                Either::Left(bottom_up.spans().sorted_by_cached_key(|child| {
                                    Reverse(value_mode.value_from_bottom_up_span(child))
                                }))
                            }
                            SortMode::Name => {
                                Either::Left(bottom_up.spans().sorted_by_cached_key(|child| {
                                    let (cat, title) = child.nice_name();
                                    (title.to_string(), cat.to_string())
                                }))
                            }
                            SortMode::ExecutionOrder => Either::Right(
                                bottom_up.spans().sorted_by_key(|child| child.start()),
                            ),
                        };
                        for child in spans {
                            let filtered = get_filter_mode(child.id());
                            add_child_item(
                                &mut children,
                                &mut current,
                                view_rect,
                                child_line_index,
                                view_mode,
                                value_mode,
                                QueueItem::SpanBottomUpSpan(child),
                                filtered,
                            );
                        }
                    }
                }
                QueueItem::SpanBottomUpSpan(_) => {
                    // no children
                }
            }

            // When span size is smaller than a pixel, we only show the deepest child.
            if placeholder {
                let child = children
                    .into_iter()
                    .max_by_key(|ChildItem { item, depth, .. }| (item.filtered.is_some(), *depth));
                if let Some(ChildItem {
                    item: mut entry, ..
                }) = child
                {
                    entry.placeholder = true;
                    queue.push(entry);
                }

                if !skipped_by_focus {
                    // add span to line
                    line.push(LineEntry {
                        start,
                        width,
                        secondary: 0,
                        ty: LineEntryType::Placeholder(filtered),
                    });
                }
            } else {
                // add children to queue
                enqueue_children(children, &mut queue);

                // check if we should filter based on width or count
                if !skipped_by_focus {
                    let count = match &span {
                        QueueItem::Span(_) => 1,
                        QueueItem::SpanGraph(span_graph) => span_graph.count(),
                        QueueItem::SpanBottomUp(bottom_up) => bottom_up.count(),
                        QueueItem::SpanBottomUpSpan(_) => 1,
                    };

                    if let Some(false) = view_rect.count_filter.as_ref().map(|filter| match filter
                        .op
                    {
                        crate::server::Op::Gt => count > filter.value as usize,
                        crate::server::Op::Lt => count < filter.value as usize,
                    }) {
                        filtered = Some(FilterMode::SelectedItem)
                    }

                    if let Some(false) = view_rect.value_filter.as_ref().map(|filter| match filter
                        .op
                    {
                        crate::server::Op::Gt => width > filter.value,
                        crate::server::Op::Lt => width < filter.value,
                    }) {
                        filtered = Some(FilterMode::SelectedItem)
                    }

                    // add span to line
                    line.push(LineEntry {
                        start,
                        width,
                        secondary,
                        ty: match span {
                            QueueItem::Span(span) => LineEntryType::Span { span, filtered },
                            QueueItem::SpanGraph(span_graph) => {
                                LineEntryType::SpanGraph(span_graph, filtered)
                            }
                            QueueItem::SpanBottomUp(bottom_up) => {
                                LineEntryType::SpanBottomUp(bottom_up, filtered)
                            }
                            QueueItem::SpanBottomUpSpan(bottom_up_span) => {
                                LineEntryType::SpanBottomUpSpan(bottom_up_span, filtered)
                            }
                        },
                    });
                }
            }
        }

        let lines = lines
            .into_iter()
            .enumerate()
            .map(|(y, line)| ViewLineUpdate {
                y: y as u64,
                spans: line
                    .into_iter()
                    .map(|entry| match entry.ty {
                        LineEntryType::Placeholder(filtered) => ViewSpan {
                            id: 0,
                            start: entry.start,
                            width: entry.width,
                            category: String::new(),
                            text: String::new(),
                            count: 1,
                            kind: match filtered {
                                Some(_) => 1,
                                None => 11,
                            },
                            start_in_parent: 0,
                            end_in_parent: 0,
                            secondary: 0,
                        },
                        LineEntryType::Span { span, filtered } => {
                            let (category, text) = span.nice_name();
                            let mut start_in_parent = 0;
                            let mut end_in_parent = 0;
                            if let Some(parent) = span.parent() {
                                let parent_start = parent.start();
                                let parent_duration = parent.end() - parent_start;
                                if !parent_duration.is_zero() {
                                    start_in_parent = ((span.start() - parent_start) * 10000
                                        / parent_duration)
                                        as u32;
                                    end_in_parent = ((span.end() - parent_start) * 10000
                                        / parent_duration)
                                        as u32;
                                } else {
                                    start_in_parent = 0;
                                    end_in_parent = 10000;
                                }
                            }
                            ViewSpan {
                                id: if !span.is_root() {
                                    span.id().get() as u64
                                } else {
                                    Default::default()
                                },
                                start: entry.start,
                                width: entry.width,
                                category: category.to_string(),
                                text: text.to_string(),
                                count: 1,
                                kind: match filtered {
                                    Some(_) => 0,
                                    None => 10,
                                },
                                start_in_parent,
                                end_in_parent,
                                secondary: entry.secondary,
                            }
                        }
                        LineEntryType::SpanGraph(graph, filtered) => {
                            let (category, text) = graph.nice_name();
                            ViewSpan {
                                id: graph.id().get() as u64,
                                start: entry.start,
                                width: entry.width,
                                category: category.to_string(),
                                text: text.to_string(),
                                count: graph.count() as u64,
                                kind: match filtered {
                                    Some(_) => 0,
                                    None => 10,
                                },
                                start_in_parent: 0,
                                end_in_parent: 0,
                                secondary: entry.secondary,
                            }
                        }
                        LineEntryType::SpanBottomUp(bottom_up, filtered) => {
                            let (category, text) = bottom_up.nice_name();
                            ViewSpan {
                                id: bottom_up.id().get() as u64,
                                start: entry.start,
                                width: entry.width,
                                category: category.to_string(),
                                text: text.to_string(),
                                count: bottom_up.count() as u64,
                                kind: match filtered {
                                    Some(_) => 2,
                                    None => 12,
                                },
                                start_in_parent: 0,
                                end_in_parent: 0,
                                secondary: entry.secondary,
                            }
                        }
                        LineEntryType::SpanBottomUpSpan(bottom_up_span, filtered) => {
                            let (category, text) = bottom_up_span.nice_name();
                            ViewSpan {
                                id: bottom_up_span.id().get() as u64,
                                start: entry.start,
                                width: entry.width,
                                category: category.to_string(),
                                text: text.to_string(),
                                count: 1,
                                kind: match filtered {
                                    Some(_) => 2,
                                    None => 12,
                                },
                                start_in_parent: 0,
                                end_in_parent: 0,
                                secondary: entry.secondary,
                            }
                        }
                    })
                    .collect(),
            })
            .collect();

        Update {
            lines,
            max: max(1, current),
        }
    }
}

#[allow(clippy::too_many_arguments)]
fn add_child_item<'a>(
    children: &mut Vec<ChildItem<'a>>,
    current: &mut u64,
    view_rect: &ViewRect,
    line_index: usize,
    view_mode: ViewMode,
    value_mode: ValueMode,
    child: QueueItem<'a>,
    filtered: Option<FilterMode>,
) -> bool {
    let child_width = child.value(value_mode);
    let max_depth = child.max_depth();
    let pixel1 = *current * view_rect.horizontal_pixels / view_rect.width;
    let pixel2 = ((*current + child_width) * view_rect.horizontal_pixels).div_ceil(view_rect.width);
    let start = *current;
    *current += child_width;

    // filter by view rect (vertical)
    if line_index > (view_rect.y + view_rect.height + EXTRA_HEIGHT) as usize {
        return false;
    }

    if line_index > 0 {
        // filter by view rect (horizontal)
        if start > view_rect.x + view_rect.width * (100 + EXTRA_WIDTH_PERCENTAGE) / 100 {
            return false;
        }
        if *current
            < view_rect
                .x
                .saturating_sub(view_rect.width * EXTRA_WIDTH_PERCENTAGE / 100)
        {
            return false;
        }
    }

    children.push(ChildItem {
        item: QueueItemWithState {
            item: child,
            line_index,
            start,
            placeholder: false,
            view_mode,
            filtered,
        },
        depth: max_depth,
        pixel_range: (pixel1, pixel2),
    });

    true
}

const MIN_VISIBLE_PIXEL_SIZE: u64 = 3;

fn enqueue_children<'a>(mut children: Vec<ChildItem<'a>>, queue: &mut Vec<QueueItemWithState<'a>>) {
    children.reverse();
    let mut last_pixel = u64::MAX;
    let mut last_max_depth = 0;
    for ChildItem {
        item: mut entry,
        depth: max_depth,
        pixel_range: (pixel1, pixel2),
    } in children
    {
        if last_pixel <= pixel1 + MIN_VISIBLE_PIXEL_SIZE {
            if last_max_depth < max_depth {
                queue.pop();
                entry.placeholder = true;
            } else {
                if let Some(entry) = queue.last_mut() {
                    entry.placeholder = true;
                }
                continue;
            }
        };
        queue.push(entry);
        last_max_depth = max_depth;
        last_pixel = pixel2;
    }
}

fn get_line<T: Default>(lines: &mut Vec<T>, i: usize) -> &mut T {
    if i >= lines.len() {
        lines.resize_with(i + 1, || Default::default());
    }
    &mut lines[i]
}

struct LineEntry<'a> {
    start: u64,
    width: u64,
    secondary: u64,
    ty: LineEntryType<'a>,
}

enum LineEntryType<'a> {
    Placeholder(Option<FilterMode>),
    Span {
        span: SpanRef<'a>,
        filtered: Option<FilterMode>,
    },
    SpanGraph(SpanGraphRef<'a>, Option<FilterMode>),
    SpanBottomUp(SpanBottomUpRef<'a>, Option<FilterMode>),
    SpanBottomUpSpan(SpanRef<'a>, Option<FilterMode>),
}