add ply resource loader

.gitignore

@@ -1,3 +1,4 @@
 # Godot 4+ specific ignores
 .godot/*
-*.ply
+*.ply
+point_cloud.ply.import

addons/point_cloud_importer/PointCloudData.gd

@@ -0,0 +1,129 @@
+class_name PointCloudData extends Resource
+
+static var num_coeffs = 45
+static var num_coeffs_per_color = num_coeffs / 3
+static var sh_degree = sqrt(num_coeffs_per_color + 1) - 1
+
+@export var positions = PackedFloat32Array()
+@export var normals = PackedFloat32Array()
+@export var opacities = PackedFloat32Array()
+@export var scales = PackedFloat32Array()
+@export var rotations = PackedFloat32Array()
+@export var sh_coeffs = PackedFloat32Array()
+@export var depths = PackedFloat32Array()
+@export var depth_index = PackedInt32Array()
+@export var num_vertex: int
+
+func _init():
+	num_coeffs = 45
+	num_coeffs_per_color = num_coeffs / 3
+	sh_degree = sqrt(num_coeffs_per_color + 1) - 1
+
+
+static func load_ply_file(filename: String) -> PointCloudData:
+	var point_cloud_data = PointCloudData.new()
+	var file = FileAccess.open(filename, FileAccess.READ)
+
+	if not file:
+		print("Failed to open file: " + filename)
+		return point_cloud_data
+
+	var num_vertex = 0
+#num_vertex = 0
+
+	# Read header
+	var line = file.get_line()
+	while not file.eof_reached():
+		if line.begins_with("element vertex"):
+			num_vertex = int(line.split(" ")[2])
+		if line.begins_with("end_header"):
+			break
+		line = file.get_line()
+
+	var coeffs = []
+
+	for i in range(num_vertex):
+		var vertex = {
+			"x": file.get_float(),
+			"y": file.get_float(),
+			"z": file.get_float(),
+			"nx": file.get_float(),
+			"ny": file.get_float(),
+			"nz": file.get_float(),
+			"f_dc_0": file.get_float(),
+			"f_dc_1": file.get_float(),
+			"f_dc_2": file.get_float(),
+			"f_rest_0": file.get_float(),
+			"f_rest_1": file.get_float(),
+			"f_rest_2": file.get_float(),
+			"f_rest_3": file.get_float(),
+			"f_rest_4": file.get_float(),
+			"f_rest_5": file.get_float(),
+			"f_rest_6": file.get_float(),
+			"f_rest_7": file.get_float(),
+			"f_rest_8": file.get_float(),
+			"f_rest_9": file.get_float(),
+			"f_rest_10": file.get_float(),
+			"f_rest_11": file.get_float(),
+			"f_rest_12": file.get_float(),
+			"f_rest_13": file.get_float(),
+			"f_rest_14": file.get_float(),
+			"f_rest_15": file.get_float(),
+			"f_rest_16": file.get_float(),
+			"f_rest_17": file.get_float(),
+			"f_rest_18": file.get_float(),
+			"f_rest_19": file.get_float(),
+			"f_rest_20": file.get_float(),
+			"f_rest_21": file.get_float(),
+			"f_rest_22": file.get_float(),
+			"f_rest_23": file.get_float(),
+			"f_rest_24": file.get_float(),
+			"f_rest_25": file.get_float(),
+			"f_rest_26": file.get_float(),
+			"f_rest_27": file.get_float(),
+			"f_rest_28": file.get_float(),
+			"f_rest_29": file.get_float(),
+			"f_rest_30": file.get_float(),
+			"f_rest_31": file.get_float(),
+			"f_rest_32": file.get_float(),
+			"f_rest_33": file.get_float(),
+			"f_rest_34": file.get_float(),
+			"f_rest_35": file.get_float(),
+			"f_rest_36": file.get_float(),
+			"f_rest_37": file.get_float(),
+			"f_rest_38": file.get_float(),
+			"f_rest_39": file.get_float(),
+			"f_rest_40": file.get_float(),
+			"f_rest_41": file.get_float(),
+			"f_rest_42": file.get_float(),
+			"f_rest_43": file.get_float(),
+			"f_rest_44": file.get_float(),
+			"opacity": file.get_float(),
+			"scale_0": file.get_float(),
+			"scale_1": file.get_float(),
+			"scale_2": file.get_float(),
+			"rot_0": file.get_float(),
+			"rot_1": file.get_float(),
+			"rot_2": file.get_float(),
+			"rot_3": file.get_float()
+		}
+
+		point_cloud_data.positions.append_array([vertex["x"], vertex["y"], vertex["z"], 0])
+		point_cloud_data.opacities.append(vertex["opacity"])
+		point_cloud_data.scales.append_array([vertex["scale_0"], vertex["scale_1"], vertex["scale_2"], 0])
+		point_cloud_data.rotations.append_array([vertex["rot_0"], vertex["rot_1"], vertex["rot_2"], vertex["rot_3"]])
+		point_cloud_data.depth_index.append(i)
+		point_cloud_data.depths.append(0)
+
+		var coeff = [vertex["f_dc_0"], vertex["f_dc_1"], vertex["f_dc_2"]]
+		for j in range(num_coeffs_per_color):
+			coeff.append_array([
+				vertex["f_rest_%d" % (0 * num_coeffs_per_color + j)],
+				vertex["f_rest_%d" % (1 * num_coeffs_per_color + j)],
+				vertex["f_rest_%d" % (2 * num_coeffs_per_color + j)]
+			])
+		point_cloud_data.sh_coeffs.append_array(coeff)
+
+	file.close()
+	point_cloud_data.num_vertex = num_vertex
+	return point_cloud_data

addons/point_cloud_importer/editor_plugin.gd

@@ -0,0 +1,15 @@
+@tool
+extends EditorPlugin
+
+
+var point_cloud_importer
+
+
+func _enter_tree():
+	point_cloud_importer = preload("point_cloud_importer.gd").new()
+	add_import_plugin(point_cloud_importer)
+
+
+func _exit_tree():
+	remove_import_plugin(point_cloud_importer)
+	point_cloud_importer = null

addons/point_cloud_importer/plugin.cfg

@@ -0,0 +1,7 @@
+[plugin]
+
+name="Point cloud importer"
+description=""
+author="SrColoma"
+version=""
+script="editor_plugin.gd"

addons/point_cloud_importer/point_cloud_importer.gd

@@ -0,0 +1,55 @@
+# Definición del plugin de importación
+@tool
+extends EditorImportPlugin
+
+func _get_import_options(path, preset_index: int) -> Array:
+	return [
+		{
+			"name": "Generate Normals",
+			"default_value": false
+		}
+	]
+
+func _get_importer_name() -> String:
+	return "point_cloud_importer"
+
+func _get_visible_name() -> String:
+	return "Point Cloud Importer"
+
+func _get_recognized_extensions() -> PackedStringArray:
+	return ["ply"]
+
+func _get_save_extension() -> String:
+	return "tres"
+
+func _get_priority() -> float:
+	return 1.0
+
+func _get_import_order() -> int:
+	return 0
+
+func _get_resource_type() -> String:
+	return "Resource"
+
+func _get_preset_count() -> int:
+	return 1
+
+func _get_preset_name(preset: int) -> String:
+	return "Default"
+
+func _import(
+	source_file: String, 
+	save_path: String, 
+	options: Dictionary, 
+	platform_variants: Array, 
+	gen_files: Array
+) -> int:
+	var point_cloud_data = PointCloudData.load_ply_file(source_file)
+
+	var save_result = ResourceSaver.save(point_cloud_data, save_path + ".tres")
+	if save_result != OK:
+		push_error("Failed to save PointCloudData resource")
+		return ERR_CANT_CREATE
+
+	return OK
+

main.gd

@@ -2,7 +2,8 @@ extends Node3D
 
 @onready var camera = get_node("Camera")
 @onready var screen_texture = get_node("TextureRect")
-@export var splat_filename: String = "point_cloud3.ply"
+#@export var splat_filename: String = "point_cloud3.ply"
+@export var splatResource: PointCloudData #= preload("res://point_cloud.ply")
 
 var rd = RenderingServer.create_local_rendering_device()
 var pipeline: RID
@@ -81,110 +82,18 @@ func _set_screen_texture_data(data: PackedByteArray):
 	screen_texture.texture.update(image)
 
 
+
 # terrible loading
 func _load_ply_file():
-	var file = FileAccess.open(splat_filename, FileAccess.READ)
-
-	if not file:
-		print("Failed to open file: " + splat_filename)
-		return
-
-	num_vertex = 0
-
-	# Read header
-	var line = file.get_line()
-	while not file.eof_reached():
-		if line.begins_with("element vertex"):
-			num_vertex = int(line.split(" ")[2])
-		if line.begins_with("end_header"):
-			break
-		line = file.get_line()
-
-	var coeffs = []
-
-	for i in range(num_vertex):
-		var vertex = {
-			"x": file.get_float(),
-			"y": file.get_float(),
-			"z": file.get_float(),
-			"nx": file.get_float(),
-			"ny": file.get_float(),
-			"nz": file.get_float(),
-			"f_dc_0": file.get_float(),
-			"f_dc_1": file.get_float(),
-			"f_dc_2": file.get_float(),
-			"f_rest_0": file.get_float(),
-			"f_rest_1": file.get_float(),
-			"f_rest_2": file.get_float(),
-			"f_rest_3": file.get_float(),
-			"f_rest_4": file.get_float(),
-			"f_rest_5": file.get_float(),
-			"f_rest_6": file.get_float(),
-			"f_rest_7": file.get_float(),
-			"f_rest_8": file.get_float(),
-			"f_rest_9": file.get_float(),
-			"f_rest_10": file.get_float(),
-			"f_rest_11": file.get_float(),
-			"f_rest_12": file.get_float(),
-			"f_rest_13": file.get_float(),
-			"f_rest_14": file.get_float(),
-			"f_rest_15": file.get_float(),
-			"f_rest_16": file.get_float(),
-			"f_rest_17": file.get_float(),
-			"f_rest_18": file.get_float(),
-			"f_rest_19": file.get_float(),
-			"f_rest_20": file.get_float(),
-			"f_rest_21": file.get_float(),
-			"f_rest_22": file.get_float(),
-			"f_rest_23": file.get_float(),
-			"f_rest_24": file.get_float(),
-			"f_rest_25": file.get_float(),
-			"f_rest_26": file.get_float(),
-			"f_rest_27": file.get_float(),
-			"f_rest_28": file.get_float(),
-			"f_rest_29": file.get_float(),
-			"f_rest_30": file.get_float(),
-			"f_rest_31": file.get_float(),
-			"f_rest_32": file.get_float(),
-			"f_rest_33": file.get_float(),
-			"f_rest_34": file.get_float(),
-			"f_rest_35": file.get_float(),
-			"f_rest_36": file.get_float(),
-			"f_rest_37": file.get_float(),
-			"f_rest_38": file.get_float(),
-			"f_rest_39": file.get_float(),
-			"f_rest_40": file.get_float(),
-			"f_rest_41": file.get_float(),
-			"f_rest_42": file.get_float(),
-			"f_rest_43": file.get_float(),
-			"f_rest_44": file.get_float(),
-			"opacity": file.get_float(),
-			"scale_0": file.get_float(),
-			"scale_1": file.get_float(),
-			"scale_2": file.get_float(),
-			"rot_0": file.get_float(),
-			"rot_1": file.get_float(),
-			"rot_2": file.get_float(),
-			"rot_3": file.get_float()
-		}
-
-		positions.append_array([vertex["x"], vertex["y"], vertex["z"], 0])
-		opacities.append(vertex["opacity"])
-		scales.append_array([vertex["scale_0"], vertex["scale_1"], vertex["scale_2"], 0])
-		rotations.append_array([vertex["rot_0"], vertex["rot_1"], vertex["rot_2"], vertex["rot_3"]])
-		depth_index.append(i)
-		depths.append(0)
-
-		var coeff = [vertex["f_dc_0"], vertex["f_dc_1"], vertex["f_dc_2"]]
-		for j in range(num_coeffs_per_color):
-			coeff.append_array([
-				vertex["f_rest_%d" % (0 * num_coeffs_per_color + j)],
-				vertex["f_rest_%d" % (1 * num_coeffs_per_color + j)],
-				vertex["f_rest_%d" % (2 * num_coeffs_per_color + j)]
-			])
-		sh_coeffs.append_array(coeff)
-
-	file.close()
+
+	depths = splatResource.depths
+	depth_index = splatResource.depth_index
+	positions = splatResource.positions
+	opacities = splatResource.opacities
+	scales = splatResource.scales
+	rotations = splatResource.rotations
+	sh_coeffs = splatResource.sh_coeffs
+	num_vertex =  splatResource.num_vertex
 
 
 func _initialise_framebuffer_format():
@@ -491,7 +400,7 @@ func render():
 	_set_screen_texture_data(byte_data)
 
 
-func _process(delta):	
+func _process(_delta):	
 	update()
 	render()
 
@@ -514,4 +423,3 @@ func _sort_splats_by_depth():
 	if angle > 0.6:
 		bitonic_sort()
 		last_direction = direction
-

main.tscn

@@ -1,11 +1,15 @@
-[gd_scene load_steps=3 format=3 uid="uid://c27tc42js165g"]
+[gd_scene load_steps=5 format=3 uid="uid://c27tc42js165g"]
 
 [ext_resource type="Script" path="res://main.gd" id="1_j5eux"]
+[ext_resource type="Resource" uid="uid://brxq4dkwnfyke" path="res://point_cloud.ply" id="2_7olva"]
 [ext_resource type="Script" path="res://Camera.gd" id="2_ppc5i"]
 
+[sub_resource type="PrismMesh" id="PrismMesh_r3wk8"]
+
 [node name="main" type="Node3D"]
 transform = Transform3D(1, 0, 0, 0, 1, 0, 0, 0, 1, 0, 0, 4.56234)
 script = ExtResource("1_j5eux")
+splatResource = ExtResource("2_7olva")
 
 [node name="Camera" type="Camera3D" parent="."]
 current = true
@@ -15,3 +19,7 @@ script = ExtResource("2_ppc5i")
 [node name="TextureRect" type="TextureRect" parent="."]
 offset_right = 40.0
 offset_bottom = 40.0
+
+[node name="MeshInstance3D" type="MeshInstance3D" parent="."]
+transform = Transform3D(1, 0, 0, 0, 1, 0, 0, 0, 1, 0, 0, -4.60119)
+mesh = SubResource("PrismMesh_r3wk8")

project.godot

@@ -15,6 +15,10 @@ run/main_scene="res://main.tscn"
 config/features=PackedStringArray("4.2", "Forward Plus")
 config/icon="res://icon.svg"
 
+[editor_plugins]
+
+enabled=PackedStringArray("res://addons/point_cloud_importer/plugin.cfg")
+
 [input]
 
 move_left={