Searching for usages of SpriteAtlas will now include its Sprites, as well (#43)

Author: arkms

Plugins/AssetUsageDetector/Editor/AssetUsageDetectorSearchFunctions.cs

@@ -211,8 +211,8 @@ private static string ExtractGUIDFromString( string str )
 		private readonly Func<Cubemap> defaultReflectionProbeGetter = (Func<Cubemap>) Delegate.CreateDelegate( typeof( Func<Cubemap> ), typeof( RenderSettings ).GetProperty( "defaultReflection", BindingFlags.Public | BindingFlags.NonPublic | BindingFlags.Static ).GetGetMethod( true ) );
 #endif
 
+		internal static readonly Func<SpriteAtlas, Sprite[]> spriteAtlasPackedSpritesGetter = (Func<SpriteAtlas, Sprite[]>) Delegate.CreateDelegate( typeof( Func<SpriteAtlas, Sprite[]> ), typeof( SpriteAtlasExtensions ).GetMethod( "GetPackedSprites", BindingFlags.Public | BindingFlags.NonPublic | BindingFlags.Static ) );
 #if ASSET_USAGE_ADDRESSABLES
-		private readonly Func<SpriteAtlas, Sprite[]> spriteAtlasPackedSpritesGetter = (Func<SpriteAtlas, Sprite[]>) Delegate.CreateDelegate( typeof( Func<SpriteAtlas, Sprite[]> ), typeof( SpriteAtlasExtensions ).GetMethod( "GetPackedSprites", BindingFlags.Public | BindingFlags.NonPublic | BindingFlags.Static ) );
 		private readonly PropertyInfo assetReferenceSubObjectTypeGetter = typeof( AssetReference ).GetProperty( "SubOjbectType", BindingFlags.Public | BindingFlags.NonPublic | BindingFlags.Instance );
 #endif
 

Plugins/AssetUsageDetector/Editor/ObjectToSearch.cs

@@ -2,6 +2,7 @@
 using System.Collections.Generic;
 using UnityEditor;
 using UnityEngine;
+using UnityEngine.U2D;
 using Object = UnityEngine.Object;
 
 namespace AssetUsageDetectorNamespace
@@ -95,6 +96,20 @@ private void AddSubAssets( Object target, bool includeTarget, bool? shouldSearch
 					}
 				}
 
+				// If is an Atlas we get sprites directly as GetSubAssets return the final texture
+				if( target is SpriteAtlas spriteAtlas )
+				{
+					Sprite[] packedSprites = AssetUsageDetector.spriteAtlasPackedSpritesGetter( spriteAtlas );
+					if( packedSprites != null )
+					{
+						for( int i = 0; i < packedSprites.Length; i++ )
+						{
+							if( packedSprites[i] != null && currentSubAssets.Add( packedSprites[i] ) )
+								subAssets.Add( new SubAsset( packedSprites[i], shouldSearchChildren ?? true ) );
+						}
+					}
+				}
+
 				// Find sub-asset(s) of the asset (if any)
 				Object[] assets = AssetDatabase.LoadAllAssetsAtPath( AssetDatabase.GetAssetPath( target ) );
 				for( int i = 0; i < assets.Length; i++ )
@@ -117,4 +132,4 @@ private void AddSubAssets( Object target, bool includeTarget, bool? shouldSearch
 			}
 		}
 	}
-}
+}