From 577fc0845718ed8ad5bebf02a277c0579a817f77 Mon Sep 17 00:00:00 2001
From: Felix Morgner <felix.morgner@gmail.com>
Date: Fri, 17 May 2024 17:58:38 +0200
Subject: wanda: restructure source layout

---
 source/lib/src/system/wallpaper.cpp | 91 +++++++++++++++++++++++++++++++++++++
 1 file changed, 91 insertions(+)
 create mode 100644 source/lib/src/system/wallpaper.cpp

(limited to 'source/lib/src/system/wallpaper.cpp')

diff --git a/source/lib/src/system/wallpaper.cpp b/source/lib/src/system/wallpaper.cpp
new file mode 100644
index 0000000..11a6402
--- /dev/null
+++ b/source/lib/src/system/wallpaper.cpp
@@ -0,0 +1,91 @@
+#include "wanda/system/wallpaper.hpp"
+
+#include "wanda/std_ext/optional.hpp"
+#include "wanda/system/logging.hpp"
+#include "wanda/system/magic.hpp"
+#include "wanda/system/setting.hpp"
+
+#include <boost/gil.hpp>
+#include <boost/gil/extension/io/jpeg.hpp>
+#include <boost/gil/extension/io/png.hpp>
+#include <boost/gil/extension/numeric/resample.hpp>
+#include <boost/gil/extension/numeric/sampler.hpp>
+#include <spdlog/fmt/fmt.h>
+
+#include <algorithm>
+#include <cmath>
+#include <memory>
+
+namespace wanda::system
+{
+  namespace
+  {
+    auto magic_instance = magic{};
+
+    auto load_image(std::filesystem::path wallpaper)
+    {
+      auto image = boost::gil::rgb8_image_t{};
+
+      switch (magic_instance.type(wallpaper))
+      {
+        case magic::mime_type::image_jpeg:
+          boost::gil::read_and_convert_image(wallpaper.native(), image, boost::gil::jpeg_tag{});
+          break;
+        case magic::mime_type::image_png:
+          boost::gil::read_and_convert_image(wallpaper.native(), image, boost::gil::png_tag{});
+          break;
+      }
+
+      return image;
+    }
+
+    auto average_colors(boost::gil::rgb8_image_t image)
+    {
+      auto accumulator = boost::gil::rgb64f_pixel_t{};
+      auto view = const_view(image);
+
+      std::ranges::for_each(view, [&](auto const & source_pixel) {
+        at_c<0>(accumulator) += std::pow(boost::gil::at_c<0>(source_pixel), 2);
+        at_c<1>(accumulator) += std::pow(boost::gil::at_c<1>(source_pixel), 2);
+        at_c<2>(accumulator) += std::pow(boost::gil::at_c<2>(source_pixel), 2);
+      });
+
+      at_c<0>(accumulator) = std::sqrt(at_c<0>(accumulator) / view.size());
+      at_c<1>(accumulator) = std::sqrt(at_c<1>(accumulator) / view.size());
+      at_c<2>(accumulator) = std::sqrt(at_c<2>(accumulator) / view.size());
+
+      return accumulator;
+    }
+
+  }  // namespace
+
+  void set_wallpaper(std::filesystem::path wallpaper)
+  {
+    using namespace wanda::system::literals;
+    using namespace wanda::std_ext;
+    using namespace std::string_literals;
+
+    auto image = load_image(wallpaper);
+    auto color = average_colors(std::move(image));
+    auto hexstring = spdlog::fmt_lib::format("#{:02X}{:02X}{:02X}",
+                                             static_cast<std::uint8_t>(at_c<0>(color)),
+                                             static_cast<std::uint8_t>(at_c<1>(color)),
+                                             static_cast<std::uint8_t>(at_c<2>(color)));
+
+    with("org.gnome.desktop.background"_setting, [&](auto & setting) {
+      with(setting["primary-color"_key], [&](auto & value) {
+        value = hexstring;
+      });
+      with(setting["picture-uri"_key], [&](auto & value) {
+        value = "file://" + wallpaper.native();
+      }) ||
+          [&] { get_logger()->error("invalid settings key"); };
+      with(setting["picture-uri-dark"_key], [&](auto & value) {
+        value = "file://" + wallpaper.native();
+      }) ||
+          [&] { get_logger()->error("invalid settings key"); };
+    }) ||
+        [&] { get_logger()->error("invalid setting"); };
+  }
+
+}  // namespace wanda::system
-- 
cgit v1.2.3