1
0
Fork 0
mirror of https://github.com/Orama-Interactive/Pixelorama.git synced 2025-02-23 05:53:14 +00:00
Pixelorama/src/Classes/ShaderImageEffect.gd

63 lines
2.4 KiB
GDScript

class_name ShaderImageEffect
extends RefCounted
## Helper class to generate image effects using shaders
signal done
func generate_image(img: Image, shader: Shader, params: Dictionary, size: Vector2i) -> void:
# duplicate shader before modifying code to avoid affecting original resource
var resized_width := false
var resized_height := false
if size.x == 1:
size.x = 2
img.crop(2, img.get_height())
resized_width = true
if size.y == 1:
size.y = 2
img.crop(img.get_width(), 2)
resized_height = true
shader = shader.duplicate()
shader.code = shader.code.replace("unshaded", "unshaded, blend_premul_alpha")
var vp := RenderingServer.viewport_create()
var canvas := RenderingServer.canvas_create()
RenderingServer.viewport_attach_canvas(vp, canvas)
RenderingServer.viewport_set_size(vp, size.x, size.y)
RenderingServer.viewport_set_disable_3d(vp, true)
RenderingServer.viewport_set_active(vp, true)
RenderingServer.viewport_set_transparent_background(vp, true)
RenderingServer.viewport_set_default_canvas_item_texture_filter(
vp, RenderingServer.CANVAS_ITEM_TEXTURE_FILTER_NEAREST
)
var ci_rid := RenderingServer.canvas_item_create()
RenderingServer.viewport_set_canvas_transform(vp, canvas, Transform3D())
RenderingServer.canvas_item_set_parent(ci_rid, canvas)
var texture := RenderingServer.texture_2d_create(img)
RenderingServer.canvas_item_add_texture_rect(ci_rid, Rect2(Vector2.ZERO, size), texture)
var mat_rid := RenderingServer.material_create()
RenderingServer.material_set_shader(mat_rid, shader.get_rid())
RenderingServer.canvas_item_set_material(ci_rid, mat_rid)
for key in params:
var param = params[key]
if param is Texture2D or param is Texture2DArray:
RenderingServer.material_set_param(mat_rid, key, [param])
else:
RenderingServer.material_set_param(mat_rid, key, param)
RenderingServer.viewport_set_update_mode(vp, RenderingServer.VIEWPORT_UPDATE_ONCE)
RenderingServer.force_draw(false)
var viewport_texture := RenderingServer.texture_2d_get(RenderingServer.viewport_get_texture(vp))
RenderingServer.free_rid(vp)
RenderingServer.free_rid(canvas)
RenderingServer.free_rid(ci_rid)
RenderingServer.free_rid(mat_rid)
RenderingServer.free_rid(texture)
viewport_texture.convert(img.get_format())
img.copy_from(viewport_texture)
if resized_width:
img.crop(img.get_width() - 1, img.get_height())
if resized_height:
img.crop(img.get_width(), img.get_height() - 1)
done.emit()