ZoeAppClass/Library/PackageCache/com.unity.scriptablebuildpi.../Editor/Tasks/StripUnusedSpriteSources.cs

57 lines
1.9 KiB
C#

using System.Collections.Generic;
using System.Linq;
using UnityEditor.Build.Content;
using UnityEditor.Build.Pipeline.Injector;
using UnityEditor.Build.Pipeline.Interfaces;
namespace UnityEditor.Build.Pipeline.Tasks
{
public class StripUnusedSpriteSources : IBuildTask
{
public int Version { get { return 2; } }
#pragma warning disable 649
[InjectContext]
IDependencyData m_DependencyData;
[InjectContext(ContextUsage.In, true)]
IBuildSpriteData m_SpriteData;
#pragma warning restore 649
public ReturnCode Run()
{
if (m_SpriteData == null || m_SpriteData.ImporterData.Count == 0)
return ReturnCode.SuccessNotRun;
if (EditorSettings.spritePackerMode == SpritePackerMode.Disabled)
return ReturnCode.SuccessNotRun;
var unusedSources = new HashSet<ObjectIdentifier>();
var textures = m_SpriteData.ImporterData.Values.Where(x => x.PackedSprite).Select(x => x.SourceTexture);
unusedSources.UnionWith(textures);
// Count refs from assets
var assetRefs = m_DependencyData.AssetInfo.SelectMany(x => x.Value.referencedObjects);
foreach (ObjectIdentifier reference in assetRefs)
unusedSources.Remove(reference);
// Count refs from scenes
var sceneRefs = m_DependencyData.SceneInfo.SelectMany(x => x.Value.referencedObjects);
foreach (ObjectIdentifier reference in sceneRefs)
unusedSources.Remove(reference);
SetOutputInformation(unusedSources);
return ReturnCode.Success;
}
public void SetOutputInformation(HashSet<ObjectIdentifier> unusedSources)
{
foreach (var source in unusedSources)
{
var assetInfo = m_DependencyData.AssetInfo[source.guid];
assetInfo.includedObjects.RemoveAt(0);
}
}
}
}