Extern/Beef
1
0
Fork 0
mirror of https://github.com/beefytech/Beef.git synced 2025-06-24 18:48:01 +02:00
Code Activity

Implement IRefEnumerator<T> in HashSet Enumerator

Browse source
This commit is contained in:
disarray2077 2024-05-17 15:17:51 -03:00 • committed by GitHub
parent 78549fa364
commit c64dc37d61
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 22 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

29
BeefLibs/corlib/src/Collections/HashSet.bf
View file

@ -1217,14 +1217,14 @@ namespace System.Collections
public int32 mNext; // Index of next entry, -1 if last
}
public struct Enumerator : IEnumerator<T>
public struct Enumerator : IRefEnumerator<T*>, IEnumerator<T>
{
private HashSet<T> mSet;
private int_cosize mIndex;
#if VERSION_HASHSET
private int32 mVersion;
#endif
private T mCurrent;
private T* mCurrent;
public this(HashSet<T> set)
{
@ -1233,7 +1233,7 @@ namespace System.Collections
#if VERSION_HASHSET
mVersion = set.mVersion;
#endif
mCurrent = default;
mCurrent = null;
}
#if VERSION_HASHSET
@ -1257,14 +1257,14 @@ namespace System.Collections
{
if (mSet.mSlots[mIndex].mHashCode >= 0)
{
mCurrent = mSet.mSlots[mIndex].mValue;
mCurrent = &mSet.mSlots[mIndex].mValue;
mIndex++;
return true;
}
mIndex++;
}
mIndex = mSet.mLastIndex + 1;
mCurrent = default(T);
mCurrent = null;
return false;
}
@ -1272,7 +1272,15 @@ namespace System.Collections
{
get
{
return mCurrent;
return *mCurrent;
}
}
public ref T CurrentRef
{
get
{
return ref *mCurrent;
}
}
@ -1292,7 +1300,7 @@ namespace System.Collections
CheckVersion();
#endif
mIndex = 0;
mCurrent = default(T);
mCurrent = null;
}
public Result<T> GetNext() mut
@ -1301,6 +1309,13 @@ namespace System.Collections
return .Err;
return Current;
}
public Result<T*> GetNextRef() mut
{
if (!MoveNext())
return .Err;
return &CurrentRef;
}
}
}