Svelto.ECS/Context/ContextNotifier.cs

using System;
using System.Collections.Generic;

namespace Svelto.Context
{
    class ContextNotifier: IContextNotifer
    {
        public ContextNotifier()
        {
            _toInitialize = new List<WeakReference<IWaitForFrameworkInitialization>>();
            _toDeinitialize = new List<WeakReference<IWaitForFrameworkDestruction>> ();
        }

        /// <summary>
        /// A Context is meant to be initialized only once in its timelife
        /// </summary>
        public void NotifyFrameworkInitialized()
        {
            foreach (WeakReference<IWaitForFrameworkInitialization> obj in _toInitialize)
            {
                if (obj.IsAlive == true)
                    obj.Target.OnFrameworkInitialized();
            }

            _toInitialize = null;
        }

        /// <summary>
        /// A Context is meant to be deinitialized only once in its timelife
        /// </summary>
        public void NotifyFrameworkDeinitialized()
        {
            foreach (WeakReference<IWaitForFrameworkDestruction> obj in _toDeinitialize)
            {
                if (obj.IsAlive == true)
                    obj.Target.OnFrameworkDestroyed();
            }

            _toDeinitialize = null;
        }

        public void AddFrameworkInitializationListener(IWaitForFrameworkInitialization obj)
        {
            if (_toInitialize != null)
                _toInitialize.Add(new WeakReference<IWaitForFrameworkInitialization>(obj));
            else
                throw new Exception("An object is expected to be initialized after the framework has been initialized. Type: " + obj.GetType());
        }

        public void AddFrameworkDestructionListener(IWaitForFrameworkDestruction obj)
        {
            if (_toDeinitialize != null)
                _toDeinitialize.Add(new WeakReference<IWaitForFrameworkDestruction>(obj));
            else
                throw new Exception("An object is expected to be initialized after the framework has been deinitialized. Type: " + obj.GetType());
        }

        List<WeakReference<IWaitForFrameworkInitialization>> _toInitialize;
        List<WeakReference<IWaitForFrameworkDestruction>> _toDeinitialize;
    }
}