Unityのプロジェクト内で、Componentや自作クラスの参照を見つけるEditor拡張をちょっと良くしました

これはなに?

以前作った､
kakovail.info
こちら､右クリックでサクッと探したいな､と思ったり､リロードの問題があったりしたので､リファクタがてら書き直していました｡

使い方と変わったところ

前回に追加して､Projectビューからお好きなスクリプトを右クリック→メニューから選択でも調べられるようになりました｡(Assets/KEditorExtensions/から)
Sceneロードでのバグも消しました｡
非アクティブなSceneの対処はちょっと保留です｡
f:id:Kakovail:20210118143638p:plain

コード

<span class="synComment">// Copyright (c) 2021 Hiroyuki Kako</span>
<span class="synComment">// This software is released under the MIT License, see LICENSE.</span>
<span class="synPreProc">#if UNITY_EDITOR</span>
<span class="synStatement">using</span> System;
<span class="synStatement">using</span> System.Collections.Generic;
<span class="synStatement">using</span> System.Linq;
<span class="synStatement">using</span> UnityEditor;
<span class="synStatement">using</span> UnityEditor.SceneManagement;
<span class="synStatement">using</span> UnityEngine;
<span class="synStatement">using</span> UnityEngine.SceneManagement;
<span class="synType">public</span> <span class="synType">class</span> ReferenceFinder : EditorWindow
{
<span class="synType">private</span> <span class="synType">static</span> <span class="synType">string</span> _targetComponentName;
<span class="synType">private</span> <span class="synType">static</span> List&lt;GameObject&gt; _foundAssets = <span class="synStatement">new</span> List&lt;GameObject&gt;();
<span class="synType">private</span> <span class="synType">static</span> IEnumerable&lt;Type&gt; _cashedComponents;
<span class="synType">private</span> <span class="synType">static</span> Dictionary&lt;<span class="synType">string</span>, List&lt;Type&gt;&gt; _typeDict;
<span class="synType">private</span> <span class="synType">static</span> <span class="synType">bool</span> _isFirstTime = <span class="synConstant">true</span>;
<span class="synType">static</span> MonoScript[] _monoScripts;
<span class="synType">private</span> Vector2 _currentScrollPosition;
[InitializeOnLoadMethod]
<span class="synType">private</span> <span class="synType">static</span> <span class="synType">void</span> InitializeEvent()
{
EditorSceneManager.sceneClosing += OnClosing;
}
<span class="synType">private</span> <span class="synType">static</span> <span class="synType">void</span> OnClosing(Scene scene, <span class="synType">bool</span> removingScene)
{
<span class="synStatement">if</span> (!HasOpenInstances&lt;ReferenceFinder&gt;()) <span class="synStatement">return</span>;
GetWindow&lt;ReferenceFinder&gt;().Close();
}
[MenuItem(<span class="synConstant">&quot;KEditorExtensions/Reference/ReferenceFinder&quot;</span>)]
<span class="synType">private</span> <span class="synType">static</span> <span class="synType">void</span> Open()
{
_cashedComponents = GetAllTypes();
GetWindow&lt;ReferenceFinder&gt;(<span class="synConstant">&quot;Components Reference Finder.&quot;</span>);
}
[MenuItem(<span class="synConstant">&quot;Assets/KEditorExtensions/Find references in active scene and all prefabs&quot;</span>)]
<span class="synType">private</span> <span class="synType">static</span> <span class="synType">void</span> FindWithRightClick()
{
var name = Selection.activeObject;
_cashedComponents = GetAllTypes();
GetWindow&lt;ReferenceFinder&gt;(<span class="synConstant">&quot;Components Reference Finder.&quot;</span>);
_targetComponentName = name.name;
SearchObjects();
}
<span class="synType">private</span> <span class="synType">void</span> OnGUI()
{
EditorGUILayout.BeginHorizontal();
EditorGUILayout.LabelField(<span class="synConstant">&quot;Never change Scene before reset this window&quot;</span>);
EditorGUILayout.EndHorizontal();
EditorGUILayout.BeginHorizontal();
EditorGUILayout.LabelField(<span class="synConstant">&quot;This window can search active scene objects and all prefabs&quot;</span>);
EditorGUILayout.EndHorizontal();
EditorGUILayout.BeginHorizontal();
_targetComponentName =
EditorGUILayout.TextField(<span class="synConstant">&quot;Target Component Name: &quot;</span>, _targetComponentName);
EditorGUILayout.EndHorizontal();
<span class="synStatement">if</span> (_foundAssets.Count &gt; <span class="synConstant">0</span>)
{
_currentScrollPosition = EditorGUILayout.BeginScrollView(_currentScrollPosition);
<span class="synStatement">foreach</span> (var asset <span class="synStatement">in</span> _foundAssets)
{
EditorGUILayout.ObjectField(asset.name, asset, <span class="synStatement">typeof</span>(GameObject), <span class="synConstant">false</span>);
}
EditorGUILayout.EndScrollView();
}
<span class="synStatement">if</span> (GUILayout.Button(<span class="synConstant">&quot;Search&quot;</span>))
{
SearchObjects();
}
}
<span class="synType">private</span> <span class="synType">static</span> <span class="synType">void</span> SearchObjects()
{
_foundAssets.Clear();
var guids = AssetDatabase.FindAssets(<span class="synConstant">&quot;t:GameObject&quot;</span>, <span class="synConstant">null</span>);
<span class="synStatement">foreach</span> (var guid <span class="synStatement">in</span> guids)
{
<span class="synType">string</span> path = AssetDatabase.GUIDToAssetPath(guid);
var loadAsset = AssetDatabase.LoadAssetAtPath&lt;GameObject&gt;(path);
var typeCash = GetType(_targetComponentName) ?? <span class="synConstant">null</span>;
<span class="synStatement">if</span> (typeCash == <span class="synConstant">null</span>)
{
Debug.Log(<span class="synConstant">&quot;No Type Found.&quot;</span>);
<span class="synStatement">return</span>;
}
var tmp = loadAsset.GetComponentsInChildren(GetType(_targetComponentName) ?? <span class="synConstant">null</span>);
<span class="synStatement">foreach</span> (var kari <span class="synStatement">in</span> tmp)
{
_foundAssets.Add(kari.gameObject);
}
}
var allObj = GetObjectsInAllScene();
<span class="synStatement">foreach</span> (var obj <span class="synStatement">in</span> allObj)
{
var typeCash = GetType(_targetComponentName) ?? <span class="synConstant">null</span>;
var tmp = obj.GetComponentsInChildren(GetType(_targetComponentName) ?? <span class="synConstant">null</span>);
<span class="synStatement">if</span> (typeCash == <span class="synConstant">null</span>)
{
Debug.Log(<span class="synConstant">&quot;No Type Found.&quot;</span>);
<span class="synStatement">return</span>;
}
<span class="synStatement">foreach</span> (var kari <span class="synStatement">in</span> tmp)
{
_foundAssets.Add(kari.gameObject);
}
}
}
<span class="synComment">/// </span><span class="synIdentifier">&lt;</span><span class="synStatement">summary</span><span class="synIdentifier">&gt;</span>
<span class="synComment">/// All script files that exist in the project</span>
<span class="synComment">/// </span><span class="synIdentifier">&lt;/</span><span class="synStatement">summary</span><span class="synIdentifier">&gt;</span>
<span class="synType">static</span> MonoScript[] MonoScripts
{
<span class="synStatement">get</span> { <span class="synStatement">return</span> _monoScripts ?? (_monoScripts = Resources.FindObjectsOfTypeAll&lt;MonoScript&gt;().ToArray()); }
}
<span class="synComment">/// </span><span class="synIdentifier">&lt;</span><span class="synStatement">summary</span><span class="synIdentifier">&gt;</span>
<span class="synComment">/// Getting the type from the class name</span>
<span class="synComment">/// </span><span class="synIdentifier">&lt;/</span><span class="synStatement">summary</span><span class="synIdentifier">&gt;</span>
<span class="synType">private</span> <span class="synType">static</span> Type GetType(<span class="synType">string</span> className)
{
<span class="synStatement">if</span> (_typeDict == <span class="synConstant">null</span>)
{
_typeDict = <span class="synStatement">new</span> Dictionary&lt;<span class="synType">string</span>, List&lt;Type&gt;&gt;();
<span class="synStatement">foreach</span> (var type <span class="synStatement">in</span> _cashedComponents)
{
<span class="synStatement">if</span> (!_typeDict.ContainsKey(type.Name))
{
_typeDict.Add(type.Name, <span class="synStatement">new</span> List&lt;Type&gt;());
}
_typeDict[type.Name].Add(type);
}
}
<span class="synComment">// If the class exists, show it in the list</span>
<span class="synStatement">if</span> (_typeDict.ContainsKey(className))
{
<span class="synStatement">return</span> _typeDict[className][<span class="synConstant">0</span>];
}
<span class="synStatement">else</span>
{
<span class="synComment">// If the class doesn't exist, get it just in case, and run again.</span>
<span class="synStatement">if</span> (_isFirstTime)
{
Debug.Log(<span class="synConstant">&quot;Not found. ReScanning...&quot;</span>);
_cashedComponents = GetAllTypes();
_isFirstTime = <span class="synConstant">false</span>;
GetType(className);
}
_isFirstTime = <span class="synConstant">true</span>;
<span class="synStatement">return</span> <span class="synConstant">null</span>;
}
}
<span class="synComment">/// </span><span class="synIdentifier">&lt;</span><span class="synStatement">summary</span><span class="synIdentifier">&gt;</span>
<span class="synComment">/// Get all class</span>
<span class="synComment">/// </span><span class="synIdentifier">&lt;/</span><span class="synStatement">summary</span><span class="synIdentifier">&gt;</span>
<span class="synType">private</span> <span class="synType">static</span> IEnumerable&lt;Type&gt; GetAllTypes()
{
var types = AppDomain.CurrentDomain.GetAssemblies()
.SelectMany(asm =&gt; asm.GetTypes())
.Where(type =&gt; type != <span class="synConstant">null</span> &amp;&amp; !<span class="synType">string</span>.IsNullOrEmpty(type.Namespace))
.Where(type =&gt; type.Namespace.Contains(<span class="synConstant">&quot;UnityEngine&quot;</span>));
var localTypes = MonoScripts
.Where(script =&gt; script != <span class="synConstant">null</span>)
.Select(script =&gt; script.GetClass())
.Where(classType =&gt; classType != <span class="synConstant">null</span>)
.Where(classType =&gt; classType.Module.Name == <span class="synConstant">&quot;Assembly-CSharp.dll&quot;</span>);
<span class="synStatement">return</span> types.Concat(localTypes).Distinct();
}
<span class="synType">private</span> <span class="synType">static</span> GameObject[] GetObjectsInAllScene(<span class="synType">bool</span> includeInactive = <span class="synConstant">true</span>)
{
var sceneNumber = SceneManager.sceneCount;
IEnumerable&lt;GameObject&gt; resultObjects = (GameObject[]) Enumerable.Empty&lt;GameObject&gt;();
<span class="synStatement">for</span> (<span class="synType">int</span> i = <span class="synConstant">0</span>; i &lt; sceneNumber; i++)
{
var obj = SceneManager.GetSceneAt(i).GetRootGameObjects();
resultObjects = resultObjects.Concat(obj);
}
<span class="synStatement">return</span> resultObjects.ToArray();
}
}
<span class="synPreProc">#endif</span>

