Построение государственной машины

---

Я смотрю на создание государственной машины для проекта. В проекте будет много алгоритмов, каждый из которых имеет в себе государственную машину. Возвращение перечисления для следующего состояния загрязнило бы перечисление базового класса, поэтому я не очень рад этому.

Ниже приведен рабочий код, который я придумал. Не без ума от возвращаемого типа объекта, но на самом деле не вижу способа обойти его.

У кого-нибудь есть какие-нибудь мысли?

Что я уже пробовал:

using System;

namespace StateMachine
{
	/// 
	public class StateMachine2BaseClass
	{
		protected void RunAlgorithm(Func<object> nextState)
		{
			while (nextState != null)
			{
				var currentState = nextState;
				nextState = (Func<object>)currentState();
			}
		}

		protected Func<object> End()
		{
			return null;
		}
	}

	/// 
	public class StateMachine2 : StateMachine2BaseClass
	{
		private int count;

		public void Run()
		{
			RunAlgorithm(Start);
		}

		private Func<object> Start()
		{
			Console.WriteLine("Start");
			return Step_1;
		}

		private Func<object> Step_1()
		{
			if (++count > 5)
				return Step_2;

			Console.WriteLine("Step 1");

			return Step_1;
		}

		private Func<object> Step_2()
		{
			Console.WriteLine("Step 2");
			return End;
		}
	}
}

---