private void SelectCopiedElements(VFXView view, VFXGroupNodeController groupNode)
{
view.ClearSelection();
var elements = view.graphElements.ToList();
newNodesUI.Clear();
newContextUIs.Clear();
FindContextUIsAndSelect(view, elements);
FindOperatorsUIsAndSelect(view, elements);
FindParameterUIsAndSelect(view, elements);
SelectEdges(view, elements);
//Select all groups that are new
SelectGroupNodes(view, elements);
// Add all copied element that are not in a copied groupNode to the potentially selected groupnode
if (groupNode != null)
{
foreach (var newSlotContainerUI in newNodesUI)
{
groupNode.AddNode(newSlotContainerUI.controller);
}
}
SelectStickyNotes(view, elements);
}
static VFXUI CopyGroupNodesAndStickyNotes(IEnumerable <Controller> elements, VFXContext[] copiedContexts, VFXModel[] copiedSlotContainers)
{
VFXGroupNodeController[] groupNodes = elements.OfType <VFXGroupNodeController>().ToArray();
VFXStickyNoteController[] stickyNotes = elements.OfType <VFXStickyNoteController>().ToArray();
VFXUI copiedGroupUI = null;
if (groupNodes.Length > 0 || stickyNotes.Length > 0)
{
copiedGroupUI = ScriptableObject.CreateInstance <VFXUI>();
var stickyNodeIndexToCopiedIndex = new Dictionary <int, int>();
if (stickyNotes.Length > 0)
{
copiedGroupUI.stickyNoteInfos = new VFXUI.StickyNoteInfo[stickyNotes.Length];
for (int i = 0; i < stickyNotes.Length; ++i)
{
VFXStickyNoteController stickyNote = stickyNotes[i];
stickyNodeIndexToCopiedIndex[stickyNote.index] = i;
VFXUI.StickyNoteInfo info = stickyNote.model.stickyNoteInfos[stickyNote.index];
copiedGroupUI.stickyNoteInfos[i] = new VFXUI.StickyNoteInfo(info);
}
}
if (groupNodes.Length > 0)
{
copiedGroupUI.groupInfos = new VFXUI.GroupInfo[groupNodes.Length];
for (int i = 0; i < groupNodes.Length; ++i)
{
VFXGroupNodeController groupNode = groupNodes[i];
VFXUI.GroupInfo info = groupNode.model.groupInfos[groupNode.index];
copiedGroupUI.groupInfos[i] = new VFXUI.GroupInfo(info);
// only keep nodes and sticky notes that are copied because a element can not be in two groups at the same time.
if (info.contents != null)
{
var groupInfo = copiedGroupUI.groupInfos[i];
groupInfo.contents = info.contents.Where(t => copiedContexts.Contains(t.model) || copiedSlotContainers.Contains(t.model) || (t.isStickyNote && stickyNodeIndexToCopiedIndex.ContainsKey(t.id))).ToArray();
for (int j = 0; j < groupInfo.contents.Length; ++j)
{
if (groupInfo.contents[j].isStickyNote)
{
groupInfo.contents[j].id = stickyNodeIndexToCopiedIndex[groupInfo.contents[j].id];
}
}
}
}
}
}
return(copiedGroupUI);
}
void CopyGroupNodesAndStickyNotes(ref SerializableGraph serializableGraph, IEnumerable <Controller> elements)
{
VFXGroupNodeController[] groupNodes = elements.OfType <VFXGroupNodeController>().ToArray();
VFXStickyNoteController[] stickyNotes = elements.OfType <VFXStickyNoteController>().ToArray();
if (groupNodes.Length > 0 || stickyNotes.Length > 0)
{
var stickyNodeIndexToCopiedIndex = new Dictionary <int, int>();
if (stickyNotes.Length > 0)
{
serializableGraph.stickyNotes = new VFXUI.StickyNoteInfo[stickyNotes.Length];
for (int i = 0; i < stickyNotes.Length; ++i)
{
VFXStickyNoteController stickyNote = stickyNotes[i];
stickyNodeIndexToCopiedIndex[stickyNote.index] = i;
VFXUI.StickyNoteInfo info = stickyNote.model.stickyNoteInfos[stickyNote.index];
serializableGraph.stickyNotes[i] = new VFXUI.StickyNoteInfo(info);
}
}
if (groupNodes.Length > 0)
{
serializableGraph.groupNodes = new GroupNode[groupNodes.Length];
for (int i = 0; i < groupNodes.Length; ++i)
{
VFXGroupNodeController groupNode = groupNodes[i];
VFXUI.GroupInfo info = groupNode.model.groupInfos[groupNode.index];
serializableGraph.groupNodes[i] = new GroupNode {
infos = new VFXUI.UIInfo(info)
};
// only keep nodes and sticky notes that are copied because a element can not be in two groups at the same time.
if (info.contents != null)
{
var nodeIndices = groupNode.nodes.OfType <VFXNodeController>().Where(t => contexts.Contains(t) || operators.Contains(t) || parameters.Contains(t)).Select(t => modelIndices[t]);
var stickNoteIndices = info.contents.Where(t => t.isStickyNote && stickyNodeIndexToCopiedIndex.ContainsKey(t.id)).Select(t => (uint)stickyNodeIndexToCopiedIndex[t.id]);
serializableGraph.groupNodes[i].contents = nodeIndices.Concat(stickNoteIndices).ToArray();
serializableGraph.groupNodes[i].stickNodeCount = stickNoteIndices.Count();
}
}
}
}
}
void DoPaste(VFXViewController viewController, Vector2 center, object data, VFXView view, VFXGroupNodeController groupNode, List <VFXNodeController> nodesInTheSameOrder)
{
SerializableGraph serializableGraph = (SerializableGraph)data;
if (serializableGraph.blocksOnly)
{
if (view != null)
{
PasteBlocks(view, ref serializableGraph);
}
}
else
{
PasteAll(viewController, center, ref serializableGraph, view, groupNode, nodesInTheSameOrder);
}
}
public static void Paste(VFXViewController viewController, Vector2 center, object data, VFXView view, VFXGroupNodeController groupNode, List <VFXNodeController> nodesInTheSameOrder = null)
{
if (s_Instance == null)
{
s_Instance = new VFXPaste();
}
s_Instance.DoPaste(viewController, center, data, view, groupNode, nodesInTheSameOrder);
}
public static void UnserializeAndPasteElements(VFXViewController viewController, Vector2 center, string data, VFXView view = null, VFXGroupNodeController groupNode = null)
{
var serializableGraph = JsonUtility.FromJson <SerializableGraph>(data);
if (s_Instance == null)
{
s_Instance = new VFXPaste();
}
s_Instance.DoPaste(viewController, center, serializableGraph, view, groupNode, null);
}
void PasteAll(VFXViewController viewController, Vector2 center, ref SerializableGraph serializableGraph, VFXView view, VFXGroupNodeController groupNode, List <VFXNodeController> nodesInTheSameOrder)
{
newControllers.Clear();
m_NodesInTheSameOrder = new VFXNodeID[serializableGraph.controllerCount];
var graph = viewController.graph;
pasteOffset = (serializableGraph.bounds.width > 0 && serializableGraph.bounds.height > 0) ? center - serializableGraph.bounds.center : Vector2.zero;
MakePasteOffsetUnique(viewController, serializableGraph);
// Paste all nodes
PasteContexts(viewController, ref serializableGraph);
PasteOperators(viewController, ref serializableGraph);
PasteParameters(viewController, ref serializableGraph);
// Create controllers for all new nodes
viewController.LightApplyChanges();
// Register all nodes for usage in groupNodes and edges
RegisterContexts(viewController);
RegisterOperators(viewController);
RegisterParameterNodes(viewController);
VFXUI ui = viewController.graph.UIInfos;
firstCopiedGroup = -1;
firstCopiedStickyNote = ui.stickyNoteInfos != null ? ui.stickyNoteInfos.Length : 0;
//Paste Everything else
PasteGroupNodes(ref serializableGraph, ui);
PasteStickyNotes(ref serializableGraph, ui);
PasteDatas(ref serializableGraph); // TODO Data settings should be pasted at context creation. This can lead to issues as blocks are added before data is initialized
PasteDataEdges(ref serializableGraph);
PasteFlowEdges(ref serializableGraph);
// Create all ui based on model
viewController.LightApplyChanges();
if (nodesInTheSameOrder != null)
{
nodesInTheSameOrder.Clear();
nodesInTheSameOrder.AddRange(m_NodesInTheSameOrder.Select(t => t.model == null ? null : viewController.GetNodeController(t.model, t.id)));
}
if (view != null)
{
SelectCopiedElements(view, groupNode);
}
}
static void PasteNodes(VFXViewController viewController, Vector2 center, Data copyData, ScriptableObject[] allSerializedObjects, VFXView view, VFXGroupNodeController groupNode)
{
var graph = viewController.graph;
Vector2 pasteOffset = (copyData.bounds.width > 0 && copyData.bounds.height > 0) ? center - copyData.bounds.center : Vector2.zero;
// look if pasting there will result in the first element beeing exactly on top of other
while (true)
{
bool foundSamePosition = false;
if (copyData.contexts != null && copyData.contexts.Length > 0)
{
VFXContext firstContext = copyData.contexts[0];
foreach (var existingContext in viewController.graph.children.OfType <VFXContext>())
{
if ((firstContext.position + pasteOffset - existingContext.position).sqrMagnitude < 1)
{
foundSamePosition = true;
break;
}
}
}
else if (copyData.slotContainers != null && copyData.slotContainers.Length > 0)
{
VFXModel firstContainer = copyData.slotContainers[0];
foreach (var existingSlotContainer in viewController.graph.children.Where(t => t is IVFXSlotContainer))
{
if ((firstContainer.position + pasteOffset - existingSlotContainer.position).sqrMagnitude < 1)
{
foundSamePosition = true;
break;
}
}
}
else
{
VFXUI ui = allSerializedObjects.OfType <VFXUI>().First();
if (ui != null)
{
if (ui.stickyNoteInfos != null && ui.stickyNoteInfos.Length > 0)
{
foreach (var stickyNote in viewController.stickyNotes)
{
if ((ui.stickyNoteInfos[0].position.position + pasteOffset - stickyNote.position.position).sqrMagnitude < 1)
{
foundSamePosition = true;
break;
}
}
}
else if (ui.groupInfos != null && ui.groupInfos.Length > 0)
{
foreach (var gn in viewController.groupNodes)
{
if ((ui.groupInfos[0].position.position + pasteOffset - gn.position.position).sqrMagnitude < 1)
{
foundSamePosition = true;
break;
}
}
}
}
}
if (foundSamePosition)
{
pasteOffset += Vector2.one * 30;
}
else
{
break;
}
}
if (copyData.contexts != null)
{
foreach (var slotContainer in copyData.contexts)
{
var newContext = slotContainer;
newContext.position += pasteOffset;
ClearLinks(newContext);
}
}
if (copyData.slotContainers != null)
{
foreach (var slotContainer in copyData.slotContainers)
{
var newSlotContainer = slotContainer;
newSlotContainer.position += pasteOffset;
ClearLinks(newSlotContainer as IVFXSlotContainer);
}
}
for (int i = 0; i < allSerializedObjects.Length; ++i)
{
ScriptableObject obj = allSerializedObjects[i];
if (obj is VFXContext || obj is VFXOperator)
{
graph.AddChild(obj as VFXModel);
}
else if (obj is VFXParameter)
{
int paramIndex = System.Array.FindIndex(copyData.parameters, t => t.index == i);
VFXParameter existingParameter = graph.children.OfType <VFXParameter>().FirstOrDefault(t => t.GetInstanceID() == copyData.parameters[paramIndex].originalInstanceID);
if (existingParameter != null)
{
// The original parameter is from the current graph, add the nodes to the original
copyData.parameters[paramIndex].parameter = existingParameter;
copyData.parameters[paramIndex].copiedParameter = obj as VFXParameter;
copyData.parameters[paramIndex].infoIndexOffset = existingParameter.nodes.Count;
foreach (var info in copyData.parameters[paramIndex].infos)
{
info.position += pasteOffset;
}
var oldIDs = copyData.parameters[paramIndex].infos.ToDictionary(t => t, t => t.id);
existingParameter.AddNodeRange(copyData.parameters[paramIndex].infos);
//keep track of new ids for groupnodes
copyData.parameters[paramIndex].idMap = copyData.parameters[paramIndex].infos.ToDictionary(t => oldIDs[t], t => t.id);
}
else
{
// The original parameter is from another graph : create the parameter in the other graph, but replace the infos with only the ones copied.
copyData.parameters[paramIndex].parameter = obj as VFXParameter;
copyData.parameters[paramIndex].parameter.SetNodes(copyData.parameters[paramIndex].infos);
graph.AddChild(obj as VFXModel);
}
}
}
VFXUI copiedUI = allSerializedObjects.OfType <VFXUI>().FirstOrDefault();
int firstCopiedGroup = -1;
int firstCopiedStickyNote = -1;
if (copiedUI != null)
{
VFXUI ui = viewController.graph.UIInfos;
firstCopiedStickyNote = ui.stickyNoteInfos != null ? ui.stickyNoteInfos.Length : 0;
if (copiedUI.groupInfos != null && copiedUI.groupInfos.Length > 0)
{
if (ui.groupInfos == null)
{
ui.groupInfos = new VFXUI.GroupInfo[0];
}
firstCopiedGroup = ui.groupInfos.Length;
foreach (var groupInfos in copiedUI.groupInfos)
{
for (int i = 0; i < groupInfos.contents.Length; ++i)
{
// if we link the parameter node to an existing parameter instead of the copied parameter we have to patch the groupnode content to point the that parameter with the correct id.
if (groupInfos.contents[i].model is VFXParameter)
{
VFXParameter parameter = groupInfos.contents[i].model as VFXParameter;
var paramInfo = copyData.parameters.FirstOrDefault(t => t.copiedParameter == parameter);
if (paramInfo.parameter != null) // parameter will not be null unless the struct returned is the default.
{
groupInfos.contents[i].model = paramInfo.parameter;
groupInfos.contents[i].id = paramInfo.idMap[groupInfos.contents[i].id];
}
}
else if (groupInfos.contents[i].isStickyNote)
{
groupInfos.contents[i].id += firstCopiedStickyNote;
}
}
}
ui.groupInfos = ui.groupInfos.Concat(copiedUI.groupInfos.Select(t => new VFXUI.GroupInfo(t)
{
position = new Rect(t.position.position + pasteOffset, t.position.size)
})).ToArray();
}
if (copiedUI.stickyNoteInfos != null && copiedUI.stickyNoteInfos.Length > 0)
{
if (ui.stickyNoteInfos == null)
{
ui.stickyNoteInfos = new VFXUI.StickyNoteInfo[0];
}
ui.stickyNoteInfos = ui.stickyNoteInfos.Concat(copiedUI.stickyNoteInfos.Select(t => new VFXUI.StickyNoteInfo(t)
{
position = new Rect(t.position.position + pasteOffset, t.position.size)
})).ToArray();
}
}
CopyDataEdges(copyData, allSerializedObjects);
if (copyData.flowEdges != null)
{
foreach (var flowEdge in copyData.flowEdges)
{
VFXContext inputContext = allSerializedObjects[flowEdge.input.contextIndex] as VFXContext;
VFXContext outputContext = allSerializedObjects[flowEdge.output.contextIndex] as VFXContext;
inputContext.LinkFrom(outputContext, flowEdge.input.flowIndex, flowEdge.output.flowIndex);
}
}
foreach (var dataAndContexts in copyData.dataAndContexts)
{
VFXData data = allSerializedObjects[dataAndContexts.dataIndex] as VFXData;
foreach (var contextIndex in dataAndContexts.contextsIndexes)
{
VFXContext context = allSerializedObjects[contextIndex] as VFXContext;
data.CopySettings(context.GetData());
}
}
// Create all ui based on model
viewController.LightApplyChanges();
if (view != null)
{
view.ClearSelection();
var elements = view.graphElements.ToList();
List <VFXNodeUI> newSlotContainerUIs = new List <VFXNodeUI>();
List <VFXContextUI> newContextUIs = new List <VFXContextUI>();
foreach (var slotContainer in allSerializedObjects.OfType <VFXContext>())
{
VFXContextUI contextUI = elements.OfType <VFXContextUI>().FirstOrDefault(t => t.controller.model == slotContainer);
if (contextUI != null)
{
newSlotContainerUIs.Add(contextUI);
newSlotContainerUIs.AddRange(contextUI.GetAllBlocks().Cast <VFXNodeUI>());
newContextUIs.Add(contextUI);
view.AddToSelection(contextUI);
}
}
foreach (var slotContainer in allSerializedObjects.OfType <VFXOperator>())
{
VFXOperatorUI slotContainerUI = elements.OfType <VFXOperatorUI>().FirstOrDefault(t => t.controller.model == slotContainer);
if (slotContainerUI != null)
{
newSlotContainerUIs.Add(slotContainerUI);
view.AddToSelection(slotContainerUI);
}
}
foreach (var param in copyData.parameters)
{
foreach (var parameterUI in elements.OfType <VFXParameterUI>().Where(t => t.controller.model == param.parameter && param.parameter.nodes.IndexOf(t.controller.infos) >= param.infoIndexOffset))
{
newSlotContainerUIs.Add(parameterUI);
view.AddToSelection(parameterUI);
}
}
// Simply selected all data edge with the context or slot container, they can be no other than the copied ones
foreach (var dataEdge in elements.OfType <VFXDataEdge>())
{
if (newSlotContainerUIs.Contains(dataEdge.input.GetFirstAncestorOfType <VFXNodeUI>()))
{
view.AddToSelection(dataEdge);
}
}
// Simply selected all data edge with the context or slot container, they can be no other than the copied ones
foreach (var flowEdge in elements.OfType <VFXFlowEdge>())
{
if (newContextUIs.Contains(flowEdge.input.GetFirstAncestorOfType <VFXContextUI>()))
{
view.AddToSelection(flowEdge);
}
}
if (groupNode != null)
{
foreach (var newSlotContainerUI in newSlotContainerUIs)
{
groupNode.AddNode(newSlotContainerUI.controller);
}
}
//Select all groups that are new
if (firstCopiedGroup >= 0)
{
foreach (var gn in elements.OfType <VFXGroupNode>())
{
if (gn.controller.index >= firstCopiedGroup)
{
view.AddToSelection(gn);
}
}
}
//Select all groups that are new
if (firstCopiedStickyNote >= 0)
{
foreach (var gn in elements.OfType <VFXStickyNote>())
{
if (gn.controller.index >= firstCopiedStickyNote)
{
view.AddToSelection(gn);
}
}
}
}
}
public static void PasteCopy(VFXViewController viewController, Vector2 center, object data, ScriptableObject[] allSerializedObjects, VFXView view, VFXGroupNodeController groupNode)
{
Data copyData = (Data)data;
if (copyData.blocksOnly)
{
if (view != null)
{
copyData.blocks = allSerializedObjects.OfType <VFXBlock>().ToArray();
PasteBlocks(view, copyData);
}
}
else
{
PasteNodes(viewController, center, copyData, allSerializedObjects, view, groupNode);
}
}
public static void UnserializeAndPasteElements(VFXViewController viewController, Vector2 center, string data, VFXView view = null, VFXGroupNodeController groupNode = null)
{
var copyData = JsonUtility.FromJson <Data>(data);
ScriptableObject[] allSerializedObjects = VFXMemorySerializer.ExtractObjects(copyData.serializedObjects, true);
copyData.contexts = allSerializedObjects.OfType <VFXContext>().ToArray();
copyData.slotContainers = allSerializedObjects.OfType <IVFXSlotContainer>().Cast <VFXModel>().Where(t => !(t is VFXContext)).ToArray();
if (copyData.contexts.Length == 0 && copyData.slotContainers.Length == 0)
{
copyData.contexts = null;
copyData.slotContainers = null;
copyData.blocks = allSerializedObjects.OfType <VFXBlock>().ToArray();
}
PasteCopy(viewController, center, copyData, allSerializedObjects, view, groupNode);
}
void PasteAll(VFXViewController viewController, Vector2 center, ref SerializableGraph serializableGraph, VFXView view, VFXGroupNodeController groupNode, List <VFXNodeController> nodesInTheSameOrder)
{
newControllers.Clear();
m_NodesInTheSameOrder = new VFXNodeID[serializableGraph.controllerCount];
// Can't paste context within subgraph block/operator
if (viewController.model.visualEffectObject is VisualEffectSubgraphOperator || viewController.model.visualEffectObject is VisualEffectSubgraphBlock)
{
if (serializableGraph.contexts != null)
{
var count = serializableGraph.contexts.Count();
if (count != 0)
{
Debug.LogWarningFormat("{0} context{1} been skipped during the paste operation. Contexts aren't available in this kind of subgraph.", count, count > 1 ? "s have" : " has");
}
}
}
else
{
PasteContexts(viewController, center, ref serializableGraph);
}
PasteOperators(viewController, center, ref serializableGraph);
PasteParameters(viewController, ref serializableGraph, center);
// Create controllers for all new nodes
viewController.LightApplyChanges();
// Register all nodes for usage in groupNodes and edges
RegisterContexts(viewController);
RegisterOperators(viewController);
RegisterParameterNodes(viewController);
VFXUI ui = viewController.graph.UIInfos;
firstCopiedGroup = -1;
firstCopiedStickyNote = ui.stickyNoteInfos != null ? ui.stickyNoteInfos.Length : 0;
//Paste Everything else
PasteGroupNodes(ref serializableGraph, center, ui);
PasteStickyNotes(ref serializableGraph, center, ui);
PasteDatas(ref serializableGraph); // TODO Data settings should be pasted at context creation. This can lead to issues as blocks are added before data is initialized
PasteDataEdges(ref serializableGraph);
PasteFlowEdges(ref serializableGraph);
// Create all ui based on model
viewController.LightApplyChanges();
if (nodesInTheSameOrder != null)
{
nodesInTheSameOrder.Clear();
nodesInTheSameOrder.AddRange(m_NodesInTheSameOrder.Select(t => t.model == null ? null : viewController.GetNodeController(t.model, t.id)));
}
if (view != null)
{
SelectCopiedElements(view, groupNode);
}
}
public void Paste(VFXViewController viewController, Vector2 center, object data, VFXView view, VFXGroupNodeController groupNode)
{
SerializableGraph serializableGraph = (SerializableGraph)data;
if (serializableGraph.blocksOnly)
{
if (view != null)
{
PasteBlocks(view, ref serializableGraph);
}
}
else
{
PasteAll(viewController, center, ref serializableGraph, view, groupNode);
}
}
void PasteAll(VFXViewController viewController, Vector2 center, ref SerializableGraph serializableGraph, VFXView view, VFXGroupNodeController groupNode)
{
newControllers.Clear();
var graph = viewController.graph;
pasteOffset = (serializableGraph.bounds.width > 0 && serializableGraph.bounds.height > 0) ? center - serializableGraph.bounds.center : Vector2.zero;
MakePasteOffsetUnique(viewController, serializableGraph);
// Paste all nodes
PasteContexts(viewController, ref serializableGraph);
PasteOperators(viewController, ref serializableGraph);
PasteParameters(viewController, ref serializableGraph);
// Create controllers for all new nodes
viewController.LightApplyChanges();
// Register all nodes for usage in groupNodes and edges
RegisterContexts(viewController);
RegisterOperators(viewController);
RegisterParameterNodes(viewController);
VFXUI ui = viewController.graph.UIInfos;
firstCopiedGroup = -1;
firstCopiedStickyNote = ui.stickyNoteInfos != null ? ui.stickyNoteInfos.Length : 0;
//Paste Everything else
PasteGroupNodes(ref serializableGraph, ui);
PasteStickyNotes(ref serializableGraph, ui);
PasteDataEdges(ref serializableGraph);
PasteFlowEdges(ref serializableGraph);
PasteDatas(ref serializableGraph);
// Create all ui based on model
viewController.LightApplyChanges();
if (view != null)
{
SelectCopiedElements(view, groupNode);
}
}
