FilterTreeModel.cpp 2.7 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465
  1. /*
  2. * Copyright (c) 2021-2022, Tobias Christiansen <tobyase@serenityos.org>
  3. * Copyright (c) 2021, Mustafa Quraish <mustafa@serenityos.org>
  4. * Copyright (c) 2022, Linus Groh <linusg@serenityos.org>
  5. *
  6. * SPDX-License-Identifier: BSD-2-Clause
  7. */
  8. #include "FilterTreeModel.h"
  9. #include "Filters/Bloom.h"
  10. #include "Filters/BoxBlur3.h"
  11. #include "Filters/BoxBlur5.h"
  12. #include "Filters/FastBoxBlur.h"
  13. #include "Filters/GaussBlur3.h"
  14. #include "Filters/GaussBlur5.h"
  15. #include "Filters/Grayscale.h"
  16. #include "Filters/Invert.h"
  17. #include "Filters/LaplaceCardinal.h"
  18. #include "Filters/LaplaceDiagonal.h"
  19. #include "Filters/Median.h"
  20. #include "Filters/Sepia.h"
  21. #include "Filters/Sharpen.h"
  22. #include <LibGUI/FileIconProvider.h>
  23. namespace PixelPaint {
  24. ErrorOr<NonnullRefPtr<GUI::TreeViewModel>> create_filter_tree_model(ImageEditor* editor)
  25. {
  26. auto directory_icon = GUI::FileIconProvider::directory_icon();
  27. auto filter_icon = GUI::Icon(TRY(Gfx::Bitmap::try_load_from_file("/res/icons/pixelpaint/filter.png"sv)));
  28. auto filter_tree_model = GUI::TreeViewModel::create();
  29. auto add_filter_node = [&]<typename FilterType>(GUI::TreeViewModel::Node& node) {
  30. auto filter = adopt_own(*new FilterType(editor));
  31. (void)node.add_node<FilterNode>(filter->filter_name(), filter_icon, move(filter));
  32. };
  33. auto artistic_category = filter_tree_model->add_node("Artistic", directory_icon);
  34. add_filter_node.template operator()<Filters::Bloom>(artistic_category);
  35. auto spatial_category = filter_tree_model->add_node("Spatial", directory_icon);
  36. auto edge_detect_category = spatial_category->add_node("Edge Detection", directory_icon);
  37. add_filter_node.template operator()<Filters::LaplaceCardinal>(edge_detect_category);
  38. add_filter_node.template operator()<Filters::LaplaceDiagonal>(edge_detect_category);
  39. auto blur_category = spatial_category->add_node("Blur & Sharpen", directory_icon);
  40. add_filter_node.template operator()<Filters::FastBoxBlur>(blur_category);
  41. add_filter_node.template operator()<Filters::GaussBlur3>(blur_category);
  42. add_filter_node.template operator()<Filters::GaussBlur5>(blur_category);
  43. add_filter_node.template operator()<Filters::BoxBlur3>(blur_category);
  44. add_filter_node.template operator()<Filters::BoxBlur5>(blur_category);
  45. add_filter_node.template operator()<Filters::Sharpen>(blur_category);
  46. add_filter_node.template operator()<Filters::Median>(blur_category);
  47. auto color_category = filter_tree_model->add_node("Color", directory_icon);
  48. add_filter_node.template operator()<Filters::Grayscale>(color_category);
  49. add_filter_node.template operator()<Filters::Invert>(color_category);
  50. add_filter_node.template operator()<Filters::Sepia>(color_category);
  51. return filter_tree_model;
  52. }
  53. }