obsqrbtz/clrsync
1
0
Fork 0
mirror of https://github.com/obsqrbtz/clrsync.git synced 2026-04-09 04:29:04 +03:00
Code Issues Packages Projects Releases Wiki Activity
Files
a200d70b14ff72c7be12dfe75223addf186ab05d
clrsync/src/core/palette/hellwal_generator.cpp

233 lines
8.5 KiB
C++
Raw Blame History

#include "hellwal_generator.hpp"
#include "core/palette/color.hpp"
#include <array>
#include <cstdio>
#include <filesystem>
#include <regex>
#include <string>
#ifdef _WIN32
#define popen _popen
#define pclose _pclose
#endif
namespace clrsync::core
{
static std::string run_command_capture_output(const std::string &cmd)
{
std::array<char, 4096> buffer;
std::string result;
FILE *pipe = popen(cmd.c_str(), "r");
if (!pipe)
return {};
while (fgets(buffer.data(), static_cast<int>(buffer.size()), pipe) != nullptr)
{
result += buffer.data();
}
int rc = pclose(pipe);
(void)rc;
return result;
}
palette hellwal_generator::generate_from_image(const std::string &image_path)
{
options default_opts{};
return generate_from_image(image_path, default_opts);
}
palette hellwal_generator::generate_from_image(const std::string &image_path, const options &opts)
{
palette pal;
std::filesystem::path p(image_path);
pal.set_name("hellwal:" + p.filename().string());
pal.set_file_path(image_path);
std::string cmd = "hellwal -i '" + image_path + "' --json";
if (opts.neon)
cmd += " --neon-mode";
if (opts.dark)
cmd += " --dark";
if (opts.light)
cmd += " --light";
if (opts.color)
cmd += " --color";
if (opts.dark_offset > 0.0f)
cmd += " --dark-offset " + std::to_string(opts.dark_offset);
if (opts.bright_offset > 0.0f)
cmd += " --bright-offset " + std::to_string(opts.bright_offset);
if (opts.invert)
cmd += " --invert";
if (opts.gray_scale > 0.0f)
cmd += " --gray-scale " + std::to_string(opts.gray_scale);
std::string out = run_command_capture_output(cmd);
if (out.empty())
return {};
std::regex special_re(
"\"(background|foreground|cursor|border)\"\\s*:\\s*\"(#[0-9A-Fa-f]{6,8})\"");
for (std::sregex_iterator it(out.begin(), out.end(), special_re), end; it != end; ++it)
{
std::smatch m = *it;
std::string key = m[1].str();
std::string hex = m[2].str();
try
{
color col;
col.from_hex_string(hex);
pal.set_color(key, col);
}
catch (...)
{
}
}
std::regex color_re("\"color(\\d{1,2})\"\\s*:\\s*\"(#[0-9A-Fa-f]{6,8})\"");
for (std::sregex_iterator it(out.begin(), out.end(), color_re), end; it != end; ++it)
{
std::smatch m = *it;
int idx = std::stoi(m[1].str());
if (idx < 0 || idx > 15)
continue;
std::string hex = m[2].str();
std::string key = "base0";
if (idx < 10)
key += std::to_string(idx);
else
key += static_cast<char>('A' + (idx - 10));
try
{
color col;
col.from_hex_string(hex);
pal.set_color(key, col);
}
catch (...)
{
}
}
auto get_color_by_index = [&](int idx) -> const color & {
std::string key = "base0";
if (idx < 10)
key += std::to_string(idx);
else
key += static_cast<char>('A' + (idx - 10));
return pal.get_color(key);
};
pal.set_color("base00", get_color_by_index(0));
pal.set_color("base01", get_color_by_index(8));
pal.set_color("base02", get_color_by_index(8));
pal.set_color("base03", get_color_by_index(8));
pal.set_color("base04", get_color_by_index(7));
pal.set_color("base05", get_color_by_index(7));
pal.set_color("base06", get_color_by_index(15));
pal.set_color("base07", get_color_by_index(15));
pal.set_color("base08", get_color_by_index(1));
pal.set_color("base09", get_color_by_index(9));
pal.set_color("base0A", get_color_by_index(3));
pal.set_color("base0B", get_color_by_index(2));
pal.set_color("base0C", get_color_by_index(6));
pal.set_color("base0D", get_color_by_index(4));
pal.set_color("base0E", get_color_by_index(5));
pal.set_color("base0F", get_color_by_index(11));
pal.set_color("accent", get_color_by_index(4));
pal.set_color("accent_secondary", get_color_by_index(6));
pal.set_color("border", get_color_by_index(8));
pal.set_color("border_focused", get_color_by_index(4));
pal.set_color("error", get_color_by_index(1));
pal.set_color("warning", get_color_by_index(3));
pal.set_color("success", get_color_by_index(2));
pal.set_color("info", get_color_by_index(4));
pal.set_color("on_error", get_color_by_index(0));
pal.set_color("on_warning", get_color_by_index(0));
pal.set_color("on_success", get_color_by_index(0));
pal.set_color("on_info", get_color_by_index(0));
pal.set_color("surface", get_color_by_index(0));
pal.set_color("surface_variant", get_color_by_index(8));
pal.set_color("on_surface", get_color_by_index(7));
pal.set_color("on_surface_variant", get_color_by_index(7));
pal.set_color("on_background", get_color_by_index(7));
// Editor - Basic
pal.set_color("editor_background", get_color_by_index(0));
pal.set_color("editor_foreground", get_color_by_index(7));
pal.set_color("editor_line_highlight", get_color_by_index(8));
pal.set_color("editor_selection", get_color_by_index(8));
pal.set_color("editor_selection_inactive", get_color_by_index(8));
pal.set_color("editor_cursor", get_color_by_index(7));
pal.set_color("editor_whitespace", get_color_by_index(8));
// Editor - Gutter
pal.set_color("editor_gutter_background", get_color_by_index(0));
pal.set_color("editor_gutter_foreground", get_color_by_index(8));
pal.set_color("editor_line_number", get_color_by_index(8));
pal.set_color("editor_line_number_active", get_color_by_index(7));
// Editor - Syntax
pal.set_color("editor_comment", get_color_by_index(8));
pal.set_color("editor_string", get_color_by_index(2));
pal.set_color("editor_number", get_color_by_index(3));
pal.set_color("editor_boolean", get_color_by_index(3));
pal.set_color("editor_keyword", get_color_by_index(5));
pal.set_color("editor_operator", get_color_by_index(7));
pal.set_color("editor_function", get_color_by_index(11));
pal.set_color("editor_variable", get_color_by_index(7));
pal.set_color("editor_parameter", get_color_by_index(4));
pal.set_color("editor_property", get_color_by_index(2));
pal.set_color("editor_constant", get_color_by_index(3));
pal.set_color("editor_type", get_color_by_index(6));
pal.set_color("editor_class", get_color_by_index(6));
pal.set_color("editor_interface", get_color_by_index(6));
pal.set_color("editor_enum", get_color_by_index(6));
pal.set_color("editor_namespace", get_color_by_index(4));
pal.set_color("editor_attribute", get_color_by_index(11));
pal.set_color("editor_decorator", get_color_by_index(11));
pal.set_color("editor_tag", get_color_by_index(1));
pal.set_color("editor_punctuation", get_color_by_index(7));
pal.set_color("editor_link", get_color_by_index(4));
pal.set_color("editor_regex", get_color_by_index(5));
pal.set_color("editor_escape_character", get_color_by_index(3));
// Editor - Diagnostics
pal.set_color("editor_invalid", get_color_by_index(1));
pal.set_color("editor_error", get_color_by_index(1));
pal.set_color("editor_error_background", get_color_by_index(0));
pal.set_color("editor_warning", get_color_by_index(3));
pal.set_color("editor_warning_background", get_color_by_index(0));
pal.set_color("editor_info", get_color_by_index(4));
pal.set_color("editor_info_background", get_color_by_index(0));
pal.set_color("editor_hint", get_color_by_index(2));
pal.set_color("editor_hint_background", get_color_by_index(0));
// Editor - UI Elements
pal.set_color("editor_active_line_border", get_color_by_index(8));
pal.set_color("editor_indent_guide", get_color_by_index(8));
pal.set_color("editor_indent_guide_active", get_color_by_index(7));
pal.set_color("editor_bracket_match", get_color_by_index(6));
pal.set_color("editor_search_match", get_color_by_index(3));
pal.set_color("editor_search_match_active", get_color_by_index(3));
pal.set_color("editor_find_range_highlight", get_color_by_index(3));
// Editor - Diff
pal.set_color("editor_deleted", get_color_by_index(1));
pal.set_color("editor_inserted", get_color_by_index(2));
pal.set_color("editor_modified", get_color_by_index(3));
pal.set_color("editor_ignored", get_color_by_index(8));
pal.set_color("editor_folded_background", get_color_by_index(8));
return pal;
}
} // namespace clrsync::core
Reference in New Issue View Git Blame Copy Permalink