扩展Unity3D编辑器的脚本模板

Posted 2020-09-30 yxysuanfa

??近期在学习Shader时感觉Shader语言參数众多、语法诡异，假设每次都从头開始写Shader一定是一件痛苦的事情。假设能够在本地定义好一组标准的Shader模板，这样当我们须要实现某些效果相似的Shader时，就能够在这个Shader模板的基础上进行改动。

由于Shader文件是一个文本文件，所以我们能够很easy地创建这样一个模板，在这个模板中我们能够进一步完好相关的參数凝视，这样就不用每次写Shader的时候都须要查文档了，从这个角度出发，就进入了这篇文章的正题：扩展Unity3D编辑器的脚本模板。

按图索骥，模板在哪里？

??Unity3D默认的脚本模版位于/Editor/Data/Resources/ScriptTemplates/文件夹下，注意该文件夹相对Unity3D的安装文件夹而言，在这个文件夹中我们能够找到Unity3D中脚本模板的某些蛛丝马迹，首先，脚本模板是一个简单的文本文件，这个文本文件里预先填充了内容，我们在编辑器中创建模脚本或者Shader的时候实际上是读取这些文件然后在写入项目中的指定路径的。

其次。这些模板文件里#SCRIPTNAME#或者#NAME#这种标记。当我们在编辑器中创建文件的时候，这个标记会被替换成指定的文件名称。比方Unity3D中继承自MonoBehaviour的脚本。有一个很重要的特性是文件名称必须和类名保持一致。这固然是Unity3D引擎的一个设定，但是在这里亦能够找到一个能够称得上理由的理由。

我们注意到这些模板的文件名称中都有一个独一无二的数字，比方C#脚本的模板中的数字是81、Shader模板中的数字是83，这些数字是什么呢，博主这里将其称为来自星星的黑科技。

来自星星的黑科技

??作为一个经常捣鼓Unity3D编辑器的人。假设说你不知道MenuItem、EditorWindow、ScriptableWizard这些黑科技。那么说明你不是一个喜欢折腾和探索的人。

从Unity3D的API文档中，我们知道MenuItem的原型为：

MenuItem(string itemName,bool isValidateFunction,int priority) 

我知道我们通常使用MenuItem常用的是它的第一个參数，即定义一个菜单项的名称，我们能够使用”/”这种分隔符来表示菜单的层级。MenuItem须要配合一个静态方法来使用，能够理解为当我们点击当前定义的菜单后就会去运行静态方法中的代码。因此MenuItem经常能够帮助我们做些编辑器扩展开发的工作。好了，第二个參数作为一个验证的标志。假设该标志为true，意味着我们定义的静态方法是一个验证方法在运行静态方法前会首先对方法进行验证，这个我们暂且无论，由于今天我们这个来自星星的黑科技主要和第三个參数有关，第三个參数表示一个优先级，它表示菜单项在菜单条中的展示顺序，优先级大的菜单项会展示在优先级小的菜单项以下，由此我们就明确了了模板文件名称中的相似81、83这种数字的真实含义，注意到模板文件的排列顺序和编辑器中的菜单项顺序是一样的，我们做一个尝试。编写以下的代码：

[MenuItem("Assets/Create/Lua Scripts", false, 85)]
static void CreateLuaScripts()
{

}

[MenuItem("Assets/Create/固定功能着色器", false, 86)]
static void CreateFixedFunctionShader()
{

}

[MenuItem("Assets/Create/表面着色器", false, 87)]
static void CreateSurfaceShader()
{

}

[MenuItem("Assets/Create/可编程着色器", false, 88)]
static void CreateVertexAndFragmentShader()
{

}

注意到我们依照已知的优先级继续写了四个方法，如今我们在编辑器中能够发现默认的菜单条发生了变化：

我们能够看到我们编写的这四个菜单都生效了，尽管它们临时什么都做不了。但顺着这个方向去探索，我们是能够实现最初的梦想的。

