//public
public void AddState(Type state)
{
if (state == null)
{
throw new ArgumentNullException("state");
}
if (!ReflectUtil.DoesExtend(typeof(FiniteState), state))
{
throw new Exception("state does not extend FiniteState.");
}
FiniteState ns = null;
try {
ns = (FiniteState)Activator.CreateInstance(state, this);
} catch (Exception ex) {
throw new Exception("Cannot create state.", ex);
}
FiniteState oldState = null;
if (states.TryGetValue(state, out oldState))
{
if (currentState == oldState)
{
currentState = states[state] = ns;
currentState.Enter();
}
else
{
states[state] = ns;
}
if (ReflectUtil.DoesExtend(typeof(IDisposable), oldState.GetType()))
{
((IDisposable)oldState).Dispose();
}
}
else
{
states.Add(state, ns);
}
if (currentState == null)
{
currentState = ns;
currentState.Enter();
}
}
public bool TrySwapStates(Type state)
{
if (state == null)
{
throw new ArgumentNullException("state");
}
if (currentState == null || !states.ContainsKey(state) || !currentState.HasExitState(state))
{
return(false);
}
currentState.Exit();
currentState = states[state];
currentState.Enter();
return(true);
}
