godot-demo-projects/viewport/screen_capture/screen_capture.gd

23 lines
604 B
GDScript

extends Node
onready var captured_image = $CapturedImage
func _on_CaptureButton_pressed():
get_viewport().set_clear_mode(Viewport.CLEAR_MODE_ONLY_NEXT_FRAME)
# Let two frames pass to make sure the screen was captured.
yield(get_tree(), "idle_frame")
yield(get_tree(), "idle_frame")
# Retrieve the captured image.
var img = get_viewport().get_texture().get_data()
# Flip it on the y-axis (because it's flipped).
img.flip_y()
# Create a texture for it.
var tex = ImageTexture.new()
tex.create_from_image(img)
# Set the texture to the captured image node.
captured_image.set_texture(tex)