123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089 |
- #region --------------- Copyright Dresser Wayne Pignone -------------
- #endregion
- #region Old change history
- #endregion
- using System;
- using System.Collections.Generic;
- using System.Linq;
- using Wayne.Lib.Log;
- namespace Wayne.Lib.StateEngine
- {
-
-
-
-
-
-
- public abstract class StateMachine : IEventConsumer, IDisposable
- {
- #region Fields
- private const bool TimerWrapperCaching = false;
- private InitialState initialState;
- private Dictionary<string, State> createdStates = new Dictionary<string, State>();
- private State currentState;
- private StateTransitionLookup stateLookup;
- internal StateFactories stateFactories = new StateFactories();
- private StateTypeContainer stateTypeContainer;
- internal List<TimerWrapper> timerWrapperList = new List<TimerWrapper>();
- private object timerWrapperListLock = new object();
- private EventHandler timerOnEnableEventHandler;
- private EventHandler timerOnDisableEventHandler;
- private StateMachine parentStateMachine;
- private int depth;
- private bool initialized;
- private object initalizeLock = new object();
- private string name;
- private StateNameKind logNameKind;
- private StateMachine mainStateMachine;
-
-
-
- protected internal bool disposed;
-
-
-
- internal protected IDebugLogger debugLogger;
-
-
-
- internal protected object logCategory;
- #endregion
- #region Events
-
-
-
- public event EventHandler<StateChangedEventArgs> OnStateChange;
-
-
-
- public event EventHandler<StateChangedEventArgs> OnStateChanged;
-
-
-
- public event EventHandler OnFinalStateEntered;
- #endregion
- #region Construction
-
-
-
- internal StateMachine(string name, IDebugLogger debugLogger, object logCategory)
- {
- this.name = name;
- this.debugLogger = debugLogger;
- this.logCategory = logCategory;
- this.stateTypeContainer = new StateTypeContainer();
- stateLookup = new StateTransitionLookup(stateTypeContainer);
- timerOnDisableEventHandler = new EventHandler(timer_OnDisable);
- timerOnEnableEventHandler = new EventHandler(timer_OnEnable);
- }
-
-
-
- ~StateMachine()
- {
- Dispose(false);
- }
- #endregion
- #region Public Methods
-
-
-
-
-
- public abstract void Start();
-
-
-
-
- public abstract void IncomingEvent(StateEngineEvent stateEngineEvent);
-
-
-
-
-
- public virtual IEnumerable<StateEngineEvent> GetPendingEventsOfType(object eventType)
- {
- return parentStateMachine != null ? parentStateMachine.GetPendingEventsOfType(eventType) : null;
- }
-
-
-
-
-
-
- public virtual void RemovePendingEventsOfType(object eventType)
- {
- if (parentStateMachine != null)
- parentStateMachine.RemovePendingEventsOfType(eventType);
- }
-
-
-
-
-
-
- public virtual void RemovePendingEvents<TComparisonObject>(StateEngineEventPredicate<TComparisonObject> predicate, TComparisonObject comparisonObject)
- {
- if (parentStateMachine != null)
- parentStateMachine.RemovePendingEvents(predicate, comparisonObject);
- }
-
-
-
-
-
-
-
- public virtual IEnumerable<StateEngineEvent> GetPendingEvents<TComparisonObject>(
- StateEngineEventPredicate<TComparisonObject> predicate, TComparisonObject comparisonObject)
- {
- return parentStateMachine != null ? parentStateMachine.GetPendingEvents(predicate, comparisonObject) : null;
- }
-
-
-
-
- public override string ToString()
- {
- return name;
- }
- #endregion
- #region State factory methods
-
-
-
- public void ClearStateFactories()
- {
- stateFactories.Clear();
- }
-
-
-
-
-
- public void AddStateFactory(IStateFactory factory)
- {
- stateFactories.AddFactory(factory);
- }
- #endregion
- #region Private Methods
-
-
-
-
-
-
-
-
-
-
- private void PerformStateChange(State newState, Wayne.Lib.StateEngine.StateEntry entry, ref Transition transition)
- {
-
- State oldState = currentState;
- CheckAndRemoveTimersForCurrentState();
-
- currentState = newState;
-
- if ((oldState != null) && oldState.Active)
- oldState.PerformExit();
- StateChangedEventArgs stateChangeEventArgs = new StateChangedEventArgs(oldState, newState, entry.SourceTransition);
-
- if (OnStateChange != null)
- OnStateChange.Invoke(this, stateChangeEventArgs);
-
- currentState.PerformEnter(entry, ref transition);
-
- if (OnStateChanged != null)
- OnStateChanged.Invoke(this, stateChangeEventArgs);
- }
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
- private State GetState(string stateFactoryName)
- {
- State result;
- if (!createdStates.TryGetValue(stateFactoryName, out result))
- {
- createdStates.Add(stateFactoryName, result = CreateState(stateFactoryName));
- }
- return result;
- }
-
-
-
-
-
- internal protected virtual State CreateState(string stateFactoryName)
- {
- State newState = stateFactories.CreateState(stateFactoryName, stateTypeContainer);
- if (newState == null && ParentStateMachine != null)
- newState = ParentStateMachine.stateFactories.CreateState(stateFactoryName, stateTypeContainer);
- if (newState == null && MainStateMachine != null)
- MainStateMachine.stateFactories.CreateState(stateFactoryName, stateTypeContainer);
- if (newState != null)
- {
- newState.WritableFactoryName = stateFactoryName;
- newState.SetParentStateMachine(this);
- newState.SetDebugLogger(debugLogger, logCategory);
- }
- return newState;
- }
-
-
-
-
-
- private void CreateStateObjects()
- {
- string[] stateFactoryNames = stateLookup.GetStateNameList(false);
- foreach (string stateFactoryName in stateFactoryNames)
- {
- State state = this.GetState(stateFactoryName);
- if (state != null)
- {
- InitialState tempInitState = state as InitialState;
- if (tempInitState != null)
- {
- initialState = tempInitState;
- }
- else
- {
- CompositeState compositeState = state as CompositeState;
- if (compositeState != null)
- {
- compositeState.Initialize();
- }
- }
- }
- else
- {
- throw new InvalidOperationException("Can not create state " + stateFactoryName);
- }
- }
- }
- #endregion
- #region Public Properties
-
-
-
- public StateTransitionLookup StateTransitionLookup
- {
- get { return stateLookup; }
- }
-
-
-
-
- public int Depth
- {
- get { return depth; }
- set { depth = value; }
- }
-
-
-
- public State CurrentState
- {
- get { return currentState; }
- }
-
-
-
- public State CurrentStateRecursive
- {
- get
- {
- return GetStateRecursively(currentState);
- }
- }
- private State GetStateRecursively(State state)
- {
- if (state is CompositeState)
- {
- var compositeState = ((CompositeState)state);
- if (compositeState.StateMachine.CurrentState == null)
- return compositeState;
- else
- return GetStateRecursively(compositeState.StateMachine.CurrentState);
- }
- else
- {
- return state;
- }
- }
-
-
-
- public System.Collections.ObjectModel.ReadOnlyCollection<State> CreatedStates
- {
- get { return createdStates.Values.ToList().AsReadOnly(); }
- }
-
-
-
- public string Name
- {
- get { return name; }
- }
-
-
-
- public abstract bool Started { get; }
-
-
-
- public void Initialize()
- {
- lock (initalizeLock)
- {
- if (!initialized)
- {
-
- CreateStateObjects();
- if (initialState == null)
- throw new StateEngineException("Must set an initial state for the state machine! In " + name);
- initialized = true;
- }
- }
- }
-
-
-
- public StateNameKind LogNameKind
- {
- get { return MainStateMachine.logNameKind; }
- set { MainStateMachine.logNameKind = value; }
- }
- private StateMachine MainStateMachine
- {
- get
- {
- if (mainStateMachine == null)
- {
- if (parentStateMachine != null)
- {
- mainStateMachine = parentStateMachine;
- while (mainStateMachine.parentStateMachine != null)
- mainStateMachine = mainStateMachine.parentStateMachine;
- }
- else
- mainStateMachine = this;
- }
- return mainStateMachine;
- }
- }
- #endregion
- #region Internal Properties
-
-
-
-
- internal StateMachine ParentStateMachine
- {
- get { return parentStateMachine; }
- set { parentStateMachine = value; }
- }
-
-
-
- internal bool Initialized
- {
- get
- {
- lock (initalizeLock)
- {
- return initialized;
- }
- }
- }
-
-
-
-
- public void SetAndMergeTypeContainer(StateTypeContainer newStateTypeContainer)
- {
- newStateTypeContainer.Register(stateTypeContainer);
- stateTypeContainer = newStateTypeContainer;
- }
- #endregion
- #region Internal Methods
-
-
-
-
-
- internal void EnterInitialState(Transition sourceTransition, ref Transition transition)
- {
- StateEntry stateEntry = null;
- if (sourceTransition == null)
- {
- stateEntry = new StateEntry(new Transition(null, BasicTransitionType.Init),
- this.initialState.FactoryName,
- HistoryType.None);
- }
- else
- {
- stateEntry = new StateEntry(sourceTransition, this.initialState.FactoryName, HistoryType.None);
- }
- PerformStateChange(this.initialState, stateEntry, ref transition);
- if (transition != null)
- this.HandleTransition(ref transition);
- }
-
-
-
-
-
-
- internal void EnterState(StateEntry stateEntry, ref Transition transition)
- {
- switch (stateEntry.HistoryType)
- {
-
- case HistoryType.None:
- {
- EnterInitialState(stateEntry.SourceTransition, ref transition);
- break;
- }
-
- case HistoryType.Shallow:
- {
- if (currentState != null)
- {
- var initEntry = new StateEntry(stateEntry.SourceTransition,
- initialState.FactoryName,
- HistoryType.None);
- PerformStateChange(currentState, initEntry, ref transition);
- }
- else
- EnterInitialState(stateEntry.SourceTransition, ref transition);
- break;
- }
-
- case HistoryType.Deep:
- {
- if (currentState != null)
- {
- var InitEntry = new StateEntry(stateEntry.SourceTransition,
- initialState.FactoryName,
- HistoryType.Deep);
- PerformStateChange(currentState, InitEntry, ref transition);
- }
- else
- EnterInitialState(stateEntry.SourceTransition, ref transition);
- break;
- }
-
- case HistoryType.Explicit:
- {
- var explicitTransition = stateEntry.SourceTransition;
- this.HandleTransition(ref explicitTransition);
- if (explicitTransition != null)
- transition = explicitTransition;
- break;
- }
- }
- }
-
-
-
-
- internal void ExitState()
- {
- lock (this)
- {
- if (currentState != null)
- currentState.PerformExit();
-
- lock (timerWrapperListLock)
- {
- if ((debugLogger != null) && debugLogger.IsActive(DebugLogLevel.Maximized))
- debugLogger.Add("Disabling all TimerWrappers", DebugLogLevel.Maximized);
- var tempTimerWrapperList = new List<TimerWrapper>(timerWrapperList);
- foreach (var wrapper in tempTimerWrapperList)
- wrapper.Disable();
- }
- }
- }
-
-
-
-
-
- internal virtual void HandleEvent(StateEngineEvent stateEngineEvent, ref Transition transition)
- {
- currentState.IncomingEvent(stateEngineEvent, ref transition);
- if (transition != null)
- this.HandleTransition(ref transition);
- }
-
-
-
-
-
-
-
-
-
-
- internal void HandleTransition(ref Transition transition)
- {
- var explicitTransition = transition as ExplicitTransition;
- bool transitionHandled;
- if (disposed)
- return;
- do
- {
- transitionHandled = false;
-
-
- if (explicitTransition == null)
- {
-
-
- var entry = StateTransitionLookup.GetNextState(currentState.FactoryName, transition);
- if ((entry != null) && !(currentState is FinalState))
- {
- transitionHandled = true;
- transition = null;
- var St = this.GetState(entry.TargetStateFactoryName);
- if (St != null)
- PerformStateChange(St, entry, ref transition);
- }
-
-
- }
-
-
- else
- {
- State nextState = null;
-
-
- foreach (var state in createdStates)
- {
- if (state.Value.LookupState(explicitTransition.TargetStateFactoryName))
- {
- nextState = state.Value;
- break;
- }
- }
-
- if (nextState != null)
- {
- var entry = new StateEntry(explicitTransition, nextState.FactoryName, HistoryType.Explicit);
- transition = null;
- transitionHandled = true;
- PerformStateChange(nextState, entry, ref transition);
- }
-
-
- }
- } while (transitionHandled && transition != null && !disposed);
- }
-
-
-
- internal virtual void ClearPendingEvents()
- {
- if (parentStateMachine != null)
- parentStateMachine.ClearPendingEvents();
- }
-
-
-
- internal void FireFinalStateEntered()
- {
- if (OnFinalStateEntered != null)
- OnFinalStateEntered(this, EventArgs.Empty);
- }
- internal void SetDebugLogger(IDebugLogger debugLogger, object logCategory)
- {
- this.debugLogger = debugLogger;
- this.logCategory = logCategory;
- }
- #endregion
- #region IDisposable Members
-
-
-
-
-
- protected virtual void Dispose(bool disposing)
- {
- if (!disposed)
- {
- disposed = true;
- if ((debugLogger != null) && debugLogger.IsActive(logCategory))
- debugLogger.Add("--------- Stopping State machine (Dispose)---------", logCategory);
- if ((currentState != null) && currentState.Active)
- {
- currentState.PerformExit();
- currentState = null;
- }
- if (disposing)
- {
-
- lock (timerWrapperListLock)
- {
- List<TimerWrapper> tempTimerWrapperList = new List<TimerWrapper>(timerWrapperList);
- foreach (TimerWrapper timerWrapper in tempTimerWrapperList)
- {
- timerWrapper.Dispose();
- }
- timerWrapperList.Clear();
- }
- createdStates.Values.ForEach(x=>x.Dispose());
- createdStates.Clear();
- timerOnDisableEventHandler -= timer_OnDisable;
- timerOnEnableEventHandler -= timer_OnEnable;
- }
- }
- }
-
-
-
- public void Dispose()
- {
- Dispose(true);
- GC.SuppressFinalize(this);
- }
- #endregion
- #region Timer methods
-
-
-
-
-
- public void ActivateTimer(Wayne.Lib.StateEngine.Timer timer)
- {
- timer.OnEnable += timerOnEnableEventHandler;
- timer.OnDisable += timerOnDisableEventHandler;
- timer.Enable();
- if ((debugLogger != null) && debugLogger.IsActive(DebugLogLevel.Maximized))
- debugLogger.Add("Activating timer for " + timer.ToString(), DebugLogLevel.Maximized);
- }
-
-
-
-
-
-
- void timer_OnDisable(object sender, EventArgs e)
- {
- Timer stateEngineTimer = (Timer)sender;
- if ((debugLogger != null) && debugLogger.IsActive(DebugLogLevel.Maximized))
- debugLogger.Add("Disabling StateEngineTimer " + stateEngineTimer.ToString(), DebugLogLevel.Maximized);
- lock (timerWrapperListLock)
- {
-
- TimerWrapper foundTimerWrapper = null;
- foreach (TimerWrapper wrapper in timerWrapperList)
- {
- if (wrapper.StateEngineTimer == stateEngineTimer)
- {
- foundTimerWrapper = wrapper;
- break;
- }
- }
- if (foundTimerWrapper != null)
- {
- foundTimerWrapper.StateEngineTimer = null;
- if (!TimerWrapperCaching)
- {
- foundTimerWrapper.Dispose();
- timerWrapperList.Remove(foundTimerWrapper);
- }
-
-
-
-
- }
- }
- }
-
-
-
-
-
- void timer_OnEnable(object sender, EventArgs e)
- {
- Timer stateEngineTimer = (Timer)sender;
- if ((debugLogger != null) && (debugLogger.IsActive(DebugLogLevel.Maximized)))
- debugLogger.Add("Enabling StateEngineTimer for " + stateEngineTimer.ToString(), DebugLogLevel.Maximized);
- lock (timerWrapperListLock)
- {
- TimerWrapper wrapper = GetTimerWrapper(stateEngineTimer);
- wrapper.Enable();
- }
- }
-
-
-
-
-
-
- private TimerWrapper GetTimerWrapper(Timer stateEngineTimer)
- {
- TimerWrapper wrapper = null;
- lock (timerWrapperListLock)
- {
-
- foreach (TimerWrapper tempWrapper in timerWrapperList)
- {
- if (tempWrapper.StateEngineTimer == stateEngineTimer)
- {
- wrapper = tempWrapper;
- break;
- }
- }
-
- if (wrapper == null)
- {
- foreach (TimerWrapper tempWrapper in timerWrapperList)
- {
- if (tempWrapper.StateEngineTimer == null)
- {
- wrapper = tempWrapper;
- wrapper.StateEngineTimer = stateEngineTimer;
- break;
- }
- }
- }
-
- if (wrapper == null)
- {
- wrapper = new TimerWrapper(TimerFired);
- wrapper.StateEngineTimer = stateEngineTimer;
- timerWrapperList.Add(wrapper);
- }
- }
- return wrapper;
- }
-
-
-
-
- void TimerFired(object state)
- {
- TimerWrapper timerWrapper = (TimerWrapper)state;
- StateEngineEvent stateEngineEvent = null;
- lock (timerWrapperListLock)
- {
- Timer stateEngineTimer = timerWrapper.StateEngineTimer;
- if (stateEngineTimer != null)
- {
- if ((debugLogger != null) && debugLogger.IsActive(DebugLogLevel.Maximized))
- debugLogger.Add("StateEngineTimer Fired " + stateEngineTimer.ToString(), DebugLogLevel.Maximized);
-
- if (!stateEngineTimer.IsPeriodic)
- {
- stateEngineTimer.Disable();
- }
- stateEngineEvent = new TimerEvent(stateEngineTimer.UserToken, stateEngineTimer.EventType);
- }
- }
- if (stateEngineEvent != null)
- IncomingEvent(stateEngineEvent);
- }
-
-
-
- private void CheckAndRemoveTimersForCurrentState()
- {
-
- lock (timerWrapperListLock)
- {
- List<TimerWrapper> tempTimerWrapperList = new List<TimerWrapper>(timerWrapperList);
- foreach (TimerWrapper timerWrapper in tempTimerWrapperList)
- {
- if (timerWrapper.StateEngineTimer != null)
- {
- if (timerWrapper.StateEngineTimer.OwnerState == currentState)
- {
- if (timerWrapper.StateEngineTimer.ClearAtStateExit)
- {
- if ((debugLogger != null) && debugLogger.IsActive(DebugLogLevel.Maximized))
- debugLogger.Add("!IsPeriodic: Disabling timer because of state change " + timerWrapper.StateEngineTimer.ToString(), DebugLogLevel.Maximized);
- timerWrapper.StateEngineTimer.Disable();
- }
- }
- }
- }
- }
- }
- #endregion
- #region Static Methods
-
-
-
-
-
-
-
-
- public static StateMachine Create(string name, IDebugLogger debugLogger, object logCategory)
- {
- return Create(name, StateMachineType.Threaded, debugLogger, logCategory);
- }
-
-
-
-
-
-
-
-
- public static StateMachine Create(string name, StateMachineType stateMachineType, IDebugLogger debugLogger, object logCategory)
- {
- switch (stateMachineType)
- {
- case StateMachineType.Threaded: return new ThreadedRootStateMachine(name, debugLogger, logCategory);
- case StateMachineType.Synchronous: return new SynchronousRootStateMachine(name, debugLogger, logCategory);
- default: throw new NotImplementedException("This statemachine type is not yet implemented.");
- }
- }
- #endregion
- }
- }
|