浏览代码

Fixed drag and drop for texture2d, cubemap, subgraph and updated the blackboardfield drag and drop

/main
Peter Bay Bastian 6 年前
当前提交
529163c8
共有 1 个文件被更改,包括 124 次插入20 次删除
  1. 144
      ShaderGraph/com.unity.shadergraph/Editor/Drawing/Views/MaterialGraphView.cs

144
ShaderGraph/com.unity.shadergraph/Editor/Drawing/Views/MaterialGraphView.cs


using UnityEngine.Experimental.UIElements;
using Edge = UnityEditor.Experimental.UIElements.GraphView.Edge;
using Node = UnityEditor.Experimental.UIElements.GraphView.Node;
using Object = UnityEngine.Object;
#if !UNITY_2018_1
using UnityEditor.Graphs;
#endif

#region Drag and drop
static bool ValidateObjectForDrop(Object obj)
{
return EditorUtility.IsPersistent(obj) && (obj is Texture2D || obj is Cubemap || obj is MaterialSubGraphAsset);
}
bool dragging = false;
if (selection != null)
{
// Blackboard
if (selection.OfType<BlackboardField>().Any())
dragging = true;
}
else
{
// Handle unity objects
var objects = DragAndDrop.objectReferences;
foreach (Object obj in objects)
{
if (ValidateObjectForDrop(obj))
{
dragging = true;
break;
}
}
}
if (selection != null && (selection.OfType<BlackboardField>().Any() ))
if (dragging)
DragAndDrop.visualMode = e.ctrlKey ? DragAndDropVisualMode.Copy : DragAndDropVisualMode.Move;
DragAndDrop.visualMode = DragAndDropVisualMode.Generic;
Vector2 localPos = (e.currentTarget as VisualElement).ChangeCoordinatesTo(contentViewContainer, e.localMousePosition);
if (selection == null)
return;
if (selection != null)
{
// Blackboard
if (selection.OfType<BlackboardField>().Any())
{
IEnumerable<BlackboardField> fields = selection.OfType<BlackboardField>();
foreach (BlackboardField field in fields)
{
CreateNode(field, localPos);
}
}
}
else
{
// Handle unity objects
var objects = DragAndDrop.objectReferences;
foreach (Object obj in objects)
{
if (ValidateObjectForDrop(obj))
{
CreateNode(obj, localPos);
}
}
}
}
void CreateNode(object obj, Vector2 nodePosition)
{
var texture2D = obj as Texture2D;
if (texture2D != null)
{
graph.owner.RegisterCompleteObjectUndo("Drag Texture");
bool isNormalMap = false;
if (EditorUtility.IsPersistent(texture2D) && !string.IsNullOrEmpty(AssetDatabase.GetAssetPath(texture2D)))
{
var importer = AssetImporter.GetAtPath(AssetDatabase.GetAssetPath(texture2D)) as TextureImporter;
if (importer != null)
isNormalMap = importer.textureType == TextureImporterType.NormalMap;
}
var node = new SampleTexture2DNode();
if (isNormalMap)
node.textureType = TextureType.Normal;
IEnumerable<BlackboardField> fields = selection.OfType<BlackboardField>();
if (!fields.Any())
return;
var drawState = node.drawState;
drawState.position = new Rect(nodePosition, drawState.position.size);
node.drawState = drawState;
graph.AddNode(node);
Vector2 localPos = (e.currentTarget as VisualElement).ChangeCoordinatesTo(contentViewContainer, e.localMousePosition);
var inputslot = node.FindInputSlot<Texture2DInputMaterialSlot>(SampleTexture2DNode.TextureInputId);
if (inputslot != null)
inputslot.texture = texture2D;
foreach (BlackboardField field in fields)
}
var cubemap = obj as Cubemap;
if (cubemap != null)
IShaderProperty property = field.userData as IShaderProperty;
if (property == null)
continue;
graph.owner.RegisterCompleteObjectUndo("Drag Cubemap");
var property = new CubemapShaderProperty { displayName = cubemap.name, value = { cubemap = cubemap } };
graph.AddShaderProperty(property);
var node = new SampleCubemapNode();
var drawState = node.drawState;
drawState.position = new Rect(nodePosition, drawState.position.size);
node.drawState = drawState;
graph.AddNode(node);
var inputslot = node.FindInputSlot<CubemapInputMaterialSlot>(SampleCubemapNode.CubemapInputId);
if (inputslot != null)
inputslot.cubemap = cubemap;
}
var node = new PropertyNode();
var subGraphAsset = obj as MaterialSubGraphAsset;
if (subGraphAsset != null)
{
graph.owner.RegisterCompleteObjectUndo("Drag Sub-Graph");
var node = new SubGraphNode();
var position = drawState.position;
position.x = localPos.x;
position.y = localPos.y;
drawState.position = position;
drawState.position = new Rect(nodePosition, drawState.position.size);
node.subGraphAsset = subGraphAsset;
graph.AddNode(node);
}
graph.owner.RegisterCompleteObjectUndo("Added Property");
graph.AddNode(node);
node.propertyGuid = property.guid;
var blackboardField = obj as BlackboardField;
if (blackboardField != null)
{
IShaderProperty property = blackboardField.userData as IShaderProperty;
if (property != null)
{
graph.owner.RegisterCompleteObjectUndo("Drag Property");
var node = new PropertyNode();
var drawState = node.drawState;
drawState.position = new Rect(nodePosition, drawState.position.size);
node.drawState = drawState;
graph.AddNode(node);
node.propertyGuid = property.guid;
}
}
}

正在加载...
取消
保存