123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245 |
- #include <LibGUI/GTreeView.h>
- #include <LibGUI/GPainter.h>
- #include <LibGUI/GScrollBar.h>
- //#define DEBUG_ITEM_RECTS
- struct Node {
- String text;
- Node* parent { nullptr };
- Vector<Node*> children;
- };
- class TestModel : public GModel {
- public:
- static Retained<TestModel> create() { return adopt(*new TestModel); }
- TestModel();
- virtual int row_count(const GModelIndex& = GModelIndex()) const override;
- virtual int column_count(const GModelIndex& = GModelIndex()) const override;
- virtual GVariant data(const GModelIndex&, Role = Role::Display) const override;
- virtual void update() override;
- virtual GModelIndex index(int row, int column = 0, const GModelIndex& parent = GModelIndex()) const override;
- virtual ColumnMetadata column_metadata(int) const override{ return { 100 }; }
- Node* m_root { nullptr };
- };
- Node* make_little_tree(int depth, Node* parent)
- {
- static int next_id = 0;
- Node* node = new Node;
- node->text = String::format("Node #%d", next_id++);
- node->parent = parent;
- if (depth)
- node->children.append(make_little_tree(depth - 1, node));
- return node;
- }
- GModelIndex TestModel::index(int row, int column, const GModelIndex& parent) const
- {
- if (!parent.is_valid())
- return create_index(row, column, m_root);
- auto& node = *(Node*)parent.internal_data();
- return create_index(row, column, node.children[row]);
- }
- TestModel::TestModel()
- {
- m_root = new Node;
- m_root->text = "Root";
- m_root->children.append(make_little_tree(3, m_root));
- m_root->children.append(make_little_tree(2, m_root));
- m_root->children.append(make_little_tree(1, m_root));
- }
- int TestModel::row_count(const GModelIndex& index) const
- {
- if (!index.is_valid())
- return 1;
- auto& node = *(const Node*)index.internal_data();
- return node.children.size();
- }
- int TestModel::column_count(const GModelIndex&) const
- {
- return 1;
- }
- void TestModel::update()
- {
- }
- GVariant TestModel::data(const GModelIndex& index, Role role) const
- {
- if (!index.is_valid())
- return { };
- auto& node = *(const Node*)index.internal_data();
- if (role == GModel::Role::Display) {
- return node.text;
- }
- if (role == GModel::Role::Icon) {
- if (node.children.is_empty())
- return GIcon::default_icon("filetype-unknown");
- return GIcon::default_icon("filetype-folder");
- }
- return { };
- }
- 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<MetadataForIndex>();
- 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(GFrame::Shape::Container);
- set_frame_shadow(GFrame::Shadow::Sunken);
- set_frame_thickness(2);
- set_model(TestModel::create());
- 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) const
- {
- if (!model())
- return { };
- GModelIndex result;
- traverse_in_paint_order([&] (const GModelIndex& index, const Rect& rect, int, bool) {
- if (rect.contains(position)) {
- result = index;
- return IterationDecision::Abort;
- }
- 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());
- auto index = index_at_content_position(adjusted_position);
- if (!index.is_valid()) {
- dbgprintf("GTV::mousedown: No valid index at %s (adjusted to: %s)\n", event.position().to_string().characters(), adjusted_position.to_string().characters());
- return;
- }
- dbgprintf("GTV::mousedown: Index %d,%d {%p}] at %s (adjusted to: %s)\n", index.row(), index.column(), index.internal_data(), event.position().to_string().characters(), adjusted_position.to_string().characters());
- auto& metadata = ensure_metadata_for_index(index);
- if (model.row_count(index)) {
- metadata.open = !metadata.open;
- dbgprintf("GTV::mousedown: toggle index %d,%d {%p} open: %d -> %d\n", index.row(), index.column(), index.internal_data(), !metadata.open, metadata.open);
- update();
- }
- }
- template<typename Callback>
- void GTreeView::traverse_in_paint_order(Callback callback) const
- {
- ASSERT(model());
- auto& model = *this->model();
- int indent_level = 0;
- int y_offset = 0;
- auto visible_content_rect = this->visible_content_rect();
- Function<IterationDecision(const GModelIndex&, bool)> traverse_index = [&] (const GModelIndex& index, bool is_last_in_parent) {
- 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() + font().width(node_text), item_height()
- };
- if (rect.intersects(visible_content_rect)) {
- if (callback(index, rect, indent_level, is_last_in_parent) == IterationDecision::Abort)
- return IterationDecision::Abort;
- }
- 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), i == row_count - 1) == IterationDecision::Abort)
- return IterationDecision::Abort;
- }
- --indent_level;
- return IterationDecision::Continue;
- };
- traverse_index(model.index(0, 0, GModelIndex()), true);
- }
- 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());
- if (!model())
- return;
- auto& model = *this->model();
- traverse_in_paint_order([&] (const GModelIndex& index, const Rect& rect, int indent_level, bool is_last_in_parent) {
- #ifdef DEBUG_ITEM_RECTS
- painter.fill_rect(rect, Color::LightGray);
- #endif
- 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(rect.location(), *bitmap, bitmap->rect());
- }
- Rect text_rect = {
- icon_rect.right() + 1 + icon_spacing(), rect.y(),
- rect.width() - icon_size() - icon_spacing(), rect.height()
- };
- auto node_text = model.data(index, GModel::Role::Display).to_string();
- painter.draw_text(text_rect, node_text, TextAlignment::CenterLeft, Color::Black);
- for (int i = 0; i <= indent_level; ++i) {
- Point a { indent_width_in_pixels() * i - icon_size() / 2, rect.y() };
- Point b { a.x(), a.y() + item_height() - 1 };
- if (i == indent_level && is_last_in_parent)
- b.set_y(rect.center().y());
- 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);
- }
- }
- return IterationDecision::Continue;
- });
- }
|