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
|
#include "client/pch.hh"
#include "client/screenshot.hh"
#include "core/io/config_map.hh"
#include "core/utils/epoch.hh"
#include "client/config/keybind.hh"
#include "client/gui/chat.hh"
#include "client/gui/language.hh"
#include "client/gui/settings.hh"
#include "client/io/glfw.hh"
#include "client/globals.hh"
#include "client/toggles.hh"
static config::KeyBind screenshot_key(GLFW_KEY_F2);
static void stbi_png_physfs_callback(void* context, void* data, int size)
{
PHYSFS_writeBytes(reinterpret_cast<PHYSFS_File*>(context), data, size);
}
static void on_glfw_key(const io::GlfwKeyEvent& event)
{
if(!globals::gui_keybind_ptr && !toggles::is_sequence_await) {
if(screenshot_key.equals(event.key) && (event.action == GLFW_PRESS)) {
screenshot::take();
return;
}
}
}
void screenshot::init(void)
{
globals::client_config.add_value("screenshot.key", screenshot_key);
settings::add_keybind(0, screenshot_key, settings_location::KEYBOARD_MISC, "key.screenshot");
globals::dispatcher.sink<io::GlfwKeyEvent>().connect<&on_glfw_key>();
}
void screenshot::take(void)
{
auto stride = 3 * globals::width;
auto length = 3 * globals::width * globals::height;
auto pixels = new std::byte[length];
glBindFramebuffer(GL_FRAMEBUFFER, 0);
GLint old_pack_alignment;
glGetIntegerv(GL_PACK_ALIGNMENT, &old_pack_alignment);
// The window can be of any size, including irregular
// values such as, say 641x480, while there is a default
// alignment value of sorts that might result in a corrupted
// image; we set GL_PACK_ALIGNMENT to 1, enabling byte-alignment
glPixelStorei(GL_PACK_ALIGNMENT, 1);
glReadPixels(0, 0, globals::width, globals::height, GL_RGB, GL_UNSIGNED_BYTE, pixels);
// Restore the old pack alignment value
glPixelStorei(GL_PACK_ALIGNMENT, old_pack_alignment);
const auto directory = std::string("screenshots");
const auto filename = std::format("{}.png", utils::unix_microseconds());
const auto filepath = std::format("{}/{}", directory, filename);
PHYSFS_mkdir(directory.c_str());
if(auto file = PHYSFS_openWrite(filepath.c_str())) {
stbi_flip_vertically_on_write(true);
stbi_write_png_to_func(&stbi_png_physfs_callback, file, globals::width, globals::height, 3, pixels, stride);
spdlog::info("screenshot: wrote {}", filepath);
gui::client_chat::print(std::format("{} {}", gui::language::resolve("chat.screenshot_message"), filename));
PHYSFS_close(file);
}
delete[] pixels;
}
|
