PDFViewerWidget.cpp 8.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235
  1. /*
  2. * Copyright (c) 2021-2022, Matthew Olsson <mattco@serenityos.org>
  3. * Copyright (c) 2021, Mustafa Quraish <mustafa@serenityos.org>
  4. *
  5. * SPDX-License-Identifier: BSD-2-Clause
  6. */
  7. #include "PDFViewerWidget.h"
  8. #include <LibCore/File.h>
  9. #include <LibFileSystemAccessClient/Client.h>
  10. #include <LibGUI/Application.h>
  11. #include <LibGUI/BoxLayout.h>
  12. #include <LibGUI/FilePicker.h>
  13. #include <LibGUI/Label.h>
  14. #include <LibGUI/Menu.h>
  15. #include <LibGUI/Menubar.h>
  16. #include <LibGUI/MessageBox.h>
  17. #include <LibGUI/Splitter.h>
  18. #include <LibGUI/Toolbar.h>
  19. #include <LibGUI/ToolbarContainer.h>
  20. PDFViewerWidget::PDFViewerWidget()
  21. {
  22. set_fill_with_background_color(true);
  23. set_layout<GUI::VerticalBoxLayout>();
  24. auto& toolbar_container = add<GUI::ToolbarContainer>();
  25. auto& toolbar = toolbar_container.add<GUI::Toolbar>();
  26. auto& splitter = add<GUI::HorizontalSplitter>();
  27. splitter.layout()->set_spacing(4);
  28. m_sidebar = splitter.add<SidebarWidget>();
  29. m_sidebar->set_preferred_width(200);
  30. m_sidebar->set_visible(false);
  31. m_viewer = splitter.add<PDFViewer>();
  32. m_viewer->on_page_change = [&](auto new_page) {
  33. m_page_text_box->set_current_number(new_page + 1, GUI::AllowCallback::No);
  34. };
  35. initialize_toolbar(toolbar);
  36. }
  37. void PDFViewerWidget::initialize_menubar(GUI::Window& window)
  38. {
  39. auto& file_menu = window.add_menu("&File");
  40. file_menu.add_action(GUI::CommonActions::make_open_action([&](auto&) {
  41. auto response = FileSystemAccessClient::Client::the().try_open_file(&window);
  42. if (!response.is_error())
  43. open_file(*response.value());
  44. }));
  45. file_menu.add_separator();
  46. file_menu.add_action(GUI::CommonActions::make_quit_action([](auto&) {
  47. GUI::Application::the()->quit();
  48. }));
  49. auto& view_menu = window.add_menu("&View");
  50. view_menu.add_action(*m_toggle_sidebar_action);
  51. view_menu.add_separator();
  52. auto& view_mode_menu = view_menu.add_submenu("View &Mode");
  53. view_mode_menu.add_action(*m_page_view_mode_single);
  54. view_mode_menu.add_action(*m_page_view_mode_multiple);
  55. view_menu.add_separator();
  56. view_menu.add_action(*m_zoom_in_action);
  57. view_menu.add_action(*m_zoom_out_action);
  58. view_menu.add_action(*m_reset_zoom_action);
  59. auto& help_menu = window.add_menu("&Help");
  60. help_menu.add_action(GUI::CommonActions::make_command_palette_action(&window));
  61. help_menu.add_action(GUI::CommonActions::make_about_action("PDF Viewer", GUI::Icon::default_icon("app-pdf-viewer"sv), &window));
  62. }
  63. void PDFViewerWidget::initialize_toolbar(GUI::Toolbar& toolbar)
  64. {
  65. auto open_outline_action = GUI::Action::create(
  66. "Toggle &Sidebar", { Mod_Ctrl, Key_S }, Gfx::Bitmap::try_load_from_file("/res/icons/16x16/sidebar.png"sv).release_value_but_fixme_should_propagate_errors(), [&](auto&) {
  67. m_sidebar_open = !m_sidebar_open;
  68. m_sidebar->set_visible(m_sidebar_open ? true : false);
  69. },
  70. nullptr);
  71. open_outline_action->set_enabled(false);
  72. m_toggle_sidebar_action = open_outline_action;
  73. toolbar.add_action(*open_outline_action);
  74. toolbar.add_separator();
  75. m_go_to_prev_page_action = GUI::Action::create("Go to &Previous Page", Gfx::Bitmap::try_load_from_file("/res/icons/16x16/go-up.png"sv).release_value_but_fixme_should_propagate_errors(), [&](auto&) {
  76. VERIFY(m_viewer->current_page() > 0);
  77. m_page_text_box->set_current_number(m_viewer->current_page());
  78. });
  79. m_go_to_prev_page_action->set_enabled(false);
  80. m_go_to_next_page_action = GUI::Action::create("Go to &Next Page", Gfx::Bitmap::try_load_from_file("/res/icons/16x16/go-down.png"sv).release_value_but_fixme_should_propagate_errors(), [&](auto&) {
  81. VERIFY(m_viewer->current_page() < m_viewer->document()->get_page_count() - 1);
  82. m_page_text_box->set_current_number(m_viewer->current_page() + 2);
  83. });
  84. m_go_to_next_page_action->set_enabled(false);
  85. toolbar.add_action(*m_go_to_prev_page_action);
  86. toolbar.add_action(*m_go_to_next_page_action);
  87. m_page_text_box = toolbar.add<NumericInput>();
  88. m_page_text_box->set_enabled(false);
  89. m_page_text_box->set_fixed_width(30);
  90. m_page_text_box->set_min_number(1);
  91. m_page_text_box->on_number_changed = [&](i32 number) {
  92. auto page_count = m_viewer->document()->get_page_count();
  93. auto new_page_number = static_cast<u32>(number);
  94. VERIFY(new_page_number >= 1 && new_page_number <= page_count);
  95. m_viewer->set_current_page(new_page_number - 1);
  96. m_go_to_prev_page_action->set_enabled(new_page_number > 1);
  97. m_go_to_next_page_action->set_enabled(new_page_number < page_count);
  98. };
  99. m_total_page_label = toolbar.add<GUI::Label>();
  100. m_total_page_label->set_autosize(true, 5);
  101. toolbar.add_separator();
  102. m_zoom_in_action = GUI::CommonActions::make_zoom_in_action([&](auto&) {
  103. m_viewer->zoom_in();
  104. });
  105. m_zoom_out_action = GUI::CommonActions::make_zoom_out_action([&](auto&) {
  106. m_viewer->zoom_out();
  107. });
  108. m_reset_zoom_action = GUI::CommonActions::make_reset_zoom_action([&](auto&) {
  109. m_viewer->reset_zoom();
  110. });
  111. m_rotate_counterclockwise_action = GUI::CommonActions::make_rotate_counterclockwise_action([&](auto&) {
  112. m_viewer->rotate(-90);
  113. });
  114. m_rotate_clockwise_action = GUI::CommonActions::make_rotate_clockwise_action([&](auto&) {
  115. m_viewer->rotate(90);
  116. });
  117. m_zoom_in_action->set_enabled(false);
  118. m_zoom_out_action->set_enabled(false);
  119. m_reset_zoom_action->set_enabled(false);
  120. m_rotate_counterclockwise_action->set_enabled(false);
  121. m_rotate_clockwise_action->set_enabled(false);
  122. m_page_view_mode_single = GUI::Action::create_checkable("Single", [&](auto&) {
  123. m_viewer->set_page_view_mode(PDFViewer::PageViewMode::Single);
  124. });
  125. m_page_view_mode_multiple = GUI::Action::create_checkable("Multiple", [&](auto&) {
  126. m_viewer->set_page_view_mode(PDFViewer::PageViewMode::Multiple);
  127. });
  128. if (m_viewer->page_view_mode() == PDFViewer::PageViewMode::Single) {
  129. m_page_view_mode_single->set_checked(true);
  130. } else {
  131. m_page_view_mode_multiple->set_checked(true);
  132. }
  133. m_page_view_action_group.add_action(*m_page_view_mode_single);
  134. m_page_view_action_group.add_action(*m_page_view_mode_multiple);
  135. m_page_view_action_group.set_exclusive(true);
  136. toolbar.add_action(*m_zoom_in_action);
  137. toolbar.add_action(*m_zoom_out_action);
  138. toolbar.add_action(*m_reset_zoom_action);
  139. toolbar.add_action(*m_rotate_counterclockwise_action);
  140. toolbar.add_action(*m_rotate_clockwise_action);
  141. toolbar.add_separator();
  142. m_show_clipping_paths = toolbar.add<GUI::CheckBox>();
  143. m_show_clipping_paths->set_text("Show clipping paths");
  144. m_show_clipping_paths->set_checked(m_viewer->show_clipping_paths(), GUI::AllowCallback::No);
  145. m_show_clipping_paths->on_checked = [&](auto checked) { m_viewer->set_show_clipping_paths(checked); };
  146. }
  147. void PDFViewerWidget::open_file(Core::File& file)
  148. {
  149. window()->set_title(String::formatted("{} - PDF Viewer", file.filename()));
  150. auto handle_error = [&]<typename T>(PDF::PDFErrorOr<T> maybe_error) {
  151. if (maybe_error.is_error()) {
  152. auto error = maybe_error.release_error();
  153. warnln("{}", error.message());
  154. GUI::MessageBox::show_error(nullptr, "Failed to load the document."sv);
  155. return true;
  156. }
  157. return false;
  158. };
  159. m_buffer = file.read_all();
  160. auto maybe_document = PDF::Document::create(m_buffer);
  161. if (handle_error(maybe_document))
  162. return;
  163. auto document = maybe_document.release_value();
  164. if (auto sh = document->security_handler(); sh && !sh->has_user_password()) {
  165. // FIXME: Prompt the user for a password
  166. VERIFY_NOT_REACHED();
  167. }
  168. if (handle_error(document->initialize()))
  169. return;
  170. if (handle_error(m_viewer->set_document(document)))
  171. return;
  172. m_total_page_label->set_text(String::formatted("of {}", document->get_page_count()));
  173. m_page_text_box->set_enabled(true);
  174. m_page_text_box->set_current_number(1, GUI::AllowCallback::No);
  175. m_page_text_box->set_max_number(document->get_page_count());
  176. m_go_to_prev_page_action->set_enabled(false);
  177. m_go_to_next_page_action->set_enabled(document->get_page_count() > 1);
  178. m_toggle_sidebar_action->set_enabled(true);
  179. m_zoom_in_action->set_enabled(true);
  180. m_zoom_out_action->set_enabled(true);
  181. m_reset_zoom_action->set_enabled(true);
  182. m_rotate_counterclockwise_action->set_enabled(true);
  183. m_rotate_clockwise_action->set_enabled(true);
  184. m_show_clipping_paths->set_checked(m_viewer->show_clipping_paths(), GUI::AllowCallback::No);
  185. if (document->outline()) {
  186. auto outline = document->outline();
  187. m_sidebar->set_outline(outline.release_nonnull());
  188. m_sidebar->set_visible(true);
  189. m_sidebar_open = true;
  190. } else {
  191. m_sidebar->set_outline({});
  192. m_sidebar->set_visible(false);
  193. m_sidebar_open = false;
  194. }
  195. }