Added Packed_*_Array support

addons/compute_worker/gpu_uniforms/gpu_packed_int32_array.gd

@@ -0,0 +1,91 @@
+# GLSL data type encoding: `int`
+
+extends GPUUniform
+class_name GPU_PackedInt32Array
+
+enum UNIFORM_TYPES{
+	UNIFORM_BUFFER,
+	STORAGE_BUFFER
+}
+
+## The initial data supplied to the uniform
+@export var data: PackedInt32Array = PackedInt32Array()
+## The size of the array as defined in the shader. Only used if `data` is not defined.
+@export var array_size: int = 0
+## The shader binding for this uniform
+@export var binding: int = 0
+## Type of uniform to create. `UNIFORM_BUFFER`s cannot be altered from within the shader
+@export var uniform_type: UNIFORM_TYPES = UNIFORM_TYPES.UNIFORM_BUFFER
+
+var data_rid: RID = RID()
+var uniform: RDUniform = RDUniform.new()
+
+
+func initialize(rd: RenderingDevice) -> RDUniform:
+
+	if data.is_empty():
+		if array_size > 0:
+			data.resize(array_size)
+		else:
+			printerr("You must define the uniform's `data` or `array_size`.")
+			return
+
+	# Create the buffer using our initial data
+	data_rid = create_rid(rd)
+
+	# Create RDUniform object using the provided binding id and data
+	return create_uniform()
+
+
+func create_uniform() -> RDUniform:
+
+	match uniform_type:
+		UNIFORM_TYPES.UNIFORM_BUFFER:
+			uniform.uniform_type = RenderingDevice.UNIFORM_TYPE_UNIFORM_BUFFER
+		UNIFORM_TYPES.STORAGE_BUFFER:
+			uniform.uniform_type = RenderingDevice.UNIFORM_TYPE_STORAGE_BUFFER
+
+	uniform.binding = binding
+	uniform.add_id(data_rid)
+
+	return uniform
+
+
+func create_rid(rd: RenderingDevice) -> RID:
+
+	var bytes = data.to_byte_array()
+
+	var buffer: RID = RID()
+
+	match uniform_type:
+		UNIFORM_TYPES.UNIFORM_BUFFER:
+			bytes = pad_byte_array_std140(bytes)
+			buffer = rd.uniform_buffer_create(bytes.size(), bytes)
+		UNIFORM_TYPES.STORAGE_BUFFER:
+			buffer = rd.storage_buffer_create(bytes.size(), bytes)
+
+	return buffer
+
+
+func get_uniform_data(rd: RenderingDevice) -> PackedInt32Array:
+	var out := rd.buffer_get_data(data_rid)
+	return out.to_int32_array()
+
+
+func set_uniform_data(rd: RenderingDevice, array: PackedInt32Array) -> void:
+	var sb_data = array.to_byte_array()
+	rd.buffer_update(data_rid, 0 , sb_data.size(), sb_data)
+
+
+func pad_byte_array_std140(arr: PackedByteArray) -> PackedByteArray:
+
+	arr.resize(arr.size() * 2)
+	var next_offset = 0
+
+	for i in range(arr.size()):
+		if next_offset + 4 > arr.size():
+			break
+		arr.encode_double(next_offset + 4, 0.0)
+		next_offset += 16
+
+	return arr

addons/compute_worker/gpu_uniforms/gpu_packed_int32_array.tres

@@ -0,0 +1,10 @@
+[gd_resource type="Resource" script_class="GPU_packed_int32_array" load_steps=2 format=3 uid="uid://b5uu7tqhsjaam"]
+
+[ext_resource type="Script" path="res://addons/compute_worker/gpu_uniforms/gpu_packed_int32_array.gd" id="1_571yc"]
+
+[resource]
+script = ExtResource("1_571yc")
+data = null
+binding = 0
+uniform_type = 0
+alias = ""

addons/compute_worker/gpu_uniforms/gpu_packed_vector2_array.gd

