GTableView.cpp 12 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310
  1. #include <LibGUI/GTableView.h>
  2. #include <LibGUI/GModel.h>
  3. #include <LibGUI/GScrollBar.h>
  4. #include <LibGUI/GPainter.h>
  5. #include <Kernel/KeyCode.h>
  6. GTableView::GTableView(GWidget* parent)
  7. : GAbstractView(parent)
  8. {
  9. set_frame_shape(GFrame::Shape::Container);
  10. set_frame_shadow(GFrame::Shadow::Sunken);
  11. set_frame_thickness(2);
  12. }
  13. GTableView::~GTableView()
  14. {
  15. }
  16. void GTableView::update_content_size()
  17. {
  18. if (!model())
  19. return set_content_size({ });
  20. int content_width = 0;
  21. int column_count = model()->column_count();
  22. for (int i = 0; i < column_count; ++i)
  23. content_width += model()->column_metadata(i).preferred_width + horizontal_padding() * 2;
  24. int content_height = item_count() * item_height();
  25. set_content_size({ content_width, content_height });
  26. set_size_occupied_by_fixed_elements({ 0, header_height() });
  27. }
  28. void GTableView::did_update_model()
  29. {
  30. GAbstractView::did_update_model();
  31. update_content_size();
  32. update();
  33. }
  34. Rect GTableView::row_rect(int item_index) const
  35. {
  36. return { 0, header_height() + (item_index * item_height()), max(content_size().width(), width()), item_height() };
  37. }
  38. int GTableView::column_width(int column_index) const
  39. {
  40. return model()->column_metadata(column_index).preferred_width;
  41. }
  42. Rect GTableView::header_rect(int column_index) const
  43. {
  44. if (is_column_hidden(column_index))
  45. return { };
  46. int x_offset = 0;
  47. for (int i = 0; i < column_index; ++i) {
  48. if (is_column_hidden(i))
  49. continue;
  50. x_offset += column_width(i) + horizontal_padding() * 2;
  51. }
  52. auto column_metadata = model()->column_metadata(column_index);
  53. int column_width = column_metadata.preferred_width;
  54. return { x_offset, 0, column_width + horizontal_padding() * 2, header_height() };
  55. }
  56. void GTableView::mousedown_event(GMouseEvent& event)
  57. {
  58. if (event.y() < header_height()) {
  59. auto adjusted_position = event.position().translated(horizontal_scrollbar().value(), 0);
  60. for (int i = 0; i < model()->column_count(); ++i) {
  61. auto header_rect = this->header_rect(i);
  62. if (header_rect.contains(adjusted_position)) {
  63. auto new_sort_order = GSortOrder::Ascending;
  64. if (model()->key_column() == i)
  65. new_sort_order = model()->sort_order() == GSortOrder::Ascending
  66. ? GSortOrder::Descending
  67. : GSortOrder::Ascending;
  68. model()->set_key_column_and_sort_order(i, new_sort_order);
  69. return;
  70. }
  71. }
  72. return;
  73. }
  74. if (event.button() == GMouseButton::Left) {
  75. auto adjusted_position = event.position().translated(0, vertical_scrollbar().value());
  76. for (int i = 0; i < item_count(); ++i) {
  77. if (row_rect(i).contains(adjusted_position)) {
  78. model()->set_selected_index(model()->index(i, 0));
  79. update();
  80. return;
  81. }
  82. }
  83. model()->set_selected_index({ });
  84. update();
  85. }
  86. }
  87. void GTableView::paint_event(GPaintEvent& event)
  88. {
  89. GFrame::paint_event(event);
  90. GPainter painter(*this);
  91. painter.add_clip_rect(frame_inner_rect());
  92. painter.add_clip_rect(event.rect());
  93. painter.save();
  94. painter.translate(-horizontal_scrollbar().value(), -vertical_scrollbar().value());
  95. int exposed_width = max(content_size().width(), width());
  96. int painted_item_index = 0;
  97. int y_offset = header_height();
  98. for (int row_index = 0; row_index < model()->row_count(); ++row_index) {
  99. bool is_selected_row = row_index == model()->selected_index().row();
  100. int y = y_offset + painted_item_index * item_height();
  101. Color background_color;
  102. Color key_column_background_color;
  103. if (is_selected_row) {
  104. background_color = is_focused() ? Color::from_rgb(0x84351a) : Color::from_rgb(0x606060);
  105. key_column_background_color = is_focused() ? Color::from_rgb(0x84351a) : Color::from_rgb(0x606060);
  106. } else {
  107. if (alternating_row_colors() && (painted_item_index % 2)) {
  108. background_color = Color(210, 210, 210);
  109. key_column_background_color = Color(190, 190, 190);
  110. } else {
  111. background_color = Color::White;
  112. key_column_background_color = Color(235, 235, 235);
  113. }
  114. }
  115. painter.fill_rect(row_rect(painted_item_index), background_color);
  116. int x_offset = 0;
  117. for (int column_index = 0; column_index < model()->column_count(); ++column_index) {
  118. if (is_column_hidden(column_index))
  119. continue;
  120. auto column_metadata = model()->column_metadata(column_index);
  121. int column_width = column_metadata.preferred_width;
  122. const Font& font = column_metadata.font ? *column_metadata.font : this->font();
  123. bool is_key_column = model()->key_column() == column_index;
  124. Rect cell_rect(horizontal_padding() + x_offset, y, column_width, item_height());
  125. if (is_key_column) {
  126. auto cell_rect_for_fill = cell_rect.inflated(horizontal_padding() * 2, 0);
  127. painter.fill_rect(cell_rect_for_fill, key_column_background_color);
  128. }
  129. auto cell_index = model()->index(row_index, column_index);
  130. auto data = model()->data(cell_index);
  131. if (data.is_bitmap()) {
  132. painter.blit(cell_rect.location(), data.as_bitmap(), data.as_bitmap().rect());
  133. } else if (data.is_icon()) {
  134. if (auto bitmap = data.as_icon().bitmap_for_size(16))
  135. painter.blit(cell_rect.location(), *bitmap, bitmap->rect());
  136. } else {
  137. Color text_color;
  138. if (is_selected_row)
  139. text_color = Color::White;
  140. else
  141. text_color = model()->data(cell_index, GModel::Role::ForegroundColor).to_color(Color::Black);
  142. painter.draw_text(cell_rect, data.to_string(), font, column_metadata.text_alignment, text_color);
  143. }
  144. x_offset += column_width + horizontal_padding() * 2;
  145. }
  146. ++painted_item_index;
  147. };
  148. Rect unpainted_rect(0, header_height() + painted_item_index * item_height(), exposed_width, height());
  149. painter.fill_rect(unpainted_rect, Color::White);
  150. // Untranslate the painter vertically and do the column headers.
  151. painter.translate(0, vertical_scrollbar().value());
  152. if (headers_visible())
  153. paint_headers(painter);
  154. painter.restore();
  155. if (is_focused()) {
  156. Rect item_area_rect {
  157. 0,
  158. header_height(),
  159. width() - vertical_scrollbar().width(),
  160. height() - header_height() - horizontal_scrollbar().height()
  161. };
  162. painter.draw_rect(item_area_rect, Color::from_rgb(0x84351a));
  163. };
  164. }
  165. void GTableView::paint_headers(Painter& painter)
  166. {
  167. int exposed_width = max(content_size().width(), width());
  168. painter.fill_rect({ 0, 0, exposed_width, header_height() }, Color::LightGray);
  169. painter.draw_line({ 0, 0 }, { exposed_width - 1, 0 }, Color::White);
  170. painter.draw_line({ 0, header_height() - 1 }, { exposed_width - 1, header_height() - 1 }, Color::MidGray);
  171. int x_offset = 0;
  172. for (int column_index = 0; column_index < model()->column_count(); ++column_index) {
  173. if (is_column_hidden(column_index))
  174. continue;
  175. auto column_metadata = model()->column_metadata(column_index);
  176. int column_width = column_metadata.preferred_width;
  177. bool is_key_column = model()->key_column() == column_index;
  178. Rect cell_rect(x_offset, 0, column_width + horizontal_padding() * 2, header_height());
  179. if (is_key_column) {
  180. painter.fill_rect(cell_rect.shrunken(2, 2), Color::from_rgb(0xdddddd));
  181. }
  182. painter.draw_text(cell_rect.translated(horizontal_padding(), 0), model()->column_name(column_index), Font::default_bold_font(), TextAlignment::CenterLeft, Color::Black);
  183. x_offset += column_width + horizontal_padding() * 2;
  184. // Draw column separator.
  185. painter.draw_line(cell_rect.top_left().translated(0, 1), cell_rect.bottom_left().translated(0, -1), Color::White);
  186. painter.draw_line(cell_rect.top_right(), cell_rect.bottom_right().translated(0, -1), Color::MidGray);
  187. }
  188. // Draw the "start" of a new column to make the last separator look right.
  189. painter.draw_line({ x_offset, 1 }, { x_offset, header_height() - 2 }, Color::White);
  190. }
  191. int GTableView::item_count() const
  192. {
  193. return model()->row_count();
  194. }
  195. void GTableView::keydown_event(GKeyEvent& event)
  196. {
  197. if (!model())
  198. return;
  199. auto& model = *this->model();
  200. if (event.key() == KeyCode::Key_Return) {
  201. model.activate(model.selected_index());
  202. return;
  203. }
  204. if (event.key() == KeyCode::Key_Up) {
  205. GModelIndex new_index;
  206. if (model.selected_index().is_valid())
  207. new_index = model.index(model.selected_index().row() - 1, model.selected_index().column());
  208. else
  209. new_index = model.index(0, 0);
  210. if (model.is_valid(new_index)) {
  211. model.set_selected_index(new_index);
  212. scroll_into_view(new_index, Orientation::Vertical);
  213. update();
  214. }
  215. return;
  216. }
  217. if (event.key() == KeyCode::Key_Down) {
  218. GModelIndex new_index;
  219. if (model.selected_index().is_valid())
  220. new_index = model.index(model.selected_index().row() + 1, model.selected_index().column());
  221. else
  222. new_index = model.index(0, 0);
  223. if (model.is_valid(new_index)) {
  224. model.set_selected_index(new_index);
  225. scroll_into_view(new_index, Orientation::Vertical);
  226. update();
  227. }
  228. return;
  229. }
  230. if (event.key() == KeyCode::Key_PageUp) {
  231. int items_per_page = visible_content_rect().height() / item_height();
  232. auto new_index = model.index(max(0, model.selected_index().row() - items_per_page), model.selected_index().column());
  233. if (model.is_valid(new_index)) {
  234. model.set_selected_index(new_index);
  235. scroll_into_view(new_index, Orientation::Vertical);
  236. update();
  237. }
  238. return;
  239. }
  240. if (event.key() == KeyCode::Key_PageDown) {
  241. int items_per_page = visible_content_rect().height() / item_height();
  242. auto new_index = model.index(min(model.row_count() - 1, model.selected_index().row() + items_per_page), model.selected_index().column());
  243. if (model.is_valid(new_index)) {
  244. model.set_selected_index(new_index);
  245. scroll_into_view(new_index, Orientation::Vertical);
  246. update();
  247. }
  248. return;
  249. }
  250. return GWidget::keydown_event(event);
  251. }
  252. void GTableView::scroll_into_view(const GModelIndex& index, Orientation orientation)
  253. {
  254. auto rect = row_rect(index.row()).translated(0, -header_height());
  255. GScrollableWidget::scroll_into_view(rect, orientation);
  256. }
  257. bool GTableView::is_column_hidden(int column) const
  258. {
  259. if (column >= 0 && column < m_column_visibility.size())
  260. return !m_column_visibility[column];
  261. return false;
  262. }
  263. void GTableView::set_column_hidden(int column, bool hidden)
  264. {
  265. ASSERT(column >= 0);
  266. if (m_column_visibility.size() <= column) {
  267. int previous_column_count = m_column_visibility.size();
  268. m_column_visibility.resize(column + 1);
  269. for (int i = previous_column_count; i < m_column_visibility.size(); ++i)
  270. m_column_visibility[i] = true;
  271. }
  272. m_column_visibility[column] = !hidden;
  273. }
  274. void GTableView::doubleclick_event(GMouseEvent& event)
  275. {
  276. if (!model())
  277. return;
  278. if (event.button() == GMouseButton::Left) {
  279. mousedown_event(event);
  280. model()->activate(model()->selected_index());
  281. }
  282. }