TypeLabeledContainer.h

Go to the documentation of this file.

#ifndef _utl_TypeLabeledContainer_h_
#define _utl_TypeLabeledContainer_h_

#include <map>
#include <vector>
#include <typeinfo>
#include <string>
#include <boost/iterator/transform_iterator.hpp>


namespace utl {

  class Own {
  public:
    static bool IsDestructive() { return true; }

    template<typename T>
    static void Delete(T*& p) { delete p; p = 0; }
  };


  class HoldOnly {
  public:
    static bool IsDestructive() { return false; }

    template<typename T>
    static void Delete(T*& p) { }
  };


  template<class Base, class OwnershipPolicy = Own>
  class TypeLabeledContainer {
  public:
    ~TypeLabeledContainer()
    {
      if (OwnershipPolicy::IsDestructive())
        for (const auto& sv : fContainer)
          for (const auto& p : sv.second)
            OwnershipPolicy::Delete(p);
    }

  private:
    template<class T>
    struct DynamicCast {
      T& operator()(Base* const b) const
      { return *dynamic_cast<T*>(b); }
    };

    typedef typename std::vector<Base*> InternalVector;
    typedef typename InternalVector::const_iterator InternalConstIterator;

  public:
    template<class T>
    class Iterator :
      public boost::transform_iterator<DynamicCast<T>, InternalConstIterator, T&, T> {

    public:
      Iterator(const InternalConstIterator it)
        : boost::transform_iterator<DynamicCast<T>, InternalConstIterator, T&, T>(it) { }
    };

    template<class T>
    class ConstIterator :
      public boost::transform_iterator<DynamicCast<const T>, InternalConstIterator, const T&, const T> {

    public:
      ConstIterator(const InternalConstIterator it)
        : boost::transform_iterator<DynamicCast<const T>, InternalConstIterator, const T&, const T>(it) { }
    };

    template<class T>
    Iterator<T>
    Begin()
    {
      const typename InternalMap::const_iterator it = fContainer.find(typeid(T).name());
      return (it != fContainer.end()) ? it->second.begin() : InternalConstIterator(0);
    }

    template<class T>
    ConstIterator<T>
    Begin()
      const
    {
      const typename InternalMap::const_iterator it = fContainer.find(typeid(T).name());
      return (it != fContainer.end()) ? it->second.begin() : InternalConstIterator(0);
    }

    template<class T>
    Iterator<T>
    End()
    {
      const typename InternalMap::const_iterator it = fContainer.find(typeid(T).name());
      return (it != fContainer.end()) ? it->second.end() : InternalConstIterator(0);
    }

    template<class T>
    ConstIterator<T>
    End()
      const
    {
      const typename InternalMap::const_iterator it = fContainer.find(typeid(T).name());
      return (it != fContainer.end()) ? it->second.end() : InternalConstIterator(0);
    }

    template<class T>
    void Insert(T* const t)
    { fContainer[std::string(typeid(T).name())].push_back(t); }

  private:
    typedef std::map<std::string, InternalVector> InternalMap;

    InternalMap fContainer;
  };

}


#endif