2017-04-14 20:38:50 +02:00
|
|
|
package armory.logicnode;
|
|
|
|
|
2017-11-20 15:16:52 +01:00
|
|
|
import iron.object.Object;
|
2017-04-14 20:38:50 +02:00
|
|
|
|
|
|
|
class AddTraitNode extends LogicNode {
|
|
|
|
|
2019-12-19 23:54:08 +01:00
|
|
|
public function new(tree: LogicTree) {
|
2017-04-14 20:38:50 +02:00
|
|
|
super(tree);
|
|
|
|
}
|
|
|
|
|
2019-12-19 23:54:08 +01:00
|
|
|
override function run(from: Int) {
|
|
|
|
var object: Object = inputs[1].get();
|
2021-09-03 23:37:13 +02:00
|
|
|
var traitName: String = inputs[2].get();
|
2019-12-19 23:54:08 +01:00
|
|
|
|
2021-09-03 23:37:13 +02:00
|
|
|
assert(Error, object != null, "Object should not be null");
|
|
|
|
assert(Error, traitName != null, "Trait name should not be null");
|
2017-04-14 20:38:50 +02:00
|
|
|
|
2021-09-03 23:37:13 +02:00
|
|
|
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!');
|
2021-09-03 23:42:43 +02:00
|
|
|
assert(Warning, object.getTrait(cname) == null, 'Object already has the trait "$traitName" applied');
|
2021-09-03 23:37:13 +02:00
|
|
|
|
|
|
|
var trait = Type.createInstance(cname, []);
|
2017-04-14 20:38:50 +02:00
|
|
|
object.addTrait(trait);
|
|
|
|
|
2018-10-22 10:10:33 +02:00
|
|
|
runOutput(0);
|
2017-04-14 20:38:50 +02:00
|
|
|
}
|
|
|
|
}
|