ParticleEffectForUGUI/Scripts/Editor/UIParticleMenu.cs

109 lines
4.3 KiB
C#

using System.IO;
using System.Text.RegularExpressions;
using UnityEditor;
using UnityEngine;
namespace Coffee.UIExtensions
{
public class UIParticleMenu
{
#if !UNITY_2019_1_OR_NEWER
static string GetPreviousSamplePath(string displayName, string sampleName)
{
string sampleRoot = string.Format("Assets/Samples/{0}", displayName);
var sampleRootInfo = new DirectoryInfo(sampleRoot);
if (!sampleRootInfo.Exists) return null;
foreach (var versionDir in sampleRootInfo.GetDirectories())
{
var samplePath = Path.Combine(versionDir.ToString(), sampleName);
if (Directory.Exists(samplePath))
return samplePath;
}
return null;
}
static void ImportSample(string packageName, string sampleName)
{
string jsonPath = string.Format("Packages/{0}/package.json", packageName);
string json = File.ReadAllText(jsonPath);
string version = Regex.Match(json, "\"version\"\\s*:\\s*\"([^\"]+)\"").Groups[1].Value;
string displayName = Regex.Match(json, "\"displayName\"\\s*:\\s*\"([^\"]+)\"").Groups[1].Value;
string src = string.Format("{0}/Samples~/{1}", Path.GetDirectoryName(jsonPath), sampleName);
string dst = string.Format("Assets/Samples/{0}/{1}/{2}", displayName, version, sampleName);
string previous = GetPreviousSamplePath(displayName, sampleName);
if (!string.IsNullOrEmpty(previous))
{
string msg = "A different version of the sample is already imported at\n\n"
+ previous
+ "\n\nIt will be deleted when you update. Are you sure you want to continue?";
if (!EditorUtility.DisplayDialog("Sample Importer", msg, "OK", "Cancel"))
return;
FileUtil.DeleteFileOrDirectory(previous + ".meta");
FileUtil.DeleteFileOrDirectory(previous);
string versionDir = Path.GetDirectoryName(previous);
if (Directory.GetFiles(versionDir, "*.meta", SearchOption.TopDirectoryOnly).Length == 0)
{
FileUtil.DeleteFileOrDirectory(versionDir + ".meta");
FileUtil.DeleteFileOrDirectory(versionDir);
}
}
Directory.CreateDirectory(Path.GetDirectoryName(dst));
FileUtil.CopyFileOrDirectory(src, dst);
AssetDatabase.ImportAsset(dst, ImportAssetOptions.ImportRecursive);
}
[MenuItem("Assets/Samples/Import UIParticle Sample")]
static void ImportSample()
{
ImportSample("com.coffee.ui-particle", "Demo");
}
#endif
[MenuItem("GameObject/UI/Particle System", false, 2019)]
public static void AddParticle(MenuCommand menuCommand)
{
// Create UI element.
EditorApplication.ExecuteMenuItem("GameObject/UI/Image");
var ui = Selection.activeGameObject;
// Create ParticleSystem.
EditorApplication.ExecuteMenuItem("GameObject/Effects/Particle System");
var ps = Selection.activeGameObject;
var transform = ps.transform;
var localRotation = transform.localRotation;
transform.SetParent(ui.transform.parent, true);
var pos = transform.localPosition;
pos.z = 0;
ps.transform.localPosition = pos;
ps.transform.localRotation = localRotation;
// Destroy UI elemant
Object.DestroyImmediate(ui);
// Assign default material.
var renderer = ps.GetComponent<ParticleSystemRenderer>();
var defaultMat = AssetDatabase.GetBuiltinExtraResource<Material>("Default-Particle.mat");
renderer.material = defaultMat ? defaultMat : renderer.material;
// Set to hierarchy mode
var particleSystem = ps.GetComponent<ParticleSystem>();
var main = particleSystem.main;
main.scalingMode = ParticleSystemScalingMode.Hierarchy;
// Add UIParticle.
var uiParticle = ps.AddComponent<UIParticle>();
uiParticle.ignoreCanvasScaler = true;
uiParticle.scale = 10;
}
}
}