73 lines
2.5 KiB
C#
73 lines
2.5 KiB
C#
using System;
|
|
using System.Collections.Generic;
|
|
using System.Linq;
|
|
using UnityEditor.Build.Content;
|
|
using UnityEditor.Build.Pipeline.Injector;
|
|
using UnityEditor.Build.Pipeline.Interfaces;
|
|
using UnityEditor.Build.Utilities;
|
|
|
|
namespace UnityEditor.Build.Pipeline.Tasks
|
|
{
|
|
/// <summary>
|
|
/// Optional build task that extracts all referenced MonoScripts and assigns them to the specified bundle
|
|
/// </summary>
|
|
public class CreateMonoScriptBundle : IBuildTask
|
|
{
|
|
static readonly GUID k_DefaultGuid = new GUID(CommonStrings.UnityDefaultResourceGuid);
|
|
/// <inheritdoc />
|
|
public int Version { get { return 1; } }
|
|
|
|
#pragma warning disable 649
|
|
[InjectContext(ContextUsage.In)]
|
|
IDependencyData m_DependencyData;
|
|
|
|
[InjectContext(ContextUsage.InOut, true)]
|
|
IBundleExplictObjectLayout m_Layout;
|
|
#pragma warning restore 649
|
|
|
|
/// <summary>
|
|
/// Stores the name for the MonoScript bundle.
|
|
/// </summary>
|
|
public string MonoScriptBundleName { get; set; }
|
|
|
|
/// <summary>
|
|
/// Create the MonoScript bundle.
|
|
/// </summary>
|
|
/// <param name="bundleName">The name of the bundle.</param>
|
|
public CreateMonoScriptBundle(string bundleName)
|
|
{
|
|
MonoScriptBundleName = bundleName;
|
|
}
|
|
|
|
/// <inheritdoc />
|
|
public ReturnCode Run()
|
|
{
|
|
HashSet<ObjectIdentifier> buildInObjects = new HashSet<ObjectIdentifier>();
|
|
foreach (AssetLoadInfo dependencyInfo in m_DependencyData.AssetInfo.Values)
|
|
buildInObjects.UnionWith(dependencyInfo.referencedObjects);
|
|
|
|
foreach (SceneDependencyInfo dependencyInfo in m_DependencyData.SceneInfo.Values)
|
|
buildInObjects.UnionWith(dependencyInfo.referencedObjects);
|
|
|
|
ObjectIdentifier[] usedSet = buildInObjects.ToArray();
|
|
Type[] usedTypes = BuildCacheUtility.GetMainTypeForObjects(usedSet);
|
|
|
|
if (m_Layout == null)
|
|
m_Layout = new BundleExplictObjectLayout();
|
|
|
|
Type monoScript = typeof(MonoScript);
|
|
for (int i = 0; i < usedTypes.Length; i++)
|
|
{
|
|
if (usedTypes[i] != monoScript || usedSet[i].guid == k_DefaultGuid)
|
|
continue;
|
|
|
|
m_Layout.ExplicitObjectLocation.Add(usedSet[i], MonoScriptBundleName);
|
|
}
|
|
|
|
if (m_Layout.ExplicitObjectLocation.Count == 0)
|
|
m_Layout = null;
|
|
|
|
return ReturnCode.Success;
|
|
}
|
|
}
|
|
}
|