quack/addons/cyclops_level_builder/commands/cmd_set_material.gd
2023-05-24 00:27:34 +03:00

154 lines
4.8 KiB
GDScript

# MIT License
#
# Copyright (c) 2023 Mark McKay
# https://github.com/blackears/cyclopsLevelBuilder
#
# Permission is hereby granted, free of charge, to any person obtaining a copy
# of this software and associated documentation files (the "Software"), to deal
# in the Software without restriction, including without limitation the rights
# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
# copies of the Software, and to permit persons to whom the Software is
# furnished to do so, subject to the following conditions:
#
# The above copyright notice and this permission notice shall be included in all
# copies or substantial portions of the Software.
#
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
# SOFTWARE.
@tool
class_name CommandSetMaterial
extends CyclopsCommand
class Target extends RefCounted:
var block_path:NodePath
var face_indices:PackedInt32Array
class BlockCache extends RefCounted:
var path:NodePath
var data:ConvexBlockData
var materials:Array[Material]
#Public
var material_path:String
#Private
var target_list:Array[Target] = []
var cache_list:Array[BlockCache] = []
func add_target(block_path:NodePath, face_indices:PackedInt32Array):
# print("add target %s %s" % [block_path.get_name(block_path.get_name_count() - 1), face_indices])
var target:Target = Target.new()
target.block_path = block_path
target.face_indices = face_indices
target_list.append(target)
func make_cache():
cache_list = []
for t in target_list:
var cache:BlockCache = BlockCache.new()
var block:CyclopsConvexBlock = builder.get_node(t.block_path)
cache.path = block.get_path()
cache.data = block.block_data
cache.materials = block.materials.duplicate()
cache_list.append(cache)
func will_change_anything()->bool:
for t in target_list:
if !t.face_indices.is_empty():
return true
return false
func _init():
command_name = "Set material"
func do_it():
make_cache()
# print("cmd set material %s" % material_path)
for t in target_list:
var block:CyclopsConvexBlock = builder.get_node(t.block_path)
var data:ConvexBlockData = block.block_data
var vol:ConvexVolume = ConvexVolume.new()
vol.init_from_convex_block_data(data)
var mat_list:Array[Material] = block.materials.duplicate()
# print("start mat list")
# for m in mat_list:
# print("cur mat %s" % "?" if m == null else m.resource_path)
var target_material:Material
for m in mat_list:
if m.resource_path == material_path:
target_material = m
break
if !target_material:
target_material = load(material_path)
mat_list.append(target_material)
# print("target mat list")
# for m in mat_list:
# print("mat %s" % "?" if m == null else m.resource_path)
var remap_face_idx_to_mat:Array[Material] = []
var ctl_mesh:ConvexVolume = ConvexVolume.new()
ctl_mesh.init_from_convex_block_data(block.control_mesh.to_convex_block_data())
for f_idx in ctl_mesh.faces.size():
var f:ConvexVolume.FaceInfo = ctl_mesh.faces[f_idx]
if t.face_indices.has(f_idx):
remap_face_idx_to_mat.append(target_material)
elif f.material_id >= 0 && f.material_id < block.materials.size():
remap_face_idx_to_mat.append(block.materials[f.material_id])
else:
remap_face_idx_to_mat.append(null)
# print("remap faceidx to mat")
# for m in remap_face_idx_to_mat:
# print("mat %s" % "?" if m == null else m.resource_path)
#Reduce material list, discarding unused materials
var mat_list_reduced:Array[Material]
for m in remap_face_idx_to_mat:
if m != null && !mat_list_reduced.has(m):
mat_list_reduced.append(m)
# print("mat_list_reduced")
# for m in mat_list_reduced:
# print("mat %s" % "?" if m == null else m.resource_path)
#Set new face materials using new material ids
for f_idx in remap_face_idx_to_mat.size():
#print("face_idx %s" % f_idx)
var face:ConvexVolume.FaceInfo = ctl_mesh.faces[f_idx]
var mat = remap_face_idx_to_mat[f_idx]
#print("mat %s" % "?" if mat == null else mat.resource_path)
#print("has %s" % mat_list_reduced.has(mat))
#print("find %s" % mat_list_reduced.find(mat))
face.material_id = -1 if mat == null else mat_list_reduced.find(mat)
#print("face.material_id %s" % face.material_id)
block.materials = mat_list_reduced
block.block_data = ctl_mesh.to_convex_block_data()
func undo_it():
for cache in cache_list:
var block:CyclopsConvexBlock = builder.get_node(cache.path)
block.materials = cache.materials.duplicate()
block.block_data = cache.data