如今我们来思考怎样依据模板来创建文件，这个对我们来说简直太简单了，通过StreamReader来读取模板。然后再用StreamWriter来生成文件就能够了。

但是这样创建的文件的文件名称是固定的。在创建文件的时候我们没法改动。并且即使改动了文件内定义的名字并不会改变啊。所以我们须要一个更好的解决方式。Unity3D提供了一个UnityEditor.ProjectWindowCallback的命名空间，在这个空间中提供了一个称为EndNameEditAction的类，我们仅仅须要继承这个类就能够完毕这个任务。这个类须要重写Action的方法，我们知道创建一个文件的完整步骤是创建文件然后使其高亮显示，因此这部分代码实现例如以下：

/// <summary>
/// 定义一个创建资源的Action类并实现其Action方法
/// </summary>
class CreateAssetAction : EndNameEditAction
{

    public override void Action(int instanceId, string pathName, string resourceFile)
    {
        //创建资源
        Object obj = CreateAssetFormTemplate(pathName, resourceFile);
        //高亮显示该资源
        ProjectWindowUtil.ShowCreatedAsset(obj);
    }

    internal static Object CreateAssetFormTemplate(string pathName, string resourceFile)
    {

        //获取要创建资源的绝对路径
        string fullName = Path.GetFullPath(pathName);
        //读取本地模版文件
        StreamReader reader = new StreamReader(resourceFile);
        string content = reader.ReadToEnd();
        reader.Close();

        //获取资源的文件名称
        string fileName = Path.GetFileNameWithoutExtension(pathName);
        //替换默认的文件名称
        content = content.Replace("#NAME", fileName);

        //写入新文件
        StreamWriter writer = new StreamWriter(fullName, false, System.Text.Encoding.UTF8);
        writer.Write(content);
        writer.Close();

        //刷新本地资源
        AssetDatabase.ImportAsset(pathName);
        AssetDatabase.Refresh();

        return AssetDatabase.LoadAssetAtPath(pathName, typeof(Object));
    }
}

这部分代码相对来说比較简单，就是读取本地模板文件然后生成新文件，在生成新文件的时候会将#NAME替换成实际的文件名称，这样我们就完毕了文件资源的创建。

如今的问题是怎样在创建文件的时候获取实际的路径，这部分代码实现例如以下：

private static string GetSelectedPath()
{
    //默认路径为Assets
    string selectedPath = "Assets";

    //获取选中的资源
    Object[] selection = Selection.GetFiltered(typeof(Object), SelectionMode.Assets);

    //遍历选中的资源以返回路径
    foreach (Object obj in selection)
    {
        selectedPath = AssetDatabase.GetAssetPath(obj);
        if (!string.IsNullOrEmpty(selectedPath) && File.Exists(selectedPath))
        {
            selectedPath = Path.GetDirectoryName(selectedPath);
            break;
        }
    }

    return selectedPath;
}

如今攻克了创建资源的问题。我们接下来仅仅要调用ProjectWindowUtil的StartNameEditingIfProjectWindowExists方法就可以。该方法须要传入一个继承自EndNameEditAction的类的实例、目标文件路径和模板文件的路径。

比如要创建一个Lua脚本能够这样实现：

[MenuItem("Assets/Create/Lua Scripts", false, 85)]
static void CreateLuaScripts()
{
    ProjectWindowUtil.StartNameEditingIfProjectWindowExists(0,
        ScriptableObject.CreateInstance<CreateAssetAction>(),
        GetSelectedPath() + "/NewLuaScript.lua", null,
        "Assets/Editor/Template/85-Lua-NewLuaScript.lua.txt");
}

小结

??如今有了这个黑科技以后，我们能够创建很多其它的模板来扩展编辑器的功能，比方对Shader而言，我们能够创建些基础性的Shader模板，然后每次须要写Shader的时候直接从模板库中选择一个功能相似的Shader然后在此基础上进行改动，这样比从头開始写一个新的Shader应该会轻松不少，这段时间学习Shader，感觉进程缓慢离图形学高手遥遥无期，行了，这篇博客就是这样了。