mirror of
https://github.com/Orama-Interactive/Pixelorama.git
synced 2025-01-18 17:19:50 +00:00
83 lines
2.1 KiB
GDScript3
83 lines
2.1 KiB
GDScript3
|
extends RefCounted
|
||
|
|
|
||
|
|
var converter := preload("../converter.gd").new()
|
||
|
|
var transparency := false
|
||
|
|
|
||
|
|
|
||
|
|
func how_many_divisions(colors_count: int) -> int:
|
||
|
|
return int(ceili(pow(colors_count, 1.0 / 4.0)))
|
||
|
|
|
||
|
|
|
||
|
|
func generate_colors(colors_count: int) -> Array:
|
||
|
|
var divisions_count: int = how_many_divisions(colors_count)
|
||
|
|
var colors: Array = []
|
||
|
|
|
||
|
|
for a in range(divisions_count):
|
||
|
|
for b in range(divisions_count):
|
||
|
|
for g in range(divisions_count):
|
||
|
|
for r in range(divisions_count):
|
||
|
|
colors.append(
|
||
|
|
[
|
||
|
|
Vector3(
|
||
|
|
(255.0 / divisions_count) * r,
|
||
|
|
(255.0 / divisions_count) * g,
|
||
|
|
(255.0 / divisions_count) * b
|
||
|
|
),
|
||
|
|
(255.0 / divisions_count) * a
|
||
|
|
]
|
||
|
|
)
|
||
|
|
|
||
|
|
return colors
|
||
|
|
|
||
|
|
|
||
|
|
func find_nearest_color(palette_color: Vector3, image_data: PackedByteArray) -> Array:
|
||
|
|
var nearest_color = null
|
||
|
|
var nearest_alpha = null
|
||
|
|
for i in range(0, image_data.size(), 4):
|
||
|
|
var color := Vector3(image_data[i], image_data[i + 1], image_data[i + 2])
|
||
|
|
# detect transparency
|
||
|
|
if image_data[3] == 0:
|
||
|
|
transparency = true
|
||
|
|
if (
|
||
|
|
(nearest_color == null)
|
||
|
|
or (
|
||
|
|
palette_color.distance_squared_to(color)
|
||
|
|
< palette_color.distance_squared_to(nearest_color)
|
||
|
|
)
|
||
|
|
):
|
||
|
|
nearest_color = color
|
||
|
|
nearest_alpha = image_data[i + 3]
|
||
|
|
return [nearest_color, nearest_alpha]
|
||
|
|
|
||
|
|
|
||
|
|
## Moves every color from palette colors to the nearest found color in image
|
||
|
|
func enhance_colors(image: Image, palette_colors: Array) -> Array:
|
||
|
|
var data := image.get_data()
|
||
|
|
|
||
|
|
for i in range(palette_colors.size()):
|
||
|
|
var nearest_color := find_nearest_color(palette_colors[i][0], data)
|
||
|
|
palette_colors[i] = nearest_color
|
||
|
|
|
||
|
|
return palette_colors
|
||
|
|
|
||
|
|
|
||
|
|
func to_color_array(colors: Array) -> Array:
|
||
|
|
var result := []
|
||
|
|
for v in colors:
|
||
|
|
result.append([v[0].x, v[0].y, v[0].z])
|
||
|
|
return result
|
||
|
|
|
||
|
|
|
||
|
|
## Quantizes to gif ready codes
|
||
|
|
func quantize(image: Image) -> Array:
|
||
|
|
var colors: Array = generate_colors(256)
|
||
|
|
var tmp_image := Image.new()
|
||
|
|
tmp_image.copy_from(image)
|
||
|
|
tmp_image.resize(32, 32)
|
||
|
|
colors = enhance_colors(tmp_image, colors)
|
||
|
|
colors = to_color_array(colors)
|
||
|
|
|
||
|
|
var data: PackedByteArray = converter.get_similar_indexed_datas(image, colors)
|
||
|
|
|
||
|
|
return [data, colors, transparency]
|