huskee/Pixelorama
1
0
Fork 0
mirror of https://github.com/Orama-Interactive/Pixelorama.git synced 2025-01-18 09:09:47 +00:00
Code Issues Wiki Activity

Update ObjParse.gd

Browse source 
Updated to c3156e2105/obj-parse/ObjParse.gd. Should fix #1165.
This commit is contained in:
Emmanouil Papadeas 2025-01-16 20:17:21 +02:00
parent 952498a2b8
commit 48130809c9
1 changed files with 144 additions and 122 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

266
src/Classes/ObjParse.gd
View file

@ -1,105 +1,111 @@
class_name ObjParse
extends Object
## Obj parser made by Ezcha, updated by Deakcor
## Created on 7/11/2018
## https://ezcha.net
## https://github.com/Ezcha/gd-obj
## MIT License
## https://github.com/Ezcha/gd-obj/blob/master/LICENSE
## Returns an array of materials from a MTL file
const DEBUG: bool = false
const DEBUG := false
# Public methods
# gd-obj
#
# Created on 7/11/2018
#
# Originally made by Ezcha
# Contributors: deakcor, kb173, jeffgamedev
#
# https://ezcha.net
# https://github.com/Ezcha/gd-obj
#
# MIT License
# https://github.com/Ezcha/gd-obj/blob/master/LICENSE
## Create mesh from obj and mtl paths
# Create mesh from obj and mtl paths
static func load_obj(obj_path: String, mtl_path: String = "") -> Mesh:
var obj_str: String = _read_file_str(obj_path)
if mtl_path == "":
mtl_path = search_mtl_path(obj_path)
var obj := get_data(obj_path)
var mats := {}
var mtl_filename: String = _get_mtl_filename(obj_str)
mtl_path = "%s/%s" % [obj_path.get_base_dir(), mtl_filename]
var mats: Dictionary = {}
if mtl_path != "":
mats = _create_mtl(get_data(mtl_path), get_mtl_tex(mtl_path))
return _create_obj(obj, mats) if obj and mats else null
mats = _create_mtl(_read_file_str(mtl_path), _get_mtl_tex(mtl_path))
if obj_str.is_empty():
return null
return _create_obj(obj_str, mats)
## Create mesh from obj, materials. Materials should be {"matname":data}
# Create mesh from obj, materials. Materials should be { "matname": data }
static func load_obj_from_buffer(obj_data: String, materials: Dictionary) -> Mesh:
return _create_obj(obj_data, materials)
## Create materials
# Create materials
static func load_mtl_from_buffer(mtl_data: String, textures: Dictionary) -> Dictionary:
return _create_mtl(mtl_data, textures)
## Get data from file path
static func get_data(path: String) -> String:
if path != "":
var file := FileAccess.open(path, FileAccess.READ)
if FileAccess.get_open_error() == OK:
var res := file.get_as_text()
file.close()
return res
return ""
# Get data from file path
static func _read_file_str(path: String) -> String:
if path == "":
return ""
var file: FileAccess = FileAccess.open(path, FileAccess.READ)
if file == null:
return ""
return file.get_as_text()
## Get textures from mtl path (return {"tex_path":data})
static func get_mtl_tex(mtl_path: String) -> Dictionary:
var file_paths := get_mtl_tex_paths(mtl_path)
var textures := {}
# Internal functions
# Get textures from mtl path (returns { "tex_path": data })
static func _get_mtl_tex(mtl_path: String) -> Dictionary:
var file_paths: Array[String] = _get_mtl_tex_paths(mtl_path)
var textures: Dictionary = {}
for k in file_paths:
textures[k] = _get_image(mtl_path, k).save_png_to_buffer()
return textures
## Get textures paths from mtl path
static func get_mtl_tex_paths(mtl_path: String) -> Array:
var file := FileAccess.open(mtl_path, FileAccess.READ)
var paths := []
if FileAccess.get_open_error() == OK:
var lines := file.get_as_text().split("\n", false)
file.close()
for line in lines:
var parts := line.split(" ", false, 1)
if parts[0] in ["map_Kd", "map_Ks", "map_Ka"]:
if !parts[1] in paths:
paths.push_back(parts[1])
# Get textures paths from mtl path
static func _get_mtl_tex_paths(mtl_path: String) -> Array[String]:
var file: FileAccess = FileAccess.open(mtl_path, FileAccess.READ)
if file == null:
return []
var paths: Array[String] = []
var lines: PackedStringArray = file.get_as_text().split("\n", false)
for line in lines:
var parts: PackedStringArray = line.split(" ", false, 1)
if ["map_Kd", "map_Ks", "map_Ka"].has(parts[0]):
if !paths.has(parts[1]):
paths.push_back(parts[1])
return paths
## Try to find mtl path from obj path
static func search_mtl_path(obj_path: String) -> String:
var mtl_path := obj_path.get_base_dir().path_join(
obj_path.get_file().rsplit(".", false, 1)[0] + ".mtl"
)
if !FileAccess.file_exists(mtl_path):
mtl_path = obj_path.get_base_dir().path_join(obj_path.get_file() + ".mtl")
if !FileAccess.file_exists(mtl_path):
return ""
return mtl_path
# Private methods
static func _get_mtl_filename(obj: String) -> String:
var lines: PackedStringArray = obj.split("\n")
for line in lines:
var split: PackedStringArray = line.split(" ", false)
if split.size() < 2:
continue
if split[0] != "mtllib":
continue
return split[1].strip_edges()
return ""
static func _create_mtl(obj: String, textures: Dictionary) -> Dictionary:
var mats := {}
var mats: Dictionary = {}
var current_mat: StandardMaterial3D = null
var lines := obj.split("\n", false)
var lines: PackedStringArray = obj.split("\n", false)
for line in lines:
var parts := line.split(" ", false)
var parts: PackedStringArray = line.split(" ", false)
match parts[0]:
"#":
# Comment
#print("Comment: "+line)
pass
"newmtl":
# Create a new material
if DEBUG:
print("Adding new material " + parts[1])
prints("Adding new material", parts[1])
current_mat = StandardMaterial3D.new()
mats[parts[1]] = current_mat
"Ka":
@ -108,117 +114,134 @@ static func _create_mtl(obj: String, textures: Dictionary) -> Dictionary:
pass
"Kd":
# Diffuse color
current_mat.albedo_color = Color(float(parts[1]), float(parts[2]), float(parts[3]))
current_mat.albedo_color = Color(
parts[1].to_float(), parts[2].to_float(), parts[3].to_float()
)
if DEBUG:
print("Setting material color " + str(current_mat.albedo_color))
prints("Setting material color", str(current_mat.albedo_color))
_:
if parts[0] in ["map_Kd", "map_Ks", "map_Ka"]:
var path := line.split(" ", false, 1)[1]
var path: String = line.split(" ", false, 1)[1]
if textures.has(path):
current_mat.albedo_texture = _create_texture(textures[path])
return mats
static func _parse_mtl_file(path) -> Dictionary:
return _create_mtl(_read_file_str(path), _get_mtl_tex(path))
static func _get_image(mtl_filepath: String, tex_filename: String) -> Image:
if DEBUG:
print(" Debug: Mapping texture file " + tex_filename)
var texfilepath := tex_filename
prints("Debug: Mapping texture file", tex_filename)
var tex_filepath: String = tex_filename
if tex_filename.is_relative_path():
texfilepath = mtl_filepath.get_base_dir().path_join(tex_filename)
var filetype := texfilepath.get_extension()
tex_filepath = "%s/%s" % [mtl_filepath.get_base_dir(), tex_filename]
var file_type: String = tex_filepath.get_extension()
if DEBUG:
print(" Debug: texture file path: " + texfilepath + " of type " + filetype)
prints("Debug: texture file path:", tex_filepath, "of type", file_type)
var img := Image.new()
img.load(texfilepath)
var img: Image = Image.new()
img.load(tex_filepath)
return img
static func _create_texture(data: PackedByteArray) -> ImageTexture:
var img := Image.new()
var img: Image = Image.new()
img.load_png_from_buffer(data)
return ImageTexture.create_from_image(img)
static func _get_texture(mtl_filepath, tex_filename) -> ImageTexture:
var tex = ImageTexture.create_from_image(_get_image(mtl_filepath, tex_filename))
if DEBUG:
prints("Debug: texture is", str(tex))
return tex
static func _create_obj(obj: String, mats: Dictionary) -> Mesh:
# Setup
var mesh := ArrayMesh.new()
var vertices := PackedVector3Array()
var normals := PackedVector3Array()
var uvs := PackedVector2Array()
var faces := {}
var mesh: ArrayMesh = ArrayMesh.new()
var vertices: PackedVector3Array = PackedVector3Array()
var normals: PackedVector3Array = PackedVector3Array()
var uvs: PackedVector2Array = PackedVector2Array()
var faces: Dictionary = {}
var mat_name := "default"
var count_mtl := 0
var mat_name: String = "default"
var count_mtl: int = 0
# Parse
var lines := obj.split("\n", false)
var lines: PackedStringArray = obj.split("\n", false)
for line in lines:
var parts := line.split(" ", false)
var parts: PackedStringArray = line.split(" ", false)
match parts[0]:
"#":
# Comment
#print("Comment: "+line)
pass
"v":
# Vertex
var n_v := Vector3(float(parts[1]), float(parts[2]), float(parts[3]))
# Vertice
var n_v: Vector3 = Vector3(
parts[1].to_float(), parts[2].to_float(), parts[3].to_float()
)
vertices.append(n_v)
"vn":
# Normal
var n_vn := Vector3(float(parts[1]), float(parts[2]), float(parts[3]))
var n_vn: Vector3 = Vector3(
parts[1].to_float(), parts[2].to_float(), parts[3].to_float()
)
normals.append(n_vn)
"vt":
# UV
var n_uv := Vector2(float(parts[1]), 1 - float(parts[2]))
var n_uv: Vector2 = Vector2(parts[1].to_float(), 1 - parts[2].to_float())
uvs.append(n_uv)
"usemtl":
# Material group
count_mtl += 1
mat_name = parts[1]
if not faces.has(mat_name):
var mats_keys := mats.keys()
mat_name = parts[1].strip_edges()
if !faces.has(mat_name):
var mats_keys: Array = mats.keys()
if !mats.has(mat_name):
if mats_keys.size() > count_mtl:
mat_name = mats_keys[count_mtl]
faces[mat_name] = []
"f":
if not faces.has(mat_name):
var mats_keys := mats.keys()
if !faces.has(mat_name):
var mats_keys: Array = mats.keys()
if mats_keys.size() > count_mtl:
mat_name = mats_keys[count_mtl]
faces[mat_name] = []
# Face
if parts.size() == 4:
# Tri
var face := {"v": [], "vt": [], "vn": []}
var face: Dictionary = {"v": [], "vt": [], "vn": []}
for map in parts:
var vertices_index := map.split("/")
if str(vertices_index[0]) != "f":
face["v"].append(int(vertices_index[0]) - 1)
face["vt"].append(int(vertices_index[1]) - 1)
if vertices_index.size() > 2:
face["vn"].append(int(vertices_index[2]) - 1)
var vertices_index: PackedStringArray = map.split("/")
if vertices_index[0] != "f":
face["v"].append(vertices_index[0].to_int() - 1)
if vertices_index.size() > 1:
face["vt"].append(vertices_index[1].to_int() - 1)
if vertices_index.size() > 2:
face["vn"].append(vertices_index[2].to_int() - 1)
if faces.has(mat_name):
faces[mat_name].append(face)
elif parts.size() > 4:
# Triangulate
var points: Array[PackedInt64Array] = []
var points: Array[Array] = []
for map in parts:
var vertices_index = map.split("/")
if str(vertices_index[0]) != "f":
var point: PackedInt64Array = []
point.append(int(vertices_index[0]) - 1)
point.append(int(vertices_index[1]) - 1)
var vertices_index: PackedStringArray = map.split("/")
if vertices_index[0] != "f":
var point: Array[int] = []
point.append(vertices_index[0].to_int() - 1)
point.append(vertices_index[1].to_int() - 1)
if vertices_index.size() > 2:
point.append(int(vertices_index[2]) - 1)
point.append(vertices_index[2].to_int() - 1)
points.append(point)
for i in points.size():
if i != 0:
var face = {"v": [], "vt": [], "vn": []}
var point0 := points[0]
var point1 := points[i]
var point2 := points[i - 1]
var point0: Array[int] = points[0]
var point1: Array[int] = points[i]
var point2: Array[int] = points[i - 1]
face["v"].append(point0[0])
face["v"].append(point2[0])
face["v"].append(point1[0])
@ -236,18 +259,16 @@ static func _create_obj(obj: String, mats: Dictionary) -> Mesh:
# Make tri
for matgroup in faces.keys():
if DEBUG:
print(
(
"Creating surface for matgroup "
+ matgroup
+ " with "
+ str(faces[matgroup].size())
+ " faces"
)
prints(
"Creating surface for matgroup",
matgroup,
"with",
str(faces[matgroup].size()),
"faces"
)
# Mesh Assembler
var st := SurfaceTool.new()
var st: SurfaceTool = SurfaceTool.new()
st.begin(Mesh.PRIMITIVE_TRIANGLES)
if !mats.has(matgroup):
mats[matgroup] = StandardMaterial3D.new()
@ -255,37 +276,38 @@ static func _create_obj(obj: String, mats: Dictionary) -> Mesh:
for face in faces[matgroup]:
if face["v"].size() == 3:
# Vertices
var fan_v := PackedVector3Array()
var fan_v: PackedVector3Array = PackedVector3Array()
fan_v.append(vertices[face["v"][0]])
fan_v.append(vertices[face["v"][2]])
fan_v.append(vertices[face["v"][1]])
# Normals
var fan_vn := PackedVector3Array()
var fan_vn: PackedVector3Array = PackedVector3Array()
if face["vn"].size() > 0:
fan_vn.append(normals[face["vn"][0]])
fan_vn.append(normals[face["vn"][2]])
fan_vn.append(normals[face["vn"][1]])
# Textures
var fan_vt := PackedVector2Array()
var fan_vt: PackedVector2Array = PackedVector2Array()
if face["vt"].size() > 0:
for k in [0, 2, 1]:
var f = face["vt"][k]
if f > -1:
var uv = uvs[f]
fan_vt.append(uv)
st.add_triangle_fan(
fan_v, fan_vt, PackedColorArray(), PackedVector2Array(), fan_vn, []
)
mesh = st.commit(mesh)
for k in mesh.get_surface_count():
var mat := mesh.surface_get_material(k)
var mat: Material = mesh.surface_get_material(k)
mat_name = ""
for m in mats:
if mats[m] == mat:
mat_name = m
mesh.surface_set_name(k, mat_name)
# Finish
return mesh