@@ -0,0 +1,79 @@
+# GLSL data type encoding: `vec4[]`, `vec3[]`
+
+extends GPUUniform
+class_name GPU_PackedVector2Array
+
+enum UNIFORM_TYPES{
+	UNIFORM_BUFFER,
+	STORAGE_BUFFER
+}
+
+## The initial data supplied to the uniform
+@export var data: PackedVector2Array = PackedVector2Array()
+## The size of the array as defined in the shader. Only used if `data` is not defined.
+@export var array_size: int = 0
+## The shader binding for this uniform
+@export var binding: int = 0
+## Type of uniform to create. `UNIFORM_BUFFER`s cannot be altered from within the shader
+@export var uniform_type: UNIFORM_TYPES = UNIFORM_TYPES.UNIFORM_BUFFER
+
+var data_rid: RID = RID()
+var uniform: RDUniform = RDUniform.new()
+
+
+func initialize(rd: RenderingDevice) -> RDUniform:
+
+	if data.is_empty():
+
+		assert(array_size > 0, "You must define the uniform's `data` or `array_size`.")
+
+		if array_size > 0:
+			data.resize(array_size)
+
+	# Create the buffer using our initial data
+	data_rid = create_rid(rd)
+
+	# Create RDUniform object using the provided binding id and data
+	return create_uniform()
+
+
+func create_uniform() -> RDUniform:
+
+	match uniform_type:
+		UNIFORM_TYPES.UNIFORM_BUFFER:
+			uniform.uniform_type = RenderingDevice.UNIFORM_TYPE_UNIFORM_BUFFER
+		UNIFORM_TYPES.STORAGE_BUFFER:
+			uniform.uniform_type = RenderingDevice.UNIFORM_TYPE_STORAGE_BUFFER
+
+	uniform.binding = binding
+	uniform.add_id(data_rid)
+
+	return uniform
+
+
+func create_rid(rd: RenderingDevice) -> RID:
+
+	var bytes = vec2_array_to_byte_array(data)
+
+	var buffer: RID = RID()
+
+	match uniform_type:
+		UNIFORM_TYPES.UNIFORM_BUFFER:
+			buffer = rd.uniform_buffer_create(bytes.size(), bytes)
+		UNIFORM_TYPES.STORAGE_BUFFER:
+			buffer = rd.storage_buffer_create(bytes.size(), bytes)
+
+	return buffer
+
+
+func get_uniform_data(rd: RenderingDevice) -> PackedVector2Array:
+	var out := rd.buffer_get_data(data_rid)
+	return byte_array_to_vec2_array(out)
+
+
+func set_uniform_data(rd: RenderingDevice, array: PackedVector2Array) -> void:
+	var sb_data = vec2_array_to_byte_array(array)
+	rd.buffer_update(data_rid, 0 , sb_data.size(), sb_data)
+
+
+

addons/compute_worker/gpu_uniforms/gpu_packed_vector2_array.tres

@@ -0,0 +1,11 @@
+[gd_resource type="Resource" script_class="GPU_PackedVector2Array" load_steps=2 format=3 uid="uid://c1pyh5mnrbuwm"]
+
+[ext_resource type="Script" path="res://addons/compute_worker/gpu_uniforms/gpu_packed_vector2_array.gd" id="1_cmbw3"]
+
+[resource]
+script = ExtResource("1_cmbw3")
+data = null
+array_size = 0
+binding = 0
+uniform_type = 0
+alias = ""

addons/compute_worker/gpu_uniforms/gpu_packed_vector3_array.gd

@@ -1,4 +1,4 @@
-# GLSL data type encoding: `vec4[]`, `vec3[]`
+# GLSL data type encoding: `vec3[]`
 
 extends GPUUniform
 class_name GPU_PackedVector3Array

addons/compute_worker/gpu_uniforms/gpu_struct.gd

@@ -58,6 +58,7 @@ func create_rid(rd: RenderingDevice) -> RID:
 
 func get_uniform_data(rd: RenderingDevice):
 	var out := rd.buffer_get_data(data_rid)
+
 	return decode_struct(out)
 
 
@@ -71,13 +72,20 @@ func set_uniform_data(rd: RenderingDevice, data: Array) -> void:
 ## Encode the contents of the passed in `data` Array to PackedByteArray. 
 ## Contents of `data` must match the order and data types defined in `struct_data`.
 func encode_struct(data: Array, init: bool = false) -> PackedByteArray:
