Stowage/godot
2
0
Fork 0
mirror of https://github.com/godotengine/godot.git synced 2025-11-07 17:11:30 +00:00
Code Issues Projects Releases Packages Wiki Activity

Restore screenshot button.

Browse source
This commit is contained in:
K. S. Ernest (iFire) Lee 2020-11-11 14:12:31 -08:00
parent 8344d27276
commit 4cff322017
1 changed files with 8 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

12
editor/editor_node.cpp
View file

@ -2705,10 +2705,14 @@ void EditorNode::_screenshot(bool p_use_utc) {
} }
void EditorNode::_save_screenshot(NodePath p_path) { void EditorNode::_save_screenshot(NodePath p_path) {
SubViewport *viewport = Object::cast_to<SubViewport>(EditorInterface::get_singleton()->get_editor_viewport()->get_viewport()); Control *editor_viewport = EditorInterface::get_singleton()->get_editor_viewport();
viewport->set_clear_mode(SubViewport::CLEAR_MODE_ONLY_NEXT_FRAME); ERR_FAIL_COND_MSG(!editor_viewport, "Cannot get editor viewport.");
Ref<Image> img = viewport->get_texture()->get_data(); Viewport *viewport = editor_viewport->get_viewport();
viewport->set_clear_mode(SubViewport::CLEAR_MODE_ALWAYS); ERR_FAIL_COND_MSG(!viewport, "Cannot get editor viewport.");
Ref<ViewportTexture> texture = viewport->get_texture();
ERR_FAIL_COND_MSG(texture.is_null(), "Cannot get editor viewport texture.");
Ref<Image> img = texture->get_data();
ERR_FAIL_COND_MSG(img.is_null(), "Cannot get editor viewport texture image.");
Error error = img->save_png(p_path); Error error = img->save_png(p_path);
ERR_FAIL_COND_MSG(error != OK, "Cannot save screenshot to file '" + p_path + "'."); ERR_FAIL_COND_MSG(error != OK, "Cannot save screenshot to file '" + p_path + "'.");
} }