EnumeratorStepHelper.cs
1.38 KB
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
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;
}
}
}