100

101

102

103

104

105

106

107

108

109

110

111

112

113

114

115

116

117

118

119

120

121

122

123

124

125

126

127

128

129

130

131

132

133

134

135

136

137

138

139

140

141

142

143

144

145

146

147

148

149

150

151

152

153

154

155

156

157

158

159

160

161

162

163

164

165

166

167

168

169

170

171

172

173

174

175

176

177

178

// This software is released under the MIT License, see LICENSE.

#if UNITY_EDITOR

using System;

using System.Collections.Generic;

using System.Linq;

using UnityEditor;

using UnityEditor.SceneManagement;

using UnityEngine;

using UnityEngine.SceneManagement;

public class ReferenceFinder : EditorWindow

{

private static string _targetComponentName;

private static List<GameObject> _foundAssets = new List<GameObject>();

private static IEnumerable<Type> _cashedComponents;

private static Dictionary<string, List<Type>> _typeDict;

private static bool _isFirstTime = true;

static MonoScript[] _monoScripts;

private Vector2 _currentScrollPosition;

[InitializeOnLoadMethod]

private static void InitializeEvent()

{

EditorSceneManager.sceneClosing += OnClosing;

}

private static void OnClosing(Scene scene, bool removingScene)

{

if (!HasOpenInstances<ReferenceFinder>()) return;

GetWindow<ReferenceFinder>().Close();

}

[MenuItem("KEditorExtensions/Reference/ReferenceFinder")]

private static void Open()

{

_cashedComponents = GetAllTypes();

GetWindow<ReferenceFinder>("Components Reference Finder.");

}

[MenuItem("Assets/KEditorExtensions/Find references in active scene and all prefabs")]

private static void FindWithRightClick()

{

var name = Selection.activeObject;

_cashedComponents = GetAllTypes();

GetWindow<ReferenceFinder>("Components Reference Finder.");

_targetComponentName = name.name;

SearchObjects();

}

private void OnGUI()

{

EditorGUILayout.BeginHorizontal();

EditorGUILayout.LabelField("Never change Scene before reset this window");

EditorGUILayout.EndHorizontal();

EditorGUILayout.BeginHorizontal();

EditorGUILayout.LabelField("This window can search active scene objects and all prefabs");

EditorGUILayout.EndHorizontal();

EditorGUILayout.BeginHorizontal();

_targetComponentName =

EditorGUILayout.TextField("Target Component Name: ", _targetComponentName);

EditorGUILayout.EndHorizontal();

if (_foundAssets.Count > 0)

{

_currentScrollPosition = EditorGUILayout.BeginScrollView(_currentScrollPosition);

foreach (var asset in _foundAssets)

{

EditorGUILayout.ObjectField(asset.name, asset, typeof(GameObject), false);

}

EditorGUILayout.EndScrollView();

}

if (GUILayout.Button("Search"))

{

SearchObjects();

}

private static void SearchObjects()

{

_foundAssets.Clear();

var guids = AssetDatabase.FindAssets("t:GameObject", null);

foreach (var guid in guids)

{

string path = AssetDatabase.GUIDToAssetPath(guid);

var loadAsset = AssetDatabase.LoadAssetAtPath<GameObject>(path);

var typeCash = GetType(_targetComponentName) ?? null;

if (typeCash == null)

{

Debug.Log("No Type Found.");

return;

}

var tmp = loadAsset.GetComponentsInChildren(GetType(_targetComponentName) ?? null);

foreach (var kari in tmp)

{

_foundAssets.Add(kari.gameObject);

}

var allObj = GetObjectsInAllScene();

foreach (var obj in allObj)

{

var typeCash = GetType(_targetComponentName) ?? null;

var tmp = obj.GetComponentsInChildren(GetType(_targetComponentName) ?? null);

if (typeCash == null)

{

Debug.Log("No Type Found.");

return;

}

foreach (var kari in tmp)

{

_foundAssets.Add(kari.gameObject);

}

/// <summary>

/// All script files that exist in the project

/// </summary>

static MonoScript[] MonoScripts

{

get { return _monoScripts ?? (_monoScripts = Resources.FindObjectsOfTypeAll<MonoScript>().ToArray()); }

}

/// <summary>

/// Getting the type from the class name

/// </summary>

private static Type GetType(string className)

{

if (_typeDict == null)

{

_typeDict = new Dictionary<string, List<Type>>();

foreach (var type in _cashedComponents)

{

if (!_typeDict.ContainsKey(type.Name))

{

_typeDict.Add(type.Name, new List<Type>());

}

_typeDict[type.Name].Add(type);

}

// If the class exists, show it in the list

if (_typeDict.ContainsKey(className))

{

return _typeDict[className][0];

}

else

{

// If the class doesn't exist, get it just in case, and run again.

if (_isFirstTime)

{

Debug.Log("Not found. ReScanning...");

_cashedComponents = GetAllTypes();

_isFirstTime = false;

GetType(className);

}

_isFirstTime = true;

return null;

}

/// <summary>

/// Get all class

/// </summary>

private static IEnumerable<Type> GetAllTypes()

{

var types = AppDomain.CurrentDomain.GetAssemblies()

.SelectMany(asm => asm.GetTypes())

.Where(type => type != null && !string.IsNullOrEmpty(type.Namespace))

.Where(type => type.Namespace.Contains("UnityEngine"));

var localTypes = MonoScripts

.Where(script => script != null)

.Select(script => script.GetClass())

.Where(classType => classType != null)

.Where(classType => classType.Module.Name == "Assembly-CSharp.dll");

return types.Concat(localTypes).Distinct();

}

private static GameObject[] GetObjectsInAllScene(bool includeInactive = true)

{

var sceneNumber = SceneManager.sceneCount;

IEnumerable<GameObject> resultObjects = (GameObject[]) Enumerable.Empty<GameObject>();

for (int i = 0; i < sceneNumber; i++)

{

var obj = SceneManager.GetSceneAt(i).GetRootGameObjects();

resultObjects = resultObjects.Concat(obj);

}

return resultObjects.ToArray();

}

#endif

これはなに?

使い方と変わったところ

コード

コメントを残す コメントをキャンセル

コメントを残すコメントをキャンセル