StlMap.hxx

Go to the documentation of this file.

 
#ifndef _UTILS_STLMAP_HXX_
#define _UTILS_STLMAP_HXX_
 
#include <map>
 
#include "utils/Allocator.hxx"
#include "utils/macros.h"
 
template <typename Key, typename Value> class StlMap
{
public:
    StlMap()
        : mappingAllocator(NULL),
          mapping(new Mapping())
    {
    }
 
    StlMap(size_t entries) 
        : mappingAllocator(entries > 0 ? new MappingAllocator(std::less<Key>(), Allocator<std::pair<const Key, Value>>(entries)) : NULL),
          mapping(entries > 0 ? NULL : new Mapping())
    {
    }
 
    ~StlMap()
    {
        if (mappingAllocator)
        {
            delete mappingAllocator;
        }
        if (mapping)
        {
            delete mapping;
        }
    }
 
    typedef pair<Key, Value> Pair;
 
    typedef typename std::map<Key, Value>::iterator Iterator;
 
    void clear() {
        if (mapping) {
            mapping->clear();
        } else {
            mappingAllocator->clear();
        }
    }
 
    size_t erase(Key key)
    {
        return mapping ? mapping->erase(key) : mappingAllocator->erase(key);
    }
    
    void erase(Iterator it)
    {
        mapping ? mapping->erase(it) : mappingAllocator->erase(it);
    }
    
    Value& operator[](const Key &key)
    {
        return mapping ? (*mapping)[key] : (*mappingAllocator)[key];
    }
 
    size_t size()
    {
        return mapping ? mapping->size() : mappingAllocator->size();
    }
 
    size_t max_size()
    {
        return mapping ? mapping->max_size() : mappingAllocator->max_size();
    }
 
    Iterator find( const Key &key )
    {
        return mapping ? mapping->find(key) : mappingAllocator->find(key);
    }
    
    Iterator end()
    {
        return mapping ? mapping->end() : mappingAllocator->end();
    }
    
    Iterator begin()
    {
        return mapping ? mapping->begin() : mappingAllocator->begin();
    }
 
private:
    typedef std::map<Key, Value, std::less<Key>, Allocator<std::pair<const Key, Value>>> MappingAllocator;
    
    typedef std::map<Key, Value> Mapping;
    
    MappingAllocator *mappingAllocator;
    
    Mapping *mapping;
 
    DISALLOW_COPY_AND_ASSIGN(StlMap);
};
 
#endif /* _UTILS_STLMAP_HXX_ */