mirror of
https://github.com/Relintai/GraphicsEditor.git
synced 2024-11-12 08:15:17 +01:00
379 lines
11 KiB
GDScript
379 lines
11 KiB
GDScript
tool
|
|
extends Control
|
|
|
|
var image = Image.new()
|
|
var last_pixel = []
|
|
onready var canvas_image_node = get_node("CanvasImage")
|
|
export var grid_size = 16
|
|
export var canvas_size = Vector2(100, 100)
|
|
export var region_size = 10
|
|
export var can_draw = true
|
|
|
|
var mouse_in_region
|
|
var mouse_on_top
|
|
|
|
#terms
|
|
#global cell - a cell that has a global grid position on the canvas
|
|
#local cell - a cell that has a local grid position in a chunk region on the canvas
|
|
#chunk region - a set of cells contained in an even number
|
|
|
|
#TODO: Maybe each chunk region can hold an image resource so that way the engine wouldn't lag at all when updating the canvas
|
|
|
|
var layers = [
|
|
{
|
|
"name": "Layer 1",
|
|
"data": [],
|
|
"chunks": []
|
|
}
|
|
]
|
|
var active_layer = "Layer 1"
|
|
|
|
var chunks = []
|
|
|
|
func get_layer_data(layer_name):
|
|
for layer in layers:
|
|
if layer.name == layer_name:
|
|
return layer.data
|
|
#return layer.chunks
|
|
|
|
func set_layer_data(layer_name, layer_data):
|
|
for layer in layers:
|
|
if layer.name == layer_name:
|
|
layer.data = layer_data
|
|
return
|
|
|
|
func remove_layer(layer_name):
|
|
for layer in layers:
|
|
if layer.name == layer_name:
|
|
layers[layer] = null
|
|
return
|
|
|
|
func add_layer(layer_name):
|
|
for layer in layers:
|
|
#TODO: Allow users to create layers with the same name, we can do this because of the indexes in the layer array
|
|
if layer.name == layer_name:
|
|
return false
|
|
layers.append({"name": layer_name, "data": []})
|
|
#layers.append({"name": layer_name, "chunks": []})
|
|
return true
|
|
|
|
var util = preload("res://addons/graphics_editor/Util.gd")
|
|
|
|
func _ready():
|
|
#canvas_image_node.rect_scale = Vector2(grid_size, grid_size)
|
|
#image.create(canvas_size.x, canvas_size.y, true, Image.FORMAT_RGBA8)
|
|
#image.lock()
|
|
#print(get_maxium_filled_chunks())
|
|
# for pixel_data in get_layer_data(active_layer):
|
|
# var cx = pixel_data[0]
|
|
# var cy = pixel_data[1]
|
|
# var ccolor = pixel_data[2]
|
|
# image.set_pixel(cx, cy, ccolor)
|
|
generate_chunks()
|
|
|
|
func _process(delta):
|
|
var mouse_position = get_local_mouse_position()
|
|
var rect = Rect2(Vector2(0, 0), self.rect_size)
|
|
mouse_in_region = rect.has_point(mouse_position)
|
|
update()
|
|
#if not Engine.editor_hint:
|
|
# print(mouse_on_canvas, " | ", has_focus())
|
|
#draw_canvas_out just updates the image constantly
|
|
#if can_draw:
|
|
# draw_canvas_out()
|
|
|
|
func generate_chunks():
|
|
var maxium_chunk_size = get_maxium_filled_chunks()
|
|
#TODO: We probably don't need to check for x and y anymore
|
|
for x in maxium_chunk_size.x:
|
|
for y in maxium_chunk_size.y:
|
|
var paint_canvas_chunk = load("res://addons/graphics_editor/PaintCanvasChunk.tscn").instance()
|
|
paint_canvas_chunk.setup(region_size)
|
|
paint_canvas_chunk.name = "C-%s-%s" % [x, y]
|
|
paint_canvas_chunk.rect_position = Vector2(x * (16 * 10), y * (16 * 10))
|
|
get_node("Chunks").add_child(paint_canvas_chunk)
|
|
|
|
func get_maxium_filled_chunks():
|
|
return Vector2(canvas_size.x / region_size, canvas_size.y / region_size).ceil()
|
|
|
|
#TODO: Remake these functions with godot's setget features
|
|
#so that we don't have to call these functions
|
|
func resize_grid(grid):
|
|
#print(grid)
|
|
if grid <= 0:
|
|
return
|
|
grid_size = grid
|
|
canvas_image_node.rect_scale = Vector2(grid, grid)
|
|
|
|
func resize_canvas(x, y):
|
|
image.unlock()
|
|
image.create(x, y, true, Image.FORMAT_RGBA8)
|
|
canvas_size = Vector2(x, y)
|
|
#setup_all_chunks()
|
|
image.lock()
|
|
|
|
#func draw_canvas_out(a = ""):
|
|
# if canvas_image_node == null:
|
|
# return
|
|
# var image_texture = ImageTexture.new()
|
|
# image_texture.create_from_image(image)
|
|
# image_texture.set_flags(0)
|
|
# canvas_image_node.texture = image_texture
|
|
|
|
func get_wrapped_region_cell(x, y):
|
|
return Vector2(wrapi(x, 0, region_size), wrapi(y, 0, region_size))
|
|
|
|
func get_region_from_cell(x, y):
|
|
return Vector2(floor(x / region_size), floor(y / region_size))
|
|
|
|
func set_local_cell_in_chunk(chunk_x, chunk_y, local_cell_x, local_cell_y, color):
|
|
var chunk_node = get_node("Chunks").get_node_or_null("C-%s-%s" % [chunk_x, chunk_y])
|
|
if chunk_node == null:
|
|
print("Can't find chunk node!")
|
|
return
|
|
chunk_node.set_cell(local_cell_x, local_cell_y, color)
|
|
|
|
func set_global_cell_in_chunk(cell_x, cell_y, color):
|
|
var chunk = get_region_from_cell(cell_x, cell_y)
|
|
var wrapped_cell = get_wrapped_region_cell(cell_x, cell_y)
|
|
set_local_cell_in_chunk(chunk.x, chunk.y, wrapped_cell.x, wrapped_cell.y, color)
|
|
|
|
#func update_chunk_region_from_cell(x, y):
|
|
# var region_to_update = get_region_from_cell(x, y)
|
|
# update_chunk_region(region_to_update.x, region_to_update.y)
|
|
|
|
func get_pixel_cell_color(x, y):
|
|
if not cell_in_canvas_region(x, y):
|
|
return null
|
|
var pixel_cell = get_pixel_cell(x, y)
|
|
if pixel_cell == null:
|
|
#We already checked that the cell can't be out of the canvas region so we can assume the pixel cell is completely transparent if it's null
|
|
return Color(0, 0, 0, 0)
|
|
else:
|
|
return util.color_from_array(pixel_cell[2])
|
|
|
|
func get_pixel_cell_color_v(vec2):
|
|
return get_pixel_cell_color(vec2.x, vec2.y)
|
|
|
|
func get_pixel_cell(x, y):
|
|
if not cell_in_canvas_region(x, y):
|
|
return null
|
|
|
|
for pixel in get_layer_data(active_layer):
|
|
if pixel[0] == x and pixel[1] == y:
|
|
return pixel
|
|
|
|
return null
|
|
|
|
func get_pixel_cell_v(vec2):
|
|
return get_pixel_cell(vec2.x, vec2.y)
|
|
|
|
#func remove_pixel_cell(x, y):
|
|
# if can_draw == false:
|
|
# return false
|
|
# if not cell_in_canvas_region(x, y):
|
|
# return false
|
|
# var layer_data = get_layer_data("Layer 1")
|
|
# for pixel in range(0, layer_data.size()):
|
|
# if layer_data[pixel][0] == x and layer_data[pixel][1] == y:
|
|
# layer_data.remove(pixel)
|
|
# #update_chunk_region_from_cell(x, y)
|
|
# #TOOD: If pixel exists in temp_pool_pixels then remove it
|
|
# image.set_pixel(x, y, Color(0, 0, 0, 0))
|
|
# return true
|
|
# return false
|
|
|
|
#func remove_pixel_cell_v(vec2):
|
|
# return remove_pixel_cell(vec2.x, vec2.y)
|
|
|
|
func set_pixel_cell(x, y, color):
|
|
if can_draw == false:
|
|
return false
|
|
if not cell_in_canvas_region(x, y):
|
|
return false
|
|
var layer_data = get_layer_data(active_layer)
|
|
var index = 0
|
|
for pixel in layer_data:
|
|
#TODO: Make a better way of accessing the array because the more pixels we have, the longer it takes to
|
|
#set the pixel
|
|
if pixel[0] == x and pixel[1] == y:
|
|
#No reason to set the pixel again if the colors are the same
|
|
if pixel[2] == color:
|
|
return
|
|
#If the color we are setting is 0, 0, 0, 0 then there is no reason to keep the information about the pixel
|
|
#so we remove it from the layer data
|
|
if color == Color(0, 0, 0, 0):
|
|
layer_data.remove(index)
|
|
else:
|
|
pixel[2] = color
|
|
#TODO: The new system is going to allow chunks to each have their own TextureRect and Image
|
|
#nodes so what we are doing in here is that we are setting the local cell in the region of that image
|
|
set_global_cell_in_chunk(x, y, color)
|
|
last_pixel = [x, y, color]
|
|
return true
|
|
index += 1
|
|
#don't append any data if the color is 0, 0, 0, 0
|
|
if color != Color(0, 0, 0, 0):
|
|
#if the pixel data doesn't exist then we add it in
|
|
layer_data.append([x, y, color])
|
|
set_global_cell_in_chunk(x, y, color)
|
|
last_pixel = [x, y, color]
|
|
return true
|
|
|
|
func set_pixel_cell_v(vec2, color):
|
|
return set_pixel_cell(vec2.x, vec2.y, color)
|
|
|
|
func set_pixels_from_line(vec2_1, vec2_2, color):
|
|
var points = get_pixels_from_line(vec2_1, vec2_2)
|
|
for i in points:
|
|
set_pixel_cell_v(i, color)
|
|
|
|
func set_random_pixels_from_line(vec2_1, vec2_2):
|
|
var points = get_pixels_from_line(vec2_1, vec2_2)
|
|
for i in points:
|
|
set_pixel_cell_v(i, util.random_color_alt())
|
|
|
|
func get_pixels_from_line(vec2_1, vec2_2):
|
|
var points = PoolVector2Array()
|
|
|
|
var dx = abs(vec2_2.x - vec2_1.x)
|
|
var dy = abs(vec2_2.y - vec2_1.y)
|
|
|
|
var x = vec2_1.x
|
|
var y = vec2_1.y
|
|
|
|
var sx = 0
|
|
if vec2_1.x > vec2_2.x:
|
|
sx = -1
|
|
else:
|
|
sx = 1
|
|
|
|
var sy = 0
|
|
if vec2_1.y > vec2_2.y:
|
|
sy = -1
|
|
else:
|
|
sy = 1
|
|
|
|
if dx > dy:
|
|
var err = dx / 2
|
|
while(true):
|
|
if x == vec2_2.x:
|
|
break
|
|
points.push_back(Vector2(x, y))
|
|
|
|
err -= dy
|
|
if err < 0:
|
|
y += sy
|
|
err += dx
|
|
x += sx
|
|
else:
|
|
var err = dy / 2
|
|
while (true):
|
|
if y == vec2_2.y:
|
|
break
|
|
points.push_back(Vector2(x, y))
|
|
|
|
err -= dx
|
|
if err < 0:
|
|
x += sx
|
|
err += dy
|
|
y += sy
|
|
points.push_back(Vector2(x, y))
|
|
return points
|
|
|
|
#even though the function checks for it, we can't afford adding more functions to the call stack
|
|
#because godot has a limit until it crashes
|
|
var flood_fill_queue = 0
|
|
func flood_fill(x, y, target_color, replacement_color):
|
|
#yield(get_tree().create_timer(1), "timeout")
|
|
flood_fill_queue += 1
|
|
if not cell_in_canvas_region(x, y):
|
|
flood_fill_queue -= 1
|
|
return
|
|
if target_color == replacement_color:
|
|
flood_fill_queue -= 1
|
|
return
|
|
elif not get_pixel_cell_color(x, y) == target_color:
|
|
flood_fill_queue -= 1
|
|
return
|
|
else:
|
|
set_pixel_cell(x, y, replacement_color)
|
|
if flood_fill_queue >= 500:
|
|
print(flood_fill_queue)
|
|
yield(get_tree().create_timer(0.01), "timeout")
|
|
#up
|
|
if get_pixel_cell_color(x, y - 1) == target_color:
|
|
flood_fill(x, y - 1, target_color, replacement_color)
|
|
#down
|
|
if get_pixel_cell_color(x, y + 1) == target_color:
|
|
flood_fill(x, y + 1, target_color, replacement_color)
|
|
#left
|
|
if get_pixel_cell_color(x - 1, y) == target_color:
|
|
flood_fill(x - 1, y, target_color, replacement_color)
|
|
#right
|
|
if get_pixel_cell_color(x + 1, y) == target_color:
|
|
flood_fill(x + 1, y, target_color, replacement_color)
|
|
flood_fill_queue -= 1
|
|
return
|
|
|
|
#func flood_fill_erase(x, y, target_color):
|
|
# yield(get_tree().create_timer(0.001), "timeout")
|
|
# if not cell_in_canvas_region(x, y):
|
|
# print("cell not in canvas")
|
|
# return
|
|
# #if target_color == replacement_color:
|
|
# # return
|
|
# elif not get_pixel_cell_color(x, y) == target_color:
|
|
# print("cell doesn't match pixel color")
|
|
# return
|
|
# elif not get_pixel_cell(x, y):
|
|
# print("cell already erased")
|
|
# return
|
|
# else:
|
|
# print("removed pixel")
|
|
# remove_pixel_cell(x, y)
|
|
# print("x: ", x, " y: ", y, " color: ", target_color)
|
|
# #up
|
|
# flood_fill_erase(x, y - 1, target_color)
|
|
# #down
|
|
# flood_fill_erase(x, y + 1, target_color)
|
|
# #left
|
|
# flood_fill_erase(x - 1, y, target_color)
|
|
# #right
|
|
# flood_fill_erase(x + 1, y, target_color)
|
|
# return
|
|
|
|
func cell_in_canvas_region(x, y):
|
|
if x > canvas_size.x - 1 or x < 0 or y > canvas_size.y - 1 or y < 0:
|
|
#out of bounds, return false
|
|
return false
|
|
else:
|
|
return true
|
|
|
|
#Both of these functions right now just return the starting position of the canvas and the last position of the canvas
|
|
func get_all_used_regions_in_canvas():
|
|
var first_used_region = get_first_used_region_in_canvas()
|
|
var last_used_region = get_last_used_region_in_canvas()
|
|
var chunk_pool = PoolVector2Array()
|
|
for chunk_x in range(first_used_region.x, last_used_region.x):
|
|
for chunk_y in range(first_used_region.y, last_used_region.y):
|
|
chunk_pool.append(Vector2(chunk_x, chunk_y))
|
|
return chunk_pool
|
|
|
|
func get_first_used_region_in_canvas():
|
|
return get_region_from_cell(0, 0)
|
|
|
|
func get_last_used_region_in_canvas():
|
|
return get_region_from_cell(canvas_size.x - 1, canvas_size.y - 1)
|
|
|
|
func get_cells_in_region(x, y):
|
|
var start_cell = Vector2(x * region_size, y * region_size)
|
|
var end_cell = Vector2((x * region_size) + region_size, (y * region_size) + region_size)
|
|
var cell_array = []
|
|
for cx in range(start_cell.x, end_cell.x):
|
|
for cy in range(start_cell.y, end_cell.y):
|
|
var pixel_cell = get_pixel_cell(cx, cy)
|
|
if pixel_cell == null:
|
|
pixel_cell = [cx, cy, Color(0, 0, 0, 0)]
|
|
cell_array.append(pixel_cell)
|
|
return cell_array |