File size: 1,415 Bytes
18a519f | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 | 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;
}
}
}
|