-
 	var arr: PackedByteArray = PackedByteArray()
 	var data_index = 0
 
 	for type_obj in struct_data:
 
 		match typeof(type_obj):
+			TYPE_PACKED_VECTOR3_ARRAY:
+				arr.append_array(vec3_array_to_byte_array(data[data_index]))
+			TYPE_PACKED_VECTOR2_ARRAY:
+				arr.append_array(vec2_array_to_byte_array(data[data_index]))
+			TYPE_VECTOR2I:
+				arr.append_array(vec2i_to_byte_array(data[data_index]))
+			TYPE_VECTOR2:
+				arr.append_array(vec2_to_byte_array(data[data_index]))
 			TYPE_VECTOR3I:
 				arr.append_array(vec3i_to_byte_array(data[data_index]))
 			TYPE_COLOR:
@@ -94,7 +102,6 @@ func encode_struct(data: Array, init: bool = false) -> PackedByteArray:
 	if !init:
 		if pad_byte_array(arr).size() != byte_length && uniform.uniform_type == RenderingDevice.UNIFORM_TYPE_UNIFORM_BUFFER:
 			printerr("Data for uniform: " + str(alias) + " does not match struct requirements. Needs: " + str(byte_length) + " was given: " + str(arr.size()))
-
 	return pad_byte_array(arr)
 
 
@@ -110,14 +117,18 @@ func pad_byte_array(arr: PackedByteArray):
 ## Decode the contents of the passed in PackedByteArray to an Array matching
 ## the order and data types defined in `struct_data`. 
 func decode_struct(data: PackedByteArray) -> Array:
-
 	var arr: Array = []
 
 	var offset: int = 0
 
 	for i in range(struct_data.size()):
-
 		match typeof(struct_data[i]):
+
+			TYPE_PACKED_VECTOR3_ARRAY:
+				var arr_size = struct_data[i].size()
+				var pvec3arr = byte_array_to_vec3_array(data.slice(offset, offset + 16*arr_size))
+				arr.push_back(pvec3arr)
+				offset += 16*arr_size
 			TYPE_VECTOR3I:
 				var vec = byte_array_to_vec3(data.slice(offset, offset + 32))
 				arr.push_back(vec)
@@ -130,6 +141,19 @@ func decode_struct(data: PackedByteArray) -> Array:
 				var vec = byte_array_to_vec3(data.slice(offset, offset + 32))
 				arr.push_back(vec)
 				offset += 32
+			TYPE_PACKED_VECTOR2_ARRAY:
+				var arr_size = struct_data[i].size()
+				var pvec2arr = byte_array_to_vec2_array(data.slice(offset, offset + 16*arr_size))
+				arr.push_back(pvec2arr)
+				offset += 16*arr_size
+			TYPE_VECTOR2I:
+				var vec = byte_array_to_vec2(data.slice(offset, offset + 16))
+				arr.push_back(vec)
+				offset += 16
+			TYPE_VECTOR2:
+				var vec = byte_array_to_vec2(data.slice(offset, offset + 16))
+				arr.push_back(vec)
+				offset += 16
 			TYPE_FLOAT:
 				var flo = byte_array_to_float(data.slice(offset, offset + 8))
 				arr.push_back(flo)
@@ -138,5 +162,4 @@ func decode_struct(data: PackedByteArray) -> Array:
 				var integer = byte_array_to_int(data.slice(offset, offset + 4))
 				arr.push_back(integer)
 				offset += 4
-
 	return arr

addons/compute_worker/gpu_uniforms/gpu_struct_array.gd

@@ -50,10 +50,9 @@ func create_rid(rd: RenderingDevice) -> RID:
 	return buffer
 
 
-func get_uniform_data(rd: RenderingDevice) -> Array[Array]:
+func get_uniform_data(rd: RenderingDevice):
 
 	var out := rd.buffer_get_data(data_rid)
-
 	var arr: Array = []
 
 	@warning_ignore("integer_division")