您最多选择25个主题
主题必须以中文或者字母或数字开头,可以包含连字符 (-),并且长度不得超过35个字符
137 行
4.1 KiB
137 行
4.1 KiB
using System;
|
|
using System.Collections.Generic;
|
|
using UnityEngine;
|
|
using UnityEngine.Rendering;
|
|
|
|
namespace UnityEditor.Rendering
|
|
{
|
|
using IProvider = FilterWindow.IProvider;
|
|
using Element = FilterWindow.Element;
|
|
using GroupElement = FilterWindow.GroupElement;
|
|
|
|
class VolumeComponentProvider : IProvider
|
|
{
|
|
class VolumeComponentElement : Element
|
|
{
|
|
public Type type;
|
|
|
|
public VolumeComponentElement(int level, string label, Type type)
|
|
{
|
|
this.level = level;
|
|
this.type = type;
|
|
// TODO: Add support for custom icons
|
|
content = new GUIContent(label);
|
|
}
|
|
}
|
|
|
|
class PathNode : IComparable<PathNode>
|
|
{
|
|
public List<PathNode> nodes = new List<PathNode>();
|
|
public string name;
|
|
public Type type;
|
|
|
|
public int CompareTo(PathNode other)
|
|
{
|
|
return name.CompareTo(other.name);
|
|
}
|
|
}
|
|
|
|
public Vector2 position { get; set; }
|
|
|
|
VolumeProfile m_Target;
|
|
VolumeComponentListEditor m_TargetEditor;
|
|
|
|
public VolumeComponentProvider(VolumeProfile target, VolumeComponentListEditor targetEditor)
|
|
{
|
|
m_Target = target;
|
|
m_TargetEditor = targetEditor;
|
|
}
|
|
|
|
public void CreateComponentTree(List<Element> tree)
|
|
{
|
|
tree.Add(new GroupElement(0, "Volume Overrides"));
|
|
|
|
var attrType = typeof(VolumeComponentMenu);
|
|
var types = VolumeManager.instance.baseComponentTypes;
|
|
var rootNode = new PathNode();
|
|
|
|
foreach (var t in types)
|
|
{
|
|
// Skip components that have already been added to the volume
|
|
if (m_Target.Has(t))
|
|
continue;
|
|
|
|
string path = string.Empty;
|
|
|
|
// Look for a VolumeComponentMenu attribute
|
|
var attrs = t.GetCustomAttributes(attrType, false);
|
|
if (attrs.Length > 0)
|
|
{
|
|
var attr = attrs[0] as VolumeComponentMenu;
|
|
if (attr != null)
|
|
path = attr.menu;
|
|
}
|
|
|
|
// If no attribute or in case something went wrong when grabbing it, fallback to a
|
|
// beautified class name
|
|
if (string.IsNullOrEmpty(path))
|
|
path = ObjectNames.NicifyVariableName(t.Name);
|
|
|
|
// Prep the categories & types tree
|
|
AddNode(rootNode, path, t);
|
|
}
|
|
|
|
// Recursively add all elements to the tree
|
|
Traverse(rootNode, 1, tree);
|
|
}
|
|
|
|
public bool GoToChild(Element element, bool addIfComponent)
|
|
{
|
|
if (element is VolumeComponentElement)
|
|
{
|
|
var e = (VolumeComponentElement)element;
|
|
m_TargetEditor.AddComponent(e.type);
|
|
return true;
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
void AddNode(PathNode root, string path, Type type)
|
|
{
|
|
var current = root;
|
|
var parts = path.Split(new[] { '/' }, StringSplitOptions.RemoveEmptyEntries);
|
|
|
|
foreach (var part in parts)
|
|
{
|
|
var child = current.nodes.Find(x => x.name == part);
|
|
|
|
if (child == null)
|
|
{
|
|
child = new PathNode { name = part, type = type };
|
|
current.nodes.Add(child);
|
|
}
|
|
|
|
current = child;
|
|
}
|
|
}
|
|
|
|
void Traverse(PathNode node, int depth, List<Element> tree)
|
|
{
|
|
node.nodes.Sort();
|
|
|
|
foreach (var n in node.nodes)
|
|
{
|
|
if (n.nodes.Count > 0) // Group
|
|
{
|
|
tree.Add(new GroupElement(depth, n.name));
|
|
Traverse(n, depth + 1, tree);
|
|
}
|
|
else // Element
|
|
{
|
|
tree.Add(new VolumeComponentElement(depth, n.name, n.type));
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|