ResourceAnalyzerController.cs 12 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324
  1. //------------------------------------------------------------
  2. // Game Framework
  3. // Copyright © 2013-2021 loyalsoft. All rights reserved.
  4. // Homepage: http://www.game7000.com/
  5. // Feedback: http://www.game7000.com/
  6. //------------------------------------------------------------
  7. using GameFramework;
  8. using System;
  9. using System.Collections.Generic;
  10. using System.Linq;
  11. using UnityEditor;
  12. using UnityEngine;
  13. namespace UnityGameFramework.Editor.ResourceTools
  14. {
  15. public sealed partial class ResourceAnalyzerController
  16. {
  17. private readonly ResourceCollection m_ResourceCollection;
  18. private readonly Dictionary<string, DependencyData> m_DependencyDatas;
  19. private readonly Dictionary<string, List<Asset>> m_ScatteredAssets;
  20. private readonly List<string[]> m_CircularDependencyDatas;
  21. private readonly HashSet<Stamp> m_AnalyzedStamps;
  22. public ResourceAnalyzerController()
  23. : this(null)
  24. {
  25. }
  26. public ResourceAnalyzerController(ResourceCollection resourceCollection)
  27. {
  28. m_ResourceCollection = resourceCollection != null ? resourceCollection : new ResourceCollection();
  29. m_ResourceCollection.OnLoadingResource += delegate (int index, int count)
  30. {
  31. if (OnLoadingResource != null)
  32. {
  33. OnLoadingResource(index, count);
  34. }
  35. };
  36. m_ResourceCollection.OnLoadingAsset += delegate (int index, int count)
  37. {
  38. if (OnLoadingAsset != null)
  39. {
  40. OnLoadingAsset(index, count);
  41. }
  42. };
  43. m_ResourceCollection.OnLoadCompleted += delegate ()
  44. {
  45. if (OnLoadCompleted != null)
  46. {
  47. OnLoadCompleted();
  48. }
  49. };
  50. m_DependencyDatas = new Dictionary<string, DependencyData>(StringComparer.Ordinal);
  51. m_ScatteredAssets = new Dictionary<string, List<Asset>>(StringComparer.Ordinal);
  52. m_AnalyzedStamps = new HashSet<Stamp>();
  53. m_CircularDependencyDatas = new List<string[]>();
  54. }
  55. public event GameFrameworkAction<int, int> OnLoadingResource = null;
  56. public event GameFrameworkAction<int, int> OnLoadingAsset = null;
  57. public event GameFrameworkAction OnLoadCompleted = null;
  58. public event GameFrameworkAction<int, int> OnAnalyzingAsset = null;
  59. public event GameFrameworkAction OnAnalyzeCompleted = null;
  60. public void Clear()
  61. {
  62. m_ResourceCollection.Clear();
  63. m_DependencyDatas.Clear();
  64. m_ScatteredAssets.Clear();
  65. m_CircularDependencyDatas.Clear();
  66. m_AnalyzedStamps.Clear();
  67. }
  68. public bool Prepare()
  69. {
  70. m_ResourceCollection.Clear();
  71. return m_ResourceCollection.Load();
  72. }
  73. public void Analyze()
  74. {
  75. m_DependencyDatas.Clear();
  76. m_ScatteredAssets.Clear();
  77. m_CircularDependencyDatas.Clear();
  78. m_AnalyzedStamps.Clear();
  79. HashSet<string> scriptAssetNames = GetFilteredAssetNames("t:Script");
  80. Asset[] assets = m_ResourceCollection.GetAssets();
  81. int count = assets.Length;
  82. for (int i = 0; i < count; i++)
  83. {
  84. if (OnAnalyzingAsset != null)
  85. {
  86. OnAnalyzingAsset(i, count);
  87. }
  88. string assetName = assets[i].Name;
  89. if (string.IsNullOrEmpty(assetName))
  90. {
  91. Debug.LogWarning(Utility.Text.Format("Can not find asset by guid '{0}'.", assets[i].Guid));
  92. continue;
  93. }
  94. DependencyData dependencyData = new DependencyData();
  95. AnalyzeAsset(assetName, assets[i], dependencyData, scriptAssetNames);
  96. dependencyData.RefreshData();
  97. m_DependencyDatas.Add(assetName, dependencyData);
  98. }
  99. foreach (List<Asset> scatteredAsset in m_ScatteredAssets.Values)
  100. {
  101. scatteredAsset.Sort((a, b) => a.Name.CompareTo(b.Name));
  102. }
  103. m_CircularDependencyDatas.AddRange(new CircularDependencyChecker(m_AnalyzedStamps.ToArray()).Check());
  104. if (OnAnalyzeCompleted != null)
  105. {
  106. OnAnalyzeCompleted();
  107. }
  108. }
  109. private void AnalyzeAsset(string assetName, Asset hostAsset, DependencyData dependencyData, HashSet<string> scriptAssetNames)
  110. {
  111. string[] dependencyAssetNames = AssetDatabase.GetDependencies(assetName, false);
  112. foreach (string dependencyAssetName in dependencyAssetNames)
  113. {
  114. if (scriptAssetNames.Contains(dependencyAssetName))
  115. {
  116. continue;
  117. }
  118. if (dependencyAssetName == assetName)
  119. {
  120. continue;
  121. }
  122. if (dependencyAssetName.EndsWith(".unity", StringComparison.Ordinal))
  123. {
  124. // 忽略对场景的依赖
  125. continue;
  126. }
  127. Stamp stamp = new Stamp(hostAsset.Name, dependencyAssetName);
  128. if (m_AnalyzedStamps.Contains(stamp))
  129. {
  130. continue;
  131. }
  132. m_AnalyzedStamps.Add(stamp);
  133. string guid = AssetDatabase.AssetPathToGUID(dependencyAssetName);
  134. if (string.IsNullOrEmpty(guid))
  135. {
  136. Debug.LogWarning(Utility.Text.Format("Can not find guid by asset '{0}'.", dependencyAssetName));
  137. continue;
  138. }
  139. Asset asset = m_ResourceCollection.GetAsset(guid);
  140. if (asset != null)
  141. {
  142. dependencyData.AddDependencyAsset(asset);
  143. }
  144. else
  145. {
  146. dependencyData.AddScatteredDependencyAsset(dependencyAssetName);
  147. List<Asset> scatteredAssets = null;
  148. if (!m_ScatteredAssets.TryGetValue(dependencyAssetName, out scatteredAssets))
  149. {
  150. scatteredAssets = new List<Asset>();
  151. m_ScatteredAssets.Add(dependencyAssetName, scatteredAssets);
  152. }
  153. scatteredAssets.Add(hostAsset);
  154. AnalyzeAsset(dependencyAssetName, hostAsset, dependencyData, scriptAssetNames);
  155. }
  156. }
  157. }
  158. public Asset GetAsset(string assetName)
  159. {
  160. return m_ResourceCollection.GetAsset(AssetDatabase.AssetPathToGUID(assetName));
  161. }
  162. public string[] GetAssetNames()
  163. {
  164. return GetAssetNames(AssetsOrder.AssetNameAsc, null);
  165. }
  166. public string[] GetAssetNames(AssetsOrder order, string filter)
  167. {
  168. HashSet<string> filteredAssetNames = GetFilteredAssetNames(filter);
  169. IEnumerable<KeyValuePair<string, DependencyData>> filteredResult = m_DependencyDatas.Where(pair => filteredAssetNames.Contains(pair.Key));
  170. IEnumerable<KeyValuePair<string, DependencyData>> orderedResult = null;
  171. switch (order)
  172. {
  173. case AssetsOrder.AssetNameAsc:
  174. orderedResult = filteredResult.OrderBy(pair => pair.Key);
  175. break;
  176. case AssetsOrder.AssetNameDesc:
  177. orderedResult = filteredResult.OrderByDescending(pair => pair.Key);
  178. break;
  179. case AssetsOrder.DependencyResourceCountAsc:
  180. orderedResult = filteredResult.OrderBy(pair => pair.Value.DependencyResourceCount);
  181. break;
  182. case AssetsOrder.DependencyResourceCountDesc:
  183. orderedResult = filteredResult.OrderByDescending(pair => pair.Value.DependencyResourceCount);
  184. break;
  185. case AssetsOrder.DependencyAssetCountAsc:
  186. orderedResult = filteredResult.OrderBy(pair => pair.Value.DependencyAssetCount);
  187. break;
  188. case AssetsOrder.DependencyAssetCountDesc:
  189. orderedResult = filteredResult.OrderByDescending(pair => pair.Value.DependencyAssetCount);
  190. break;
  191. case AssetsOrder.ScatteredDependencyAssetCountAsc:
  192. orderedResult = filteredResult.OrderBy(pair => pair.Value.ScatteredDependencyAssetCount);
  193. break;
  194. case AssetsOrder.ScatteredDependencyAssetCountDesc:
  195. orderedResult = filteredResult.OrderByDescending(pair => pair.Value.ScatteredDependencyAssetCount);
  196. break;
  197. default:
  198. orderedResult = filteredResult;
  199. break;
  200. }
  201. return orderedResult.Select(pair => pair.Key).ToArray();
  202. }
  203. public DependencyData GetDependencyData(string assetName)
  204. {
  205. DependencyData dependencyData = null;
  206. if (m_DependencyDatas.TryGetValue(assetName, out dependencyData))
  207. {
  208. return dependencyData;
  209. }
  210. return dependencyData;
  211. }
  212. public string[] GetScatteredAssetNames()
  213. {
  214. return GetScatteredAssetNames(ScatteredAssetsOrder.HostAssetCountDesc, null);
  215. }
  216. public string[] GetScatteredAssetNames(ScatteredAssetsOrder order, string filter)
  217. {
  218. HashSet<string> filterAssetNames = GetFilteredAssetNames(filter);
  219. IEnumerable<KeyValuePair<string, List<Asset>>> filteredResult = m_ScatteredAssets.Where(pair => filterAssetNames.Contains(pair.Key) && pair.Value.Count > 1);
  220. IEnumerable<KeyValuePair<string, List<Asset>>> orderedResult = null;
  221. switch (order)
  222. {
  223. case ScatteredAssetsOrder.AssetNameAsc:
  224. orderedResult = filteredResult.OrderBy(pair => pair.Key);
  225. break;
  226. case ScatteredAssetsOrder.AssetNameDesc:
  227. orderedResult = filteredResult.OrderByDescending(pair => pair.Key);
  228. break;
  229. case ScatteredAssetsOrder.HostAssetCountAsc:
  230. orderedResult = filteredResult.OrderBy(pair => pair.Value.Count);
  231. break;
  232. case ScatteredAssetsOrder.HostAssetCountDesc:
  233. orderedResult = filteredResult.OrderByDescending(pair => pair.Value.Count);
  234. break;
  235. default:
  236. orderedResult = filteredResult;
  237. break;
  238. }
  239. return orderedResult.Select(pair => pair.Key).ToArray();
  240. }
  241. public Asset[] GetHostAssets(string scatteredAssetName)
  242. {
  243. List<Asset> assets = null;
  244. if (m_ScatteredAssets.TryGetValue(scatteredAssetName, out assets))
  245. {
  246. return assets.ToArray();
  247. }
  248. return null;
  249. }
  250. public string[][] GetCircularDependencyDatas()
  251. {
  252. return m_CircularDependencyDatas.ToArray();
  253. }
  254. private HashSet<string> GetFilteredAssetNames(string filter)
  255. {
  256. string[] filterAssetGuids = AssetDatabase.FindAssets(filter);
  257. HashSet<string> filterAssetNames = new HashSet<string>();
  258. foreach (string filterAssetGuid in filterAssetGuids)
  259. {
  260. filterAssetNames.Add(AssetDatabase.GUIDToAssetPath(filterAssetGuid));
  261. }
  262. return filterAssetNames;
  263. }
  264. }
  265. }