feat: use new IEnumeratable type
This commit is contained in:
64
BallanceTasEditor/Utils/CountableEnumerable.cs
Normal file
64
BallanceTasEditor/Utils/CountableEnumerable.cs
Normal file
@ -0,0 +1,64 @@
|
||||
using System;
|
||||
using System.Collections.Generic;
|
||||
using System.Linq;
|
||||
using System.Text;
|
||||
using System.Threading.Tasks;
|
||||
|
||||
namespace BallanceTasEditor.Utils {
|
||||
/// <summary>
|
||||
/// 一种提前给定元素个数的的IEnumerable。
|
||||
/// </summary>
|
||||
public sealed class CountableEnumerable<T> {
|
||||
/// <summary>
|
||||
/// 以直接方式构建。
|
||||
/// </summary>
|
||||
/// <param name="enumerable">一个迭代器,其最多只能迭代给定次数。</param>
|
||||
/// <param name="count">迭代器会迭代的次数。</param>
|
||||
public CountableEnumerable(IEnumerable<T> enumerable, int count) {
|
||||
m_Inner = enumerable;
|
||||
m_Count = count;
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// 从数组便捷构建。
|
||||
/// </summary>
|
||||
/// <param name="array">要使用的数组。</param>
|
||||
public CountableEnumerable(T[] array) {
|
||||
m_Inner = array;
|
||||
m_Count = array.Length;
|
||||
}
|
||||
|
||||
private IEnumerable<T> m_Inner;
|
||||
private int m_Count;
|
||||
|
||||
/// <summary>
|
||||
/// 获取迭代器对象。
|
||||
/// </summary>
|
||||
/// <returns>用于迭代的迭代器。</returns>
|
||||
/// <exception cref="ArgumentException">当迭代器迭代次数与给定次数不匹配时。</exception>
|
||||
public IEnumerable<T> GetInner() {
|
||||
int counter = 0;
|
||||
foreach (var item in m_Inner) {
|
||||
if (counter >= m_Count) {
|
||||
throw new ArgumentException("Given IEnumerable<T> is not stopped at given count.");
|
||||
} else {
|
||||
yield return item;
|
||||
++counter;
|
||||
}
|
||||
}
|
||||
|
||||
if (counter != m_Count) {
|
||||
throw new ArgumentException("Given IEnumerable<T> is not stopped at given count.");
|
||||
}
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// 获取该迭代器会迭代的次数。
|
||||
/// </summary>
|
||||
/// <returns>迭代器会迭代的次数,用于给使用该结构的方法提前分配必要的空间。</returns>
|
||||
public int GetCount() {
|
||||
return m_Count;
|
||||
}
|
||||
|
||||
}
|
||||
}
|
||||
@ -30,7 +30,7 @@ namespace BallanceTasEditor.Utils {
|
||||
/// <param name="index">要在前方插入数据的元素的索引。</param>
|
||||
/// <param name="items">要插入的元素的迭代器。</param>
|
||||
/// <exception cref="ArgumentException">给定的索引超出范围。</exception>
|
||||
void Insert(int index, IEnumerable<T> items);
|
||||
void Insert(int index, CountableEnumerable<T> items);
|
||||
/// <summary>
|
||||
/// 从给定单元开始,移除给定个数的元素。
|
||||
/// </summary>
|
||||
@ -70,7 +70,7 @@ namespace BallanceTasEditor.Utils {
|
||||
throw new NotImplementedException();
|
||||
}
|
||||
|
||||
public void Insert(int index, IEnumerable<T> items) {
|
||||
public void Insert(int index, CountableEnumerable<T> items) {
|
||||
throw new NotImplementedException();
|
||||
}
|
||||
|
||||
@ -108,8 +108,8 @@ namespace BallanceTasEditor.Utils {
|
||||
return m_Container[index];
|
||||
}
|
||||
|
||||
public void Insert(int index, IEnumerable<T> items) {
|
||||
m_Container.InsertRange(index, items);
|
||||
public void Insert(int index, CountableEnumerable<T> items) {
|
||||
m_Container.InsertRange(index, items.GetInner());
|
||||
}
|
||||
|
||||
public void Remove(int index, int count) {
|
||||
@ -219,11 +219,11 @@ namespace BallanceTasEditor.Utils {
|
||||
}
|
||||
}
|
||||
|
||||
public void Insert(int index, IEnumerable<T> items) {
|
||||
public void Insert(int index, CountableEnumerable<T> items) {
|
||||
if (index < 0 || index > GetCount()) {
|
||||
throw new ArgumentOutOfRangeException("Index out of range.");
|
||||
} else if (index == GetCount()) {
|
||||
foreach (T item in items) {
|
||||
foreach (T item in items.GetInner()) {
|
||||
m_Container.AddLast(item);
|
||||
}
|
||||
|
||||
@ -234,7 +234,7 @@ namespace BallanceTasEditor.Utils {
|
||||
MoveToIndex(index);
|
||||
|
||||
int count = 0;
|
||||
foreach (T item in items) {
|
||||
foreach (T item in items.GetInner()) {
|
||||
m_Container.AddBefore(m_Cursor, item);
|
||||
++count;
|
||||
}
|
||||
|
||||
@ -13,6 +13,14 @@ namespace BallanceTasEditorTests.Utils {
|
||||
private static readonly int[] BLANK = { };
|
||||
private static readonly int[] PROBE = { 10, 20, 30, 40, 50 };
|
||||
|
||||
private static CountableEnumerable<int> GetCountableProbe() {
|
||||
return new CountableEnumerable<int>(PROBE);
|
||||
}
|
||||
|
||||
private static CountableEnumerable<int> GetCountableBlank() {
|
||||
return new CountableEnumerable<int>(BLANK);
|
||||
}
|
||||
|
||||
private static IEnumerable<object[]> TasStorageInstanceProvider {
|
||||
get {
|
||||
yield return new object[] { new ListTasStorage<int>() };
|
||||
@ -34,7 +42,7 @@ namespace BallanceTasEditorTests.Utils {
|
||||
AssertExtension.ThrowsDerivedException<ArgumentException>(() => storage.Visit(1));
|
||||
|
||||
// 设置数据
|
||||
storage.Insert(0, PROBE);
|
||||
storage.Insert(0, GetCountableProbe());
|
||||
// 访问数据
|
||||
AssertExtension.ThrowsDerivedException<ArgumentException>(() => storage.Visit(-1));
|
||||
for (int i = 0; i < PROBE.Length; i++) {
|
||||
@ -53,9 +61,9 @@ namespace BallanceTasEditorTests.Utils {
|
||||
// 和在非空时的头,中,尾分别插入的结果。
|
||||
|
||||
// 先检测空插入
|
||||
AssertExtension.ThrowsDerivedException<ArgumentException>(() => storage.Insert(-1, PROBE));
|
||||
AssertExtension.ThrowsDerivedException<ArgumentException>(() => storage.Insert(1, PROBE));
|
||||
storage.Insert(0, PROBE);
|
||||
AssertExtension.ThrowsDerivedException<ArgumentException>(() => storage.Insert(-1, GetCountableProbe()));
|
||||
AssertExtension.ThrowsDerivedException<ArgumentException>(() => storage.Insert(1, GetCountableProbe()));
|
||||
storage.Insert(0, GetCountableProbe());
|
||||
for (int i = 0; i < PROBE.Length; i++) {
|
||||
Assert.AreEqual(storage.Visit(i), PROBE[i]);
|
||||
}
|
||||
@ -65,8 +73,8 @@ namespace BallanceTasEditorTests.Utils {
|
||||
foreach (var index in indices) {
|
||||
// 清空,一次插入,然后二次插入
|
||||
storage.Clear();
|
||||
storage.Insert(0, PROBE);
|
||||
storage.Insert(index, PROBE);
|
||||
storage.Insert(0, GetCountableProbe());
|
||||
storage.Insert(index, GetCountableProbe());
|
||||
|
||||
// 用List做正确模拟
|
||||
var expected = new List<int>();
|
||||
@ -96,7 +104,7 @@ namespace BallanceTasEditorTests.Utils {
|
||||
foreach (var index in indices) {
|
||||
// 清空,插入,删除
|
||||
storage.Clear();
|
||||
storage.Insert(0, PROBE);
|
||||
storage.Insert(0, GetCountableProbe());
|
||||
storage.Remove(index, 1);
|
||||
|
||||
// 用List做正确模拟
|
||||
@ -119,7 +127,7 @@ namespace BallanceTasEditorTests.Utils {
|
||||
[DynamicData(nameof(TasStorageInstanceProvider))]
|
||||
public void ClearTest(ITasStorage<int> storage) {
|
||||
// 设置数据后清空
|
||||
storage.Insert(0, PROBE);
|
||||
storage.Insert(0, GetCountableProbe());
|
||||
storage.Clear();
|
||||
|
||||
// 检查是否为空
|
||||
@ -136,7 +144,7 @@ namespace BallanceTasEditorTests.Utils {
|
||||
Assert.IsTrue(storage.IsEmpty());
|
||||
|
||||
// 插入数据后再检查
|
||||
storage.Insert(0, PROBE);
|
||||
storage.Insert(0, GetCountableProbe());
|
||||
Assert.IsFalse(storage.IsEmpty());
|
||||
}
|
||||
|
||||
@ -150,7 +158,7 @@ namespace BallanceTasEditorTests.Utils {
|
||||
Assert.AreEqual(storage.GetCount(), 0);
|
||||
|
||||
// 插入数据后再检查
|
||||
storage.Insert(0, PROBE);
|
||||
storage.Insert(0, GetCountableProbe());
|
||||
Assert.AreEqual(storage.GetCount(), PROBE.Length);
|
||||
}
|
||||
|
||||
@ -166,7 +174,7 @@ namespace BallanceTasEditorTests.Utils {
|
||||
Assert.AreEqual(storage.GetCount(), 0);
|
||||
|
||||
// 设置内容
|
||||
storage.Insert(0, PROBE);
|
||||
storage.Insert(0, GetCountableProbe());
|
||||
// 并再次检查大小
|
||||
Assert.IsFalse(storage.IsEmpty());
|
||||
Assert.AreEqual(storage.GetCount(), PROBE.Length);
|
||||
@ -185,7 +193,7 @@ namespace BallanceTasEditorTests.Utils {
|
||||
|
||||
// 清空后插入0项,然后确认
|
||||
storage.Clear();
|
||||
storage.Insert(0, BLANK);
|
||||
storage.Insert(0, GetCountableBlank());
|
||||
AssertExtension.ThrowsDerivedException<ArgumentException>(() => storage.Visit(-1));
|
||||
AssertExtension.ThrowsDerivedException<ArgumentException>(() => storage.Visit(0));
|
||||
AssertExtension.ThrowsDerivedException<ArgumentException>(() => storage.Visit(1));
|
||||
|
||||
Reference in New Issue
Block a user