12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152 |
- using System;
- 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;
- }
- }
- }
|