armory/Sources/armory/logicnode/AddTraitNode.hx
2021-09-03 23:37:13 +02:00

28 lines
781 B
Haxe

package armory.logicnode;
import iron.object.Object;
class AddTraitNode extends LogicNode {
public function new(tree: LogicTree) {
super(tree);
}
override function run(from: Int) {
var object: Object = inputs[1].get();
var traitName: String = inputs[2].get();
assert(Error, object != null, "Object should not be null");
assert(Error, traitName != null, "Trait name should not be null");
var cname = Type.resolveClass(Main.projectPackage + "." + traitName);
if (cname == null) cname = Type.resolveClass(Main.projectPackage + ".node." + traitName);
assert(Error, cname != null, 'No trait with the name "$traitName" found, make sure that the trait is exported!');
var trait = Type.createInstance(cname, []);
object.addTrait(trait);
runOutput(0);
}
}