PendingItemsContainer.h

// Copyright eeGeo Ltd (2012-2014), All Rights Reserved

#pragma once

#include "Types.h"
#include <set>

namespace Eegeo
{
    namespace Helpers
    {
        template <typename TItemType>
        class PendingItemsContainer
        {
        private:
            bool m_cancellingAndRemovingAll;

        protected:
            typedef std::set<TItemType*> TContainer;
            typedef typename TContainer::iterator TIterator;
            typedef typename TContainer::size_type TSize;
            
            TContainer m_items;
            
        public:
            PendingItemsContainer()
            : m_cancellingAndRemovingAll(false)
            {}

            ~PendingItemsContainer()
            {
                CancelAndRemoveAllItems();
            }
            
            void InsertItem(TItemType& item)
            {
                std::pair<TIterator, bool> result = m_items.insert(&item);
                Eegeo_ASSERT(result.second, "Failed to insert item into PendingItemsContainer.\n");
            }
            
            void RemoveItem(TItemType& item)
            {
                if (!m_cancellingAndRemovingAll)
                {
                    size_t numRemoved = m_items.erase(&item);
                    Eegeo_ASSERT(numRemoved == 1, "Failed to remove item from PendingItemsContainer.\n");
                }
            }
            
            void CancelAndRemoveAllItems()
            {
                m_cancellingAndRemovingAll = true;
                for(TIterator it = m_items.begin(); it != m_items.end(); ++ it)
                {
                    TItemType& item = **it;
                    item.Cancel();
                }
                
                m_items.clear();
                m_cancellingAndRemovingAll = false;
            }

            TSize Size() const
            {
                return m_items.size();
            }
        };
    }
}