mirror of https://github.com/ogoun/Zero.git
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
95 lines
2.9 KiB
95 lines
2.9 KiB
using System.Collections.Generic;
|
|
|
|
namespace System.Linq
|
|
{
|
|
public static class LinqExtension
|
|
{
|
|
public static IEnumerable<T[]> ZipLongest<T>(this IEnumerable<T> left, IEnumerable<T> right)
|
|
{
|
|
IEnumerator<T> leftEnumerator = left.GetEnumerator();
|
|
IEnumerator<T> rightEnumerator = right.GetEnumerator();
|
|
|
|
bool hasLeft = leftEnumerator.MoveNext();
|
|
bool hasRight = rightEnumerator.MoveNext();
|
|
|
|
while (hasLeft || hasRight)
|
|
{
|
|
if (hasLeft && hasRight)
|
|
{
|
|
yield return new T[] { leftEnumerator.Current, rightEnumerator.Current };
|
|
}
|
|
else if (hasLeft)
|
|
{
|
|
yield return new T[] { leftEnumerator.Current, default(T) };
|
|
}
|
|
else if (hasRight)
|
|
{
|
|
yield return new T[] { default(T), rightEnumerator.Current };
|
|
}
|
|
|
|
hasLeft = leftEnumerator.MoveNext();
|
|
hasRight = rightEnumerator.MoveNext();
|
|
}
|
|
}
|
|
|
|
public static IEnumerable<TSource> DistinctBy<TSource, TKey>
|
|
(this IEnumerable<TSource> source, Func<TSource, TKey> keySelector)
|
|
{
|
|
if (source != null)
|
|
{
|
|
var seenKeys = new HashSet<TKey>();
|
|
foreach (TSource element in source)
|
|
{
|
|
if (seenKeys.Add(keySelector(element)))
|
|
{
|
|
yield return element;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
public static IList<TSource> Materialize<TSource>(this IEnumerable<TSource> source)
|
|
{
|
|
if (source is IList<TSource>)
|
|
{
|
|
// Already a list, use it as is
|
|
return (IList<TSource>)source;
|
|
}
|
|
else
|
|
{
|
|
// Not a list, materialize it to a list
|
|
return source.ToList();
|
|
}
|
|
}
|
|
public static IEnumerable<IEnumerable<T>> Chunkify<T>(this IEnumerable<T> source, int size)
|
|
{
|
|
if (source == null)
|
|
{
|
|
yield break;
|
|
}
|
|
if (size <= 0)
|
|
{
|
|
throw new ArgumentException("chunkSize must be greater than 0.");
|
|
}
|
|
T[] arr = new T[size];
|
|
int index = 0;
|
|
foreach (var obj in source)
|
|
{
|
|
arr[index] = obj;
|
|
index++;
|
|
if (index >= size)
|
|
{
|
|
yield return arr;
|
|
index = 0;
|
|
arr = new T[size];
|
|
}
|
|
}
|
|
if (index > 0)
|
|
{
|
|
var narr = new T[index];
|
|
Array.Copy(arr, narr, index);
|
|
yield return narr;
|
|
}
|
|
}
|
|
}
|
|
} |