static void CreateItemizedNode(State state, VSGraphModel graphModel, ref IPortModel outputPortModel)
{
ItemizeOptions currentItemizeOptions = state.Preferences.CurrentItemizeOptions;
// automatically itemize, i.e. duplicate variables as they get connected
if (outputPortModel.Connected && currentItemizeOptions != ItemizeOptions.Nothing)
{
var nodeToConnect = outputPortModel.NodeModel;
var offset = Vector2.up * k_NodeOffset;
if (currentItemizeOptions.HasFlag(ItemizeOptions.Constants) &&
nodeToConnect is ConstantNodeModel constantModel)
{
string newName = string.IsNullOrEmpty(constantModel.Title)
? "Temporary"
: constantModel.Title + "Copy";
nodeToConnect = graphModel.CreateConstantNode(
newName,
constantModel.Type.GenerateTypeHandle(graphModel.Stencil),
constantModel.Position + offset
);
((ConstantNodeModel)nodeToConnect).ObjectValue = constantModel.ObjectValue;
}
else if (currentItemizeOptions.HasFlag(ItemizeOptions.Variables) &&
nodeToConnect is VariableNodeModel variableModel)
{
nodeToConnect = graphModel.CreateVariableNode(variableModel.DeclarationModel,
variableModel.Position + offset);
}
else if (currentItemizeOptions.HasFlag(ItemizeOptions.Variables) &&
nodeToConnect is ThisNodeModel thisModel)
{
nodeToConnect = graphModel.CreateNode <ThisNodeModel>("this", thisModel.Position + offset);
}
else if (currentItemizeOptions.HasFlag(ItemizeOptions.SystemConstants) &&
nodeToConnect is SystemConstantNodeModel sysConstModel)
{
Action <SystemConstantNodeModel> preDefineSetup = m =>
{
m.ReturnType = sysConstModel.ReturnType;
m.DeclaringType = sysConstModel.DeclaringType;
m.Identifier = sysConstModel.Identifier;
};
nodeToConnect = graphModel.CreateNode(sysConstModel.Title, sysConstModel.Position + offset, SpawnFlags.Default, preDefineSetup);
}
outputPortModel = nodeToConnect.OutputsById[outputPortModel.UniqueId];
}
}
void PrepareGraph(VSGraphModel graphModel)
{
var entityTypeHandle = typeof(Entity).GenerateTypeHandle(graphModel.Stencil);
// Component creation
var scale = typeof(Scale).GenerateTypeHandle(Stencil);
var group = graphModel.CreateComponentQuery("g");
group.AddComponent(graphModel.Stencil, scale, ComponentDefinitionFlags.None);
// On update
var groupInstance = graphModel.CreateVariableNode(group, Vector2.zero);
m_OnUpdateEntities = graphModel.CreateNode <OnUpdateEntitiesNodeModel>("update", Vector2.zero);
graphModel.CreateEdge(m_OnUpdateEntities.InstancePort, groupInstance.OutputPort);
m_CreateEntityModel = m_OnUpdateEntities.CreateStackedNode <CreateEntityNodeModel>("instantiate");
// Variable containing the new entity
m_OnUpdateEntities.CreateFunctionVariableDeclaration("newEntity", entityTypeHandle);
m_NewEntityVariable = graphModel.CreateVariableNode(
m_OnUpdateEntities.FunctionParameterModels.Single(p => p.DataType == entityTypeHandle),
Vector2.zero);
m_FloatConstantNode = (FloatConstantModel)graphModel.CreateConstantNode("float", TypeHandle.Float, Vector2.zero);
m_FloatConstantNode.value = 10f;
graphModel.CreateEdge(m_CreateEntityModel.InstancePort, m_NewEntityVariable.OutputPort);
}
public void InitBasicGraph(VSGraphModel graphModel)
{
var query = graphModel.CreateComponentQuery("myQuery");
var node = graphModel.CreateNode <OnUpdateEntitiesNodeModel>("On Update Entities", Vector2.zero);
var queryInstance = graphModel.CreateVariableNode(query, new Vector2(-145, 8));
graphModel.CreateEdge(node.InstancePort, queryInstance.OutputPort);
}
public void InitBasicGraph(VSGraphModel graphModel)
{
var query = graphModel.CreateQueryFromGameObject(GameObject);
var groupInstance = graphModel.CreateVariableNode(query, m_Position);
var node = graphModel.CreateNode <OnUpdateEntitiesNodeModel>("update", m_Position + k_GroupOffset);
graphModel.CreateEdge(node.InstancePort, groupInstance.OutputPort);
}
public void PortModelsCanHaveDefaultValues(object expectedValue, Func <TestNodeModelWithCustomPorts, Func <IPortModel> > createPort, Func <ConstantNodeModel, object> getValue)
{
VSGraphAssetModel asset = ScriptableObject.CreateInstance <VSGraphAssetModel>();
VSGraphModel g = asset.CreateVSGraph <ClassStencil>("asd");
m_Node = g.CreateNode <TestNodeModelWithCustomPorts>("test", Vector2.zero, preDefineSetup: ports => ports.CreatePortFunc = createPort(ports));
Assert.That(getValue(m_Node.InputsByDisplayOrder.Single().EmbeddedValue), Is.EqualTo(expectedValue));
}
static OnUpdateEntitiesNodeModel CreateOnUpdateAndConnectGroup(VSGraphModel graphModel, IVariableDeclarationModel group)
{
var groupInstance = graphModel.CreateVariableNode(group, Vector2.zero);
var onUpdateEntities = graphModel.CreateNode <OnUpdateEntitiesNodeModel>("update", Vector2.zero);
graphModel.CreateEdge(onUpdateEntities.InstancePort, groupInstance.OutputPort);
return(onUpdateEntities);
}
static OnUpdateEntitiesNodeModel CreateOnUpdateAndConnectQuery(VSGraphModel graphModel, IVariableDeclarationModel query)
{
var queryInstance = graphModel.CreateVariableNode(query, Vector2.zero);
var onUpdateEntities = graphModel.CreateNode <OnUpdateEntitiesNodeModel>("update", Vector2.zero);
graphModel.CreateEdge(onUpdateEntities.InstancePort, queryInstance.OutputPort);
return(onUpdateEntities);
}
public static void CreateSubGraphReference(VSGraphModel graphModel, IEnumerable <VSGraphAssetModel> subgraphAssets, Vector2 position)
{
foreach (var subgraphModel in subgraphAssets)
{
var subgraphReference = graphModel.CreateNode <SubgraphReferenceNodeModel>(subgraphModel.Name, position,
SpawnFlags.Default,
v => { v.GraphReference = subgraphModel; });
position += new Vector2(20, 25);
}
}
public void CallingDefineTwiceCreatesOneEmbeddedConstant()
{
VSGraphAssetModel asset = ScriptableObject.CreateInstance <VSGraphAssetModel>();
VSGraphModel g = asset.CreateVSGraph <ClassStencil>("asd");
m_Node = g.CreateNode <TestNodeModel>("test", Vector2.zero);
Assert.That(m_Node.InputConstantsById.Count, Is.EqualTo(1));
m_Node.DefineNode();
Assert.That(m_Node.InputConstantsById.Count, Is.EqualTo(1));
}
static State DotsCreateGetSetVariableNodes(State prevState, DotsCreateGetSetVariableNodesAction action)
{
VSGraphModel vsGraphModel = ((VSGraphModel)prevState.CurrentGraphModel);
foreach (var tuple in action.VariablesToCreate)
{
vsGraphModel.CreateNode <SetVariableNodeModel>(tuple.Item1.Title, tuple.Item2, SpawnFlags.Default, v =>
{
v.DeclarationModel = tuple.Item1;
v.IsGetter = action.CreateGetters;
});
}
return(prevState);
}
static void CreateUpdateAndLogEntity(VSGraphModel graphModel, IVariableModel variable)
{
// update entities
var update = graphModel.CreateNode <OnUpdateEntitiesNodeModel>("update", Vector2.zero);
graphModel.CreateEdge(update.InstancePort, variable.OutputPort);
// Create entity from update
var entity = graphModel.CreateVariableNode(
update.FunctionParameterModels.Single(
p => p.DataType == typeof(Entity).GenerateTypeHandle(graphModel.Stencil)),
Vector2.zero);
// Log the entity
var log = update.CreateFunctionCallNode(typeof(Debug).GetMethod("Log", new[] { typeof(object) }), 0);
graphModel.CreateEdge(log.GetParameterPorts().First(), entity.OutputPort);
}
static AddComponentNodeModel CreateAddComponentInGraph(
VSGraphModel graphModel,
IVariableDeclarationModel group,
Type componentToAdd)
{
var groupInstance = graphModel.CreateVariableNode(group, Vector2.zero);
var onUpdateEntities = graphModel.CreateNode <OnUpdateEntitiesNodeModel>("update", Vector2.zero);
graphModel.CreateEdge(onUpdateEntities.InstancePort, groupInstance.OutputPort);
var entityInstance = graphModel.CreateVariableNode(
onUpdateEntities.FunctionParameterModels.Single(
p => p.DataType == typeof(Entity).GenerateTypeHandle(graphModel.Stencil)
),
Vector2.zero);
var addComponent = onUpdateEntities.CreateStackedNode <AddComponentNodeModel>("add");
addComponent.ComponentType = componentToAdd.GenerateTypeHandle(graphModel.Stencil);
graphModel.CreateEdge(addComponent.EntityPort, entityInstance.OutputPort);
return(addComponent);
}
static void CreateRemoveComponentInGraph(VSGraphModel graphModel, Type componentToRemove)
{
var component = componentToRemove.GenerateTypeHandle(graphModel.Stencil);
var query = graphModel.CreateComponentQuery("g");
query.AddComponent(graphModel.Stencil, component, ComponentDefinitionFlags.None);
var queryInstance = graphModel.CreateVariableNode(query, Vector2.zero);
var onUpdateEntities = graphModel.CreateNode <OnUpdateEntitiesNodeModel>("update", Vector2.zero);
graphModel.CreateEdge(onUpdateEntities.InstancePort, queryInstance.OutputPort);
var entityInstance = graphModel.CreateVariableNode(
onUpdateEntities.FunctionParameterModels.Single(
p => p.DataType == typeof(Entity).GenerateTypeHandle(graphModel.Stencil)
),
Vector2.zero);
var removeComponent = onUpdateEntities.CreateStackedNode <RemoveComponentNodeModel>("remove");
removeComponent.ComponentType = component;
graphModel.CreateEdge(removeComponent.EntityPort, entityInstance.OutputPort);
}
public static List <NodeModel> SpawnAllNodeModelsInGraph(VSGraphModel graphModel)
{
List <NodeModel> spawnedNodes = new List <NodeModel>();
Stencil stencil = graphModel.Stencil;
StackModel stack;
FunctionModel funcModel;
OnUpdateEntitiesNodeModel onUpdateModel;
//--Floating Nodes--
//Stack-Derived NodeModels
spawnedNodes.Add(stack = graphModel.CreateNode <StackModel>("StackModel"));
spawnedNodes.Add(funcModel = graphModel.CreateNode <FunctionModel>("FunctionModel"));
spawnedNodes.Add(onUpdateModel = graphModel.CreateNode <OnUpdateEntitiesNodeModel>("OnUpdateEntitiesNodeModel"));
var methodInfo = TypeSystem.GetMethod(typeof(Debug), nameof(Debug.Log), true);
spawnedNodes.Add(graphModel.CreateEventFunction(methodInfo, Vector2.zero));
spawnedNodes.Add(graphModel.CreateNode <OnEndEntitiesNodeModel>("OnEndEntitiesNodeModel"));
spawnedNodes.Add(graphModel.CreateNode <OnEventNodeModel>("OnEventNodeModel"));
spawnedNodes.Add(graphModel.CreateNode <OnStartEntitiesNodeModel>("OnStartEntitiesNodeModel"));
spawnedNodes.Add(graphModel.CreateNode <PostUpdate>("PostUpdate"));
spawnedNodes.Add(graphModel.CreateNode <PreUpdate>("PreUpdate"));
spawnedNodes.Add(graphModel.CreateNode <KeyDownEventModel>("KeyDownEventModel"));
spawnedNodes.Add(graphModel.CreateNode <CountEntitiesNodeModel>("CountEntitiesNodeModel"));
spawnedNodes.Add(graphModel.CreateLoopStack(typeof(ForEachHeaderModel), Vector2.zero));
spawnedNodes.Add(graphModel.CreateLoopStack(typeof(WhileHeaderModel), Vector2.zero));
spawnedNodes.Add(graphModel.CreateLoopStack(typeof(ForAllEntitiesStackModel), Vector2.zero));
spawnedNodes.Add(graphModel.CreateLoopStack(typeof(CoroutineStackModel), Vector2.zero));
//Constant-typed NodeModels
spawnedNodes.Add(graphModel.CreateNode <BooleanConstantNodeModel>("BooleanConstantNodeModel"));
spawnedNodes.Add(graphModel.CreateNode <ColorConstantModel>("ColorConstantModel"));
spawnedNodes.Add(graphModel.CreateNode <CurveConstantNodeModel>("CurveConstantNodeModel"));
spawnedNodes.Add(graphModel.CreateNode <DoubleConstantModel>("DoubleConstantModel"));
spawnedNodes.Add(graphModel.CreateNode <EnumConstantNodeModel>("EnumConstantNodeModel"));
spawnedNodes.Add(graphModel.CreateNode <FloatConstantModel>("FloatConstantModel"));
spawnedNodes.Add(graphModel.CreateNode <GetPropertyGroupNodeModel>("GetPropertyGroupNodeModel"));
spawnedNodes.Add(graphModel.CreateNode <InputConstantModel>("InputConstantModel"));
spawnedNodes.Add(graphModel.CreateNode <IntConstantModel>("IntConstantModel"));
spawnedNodes.Add(graphModel.CreateNode <LayerConstantModel>("LayerConstantModel"));
spawnedNodes.Add(graphModel.CreateNode <LayerMaskConstantModel>("LayerMaskConstantModel"));
spawnedNodes.Add(graphModel.CreateNode <ObjectConstantModel>("ObjectConstantModel"));
spawnedNodes.Add(graphModel.CreateNode <QuaternionConstantModel>("QuaternionConstantModel"));
spawnedNodes.Add(graphModel.CreateNode <StringConstantModel>("StringConstantModel"));
spawnedNodes.Add(graphModel.CreateNode <TagConstantModel>("TagConstantModel"));
spawnedNodes.Add(graphModel.CreateNode <Vector2ConstantModel>("Vector2ConstantModel"));
spawnedNodes.Add(graphModel.CreateNode <Vector3ConstantModel>("Vector3ConstantModel"));
spawnedNodes.Add(graphModel.CreateNode <Vector4ConstantModel>("Vector4ConstantModel"));
spawnedNodes.Add(graphModel.CreateNode <ConstantSceneAssetNodeModel>("ConstantSceneAssetNodeModel"));
spawnedNodes.Add(graphModel.CreateNode <Float2ConstantModel>("Float2ConstantModel"));
spawnedNodes.Add(graphModel.CreateNode <Float3ConstantModel>("Float3ConstantModel"));
spawnedNodes.Add(graphModel.CreateNode <Float4ConstantModel>("Float4ConstantModel"));
//Misc
void DefineSystemConstant(SystemConstantNodeModel m)
{
m.ReturnType = typeof(float).GenerateTypeHandle(stencil);
m.DeclaringType = typeof(Mathf).GenerateTypeHandle(stencil);
m.Identifier = "PI";
}
spawnedNodes.Add(graphModel.CreateNode <SystemConstantNodeModel>("SystemConstantNodeModel", Vector2.zero, SpawnFlags.Default, DefineSystemConstant));
spawnedNodes.Add(graphModel.CreateNode <GetInputNodeModel>("GetInputNodeModel"));
spawnedNodes.Add(graphModel.CreateNode <GetOrCreateComponentNodeModel>("GetOrCreateComponentNodeModel"));
spawnedNodes.Add(graphModel.CreateNode <GetSingletonNodeModel>("GetSingletonNodeModel"));
spawnedNodes.Add(graphModel.CreateNode <ThisNodeModel>("ThisNodeModel"));
VariableDeclarationModel decl = graphModel.CreateGraphVariableDeclaration("MyVariableName", typeof(int).GenerateTypeHandle(graphModel.Stencil), true);
spawnedNodes.Add((NodeModel)graphModel.CreateVariableNode(decl, Vector2.zero));
spawnedNodes.Add(graphModel.CreateNode <MacroRefNodeModel>("MacroRefNodeModel"));
spawnedNodes.Add(graphModel.CreateInlineExpressionNode("2+2", Vector2.zero));
spawnedNodes.Add(graphModel.CreateBinaryOperatorNode(BinaryOperatorKind.Add, Vector2.zero));
spawnedNodes.Add(graphModel.CreateUnaryOperatorNode(UnaryOperatorKind.PostIncrement, Vector2.zero));
//--Stack-Contained Nodes--
spawnedNodes.Add(stack.CreateStackedNode <AddComponentNodeModel>());
spawnedNodes.Add(stack.CreateStackedNode <DestroyEntityNodeModel>());
spawnedNodes.Add(stack.CreateStackedNode <ForAllEntitiesNodeModel>());
spawnedNodes.Add(stack.CreateStackedNode <ForEachNodeModel>());
spawnedNodes.Add(stack.CreateFunctionCallNode(TypeSystem.GetMethod(typeof(Debug), nameof(Debug.Log), true)));
spawnedNodes.Add(stack.CreateFunctionRefCallNode(funcModel));
spawnedNodes.Add(stack.CreateStackedNode <InstantiateNodeModel>());
spawnedNodes.Add(stack.CreateStackedNode <CreateEntityNodeModel>());
spawnedNodes.Add(stack.CreateStackedNode <IfConditionNodeModel>());
spawnedNodes.Add(stack.CreateStackedNode <LogNodeModel>());
spawnedNodes.Add(stack.CreateStackedNode <RemoveComponentNodeModel>());
spawnedNodes.Add(stack.CreateStackedNode <SetComponentNodeModel>());
spawnedNodes.Add(stack.CreateStackedNode <SetPositionNodeModel>());
spawnedNodes.Add(stack.CreateStackedNode <SetRotationNodeModel>());
spawnedNodes.Add(stack.CreateStackedNode <WhileNodeModel>());
spawnedNodes.Add(stack.CreateStackedNode <SetPropertyGroupNodeModel>());
spawnedNodes.Add(stack.CreateStackedNode <SetVariableNodeModel>());
spawnedNodes.Add(stack.CreateStackedNode <CoroutineNodeModel>());
spawnedNodes.Add(funcModel.CreateStackedNode <ReturnNodeModel>());
TypeHandle eventTypeHandle = typeof(DummyEvent).GenerateTypeHandle(stencil);
spawnedNodes.Add(onUpdateModel.CreateStackedNode <SendEventNodeModel>("SendEventNodeModel", 0, SpawnFlags.Default, n => n.EventType = eventTypeHandle));
return(spawnedNodes);
}
