|
|
|
|
using System.Collections.Generic;
|
|
|
|
|
using MemoryPools.Collections.Specialized;
|
|
|
|
|
|
|
|
|
|
namespace MemoryPools.Collections.Linq
|
|
|
|
|
{
|
|
|
|
|
internal class ExceptExprEnumerable<T> : IPoolingEnumerable<T>
|
|
|
|
|
{
|
|
|
|
|
private int _count;
|
|
|
|
|
private IPoolingEnumerable<T> _src;
|
|
|
|
|
private IEqualityComparer<T> _comparer;
|
|
|
|
|
private PoolingDictionary<T, int> _except;
|
|
|
|
|
|
|
|
|
|
public ExceptExprEnumerable<T> Init(IPoolingEnumerable<T> src, PoolingDictionary<T, int> except, IEqualityComparer<T> comparer = default)
|
|
|
|
|
{
|
|
|
|
|
_src = src;
|
|
|
|
|
_except = except;
|
|
|
|
|
_comparer = comparer;
|
|
|
|
|
_count = 0;
|
|
|
|
|
return this;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public IPoolingEnumerator<T> GetEnumerator()
|
|
|
|
|
{
|
|
|
|
|
_count++;
|
|
|
|
|
return Pool<ExceptExprEnumerator>.Get().Init(this, _src.GetEnumerator());
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
private void Dispose()
|
|
|
|
|
{
|
|
|
|
|
if(_count == 0) return;
|
|
|
|
|
_count--;
|
|
|
|
|
if (_count == 0)
|
|
|
|
|
{
|
|
|
|
|
_src = default;
|
|
|
|
|
_except?.Dispose();
|
|
|
|
|
Pool<PoolingDictionary<T, int>>.Return(_except);
|
|
|
|
|
_except = default;
|
|
|
|
|
Pool<ExceptExprEnumerable<T>>.Return(this);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
internal class ExceptExprEnumerator : IPoolingEnumerator<T>
|
|
|
|
|
{
|
|
|
|
|
private ExceptExprEnumerable<T> _parent;
|
|
|
|
|
private IPoolingEnumerator<T> _src;
|
|
|
|
|
|
|
|
|
|
public ExceptExprEnumerator Init(ExceptExprEnumerable<T> parent, IPoolingEnumerator<T> src)
|
|
|
|
|
{
|
|
|
|
|
_src = src;
|
|
|
|
|
_parent = parent;
|
|
|
|
|
return this;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public bool MoveNext()
|
|
|
|
|
{
|
|
|
|
|
while (_src.MoveNext())
|
|
|
|
|
{
|
|
|
|
|
if(_parent._except.ContainsKey(_src.Current)) continue;
|
|
|
|
|
return true;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return false;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public void Reset() => _src.Reset();
|
|
|
|
|
|
|
|
|
|
object IPoolingEnumerator.Current => Current;
|
|
|
|
|
|
|
|
|
|
public T Current => _src.Current;
|
|
|
|
|
|
|
|
|
|
public void Dispose()
|
|
|
|
|
{
|
|
|
|
|
_src?.Dispose();
|
|
|
|
|
_src = null;
|
|
|
|
|
|
|
|
|
|
_parent?.Dispose();
|
|
|
|
|
_parent = default;
|
|
|
|
|
|
|
|
|
|
Pool<ExceptExprEnumerator>.Return(this);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
IPoolingEnumerator IPoolingEnumerable.GetEnumerator() => GetEnumerator();
|
|
|
|
|
}
|
|
|
|
|
}
|