public static IGraphElement CreateSetVar(this ElementBuilder elementBuilder, IStore store, SetVariableNodeModel model) { if (model.DeclarationModel.IsObjectReference()) { return(elementBuilder.CreateToken(store, model)); } if (!model.IsGetter) { var ui = new SetVariableNode(); ui.AddToClassList(Unity.Modifier.GraphElements.Node.k_UssClassName + "--setvar-node"); ui.Setup(model, store, elementBuilder.GraphView); return(ui); } var token = new Token(); token.AddToClassList("dots-variable-token"); token.Setup(model, store, elementBuilder.GraphView); return(token); }