LineTool.cpp 5.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163
  1. /*
  2. * Copyright (c) 2018-2020, Andreas Kling <kling@serenityos.org>
  3. * Copyright (c) 2021, Mustafa Quraish <mustafa@serenityos.org>
  4. * Copyright (c) 2022, MacDue <macdue@dueutil.tech>
  5. * Copyright (c) 2022, the SerenityOS developers.
  6. *
  7. * SPDX-License-Identifier: BSD-2-Clause
  8. */
  9. #include "LineTool.h"
  10. #include "../ImageEditor.h"
  11. #include "../Layer.h"
  12. #include <AK/Math.h>
  13. #include <LibGUI/Action.h>
  14. #include <LibGUI/BoxLayout.h>
  15. #include <LibGUI/CheckBox.h>
  16. #include <LibGUI/Label.h>
  17. #include <LibGUI/Menu.h>
  18. #include <LibGUI/Painter.h>
  19. #include <LibGUI/ValueSlider.h>
  20. #include <LibGfx/AntiAliasingPainter.h>
  21. namespace PixelPaint {
  22. void LineTool::on_mousedown(Layer* layer, MouseEvent& event)
  23. {
  24. if (!layer)
  25. return;
  26. auto& layer_event = event.layer_event();
  27. if (layer_event.button() != GUI::MouseButton::Primary && layer_event.button() != GUI::MouseButton::Secondary)
  28. return;
  29. if (m_drawing_button != GUI::MouseButton::None)
  30. return;
  31. m_drawing_button = layer_event.button();
  32. m_drag_start_position = layer_event.position();
  33. m_line_start_position = layer_event.position();
  34. m_line_end_position = layer_event.position();
  35. m_editor->update();
  36. }
  37. void LineTool::draw_using(GUI::Painter& painter, Gfx::IntPoint start_position, Gfx::IntPoint end_position, Color color, int thickness)
  38. {
  39. if (m_antialias_enabled) {
  40. Gfx::AntiAliasingPainter aa_painter { painter };
  41. auto as_float_point = [](auto const& point) {
  42. return Gfx::FloatPoint { point.x(), point.y() };
  43. };
  44. aa_painter.draw_line(as_float_point(start_position), as_float_point(end_position), color, thickness);
  45. } else {
  46. painter.draw_line(start_position, end_position, color, thickness);
  47. }
  48. }
  49. void LineTool::on_mouseup(Layer* layer, MouseEvent& event)
  50. {
  51. if (!layer)
  52. return;
  53. auto& layer_event = event.layer_event();
  54. if (layer_event.button() == m_drawing_button) {
  55. GUI::Painter painter(layer->get_scratch_edited_bitmap());
  56. draw_using(painter, m_line_start_position, m_line_end_position, m_editor->color_for(m_drawing_button), m_thickness);
  57. m_drawing_button = GUI::MouseButton::None;
  58. auto modified_rect = Gfx::IntRect::from_two_points(m_line_start_position, m_line_end_position).inflated(m_thickness * 2, m_thickness * 2);
  59. layer->did_modify_bitmap(modified_rect);
  60. m_editor->update();
  61. m_editor->did_complete_action(tool_name());
  62. }
  63. }
  64. void LineTool::on_mousemove(Layer* layer, MouseEvent& event)
  65. {
  66. if (!layer)
  67. return;
  68. auto& layer_event = event.layer_event();
  69. if (m_drawing_button == GUI::MouseButton::None)
  70. return;
  71. if (layer_event.shift())
  72. m_line_end_position = constrain_line_angle(m_drag_start_position, layer_event.position());
  73. else
  74. m_line_end_position = layer_event.position();
  75. if (layer_event.alt()) {
  76. m_line_start_position = m_drag_start_position + (m_drag_start_position - m_line_end_position);
  77. } else {
  78. m_line_start_position = m_drag_start_position;
  79. }
  80. m_editor->update();
  81. }
  82. void LineTool::on_second_paint(Layer const* layer, GUI::PaintEvent& event)
  83. {
  84. if (!layer || m_drawing_button == GUI::MouseButton::None)
  85. return;
  86. GUI::Painter painter(*m_editor);
  87. painter.add_clip_rect(event.rect());
  88. painter.translate(editor_layer_location(*layer));
  89. auto preview_start = editor_stroke_position(m_line_start_position, m_thickness);
  90. auto preview_end = editor_stroke_position(m_line_end_position, m_thickness);
  91. draw_using(painter, preview_start, preview_end, m_editor->color_for(m_drawing_button), AK::max(m_thickness * m_editor->scale(), 1));
  92. }
  93. bool LineTool::on_keydown(GUI::KeyEvent& event)
  94. {
  95. if (event.key() == Key_Escape && m_drawing_button != GUI::MouseButton::None) {
  96. m_drawing_button = GUI::MouseButton::None;
  97. m_editor->update();
  98. return true;
  99. }
  100. return Tool::on_keydown(event);
  101. }
  102. GUI::Widget* LineTool::get_properties_widget()
  103. {
  104. if (!m_properties_widget) {
  105. m_properties_widget = GUI::Widget::construct();
  106. m_properties_widget->set_layout<GUI::VerticalBoxLayout>();
  107. auto& thickness_container = m_properties_widget->add<GUI::Widget>();
  108. thickness_container.set_fixed_height(20);
  109. thickness_container.set_layout<GUI::HorizontalBoxLayout>();
  110. auto& thickness_label = thickness_container.add<GUI::Label>("Thickness:");
  111. thickness_label.set_text_alignment(Gfx::TextAlignment::CenterLeft);
  112. thickness_label.set_fixed_size(80, 20);
  113. auto& thickness_slider = thickness_container.add<GUI::ValueSlider>(Orientation::Horizontal, "px");
  114. thickness_slider.set_range(1, 10);
  115. thickness_slider.set_value(m_thickness);
  116. thickness_slider.on_change = [&](int value) {
  117. m_thickness = value;
  118. };
  119. set_primary_slider(&thickness_slider);
  120. auto& mode_container = m_properties_widget->add<GUI::Widget>();
  121. mode_container.set_fixed_height(20);
  122. mode_container.set_layout<GUI::HorizontalBoxLayout>();
  123. auto& mode_label = mode_container.add<GUI::Label>("Mode:");
  124. mode_label.set_text_alignment(Gfx::TextAlignment::CenterLeft);
  125. mode_label.set_fixed_size(80, 20);
  126. auto& aa_enable_checkbox = mode_container.add<GUI::CheckBox>("Anti-alias");
  127. aa_enable_checkbox.on_checked = [&](bool checked) {
  128. m_antialias_enabled = checked;
  129. };
  130. aa_enable_checkbox.set_checked(true);
  131. }
  132. return m_properties_widget.ptr();
  133. }
  134. }