52 lines
1.4 KiB
C#
52 lines
1.4 KiB
C#
|
using System.Collections;
|
||
|
|
using System.Reflection;
|
||
|
|
|
||
|
|
namespace UnityEditor.TestTools.TestRunner
|
||
|
|
{
|
||
|
|
internal class EnumeratorStepHelper
|
||
|
|
{
|
||
|
|
private static int m_PC;
|
||
|
|
|
||
|
|
public static void SetEnumeratorPC(int pc)
|
||
|
|
{
|
||
|
|
m_PC = pc;
|
||
|
|
}
|
||
|
|
|
||
|
|
/// <summary>
|
||
|
|
/// Gets the current enumerator PC
|
||
|
|
/// </summary>
|
||
|
|
/// <returns>
|
||
|
|
/// The PC
|
||
|
|
/// 0 if no current Enumeration
|
||
|
|
/// </returns>
|
||
|
|
public static int GetEnumeratorPC(IEnumerator enumerator)
|
||
|
|
{
|
||
|
|
if (enumerator == null)
|
||
|
|
{
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
return (int)GetPCFieldInfo(enumerator).GetValue(enumerator);
|
||
|
|
}
|
||
|
|
|
||
|
|
public static bool UpdateEnumeratorPcIfNeeded(IEnumerator enumerator)
|
||
|
|
{
|
||
|
|
if (m_PC != 0)
|
||
|
|
{
|
||
|
|
GetPCFieldInfo(enumerator).SetValue(enumerator, m_PC);
|
||
|
|
m_PC = 0;
|
||
|
|
return true;
|
||
|
|
}
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
|
||
|
|
private static FieldInfo GetPCFieldInfo(IEnumerator enumerator)
|
||
|
|
{
|
||
|
|
var field = enumerator.GetType().GetField("$PC", BindingFlags.NonPublic | BindingFlags.Instance);
|
||
|
|
if (field == null) // Roslyn
|
||
|
|
field = enumerator.GetType().GetField("<>1__state", BindingFlags.NonPublic | BindingFlags.Instance);
|
||
|
|
|
||
|
|
return field;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|