GItemView.cpp 12 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363
  1. #include <AK/StringBuilder.h>
  2. #include <Kernel/KeyCode.h>
  3. #include <LibDraw/Palette.h>
  4. #include <LibGUI/GDragOperation.h>
  5. #include <LibGUI/GItemView.h>
  6. #include <LibGUI/GModel.h>
  7. #include <LibGUI/GPainter.h>
  8. #include <LibGUI/GScrollBar.h>
  9. GItemView::GItemView(GWidget* parent)
  10. : GAbstractView(parent)
  11. {
  12. set_frame_shape(FrameShape::Container);
  13. set_frame_shadow(FrameShadow::Sunken);
  14. set_frame_thickness(2);
  15. horizontal_scrollbar().set_visible(false);
  16. }
  17. GItemView::~GItemView()
  18. {
  19. }
  20. void GItemView::scroll_into_view(const GModelIndex& index, Orientation orientation)
  21. {
  22. GScrollableWidget::scroll_into_view(item_rect(index.row()), orientation);
  23. }
  24. void GItemView::resize_event(GResizeEvent& event)
  25. {
  26. GAbstractView::resize_event(event);
  27. update_content_size();
  28. }
  29. void GItemView::did_update_model()
  30. {
  31. GAbstractView::did_update_model();
  32. update_content_size();
  33. update();
  34. }
  35. void GItemView::update_content_size()
  36. {
  37. if (!model())
  38. return set_content_size({});
  39. m_visual_column_count = available_size().width() / effective_item_size().width();
  40. if (m_visual_column_count)
  41. m_visual_row_count = ceil_div(model()->row_count(), m_visual_column_count);
  42. else
  43. m_visual_row_count = 0;
  44. int content_width = available_size().width();
  45. int content_height = m_visual_row_count * effective_item_size().height();
  46. set_content_size({ content_width, content_height });
  47. }
  48. Rect GItemView::item_rect(int item_index) const
  49. {
  50. if (!m_visual_row_count || !m_visual_column_count)
  51. return {};
  52. int visual_row_index = item_index / m_visual_column_count;
  53. int visual_column_index = item_index % m_visual_column_count;
  54. return {
  55. visual_column_index * effective_item_size().width(),
  56. visual_row_index * effective_item_size().height(),
  57. effective_item_size().width(),
  58. effective_item_size().height()
  59. };
  60. }
  61. int GItemView::item_at_event_position(const Point& position) const
  62. {
  63. // FIXME: Since all items are the same size, just compute the clicked item index
  64. // instead of iterating over everything.
  65. auto adjusted_position = position.translated(0, vertical_scrollbar().value());
  66. for (int i = 0; i < item_count(); ++i) {
  67. if (item_rect(i).contains(adjusted_position))
  68. return i;
  69. }
  70. return -1;
  71. }
  72. void GItemView::mousedown_event(GMouseEvent& event)
  73. {
  74. int item_index = item_at_event_position(event.position());
  75. if (event.button() == GMouseButton::Left) {
  76. m_left_mousedown_position = event.position();
  77. if (item_index == -1) {
  78. selection().clear();
  79. } else {
  80. auto index = model()->index(item_index, m_model_column);
  81. if (event.modifiers() & Mod_Ctrl)
  82. selection().toggle(index);
  83. else
  84. selection().set(index);
  85. }
  86. }
  87. GAbstractView::mousedown_event(event);
  88. }
  89. void GItemView::mousemove_event(GMouseEvent& event)
  90. {
  91. if (!model())
  92. return GAbstractView::mousemove_event(event);
  93. if (event.buttons() & GMouseButton::Left && !selection().is_empty()) {
  94. auto diff = event.position() - m_left_mousedown_position;
  95. auto distance_travelled_squared = diff.x() * diff.x() + diff.y() * diff.y();
  96. constexpr int drag_distance_threshold = 5;
  97. if (distance_travelled_squared > (drag_distance_threshold)) {
  98. dbg() << "Initiate drag!";
  99. auto drag_operation = GDragOperation::construct();
  100. RefPtr<GraphicsBitmap> bitmap;
  101. StringBuilder text_builder;
  102. StringBuilder data_builder;
  103. selection().for_each_index([&](auto& index) {
  104. auto text_data = model()->data(index);
  105. text_builder.append(text_data.to_string());
  106. text_builder.append(" ");
  107. auto drag_data = model()->data(index, GModel::Role::DragData);
  108. data_builder.append(drag_data.to_string());
  109. data_builder.append('\n');
  110. if (!bitmap) {
  111. GVariant icon_data = model()->data(index, GModel::Role::Icon);
  112. if (icon_data.is_icon())
  113. bitmap = icon_data.as_icon().bitmap_for_size(32);
  114. }
  115. });
  116. drag_operation->set_text(text_builder.to_string());
  117. drag_operation->set_bitmap(bitmap);
  118. drag_operation->set_data("url-list", data_builder.to_string());
  119. auto outcome = drag_operation->exec();
  120. switch (outcome) {
  121. case GDragOperation::Outcome::Accepted:
  122. dbg() << "Drag was accepted!";
  123. break;
  124. case GDragOperation::Outcome::Cancelled:
  125. dbg() << "Drag was cancelled!";
  126. break;
  127. default:
  128. ASSERT_NOT_REACHED();
  129. break;
  130. }
  131. }
  132. }
  133. GAbstractView::mousemove_event(event);
  134. }
  135. void GItemView::context_menu_event(GContextMenuEvent& event)
  136. {
  137. if (!model())
  138. return;
  139. auto item_index = item_at_event_position(event.position());
  140. GModelIndex index;
  141. if (item_index != -1) {
  142. index = model()->index(item_index, m_model_column);
  143. if (!selection().contains(index))
  144. selection().set(index);
  145. } else {
  146. selection().clear();
  147. }
  148. if (on_context_menu_request)
  149. on_context_menu_request(index, event);
  150. GAbstractView::context_menu_event(event);
  151. }
  152. void GItemView::doubleclick_event(GMouseEvent& event)
  153. {
  154. if (!model())
  155. return;
  156. if (event.button() == GMouseButton::Left) {
  157. mousedown_event(event);
  158. selection().for_each_index([this](auto& index) {
  159. activate(index);
  160. });
  161. }
  162. }
  163. void GItemView::paint_event(GPaintEvent& event)
  164. {
  165. Color widget_background_color = palette().color(background_role());
  166. GFrame::paint_event(event);
  167. GPainter painter(*this);
  168. painter.add_clip_rect(widget_inner_rect());
  169. painter.add_clip_rect(event.rect());
  170. painter.fill_rect(event.rect(), widget_background_color);
  171. painter.translate(-horizontal_scrollbar().value(), -vertical_scrollbar().value());
  172. auto column_metadata = model()->column_metadata(m_model_column);
  173. const Font& font = column_metadata.font ? *column_metadata.font : this->font();
  174. for (int item_index = 0; item_index < model()->row_count(); ++item_index) {
  175. bool is_selected_item = selection().contains(model()->index(item_index, m_model_column));
  176. Color background_color;
  177. if (is_selected_item) {
  178. background_color = is_focused() ? palette().selection() : Color::from_rgb(0x606060);
  179. } else {
  180. background_color = widget_background_color;
  181. }
  182. Rect item_rect = this->item_rect(item_index);
  183. auto model_index = model()->index(item_index, m_model_column);
  184. auto icon = model()->data(model_index, GModel::Role::Icon);
  185. auto item_text = model()->data(model_index, GModel::Role::Display);
  186. Rect icon_rect = { 0, 0, 32, 32 };
  187. icon_rect.center_within(item_rect);
  188. icon_rect.move_by(0, -font.glyph_height() - 6);
  189. if (icon.is_icon()) {
  190. if (auto bitmap = icon.as_icon().bitmap_for_size(icon_rect.width()))
  191. painter.draw_scaled_bitmap(icon_rect, *bitmap, bitmap->rect());
  192. }
  193. Rect text_rect { 0, icon_rect.bottom() + 6 + 1, font.width(item_text.to_string()), font.glyph_height() };
  194. text_rect.center_horizontally_within(item_rect);
  195. text_rect.inflate(6, 4);
  196. text_rect.intersect(item_rect);
  197. Color text_color;
  198. if (is_selected_item)
  199. text_color = palette().selection_text();
  200. else
  201. text_color = model()->data(model_index, GModel::Role::ForegroundColor).to_color(palette().color(foreground_role()));
  202. painter.fill_rect(text_rect, background_color);
  203. painter.draw_text(text_rect, item_text.to_string(), font, TextAlignment::Center, text_color, TextElision::Right);
  204. };
  205. }
  206. int GItemView::item_count() const
  207. {
  208. if (!model())
  209. return 0;
  210. return model()->row_count();
  211. }
  212. void GItemView::keydown_event(GKeyEvent& event)
  213. {
  214. if (!model())
  215. return;
  216. if (!m_visual_row_count || !m_visual_column_count)
  217. return;
  218. auto& model = *this->model();
  219. if (event.key() == KeyCode::Key_Return) {
  220. selection().for_each_index([this](auto& index) {
  221. activate(index);
  222. });
  223. return;
  224. }
  225. if (event.key() == KeyCode::Key_Home) {
  226. auto new_index = model.index(0, 0);
  227. if (model.is_valid(new_index)) {
  228. selection().set(new_index);
  229. scroll_into_view(new_index, Orientation::Vertical);
  230. update();
  231. }
  232. return;
  233. }
  234. if (event.key() == KeyCode::Key_End) {
  235. auto new_index = model.index(model.row_count() - 1, 0);
  236. if (model.is_valid(new_index)) {
  237. selection().set(new_index);
  238. scroll_into_view(new_index, Orientation::Vertical);
  239. update();
  240. }
  241. return;
  242. }
  243. if (event.key() == KeyCode::Key_Up) {
  244. GModelIndex new_index;
  245. if (!selection().is_empty()) {
  246. auto old_index = selection().first();
  247. new_index = model.index(old_index.row() - m_visual_column_count, old_index.column());
  248. } else {
  249. new_index = model.index(0, 0);
  250. }
  251. if (model.is_valid(new_index)) {
  252. selection().set(new_index);
  253. scroll_into_view(new_index, Orientation::Vertical);
  254. update();
  255. }
  256. return;
  257. }
  258. if (event.key() == KeyCode::Key_Down) {
  259. GModelIndex new_index;
  260. if (!selection().is_empty()) {
  261. auto old_index = selection().first();
  262. new_index = model.index(old_index.row() + m_visual_column_count, old_index.column());
  263. } else {
  264. new_index = model.index(0, 0);
  265. }
  266. if (model.is_valid(new_index)) {
  267. selection().set(new_index);
  268. scroll_into_view(new_index, Orientation::Vertical);
  269. update();
  270. }
  271. return;
  272. }
  273. if (event.key() == KeyCode::Key_Left) {
  274. GModelIndex new_index;
  275. if (!selection().is_empty()) {
  276. auto old_index = selection().first();
  277. new_index = model.index(old_index.row() - 1, old_index.column());
  278. } else {
  279. new_index = model.index(0, 0);
  280. }
  281. if (model.is_valid(new_index)) {
  282. selection().set(new_index);
  283. scroll_into_view(new_index, Orientation::Vertical);
  284. update();
  285. }
  286. return;
  287. }
  288. if (event.key() == KeyCode::Key_Right) {
  289. GModelIndex new_index;
  290. if (!selection().is_empty()) {
  291. auto old_index = selection().first();
  292. new_index = model.index(old_index.row() + 1, old_index.column());
  293. } else {
  294. new_index = model.index(0, 0);
  295. }
  296. if (model.is_valid(new_index)) {
  297. selection().set(new_index);
  298. scroll_into_view(new_index, Orientation::Vertical);
  299. update();
  300. }
  301. return;
  302. }
  303. if (event.key() == KeyCode::Key_PageUp) {
  304. int items_per_page = (visible_content_rect().height() / effective_item_size().height()) * m_visual_column_count;
  305. auto old_index = selection().first();
  306. auto new_index = model.index(max(0, old_index.row() - items_per_page), old_index.column());
  307. if (model.is_valid(new_index)) {
  308. selection().set(new_index);
  309. scroll_into_view(new_index, Orientation::Vertical);
  310. update();
  311. }
  312. return;
  313. }
  314. if (event.key() == KeyCode::Key_PageDown) {
  315. int items_per_page = (visible_content_rect().height() / effective_item_size().height()) * m_visual_column_count;
  316. auto old_index = selection().first();
  317. auto new_index = model.index(min(model.row_count() - 1, old_index.row() + items_per_page), old_index.column());
  318. if (model.is_valid(new_index)) {
  319. selection().set(new_index);
  320. scroll_into_view(new_index, Orientation::Vertical);
  321. update();
  322. }
  323. return;
  324. }
  325. return GWidget::keydown_event(event);
  326. }