initial commit

Library/PackageCache/com.unity.scriptablebuildpipeline@1.21.5/Editor/Injector/ContextInjector.cs

@@ -0,0 +1,109 @@
+using System;
+using System.Reflection;
+using UnityEditor.Build.Pipeline.Interfaces;
+
+namespace UnityEditor.Build.Pipeline.Injector
+{
+    /// <summary>
+    /// Use to pass around information between build tasks.
+    /// </summary>
+    [AttributeUsage(AttributeTargets.Field)]
+    public class InjectContextAttribute : Attribute
+    {
+        //public string Identifier { get; set; }
+        /// <summary>
+        /// Stores the how the attribute is used among build tasks.
+        /// </summary>
+        public ContextUsage Usage { get; set; }
+        /// <summary>
+        /// Stores whether using the context attribute is optional.
+        /// </summary>
+        public bool Optional { get; set; }
+
+        /// <summary>
+        /// Creates a new context attribute that stores information that can be passed between build tasks.
+        /// </summary>
+        /// <param name="usage">The usage behavior for the attribute. By default it is set to <see cref="ContextUsage.InOut"/>.</param>
+        /// <param name="optional">Set to true if using the attribute is optional. Set to false otherwise.</param>
+        public InjectContextAttribute(ContextUsage usage = ContextUsage.InOut, bool optional = false)
+        {
+            this.Usage = usage;
+            Optional = optional;
+        }
+    }
+
+    /// <summary>
+    /// Options for how the attribute is used among build tasks. It can be either injected to and or extracted from a build task.
+    /// </summary>
+    public enum ContextUsage
+    {
+        /// <summary>
+        /// Use to indicate that the attribute can be injected to and extracted from a build task.
+        /// </summary>
+        InOut,
+        /// <summary>
+        /// Use to indicate that the attribute can only be injected to a build task.
+        /// </summary>
+        In,
+        /// <summary>
+        /// Use to indicate that the attribute can only be extracted from a build task.
+        /// </summary>
+        Out
+    }
+
+    class ContextInjector
+    {
+        public static void Inject(IBuildContext context, object obj)
+        {
+            FieldInfo[] fields = obj.GetType().GetFields(BindingFlags.Instance | BindingFlags.NonPublic);
+            foreach (FieldInfo field in fields)
+            {
+                object[] attrs = field.GetCustomAttributes(typeof(InjectContextAttribute), true);
+                if (attrs.Length == 0)
+                    continue;
+
+                InjectContextAttribute attr = attrs[0] as InjectContextAttribute;
+                if (attr == null || attr.Usage == ContextUsage.Out)
+                    continue;
+
+                object injectionObject;
+                if (field.FieldType == typeof(IBuildContext))
+                    injectionObject = context;
+                else if (!attr.Optional)
+                    injectionObject = context.GetContextObject(field.FieldType);
+                else
+                {
+                    IContextObject contextObject;
+                    context.TryGetContextObject(field.FieldType, out contextObject);
+                    injectionObject = contextObject;
+                }
+
+                field.SetValue(obj, injectionObject);
+            }
+        }
+
+        public static void Extract(IBuildContext context, object obj)
+        {
+            FieldInfo[] fields = obj.GetType().GetFields(BindingFlags.Instance | BindingFlags.NonPublic);
+            foreach (FieldInfo field in fields)
+            {
+                object[] attrs = field.GetCustomAttributes(typeof(InjectContextAttribute), true);
+                if (attrs.Length == 0)
+                    continue;
+
+                InjectContextAttribute attr = attrs[0] as InjectContextAttribute;
+                if (attr == null || attr.Usage == ContextUsage.In)
+                    continue;
+
+                if (field.FieldType == typeof(IBuildContext))
+                    throw new InvalidOperationException("IBuildContext can only be used with the ContextUsage.In option.");
+
+                IContextObject contextObject = field.GetValue(obj) as IContextObject;
+                if (!attr.Optional)
+                    context.SetContextObject(field.FieldType, contextObject);
+                else if (contextObject != null)
+                    context.SetContextObject(field.FieldType, contextObject);
+            }
+        }
+    }
+}

Library/PackageCache/com.unity.scriptablebuildpipeline@1.21.5/Editor/Injector/ContextInjector.cs.meta

@@ -0,0 +1,11 @@
+fileFormatVersion: 2
+guid: 29402082393e3a1439cfaa6cc77aa31b
+MonoImporter:
+  externalObjects: {}
+  serializedVersion: 2
+  defaultReferences: []
+  executionOrder: 0
+  icon: {instanceID: 0}
+  userData: 
+  assetBundleName: 
+  assetBundleVariant: