1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
|
// Display images inside a terminal
// Copyright (C) 2023 JustKidding
//
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with this program. If not, see <https://www.gnu.org/licenses/>.
#include "chafa.hpp"
#include "dimensions.hpp"
#include "terminal.hpp"
#include "util.hpp"
#include "util/ptr.hpp"
#include <algorithm>
#include <cmath>
#include <iostream>
#include <range/v3/all.hpp>
#include <spdlog/spdlog.h>
void gstring_delete(GString *str)
{
g_string_free(str, true);
}
Chafa::Chafa(std::unique_ptr<Image> new_image, std::mutex *stdout_mutex)
: symbol_map(chafa_symbol_map_new()),
config(chafa_canvas_config_new()),
image(std::move(new_image)),
stdout_mutex(stdout_mutex)
{
const auto envp = c_unique_ptr<gchar *, g_strfreev>{g_get_environ()};
term_info = chafa_term_db_detect(chafa_term_db_get_default(), envp.get());
const auto dims = image->dimensions();
x = dims.x + 1;
y = dims.y + 1;
horizontal_cells = std::ceil(static_cast<double>(image->width()) / dims.terminal->font_width);
vertical_cells = std::ceil(static_cast<double>(image->height()) / dims.terminal->font_height);
chafa_symbol_map_add_by_tags(symbol_map, CHAFA_SYMBOL_TAG_BLOCK);
chafa_symbol_map_add_by_tags(symbol_map, CHAFA_SYMBOL_TAG_BORDER);
chafa_symbol_map_add_by_tags(symbol_map, CHAFA_SYMBOL_TAG_SPACE);
chafa_symbol_map_remove_by_tags(symbol_map, CHAFA_SYMBOL_TAG_WIDE);
chafa_canvas_config_set_symbol_map(config, symbol_map);
chafa_canvas_config_set_pixel_mode(config, CHAFA_PIXEL_MODE_SYMBOLS);
chafa_canvas_config_set_geometry(config, horizontal_cells, vertical_cells);
}
Chafa::~Chafa()
{
chafa_canvas_unref(canvas);
chafa_canvas_config_unref(config);
chafa_symbol_map_unref(symbol_map);
chafa_term_info_unref(term_info);
const std::scoped_lock lock{*stdout_mutex};
util::clear_terminal_area(x, y, horizontal_cells, vertical_cells);
}
void Chafa::draw()
{
canvas = chafa_canvas_new(config);
chafa_canvas_draw_all_pixels(canvas, CHAFA_PIXEL_BGRA8_UNASSOCIATED, image->data(), image->width(), image->height(),
image->width() * 4);
#ifdef CHAFA_VERSION_1_14
GString **lines = nullptr;
gint lines_length = 0;
chafa_canvas_print_rows(canvas, term_info, &lines, &lines_length);
auto ycoord = y;
const std::scoped_lock lock{*stdout_mutex};
util::save_cursor_position();
for (int i = 0; i < lines_length; ++i) {
const auto line = c_unique_ptr<GString, gstring_delete>{lines[i]};
util::move_cursor(ycoord++, x);
std::cout << line->str;
}
g_free(lines);
#else
const auto result = c_unique_ptr<GString, gstring_delete>{chafa_canvas_print(canvas, term_info)};
auto ycoord = y;
const auto lines = util::str_split(result->str, "\n");
const std::scoped_lock lock{*stdout_mutex};
util::save_cursor_position();
ranges::for_each(lines, [this, &ycoord](const std::string &line) {
util::move_cursor(ycoord++, x);
std::cout << line;
});
#endif
std::cout << std::flush;
util::restore_cursor_position();
}
|
