AloneGame / Library / PackageCache / com.unity.collections / Unity.Collections / NativeParallelHashSetExtensions.tt

<#/*THIS IS A T4 FILE - see t4_text_templating.md for what it is and how to run codegen*/#>
<#@ template debug="True" #>
<#@ output extension=".gen.cs" #>
<#@ assembly name="System.Core" #>
using System;
using Unity.Collections.LowLevel.Unsafe;

namespace Unity.Collections
{
    /// <summary>
    /// Provides extension methods for sets.
    /// </summary>
    public unsafe static class HashSetExtensions
    {
<#
{
    foreach (var ContainerType in new[] {
        ( new[] { "NativeHashSet", "Count" } ),
        ( new[] { "NativeParallelHashSet", "Count()" } ),
    }) {

    foreach (var OtherContainerType in new[] {
        ( "FixedList128Bytes<T>" ),
        ( "FixedList32Bytes<T>" ),
        ( "FixedList4096Bytes<T>" ),
        ( "FixedList512Bytes<T>" ),
        ( "FixedList64Bytes<T>" ),
        ( "NativeArray<T>" ),
        ( "NativeHashSet<T>" ),
        ( "NativeHashSet<T>.ReadOnly" ),
        ( "NativeParallelHashSet<T>" ),
        ( "NativeParallelHashSet<T>.ReadOnly" ),
        ( "NativeList<T>" ),
    }) {
#>
        /// <summary>
        /// Removes the values from this set which are also present in another collection.
        /// </summary>
        /// <typeparam name="T">The type of values.</typeparam>
        /// <param name="container">The set to remove values from.</param>
        /// <param name="other">The collection to compare with.</param>
        public static void ExceptWith<T>(this ref <#=ContainerType[0]#><T> container, <#=OtherContainerType#> other)
            where T : unmanaged, IEquatable<T>
        {
            foreach (var item in other)
            {
                container.Remove(item);
            }
        }

        /// <summary>
        /// Removes the values from this set which are absent in another collection.
        /// </summary>
        /// <typeparam name="T">The type of values.</typeparam>
        /// <param name="container">The set to remove values from.</param>
        /// <param name="other">The collection to compare with.</param>
        public static void IntersectWith<T>(this ref <#=ContainerType[0]#><T> container, <#=OtherContainerType#> other)
            where T : unmanaged, IEquatable<T>
        {
            var result = new UnsafeList<T>(container.<#=ContainerType[1]#>, Allocator.Temp);

            foreach (var item in other)
            {
                if (container.Contains(item))
                {
                    result.Add(item);
                }
            }

            container.Clear();
            container.UnionWith(result);

            result.Dispose();
        }

        /// <summary>
        /// Adds all values from a collection to this set.
        /// </summary>
        /// <typeparam name="T">The type of values.</typeparam>
        /// <param name="container">The set to add values to.</param>
        /// <param name="other">The collection to copy values from.</param>
        public static void UnionWith<T>(this ref <#=ContainerType[0]#><T> container, <#=OtherContainerType#> other)
            where T : unmanaged, IEquatable<T>
        {
            foreach (var item in other)
            {
                container.Add(item);
            }
        }
<#}}}#>
    }
}