From 04b9dc2d30cfc9b383029f6a4b02e2725108b0ae Mon Sep 17 00:00:00 2001 From: Andreas Kling Date: Thu, 4 Jul 2019 16:16:50 +0200 Subject: Libraries: Create top level directory for libraries. Things were getting a little crowded in the project root, so this patch moves the Lib*/ directories into Libraries/. --- Libraries/LibGUI/GTreeView.cpp | 299 +++++++++++++++++++++++++++++++++++++++++ 1 file changed, 299 insertions(+) create mode 100644 Libraries/LibGUI/GTreeView.cpp (limited to 'Libraries/LibGUI/GTreeView.cpp') diff --git a/Libraries/LibGUI/GTreeView.cpp b/Libraries/LibGUI/GTreeView.cpp new file mode 100644 index 0000000000..59e35426ff --- /dev/null +++ b/Libraries/LibGUI/GTreeView.cpp @@ -0,0 +1,299 @@ +#include +#include +#include + +//#define DEBUG_ITEM_RECTS + +struct GTreeView::MetadataForIndex { + bool open { false }; +}; + +GTreeView::MetadataForIndex& GTreeView::ensure_metadata_for_index(const GModelIndex& index) const +{ + ASSERT(index.is_valid()); + auto it = m_view_metadata.find(index.internal_data()); + if (it != m_view_metadata.end()) + return *it->value; + auto new_metadata = make(); + auto& new_metadata_ref = *new_metadata; + m_view_metadata.set(index.internal_data(), move(new_metadata)); + return new_metadata_ref; +} + +GTreeView::GTreeView(GWidget* parent) + : GAbstractView(parent) +{ + set_frame_shape(FrameShape::Container); + set_frame_shadow(FrameShadow::Sunken); + set_frame_thickness(2); + + m_expand_bitmap = GraphicsBitmap::load_from_file("/res/icons/treeview-expand.png"); + m_collapse_bitmap = GraphicsBitmap::load_from_file("/res/icons/treeview-collapse.png"); +} + +GTreeView::~GTreeView() +{ +} + +GModelIndex GTreeView::index_at_content_position(const Point& position, bool& is_toggle) const +{ + is_toggle = false; + if (!model()) + return {}; + GModelIndex result; + traverse_in_paint_order([&](const GModelIndex& index, const Rect& rect, const Rect& toggle_rect, int) { + if (rect.contains(position)) { + result = index; + return IterationDecision::Break; + } + if (toggle_rect.contains(position)) { + result = index; + is_toggle = true; + return IterationDecision::Break; + } + return IterationDecision::Continue; + }); + return result; +} + +void GTreeView::mousedown_event(GMouseEvent& event) +{ + if (!model()) + return; + auto& model = *this->model(); + auto adjusted_position = event.position().translated(horizontal_scrollbar().value() - frame_thickness(), vertical_scrollbar().value() - frame_thickness()); + bool is_toggle; + auto index = index_at_content_position(adjusted_position, is_toggle); + if (!index.is_valid()) + return; + + if (model.selected_index() != index) { + model.set_selected_index(index); + update(); + } + + if (is_toggle && model.row_count(index)) { + auto& metadata = ensure_metadata_for_index(index); + metadata.open = !metadata.open; + update_content_size(); + update(); + } +} + +template +void GTreeView::traverse_in_paint_order(Callback callback) const +{ + ASSERT(model()); + auto& model = *this->model(); + int indent_level = 0; + int y_offset = 0; + + Function traverse_index = [&](const GModelIndex& index) { + int row_count_at_index = model.row_count(index); + if (index.is_valid()) { + auto& metadata = ensure_metadata_for_index(index); + int x_offset = indent_level * indent_width_in_pixels(); + auto node_text = model.data(index, GModel::Role::Display).to_string(); + Rect rect = { + x_offset, y_offset, + icon_size() + icon_spacing() + text_padding() + font().width(node_text) + text_padding(), item_height() + }; + Rect toggle_rect; + if (row_count_at_index > 0) { + int toggle_x = indent_width_in_pixels() * indent_level - icon_size() / 2 - 4; + toggle_rect = { toggle_x, rect.y(), toggle_size(), toggle_size() }; + toggle_rect.center_vertically_within(rect); + } + if (callback(index, rect, toggle_rect, indent_level) == IterationDecision::Break) + return IterationDecision::Break; + y_offset += item_height(); + // NOTE: Skip traversing children if this index is closed! + if (!metadata.open) + return IterationDecision::Continue; + } + + ++indent_level; + int row_count = model.row_count(index); + for (int i = 0; i < row_count; ++i) { + if (traverse_index(model.index(i, 0, index)) == IterationDecision::Break) + return IterationDecision::Break; + } + --indent_level; + return IterationDecision::Continue; + }; + traverse_index(model.index(0, 0, GModelIndex())); +} + +void GTreeView::paint_event(GPaintEvent& event) +{ + GFrame::paint_event(event); + GPainter painter(*this); + painter.add_clip_rect(frame_inner_rect()); + painter.add_clip_rect(event.rect()); + painter.fill_rect(event.rect(), Color::White); + painter.translate(frame_inner_rect().location()); + painter.translate(-horizontal_scrollbar().value(), -vertical_scrollbar().value()); + + if (!model()) + return; + auto& model = *this->model(); + auto visible_content_rect = this->visible_content_rect(); + + traverse_in_paint_order([&](const GModelIndex& index, const Rect& rect, const Rect& toggle_rect, int indent_level) { + if (!rect.intersects(visible_content_rect)) + return IterationDecision::Continue; +#ifdef DEBUG_ITEM_RECTS + painter.fill_rect(rect, Color::WarmGray); +#endif + + Color background_color = Color::from_rgb(0xffffff); + Color text_color = Color::from_rgb(0x000000); + + Rect icon_rect = { rect.x(), rect.y(), icon_size(), icon_size() }; + auto icon = model.data(index, GModel::Role::Icon); + if (icon.is_icon()) { + if (auto* bitmap = icon.as_icon().bitmap_for_size(icon_size())) + painter.blit(icon_rect.location(), *bitmap, bitmap->rect()); + } + Rect text_rect = { + icon_rect.right() + 1 + icon_spacing(), rect.y(), + rect.width() - icon_size() - icon_spacing(), rect.height() + }; + if (index == model.selected_index()) { + background_color = is_focused() ? Color::from_rgb(0x84351a) : Color::from_rgb(0x606060); + text_color = Color::from_rgb(0xffffff); + painter.fill_rect(text_rect, background_color); + } + auto node_text = model.data(index, GModel::Role::Display).to_string(); + painter.draw_text(text_rect, node_text, TextAlignment::Center, text_color); + auto index_at_indent = index; + for (int i = indent_level; i >= 0; --i) { + auto parent_of_index_at_indent = index_at_indent.parent(); + bool index_at_indent_is_last_in_parent = index_at_indent.row() == model.row_count(parent_of_index_at_indent) - 1; + Point a { indent_width_in_pixels() * i - icon_size() / 2, rect.y() - 2 }; + Point b { a.x(), a.y() + item_height() - 1 }; + if (index_at_indent_is_last_in_parent) + b.set_y(rect.center().y()); + if (!(i != indent_level && index_at_indent_is_last_in_parent)) + painter.draw_line(a, b, Color::MidGray); + + if (i == indent_level) { + Point c { a.x(), rect.center().y() }; + Point d { c.x() + icon_size() / 2, c.y() }; + painter.draw_line(c, d, Color::MidGray); + } + index_at_indent = parent_of_index_at_indent; + } + + if (!toggle_rect.is_empty()) { + auto& metadata = ensure_metadata_for_index(index); + if (metadata.open) + painter.blit(toggle_rect.location(), *m_collapse_bitmap, m_collapse_bitmap->rect()); + else + painter.blit(toggle_rect.location(), *m_expand_bitmap, m_expand_bitmap->rect()); + } + + return IterationDecision::Continue; + }); +} + +void GTreeView::scroll_into_view(const GModelIndex& a_index, Orientation orientation) +{ + if (!a_index.is_valid()) + return; + Rect found_rect; + traverse_in_paint_order([&](const GModelIndex& index, const Rect& rect, const Rect&, int) { + if (index == a_index) { + found_rect = rect; + return IterationDecision::Break; + } + return IterationDecision::Continue; + }); + GScrollableWidget::scroll_into_view(found_rect, orientation); +} + +void GTreeView::did_update_model() +{ + GAbstractView::did_update_model(); + update_content_size(); +} + +void GTreeView::did_update_selection() +{ + ASSERT(model()); + auto& model = *this->model(); + auto index = model.selected_index(); + if (!index.is_valid()) + return; + bool opened_any = false; + auto& metadata_for_index = ensure_metadata_for_index(index); + if (!metadata_for_index.open) { + opened_any = true; + metadata_for_index.open = true; + } + auto ancestor = index.parent(); + while (ancestor.is_valid()) { + auto& metadata_for_ancestor = ensure_metadata_for_index(ancestor); + if (!metadata_for_ancestor.open) { + metadata_for_ancestor.open = true; + opened_any = true; + } + ancestor = ancestor.parent(); + } + if (opened_any) + update_content_size(); + update(); + if (activates_on_selection()) + activate(index); +} + +void GTreeView::update_content_size() +{ + int height = 0; + int width = 0; + traverse_in_paint_order([&](const GModelIndex&, const Rect& rect, const Rect&, int) { + width = max(width, rect.right()); + height += rect.height(); + return IterationDecision::Continue; + }); + set_content_size({ width, height }); +} + +void GTreeView::keydown_event(GKeyEvent& event) +{ + if (!model()) + return; + auto cursor_index = model()->selected_index(); + if (event.key() == KeyCode::Key_Up) { + GModelIndex previous_index; + GModelIndex found_index; + traverse_in_paint_order([&](const GModelIndex& index, const Rect&, const Rect&, int) { + if (index == cursor_index) { + found_index = previous_index; + return IterationDecision::Break; + } + previous_index = index; + return IterationDecision::Continue; + }); + if (found_index.is_valid()) { + model()->set_selected_index(found_index); + update(); + } + return; + } + if (event.key() == KeyCode::Key_Down) { + GModelIndex previous_index; + GModelIndex found_index; + traverse_in_paint_order([&](const GModelIndex& index, const Rect&, const Rect&, int) { + if (previous_index == cursor_index) { + found_index = index; + return IterationDecision::Break; + } + previous_index = index; + return IterationDecision::Continue; + }); + if (found_index.is_valid()) + model()->set_selected_index(found_index); + return; + } +} -- cgit v1.2.3